diff --git a/trie/trie.c b/trie/trie.c index 3da1ef6..1c542b5 100644 --- a/trie/trie.c +++ b/trie/trie.c @@ -1,8 +1,11 @@ #include "trie.h" +#include -#define INPUT_FILE "../data/routing-data" +#define RULES_FILE "../data/routing-data" +#define TEST_FILE "../data/test-data" +#define TEST_ROWS 17217 #define READ_MODE "r" -#define GET_BIT(key, bit) ( ( key.l[ bit / 64 ] & (1ul << (63 - bit % 64)) ) != 0 ) +#define GET_BIT(key, bit) ( ( key.l[ bit / 64 ] & (1ul << (bit % 64)) ) != 0 ) trie_node_t create_node() { @@ -20,12 +23,12 @@ trie_holder_t create_holder() { int trie_insert(trie_holder_t t, ipv6_t key, int val, int mask) { trie_node_t iter = t->_root; trie_node_t* next = &t->_root; - int i = 0; + int i; - for (;i <= 128 - mask; ++i) { + for (i = 0; i < mask; ++i) { if (!*(next = GET_BIT(key, i) ? &iter->_l : &iter->_r)) *next = create_node(); - + iter = *next; } @@ -45,7 +48,21 @@ int trie_insert(trie_holder_t t, ipv6_t key, int val, int mask) { } int trie_search(trie_holder_t t, ipv6_t key) { - return 0; + trie_node_t iter = t->_root; + trie_node_t next; + int i = 0; + int ret = -1; + + for (;i < 128; ++i) { + if (iter->_rule_valid) + ret = iter->_rule; + + if (!(next = GET_BIT(key, i) ? iter->_l : iter->_r)) + break; + + iter = next; + } + return ret; } @@ -56,8 +73,8 @@ int parse_line(ipv6_t* key, int* val, int* mask, FILE* f) { /*! assumes only goo 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]); + &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) != '/'); @@ -70,12 +87,13 @@ int load_input(trie_holder_t t) { int val; int mask; ipv6_t key; - FILE* f; int cnt = 0; - if (!(f = fopen(INPUT_FILE, READ_MODE))) - perror("cant open input file"); + if (!(f = fopen(RULES_FILE, READ_MODE))) { + puts("cant open input file"); + exit(1); + } while (parse_line(&key, &val, &mask, f)) { trie_insert(t, key, val, mask); @@ -84,22 +102,53 @@ int load_input(trie_holder_t t) { return cnt; } -int main() { - trie_holder_t t = create_holder(); +int load_tests(ipv6_t** rips, int** rref) { + ipv6_t* ips; + int* ref; + int row; + FILE* f; - /* - unsigned key[4] = {8927349, 9872346, 898097878, 928398629}; - int i; - 00000000100010000011100001110101 00000000100101101010001111011010 00110101100001111110001011010110 00110111010101100011110100100101 - 00000000100010000011100001110101 00000000100101101010001111011010 00110101100001111110001011010110 00110111010101100011110100100101 - - for (i = 0; i < 128; ++i) { - printf("%d", GET_BIT(key, i)); + if (!(f = fopen( TEST_FILE, READ_MODE ))) { + puts("cant open test file"); + exit(1); } - printf("\n"); - return 0; + + 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])); + } + + *rips = ips; + *rref = ref; + return TEST_ROWS; +} + +int fire_tests(trie_holder_t t, ipv6_t* ips, int* ref, int num_tests ) { + int row; + for (row = 0; row < TEST_ROWS; ++ row) { + printf("%d %d\n", trie_search(t, ips[row]), ref[row] ); + } + + return 1; +} + +int main() { + ipv6_t* ips; + int* ref; + + trie_holder_t t = create_holder(); + printf("Loaded %d rules\n", load_input(t) ); + printf("Loaded %d tests\n", load_tests(&ips, &ref) ); + /* + fire_tests(t, ips, ref, TEST_ROWS); */ - printf("Loaded %d rules\n", load_input(t) ); return 0; } \ No newline at end of file