hladu357
/
CDN_DNS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tests added

This commit is contained in:
hladu357 2024-06-06 10:33:28 +02:00
parent 657b8b17b9
commit 19304a2c57
1 changed files with 73 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

95
trie/trie.c
View File

@ -1,8 +1,11 @@
#include "trie.h"
#include <assert.h>
#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,9 +23,9 @@ 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();
@ -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;
}