mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-15 02:41:59 +00:00
9520234365
This commit implements a basic IP prefix dictionary that allows storing IPv4/IPv6 prefixes and matching them against a single IP address on query time. This allows for doing IP -> ASN matching and other similar things on query time. The implementation is basic for start, using a simple bitwise trie and reusing interface for complex key dictionaries (so using tuple instead of UInt32/FixedString(16) as the key). A faster bitwise trie implementation (like poptrie) is desired to improve lookup performance and memory consumption with large prefix tables.
95 lines
2.0 KiB
C
95 lines
2.0 KiB
C
#include <stdio.h>
|
|
#include <btrie.h>
|
|
|
|
int main()
|
|
{
|
|
btrie_t *it;
|
|
int ret;
|
|
|
|
uint8_t prefix_v6[16] = {0xde, 0xad, 0xbe, 0xef};
|
|
uint8_t mask_v6[16] = {0xff, 0xff, 0xff};
|
|
uint8_t ip_v6[16] = {0xde, 0xad, 0xbe, 0xef, 0xde};
|
|
|
|
it = btrie_create();
|
|
if (it == NULL) {
|
|
printf("create error!\n");
|
|
return 0;
|
|
}
|
|
|
|
//add 101.45.69.50/16
|
|
ret = btrie_insert(it, 1697465650, 0xffff0000, 1);
|
|
if (ret != 0) {
|
|
printf("insert 1 error.\n");
|
|
goto error;
|
|
}
|
|
|
|
//add 10.45.69.50/16
|
|
ret = btrie_insert(it, 170738994, 0xffff0000, 1);
|
|
if (ret != 0) {
|
|
printf("insert 2 error.\n");
|
|
goto error;
|
|
}
|
|
|
|
//add 10.45.79.50/16
|
|
ret = btrie_insert(it, 170741554, 0xffff0000, 1);
|
|
if (ret == 0) {
|
|
printf("insert 3 error.\n");
|
|
goto error;
|
|
}
|
|
|
|
//add 102.45.79.50/24
|
|
ret = btrie_insert(it, 1714245426, 0xffffff00, 1);
|
|
if (ret != 0) {
|
|
printf("insert 4 error.\n");
|
|
goto error;
|
|
}
|
|
|
|
ret = btrie_find(it, 170741554);
|
|
if (ret == 1) {
|
|
printf("test case 1 passed\n");
|
|
} else {
|
|
printf("test case 1 error\n");
|
|
}
|
|
|
|
ret = btrie_find(it, 170786817);
|
|
if (ret != 1) {
|
|
printf("test case 2 passed\n");
|
|
} else {
|
|
printf("test case 2 error\n");
|
|
}
|
|
|
|
ret = btrie_delete(it, 1714245426, 0xffffff00);
|
|
if (ret != 0) {
|
|
printf("delete 1 error\n");
|
|
goto error;
|
|
}
|
|
|
|
ret = btrie_find(it, 1714245426);
|
|
if (ret != 1) {
|
|
printf("test case 3 passed\n");
|
|
} else {
|
|
printf("test case 3 error\n");
|
|
}
|
|
|
|
//add dead:beef::/32
|
|
ret = btrie_insert_a6(it, prefix_v6, mask_v6, 1);
|
|
if (ret != 0) {
|
|
printf("insert 5 error\n");
|
|
goto error;
|
|
}
|
|
|
|
ret = btrie_find_a6(it, ip_v6);
|
|
if (ret == 1) {
|
|
printf("test case 4 passed\n");
|
|
} else {
|
|
printf("test case 4 error\n");
|
|
}
|
|
|
|
return 0;
|
|
|
|
error:
|
|
btrie_destroy(it);
|
|
printf("test failed\n");
|
|
return 1;
|
|
}
|