dtstream.c revision 1.1.1.5 1 1.1 christos /*
2 1.1 christos * dnstap/dtstream.c - Frame Streams thread for unbound DNSTAP
3 1.1 christos *
4 1.1 christos * Copyright (c) 2020, NLnet Labs. All rights reserved.
5 1.1 christos *
6 1.1 christos * This software is open source.
7 1.1 christos *
8 1.1 christos * Redistribution and use in source and binary forms, with or without
9 1.1 christos * modification, are permitted provided that the following conditions
10 1.1 christos * are met:
11 1.1 christos *
12 1.1 christos * Redistributions of source code must retain the above copyright notice,
13 1.1 christos * this list of conditions and the following disclaimer.
14 1.1 christos *
15 1.1 christos * Redistributions in binary form must reproduce the above copyright notice,
16 1.1 christos * this list of conditions and the following disclaimer in the documentation
17 1.1 christos * and/or other materials provided with the distribution.
18 1.1 christos *
19 1.1 christos * Neither the name of the NLNET LABS nor the names of its contributors may
20 1.1 christos * be used to endorse or promote products derived from this software without
21 1.1 christos * specific prior written permission.
22 1.1 christos *
23 1.1 christos * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24 1.1 christos * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25 1.1 christos * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26 1.1 christos * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27 1.1 christos * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28 1.1 christos * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
29 1.1 christos * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
30 1.1 christos * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
31 1.1 christos * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
32 1.1 christos * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
33 1.1 christos * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34 1.1 christos *
35 1.1 christos */
36 1.1 christos
37 1.1 christos /**
38 1.1 christos * \file
39 1.1 christos *
40 1.1 christos * An implementation of the Frame Streams data transport protocol for
41 1.1 christos * the Unbound DNSTAP message logging facility.
42 1.1 christos */
43 1.1 christos
44 1.1 christos #include "config.h"
45 1.1 christos #include "dnstap/dtstream.h"
46 1.1 christos #include "dnstap/dnstap_fstrm.h"
47 1.1 christos #include "util/config_file.h"
48 1.1 christos #include "util/ub_event.h"
49 1.1 christos #include "util/net_help.h"
50 1.1 christos #include "services/outside_network.h"
51 1.1 christos #include "sldns/sbuffer.h"
52 1.1 christos #ifdef HAVE_SYS_UN_H
53 1.1 christos #include <sys/un.h>
54 1.1 christos #endif
55 1.1 christos #include <fcntl.h>
56 1.1 christos #ifdef HAVE_OPENSSL_SSL_H
57 1.1 christos #include <openssl/ssl.h>
58 1.1 christos #endif
59 1.1 christos #ifdef HAVE_OPENSSL_ERR_H
60 1.1 christos #include <openssl/err.h>
61 1.1 christos #endif
62 1.1 christos
63 1.1 christos /** number of messages to process in one output callback */
64 1.1 christos #define DTIO_MESSAGES_PER_CALLBACK 100
65 1.1 christos /** the msec to wait for reconnect (if not immediate, the first attempt) */
66 1.1 christos #define DTIO_RECONNECT_TIMEOUT_MIN 10
67 1.1 christos /** the msec to wait for reconnect max after backoff */
68 1.1 christos #define DTIO_RECONNECT_TIMEOUT_MAX 1000
69 1.1 christos /** the msec to wait for reconnect slow, to stop busy spinning on reconnect */
70 1.1 christos #define DTIO_RECONNECT_TIMEOUT_SLOW 1000
71 1.1 christos /** number of messages before wakeup of thread */
72 1.1 christos #define DTIO_MSG_FOR_WAKEUP 32
73 1.1 christos
74 1.1 christos /** maximum length of received frame */
75 1.1 christos #define DTIO_RECV_FRAME_MAX_LEN 1000
76 1.1 christos
77 1.1 christos struct stop_flush_info;
78 1.1 christos /** DTIO command channel commands */
79 1.1 christos enum {
80 1.1 christos /** DTIO command channel stop */
81 1.1 christos DTIO_COMMAND_STOP = 0,
82 1.1 christos /** DTIO command channel wakeup */
83 1.1 christos DTIO_COMMAND_WAKEUP = 1
84 1.1 christos } dtio_channel_command;
85 1.1 christos
86 1.1 christos /** open the output channel */
87 1.1 christos static void dtio_open_output(struct dt_io_thread* dtio);
88 1.1 christos /** add output event for read and write */
89 1.1 christos static int dtio_add_output_event_write(struct dt_io_thread* dtio);
90 1.1 christos /** start reconnection attempts */
91 1.1 christos static void dtio_reconnect_enable(struct dt_io_thread* dtio);
92 1.1 christos /** stop from stop_flush event loop */
93 1.1 christos static void dtio_stop_flush_exit(struct stop_flush_info* info);
94 1.1 christos /** setup a start control message */
95 1.1 christos static int dtio_control_start_send(struct dt_io_thread* dtio);
96 1.1 christos #ifdef HAVE_SSL
97 1.1 christos /** enable briefly waiting for a read event, for SSL negotiation */
98 1.1 christos static int dtio_enable_brief_read(struct dt_io_thread* dtio);
99 1.1 christos /** enable briefly waiting for a write event, for SSL negotiation */
100 1.1 christos static int dtio_enable_brief_write(struct dt_io_thread* dtio);
101 1.1 christos #endif
102 1.1 christos
103 1.1 christos struct dt_msg_queue*
104 1.1 christos dt_msg_queue_create(struct comm_base* base)
105 1.1 christos {
106 1.1 christos struct dt_msg_queue* mq = calloc(1, sizeof(*mq));
107 1.1 christos if(!mq) return NULL;
108 1.1 christos mq->maxsize = 1*1024*1024; /* set max size of buffer, per worker,
109 1.1 christos about 1 M should contain 64K messages with some overhead,
110 1.1 christos or a whole bunch smaller ones */
111 1.1 christos mq->wakeup_timer = comm_timer_create(base, mq_wakeup_cb, mq);
112 1.1 christos if(!mq->wakeup_timer) {
113 1.1 christos free(mq);
114 1.1 christos return NULL;
115 1.1 christos }
116 1.1 christos lock_basic_init(&mq->lock);
117 1.1 christos lock_protect(&mq->lock, mq, sizeof(*mq));
118 1.1 christos return mq;
119 1.1 christos }
120 1.1 christos
121 1.1 christos /** clear the message list, caller must hold the lock */
122 1.1 christos static void
123 1.1 christos dt_msg_queue_clear(struct dt_msg_queue* mq)
124 1.1 christos {
125 1.1 christos struct dt_msg_entry* e = mq->first, *next=NULL;
126 1.1 christos while(e) {
127 1.1 christos next = e->next;
128 1.1 christos free(e->buf);
129 1.1 christos free(e);
130 1.1 christos e = next;
131 1.1 christos }
132 1.1 christos mq->first = NULL;
133 1.1 christos mq->last = NULL;
134 1.1 christos mq->cursize = 0;
135 1.1 christos mq->msgcount = 0;
136 1.1 christos }
137 1.1 christos
138 1.1 christos void
139 1.1 christos dt_msg_queue_delete(struct dt_msg_queue* mq)
140 1.1 christos {
141 1.1 christos if(!mq) return;
142 1.1 christos lock_basic_destroy(&mq->lock);
143 1.1 christos dt_msg_queue_clear(mq);
144 1.1 christos comm_timer_delete(mq->wakeup_timer);
145 1.1 christos free(mq);
146 1.1 christos }
147 1.1 christos
148 1.1 christos /** make the dtio wake up by sending a wakeup command */
149 1.1 christos static void dtio_wakeup(struct dt_io_thread* dtio)
150 1.1 christos {
151 1.1 christos uint8_t cmd = DTIO_COMMAND_WAKEUP;
152 1.1 christos if(!dtio) return;
153 1.1 christos if(!dtio->started) return;
154 1.1 christos
155 1.1 christos while(1) {
156 1.1 christos ssize_t r = write(dtio->commandpipe[1], &cmd, sizeof(cmd));
157 1.1 christos if(r == -1) {
158 1.1 christos #ifndef USE_WINSOCK
159 1.1 christos if(errno == EINTR || errno == EAGAIN)
160 1.1 christos continue;
161 1.1 christos #else
162 1.1 christos if(WSAGetLastError() == WSAEINPROGRESS)
163 1.1 christos continue;
164 1.1 christos if(WSAGetLastError() == WSAEWOULDBLOCK)
165 1.1 christos continue;
166 1.1 christos #endif
167 1.1 christos log_err("dnstap io wakeup: write: %s",
168 1.1 christos sock_strerror(errno));
169 1.1 christos break;
170 1.1 christos }
171 1.1 christos break;
172 1.1 christos }
173 1.1 christos }
174 1.1 christos
175 1.1 christos void
176 1.1 christos mq_wakeup_cb(void* arg)
177 1.1 christos {
178 1.1 christos struct dt_msg_queue* mq = (struct dt_msg_queue*)arg;
179 1.1.1.4 christos
180 1.1 christos lock_basic_lock(&mq->dtio->wakeup_timer_lock);
181 1.1 christos mq->dtio->wakeup_timer_enabled = 0;
182 1.1 christos lock_basic_unlock(&mq->dtio->wakeup_timer_lock);
183 1.1 christos dtio_wakeup(mq->dtio);
184 1.1 christos }
185 1.1 christos
186 1.1 christos /** start timer to wakeup dtio because there is content in the queue */
187 1.1 christos static void
188 1.1.1.2 christos dt_msg_queue_start_timer(struct dt_msg_queue* mq, int wakeupnow)
189 1.1 christos {
190 1.1.1.2 christos struct timeval tv = {0};
191 1.1 christos /* Start a timer to process messages to be logged.
192 1.1 christos * If we woke up the dtio thread for every message, the wakeup
193 1.1 christos * messages take up too much processing power. If the queue
194 1.1 christos * fills up the wakeup happens immediately. The timer wakes it up
195 1.1 christos * if there are infrequent messages to log. */
196 1.1 christos
197 1.1 christos /* we cannot start a timer in dtio thread, because it is a different
198 1.1 christos * thread and its event base is in use by the other thread, it would
199 1.1 christos * give race conditions if we tried to modify its event base,
200 1.1 christos * and locks would wait until it woke up, and this is what we do. */
201 1.1 christos
202 1.1 christos /* do not start the timer if a timer already exists, perhaps
203 1.1 christos * in another worker. So this variable is protected by a lock in
204 1.1.1.2 christos * dtio. */
205 1.1.1.2 christos
206 1.1.1.2 christos /* If we need to wakeupnow, 0 the timer to force the callback. */
207 1.1 christos lock_basic_lock(&mq->dtio->wakeup_timer_lock);
208 1.1 christos if(mq->dtio->wakeup_timer_enabled) {
209 1.1.1.2 christos if(wakeupnow) {
210 1.1.1.4 christos tv.tv_sec = 0;
211 1.1.1.4 christos tv.tv_usec = 0;
212 1.1.1.2 christos comm_timer_set(mq->wakeup_timer, &tv);
213 1.1.1.2 christos }
214 1.1 christos lock_basic_unlock(&mq->dtio->wakeup_timer_lock);
215 1.1 christos return;
216 1.1 christos }
217 1.1 christos mq->dtio->wakeup_timer_enabled = 1; /* we are going to start one */
218 1.1 christos
219 1.1 christos /* start the timer, in mq, in the event base of our worker */
220 1.1.1.2 christos if(!wakeupnow) {
221 1.1.1.2 christos tv.tv_sec = 1;
222 1.1.1.2 christos tv.tv_usec = 0;
223 1.1.1.4 christos /* If it is already set, keep it running. */
224 1.1.1.4 christos if(!comm_timer_is_set(mq->wakeup_timer))
225 1.1.1.4 christos comm_timer_set(mq->wakeup_timer, &tv);
226 1.1.1.4 christos } else {
227 1.1.1.4 christos tv.tv_sec = 0;
228 1.1.1.4 christos tv.tv_usec = 0;
229 1.1.1.4 christos comm_timer_set(mq->wakeup_timer, &tv);
230 1.1.1.2 christos }
231 1.1.1.2 christos lock_basic_unlock(&mq->dtio->wakeup_timer_lock);
232 1.1 christos }
233 1.1 christos
234 1.1 christos void
235 1.1 christos dt_msg_queue_submit(struct dt_msg_queue* mq, void* buf, size_t len)
236 1.1 christos {
237 1.1 christos int wakeupnow = 0, wakeupstarttimer = 0;
238 1.1 christos struct dt_msg_entry* entry;
239 1.1 christos
240 1.1 christos /* check conditions */
241 1.1 christos if(!buf) return;
242 1.1 christos if(len == 0) {
243 1.1 christos /* it is not possible to log entries with zero length,
244 1.1 christos * because the framestream protocol does not carry it.
245 1.1 christos * However the protobuf serialization does not create zero
246 1.1 christos * length datagrams for dnstap, so this should not happen. */
247 1.1 christos free(buf);
248 1.1 christos return;
249 1.1 christos }
250 1.1 christos if(!mq) {
251 1.1 christos free(buf);
252 1.1 christos return;
253 1.1 christos }
254 1.1 christos
255 1.1 christos /* allocate memory for queue entry */
256 1.1 christos entry = malloc(sizeof(*entry));
257 1.1 christos if(!entry) {
258 1.1 christos log_err("out of memory logging dnstap");
259 1.1 christos free(buf);
260 1.1 christos return;
261 1.1 christos }
262 1.1 christos entry->next = NULL;
263 1.1 christos entry->buf = buf;
264 1.1 christos entry->len = len;
265 1.1 christos
266 1.1.1.2 christos /* acquire lock */
267 1.1 christos lock_basic_lock(&mq->lock);
268 1.1.1.4 christos /* if list was empty, start timer for (eventual) wakeup,
269 1.1.1.4 christos * or if dtio is not writing now an eventual wakeup is needed. */
270 1.1.1.4 christos if(mq->first == NULL || !mq->dtio->event_added_is_write)
271 1.1 christos wakeupstarttimer = 1;
272 1.1 christos /* if list contains more than wakeupnum elements, wakeup now,
273 1.1 christos * or if list is (going to be) almost full */
274 1.1 christos if(mq->msgcount == DTIO_MSG_FOR_WAKEUP ||
275 1.1 christos (mq->cursize < mq->maxsize * 9 / 10 &&
276 1.1 christos mq->cursize+len >= mq->maxsize * 9 / 10))
277 1.1 christos wakeupnow = 1;
278 1.1 christos /* see if it is going to fit */
279 1.1 christos if(mq->cursize + len > mq->maxsize) {
280 1.1 christos /* buffer full, or congested. */
281 1.1 christos /* drop */
282 1.1 christos lock_basic_unlock(&mq->lock);
283 1.1 christos free(buf);
284 1.1 christos free(entry);
285 1.1 christos return;
286 1.1 christos }
287 1.1 christos mq->cursize += len;
288 1.1 christos mq->msgcount ++;
289 1.1 christos /* append to list */
290 1.1 christos if(mq->last) {
291 1.1 christos mq->last->next = entry;
292 1.1 christos } else {
293 1.1 christos mq->first = entry;
294 1.1 christos }
295 1.1 christos mq->last = entry;
296 1.1 christos /* release lock */
297 1.1 christos lock_basic_unlock(&mq->lock);
298 1.1 christos
299 1.1.1.2 christos if(wakeupnow || wakeupstarttimer) {
300 1.1.1.2 christos dt_msg_queue_start_timer(mq, wakeupnow);
301 1.1 christos }
302 1.1 christos }
303 1.1 christos
304 1.1 christos struct dt_io_thread* dt_io_thread_create(void)
305 1.1 christos {
306 1.1 christos struct dt_io_thread* dtio = calloc(1, sizeof(*dtio));
307 1.1 christos lock_basic_init(&dtio->wakeup_timer_lock);
308 1.1 christos lock_protect(&dtio->wakeup_timer_lock, &dtio->wakeup_timer_enabled,
309 1.1 christos sizeof(dtio->wakeup_timer_enabled));
310 1.1 christos return dtio;
311 1.1 christos }
312 1.1 christos
313 1.1 christos void dt_io_thread_delete(struct dt_io_thread* dtio)
314 1.1 christos {
315 1.1 christos struct dt_io_list_item* item, *nextitem;
316 1.1 christos if(!dtio) return;
317 1.1 christos lock_basic_destroy(&dtio->wakeup_timer_lock);
318 1.1 christos item=dtio->io_list;
319 1.1 christos while(item) {
320 1.1 christos nextitem = item->next;
321 1.1 christos free(item);
322 1.1 christos item = nextitem;
323 1.1 christos }
324 1.1 christos free(dtio->socket_path);
325 1.1 christos free(dtio->ip_str);
326 1.1 christos free(dtio->tls_server_name);
327 1.1 christos free(dtio->client_key_file);
328 1.1 christos free(dtio->client_cert_file);
329 1.1 christos if(dtio->ssl_ctx) {
330 1.1 christos #ifdef HAVE_SSL
331 1.1 christos SSL_CTX_free(dtio->ssl_ctx);
332 1.1 christos #endif
333 1.1 christos }
334 1.1 christos free(dtio);
335 1.1 christos }
336 1.1 christos
337 1.1 christos int dt_io_thread_apply_cfg(struct dt_io_thread* dtio, struct config_file *cfg)
338 1.1 christos {
339 1.1 christos if(!cfg->dnstap) {
340 1.1 christos log_warn("cannot setup dnstap because dnstap-enable is no");
341 1.1 christos return 0;
342 1.1 christos }
343 1.1 christos
344 1.1 christos /* what type of connectivity do we have */
345 1.1 christos if(cfg->dnstap_ip && cfg->dnstap_ip[0]) {
346 1.1 christos if(cfg->dnstap_tls)
347 1.1 christos dtio->upstream_is_tls = 1;
348 1.1 christos else dtio->upstream_is_tcp = 1;
349 1.1 christos } else {
350 1.1 christos dtio->upstream_is_unix = 1;
351 1.1 christos }
352 1.1 christos dtio->is_bidirectional = cfg->dnstap_bidirectional;
353 1.1 christos
354 1.1 christos if(dtio->upstream_is_unix) {
355 1.1 christos char* nm;
356 1.1 christos if(!cfg->dnstap_socket_path ||
357 1.1 christos cfg->dnstap_socket_path[0]==0) {
358 1.1 christos log_err("dnstap setup: no dnstap-socket-path for "
359 1.1 christos "socket connect");
360 1.1 christos return 0;
361 1.1 christos }
362 1.1 christos nm = cfg->dnstap_socket_path;
363 1.1 christos if(cfg->chrootdir && cfg->chrootdir[0] && strncmp(nm,
364 1.1 christos cfg->chrootdir, strlen(cfg->chrootdir)) == 0)
365 1.1 christos nm += strlen(cfg->chrootdir);
366 1.1 christos free(dtio->socket_path);
367 1.1 christos dtio->socket_path = strdup(nm);
368 1.1 christos if(!dtio->socket_path) {
369 1.1 christos log_err("dnstap setup: malloc failure");
370 1.1 christos return 0;
371 1.1 christos }
372 1.1 christos }
373 1.1 christos
374 1.1 christos if(dtio->upstream_is_tcp || dtio->upstream_is_tls) {
375 1.1 christos if(!cfg->dnstap_ip || cfg->dnstap_ip[0] == 0) {
376 1.1 christos log_err("dnstap setup: no dnstap-ip for TCP connect");
377 1.1 christos return 0;
378 1.1 christos }
379 1.1 christos free(dtio->ip_str);
380 1.1 christos dtio->ip_str = strdup(cfg->dnstap_ip);
381 1.1 christos if(!dtio->ip_str) {
382 1.1 christos log_err("dnstap setup: malloc failure");
383 1.1 christos return 0;
384 1.1 christos }
385 1.1 christos }
386 1.1 christos
387 1.1 christos if(dtio->upstream_is_tls) {
388 1.1 christos #ifdef HAVE_SSL
389 1.1 christos if(cfg->dnstap_tls_server_name &&
390 1.1 christos cfg->dnstap_tls_server_name[0]) {
391 1.1 christos free(dtio->tls_server_name);
392 1.1 christos dtio->tls_server_name = strdup(
393 1.1 christos cfg->dnstap_tls_server_name);
394 1.1 christos if(!dtio->tls_server_name) {
395 1.1 christos log_err("dnstap setup: malloc failure");
396 1.1 christos return 0;
397 1.1 christos }
398 1.1 christos if(!check_auth_name_for_ssl(dtio->tls_server_name))
399 1.1 christos return 0;
400 1.1 christos }
401 1.1 christos if(cfg->dnstap_tls_client_key_file &&
402 1.1 christos cfg->dnstap_tls_client_key_file[0]) {
403 1.1 christos dtio->use_client_certs = 1;
404 1.1 christos free(dtio->client_key_file);
405 1.1 christos dtio->client_key_file = strdup(
406 1.1 christos cfg->dnstap_tls_client_key_file);
407 1.1 christos if(!dtio->client_key_file) {
408 1.1 christos log_err("dnstap setup: malloc failure");
409 1.1 christos return 0;
410 1.1 christos }
411 1.1 christos if(!cfg->dnstap_tls_client_cert_file ||
412 1.1 christos cfg->dnstap_tls_client_cert_file[0]==0) {
413 1.1 christos log_err("dnstap setup: client key "
414 1.1 christos "authentication enabled with "
415 1.1 christos "dnstap-tls-client-key-file, but "
416 1.1 christos "no dnstap-tls-client-cert-file "
417 1.1 christos "is given");
418 1.1 christos return 0;
419 1.1 christos }
420 1.1 christos free(dtio->client_cert_file);
421 1.1 christos dtio->client_cert_file = strdup(
422 1.1 christos cfg->dnstap_tls_client_cert_file);
423 1.1 christos if(!dtio->client_cert_file) {
424 1.1 christos log_err("dnstap setup: malloc failure");
425 1.1 christos return 0;
426 1.1 christos }
427 1.1 christos } else {
428 1.1 christos dtio->use_client_certs = 0;
429 1.1 christos dtio->client_key_file = NULL;
430 1.1 christos dtio->client_cert_file = NULL;
431 1.1 christos }
432 1.1 christos
433 1.1 christos if(cfg->dnstap_tls_cert_bundle) {
434 1.1 christos dtio->ssl_ctx = connect_sslctx_create(
435 1.1 christos dtio->client_key_file,
436 1.1 christos dtio->client_cert_file,
437 1.1 christos cfg->dnstap_tls_cert_bundle, 0);
438 1.1 christos } else {
439 1.1 christos dtio->ssl_ctx = connect_sslctx_create(
440 1.1 christos dtio->client_key_file,
441 1.1 christos dtio->client_cert_file,
442 1.1 christos cfg->tls_cert_bundle, cfg->tls_win_cert);
443 1.1 christos }
444 1.1 christos if(!dtio->ssl_ctx) {
445 1.1 christos log_err("could not setup SSL CTX");
446 1.1 christos return 0;
447 1.1 christos }
448 1.1 christos dtio->tls_use_sni = cfg->tls_use_sni;
449 1.1 christos #endif /* HAVE_SSL */
450 1.1 christos }
451 1.1 christos return 1;
452 1.1 christos }
453 1.1 christos
454 1.1 christos int dt_io_thread_register_queue(struct dt_io_thread* dtio,
455 1.1 christos struct dt_msg_queue* mq)
456 1.1 christos {
457 1.1 christos struct dt_io_list_item* item = malloc(sizeof(*item));
458 1.1 christos if(!item) return 0;
459 1.1 christos lock_basic_lock(&mq->lock);
460 1.1 christos mq->dtio = dtio;
461 1.1 christos lock_basic_unlock(&mq->lock);
462 1.1 christos item->queue = mq;
463 1.1 christos item->next = dtio->io_list;
464 1.1 christos dtio->io_list = item;
465 1.1 christos dtio->io_list_iter = NULL;
466 1.1 christos return 1;
467 1.1 christos }
468 1.1 christos
469 1.1 christos void dt_io_thread_unregister_queue(struct dt_io_thread* dtio,
470 1.1 christos struct dt_msg_queue* mq)
471 1.1 christos {
472 1.1 christos struct dt_io_list_item* item, *prev=NULL;
473 1.1 christos if(!dtio) return;
474 1.1 christos item = dtio->io_list;
475 1.1 christos while(item) {
476 1.1 christos if(item->queue == mq) {
477 1.1 christos /* found it */
478 1.1 christos if(prev) prev->next = item->next;
479 1.1 christos else dtio->io_list = item->next;
480 1.1 christos /* the queue itself only registered, not deleted */
481 1.1 christos lock_basic_lock(&item->queue->lock);
482 1.1 christos item->queue->dtio = NULL;
483 1.1 christos lock_basic_unlock(&item->queue->lock);
484 1.1 christos free(item);
485 1.1 christos dtio->io_list_iter = NULL;
486 1.1 christos return;
487 1.1 christos }
488 1.1 christos prev = item;
489 1.1 christos item = item->next;
490 1.1 christos }
491 1.1 christos }
492 1.1 christos
493 1.1 christos /** pick a message from the queue, the routine locks and unlocks,
494 1.1 christos * returns true if there is a message */
495 1.1 christos static int dt_msg_queue_pop(struct dt_msg_queue* mq, void** buf,
496 1.1 christos size_t* len)
497 1.1 christos {
498 1.1 christos lock_basic_lock(&mq->lock);
499 1.1 christos if(mq->first) {
500 1.1 christos struct dt_msg_entry* entry = mq->first;
501 1.1 christos mq->first = entry->next;
502 1.1 christos if(!entry->next) mq->last = NULL;
503 1.1 christos mq->cursize -= entry->len;
504 1.1 christos mq->msgcount --;
505 1.1 christos lock_basic_unlock(&mq->lock);
506 1.1 christos
507 1.1 christos *buf = entry->buf;
508 1.1 christos *len = entry->len;
509 1.1 christos free(entry);
510 1.1 christos return 1;
511 1.1 christos }
512 1.1 christos lock_basic_unlock(&mq->lock);
513 1.1 christos return 0;
514 1.1 christos }
515 1.1 christos
516 1.1 christos /** find message in queue, false if no message, true if message to send */
517 1.1 christos static int dtio_find_in_queue(struct dt_io_thread* dtio,
518 1.1 christos struct dt_msg_queue* mq)
519 1.1 christos {
520 1.1 christos void* buf=NULL;
521 1.1 christos size_t len=0;
522 1.1 christos if(dt_msg_queue_pop(mq, &buf, &len)) {
523 1.1 christos dtio->cur_msg = buf;
524 1.1 christos dtio->cur_msg_len = len;
525 1.1 christos dtio->cur_msg_done = 0;
526 1.1 christos dtio->cur_msg_len_done = 0;
527 1.1 christos return 1;
528 1.1 christos }
529 1.1 christos return 0;
530 1.1 christos }
531 1.1 christos
532 1.1 christos /** find a new message to write, search message queues, false if none */
533 1.1 christos static int dtio_find_msg(struct dt_io_thread* dtio)
534 1.1 christos {
535 1.1 christos struct dt_io_list_item *spot, *item;
536 1.1 christos
537 1.1 christos spot = dtio->io_list_iter;
538 1.1 christos /* use the next queue for the next message lookup,
539 1.1 christos * if we hit the end(NULL) the NULL restarts the iter at start. */
540 1.1 christos if(spot)
541 1.1 christos dtio->io_list_iter = spot->next;
542 1.1 christos else if(dtio->io_list)
543 1.1 christos dtio->io_list_iter = dtio->io_list->next;
544 1.1 christos
545 1.1 christos /* scan from spot to end-of-io_list */
546 1.1 christos item = spot;
547 1.1 christos while(item) {
548 1.1 christos if(dtio_find_in_queue(dtio, item->queue))
549 1.1 christos return 1;
550 1.1 christos item = item->next;
551 1.1 christos }
552 1.1 christos /* scan starting at the start-of-list (to wrap around the end) */
553 1.1 christos item = dtio->io_list;
554 1.1 christos while(item) {
555 1.1 christos if(dtio_find_in_queue(dtio, item->queue))
556 1.1 christos return 1;
557 1.1 christos item = item->next;
558 1.1 christos }
559 1.1 christos return 0;
560 1.1 christos }
561 1.1 christos
562 1.1 christos /** callback for the dnstap reconnect, to start reconnecting to output */
563 1.1 christos void dtio_reconnect_timeout_cb(int ATTR_UNUSED(fd),
564 1.1 christos short ATTR_UNUSED(bits), void* arg)
565 1.1 christos {
566 1.1 christos struct dt_io_thread* dtio = (struct dt_io_thread*)arg;
567 1.1 christos dtio->reconnect_is_added = 0;
568 1.1 christos verbose(VERB_ALGO, "dnstap io: reconnect timer");
569 1.1 christos
570 1.1 christos dtio_open_output(dtio);
571 1.1 christos if(dtio->event) {
572 1.1 christos if(!dtio_add_output_event_write(dtio))
573 1.1 christos return;
574 1.1 christos /* nothing wrong so far, wait on the output event */
575 1.1 christos return;
576 1.1 christos }
577 1.1 christos /* exponential backoff and retry on timer */
578 1.1 christos dtio_reconnect_enable(dtio);
579 1.1 christos }
580 1.1 christos
581 1.1 christos /** attempt to reconnect to the output, after a timeout */
582 1.1 christos static void dtio_reconnect_enable(struct dt_io_thread* dtio)
583 1.1 christos {
584 1.1 christos struct timeval tv;
585 1.1 christos int msec;
586 1.1 christos if(dtio->want_to_exit) return;
587 1.1 christos if(dtio->reconnect_is_added)
588 1.1 christos return; /* already done */
589 1.1 christos
590 1.1 christos /* exponential backoff, store the value for next timeout */
591 1.1 christos msec = dtio->reconnect_timeout;
592 1.1 christos if(msec == 0) {
593 1.1 christos dtio->reconnect_timeout = DTIO_RECONNECT_TIMEOUT_MIN;
594 1.1 christos } else {
595 1.1 christos dtio->reconnect_timeout = msec*2;
596 1.1 christos if(dtio->reconnect_timeout > DTIO_RECONNECT_TIMEOUT_MAX)
597 1.1 christos dtio->reconnect_timeout = DTIO_RECONNECT_TIMEOUT_MAX;
598 1.1 christos }
599 1.1 christos verbose(VERB_ALGO, "dnstap io: set reconnect attempt after %d msec",
600 1.1 christos msec);
601 1.1 christos
602 1.1 christos /* setup wait timer */
603 1.1 christos memset(&tv, 0, sizeof(tv));
604 1.1 christos tv.tv_sec = msec/1000;
605 1.1 christos tv.tv_usec = (msec%1000)*1000;
606 1.1 christos if(ub_timer_add(dtio->reconnect_timer, dtio->event_base,
607 1.1 christos &dtio_reconnect_timeout_cb, dtio, &tv) != 0) {
608 1.1 christos log_err("dnstap io: could not reconnect ev timer add");
609 1.1 christos return;
610 1.1 christos }
611 1.1 christos dtio->reconnect_is_added = 1;
612 1.1 christos }
613 1.1 christos
614 1.1 christos /** remove dtio reconnect timer */
615 1.1 christos static void dtio_reconnect_del(struct dt_io_thread* dtio)
616 1.1 christos {
617 1.1 christos if(!dtio->reconnect_is_added)
618 1.1 christos return;
619 1.1 christos ub_timer_del(dtio->reconnect_timer);
620 1.1 christos dtio->reconnect_is_added = 0;
621 1.1 christos }
622 1.1 christos
623 1.1 christos /** clear the reconnect exponential backoff timer.
624 1.1 christos * We have successfully connected so we can try again with short timeouts. */
625 1.1 christos static void dtio_reconnect_clear(struct dt_io_thread* dtio)
626 1.1 christos {
627 1.1 christos dtio->reconnect_timeout = 0;
628 1.1 christos dtio_reconnect_del(dtio);
629 1.1 christos }
630 1.1 christos
631 1.1 christos /** reconnect slowly, because we already know we have to wait for a bit */
632 1.1 christos static void dtio_reconnect_slow(struct dt_io_thread* dtio, int msec)
633 1.1 christos {
634 1.1 christos dtio_reconnect_del(dtio);
635 1.1 christos dtio->reconnect_timeout = msec;
636 1.1 christos dtio_reconnect_enable(dtio);
637 1.1 christos }
638 1.1 christos
639 1.1 christos /** delete the current message in the dtio, and reset counters */
640 1.1 christos static void dtio_cur_msg_free(struct dt_io_thread* dtio)
641 1.1 christos {
642 1.1 christos free(dtio->cur_msg);
643 1.1 christos dtio->cur_msg = NULL;
644 1.1 christos dtio->cur_msg_len = 0;
645 1.1 christos dtio->cur_msg_done = 0;
646 1.1 christos dtio->cur_msg_len_done = 0;
647 1.1 christos }
648 1.1 christos
649 1.1 christos /** delete the buffer and counters used to read frame */
650 1.1 christos static void dtio_read_frame_free(struct dt_frame_read_buf* rb)
651 1.1 christos {
652 1.1 christos if(rb->buf) {
653 1.1 christos free(rb->buf);
654 1.1 christos rb->buf = NULL;
655 1.1 christos }
656 1.1 christos rb->buf_count = 0;
657 1.1 christos rb->buf_cap = 0;
658 1.1 christos rb->frame_len = 0;
659 1.1 christos rb->frame_len_done = 0;
660 1.1 christos rb->control_frame = 0;
661 1.1 christos }
662 1.1 christos
663 1.1 christos /** del the output file descriptor event for listening */
664 1.1 christos static void dtio_del_output_event(struct dt_io_thread* dtio)
665 1.1 christos {
666 1.1 christos if(!dtio->event_added)
667 1.1 christos return;
668 1.1 christos ub_event_del(dtio->event);
669 1.1 christos dtio->event_added = 0;
670 1.1 christos dtio->event_added_is_write = 0;
671 1.1 christos }
672 1.1 christos
673 1.1 christos /** close dtio socket and set it to -1 */
674 1.1 christos static void dtio_close_fd(struct dt_io_thread* dtio)
675 1.1 christos {
676 1.1 christos sock_close(dtio->fd);
677 1.1 christos dtio->fd = -1;
678 1.1 christos }
679 1.1 christos
680 1.1 christos /** close and stop the output file descriptor event */
681 1.1 christos static void dtio_close_output(struct dt_io_thread* dtio)
682 1.1 christos {
683 1.1 christos if(!dtio->event)
684 1.1 christos return;
685 1.1 christos ub_event_free(dtio->event);
686 1.1 christos dtio->event = NULL;
687 1.1 christos if(dtio->ssl) {
688 1.1 christos #ifdef HAVE_SSL
689 1.1 christos SSL_shutdown(dtio->ssl);
690 1.1 christos SSL_free(dtio->ssl);
691 1.1 christos dtio->ssl = NULL;
692 1.1 christos #endif
693 1.1 christos }
694 1.1 christos dtio_close_fd(dtio);
695 1.1 christos
696 1.1 christos /* if there is a (partial) message, discard it
697 1.1 christos * we cannot send (the remainder of) it, and a new
698 1.1 christos * connection needs to start with a control frame. */
699 1.1 christos if(dtio->cur_msg) {
700 1.1 christos dtio_cur_msg_free(dtio);
701 1.1 christos }
702 1.1 christos
703 1.1 christos dtio->ready_frame_sent = 0;
704 1.1 christos dtio->accept_frame_received = 0;
705 1.1 christos dtio_read_frame_free(&dtio->read_frame);
706 1.1 christos
707 1.1 christos dtio_reconnect_enable(dtio);
708 1.1 christos }
709 1.1 christos
710 1.1 christos /** check for pending nonblocking connect errors,
711 1.1 christos * returns 1 if it is okay. -1 on error (close it), 0 to try later */
712 1.1 christos static int dtio_check_nb_connect(struct dt_io_thread* dtio)
713 1.1 christos {
714 1.1 christos int error = 0;
715 1.1 christos socklen_t len = (socklen_t)sizeof(error);
716 1.1 christos if(!dtio->check_nb_connect)
717 1.1 christos return 1; /* everything okay */
718 1.1 christos if(getsockopt(dtio->fd, SOL_SOCKET, SO_ERROR, (void*)&error,
719 1.1 christos &len) < 0) {
720 1.1 christos #ifndef USE_WINSOCK
721 1.1 christos error = errno; /* on solaris errno is error */
722 1.1 christos #else
723 1.1 christos error = WSAGetLastError();
724 1.1 christos #endif
725 1.1 christos }
726 1.1 christos #ifndef USE_WINSOCK
727 1.1 christos #if defined(EINPROGRESS) && defined(EWOULDBLOCK)
728 1.1 christos if(error == EINPROGRESS || error == EWOULDBLOCK)
729 1.1 christos return 0; /* try again later */
730 1.1 christos #endif
731 1.1 christos #else
732 1.1 christos if(error == WSAEINPROGRESS) {
733 1.1 christos return 0; /* try again later */
734 1.1 christos } else if(error == WSAEWOULDBLOCK) {
735 1.1 christos ub_winsock_tcp_wouldblock((dtio->stop_flush_event?
736 1.1 christos dtio->stop_flush_event:dtio->event), UB_EV_WRITE);
737 1.1 christos return 0; /* try again later */
738 1.1 christos }
739 1.1 christos #endif
740 1.1 christos if(error != 0) {
741 1.1 christos char* to = dtio->socket_path;
742 1.1 christos if(!to) to = dtio->ip_str;
743 1.1 christos if(!to) to = "";
744 1.1 christos log_err("dnstap io: failed to connect to \"%s\": %s",
745 1.1 christos to, sock_strerror(error));
746 1.1 christos return -1; /* error, close it */
747 1.1 christos }
748 1.1 christos
749 1.1 christos if(dtio->ip_str)
750 1.1 christos verbose(VERB_DETAIL, "dnstap io: connected to %s",
751 1.1 christos dtio->ip_str);
752 1.1 christos else if(dtio->socket_path)
753 1.1 christos verbose(VERB_DETAIL, "dnstap io: connected to \"%s\"",
754 1.1 christos dtio->socket_path);
755 1.1 christos dtio_reconnect_clear(dtio);
756 1.1 christos dtio->check_nb_connect = 0;
757 1.1 christos return 1; /* everything okay */
758 1.1 christos }
759 1.1 christos
760 1.1 christos #ifdef HAVE_SSL
761 1.1 christos /** write to ssl output
762 1.1 christos * returns number of bytes written, 0 if nothing happened,
763 1.1 christos * try again later, or -1 if the channel is to be closed. */
764 1.1 christos static int dtio_write_ssl(struct dt_io_thread* dtio, uint8_t* buf,
765 1.1 christos size_t len)
766 1.1 christos {
767 1.1 christos int r;
768 1.1 christos ERR_clear_error();
769 1.1 christos r = SSL_write(dtio->ssl, buf, len);
770 1.1 christos if(r <= 0) {
771 1.1 christos int want = SSL_get_error(dtio->ssl, r);
772 1.1 christos if(want == SSL_ERROR_ZERO_RETURN) {
773 1.1 christos /* closed */
774 1.1 christos return -1;
775 1.1 christos } else if(want == SSL_ERROR_WANT_READ) {
776 1.1 christos /* we want a brief read event */
777 1.1 christos dtio_enable_brief_read(dtio);
778 1.1 christos return 0;
779 1.1 christos } else if(want == SSL_ERROR_WANT_WRITE) {
780 1.1 christos /* write again later */
781 1.1 christos return 0;
782 1.1 christos } else if(want == SSL_ERROR_SYSCALL) {
783 1.1 christos #ifdef EPIPE
784 1.1 christos if(errno == EPIPE && verbosity < 2)
785 1.1 christos return -1; /* silence 'broken pipe' */
786 1.1 christos #endif
787 1.1 christos #ifdef ECONNRESET
788 1.1 christos if(errno == ECONNRESET && verbosity < 2)
789 1.1 christos return -1; /* silence reset by peer */
790 1.1 christos #endif
791 1.1 christos if(errno != 0) {
792 1.1 christos log_err("dnstap io, SSL_write syscall: %s",
793 1.1 christos strerror(errno));
794 1.1 christos }
795 1.1 christos return -1;
796 1.1 christos }
797 1.1.1.3 christos log_crypto_err_io("dnstap io, could not SSL_write", want);
798 1.1 christos return -1;
799 1.1 christos }
800 1.1 christos return r;
801 1.1 christos }
802 1.1 christos #endif /* HAVE_SSL */
803 1.1 christos
804 1.1 christos /** write buffer to output.
805 1.1 christos * returns number of bytes written, 0 if nothing happened,
806 1.1 christos * try again later, or -1 if the channel is to be closed. */
807 1.1 christos static int dtio_write_buf(struct dt_io_thread* dtio, uint8_t* buf,
808 1.1 christos size_t len)
809 1.1 christos {
810 1.1 christos ssize_t ret;
811 1.1 christos if(dtio->fd == -1)
812 1.1 christos return -1;
813 1.1 christos #ifdef HAVE_SSL
814 1.1 christos if(dtio->ssl)
815 1.1 christos return dtio_write_ssl(dtio, buf, len);
816 1.1 christos #endif
817 1.1 christos ret = send(dtio->fd, (void*)buf, len, 0);
818 1.1 christos if(ret == -1) {
819 1.1 christos #ifndef USE_WINSOCK
820 1.1 christos if(errno == EINTR || errno == EAGAIN)
821 1.1 christos return 0;
822 1.1 christos #else
823 1.1 christos if(WSAGetLastError() == WSAEINPROGRESS)
824 1.1 christos return 0;
825 1.1 christos if(WSAGetLastError() == WSAEWOULDBLOCK) {
826 1.1 christos ub_winsock_tcp_wouldblock((dtio->stop_flush_event?
827 1.1 christos dtio->stop_flush_event:dtio->event),
828 1.1 christos UB_EV_WRITE);
829 1.1 christos return 0;
830 1.1 christos }
831 1.1 christos #endif
832 1.1 christos log_err("dnstap io: failed send: %s", sock_strerror(errno));
833 1.1 christos return -1;
834 1.1 christos }
835 1.1 christos return ret;
836 1.1 christos }
837 1.1 christos
838 1.1 christos #ifdef HAVE_WRITEV
839 1.1 christos /** write with writev, len and message, in one write, if possible.
840 1.1 christos * return true if message is done, false if incomplete */
841 1.1 christos static int dtio_write_with_writev(struct dt_io_thread* dtio)
842 1.1 christos {
843 1.1 christos uint32_t sendlen = htonl(dtio->cur_msg_len);
844 1.1 christos struct iovec iov[2];
845 1.1 christos ssize_t r;
846 1.1 christos iov[0].iov_base = ((uint8_t*)&sendlen)+dtio->cur_msg_len_done;
847 1.1 christos iov[0].iov_len = sizeof(sendlen)-dtio->cur_msg_len_done;
848 1.1 christos iov[1].iov_base = dtio->cur_msg;
849 1.1 christos iov[1].iov_len = dtio->cur_msg_len;
850 1.1 christos log_assert(iov[0].iov_len > 0);
851 1.1 christos r = writev(dtio->fd, iov, 2);
852 1.1 christos if(r == -1) {
853 1.1 christos #ifndef USE_WINSOCK
854 1.1 christos if(errno == EINTR || errno == EAGAIN)
855 1.1 christos return 0;
856 1.1 christos #else
857 1.1 christos if(WSAGetLastError() == WSAEINPROGRESS)
858 1.1 christos return 0;
859 1.1 christos if(WSAGetLastError() == WSAEWOULDBLOCK) {
860 1.1 christos ub_winsock_tcp_wouldblock((dtio->stop_flush_event?
861 1.1 christos dtio->stop_flush_event:dtio->event),
862 1.1 christos UB_EV_WRITE);
863 1.1 christos return 0;
864 1.1 christos }
865 1.1 christos #endif
866 1.1 christos log_err("dnstap io: failed writev: %s", sock_strerror(errno));
867 1.1 christos /* close the channel */
868 1.1 christos dtio_del_output_event(dtio);
869 1.1 christos dtio_close_output(dtio);
870 1.1 christos return 0;
871 1.1 christos }
872 1.1 christos /* written r bytes */
873 1.1 christos dtio->cur_msg_len_done += r;
874 1.1 christos if(dtio->cur_msg_len_done < 4)
875 1.1 christos return 0;
876 1.1 christos if(dtio->cur_msg_len_done > 4) {
877 1.1 christos dtio->cur_msg_done = dtio->cur_msg_len_done-4;
878 1.1 christos dtio->cur_msg_len_done = 4;
879 1.1 christos }
880 1.1 christos if(dtio->cur_msg_done < dtio->cur_msg_len)
881 1.1 christos return 0;
882 1.1 christos return 1;
883 1.1 christos }
884 1.1 christos #endif /* HAVE_WRITEV */
885 1.1 christos
886 1.1 christos /** write more of the length, preceding the data frame.
887 1.1 christos * return true if message is done, false if incomplete. */
888 1.1 christos static int dtio_write_more_of_len(struct dt_io_thread* dtio)
889 1.1 christos {
890 1.1 christos uint32_t sendlen;
891 1.1 christos int r;
892 1.1 christos if(dtio->cur_msg_len_done >= 4)
893 1.1 christos return 1;
894 1.1 christos #ifdef HAVE_WRITEV
895 1.1 christos if(!dtio->ssl) {
896 1.1 christos /* we try writev for everything.*/
897 1.1 christos return dtio_write_with_writev(dtio);
898 1.1 christos }
899 1.1 christos #endif /* HAVE_WRITEV */
900 1.1 christos sendlen = htonl(dtio->cur_msg_len);
901 1.1 christos r = dtio_write_buf(dtio,
902 1.1 christos ((uint8_t*)&sendlen)+dtio->cur_msg_len_done,
903 1.1 christos sizeof(sendlen)-dtio->cur_msg_len_done);
904 1.1 christos if(r == -1) {
905 1.1 christos /* close the channel */
906 1.1 christos dtio_del_output_event(dtio);
907 1.1 christos dtio_close_output(dtio);
908 1.1 christos return 0;
909 1.1 christos } else if(r == 0) {
910 1.1 christos /* try again later */
911 1.1 christos return 0;
912 1.1 christos }
913 1.1 christos dtio->cur_msg_len_done += r;
914 1.1 christos if(dtio->cur_msg_len_done < 4)
915 1.1 christos return 0;
916 1.1 christos return 1;
917 1.1 christos }
918 1.1 christos
919 1.1 christos /** write more of the data frame.
920 1.1 christos * return true if message is done, false if incomplete. */
921 1.1 christos static int dtio_write_more_of_data(struct dt_io_thread* dtio)
922 1.1 christos {
923 1.1 christos int r;
924 1.1 christos if(dtio->cur_msg_done >= dtio->cur_msg_len)
925 1.1 christos return 1;
926 1.1 christos r = dtio_write_buf(dtio,
927 1.1 christos ((uint8_t*)dtio->cur_msg)+dtio->cur_msg_done,
928 1.1 christos dtio->cur_msg_len - dtio->cur_msg_done);
929 1.1 christos if(r == -1) {
930 1.1 christos /* close the channel */
931 1.1 christos dtio_del_output_event(dtio);
932 1.1 christos dtio_close_output(dtio);
933 1.1 christos return 0;
934 1.1 christos } else if(r == 0) {
935 1.1 christos /* try again later */
936 1.1 christos return 0;
937 1.1 christos }
938 1.1 christos dtio->cur_msg_done += r;
939 1.1 christos if(dtio->cur_msg_done < dtio->cur_msg_len)
940 1.1 christos return 0;
941 1.1 christos return 1;
942 1.1 christos }
943 1.1 christos
944 1.1.1.2 christos /** write more of the current message. false if incomplete, true if
945 1.1 christos * the message is done */
946 1.1 christos static int dtio_write_more(struct dt_io_thread* dtio)
947 1.1 christos {
948 1.1 christos if(dtio->cur_msg_len_done < 4) {
949 1.1 christos if(!dtio_write_more_of_len(dtio))
950 1.1 christos return 0;
951 1.1 christos }
952 1.1 christos if(dtio->cur_msg_done < dtio->cur_msg_len) {
953 1.1 christos if(!dtio_write_more_of_data(dtio))
954 1.1 christos return 0;
955 1.1 christos }
956 1.1 christos return 1;
957 1.1 christos }
958 1.1 christos
959 1.1 christos /** Receive bytes from dtio->fd, store in buffer. Returns 0: closed,
960 1.1 christos * -1: continue, >0: number of bytes read into buffer */
961 1.1 christos static ssize_t receive_bytes(struct dt_io_thread* dtio, void* buf, size_t len) {
962 1.1 christos ssize_t r;
963 1.1.1.3 christos r = recv(dtio->fd, (void*)buf, len, MSG_DONTWAIT);
964 1.1 christos if(r == -1) {
965 1.1 christos char* to = dtio->socket_path;
966 1.1 christos if(!to) to = dtio->ip_str;
967 1.1 christos if(!to) to = "";
968 1.1 christos #ifndef USE_WINSOCK
969 1.1 christos if(errno == EINTR || errno == EAGAIN)
970 1.1 christos return -1; /* try later */
971 1.1 christos #else
972 1.1 christos if(WSAGetLastError() == WSAEINPROGRESS) {
973 1.1 christos return -1; /* try later */
974 1.1 christos } else if(WSAGetLastError() == WSAEWOULDBLOCK) {
975 1.1 christos ub_winsock_tcp_wouldblock(
976 1.1 christos (dtio->stop_flush_event?
977 1.1 christos dtio->stop_flush_event:dtio->event),
978 1.1 christos UB_EV_READ);
979 1.1 christos return -1; /* try later */
980 1.1 christos }
981 1.1 christos #endif
982 1.1 christos if(dtio->reconnect_timeout > DTIO_RECONNECT_TIMEOUT_MIN &&
983 1.1 christos verbosity < 4)
984 1.1 christos return 0; /* no log retries on low verbosity */
985 1.1 christos log_err("dnstap io: output closed, recv %s: %s", to,
986 1.1 christos strerror(errno));
987 1.1 christos /* and close below */
988 1.1 christos return 0;
989 1.1 christos }
990 1.1 christos if(r == 0) {
991 1.1 christos if(dtio->reconnect_timeout > DTIO_RECONNECT_TIMEOUT_MIN &&
992 1.1 christos verbosity < 4)
993 1.1 christos return 0; /* no log retries on low verbosity */
994 1.1 christos verbose(VERB_DETAIL, "dnstap io: output closed by the other side");
995 1.1 christos /* and close below */
996 1.1 christos return 0;
997 1.1 christos }
998 1.1 christos /* something was received */
999 1.1 christos return r;
1000 1.1 christos }
1001 1.1 christos
1002 1.1 christos #ifdef HAVE_SSL
1003 1.1 christos /** Receive bytes over TLS from dtio->fd, store in buffer. Returns 0: closed,
1004 1.1 christos * -1: continue, >0: number of bytes read into buffer */
1005 1.1 christos static int ssl_read_bytes(struct dt_io_thread* dtio, void* buf, size_t len)
1006 1.1 christos {
1007 1.1 christos int r;
1008 1.1 christos ERR_clear_error();
1009 1.1 christos r = SSL_read(dtio->ssl, buf, len);
1010 1.1 christos if(r <= 0) {
1011 1.1 christos int want = SSL_get_error(dtio->ssl, r);
1012 1.1 christos if(want == SSL_ERROR_ZERO_RETURN) {
1013 1.1 christos if(dtio->reconnect_timeout > DTIO_RECONNECT_TIMEOUT_MIN &&
1014 1.1 christos verbosity < 4)
1015 1.1 christos return 0; /* no log retries on low verbosity */
1016 1.1 christos verbose(VERB_DETAIL, "dnstap io: output closed by the "
1017 1.1 christos "other side");
1018 1.1 christos return 0;
1019 1.1 christos } else if(want == SSL_ERROR_WANT_READ) {
1020 1.1 christos /* continue later */
1021 1.1 christos return -1;
1022 1.1 christos } else if(want == SSL_ERROR_WANT_WRITE) {
1023 1.1 christos (void)dtio_enable_brief_write(dtio);
1024 1.1 christos return -1;
1025 1.1 christos } else if(want == SSL_ERROR_SYSCALL) {
1026 1.1 christos #ifdef ECONNRESET
1027 1.1 christos if(dtio->reconnect_timeout > DTIO_RECONNECT_TIMEOUT_MIN &&
1028 1.1 christos errno == ECONNRESET && verbosity < 4)
1029 1.1 christos return 0; /* silence reset by peer */
1030 1.1 christos #endif
1031 1.1 christos if(errno != 0)
1032 1.1 christos log_err("SSL_read syscall: %s",
1033 1.1 christos strerror(errno));
1034 1.1 christos verbose(VERB_DETAIL, "dnstap io: output closed by the "
1035 1.1 christos "other side");
1036 1.1 christos return 0;
1037 1.1 christos }
1038 1.1.1.3 christos log_crypto_err_io("could not SSL_read", want);
1039 1.1 christos verbose(VERB_DETAIL, "dnstap io: output closed by the "
1040 1.1 christos "other side");
1041 1.1 christos return 0;
1042 1.1 christos }
1043 1.1 christos return r;
1044 1.1 christos }
1045 1.1 christos #endif /* HAVE_SSL */
1046 1.1 christos
1047 1.1 christos /** check if the output fd has been closed,
1048 1.1 christos * it returns false if the stream is closed. */
1049 1.1 christos static int dtio_check_close(struct dt_io_thread* dtio)
1050 1.1 christos {
1051 1.1 christos /* we don't want to read any packets, but if there are we can
1052 1.1 christos * discard the input (ignore it). Ignore of unknown (control)
1053 1.1 christos * packets is okay for the framestream protocol. And also, the
1054 1.1 christos * read call can return that the stream has been closed by the
1055 1.1 christos * other side. */
1056 1.1 christos uint8_t buf[1024];
1057 1.1 christos int r = -1;
1058 1.1 christos
1059 1.1 christos
1060 1.1 christos if(dtio->fd == -1) return 0;
1061 1.1 christos
1062 1.1 christos while(r != 0) {
1063 1.1 christos /* not interested in buffer content, overwrite */
1064 1.1 christos r = receive_bytes(dtio, (void*)buf, sizeof(buf));
1065 1.1 christos if(r == -1)
1066 1.1 christos return 1;
1067 1.1 christos }
1068 1.1 christos /* the other end has been closed */
1069 1.1 christos /* close the channel */
1070 1.1 christos dtio_del_output_event(dtio);
1071 1.1 christos dtio_close_output(dtio);
1072 1.1 christos return 0;
1073 1.1 christos }
1074 1.1 christos
1075 1.1 christos /** Read accept frame. Returns -1: continue reading, 0: closed,
1076 1.1 christos * 1: valid accept received. */
1077 1.1 christos static int dtio_read_accept_frame(struct dt_io_thread* dtio)
1078 1.1 christos {
1079 1.1 christos int r;
1080 1.1 christos size_t read_frame_done;
1081 1.1 christos while(dtio->read_frame.frame_len_done < 4) {
1082 1.1 christos #ifdef HAVE_SSL
1083 1.1 christos if(dtio->ssl) {
1084 1.1 christos r = ssl_read_bytes(dtio,
1085 1.1 christos (uint8_t*)&dtio->read_frame.frame_len+
1086 1.1 christos dtio->read_frame.frame_len_done,
1087 1.1 christos 4-dtio->read_frame.frame_len_done);
1088 1.1 christos } else {
1089 1.1 christos #endif
1090 1.1 christos r = receive_bytes(dtio,
1091 1.1 christos (uint8_t*)&dtio->read_frame.frame_len+
1092 1.1 christos dtio->read_frame.frame_len_done,
1093 1.1 christos 4-dtio->read_frame.frame_len_done);
1094 1.1 christos #ifdef HAVE_SSL
1095 1.1 christos }
1096 1.1 christos #endif
1097 1.1 christos if(r == -1)
1098 1.1 christos return -1; /* continue reading */
1099 1.1 christos if(r == 0) {
1100 1.1 christos /* connection closed */
1101 1.1 christos goto close_connection;
1102 1.1 christos }
1103 1.1 christos dtio->read_frame.frame_len_done += r;
1104 1.1 christos if(dtio->read_frame.frame_len_done < 4)
1105 1.1 christos return -1; /* continue reading */
1106 1.1 christos
1107 1.1 christos if(dtio->read_frame.frame_len == 0) {
1108 1.1 christos dtio->read_frame.frame_len_done = 0;
1109 1.1 christos dtio->read_frame.control_frame = 1;
1110 1.1 christos continue;
1111 1.1 christos }
1112 1.1 christos dtio->read_frame.frame_len = ntohl(dtio->read_frame.frame_len);
1113 1.1 christos if(dtio->read_frame.frame_len > DTIO_RECV_FRAME_MAX_LEN) {
1114 1.1 christos verbose(VERB_OPS, "dnstap: received frame exceeds max "
1115 1.1 christos "length of %d bytes, closing connection",
1116 1.1 christos DTIO_RECV_FRAME_MAX_LEN);
1117 1.1 christos goto close_connection;
1118 1.1 christos }
1119 1.1 christos dtio->read_frame.buf = calloc(1, dtio->read_frame.frame_len);
1120 1.1 christos dtio->read_frame.buf_cap = dtio->read_frame.frame_len;
1121 1.1 christos if(!dtio->read_frame.buf) {
1122 1.1 christos log_err("dnstap io: out of memory (creating read "
1123 1.1 christos "buffer)");
1124 1.1 christos goto close_connection;
1125 1.1 christos }
1126 1.1 christos }
1127 1.1 christos if(dtio->read_frame.buf_count < dtio->read_frame.frame_len) {
1128 1.1 christos #ifdef HAVE_SSL
1129 1.1 christos if(dtio->ssl) {
1130 1.1 christos r = ssl_read_bytes(dtio, dtio->read_frame.buf+
1131 1.1 christos dtio->read_frame.buf_count,
1132 1.1 christos dtio->read_frame.buf_cap-
1133 1.1 christos dtio->read_frame.buf_count);
1134 1.1 christos } else {
1135 1.1 christos #endif
1136 1.1 christos r = receive_bytes(dtio, dtio->read_frame.buf+
1137 1.1 christos dtio->read_frame.buf_count,
1138 1.1 christos dtio->read_frame.buf_cap-
1139 1.1 christos dtio->read_frame.buf_count);
1140 1.1 christos #ifdef HAVE_SSL
1141 1.1 christos }
1142 1.1 christos #endif
1143 1.1 christos if(r == -1)
1144 1.1 christos return -1; /* continue reading */
1145 1.1 christos if(r == 0) {
1146 1.1 christos /* connection closed */
1147 1.1 christos goto close_connection;
1148 1.1 christos }
1149 1.1 christos dtio->read_frame.buf_count += r;
1150 1.1 christos if(dtio->read_frame.buf_count < dtio->read_frame.frame_len)
1151 1.1 christos return -1; /* continue reading */
1152 1.1 christos }
1153 1.1 christos
1154 1.1 christos /* Complete frame received, check if this is a valid ACCEPT control
1155 1.1 christos * frame. */
1156 1.1 christos if(dtio->read_frame.frame_len < 4) {
1157 1.1 christos verbose(VERB_OPS, "dnstap: invalid data received");
1158 1.1 christos goto close_connection;
1159 1.1 christos }
1160 1.1 christos if(sldns_read_uint32(dtio->read_frame.buf) !=
1161 1.1 christos FSTRM_CONTROL_FRAME_ACCEPT) {
1162 1.1 christos verbose(VERB_ALGO, "dnstap: invalid control type received, "
1163 1.1 christos "ignored");
1164 1.1 christos dtio->ready_frame_sent = 0;
1165 1.1 christos dtio->accept_frame_received = 0;
1166 1.1 christos dtio_read_frame_free(&dtio->read_frame);
1167 1.1 christos return -1;
1168 1.1 christos }
1169 1.1 christos read_frame_done = 4; /* control frame type */
1170 1.1 christos
1171 1.1 christos /* Iterate over control fields, ignore unknown types.
1172 1.1 christos * Need to be able to read at least 8 bytes (control field type +
1173 1.1 christos * length). */
1174 1.1 christos while(read_frame_done+8 < dtio->read_frame.frame_len) {
1175 1.1 christos uint32_t type = sldns_read_uint32(dtio->read_frame.buf +
1176 1.1 christos read_frame_done);
1177 1.1 christos uint32_t len = sldns_read_uint32(dtio->read_frame.buf +
1178 1.1 christos read_frame_done + 4);
1179 1.1 christos if(type == FSTRM_CONTROL_FIELD_TYPE_CONTENT_TYPE) {
1180 1.1 christos if(len == strlen(DNSTAP_CONTENT_TYPE) &&
1181 1.1 christos read_frame_done+8+len <=
1182 1.1 christos dtio->read_frame.frame_len &&
1183 1.1 christos memcmp(dtio->read_frame.buf + read_frame_done +
1184 1.1 christos + 8, DNSTAP_CONTENT_TYPE, len) == 0) {
1185 1.1 christos if(!dtio_control_start_send(dtio)) {
1186 1.1 christos verbose(VERB_OPS, "dnstap io: out of "
1187 1.1 christos "memory while sending START frame");
1188 1.1 christos goto close_connection;
1189 1.1 christos }
1190 1.1 christos dtio->accept_frame_received = 1;
1191 1.1 christos if(!dtio_add_output_event_write(dtio))
1192 1.1 christos goto close_connection;
1193 1.1 christos return 1;
1194 1.1 christos } else {
1195 1.1.1.2 christos /* unknown content type */
1196 1.1 christos verbose(VERB_ALGO, "dnstap: ACCEPT frame "
1197 1.1 christos "contains unknown content type, "
1198 1.1 christos "closing connection");
1199 1.1 christos goto close_connection;
1200 1.1 christos }
1201 1.1 christos }
1202 1.1 christos /* unknown option, try next */
1203 1.1 christos read_frame_done += 8+len;
1204 1.1 christos }
1205 1.1 christos
1206 1.1 christos
1207 1.1 christos close_connection:
1208 1.1 christos dtio_del_output_event(dtio);
1209 1.1 christos dtio_reconnect_slow(dtio, DTIO_RECONNECT_TIMEOUT_SLOW);
1210 1.1 christos dtio_close_output(dtio);
1211 1.1 christos return 0;
1212 1.1 christos }
1213 1.1 christos
1214 1.1 christos /** add the output file descriptor event for listening, read only */
1215 1.1 christos static int dtio_add_output_event_read(struct dt_io_thread* dtio)
1216 1.1 christos {
1217 1.1 christos if(!dtio->event)
1218 1.1 christos return 0;
1219 1.1 christos if(dtio->event_added && !dtio->event_added_is_write)
1220 1.1 christos return 1;
1221 1.1 christos /* we have to (re-)register the event */
1222 1.1 christos if(dtio->event_added)
1223 1.1 christos ub_event_del(dtio->event);
1224 1.1 christos ub_event_del_bits(dtio->event, UB_EV_WRITE);
1225 1.1 christos if(ub_event_add(dtio->event, NULL) != 0) {
1226 1.1 christos log_err("dnstap io: out of memory (adding event)");
1227 1.1 christos dtio->event_added = 0;
1228 1.1 christos dtio->event_added_is_write = 0;
1229 1.1 christos /* close output and start reattempts to open it */
1230 1.1 christos dtio_close_output(dtio);
1231 1.1 christos return 0;
1232 1.1 christos }
1233 1.1 christos dtio->event_added = 1;
1234 1.1 christos dtio->event_added_is_write = 0;
1235 1.1 christos return 1;
1236 1.1 christos }
1237 1.1 christos
1238 1.1 christos /** add the output file descriptor event for listening, read and write */
1239 1.1 christos static int dtio_add_output_event_write(struct dt_io_thread* dtio)
1240 1.1 christos {
1241 1.1 christos if(!dtio->event)
1242 1.1 christos return 0;
1243 1.1 christos if(dtio->event_added && dtio->event_added_is_write)
1244 1.1 christos return 1;
1245 1.1 christos /* we have to (re-)register the event */
1246 1.1 christos if(dtio->event_added)
1247 1.1 christos ub_event_del(dtio->event);
1248 1.1 christos ub_event_add_bits(dtio->event, UB_EV_WRITE);
1249 1.1 christos if(ub_event_add(dtio->event, NULL) != 0) {
1250 1.1 christos log_err("dnstap io: out of memory (adding event)");
1251 1.1 christos dtio->event_added = 0;
1252 1.1 christos dtio->event_added_is_write = 0;
1253 1.1 christos /* close output and start reattempts to open it */
1254 1.1 christos dtio_close_output(dtio);
1255 1.1 christos return 0;
1256 1.1 christos }
1257 1.1 christos dtio->event_added = 1;
1258 1.1 christos dtio->event_added_is_write = 1;
1259 1.1 christos return 1;
1260 1.1 christos }
1261 1.1 christos
1262 1.1 christos /** put the dtio thread to sleep */
1263 1.1 christos static void dtio_sleep(struct dt_io_thread* dtio)
1264 1.1 christos {
1265 1.1 christos /* unregister the event polling for write, because there is
1266 1.1 christos * nothing to be written */
1267 1.1 christos (void)dtio_add_output_event_read(dtio);
1268 1.1.1.4 christos
1269 1.1.1.4 christos /* Set wakeuptimer enabled off; so that the next worker thread that
1270 1.1.1.4 christos * wants to log starts a timer if needed, since the writer thread
1271 1.1.1.4 christos * has gone to sleep. */
1272 1.1.1.4 christos lock_basic_lock(&dtio->wakeup_timer_lock);
1273 1.1.1.4 christos dtio->wakeup_timer_enabled = 0;
1274 1.1.1.4 christos lock_basic_unlock(&dtio->wakeup_timer_lock);
1275 1.1 christos }
1276 1.1 christos
1277 1.1 christos #ifdef HAVE_SSL
1278 1.1 christos /** enable the brief read condition */
1279 1.1 christos static int dtio_enable_brief_read(struct dt_io_thread* dtio)
1280 1.1 christos {
1281 1.1 christos dtio->ssl_brief_read = 1;
1282 1.1 christos if(dtio->stop_flush_event) {
1283 1.1 christos ub_event_del(dtio->stop_flush_event);
1284 1.1 christos ub_event_del_bits(dtio->stop_flush_event, UB_EV_WRITE);
1285 1.1 christos if(ub_event_add(dtio->stop_flush_event, NULL) != 0) {
1286 1.1 christos log_err("dnstap io, stop flush, could not ub_event_add");
1287 1.1 christos return 0;
1288 1.1 christos }
1289 1.1 christos return 1;
1290 1.1 christos }
1291 1.1 christos return dtio_add_output_event_read(dtio);
1292 1.1 christos }
1293 1.1 christos #endif /* HAVE_SSL */
1294 1.1 christos
1295 1.1 christos #ifdef HAVE_SSL
1296 1.1 christos /** disable the brief read condition */
1297 1.1 christos static int dtio_disable_brief_read(struct dt_io_thread* dtio)
1298 1.1 christos {
1299 1.1 christos dtio->ssl_brief_read = 0;
1300 1.1 christos if(dtio->stop_flush_event) {
1301 1.1 christos ub_event_del(dtio->stop_flush_event);
1302 1.1 christos ub_event_add_bits(dtio->stop_flush_event, UB_EV_WRITE);
1303 1.1 christos if(ub_event_add(dtio->stop_flush_event, NULL) != 0) {
1304 1.1 christos log_err("dnstap io, stop flush, could not ub_event_add");
1305 1.1 christos return 0;
1306 1.1 christos }
1307 1.1 christos return 1;
1308 1.1 christos }
1309 1.1 christos return dtio_add_output_event_write(dtio);
1310 1.1 christos }
1311 1.1 christos #endif /* HAVE_SSL */
1312 1.1 christos
1313 1.1 christos #ifdef HAVE_SSL
1314 1.1 christos /** enable the brief write condition */
1315 1.1 christos static int dtio_enable_brief_write(struct dt_io_thread* dtio)
1316 1.1 christos {
1317 1.1 christos dtio->ssl_brief_write = 1;
1318 1.1 christos return dtio_add_output_event_write(dtio);
1319 1.1 christos }
1320 1.1 christos #endif /* HAVE_SSL */
1321 1.1 christos
1322 1.1 christos #ifdef HAVE_SSL
1323 1.1 christos /** disable the brief write condition */
1324 1.1 christos static int dtio_disable_brief_write(struct dt_io_thread* dtio)
1325 1.1 christos {
1326 1.1 christos dtio->ssl_brief_write = 0;
1327 1.1 christos return dtio_add_output_event_read(dtio);
1328 1.1 christos }
1329 1.1 christos #endif /* HAVE_SSL */
1330 1.1 christos
1331 1.1 christos #ifdef HAVE_SSL
1332 1.1 christos /** check peer verification after ssl handshake connection, false if closed*/
1333 1.1 christos static int dtio_ssl_check_peer(struct dt_io_thread* dtio)
1334 1.1 christos {
1335 1.1 christos if((SSL_get_verify_mode(dtio->ssl)&SSL_VERIFY_PEER)) {
1336 1.1 christos /* verification */
1337 1.1 christos if(SSL_get_verify_result(dtio->ssl) == X509_V_OK) {
1338 1.1.1.4 christos #ifdef HAVE_SSL_GET1_PEER_CERTIFICATE
1339 1.1.1.4 christos X509* x = SSL_get1_peer_certificate(dtio->ssl);
1340 1.1.1.4 christos #else
1341 1.1 christos X509* x = SSL_get_peer_certificate(dtio->ssl);
1342 1.1.1.4 christos #endif
1343 1.1 christos if(!x) {
1344 1.1 christos verbose(VERB_ALGO, "dnstap io, %s, SSL "
1345 1.1 christos "connection failed no certificate",
1346 1.1 christos dtio->ip_str);
1347 1.1 christos return 0;
1348 1.1 christos }
1349 1.1 christos log_cert(VERB_ALGO, "dnstap io, peer certificate",
1350 1.1 christos x);
1351 1.1 christos #ifdef HAVE_SSL_GET0_PEERNAME
1352 1.1 christos if(SSL_get0_peername(dtio->ssl)) {
1353 1.1 christos verbose(VERB_ALGO, "dnstap io, %s, SSL "
1354 1.1 christos "connection to %s authenticated",
1355 1.1 christos dtio->ip_str,
1356 1.1 christos SSL_get0_peername(dtio->ssl));
1357 1.1 christos } else {
1358 1.1 christos #endif
1359 1.1 christos verbose(VERB_ALGO, "dnstap io, %s, SSL "
1360 1.1 christos "connection authenticated",
1361 1.1 christos dtio->ip_str);
1362 1.1 christos #ifdef HAVE_SSL_GET0_PEERNAME
1363 1.1 christos }
1364 1.1 christos #endif
1365 1.1 christos X509_free(x);
1366 1.1 christos } else {
1367 1.1.1.4 christos #ifdef HAVE_SSL_GET1_PEER_CERTIFICATE
1368 1.1.1.4 christos X509* x = SSL_get1_peer_certificate(dtio->ssl);
1369 1.1.1.4 christos #else
1370 1.1 christos X509* x = SSL_get_peer_certificate(dtio->ssl);
1371 1.1.1.4 christos #endif
1372 1.1 christos if(x) {
1373 1.1 christos log_cert(VERB_ALGO, "dnstap io, peer "
1374 1.1 christos "certificate", x);
1375 1.1 christos X509_free(x);
1376 1.1 christos }
1377 1.1 christos verbose(VERB_ALGO, "dnstap io, %s, SSL connection "
1378 1.1 christos "failed: failed to authenticate",
1379 1.1 christos dtio->ip_str);
1380 1.1 christos return 0;
1381 1.1 christos }
1382 1.1 christos } else {
1383 1.1 christos /* unauthenticated, the verify peer flag was not set
1384 1.1 christos * in ssl when the ssl object was created from ssl_ctx */
1385 1.1 christos verbose(VERB_ALGO, "dnstap io, %s, SSL connection",
1386 1.1 christos dtio->ip_str);
1387 1.1 christos }
1388 1.1 christos return 1;
1389 1.1 christos }
1390 1.1 christos #endif /* HAVE_SSL */
1391 1.1 christos
1392 1.1 christos #ifdef HAVE_SSL
1393 1.1 christos /** perform ssl handshake, returns 1 if okay, 0 to stop */
1394 1.1 christos static int dtio_ssl_handshake(struct dt_io_thread* dtio,
1395 1.1 christos struct stop_flush_info* info)
1396 1.1 christos {
1397 1.1 christos int r;
1398 1.1 christos if(dtio->ssl_brief_read) {
1399 1.1 christos /* assume the brief read condition is satisfied,
1400 1.1 christos * if we need more or again, we can set it again */
1401 1.1 christos if(!dtio_disable_brief_read(dtio)) {
1402 1.1 christos if(info) dtio_stop_flush_exit(info);
1403 1.1 christos return 0;
1404 1.1 christos }
1405 1.1 christos }
1406 1.1 christos if(dtio->ssl_handshake_done)
1407 1.1 christos return 1;
1408 1.1 christos
1409 1.1 christos ERR_clear_error();
1410 1.1 christos r = SSL_do_handshake(dtio->ssl);
1411 1.1 christos if(r != 1) {
1412 1.1 christos int want = SSL_get_error(dtio->ssl, r);
1413 1.1 christos if(want == SSL_ERROR_WANT_READ) {
1414 1.1 christos /* we want to read on the connection */
1415 1.1 christos if(!dtio_enable_brief_read(dtio)) {
1416 1.1 christos if(info) dtio_stop_flush_exit(info);
1417 1.1 christos return 0;
1418 1.1 christos }
1419 1.1 christos return 0;
1420 1.1 christos } else if(want == SSL_ERROR_WANT_WRITE) {
1421 1.1 christos /* we want to write on the connection */
1422 1.1 christos return 0;
1423 1.1 christos } else if(r == 0) {
1424 1.1 christos /* closed */
1425 1.1 christos if(info) dtio_stop_flush_exit(info);
1426 1.1 christos dtio_del_output_event(dtio);
1427 1.1 christos dtio_reconnect_slow(dtio, DTIO_RECONNECT_TIMEOUT_SLOW);
1428 1.1 christos dtio_close_output(dtio);
1429 1.1 christos return 0;
1430 1.1 christos } else if(want == SSL_ERROR_SYSCALL) {
1431 1.1 christos /* SYSCALL and errno==0 means closed uncleanly */
1432 1.1 christos int silent = 0;
1433 1.1 christos #ifdef EPIPE
1434 1.1 christos if(errno == EPIPE && verbosity < 2)
1435 1.1 christos silent = 1; /* silence 'broken pipe' */
1436 1.1 christos #endif
1437 1.1 christos #ifdef ECONNRESET
1438 1.1 christos if(errno == ECONNRESET && verbosity < 2)
1439 1.1 christos silent = 1; /* silence reset by peer */
1440 1.1 christos #endif
1441 1.1 christos if(errno == 0)
1442 1.1 christos silent = 1;
1443 1.1 christos if(!silent)
1444 1.1 christos log_err("dnstap io, SSL_handshake syscall: %s",
1445 1.1 christos strerror(errno));
1446 1.1 christos /* closed */
1447 1.1 christos if(info) dtio_stop_flush_exit(info);
1448 1.1 christos dtio_del_output_event(dtio);
1449 1.1 christos dtio_reconnect_slow(dtio, DTIO_RECONNECT_TIMEOUT_SLOW);
1450 1.1 christos dtio_close_output(dtio);
1451 1.1 christos return 0;
1452 1.1 christos } else {
1453 1.1 christos unsigned long err = ERR_get_error();
1454 1.1 christos if(!squelch_err_ssl_handshake(err)) {
1455 1.1.1.3 christos log_crypto_err_io_code("dnstap io, ssl handshake failed",
1456 1.1.1.3 christos want, err);
1457 1.1 christos verbose(VERB_OPS, "dnstap io, ssl handshake failed "
1458 1.1 christos "from %s", dtio->ip_str);
1459 1.1 christos }
1460 1.1 christos /* closed */
1461 1.1 christos if(info) dtio_stop_flush_exit(info);
1462 1.1 christos dtio_del_output_event(dtio);
1463 1.1 christos dtio_reconnect_slow(dtio, DTIO_RECONNECT_TIMEOUT_SLOW);
1464 1.1 christos dtio_close_output(dtio);
1465 1.1 christos return 0;
1466 1.1 christos }
1467 1.1 christos
1468 1.1 christos }
1469 1.1 christos /* check peer verification */
1470 1.1 christos dtio->ssl_handshake_done = 1;
1471 1.1 christos
1472 1.1 christos if(!dtio_ssl_check_peer(dtio)) {
1473 1.1 christos /* closed */
1474 1.1 christos if(info) dtio_stop_flush_exit(info);
1475 1.1 christos dtio_del_output_event(dtio);
1476 1.1 christos dtio_reconnect_slow(dtio, DTIO_RECONNECT_TIMEOUT_SLOW);
1477 1.1 christos dtio_close_output(dtio);
1478 1.1 christos return 0;
1479 1.1 christos }
1480 1.1 christos return 1;
1481 1.1 christos }
1482 1.1 christos #endif /* HAVE_SSL */
1483 1.1 christos
1484 1.1 christos /** callback for the dnstap events, to write to the output */
1485 1.1 christos void dtio_output_cb(int ATTR_UNUSED(fd), short bits, void* arg)
1486 1.1 christos {
1487 1.1 christos struct dt_io_thread* dtio = (struct dt_io_thread*)arg;
1488 1.1 christos int i;
1489 1.1 christos
1490 1.1 christos if(dtio->check_nb_connect) {
1491 1.1 christos int connect_err = dtio_check_nb_connect(dtio);
1492 1.1 christos if(connect_err == -1) {
1493 1.1 christos /* close the channel */
1494 1.1 christos dtio_del_output_event(dtio);
1495 1.1 christos dtio_close_output(dtio);
1496 1.1 christos return;
1497 1.1 christos } else if(connect_err == 0) {
1498 1.1 christos /* try again later */
1499 1.1 christos return;
1500 1.1 christos }
1501 1.1 christos /* nonblocking connect check passed, continue */
1502 1.1 christos }
1503 1.1 christos
1504 1.1 christos #ifdef HAVE_SSL
1505 1.1 christos if(dtio->ssl &&
1506 1.1 christos (!dtio->ssl_handshake_done || dtio->ssl_brief_read)) {
1507 1.1 christos if(!dtio_ssl_handshake(dtio, NULL))
1508 1.1 christos return;
1509 1.1 christos }
1510 1.1 christos #endif
1511 1.1 christos
1512 1.1.1.5 christos if((bits&UB_EV_READ) || dtio->ssl_brief_write) {
1513 1.1.1.4 christos #ifdef HAVE_SSL
1514 1.1 christos if(dtio->ssl_brief_write)
1515 1.1 christos (void)dtio_disable_brief_write(dtio);
1516 1.1.1.4 christos #endif
1517 1.1 christos if(dtio->ready_frame_sent && !dtio->accept_frame_received) {
1518 1.1 christos if(dtio_read_accept_frame(dtio) <= 0)
1519 1.1 christos return;
1520 1.1 christos } else if(!dtio_check_close(dtio))
1521 1.1 christos return;
1522 1.1 christos }
1523 1.1 christos
1524 1.1 christos /* loop to process a number of messages. This improves throughput,
1525 1.1 christos * because selecting on write-event if not needed for busy messages
1526 1.1 christos * (dnstap log) generation and if they need to all be written back.
1527 1.1 christos * The write event is usually not blocked up. But not forever,
1528 1.1 christos * because the event loop needs to stay responsive for other events.
1529 1.1 christos * If there are no (more) messages, or if the output buffers get
1530 1.1 christos * full, it returns out of the loop. */
1531 1.1 christos for(i=0; i<DTIO_MESSAGES_PER_CALLBACK; i++) {
1532 1.1 christos /* see if there are messages that need writing */
1533 1.1 christos if(!dtio->cur_msg) {
1534 1.1 christos if(!dtio_find_msg(dtio)) {
1535 1.1 christos if(i == 0) {
1536 1.1 christos /* no messages on the first iteration,
1537 1.1 christos * the queues are all empty */
1538 1.1 christos dtio_sleep(dtio);
1539 1.1.1.4 christos /* After putting to sleep, see if
1540 1.1.1.4 christos * a message is in a message queue,
1541 1.1.1.4 christos * if so, resume service. Stops a
1542 1.1.1.4 christos * race condition where a thread could
1543 1.1.1.4 christos * have one message but the dtio
1544 1.1.1.4 christos * also just went to sleep. With the
1545 1.1.1.4 christos * message queued between the
1546 1.1.1.4 christos * dtio_find_msg and dtio_sleep
1547 1.1.1.4 christos * calls. */
1548 1.1.1.4 christos if(dtio_find_msg(dtio)) {
1549 1.1.1.4 christos if(!dtio_add_output_event_write(dtio))
1550 1.1.1.4 christos return;
1551 1.1.1.4 christos }
1552 1.1 christos }
1553 1.1.1.4 christos if(!dtio->cur_msg)
1554 1.1.1.4 christos return; /* nothing to do */
1555 1.1 christos }
1556 1.1 christos }
1557 1.1 christos
1558 1.1 christos /* write it */
1559 1.1 christos if(dtio->cur_msg_done < dtio->cur_msg_len) {
1560 1.1 christos if(!dtio_write_more(dtio))
1561 1.1 christos return;
1562 1.1 christos }
1563 1.1 christos
1564 1.1 christos /* done with the current message */
1565 1.1 christos dtio_cur_msg_free(dtio);
1566 1.1 christos
1567 1.1 christos /* If this is a bidirectional stream the first message will be
1568 1.1 christos * the READY control frame. We can only continue writing after
1569 1.1 christos * receiving an ACCEPT control frame. */
1570 1.1 christos if(dtio->is_bidirectional && !dtio->ready_frame_sent) {
1571 1.1 christos dtio->ready_frame_sent = 1;
1572 1.1 christos (void)dtio_add_output_event_read(dtio);
1573 1.1 christos break;
1574 1.1 christos }
1575 1.1 christos }
1576 1.1 christos }
1577 1.1 christos
1578 1.1 christos /** callback for the dnstap commandpipe, to stop the dnstap IO */
1579 1.1 christos void dtio_cmd_cb(int fd, short ATTR_UNUSED(bits), void* arg)
1580 1.1 christos {
1581 1.1 christos struct dt_io_thread* dtio = (struct dt_io_thread*)arg;
1582 1.1 christos uint8_t cmd;
1583 1.1 christos ssize_t r;
1584 1.1 christos if(dtio->want_to_exit)
1585 1.1 christos return;
1586 1.1 christos r = read(fd, &cmd, sizeof(cmd));
1587 1.1 christos if(r == -1) {
1588 1.1 christos #ifndef USE_WINSOCK
1589 1.1 christos if(errno == EINTR || errno == EAGAIN)
1590 1.1 christos return; /* ignore this */
1591 1.1 christos #else
1592 1.1 christos if(WSAGetLastError() == WSAEINPROGRESS)
1593 1.1 christos return;
1594 1.1 christos if(WSAGetLastError() == WSAEWOULDBLOCK)
1595 1.1 christos return;
1596 1.1 christos #endif
1597 1.1 christos log_err("dnstap io: failed to read: %s", sock_strerror(errno));
1598 1.1 christos /* and then fall through to quit the thread */
1599 1.1 christos } else if(r == 0) {
1600 1.1 christos verbose(VERB_ALGO, "dnstap io: cmd channel closed");
1601 1.1 christos } else if(r == 1 && cmd == DTIO_COMMAND_STOP) {
1602 1.1 christos verbose(VERB_ALGO, "dnstap io: cmd channel cmd quit");
1603 1.1 christos } else if(r == 1 && cmd == DTIO_COMMAND_WAKEUP) {
1604 1.1 christos verbose(VERB_ALGO, "dnstap io: cmd channel cmd wakeup");
1605 1.1 christos
1606 1.1 christos if(dtio->is_bidirectional && !dtio->accept_frame_received) {
1607 1.1 christos verbose(VERB_ALGO, "dnstap io: cmd wakeup ignored, "
1608 1.1 christos "waiting for ACCEPT control frame");
1609 1.1 christos return;
1610 1.1 christos }
1611 1.1 christos
1612 1.1 christos /* reregister event */
1613 1.1 christos if(!dtio_add_output_event_write(dtio))
1614 1.1 christos return;
1615 1.1 christos return;
1616 1.1 christos } else if(r == 1) {
1617 1.1 christos verbose(VERB_ALGO, "dnstap io: cmd channel unknown command");
1618 1.1 christos }
1619 1.1 christos dtio->want_to_exit = 1;
1620 1.1 christos if(ub_event_base_loopexit((struct ub_event_base*)dtio->event_base)
1621 1.1 christos != 0) {
1622 1.1 christos log_err("dnstap io: could not loopexit");
1623 1.1 christos }
1624 1.1 christos }
1625 1.1 christos
1626 1.1 christos #ifndef THREADS_DISABLED
1627 1.1 christos /** setup the event base for the dnstap io thread */
1628 1.1 christos static void dtio_setup_base(struct dt_io_thread* dtio, time_t* secs,
1629 1.1 christos struct timeval* now)
1630 1.1 christos {
1631 1.1 christos memset(now, 0, sizeof(*now));
1632 1.1 christos dtio->event_base = ub_default_event_base(0, secs, now);
1633 1.1 christos if(!dtio->event_base) {
1634 1.1 christos fatal_exit("dnstap io: could not create event_base");
1635 1.1 christos }
1636 1.1 christos }
1637 1.1 christos #endif /* THREADS_DISABLED */
1638 1.1 christos
1639 1.1 christos /** setup the cmd event for dnstap io */
1640 1.1 christos static void dtio_setup_cmd(struct dt_io_thread* dtio)
1641 1.1 christos {
1642 1.1 christos struct ub_event* cmdev;
1643 1.1 christos fd_set_nonblock(dtio->commandpipe[0]);
1644 1.1 christos cmdev = ub_event_new(dtio->event_base, dtio->commandpipe[0],
1645 1.1 christos UB_EV_READ | UB_EV_PERSIST, &dtio_cmd_cb, dtio);
1646 1.1 christos if(!cmdev) {
1647 1.1 christos fatal_exit("dnstap io: out of memory");
1648 1.1 christos }
1649 1.1 christos dtio->command_event = cmdev;
1650 1.1 christos if(ub_event_add(cmdev, NULL) != 0) {
1651 1.1 christos fatal_exit("dnstap io: out of memory (adding event)");
1652 1.1 christos }
1653 1.1 christos }
1654 1.1 christos
1655 1.1 christos /** setup the reconnect event for dnstap io */
1656 1.1 christos static void dtio_setup_reconnect(struct dt_io_thread* dtio)
1657 1.1 christos {
1658 1.1 christos dtio_reconnect_clear(dtio);
1659 1.1 christos dtio->reconnect_timer = ub_event_new(dtio->event_base, -1,
1660 1.1 christos UB_EV_TIMEOUT, &dtio_reconnect_timeout_cb, dtio);
1661 1.1 christos if(!dtio->reconnect_timer) {
1662 1.1 christos fatal_exit("dnstap io: out of memory");
1663 1.1 christos }
1664 1.1 christos }
1665 1.1 christos
1666 1.1 christos /**
1667 1.1 christos * structure to keep track of information during stop flush
1668 1.1 christos */
1669 1.1 christos struct stop_flush_info {
1670 1.1 christos /** the event base during stop flush */
1671 1.1 christos struct ub_event_base* base;
1672 1.1 christos /** did we already want to exit this stop-flush event base */
1673 1.1 christos int want_to_exit_flush;
1674 1.1 christos /** has the timer fired */
1675 1.1 christos int timer_done;
1676 1.1 christos /** the dtio */
1677 1.1 christos struct dt_io_thread* dtio;
1678 1.1 christos /** the stop control frame */
1679 1.1 christos void* stop_frame;
1680 1.1 christos /** length of the stop frame */
1681 1.1 christos size_t stop_frame_len;
1682 1.1 christos /** how much we have done of the stop frame */
1683 1.1 christos size_t stop_frame_done;
1684 1.1 christos };
1685 1.1 christos
1686 1.1 christos /** exit the stop flush base */
1687 1.1 christos static void dtio_stop_flush_exit(struct stop_flush_info* info)
1688 1.1 christos {
1689 1.1 christos if(info->want_to_exit_flush)
1690 1.1 christos return;
1691 1.1 christos info->want_to_exit_flush = 1;
1692 1.1 christos if(ub_event_base_loopexit(info->base) != 0) {
1693 1.1 christos log_err("dnstap io: could not loopexit");
1694 1.1 christos }
1695 1.1 christos }
1696 1.1 christos
1697 1.1 christos /** send the stop control,
1698 1.1 christos * return true if completed the frame. */
1699 1.1 christos static int dtio_control_stop_send(struct stop_flush_info* info)
1700 1.1 christos {
1701 1.1 christos struct dt_io_thread* dtio = info->dtio;
1702 1.1 christos int r;
1703 1.1 christos if(info->stop_frame_done >= info->stop_frame_len)
1704 1.1 christos return 1;
1705 1.1 christos r = dtio_write_buf(dtio, ((uint8_t*)info->stop_frame) +
1706 1.1 christos info->stop_frame_done, info->stop_frame_len -
1707 1.1 christos info->stop_frame_done);
1708 1.1 christos if(r == -1) {
1709 1.1 christos verbose(VERB_ALGO, "dnstap io: stop flush: output closed");
1710 1.1 christos dtio_stop_flush_exit(info);
1711 1.1 christos return 0;
1712 1.1 christos }
1713 1.1 christos if(r == 0) {
1714 1.1 christos /* try again later, or timeout */
1715 1.1 christos return 0;
1716 1.1 christos }
1717 1.1 christos info->stop_frame_done += r;
1718 1.1 christos if(info->stop_frame_done < info->stop_frame_len)
1719 1.1 christos return 0; /* not done yet */
1720 1.1 christos return 1;
1721 1.1 christos }
1722 1.1 christos
1723 1.1 christos void dtio_stop_timer_cb(int ATTR_UNUSED(fd), short ATTR_UNUSED(bits),
1724 1.1 christos void* arg)
1725 1.1 christos {
1726 1.1 christos struct stop_flush_info* info = (struct stop_flush_info*)arg;
1727 1.1 christos if(info->want_to_exit_flush)
1728 1.1 christos return;
1729 1.1 christos verbose(VERB_ALGO, "dnstap io: stop flush timer expired, stop flush");
1730 1.1 christos info->timer_done = 1;
1731 1.1 christos dtio_stop_flush_exit(info);
1732 1.1 christos }
1733 1.1 christos
1734 1.1 christos void dtio_stop_ev_cb(int ATTR_UNUSED(fd), short bits, void* arg)
1735 1.1 christos {
1736 1.1 christos struct stop_flush_info* info = (struct stop_flush_info*)arg;
1737 1.1 christos struct dt_io_thread* dtio = info->dtio;
1738 1.1 christos if(info->want_to_exit_flush)
1739 1.1 christos return;
1740 1.1 christos if(dtio->check_nb_connect) {
1741 1.1 christos /* we don't start the stop_flush if connect still
1742 1.1 christos * in progress, but the check code is here, just in case */
1743 1.1 christos int connect_err = dtio_check_nb_connect(dtio);
1744 1.1 christos if(connect_err == -1) {
1745 1.1 christos /* close the channel, exit the stop flush */
1746 1.1 christos dtio_stop_flush_exit(info);
1747 1.1 christos dtio_del_output_event(dtio);
1748 1.1 christos dtio_close_output(dtio);
1749 1.1 christos return;
1750 1.1 christos } else if(connect_err == 0) {
1751 1.1 christos /* try again later */
1752 1.1 christos return;
1753 1.1 christos }
1754 1.1 christos /* nonblocking connect check passed, continue */
1755 1.1 christos }
1756 1.1 christos #ifdef HAVE_SSL
1757 1.1 christos if(dtio->ssl &&
1758 1.1 christos (!dtio->ssl_handshake_done || dtio->ssl_brief_read)) {
1759 1.1 christos if(!dtio_ssl_handshake(dtio, info))
1760 1.1 christos return;
1761 1.1 christos }
1762 1.1 christos #endif
1763 1.1 christos
1764 1.1 christos if((bits&UB_EV_READ)) {
1765 1.1 christos if(!dtio_check_close(dtio)) {
1766 1.1 christos if(dtio->fd == -1) {
1767 1.1 christos verbose(VERB_ALGO, "dnstap io: "
1768 1.1 christos "stop flush: output closed");
1769 1.1 christos dtio_stop_flush_exit(info);
1770 1.1 christos }
1771 1.1 christos return;
1772 1.1 christos }
1773 1.1 christos }
1774 1.1 christos /* write remainder of last frame */
1775 1.1 christos if(dtio->cur_msg) {
1776 1.1 christos if(dtio->cur_msg_done < dtio->cur_msg_len) {
1777 1.1 christos if(!dtio_write_more(dtio)) {
1778 1.1 christos if(dtio->fd == -1) {
1779 1.1 christos verbose(VERB_ALGO, "dnstap io: "
1780 1.1 christos "stop flush: output closed");
1781 1.1 christos dtio_stop_flush_exit(info);
1782 1.1 christos }
1783 1.1 christos return;
1784 1.1 christos }
1785 1.1 christos }
1786 1.1 christos verbose(VERB_ALGO, "dnstap io: stop flush completed "
1787 1.1 christos "last frame");
1788 1.1 christos dtio_cur_msg_free(dtio);
1789 1.1 christos }
1790 1.1 christos /* write stop frame */
1791 1.1 christos if(info->stop_frame_done < info->stop_frame_len) {
1792 1.1 christos if(!dtio_control_stop_send(info))
1793 1.1 christos return;
1794 1.1 christos verbose(VERB_ALGO, "dnstap io: stop flush completed "
1795 1.1 christos "stop control frame");
1796 1.1 christos }
1797 1.1 christos /* when last frame and stop frame are sent, exit */
1798 1.1 christos dtio_stop_flush_exit(info);
1799 1.1 christos }
1800 1.1 christos
1801 1.1 christos /** flush at end, last packet and stop control */
1802 1.1 christos static void dtio_control_stop_flush(struct dt_io_thread* dtio)
1803 1.1 christos {
1804 1.1 christos /* briefly attempt to flush the previous packet to the output,
1805 1.1 christos * this could be a partial packet, or even the start control frame */
1806 1.1 christos time_t secs = 0;
1807 1.1 christos struct timeval now;
1808 1.1 christos struct stop_flush_info info;
1809 1.1 christos struct timeval tv;
1810 1.1 christos struct ub_event* timer, *stopev;
1811 1.1 christos
1812 1.1 christos if(dtio->fd == -1 || dtio->check_nb_connect) {
1813 1.1 christos /* no connection or we have just connected, so nothing is
1814 1.1 christos * sent yet, so nothing to stop or flush */
1815 1.1 christos return;
1816 1.1 christos }
1817 1.1 christos if(dtio->ssl && !dtio->ssl_handshake_done) {
1818 1.1 christos /* no SSL connection has been established yet */
1819 1.1 christos return;
1820 1.1 christos }
1821 1.1 christos
1822 1.1 christos memset(&info, 0, sizeof(info));
1823 1.1 christos memset(&now, 0, sizeof(now));
1824 1.1 christos info.dtio = dtio;
1825 1.1 christos info.base = ub_default_event_base(0, &secs, &now);
1826 1.1 christos if(!info.base) {
1827 1.1 christos log_err("dnstap io: malloc failure");
1828 1.1 christos return;
1829 1.1 christos }
1830 1.1 christos timer = ub_event_new(info.base, -1, UB_EV_TIMEOUT,
1831 1.1 christos &dtio_stop_timer_cb, &info);
1832 1.1 christos if(!timer) {
1833 1.1 christos log_err("dnstap io: malloc failure");
1834 1.1 christos ub_event_base_free(info.base);
1835 1.1 christos return;
1836 1.1 christos }
1837 1.1 christos memset(&tv, 0, sizeof(tv));
1838 1.1 christos tv.tv_sec = 2;
1839 1.1 christos if(ub_timer_add(timer, info.base, &dtio_stop_timer_cb, &info,
1840 1.1 christos &tv) != 0) {
1841 1.1 christos log_err("dnstap io: cannot event_timer_add");
1842 1.1 christos ub_event_free(timer);
1843 1.1 christos ub_event_base_free(info.base);
1844 1.1 christos return;
1845 1.1 christos }
1846 1.1 christos stopev = ub_event_new(info.base, dtio->fd, UB_EV_READ |
1847 1.1 christos UB_EV_WRITE | UB_EV_PERSIST, &dtio_stop_ev_cb, &info);
1848 1.1 christos if(!stopev) {
1849 1.1 christos log_err("dnstap io: malloc failure");
1850 1.1 christos ub_timer_del(timer);
1851 1.1 christos ub_event_free(timer);
1852 1.1 christos ub_event_base_free(info.base);
1853 1.1 christos return;
1854 1.1 christos }
1855 1.1 christos if(ub_event_add(stopev, NULL) != 0) {
1856 1.1 christos log_err("dnstap io: cannot event_add");
1857 1.1 christos ub_event_free(stopev);
1858 1.1 christos ub_timer_del(timer);
1859 1.1 christos ub_event_free(timer);
1860 1.1 christos ub_event_base_free(info.base);
1861 1.1 christos return;
1862 1.1 christos }
1863 1.1 christos info.stop_frame = fstrm_create_control_frame_stop(
1864 1.1 christos &info.stop_frame_len);
1865 1.1 christos if(!info.stop_frame) {
1866 1.1 christos log_err("dnstap io: malloc failure");
1867 1.1 christos ub_event_del(stopev);
1868 1.1 christos ub_event_free(stopev);
1869 1.1 christos ub_timer_del(timer);
1870 1.1 christos ub_event_free(timer);
1871 1.1 christos ub_event_base_free(info.base);
1872 1.1 christos return;
1873 1.1 christos }
1874 1.1 christos dtio->stop_flush_event = stopev;
1875 1.1 christos
1876 1.1 christos /* wait briefly, or until finished */
1877 1.1 christos verbose(VERB_ALGO, "dnstap io: stop flush started");
1878 1.1 christos if(ub_event_base_dispatch(info.base) < 0) {
1879 1.1 christos log_err("dnstap io: dispatch flush failed, errno is %s",
1880 1.1 christos strerror(errno));
1881 1.1 christos }
1882 1.1 christos verbose(VERB_ALGO, "dnstap io: stop flush ended");
1883 1.1 christos free(info.stop_frame);
1884 1.1 christos dtio->stop_flush_event = NULL;
1885 1.1 christos ub_event_del(stopev);
1886 1.1 christos ub_event_free(stopev);
1887 1.1 christos ub_timer_del(timer);
1888 1.1 christos ub_event_free(timer);
1889 1.1 christos ub_event_base_free(info.base);
1890 1.1 christos }
1891 1.1 christos
1892 1.1 christos /** perform desetup and free stuff when the dnstap io thread exits */
1893 1.1 christos static void dtio_desetup(struct dt_io_thread* dtio)
1894 1.1 christos {
1895 1.1 christos dtio_control_stop_flush(dtio);
1896 1.1 christos dtio_del_output_event(dtio);
1897 1.1 christos dtio_close_output(dtio);
1898 1.1 christos ub_event_del(dtio->command_event);
1899 1.1 christos ub_event_free(dtio->command_event);
1900 1.1 christos #ifndef USE_WINSOCK
1901 1.1 christos close(dtio->commandpipe[0]);
1902 1.1 christos #else
1903 1.1 christos _close(dtio->commandpipe[0]);
1904 1.1 christos #endif
1905 1.1 christos dtio->commandpipe[0] = -1;
1906 1.1 christos dtio_reconnect_del(dtio);
1907 1.1 christos ub_event_free(dtio->reconnect_timer);
1908 1.1 christos dtio_cur_msg_free(dtio);
1909 1.1 christos #ifndef THREADS_DISABLED
1910 1.1 christos ub_event_base_free(dtio->event_base);
1911 1.1 christos #endif
1912 1.1 christos }
1913 1.1 christos
1914 1.1 christos /** setup a start control message */
1915 1.1 christos static int dtio_control_start_send(struct dt_io_thread* dtio)
1916 1.1 christos {
1917 1.1 christos log_assert(dtio->cur_msg == NULL && dtio->cur_msg_len == 0);
1918 1.1 christos dtio->cur_msg = fstrm_create_control_frame_start(DNSTAP_CONTENT_TYPE,
1919 1.1 christos &dtio->cur_msg_len);
1920 1.1 christos if(!dtio->cur_msg) {
1921 1.1 christos return 0;
1922 1.1 christos }
1923 1.1 christos /* setup to send the control message */
1924 1.1 christos /* set that the buffer needs to be sent, but the length
1925 1.1 christos * of that buffer is already written, that way the buffer can
1926 1.1 christos * start with 0 length and then the length of the control frame
1927 1.1 christos * in it */
1928 1.1 christos dtio->cur_msg_done = 0;
1929 1.1 christos dtio->cur_msg_len_done = 4;
1930 1.1 christos return 1;
1931 1.1 christos }
1932 1.1 christos
1933 1.1 christos /** setup a ready control message */
1934 1.1 christos static int dtio_control_ready_send(struct dt_io_thread* dtio)
1935 1.1 christos {
1936 1.1 christos log_assert(dtio->cur_msg == NULL && dtio->cur_msg_len == 0);
1937 1.1 christos dtio->cur_msg = fstrm_create_control_frame_ready(DNSTAP_CONTENT_TYPE,
1938 1.1 christos &dtio->cur_msg_len);
1939 1.1 christos if(!dtio->cur_msg) {
1940 1.1 christos return 0;
1941 1.1 christos }
1942 1.1 christos /* setup to send the control message */
1943 1.1 christos /* set that the buffer needs to be sent, but the length
1944 1.1 christos * of that buffer is already written, that way the buffer can
1945 1.1 christos * start with 0 length and then the length of the control frame
1946 1.1 christos * in it */
1947 1.1 christos dtio->cur_msg_done = 0;
1948 1.1 christos dtio->cur_msg_len_done = 4;
1949 1.1 christos return 1;
1950 1.1 christos }
1951 1.1 christos
1952 1.1 christos /** open the output file descriptor for af_local */
1953 1.1 christos static int dtio_open_output_local(struct dt_io_thread* dtio)
1954 1.1 christos {
1955 1.1 christos #ifdef HAVE_SYS_UN_H
1956 1.1 christos struct sockaddr_un s;
1957 1.1 christos dtio->fd = socket(AF_LOCAL, SOCK_STREAM, 0);
1958 1.1 christos if(dtio->fd == -1) {
1959 1.1 christos log_err("dnstap io: failed to create socket: %s",
1960 1.1 christos sock_strerror(errno));
1961 1.1 christos return 0;
1962 1.1 christos }
1963 1.1 christos memset(&s, 0, sizeof(s));
1964 1.1 christos #ifdef HAVE_STRUCT_SOCKADDR_UN_SUN_LEN
1965 1.1 christos /* this member exists on BSDs, not Linux */
1966 1.1 christos s.sun_len = (unsigned)sizeof(s);
1967 1.1 christos #endif
1968 1.1 christos s.sun_family = AF_LOCAL;
1969 1.1 christos /* length is 92-108, 104 on FreeBSD */
1970 1.1 christos (void)strlcpy(s.sun_path, dtio->socket_path, sizeof(s.sun_path));
1971 1.1 christos fd_set_nonblock(dtio->fd);
1972 1.1 christos if(connect(dtio->fd, (struct sockaddr*)&s, (socklen_t)sizeof(s))
1973 1.1 christos == -1) {
1974 1.1 christos char* to = dtio->socket_path;
1975 1.1 christos if(dtio->reconnect_timeout > DTIO_RECONNECT_TIMEOUT_MIN &&
1976 1.1 christos verbosity < 4) {
1977 1.1 christos dtio_close_fd(dtio);
1978 1.1 christos return 0; /* no log retries on low verbosity */
1979 1.1 christos }
1980 1.1 christos log_err("dnstap io: failed to connect to \"%s\": %s",
1981 1.1 christos to, sock_strerror(errno));
1982 1.1 christos dtio_close_fd(dtio);
1983 1.1 christos return 0;
1984 1.1 christos }
1985 1.1 christos return 1;
1986 1.1 christos #else
1987 1.1 christos log_err("cannot create af_local socket");
1988 1.1 christos return 0;
1989 1.1 christos #endif /* HAVE_SYS_UN_H */
1990 1.1 christos }
1991 1.1 christos
1992 1.1 christos /** open the output file descriptor for af_inet and af_inet6 */
1993 1.1 christos static int dtio_open_output_tcp(struct dt_io_thread* dtio)
1994 1.1 christos {
1995 1.1 christos struct sockaddr_storage addr;
1996 1.1 christos socklen_t addrlen;
1997 1.1 christos memset(&addr, 0, sizeof(addr));
1998 1.1 christos addrlen = (socklen_t)sizeof(addr);
1999 1.1 christos
2000 1.1.1.3 christos if(!extstrtoaddr(dtio->ip_str, &addr, &addrlen, UNBOUND_DNS_PORT)) {
2001 1.1 christos log_err("could not parse IP '%s'", dtio->ip_str);
2002 1.1 christos return 0;
2003 1.1 christos }
2004 1.1 christos dtio->fd = socket(addr.ss_family, SOCK_STREAM, 0);
2005 1.1 christos if(dtio->fd == -1) {
2006 1.1 christos log_err("can't create socket: %s", sock_strerror(errno));
2007 1.1 christos return 0;
2008 1.1 christos }
2009 1.1 christos fd_set_nonblock(dtio->fd);
2010 1.1 christos if(connect(dtio->fd, (struct sockaddr*)&addr, addrlen) == -1) {
2011 1.1 christos if(errno == EINPROGRESS)
2012 1.1 christos return 1; /* wait until connect done*/
2013 1.1 christos if(dtio->reconnect_timeout > DTIO_RECONNECT_TIMEOUT_MIN &&
2014 1.1 christos verbosity < 4) {
2015 1.1 christos dtio_close_fd(dtio);
2016 1.1 christos return 0; /* no log retries on low verbosity */
2017 1.1 christos }
2018 1.1 christos #ifndef USE_WINSOCK
2019 1.1 christos if(tcp_connect_errno_needs_log(
2020 1.1 christos (struct sockaddr *)&addr, addrlen)) {
2021 1.1 christos log_err("dnstap io: failed to connect to %s: %s",
2022 1.1 christos dtio->ip_str, strerror(errno));
2023 1.1 christos }
2024 1.1 christos #else
2025 1.1 christos if(WSAGetLastError() == WSAEINPROGRESS ||
2026 1.1 christos WSAGetLastError() == WSAEWOULDBLOCK)
2027 1.1 christos return 1; /* wait until connect done*/
2028 1.1 christos if(tcp_connect_errno_needs_log(
2029 1.1 christos (struct sockaddr *)&addr, addrlen)) {
2030 1.1 christos log_err("dnstap io: failed to connect to %s: %s",
2031 1.1 christos dtio->ip_str, wsa_strerror(WSAGetLastError()));
2032 1.1 christos }
2033 1.1 christos #endif
2034 1.1 christos dtio_close_fd(dtio);
2035 1.1 christos return 0;
2036 1.1 christos }
2037 1.1 christos return 1;
2038 1.1 christos }
2039 1.1 christos
2040 1.1 christos /** setup the SSL structure for new connection */
2041 1.1 christos static int dtio_setup_ssl(struct dt_io_thread* dtio)
2042 1.1 christos {
2043 1.1 christos dtio->ssl = outgoing_ssl_fd(dtio->ssl_ctx, dtio->fd);
2044 1.1 christos if(!dtio->ssl) return 0;
2045 1.1 christos dtio->ssl_handshake_done = 0;
2046 1.1 christos dtio->ssl_brief_read = 0;
2047 1.1 christos
2048 1.1 christos if(!set_auth_name_on_ssl(dtio->ssl, dtio->tls_server_name,
2049 1.1 christos dtio->tls_use_sni)) {
2050 1.1 christos return 0;
2051 1.1 christos }
2052 1.1 christos return 1;
2053 1.1 christos }
2054 1.1 christos
2055 1.1 christos /** open the output file descriptor */
2056 1.1 christos static void dtio_open_output(struct dt_io_thread* dtio)
2057 1.1 christos {
2058 1.1 christos struct ub_event* ev;
2059 1.1 christos if(dtio->upstream_is_unix) {
2060 1.1 christos if(!dtio_open_output_local(dtio)) {
2061 1.1 christos dtio_reconnect_enable(dtio);
2062 1.1 christos return;
2063 1.1 christos }
2064 1.1 christos } else if(dtio->upstream_is_tcp || dtio->upstream_is_tls) {
2065 1.1 christos if(!dtio_open_output_tcp(dtio)) {
2066 1.1 christos dtio_reconnect_enable(dtio);
2067 1.1 christos return;
2068 1.1 christos }
2069 1.1 christos if(dtio->upstream_is_tls) {
2070 1.1 christos if(!dtio_setup_ssl(dtio)) {
2071 1.1 christos dtio_close_fd(dtio);
2072 1.1 christos dtio_reconnect_enable(dtio);
2073 1.1 christos return;
2074 1.1 christos }
2075 1.1 christos }
2076 1.1 christos }
2077 1.1 christos dtio->check_nb_connect = 1;
2078 1.1 christos
2079 1.1 christos /* the EV_READ is to read ACCEPT control messages, and catch channel
2080 1.1 christos * close. EV_WRITE is to write packets */
2081 1.1 christos ev = ub_event_new(dtio->event_base, dtio->fd,
2082 1.1 christos UB_EV_READ | UB_EV_WRITE | UB_EV_PERSIST, &dtio_output_cb,
2083 1.1 christos dtio);
2084 1.1 christos if(!ev) {
2085 1.1 christos log_err("dnstap io: out of memory");
2086 1.1 christos if(dtio->ssl) {
2087 1.1 christos #ifdef HAVE_SSL
2088 1.1 christos SSL_free(dtio->ssl);
2089 1.1 christos dtio->ssl = NULL;
2090 1.1 christos #endif
2091 1.1 christos }
2092 1.1 christos dtio_close_fd(dtio);
2093 1.1 christos dtio_reconnect_enable(dtio);
2094 1.1 christos return;
2095 1.1 christos }
2096 1.1 christos dtio->event = ev;
2097 1.1 christos
2098 1.1 christos /* setup protocol control message to start */
2099 1.1 christos if((!dtio->is_bidirectional && !dtio_control_start_send(dtio)) ||
2100 1.1 christos (dtio->is_bidirectional && !dtio_control_ready_send(dtio)) ) {
2101 1.1 christos log_err("dnstap io: out of memory");
2102 1.1 christos ub_event_free(dtio->event);
2103 1.1 christos dtio->event = NULL;
2104 1.1 christos if(dtio->ssl) {
2105 1.1 christos #ifdef HAVE_SSL
2106 1.1 christos SSL_free(dtio->ssl);
2107 1.1 christos dtio->ssl = NULL;
2108 1.1 christos #endif
2109 1.1 christos }
2110 1.1 christos dtio_close_fd(dtio);
2111 1.1 christos dtio_reconnect_enable(dtio);
2112 1.1 christos return;
2113 1.1 christos }
2114 1.1 christos }
2115 1.1 christos
2116 1.1 christos /** perform the setup of the writer thread on the established event_base */
2117 1.1 christos static void dtio_setup_on_base(struct dt_io_thread* dtio)
2118 1.1 christos {
2119 1.1 christos dtio_setup_cmd(dtio);
2120 1.1 christos dtio_setup_reconnect(dtio);
2121 1.1 christos dtio_open_output(dtio);
2122 1.1 christos if(!dtio_add_output_event_write(dtio))
2123 1.1 christos return;
2124 1.1 christos }
2125 1.1 christos
2126 1.1 christos #ifndef THREADS_DISABLED
2127 1.1 christos /** the IO thread function for the DNSTAP IO */
2128 1.1 christos static void* dnstap_io(void* arg)
2129 1.1 christos {
2130 1.1 christos struct dt_io_thread* dtio = (struct dt_io_thread*)arg;
2131 1.1 christos time_t secs = 0;
2132 1.1 christos struct timeval now;
2133 1.1 christos log_thread_set(&dtio->threadnum);
2134 1.1 christos
2135 1.1 christos /* setup */
2136 1.1 christos verbose(VERB_ALGO, "start dnstap io thread");
2137 1.1 christos dtio_setup_base(dtio, &secs, &now);
2138 1.1 christos dtio_setup_on_base(dtio);
2139 1.1 christos
2140 1.1 christos /* run */
2141 1.1 christos if(ub_event_base_dispatch(dtio->event_base) < 0) {
2142 1.1 christos log_err("dnstap io: dispatch failed, errno is %s",
2143 1.1 christos strerror(errno));
2144 1.1 christos }
2145 1.1 christos
2146 1.1 christos /* cleanup */
2147 1.1 christos verbose(VERB_ALGO, "stop dnstap io thread");
2148 1.1 christos dtio_desetup(dtio);
2149 1.1 christos return NULL;
2150 1.1 christos }
2151 1.1 christos #endif /* THREADS_DISABLED */
2152 1.1 christos
2153 1.1 christos int dt_io_thread_start(struct dt_io_thread* dtio, void* event_base_nothr,
2154 1.1 christos int numworkers)
2155 1.1 christos {
2156 1.1 christos /* set up the thread, can fail */
2157 1.1 christos #ifndef USE_WINSOCK
2158 1.1 christos if(pipe(dtio->commandpipe) == -1) {
2159 1.1 christos log_err("failed to create pipe: %s", strerror(errno));
2160 1.1 christos return 0;
2161 1.1 christos }
2162 1.1 christos #else
2163 1.1 christos if(_pipe(dtio->commandpipe, 4096, _O_BINARY) == -1) {
2164 1.1 christos log_err("failed to create _pipe: %s",
2165 1.1 christos wsa_strerror(WSAGetLastError()));
2166 1.1 christos return 0;
2167 1.1 christos }
2168 1.1 christos #endif
2169 1.1 christos
2170 1.1 christos /* start the thread */
2171 1.1 christos dtio->threadnum = numworkers+1;
2172 1.1 christos dtio->started = 1;
2173 1.1 christos #ifndef THREADS_DISABLED
2174 1.1 christos ub_thread_create(&dtio->tid, dnstap_io, dtio);
2175 1.1 christos (void)event_base_nothr;
2176 1.1 christos #else
2177 1.1 christos dtio->event_base = event_base_nothr;
2178 1.1 christos dtio_setup_on_base(dtio);
2179 1.1 christos #endif
2180 1.1 christos return 1;
2181 1.1 christos }
2182 1.1 christos
2183 1.1 christos void dt_io_thread_stop(struct dt_io_thread* dtio)
2184 1.1 christos {
2185 1.1 christos #ifndef THREADS_DISABLED
2186 1.1 christos uint8_t cmd = DTIO_COMMAND_STOP;
2187 1.1 christos #endif
2188 1.1 christos if(!dtio) return;
2189 1.1 christos if(!dtio->started) return;
2190 1.1 christos verbose(VERB_ALGO, "dnstap io: send stop cmd");
2191 1.1 christos
2192 1.1 christos #ifndef THREADS_DISABLED
2193 1.1 christos while(1) {
2194 1.1 christos ssize_t r = write(dtio->commandpipe[1], &cmd, sizeof(cmd));
2195 1.1 christos if(r == -1) {
2196 1.1 christos #ifndef USE_WINSOCK
2197 1.1 christos if(errno == EINTR || errno == EAGAIN)
2198 1.1 christos continue;
2199 1.1 christos #else
2200 1.1 christos if(WSAGetLastError() == WSAEINPROGRESS)
2201 1.1 christos continue;
2202 1.1 christos if(WSAGetLastError() == WSAEWOULDBLOCK)
2203 1.1 christos continue;
2204 1.1 christos #endif
2205 1.1 christos log_err("dnstap io stop: write: %s",
2206 1.1 christos sock_strerror(errno));
2207 1.1 christos break;
2208 1.1 christos }
2209 1.1 christos break;
2210 1.1 christos }
2211 1.1 christos dtio->started = 0;
2212 1.1 christos #endif /* THREADS_DISABLED */
2213 1.1 christos
2214 1.1 christos #ifndef USE_WINSOCK
2215 1.1 christos close(dtio->commandpipe[1]);
2216 1.1 christos #else
2217 1.1 christos _close(dtio->commandpipe[1]);
2218 1.1 christos #endif
2219 1.1 christos dtio->commandpipe[1] = -1;
2220 1.1 christos #ifndef THREADS_DISABLED
2221 1.1 christos ub_thread_join(dtio->tid);
2222 1.1 christos #else
2223 1.1 christos dtio->want_to_exit = 1;
2224 1.1 christos dtio_desetup(dtio);
2225 1.1 christos #endif
2226 1.1 christos }
2227