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
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
|
/* ISC license. */
#ifndef SHIBARI_CACHE_INTERNAL_H
#define SHIBARI_CACHE_INTERNAL_H
#include <stddef.h>
#include <stdint.h>
#include <skalibs/uint64.h>
#include <skalibs/cdb.h>
#include <skalibs/tai.h>
#include <skalibs/strerr.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-domain.h>
#include <s6-dns/s6dns-engine.h>
#include <shibari/dcache.h>
#define dienomem() strerr_diefu1sys(111, "concatenate data") ;
#define dienewquery() strerr_diefu1sys(111, "prepare new query") ;
/* cache */
extern void cache_init (uint64_t) ;
extern void cache_dump (void) ;
extern void cache_load (void) ;
extern int cache_search (s6dns_domain_t const *, uint16_t, dcache_key_t *) ;
/* clientaccess */
extern int clientaccess_ip4 (char const *) ;
#ifdef SKALIBS_IPV6_ENABLED
extern int clientaccess_ip6 (char const *) ;
#endif
/* conf */
extern int conf_getb (char const *, size_t, cdb_data *) ;
extern int conf_get (char const *, cdb_data *) ;
extern int conf_get_uint16 (char const *, uint16_t *) ;
extern int conf_get_uint32 (char const *, uint32_t *) ;
extern int conf_get_uint64 (char const *, uint64_t *) ;
extern char const *conf_get_string (char const *) ;
/* dns */
typedef struct dnstask_s dnstask, *dnstask_ref ;
struct dnstask_s
{
stralloc sa ;
char ip[SKALIBS_IP_SIZE] ;
uint16_t qtype ;
uint16_t prefixlen ;
uint16_t state : 13 ;
uint16_t spin : 1 ;
uint16_t source : 2 ;
uint16_t sid ;
uint16 query[2] ;
} ;
#define DNSTASK_ZERO { .sa = STRALLOC_ZERO, .ip = { 0 }, .qtype = 0, .prefixlen = 0, .state = 0, .spin = 0, .source = 0, .sid = 0, query = { 0, 0 } }
#define ntasks genset_n(&g->tasks)
#define DNSTASK(i) genset_p(dnstask, &g->dnstasks, (i))
extern void dnstask_wakeup (uint16_t, uint16_t, uint32_t) ;
extern int dns_start (uint8_t, uint16_t, char const *, uint16_t, char const *, uint16_t) ;
/* log */
extern void log_udp4bad (char const *, uint16_t) ;
extern void log_newtcp4 (char const *, uint16_t) ;
extern void log_tcpbad (uint16_t) ;
extern void log_tcptimeout (uint16_t) ;
#ifdef SKALIBS_IPV6_ENABLED
extern void log_udp6bad (char const *, uint16_t) ;
extern void log_newtcp6 (char const *, uint16_t) ;
#endif
/* query */
typedef struct query_s query, *query_ref ;
struct query_s
{
s6dns_engine_t dt ;
uint16_t prev ;
uint16_t next ;
uint16_t xindex ;
} ;
#define QUERY_ZERO { .dt = S6DNS_ENGINE_ZERO, .prev = 0, .next = 0, .xindex = UINT16_MAX }
#define nq (gensetdyn_n(&g->queries) - 1)
#define QUERY(i) GENSETDYN_P(query, &g->queries, (i))
#define qstart (QUERY(g->qsentinel)->next)
extern uint16_t query_abort (uint16_t) ;
extern uint16_t query_event (uint16_t) ;
extern int query_start (uint16_t, char const *, uint16_t, uint16_t, char const *, uint16_t, char const *, uint16_t, uint32_t) ;
extern int query_end (uint8_t, uint16_t, char const *, uint16_t, char const *, uint16_t) ;
/* dns */
extern void dns_start (query *) ;
/* tcpconnection */
typedef struct tcpconnection_s tcpconnection, *tcpconnection_ref ;
struct tcpconnection_s
{
bufalloc out ;
stralloc in ;
uint32_t instate ;
tain rdeadline ;
tain wdeadline ;
genalloc tasks ; /* uint16_t */
uint16_t prev ;
uint16_t next ;
uint16_t xindex ;
} ;
#define TCPCONNECTION_ZERO { .out = BUFALLOC_ZERO, .in = STRALLOC_ZERO, .instate = 0, .rdeadline = TAIN_INFINITE, .wdeadline = TAIN_INFINITE, .queries = GENALLOC_ZERO, .prev = 0, .next = 0, .xindex = UINT16_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_removequery (tcpconnection *, uint16_t) ;
extern uint16_t tcpconnection_delete (tcpconnection *) ;
extern int tcpconnection_flush (tcpconnection *) ;
extern void tcpconnection_new (int) ;
/* udpqueue */
typedef struct udpaux_s udpaux, *udpaux_ref ;
struct udpaux_s
{
uint16_t port ;
uint16_t len ;
} ;
#define UDPAUX_ZERO { .port = 0, .len = 0 }
typedef struct udpqueue_s udpqueue, *udpqueue_ref ;
struct udpqueue_s
{
int fd ;
stralloc storage ;
genalloc messages ; /* udpaux */
tain deadline ;
uint16_t xindex ;
} ;
#define UDPQUEUE_ZERO { .fd = -1, .storage = STRALLOC_ZERO, .messages = GENALLOC_ZERO, .deadline = TAIN_INFINITE, .xindex = UINT16_MAX }
extern void udpqueue_drop (udpqueue *) ;
extern int udpqueue_add (udpqueue *, uint8_t, char const *, uint16_t, char const *, uint16_t) ;
extern int udpqueue_flush (udpqueue *, uint8_t) ;
/* main */
typedef struct global_s global, *global_ref ;
struct global_s
{
cdb confdb ;
char const *dumpfile ;
tain rtto ;
tain wtto ;
tain qtto ;
udpqueue *udpqueues[2] ;
genset tcpconnections ;
genset dnstasks ;
gensetdyn queries ;
uint16_t tcpsentinel ;
uint16_t qsentinel ;
uint16_t verbosity ;
} ;
#define GLOBAL_ZERO { \
.confdb = CDB_ZERO, \
.dumpfile = 0, \
.rtto = TAIN_INFINITE, \
.wtto = TAIN_INFINITE, \
.qtto = TAIN_INFINITE, \
.udpqueues = { 0, 0 }, \
.tcpconnections = GENSET_ZERO, \
.dnstasks = GENSET_ZERO, \
.queries = GENSETDYN_INIT(query, 3, 3, 8), \
.tcpsentinel = 0, \
.qsentinel = 0, \
.verbosity = 1, \
}
extern global *g ;
#endif
|
