/**
* Create a BGP domain (an Autonomous System).
*/
bgp_domain_t * bgp_domain_create(asn_t asn)
{
bgp_domain_t * domain= (bgp_domain_t *) MALLOC(sizeof(bgp_domain_t));
domain->asn= asn;
domain->name= NULL;
/* Radix-tree with all routers. Destroy function is NULL. */
domain->routers= radix_tree_create(32, NULL);
return domain;
}
void
rspamd_radix_test_func (void)
{
#if 0
radix_tree_t *tree = radix_tree_create ();
#endif
radix_compressed_t *comp_tree = radix_create_compressed ();
struct {
guint32 addr;
guint32 mask;
guint8 addr6[16];
guint32 mask6;
} *addrs;
gsize nelts, i;
gint lc;
gboolean all_good = TRUE;
gdouble ts1, ts2;
double diff;
/* Test suite for the compressed trie */
rspamd_radix_text_vec ();
nelts = max_elts;
/* First of all we generate many elements and push them to the array */
addrs = g_malloc (nelts * sizeof (addrs[0]));
for (i = 0; i < nelts; i ++) {
addrs[i].addr = ottery_rand_uint32 ();
addrs[i].mask = masks[ottery_rand_range(G_N_ELEMENTS (masks) - 1)];
ottery_rand_bytes (addrs[i].addr6, sizeof(addrs[i].addr6));
addrs[i].mask6 = ottery_rand_range(128);
}
#if 0
msg_info ("old radix performance (%z elts)", nelts);
ts1 = rspamd_get_ticks ();
for (i = 0; i < nelts; i ++) {
guint32 mask = G_MAXUINT32 << (32 - addrs[i].mask);
radix32tree_insert (tree, addrs[i].addr, mask, 1);
}
ts2 = rspamd_get_ticks ();
diff = (ts2 - ts1) * 1000.0;
msg_info ("Added %z elements in %.6f ms", nelts, diff);
ts1 = rspamd_get_ticks ();
for (lc = 0; lc < lookup_cycles; lc ++) {
for (i = 0; i < nelts; i ++) {
g_assert (radix32tree_find (tree, addrs[i].addr) != RADIX_NO_VALUE);
}
}
ts2 = rspamd_get_ticks ();
diff = (ts2 - ts1) * 1000.0;
msg_info ("Checked %z elements in %.6f ms", nelts, diff);
ts1 = rspamd_get_ticks ();
for (i = 0; i < nelts; i ++) {
radix32tree_delete (tree, addrs[i].addr, addrs[i].mask);
}
ts2 = rspamd_get_ticks ();
diff = (ts2 - ts1) * 1000.;
msg_info ("Deleted %z elements in %.6f ms", nelts, diff);
radix_tree_free (tree);
#endif
msg_info ("new radix performance (%z elts)", nelts);
ts1 = rspamd_get_ticks ();
for (i = 0; i < nelts; i ++) {
radix_insert_compressed (comp_tree, addrs[i].addr6, sizeof (addrs[i].addr6),
128 - addrs[i].mask6, i);
}
ts2 = rspamd_get_ticks ();
diff = (ts2 - ts1) * 1000.0;
msg_info ("Added %z elements in %.6f ms", nelts, diff);
ts1 = rspamd_get_ticks ();
for (lc = 0; lc < lookup_cycles; lc ++) {
for (i = 0; i < nelts; i ++) {
if (radix_find_compressed (comp_tree, addrs[i].addr6, sizeof (addrs[i].addr6))
== RADIX_NO_VALUE) {
all_good = FALSE;
}
}
}
#if 1
if (!all_good) {
for (i = 0; i < nelts; i ++) {
/* Used to write bad random vector */
char ipbuf[INET6_ADDRSTRLEN + 1];
inet_ntop(AF_INET6, addrs[i].addr6, ipbuf, sizeof(ipbuf));
msg_info("{\"%s\", NULL, \"%ud\", 0, 0, 0, 0},",
ipbuf,
addrs[i].mask6);
}
}
#endif
g_assert (all_good);
ts2 = rspamd_get_ticks ();
diff = (ts2 - ts1) * 1000.0;
msg_info ("Checked %z elements in %.6f ms", nelts, diff);
radix_destroy_compressed (comp_tree);
g_free (addrs);
}
geo_t *geo_load(geo_t *g, char *filename, int geo_mode) {
int n;
FILE *fp;
char buf[MAX_LINE];
char *p;
unsigned char *field[2];
if (!filename) {
return NULL;
}
if (geo_mode != GEO_RANGE && geo_mode != GEO_CIDR) {
return NULL;
}
if (!g) {
g = (geo_t *)malloc(sizeof(geo_t));
if (!g) {
return NULL;
}
g->geo_mode = geo_mode;
if (geo_mode == GEO_RANGE) {
g->u.range = range_create();
if (!g->u.range) {
free(g);
return NULL;
}
} else if (geo_mode == GEO_CIDR) {
g->u.tree = radix_tree_create();
if (!g->u.tree) {
free(g);
return NULL;
}
} else {
return NULL;
}
}
fp = fopen(filename, "r");
if (!fp) {
return NULL;
}
while (fgets(buf, MAX_LINE, fp)) {
n = strlen(buf);
if (buf[n - 1] == '\n') {
buf[n - 1] = '\0';
}
p = buf;
while (*p == ' ' || *p == '\t') {
++p;
}
if (*p == '\0' || *p == '#' || (*p == '/' && *(p + 1) == '/')) {
continue;
}
if (str_explode(NULL, (unsigned char *)p, field, 2) != 2) {
goto error;
}
if (strcmp((char *)field[0], "include") == 0) {
if (geo_load(g, (char *)field[1], geo_mode) == NULL) {
goto error;
}
continue;
}
if (geo_mode == GEO_RANGE) {
if (geo_parse_range(g, (char *)field[0], (char *)field[1]) != 0) {
goto error;
}
} else if (geo_mode == GEO_CIDR) {
if (geo_parse_cidr(g, (char *)field[0], (char *)field[1]) != 0) {
goto error;
}
} else {
/* should NOT get here */
assert(0);
}
}
fclose(fp);
return g;
error:
fclose(fp);
geo_unload(g);
return NULL;
}
