Add routine to find rspamd_inet_addr in radix trie.

2 files changed, 28 insertions, 0 deletions

diff --git a/src/libutil/radix.c b/src/libutil/radix.c
index 99e25d0ba..40cccb46b 100644
--- a/src/libutil/radix.c
+++ b/src/libutil/radix.c
@@ -868,6 +868,25 @@ radix_tree_destroy_compressed (radix_compressed_t *tree)
 	g_slice_free1 (sizeof (*tree), tree);
 }
 
+uintptr_t
+radix_find_compressed_addr (radix_compressed_t *tree, rspamd_inet_addr_t *addr)
+{
+	if (addr == NULL) {
+		return RADIX_NO_VALUE;
+	}
+
+	if (addr->af == AF_INET) {
+		return radix_find_compressed (tree, (guint8 *)&addr->addr.s4.sin_addr,
+				sizeof (addr->addr.s4.sin_addr));
+	}
+	else if (addr->af == AF_INET6) {
+		return radix_find_compressed (tree, (guint8 *)&addr->addr.s6.sin6_addr,
+				sizeof (addr->addr.s6.sin6_addr));
+	}
+
+	return RADIX_NO_VALUE;
+}
+
 /*
  * vi:ts=4
  */
diff --git a/src/libutil/radix.h b/src/libutil/radix.h
index 52dc96952..f89d30103 100644
--- a/src/libutil/radix.h
+++ b/src/libutil/radix.h
@@ -82,6 +82,15 @@ uintptr_t radix32tree_find (radix_tree_t *tree, guint32 key);
 uintptr_t radix32_tree_find_addr (radix_tree_t *tree, rspamd_inet_addr_t *addr);
 
 /**
+ * Find specified address in tree (works for any address)
+ * @param tree
+ * @param addr
+ * @return
+ */
+uintptr_t radix_find_compressed_addr (radix_compressed_t *tree,
+		rspamd_inet_addr_t *addr);
+
+/**
  * Traverse via the whole tree calling specified callback
  */
 void radix32tree_traverse (radix_tree_t *tree,