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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
/* ISC license. */
#ifndef SHIBARI_CACHE_INTERNAL_H
#define SHIBARI_CACHE_INTERNAL_H
#include <stdint.h>
#include <skalibs/uint64.h>
#include <skalibs/cdb.h>
#include <skalibs/tai.h>
#include <skalibs/stralloc.h>
#include <skalibs/genalloc.h>
#include <skalibs/bufalloc.h>
#include <skalibs/genset.h>
#include <skalibs/ip46.h>
#include <s6-dns/s6dns-engine.h>
/* cache */
extern void cache_init (uint64_t) ;
extern void cache_dump (void) ;
extern void cache_load (void) ;
/* conf */
extern int conf_getb (cdb const *, char const *, size_t, cdb_data *) ;
extern int conf_get (cdb const *, char const *, cdb_data *) ;
extern int conf_get_uint32 (cdb const *, char const *, uint32_t *) ;
extern int conf_get_uint64 (cdb const *, char const *, uint64_t *) ;
extern char const *conf_get_string (cdb const *, char const *) ;
/* tcpconnection */
typedef struct tcpconnection_s tcpconnection, *tcpconnection_ref ;
struct tcpconnection_s
{
bufalloc out ;
stralloc in ;
uint32_t instate ;
tain rdeadline ;
tain wdeadline ;
uint32_t prev ;
uint32_t next ;
uint32_t xindex ;
} ;
#define TCPCONNECTION_ZERO { .out = BUFALLOC_ZERO, .in = STRALLOC_ZERO, .instate = 0, .rdeadline = TAIN_INFINITE, .wdeadline = TAIN_INFINITE, .prev = 0, .next = 0. .xindex = UINT32_MAX }
#define ntcp (genset_n(&g->tcpconnections) - 1)
#define TCPCONNECTION(i) genset_p(tcpconnection, &g->tcpconnections, (i))
#define tcpstart (TCPCONNECTION(g->tcpsentinel)->next)
extern void tcpconnection_drop (tcpconnection *) ;
/* udpqueue */
typedef struct udp4msg_s udp4msg, *udp4msg_ref ;
struct udp4msg_s
{
char ip[4] ;
uint16_t port ;
uint16_t len ;
} ;
#define UDP4MSG_ZERO { .ip = { 0 }, .port = 0, .len = 0 }
typedef struct udp6msg_s udp6msg, *udp6msg_ref ;
struct udp4msg_s
{
char ip[16] ;
uint16_t port ;
uint16_t len ;
} ;
#define UDP6MSG_ZERO { .ip = { 0 }, .port = 0, .len = 0 }
typedef struct udpqueue_s udpqueue, *udpqueue_ref ;
struct udpqueue_s
{
int fd ;
stralloc storage ;
genalloc messages ; /* udp[46]msg */
tain deadline ;
uint32_t xindex ;
} ;
#define UDPQUEUE_ZERO { .fd = -1, .storage = STRALLOC_ZERO, .messages = GENALLOC_ZERO, .deadline = TAIN_INFINITE, .xindex = UINT32_MAX }
extern void udpqueue_drop (udpqueue *) ;
extern int udpqueue_add4 (udpqueue *, char const *, uint16_t) ;
extern int udpqueue_flush4 (udpqueue *) ;
#ifdef SKALIBS_IPV6_ENABLED
extern int udpqueue_add6 (udpqueue *, char const *, uint16_t) ;
extern int udpqueue_flush6 (udpqueue *) ;
#endif
/* query */
typedef struct query_s query, *query_ref ;
struct query_s
{
s6dns_engine_t dt ;
uint32_t origin ;
uint32_t prev ;
uint32_t next ;
uint32_t xindex ;
char ip[16] ;
uint16_t port ;
} ;
#define QUERY_ZERO { .dt = S6DNS_ENGINE_ZERO, .origin = 0, .prev = 0, .next = 0, .xindex = UINT32_MAX, .ip = { 0 }, .port = 0 }
#define nq (genset_n(&g->queries) - 1)
#define QUERY(i) genset_p(query, &g->queries, (i))
#define qstart (QUERY(g->qsentinel)->next)
extern void query_fail (query *) ;
extern void query_success (query *) ;
extern void query_new (uint32_t, char const *, uint8_t, uint16_t, char const *, uint16_t) ;
/* main */
typedef struct global_s global, *global_ref ;
struct global_s
{
cdb confdb ;
char const *dumpfile ;
uint32_t verbosity ;
tain rtto ;
tain wtto ;
genset tcpconnections ; /* tcpconnection */
uint32_t tcpsentinel ;
genset queries ; /* query */
uint32_t qsentinel ;
} ;
#define GLOBAL_ZERO { \
.confdb = CDB_ZERO, \
.dumpfile = 0, \
.verbosity = 1, \
.rtto = TAIN_INFINITE, \
.wtto = TAIN_INFINITE, \
.tcpconnections = GENSET_ZERO, \
.tcpsentinel = 0, \
.queries = GENSET_ZERO, \
.qsentinel = 0, \
}
extern global *g ;
#endif
|
