From: Vsevolod Stakhov <vsevolod@rspamd.com>
Date: Wed, 14 May 2025 11:00:22 +0000 (+0100)
Subject: [Feature] Add some convenience methods
X-Git-Tag: 3.12.0~25
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=5cf2bdb83c8bec117e787cea30e3b1f79d8dedee;p=thirdparty%2Frspamd.git

[Feature] Add some convenience methods
---

diff --git a/src/libutil/radix.c b/src/libutil/radix.c
index 2cae8e34af..bdd722b499 100644
--- a/src/libutil/radix.c
+++ b/src/libutil/radix.c
@@ -1,5 +1,5 @@
 /*
- * Copyright 2024 Vsevolod Stakhov
+ * Copyright 2025 Vsevolod Stakhov
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -66,7 +66,7 @@ radix_find_compressed(radix_compressed_t *tree, const uint8_t *key, gsize keylen
 
 uintptr_t
 radix_insert_compressed(radix_compressed_t *tree,
-						uint8_t *key, gsize keylen,
+						const uint8_t *key, gsize keylen,
 						gsize masklen,
 						uintptr_t value)
 {
@@ -128,6 +128,39 @@ radix_insert_compressed(radix_compressed_t *tree,
 	return old;
 }
 
+uintptr_t
+radix_insert_compressed_addr(radix_compressed_t *tree,
+							 const rspamd_inet_addr_t *addr,
+							 uintptr_t value)
+{
+	const unsigned char *key;
+	unsigned int klen = 0;
+	unsigned char buf[16];
+
+	if (addr == NULL) {
+		return RADIX_NO_VALUE;
+	}
+
+	key = rspamd_inet_address_get_hash_key(addr, &klen);
+
+	if (key && klen) {
+		if (klen == 4) {
+			/* Map to ipv6 */
+			memset(buf, 0, 10);
+			buf[10] = 0xffu;
+			buf[11] = 0xffu;
+			memcpy(buf + 12, key, klen);
+
+			key = buf;
+			klen = sizeof(buf);
+		}
+
+		return radix_insert_compressed(tree, key, klen, 0, value);
+	}
+
+	return RADIX_NO_VALUE;
+}
+
 
 radix_compressed_t *
 radix_create_compressed(const char *tree_name)
diff --git a/src/libutil/radix.h b/src/libutil/radix.h
index c4fe964415..8c12247070 100644
--- a/src/libutil/radix.h
+++ b/src/libutil/radix.h
@@ -1,5 +1,5 @@
 /*
- * Copyright 2024 Vsevolod Stakhov
+ * Copyright 2025 Vsevolod Stakhov
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -20,7 +20,7 @@
 #include "mem_pool.h"
 #include "util.h"
 
-#define RADIX_NO_VALUE (uintptr_t) - 1
+#define RADIX_NO_VALUE (uintptr_t) -1
 
 #ifdef __cplusplus
 extern "C" {
@@ -39,10 +39,22 @@ typedef struct radix_tree_compressed radix_compressed_t;
  */
 uintptr_t
 radix_insert_compressed(radix_compressed_t *tree,
-						uint8_t *key, gsize keylen,
+						const uint8_t *key, gsize keylen,
 						gsize masklen,
 						uintptr_t value);
 
+/**
+ * Insert new address to the radix trie (works for IPv4 or IPv6 addresses)
+ * @param tree radix trie
+ * @param addr address to insert
+ * @param value opaque value pointer
+ * @return previous value of the key or `RADIX_NO_VALUE`
+ */
+uintptr_t
+radix_insert_compressed_addr(radix_compressed_t *tree,
+							 const rspamd_inet_addr_t *addr,
+							 uintptr_t value);
+
 /**
  * Find a key in a radix trie
  * @param tree radix trie