-
Notifications
You must be signed in to change notification settings - Fork 180
/
cidr-trie.h
65 lines (53 loc) · 1.67 KB
/
cidr-trie.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
/*
* graftcp
* Copyright (C) 2023 Hmgle <[email protected]>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*/
/* Inspired by nginx: ngx_radix_tree.h */
#ifndef CIDR_TRIE_H
#define CIDR_TRIE_H
#include <netinet/in.h>
#include <stdint.h>
#define TRIE_NO_VALUE 0
typedef struct trie_node_s trie_node_t;
struct trie_node_s {
trie_node_t *left;
trie_node_t *right;
int value;
};
typedef struct {
trie_node_t *root;
} trie_t;
struct cidr_s {
uint32_t addr;
uint32_t mask;
};
struct cidr6_s {
struct in6_addr addr;
struct in6_addr mask;
};
trie_t *trie_new();
void trie32_insert(trie_t *trie, struct cidr_s *cidr, int value);
int trie32_insert_str(trie_t *trie, const char *ipstr, int value);
int trie32_lookup(trie_t *trie, uint32_t ip);
void trie128_insert(trie_t *trie, struct cidr6_s *cidr6, int value);
int trie128_insert_str(trie_t *trie, const char *ipstr, int value);
int trie128_lookup(trie_t *trie, uint8_t *ip);
typedef struct {
trie_t *cidr4_trie;
trie_t *cidr6_trie;
} cidr_trie_t;
cidr_trie_t *cidr_trie_new();
void cidr_trie_insert_str(cidr_trie_t *cidr_trie, const char *ipstr, int value);
int cidr4_trie_lookup(cidr_trie_t *cidr_trie, uint32_t ip);
int cidr6_trie_lookup(cidr_trie_t *cidr_trie, uint8_t *ip);
#endif