mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-10 08:22:00 +00:00
ddb0dcd2fd
* Increase default maximum size of brtie dictionary to 128m ips pbuilder: use newer cmake on ubuntu trusty * no cmake
104 lines
2.2 KiB
C
104 lines
2.2 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");
|
|
}
|
|
|
|
// insert 4m ips
|
|
for (size_t ip = 1; ip < 1024 * 1024 * 4; ++ip) {
|
|
ret = btrie_insert(it, ip, 0xffffffff, 1);
|
|
if (ret != 0) {
|
|
printf("insert 5 error (%d) (%zu) .\n", ret, ip);
|
|
goto error;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
|
|
error:
|
|
btrie_destroy(it);
|
|
printf("test failed\n");
|
|
return 1;
|
|
}
|