Compare commits
No commits in common. "6474eb8a5d9593ed804fdc2b663a414a63018f42" and "3ca4216b581be6444c7366d38f6a83f2a6fad612" have entirely different histories.
6474eb8a5d
...
3ca4216b58
210
radix/radix.c
210
radix/radix.c
|
@ -1,210 +0,0 @@
|
|||
#include "radix.h"
|
||||
|
||||
#define GET_BIT(key, bit) ( ( key.s[ bit / 16 ] & (1u << (15 - bit % 16)) ) != 0 )
|
||||
|
||||
radix_node_t create_node(ipv6_t key, unsigned end_bit, unsigned rule) {
|
||||
radix_node_t ret = calloc(1, sizeof(struct radix_node));
|
||||
ret->_ip = key;
|
||||
ret->_ip_end_bit = end_bit;
|
||||
ret->_rule_valid = VALID_RULE;
|
||||
ret->_rule = rule;
|
||||
return ret;
|
||||
}
|
||||
|
||||
static inline radix_node_t clone_node(radix_node_t n) {
|
||||
radix_node_t ret = calloc(1, sizeof(struct radix_node));
|
||||
memcpy(ret, n, sizeof(struct radix_node));
|
||||
return ret;
|
||||
}
|
||||
|
||||
static inline void split_node(radix_node_t n, unsigned bit) {
|
||||
radix_node_t tmp = clone_node(n);
|
||||
n->_ip_end_bit = bit;
|
||||
n->_l = n->_r = tmp;
|
||||
n->_rule_valid = INVALID_RULE;
|
||||
}
|
||||
|
||||
radix_holder_t create_holder() {
|
||||
radix_holder_t ret = calloc(1, sizeof(struct radix_holder));
|
||||
return ret;
|
||||
}
|
||||
|
||||
int destory_holder(struct radix_holder* t) {
|
||||
destory_node(t->_root);
|
||||
free(t);
|
||||
return 1;
|
||||
}
|
||||
|
||||
int destory_node(struct radix_node* n) {
|
||||
int ret = 0;
|
||||
if (!n) return ret;
|
||||
ret = destory_node(n->_l) + destory_node(n->_r);
|
||||
free(n);
|
||||
return ret;
|
||||
}
|
||||
|
||||
int radix_insert(radix_holder_t t, ipv6_t key, int val, int mask) {
|
||||
unsigned char b;
|
||||
int ibit;
|
||||
radix_node_t node = t->_root;
|
||||
radix_node_t* next = &t->_root;
|
||||
|
||||
if (!t->_root) {
|
||||
t->_root = create_node(key, mask, val);
|
||||
return t->_size = 1;
|
||||
}
|
||||
|
||||
for (ibit = 0; ibit < mask; ++ibit) {
|
||||
b = GET_BIT(key, ibit);
|
||||
if (ibit >= node->_ip_end_bit) { /* node key ends */
|
||||
next = b ? &node->_l : &node->_r;
|
||||
|
||||
if (!*next) {
|
||||
*next = create_node(key, mask, val);
|
||||
++t->_size;
|
||||
return 1;
|
||||
}
|
||||
|
||||
node = *next;
|
||||
continue;
|
||||
}
|
||||
|
||||
if (b == GET_BIT(node->_ip, ibit)) /* bit match */
|
||||
continue;
|
||||
|
||||
next = b ? &node->_l : &node->_r;
|
||||
if (*next && ibit == mask - 1) { /* last bit mismatch */
|
||||
node = *next;
|
||||
continue;
|
||||
}
|
||||
|
||||
split_node(node, ibit); /* split node */
|
||||
*next = create_node(key, mask, val);
|
||||
++t->_size;
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* exact match */
|
||||
if (node->_rule_valid) {
|
||||
printf("WARNING: conflicting rule %x:%x:%x:%x:%x:%x:%x:%x",
|
||||
key.s[0], key.s[1], key.s[2], key.s[3],
|
||||
key.s[4], key.s[5], key.s[6], key.s[7]);
|
||||
|
||||
printf("/%d %d\n", mask, val);
|
||||
return 0;
|
||||
}
|
||||
|
||||
node->_rule_valid = VALID_RULE;
|
||||
node->_rule = val;
|
||||
return 1;
|
||||
}
|
||||
|
||||
int radix_search(radix_holder_t t, ipv6_t key) {
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
/*--------------------------------------------------------------------------------*/
|
||||
/* TESTING */
|
||||
/*--------------------------------------------------------------------------------*/
|
||||
#include <assert.h>
|
||||
|
||||
#define RULES_FILE "../data/routing-data"
|
||||
#define TEST_FILE "../data/test-data"
|
||||
#define TEST_ROWS 5721
|
||||
#define READ_MODE "r"
|
||||
|
||||
int parse_line(ipv6_t* key, int* val, int* mask, FILE* f) { /*! assumes only good input */
|
||||
memset(key, 0, sizeof(ipv6_t));
|
||||
|
||||
fscanf(f, "%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx",
|
||||
&key->s[0], &key->s[1], &key->s[2], &key->s[3],
|
||||
&key->s[4], &key->s[5], &key->s[6], &key->s[7]);
|
||||
|
||||
if (feof(f)) return 0; /* ends with newline */
|
||||
while (fgetc(f) != '/');
|
||||
|
||||
fscanf(f, "%d %d", mask, val);
|
||||
return !feof(f);
|
||||
}
|
||||
|
||||
int load_input(radix_holder_t t) {
|
||||
int val;
|
||||
int mask;
|
||||
ipv6_t key;
|
||||
FILE* f;
|
||||
int cnt = 0;
|
||||
|
||||
if (!(f = fopen(RULES_FILE, READ_MODE))) {
|
||||
puts("cant open input file");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
while (parse_line(&key, &val, &mask, f)) {
|
||||
cnt += radix_insert(t, key, val, mask);
|
||||
}
|
||||
fclose(f);
|
||||
return cnt;
|
||||
}
|
||||
|
||||
int load_tests(ipv6_t** rips, int** rref) {
|
||||
ipv6_t* ips;
|
||||
int* ref;
|
||||
int row;
|
||||
FILE* f;
|
||||
|
||||
if (!(f = fopen( TEST_FILE, READ_MODE ))) {
|
||||
puts("cant open test file");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
ips = calloc( TEST_ROWS, sizeof(ipv6_t) );
|
||||
ref = calloc( TEST_ROWS, sizeof(int) );
|
||||
|
||||
for (row = 0; row < TEST_ROWS; ++ row) {
|
||||
fscanf(f, "%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx",
|
||||
&ips[row].s[0], &ips[row].s[1], &ips[row].s[2], &ips[row].s[3],
|
||||
&ips[row].s[4], &ips[row].s[5], &ips[row].s[6], &ips[row].s[7]);
|
||||
|
||||
|
||||
while (fgetc(f) != ' ');
|
||||
fscanf(f, "%d", &(ref[row]));
|
||||
}
|
||||
fclose(f);
|
||||
|
||||
*rips = ips;
|
||||
*rref = ref;
|
||||
return TEST_ROWS;
|
||||
}
|
||||
|
||||
int fire_tests(radix_holder_t t, ipv6_t* ips, int* ref, int num_tests ) {
|
||||
int row;
|
||||
int wrong = 0;
|
||||
|
||||
for (row = 0; row < TEST_ROWS; ++ row) {
|
||||
if (radix_search(t, ips[row]) != ref[row]) {
|
||||
printf("ref: %d got: %d @ %d\n", ref[row], radix_search(t, ips[row]), row );
|
||||
++wrong;
|
||||
}
|
||||
}
|
||||
|
||||
free(ips);
|
||||
free(ref);
|
||||
return wrong;
|
||||
}
|
||||
|
||||
int main() {
|
||||
ipv6_t* ips;
|
||||
int* ref;
|
||||
|
||||
radix_holder_t t = create_holder();
|
||||
|
||||
printf("Loaded %d rules\n", load_input(t) );
|
||||
/*
|
||||
printf("Loaded %d tests\n", load_tests(&ips, &ref) );
|
||||
printf("%d tests wrong\n", fire_tests(t, ips, ref, TEST_ROWS) );
|
||||
*/
|
||||
destory_holder(t);
|
||||
return 0;
|
||||
}
|
|
@ -1,48 +0,0 @@
|
|||
#ifndef RADIX_HEADER_736213980973149671328765
|
||||
#define RADIX_HEADER_736213980973149671328765
|
||||
|
||||
#include "stdio.h"
|
||||
#include "stdlib.h"
|
||||
#include "string.h"
|
||||
|
||||
#define VALID_RULE 1
|
||||
#define INVALID_RULE 0
|
||||
#define RULE_NOT_FOUND -1
|
||||
|
||||
union ipv6 {
|
||||
unsigned short s[8];
|
||||
unsigned int i[4];
|
||||
unsigned long l[2];
|
||||
};
|
||||
|
||||
struct radix_node {
|
||||
struct radix_node* _l;
|
||||
struct radix_node* _r;
|
||||
union ipv6 _ip;
|
||||
unsigned short _rule;
|
||||
unsigned _ip_end_bit:7;
|
||||
unsigned _rule_valid:1;
|
||||
};
|
||||
|
||||
struct radix_holder {
|
||||
struct radix_node* _root;
|
||||
size_t _size;
|
||||
};
|
||||
|
||||
typedef struct radix_node* radix_node_t;
|
||||
typedef struct radix_holder* radix_holder_t;
|
||||
typedef union ipv6 ipv6_t;
|
||||
typedef int rule_t;
|
||||
|
||||
struct radix_holder* create_holder();
|
||||
struct radix_node* create_node(ipv6_t key, unsigned end_bit, unsigned rule);
|
||||
static struct radix_node* clone_node(struct radix_node* n);
|
||||
static void split_node(struct radix_node* n, unsigned bit);
|
||||
|
||||
int destory_holder(struct radix_holder*);
|
||||
int destory_node(struct radix_node*);
|
||||
|
||||
int radix_insert(struct radix_holder*, ipv6_t key, int val, int mask);
|
||||
rule_t radix_search(struct radix_holder*, ipv6_t key);
|
||||
|
||||
#endif
|
|
@ -18,7 +18,7 @@ union ipv6 {
|
|||
struct trie_node {
|
||||
struct trie_node* _l;
|
||||
struct trie_node* _r;
|
||||
unsigned _rule:16;
|
||||
int _rule:31;
|
||||
unsigned _rule_valid:1;
|
||||
};
|
||||
|
||||
|
|
Loading…
Reference in New Issue