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
|
/* 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/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) ;
/* 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 *) ;
/* log */
extern void log_newtcp4 (char const *, uint16_t) ;
#ifdef SKALIBS_IPV6_ENABLED
extern void log_newtcp6 (char const *, uint16_t) ;
extern void log_tcptimeout (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 ;
uint16_t i ;
uint16_t port ;
uint8_t source ;
char ip[SKALIBS_IP_SIZE] ;
} ;
#define QUERY_ZERO { .dt = S6DNS_ENGINE_ZERO, .prev = 0, .next = 0, .xindex = UINT16_MAX, .i = 0, .port = 0, .source = 0, .ip = { 0 } }
#define nq (genset_n(&g->queries) - 1)
#define QUERY(i) genset_p(query, &g->queries, (i))
#define qstart (QUERY(g->qsentinel)->next)
extern uint16_t query_abort (uint16_t) ;
extern uint16_t query_fail (uint16_t) ;
extern uint16_t query_succeed (uint16_t) ;
extern int query_new (uint8_t, uint16_t, char const *, uint16_t, char const *, uint16_t) ;
/* tcpconnection */
typedef struct tcpconnection_s tcpconnection, *tcpconnection_ref ;
struct tcpconnection_s
{
bufalloc out ;
stralloc in ;
uint32_t instate ;
tain rdeadline ;
tain wdeadline ;
genalloc queries ; /* 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 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 }
#ifdef SKALIBS_IPV6_ENABLED
typedef struct udp6msg_s udp6msg, *udp6msg_ref ;
struct udp6msg_s
{
char ip[16] ;
uint16_t port ;
uint16_t len ;
} ;
#define UDP6MSG_ZERO { .ip = { 0 }, .port = 0, .len = 0 }
#endif
typedef struct udpqueue_s udpqueue, *udpqueue_ref ;
struct udpqueue_s
{
int fd ;
stralloc storage ;
genalloc messages ; /* udp[46]msg */
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_add4 (udpqueue *, char const *, uint16_t, char const *, uint16_t) ;
extern int udpqueue_flush4 (udpqueue *) ;
#ifdef SKALIBS_IPV6_ENABLED
extern int udpqueue_add6 (udpqueue *, char const *, uint16_t, char const *, uint16_t) ;
extern int udpqueue_flush6 (udpqueue *) ;
#endif
/* main */
typedef struct global_s global, *global_ref ;
struct global_s
{
cdb confdb ;
char const *dumpfile ;
uint16_t verbosity ;
tain rtto ;
tain wtto ;
genset tcpconnections ; /* tcpconnection */
genset queries ; /* query */
uint16_t tcpsentinel ;
uint16_t qsentinel ;
} ;
#define GLOBAL_ZERO { \
.confdb = CDB_ZERO, \
.dumpfile = 0, \
.verbosity = 1, \
.rtto = TAIN_INFINITE, \
.wtto = TAIN_INFINITE, \
.tcpconnections = GENSET_ZERO, \
.queries = GENSET_ZERO, \
.tcpsentinel = 0, \
.qsentinel = 0, \
}
extern global *g ;
#endif
|
