diff options
Diffstat (limited to 'src/cache')
-rw-r--r-- | src/cache/access.c | 47 | ||||
-rw-r--r-- | src/cache/tcpconnection.c | 44 |
2 files changed, 47 insertions, 44 deletions
diff --git a/src/cache/access.c b/src/cache/access.c new file mode 100644 index 0000000..8f87125 --- /dev/null +++ b/src/cache/access.c @@ -0,0 +1,47 @@ +/* ISC license. */ + +#include <stdint.h> + +#include <skalibs/cdb.h> + +#include "shibari-cache-internal.h" + +static inline int check (char const *key, size_t keylen) +{ + cdb_data data ; + return cdb_find(&confdb, &data, key, keylen) ; +} + +int ip4_access (char const *ip) +{ + int r ; + char key[9] = "A4:" ; + uint8_t i = 33 ; + memcpy(key+4, ip, 4) ; + key[8] = 0 ; + while (i--) + { + key[3] = i ; + key[4 + (i>>3)] &= ~(1U << (7 - (i & 7))) ; + r = check(key, 8) ; + if (r) return r ; + } + return 0 ; +} + +int ip6_access (char const *ip) +{ + int r ; + char key[21] = "A6:" ; + uint8_t i = 129 ; + memcpy(key+4, ip, 16) ; + key[20] = 0 ; + while (i--) + { + key[3] = i ; + key[4 + (i>>3)] &= ~(1U << (7 - (i & 7))) ; + r = check(key, 20) ; + if (r) return r ; + } + return 0 ; +} diff --git a/src/cache/tcpconnection.c b/src/cache/tcpconnection.c index 7c7f714..bd72fb4 100644 --- a/src/cache/tcpconnection.c +++ b/src/cache/tcpconnection.c @@ -1,49 +1,5 @@ /* ISC license. */ -#include <stdint.h> - -#include <skalibs/cdb.h> - #include "shibari-cache-internal.h" genset *tcpconn = 0 ; - -static inline int check (char const *key, size_t keylen) -{ - cdb_data data ; - return cdb_find(&confdb, &data, key, keylen) ; -} - -int tcp4_access (char const *ip) -{ - int r ; - char key[9] = "A4:" ; - uint8_t i = 33 ; - memcpy(key+4, ip, 4) ; - key[8] = 0 ; - while (i--) - { - key[3] = i ; - key[4 + (i>>3)] &= ~(1U << (7 - (i & 7))) ; - r = check(key, 8) ; - if (r) return r ; - } - return 0 ; -} - -int tcp6_access (char const *ip) -{ - int r ; - char key[21] = "A6:" ; - uint8_t i = 129 ; - memcpy(key+4, ip, 16) ; - key[20] = 0 ; - while (i--) - { - key[3] = i ; - key[4 + (i>>3)] &= ~(1U << (7 - (i & 7))) ; - r = check(key, 20) ; - if (r) return r ; - } - return 0 ; -} |