aboutsummaryrefslogtreecommitdiff
path: root/src/lib/ndpi_bitmap.c
diff options
context:
space:
mode:
authorLuca Deri <lucaderi@users.noreply.github.com>2024-03-25 08:15:19 +0100
committerGitHub <noreply@github.com>2024-03-25 08:15:19 +0100
commit51f5fc714015c312622ed4ee1820518d26edee65 (patch)
tree9cd1cee32e3fadada62414f1821210fb4bb83b95 /src/lib/ndpi_bitmap.c
parentbe0fd7444ff714f3c654aca08b3bf151ac801bbf (diff)
Added support for roaring bitmap v3 (#2355)
* Integrated RoaringBitmap v3 * Renamed ndpi_bitmap64 ro ndpi_bitmap64_fuse * Fixes to ndpi_bitmap for new roaring library * Fixes for bitmap serialization * Fixed format * Warning fix * Conversion fix * Warning fix * Added check for roaring v3 support * Updated file name * Updated path * Uses clang-9 (instead of clang-7) for builds * Fixed fuzz_ds_bitmap64_fuse * Fixes nDPI printf handling * Disabled printf * Yet another printf fix * Cleaup * Fx for compiling on older platforms * Fixes for old compilers * Initialization changes * Added compiler check * Fixes for old compilers * Inline function is not static inline * Added missing include
Diffstat (limited to 'src/lib/ndpi_bitmap.c')
-rw-r--r--src/lib/ndpi_bitmap.c138
1 files changed, 113 insertions, 25 deletions
diff --git a/src/lib/ndpi_bitmap.c b/src/lib/ndpi_bitmap.c
index 216251b21..6caf174bf 100644
--- a/src/lib/ndpi_bitmap.c
+++ b/src/lib/ndpi_bitmap.c
@@ -1,7 +1,7 @@
/*
* ndpi_bitmap.c
*
- * Copyright (C) 2011-23 - ntop.org and contributors
+ * Copyright (C) 2011-24 - ntop.org and contributors
*
* This file is part of nDPI, an open source deep packet inspection
* library based on the OpenDPI and PACE technology by ipoque GmbH
@@ -35,145 +35,226 @@
#include "ndpi_includes.h"
#include "ndpi_encryption.h"
+#ifdef USE_ROARING_V2
+#include "third_party/include/roaring_v2.h"
+#else
#include "third_party/include/roaring.h"
+#endif
/* ******************************************* */
ndpi_bitmap* ndpi_bitmap_alloc() {
+#ifdef USE_ROARING_V2
return((ndpi_bitmap*)roaring_bitmap_create());
-}
-
-/* ******************************************* */
-
-ndpi_bitmap* ndpi_bitmap_alloc_size(u_int32_t size) {
- return((ndpi_bitmap*)roaring_bitmap_create_with_capacity(size));
+#else
+ return((ndpi_bitmap*)roaring64_bitmap_create());
+#endif
}
/* ******************************************* */
void ndpi_bitmap_free(ndpi_bitmap* b) {
+#ifdef USE_ROARING_V2
roaring_bitmap_free((const roaring_bitmap_t *)b);
+#else
+ roaring64_bitmap_free((roaring64_bitmap_t *)b);
+#endif
}
/* ******************************************* */
ndpi_bitmap* ndpi_bitmap_copy(ndpi_bitmap* b) {
+#ifdef USE_ROARING_V2
return(roaring_bitmap_copy(b));
+#else
+ return(roaring64_bitmap_copy(b));
+#endif
}
/* ******************************************* */
u_int64_t ndpi_bitmap_cardinality(ndpi_bitmap* b) {
+#ifdef USE_ROARING_V2
return(roaring_bitmap_get_cardinality((const roaring_bitmap_t *)b));
+#else
+ return(roaring64_bitmap_get_cardinality((roaring64_bitmap_t *)b));
+#endif
}
/* ******************************************* */
-void ndpi_bitmap_set(ndpi_bitmap* b, u_int32_t value) {
+void ndpi_bitmap_set(ndpi_bitmap* b, u_int64_t value) {
+#ifdef USE_ROARING_V2
roaring_bitmap_add((roaring_bitmap_t *)b, value);
+#else
+ roaring64_bitmap_add((roaring64_bitmap_t *)b, value);
+#endif
}
/* ******************************************* */
-void ndpi_bitmap_unset(ndpi_bitmap* b, u_int32_t value) {
+void ndpi_bitmap_unset(ndpi_bitmap* b, u_int64_t value) {
+#ifdef USE_ROARING_V2
roaring_bitmap_remove((roaring_bitmap_t *)b, value);
+#else
+ roaring64_bitmap_remove((roaring64_bitmap_t *)b, value);
+#endif
}
/* ******************************************* */
-bool ndpi_bitmap_isset(ndpi_bitmap* b, u_int32_t value) {
- return(roaring_bitmap_contains((const roaring_bitmap_t *)b, value));
-}
-
-/* ******************************************* */
+bool ndpi_bitmap_isset(ndpi_bitmap* b, u_int64_t value) {
+ bool ret;
+
+#ifdef USE_ROARING_V2
+ ret = roaring_bitmap_contains((const roaring_bitmap_t *)b, value);
+#else
+ ret = roaring64_bitmap_contains((const roaring64_bitmap_t *)b, value);
+#endif
-void ndpi_bitmap_clear(ndpi_bitmap* b) {
- roaring_bitmap_clear((roaring_bitmap_t *)b);
+ return(ret);
}
/* ******************************************* */
size_t ndpi_bitmap_serialize(ndpi_bitmap* b, char **buf) {
- const roaring_bitmap_t *r = (const roaring_bitmap_t *)b;
- size_t s = roaring_bitmap_size_in_bytes(r);
+ size_t s;
+#ifdef USE_ROARING_V2
+ const roaring_bitmap_t *r = (const roaring_bitmap_t *)b;
+
+ s = roaring_bitmap_portable_size_in_bytes(r);
+#else
+ const roaring64_bitmap_t *r = (const roaring64_bitmap_t *)b;
+
+ s = roaring64_bitmap_portable_size_in_bytes(r);
+#endif
+
*buf = (char*)ndpi_malloc(s);
if((*buf) == NULL) return(0);
- return(roaring_bitmap_serialize(r, *buf));
+#ifdef USE_ROARING_V2
+ return(roaring_bitmap_portable_serialize(r, *buf));
+#else
+ return(roaring64_bitmap_portable_serialize(r, *buf));
+#endif
}
/* ******************************************* */
-ndpi_bitmap* ndpi_bitmap_deserialize(char *buf) {
- return((ndpi_bitmap*)roaring_bitmap_deserialize(buf));
+ndpi_bitmap* ndpi_bitmap_deserialize(char *buf, size_t buf_len) {
+#ifdef USE_ROARING_V2
+ return((ndpi_bitmap*)roaring_bitmap_portable_deserialize_safe(buf, buf_len));
+#else
+ return((ndpi_bitmap*)roaring64_bitmap_portable_deserialize_safe(buf, buf_len));
+#endif
}
/* ******************************************* */
/* b = b & b_and */
void ndpi_bitmap_and(ndpi_bitmap* a, ndpi_bitmap* b_and) {
+#ifdef USE_ROARING_V2
roaring_bitmap_and_inplace((roaring_bitmap_t*)a, (roaring_bitmap_t*)b_and);
+#else
+ roaring64_bitmap_and_inplace((roaring64_bitmap_t*)a, (roaring64_bitmap_t*)b_and);
+#endif
}
/* ******************************************* */
/* b = b & b_and */
ndpi_bitmap* ndpi_bitmap_and_alloc(ndpi_bitmap* a, ndpi_bitmap* b_and) {
+#ifdef USE_ROARING_V2
return((ndpi_bitmap*)roaring_bitmap_and((roaring_bitmap_t*)a, (roaring_bitmap_t*)b_and));
+#else
+ return((ndpi_bitmap*)roaring64_bitmap_and((roaring64_bitmap_t*)a, (roaring64_bitmap_t*)b_and));
+#endif
}
/* ******************************************* */
/* b = b & !b_and */
void ndpi_bitmap_andnot(ndpi_bitmap* a, ndpi_bitmap* b_and) {
+#ifdef USE_ROARING_V2
roaring_bitmap_andnot_inplace((roaring_bitmap_t*)a, (roaring_bitmap_t*)b_and);
+#else
+ roaring64_bitmap_andnot_inplace((roaring64_bitmap_t*)a, (roaring64_bitmap_t*)b_and);
+#endif
}
/* ******************************************* */
/* b = b | b_or */
void ndpi_bitmap_or(ndpi_bitmap* a, ndpi_bitmap* b_or) {
+#ifdef USE_ROARING_V2
roaring_bitmap_or_inplace((roaring_bitmap_t*)a, (roaring_bitmap_t*)b_or);
+#else
+ roaring64_bitmap_or_inplace((roaring64_bitmap_t*)a, (roaring64_bitmap_t*)b_or);
+#endif
}
/* ******************************************* */
/* b = b | b_or */
ndpi_bitmap* ndpi_bitmap_or_alloc(ndpi_bitmap* a, ndpi_bitmap* b_or) {
+#ifdef USE_ROARING_V2
return((ndpi_bitmap*)roaring_bitmap_or((roaring_bitmap_t*)a, (roaring_bitmap_t*)b_or));
+#else
+ return((ndpi_bitmap*)roaring64_bitmap_or((roaring64_bitmap_t*)a, (roaring64_bitmap_t*)b_or));
+#endif
}
/* ******************************************* */
/* b = b ^ b_xor */
void ndpi_bitmap_xor(ndpi_bitmap* a, ndpi_bitmap* b_xor) {
+#ifdef USE_ROARING_V2
roaring_bitmap_xor_inplace((roaring_bitmap_t*)a, (roaring_bitmap_t*)b_xor);
+#else
+ roaring64_bitmap_xor_inplace((roaring64_bitmap_t*)a, (roaring64_bitmap_t*)b_xor);
+#endif
}
/* ******************************************* */
void ndpi_bitmap_optimize(ndpi_bitmap* a) {
+#ifdef USE_ROARING_V2
roaring_bitmap_run_optimize(a);
+#else
+ roaring64_bitmap_run_optimize(a);
+#endif
}
/* ******************************************* */
ndpi_bitmap_iterator* ndpi_bitmap_iterator_alloc(ndpi_bitmap* b) {
- return(roaring_create_iterator((roaring_bitmap_t*)b));
+#ifdef USE_ROARING_V2
+ return((ndpi_bitmap_iterator*)roaring_create_iterator((roaring_bitmap_t*)b));
+#else
+ return((ndpi_bitmap_iterator*)roaring64_iterator_create((const roaring64_bitmap_t*)b));
+#endif
}
/* ******************************************* */
void ndpi_bitmap_iterator_free(ndpi_bitmap* b) {
+#ifdef USE_ROARING_V2
roaring_free_uint32_iterator((roaring_uint32_iterator_t*)b);
+#else
+ roaring64_iterator_free((roaring64_iterator_t*)b);
+#endif
}
/* ******************************************* */
bool ndpi_bitmap_is_empty(ndpi_bitmap* b) {
+#ifdef USE_ROARING_V2
return(roaring_bitmap_is_empty((roaring_bitmap_t*)b));
+#else
+ return(roaring64_bitmap_is_empty((roaring64_bitmap_t*)b));
+#endif
}
/* ******************************************* */
@@ -182,8 +263,15 @@ bool ndpi_bitmap_is_empty(ndpi_bitmap* b) {
true is returned when a value is present, false when we reached the end
*/
-bool ndpi_bitmap_iterator_next(ndpi_bitmap_iterator* i, uint32_t *value) {
- uint32_t num = roaring_read_uint32_iterator((roaring_uint32_iterator_t*)i, value, 1);
-
+bool ndpi_bitmap_iterator_next(ndpi_bitmap_iterator* i, u_int64_t *value) {
+#ifdef USE_ROARING_V2
+ uint32_t ret;
+ uint32_t num = roaring_read_uint32_iterator((roaring_uint32_iterator_t*)i, &ret, 1);
+
+ *value = (uint32_t)ret;
+#else
+ uint64_t num = roaring64_iterator_read((roaring64_iterator_t*)i, value, 1);
+#endif
+
return((num == 1) ? true /* found */ : false /* not found */);
}