xref: /freebsd/contrib/unbound/util/netevent.c (revision 8f76bb7dad48538c6832c2fb466a433d2a3f8cd5)
1b7579f77SDag-Erling Smørgrav /*
2b7579f77SDag-Erling Smørgrav  * util/netevent.c - event notification
3b7579f77SDag-Erling Smørgrav  *
4b7579f77SDag-Erling Smørgrav  * Copyright (c) 2007, NLnet Labs. All rights reserved.
5b7579f77SDag-Erling Smørgrav  *
6b7579f77SDag-Erling Smørgrav  * This software is open source.
7b7579f77SDag-Erling Smørgrav  *
8b7579f77SDag-Erling Smørgrav  * Redistribution and use in source and binary forms, with or without
9b7579f77SDag-Erling Smørgrav  * modification, are permitted provided that the following conditions
10b7579f77SDag-Erling Smørgrav  * are met:
11b7579f77SDag-Erling Smørgrav  *
12b7579f77SDag-Erling Smørgrav  * Redistributions of source code must retain the above copyright notice,
13b7579f77SDag-Erling Smørgrav  * this list of conditions and the following disclaimer.
14b7579f77SDag-Erling Smørgrav  *
15b7579f77SDag-Erling Smørgrav  * Redistributions in binary form must reproduce the above copyright notice,
16b7579f77SDag-Erling Smørgrav  * this list of conditions and the following disclaimer in the documentation
17b7579f77SDag-Erling Smørgrav  * and/or other materials provided with the distribution.
18b7579f77SDag-Erling Smørgrav  *
19b7579f77SDag-Erling Smørgrav  * Neither the name of the NLNET LABS nor the names of its contributors may
20b7579f77SDag-Erling Smørgrav  * be used to endorse or promote products derived from this software without
21b7579f77SDag-Erling Smørgrav  * specific prior written permission.
22b7579f77SDag-Erling Smørgrav  *
23b7579f77SDag-Erling Smørgrav  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
2417d15b25SDag-Erling Smørgrav  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
2517d15b25SDag-Erling Smørgrav  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
2617d15b25SDag-Erling Smørgrav  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
2717d15b25SDag-Erling Smørgrav  * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
2817d15b25SDag-Erling Smørgrav  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
2917d15b25SDag-Erling Smørgrav  * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
3017d15b25SDag-Erling Smørgrav  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
3117d15b25SDag-Erling Smørgrav  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
3217d15b25SDag-Erling Smørgrav  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
3317d15b25SDag-Erling Smørgrav  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34b7579f77SDag-Erling Smørgrav  */
35b7579f77SDag-Erling Smørgrav 
36b7579f77SDag-Erling Smørgrav /**
37b7579f77SDag-Erling Smørgrav  * \file
38b7579f77SDag-Erling Smørgrav  *
39b7579f77SDag-Erling Smørgrav  * This file contains event notification functions.
40b7579f77SDag-Erling Smørgrav  */
41b7579f77SDag-Erling Smørgrav #include "config.h"
42b7579f77SDag-Erling Smørgrav #include "util/netevent.h"
43e2d15004SDag-Erling Smørgrav #include "util/ub_event.h"
44b7579f77SDag-Erling Smørgrav #include "util/log.h"
45b7579f77SDag-Erling Smørgrav #include "util/net_help.h"
464c75e3aaSDag-Erling Smørgrav #include "util/tcp_conn_limit.h"
47b7579f77SDag-Erling Smørgrav #include "util/fptr_wlist.h"
48865f46b2SCy Schubert #include "util/proxy_protocol.h"
49*8f76bb7dSCy Schubert #include "util/timeval_func.h"
5009a3aaf3SDag-Erling Smørgrav #include "sldns/pkthdr.h"
5109a3aaf3SDag-Erling Smørgrav #include "sldns/sbuffer.h"
520fb34990SDag-Erling Smørgrav #include "sldns/str2wire.h"
53ff825849SDag-Erling Smørgrav #include "dnstap/dnstap.h"
5465b390aaSDag-Erling Smørgrav #include "dnscrypt/dnscrypt.h"
55e86b9096SDag-Erling Smørgrav #include "services/listen_dnsport.h"
565469a995SCy Schubert #ifdef HAVE_SYS_TYPES_H
575469a995SCy Schubert #include <sys/types.h>
585469a995SCy Schubert #endif
595469a995SCy Schubert #ifdef HAVE_SYS_SOCKET_H
605469a995SCy Schubert #include <sys/socket.h>
615469a995SCy Schubert #endif
625469a995SCy Schubert #ifdef HAVE_NETDB_H
635469a995SCy Schubert #include <netdb.h>
645469a995SCy Schubert #endif
65865f46b2SCy Schubert #ifdef HAVE_POLL_H
66865f46b2SCy Schubert #include <poll.h>
67865f46b2SCy Schubert #endif
685469a995SCy Schubert 
698ed2b524SDag-Erling Smørgrav #ifdef HAVE_OPENSSL_SSL_H
70b7579f77SDag-Erling Smørgrav #include <openssl/ssl.h>
718ed2b524SDag-Erling Smørgrav #endif
728ed2b524SDag-Erling Smørgrav #ifdef HAVE_OPENSSL_ERR_H
73b7579f77SDag-Erling Smørgrav #include <openssl/err.h>
748ed2b524SDag-Erling Smørgrav #endif
75*8f76bb7dSCy Schubert #ifdef HAVE_LINUX_NET_TSTAMP_H
76*8f76bb7dSCy Schubert #include <linux/net_tstamp.h>
77*8f76bb7dSCy Schubert #endif
78b7579f77SDag-Erling Smørgrav /* -------- Start of local definitions -------- */
79b7579f77SDag-Erling Smørgrav /** if CMSG_ALIGN is not defined on this platform, a workaround */
80b7579f77SDag-Erling Smørgrav #ifndef CMSG_ALIGN
81f61ef7f6SDag-Erling Smørgrav #  ifdef __CMSG_ALIGN
82f61ef7f6SDag-Erling Smørgrav #    define CMSG_ALIGN(n) __CMSG_ALIGN(n)
83f61ef7f6SDag-Erling Smørgrav #  elif defined(CMSG_DATA_ALIGN)
84b7579f77SDag-Erling Smørgrav #    define CMSG_ALIGN _CMSG_DATA_ALIGN
85b7579f77SDag-Erling Smørgrav #  else
86b7579f77SDag-Erling Smørgrav #    define CMSG_ALIGN(len) (((len)+sizeof(long)-1) & ~(sizeof(long)-1))
87b7579f77SDag-Erling Smørgrav #  endif
88b7579f77SDag-Erling Smørgrav #endif
89b7579f77SDag-Erling Smørgrav 
90b7579f77SDag-Erling Smørgrav /** if CMSG_LEN is not defined on this platform, a workaround */
91b7579f77SDag-Erling Smørgrav #ifndef CMSG_LEN
92b7579f77SDag-Erling Smørgrav #  define CMSG_LEN(len) (CMSG_ALIGN(sizeof(struct cmsghdr))+(len))
93b7579f77SDag-Erling Smørgrav #endif
94b7579f77SDag-Erling Smørgrav 
95b7579f77SDag-Erling Smørgrav /** if CMSG_SPACE is not defined on this platform, a workaround */
96b7579f77SDag-Erling Smørgrav #ifndef CMSG_SPACE
97b7579f77SDag-Erling Smørgrav #  ifdef _CMSG_HDR_ALIGN
98b7579f77SDag-Erling Smørgrav #    define CMSG_SPACE(l) (CMSG_ALIGN(l)+_CMSG_HDR_ALIGN(sizeof(struct cmsghdr)))
99b7579f77SDag-Erling Smørgrav #  else
100b7579f77SDag-Erling Smørgrav #    define CMSG_SPACE(l) (CMSG_ALIGN(l)+CMSG_ALIGN(sizeof(struct cmsghdr)))
101b7579f77SDag-Erling Smørgrav #  endif
102b7579f77SDag-Erling Smørgrav #endif
103b7579f77SDag-Erling Smørgrav 
1044c75e3aaSDag-Erling Smørgrav /** The TCP writing query timeout in milliseconds */
105b5663de9SDag-Erling Smørgrav #define TCP_QUERY_TIMEOUT 120000
1064c75e3aaSDag-Erling Smørgrav /** The minimum actual TCP timeout to use, regardless of what we advertise,
1074c75e3aaSDag-Erling Smørgrav  * in msec */
1084c75e3aaSDag-Erling Smørgrav #define TCP_QUERY_TIMEOUT_MINIMUM 200
109b7579f77SDag-Erling Smørgrav 
110b7579f77SDag-Erling Smørgrav #ifndef NONBLOCKING_IS_BROKEN
111b7579f77SDag-Erling Smørgrav /** number of UDP reads to perform per read indication from select */
112b7579f77SDag-Erling Smørgrav #define NUM_UDP_PER_SELECT 100
113b7579f77SDag-Erling Smørgrav #else
114b7579f77SDag-Erling Smørgrav #define NUM_UDP_PER_SELECT 1
115b7579f77SDag-Erling Smørgrav #endif
116b7579f77SDag-Erling Smørgrav 
117865f46b2SCy Schubert /** timeout in millisec to wait for write to unblock, packets dropped after.*/
118865f46b2SCy Schubert #define SEND_BLOCKED_WAIT_TIMEOUT 200
119865f46b2SCy Schubert 
120*8f76bb7dSCy Schubert /** Let's make timestamping code cleaner and redefine SO_TIMESTAMP* */
121*8f76bb7dSCy Schubert #ifndef SO_TIMESTAMP
122*8f76bb7dSCy Schubert #define SO_TIMESTAMP 29
123*8f76bb7dSCy Schubert #endif
124*8f76bb7dSCy Schubert #ifndef SO_TIMESTAMPNS
125*8f76bb7dSCy Schubert #define SO_TIMESTAMPNS 35
126*8f76bb7dSCy Schubert #endif
127*8f76bb7dSCy Schubert #ifndef SO_TIMESTAMPING
128*8f76bb7dSCy Schubert #define SO_TIMESTAMPING 37
129*8f76bb7dSCy Schubert #endif
130b7579f77SDag-Erling Smørgrav /**
131e2d15004SDag-Erling Smørgrav  * The internal event structure for keeping ub_event info for the event.
132b7579f77SDag-Erling Smørgrav  * Possibly other structures (list, tree) this is part of.
133b7579f77SDag-Erling Smørgrav  */
134b7579f77SDag-Erling Smørgrav struct internal_event {
135b7579f77SDag-Erling Smørgrav 	/** the comm base */
136b7579f77SDag-Erling Smørgrav 	struct comm_base* base;
137e2d15004SDag-Erling Smørgrav 	/** ub_event event type */
138e2d15004SDag-Erling Smørgrav 	struct ub_event* ev;
139b7579f77SDag-Erling Smørgrav };
140b7579f77SDag-Erling Smørgrav 
141b7579f77SDag-Erling Smørgrav /**
142b7579f77SDag-Erling Smørgrav  * Internal base structure, so that every thread has its own events.
143b7579f77SDag-Erling Smørgrav  */
144b7579f77SDag-Erling Smørgrav struct internal_base {
145e2d15004SDag-Erling Smørgrav 	/** ub_event event_base type. */
146e2d15004SDag-Erling Smørgrav 	struct ub_event_base* base;
147b7579f77SDag-Erling Smørgrav 	/** seconds time pointer points here */
14817d15b25SDag-Erling Smørgrav 	time_t secs;
149b7579f77SDag-Erling Smørgrav 	/** timeval with current time */
150b7579f77SDag-Erling Smørgrav 	struct timeval now;
151b7579f77SDag-Erling Smørgrav 	/** the event used for slow_accept timeouts */
152e2d15004SDag-Erling Smørgrav 	struct ub_event* slow_accept;
153b7579f77SDag-Erling Smørgrav 	/** true if slow_accept is enabled */
154b7579f77SDag-Erling Smørgrav 	int slow_accept_enabled;
155865f46b2SCy Schubert 	/** last log time for slow logging of file descriptor errors */
156865f46b2SCy Schubert 	time_t last_slow_log;
157865f46b2SCy Schubert 	/** last log time for slow logging of write wait failures */
158865f46b2SCy Schubert 	time_t last_writewait_log;
159b7579f77SDag-Erling Smørgrav };
160b7579f77SDag-Erling Smørgrav 
161b7579f77SDag-Erling Smørgrav /**
162b7579f77SDag-Erling Smørgrav  * Internal timer structure, to store timer event in.
163b7579f77SDag-Erling Smørgrav  */
164b7579f77SDag-Erling Smørgrav struct internal_timer {
165e2d15004SDag-Erling Smørgrav 	/** the super struct from which derived */
166e2d15004SDag-Erling Smørgrav 	struct comm_timer super;
167b7579f77SDag-Erling Smørgrav 	/** the comm base */
168b7579f77SDag-Erling Smørgrav 	struct comm_base* base;
169e2d15004SDag-Erling Smørgrav 	/** ub_event event type */
170e2d15004SDag-Erling Smørgrav 	struct ub_event* ev;
171b7579f77SDag-Erling Smørgrav 	/** is timer enabled */
172b7579f77SDag-Erling Smørgrav 	uint8_t enabled;
173b7579f77SDag-Erling Smørgrav };
174b7579f77SDag-Erling Smørgrav 
175b7579f77SDag-Erling Smørgrav /**
176b7579f77SDag-Erling Smørgrav  * Internal signal structure, to store signal event in.
177b7579f77SDag-Erling Smørgrav  */
178b7579f77SDag-Erling Smørgrav struct internal_signal {
179e2d15004SDag-Erling Smørgrav 	/** ub_event event type */
180e2d15004SDag-Erling Smørgrav 	struct ub_event* ev;
181b7579f77SDag-Erling Smørgrav 	/** next in signal list */
182b7579f77SDag-Erling Smørgrav 	struct internal_signal* next;
183b7579f77SDag-Erling Smørgrav };
184b7579f77SDag-Erling Smørgrav 
185b7579f77SDag-Erling Smørgrav /** create a tcp handler with a parent */
186b7579f77SDag-Erling Smørgrav static struct comm_point* comm_point_create_tcp_handler(
187b7579f77SDag-Erling Smørgrav 	struct comm_base *base, struct comm_point* parent, size_t bufsize,
188e86b9096SDag-Erling Smørgrav 	struct sldns_buffer* spoolbuf, comm_point_callback_type* callback,
1895469a995SCy Schubert 	void* callback_arg, struct unbound_socket* socket);
190b7579f77SDag-Erling Smørgrav 
191b7579f77SDag-Erling Smørgrav /* -------- End of local definitions -------- */
192b7579f77SDag-Erling Smørgrav 
193b7579f77SDag-Erling Smørgrav struct comm_base*
194b7579f77SDag-Erling Smørgrav comm_base_create(int sigs)
195b7579f77SDag-Erling Smørgrav {
196b7579f77SDag-Erling Smørgrav 	struct comm_base* b = (struct comm_base*)calloc(1,
197b7579f77SDag-Erling Smørgrav 		sizeof(struct comm_base));
198e2d15004SDag-Erling Smørgrav 	const char *evnm="event", *evsys="", *evmethod="";
199e2d15004SDag-Erling Smørgrav 
200b7579f77SDag-Erling Smørgrav 	if(!b)
201b7579f77SDag-Erling Smørgrav 		return NULL;
202b7579f77SDag-Erling Smørgrav 	b->eb = (struct internal_base*)calloc(1, sizeof(struct internal_base));
203b7579f77SDag-Erling Smørgrav 	if(!b->eb) {
204b7579f77SDag-Erling Smørgrav 		free(b);
205b7579f77SDag-Erling Smørgrav 		return NULL;
206b7579f77SDag-Erling Smørgrav 	}
207e2d15004SDag-Erling Smørgrav 	b->eb->base = ub_default_event_base(sigs, &b->eb->secs, &b->eb->now);
208b7579f77SDag-Erling Smørgrav 	if(!b->eb->base) {
209b7579f77SDag-Erling Smørgrav 		free(b->eb);
210b7579f77SDag-Erling Smørgrav 		free(b);
211b7579f77SDag-Erling Smørgrav 		return NULL;
212b7579f77SDag-Erling Smørgrav 	}
213e2d15004SDag-Erling Smørgrav 	ub_comm_base_now(b);
214e2d15004SDag-Erling Smørgrav 	ub_get_event_sys(b->eb->base, &evnm, &evsys, &evmethod);
215e86b9096SDag-Erling Smørgrav 	verbose(VERB_ALGO, "%s %s uses %s method.", evnm, evsys, evmethod);
216b7579f77SDag-Erling Smørgrav 	return b;
217b7579f77SDag-Erling Smørgrav }
218b7579f77SDag-Erling Smørgrav 
21917d15b25SDag-Erling Smørgrav struct comm_base*
220e2d15004SDag-Erling Smørgrav comm_base_create_event(struct ub_event_base* base)
22117d15b25SDag-Erling Smørgrav {
22217d15b25SDag-Erling Smørgrav 	struct comm_base* b = (struct comm_base*)calloc(1,
22317d15b25SDag-Erling Smørgrav 		sizeof(struct comm_base));
22417d15b25SDag-Erling Smørgrav 	if(!b)
22517d15b25SDag-Erling Smørgrav 		return NULL;
22617d15b25SDag-Erling Smørgrav 	b->eb = (struct internal_base*)calloc(1, sizeof(struct internal_base));
22717d15b25SDag-Erling Smørgrav 	if(!b->eb) {
22817d15b25SDag-Erling Smørgrav 		free(b);
22917d15b25SDag-Erling Smørgrav 		return NULL;
23017d15b25SDag-Erling Smørgrav 	}
23117d15b25SDag-Erling Smørgrav 	b->eb->base = base;
232e2d15004SDag-Erling Smørgrav 	ub_comm_base_now(b);
23317d15b25SDag-Erling Smørgrav 	return b;
23417d15b25SDag-Erling Smørgrav }
23517d15b25SDag-Erling Smørgrav 
236b7579f77SDag-Erling Smørgrav void
237b7579f77SDag-Erling Smørgrav comm_base_delete(struct comm_base* b)
238b7579f77SDag-Erling Smørgrav {
239b7579f77SDag-Erling Smørgrav 	if(!b)
240b7579f77SDag-Erling Smørgrav 		return;
241b7579f77SDag-Erling Smørgrav 	if(b->eb->slow_accept_enabled) {
242e2d15004SDag-Erling Smørgrav 		if(ub_event_del(b->eb->slow_accept) != 0) {
243b7579f77SDag-Erling Smørgrav 			log_err("could not event_del slow_accept");
244b7579f77SDag-Erling Smørgrav 		}
245e2d15004SDag-Erling Smørgrav 		ub_event_free(b->eb->slow_accept);
246b7579f77SDag-Erling Smørgrav 	}
247e2d15004SDag-Erling Smørgrav 	ub_event_base_free(b->eb->base);
248b7579f77SDag-Erling Smørgrav 	b->eb->base = NULL;
249b7579f77SDag-Erling Smørgrav 	free(b->eb);
250b7579f77SDag-Erling Smørgrav 	free(b);
251b7579f77SDag-Erling Smørgrav }
252b7579f77SDag-Erling Smørgrav 
253b7579f77SDag-Erling Smørgrav void
25417d15b25SDag-Erling Smørgrav comm_base_delete_no_base(struct comm_base* b)
25517d15b25SDag-Erling Smørgrav {
25617d15b25SDag-Erling Smørgrav 	if(!b)
25717d15b25SDag-Erling Smørgrav 		return;
25817d15b25SDag-Erling Smørgrav 	if(b->eb->slow_accept_enabled) {
259e2d15004SDag-Erling Smørgrav 		if(ub_event_del(b->eb->slow_accept) != 0) {
26017d15b25SDag-Erling Smørgrav 			log_err("could not event_del slow_accept");
26117d15b25SDag-Erling Smørgrav 		}
262e2d15004SDag-Erling Smørgrav 		ub_event_free(b->eb->slow_accept);
26317d15b25SDag-Erling Smørgrav 	}
26417d15b25SDag-Erling Smørgrav 	b->eb->base = NULL;
26517d15b25SDag-Erling Smørgrav 	free(b->eb);
26617d15b25SDag-Erling Smørgrav 	free(b);
26717d15b25SDag-Erling Smørgrav }
26817d15b25SDag-Erling Smørgrav 
26917d15b25SDag-Erling Smørgrav void
27017d15b25SDag-Erling Smørgrav comm_base_timept(struct comm_base* b, time_t** tt, struct timeval** tv)
271b7579f77SDag-Erling Smørgrav {
272b7579f77SDag-Erling Smørgrav 	*tt = &b->eb->secs;
273b7579f77SDag-Erling Smørgrav 	*tv = &b->eb->now;
274b7579f77SDag-Erling Smørgrav }
275b7579f77SDag-Erling Smørgrav 
276b7579f77SDag-Erling Smørgrav void
277b7579f77SDag-Erling Smørgrav comm_base_dispatch(struct comm_base* b)
278b7579f77SDag-Erling Smørgrav {
279b7579f77SDag-Erling Smørgrav 	int retval;
280e2d15004SDag-Erling Smørgrav 	retval = ub_event_base_dispatch(b->eb->base);
281e2d15004SDag-Erling Smørgrav 	if(retval < 0) {
282b7579f77SDag-Erling Smørgrav 		fatal_exit("event_dispatch returned error %d, "
283b7579f77SDag-Erling Smørgrav 			"errno is %s", retval, strerror(errno));
284b7579f77SDag-Erling Smørgrav 	}
285b7579f77SDag-Erling Smørgrav }
286b7579f77SDag-Erling Smørgrav 
287b7579f77SDag-Erling Smørgrav void comm_base_exit(struct comm_base* b)
288b7579f77SDag-Erling Smørgrav {
289e2d15004SDag-Erling Smørgrav 	if(ub_event_base_loopexit(b->eb->base) != 0) {
290b7579f77SDag-Erling Smørgrav 		log_err("Could not loopexit");
291b7579f77SDag-Erling Smørgrav 	}
292b7579f77SDag-Erling Smørgrav }
293b7579f77SDag-Erling Smørgrav 
294b7579f77SDag-Erling Smørgrav void comm_base_set_slow_accept_handlers(struct comm_base* b,
295b7579f77SDag-Erling Smørgrav 	void (*stop_acc)(void*), void (*start_acc)(void*), void* arg)
296b7579f77SDag-Erling Smørgrav {
297b7579f77SDag-Erling Smørgrav 	b->stop_accept = stop_acc;
298b7579f77SDag-Erling Smørgrav 	b->start_accept = start_acc;
299b7579f77SDag-Erling Smørgrav 	b->cb_arg = arg;
300b7579f77SDag-Erling Smørgrav }
301b7579f77SDag-Erling Smørgrav 
302e2d15004SDag-Erling Smørgrav struct ub_event_base* comm_base_internal(struct comm_base* b)
303b7579f77SDag-Erling Smørgrav {
304b7579f77SDag-Erling Smørgrav 	return b->eb->base;
305b7579f77SDag-Erling Smørgrav }
306b7579f77SDag-Erling Smørgrav 
307b7579f77SDag-Erling Smørgrav /** see if errno for udp has to be logged or not uses globals */
308b7579f77SDag-Erling Smørgrav static int
309b7579f77SDag-Erling Smørgrav udp_send_errno_needs_log(struct sockaddr* addr, socklen_t addrlen)
310b7579f77SDag-Erling Smørgrav {
311b7579f77SDag-Erling Smørgrav 	/* do not log transient errors (unless high verbosity) */
312b7579f77SDag-Erling Smørgrav #if defined(ENETUNREACH) || defined(EHOSTDOWN) || defined(EHOSTUNREACH) || defined(ENETDOWN)
313b7579f77SDag-Erling Smørgrav 	switch(errno) {
314b7579f77SDag-Erling Smørgrav #  ifdef ENETUNREACH
315b7579f77SDag-Erling Smørgrav 		case ENETUNREACH:
316b7579f77SDag-Erling Smørgrav #  endif
317b7579f77SDag-Erling Smørgrav #  ifdef EHOSTDOWN
318b7579f77SDag-Erling Smørgrav 		case EHOSTDOWN:
319b7579f77SDag-Erling Smørgrav #  endif
320b7579f77SDag-Erling Smørgrav #  ifdef EHOSTUNREACH
321b7579f77SDag-Erling Smørgrav 		case EHOSTUNREACH:
322b7579f77SDag-Erling Smørgrav #  endif
323b7579f77SDag-Erling Smørgrav #  ifdef ENETDOWN
324b7579f77SDag-Erling Smørgrav 		case ENETDOWN:
325b7579f77SDag-Erling Smørgrav #  endif
3265469a995SCy Schubert 		case EPERM:
32724e36522SCy Schubert 		case EACCES:
328b7579f77SDag-Erling Smørgrav 			if(verbosity < VERB_ALGO)
329b7579f77SDag-Erling Smørgrav 				return 0;
330b7579f77SDag-Erling Smørgrav 		default:
331b7579f77SDag-Erling Smørgrav 			break;
332b7579f77SDag-Erling Smørgrav 	}
333b7579f77SDag-Erling Smørgrav #endif
33417d15b25SDag-Erling Smørgrav 	/* permission denied is gotten for every send if the
33517d15b25SDag-Erling Smørgrav 	 * network is disconnected (on some OS), squelch it */
336f61ef7f6SDag-Erling Smørgrav 	if( ((errno == EPERM)
337f61ef7f6SDag-Erling Smørgrav #  ifdef EADDRNOTAVAIL
338f61ef7f6SDag-Erling Smørgrav 		/* 'Cannot assign requested address' also when disconnected */
339f61ef7f6SDag-Erling Smørgrav 		|| (errno == EADDRNOTAVAIL)
340f61ef7f6SDag-Erling Smørgrav #  endif
3415469a995SCy Schubert 		) && verbosity < VERB_ALGO)
34217d15b25SDag-Erling Smørgrav 		return 0;
34357bddd21SDag-Erling Smørgrav #  ifdef EADDRINUSE
34457bddd21SDag-Erling Smørgrav 	/* If SO_REUSEADDR is set, we could try to connect to the same server
34557bddd21SDag-Erling Smørgrav 	 * from the same source port twice. */
34657bddd21SDag-Erling Smørgrav 	if(errno == EADDRINUSE && verbosity < VERB_DETAIL)
34757bddd21SDag-Erling Smørgrav 		return 0;
34857bddd21SDag-Erling Smørgrav #  endif
349b7579f77SDag-Erling Smørgrav 	/* squelch errors where people deploy AAAA ::ffff:bla for
350b7579f77SDag-Erling Smørgrav 	 * authority servers, which we try for intranets. */
351b7579f77SDag-Erling Smørgrav 	if(errno == EINVAL && addr_is_ip4mapped(
352b7579f77SDag-Erling Smørgrav 		(struct sockaddr_storage*)addr, addrlen) &&
353b7579f77SDag-Erling Smørgrav 		verbosity < VERB_DETAIL)
354b7579f77SDag-Erling Smørgrav 		return 0;
355b7579f77SDag-Erling Smørgrav 	/* SO_BROADCAST sockopt can give access to 255.255.255.255,
356b7579f77SDag-Erling Smørgrav 	 * but a dns cache does not need it. */
357b7579f77SDag-Erling Smørgrav 	if(errno == EACCES && addr_is_broadcast(
358b7579f77SDag-Erling Smørgrav 		(struct sockaddr_storage*)addr, addrlen) &&
359b7579f77SDag-Erling Smørgrav 		verbosity < VERB_DETAIL)
360b7579f77SDag-Erling Smørgrav 		return 0;
361b7579f77SDag-Erling Smørgrav 	return 1;
362b7579f77SDag-Erling Smørgrav }
363b7579f77SDag-Erling Smørgrav 
364b7579f77SDag-Erling Smørgrav int tcp_connect_errno_needs_log(struct sockaddr* addr, socklen_t addrlen)
365b7579f77SDag-Erling Smørgrav {
366b7579f77SDag-Erling Smørgrav 	return udp_send_errno_needs_log(addr, addrlen);
367b7579f77SDag-Erling Smørgrav }
368b7579f77SDag-Erling Smørgrav 
369b7579f77SDag-Erling Smørgrav /* send a UDP reply */
370b7579f77SDag-Erling Smørgrav int
37117d15b25SDag-Erling Smørgrav comm_point_send_udp_msg(struct comm_point *c, sldns_buffer* packet,
3727341cb0cSXin LI 	struct sockaddr* addr, socklen_t addrlen, int is_connected)
373b7579f77SDag-Erling Smørgrav {
374b7579f77SDag-Erling Smørgrav 	ssize_t sent;
375b7579f77SDag-Erling Smørgrav 	log_assert(c->fd != -1);
376b7579f77SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG
37717d15b25SDag-Erling Smørgrav 	if(sldns_buffer_remaining(packet) == 0)
378b7579f77SDag-Erling Smørgrav 		log_err("error: send empty UDP packet");
379b7579f77SDag-Erling Smørgrav #endif
380b7579f77SDag-Erling Smørgrav 	log_assert(addr && addrlen > 0);
3817341cb0cSXin LI 	if(!is_connected) {
38217d15b25SDag-Erling Smørgrav 		sent = sendto(c->fd, (void*)sldns_buffer_begin(packet),
38317d15b25SDag-Erling Smørgrav 			sldns_buffer_remaining(packet), 0,
384b7579f77SDag-Erling Smørgrav 			addr, addrlen);
385369c6923SCy Schubert 	} else {
386369c6923SCy Schubert 		sent = send(c->fd, (void*)sldns_buffer_begin(packet),
387369c6923SCy Schubert 			sldns_buffer_remaining(packet), 0);
388369c6923SCy Schubert 	}
389b7579f77SDag-Erling Smørgrav 	if(sent == -1) {
390f61ef7f6SDag-Erling Smørgrav 		/* try again and block, waiting for IO to complete,
391f61ef7f6SDag-Erling Smørgrav 		 * we want to send the answer, and we will wait for
392f61ef7f6SDag-Erling Smørgrav 		 * the ethernet interface buffer to have space. */
393f61ef7f6SDag-Erling Smørgrav #ifndef USE_WINSOCK
394865f46b2SCy Schubert 		if(errno == EAGAIN || errno == EINTR ||
395f61ef7f6SDag-Erling Smørgrav #  ifdef EWOULDBLOCK
396f61ef7f6SDag-Erling Smørgrav 			errno == EWOULDBLOCK ||
397f61ef7f6SDag-Erling Smørgrav #  endif
398f61ef7f6SDag-Erling Smørgrav 			errno == ENOBUFS) {
399f61ef7f6SDag-Erling Smørgrav #else
400f61ef7f6SDag-Erling Smørgrav 		if(WSAGetLastError() == WSAEINPROGRESS ||
401865f46b2SCy Schubert 			WSAGetLastError() == WSAEINTR ||
402f61ef7f6SDag-Erling Smørgrav 			WSAGetLastError() == WSAENOBUFS ||
403f61ef7f6SDag-Erling Smørgrav 			WSAGetLastError() == WSAEWOULDBLOCK) {
404f61ef7f6SDag-Erling Smørgrav #endif
405865f46b2SCy Schubert 			/* if we set the fd blocking, other threads suddenly
406865f46b2SCy Schubert 			 * have a blocking fd that they operate on */
407865f46b2SCy Schubert 			while(sent == -1 && (
408865f46b2SCy Schubert #ifndef USE_WINSOCK
409865f46b2SCy Schubert 				errno == EAGAIN || errno == EINTR ||
410865f46b2SCy Schubert #  ifdef EWOULDBLOCK
411865f46b2SCy Schubert 				errno == EWOULDBLOCK ||
412865f46b2SCy Schubert #  endif
413865f46b2SCy Schubert 				errno == ENOBUFS
414865f46b2SCy Schubert #else
415865f46b2SCy Schubert 				WSAGetLastError() == WSAEINPROGRESS ||
416865f46b2SCy Schubert 				WSAGetLastError() == WSAEINTR ||
417865f46b2SCy Schubert 				WSAGetLastError() == WSAENOBUFS ||
418865f46b2SCy Schubert 				WSAGetLastError() == WSAEWOULDBLOCK
419865f46b2SCy Schubert #endif
420865f46b2SCy Schubert 			)) {
421865f46b2SCy Schubert #if defined(HAVE_POLL) || defined(USE_WINSOCK)
422865f46b2SCy Schubert 				struct pollfd p;
423865f46b2SCy Schubert 				int pret;
424865f46b2SCy Schubert 				memset(&p, 0, sizeof(p));
425865f46b2SCy Schubert 				p.fd = c->fd;
426865f46b2SCy Schubert 				p.events = POLLOUT | POLLERR | POLLHUP;
427865f46b2SCy Schubert #  ifndef USE_WINSOCK
428865f46b2SCy Schubert 				pret = poll(&p, 1, SEND_BLOCKED_WAIT_TIMEOUT);
429865f46b2SCy Schubert #  else
430865f46b2SCy Schubert 				pret = WSAPoll(&p, 1,
431865f46b2SCy Schubert 					SEND_BLOCKED_WAIT_TIMEOUT);
432865f46b2SCy Schubert #  endif
433865f46b2SCy Schubert 				if(pret == 0) {
434865f46b2SCy Schubert 					/* timer expired */
435865f46b2SCy Schubert 					struct comm_base* b = c->ev->base;
436865f46b2SCy Schubert 					if(b->eb->last_writewait_log+SLOW_LOG_TIME <=
437865f46b2SCy Schubert 						b->eb->secs) {
438865f46b2SCy Schubert 						b->eb->last_writewait_log = b->eb->secs;
439865f46b2SCy Schubert 						verbose(VERB_OPS, "send udp blocked "
440865f46b2SCy Schubert 							"for long, dropping packet.");
441865f46b2SCy Schubert 					}
442865f46b2SCy Schubert 					return 0;
443865f46b2SCy Schubert 				} else if(pret < 0 &&
444865f46b2SCy Schubert #ifndef USE_WINSOCK
445865f46b2SCy Schubert 					errno != EAGAIN && errno != EINTR &&
446865f46b2SCy Schubert #  ifdef EWOULDBLOCK
447865f46b2SCy Schubert 					errno != EWOULDBLOCK &&
448865f46b2SCy Schubert #  endif
449865f46b2SCy Schubert 					errno != ENOBUFS
450865f46b2SCy Schubert #else
451865f46b2SCy Schubert 					WSAGetLastError() != WSAEINPROGRESS &&
452865f46b2SCy Schubert 					WSAGetLastError() != WSAEINTR &&
453865f46b2SCy Schubert 					WSAGetLastError() != WSAENOBUFS &&
454865f46b2SCy Schubert 					WSAGetLastError() != WSAEWOULDBLOCK
455865f46b2SCy Schubert #endif
456865f46b2SCy Schubert 					) {
457865f46b2SCy Schubert 					log_err("poll udp out failed: %s",
458865f46b2SCy Schubert 						sock_strerror(errno));
459865f46b2SCy Schubert 					return 0;
460865f46b2SCy Schubert 				}
461865f46b2SCy Schubert #endif /* defined(HAVE_POLL) || defined(USE_WINSOCK) */
4627341cb0cSXin LI 				if (!is_connected) {
463f61ef7f6SDag-Erling Smørgrav 					sent = sendto(c->fd, (void*)sldns_buffer_begin(packet),
464f61ef7f6SDag-Erling Smørgrav 						sldns_buffer_remaining(packet), 0,
465f61ef7f6SDag-Erling Smørgrav 						addr, addrlen);
4667341cb0cSXin LI 				} else {
4677341cb0cSXin LI 					sent = send(c->fd, (void*)sldns_buffer_begin(packet),
4687341cb0cSXin LI 						sldns_buffer_remaining(packet), 0);
4697341cb0cSXin LI 				}
470865f46b2SCy Schubert 			}
471f61ef7f6SDag-Erling Smørgrav 		}
472f61ef7f6SDag-Erling Smørgrav 	}
473f61ef7f6SDag-Erling Smørgrav 	if(sent == -1) {
474b7579f77SDag-Erling Smørgrav 		if(!udp_send_errno_needs_log(addr, addrlen))
475b7579f77SDag-Erling Smørgrav 			return 0;
4767341cb0cSXin LI 		if (!is_connected) {
477c0caa2e2SCy Schubert 			verbose(VERB_OPS, "sendto failed: %s", sock_strerror(errno));
4787341cb0cSXin LI 		} else {
4797341cb0cSXin LI 			verbose(VERB_OPS, "send failed: %s", sock_strerror(errno));
4807341cb0cSXin LI 		}
481f44e67d1SCy Schubert 		if(addr)
482b7579f77SDag-Erling Smørgrav 			log_addr(VERB_OPS, "remote address is",
483b7579f77SDag-Erling Smørgrav 				(struct sockaddr_storage*)addr, addrlen);
484b7579f77SDag-Erling Smørgrav 		return 0;
48517d15b25SDag-Erling Smørgrav 	} else if((size_t)sent != sldns_buffer_remaining(packet)) {
486b7579f77SDag-Erling Smørgrav 		log_err("sent %d in place of %d bytes",
48717d15b25SDag-Erling Smørgrav 			(int)sent, (int)sldns_buffer_remaining(packet));
488b7579f77SDag-Erling Smørgrav 		return 0;
489b7579f77SDag-Erling Smørgrav 	}
490b7579f77SDag-Erling Smørgrav 	return 1;
491b7579f77SDag-Erling Smørgrav }
492b7579f77SDag-Erling Smørgrav 
493b7579f77SDag-Erling Smørgrav #if defined(AF_INET6) && defined(IPV6_PKTINFO) && (defined(HAVE_RECVMSG) || defined(HAVE_SENDMSG))
494b7579f77SDag-Erling Smørgrav /** print debug ancillary info */
495b7579f77SDag-Erling Smørgrav static void p_ancil(const char* str, struct comm_reply* r)
496b7579f77SDag-Erling Smørgrav {
497b7579f77SDag-Erling Smørgrav 	if(r->srctype != 4 && r->srctype != 6) {
498b7579f77SDag-Erling Smørgrav 		log_info("%s: unknown srctype %d", str, r->srctype);
499b7579f77SDag-Erling Smørgrav 		return;
500b7579f77SDag-Erling Smørgrav 	}
5015469a995SCy Schubert 
502b7579f77SDag-Erling Smørgrav 	if(r->srctype == 6) {
5035469a995SCy Schubert #ifdef IPV6_PKTINFO
504b7579f77SDag-Erling Smørgrav 		char buf[1024];
505b7579f77SDag-Erling Smørgrav 		if(inet_ntop(AF_INET6, &r->pktinfo.v6info.ipi6_addr,
506b7579f77SDag-Erling Smørgrav 			buf, (socklen_t)sizeof(buf)) == 0) {
50717d15b25SDag-Erling Smørgrav 			(void)strlcpy(buf, "(inet_ntop error)", sizeof(buf));
508b7579f77SDag-Erling Smørgrav 		}
509b7579f77SDag-Erling Smørgrav 		buf[sizeof(buf)-1]=0;
510b7579f77SDag-Erling Smørgrav 		log_info("%s: %s %d", str, buf, r->pktinfo.v6info.ipi6_ifindex);
5115469a995SCy Schubert #endif
512b7579f77SDag-Erling Smørgrav 	} else if(r->srctype == 4) {
513b7579f77SDag-Erling Smørgrav #ifdef IP_PKTINFO
514b7579f77SDag-Erling Smørgrav 		char buf1[1024], buf2[1024];
515b7579f77SDag-Erling Smørgrav 		if(inet_ntop(AF_INET, &r->pktinfo.v4info.ipi_addr,
516b7579f77SDag-Erling Smørgrav 			buf1, (socklen_t)sizeof(buf1)) == 0) {
51717d15b25SDag-Erling Smørgrav 			(void)strlcpy(buf1, "(inet_ntop error)", sizeof(buf1));
518b7579f77SDag-Erling Smørgrav 		}
519b7579f77SDag-Erling Smørgrav 		buf1[sizeof(buf1)-1]=0;
520b7579f77SDag-Erling Smørgrav #ifdef HAVE_STRUCT_IN_PKTINFO_IPI_SPEC_DST
521b7579f77SDag-Erling Smørgrav 		if(inet_ntop(AF_INET, &r->pktinfo.v4info.ipi_spec_dst,
522b7579f77SDag-Erling Smørgrav 			buf2, (socklen_t)sizeof(buf2)) == 0) {
52317d15b25SDag-Erling Smørgrav 			(void)strlcpy(buf2, "(inet_ntop error)", sizeof(buf2));
524b7579f77SDag-Erling Smørgrav 		}
525b7579f77SDag-Erling Smørgrav 		buf2[sizeof(buf2)-1]=0;
526b7579f77SDag-Erling Smørgrav #else
527b7579f77SDag-Erling Smørgrav 		buf2[0]=0;
528b7579f77SDag-Erling Smørgrav #endif
529b7579f77SDag-Erling Smørgrav 		log_info("%s: %d %s %s", str, r->pktinfo.v4info.ipi_ifindex,
530b7579f77SDag-Erling Smørgrav 			buf1, buf2);
531b7579f77SDag-Erling Smørgrav #elif defined(IP_RECVDSTADDR)
532b7579f77SDag-Erling Smørgrav 		char buf1[1024];
533b7579f77SDag-Erling Smørgrav 		if(inet_ntop(AF_INET, &r->pktinfo.v4addr,
534b7579f77SDag-Erling Smørgrav 			buf1, (socklen_t)sizeof(buf1)) == 0) {
53517d15b25SDag-Erling Smørgrav 			(void)strlcpy(buf1, "(inet_ntop error)", sizeof(buf1));
536b7579f77SDag-Erling Smørgrav 		}
537b7579f77SDag-Erling Smørgrav 		buf1[sizeof(buf1)-1]=0;
538b7579f77SDag-Erling Smørgrav 		log_info("%s: %s", str, buf1);
539b7579f77SDag-Erling Smørgrav #endif /* IP_PKTINFO or PI_RECVDSTDADDR */
540b7579f77SDag-Erling Smørgrav 	}
541b7579f77SDag-Erling Smørgrav }
542b7579f77SDag-Erling Smørgrav #endif /* AF_INET6 && IPV6_PKTINFO && HAVE_RECVMSG||HAVE_SENDMSG */
543b7579f77SDag-Erling Smørgrav 
544b7579f77SDag-Erling Smørgrav /** send a UDP reply over specified interface*/
545b7579f77SDag-Erling Smørgrav static int
54617d15b25SDag-Erling Smørgrav comm_point_send_udp_msg_if(struct comm_point *c, sldns_buffer* packet,
547b7579f77SDag-Erling Smørgrav 	struct sockaddr* addr, socklen_t addrlen, struct comm_reply* r)
548b7579f77SDag-Erling Smørgrav {
549b7579f77SDag-Erling Smørgrav #if defined(AF_INET6) && defined(IPV6_PKTINFO) && defined(HAVE_SENDMSG)
550b7579f77SDag-Erling Smørgrav 	ssize_t sent;
551b7579f77SDag-Erling Smørgrav 	struct msghdr msg;
552b7579f77SDag-Erling Smørgrav 	struct iovec iov[1];
55325039b37SCy Schubert 	union {
55425039b37SCy Schubert 		struct cmsghdr hdr;
55525039b37SCy Schubert 		char buf[256];
55625039b37SCy Schubert 	} control;
557b7579f77SDag-Erling Smørgrav #ifndef S_SPLINT_S
558b7579f77SDag-Erling Smørgrav 	struct cmsghdr *cmsg;
559b7579f77SDag-Erling Smørgrav #endif /* S_SPLINT_S */
560b7579f77SDag-Erling Smørgrav 
561b7579f77SDag-Erling Smørgrav 	log_assert(c->fd != -1);
562b7579f77SDag-Erling Smørgrav #ifdef UNBOUND_DEBUG
56317d15b25SDag-Erling Smørgrav 	if(sldns_buffer_remaining(packet) == 0)
564b7579f77SDag-Erling Smørgrav 		log_err("error: send empty UDP packet");
565b7579f77SDag-Erling Smørgrav #endif
566b7579f77SDag-Erling Smørgrav 	log_assert(addr && addrlen > 0);
567b7579f77SDag-Erling Smørgrav 
568b7579f77SDag-Erling Smørgrav 	msg.msg_name = addr;
569b7579f77SDag-Erling Smørgrav 	msg.msg_namelen = addrlen;
57017d15b25SDag-Erling Smørgrav 	iov[0].iov_base = sldns_buffer_begin(packet);
57117d15b25SDag-Erling Smørgrav 	iov[0].iov_len = sldns_buffer_remaining(packet);
572b7579f77SDag-Erling Smørgrav 	msg.msg_iov = iov;
573b7579f77SDag-Erling Smørgrav 	msg.msg_iovlen = 1;
57425039b37SCy Schubert 	msg.msg_control = control.buf;
575b7579f77SDag-Erling Smørgrav #ifndef S_SPLINT_S
57625039b37SCy Schubert 	msg.msg_controllen = sizeof(control.buf);
577b7579f77SDag-Erling Smørgrav #endif /* S_SPLINT_S */
578b7579f77SDag-Erling Smørgrav 	msg.msg_flags = 0;
579b7579f77SDag-Erling Smørgrav 
580b7579f77SDag-Erling Smørgrav #ifndef S_SPLINT_S
581b7579f77SDag-Erling Smørgrav 	cmsg = CMSG_FIRSTHDR(&msg);
582b7579f77SDag-Erling Smørgrav 	if(r->srctype == 4) {
583b7579f77SDag-Erling Smørgrav #ifdef IP_PKTINFO
58409a3aaf3SDag-Erling Smørgrav 		void* cmsg_data;
585b7579f77SDag-Erling Smørgrav 		msg.msg_controllen = CMSG_SPACE(sizeof(struct in_pktinfo));
58625039b37SCy Schubert 		log_assert(msg.msg_controllen <= sizeof(control.buf));
587b7579f77SDag-Erling Smørgrav 		cmsg->cmsg_level = IPPROTO_IP;
588b7579f77SDag-Erling Smørgrav 		cmsg->cmsg_type = IP_PKTINFO;
589b7579f77SDag-Erling Smørgrav 		memmove(CMSG_DATA(cmsg), &r->pktinfo.v4info,
590b7579f77SDag-Erling Smørgrav 			sizeof(struct in_pktinfo));
59109a3aaf3SDag-Erling Smørgrav 		/* unset the ifindex to not bypass the routing tables */
59209a3aaf3SDag-Erling Smørgrav 		cmsg_data = CMSG_DATA(cmsg);
59309a3aaf3SDag-Erling Smørgrav 		((struct in_pktinfo *) cmsg_data)->ipi_ifindex = 0;
594b7579f77SDag-Erling Smørgrav 		cmsg->cmsg_len = CMSG_LEN(sizeof(struct in_pktinfo));
595*8f76bb7dSCy Schubert 		/* zero the padding bytes inserted by the CMSG_LEN */
596*8f76bb7dSCy Schubert 		if(sizeof(struct in_pktinfo) < cmsg->cmsg_len)
597*8f76bb7dSCy Schubert 			memset(((uint8_t*)(CMSG_DATA(cmsg))) +
598*8f76bb7dSCy Schubert 				sizeof(struct in_pktinfo), 0, cmsg->cmsg_len
599*8f76bb7dSCy Schubert 				- sizeof(struct in_pktinfo));
600b7579f77SDag-Erling Smørgrav #elif defined(IP_SENDSRCADDR)
601b7579f77SDag-Erling Smørgrav 		msg.msg_controllen = CMSG_SPACE(sizeof(struct in_addr));
60225039b37SCy Schubert 		log_assert(msg.msg_controllen <= sizeof(control.buf));
603b7579f77SDag-Erling Smørgrav 		cmsg->cmsg_level = IPPROTO_IP;
604b7579f77SDag-Erling Smørgrav 		cmsg->cmsg_type = IP_SENDSRCADDR;
605b7579f77SDag-Erling Smørgrav 		memmove(CMSG_DATA(cmsg), &r->pktinfo.v4addr,
606b7579f77SDag-Erling Smørgrav 			sizeof(struct in_addr));
607b7579f77SDag-Erling Smørgrav 		cmsg->cmsg_len = CMSG_LEN(sizeof(struct in_addr));
608*8f76bb7dSCy Schubert 		/* zero the padding bytes inserted by the CMSG_LEN */
609*8f76bb7dSCy Schubert 		if(sizeof(struct in_addr) < cmsg->cmsg_len)
610*8f76bb7dSCy Schubert 			memset(((uint8_t*)(CMSG_DATA(cmsg))) +
611*8f76bb7dSCy Schubert 				sizeof(struct in_addr), 0, cmsg->cmsg_len
612*8f76bb7dSCy Schubert 				- sizeof(struct in_addr));
613b7579f77SDag-Erling Smørgrav #else
614b7579f77SDag-Erling Smørgrav 		verbose(VERB_ALGO, "no IP_PKTINFO or IP_SENDSRCADDR");
615b7579f77SDag-Erling Smørgrav 		msg.msg_control = NULL;
616b7579f77SDag-Erling Smørgrav #endif /* IP_PKTINFO or IP_SENDSRCADDR */
617b7579f77SDag-Erling Smørgrav 	} else if(r->srctype == 6) {
61809a3aaf3SDag-Erling Smørgrav 		void* cmsg_data;
619b7579f77SDag-Erling Smørgrav 		msg.msg_controllen = CMSG_SPACE(sizeof(struct in6_pktinfo));
62025039b37SCy Schubert 		log_assert(msg.msg_controllen <= sizeof(control.buf));
621b7579f77SDag-Erling Smørgrav 		cmsg->cmsg_level = IPPROTO_IPV6;
622b7579f77SDag-Erling Smørgrav 		cmsg->cmsg_type = IPV6_PKTINFO;
623b7579f77SDag-Erling Smørgrav 		memmove(CMSG_DATA(cmsg), &r->pktinfo.v6info,
624b7579f77SDag-Erling Smørgrav 			sizeof(struct in6_pktinfo));
62509a3aaf3SDag-Erling Smørgrav 		/* unset the ifindex to not bypass the routing tables */
62609a3aaf3SDag-Erling Smørgrav 		cmsg_data = CMSG_DATA(cmsg);
62709a3aaf3SDag-Erling Smørgrav 		((struct in6_pktinfo *) cmsg_data)->ipi6_ifindex = 0;
628b7579f77SDag-Erling Smørgrav 		cmsg->cmsg_len = CMSG_LEN(sizeof(struct in6_pktinfo));
629*8f76bb7dSCy Schubert 		/* zero the padding bytes inserted by the CMSG_LEN */
630*8f76bb7dSCy Schubert 		if(sizeof(struct in6_pktinfo) < cmsg->cmsg_len)
631*8f76bb7dSCy Schubert 			memset(((uint8_t*)(CMSG_DATA(cmsg))) +
632*8f76bb7dSCy Schubert 				sizeof(struct in6_pktinfo), 0, cmsg->cmsg_len
633*8f76bb7dSCy Schubert 				- sizeof(struct in6_pktinfo));
634b7579f77SDag-Erling Smørgrav 	} else {
635b7579f77SDag-Erling Smørgrav 		/* try to pass all 0 to use default route */
636b7579f77SDag-Erling Smørgrav 		msg.msg_controllen = CMSG_SPACE(sizeof(struct in6_pktinfo));
63725039b37SCy Schubert 		log_assert(msg.msg_controllen <= sizeof(control.buf));
638b7579f77SDag-Erling Smørgrav 		cmsg->cmsg_level = IPPROTO_IPV6;
639b7579f77SDag-Erling Smørgrav 		cmsg->cmsg_type = IPV6_PKTINFO;
640b7579f77SDag-Erling Smørgrav 		memset(CMSG_DATA(cmsg), 0, sizeof(struct in6_pktinfo));
641b7579f77SDag-Erling Smørgrav 		cmsg->cmsg_len = CMSG_LEN(sizeof(struct in6_pktinfo));
642*8f76bb7dSCy Schubert 		/* zero the padding bytes inserted by the CMSG_LEN */
643*8f76bb7dSCy Schubert 		if(sizeof(struct in6_pktinfo) < cmsg->cmsg_len)
644*8f76bb7dSCy Schubert 			memset(((uint8_t*)(CMSG_DATA(cmsg))) +
645*8f76bb7dSCy Schubert 				sizeof(struct in6_pktinfo), 0, cmsg->cmsg_len
646*8f76bb7dSCy Schubert 				- sizeof(struct in6_pktinfo));
647b7579f77SDag-Erling Smørgrav 	}
648b7579f77SDag-Erling Smørgrav #endif /* S_SPLINT_S */
649*8f76bb7dSCy Schubert 	if(verbosity >= VERB_ALGO && r->srctype != 0)
650b7579f77SDag-Erling Smørgrav 		p_ancil("send_udp over interface", r);
651b7579f77SDag-Erling Smørgrav 	sent = sendmsg(c->fd, &msg, 0);
652b7579f77SDag-Erling Smørgrav 	if(sent == -1) {
653f61ef7f6SDag-Erling Smørgrav 		/* try again and block, waiting for IO to complete,
654f61ef7f6SDag-Erling Smørgrav 		 * we want to send the answer, and we will wait for
655f61ef7f6SDag-Erling Smørgrav 		 * the ethernet interface buffer to have space. */
656f61ef7f6SDag-Erling Smørgrav #ifndef USE_WINSOCK
657865f46b2SCy Schubert 		if(errno == EAGAIN || errno == EINTR ||
658f61ef7f6SDag-Erling Smørgrav #  ifdef EWOULDBLOCK
659f61ef7f6SDag-Erling Smørgrav 			errno == EWOULDBLOCK ||
660f61ef7f6SDag-Erling Smørgrav #  endif
661f61ef7f6SDag-Erling Smørgrav 			errno == ENOBUFS) {
662f61ef7f6SDag-Erling Smørgrav #else
663f61ef7f6SDag-Erling Smørgrav 		if(WSAGetLastError() == WSAEINPROGRESS ||
664865f46b2SCy Schubert 			WSAGetLastError() == WSAEINTR ||
665f61ef7f6SDag-Erling Smørgrav 			WSAGetLastError() == WSAENOBUFS ||
666f61ef7f6SDag-Erling Smørgrav 			WSAGetLastError() == WSAEWOULDBLOCK) {
667f61ef7f6SDag-Erling Smørgrav #endif
668865f46b2SCy Schubert 			while(sent == -1 && (
669865f46b2SCy Schubert #ifndef USE_WINSOCK
670865f46b2SCy Schubert 				errno == EAGAIN || errno == EINTR ||
671865f46b2SCy Schubert #  ifdef EWOULDBLOCK
672865f46b2SCy Schubert 				errno == EWOULDBLOCK ||
673865f46b2SCy Schubert #  endif
674865f46b2SCy Schubert 				errno == ENOBUFS
675865f46b2SCy Schubert #else
676865f46b2SCy Schubert 				WSAGetLastError() == WSAEINPROGRESS ||
677865f46b2SCy Schubert 				WSAGetLastError() == WSAEINTR ||
678865f46b2SCy Schubert 				WSAGetLastError() == WSAENOBUFS ||
679865f46b2SCy Schubert 				WSAGetLastError() == WSAEWOULDBLOCK
680865f46b2SCy Schubert #endif
681865f46b2SCy Schubert 			)) {
682865f46b2SCy Schubert #if defined(HAVE_POLL) || defined(USE_WINSOCK)
683865f46b2SCy Schubert 				struct pollfd p;
684865f46b2SCy Schubert 				int pret;
685865f46b2SCy Schubert 				memset(&p, 0, sizeof(p));
686865f46b2SCy Schubert 				p.fd = c->fd;
687865f46b2SCy Schubert 				p.events = POLLOUT | POLLERR | POLLHUP;
688865f46b2SCy Schubert #  ifndef USE_WINSOCK
689865f46b2SCy Schubert 				pret = poll(&p, 1, SEND_BLOCKED_WAIT_TIMEOUT);
690865f46b2SCy Schubert #  else
691865f46b2SCy Schubert 				pret = WSAPoll(&p, 1,
692865f46b2SCy Schubert 					SEND_BLOCKED_WAIT_TIMEOUT);
693865f46b2SCy Schubert #  endif
694865f46b2SCy Schubert 				if(pret == 0) {
695865f46b2SCy Schubert 					/* timer expired */
696865f46b2SCy Schubert 					struct comm_base* b = c->ev->base;
697865f46b2SCy Schubert 					if(b->eb->last_writewait_log+SLOW_LOG_TIME <=
698865f46b2SCy Schubert 						b->eb->secs) {
699865f46b2SCy Schubert 						b->eb->last_writewait_log = b->eb->secs;
700865f46b2SCy Schubert 						verbose(VERB_OPS, "send udp blocked "
701865f46b2SCy Schubert 							"for long, dropping packet.");
702865f46b2SCy Schubert 					}
703865f46b2SCy Schubert 					return 0;
704865f46b2SCy Schubert 				} else if(pret < 0 &&
705865f46b2SCy Schubert #ifndef USE_WINSOCK
706865f46b2SCy Schubert 					errno != EAGAIN && errno != EINTR &&
707865f46b2SCy Schubert #  ifdef EWOULDBLOCK
708865f46b2SCy Schubert 					errno != EWOULDBLOCK &&
709865f46b2SCy Schubert #  endif
710865f46b2SCy Schubert 					errno != ENOBUFS
711865f46b2SCy Schubert #else
712865f46b2SCy Schubert 					WSAGetLastError() != WSAEINPROGRESS &&
713865f46b2SCy Schubert 					WSAGetLastError() != WSAEINTR &&
714865f46b2SCy Schubert 					WSAGetLastError() != WSAENOBUFS &&
715865f46b2SCy Schubert 					WSAGetLastError() != WSAEWOULDBLOCK
716865f46b2SCy Schubert #endif
717865f46b2SCy Schubert 					) {
718865f46b2SCy Schubert 					log_err("poll udp out failed: %s",
719865f46b2SCy Schubert 						sock_strerror(errno));
720865f46b2SCy Schubert 					return 0;
721865f46b2SCy Schubert 				}
722865f46b2SCy Schubert #endif /* defined(HAVE_POLL) || defined(USE_WINSOCK) */
723f61ef7f6SDag-Erling Smørgrav 				sent = sendmsg(c->fd, &msg, 0);
724865f46b2SCy Schubert 			}
725f61ef7f6SDag-Erling Smørgrav 		}
726f61ef7f6SDag-Erling Smørgrav 	}
727f61ef7f6SDag-Erling Smørgrav 	if(sent == -1) {
728b7579f77SDag-Erling Smørgrav 		if(!udp_send_errno_needs_log(addr, addrlen))
729b7579f77SDag-Erling Smørgrav 			return 0;
730b7579f77SDag-Erling Smørgrav 		verbose(VERB_OPS, "sendmsg failed: %s", strerror(errno));
731b7579f77SDag-Erling Smørgrav 		log_addr(VERB_OPS, "remote address is",
732b7579f77SDag-Erling Smørgrav 			(struct sockaddr_storage*)addr, addrlen);
733f61ef7f6SDag-Erling Smørgrav #ifdef __NetBSD__
734f61ef7f6SDag-Erling Smørgrav 		/* netbsd 7 has IP_PKTINFO for recv but not send */
735f61ef7f6SDag-Erling Smørgrav 		if(errno == EINVAL && r->srctype == 4)
736f61ef7f6SDag-Erling Smørgrav 			log_err("sendmsg: No support for sendmsg(IP_PKTINFO). "
737f61ef7f6SDag-Erling Smørgrav 				"Please disable interface-automatic");
738f61ef7f6SDag-Erling Smørgrav #endif
739b7579f77SDag-Erling Smørgrav 		return 0;
74017d15b25SDag-Erling Smørgrav 	} else if((size_t)sent != sldns_buffer_remaining(packet)) {
741b7579f77SDag-Erling Smørgrav 		log_err("sent %d in place of %d bytes",
74217d15b25SDag-Erling Smørgrav 			(int)sent, (int)sldns_buffer_remaining(packet));
743b7579f77SDag-Erling Smørgrav 		return 0;
744b7579f77SDag-Erling Smørgrav 	}
745b7579f77SDag-Erling Smørgrav 	return 1;
746b7579f77SDag-Erling Smørgrav #else
747b7579f77SDag-Erling Smørgrav 	(void)c;
748b7579f77SDag-Erling Smørgrav 	(void)packet;
749b7579f77SDag-Erling Smørgrav 	(void)addr;
750b7579f77SDag-Erling Smørgrav 	(void)addrlen;
751b7579f77SDag-Erling Smørgrav 	(void)r;
752b7579f77SDag-Erling Smørgrav 	log_err("sendmsg: IPV6_PKTINFO not supported");
753b7579f77SDag-Erling Smørgrav 	return 0;
754b7579f77SDag-Erling Smørgrav #endif /* AF_INET6 && IPV6_PKTINFO && HAVE_SENDMSG */
755b7579f77SDag-Erling Smørgrav }
756b7579f77SDag-Erling Smørgrav 
757369c6923SCy Schubert /** return true is UDP receive error needs to be logged */
758369c6923SCy Schubert static int udp_recv_needs_log(int err)
759369c6923SCy Schubert {
760369c6923SCy Schubert 	switch(err) {
761f44e67d1SCy Schubert 	case EACCES: /* some hosts send ICMP 'Permission Denied' */
762f44e67d1SCy Schubert #ifndef USE_WINSOCK
763369c6923SCy Schubert 	case ECONNREFUSED:
764369c6923SCy Schubert #  ifdef ENETUNREACH
765369c6923SCy Schubert 	case ENETUNREACH:
766369c6923SCy Schubert #  endif
767369c6923SCy Schubert #  ifdef EHOSTDOWN
768369c6923SCy Schubert 	case EHOSTDOWN:
769369c6923SCy Schubert #  endif
770369c6923SCy Schubert #  ifdef EHOSTUNREACH
771369c6923SCy Schubert 	case EHOSTUNREACH:
772369c6923SCy Schubert #  endif
773369c6923SCy Schubert #  ifdef ENETDOWN
774369c6923SCy Schubert 	case ENETDOWN:
775369c6923SCy Schubert #  endif
776f44e67d1SCy Schubert #else /* USE_WINSOCK */
777f44e67d1SCy Schubert 	case WSAECONNREFUSED:
778f44e67d1SCy Schubert 	case WSAENETUNREACH:
779f44e67d1SCy Schubert 	case WSAEHOSTDOWN:
780f44e67d1SCy Schubert 	case WSAEHOSTUNREACH:
781f44e67d1SCy Schubert 	case WSAENETDOWN:
782f44e67d1SCy Schubert #endif
783369c6923SCy Schubert 		if(verbosity >= VERB_ALGO)
784369c6923SCy Schubert 			return 1;
785369c6923SCy Schubert 		return 0;
786369c6923SCy Schubert 	default:
787369c6923SCy Schubert 		break;
788369c6923SCy Schubert 	}
789369c6923SCy Schubert 	return 1;
790369c6923SCy Schubert }
791369c6923SCy Schubert 
792865f46b2SCy Schubert /** Parses the PROXYv2 header from buf and updates the comm_reply struct.
793865f46b2SCy Schubert  *  Returns 1 on success, 0 on failure. */
794865f46b2SCy Schubert static int consume_pp2_header(struct sldns_buffer* buf, struct comm_reply* rep,
795865f46b2SCy Schubert 	int stream) {
796865f46b2SCy Schubert 	size_t size;
797865f46b2SCy Schubert 	struct pp2_header *header = pp2_read_header(buf);
798865f46b2SCy Schubert 	if(header == NULL) return 0;
799865f46b2SCy Schubert 	size = PP2_HEADER_SIZE + ntohs(header->len);
800865f46b2SCy Schubert 	if((header->ver_cmd & 0xF) == PP2_CMD_LOCAL) {
801865f46b2SCy Schubert 		/* A connection from the proxy itself.
802865f46b2SCy Schubert 		 * No need to do anything with addresses. */
803865f46b2SCy Schubert 		goto done;
804865f46b2SCy Schubert 	}
805865f46b2SCy Schubert 	if(header->fam_prot == 0x00) {
806865f46b2SCy Schubert 		/* Unspecified family and protocol. This could be used for
807865f46b2SCy Schubert 		 * health checks by proxies.
808865f46b2SCy Schubert 		 * No need to do anything with addresses. */
809865f46b2SCy Schubert 		goto done;
810865f46b2SCy Schubert 	}
811865f46b2SCy Schubert 	/* Read the proxied address */
812865f46b2SCy Schubert 	switch(header->fam_prot) {
813865f46b2SCy Schubert 		case 0x11: /* AF_INET|STREAM */
814865f46b2SCy Schubert 		case 0x12: /* AF_INET|DGRAM */
815865f46b2SCy Schubert 			{
816865f46b2SCy Schubert 			struct sockaddr_in* addr =
817865f46b2SCy Schubert 				(struct sockaddr_in*)&rep->client_addr;
818865f46b2SCy Schubert 			addr->sin_family = AF_INET;
819865f46b2SCy Schubert 			addr->sin_addr.s_addr = header->addr.addr4.src_addr;
820865f46b2SCy Schubert 			addr->sin_port = header->addr.addr4.src_port;
821865f46b2SCy Schubert 			rep->client_addrlen = (socklen_t)sizeof(struct sockaddr_in);
822865f46b2SCy Schubert 			}
823865f46b2SCy Schubert 			/* Ignore the destination address; it should be us. */
824865f46b2SCy Schubert 			break;
825865f46b2SCy Schubert 		case 0x21: /* AF_INET6|STREAM */
826865f46b2SCy Schubert 		case 0x22: /* AF_INET6|DGRAM */
827865f46b2SCy Schubert 			{
828865f46b2SCy Schubert 			struct sockaddr_in6* addr =
829865f46b2SCy Schubert 				(struct sockaddr_in6*)&rep->client_addr;
830865f46b2SCy Schubert 			memset(addr, 0, sizeof(*addr));
831865f46b2SCy Schubert 			addr->sin6_family = AF_INET6;
832865f46b2SCy Schubert 			memcpy(&addr->sin6_addr,
833865f46b2SCy Schubert 				header->addr.addr6.src_addr, 16);
834865f46b2SCy Schubert 			addr->sin6_port = header->addr.addr6.src_port;
835865f46b2SCy Schubert 			rep->client_addrlen = (socklen_t)sizeof(struct sockaddr_in6);
836865f46b2SCy Schubert 			}
837865f46b2SCy Schubert 			/* Ignore the destination address; it should be us. */
838865f46b2SCy Schubert 			break;
839865f46b2SCy Schubert 	}
840865f46b2SCy Schubert 	rep->is_proxied = 1;
841865f46b2SCy Schubert done:
842865f46b2SCy Schubert 	if(!stream) {
843865f46b2SCy Schubert 		/* We are reading a whole packet;
844865f46b2SCy Schubert 		 * Move the rest of the data to overwrite the PROXYv2 header */
845865f46b2SCy Schubert 		/* XXX can we do better to avoid memmove? */
8461838dec3SCy Schubert 		memmove(header, ((char*)header)+size,
847865f46b2SCy Schubert 			sldns_buffer_limit(buf)-size);
848865f46b2SCy Schubert 		sldns_buffer_set_limit(buf, sldns_buffer_limit(buf)-size);
849865f46b2SCy Schubert 	}
850865f46b2SCy Schubert 	return 1;
851865f46b2SCy Schubert }
852865f46b2SCy Schubert 
853b7579f77SDag-Erling Smørgrav void
854b7579f77SDag-Erling Smørgrav comm_point_udp_ancil_callback(int fd, short event, void* arg)
855b7579f77SDag-Erling Smørgrav {
856b7579f77SDag-Erling Smørgrav #if defined(AF_INET6) && defined(IPV6_PKTINFO) && defined(HAVE_RECVMSG)
857b7579f77SDag-Erling Smørgrav 	struct comm_reply rep;
858b7579f77SDag-Erling Smørgrav 	struct msghdr msg;
859b7579f77SDag-Erling Smørgrav 	struct iovec iov[1];
860b7579f77SDag-Erling Smørgrav 	ssize_t rcv;
86125039b37SCy Schubert 	union {
86225039b37SCy Schubert 		struct cmsghdr hdr;
86325039b37SCy Schubert 		char buf[256];
86425039b37SCy Schubert 	} ancil;
865b7579f77SDag-Erling Smørgrav 	int i;
866b7579f77SDag-Erling Smørgrav #ifndef S_SPLINT_S
867b7579f77SDag-Erling Smørgrav 	struct cmsghdr* cmsg;
868b7579f77SDag-Erling Smørgrav #endif /* S_SPLINT_S */
869*8f76bb7dSCy Schubert #ifdef HAVE_LINUX_NET_TSTAMP_H
870*8f76bb7dSCy Schubert 	struct timespec *ts;
871*8f76bb7dSCy Schubert #endif /* HAVE_LINUX_NET_TSTAMP_H */
872b7579f77SDag-Erling Smørgrav 
873b7579f77SDag-Erling Smørgrav 	rep.c = (struct comm_point*)arg;
874b7579f77SDag-Erling Smørgrav 	log_assert(rep.c->type == comm_udp);
875b7579f77SDag-Erling Smørgrav 
876e2d15004SDag-Erling Smørgrav 	if(!(event&UB_EV_READ))
877b7579f77SDag-Erling Smørgrav 		return;
878b7579f77SDag-Erling Smørgrav 	log_assert(rep.c && rep.c->buffer && rep.c->fd == fd);
879e2d15004SDag-Erling Smørgrav 	ub_comm_base_now(rep.c->ev->base);
880b7579f77SDag-Erling Smørgrav 	for(i=0; i<NUM_UDP_PER_SELECT; i++) {
88117d15b25SDag-Erling Smørgrav 		sldns_buffer_clear(rep.c->buffer);
882*8f76bb7dSCy Schubert 		timeval_clear(&rep.c->recv_tv);
883865f46b2SCy Schubert 		rep.remote_addrlen = (socklen_t)sizeof(rep.remote_addr);
884b7579f77SDag-Erling Smørgrav 		log_assert(fd != -1);
88517d15b25SDag-Erling Smørgrav 		log_assert(sldns_buffer_remaining(rep.c->buffer) > 0);
886865f46b2SCy Schubert 		msg.msg_name = &rep.remote_addr;
887865f46b2SCy Schubert 		msg.msg_namelen = (socklen_t)sizeof(rep.remote_addr);
88817d15b25SDag-Erling Smørgrav 		iov[0].iov_base = sldns_buffer_begin(rep.c->buffer);
88917d15b25SDag-Erling Smørgrav 		iov[0].iov_len = sldns_buffer_remaining(rep.c->buffer);
890b7579f77SDag-Erling Smørgrav 		msg.msg_iov = iov;
891b7579f77SDag-Erling Smørgrav 		msg.msg_iovlen = 1;
89225039b37SCy Schubert 		msg.msg_control = ancil.buf;
893b7579f77SDag-Erling Smørgrav #ifndef S_SPLINT_S
89425039b37SCy Schubert 		msg.msg_controllen = sizeof(ancil.buf);
895b7579f77SDag-Erling Smørgrav #endif /* S_SPLINT_S */
896b7579f77SDag-Erling Smørgrav 		msg.msg_flags = 0;
897865f46b2SCy Schubert 		rcv = recvmsg(fd, &msg, MSG_DONTWAIT);
898b7579f77SDag-Erling Smørgrav 		if(rcv == -1) {
899369c6923SCy Schubert 			if(errno != EAGAIN && errno != EINTR
900369c6923SCy Schubert 				&& udp_recv_needs_log(errno)) {
901b7579f77SDag-Erling Smørgrav 				log_err("recvmsg failed: %s", strerror(errno));
902b7579f77SDag-Erling Smørgrav 			}
903b7579f77SDag-Erling Smørgrav 			return;
904b7579f77SDag-Erling Smørgrav 		}
905865f46b2SCy Schubert 		rep.remote_addrlen = msg.msg_namelen;
90617d15b25SDag-Erling Smørgrav 		sldns_buffer_skip(rep.c->buffer, rcv);
90717d15b25SDag-Erling Smørgrav 		sldns_buffer_flip(rep.c->buffer);
908b7579f77SDag-Erling Smørgrav 		rep.srctype = 0;
909865f46b2SCy Schubert 		rep.is_proxied = 0;
910b7579f77SDag-Erling Smørgrav #ifndef S_SPLINT_S
911b7579f77SDag-Erling Smørgrav 		for(cmsg = CMSG_FIRSTHDR(&msg); cmsg != NULL;
912b7579f77SDag-Erling Smørgrav 			cmsg = CMSG_NXTHDR(&msg, cmsg)) {
913b7579f77SDag-Erling Smørgrav 			if( cmsg->cmsg_level == IPPROTO_IPV6 &&
914b7579f77SDag-Erling Smørgrav 				cmsg->cmsg_type == IPV6_PKTINFO) {
915b7579f77SDag-Erling Smørgrav 				rep.srctype = 6;
916b7579f77SDag-Erling Smørgrav 				memmove(&rep.pktinfo.v6info, CMSG_DATA(cmsg),
917b7579f77SDag-Erling Smørgrav 					sizeof(struct in6_pktinfo));
918b7579f77SDag-Erling Smørgrav 				break;
919b7579f77SDag-Erling Smørgrav #ifdef IP_PKTINFO
920b7579f77SDag-Erling Smørgrav 			} else if( cmsg->cmsg_level == IPPROTO_IP &&
921b7579f77SDag-Erling Smørgrav 				cmsg->cmsg_type == IP_PKTINFO) {
922b7579f77SDag-Erling Smørgrav 				rep.srctype = 4;
923b7579f77SDag-Erling Smørgrav 				memmove(&rep.pktinfo.v4info, CMSG_DATA(cmsg),
924b7579f77SDag-Erling Smørgrav 					sizeof(struct in_pktinfo));
925b7579f77SDag-Erling Smørgrav 				break;
926b7579f77SDag-Erling Smørgrav #elif defined(IP_RECVDSTADDR)
927b7579f77SDag-Erling Smørgrav 			} else if( cmsg->cmsg_level == IPPROTO_IP &&
928b7579f77SDag-Erling Smørgrav 				cmsg->cmsg_type == IP_RECVDSTADDR) {
929b7579f77SDag-Erling Smørgrav 				rep.srctype = 4;
930b7579f77SDag-Erling Smørgrav 				memmove(&rep.pktinfo.v4addr, CMSG_DATA(cmsg),
931b7579f77SDag-Erling Smørgrav 					sizeof(struct in_addr));
932b7579f77SDag-Erling Smørgrav 				break;
933b7579f77SDag-Erling Smørgrav #endif /* IP_PKTINFO or IP_RECVDSTADDR */
934*8f76bb7dSCy Schubert #ifdef HAVE_LINUX_NET_TSTAMP_H
935*8f76bb7dSCy Schubert 			} else if( cmsg->cmsg_level == SOL_SOCKET &&
936*8f76bb7dSCy Schubert 				cmsg->cmsg_type == SO_TIMESTAMPNS) {
937*8f76bb7dSCy Schubert 				ts = (struct timespec *)CMSG_DATA(cmsg);
938*8f76bb7dSCy Schubert 				TIMESPEC_TO_TIMEVAL(&rep.c->recv_tv, ts);
939*8f76bb7dSCy Schubert 			} else if( cmsg->cmsg_level == SOL_SOCKET &&
940*8f76bb7dSCy Schubert 				cmsg->cmsg_type == SO_TIMESTAMPING) {
941*8f76bb7dSCy Schubert 				ts = (struct timespec *)CMSG_DATA(cmsg);
942*8f76bb7dSCy Schubert 				TIMESPEC_TO_TIMEVAL(&rep.c->recv_tv, ts);
943*8f76bb7dSCy Schubert 			} else if( cmsg->cmsg_level == SOL_SOCKET &&
944*8f76bb7dSCy Schubert 				cmsg->cmsg_type == SO_TIMESTAMP) {
945*8f76bb7dSCy Schubert 				memmove(&rep.c->recv_tv, CMSG_DATA(cmsg), sizeof(struct timeval));
946*8f76bb7dSCy Schubert #endif /* HAVE_LINUX_NET_TSTAMP_H */
947b7579f77SDag-Erling Smørgrav 			}
948b7579f77SDag-Erling Smørgrav 		}
949*8f76bb7dSCy Schubert 
950*8f76bb7dSCy Schubert 		if(verbosity >= VERB_ALGO && rep.srctype != 0)
951b7579f77SDag-Erling Smørgrav 			p_ancil("receive_udp on interface", &rep);
952b7579f77SDag-Erling Smørgrav #endif /* S_SPLINT_S */
953865f46b2SCy Schubert 
954865f46b2SCy Schubert 		if(rep.c->pp2_enabled && !consume_pp2_header(rep.c->buffer,
955865f46b2SCy Schubert 			&rep, 0)) {
956865f46b2SCy Schubert 			log_err("proxy_protocol: could not consume PROXYv2 header");
957865f46b2SCy Schubert 			return;
958865f46b2SCy Schubert 		}
959865f46b2SCy Schubert 		if(!rep.is_proxied) {
960865f46b2SCy Schubert 			rep.client_addrlen = rep.remote_addrlen;
961865f46b2SCy Schubert 			memmove(&rep.client_addr, &rep.remote_addr,
962865f46b2SCy Schubert 				rep.remote_addrlen);
963865f46b2SCy Schubert 		}
964865f46b2SCy Schubert 
965b7579f77SDag-Erling Smørgrav 		fptr_ok(fptr_whitelist_comm_point(rep.c->callback));
966b7579f77SDag-Erling Smørgrav 		if((*rep.c->callback)(rep.c, rep.c->cb_arg, NETEVENT_NOERROR, &rep)) {
967b7579f77SDag-Erling Smørgrav 			/* send back immediate reply */
968b7579f77SDag-Erling Smørgrav 			(void)comm_point_send_udp_msg_if(rep.c, rep.c->buffer,
969865f46b2SCy Schubert 				(struct sockaddr*)&rep.remote_addr,
970865f46b2SCy Schubert 				rep.remote_addrlen, &rep);
971b7579f77SDag-Erling Smørgrav 		}
97257bddd21SDag-Erling Smørgrav 		if(!rep.c || rep.c->fd == -1) /* commpoint closed */
973b7579f77SDag-Erling Smørgrav 			break;
974b7579f77SDag-Erling Smørgrav 	}
975b7579f77SDag-Erling Smørgrav #else
976b7579f77SDag-Erling Smørgrav 	(void)fd;
977b7579f77SDag-Erling Smørgrav 	(void)event;
978b7579f77SDag-Erling Smørgrav 	(void)arg;
9793005e0a3SDag-Erling Smørgrav 	fatal_exit("recvmsg: No support for IPV6_PKTINFO; IP_PKTINFO or IP_RECVDSTADDR. "
980b7579f77SDag-Erling Smørgrav 		"Please disable interface-automatic");
981b7579f77SDag-Erling Smørgrav #endif /* AF_INET6 && IPV6_PKTINFO && HAVE_RECVMSG */
982b7579f77SDag-Erling Smørgrav }
983b7579f77SDag-Erling Smørgrav 
984b7579f77SDag-Erling Smørgrav void
985b7579f77SDag-Erling Smørgrav comm_point_udp_callback(int fd, short event, void* arg)
986b7579f77SDag-Erling Smørgrav {
987b7579f77SDag-Erling Smørgrav 	struct comm_reply rep;
988b7579f77SDag-Erling Smørgrav 	ssize_t rcv;
989b7579f77SDag-Erling Smørgrav 	int i;
99065b390aaSDag-Erling Smørgrav 	struct sldns_buffer *buffer;
991b7579f77SDag-Erling Smørgrav 
992b7579f77SDag-Erling Smørgrav 	rep.c = (struct comm_point*)arg;
993b7579f77SDag-Erling Smørgrav 	log_assert(rep.c->type == comm_udp);
994b7579f77SDag-Erling Smørgrav 
995e2d15004SDag-Erling Smørgrav 	if(!(event&UB_EV_READ))
996b7579f77SDag-Erling Smørgrav 		return;
997b7579f77SDag-Erling Smørgrav 	log_assert(rep.c && rep.c->buffer && rep.c->fd == fd);
998e2d15004SDag-Erling Smørgrav 	ub_comm_base_now(rep.c->ev->base);
999b7579f77SDag-Erling Smørgrav 	for(i=0; i<NUM_UDP_PER_SELECT; i++) {
100017d15b25SDag-Erling Smørgrav 		sldns_buffer_clear(rep.c->buffer);
1001865f46b2SCy Schubert 		rep.remote_addrlen = (socklen_t)sizeof(rep.remote_addr);
1002b7579f77SDag-Erling Smørgrav 		log_assert(fd != -1);
100317d15b25SDag-Erling Smørgrav 		log_assert(sldns_buffer_remaining(rep.c->buffer) > 0);
100417d15b25SDag-Erling Smørgrav 		rcv = recvfrom(fd, (void*)sldns_buffer_begin(rep.c->buffer),
1005865f46b2SCy Schubert 			sldns_buffer_remaining(rep.c->buffer), MSG_DONTWAIT,
1006865f46b2SCy Schubert 			(struct sockaddr*)&rep.remote_addr, &rep.remote_addrlen);
1007b7579f77SDag-Erling Smørgrav 		if(rcv == -1) {
1008b7579f77SDag-Erling Smørgrav #ifndef USE_WINSOCK
1009369c6923SCy Schubert 			if(errno != EAGAIN && errno != EINTR
1010369c6923SCy Schubert 				&& udp_recv_needs_log(errno))
1011b7579f77SDag-Erling Smørgrav 				log_err("recvfrom %d failed: %s",
1012b7579f77SDag-Erling Smørgrav 					fd, strerror(errno));
1013b7579f77SDag-Erling Smørgrav #else
1014b7579f77SDag-Erling Smørgrav 			if(WSAGetLastError() != WSAEINPROGRESS &&
1015b7579f77SDag-Erling Smørgrav 				WSAGetLastError() != WSAECONNRESET &&
1016f44e67d1SCy Schubert 				WSAGetLastError()!= WSAEWOULDBLOCK &&
1017f44e67d1SCy Schubert 				udp_recv_needs_log(WSAGetLastError()))
1018b7579f77SDag-Erling Smørgrav 				log_err("recvfrom failed: %s",
1019b7579f77SDag-Erling Smørgrav 					wsa_strerror(WSAGetLastError()));
1020b7579f77SDag-Erling Smørgrav #endif
1021b7579f77SDag-Erling Smørgrav 			return;
1022b7579f77SDag-Erling Smørgrav 		}
102317d15b25SDag-Erling Smørgrav 		sldns_buffer_skip(rep.c->buffer, rcv);
102417d15b25SDag-Erling Smørgrav 		sldns_buffer_flip(rep.c->buffer);
1025b7579f77SDag-Erling Smørgrav 		rep.srctype = 0;
1026865f46b2SCy Schubert 		rep.is_proxied = 0;
1027865f46b2SCy Schubert 
1028865f46b2SCy Schubert 		if(rep.c->pp2_enabled && !consume_pp2_header(rep.c->buffer,
1029865f46b2SCy Schubert 			&rep, 0)) {
1030865f46b2SCy Schubert 			log_err("proxy_protocol: could not consume PROXYv2 header");
1031865f46b2SCy Schubert 			return;
1032865f46b2SCy Schubert 		}
1033865f46b2SCy Schubert 		if(!rep.is_proxied) {
1034865f46b2SCy Schubert 			rep.client_addrlen = rep.remote_addrlen;
1035865f46b2SCy Schubert 			memmove(&rep.client_addr, &rep.remote_addr,
1036865f46b2SCy Schubert 				rep.remote_addrlen);
1037865f46b2SCy Schubert 		}
1038865f46b2SCy Schubert 
1039b7579f77SDag-Erling Smørgrav 		fptr_ok(fptr_whitelist_comm_point(rep.c->callback));
1040b7579f77SDag-Erling Smørgrav 		if((*rep.c->callback)(rep.c, rep.c->cb_arg, NETEVENT_NOERROR, &rep)) {
1041b7579f77SDag-Erling Smørgrav 			/* send back immediate reply */
104265b390aaSDag-Erling Smørgrav #ifdef USE_DNSCRYPT
104365b390aaSDag-Erling Smørgrav 			buffer = rep.c->dnscrypt_buffer;
104465b390aaSDag-Erling Smørgrav #else
104565b390aaSDag-Erling Smørgrav 			buffer = rep.c->buffer;
104665b390aaSDag-Erling Smørgrav #endif
104765b390aaSDag-Erling Smørgrav 			(void)comm_point_send_udp_msg(rep.c, buffer,
1048865f46b2SCy Schubert 				(struct sockaddr*)&rep.remote_addr,
1049865f46b2SCy Schubert 				rep.remote_addrlen, 0);
1050b7579f77SDag-Erling Smørgrav 		}
105157bddd21SDag-Erling Smørgrav 		if(!rep.c || rep.c->fd != fd) /* commpoint closed to -1 or reused for
1052b7579f77SDag-Erling Smørgrav 		another UDP port. Note rep.c cannot be reused with TCP fd. */
1053b7579f77SDag-Erling Smørgrav 			break;
1054b7579f77SDag-Erling Smørgrav 	}
1055b7579f77SDag-Erling Smørgrav }
1056b7579f77SDag-Erling Smørgrav 
1057f44e67d1SCy Schubert int adjusted_tcp_timeout(struct comm_point* c)
1058f44e67d1SCy Schubert {
1059f44e67d1SCy Schubert 	if(c->tcp_timeout_msec < TCP_QUERY_TIMEOUT_MINIMUM)
1060f44e67d1SCy Schubert 		return TCP_QUERY_TIMEOUT_MINIMUM;
1061f44e67d1SCy Schubert 	return c->tcp_timeout_msec;
1062f44e67d1SCy Schubert }
1063f44e67d1SCy Schubert 
1064b7579f77SDag-Erling Smørgrav /** Use a new tcp handler for new query fd, set to read query */
1065b7579f77SDag-Erling Smørgrav static void
1066b5663de9SDag-Erling Smørgrav setup_tcp_handler(struct comm_point* c, int fd, int cur, int max)
1067b7579f77SDag-Erling Smørgrav {
10684c75e3aaSDag-Erling Smørgrav 	int handler_usage;
1069c0caa2e2SCy Schubert 	log_assert(c->type == comm_tcp || c->type == comm_http);
1070b7579f77SDag-Erling Smørgrav 	log_assert(c->fd == -1);
107117d15b25SDag-Erling Smørgrav 	sldns_buffer_clear(c->buffer);
107265b390aaSDag-Erling Smørgrav #ifdef USE_DNSCRYPT
107365b390aaSDag-Erling Smørgrav 	if (c->dnscrypt)
107465b390aaSDag-Erling Smørgrav 		sldns_buffer_clear(c->dnscrypt_buffer);
107565b390aaSDag-Erling Smørgrav #endif
1076b7579f77SDag-Erling Smørgrav 	c->tcp_is_reading = 1;
1077b7579f77SDag-Erling Smørgrav 	c->tcp_byte_count = 0;
107824e36522SCy Schubert 	c->tcp_keepalive = 0;
1079b5663de9SDag-Erling Smørgrav 	/* if more than half the tcp handlers are in use, use a shorter
1080b5663de9SDag-Erling Smørgrav 	 * timeout for this TCP connection, we need to make space for
1081b5663de9SDag-Erling Smørgrav 	 * other connections to be able to get attention */
10824c75e3aaSDag-Erling Smørgrav 	/* If > 50% TCP handler structures in use, set timeout to 1/100th
10834c75e3aaSDag-Erling Smørgrav 	 * 	configured value.
10844c75e3aaSDag-Erling Smørgrav 	 * If > 65%TCP handler structures in use, set to 1/500th configured
10854c75e3aaSDag-Erling Smørgrav 	 * 	value.
10864c75e3aaSDag-Erling Smørgrav 	 * If > 80% TCP handler structures in use, set to 0.
10874c75e3aaSDag-Erling Smørgrav 	 *
10884c75e3aaSDag-Erling Smørgrav 	 * If the timeout to use falls below 200 milliseconds, an actual
10894c75e3aaSDag-Erling Smørgrav 	 * timeout of 200ms is used.
10904c75e3aaSDag-Erling Smørgrav 	 */
10914c75e3aaSDag-Erling Smørgrav 	handler_usage = (cur * 100) / max;
10924c75e3aaSDag-Erling Smørgrav 	if(handler_usage > 50 && handler_usage <= 65)
10934c75e3aaSDag-Erling Smørgrav 		c->tcp_timeout_msec /= 100;
10944c75e3aaSDag-Erling Smørgrav 	else if (handler_usage > 65 && handler_usage <= 80)
10954c75e3aaSDag-Erling Smørgrav 		c->tcp_timeout_msec /= 500;
10964c75e3aaSDag-Erling Smørgrav 	else if (handler_usage > 80)
10974c75e3aaSDag-Erling Smørgrav 		c->tcp_timeout_msec = 0;
1098f44e67d1SCy Schubert 	comm_point_start_listening(c, fd, adjusted_tcp_timeout(c));
1099b7579f77SDag-Erling Smørgrav }
1100b7579f77SDag-Erling Smørgrav 
1101b7579f77SDag-Erling Smørgrav void comm_base_handle_slow_accept(int ATTR_UNUSED(fd),
1102b7579f77SDag-Erling Smørgrav 	short ATTR_UNUSED(event), void* arg)
1103b7579f77SDag-Erling Smørgrav {
1104b7579f77SDag-Erling Smørgrav 	struct comm_base* b = (struct comm_base*)arg;
1105b7579f77SDag-Erling Smørgrav 	/* timeout for the slow accept, re-enable accepts again */
1106b7579f77SDag-Erling Smørgrav 	if(b->start_accept) {
1107b7579f77SDag-Erling Smørgrav 		verbose(VERB_ALGO, "wait is over, slow accept disabled");
1108b7579f77SDag-Erling Smørgrav 		fptr_ok(fptr_whitelist_start_accept(b->start_accept));
1109b7579f77SDag-Erling Smørgrav 		(*b->start_accept)(b->cb_arg);
1110b7579f77SDag-Erling Smørgrav 		b->eb->slow_accept_enabled = 0;
1111b7579f77SDag-Erling Smørgrav 	}
1112b7579f77SDag-Erling Smørgrav }
1113b7579f77SDag-Erling Smørgrav 
1114b7579f77SDag-Erling Smørgrav int comm_point_perform_accept(struct comm_point* c,
1115b7579f77SDag-Erling Smørgrav 	struct sockaddr_storage* addr, socklen_t* addrlen)
1116b7579f77SDag-Erling Smørgrav {
1117b7579f77SDag-Erling Smørgrav 	int new_fd;
1118b7579f77SDag-Erling Smørgrav 	*addrlen = (socklen_t)sizeof(*addr);
11193bd4df0aSDag-Erling Smørgrav #ifndef HAVE_ACCEPT4
1120b7579f77SDag-Erling Smørgrav 	new_fd = accept(c->fd, (struct sockaddr*)addr, addrlen);
11213bd4df0aSDag-Erling Smørgrav #else
11223bd4df0aSDag-Erling Smørgrav 	/* SOCK_NONBLOCK saves extra calls to fcntl for the same result */
11233bd4df0aSDag-Erling Smørgrav 	new_fd = accept4(c->fd, (struct sockaddr*)addr, addrlen, SOCK_NONBLOCK);
11243bd4df0aSDag-Erling Smørgrav #endif
1125b7579f77SDag-Erling Smørgrav 	if(new_fd == -1) {
1126b7579f77SDag-Erling Smørgrav #ifndef USE_WINSOCK
1127b7579f77SDag-Erling Smørgrav 		/* EINTR is signal interrupt. others are closed connection. */
1128b7579f77SDag-Erling Smørgrav 		if(	errno == EINTR || errno == EAGAIN
1129b7579f77SDag-Erling Smørgrav #ifdef EWOULDBLOCK
1130b7579f77SDag-Erling Smørgrav 			|| errno == EWOULDBLOCK
1131b7579f77SDag-Erling Smørgrav #endif
1132b7579f77SDag-Erling Smørgrav #ifdef ECONNABORTED
1133b7579f77SDag-Erling Smørgrav 			|| errno == ECONNABORTED
1134b7579f77SDag-Erling Smørgrav #endif
1135b7579f77SDag-Erling Smørgrav #ifdef EPROTO
1136b7579f77SDag-Erling Smørgrav 			|| errno == EPROTO
1137b7579f77SDag-Erling Smørgrav #endif /* EPROTO */
1138b7579f77SDag-Erling Smørgrav 			)
1139b7579f77SDag-Erling Smørgrav 			return -1;
1140b7579f77SDag-Erling Smørgrav #if defined(ENFILE) && defined(EMFILE)
1141b7579f77SDag-Erling Smørgrav 		if(errno == ENFILE || errno == EMFILE) {
1142b7579f77SDag-Erling Smørgrav 			/* out of file descriptors, likely outside of our
1143b7579f77SDag-Erling Smørgrav 			 * control. stop accept() calls for some time */
1144b7579f77SDag-Erling Smørgrav 			if(c->ev->base->stop_accept) {
1145b7579f77SDag-Erling Smørgrav 				struct comm_base* b = c->ev->base;
1146b7579f77SDag-Erling Smørgrav 				struct timeval tv;
1147b7579f77SDag-Erling Smørgrav 				verbose(VERB_ALGO, "out of file descriptors: "
1148b7579f77SDag-Erling Smørgrav 					"slow accept");
1149865f46b2SCy Schubert 				ub_comm_base_now(b);
1150865f46b2SCy Schubert 				if(b->eb->last_slow_log+SLOW_LOG_TIME <=
1151865f46b2SCy Schubert 					b->eb->secs) {
1152865f46b2SCy Schubert 					b->eb->last_slow_log = b->eb->secs;
1153865f46b2SCy Schubert 					verbose(VERB_OPS, "accept failed, "
1154865f46b2SCy Schubert 						"slow down accept for %d "
1155865f46b2SCy Schubert 						"msec: %s",
1156865f46b2SCy Schubert 						NETEVENT_SLOW_ACCEPT_TIME,
1157865f46b2SCy Schubert 						sock_strerror(errno));
1158865f46b2SCy Schubert 				}
1159b7579f77SDag-Erling Smørgrav 				b->eb->slow_accept_enabled = 1;
1160b7579f77SDag-Erling Smørgrav 				fptr_ok(fptr_whitelist_stop_accept(
1161b7579f77SDag-Erling Smørgrav 					b->stop_accept));
1162b7579f77SDag-Erling Smørgrav 				(*b->stop_accept)(b->cb_arg);
1163b7579f77SDag-Erling Smørgrav 				/* set timeout, no mallocs */
1164b7579f77SDag-Erling Smørgrav 				tv.tv_sec = NETEVENT_SLOW_ACCEPT_TIME/1000;
1165b5663de9SDag-Erling Smørgrav 				tv.tv_usec = (NETEVENT_SLOW_ACCEPT_TIME%1000)*1000;
1166e2d15004SDag-Erling Smørgrav 				b->eb->slow_accept = ub_event_new(b->eb->base,
1167e2d15004SDag-Erling Smørgrav 					-1, UB_EV_TIMEOUT,
1168b7579f77SDag-Erling Smørgrav 					comm_base_handle_slow_accept, b);
1169e2d15004SDag-Erling Smørgrav 				if(b->eb->slow_accept == NULL) {
1170b7579f77SDag-Erling Smørgrav 					/* we do not want to log here, because
1171b7579f77SDag-Erling Smørgrav 					 * that would spam the logfiles.
1172b7579f77SDag-Erling Smørgrav 					 * error: "event_base_set failed." */
1173b7579f77SDag-Erling Smørgrav 				}
1174e2d15004SDag-Erling Smørgrav 				else if(ub_event_add(b->eb->slow_accept, &tv)
1175e2d15004SDag-Erling Smørgrav 					!= 0) {
1176b7579f77SDag-Erling Smørgrav 					/* we do not want to log here,
1177b7579f77SDag-Erling Smørgrav 					 * error: "event_add failed." */
1178b7579f77SDag-Erling Smørgrav 				}
1179865f46b2SCy Schubert 			} else {
1180865f46b2SCy Schubert 				log_err("accept, with no slow down, "
1181865f46b2SCy Schubert 					"failed: %s", sock_strerror(errno));
1182b7579f77SDag-Erling Smørgrav 			}
1183b7579f77SDag-Erling Smørgrav 			return -1;
1184b7579f77SDag-Erling Smørgrav 		}
1185b7579f77SDag-Erling Smørgrav #endif
1186b7579f77SDag-Erling Smørgrav #else /* USE_WINSOCK */
1187b7579f77SDag-Erling Smørgrav 		if(WSAGetLastError() == WSAEINPROGRESS ||
1188b7579f77SDag-Erling Smørgrav 			WSAGetLastError() == WSAECONNRESET)
1189b7579f77SDag-Erling Smørgrav 			return -1;
1190b7579f77SDag-Erling Smørgrav 		if(WSAGetLastError() == WSAEWOULDBLOCK) {
1191e2d15004SDag-Erling Smørgrav 			ub_winsock_tcp_wouldblock(c->ev->ev, UB_EV_READ);
1192b7579f77SDag-Erling Smørgrav 			return -1;
1193b7579f77SDag-Erling Smørgrav 		}
1194b7579f77SDag-Erling Smørgrav #endif
1195c0caa2e2SCy Schubert 		log_err_addr("accept failed", sock_strerror(errno), addr,
1196c0caa2e2SCy Schubert 			*addrlen);
1197b7579f77SDag-Erling Smørgrav 		return -1;
1198b7579f77SDag-Erling Smørgrav 	}
11994c75e3aaSDag-Erling Smørgrav 	if(c->tcp_conn_limit && c->type == comm_tcp_accept) {
12004c75e3aaSDag-Erling Smørgrav 		c->tcl_addr = tcl_addr_lookup(c->tcp_conn_limit, addr, *addrlen);
12014c75e3aaSDag-Erling Smørgrav 		if(!tcl_new_connection(c->tcl_addr)) {
12024c75e3aaSDag-Erling Smørgrav 			if(verbosity >= 3)
12034c75e3aaSDag-Erling Smørgrav 				log_err_addr("accept rejected",
12044c75e3aaSDag-Erling Smørgrav 				"connection limit exceeded", addr, *addrlen);
12054c75e3aaSDag-Erling Smørgrav 			close(new_fd);
12064c75e3aaSDag-Erling Smørgrav 			return -1;
12074c75e3aaSDag-Erling Smørgrav 		}
12084c75e3aaSDag-Erling Smørgrav 	}
12093bd4df0aSDag-Erling Smørgrav #ifndef HAVE_ACCEPT4
1210b7579f77SDag-Erling Smørgrav 	fd_set_nonblock(new_fd);
12113bd4df0aSDag-Erling Smørgrav #endif
1212b7579f77SDag-Erling Smørgrav 	return new_fd;
1213b7579f77SDag-Erling Smørgrav }
1214b7579f77SDag-Erling Smørgrav 
1215b7579f77SDag-Erling Smørgrav #ifdef USE_WINSOCK
1216b7579f77SDag-Erling Smørgrav static long win_bio_cb(BIO *b, int oper, const char* ATTR_UNUSED(argp),
121724e36522SCy Schubert #ifdef HAVE_BIO_SET_CALLBACK_EX
121824e36522SCy Schubert 	size_t ATTR_UNUSED(len),
121924e36522SCy Schubert #endif
122024e36522SCy Schubert         int ATTR_UNUSED(argi), long argl,
122124e36522SCy Schubert #ifndef HAVE_BIO_SET_CALLBACK_EX
122224e36522SCy Schubert 	long retvalue
122324e36522SCy Schubert #else
122424e36522SCy Schubert 	int retvalue, size_t* ATTR_UNUSED(processed)
122524e36522SCy Schubert #endif
122624e36522SCy Schubert 	)
1227b7579f77SDag-Erling Smørgrav {
12283bd4df0aSDag-Erling Smørgrav 	int wsa_err = WSAGetLastError(); /* store errcode before it is gone */
1229b7579f77SDag-Erling Smørgrav 	verbose(VERB_ALGO, "bio_cb %d, %s %s %s", oper,
1230b7579f77SDag-Erling Smørgrav 		(oper&BIO_CB_RETURN)?"return":"before",
1231b7579f77SDag-Erling Smørgrav 		(oper&BIO_CB_READ)?"read":((oper&BIO_CB_WRITE)?"write":"other"),
12323bd4df0aSDag-Erling Smørgrav 		wsa_err==WSAEWOULDBLOCK?"wsawb":"");
1233b7579f77SDag-Erling Smørgrav 	/* on windows, check if previous operation caused EWOULDBLOCK */
1234b7579f77SDag-Erling Smørgrav 	if( (oper == (BIO_CB_READ|BIO_CB_RETURN) && argl == 0) ||
1235b7579f77SDag-Erling Smørgrav 		(oper == (BIO_CB_GETS|BIO_CB_RETURN) && argl == 0)) {
12363bd4df0aSDag-Erling Smørgrav 		if(wsa_err == WSAEWOULDBLOCK)
1237e2d15004SDag-Erling Smørgrav 			ub_winsock_tcp_wouldblock((struct ub_event*)
1238e2d15004SDag-Erling Smørgrav 				BIO_get_callback_arg(b), UB_EV_READ);
1239b7579f77SDag-Erling Smørgrav 	}
1240b7579f77SDag-Erling Smørgrav 	if( (oper == (BIO_CB_WRITE|BIO_CB_RETURN) && argl == 0) ||
1241b7579f77SDag-Erling Smørgrav 		(oper == (BIO_CB_PUTS|BIO_CB_RETURN) && argl == 0)) {
12423bd4df0aSDag-Erling Smørgrav 		if(wsa_err == WSAEWOULDBLOCK)
1243e2d15004SDag-Erling Smørgrav 			ub_winsock_tcp_wouldblock((struct ub_event*)
1244e2d15004SDag-Erling Smørgrav 				BIO_get_callback_arg(b), UB_EV_WRITE);
1245b7579f77SDag-Erling Smørgrav 	}
1246b7579f77SDag-Erling Smørgrav 	/* return original return value */
1247b7579f77SDag-Erling Smørgrav 	return retvalue;
1248b7579f77SDag-Erling Smørgrav }
1249b7579f77SDag-Erling Smørgrav 
1250b7579f77SDag-Erling Smørgrav /** set win bio callbacks for nonblocking operations */
1251b7579f77SDag-Erling Smørgrav void
1252b7579f77SDag-Erling Smørgrav comm_point_tcp_win_bio_cb(struct comm_point* c, void* thessl)
1253b7579f77SDag-Erling Smørgrav {
1254b7579f77SDag-Erling Smørgrav 	SSL* ssl = (SSL*)thessl;
1255b7579f77SDag-Erling Smørgrav 	/* set them both just in case, but usually they are the same BIO */
125624e36522SCy Schubert #ifdef HAVE_BIO_SET_CALLBACK_EX
125724e36522SCy Schubert 	BIO_set_callback_ex(SSL_get_rbio(ssl), &win_bio_cb);
125824e36522SCy Schubert #else
1259b7579f77SDag-Erling Smørgrav 	BIO_set_callback(SSL_get_rbio(ssl), &win_bio_cb);
126024e36522SCy Schubert #endif
1261e2d15004SDag-Erling Smørgrav 	BIO_set_callback_arg(SSL_get_rbio(ssl), (char*)c->ev->ev);
126224e36522SCy Schubert #ifdef HAVE_BIO_SET_CALLBACK_EX
126324e36522SCy Schubert 	BIO_set_callback_ex(SSL_get_wbio(ssl), &win_bio_cb);
126424e36522SCy Schubert #else
1265b7579f77SDag-Erling Smørgrav 	BIO_set_callback(SSL_get_wbio(ssl), &win_bio_cb);
126624e36522SCy Schubert #endif
1267e2d15004SDag-Erling Smørgrav 	BIO_set_callback_arg(SSL_get_wbio(ssl), (char*)c->ev->ev);
1268b7579f77SDag-Erling Smørgrav }
1269b7579f77SDag-Erling Smørgrav #endif
1270b7579f77SDag-Erling Smørgrav 
1271c0caa2e2SCy Schubert #ifdef HAVE_NGHTTP2
1272c0caa2e2SCy Schubert /** Create http2 session server.  Per connection, after TCP accepted.*/
1273c0caa2e2SCy Schubert static int http2_session_server_create(struct http2_session* h2_session)
1274c0caa2e2SCy Schubert {
1275c0caa2e2SCy Schubert 	log_assert(h2_session->callbacks);
1276c0caa2e2SCy Schubert 	h2_session->is_drop = 0;
1277c0caa2e2SCy Schubert 	if(nghttp2_session_server_new(&h2_session->session,
1278c0caa2e2SCy Schubert 			h2_session->callbacks,
1279c0caa2e2SCy Schubert 		h2_session) == NGHTTP2_ERR_NOMEM) {
1280c0caa2e2SCy Schubert 		log_err("failed to create nghttp2 session server");
1281c0caa2e2SCy Schubert 		return 0;
1282c0caa2e2SCy Schubert 	}
1283c0caa2e2SCy Schubert 
1284c0caa2e2SCy Schubert 	return 1;
1285c0caa2e2SCy Schubert }
1286c0caa2e2SCy Schubert 
1287c0caa2e2SCy Schubert /** Submit http2 setting to session. Once per session. */
1288c0caa2e2SCy Schubert static int http2_submit_settings(struct http2_session* h2_session)
1289c0caa2e2SCy Schubert {
1290c0caa2e2SCy Schubert 	int ret;
1291c0caa2e2SCy Schubert 	nghttp2_settings_entry settings[1] = {
1292c0caa2e2SCy Schubert 		{NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS,
1293c0caa2e2SCy Schubert 		 h2_session->c->http2_max_streams}};
1294c0caa2e2SCy Schubert 
1295c0caa2e2SCy Schubert 	ret = nghttp2_submit_settings(h2_session->session, NGHTTP2_FLAG_NONE,
1296c0caa2e2SCy Schubert 		settings, 1);
1297c0caa2e2SCy Schubert 	if(ret) {
1298c0caa2e2SCy Schubert 		verbose(VERB_QUERY, "http2: submit_settings failed, "
1299c0caa2e2SCy Schubert 			"error: %s", nghttp2_strerror(ret));
1300c0caa2e2SCy Schubert 		return 0;
1301c0caa2e2SCy Schubert 	}
1302c0caa2e2SCy Schubert 	return 1;
1303c0caa2e2SCy Schubert }
1304c0caa2e2SCy Schubert #endif /* HAVE_NGHTTP2 */
1305c0caa2e2SCy Schubert 
1306c0caa2e2SCy Schubert 
1307b7579f77SDag-Erling Smørgrav void
1308b7579f77SDag-Erling Smørgrav comm_point_tcp_accept_callback(int fd, short event, void* arg)
1309b7579f77SDag-Erling Smørgrav {
1310b7579f77SDag-Erling Smørgrav 	struct comm_point* c = (struct comm_point*)arg, *c_hdl;
1311b7579f77SDag-Erling Smørgrav 	int new_fd;
1312b7579f77SDag-Erling Smørgrav 	log_assert(c->type == comm_tcp_accept);
1313e2d15004SDag-Erling Smørgrav 	if(!(event & UB_EV_READ)) {
1314b7579f77SDag-Erling Smørgrav 		log_info("ignoring tcp accept event %d", (int)event);
1315b7579f77SDag-Erling Smørgrav 		return;
1316b7579f77SDag-Erling Smørgrav 	}
1317e2d15004SDag-Erling Smørgrav 	ub_comm_base_now(c->ev->base);
1318b7579f77SDag-Erling Smørgrav 	/* find free tcp handler. */
1319b7579f77SDag-Erling Smørgrav 	if(!c->tcp_free) {
1320b7579f77SDag-Erling Smørgrav 		log_warn("accepted too many tcp, connections full");
1321b7579f77SDag-Erling Smørgrav 		return;
1322b7579f77SDag-Erling Smørgrav 	}
1323b7579f77SDag-Erling Smørgrav 	/* accept incoming connection. */
1324b7579f77SDag-Erling Smørgrav 	c_hdl = c->tcp_free;
1325e86b9096SDag-Erling Smørgrav 	/* clear leftover flags from previous use, and then set the
1326e86b9096SDag-Erling Smørgrav 	 * correct event base for the event structure for libevent */
1327e86b9096SDag-Erling Smørgrav 	ub_event_free(c_hdl->ev->ev);
132824e36522SCy Schubert 	c_hdl->ev->ev = NULL;
1329369c6923SCy Schubert 	if((c_hdl->type == comm_tcp && c_hdl->tcp_req_info) ||
1330369c6923SCy Schubert 		c_hdl->type == comm_local || c_hdl->type == comm_raw)
1331369c6923SCy Schubert 		c_hdl->tcp_do_toggle_rw = 0;
1332369c6923SCy Schubert 	else	c_hdl->tcp_do_toggle_rw = 1;
1333c0caa2e2SCy Schubert 
1334c0caa2e2SCy Schubert 	if(c_hdl->type == comm_http) {
1335c0caa2e2SCy Schubert #ifdef HAVE_NGHTTP2
1336c0caa2e2SCy Schubert 		if(!c_hdl->h2_session ||
1337c0caa2e2SCy Schubert 			!http2_session_server_create(c_hdl->h2_session)) {
1338c0caa2e2SCy Schubert 			log_warn("failed to create nghttp2");
1339c0caa2e2SCy Schubert 			return;
1340c0caa2e2SCy Schubert 		}
1341c0caa2e2SCy Schubert 		if(!c_hdl->h2_session ||
1342c0caa2e2SCy Schubert 			!http2_submit_settings(c_hdl->h2_session)) {
1343c0caa2e2SCy Schubert 			log_warn("failed to submit http2 settings");
1344c0caa2e2SCy Schubert 			return;
1345c0caa2e2SCy Schubert 		}
1346369c6923SCy Schubert 		if(!c->ssl) {
1347369c6923SCy Schubert 			c_hdl->tcp_do_toggle_rw = 0;
1348369c6923SCy Schubert 			c_hdl->use_h2 = 1;
1349369c6923SCy Schubert 		}
1350c0caa2e2SCy Schubert #endif
1351c0caa2e2SCy Schubert 		c_hdl->ev->ev = ub_event_new(c_hdl->ev->base->eb->base, -1,
1352c0caa2e2SCy Schubert 			UB_EV_PERSIST | UB_EV_READ | UB_EV_TIMEOUT,
1353c0caa2e2SCy Schubert 			comm_point_http_handle_callback, c_hdl);
1354c0caa2e2SCy Schubert 	} else {
1355c0caa2e2SCy Schubert 		c_hdl->ev->ev = ub_event_new(c_hdl->ev->base->eb->base, -1,
1356c0caa2e2SCy Schubert 			UB_EV_PERSIST | UB_EV_READ | UB_EV_TIMEOUT,
1357c0caa2e2SCy Schubert 			comm_point_tcp_handle_callback, c_hdl);
1358c0caa2e2SCy Schubert 	}
1359e86b9096SDag-Erling Smørgrav 	if(!c_hdl->ev->ev) {
1360e86b9096SDag-Erling Smørgrav 		log_warn("could not ub_event_new, dropped tcp");
1361e86b9096SDag-Erling Smørgrav 		return;
1362e86b9096SDag-Erling Smørgrav 	}
1363b7579f77SDag-Erling Smørgrav 	log_assert(fd != -1);
1364b5663de9SDag-Erling Smørgrav 	(void)fd;
1365865f46b2SCy Schubert 	new_fd = comm_point_perform_accept(c, &c_hdl->repinfo.remote_addr,
1366865f46b2SCy Schubert 		&c_hdl->repinfo.remote_addrlen);
1367b7579f77SDag-Erling Smørgrav 	if(new_fd == -1)
1368b7579f77SDag-Erling Smørgrav 		return;
1369865f46b2SCy Schubert 	/* Copy remote_address to client_address.
1370865f46b2SCy Schubert 	 * Simplest way/time for streams to do that. */
1371865f46b2SCy Schubert 	c_hdl->repinfo.client_addrlen = c_hdl->repinfo.remote_addrlen;
1372865f46b2SCy Schubert 	memmove(&c_hdl->repinfo.client_addr,
1373865f46b2SCy Schubert 		&c_hdl->repinfo.remote_addr,
1374865f46b2SCy Schubert 		c_hdl->repinfo.remote_addrlen);
1375b7579f77SDag-Erling Smørgrav 	if(c->ssl) {
1376b7579f77SDag-Erling Smørgrav 		c_hdl->ssl = incoming_ssl_fd(c->ssl, new_fd);
1377b7579f77SDag-Erling Smørgrav 		if(!c_hdl->ssl) {
1378b7579f77SDag-Erling Smørgrav 			c_hdl->fd = new_fd;
1379b7579f77SDag-Erling Smørgrav 			comm_point_close(c_hdl);
1380b7579f77SDag-Erling Smørgrav 			return;
1381b7579f77SDag-Erling Smørgrav 		}
1382b7579f77SDag-Erling Smørgrav 		c_hdl->ssl_shake_state = comm_ssl_shake_read;
1383b7579f77SDag-Erling Smørgrav #ifdef USE_WINSOCK
1384b7579f77SDag-Erling Smørgrav 		comm_point_tcp_win_bio_cb(c_hdl, c_hdl->ssl);
1385b7579f77SDag-Erling Smørgrav #endif
1386b7579f77SDag-Erling Smørgrav 	}
1387b7579f77SDag-Erling Smørgrav 
1388b7579f77SDag-Erling Smørgrav 	/* grab the tcp handler buffers */
138909a3aaf3SDag-Erling Smørgrav 	c->cur_tcp_count++;
1390b7579f77SDag-Erling Smørgrav 	c->tcp_free = c_hdl->tcp_free;
139124e36522SCy Schubert 	c_hdl->tcp_free = NULL;
1392b7579f77SDag-Erling Smørgrav 	if(!c->tcp_free) {
1393b7579f77SDag-Erling Smørgrav 		/* stop accepting incoming queries for now. */
1394b7579f77SDag-Erling Smørgrav 		comm_point_stop_listening(c);
1395b7579f77SDag-Erling Smørgrav 	}
1396b5663de9SDag-Erling Smørgrav 	setup_tcp_handler(c_hdl, new_fd, c->cur_tcp_count, c->max_tcp_count);
1397b7579f77SDag-Erling Smørgrav }
1398b7579f77SDag-Erling Smørgrav 
1399b7579f77SDag-Erling Smørgrav /** Make tcp handler free for next assignment */
1400b7579f77SDag-Erling Smørgrav static void
1401b7579f77SDag-Erling Smørgrav reclaim_tcp_handler(struct comm_point* c)
1402b7579f77SDag-Erling Smørgrav {
1403b7579f77SDag-Erling Smørgrav 	log_assert(c->type == comm_tcp);
1404b7579f77SDag-Erling Smørgrav 	if(c->ssl) {
14058ed2b524SDag-Erling Smørgrav #ifdef HAVE_SSL
1406b7579f77SDag-Erling Smørgrav 		SSL_shutdown(c->ssl);
1407b7579f77SDag-Erling Smørgrav 		SSL_free(c->ssl);
1408b7579f77SDag-Erling Smørgrav 		c->ssl = NULL;
14098ed2b524SDag-Erling Smørgrav #endif
1410b7579f77SDag-Erling Smørgrav 	}
1411b7579f77SDag-Erling Smørgrav 	comm_point_close(c);
1412b7579f77SDag-Erling Smørgrav 	if(c->tcp_parent) {
141324e36522SCy Schubert 		if(c != c->tcp_parent->tcp_free) {
141409a3aaf3SDag-Erling Smørgrav 			c->tcp_parent->cur_tcp_count--;
1415b7579f77SDag-Erling Smørgrav 			c->tcp_free = c->tcp_parent->tcp_free;
1416b7579f77SDag-Erling Smørgrav 			c->tcp_parent->tcp_free = c;
141724e36522SCy Schubert 		}
1418b7579f77SDag-Erling Smørgrav 		if(!c->tcp_free) {
1419b7579f77SDag-Erling Smørgrav 			/* re-enable listening on accept socket */
1420b7579f77SDag-Erling Smørgrav 			comm_point_start_listening(c->tcp_parent, -1, -1);
1421b7579f77SDag-Erling Smørgrav 		}
1422b7579f77SDag-Erling Smørgrav 	}
1423369c6923SCy Schubert 	c->tcp_more_read_again = NULL;
1424369c6923SCy Schubert 	c->tcp_more_write_again = NULL;
14259cf5bc93SCy Schubert 	c->tcp_byte_count = 0;
1426865f46b2SCy Schubert 	c->pp2_header_state = pp2_header_none;
14279cf5bc93SCy Schubert 	sldns_buffer_clear(c->buffer);
1428b7579f77SDag-Erling Smørgrav }
1429b7579f77SDag-Erling Smørgrav 
1430b7579f77SDag-Erling Smørgrav /** do the callback when writing is done */
1431b7579f77SDag-Erling Smørgrav static void
1432b7579f77SDag-Erling Smørgrav tcp_callback_writer(struct comm_point* c)
1433b7579f77SDag-Erling Smørgrav {
1434b7579f77SDag-Erling Smørgrav 	log_assert(c->type == comm_tcp);
1435369c6923SCy Schubert 	if(!c->tcp_write_and_read) {
143617d15b25SDag-Erling Smørgrav 		sldns_buffer_clear(c->buffer);
1437369c6923SCy Schubert 		c->tcp_byte_count = 0;
1438369c6923SCy Schubert 	}
1439b7579f77SDag-Erling Smørgrav 	if(c->tcp_do_toggle_rw)
1440b7579f77SDag-Erling Smørgrav 		c->tcp_is_reading = 1;
1441b7579f77SDag-Erling Smørgrav 	/* switch from listening(write) to listening(read) */
1442e86b9096SDag-Erling Smørgrav 	if(c->tcp_req_info) {
1443e86b9096SDag-Erling Smørgrav 		tcp_req_info_handle_writedone(c->tcp_req_info);
1444e86b9096SDag-Erling Smørgrav 	} else {
1445b7579f77SDag-Erling Smørgrav 		comm_point_stop_listening(c);
1446369c6923SCy Schubert 		if(c->tcp_write_and_read) {
1447369c6923SCy Schubert 			fptr_ok(fptr_whitelist_comm_point(c->callback));
1448369c6923SCy Schubert 			if( (*c->callback)(c, c->cb_arg, NETEVENT_PKT_WRITTEN,
1449369c6923SCy Schubert 				&c->repinfo) ) {
1450369c6923SCy Schubert 				comm_point_start_listening(c, -1,
1451f44e67d1SCy Schubert 					adjusted_tcp_timeout(c));
1452369c6923SCy Schubert 			}
1453369c6923SCy Schubert 		} else {
1454f44e67d1SCy Schubert 			comm_point_start_listening(c, -1,
1455f44e67d1SCy Schubert 					adjusted_tcp_timeout(c));
1456b7579f77SDag-Erling Smørgrav 		}
1457e86b9096SDag-Erling Smørgrav 	}
1458369c6923SCy Schubert }
1459b7579f77SDag-Erling Smørgrav 
1460b7579f77SDag-Erling Smørgrav /** do the callback when reading is done */
1461b7579f77SDag-Erling Smørgrav static void
1462b7579f77SDag-Erling Smørgrav tcp_callback_reader(struct comm_point* c)
1463b7579f77SDag-Erling Smørgrav {
1464b7579f77SDag-Erling Smørgrav 	log_assert(c->type == comm_tcp || c->type == comm_local);
146517d15b25SDag-Erling Smørgrav 	sldns_buffer_flip(c->buffer);
1466b7579f77SDag-Erling Smørgrav 	if(c->tcp_do_toggle_rw)
1467b7579f77SDag-Erling Smørgrav 		c->tcp_is_reading = 0;
1468b7579f77SDag-Erling Smørgrav 	c->tcp_byte_count = 0;
1469e86b9096SDag-Erling Smørgrav 	if(c->tcp_req_info) {
1470e86b9096SDag-Erling Smørgrav 		tcp_req_info_handle_readdone(c->tcp_req_info);
1471e86b9096SDag-Erling Smørgrav 	} else {
1472b7579f77SDag-Erling Smørgrav 		if(c->type == comm_tcp)
1473b7579f77SDag-Erling Smørgrav 			comm_point_stop_listening(c);
1474b7579f77SDag-Erling Smørgrav 		fptr_ok(fptr_whitelist_comm_point(c->callback));
1475b7579f77SDag-Erling Smørgrav 		if( (*c->callback)(c, c->cb_arg, NETEVENT_NOERROR, &c->repinfo) ) {
1476f44e67d1SCy Schubert 			comm_point_start_listening(c, -1,
1477f44e67d1SCy Schubert 					adjusted_tcp_timeout(c));
1478b7579f77SDag-Erling Smørgrav 		}
1479b7579f77SDag-Erling Smørgrav 	}
1480e86b9096SDag-Erling Smørgrav }
1481b7579f77SDag-Erling Smørgrav 
148257bddd21SDag-Erling Smørgrav #ifdef HAVE_SSL
14830eefd307SCy Schubert /** true if the ssl handshake error has to be squelched from the logs */
148425039b37SCy Schubert int
14850eefd307SCy Schubert squelch_err_ssl_handshake(unsigned long err)
14860eefd307SCy Schubert {
14870eefd307SCy Schubert 	if(verbosity >= VERB_QUERY)
14880eefd307SCy Schubert 		return 0; /* only squelch on low verbosity */
1489a39a5a69SCy Schubert 	if(ERR_GET_LIB(err) == ERR_LIB_SSL &&
1490a39a5a69SCy Schubert 		(ERR_GET_REASON(err) == SSL_R_HTTPS_PROXY_REQUEST ||
1491a39a5a69SCy Schubert 		 ERR_GET_REASON(err) == SSL_R_HTTP_REQUEST ||
1492a39a5a69SCy Schubert 		 ERR_GET_REASON(err) == SSL_R_WRONG_VERSION_NUMBER ||
1493a39a5a69SCy Schubert 		 ERR_GET_REASON(err) == SSL_R_SSLV3_ALERT_BAD_CERTIFICATE
14940eefd307SCy Schubert #ifdef SSL_F_TLS_POST_PROCESS_CLIENT_HELLO
1495a39a5a69SCy Schubert 		 || ERR_GET_REASON(err) == SSL_R_NO_SHARED_CIPHER
14960eefd307SCy Schubert #endif
14970eefd307SCy Schubert #ifdef SSL_F_TLS_EARLY_POST_PROCESS_CLIENT_HELLO
1498a39a5a69SCy Schubert 		 || ERR_GET_REASON(err) == SSL_R_UNKNOWN_PROTOCOL
1499a39a5a69SCy Schubert 		 || ERR_GET_REASON(err) == SSL_R_UNSUPPORTED_PROTOCOL
15000eefd307SCy Schubert #  ifdef SSL_R_VERSION_TOO_LOW
1501a39a5a69SCy Schubert 		 || ERR_GET_REASON(err) == SSL_R_VERSION_TOO_LOW
15020eefd307SCy Schubert #  endif
15030eefd307SCy Schubert #endif
1504a39a5a69SCy Schubert 		))
15050eefd307SCy Schubert 		return 1;
15060eefd307SCy Schubert 	return 0;
15070eefd307SCy Schubert }
15080eefd307SCy Schubert #endif /* HAVE_SSL */
15090eefd307SCy Schubert 
1510b7579f77SDag-Erling Smørgrav /** continue ssl handshake */
15118ed2b524SDag-Erling Smørgrav #ifdef HAVE_SSL
1512b7579f77SDag-Erling Smørgrav static int
1513b7579f77SDag-Erling Smørgrav ssl_handshake(struct comm_point* c)
1514b7579f77SDag-Erling Smørgrav {
1515b7579f77SDag-Erling Smørgrav 	int r;
1516b7579f77SDag-Erling Smørgrav 	if(c->ssl_shake_state == comm_ssl_shake_hs_read) {
1517b7579f77SDag-Erling Smørgrav 		/* read condition satisfied back to writing */
15185469a995SCy Schubert 		comm_point_listen_for_rw(c, 0, 1);
1519b7579f77SDag-Erling Smørgrav 		c->ssl_shake_state = comm_ssl_shake_none;
1520b7579f77SDag-Erling Smørgrav 		return 1;
1521b7579f77SDag-Erling Smørgrav 	}
1522b7579f77SDag-Erling Smørgrav 	if(c->ssl_shake_state == comm_ssl_shake_hs_write) {
1523b7579f77SDag-Erling Smørgrav 		/* write condition satisfied, back to reading */
1524b7579f77SDag-Erling Smørgrav 		comm_point_listen_for_rw(c, 1, 0);
1525b7579f77SDag-Erling Smørgrav 		c->ssl_shake_state = comm_ssl_shake_none;
1526b7579f77SDag-Erling Smørgrav 		return 1;
1527b7579f77SDag-Erling Smørgrav 	}
1528b7579f77SDag-Erling Smørgrav 
1529b7579f77SDag-Erling Smørgrav 	ERR_clear_error();
1530b7579f77SDag-Erling Smørgrav 	r = SSL_do_handshake(c->ssl);
1531b7579f77SDag-Erling Smørgrav 	if(r != 1) {
1532b7579f77SDag-Erling Smørgrav 		int want = SSL_get_error(c->ssl, r);
1533b7579f77SDag-Erling Smørgrav 		if(want == SSL_ERROR_WANT_READ) {
1534b7579f77SDag-Erling Smørgrav 			if(c->ssl_shake_state == comm_ssl_shake_read)
1535b7579f77SDag-Erling Smørgrav 				return 1;
1536b7579f77SDag-Erling Smørgrav 			c->ssl_shake_state = comm_ssl_shake_read;
1537b7579f77SDag-Erling Smørgrav 			comm_point_listen_for_rw(c, 1, 0);
1538b7579f77SDag-Erling Smørgrav 			return 1;
1539b7579f77SDag-Erling Smørgrav 		} else if(want == SSL_ERROR_WANT_WRITE) {
1540b7579f77SDag-Erling Smørgrav 			if(c->ssl_shake_state == comm_ssl_shake_write)
1541b7579f77SDag-Erling Smørgrav 				return 1;
1542b7579f77SDag-Erling Smørgrav 			c->ssl_shake_state = comm_ssl_shake_write;
1543b7579f77SDag-Erling Smørgrav 			comm_point_listen_for_rw(c, 0, 1);
1544b7579f77SDag-Erling Smørgrav 			return 1;
1545b7579f77SDag-Erling Smørgrav 		} else if(r == 0) {
1546b7579f77SDag-Erling Smørgrav 			return 0; /* closed */
1547b7579f77SDag-Erling Smørgrav 		} else if(want == SSL_ERROR_SYSCALL) {
1548b7579f77SDag-Erling Smørgrav 			/* SYSCALL and errno==0 means closed uncleanly */
1549091e9e46SCy Schubert #ifdef EPIPE
1550091e9e46SCy Schubert 			if(errno == EPIPE && verbosity < 2)
1551091e9e46SCy Schubert 				return 0; /* silence 'broken pipe' */
1552091e9e46SCy Schubert #endif
1553091e9e46SCy Schubert #ifdef ECONNRESET
1554091e9e46SCy Schubert 			if(errno == ECONNRESET && verbosity < 2)
1555091e9e46SCy Schubert 				return 0; /* silence reset by peer */
1556091e9e46SCy Schubert #endif
1557a39a5a69SCy Schubert 			if(!tcp_connect_errno_needs_log(
1558865f46b2SCy Schubert 				(struct sockaddr*)&c->repinfo.remote_addr,
1559865f46b2SCy Schubert 				c->repinfo.remote_addrlen))
1560a39a5a69SCy Schubert 				return 0; /* silence connect failures that
1561a39a5a69SCy Schubert 				show up because after connect this is the
1562a39a5a69SCy Schubert 				first system call that accesses the socket */
1563b7579f77SDag-Erling Smørgrav 			if(errno != 0)
1564b7579f77SDag-Erling Smørgrav 				log_err("SSL_handshake syscall: %s",
1565b7579f77SDag-Erling Smørgrav 					strerror(errno));
1566b7579f77SDag-Erling Smørgrav 			return 0;
1567b7579f77SDag-Erling Smørgrav 		} else {
15680eefd307SCy Schubert 			unsigned long err = ERR_get_error();
15690eefd307SCy Schubert 			if(!squelch_err_ssl_handshake(err)) {
15700eefd307SCy Schubert 				log_crypto_err_code("ssl handshake failed", err);
1571865f46b2SCy Schubert 				log_addr(VERB_OPS, "ssl handshake failed",
1572865f46b2SCy Schubert 					&c->repinfo.remote_addr,
1573865f46b2SCy Schubert 					c->repinfo.remote_addrlen);
15740eefd307SCy Schubert 			}
1575b7579f77SDag-Erling Smørgrav 			return 0;
1576b7579f77SDag-Erling Smørgrav 		}
1577b7579f77SDag-Erling Smørgrav 	}
1578b7579f77SDag-Erling Smørgrav 	/* this is where peer verification could take place */
157957bddd21SDag-Erling Smørgrav 	if((SSL_get_verify_mode(c->ssl)&SSL_VERIFY_PEER)) {
158057bddd21SDag-Erling Smørgrav 		/* verification */
158157bddd21SDag-Erling Smørgrav 		if(SSL_get_verify_result(c->ssl) == X509_V_OK) {
15825469a995SCy Schubert #ifdef HAVE_SSL_GET1_PEER_CERTIFICATE
15835469a995SCy Schubert 			X509* x = SSL_get1_peer_certificate(c->ssl);
15845469a995SCy Schubert #else
158557bddd21SDag-Erling Smørgrav 			X509* x = SSL_get_peer_certificate(c->ssl);
15865469a995SCy Schubert #endif
158757bddd21SDag-Erling Smørgrav 			if(!x) {
158857bddd21SDag-Erling Smørgrav 				log_addr(VERB_ALGO, "SSL connection failed: "
158957bddd21SDag-Erling Smørgrav 					"no certificate",
1590865f46b2SCy Schubert 					&c->repinfo.remote_addr,
1591865f46b2SCy Schubert 					c->repinfo.remote_addrlen);
159257bddd21SDag-Erling Smørgrav 				return 0;
159357bddd21SDag-Erling Smørgrav 			}
159457bddd21SDag-Erling Smørgrav 			log_cert(VERB_ALGO, "peer certificate", x);
159557bddd21SDag-Erling Smørgrav #ifdef HAVE_SSL_GET0_PEERNAME
159657bddd21SDag-Erling Smørgrav 			if(SSL_get0_peername(c->ssl)) {
159757bddd21SDag-Erling Smørgrav 				char buf[255];
159857bddd21SDag-Erling Smørgrav 				snprintf(buf, sizeof(buf), "SSL connection "
159957bddd21SDag-Erling Smørgrav 					"to %s authenticated",
160057bddd21SDag-Erling Smørgrav 					SSL_get0_peername(c->ssl));
1601865f46b2SCy Schubert 				log_addr(VERB_ALGO, buf, &c->repinfo.remote_addr,
1602865f46b2SCy Schubert 					c->repinfo.remote_addrlen);
160357bddd21SDag-Erling Smørgrav 			} else {
160457bddd21SDag-Erling Smørgrav #endif
160557bddd21SDag-Erling Smørgrav 				log_addr(VERB_ALGO, "SSL connection "
1606865f46b2SCy Schubert 					"authenticated", &c->repinfo.remote_addr,
1607865f46b2SCy Schubert 					c->repinfo.remote_addrlen);
160857bddd21SDag-Erling Smørgrav #ifdef HAVE_SSL_GET0_PEERNAME
160957bddd21SDag-Erling Smørgrav 			}
161057bddd21SDag-Erling Smørgrav #endif
161157bddd21SDag-Erling Smørgrav 			X509_free(x);
161257bddd21SDag-Erling Smørgrav 		} else {
16135469a995SCy Schubert #ifdef HAVE_SSL_GET1_PEER_CERTIFICATE
16145469a995SCy Schubert 			X509* x = SSL_get1_peer_certificate(c->ssl);
16155469a995SCy Schubert #else
161657bddd21SDag-Erling Smørgrav 			X509* x = SSL_get_peer_certificate(c->ssl);
16175469a995SCy Schubert #endif
161857bddd21SDag-Erling Smørgrav 			if(x) {
161957bddd21SDag-Erling Smørgrav 				log_cert(VERB_ALGO, "peer certificate", x);
162057bddd21SDag-Erling Smørgrav 				X509_free(x);
162157bddd21SDag-Erling Smørgrav 			}
162257bddd21SDag-Erling Smørgrav 			log_addr(VERB_ALGO, "SSL connection failed: "
162357bddd21SDag-Erling Smørgrav 				"failed to authenticate",
1624865f46b2SCy Schubert 				&c->repinfo.remote_addr,
1625865f46b2SCy Schubert 				c->repinfo.remote_addrlen);
162657bddd21SDag-Erling Smørgrav 			return 0;
162757bddd21SDag-Erling Smørgrav 		}
162857bddd21SDag-Erling Smørgrav 	} else {
162957bddd21SDag-Erling Smørgrav 		/* unauthenticated, the verify peer flag was not set
163057bddd21SDag-Erling Smørgrav 		 * in c->ssl when the ssl object was created from ssl_ctx */
1631865f46b2SCy Schubert 		log_addr(VERB_ALGO, "SSL connection", &c->repinfo.remote_addr,
1632865f46b2SCy Schubert 			c->repinfo.remote_addrlen);
163357bddd21SDag-Erling Smørgrav 	}
1634b7579f77SDag-Erling Smørgrav 
16355469a995SCy Schubert #ifdef HAVE_SSL_GET0_ALPN_SELECTED
1636c0caa2e2SCy Schubert 	/* check if http2 use is negotiated */
1637c0caa2e2SCy Schubert 	if(c->type == comm_http && c->h2_session) {
1638c0caa2e2SCy Schubert 		const unsigned char *alpn;
1639c0caa2e2SCy Schubert 		unsigned int alpnlen = 0;
1640c0caa2e2SCy Schubert 		SSL_get0_alpn_selected(c->ssl, &alpn, &alpnlen);
1641c0caa2e2SCy Schubert 		if(alpnlen == 2 && memcmp("h2", alpn, 2) == 0) {
1642c0caa2e2SCy Schubert 			/* connection upgraded to HTTP2 */
1643c0caa2e2SCy Schubert 			c->tcp_do_toggle_rw = 0;
1644c0caa2e2SCy Schubert 			c->use_h2 = 1;
1645c0caa2e2SCy Schubert 		}
1646c0caa2e2SCy Schubert 	}
16475469a995SCy Schubert #endif
1648c0caa2e2SCy Schubert 
1649b7579f77SDag-Erling Smørgrav 	/* setup listen rw correctly */
1650b7579f77SDag-Erling Smørgrav 	if(c->tcp_is_reading) {
1651b7579f77SDag-Erling Smørgrav 		if(c->ssl_shake_state != comm_ssl_shake_read)
1652b7579f77SDag-Erling Smørgrav 			comm_point_listen_for_rw(c, 1, 0);
1653b7579f77SDag-Erling Smørgrav 	} else {
16545469a995SCy Schubert 		comm_point_listen_for_rw(c, 0, 1);
1655b7579f77SDag-Erling Smørgrav 	}
1656b7579f77SDag-Erling Smørgrav 	c->ssl_shake_state = comm_ssl_shake_none;
1657b7579f77SDag-Erling Smørgrav 	return 1;
1658b7579f77SDag-Erling Smørgrav }
16598ed2b524SDag-Erling Smørgrav #endif /* HAVE_SSL */
1660b7579f77SDag-Erling Smørgrav 
1661b7579f77SDag-Erling Smørgrav /** ssl read callback on TCP */
1662b7579f77SDag-Erling Smørgrav static int
1663b7579f77SDag-Erling Smørgrav ssl_handle_read(struct comm_point* c)
1664b7579f77SDag-Erling Smørgrav {
16658ed2b524SDag-Erling Smørgrav #ifdef HAVE_SSL
1666b7579f77SDag-Erling Smørgrav 	int r;
1667b7579f77SDag-Erling Smørgrav 	if(c->ssl_shake_state != comm_ssl_shake_none) {
1668b7579f77SDag-Erling Smørgrav 		if(!ssl_handshake(c))
1669b7579f77SDag-Erling Smørgrav 			return 0;
1670b7579f77SDag-Erling Smørgrav 		if(c->ssl_shake_state != comm_ssl_shake_none)
1671b7579f77SDag-Erling Smørgrav 			return 1;
1672b7579f77SDag-Erling Smørgrav 	}
1673865f46b2SCy Schubert 	if(c->pp2_enabled && c->pp2_header_state != pp2_header_done) {
1674865f46b2SCy Schubert 		struct pp2_header* header = NULL;
1675865f46b2SCy Schubert 		size_t want_read_size = 0;
1676865f46b2SCy Schubert 		size_t current_read_size = 0;
1677865f46b2SCy Schubert 		if(c->pp2_header_state == pp2_header_none) {
1678865f46b2SCy Schubert 			want_read_size = PP2_HEADER_SIZE;
1679865f46b2SCy Schubert 			if(sldns_buffer_remaining(c->buffer)<want_read_size) {
1680865f46b2SCy Schubert 				log_err_addr("proxy_protocol: not enough "
1681865f46b2SCy Schubert 					"buffer size to read PROXYv2 header", "",
1682865f46b2SCy Schubert 					&c->repinfo.remote_addr,
1683865f46b2SCy Schubert 					c->repinfo.remote_addrlen);
1684865f46b2SCy Schubert 				return 0;
1685865f46b2SCy Schubert 			}
1686865f46b2SCy Schubert 			verbose(VERB_ALGO, "proxy_protocol: reading fixed "
1687865f46b2SCy Schubert 				"part of PROXYv2 header (len %lu)",
1688865f46b2SCy Schubert 				(unsigned long)want_read_size);
1689865f46b2SCy Schubert 			current_read_size = want_read_size;
1690865f46b2SCy Schubert 			if(c->tcp_byte_count < current_read_size) {
1691865f46b2SCy Schubert 				ERR_clear_error();
1692865f46b2SCy Schubert 				if((r=SSL_read(c->ssl, (void*)sldns_buffer_at(
1693865f46b2SCy Schubert 					c->buffer, c->tcp_byte_count),
1694865f46b2SCy Schubert 					current_read_size -
1695865f46b2SCy Schubert 					c->tcp_byte_count)) <= 0) {
1696865f46b2SCy Schubert 					int want = SSL_get_error(c->ssl, r);
1697865f46b2SCy Schubert 					if(want == SSL_ERROR_ZERO_RETURN) {
1698865f46b2SCy Schubert 						if(c->tcp_req_info)
1699865f46b2SCy Schubert 							return tcp_req_info_handle_read_close(c->tcp_req_info);
1700865f46b2SCy Schubert 						return 0; /* shutdown, closed */
1701865f46b2SCy Schubert 					} else if(want == SSL_ERROR_WANT_READ) {
1702865f46b2SCy Schubert #ifdef USE_WINSOCK
1703865f46b2SCy Schubert 						ub_winsock_tcp_wouldblock(c->ev->ev, UB_EV_READ);
1704865f46b2SCy Schubert #endif
1705865f46b2SCy Schubert 						return 1; /* read more later */
1706865f46b2SCy Schubert 					} else if(want == SSL_ERROR_WANT_WRITE) {
1707865f46b2SCy Schubert 						c->ssl_shake_state = comm_ssl_shake_hs_write;
1708865f46b2SCy Schubert 						comm_point_listen_for_rw(c, 0, 1);
1709865f46b2SCy Schubert 						return 1;
1710865f46b2SCy Schubert 					} else if(want == SSL_ERROR_SYSCALL) {
1711865f46b2SCy Schubert #ifdef ECONNRESET
1712865f46b2SCy Schubert 						if(errno == ECONNRESET && verbosity < 2)
1713865f46b2SCy Schubert 							return 0; /* silence reset by peer */
1714865f46b2SCy Schubert #endif
1715865f46b2SCy Schubert 						if(errno != 0)
1716865f46b2SCy Schubert 							log_err("SSL_read syscall: %s",
1717865f46b2SCy Schubert 								strerror(errno));
1718865f46b2SCy Schubert 						return 0;
1719865f46b2SCy Schubert 					}
1720865f46b2SCy Schubert 					log_crypto_err("could not SSL_read");
1721865f46b2SCy Schubert 					return 0;
1722865f46b2SCy Schubert 				}
1723865f46b2SCy Schubert 				c->tcp_byte_count += r;
1724865f46b2SCy Schubert 				if(c->tcp_byte_count != current_read_size) return 1;
1725865f46b2SCy Schubert 				c->pp2_header_state = pp2_header_init;
1726865f46b2SCy Schubert 			}
1727865f46b2SCy Schubert 		}
1728865f46b2SCy Schubert 		if(c->pp2_header_state == pp2_header_init) {
1729865f46b2SCy Schubert 			header = pp2_read_header(c->buffer);
1730865f46b2SCy Schubert 			if(!header) {
1731865f46b2SCy Schubert 				log_err("proxy_protocol: could not parse "
1732865f46b2SCy Schubert 					"PROXYv2 header");
1733865f46b2SCy Schubert 				return 0;
1734865f46b2SCy Schubert 			}
1735865f46b2SCy Schubert 			want_read_size = ntohs(header->len);
1736865f46b2SCy Schubert 			if(sldns_buffer_remaining(c->buffer) <
1737865f46b2SCy Schubert 				PP2_HEADER_SIZE + want_read_size) {
1738865f46b2SCy Schubert 				log_err_addr("proxy_protocol: not enough "
1739865f46b2SCy Schubert 					"buffer size to read PROXYv2 header", "",
1740865f46b2SCy Schubert 					&c->repinfo.remote_addr,
1741865f46b2SCy Schubert 					c->repinfo.remote_addrlen);
1742865f46b2SCy Schubert 				return 0;
1743865f46b2SCy Schubert 			}
1744865f46b2SCy Schubert 			verbose(VERB_ALGO, "proxy_protocol: reading variable "
1745865f46b2SCy Schubert 				"part of PROXYv2 header (len %lu)",
1746865f46b2SCy Schubert 				(unsigned long)want_read_size);
1747865f46b2SCy Schubert 			current_read_size = PP2_HEADER_SIZE + want_read_size;
1748865f46b2SCy Schubert 			if(want_read_size == 0) {
1749865f46b2SCy Schubert 				/* nothing more to read; header is complete */
1750865f46b2SCy Schubert 				c->pp2_header_state = pp2_header_done;
1751865f46b2SCy Schubert 			} else if(c->tcp_byte_count < current_read_size) {
1752865f46b2SCy Schubert 				ERR_clear_error();
1753865f46b2SCy Schubert 				if((r=SSL_read(c->ssl, (void*)sldns_buffer_at(
1754865f46b2SCy Schubert 					c->buffer, c->tcp_byte_count),
1755865f46b2SCy Schubert 					current_read_size -
1756865f46b2SCy Schubert 					c->tcp_byte_count)) <= 0) {
1757865f46b2SCy Schubert 					int want = SSL_get_error(c->ssl, r);
1758865f46b2SCy Schubert 					if(want == SSL_ERROR_ZERO_RETURN) {
1759865f46b2SCy Schubert 						if(c->tcp_req_info)
1760865f46b2SCy Schubert 							return tcp_req_info_handle_read_close(c->tcp_req_info);
1761865f46b2SCy Schubert 						return 0; /* shutdown, closed */
1762865f46b2SCy Schubert 					} else if(want == SSL_ERROR_WANT_READ) {
1763865f46b2SCy Schubert #ifdef USE_WINSOCK
1764865f46b2SCy Schubert 						ub_winsock_tcp_wouldblock(c->ev->ev, UB_EV_READ);
1765865f46b2SCy Schubert #endif
1766865f46b2SCy Schubert 						return 1; /* read more later */
1767865f46b2SCy Schubert 					} else if(want == SSL_ERROR_WANT_WRITE) {
1768865f46b2SCy Schubert 						c->ssl_shake_state = comm_ssl_shake_hs_write;
1769865f46b2SCy Schubert 						comm_point_listen_for_rw(c, 0, 1);
1770865f46b2SCy Schubert 						return 1;
1771865f46b2SCy Schubert 					} else if(want == SSL_ERROR_SYSCALL) {
1772865f46b2SCy Schubert #ifdef ECONNRESET
1773865f46b2SCy Schubert 						if(errno == ECONNRESET && verbosity < 2)
1774865f46b2SCy Schubert 							return 0; /* silence reset by peer */
1775865f46b2SCy Schubert #endif
1776865f46b2SCy Schubert 						if(errno != 0)
1777865f46b2SCy Schubert 							log_err("SSL_read syscall: %s",
1778865f46b2SCy Schubert 								strerror(errno));
1779865f46b2SCy Schubert 						return 0;
1780865f46b2SCy Schubert 					}
1781865f46b2SCy Schubert 					log_crypto_err("could not SSL_read");
1782865f46b2SCy Schubert 					return 0;
1783865f46b2SCy Schubert 				}
1784865f46b2SCy Schubert 				c->tcp_byte_count += r;
1785865f46b2SCy Schubert 				if(c->tcp_byte_count != current_read_size) return 1;
1786865f46b2SCy Schubert 				c->pp2_header_state = pp2_header_done;
1787865f46b2SCy Schubert 			}
1788865f46b2SCy Schubert 		}
1789865f46b2SCy Schubert 		if(c->pp2_header_state != pp2_header_done || !header) {
1790865f46b2SCy Schubert 			log_err_addr("proxy_protocol: wrong state for the "
1791865f46b2SCy Schubert 				"PROXYv2 header", "", &c->repinfo.remote_addr,
1792865f46b2SCy Schubert 				c->repinfo.remote_addrlen);
1793865f46b2SCy Schubert 			return 0;
1794865f46b2SCy Schubert 		}
1795865f46b2SCy Schubert 		if(!consume_pp2_header(c->buffer, &c->repinfo, 1)) {
1796865f46b2SCy Schubert 			log_err_addr("proxy_protocol: could not consume "
1797865f46b2SCy Schubert 				"PROXYv2 header", "", &c->repinfo.remote_addr,
1798865f46b2SCy Schubert 				c->repinfo.remote_addrlen);
1799865f46b2SCy Schubert 			return 0;
1800865f46b2SCy Schubert 		}
1801865f46b2SCy Schubert 		verbose(VERB_ALGO, "proxy_protocol: successful read of "
1802865f46b2SCy Schubert 			"PROXYv2 header");
1803865f46b2SCy Schubert 		/* Clear and reset the buffer to read the following
1804865f46b2SCy Schubert 		 * DNS packet(s). */
1805865f46b2SCy Schubert 		sldns_buffer_clear(c->buffer);
1806865f46b2SCy Schubert 		c->tcp_byte_count = 0;
1807865f46b2SCy Schubert 		return 1;
1808865f46b2SCy Schubert 	}
1809b7579f77SDag-Erling Smørgrav 	if(c->tcp_byte_count < sizeof(uint16_t)) {
1810b7579f77SDag-Erling Smørgrav 		/* read length bytes */
1811b7579f77SDag-Erling Smørgrav 		ERR_clear_error();
181217d15b25SDag-Erling Smørgrav 		if((r=SSL_read(c->ssl, (void*)sldns_buffer_at(c->buffer,
1813b7579f77SDag-Erling Smørgrav 			c->tcp_byte_count), (int)(sizeof(uint16_t) -
1814b7579f77SDag-Erling Smørgrav 			c->tcp_byte_count))) <= 0) {
1815b7579f77SDag-Erling Smørgrav 			int want = SSL_get_error(c->ssl, r);
1816b7579f77SDag-Erling Smørgrav 			if(want == SSL_ERROR_ZERO_RETURN) {
1817e86b9096SDag-Erling Smørgrav 				if(c->tcp_req_info)
1818e86b9096SDag-Erling Smørgrav 					return tcp_req_info_handle_read_close(c->tcp_req_info);
1819b7579f77SDag-Erling Smørgrav 				return 0; /* shutdown, closed */
1820b7579f77SDag-Erling Smørgrav 			} else if(want == SSL_ERROR_WANT_READ) {
18215469a995SCy Schubert #ifdef USE_WINSOCK
18223bd4df0aSDag-Erling Smørgrav 				ub_winsock_tcp_wouldblock(c->ev->ev, UB_EV_READ);
18235469a995SCy Schubert #endif
1824b7579f77SDag-Erling Smørgrav 				return 1; /* read more later */
1825b7579f77SDag-Erling Smørgrav 			} else if(want == SSL_ERROR_WANT_WRITE) {
1826b7579f77SDag-Erling Smørgrav 				c->ssl_shake_state = comm_ssl_shake_hs_write;
1827b7579f77SDag-Erling Smørgrav 				comm_point_listen_for_rw(c, 0, 1);
1828b7579f77SDag-Erling Smørgrav 				return 1;
1829b7579f77SDag-Erling Smørgrav 			} else if(want == SSL_ERROR_SYSCALL) {
1830e86b9096SDag-Erling Smørgrav #ifdef ECONNRESET
1831e86b9096SDag-Erling Smørgrav 				if(errno == ECONNRESET && verbosity < 2)
1832e86b9096SDag-Erling Smørgrav 					return 0; /* silence reset by peer */
1833e86b9096SDag-Erling Smørgrav #endif
1834b7579f77SDag-Erling Smørgrav 				if(errno != 0)
1835b7579f77SDag-Erling Smørgrav 					log_err("SSL_read syscall: %s",
1836b7579f77SDag-Erling Smørgrav 						strerror(errno));
1837b7579f77SDag-Erling Smørgrav 				return 0;
1838b7579f77SDag-Erling Smørgrav 			}
1839b7579f77SDag-Erling Smørgrav 			log_crypto_err("could not SSL_read");
1840b7579f77SDag-Erling Smørgrav 			return 0;
1841b7579f77SDag-Erling Smørgrav 		}
1842b7579f77SDag-Erling Smørgrav 		c->tcp_byte_count += r;
18433bd4df0aSDag-Erling Smørgrav 		if(c->tcp_byte_count < sizeof(uint16_t))
1844b7579f77SDag-Erling Smørgrav 			return 1;
184517d15b25SDag-Erling Smørgrav 		if(sldns_buffer_read_u16_at(c->buffer, 0) >
184617d15b25SDag-Erling Smørgrav 			sldns_buffer_capacity(c->buffer)) {
1847b7579f77SDag-Erling Smørgrav 			verbose(VERB_QUERY, "ssl: dropped larger than buffer");
1848b7579f77SDag-Erling Smørgrav 			return 0;
1849b7579f77SDag-Erling Smørgrav 		}
185017d15b25SDag-Erling Smørgrav 		sldns_buffer_set_limit(c->buffer,
185117d15b25SDag-Erling Smørgrav 			sldns_buffer_read_u16_at(c->buffer, 0));
185217d15b25SDag-Erling Smørgrav 		if(sldns_buffer_limit(c->buffer) < LDNS_HEADER_SIZE) {
1853b7579f77SDag-Erling Smørgrav 			verbose(VERB_QUERY, "ssl: dropped bogus too short.");
1854b7579f77SDag-Erling Smørgrav 			return 0;
1855b7579f77SDag-Erling Smørgrav 		}
18563bd4df0aSDag-Erling Smørgrav 		sldns_buffer_skip(c->buffer, (ssize_t)(c->tcp_byte_count-sizeof(uint16_t)));
1857b7579f77SDag-Erling Smørgrav 		verbose(VERB_ALGO, "Reading ssl tcp query of length %d",
185817d15b25SDag-Erling Smørgrav 			(int)sldns_buffer_limit(c->buffer));
1859b7579f77SDag-Erling Smørgrav 	}
18603bd4df0aSDag-Erling Smørgrav 	if(sldns_buffer_remaining(c->buffer) > 0) {
1861b7579f77SDag-Erling Smørgrav 		ERR_clear_error();
186217d15b25SDag-Erling Smørgrav 		r = SSL_read(c->ssl, (void*)sldns_buffer_current(c->buffer),
186317d15b25SDag-Erling Smørgrav 			(int)sldns_buffer_remaining(c->buffer));
1864b7579f77SDag-Erling Smørgrav 		if(r <= 0) {
1865b7579f77SDag-Erling Smørgrav 			int want = SSL_get_error(c->ssl, r);
1866b7579f77SDag-Erling Smørgrav 			if(want == SSL_ERROR_ZERO_RETURN) {
1867e86b9096SDag-Erling Smørgrav 				if(c->tcp_req_info)
1868e86b9096SDag-Erling Smørgrav 					return tcp_req_info_handle_read_close(c->tcp_req_info);
1869b7579f77SDag-Erling Smørgrav 				return 0; /* shutdown, closed */
1870b7579f77SDag-Erling Smørgrav 			} else if(want == SSL_ERROR_WANT_READ) {
18715469a995SCy Schubert #ifdef USE_WINSOCK
18723bd4df0aSDag-Erling Smørgrav 				ub_winsock_tcp_wouldblock(c->ev->ev, UB_EV_READ);
18735469a995SCy Schubert #endif
1874b7579f77SDag-Erling Smørgrav 				return 1; /* read more later */
1875b7579f77SDag-Erling Smørgrav 			} else if(want == SSL_ERROR_WANT_WRITE) {
1876b7579f77SDag-Erling Smørgrav 				c->ssl_shake_state = comm_ssl_shake_hs_write;
1877b7579f77SDag-Erling Smørgrav 				comm_point_listen_for_rw(c, 0, 1);
1878b7579f77SDag-Erling Smørgrav 				return 1;
1879b7579f77SDag-Erling Smørgrav 			} else if(want == SSL_ERROR_SYSCALL) {
1880e86b9096SDag-Erling Smørgrav #ifdef ECONNRESET
1881e86b9096SDag-Erling Smørgrav 				if(errno == ECONNRESET && verbosity < 2)
1882e86b9096SDag-Erling Smørgrav 					return 0; /* silence reset by peer */
1883e86b9096SDag-Erling Smørgrav #endif
1884b7579f77SDag-Erling Smørgrav 				if(errno != 0)
1885b7579f77SDag-Erling Smørgrav 					log_err("SSL_read syscall: %s",
1886b7579f77SDag-Erling Smørgrav 						strerror(errno));
1887b7579f77SDag-Erling Smørgrav 				return 0;
1888b7579f77SDag-Erling Smørgrav 			}
1889b7579f77SDag-Erling Smørgrav 			log_crypto_err("could not SSL_read");
1890b7579f77SDag-Erling Smørgrav 			return 0;
1891b7579f77SDag-Erling Smørgrav 		}
189217d15b25SDag-Erling Smørgrav 		sldns_buffer_skip(c->buffer, (ssize_t)r);
18933bd4df0aSDag-Erling Smørgrav 	}
189417d15b25SDag-Erling Smørgrav 	if(sldns_buffer_remaining(c->buffer) <= 0) {
1895b7579f77SDag-Erling Smørgrav 		tcp_callback_reader(c);
1896b7579f77SDag-Erling Smørgrav 	}
1897b7579f77SDag-Erling Smørgrav 	return 1;
18988ed2b524SDag-Erling Smørgrav #else
18998ed2b524SDag-Erling Smørgrav 	(void)c;
19008ed2b524SDag-Erling Smørgrav 	return 0;
19018ed2b524SDag-Erling Smørgrav #endif /* HAVE_SSL */
1902b7579f77SDag-Erling Smørgrav }
1903b7579f77SDag-Erling Smørgrav 
1904b7579f77SDag-Erling Smørgrav /** ssl write callback on TCP */
1905b7579f77SDag-Erling Smørgrav static int
1906b7579f77SDag-Erling Smørgrav ssl_handle_write(struct comm_point* c)
1907b7579f77SDag-Erling Smørgrav {
19088ed2b524SDag-Erling Smørgrav #ifdef HAVE_SSL
1909b7579f77SDag-Erling Smørgrav 	int r;
1910b7579f77SDag-Erling Smørgrav 	if(c->ssl_shake_state != comm_ssl_shake_none) {
1911b7579f77SDag-Erling Smørgrav 		if(!ssl_handshake(c))
1912b7579f77SDag-Erling Smørgrav 			return 0;
1913b7579f77SDag-Erling Smørgrav 		if(c->ssl_shake_state != comm_ssl_shake_none)
1914b7579f77SDag-Erling Smørgrav 			return 1;
1915b7579f77SDag-Erling Smørgrav 	}
1916b7579f77SDag-Erling Smørgrav 	/* ignore return, if fails we may simply block */
19170eefd307SCy Schubert 	(void)SSL_set_mode(c->ssl, (long)SSL_MODE_ENABLE_PARTIAL_WRITE);
1918369c6923SCy Schubert 	if((c->tcp_write_and_read?c->tcp_write_byte_count:c->tcp_byte_count) < sizeof(uint16_t)) {
1919369c6923SCy Schubert 		uint16_t len = htons(c->tcp_write_and_read?c->tcp_write_pkt_len:sldns_buffer_limit(c->buffer));
1920b7579f77SDag-Erling Smørgrav 		ERR_clear_error();
1921369c6923SCy Schubert 		if(c->tcp_write_and_read) {
1922369c6923SCy Schubert 			if(c->tcp_write_pkt_len + 2 < LDNS_RR_BUF_SIZE) {
1923369c6923SCy Schubert 				/* combine the tcp length and the query for
1924369c6923SCy Schubert 				 * write, this emulates writev */
1925369c6923SCy Schubert 				uint8_t buf[LDNS_RR_BUF_SIZE];
1926369c6923SCy Schubert 				memmove(buf, &len, sizeof(uint16_t));
1927369c6923SCy Schubert 				memmove(buf+sizeof(uint16_t),
1928369c6923SCy Schubert 					c->tcp_write_pkt,
1929369c6923SCy Schubert 					c->tcp_write_pkt_len);
1930369c6923SCy Schubert 				r = SSL_write(c->ssl,
1931369c6923SCy Schubert 					(void*)(buf+c->tcp_write_byte_count),
1932369c6923SCy Schubert 					c->tcp_write_pkt_len + 2 -
1933369c6923SCy Schubert 					c->tcp_write_byte_count);
1934369c6923SCy Schubert 			} else {
1935369c6923SCy Schubert 				r = SSL_write(c->ssl,
1936369c6923SCy Schubert 					(void*)(((uint8_t*)&len)+c->tcp_write_byte_count),
1937369c6923SCy Schubert 					(int)(sizeof(uint16_t)-c->tcp_write_byte_count));
1938369c6923SCy Schubert 			}
1939369c6923SCy Schubert 		} else if(sizeof(uint16_t)+sldns_buffer_remaining(c->buffer) <
19400fb34990SDag-Erling Smørgrav 			LDNS_RR_BUF_SIZE) {
19410fb34990SDag-Erling Smørgrav 			/* combine the tcp length and the query for write,
19420fb34990SDag-Erling Smørgrav 			 * this emulates writev */
19430fb34990SDag-Erling Smørgrav 			uint8_t buf[LDNS_RR_BUF_SIZE];
19440fb34990SDag-Erling Smørgrav 			memmove(buf, &len, sizeof(uint16_t));
19450fb34990SDag-Erling Smørgrav 			memmove(buf+sizeof(uint16_t),
19460fb34990SDag-Erling Smørgrav 				sldns_buffer_current(c->buffer),
19470fb34990SDag-Erling Smørgrav 				sldns_buffer_remaining(c->buffer));
19480fb34990SDag-Erling Smørgrav 			r = SSL_write(c->ssl, (void*)(buf+c->tcp_byte_count),
19490fb34990SDag-Erling Smørgrav 				(int)(sizeof(uint16_t)+
19500fb34990SDag-Erling Smørgrav 				sldns_buffer_remaining(c->buffer)
19510fb34990SDag-Erling Smørgrav 				- c->tcp_byte_count));
19520fb34990SDag-Erling Smørgrav 		} else {
1953b7579f77SDag-Erling Smørgrav 			r = SSL_write(c->ssl,
1954b7579f77SDag-Erling Smørgrav 				(void*)(((uint8_t*)&len)+c->tcp_byte_count),
1955b7579f77SDag-Erling Smørgrav 				(int)(sizeof(uint16_t)-c->tcp_byte_count));
19560fb34990SDag-Erling Smørgrav 		}
1957b7579f77SDag-Erling Smørgrav 		if(r <= 0) {
1958b7579f77SDag-Erling Smørgrav 			int want = SSL_get_error(c->ssl, r);
1959b7579f77SDag-Erling Smørgrav 			if(want == SSL_ERROR_ZERO_RETURN) {
1960b7579f77SDag-Erling Smørgrav 				return 0; /* closed */
1961b7579f77SDag-Erling Smørgrav 			} else if(want == SSL_ERROR_WANT_READ) {
1962e86b9096SDag-Erling Smørgrav 				c->ssl_shake_state = comm_ssl_shake_hs_read;
1963b7579f77SDag-Erling Smørgrav 				comm_point_listen_for_rw(c, 1, 0);
1964b7579f77SDag-Erling Smørgrav 				return 1; /* wait for read condition */
1965b7579f77SDag-Erling Smørgrav 			} else if(want == SSL_ERROR_WANT_WRITE) {
19665469a995SCy Schubert #ifdef USE_WINSOCK
19673bd4df0aSDag-Erling Smørgrav 				ub_winsock_tcp_wouldblock(c->ev->ev, UB_EV_WRITE);
19685469a995SCy Schubert #endif
1969b7579f77SDag-Erling Smørgrav 				return 1; /* write more later */
1970b7579f77SDag-Erling Smørgrav 			} else if(want == SSL_ERROR_SYSCALL) {
1971e86b9096SDag-Erling Smørgrav #ifdef EPIPE
1972e86b9096SDag-Erling Smørgrav 				if(errno == EPIPE && verbosity < 2)
1973e86b9096SDag-Erling Smørgrav 					return 0; /* silence 'broken pipe' */
1974e86b9096SDag-Erling Smørgrav #endif
1975b7579f77SDag-Erling Smørgrav 				if(errno != 0)
1976b7579f77SDag-Erling Smørgrav 					log_err("SSL_write syscall: %s",
1977b7579f77SDag-Erling Smørgrav 						strerror(errno));
1978b7579f77SDag-Erling Smørgrav 				return 0;
1979b7579f77SDag-Erling Smørgrav 			}
1980b7579f77SDag-Erling Smørgrav 			log_crypto_err("could not SSL_write");
1981b7579f77SDag-Erling Smørgrav 			return 0;
1982b7579f77SDag-Erling Smørgrav 		}
1983369c6923SCy Schubert 		if(c->tcp_write_and_read) {
1984369c6923SCy Schubert 			c->tcp_write_byte_count += r;
1985369c6923SCy Schubert 			if(c->tcp_write_byte_count < sizeof(uint16_t))
1986369c6923SCy Schubert 				return 1;
1987369c6923SCy Schubert 		} else {
1988b7579f77SDag-Erling Smørgrav 			c->tcp_byte_count += r;
1989b7579f77SDag-Erling Smørgrav 			if(c->tcp_byte_count < sizeof(uint16_t))
1990b7579f77SDag-Erling Smørgrav 				return 1;
199117d15b25SDag-Erling Smørgrav 			sldns_buffer_set_position(c->buffer, c->tcp_byte_count -
1992b7579f77SDag-Erling Smørgrav 				sizeof(uint16_t));
1993369c6923SCy Schubert 		}
1994369c6923SCy Schubert 		if((!c->tcp_write_and_read && sldns_buffer_remaining(c->buffer) == 0) || (c->tcp_write_and_read && c->tcp_write_byte_count == c->tcp_write_pkt_len + 2)) {
1995b7579f77SDag-Erling Smørgrav 			tcp_callback_writer(c);
1996b7579f77SDag-Erling Smørgrav 			return 1;
1997b7579f77SDag-Erling Smørgrav 		}
1998b7579f77SDag-Erling Smørgrav 	}
1999369c6923SCy Schubert 	log_assert(c->tcp_write_and_read || sldns_buffer_remaining(c->buffer) > 0);
2000369c6923SCy Schubert 	log_assert(!c->tcp_write_and_read || c->tcp_write_byte_count < c->tcp_write_pkt_len + 2);
2001b7579f77SDag-Erling Smørgrav 	ERR_clear_error();
2002369c6923SCy Schubert 	if(c->tcp_write_and_read) {
2003369c6923SCy Schubert 		r = SSL_write(c->ssl, (void*)(c->tcp_write_pkt + c->tcp_write_byte_count - 2),
2004369c6923SCy Schubert 			(int)(c->tcp_write_pkt_len + 2 - c->tcp_write_byte_count));
2005369c6923SCy Schubert 	} else {
200617d15b25SDag-Erling Smørgrav 		r = SSL_write(c->ssl, (void*)sldns_buffer_current(c->buffer),
200717d15b25SDag-Erling Smørgrav 			(int)sldns_buffer_remaining(c->buffer));
2008369c6923SCy Schubert 	}
2009b7579f77SDag-Erling Smørgrav 	if(r <= 0) {
2010b7579f77SDag-Erling Smørgrav 		int want = SSL_get_error(c->ssl, r);
2011b7579f77SDag-Erling Smørgrav 		if(want == SSL_ERROR_ZERO_RETURN) {
2012b7579f77SDag-Erling Smørgrav 			return 0; /* closed */
2013b7579f77SDag-Erling Smørgrav 		} else if(want == SSL_ERROR_WANT_READ) {
2014e86b9096SDag-Erling Smørgrav 			c->ssl_shake_state = comm_ssl_shake_hs_read;
2015b7579f77SDag-Erling Smørgrav 			comm_point_listen_for_rw(c, 1, 0);
2016b7579f77SDag-Erling Smørgrav 			return 1; /* wait for read condition */
2017b7579f77SDag-Erling Smørgrav 		} else if(want == SSL_ERROR_WANT_WRITE) {
20185469a995SCy Schubert #ifdef USE_WINSOCK
20193bd4df0aSDag-Erling Smørgrav 			ub_winsock_tcp_wouldblock(c->ev->ev, UB_EV_WRITE);
20205469a995SCy Schubert #endif
2021b7579f77SDag-Erling Smørgrav 			return 1; /* write more later */
2022b7579f77SDag-Erling Smørgrav 		} else if(want == SSL_ERROR_SYSCALL) {
2023e86b9096SDag-Erling Smørgrav #ifdef EPIPE
2024e86b9096SDag-Erling Smørgrav 			if(errno == EPIPE && verbosity < 2)
2025e86b9096SDag-Erling Smørgrav 				return 0; /* silence 'broken pipe' */
2026e86b9096SDag-Erling Smørgrav #endif
2027b7579f77SDag-Erling Smørgrav 			if(errno != 0)
2028b7579f77SDag-Erling Smørgrav 				log_err("SSL_write syscall: %s",
2029b7579f77SDag-Erling Smørgrav 					strerror(errno));
2030b7579f77SDag-Erling Smørgrav 			return 0;
2031b7579f77SDag-Erling Smørgrav 		}
2032b7579f77SDag-Erling Smørgrav 		log_crypto_err("could not SSL_write");
2033b7579f77SDag-Erling Smørgrav 		return 0;
2034b7579f77SDag-Erling Smørgrav 	}
2035369c6923SCy Schubert 	if(c->tcp_write_and_read) {
2036369c6923SCy Schubert 		c->tcp_write_byte_count += r;
2037369c6923SCy Schubert 	} else {
203817d15b25SDag-Erling Smørgrav 		sldns_buffer_skip(c->buffer, (ssize_t)r);
2039369c6923SCy Schubert 	}
2040b7579f77SDag-Erling Smørgrav 
2041369c6923SCy Schubert 	if((!c->tcp_write_and_read && sldns_buffer_remaining(c->buffer) == 0) || (c->tcp_write_and_read && c->tcp_write_byte_count == c->tcp_write_pkt_len + 2)) {
2042b7579f77SDag-Erling Smørgrav 		tcp_callback_writer(c);
2043b7579f77SDag-Erling Smørgrav 	}
2044b7579f77SDag-Erling Smørgrav 	return 1;
20458ed2b524SDag-Erling Smørgrav #else
20468ed2b524SDag-Erling Smørgrav 	(void)c;
20478ed2b524SDag-Erling Smørgrav 	return 0;
20488ed2b524SDag-Erling Smørgrav #endif /* HAVE_SSL */
2049b7579f77SDag-Erling Smørgrav }
2050b7579f77SDag-Erling Smørgrav 
2051b7579f77SDag-Erling Smørgrav /** handle ssl tcp connection with dns contents */
2052b7579f77SDag-Erling Smørgrav static int
2053369c6923SCy Schubert ssl_handle_it(struct comm_point* c, int is_write)
2054b7579f77SDag-Erling Smørgrav {
2055369c6923SCy Schubert 	/* handle case where renegotiation wants read during write call
2056369c6923SCy Schubert 	 * or write during read calls */
2057369c6923SCy Schubert 	if(is_write && c->ssl_shake_state == comm_ssl_shake_hs_write)
2058369c6923SCy Schubert 		return ssl_handle_read(c);
2059369c6923SCy Schubert 	else if(!is_write && c->ssl_shake_state == comm_ssl_shake_hs_read)
2060369c6923SCy Schubert 		return ssl_handle_write(c);
2061369c6923SCy Schubert 	/* handle read events for read operation and write events for a
2062369c6923SCy Schubert 	 * write operation */
2063369c6923SCy Schubert 	else if(!is_write)
2064b7579f77SDag-Erling Smørgrav 		return ssl_handle_read(c);
2065b7579f77SDag-Erling Smørgrav 	return ssl_handle_write(c);
2066b7579f77SDag-Erling Smørgrav }
2067b7579f77SDag-Erling Smørgrav 
2068865f46b2SCy Schubert /**
2069865f46b2SCy Schubert  * Handle tcp reading callback.
2070b7579f77SDag-Erling Smørgrav  * @param fd: file descriptor of socket.
2071b7579f77SDag-Erling Smørgrav  * @param c: comm point to read from into buffer.
2072b7579f77SDag-Erling Smørgrav  * @param short_ok: if true, very short packets are OK (for comm_local).
2073b7579f77SDag-Erling Smørgrav  * @return: 0 on error
2074b7579f77SDag-Erling Smørgrav  */
2075b7579f77SDag-Erling Smørgrav static int
2076b7579f77SDag-Erling Smørgrav comm_point_tcp_handle_read(int fd, struct comm_point* c, int short_ok)
2077b7579f77SDag-Erling Smørgrav {
2078b7579f77SDag-Erling Smørgrav 	ssize_t r;
2079865f46b2SCy Schubert 	int recv_initial = 0;
2080b7579f77SDag-Erling Smørgrav 	log_assert(c->type == comm_tcp || c->type == comm_local);
2081b7579f77SDag-Erling Smørgrav 	if(c->ssl)
2082369c6923SCy Schubert 		return ssl_handle_it(c, 0);
2083369c6923SCy Schubert 	if(!c->tcp_is_reading && !c->tcp_write_and_read)
2084b7579f77SDag-Erling Smørgrav 		return 0;
2085b7579f77SDag-Erling Smørgrav 
2086b7579f77SDag-Erling Smørgrav 	log_assert(fd != -1);
2087865f46b2SCy Schubert 	if(c->pp2_enabled && c->pp2_header_state != pp2_header_done) {
2088865f46b2SCy Schubert 		struct pp2_header* header = NULL;
2089865f46b2SCy Schubert 		size_t want_read_size = 0;
2090865f46b2SCy Schubert 		size_t current_read_size = 0;
2091865f46b2SCy Schubert 		if(c->pp2_header_state == pp2_header_none) {
2092865f46b2SCy Schubert 			want_read_size = PP2_HEADER_SIZE;
2093865f46b2SCy Schubert 			if(sldns_buffer_remaining(c->buffer)<want_read_size) {
2094865f46b2SCy Schubert 				log_err_addr("proxy_protocol: not enough "
2095865f46b2SCy Schubert 					"buffer size to read PROXYv2 header", "",
2096865f46b2SCy Schubert 					&c->repinfo.remote_addr,
2097865f46b2SCy Schubert 					c->repinfo.remote_addrlen);
2098865f46b2SCy Schubert 				return 0;
2099865f46b2SCy Schubert 			}
2100865f46b2SCy Schubert 			verbose(VERB_ALGO, "proxy_protocol: reading fixed "
2101865f46b2SCy Schubert 				"part of PROXYv2 header (len %lu)",
2102865f46b2SCy Schubert 				(unsigned long)want_read_size);
2103865f46b2SCy Schubert 			current_read_size = want_read_size;
2104865f46b2SCy Schubert 			if(c->tcp_byte_count < current_read_size) {
2105865f46b2SCy Schubert 				r = recv(fd, (void*)sldns_buffer_at(c->buffer,
2106865f46b2SCy Schubert 					c->tcp_byte_count),
2107865f46b2SCy Schubert 					current_read_size-c->tcp_byte_count, MSG_DONTWAIT);
210864d318eaSCy Schubert 				if(r == 0) {
210964d318eaSCy Schubert 					if(c->tcp_req_info)
211064d318eaSCy Schubert 						return tcp_req_info_handle_read_close(c->tcp_req_info);
211164d318eaSCy Schubert 					return 0;
211264d318eaSCy Schubert 				} else if(r == -1) {
2113865f46b2SCy Schubert 					goto recv_error_initial;
2114865f46b2SCy Schubert 				}
2115865f46b2SCy Schubert 				c->tcp_byte_count += r;
2116865f46b2SCy Schubert 				if(c->tcp_byte_count != current_read_size) return 1;
2117865f46b2SCy Schubert 				c->pp2_header_state = pp2_header_init;
2118865f46b2SCy Schubert 			}
2119865f46b2SCy Schubert 		}
2120865f46b2SCy Schubert 		if(c->pp2_header_state == pp2_header_init) {
2121865f46b2SCy Schubert 			header = pp2_read_header(c->buffer);
2122865f46b2SCy Schubert 			if(!header) {
2123865f46b2SCy Schubert 				log_err("proxy_protocol: could not parse "
2124865f46b2SCy Schubert 					"PROXYv2 header");
2125865f46b2SCy Schubert 				return 0;
2126865f46b2SCy Schubert 			}
2127865f46b2SCy Schubert 			want_read_size = ntohs(header->len);
2128865f46b2SCy Schubert 			if(sldns_buffer_remaining(c->buffer) <
2129865f46b2SCy Schubert 				PP2_HEADER_SIZE + want_read_size) {
2130865f46b2SCy Schubert 				log_err_addr("proxy_protocol: not enough "
2131865f46b2SCy Schubert 					"buffer size to read PROXYv2 header", "",
2132865f46b2SCy Schubert 					&c->repinfo.remote_addr,
2133865f46b2SCy Schubert 					c->repinfo.remote_addrlen);
2134865f46b2SCy Schubert 				return 0;
2135865f46b2SCy Schubert 			}
2136865f46b2SCy Schubert 			verbose(VERB_ALGO, "proxy_protocol: reading variable "
2137865f46b2SCy Schubert 				"part of PROXYv2 header (len %lu)",
2138865f46b2SCy Schubert 				(unsigned long)want_read_size);
2139865f46b2SCy Schubert 			current_read_size = PP2_HEADER_SIZE + want_read_size;
2140865f46b2SCy Schubert 			if(want_read_size == 0) {
2141865f46b2SCy Schubert 				/* nothing more to read; header is complete */
2142865f46b2SCy Schubert 				c->pp2_header_state = pp2_header_done;
2143865f46b2SCy Schubert 			} else if(c->tcp_byte_count < current_read_size) {
2144865f46b2SCy Schubert 				r = recv(fd, (void*)sldns_buffer_at(c->buffer,
2145865f46b2SCy Schubert 					c->tcp_byte_count),
2146865f46b2SCy Schubert 					current_read_size-c->tcp_byte_count, MSG_DONTWAIT);
2147865f46b2SCy Schubert 				if(r == 0) {
2148865f46b2SCy Schubert 					if(c->tcp_req_info)
2149865f46b2SCy Schubert 						return tcp_req_info_handle_read_close(c->tcp_req_info);
2150865f46b2SCy Schubert 					return 0;
2151865f46b2SCy Schubert 				} else if(r == -1) {
2152865f46b2SCy Schubert 					goto recv_error;
2153865f46b2SCy Schubert 				}
2154865f46b2SCy Schubert 				c->tcp_byte_count += r;
2155865f46b2SCy Schubert 				if(c->tcp_byte_count != current_read_size) return 1;
2156865f46b2SCy Schubert 				c->pp2_header_state = pp2_header_done;
2157865f46b2SCy Schubert 			}
2158865f46b2SCy Schubert 		}
2159865f46b2SCy Schubert 		if(c->pp2_header_state != pp2_header_done || !header) {
2160865f46b2SCy Schubert 			log_err_addr("proxy_protocol: wrong state for the "
2161865f46b2SCy Schubert 				"PROXYv2 header", "", &c->repinfo.remote_addr,
2162865f46b2SCy Schubert 				c->repinfo.remote_addrlen);
2163865f46b2SCy Schubert 			return 0;
2164865f46b2SCy Schubert 		}
2165865f46b2SCy Schubert 		if(!consume_pp2_header(c->buffer, &c->repinfo, 1)) {
2166865f46b2SCy Schubert 			log_err_addr("proxy_protocol: could not consume "
2167865f46b2SCy Schubert 				"PROXYv2 header", "", &c->repinfo.remote_addr,
2168865f46b2SCy Schubert 				c->repinfo.remote_addrlen);
2169865f46b2SCy Schubert 			return 0;
2170865f46b2SCy Schubert 		}
2171865f46b2SCy Schubert 		verbose(VERB_ALGO, "proxy_protocol: successful read of "
2172865f46b2SCy Schubert 			"PROXYv2 header");
2173865f46b2SCy Schubert 		/* Clear and reset the buffer to read the following
2174865f46b2SCy Schubert 		    * DNS packet(s). */
2175865f46b2SCy Schubert 		sldns_buffer_clear(c->buffer);
2176865f46b2SCy Schubert 		c->tcp_byte_count = 0;
2177865f46b2SCy Schubert 		return 1;
2178865f46b2SCy Schubert 	}
2179865f46b2SCy Schubert 
2180865f46b2SCy Schubert 	if(c->tcp_byte_count < sizeof(uint16_t)) {
2181865f46b2SCy Schubert 		/* read length bytes */
2182865f46b2SCy Schubert 		r = recv(fd,(void*)sldns_buffer_at(c->buffer,c->tcp_byte_count),
2183865f46b2SCy Schubert 			sizeof(uint16_t)-c->tcp_byte_count, MSG_DONTWAIT);
2184865f46b2SCy Schubert 		if(r == 0) {
2185865f46b2SCy Schubert 			if(c->tcp_req_info)
2186865f46b2SCy Schubert 				return tcp_req_info_handle_read_close(c->tcp_req_info);
2187865f46b2SCy Schubert 			return 0;
2188865f46b2SCy Schubert 		} else if(r == -1) {
2189865f46b2SCy Schubert 			if(c->pp2_enabled) goto recv_error;
2190865f46b2SCy Schubert 			goto recv_error_initial;
2191865f46b2SCy Schubert 		}
2192865f46b2SCy Schubert 		c->tcp_byte_count += r;
2193865f46b2SCy Schubert 		if(c->tcp_byte_count != sizeof(uint16_t))
2194865f46b2SCy Schubert 			return 1;
2195865f46b2SCy Schubert 		if(sldns_buffer_read_u16_at(c->buffer, 0) >
2196865f46b2SCy Schubert 			sldns_buffer_capacity(c->buffer)) {
2197865f46b2SCy Schubert 			verbose(VERB_QUERY, "tcp: dropped larger than buffer");
2198865f46b2SCy Schubert 			return 0;
2199865f46b2SCy Schubert 		}
2200865f46b2SCy Schubert 		sldns_buffer_set_limit(c->buffer,
2201865f46b2SCy Schubert 			sldns_buffer_read_u16_at(c->buffer, 0));
2202865f46b2SCy Schubert 		if(!short_ok &&
2203865f46b2SCy Schubert 			sldns_buffer_limit(c->buffer) < LDNS_HEADER_SIZE) {
2204865f46b2SCy Schubert 			verbose(VERB_QUERY, "tcp: dropped bogus too short.");
2205865f46b2SCy Schubert 			return 0;
2206865f46b2SCy Schubert 		}
2207865f46b2SCy Schubert 		verbose(VERB_ALGO, "Reading tcp query of length %d",
2208865f46b2SCy Schubert 			(int)sldns_buffer_limit(c->buffer));
2209865f46b2SCy Schubert 	}
2210865f46b2SCy Schubert 
2211865f46b2SCy Schubert 	if(sldns_buffer_remaining(c->buffer) == 0)
2212865f46b2SCy Schubert 		log_err("in comm_point_tcp_handle_read buffer_remaining is "
2213865f46b2SCy Schubert 			"not > 0 as expected, continuing with (harmless) 0 "
2214865f46b2SCy Schubert 			"length recv");
2215865f46b2SCy Schubert 	r = recv(fd, (void*)sldns_buffer_current(c->buffer),
2216865f46b2SCy Schubert 		sldns_buffer_remaining(c->buffer), MSG_DONTWAIT);
2217865f46b2SCy Schubert 	if(r == 0) {
2218865f46b2SCy Schubert 		if(c->tcp_req_info)
2219865f46b2SCy Schubert 			return tcp_req_info_handle_read_close(c->tcp_req_info);
2220865f46b2SCy Schubert 		return 0;
2221865f46b2SCy Schubert 	} else if(r == -1) {
2222865f46b2SCy Schubert 		goto recv_error;
2223865f46b2SCy Schubert 	}
2224865f46b2SCy Schubert 	sldns_buffer_skip(c->buffer, r);
2225865f46b2SCy Schubert 	if(sldns_buffer_remaining(c->buffer) <= 0) {
2226865f46b2SCy Schubert 		tcp_callback_reader(c);
2227865f46b2SCy Schubert 	}
2228865f46b2SCy Schubert 	return 1;
2229865f46b2SCy Schubert 
2230865f46b2SCy Schubert recv_error_initial:
2231865f46b2SCy Schubert 	recv_initial = 1;
2232865f46b2SCy Schubert recv_error:
2233b7579f77SDag-Erling Smørgrav #ifndef USE_WINSOCK
2234b7579f77SDag-Erling Smørgrav 	if(errno == EINTR || errno == EAGAIN)
2235b7579f77SDag-Erling Smørgrav 		return 1;
2236865f46b2SCy Schubert 	if(recv_initial) {
2237b7579f77SDag-Erling Smørgrav #ifdef ECONNRESET
2238b7579f77SDag-Erling Smørgrav 		if(errno == ECONNRESET && verbosity < 2)
2239b7579f77SDag-Erling Smørgrav 			return 0; /* silence reset by peer */
2240b7579f77SDag-Erling Smørgrav #endif
22415469a995SCy Schubert #ifdef ECONNREFUSED
22425469a995SCy Schubert 		if(errno == ECONNREFUSED && verbosity < 2)
22435469a995SCy Schubert 			return 0; /* silence reset by peer */
22445469a995SCy Schubert #endif
2245f44e67d1SCy Schubert #ifdef ENETUNREACH
2246f44e67d1SCy Schubert 		if(errno == ENETUNREACH && verbosity < 2)
2247f44e67d1SCy Schubert 			return 0; /* silence it */
2248f44e67d1SCy Schubert #endif
2249f44e67d1SCy Schubert #ifdef EHOSTDOWN
2250f44e67d1SCy Schubert 		if(errno == EHOSTDOWN && verbosity < 2)
2251f44e67d1SCy Schubert 			return 0; /* silence it */
2252f44e67d1SCy Schubert #endif
2253f44e67d1SCy Schubert #ifdef EHOSTUNREACH
2254f44e67d1SCy Schubert 		if(errno == EHOSTUNREACH && verbosity < 2)
2255f44e67d1SCy Schubert 			return 0; /* silence it */
2256f44e67d1SCy Schubert #endif
2257f44e67d1SCy Schubert #ifdef ENETDOWN
2258f44e67d1SCy Schubert 		if(errno == ENETDOWN && verbosity < 2)
2259f44e67d1SCy Schubert 			return 0; /* silence it */
2260f44e67d1SCy Schubert #endif
2261f44e67d1SCy Schubert #ifdef EACCES
2262f44e67d1SCy Schubert 		if(errno == EACCES && verbosity < 2)
2263f44e67d1SCy Schubert 			return 0; /* silence it */
2264f44e67d1SCy Schubert #endif
2265f44e67d1SCy Schubert #ifdef ENOTCONN
2266f44e67d1SCy Schubert 		if(errno == ENOTCONN) {
2267865f46b2SCy Schubert 			log_err_addr("read (in tcp s) failed and this "
2268865f46b2SCy Schubert 				"could be because TCP Fast Open is "
2269865f46b2SCy Schubert 				"enabled [--disable-tfo-client "
2270865f46b2SCy Schubert 				"--disable-tfo-server] but does not "
2271865f46b2SCy Schubert 				"work", sock_strerror(errno),
2272865f46b2SCy Schubert 				&c->repinfo.remote_addr,
2273865f46b2SCy Schubert 				c->repinfo.remote_addrlen);
2274f44e67d1SCy Schubert 			return 0;
2275f44e67d1SCy Schubert 		}
2276f44e67d1SCy Schubert #endif
2277865f46b2SCy Schubert 	}
2278b7579f77SDag-Erling Smørgrav #else /* USE_WINSOCK */
2279865f46b2SCy Schubert 	if(recv_initial) {
22805469a995SCy Schubert 		if(WSAGetLastError() == WSAECONNREFUSED && verbosity < 2)
22815469a995SCy Schubert 			return 0;
22825469a995SCy Schubert 		if(WSAGetLastError() == WSAEHOSTDOWN && verbosity < 2)
22835469a995SCy Schubert 			return 0;
22845469a995SCy Schubert 		if(WSAGetLastError() == WSAEHOSTUNREACH && verbosity < 2)
22855469a995SCy Schubert 			return 0;
22865469a995SCy Schubert 		if(WSAGetLastError() == WSAENETDOWN && verbosity < 2)
22875469a995SCy Schubert 			return 0;
22885469a995SCy Schubert 		if(WSAGetLastError() == WSAENETUNREACH && verbosity < 2)
22895469a995SCy Schubert 			return 0;
2290865f46b2SCy Schubert 	}
2291b7579f77SDag-Erling Smørgrav 	if(WSAGetLastError() == WSAECONNRESET)
2292b7579f77SDag-Erling Smørgrav 		return 0;
2293b7579f77SDag-Erling Smørgrav 	if(WSAGetLastError() == WSAEINPROGRESS)
2294b7579f77SDag-Erling Smørgrav 		return 1;
2295b7579f77SDag-Erling Smørgrav 	if(WSAGetLastError() == WSAEWOULDBLOCK) {
2296e2d15004SDag-Erling Smørgrav 		ub_winsock_tcp_wouldblock(c->ev->ev,
2297e2d15004SDag-Erling Smørgrav 			UB_EV_READ);
2298b7579f77SDag-Erling Smørgrav 		return 1;
2299b7579f77SDag-Erling Smørgrav 	}
2300b7579f77SDag-Erling Smørgrav #endif
2301c0caa2e2SCy Schubert 	log_err_addr("read (in tcp s)", sock_strerror(errno),
2302865f46b2SCy Schubert 		&c->repinfo.remote_addr, c->repinfo.remote_addrlen);
2303b7579f77SDag-Erling Smørgrav 	return 0;
2304b7579f77SDag-Erling Smørgrav }
2305b7579f77SDag-Erling Smørgrav 
2306b7579f77SDag-Erling Smørgrav /**
2307b7579f77SDag-Erling Smørgrav  * Handle tcp writing callback.
2308b7579f77SDag-Erling Smørgrav  * @param fd: file descriptor of socket.
2309b7579f77SDag-Erling Smørgrav  * @param c: comm point to write buffer out of.
2310b7579f77SDag-Erling Smørgrav  * @return: 0 on error
2311b7579f77SDag-Erling Smørgrav  */
2312b7579f77SDag-Erling Smørgrav static int
2313b7579f77SDag-Erling Smørgrav comm_point_tcp_handle_write(int fd, struct comm_point* c)
2314b7579f77SDag-Erling Smørgrav {
2315b7579f77SDag-Erling Smørgrav 	ssize_t r;
231665b390aaSDag-Erling Smørgrav 	struct sldns_buffer *buffer;
2317b7579f77SDag-Erling Smørgrav 	log_assert(c->type == comm_tcp);
231865b390aaSDag-Erling Smørgrav #ifdef USE_DNSCRYPT
231965b390aaSDag-Erling Smørgrav 	buffer = c->dnscrypt_buffer;
232065b390aaSDag-Erling Smørgrav #else
232165b390aaSDag-Erling Smørgrav 	buffer = c->buffer;
232265b390aaSDag-Erling Smørgrav #endif
2323369c6923SCy Schubert 	if(c->tcp_is_reading && !c->ssl && !c->tcp_write_and_read)
2324b7579f77SDag-Erling Smørgrav 		return 0;
2325b7579f77SDag-Erling Smørgrav 	log_assert(fd != -1);
2326369c6923SCy Schubert 	if(((!c->tcp_write_and_read && c->tcp_byte_count == 0) || (c->tcp_write_and_read && c->tcp_write_byte_count == 0)) && c->tcp_check_nb_connect) {
2327b7579f77SDag-Erling Smørgrav 		/* check for pending error from nonblocking connect */
2328b7579f77SDag-Erling Smørgrav 		/* from Stevens, unix network programming, vol1, 3rd ed, p450*/
2329b7579f77SDag-Erling Smørgrav 		int error = 0;
2330b7579f77SDag-Erling Smørgrav 		socklen_t len = (socklen_t)sizeof(error);
2331b7579f77SDag-Erling Smørgrav 		if(getsockopt(fd, SOL_SOCKET, SO_ERROR, (void*)&error,
2332b7579f77SDag-Erling Smørgrav 			&len) < 0){
2333b7579f77SDag-Erling Smørgrav #ifndef USE_WINSOCK
2334b7579f77SDag-Erling Smørgrav 			error = errno; /* on solaris errno is error */
2335b7579f77SDag-Erling Smørgrav #else /* USE_WINSOCK */
2336b7579f77SDag-Erling Smørgrav 			error = WSAGetLastError();
2337b7579f77SDag-Erling Smørgrav #endif
2338b7579f77SDag-Erling Smørgrav 		}
2339b7579f77SDag-Erling Smørgrav #ifndef USE_WINSOCK
2340b7579f77SDag-Erling Smørgrav #if defined(EINPROGRESS) && defined(EWOULDBLOCK)
2341b7579f77SDag-Erling Smørgrav 		if(error == EINPROGRESS || error == EWOULDBLOCK)
2342b7579f77SDag-Erling Smørgrav 			return 1; /* try again later */
2343b7579f77SDag-Erling Smørgrav 		else
2344b7579f77SDag-Erling Smørgrav #endif
2345b7579f77SDag-Erling Smørgrav 		if(error != 0 && verbosity < 2)
2346b7579f77SDag-Erling Smørgrav 			return 0; /* silence lots of chatter in the logs */
2347b7579f77SDag-Erling Smørgrav                 else if(error != 0) {
2348ff825849SDag-Erling Smørgrav 			log_err_addr("tcp connect", strerror(error),
2349865f46b2SCy Schubert 				&c->repinfo.remote_addr,
2350865f46b2SCy Schubert 				c->repinfo.remote_addrlen);
2351b7579f77SDag-Erling Smørgrav #else /* USE_WINSOCK */
2352b7579f77SDag-Erling Smørgrav 		/* examine error */
2353b7579f77SDag-Erling Smørgrav 		if(error == WSAEINPROGRESS)
2354b7579f77SDag-Erling Smørgrav 			return 1;
2355b7579f77SDag-Erling Smørgrav 		else if(error == WSAEWOULDBLOCK) {
2356e2d15004SDag-Erling Smørgrav 			ub_winsock_tcp_wouldblock(c->ev->ev, UB_EV_WRITE);
2357b7579f77SDag-Erling Smørgrav 			return 1;
2358b7579f77SDag-Erling Smørgrav 		} else if(error != 0 && verbosity < 2)
2359b7579f77SDag-Erling Smørgrav 			return 0;
2360b7579f77SDag-Erling Smørgrav 		else if(error != 0) {
2361ff825849SDag-Erling Smørgrav 			log_err_addr("tcp connect", wsa_strerror(error),
2362865f46b2SCy Schubert 				&c->repinfo.remote_addr,
2363865f46b2SCy Schubert 				c->repinfo.remote_addrlen);
2364b7579f77SDag-Erling Smørgrav #endif /* USE_WINSOCK */
2365b7579f77SDag-Erling Smørgrav 			return 0;
2366b7579f77SDag-Erling Smørgrav 		}
2367b7579f77SDag-Erling Smørgrav 	}
2368b7579f77SDag-Erling Smørgrav 	if(c->ssl)
2369369c6923SCy Schubert 		return ssl_handle_it(c, 1);
2370b7579f77SDag-Erling Smørgrav 
2371b5663de9SDag-Erling Smørgrav #ifdef USE_MSG_FASTOPEN
2372b5663de9SDag-Erling Smørgrav 	/* Only try this on first use of a connection that uses tfo,
2373b5663de9SDag-Erling Smørgrav 	   otherwise fall through to normal write */
2374b5663de9SDag-Erling Smørgrav 	/* Also, TFO support on WINDOWS not implemented at the moment */
2375b5663de9SDag-Erling Smørgrav 	if(c->tcp_do_fastopen == 1) {
2376b5663de9SDag-Erling Smørgrav 		/* this form of sendmsg() does both a connect() and send() so need to
2377b5663de9SDag-Erling Smørgrav 		   look for various flavours of error*/
2378369c6923SCy Schubert 		uint16_t len = htons(c->tcp_write_and_read?c->tcp_write_pkt_len:sldns_buffer_limit(buffer));
2379b5663de9SDag-Erling Smørgrav 		struct msghdr msg;
2380b5663de9SDag-Erling Smørgrav 		struct iovec iov[2];
2381b5663de9SDag-Erling Smørgrav 		c->tcp_do_fastopen = 0;
2382b5663de9SDag-Erling Smørgrav 		memset(&msg, 0, sizeof(msg));
2383369c6923SCy Schubert 		if(c->tcp_write_and_read) {
2384369c6923SCy Schubert 			iov[0].iov_base = (uint8_t*)&len + c->tcp_write_byte_count;
2385369c6923SCy Schubert 			iov[0].iov_len = sizeof(uint16_t) - c->tcp_write_byte_count;
2386369c6923SCy Schubert 			iov[1].iov_base = c->tcp_write_pkt;
2387369c6923SCy Schubert 			iov[1].iov_len = c->tcp_write_pkt_len;
2388369c6923SCy Schubert 		} else {
2389b5663de9SDag-Erling Smørgrav 			iov[0].iov_base = (uint8_t*)&len + c->tcp_byte_count;
2390b5663de9SDag-Erling Smørgrav 			iov[0].iov_len = sizeof(uint16_t) - c->tcp_byte_count;
239165b390aaSDag-Erling Smørgrav 			iov[1].iov_base = sldns_buffer_begin(buffer);
239265b390aaSDag-Erling Smørgrav 			iov[1].iov_len = sldns_buffer_limit(buffer);
2393369c6923SCy Schubert 		}
2394b5663de9SDag-Erling Smørgrav 		log_assert(iov[0].iov_len > 0);
2395865f46b2SCy Schubert 		msg.msg_name = &c->repinfo.remote_addr;
2396865f46b2SCy Schubert 		msg.msg_namelen = c->repinfo.remote_addrlen;
2397b5663de9SDag-Erling Smørgrav 		msg.msg_iov = iov;
2398b5663de9SDag-Erling Smørgrav 		msg.msg_iovlen = 2;
2399b5663de9SDag-Erling Smørgrav 		r = sendmsg(fd, &msg, MSG_FASTOPEN);
2400b5663de9SDag-Erling Smørgrav 		if (r == -1) {
2401b5663de9SDag-Erling Smørgrav #if defined(EINPROGRESS) && defined(EWOULDBLOCK)
2402b5663de9SDag-Erling Smørgrav 			/* Handshake is underway, maybe because no TFO cookie available.
24038a384985SDag-Erling Smørgrav 			   Come back to write the message*/
2404b5663de9SDag-Erling Smørgrav 			if(errno == EINPROGRESS || errno == EWOULDBLOCK)
2405b5663de9SDag-Erling Smørgrav 				return 1;
2406b5663de9SDag-Erling Smørgrav #endif
2407b5663de9SDag-Erling Smørgrav 			if(errno == EINTR || errno == EAGAIN)
2408b5663de9SDag-Erling Smørgrav 				return 1;
2409b5663de9SDag-Erling Smørgrav 			/* Not handling EISCONN here as shouldn't ever hit that case.*/
241024e36522SCy Schubert 			if(errno != EPIPE
241124e36522SCy Schubert #ifdef EOPNOTSUPP
241224e36522SCy Schubert 				/* if /proc/sys/net/ipv4/tcp_fastopen is
241324e36522SCy Schubert 				 * disabled on Linux, sendmsg may return
241424e36522SCy Schubert 				 * 'Operation not supported', if so
241524e36522SCy Schubert 				 * fallthrough to ordinary connect. */
241624e36522SCy Schubert 				&& errno != EOPNOTSUPP
241724e36522SCy Schubert #endif
241824e36522SCy Schubert 				&& errno != 0) {
241924e36522SCy Schubert 				if(verbosity < 2)
2420b5663de9SDag-Erling Smørgrav 					return 0; /* silence lots of chatter in the logs */
2421b5663de9SDag-Erling Smørgrav 				log_err_addr("tcp sendmsg", strerror(errno),
2422865f46b2SCy Schubert 					&c->repinfo.remote_addr,
2423865f46b2SCy Schubert 					c->repinfo.remote_addrlen);
2424b5663de9SDag-Erling Smørgrav 				return 0;
2425c7f4d7adSDag-Erling Smørgrav 			}
242624e36522SCy Schubert 			verbose(VERB_ALGO, "tcp sendmsg for fastopen failed (with %s), try normal connect", strerror(errno));
2427c7f4d7adSDag-Erling Smørgrav 			/* fallthrough to nonFASTOPEN
2428c7f4d7adSDag-Erling Smørgrav 			 * (MSG_FASTOPEN on Linux 3 produces EPIPE)
2429c7f4d7adSDag-Erling Smørgrav 			 * we need to perform connect() */
2430865f46b2SCy Schubert 			if(connect(fd, (struct sockaddr *)&c->repinfo.remote_addr,
2431865f46b2SCy Schubert 				c->repinfo.remote_addrlen) == -1) {
2432c7f4d7adSDag-Erling Smørgrav #ifdef EINPROGRESS
2433c7f4d7adSDag-Erling Smørgrav 				if(errno == EINPROGRESS)
2434c7f4d7adSDag-Erling Smørgrav 					return 1; /* wait until connect done*/
2435c7f4d7adSDag-Erling Smørgrav #endif
2436c7f4d7adSDag-Erling Smørgrav #ifdef USE_WINSOCK
2437c7f4d7adSDag-Erling Smørgrav 				if(WSAGetLastError() == WSAEINPROGRESS ||
2438c7f4d7adSDag-Erling Smørgrav 					WSAGetLastError() == WSAEWOULDBLOCK)
2439c7f4d7adSDag-Erling Smørgrav 					return 1; /* wait until connect done*/
2440c7f4d7adSDag-Erling Smørgrav #endif
2441c7f4d7adSDag-Erling Smørgrav 				if(tcp_connect_errno_needs_log(
2442865f46b2SCy Schubert 					(struct sockaddr *)&c->repinfo.remote_addr,
2443865f46b2SCy Schubert 					c->repinfo.remote_addrlen)) {
2444c7f4d7adSDag-Erling Smørgrav 					log_err_addr("outgoing tcp: connect after EPIPE for fastopen",
2445865f46b2SCy Schubert 						strerror(errno),
2446865f46b2SCy Schubert 						&c->repinfo.remote_addr,
2447865f46b2SCy Schubert 						c->repinfo.remote_addrlen);
2448c7f4d7adSDag-Erling Smørgrav 				}
2449c7f4d7adSDag-Erling Smørgrav 				return 0;
2450c7f4d7adSDag-Erling Smørgrav 			}
2451c7f4d7adSDag-Erling Smørgrav 
2452b5663de9SDag-Erling Smørgrav 		} else {
2453369c6923SCy Schubert 			if(c->tcp_write_and_read) {
2454369c6923SCy Schubert 				c->tcp_write_byte_count += r;
2455369c6923SCy Schubert 				if(c->tcp_write_byte_count < sizeof(uint16_t))
2456369c6923SCy Schubert 					return 1;
2457369c6923SCy Schubert 			} else {
2458b5663de9SDag-Erling Smørgrav 				c->tcp_byte_count += r;
2459b5663de9SDag-Erling Smørgrav 				if(c->tcp_byte_count < sizeof(uint16_t))
2460b5663de9SDag-Erling Smørgrav 					return 1;
246165b390aaSDag-Erling Smørgrav 				sldns_buffer_set_position(buffer, c->tcp_byte_count -
2462b5663de9SDag-Erling Smørgrav 					sizeof(uint16_t));
2463369c6923SCy Schubert 			}
2464369c6923SCy Schubert 			if((!c->tcp_write_and_read && sldns_buffer_remaining(buffer) == 0) || (c->tcp_write_and_read && c->tcp_write_byte_count == c->tcp_write_pkt_len + 2)) {
2465b5663de9SDag-Erling Smørgrav 				tcp_callback_writer(c);
2466b5663de9SDag-Erling Smørgrav 				return 1;
2467b5663de9SDag-Erling Smørgrav 			}
2468b5663de9SDag-Erling Smørgrav 		}
2469b5663de9SDag-Erling Smørgrav 	}
2470b5663de9SDag-Erling Smørgrav #endif /* USE_MSG_FASTOPEN */
2471b5663de9SDag-Erling Smørgrav 
2472369c6923SCy Schubert 	if((c->tcp_write_and_read?c->tcp_write_byte_count:c->tcp_byte_count) < sizeof(uint16_t)) {
2473369c6923SCy Schubert 		uint16_t len = htons(c->tcp_write_and_read?c->tcp_write_pkt_len:sldns_buffer_limit(buffer));
2474b7579f77SDag-Erling Smørgrav #ifdef HAVE_WRITEV
2475b7579f77SDag-Erling Smørgrav 		struct iovec iov[2];
2476369c6923SCy Schubert 		if(c->tcp_write_and_read) {
2477369c6923SCy Schubert 			iov[0].iov_base = (uint8_t*)&len + c->tcp_write_byte_count;
2478369c6923SCy Schubert 			iov[0].iov_len = sizeof(uint16_t) - c->tcp_write_byte_count;
2479369c6923SCy Schubert 			iov[1].iov_base = c->tcp_write_pkt;
2480369c6923SCy Schubert 			iov[1].iov_len = c->tcp_write_pkt_len;
2481369c6923SCy Schubert 		} else {
2482b7579f77SDag-Erling Smørgrav 			iov[0].iov_base = (uint8_t*)&len + c->tcp_byte_count;
2483b7579f77SDag-Erling Smørgrav 			iov[0].iov_len = sizeof(uint16_t) - c->tcp_byte_count;
248465b390aaSDag-Erling Smørgrav 			iov[1].iov_base = sldns_buffer_begin(buffer);
248565b390aaSDag-Erling Smørgrav 			iov[1].iov_len = sldns_buffer_limit(buffer);
2486369c6923SCy Schubert 		}
2487b7579f77SDag-Erling Smørgrav 		log_assert(iov[0].iov_len > 0);
2488b7579f77SDag-Erling Smørgrav 		r = writev(fd, iov, 2);
2489b7579f77SDag-Erling Smørgrav #else /* HAVE_WRITEV */
2490369c6923SCy Schubert 		if(c->tcp_write_and_read) {
2491369c6923SCy Schubert 			r = send(fd, (void*)(((uint8_t*)&len)+c->tcp_write_byte_count),
2492369c6923SCy Schubert 				sizeof(uint16_t)-c->tcp_write_byte_count, 0);
2493369c6923SCy Schubert 		} else {
2494b7579f77SDag-Erling Smørgrav 			r = send(fd, (void*)(((uint8_t*)&len)+c->tcp_byte_count),
2495b7579f77SDag-Erling Smørgrav 				sizeof(uint16_t)-c->tcp_byte_count, 0);
2496369c6923SCy Schubert 		}
2497b7579f77SDag-Erling Smørgrav #endif /* HAVE_WRITEV */
2498b7579f77SDag-Erling Smørgrav 		if(r == -1) {
2499b7579f77SDag-Erling Smørgrav #ifndef USE_WINSOCK
2500b7579f77SDag-Erling Smørgrav #  ifdef EPIPE
2501b7579f77SDag-Erling Smørgrav                 	if(errno == EPIPE && verbosity < 2)
2502b7579f77SDag-Erling Smørgrav                         	return 0; /* silence 'broken pipe' */
2503b7579f77SDag-Erling Smørgrav   #endif
2504b7579f77SDag-Erling Smørgrav 			if(errno == EINTR || errno == EAGAIN)
2505b7579f77SDag-Erling Smørgrav 				return 1;
2506e86b9096SDag-Erling Smørgrav #ifdef ECONNRESET
2507e86b9096SDag-Erling Smørgrav 			if(errno == ECONNRESET && verbosity < 2)
2508e86b9096SDag-Erling Smørgrav 				return 0; /* silence reset by peer */
2509e86b9096SDag-Erling Smørgrav #endif
2510ff825849SDag-Erling Smørgrav #  ifdef HAVE_WRITEV
2511ff825849SDag-Erling Smørgrav 			log_err_addr("tcp writev", strerror(errno),
2512865f46b2SCy Schubert 				&c->repinfo.remote_addr,
2513865f46b2SCy Schubert 				c->repinfo.remote_addrlen);
2514ff825849SDag-Erling Smørgrav #  else /* HAVE_WRITEV */
2515ff825849SDag-Erling Smørgrav 			log_err_addr("tcp send s", strerror(errno),
2516865f46b2SCy Schubert 				&c->repinfo.remote_addr,
2517865f46b2SCy Schubert 				c->repinfo.remote_addrlen);
2518ff825849SDag-Erling Smørgrav #  endif /* HAVE_WRITEV */
2519b7579f77SDag-Erling Smørgrav #else
2520b7579f77SDag-Erling Smørgrav 			if(WSAGetLastError() == WSAENOTCONN)
2521b7579f77SDag-Erling Smørgrav 				return 1;
2522b7579f77SDag-Erling Smørgrav 			if(WSAGetLastError() == WSAEINPROGRESS)
2523b7579f77SDag-Erling Smørgrav 				return 1;
2524b7579f77SDag-Erling Smørgrav 			if(WSAGetLastError() == WSAEWOULDBLOCK) {
2525e2d15004SDag-Erling Smørgrav 				ub_winsock_tcp_wouldblock(c->ev->ev,
2526e2d15004SDag-Erling Smørgrav 					UB_EV_WRITE);
2527b7579f77SDag-Erling Smørgrav 				return 1;
2528b7579f77SDag-Erling Smørgrav 			}
2529e86b9096SDag-Erling Smørgrav 			if(WSAGetLastError() == WSAECONNRESET && verbosity < 2)
2530e86b9096SDag-Erling Smørgrav 				return 0; /* silence reset by peer */
2531ff825849SDag-Erling Smørgrav 			log_err_addr("tcp send s",
2532ff825849SDag-Erling Smørgrav 				wsa_strerror(WSAGetLastError()),
2533865f46b2SCy Schubert 				&c->repinfo.remote_addr,
2534865f46b2SCy Schubert 				c->repinfo.remote_addrlen);
2535b7579f77SDag-Erling Smørgrav #endif
2536b7579f77SDag-Erling Smørgrav 			return 0;
2537b7579f77SDag-Erling Smørgrav 		}
2538369c6923SCy Schubert 		if(c->tcp_write_and_read) {
2539369c6923SCy Schubert 			c->tcp_write_byte_count += r;
2540369c6923SCy Schubert 			if(c->tcp_write_byte_count < sizeof(uint16_t))
2541369c6923SCy Schubert 				return 1;
2542369c6923SCy Schubert 		} else {
2543b7579f77SDag-Erling Smørgrav 			c->tcp_byte_count += r;
2544b7579f77SDag-Erling Smørgrav 			if(c->tcp_byte_count < sizeof(uint16_t))
2545b7579f77SDag-Erling Smørgrav 				return 1;
254665b390aaSDag-Erling Smørgrav 			sldns_buffer_set_position(buffer, c->tcp_byte_count -
2547b7579f77SDag-Erling Smørgrav 				sizeof(uint16_t));
2548369c6923SCy Schubert 		}
2549369c6923SCy Schubert 		if((!c->tcp_write_and_read && sldns_buffer_remaining(buffer) == 0) || (c->tcp_write_and_read && c->tcp_write_byte_count == c->tcp_write_pkt_len + 2)) {
2550b7579f77SDag-Erling Smørgrav 			tcp_callback_writer(c);
2551b7579f77SDag-Erling Smørgrav 			return 1;
2552b7579f77SDag-Erling Smørgrav 		}
2553b7579f77SDag-Erling Smørgrav 	}
2554369c6923SCy Schubert 	log_assert(c->tcp_write_and_read || sldns_buffer_remaining(buffer) > 0);
2555369c6923SCy Schubert 	log_assert(!c->tcp_write_and_read || c->tcp_write_byte_count < c->tcp_write_pkt_len + 2);
2556369c6923SCy Schubert 	if(c->tcp_write_and_read) {
2557f44e67d1SCy Schubert 		r = send(fd, (void*)(c->tcp_write_pkt + c->tcp_write_byte_count - 2),
2558369c6923SCy Schubert 			c->tcp_write_pkt_len + 2 - c->tcp_write_byte_count, 0);
2559369c6923SCy Schubert 	} else {
256065b390aaSDag-Erling Smørgrav 		r = send(fd, (void*)sldns_buffer_current(buffer),
256165b390aaSDag-Erling Smørgrav 			sldns_buffer_remaining(buffer), 0);
2562369c6923SCy Schubert 	}
2563b7579f77SDag-Erling Smørgrav 	if(r == -1) {
2564b7579f77SDag-Erling Smørgrav #ifndef USE_WINSOCK
2565b7579f77SDag-Erling Smørgrav 		if(errno == EINTR || errno == EAGAIN)
2566b7579f77SDag-Erling Smørgrav 			return 1;
2567e86b9096SDag-Erling Smørgrav #ifdef ECONNRESET
2568e86b9096SDag-Erling Smørgrav 		if(errno == ECONNRESET && verbosity < 2)
2569e86b9096SDag-Erling Smørgrav 			return 0; /* silence reset by peer */
2570e86b9096SDag-Erling Smørgrav #endif
2571b7579f77SDag-Erling Smørgrav #else
2572b7579f77SDag-Erling Smørgrav 		if(WSAGetLastError() == WSAEINPROGRESS)
2573b7579f77SDag-Erling Smørgrav 			return 1;
2574b7579f77SDag-Erling Smørgrav 		if(WSAGetLastError() == WSAEWOULDBLOCK) {
2575e2d15004SDag-Erling Smørgrav 			ub_winsock_tcp_wouldblock(c->ev->ev, UB_EV_WRITE);
2576b7579f77SDag-Erling Smørgrav 			return 1;
2577b7579f77SDag-Erling Smørgrav 		}
2578e86b9096SDag-Erling Smørgrav 		if(WSAGetLastError() == WSAECONNRESET && verbosity < 2)
2579e86b9096SDag-Erling Smørgrav 			return 0; /* silence reset by peer */
2580b7579f77SDag-Erling Smørgrav #endif
2581c0caa2e2SCy Schubert 		log_err_addr("tcp send r", sock_strerror(errno),
2582865f46b2SCy Schubert 			&c->repinfo.remote_addr,
2583865f46b2SCy Schubert 			c->repinfo.remote_addrlen);
2584b7579f77SDag-Erling Smørgrav 		return 0;
2585b7579f77SDag-Erling Smørgrav 	}
2586369c6923SCy Schubert 	if(c->tcp_write_and_read) {
2587369c6923SCy Schubert 		c->tcp_write_byte_count += r;
2588369c6923SCy Schubert 	} else {
258965b390aaSDag-Erling Smørgrav 		sldns_buffer_skip(buffer, r);
2590369c6923SCy Schubert 	}
2591b7579f77SDag-Erling Smørgrav 
2592369c6923SCy Schubert 	if((!c->tcp_write_and_read && sldns_buffer_remaining(buffer) == 0) || (c->tcp_write_and_read && c->tcp_write_byte_count == c->tcp_write_pkt_len + 2)) {
2593b7579f77SDag-Erling Smørgrav 		tcp_callback_writer(c);
2594b7579f77SDag-Erling Smørgrav 	}
2595b7579f77SDag-Erling Smørgrav 
2596b7579f77SDag-Erling Smørgrav 	return 1;
2597b7579f77SDag-Erling Smørgrav }
2598b7579f77SDag-Erling Smørgrav 
25991838dec3SCy Schubert /** read again to drain buffers when there could be more to read, returns 0
26001838dec3SCy Schubert  * on failure which means the comm point is closed. */
26011838dec3SCy Schubert static int
2602e86b9096SDag-Erling Smørgrav tcp_req_info_read_again(int fd, struct comm_point* c)
2603e86b9096SDag-Erling Smørgrav {
2604e86b9096SDag-Erling Smørgrav 	while(c->tcp_req_info->read_again) {
2605e86b9096SDag-Erling Smørgrav 		int r;
2606e86b9096SDag-Erling Smørgrav 		c->tcp_req_info->read_again = 0;
2607e86b9096SDag-Erling Smørgrav 		if(c->tcp_is_reading)
2608e86b9096SDag-Erling Smørgrav 			r = comm_point_tcp_handle_read(fd, c, 0);
2609e86b9096SDag-Erling Smørgrav 		else 	r = comm_point_tcp_handle_write(fd, c);
2610e86b9096SDag-Erling Smørgrav 		if(!r) {
2611e86b9096SDag-Erling Smørgrav 			reclaim_tcp_handler(c);
2612e86b9096SDag-Erling Smørgrav 			if(!c->tcp_do_close) {
2613e86b9096SDag-Erling Smørgrav 				fptr_ok(fptr_whitelist_comm_point(
2614e86b9096SDag-Erling Smørgrav 					c->callback));
2615e86b9096SDag-Erling Smørgrav 				(void)(*c->callback)(c, c->cb_arg,
2616e86b9096SDag-Erling Smørgrav 					NETEVENT_CLOSED, NULL);
2617e86b9096SDag-Erling Smørgrav 			}
26181838dec3SCy Schubert 			return 0;
2619e86b9096SDag-Erling Smørgrav 		}
2620e86b9096SDag-Erling Smørgrav 	}
26211838dec3SCy Schubert 	return 1;
2622e86b9096SDag-Erling Smørgrav }
2623e86b9096SDag-Erling Smørgrav 
2624369c6923SCy Schubert /** read again to drain buffers when there could be more to read */
2625369c6923SCy Schubert static void
2626369c6923SCy Schubert tcp_more_read_again(int fd, struct comm_point* c)
2627369c6923SCy Schubert {
2628369c6923SCy Schubert 	/* if the packet is done, but another one could be waiting on
2629369c6923SCy Schubert 	 * the connection, the callback signals this, and we try again */
2630369c6923SCy Schubert 	/* this continues until the read routines get EAGAIN or so,
2631369c6923SCy Schubert 	 * and thus does not call the callback, and the bool is 0 */
2632369c6923SCy Schubert 	int* moreread = c->tcp_more_read_again;
2633369c6923SCy Schubert 	while(moreread && *moreread) {
2634369c6923SCy Schubert 		*moreread = 0;
2635369c6923SCy Schubert 		if(!comm_point_tcp_handle_read(fd, c, 0)) {
2636369c6923SCy Schubert 			reclaim_tcp_handler(c);
2637369c6923SCy Schubert 			if(!c->tcp_do_close) {
2638369c6923SCy Schubert 				fptr_ok(fptr_whitelist_comm_point(
2639369c6923SCy Schubert 					c->callback));
2640369c6923SCy Schubert 				(void)(*c->callback)(c, c->cb_arg,
2641369c6923SCy Schubert 					NETEVENT_CLOSED, NULL);
2642369c6923SCy Schubert 			}
2643369c6923SCy Schubert 			return;
2644369c6923SCy Schubert 		}
2645369c6923SCy Schubert 	}
2646369c6923SCy Schubert }
2647369c6923SCy Schubert 
2648369c6923SCy Schubert /** write again to fill up when there could be more to write */
2649369c6923SCy Schubert static void
2650369c6923SCy Schubert tcp_more_write_again(int fd, struct comm_point* c)
2651369c6923SCy Schubert {
2652369c6923SCy Schubert 	/* if the packet is done, but another is waiting to be written,
2653369c6923SCy Schubert 	 * the callback signals it and we try again. */
2654369c6923SCy Schubert 	/* this continues until the write routines get EAGAIN or so,
2655369c6923SCy Schubert 	 * and thus does not call the callback, and the bool is 0 */
2656369c6923SCy Schubert 	int* morewrite = c->tcp_more_write_again;
2657369c6923SCy Schubert 	while(morewrite && *morewrite) {
2658369c6923SCy Schubert 		*morewrite = 0;
2659369c6923SCy Schubert 		if(!comm_point_tcp_handle_write(fd, c)) {
2660369c6923SCy Schubert 			reclaim_tcp_handler(c);
2661369c6923SCy Schubert 			if(!c->tcp_do_close) {
2662369c6923SCy Schubert 				fptr_ok(fptr_whitelist_comm_point(
2663369c6923SCy Schubert 					c->callback));
2664369c6923SCy Schubert 				(void)(*c->callback)(c, c->cb_arg,
2665369c6923SCy Schubert 					NETEVENT_CLOSED, NULL);
2666369c6923SCy Schubert 			}
2667369c6923SCy Schubert 			return;
2668369c6923SCy Schubert 		}
2669369c6923SCy Schubert 	}
2670369c6923SCy Schubert }
2671369c6923SCy Schubert 
2672b7579f77SDag-Erling Smørgrav void
2673b7579f77SDag-Erling Smørgrav comm_point_tcp_handle_callback(int fd, short event, void* arg)
2674b7579f77SDag-Erling Smørgrav {
2675b7579f77SDag-Erling Smørgrav 	struct comm_point* c = (struct comm_point*)arg;
2676b7579f77SDag-Erling Smørgrav 	log_assert(c->type == comm_tcp);
2677e2d15004SDag-Erling Smørgrav 	ub_comm_base_now(c->ev->base);
2678b7579f77SDag-Erling Smørgrav 
26791838dec3SCy Schubert 	if(c->fd == -1 || c->fd != fd)
26801838dec3SCy Schubert 		return; /* duplicate event, but commpoint closed. */
26811838dec3SCy Schubert 
268265b390aaSDag-Erling Smørgrav #ifdef USE_DNSCRYPT
268365b390aaSDag-Erling Smørgrav 	/* Initialize if this is a dnscrypt socket */
268465b390aaSDag-Erling Smørgrav 	if(c->tcp_parent) {
268565b390aaSDag-Erling Smørgrav 		c->dnscrypt = c->tcp_parent->dnscrypt;
268665b390aaSDag-Erling Smørgrav 	}
268765b390aaSDag-Erling Smørgrav 	if(c->dnscrypt && c->dnscrypt_buffer == c->buffer) {
268865b390aaSDag-Erling Smørgrav 		c->dnscrypt_buffer = sldns_buffer_new(sldns_buffer_capacity(c->buffer));
268965b390aaSDag-Erling Smørgrav 		if(!c->dnscrypt_buffer) {
269065b390aaSDag-Erling Smørgrav 			log_err("Could not allocate dnscrypt buffer");
2691971980c3SDag-Erling Smørgrav 			reclaim_tcp_handler(c);
2692971980c3SDag-Erling Smørgrav 			if(!c->tcp_do_close) {
2693971980c3SDag-Erling Smørgrav 				fptr_ok(fptr_whitelist_comm_point(
2694971980c3SDag-Erling Smørgrav 					c->callback));
2695971980c3SDag-Erling Smørgrav 				(void)(*c->callback)(c, c->cb_arg,
2696971980c3SDag-Erling Smørgrav 					NETEVENT_CLOSED, NULL);
2697971980c3SDag-Erling Smørgrav 			}
269865b390aaSDag-Erling Smørgrav 			return;
269965b390aaSDag-Erling Smørgrav 		}
270065b390aaSDag-Erling Smørgrav 	}
270165b390aaSDag-Erling Smørgrav #endif
270265b390aaSDag-Erling Smørgrav 
2703e86b9096SDag-Erling Smørgrav 	if(event&UB_EV_TIMEOUT) {
2704e86b9096SDag-Erling Smørgrav 		verbose(VERB_QUERY, "tcp took too long, dropped");
2705e86b9096SDag-Erling Smørgrav 		reclaim_tcp_handler(c);
2706e86b9096SDag-Erling Smørgrav 		if(!c->tcp_do_close) {
2707e86b9096SDag-Erling Smørgrav 			fptr_ok(fptr_whitelist_comm_point(c->callback));
2708e86b9096SDag-Erling Smørgrav 			(void)(*c->callback)(c, c->cb_arg,
2709e86b9096SDag-Erling Smørgrav 				NETEVENT_TIMEOUT, NULL);
2710e86b9096SDag-Erling Smørgrav 		}
2711e86b9096SDag-Erling Smørgrav 		return;
2712e86b9096SDag-Erling Smørgrav 	}
2713369c6923SCy Schubert 	if(event&UB_EV_READ
2714369c6923SCy Schubert #ifdef USE_MSG_FASTOPEN
2715369c6923SCy Schubert 		&& !(c->tcp_do_fastopen && (event&UB_EV_WRITE))
2716369c6923SCy Schubert #endif
2717369c6923SCy Schubert 		) {
2718e86b9096SDag-Erling Smørgrav 		int has_tcpq = (c->tcp_req_info != NULL);
2719369c6923SCy Schubert 		int* moreread = c->tcp_more_read_again;
2720b7579f77SDag-Erling Smørgrav 		if(!comm_point_tcp_handle_read(fd, c, 0)) {
2721b7579f77SDag-Erling Smørgrav 			reclaim_tcp_handler(c);
2722b7579f77SDag-Erling Smørgrav 			if(!c->tcp_do_close) {
2723b7579f77SDag-Erling Smørgrav 				fptr_ok(fptr_whitelist_comm_point(
2724b7579f77SDag-Erling Smørgrav 					c->callback));
2725b7579f77SDag-Erling Smørgrav 				(void)(*c->callback)(c, c->cb_arg,
2726b7579f77SDag-Erling Smørgrav 					NETEVENT_CLOSED, NULL);
2727b7579f77SDag-Erling Smørgrav 			}
2728369c6923SCy Schubert 			return;
2729b7579f77SDag-Erling Smørgrav 		}
27301838dec3SCy Schubert 		if(has_tcpq && c->tcp_req_info && c->tcp_req_info->read_again) {
27311838dec3SCy Schubert 			if(!tcp_req_info_read_again(fd, c))
27321838dec3SCy Schubert 				return;
27331838dec3SCy Schubert 		}
2734369c6923SCy Schubert 		if(moreread && *moreread)
2735369c6923SCy Schubert 			tcp_more_read_again(fd, c);
2736b7579f77SDag-Erling Smørgrav 		return;
2737b7579f77SDag-Erling Smørgrav 	}
2738e2d15004SDag-Erling Smørgrav 	if(event&UB_EV_WRITE) {
2739e86b9096SDag-Erling Smørgrav 		int has_tcpq = (c->tcp_req_info != NULL);
2740369c6923SCy Schubert 		int* morewrite = c->tcp_more_write_again;
2741b7579f77SDag-Erling Smørgrav 		if(!comm_point_tcp_handle_write(fd, c)) {
2742b7579f77SDag-Erling Smørgrav 			reclaim_tcp_handler(c);
2743b7579f77SDag-Erling Smørgrav 			if(!c->tcp_do_close) {
2744b7579f77SDag-Erling Smørgrav 				fptr_ok(fptr_whitelist_comm_point(
2745b7579f77SDag-Erling Smørgrav 					c->callback));
2746b7579f77SDag-Erling Smørgrav 				(void)(*c->callback)(c, c->cb_arg,
2747b7579f77SDag-Erling Smørgrav 					NETEVENT_CLOSED, NULL);
2748b7579f77SDag-Erling Smørgrav 			}
2749369c6923SCy Schubert 			return;
2750b7579f77SDag-Erling Smørgrav 		}
27511838dec3SCy Schubert 		if(has_tcpq && c->tcp_req_info && c->tcp_req_info->read_again) {
27521838dec3SCy Schubert 			if(!tcp_req_info_read_again(fd, c))
27531838dec3SCy Schubert 				return;
27541838dec3SCy Schubert 		}
2755369c6923SCy Schubert 		if(morewrite && *morewrite)
2756369c6923SCy Schubert 			tcp_more_write_again(fd, c);
2757b7579f77SDag-Erling Smørgrav 		return;
2758b7579f77SDag-Erling Smørgrav 	}
2759b7579f77SDag-Erling Smørgrav 	log_err("Ignored event %d for tcphdl.", event);
2760b7579f77SDag-Erling Smørgrav }
2761b7579f77SDag-Erling Smørgrav 
276257bddd21SDag-Erling Smørgrav /** Make http handler free for next assignment */
276357bddd21SDag-Erling Smørgrav static void
276457bddd21SDag-Erling Smørgrav reclaim_http_handler(struct comm_point* c)
276557bddd21SDag-Erling Smørgrav {
276657bddd21SDag-Erling Smørgrav 	log_assert(c->type == comm_http);
276757bddd21SDag-Erling Smørgrav 	if(c->ssl) {
276857bddd21SDag-Erling Smørgrav #ifdef HAVE_SSL
276957bddd21SDag-Erling Smørgrav 		SSL_shutdown(c->ssl);
277057bddd21SDag-Erling Smørgrav 		SSL_free(c->ssl);
277157bddd21SDag-Erling Smørgrav 		c->ssl = NULL;
277257bddd21SDag-Erling Smørgrav #endif
277357bddd21SDag-Erling Smørgrav 	}
277457bddd21SDag-Erling Smørgrav 	comm_point_close(c);
277557bddd21SDag-Erling Smørgrav 	if(c->tcp_parent) {
277624e36522SCy Schubert 		if(c != c->tcp_parent->tcp_free) {
277757bddd21SDag-Erling Smørgrav 			c->tcp_parent->cur_tcp_count--;
277857bddd21SDag-Erling Smørgrav 			c->tcp_free = c->tcp_parent->tcp_free;
277957bddd21SDag-Erling Smørgrav 			c->tcp_parent->tcp_free = c;
278024e36522SCy Schubert 		}
278157bddd21SDag-Erling Smørgrav 		if(!c->tcp_free) {
278257bddd21SDag-Erling Smørgrav 			/* re-enable listening on accept socket */
278357bddd21SDag-Erling Smørgrav 			comm_point_start_listening(c->tcp_parent, -1, -1);
278457bddd21SDag-Erling Smørgrav 		}
278557bddd21SDag-Erling Smørgrav 	}
278657bddd21SDag-Erling Smørgrav }
278757bddd21SDag-Erling Smørgrav 
278857bddd21SDag-Erling Smørgrav /** read more data for http (with ssl) */
278957bddd21SDag-Erling Smørgrav static int
279057bddd21SDag-Erling Smørgrav ssl_http_read_more(struct comm_point* c)
279157bddd21SDag-Erling Smørgrav {
279257bddd21SDag-Erling Smørgrav #ifdef HAVE_SSL
279357bddd21SDag-Erling Smørgrav 	int r;
279457bddd21SDag-Erling Smørgrav 	log_assert(sldns_buffer_remaining(c->buffer) > 0);
279557bddd21SDag-Erling Smørgrav 	ERR_clear_error();
279657bddd21SDag-Erling Smørgrav 	r = SSL_read(c->ssl, (void*)sldns_buffer_current(c->buffer),
279757bddd21SDag-Erling Smørgrav 		(int)sldns_buffer_remaining(c->buffer));
279857bddd21SDag-Erling Smørgrav 	if(r <= 0) {
279957bddd21SDag-Erling Smørgrav 		int want = SSL_get_error(c->ssl, r);
280057bddd21SDag-Erling Smørgrav 		if(want == SSL_ERROR_ZERO_RETURN) {
280157bddd21SDag-Erling Smørgrav 			return 0; /* shutdown, closed */
280257bddd21SDag-Erling Smørgrav 		} else if(want == SSL_ERROR_WANT_READ) {
280357bddd21SDag-Erling Smørgrav 			return 1; /* read more later */
280457bddd21SDag-Erling Smørgrav 		} else if(want == SSL_ERROR_WANT_WRITE) {
280557bddd21SDag-Erling Smørgrav 			c->ssl_shake_state = comm_ssl_shake_hs_write;
280657bddd21SDag-Erling Smørgrav 			comm_point_listen_for_rw(c, 0, 1);
280757bddd21SDag-Erling Smørgrav 			return 1;
280857bddd21SDag-Erling Smørgrav 		} else if(want == SSL_ERROR_SYSCALL) {
2809e86b9096SDag-Erling Smørgrav #ifdef ECONNRESET
2810e86b9096SDag-Erling Smørgrav 			if(errno == ECONNRESET && verbosity < 2)
2811e86b9096SDag-Erling Smørgrav 				return 0; /* silence reset by peer */
2812e86b9096SDag-Erling Smørgrav #endif
281357bddd21SDag-Erling Smørgrav 			if(errno != 0)
281457bddd21SDag-Erling Smørgrav 				log_err("SSL_read syscall: %s",
281557bddd21SDag-Erling Smørgrav 					strerror(errno));
281657bddd21SDag-Erling Smørgrav 			return 0;
281757bddd21SDag-Erling Smørgrav 		}
281857bddd21SDag-Erling Smørgrav 		log_crypto_err("could not SSL_read");
281957bddd21SDag-Erling Smørgrav 		return 0;
282057bddd21SDag-Erling Smørgrav 	}
28215469a995SCy Schubert 	verbose(VERB_ALGO, "ssl http read more skip to %d + %d",
28225469a995SCy Schubert 		(int)sldns_buffer_position(c->buffer), (int)r);
282357bddd21SDag-Erling Smørgrav 	sldns_buffer_skip(c->buffer, (ssize_t)r);
282457bddd21SDag-Erling Smørgrav 	return 1;
282557bddd21SDag-Erling Smørgrav #else
282657bddd21SDag-Erling Smørgrav 	(void)c;
282757bddd21SDag-Erling Smørgrav 	return 0;
282857bddd21SDag-Erling Smørgrav #endif /* HAVE_SSL */
282957bddd21SDag-Erling Smørgrav }
283057bddd21SDag-Erling Smørgrav 
283157bddd21SDag-Erling Smørgrav /** read more data for http */
283257bddd21SDag-Erling Smørgrav static int
283357bddd21SDag-Erling Smørgrav http_read_more(int fd, struct comm_point* c)
283457bddd21SDag-Erling Smørgrav {
283557bddd21SDag-Erling Smørgrav 	ssize_t r;
283657bddd21SDag-Erling Smørgrav 	log_assert(sldns_buffer_remaining(c->buffer) > 0);
283757bddd21SDag-Erling Smørgrav 	r = recv(fd, (void*)sldns_buffer_current(c->buffer),
2838865f46b2SCy Schubert 		sldns_buffer_remaining(c->buffer), MSG_DONTWAIT);
283957bddd21SDag-Erling Smørgrav 	if(r == 0) {
284057bddd21SDag-Erling Smørgrav 		return 0;
284157bddd21SDag-Erling Smørgrav 	} else if(r == -1) {
284257bddd21SDag-Erling Smørgrav #ifndef USE_WINSOCK
284357bddd21SDag-Erling Smørgrav 		if(errno == EINTR || errno == EAGAIN)
284457bddd21SDag-Erling Smørgrav 			return 1;
284557bddd21SDag-Erling Smørgrav #else /* USE_WINSOCK */
284657bddd21SDag-Erling Smørgrav 		if(WSAGetLastError() == WSAECONNRESET)
284757bddd21SDag-Erling Smørgrav 			return 0;
284857bddd21SDag-Erling Smørgrav 		if(WSAGetLastError() == WSAEINPROGRESS)
284957bddd21SDag-Erling Smørgrav 			return 1;
285057bddd21SDag-Erling Smørgrav 		if(WSAGetLastError() == WSAEWOULDBLOCK) {
285157bddd21SDag-Erling Smørgrav 			ub_winsock_tcp_wouldblock(c->ev->ev, UB_EV_READ);
285257bddd21SDag-Erling Smørgrav 			return 1;
285357bddd21SDag-Erling Smørgrav 		}
285457bddd21SDag-Erling Smørgrav #endif
2855c0caa2e2SCy Schubert 		log_err_addr("read (in http r)", sock_strerror(errno),
2856865f46b2SCy Schubert 			&c->repinfo.remote_addr, c->repinfo.remote_addrlen);
285757bddd21SDag-Erling Smørgrav 		return 0;
285857bddd21SDag-Erling Smørgrav 	}
28595469a995SCy Schubert 	verbose(VERB_ALGO, "http read more skip to %d + %d",
28605469a995SCy Schubert 		(int)sldns_buffer_position(c->buffer), (int)r);
286157bddd21SDag-Erling Smørgrav 	sldns_buffer_skip(c->buffer, r);
286257bddd21SDag-Erling Smørgrav 	return 1;
286357bddd21SDag-Erling Smørgrav }
286457bddd21SDag-Erling Smørgrav 
286557bddd21SDag-Erling Smørgrav /** return true if http header has been read (one line complete) */
286657bddd21SDag-Erling Smørgrav static int
286757bddd21SDag-Erling Smørgrav http_header_done(sldns_buffer* buf)
286857bddd21SDag-Erling Smørgrav {
286957bddd21SDag-Erling Smørgrav 	size_t i;
287057bddd21SDag-Erling Smørgrav 	for(i=sldns_buffer_position(buf); i<sldns_buffer_limit(buf); i++) {
287157bddd21SDag-Erling Smørgrav 		/* there was a \r before the \n, but we ignore that */
287257bddd21SDag-Erling Smørgrav 		if((char)sldns_buffer_read_u8_at(buf, i) == '\n')
287357bddd21SDag-Erling Smørgrav 			return 1;
287457bddd21SDag-Erling Smørgrav 	}
287557bddd21SDag-Erling Smørgrav 	return 0;
287657bddd21SDag-Erling Smørgrav }
287757bddd21SDag-Erling Smørgrav 
287857bddd21SDag-Erling Smørgrav /** return character string into buffer for header line, moves buffer
287957bddd21SDag-Erling Smørgrav  * past that line and puts zero terminator into linefeed-newline */
288057bddd21SDag-Erling Smørgrav static char*
288157bddd21SDag-Erling Smørgrav http_header_line(sldns_buffer* buf)
288257bddd21SDag-Erling Smørgrav {
288357bddd21SDag-Erling Smørgrav 	char* result = (char*)sldns_buffer_current(buf);
288457bddd21SDag-Erling Smørgrav 	size_t i;
288557bddd21SDag-Erling Smørgrav 	for(i=sldns_buffer_position(buf); i<sldns_buffer_limit(buf); i++) {
288657bddd21SDag-Erling Smørgrav 		/* terminate the string on the \r */
288757bddd21SDag-Erling Smørgrav 		if((char)sldns_buffer_read_u8_at(buf, i) == '\r')
288857bddd21SDag-Erling Smørgrav 			sldns_buffer_write_u8_at(buf, i, 0);
288957bddd21SDag-Erling Smørgrav 		/* terminate on the \n and skip past the it and done */
289057bddd21SDag-Erling Smørgrav 		if((char)sldns_buffer_read_u8_at(buf, i) == '\n') {
289157bddd21SDag-Erling Smørgrav 			sldns_buffer_write_u8_at(buf, i, 0);
289257bddd21SDag-Erling Smørgrav 			sldns_buffer_set_position(buf, i+1);
289357bddd21SDag-Erling Smørgrav 			return result;
289457bddd21SDag-Erling Smørgrav 		}
289557bddd21SDag-Erling Smørgrav 	}
289657bddd21SDag-Erling Smørgrav 	return NULL;
289757bddd21SDag-Erling Smørgrav }
289857bddd21SDag-Erling Smørgrav 
289957bddd21SDag-Erling Smørgrav /** move unread buffer to start and clear rest for putting the rest into it */
290057bddd21SDag-Erling Smørgrav static void
290157bddd21SDag-Erling Smørgrav http_moveover_buffer(sldns_buffer* buf)
290257bddd21SDag-Erling Smørgrav {
290357bddd21SDag-Erling Smørgrav 	size_t pos = sldns_buffer_position(buf);
290457bddd21SDag-Erling Smørgrav 	size_t len = sldns_buffer_remaining(buf);
290557bddd21SDag-Erling Smørgrav 	sldns_buffer_clear(buf);
290657bddd21SDag-Erling Smørgrav 	memmove(sldns_buffer_begin(buf), sldns_buffer_at(buf, pos), len);
290757bddd21SDag-Erling Smørgrav 	sldns_buffer_set_position(buf, len);
290857bddd21SDag-Erling Smørgrav }
290957bddd21SDag-Erling Smørgrav 
291057bddd21SDag-Erling Smørgrav /** a http header is complete, process it */
291157bddd21SDag-Erling Smørgrav static int
291257bddd21SDag-Erling Smørgrav http_process_initial_header(struct comm_point* c)
291357bddd21SDag-Erling Smørgrav {
291457bddd21SDag-Erling Smørgrav 	char* line = http_header_line(c->buffer);
291557bddd21SDag-Erling Smørgrav 	if(!line) return 1;
291657bddd21SDag-Erling Smørgrav 	verbose(VERB_ALGO, "http header: %s", line);
291757bddd21SDag-Erling Smørgrav 	if(strncasecmp(line, "HTTP/1.1 ", 9) == 0) {
291857bddd21SDag-Erling Smørgrav 		/* check returncode */
291957bddd21SDag-Erling Smørgrav 		if(line[9] != '2') {
292057bddd21SDag-Erling Smørgrav 			verbose(VERB_ALGO, "http bad status %s", line+9);
292157bddd21SDag-Erling Smørgrav 			return 0;
292257bddd21SDag-Erling Smørgrav 		}
292357bddd21SDag-Erling Smørgrav 	} else if(strncasecmp(line, "Content-Length: ", 16) == 0) {
292457bddd21SDag-Erling Smørgrav 		if(!c->http_is_chunked)
292557bddd21SDag-Erling Smørgrav 			c->tcp_byte_count = (size_t)atoi(line+16);
292657bddd21SDag-Erling Smørgrav 	} else if(strncasecmp(line, "Transfer-Encoding: chunked", 19+7) == 0) {
292757bddd21SDag-Erling Smørgrav 		c->tcp_byte_count = 0;
292857bddd21SDag-Erling Smørgrav 		c->http_is_chunked = 1;
292957bddd21SDag-Erling Smørgrav 	} else if(line[0] == 0) {
293057bddd21SDag-Erling Smørgrav 		/* end of initial headers */
293157bddd21SDag-Erling Smørgrav 		c->http_in_headers = 0;
293257bddd21SDag-Erling Smørgrav 		if(c->http_is_chunked)
293357bddd21SDag-Erling Smørgrav 			c->http_in_chunk_headers = 1;
293457bddd21SDag-Erling Smørgrav 		/* remove header text from front of buffer
293557bddd21SDag-Erling Smørgrav 		 * the buffer is going to be used to return the data segment
293657bddd21SDag-Erling Smørgrav 		 * itself and we don't want the header to get returned
293757bddd21SDag-Erling Smørgrav 		 * prepended with it */
293857bddd21SDag-Erling Smørgrav 		http_moveover_buffer(c->buffer);
293957bddd21SDag-Erling Smørgrav 		sldns_buffer_flip(c->buffer);
294057bddd21SDag-Erling Smørgrav 		return 1;
294157bddd21SDag-Erling Smørgrav 	}
294257bddd21SDag-Erling Smørgrav 	/* ignore other headers */
294357bddd21SDag-Erling Smørgrav 	return 1;
294457bddd21SDag-Erling Smørgrav }
294557bddd21SDag-Erling Smørgrav 
294657bddd21SDag-Erling Smørgrav /** a chunk header is complete, process it, return 0=fail, 1=continue next
294757bddd21SDag-Erling Smørgrav  * header line, 2=done with chunked transfer*/
294857bddd21SDag-Erling Smørgrav static int
294957bddd21SDag-Erling Smørgrav http_process_chunk_header(struct comm_point* c)
295057bddd21SDag-Erling Smørgrav {
295157bddd21SDag-Erling Smørgrav 	char* line = http_header_line(c->buffer);
295257bddd21SDag-Erling Smørgrav 	if(!line) return 1;
295357bddd21SDag-Erling Smørgrav 	if(c->http_in_chunk_headers == 3) {
295457bddd21SDag-Erling Smørgrav 		verbose(VERB_ALGO, "http chunk trailer: %s", line);
295557bddd21SDag-Erling Smørgrav 		/* are we done ? */
295657bddd21SDag-Erling Smørgrav 		if(line[0] == 0 && c->tcp_byte_count == 0) {
295757bddd21SDag-Erling Smørgrav 			/* callback of http reader when NETEVENT_DONE,
295857bddd21SDag-Erling Smørgrav 			 * end of data, with no data in buffer */
295957bddd21SDag-Erling Smørgrav 			sldns_buffer_set_position(c->buffer, 0);
296057bddd21SDag-Erling Smørgrav 			sldns_buffer_set_limit(c->buffer, 0);
296157bddd21SDag-Erling Smørgrav 			fptr_ok(fptr_whitelist_comm_point(c->callback));
296257bddd21SDag-Erling Smørgrav 			(void)(*c->callback)(c, c->cb_arg, NETEVENT_DONE, NULL);
296357bddd21SDag-Erling Smørgrav 			/* return that we are done */
296457bddd21SDag-Erling Smørgrav 			return 2;
296557bddd21SDag-Erling Smørgrav 		}
296657bddd21SDag-Erling Smørgrav 		if(line[0] == 0) {
296757bddd21SDag-Erling Smørgrav 			/* continue with header of the next chunk */
296857bddd21SDag-Erling Smørgrav 			c->http_in_chunk_headers = 1;
296957bddd21SDag-Erling Smørgrav 			/* remove header text from front of buffer */
297057bddd21SDag-Erling Smørgrav 			http_moveover_buffer(c->buffer);
297157bddd21SDag-Erling Smørgrav 			sldns_buffer_flip(c->buffer);
297257bddd21SDag-Erling Smørgrav 			return 1;
297357bddd21SDag-Erling Smørgrav 		}
297457bddd21SDag-Erling Smørgrav 		/* ignore further trail headers */
297557bddd21SDag-Erling Smørgrav 		return 1;
297657bddd21SDag-Erling Smørgrav 	}
297757bddd21SDag-Erling Smørgrav 	verbose(VERB_ALGO, "http chunk header: %s", line);
297857bddd21SDag-Erling Smørgrav 	if(c->http_in_chunk_headers == 1) {
297957bddd21SDag-Erling Smørgrav 		/* read chunked start line */
298057bddd21SDag-Erling Smørgrav 		char* end = NULL;
298157bddd21SDag-Erling Smørgrav 		c->tcp_byte_count = (size_t)strtol(line, &end, 16);
298257bddd21SDag-Erling Smørgrav 		if(end == line)
298357bddd21SDag-Erling Smørgrav 			return 0;
298457bddd21SDag-Erling Smørgrav 		c->http_in_chunk_headers = 0;
298557bddd21SDag-Erling Smørgrav 		/* remove header text from front of buffer */
298657bddd21SDag-Erling Smørgrav 		http_moveover_buffer(c->buffer);
298757bddd21SDag-Erling Smørgrav 		sldns_buffer_flip(c->buffer);
298857bddd21SDag-Erling Smørgrav 		if(c->tcp_byte_count == 0) {
298957bddd21SDag-Erling Smørgrav 			/* done with chunks, process chunk_trailer lines */
299057bddd21SDag-Erling Smørgrav 			c->http_in_chunk_headers = 3;
299157bddd21SDag-Erling Smørgrav 		}
299257bddd21SDag-Erling Smørgrav 		return 1;
299357bddd21SDag-Erling Smørgrav 	}
299457bddd21SDag-Erling Smørgrav 	/* ignore other headers */
299557bddd21SDag-Erling Smørgrav 	return 1;
299657bddd21SDag-Erling Smørgrav }
299757bddd21SDag-Erling Smørgrav 
29985469a995SCy Schubert /** handle nonchunked data segment, 0=fail, 1=wait */
299957bddd21SDag-Erling Smørgrav static int
300057bddd21SDag-Erling Smørgrav http_nonchunk_segment(struct comm_point* c)
300157bddd21SDag-Erling Smørgrav {
300257bddd21SDag-Erling Smørgrav 	/* c->buffer at position..limit has new data we read in.
300357bddd21SDag-Erling Smørgrav 	 * the buffer itself is full of nonchunked data.
300457bddd21SDag-Erling Smørgrav 	 * we are looking to read tcp_byte_count more data
300557bddd21SDag-Erling Smørgrav 	 * and then the transfer is done. */
300657bddd21SDag-Erling Smørgrav 	size_t remainbufferlen;
30075469a995SCy Schubert 	size_t got_now = sldns_buffer_limit(c->buffer);
300857bddd21SDag-Erling Smørgrav 	if(c->tcp_byte_count <= got_now) {
300957bddd21SDag-Erling Smørgrav 		/* done, this is the last data fragment */
301057bddd21SDag-Erling Smørgrav 		c->http_stored = 0;
301157bddd21SDag-Erling Smørgrav 		sldns_buffer_set_position(c->buffer, 0);
301257bddd21SDag-Erling Smørgrav 		fptr_ok(fptr_whitelist_comm_point(c->callback));
301357bddd21SDag-Erling Smørgrav 		(void)(*c->callback)(c, c->cb_arg, NETEVENT_DONE, NULL);
301457bddd21SDag-Erling Smørgrav 		return 1;
301557bddd21SDag-Erling Smørgrav 	}
301657bddd21SDag-Erling Smørgrav 	/* if we have the buffer space,
301757bddd21SDag-Erling Smørgrav 	 * read more data collected into the buffer */
301857bddd21SDag-Erling Smørgrav 	remainbufferlen = sldns_buffer_capacity(c->buffer) -
301957bddd21SDag-Erling Smørgrav 		sldns_buffer_limit(c->buffer);
30205469a995SCy Schubert 	if(remainbufferlen+got_now >= c->tcp_byte_count ||
3021a39a5a69SCy Schubert 		remainbufferlen >= (size_t)(c->ssl?16384:2048)) {
302257bddd21SDag-Erling Smørgrav 		size_t total = sldns_buffer_limit(c->buffer);
302357bddd21SDag-Erling Smørgrav 		sldns_buffer_clear(c->buffer);
302457bddd21SDag-Erling Smørgrav 		sldns_buffer_set_position(c->buffer, total);
302557bddd21SDag-Erling Smørgrav 		c->http_stored = total;
302657bddd21SDag-Erling Smørgrav 		/* return and wait to read more */
302757bddd21SDag-Erling Smørgrav 		return 1;
302857bddd21SDag-Erling Smørgrav 	}
302957bddd21SDag-Erling Smørgrav 	/* call callback with this data amount, then
303057bddd21SDag-Erling Smørgrav 	 * wait for more */
30315469a995SCy Schubert 	c->tcp_byte_count -= got_now;
303257bddd21SDag-Erling Smørgrav 	c->http_stored = 0;
303357bddd21SDag-Erling Smørgrav 	sldns_buffer_set_position(c->buffer, 0);
303457bddd21SDag-Erling Smørgrav 	fptr_ok(fptr_whitelist_comm_point(c->callback));
303557bddd21SDag-Erling Smørgrav 	(void)(*c->callback)(c, c->cb_arg, NETEVENT_NOERROR, NULL);
303657bddd21SDag-Erling Smørgrav 	/* c->callback has to buffer_clear(c->buffer). */
303757bddd21SDag-Erling Smørgrav 	/* return and wait to read more */
303857bddd21SDag-Erling Smørgrav 	return 1;
303957bddd21SDag-Erling Smørgrav }
304057bddd21SDag-Erling Smørgrav 
3041f44e67d1SCy Schubert /** handle chunked data segment, return 0=fail, 1=wait, 2=process more */
304257bddd21SDag-Erling Smørgrav static int
304357bddd21SDag-Erling Smørgrav http_chunked_segment(struct comm_point* c)
304457bddd21SDag-Erling Smørgrav {
304557bddd21SDag-Erling Smørgrav 	/* the c->buffer has from position..limit new data we read. */
304657bddd21SDag-Erling Smørgrav 	/* the current chunk has length tcp_byte_count.
304757bddd21SDag-Erling Smørgrav 	 * once we read that read more chunk headers.
304857bddd21SDag-Erling Smørgrav 	 */
304957bddd21SDag-Erling Smørgrav 	size_t remainbufferlen;
305057bddd21SDag-Erling Smørgrav 	size_t got_now = sldns_buffer_limit(c->buffer) - c->http_stored;
3051f44e67d1SCy Schubert 	verbose(VERB_ALGO, "http_chunked_segment: got now %d, tcpbytcount %d, http_stored %d, buffer pos %d, buffer limit %d", (int)got_now, (int)c->tcp_byte_count, (int)c->http_stored, (int)sldns_buffer_position(c->buffer), (int)sldns_buffer_limit(c->buffer));
305257bddd21SDag-Erling Smørgrav 	if(c->tcp_byte_count <= got_now) {
305357bddd21SDag-Erling Smørgrav 		/* the chunk has completed (with perhaps some extra data
305457bddd21SDag-Erling Smørgrav 		 * from next chunk header and next chunk) */
305557bddd21SDag-Erling Smørgrav 		/* save too much info into temp buffer */
305657bddd21SDag-Erling Smørgrav 		size_t fraglen;
305757bddd21SDag-Erling Smørgrav 		struct comm_reply repinfo;
305857bddd21SDag-Erling Smørgrav 		c->http_stored = 0;
305957bddd21SDag-Erling Smørgrav 		sldns_buffer_skip(c->buffer, (ssize_t)c->tcp_byte_count);
306057bddd21SDag-Erling Smørgrav 		sldns_buffer_clear(c->http_temp);
306157bddd21SDag-Erling Smørgrav 		sldns_buffer_write(c->http_temp,
306257bddd21SDag-Erling Smørgrav 			sldns_buffer_current(c->buffer),
306357bddd21SDag-Erling Smørgrav 			sldns_buffer_remaining(c->buffer));
306457bddd21SDag-Erling Smørgrav 		sldns_buffer_flip(c->http_temp);
306557bddd21SDag-Erling Smørgrav 
306657bddd21SDag-Erling Smørgrav 		/* callback with this fragment */
306757bddd21SDag-Erling Smørgrav 		fraglen = sldns_buffer_position(c->buffer);
306857bddd21SDag-Erling Smørgrav 		sldns_buffer_set_position(c->buffer, 0);
306957bddd21SDag-Erling Smørgrav 		sldns_buffer_set_limit(c->buffer, fraglen);
307057bddd21SDag-Erling Smørgrav 		repinfo = c->repinfo;
307157bddd21SDag-Erling Smørgrav 		fptr_ok(fptr_whitelist_comm_point(c->callback));
307257bddd21SDag-Erling Smørgrav 		(void)(*c->callback)(c, c->cb_arg, NETEVENT_NOERROR, &repinfo);
307357bddd21SDag-Erling Smørgrav 		/* c->callback has to buffer_clear(). */
307457bddd21SDag-Erling Smørgrav 
307557bddd21SDag-Erling Smørgrav 		/* is commpoint deleted? */
307657bddd21SDag-Erling Smørgrav 		if(!repinfo.c) {
307757bddd21SDag-Erling Smørgrav 			return 1;
307857bddd21SDag-Erling Smørgrav 		}
307957bddd21SDag-Erling Smørgrav 		/* copy waiting info */
308057bddd21SDag-Erling Smørgrav 		sldns_buffer_clear(c->buffer);
308157bddd21SDag-Erling Smørgrav 		sldns_buffer_write(c->buffer,
308257bddd21SDag-Erling Smørgrav 			sldns_buffer_begin(c->http_temp),
308357bddd21SDag-Erling Smørgrav 			sldns_buffer_remaining(c->http_temp));
308457bddd21SDag-Erling Smørgrav 		sldns_buffer_flip(c->buffer);
308557bddd21SDag-Erling Smørgrav 		/* process end of chunk trailer header lines, until
308657bddd21SDag-Erling Smørgrav 		 * an empty line */
308757bddd21SDag-Erling Smørgrav 		c->http_in_chunk_headers = 3;
308857bddd21SDag-Erling Smørgrav 		/* process more data in buffer (if any) */
308957bddd21SDag-Erling Smørgrav 		return 2;
309057bddd21SDag-Erling Smørgrav 	}
309157bddd21SDag-Erling Smørgrav 	c->tcp_byte_count -= got_now;
309257bddd21SDag-Erling Smørgrav 
309357bddd21SDag-Erling Smørgrav 	/* if we have the buffer space,
309457bddd21SDag-Erling Smørgrav 	 * read more data collected into the buffer */
309557bddd21SDag-Erling Smørgrav 	remainbufferlen = sldns_buffer_capacity(c->buffer) -
309657bddd21SDag-Erling Smørgrav 		sldns_buffer_limit(c->buffer);
309757bddd21SDag-Erling Smørgrav 	if(remainbufferlen >= c->tcp_byte_count ||
309857bddd21SDag-Erling Smørgrav 		remainbufferlen >= 2048) {
309957bddd21SDag-Erling Smørgrav 		size_t total = sldns_buffer_limit(c->buffer);
310057bddd21SDag-Erling Smørgrav 		sldns_buffer_clear(c->buffer);
310157bddd21SDag-Erling Smørgrav 		sldns_buffer_set_position(c->buffer, total);
310257bddd21SDag-Erling Smørgrav 		c->http_stored = total;
310357bddd21SDag-Erling Smørgrav 		/* return and wait to read more */
310457bddd21SDag-Erling Smørgrav 		return 1;
310557bddd21SDag-Erling Smørgrav 	}
310657bddd21SDag-Erling Smørgrav 
310757bddd21SDag-Erling Smørgrav 	/* callback of http reader for a new part of the data */
310857bddd21SDag-Erling Smørgrav 	c->http_stored = 0;
310957bddd21SDag-Erling Smørgrav 	sldns_buffer_set_position(c->buffer, 0);
311057bddd21SDag-Erling Smørgrav 	fptr_ok(fptr_whitelist_comm_point(c->callback));
311157bddd21SDag-Erling Smørgrav 	(void)(*c->callback)(c, c->cb_arg, NETEVENT_NOERROR, NULL);
311257bddd21SDag-Erling Smørgrav 	/* c->callback has to buffer_clear(c->buffer). */
311357bddd21SDag-Erling Smørgrav 	/* return and wait to read more */
311457bddd21SDag-Erling Smørgrav 	return 1;
311557bddd21SDag-Erling Smørgrav }
311657bddd21SDag-Erling Smørgrav 
3117c0caa2e2SCy Schubert #ifdef HAVE_NGHTTP2
3118c0caa2e2SCy Schubert /** Create new http2 session. Called when creating handling comm point. */
3119f44e67d1SCy Schubert static struct http2_session* http2_session_create(struct comm_point* c)
3120c0caa2e2SCy Schubert {
3121c0caa2e2SCy Schubert 	struct http2_session* session = calloc(1, sizeof(*session));
3122c0caa2e2SCy Schubert 	if(!session) {
3123c0caa2e2SCy Schubert 		log_err("malloc failure while creating http2 session");
3124c0caa2e2SCy Schubert 		return NULL;
3125c0caa2e2SCy Schubert 	}
3126c0caa2e2SCy Schubert 	session->c = c;
3127c0caa2e2SCy Schubert 
3128c0caa2e2SCy Schubert 	return session;
3129c0caa2e2SCy Schubert }
3130c0caa2e2SCy Schubert #endif
3131c0caa2e2SCy Schubert 
3132c0caa2e2SCy Schubert /** Delete http2 session. After closing connection or on error */
3133f44e67d1SCy Schubert static void http2_session_delete(struct http2_session* h2_session)
3134c0caa2e2SCy Schubert {
3135c0caa2e2SCy Schubert #ifdef HAVE_NGHTTP2
3136c0caa2e2SCy Schubert 	if(h2_session->callbacks)
3137c0caa2e2SCy Schubert 		nghttp2_session_callbacks_del(h2_session->callbacks);
3138c0caa2e2SCy Schubert 	free(h2_session);
3139c0caa2e2SCy Schubert #else
3140c0caa2e2SCy Schubert 	(void)h2_session;
3141c0caa2e2SCy Schubert #endif
3142c0caa2e2SCy Schubert }
3143c0caa2e2SCy Schubert 
3144c0caa2e2SCy Schubert #ifdef HAVE_NGHTTP2
3145c0caa2e2SCy Schubert struct http2_stream* http2_stream_create(int32_t stream_id)
3146c0caa2e2SCy Schubert {
3147c0caa2e2SCy Schubert 	struct http2_stream* h2_stream = calloc(1, sizeof(*h2_stream));
3148c0caa2e2SCy Schubert 	if(!h2_stream) {
3149c0caa2e2SCy Schubert 		log_err("malloc failure while creating http2 stream");
3150c0caa2e2SCy Schubert 		return NULL;
3151c0caa2e2SCy Schubert 	}
3152c0caa2e2SCy Schubert 	h2_stream->stream_id = stream_id;
3153c0caa2e2SCy Schubert 	return h2_stream;
3154c0caa2e2SCy Schubert }
3155c0caa2e2SCy Schubert 
3156c0caa2e2SCy Schubert /** Delete http2 stream. After session delete or stream close callback */
3157c0caa2e2SCy Schubert static void http2_stream_delete(struct http2_session* h2_session,
3158c0caa2e2SCy Schubert 	struct http2_stream* h2_stream)
3159c0caa2e2SCy Schubert {
3160c0caa2e2SCy Schubert 	if(h2_stream->mesh_state) {
3161c0caa2e2SCy Schubert 		mesh_state_remove_reply(h2_stream->mesh, h2_stream->mesh_state,
3162c0caa2e2SCy Schubert 			h2_session->c);
3163c0caa2e2SCy Schubert 		h2_stream->mesh_state = NULL;
3164c0caa2e2SCy Schubert 	}
3165c0caa2e2SCy Schubert 	http2_req_stream_clear(h2_stream);
3166c0caa2e2SCy Schubert 	free(h2_stream);
3167c0caa2e2SCy Schubert }
3168c0caa2e2SCy Schubert #endif
3169c0caa2e2SCy Schubert 
3170c0caa2e2SCy Schubert void http2_stream_add_meshstate(struct http2_stream* h2_stream,
3171c0caa2e2SCy Schubert 	struct mesh_area* mesh, struct mesh_state* m)
3172c0caa2e2SCy Schubert {
3173c0caa2e2SCy Schubert 	h2_stream->mesh = mesh;
3174c0caa2e2SCy Schubert 	h2_stream->mesh_state = m;
3175c0caa2e2SCy Schubert }
3176c0caa2e2SCy Schubert 
3177c0caa2e2SCy Schubert /** delete http2 session server. After closing connection. */
3178c0caa2e2SCy Schubert static void http2_session_server_delete(struct http2_session* h2_session)
3179c0caa2e2SCy Schubert {
3180c0caa2e2SCy Schubert #ifdef HAVE_NGHTTP2
3181c0caa2e2SCy Schubert 	struct http2_stream* h2_stream, *next;
3182c0caa2e2SCy Schubert 	nghttp2_session_del(h2_session->session); /* NULL input is fine */
3183c0caa2e2SCy Schubert 	h2_session->session = NULL;
3184c0caa2e2SCy Schubert 	for(h2_stream = h2_session->first_stream; h2_stream;) {
3185c0caa2e2SCy Schubert 		next = h2_stream->next;
3186c0caa2e2SCy Schubert 		http2_stream_delete(h2_session, h2_stream);
3187c0caa2e2SCy Schubert 		h2_stream = next;
3188c0caa2e2SCy Schubert 	}
3189c0caa2e2SCy Schubert 	h2_session->first_stream = NULL;
3190c0caa2e2SCy Schubert 	h2_session->is_drop = 0;
3191c0caa2e2SCy Schubert 	h2_session->postpone_drop = 0;
3192c0caa2e2SCy Schubert 	h2_session->c->h2_stream = NULL;
3193c0caa2e2SCy Schubert #endif
3194c0caa2e2SCy Schubert 	(void)h2_session;
3195c0caa2e2SCy Schubert }
3196c0caa2e2SCy Schubert 
3197c0caa2e2SCy Schubert #ifdef HAVE_NGHTTP2
3198c0caa2e2SCy Schubert void http2_session_add_stream(struct http2_session* h2_session,
3199c0caa2e2SCy Schubert 	struct http2_stream* h2_stream)
3200c0caa2e2SCy Schubert {
3201c0caa2e2SCy Schubert 	if(h2_session->first_stream)
3202c0caa2e2SCy Schubert 		h2_session->first_stream->prev = h2_stream;
3203c0caa2e2SCy Schubert 	h2_stream->next = h2_session->first_stream;
3204c0caa2e2SCy Schubert 	h2_session->first_stream = h2_stream;
3205c0caa2e2SCy Schubert }
3206c0caa2e2SCy Schubert 
3207c0caa2e2SCy Schubert /** remove stream from session linked list. After stream close callback or
3208c0caa2e2SCy Schubert  * closing connection */
3209f44e67d1SCy Schubert static void http2_session_remove_stream(struct http2_session* h2_session,
3210c0caa2e2SCy Schubert 	struct http2_stream* h2_stream)
3211c0caa2e2SCy Schubert {
3212c0caa2e2SCy Schubert 	if(h2_stream->prev)
3213c0caa2e2SCy Schubert 		h2_stream->prev->next = h2_stream->next;
3214c0caa2e2SCy Schubert 	else
3215c0caa2e2SCy Schubert 		h2_session->first_stream = h2_stream->next;
3216c0caa2e2SCy Schubert 	if(h2_stream->next)
3217c0caa2e2SCy Schubert 		h2_stream->next->prev = h2_stream->prev;
3218c0caa2e2SCy Schubert 
3219c0caa2e2SCy Schubert }
3220c0caa2e2SCy Schubert 
3221c0caa2e2SCy Schubert int http2_stream_close_cb(nghttp2_session* ATTR_UNUSED(session),
3222c0caa2e2SCy Schubert 	int32_t stream_id, uint32_t ATTR_UNUSED(error_code), void* cb_arg)
3223c0caa2e2SCy Schubert {
3224c0caa2e2SCy Schubert 	struct http2_stream* h2_stream;
3225c0caa2e2SCy Schubert 	struct http2_session* h2_session = (struct http2_session*)cb_arg;
3226c0caa2e2SCy Schubert 	if(!(h2_stream = nghttp2_session_get_stream_user_data(
3227c0caa2e2SCy Schubert 		h2_session->session, stream_id))) {
3228c0caa2e2SCy Schubert 		return 0;
3229c0caa2e2SCy Schubert 	}
3230c0caa2e2SCy Schubert 	http2_session_remove_stream(h2_session, h2_stream);
3231c0caa2e2SCy Schubert 	http2_stream_delete(h2_session, h2_stream);
3232c0caa2e2SCy Schubert 	return 0;
3233c0caa2e2SCy Schubert }
3234c0caa2e2SCy Schubert 
3235c0caa2e2SCy Schubert ssize_t http2_recv_cb(nghttp2_session* ATTR_UNUSED(session), uint8_t* buf,
3236c0caa2e2SCy Schubert 	size_t len, int ATTR_UNUSED(flags), void* cb_arg)
3237c0caa2e2SCy Schubert {
3238c0caa2e2SCy Schubert 	struct http2_session* h2_session = (struct http2_session*)cb_arg;
3239369c6923SCy Schubert 	ssize_t ret;
3240c0caa2e2SCy Schubert 
3241c0caa2e2SCy Schubert 	log_assert(h2_session->c->type == comm_http);
3242c0caa2e2SCy Schubert 	log_assert(h2_session->c->h2_session);
3243c0caa2e2SCy Schubert 
3244369c6923SCy Schubert #ifdef HAVE_SSL
3245369c6923SCy Schubert 	if(h2_session->c->ssl) {
3246369c6923SCy Schubert 		int r;
3247c0caa2e2SCy Schubert 		ERR_clear_error();
3248c0caa2e2SCy Schubert 		r = SSL_read(h2_session->c->ssl, buf, len);
3249c0caa2e2SCy Schubert 		if(r <= 0) {
3250c0caa2e2SCy Schubert 			int want = SSL_get_error(h2_session->c->ssl, r);
3251c0caa2e2SCy Schubert 			if(want == SSL_ERROR_ZERO_RETURN) {
3252c0caa2e2SCy Schubert 				return NGHTTP2_ERR_EOF;
3253c0caa2e2SCy Schubert 			} else if(want == SSL_ERROR_WANT_READ) {
3254c0caa2e2SCy Schubert 				return NGHTTP2_ERR_WOULDBLOCK;
3255c0caa2e2SCy Schubert 			} else if(want == SSL_ERROR_WANT_WRITE) {
3256c0caa2e2SCy Schubert 				h2_session->c->ssl_shake_state = comm_ssl_shake_hs_write;
3257c0caa2e2SCy Schubert 				comm_point_listen_for_rw(h2_session->c, 0, 1);
3258c0caa2e2SCy Schubert 				return NGHTTP2_ERR_WOULDBLOCK;
3259c0caa2e2SCy Schubert 			} else if(want == SSL_ERROR_SYSCALL) {
3260c0caa2e2SCy Schubert #ifdef ECONNRESET
3261c0caa2e2SCy Schubert 				if(errno == ECONNRESET && verbosity < 2)
3262c0caa2e2SCy Schubert 					return NGHTTP2_ERR_CALLBACK_FAILURE;
3263c0caa2e2SCy Schubert #endif
3264c0caa2e2SCy Schubert 				if(errno != 0)
3265c0caa2e2SCy Schubert 					log_err("SSL_read syscall: %s",
3266c0caa2e2SCy Schubert 						strerror(errno));
3267c0caa2e2SCy Schubert 				return NGHTTP2_ERR_CALLBACK_FAILURE;
3268c0caa2e2SCy Schubert 			}
3269c0caa2e2SCy Schubert 			log_crypto_err("could not SSL_read");
3270c0caa2e2SCy Schubert 			return NGHTTP2_ERR_CALLBACK_FAILURE;
3271c0caa2e2SCy Schubert 		}
3272c0caa2e2SCy Schubert 		return r;
3273369c6923SCy Schubert 	}
3274369c6923SCy Schubert #endif /* HAVE_SSL */
3275369c6923SCy Schubert 
3276865f46b2SCy Schubert 	ret = recv(h2_session->c->fd, buf, len, MSG_DONTWAIT);
3277369c6923SCy Schubert 	if(ret == 0) {
3278369c6923SCy Schubert 		return NGHTTP2_ERR_EOF;
3279369c6923SCy Schubert 	} else if(ret < 0) {
3280369c6923SCy Schubert #ifndef USE_WINSOCK
3281369c6923SCy Schubert 		if(errno == EINTR || errno == EAGAIN)
3282369c6923SCy Schubert 			return NGHTTP2_ERR_WOULDBLOCK;
3283369c6923SCy Schubert #ifdef ECONNRESET
3284369c6923SCy Schubert 		if(errno == ECONNRESET && verbosity < 2)
3285369c6923SCy Schubert 			return NGHTTP2_ERR_CALLBACK_FAILURE;
3286c0caa2e2SCy Schubert #endif
3287369c6923SCy Schubert 		log_err_addr("could not http2 recv: %s", strerror(errno),
3288865f46b2SCy Schubert 			&h2_session->c->repinfo.remote_addr,
3289865f46b2SCy Schubert 			h2_session->c->repinfo.remote_addrlen);
3290369c6923SCy Schubert #else /* USE_WINSOCK */
3291369c6923SCy Schubert 		if(WSAGetLastError() == WSAECONNRESET)
3292369c6923SCy Schubert 			return NGHTTP2_ERR_CALLBACK_FAILURE;
3293369c6923SCy Schubert 		if(WSAGetLastError() == WSAEINPROGRESS)
3294369c6923SCy Schubert 			return NGHTTP2_ERR_WOULDBLOCK;
3295369c6923SCy Schubert 		if(WSAGetLastError() == WSAEWOULDBLOCK) {
3296369c6923SCy Schubert 			ub_winsock_tcp_wouldblock(h2_session->c->ev->ev,
3297369c6923SCy Schubert 				UB_EV_READ);
3298369c6923SCy Schubert 			return NGHTTP2_ERR_WOULDBLOCK;
3299369c6923SCy Schubert 		}
3300369c6923SCy Schubert 		log_err_addr("could not http2 recv: %s",
3301369c6923SCy Schubert 			wsa_strerror(WSAGetLastError()),
3302865f46b2SCy Schubert 			&h2_session->c->repinfo.remote_addr,
3303865f46b2SCy Schubert 			h2_session->c->repinfo.remote_addrlen);
3304369c6923SCy Schubert #endif
3305369c6923SCy Schubert 		return NGHTTP2_ERR_CALLBACK_FAILURE;
3306369c6923SCy Schubert 	}
3307369c6923SCy Schubert 	return ret;
3308c0caa2e2SCy Schubert }
3309c0caa2e2SCy Schubert #endif /* HAVE_NGHTTP2 */
3310c0caa2e2SCy Schubert 
3311c0caa2e2SCy Schubert /** Handle http2 read */
3312c0caa2e2SCy Schubert static int
3313c0caa2e2SCy Schubert comm_point_http2_handle_read(int ATTR_UNUSED(fd), struct comm_point* c)
3314c0caa2e2SCy Schubert {
3315c0caa2e2SCy Schubert #ifdef HAVE_NGHTTP2
3316c0caa2e2SCy Schubert 	int ret;
3317c0caa2e2SCy Schubert 	log_assert(c->h2_session);
3318c0caa2e2SCy Schubert 
3319c0caa2e2SCy Schubert 	/* reading until recv cb returns NGHTTP2_ERR_WOULDBLOCK */
3320c0caa2e2SCy Schubert 	ret = nghttp2_session_recv(c->h2_session->session);
3321c0caa2e2SCy Schubert 	if(ret) {
3322c0caa2e2SCy Schubert 		if(ret != NGHTTP2_ERR_EOF &&
3323c0caa2e2SCy Schubert 			ret != NGHTTP2_ERR_CALLBACK_FAILURE) {
3324369c6923SCy Schubert 			char a[256];
3325865f46b2SCy Schubert 			addr_to_str(&c->repinfo.remote_addr,
3326865f46b2SCy Schubert 				c->repinfo.remote_addrlen, a, sizeof(a));
3327369c6923SCy Schubert 			verbose(VERB_QUERY, "http2: session_recv from %s failed, "
3328369c6923SCy Schubert 				"error: %s", a, nghttp2_strerror(ret));
3329c0caa2e2SCy Schubert 		}
3330c0caa2e2SCy Schubert 		return 0;
3331c0caa2e2SCy Schubert 	}
3332c0caa2e2SCy Schubert 	if(nghttp2_session_want_write(c->h2_session->session)) {
3333c0caa2e2SCy Schubert 		c->tcp_is_reading = 0;
3334c0caa2e2SCy Schubert 		comm_point_stop_listening(c);
3335f44e67d1SCy Schubert 		comm_point_start_listening(c, -1, adjusted_tcp_timeout(c));
3336c0caa2e2SCy Schubert 	} else if(!nghttp2_session_want_read(c->h2_session->session))
3337c0caa2e2SCy Schubert 		return 0; /* connection can be closed */
3338c0caa2e2SCy Schubert 	return 1;
3339c0caa2e2SCy Schubert #else
3340c0caa2e2SCy Schubert 	(void)c;
3341c0caa2e2SCy Schubert 	return 0;
3342c0caa2e2SCy Schubert #endif
3343c0caa2e2SCy Schubert }
3344c0caa2e2SCy Schubert 
334557bddd21SDag-Erling Smørgrav /**
334657bddd21SDag-Erling Smørgrav  * Handle http reading callback.
334757bddd21SDag-Erling Smørgrav  * @param fd: file descriptor of socket.
334857bddd21SDag-Erling Smørgrav  * @param c: comm point to read from into buffer.
334957bddd21SDag-Erling Smørgrav  * @return: 0 on error
335057bddd21SDag-Erling Smørgrav  */
335157bddd21SDag-Erling Smørgrav static int
335257bddd21SDag-Erling Smørgrav comm_point_http_handle_read(int fd, struct comm_point* c)
335357bddd21SDag-Erling Smørgrav {
335457bddd21SDag-Erling Smørgrav 	log_assert(c->type == comm_http);
335557bddd21SDag-Erling Smørgrav 	log_assert(fd != -1);
335657bddd21SDag-Erling Smørgrav 
335757bddd21SDag-Erling Smørgrav 	/* if we are in ssl handshake, handle SSL handshake */
335857bddd21SDag-Erling Smørgrav #ifdef HAVE_SSL
335957bddd21SDag-Erling Smørgrav 	if(c->ssl && c->ssl_shake_state != comm_ssl_shake_none) {
336057bddd21SDag-Erling Smørgrav 		if(!ssl_handshake(c))
336157bddd21SDag-Erling Smørgrav 			return 0;
336257bddd21SDag-Erling Smørgrav 		if(c->ssl_shake_state != comm_ssl_shake_none)
336357bddd21SDag-Erling Smørgrav 			return 1;
336457bddd21SDag-Erling Smørgrav 	}
336557bddd21SDag-Erling Smørgrav #endif /* HAVE_SSL */
336657bddd21SDag-Erling Smørgrav 
336757bddd21SDag-Erling Smørgrav 	if(!c->tcp_is_reading)
336857bddd21SDag-Erling Smørgrav 		return 1;
3369c0caa2e2SCy Schubert 
3370c0caa2e2SCy Schubert 	if(c->use_h2) {
3371c0caa2e2SCy Schubert 		return comm_point_http2_handle_read(fd, c);
3372c0caa2e2SCy Schubert 	}
3373c0caa2e2SCy Schubert 
3374c0caa2e2SCy Schubert 	/* http version is <= http/1.1 */
3375c0caa2e2SCy Schubert 
3376c0caa2e2SCy Schubert 	if(c->http_min_version >= http_version_2) {
3377c0caa2e2SCy Schubert 		/* HTTP/2 failed, not allowed to use lower version. */
3378c0caa2e2SCy Schubert 		return 0;
3379c0caa2e2SCy Schubert 	}
3380c0caa2e2SCy Schubert 
338157bddd21SDag-Erling Smørgrav 	/* read more data */
338257bddd21SDag-Erling Smørgrav 	if(c->ssl) {
338357bddd21SDag-Erling Smørgrav 		if(!ssl_http_read_more(c))
338457bddd21SDag-Erling Smørgrav 			return 0;
338557bddd21SDag-Erling Smørgrav 	} else {
338657bddd21SDag-Erling Smørgrav 		if(!http_read_more(fd, c))
338757bddd21SDag-Erling Smørgrav 			return 0;
338857bddd21SDag-Erling Smørgrav 	}
338957bddd21SDag-Erling Smørgrav 
33905469a995SCy Schubert 	if(c->http_stored >= sldns_buffer_position(c->buffer)) {
33915469a995SCy Schubert 		/* read did not work but we wanted more data, there is
33925469a995SCy Schubert 		 * no bytes to process now. */
33935469a995SCy Schubert 		return 1;
33945469a995SCy Schubert 	}
339557bddd21SDag-Erling Smørgrav 	sldns_buffer_flip(c->buffer);
3396f44e67d1SCy Schubert 	/* if we are partway in a segment of data, position us at the point
3397f44e67d1SCy Schubert 	 * where we left off previously */
3398f44e67d1SCy Schubert 	if(c->http_stored < sldns_buffer_limit(c->buffer))
3399f44e67d1SCy Schubert 		sldns_buffer_set_position(c->buffer, c->http_stored);
3400f44e67d1SCy Schubert 	else	sldns_buffer_set_position(c->buffer, sldns_buffer_limit(c->buffer));
3401c0caa2e2SCy Schubert 
340257bddd21SDag-Erling Smørgrav 	while(sldns_buffer_remaining(c->buffer) > 0) {
3403c0caa2e2SCy Schubert 		/* Handle HTTP/1.x data */
340457bddd21SDag-Erling Smørgrav 		/* if we are reading headers, read more headers */
340557bddd21SDag-Erling Smørgrav 		if(c->http_in_headers || c->http_in_chunk_headers) {
340657bddd21SDag-Erling Smørgrav 			/* if header is done, process the header */
340757bddd21SDag-Erling Smørgrav 			if(!http_header_done(c->buffer)) {
340857bddd21SDag-Erling Smørgrav 				/* copy remaining data to front of buffer
340957bddd21SDag-Erling Smørgrav 				 * and set rest for writing into it */
341057bddd21SDag-Erling Smørgrav 				http_moveover_buffer(c->buffer);
341157bddd21SDag-Erling Smørgrav 				/* return and wait to read more */
341257bddd21SDag-Erling Smørgrav 				return 1;
341357bddd21SDag-Erling Smørgrav 			}
341457bddd21SDag-Erling Smørgrav 			if(!c->http_in_chunk_headers) {
341557bddd21SDag-Erling Smørgrav 				/* process initial headers */
341657bddd21SDag-Erling Smørgrav 				if(!http_process_initial_header(c))
341757bddd21SDag-Erling Smørgrav 					return 0;
341857bddd21SDag-Erling Smørgrav 			} else {
341957bddd21SDag-Erling Smørgrav 				/* process chunk headers */
342057bddd21SDag-Erling Smørgrav 				int r = http_process_chunk_header(c);
342157bddd21SDag-Erling Smørgrav 				if(r == 0) return 0;
342257bddd21SDag-Erling Smørgrav 				if(r == 2) return 1; /* done */
342357bddd21SDag-Erling Smørgrav 				/* r == 1, continue */
342457bddd21SDag-Erling Smørgrav 			}
342557bddd21SDag-Erling Smørgrav 			/* see if we have more to process */
342657bddd21SDag-Erling Smørgrav 			continue;
342757bddd21SDag-Erling Smørgrav 		}
342857bddd21SDag-Erling Smørgrav 
342957bddd21SDag-Erling Smørgrav 		if(!c->http_is_chunked) {
343057bddd21SDag-Erling Smørgrav 			/* if we are reading nonchunks, process that*/
343157bddd21SDag-Erling Smørgrav 			return http_nonchunk_segment(c);
343257bddd21SDag-Erling Smørgrav 		} else {
343357bddd21SDag-Erling Smørgrav 			/* if we are reading chunks, read the chunk */
343457bddd21SDag-Erling Smørgrav 			int r = http_chunked_segment(c);
343557bddd21SDag-Erling Smørgrav 			if(r == 0) return 0;
343657bddd21SDag-Erling Smørgrav 			if(r == 1) return 1;
343757bddd21SDag-Erling Smørgrav 			continue;
343857bddd21SDag-Erling Smørgrav 		}
343957bddd21SDag-Erling Smørgrav 	}
344057bddd21SDag-Erling Smørgrav 	/* broke out of the loop; could not process header instead need
344157bddd21SDag-Erling Smørgrav 	 * to read more */
344257bddd21SDag-Erling Smørgrav 	/* moveover any remaining data and read more data */
344357bddd21SDag-Erling Smørgrav 	http_moveover_buffer(c->buffer);
344457bddd21SDag-Erling Smørgrav 	/* return and wait to read more */
344557bddd21SDag-Erling Smørgrav 	return 1;
344657bddd21SDag-Erling Smørgrav }
344757bddd21SDag-Erling Smørgrav 
344857bddd21SDag-Erling Smørgrav /** check pending connect for http */
344957bddd21SDag-Erling Smørgrav static int
345057bddd21SDag-Erling Smørgrav http_check_connect(int fd, struct comm_point* c)
345157bddd21SDag-Erling Smørgrav {
345257bddd21SDag-Erling Smørgrav 	/* check for pending error from nonblocking connect */
345357bddd21SDag-Erling Smørgrav 	/* from Stevens, unix network programming, vol1, 3rd ed, p450*/
345457bddd21SDag-Erling Smørgrav 	int error = 0;
345557bddd21SDag-Erling Smørgrav 	socklen_t len = (socklen_t)sizeof(error);
345657bddd21SDag-Erling Smørgrav 	if(getsockopt(fd, SOL_SOCKET, SO_ERROR, (void*)&error,
345757bddd21SDag-Erling Smørgrav 		&len) < 0){
345857bddd21SDag-Erling Smørgrav #ifndef USE_WINSOCK
345957bddd21SDag-Erling Smørgrav 		error = errno; /* on solaris errno is error */
346057bddd21SDag-Erling Smørgrav #else /* USE_WINSOCK */
346157bddd21SDag-Erling Smørgrav 		error = WSAGetLastError();
346257bddd21SDag-Erling Smørgrav #endif
346357bddd21SDag-Erling Smørgrav 	}
346457bddd21SDag-Erling Smørgrav #ifndef USE_WINSOCK
346557bddd21SDag-Erling Smørgrav #if defined(EINPROGRESS) && defined(EWOULDBLOCK)
346657bddd21SDag-Erling Smørgrav 	if(error == EINPROGRESS || error == EWOULDBLOCK)
346757bddd21SDag-Erling Smørgrav 		return 1; /* try again later */
346857bddd21SDag-Erling Smørgrav 	else
346957bddd21SDag-Erling Smørgrav #endif
347057bddd21SDag-Erling Smørgrav 	if(error != 0 && verbosity < 2)
347157bddd21SDag-Erling Smørgrav 		return 0; /* silence lots of chatter in the logs */
347257bddd21SDag-Erling Smørgrav 	else if(error != 0) {
347357bddd21SDag-Erling Smørgrav 		log_err_addr("http connect", strerror(error),
3474865f46b2SCy Schubert 			&c->repinfo.remote_addr, c->repinfo.remote_addrlen);
347557bddd21SDag-Erling Smørgrav #else /* USE_WINSOCK */
347657bddd21SDag-Erling Smørgrav 	/* examine error */
347757bddd21SDag-Erling Smørgrav 	if(error == WSAEINPROGRESS)
347857bddd21SDag-Erling Smørgrav 		return 1;
347957bddd21SDag-Erling Smørgrav 	else if(error == WSAEWOULDBLOCK) {
348057bddd21SDag-Erling Smørgrav 		ub_winsock_tcp_wouldblock(c->ev->ev, UB_EV_WRITE);
348157bddd21SDag-Erling Smørgrav 		return 1;
348257bddd21SDag-Erling Smørgrav 	} else if(error != 0 && verbosity < 2)
348357bddd21SDag-Erling Smørgrav 		return 0;
348457bddd21SDag-Erling Smørgrav 	else if(error != 0) {
348557bddd21SDag-Erling Smørgrav 		log_err_addr("http connect", wsa_strerror(error),
3486865f46b2SCy Schubert 			&c->repinfo.remote_addr, c->repinfo.remote_addrlen);
348757bddd21SDag-Erling Smørgrav #endif /* USE_WINSOCK */
348857bddd21SDag-Erling Smørgrav 		return 0;
348957bddd21SDag-Erling Smørgrav 	}
349057bddd21SDag-Erling Smørgrav 	/* keep on processing this socket */
349157bddd21SDag-Erling Smørgrav 	return 2;
349257bddd21SDag-Erling Smørgrav }
349357bddd21SDag-Erling Smørgrav 
349457bddd21SDag-Erling Smørgrav /** write more data for http (with ssl) */
349557bddd21SDag-Erling Smørgrav static int
349657bddd21SDag-Erling Smørgrav ssl_http_write_more(struct comm_point* c)
349757bddd21SDag-Erling Smørgrav {
349857bddd21SDag-Erling Smørgrav #ifdef HAVE_SSL
349957bddd21SDag-Erling Smørgrav 	int r;
350057bddd21SDag-Erling Smørgrav 	log_assert(sldns_buffer_remaining(c->buffer) > 0);
350157bddd21SDag-Erling Smørgrav 	ERR_clear_error();
350257bddd21SDag-Erling Smørgrav 	r = SSL_write(c->ssl, (void*)sldns_buffer_current(c->buffer),
350357bddd21SDag-Erling Smørgrav 		(int)sldns_buffer_remaining(c->buffer));
350457bddd21SDag-Erling Smørgrav 	if(r <= 0) {
350557bddd21SDag-Erling Smørgrav 		int want = SSL_get_error(c->ssl, r);
350657bddd21SDag-Erling Smørgrav 		if(want == SSL_ERROR_ZERO_RETURN) {
350757bddd21SDag-Erling Smørgrav 			return 0; /* closed */
350857bddd21SDag-Erling Smørgrav 		} else if(want == SSL_ERROR_WANT_READ) {
3509e86b9096SDag-Erling Smørgrav 			c->ssl_shake_state = comm_ssl_shake_hs_read;
351057bddd21SDag-Erling Smørgrav 			comm_point_listen_for_rw(c, 1, 0);
351157bddd21SDag-Erling Smørgrav 			return 1; /* wait for read condition */
351257bddd21SDag-Erling Smørgrav 		} else if(want == SSL_ERROR_WANT_WRITE) {
351357bddd21SDag-Erling Smørgrav 			return 1; /* write more later */
351457bddd21SDag-Erling Smørgrav 		} else if(want == SSL_ERROR_SYSCALL) {
3515e86b9096SDag-Erling Smørgrav #ifdef EPIPE
3516e86b9096SDag-Erling Smørgrav 			if(errno == EPIPE && verbosity < 2)
3517e86b9096SDag-Erling Smørgrav 				return 0; /* silence 'broken pipe' */
3518e86b9096SDag-Erling Smørgrav #endif
351957bddd21SDag-Erling Smørgrav 			if(errno != 0)
352057bddd21SDag-Erling Smørgrav 				log_err("SSL_write syscall: %s",
352157bddd21SDag-Erling Smørgrav 					strerror(errno));
352257bddd21SDag-Erling Smørgrav 			return 0;
352357bddd21SDag-Erling Smørgrav 		}
352457bddd21SDag-Erling Smørgrav 		log_crypto_err("could not SSL_write");
352557bddd21SDag-Erling Smørgrav 		return 0;
352657bddd21SDag-Erling Smørgrav 	}
352757bddd21SDag-Erling Smørgrav 	sldns_buffer_skip(c->buffer, (ssize_t)r);
352857bddd21SDag-Erling Smørgrav 	return 1;
352957bddd21SDag-Erling Smørgrav #else
353057bddd21SDag-Erling Smørgrav 	(void)c;
353157bddd21SDag-Erling Smørgrav 	return 0;
353257bddd21SDag-Erling Smørgrav #endif /* HAVE_SSL */
353357bddd21SDag-Erling Smørgrav }
353457bddd21SDag-Erling Smørgrav 
353557bddd21SDag-Erling Smørgrav /** write more data for http */
353657bddd21SDag-Erling Smørgrav static int
353757bddd21SDag-Erling Smørgrav http_write_more(int fd, struct comm_point* c)
353857bddd21SDag-Erling Smørgrav {
353957bddd21SDag-Erling Smørgrav 	ssize_t r;
354057bddd21SDag-Erling Smørgrav 	log_assert(sldns_buffer_remaining(c->buffer) > 0);
354157bddd21SDag-Erling Smørgrav 	r = send(fd, (void*)sldns_buffer_current(c->buffer),
354257bddd21SDag-Erling Smørgrav 		sldns_buffer_remaining(c->buffer), 0);
354357bddd21SDag-Erling Smørgrav 	if(r == -1) {
354457bddd21SDag-Erling Smørgrav #ifndef USE_WINSOCK
354557bddd21SDag-Erling Smørgrav 		if(errno == EINTR || errno == EAGAIN)
354657bddd21SDag-Erling Smørgrav 			return 1;
354757bddd21SDag-Erling Smørgrav #else
354857bddd21SDag-Erling Smørgrav 		if(WSAGetLastError() == WSAEINPROGRESS)
354957bddd21SDag-Erling Smørgrav 			return 1;
355057bddd21SDag-Erling Smørgrav 		if(WSAGetLastError() == WSAEWOULDBLOCK) {
355157bddd21SDag-Erling Smørgrav 			ub_winsock_tcp_wouldblock(c->ev->ev, UB_EV_WRITE);
355257bddd21SDag-Erling Smørgrav 			return 1;
355357bddd21SDag-Erling Smørgrav 		}
355457bddd21SDag-Erling Smørgrav #endif
3555c0caa2e2SCy Schubert 		log_err_addr("http send r", sock_strerror(errno),
3556865f46b2SCy Schubert 			&c->repinfo.remote_addr, c->repinfo.remote_addrlen);
355757bddd21SDag-Erling Smørgrav 		return 0;
355857bddd21SDag-Erling Smørgrav 	}
355957bddd21SDag-Erling Smørgrav 	sldns_buffer_skip(c->buffer, r);
356057bddd21SDag-Erling Smørgrav 	return 1;
356157bddd21SDag-Erling Smørgrav }
356257bddd21SDag-Erling Smørgrav 
3563c0caa2e2SCy Schubert #ifdef HAVE_NGHTTP2
3564c0caa2e2SCy Schubert ssize_t http2_send_cb(nghttp2_session* ATTR_UNUSED(session), const uint8_t* buf,
3565c0caa2e2SCy Schubert 	size_t len, int ATTR_UNUSED(flags), void* cb_arg)
3566c0caa2e2SCy Schubert {
3567369c6923SCy Schubert 	ssize_t ret;
3568c0caa2e2SCy Schubert 	struct http2_session* h2_session = (struct http2_session*)cb_arg;
3569c0caa2e2SCy Schubert 	log_assert(h2_session->c->type == comm_http);
3570c0caa2e2SCy Schubert 	log_assert(h2_session->c->h2_session);
3571c0caa2e2SCy Schubert 
3572369c6923SCy Schubert #ifdef HAVE_SSL
3573369c6923SCy Schubert 	if(h2_session->c->ssl) {
3574369c6923SCy Schubert 		int r;
3575c0caa2e2SCy Schubert 		ERR_clear_error();
3576c0caa2e2SCy Schubert 		r = SSL_write(h2_session->c->ssl, buf, len);
3577c0caa2e2SCy Schubert 		if(r <= 0) {
3578c0caa2e2SCy Schubert 			int want = SSL_get_error(h2_session->c->ssl, r);
3579c0caa2e2SCy Schubert 			if(want == SSL_ERROR_ZERO_RETURN) {
3580c0caa2e2SCy Schubert 				return NGHTTP2_ERR_CALLBACK_FAILURE;
3581c0caa2e2SCy Schubert 			} else if(want == SSL_ERROR_WANT_READ) {
3582c0caa2e2SCy Schubert 				h2_session->c->ssl_shake_state = comm_ssl_shake_hs_read;
3583c0caa2e2SCy Schubert 				comm_point_listen_for_rw(h2_session->c, 1, 0);
3584c0caa2e2SCy Schubert 				return NGHTTP2_ERR_WOULDBLOCK;
3585c0caa2e2SCy Schubert 			} else if(want == SSL_ERROR_WANT_WRITE) {
3586c0caa2e2SCy Schubert 				return NGHTTP2_ERR_WOULDBLOCK;
3587c0caa2e2SCy Schubert 			} else if(want == SSL_ERROR_SYSCALL) {
3588c0caa2e2SCy Schubert #ifdef EPIPE
3589c0caa2e2SCy Schubert 				if(errno == EPIPE && verbosity < 2)
3590c0caa2e2SCy Schubert 					return NGHTTP2_ERR_CALLBACK_FAILURE;
3591c0caa2e2SCy Schubert #endif
3592c0caa2e2SCy Schubert 				if(errno != 0)
3593c0caa2e2SCy Schubert 					log_err("SSL_write syscall: %s",
3594c0caa2e2SCy Schubert 						strerror(errno));
3595c0caa2e2SCy Schubert 				return NGHTTP2_ERR_CALLBACK_FAILURE;
3596c0caa2e2SCy Schubert 			}
3597c0caa2e2SCy Schubert 			log_crypto_err("could not SSL_write");
3598c0caa2e2SCy Schubert 			return NGHTTP2_ERR_CALLBACK_FAILURE;
3599c0caa2e2SCy Schubert 		}
3600c0caa2e2SCy Schubert 		return r;
3601369c6923SCy Schubert 	}
3602369c6923SCy Schubert #endif /* HAVE_SSL */
3603369c6923SCy Schubert 
3604369c6923SCy Schubert 	ret = send(h2_session->c->fd, buf, len, 0);
3605369c6923SCy Schubert 	if(ret == 0) {
3606369c6923SCy Schubert 		return NGHTTP2_ERR_CALLBACK_FAILURE;
3607369c6923SCy Schubert 	} else if(ret < 0) {
3608369c6923SCy Schubert #ifndef USE_WINSOCK
3609369c6923SCy Schubert 		if(errno == EINTR || errno == EAGAIN)
3610369c6923SCy Schubert 			return NGHTTP2_ERR_WOULDBLOCK;
3611369c6923SCy Schubert #ifdef EPIPE
3612369c6923SCy Schubert 		if(errno == EPIPE && verbosity < 2)
3613369c6923SCy Schubert 			return NGHTTP2_ERR_CALLBACK_FAILURE;
3614c0caa2e2SCy Schubert #endif
3615369c6923SCy Schubert #ifdef ECONNRESET
3616369c6923SCy Schubert 		if(errno == ECONNRESET && verbosity < 2)
3617369c6923SCy Schubert 			return NGHTTP2_ERR_CALLBACK_FAILURE;
3618369c6923SCy Schubert #endif
3619369c6923SCy Schubert 		log_err_addr("could not http2 write: %s", strerror(errno),
3620865f46b2SCy Schubert 			&h2_session->c->repinfo.remote_addr,
3621865f46b2SCy Schubert 			h2_session->c->repinfo.remote_addrlen);
3622369c6923SCy Schubert #else /* USE_WINSOCK */
3623369c6923SCy Schubert 		if(WSAGetLastError() == WSAENOTCONN)
3624369c6923SCy Schubert 			return NGHTTP2_ERR_WOULDBLOCK;
3625369c6923SCy Schubert 		if(WSAGetLastError() == WSAEINPROGRESS)
3626369c6923SCy Schubert 			return NGHTTP2_ERR_WOULDBLOCK;
3627369c6923SCy Schubert 		if(WSAGetLastError() == WSAEWOULDBLOCK) {
3628369c6923SCy Schubert 			ub_winsock_tcp_wouldblock(h2_session->c->ev->ev,
3629369c6923SCy Schubert 				UB_EV_WRITE);
3630369c6923SCy Schubert 			return NGHTTP2_ERR_WOULDBLOCK;
3631369c6923SCy Schubert 		}
3632369c6923SCy Schubert 		if(WSAGetLastError() == WSAECONNRESET && verbosity < 2)
3633369c6923SCy Schubert 			return NGHTTP2_ERR_CALLBACK_FAILURE;
3634369c6923SCy Schubert 		log_err_addr("could not http2 write: %s",
3635369c6923SCy Schubert 			wsa_strerror(WSAGetLastError()),
3636865f46b2SCy Schubert 			&h2_session->c->repinfo.remote_addr,
3637865f46b2SCy Schubert 			h2_session->c->repinfo.remote_addrlen);
3638369c6923SCy Schubert #endif
3639369c6923SCy Schubert 		return NGHTTP2_ERR_CALLBACK_FAILURE;
3640369c6923SCy Schubert 	}
3641369c6923SCy Schubert 	return ret;
3642c0caa2e2SCy Schubert }
3643c0caa2e2SCy Schubert #endif /* HAVE_NGHTTP2 */
3644c0caa2e2SCy Schubert 
3645c0caa2e2SCy Schubert /** Handle http2 writing */
3646c0caa2e2SCy Schubert static int
3647c0caa2e2SCy Schubert comm_point_http2_handle_write(int ATTR_UNUSED(fd), struct comm_point* c)
3648c0caa2e2SCy Schubert {
3649c0caa2e2SCy Schubert #ifdef HAVE_NGHTTP2
3650c0caa2e2SCy Schubert 	int ret;
3651c0caa2e2SCy Schubert 	log_assert(c->h2_session);
3652c0caa2e2SCy Schubert 
3653c0caa2e2SCy Schubert 	ret = nghttp2_session_send(c->h2_session->session);
3654c0caa2e2SCy Schubert 	if(ret) {
3655c0caa2e2SCy Schubert 		verbose(VERB_QUERY, "http2: session_send failed, "
3656c0caa2e2SCy Schubert 			"error: %s", nghttp2_strerror(ret));
3657c0caa2e2SCy Schubert 		return 0;
3658c0caa2e2SCy Schubert 	}
3659c0caa2e2SCy Schubert 
3660c0caa2e2SCy Schubert 	if(nghttp2_session_want_read(c->h2_session->session)) {
3661c0caa2e2SCy Schubert 		c->tcp_is_reading = 1;
3662c0caa2e2SCy Schubert 		comm_point_stop_listening(c);
3663f44e67d1SCy Schubert 		comm_point_start_listening(c, -1, adjusted_tcp_timeout(c));
3664c0caa2e2SCy Schubert 	} else if(!nghttp2_session_want_write(c->h2_session->session))
3665c0caa2e2SCy Schubert 		return 0; /* connection can be closed */
3666c0caa2e2SCy Schubert 	return 1;
3667c0caa2e2SCy Schubert #else
3668c0caa2e2SCy Schubert 	(void)c;
3669c0caa2e2SCy Schubert 	return 0;
3670c0caa2e2SCy Schubert #endif
3671c0caa2e2SCy Schubert }
3672c0caa2e2SCy Schubert 
367357bddd21SDag-Erling Smørgrav /**
367457bddd21SDag-Erling Smørgrav  * Handle http writing callback.
367557bddd21SDag-Erling Smørgrav  * @param fd: file descriptor of socket.
367657bddd21SDag-Erling Smørgrav  * @param c: comm point to write buffer out of.
367757bddd21SDag-Erling Smørgrav  * @return: 0 on error
367857bddd21SDag-Erling Smørgrav  */
367957bddd21SDag-Erling Smørgrav static int
368057bddd21SDag-Erling Smørgrav comm_point_http_handle_write(int fd, struct comm_point* c)
368157bddd21SDag-Erling Smørgrav {
368257bddd21SDag-Erling Smørgrav 	log_assert(c->type == comm_http);
368357bddd21SDag-Erling Smørgrav 	log_assert(fd != -1);
368457bddd21SDag-Erling Smørgrav 
368557bddd21SDag-Erling Smørgrav 	/* check pending connect errors, if that fails, we wait for more,
368657bddd21SDag-Erling Smørgrav 	 * or we can continue to write contents */
368757bddd21SDag-Erling Smørgrav 	if(c->tcp_check_nb_connect) {
368857bddd21SDag-Erling Smørgrav 		int r = http_check_connect(fd, c);
368957bddd21SDag-Erling Smørgrav 		if(r == 0) return 0;
369057bddd21SDag-Erling Smørgrav 		if(r == 1) return 1;
369157bddd21SDag-Erling Smørgrav 		c->tcp_check_nb_connect = 0;
369257bddd21SDag-Erling Smørgrav 	}
369357bddd21SDag-Erling Smørgrav 	/* if we are in ssl handshake, handle SSL handshake */
369457bddd21SDag-Erling Smørgrav #ifdef HAVE_SSL
369557bddd21SDag-Erling Smørgrav 	if(c->ssl && c->ssl_shake_state != comm_ssl_shake_none) {
369657bddd21SDag-Erling Smørgrav 		if(!ssl_handshake(c))
369757bddd21SDag-Erling Smørgrav 			return 0;
369857bddd21SDag-Erling Smørgrav 		if(c->ssl_shake_state != comm_ssl_shake_none)
369957bddd21SDag-Erling Smørgrav 			return 1;
370057bddd21SDag-Erling Smørgrav 	}
370157bddd21SDag-Erling Smørgrav #endif /* HAVE_SSL */
370257bddd21SDag-Erling Smørgrav 	if(c->tcp_is_reading)
370357bddd21SDag-Erling Smørgrav 		return 1;
3704c0caa2e2SCy Schubert 
3705c0caa2e2SCy Schubert 	if(c->use_h2) {
3706c0caa2e2SCy Schubert 		return comm_point_http2_handle_write(fd, c);
3707c0caa2e2SCy Schubert 	}
3708c0caa2e2SCy Schubert 
3709c0caa2e2SCy Schubert 	/* http version is <= http/1.1 */
3710c0caa2e2SCy Schubert 
3711c0caa2e2SCy Schubert 	if(c->http_min_version >= http_version_2) {
3712c0caa2e2SCy Schubert 		/* HTTP/2 failed, not allowed to use lower version. */
3713c0caa2e2SCy Schubert 		return 0;
3714c0caa2e2SCy Schubert 	}
3715c0caa2e2SCy Schubert 
371657bddd21SDag-Erling Smørgrav 	/* if we are writing, write more */
371757bddd21SDag-Erling Smørgrav 	if(c->ssl) {
371857bddd21SDag-Erling Smørgrav 		if(!ssl_http_write_more(c))
371957bddd21SDag-Erling Smørgrav 			return 0;
372057bddd21SDag-Erling Smørgrav 	} else {
372157bddd21SDag-Erling Smørgrav 		if(!http_write_more(fd, c))
372257bddd21SDag-Erling Smørgrav 			return 0;
372357bddd21SDag-Erling Smørgrav 	}
372457bddd21SDag-Erling Smørgrav 
372557bddd21SDag-Erling Smørgrav 	/* we write a single buffer contents, that can contain
372657bddd21SDag-Erling Smørgrav 	 * the http request, and then flip to read the results */
372757bddd21SDag-Erling Smørgrav 	/* see if write is done */
372857bddd21SDag-Erling Smørgrav 	if(sldns_buffer_remaining(c->buffer) == 0) {
372957bddd21SDag-Erling Smørgrav 		sldns_buffer_clear(c->buffer);
373057bddd21SDag-Erling Smørgrav 		if(c->tcp_do_toggle_rw)
373157bddd21SDag-Erling Smørgrav 			c->tcp_is_reading = 1;
373257bddd21SDag-Erling Smørgrav 		c->tcp_byte_count = 0;
373357bddd21SDag-Erling Smørgrav 		/* switch from listening(write) to listening(read) */
373457bddd21SDag-Erling Smørgrav 		comm_point_stop_listening(c);
373557bddd21SDag-Erling Smørgrav 		comm_point_start_listening(c, -1, -1);
373657bddd21SDag-Erling Smørgrav 	}
373757bddd21SDag-Erling Smørgrav 	return 1;
373857bddd21SDag-Erling Smørgrav }
373957bddd21SDag-Erling Smørgrav 
374057bddd21SDag-Erling Smørgrav void
374157bddd21SDag-Erling Smørgrav comm_point_http_handle_callback(int fd, short event, void* arg)
374257bddd21SDag-Erling Smørgrav {
374357bddd21SDag-Erling Smørgrav 	struct comm_point* c = (struct comm_point*)arg;
374457bddd21SDag-Erling Smørgrav 	log_assert(c->type == comm_http);
374557bddd21SDag-Erling Smørgrav 	ub_comm_base_now(c->ev->base);
374657bddd21SDag-Erling Smørgrav 
3747e86b9096SDag-Erling Smørgrav 	if(event&UB_EV_TIMEOUT) {
3748e86b9096SDag-Erling Smørgrav 		verbose(VERB_QUERY, "http took too long, dropped");
3749e86b9096SDag-Erling Smørgrav 		reclaim_http_handler(c);
3750e86b9096SDag-Erling Smørgrav 		if(!c->tcp_do_close) {
3751e86b9096SDag-Erling Smørgrav 			fptr_ok(fptr_whitelist_comm_point(c->callback));
3752e86b9096SDag-Erling Smørgrav 			(void)(*c->callback)(c, c->cb_arg,
3753e86b9096SDag-Erling Smørgrav 				NETEVENT_TIMEOUT, NULL);
3754e86b9096SDag-Erling Smørgrav 		}
3755e86b9096SDag-Erling Smørgrav 		return;
3756e86b9096SDag-Erling Smørgrav 	}
375757bddd21SDag-Erling Smørgrav 	if(event&UB_EV_READ) {
375857bddd21SDag-Erling Smørgrav 		if(!comm_point_http_handle_read(fd, c)) {
375957bddd21SDag-Erling Smørgrav 			reclaim_http_handler(c);
376057bddd21SDag-Erling Smørgrav 			if(!c->tcp_do_close) {
376157bddd21SDag-Erling Smørgrav 				fptr_ok(fptr_whitelist_comm_point(
376257bddd21SDag-Erling Smørgrav 					c->callback));
376357bddd21SDag-Erling Smørgrav 				(void)(*c->callback)(c, c->cb_arg,
376457bddd21SDag-Erling Smørgrav 					NETEVENT_CLOSED, NULL);
376557bddd21SDag-Erling Smørgrav 			}
376657bddd21SDag-Erling Smørgrav 		}
376757bddd21SDag-Erling Smørgrav 		return;
376857bddd21SDag-Erling Smørgrav 	}
376957bddd21SDag-Erling Smørgrav 	if(event&UB_EV_WRITE) {
377057bddd21SDag-Erling Smørgrav 		if(!comm_point_http_handle_write(fd, c)) {
377157bddd21SDag-Erling Smørgrav 			reclaim_http_handler(c);
377257bddd21SDag-Erling Smørgrav 			if(!c->tcp_do_close) {
377357bddd21SDag-Erling Smørgrav 				fptr_ok(fptr_whitelist_comm_point(
377457bddd21SDag-Erling Smørgrav 					c->callback));
377557bddd21SDag-Erling Smørgrav 				(void)(*c->callback)(c, c->cb_arg,
377657bddd21SDag-Erling Smørgrav 					NETEVENT_CLOSED, NULL);
377757bddd21SDag-Erling Smørgrav 			}
377857bddd21SDag-Erling Smørgrav 		}
377957bddd21SDag-Erling Smørgrav 		return;
378057bddd21SDag-Erling Smørgrav 	}
378157bddd21SDag-Erling Smørgrav 	log_err("Ignored event %d for httphdl.", event);
378257bddd21SDag-Erling Smørgrav }
378357bddd21SDag-Erling Smørgrav 
3784b7579f77SDag-Erling Smørgrav void comm_point_local_handle_callback(int fd, short event, void* arg)
3785b7579f77SDag-Erling Smørgrav {
3786b7579f77SDag-Erling Smørgrav 	struct comm_point* c = (struct comm_point*)arg;
3787b7579f77SDag-Erling Smørgrav 	log_assert(c->type == comm_local);
3788e2d15004SDag-Erling Smørgrav 	ub_comm_base_now(c->ev->base);
3789b7579f77SDag-Erling Smørgrav 
3790e2d15004SDag-Erling Smørgrav 	if(event&UB_EV_READ) {
3791b7579f77SDag-Erling Smørgrav 		if(!comm_point_tcp_handle_read(fd, c, 1)) {
3792b7579f77SDag-Erling Smørgrav 			fptr_ok(fptr_whitelist_comm_point(c->callback));
3793b7579f77SDag-Erling Smørgrav 			(void)(*c->callback)(c, c->cb_arg, NETEVENT_CLOSED,
3794b7579f77SDag-Erling Smørgrav 				NULL);
3795b7579f77SDag-Erling Smørgrav 		}
3796b7579f77SDag-Erling Smørgrav 		return;
3797b7579f77SDag-Erling Smørgrav 	}
3798b7579f77SDag-Erling Smørgrav 	log_err("Ignored event %d for localhdl.", event);
3799b7579f77SDag-Erling Smørgrav }
3800b7579f77SDag-Erling Smørgrav 
3801b7579f77SDag-Erling Smørgrav void comm_point_raw_handle_callback(int ATTR_UNUSED(fd),
3802b7579f77SDag-Erling Smørgrav 	short event, void* arg)
3803b7579f77SDag-Erling Smørgrav {
3804b7579f77SDag-Erling Smørgrav 	struct comm_point* c = (struct comm_point*)arg;
3805b7579f77SDag-Erling Smørgrav 	int err = NETEVENT_NOERROR;
3806b7579f77SDag-Erling Smørgrav 	log_assert(c->type == comm_raw);
3807e2d15004SDag-Erling Smørgrav 	ub_comm_base_now(c->ev->base);
3808b7579f77SDag-Erling Smørgrav 
3809e2d15004SDag-Erling Smørgrav 	if(event&UB_EV_TIMEOUT)
3810b7579f77SDag-Erling Smørgrav 		err = NETEVENT_TIMEOUT;
3811b7579f77SDag-Erling Smørgrav 	fptr_ok(fptr_whitelist_comm_point_raw(c->callback));
3812b7579f77SDag-Erling Smørgrav 	(void)(*c->callback)(c, c->cb_arg, err, NULL);
3813b7579f77SDag-Erling Smørgrav }
3814b7579f77SDag-Erling Smørgrav 
3815b7579f77SDag-Erling Smørgrav struct comm_point*
381617d15b25SDag-Erling Smørgrav comm_point_create_udp(struct comm_base *base, int fd, sldns_buffer* buffer,
3817865f46b2SCy Schubert 	int pp2_enabled, comm_point_callback_type* callback,
3818865f46b2SCy Schubert 	void* callback_arg, struct unbound_socket* socket)
3819b7579f77SDag-Erling Smørgrav {
3820b7579f77SDag-Erling Smørgrav 	struct comm_point* c = (struct comm_point*)calloc(1,
3821b7579f77SDag-Erling Smørgrav 		sizeof(struct comm_point));
3822b7579f77SDag-Erling Smørgrav 	short evbits;
3823b7579f77SDag-Erling Smørgrav 	if(!c)
3824b7579f77SDag-Erling Smørgrav 		return NULL;
3825b7579f77SDag-Erling Smørgrav 	c->ev = (struct internal_event*)calloc(1,
3826b7579f77SDag-Erling Smørgrav 		sizeof(struct internal_event));
3827b7579f77SDag-Erling Smørgrav 	if(!c->ev) {
3828b7579f77SDag-Erling Smørgrav 		free(c);
3829b7579f77SDag-Erling Smørgrav 		return NULL;
3830b7579f77SDag-Erling Smørgrav 	}
3831b7579f77SDag-Erling Smørgrav 	c->ev->base = base;
3832b7579f77SDag-Erling Smørgrav 	c->fd = fd;
3833b7579f77SDag-Erling Smørgrav 	c->buffer = buffer;
3834b7579f77SDag-Erling Smørgrav 	c->timeout = NULL;
3835b7579f77SDag-Erling Smørgrav 	c->tcp_is_reading = 0;
3836b7579f77SDag-Erling Smørgrav 	c->tcp_byte_count = 0;
3837b7579f77SDag-Erling Smørgrav 	c->tcp_parent = NULL;
3838b7579f77SDag-Erling Smørgrav 	c->max_tcp_count = 0;
383909a3aaf3SDag-Erling Smørgrav 	c->cur_tcp_count = 0;
3840b7579f77SDag-Erling Smørgrav 	c->tcp_handlers = NULL;
3841b7579f77SDag-Erling Smørgrav 	c->tcp_free = NULL;
3842b7579f77SDag-Erling Smørgrav 	c->type = comm_udp;
3843b7579f77SDag-Erling Smørgrav 	c->tcp_do_close = 0;
3844b7579f77SDag-Erling Smørgrav 	c->do_not_close = 0;
3845b7579f77SDag-Erling Smørgrav 	c->tcp_do_toggle_rw = 0;
3846b7579f77SDag-Erling Smørgrav 	c->tcp_check_nb_connect = 0;
3847b5663de9SDag-Erling Smørgrav #ifdef USE_MSG_FASTOPEN
3848b5663de9SDag-Erling Smørgrav 	c->tcp_do_fastopen = 0;
3849b5663de9SDag-Erling Smørgrav #endif
385065b390aaSDag-Erling Smørgrav #ifdef USE_DNSCRYPT
385165b390aaSDag-Erling Smørgrav 	c->dnscrypt = 0;
385265b390aaSDag-Erling Smørgrav 	c->dnscrypt_buffer = buffer;
385365b390aaSDag-Erling Smørgrav #endif
3854b7579f77SDag-Erling Smørgrav 	c->inuse = 0;
3855b7579f77SDag-Erling Smørgrav 	c->callback = callback;
3856b7579f77SDag-Erling Smørgrav 	c->cb_arg = callback_arg;
38575469a995SCy Schubert 	c->socket = socket;
3858865f46b2SCy Schubert 	c->pp2_enabled = pp2_enabled;
3859865f46b2SCy Schubert 	c->pp2_header_state = pp2_header_none;
3860e2d15004SDag-Erling Smørgrav 	evbits = UB_EV_READ | UB_EV_PERSIST;
3861e2d15004SDag-Erling Smørgrav 	/* ub_event stuff */
3862e2d15004SDag-Erling Smørgrav 	c->ev->ev = ub_event_new(base->eb->base, c->fd, evbits,
3863*8f76bb7dSCy Schubert #ifdef USE_WINSOCK
3864e2d15004SDag-Erling Smørgrav 		comm_point_udp_callback, c);
3865*8f76bb7dSCy Schubert #else
3866*8f76bb7dSCy Schubert 		comm_point_udp_ancil_callback, c);
3867*8f76bb7dSCy Schubert #endif
3868e2d15004SDag-Erling Smørgrav 	if(c->ev->ev == NULL) {
3869b7579f77SDag-Erling Smørgrav 		log_err("could not baseset udp event");
3870b7579f77SDag-Erling Smørgrav 		comm_point_delete(c);
3871b7579f77SDag-Erling Smørgrav 		return NULL;
3872b7579f77SDag-Erling Smørgrav 	}
3873e2d15004SDag-Erling Smørgrav 	if(fd!=-1 && ub_event_add(c->ev->ev, c->timeout) != 0 ) {
3874b7579f77SDag-Erling Smørgrav 		log_err("could not add udp event");
3875b7579f77SDag-Erling Smørgrav 		comm_point_delete(c);
3876b7579f77SDag-Erling Smørgrav 		return NULL;
3877b7579f77SDag-Erling Smørgrav 	}
3878f44e67d1SCy Schubert 	c->event_added = 1;
3879b7579f77SDag-Erling Smørgrav 	return c;
3880b7579f77SDag-Erling Smørgrav }
3881b7579f77SDag-Erling Smørgrav 
3882b7579f77SDag-Erling Smørgrav struct comm_point*
3883b7579f77SDag-Erling Smørgrav comm_point_create_udp_ancil(struct comm_base *base, int fd,
3884865f46b2SCy Schubert 	sldns_buffer* buffer, int pp2_enabled,
38855469a995SCy Schubert 	comm_point_callback_type* callback, void* callback_arg, struct unbound_socket* socket)
3886b7579f77SDag-Erling Smørgrav {
3887b7579f77SDag-Erling Smørgrav 	struct comm_point* c = (struct comm_point*)calloc(1,
3888b7579f77SDag-Erling Smørgrav 		sizeof(struct comm_point));
3889b7579f77SDag-Erling Smørgrav 	short evbits;
3890b7579f77SDag-Erling Smørgrav 	if(!c)
3891b7579f77SDag-Erling Smørgrav 		return NULL;
3892b7579f77SDag-Erling Smørgrav 	c->ev = (struct internal_event*)calloc(1,
3893b7579f77SDag-Erling Smørgrav 		sizeof(struct internal_event));
3894b7579f77SDag-Erling Smørgrav 	if(!c->ev) {
3895b7579f77SDag-Erling Smørgrav 		free(c);
3896b7579f77SDag-Erling Smørgrav 		return NULL;
3897b7579f77SDag-Erling Smørgrav 	}
3898b7579f77SDag-Erling Smørgrav 	c->ev->base = base;
3899b7579f77SDag-Erling Smørgrav 	c->fd = fd;
3900b7579f77SDag-Erling Smørgrav 	c->buffer = buffer;
3901b7579f77SDag-Erling Smørgrav 	c->timeout = NULL;
3902b7579f77SDag-Erling Smørgrav 	c->tcp_is_reading = 0;
3903b7579f77SDag-Erling Smørgrav 	c->tcp_byte_count = 0;
3904b7579f77SDag-Erling Smørgrav 	c->tcp_parent = NULL;
3905b7579f77SDag-Erling Smørgrav 	c->max_tcp_count = 0;
390609a3aaf3SDag-Erling Smørgrav 	c->cur_tcp_count = 0;
3907b7579f77SDag-Erling Smørgrav 	c->tcp_handlers = NULL;
3908b7579f77SDag-Erling Smørgrav 	c->tcp_free = NULL;
3909b7579f77SDag-Erling Smørgrav 	c->type = comm_udp;
3910b7579f77SDag-Erling Smørgrav 	c->tcp_do_close = 0;
3911b7579f77SDag-Erling Smørgrav 	c->do_not_close = 0;
391265b390aaSDag-Erling Smørgrav #ifdef USE_DNSCRYPT
391365b390aaSDag-Erling Smørgrav 	c->dnscrypt = 0;
391465b390aaSDag-Erling Smørgrav 	c->dnscrypt_buffer = buffer;
391565b390aaSDag-Erling Smørgrav #endif
3916b7579f77SDag-Erling Smørgrav 	c->inuse = 0;
3917b7579f77SDag-Erling Smørgrav 	c->tcp_do_toggle_rw = 0;
3918b7579f77SDag-Erling Smørgrav 	c->tcp_check_nb_connect = 0;
3919b5663de9SDag-Erling Smørgrav #ifdef USE_MSG_FASTOPEN
3920b5663de9SDag-Erling Smørgrav 	c->tcp_do_fastopen = 0;
3921b5663de9SDag-Erling Smørgrav #endif
3922b7579f77SDag-Erling Smørgrav 	c->callback = callback;
3923b7579f77SDag-Erling Smørgrav 	c->cb_arg = callback_arg;
39245469a995SCy Schubert 	c->socket = socket;
3925865f46b2SCy Schubert 	c->pp2_enabled = pp2_enabled;
3926865f46b2SCy Schubert 	c->pp2_header_state = pp2_header_none;
3927e2d15004SDag-Erling Smørgrav 	evbits = UB_EV_READ | UB_EV_PERSIST;
3928e2d15004SDag-Erling Smørgrav 	/* ub_event stuff */
3929e2d15004SDag-Erling Smørgrav 	c->ev->ev = ub_event_new(base->eb->base, c->fd, evbits,
3930e2d15004SDag-Erling Smørgrav 		comm_point_udp_ancil_callback, c);
3931e2d15004SDag-Erling Smørgrav 	if(c->ev->ev == NULL) {
3932b7579f77SDag-Erling Smørgrav 		log_err("could not baseset udp event");
3933b7579f77SDag-Erling Smørgrav 		comm_point_delete(c);
3934b7579f77SDag-Erling Smørgrav 		return NULL;
3935b7579f77SDag-Erling Smørgrav 	}
3936e2d15004SDag-Erling Smørgrav 	if(fd!=-1 && ub_event_add(c->ev->ev, c->timeout) != 0 ) {
3937b7579f77SDag-Erling Smørgrav 		log_err("could not add udp event");
3938b7579f77SDag-Erling Smørgrav 		comm_point_delete(c);
3939b7579f77SDag-Erling Smørgrav 		return NULL;
3940b7579f77SDag-Erling Smørgrav 	}
3941f44e67d1SCy Schubert 	c->event_added = 1;
3942b7579f77SDag-Erling Smørgrav 	return c;
3943b7579f77SDag-Erling Smørgrav }
3944b7579f77SDag-Erling Smørgrav 
3945b7579f77SDag-Erling Smørgrav static struct comm_point*
3946b7579f77SDag-Erling Smørgrav comm_point_create_tcp_handler(struct comm_base *base,
3947b7579f77SDag-Erling Smørgrav 	struct comm_point* parent, size_t bufsize,
3948e86b9096SDag-Erling Smørgrav 	struct sldns_buffer* spoolbuf, comm_point_callback_type* callback,
39495469a995SCy Schubert 	void* callback_arg, struct unbound_socket* socket)
3950b7579f77SDag-Erling Smørgrav {
3951b7579f77SDag-Erling Smørgrav 	struct comm_point* c = (struct comm_point*)calloc(1,
3952b7579f77SDag-Erling Smørgrav 		sizeof(struct comm_point));
3953b7579f77SDag-Erling Smørgrav 	short evbits;
3954b7579f77SDag-Erling Smørgrav 	if(!c)
3955b7579f77SDag-Erling Smørgrav 		return NULL;
3956b7579f77SDag-Erling Smørgrav 	c->ev = (struct internal_event*)calloc(1,
3957b7579f77SDag-Erling Smørgrav 		sizeof(struct internal_event));
3958b7579f77SDag-Erling Smørgrav 	if(!c->ev) {
3959b7579f77SDag-Erling Smørgrav 		free(c);
3960b7579f77SDag-Erling Smørgrav 		return NULL;
3961b7579f77SDag-Erling Smørgrav 	}
3962b7579f77SDag-Erling Smørgrav 	c->ev->base = base;
3963b7579f77SDag-Erling Smørgrav 	c->fd = -1;
396417d15b25SDag-Erling Smørgrav 	c->buffer = sldns_buffer_new(bufsize);
3965b7579f77SDag-Erling Smørgrav 	if(!c->buffer) {
3966b7579f77SDag-Erling Smørgrav 		free(c->ev);
3967b7579f77SDag-Erling Smørgrav 		free(c);
3968b7579f77SDag-Erling Smørgrav 		return NULL;
3969b7579f77SDag-Erling Smørgrav 	}
3970b7579f77SDag-Erling Smørgrav 	c->timeout = (struct timeval*)malloc(sizeof(struct timeval));
3971b7579f77SDag-Erling Smørgrav 	if(!c->timeout) {
397217d15b25SDag-Erling Smørgrav 		sldns_buffer_free(c->buffer);
3973b7579f77SDag-Erling Smørgrav 		free(c->ev);
3974b7579f77SDag-Erling Smørgrav 		free(c);
3975b7579f77SDag-Erling Smørgrav 		return NULL;
3976b7579f77SDag-Erling Smørgrav 	}
3977b7579f77SDag-Erling Smørgrav 	c->tcp_is_reading = 0;
3978b7579f77SDag-Erling Smørgrav 	c->tcp_byte_count = 0;
3979b7579f77SDag-Erling Smørgrav 	c->tcp_parent = parent;
39804c75e3aaSDag-Erling Smørgrav 	c->tcp_timeout_msec = parent->tcp_timeout_msec;
39814c75e3aaSDag-Erling Smørgrav 	c->tcp_conn_limit = parent->tcp_conn_limit;
39824c75e3aaSDag-Erling Smørgrav 	c->tcl_addr = NULL;
39834c75e3aaSDag-Erling Smørgrav 	c->tcp_keepalive = 0;
3984b7579f77SDag-Erling Smørgrav 	c->max_tcp_count = 0;
398509a3aaf3SDag-Erling Smørgrav 	c->cur_tcp_count = 0;
3986b7579f77SDag-Erling Smørgrav 	c->tcp_handlers = NULL;
3987b7579f77SDag-Erling Smørgrav 	c->tcp_free = NULL;
3988b7579f77SDag-Erling Smørgrav 	c->type = comm_tcp;
3989b7579f77SDag-Erling Smørgrav 	c->tcp_do_close = 0;
3990b7579f77SDag-Erling Smørgrav 	c->do_not_close = 0;
3991b7579f77SDag-Erling Smørgrav 	c->tcp_do_toggle_rw = 1;
3992b7579f77SDag-Erling Smørgrav 	c->tcp_check_nb_connect = 0;
3993b5663de9SDag-Erling Smørgrav #ifdef USE_MSG_FASTOPEN
3994b5663de9SDag-Erling Smørgrav 	c->tcp_do_fastopen = 0;
3995b5663de9SDag-Erling Smørgrav #endif
399665b390aaSDag-Erling Smørgrav #ifdef USE_DNSCRYPT
399765b390aaSDag-Erling Smørgrav 	c->dnscrypt = 0;
3998c7f4d7adSDag-Erling Smørgrav 	/* We don't know just yet if this is a dnscrypt channel. Allocation
3999c7f4d7adSDag-Erling Smørgrav 	 * will be done when handling the callback. */
400065b390aaSDag-Erling Smørgrav 	c->dnscrypt_buffer = c->buffer;
400165b390aaSDag-Erling Smørgrav #endif
4002b7579f77SDag-Erling Smørgrav 	c->repinfo.c = c;
4003b7579f77SDag-Erling Smørgrav 	c->callback = callback;
4004b7579f77SDag-Erling Smørgrav 	c->cb_arg = callback_arg;
40055469a995SCy Schubert 	c->socket = socket;
4006865f46b2SCy Schubert 	c->pp2_enabled = parent->pp2_enabled;
4007865f46b2SCy Schubert 	c->pp2_header_state = pp2_header_none;
4008e86b9096SDag-Erling Smørgrav 	if(spoolbuf) {
4009e86b9096SDag-Erling Smørgrav 		c->tcp_req_info = tcp_req_info_create(spoolbuf);
4010e86b9096SDag-Erling Smørgrav 		if(!c->tcp_req_info) {
4011e86b9096SDag-Erling Smørgrav 			log_err("could not create tcp commpoint");
4012e86b9096SDag-Erling Smørgrav 			sldns_buffer_free(c->buffer);
4013e86b9096SDag-Erling Smørgrav 			free(c->timeout);
4014e86b9096SDag-Erling Smørgrav 			free(c->ev);
4015e86b9096SDag-Erling Smørgrav 			free(c);
4016e86b9096SDag-Erling Smørgrav 			return NULL;
4017e86b9096SDag-Erling Smørgrav 		}
4018e86b9096SDag-Erling Smørgrav 		c->tcp_req_info->cp = c;
4019e86b9096SDag-Erling Smørgrav 		c->tcp_do_close = 1;
4020e86b9096SDag-Erling Smørgrav 		c->tcp_do_toggle_rw = 0;
4021e86b9096SDag-Erling Smørgrav 	}
4022b7579f77SDag-Erling Smørgrav 	/* add to parent free list */
4023b7579f77SDag-Erling Smørgrav 	c->tcp_free = parent->tcp_free;
4024b7579f77SDag-Erling Smørgrav 	parent->tcp_free = c;
4025e2d15004SDag-Erling Smørgrav 	/* ub_event stuff */
4026e2d15004SDag-Erling Smørgrav 	evbits = UB_EV_PERSIST | UB_EV_READ | UB_EV_TIMEOUT;
4027e2d15004SDag-Erling Smørgrav 	c->ev->ev = ub_event_new(base->eb->base, c->fd, evbits,
4028e2d15004SDag-Erling Smørgrav 		comm_point_tcp_handle_callback, c);
4029e2d15004SDag-Erling Smørgrav 	if(c->ev->ev == NULL)
4030b7579f77SDag-Erling Smørgrav 	{
4031b7579f77SDag-Erling Smørgrav 		log_err("could not basetset tcphdl event");
4032b7579f77SDag-Erling Smørgrav 		parent->tcp_free = c->tcp_free;
4033e86b9096SDag-Erling Smørgrav 		tcp_req_info_delete(c->tcp_req_info);
4034e86b9096SDag-Erling Smørgrav 		sldns_buffer_free(c->buffer);
4035e86b9096SDag-Erling Smørgrav 		free(c->timeout);
4036b7579f77SDag-Erling Smørgrav 		free(c->ev);
4037b7579f77SDag-Erling Smørgrav 		free(c);
4038b7579f77SDag-Erling Smørgrav 		return NULL;
4039b7579f77SDag-Erling Smørgrav 	}
4040b7579f77SDag-Erling Smørgrav 	return c;
4041b7579f77SDag-Erling Smørgrav }
4042b7579f77SDag-Erling Smørgrav 
4043c0caa2e2SCy Schubert static struct comm_point*
4044c0caa2e2SCy Schubert comm_point_create_http_handler(struct comm_base *base,
4045c0caa2e2SCy Schubert 	struct comm_point* parent, size_t bufsize, int harden_large_queries,
4046c0caa2e2SCy Schubert 	uint32_t http_max_streams, char* http_endpoint,
40475469a995SCy Schubert 	comm_point_callback_type* callback, void* callback_arg,
40485469a995SCy Schubert 	struct unbound_socket* socket)
4049c0caa2e2SCy Schubert {
4050c0caa2e2SCy Schubert 	struct comm_point* c = (struct comm_point*)calloc(1,
4051c0caa2e2SCy Schubert 		sizeof(struct comm_point));
4052c0caa2e2SCy Schubert 	short evbits;
4053c0caa2e2SCy Schubert 	if(!c)
4054c0caa2e2SCy Schubert 		return NULL;
4055c0caa2e2SCy Schubert 	c->ev = (struct internal_event*)calloc(1,
4056c0caa2e2SCy Schubert 		sizeof(struct internal_event));
4057c0caa2e2SCy Schubert 	if(!c->ev) {
4058c0caa2e2SCy Schubert 		free(c);
4059c0caa2e2SCy Schubert 		return NULL;
4060c0caa2e2SCy Schubert 	}
4061c0caa2e2SCy Schubert 	c->ev->base = base;
4062c0caa2e2SCy Schubert 	c->fd = -1;
4063c0caa2e2SCy Schubert 	c->buffer = sldns_buffer_new(bufsize);
4064c0caa2e2SCy Schubert 	if(!c->buffer) {
4065c0caa2e2SCy Schubert 		free(c->ev);
4066c0caa2e2SCy Schubert 		free(c);
4067c0caa2e2SCy Schubert 		return NULL;
4068c0caa2e2SCy Schubert 	}
4069c0caa2e2SCy Schubert 	c->timeout = (struct timeval*)malloc(sizeof(struct timeval));
4070c0caa2e2SCy Schubert 	if(!c->timeout) {
4071c0caa2e2SCy Schubert 		sldns_buffer_free(c->buffer);
4072c0caa2e2SCy Schubert 		free(c->ev);
4073c0caa2e2SCy Schubert 		free(c);
4074c0caa2e2SCy Schubert 		return NULL;
4075c0caa2e2SCy Schubert 	}
4076c0caa2e2SCy Schubert 	c->tcp_is_reading = 0;
4077c0caa2e2SCy Schubert 	c->tcp_byte_count = 0;
4078c0caa2e2SCy Schubert 	c->tcp_parent = parent;
4079c0caa2e2SCy Schubert 	c->tcp_timeout_msec = parent->tcp_timeout_msec;
4080c0caa2e2SCy Schubert 	c->tcp_conn_limit = parent->tcp_conn_limit;
4081c0caa2e2SCy Schubert 	c->tcl_addr = NULL;
4082c0caa2e2SCy Schubert 	c->tcp_keepalive = 0;
4083c0caa2e2SCy Schubert 	c->max_tcp_count = 0;
4084c0caa2e2SCy Schubert 	c->cur_tcp_count = 0;
4085c0caa2e2SCy Schubert 	c->tcp_handlers = NULL;
4086c0caa2e2SCy Schubert 	c->tcp_free = NULL;
4087c0caa2e2SCy Schubert 	c->type = comm_http;
4088c0caa2e2SCy Schubert 	c->tcp_do_close = 1;
4089c0caa2e2SCy Schubert 	c->do_not_close = 0;
4090c0caa2e2SCy Schubert 	c->tcp_do_toggle_rw = 1; /* will be set to 0 after http2 upgrade */
4091c0caa2e2SCy Schubert 	c->tcp_check_nb_connect = 0;
4092c0caa2e2SCy Schubert #ifdef USE_MSG_FASTOPEN
4093c0caa2e2SCy Schubert 	c->tcp_do_fastopen = 0;
4094c0caa2e2SCy Schubert #endif
4095c0caa2e2SCy Schubert #ifdef USE_DNSCRYPT
4096c0caa2e2SCy Schubert 	c->dnscrypt = 0;
4097c0caa2e2SCy Schubert 	c->dnscrypt_buffer = NULL;
4098c0caa2e2SCy Schubert #endif
4099c0caa2e2SCy Schubert 	c->repinfo.c = c;
4100c0caa2e2SCy Schubert 	c->callback = callback;
4101c0caa2e2SCy Schubert 	c->cb_arg = callback_arg;
41025469a995SCy Schubert 	c->socket = socket;
4103865f46b2SCy Schubert 	c->pp2_enabled = 0;
4104865f46b2SCy Schubert 	c->pp2_header_state = pp2_header_none;
4105c0caa2e2SCy Schubert 
4106c0caa2e2SCy Schubert 	c->http_min_version = http_version_2;
4107c0caa2e2SCy Schubert 	c->http2_stream_max_qbuffer_size = bufsize;
4108c0caa2e2SCy Schubert 	if(harden_large_queries && bufsize > 512)
4109c0caa2e2SCy Schubert 		c->http2_stream_max_qbuffer_size = 512;
4110c0caa2e2SCy Schubert 	c->http2_max_streams = http_max_streams;
4111c0caa2e2SCy Schubert 	if(!(c->http_endpoint = strdup(http_endpoint))) {
4112c0caa2e2SCy Schubert 		log_err("could not strdup http_endpoint");
4113c0caa2e2SCy Schubert 		sldns_buffer_free(c->buffer);
4114c0caa2e2SCy Schubert 		free(c->timeout);
4115c0caa2e2SCy Schubert 		free(c->ev);
4116c0caa2e2SCy Schubert 		free(c);
4117c0caa2e2SCy Schubert 		return NULL;
4118c0caa2e2SCy Schubert 	}
4119c0caa2e2SCy Schubert 	c->use_h2 = 0;
4120c0caa2e2SCy Schubert #ifdef HAVE_NGHTTP2
4121c0caa2e2SCy Schubert 	if(!(c->h2_session = http2_session_create(c))) {
4122c0caa2e2SCy Schubert 		log_err("could not create http2 session");
4123c0caa2e2SCy Schubert 		free(c->http_endpoint);
4124c0caa2e2SCy Schubert 		sldns_buffer_free(c->buffer);
4125c0caa2e2SCy Schubert 		free(c->timeout);
4126c0caa2e2SCy Schubert 		free(c->ev);
4127c0caa2e2SCy Schubert 		free(c);
4128c0caa2e2SCy Schubert 		return NULL;
4129c0caa2e2SCy Schubert 	}
4130c0caa2e2SCy Schubert 	if(!(c->h2_session->callbacks = http2_req_callbacks_create())) {
4131c0caa2e2SCy Schubert 		log_err("could not create http2 callbacks");
4132c0caa2e2SCy Schubert 		http2_session_delete(c->h2_session);
4133c0caa2e2SCy Schubert 		free(c->http_endpoint);
4134c0caa2e2SCy Schubert 		sldns_buffer_free(c->buffer);
4135c0caa2e2SCy Schubert 		free(c->timeout);
4136c0caa2e2SCy Schubert 		free(c->ev);
4137c0caa2e2SCy Schubert 		free(c);
4138c0caa2e2SCy Schubert 		return NULL;
4139c0caa2e2SCy Schubert 	}
4140c0caa2e2SCy Schubert #endif
4141c0caa2e2SCy Schubert 
4142c0caa2e2SCy Schubert 	/* add to parent free list */
4143c0caa2e2SCy Schubert 	c->tcp_free = parent->tcp_free;
4144c0caa2e2SCy Schubert 	parent->tcp_free = c;
4145c0caa2e2SCy Schubert 	/* ub_event stuff */
4146c0caa2e2SCy Schubert 	evbits = UB_EV_PERSIST | UB_EV_READ | UB_EV_TIMEOUT;
4147c0caa2e2SCy Schubert 	c->ev->ev = ub_event_new(base->eb->base, c->fd, evbits,
4148c0caa2e2SCy Schubert 		comm_point_http_handle_callback, c);
4149c0caa2e2SCy Schubert 	if(c->ev->ev == NULL)
4150c0caa2e2SCy Schubert 	{
4151c0caa2e2SCy Schubert 		log_err("could not set http handler event");
4152c0caa2e2SCy Schubert 		parent->tcp_free = c->tcp_free;
4153c0caa2e2SCy Schubert 		http2_session_delete(c->h2_session);
4154c0caa2e2SCy Schubert 		sldns_buffer_free(c->buffer);
4155c0caa2e2SCy Schubert 		free(c->timeout);
4156c0caa2e2SCy Schubert 		free(c->ev);
4157c0caa2e2SCy Schubert 		free(c);
4158c0caa2e2SCy Schubert 		return NULL;
4159c0caa2e2SCy Schubert 	}
4160c0caa2e2SCy Schubert 	return c;
4161c0caa2e2SCy Schubert }
4162c0caa2e2SCy Schubert 
4163b7579f77SDag-Erling Smørgrav struct comm_point*
41644c75e3aaSDag-Erling Smørgrav comm_point_create_tcp(struct comm_base *base, int fd, int num,
4165c0caa2e2SCy Schubert 	int idle_timeout, int harden_large_queries,
4166c0caa2e2SCy Schubert 	uint32_t http_max_streams, char* http_endpoint,
4167c0caa2e2SCy Schubert 	struct tcl_list* tcp_conn_limit, size_t bufsize,
4168c0caa2e2SCy Schubert 	struct sldns_buffer* spoolbuf, enum listen_type port_type,
4169865f46b2SCy Schubert 	int pp2_enabled, comm_point_callback_type* callback,
4170865f46b2SCy Schubert 	void* callback_arg, struct unbound_socket* socket)
4171b7579f77SDag-Erling Smørgrav {
4172b7579f77SDag-Erling Smørgrav 	struct comm_point* c = (struct comm_point*)calloc(1,
4173b7579f77SDag-Erling Smørgrav 		sizeof(struct comm_point));
4174b7579f77SDag-Erling Smørgrav 	short evbits;
4175b7579f77SDag-Erling Smørgrav 	int i;
4176b7579f77SDag-Erling Smørgrav 	/* first allocate the TCP accept listener */
4177b7579f77SDag-Erling Smørgrav 	if(!c)
4178b7579f77SDag-Erling Smørgrav 		return NULL;
4179b7579f77SDag-Erling Smørgrav 	c->ev = (struct internal_event*)calloc(1,
4180b7579f77SDag-Erling Smørgrav 		sizeof(struct internal_event));
4181b7579f77SDag-Erling Smørgrav 	if(!c->ev) {
4182b7579f77SDag-Erling Smørgrav 		free(c);
4183b7579f77SDag-Erling Smørgrav 		return NULL;
4184b7579f77SDag-Erling Smørgrav 	}
4185b7579f77SDag-Erling Smørgrav 	c->ev->base = base;
4186b7579f77SDag-Erling Smørgrav 	c->fd = fd;
4187b7579f77SDag-Erling Smørgrav 	c->buffer = NULL;
4188b7579f77SDag-Erling Smørgrav 	c->timeout = NULL;
4189b7579f77SDag-Erling Smørgrav 	c->tcp_is_reading = 0;
4190b7579f77SDag-Erling Smørgrav 	c->tcp_byte_count = 0;
41914c75e3aaSDag-Erling Smørgrav 	c->tcp_timeout_msec = idle_timeout;
41924c75e3aaSDag-Erling Smørgrav 	c->tcp_conn_limit = tcp_conn_limit;
41934c75e3aaSDag-Erling Smørgrav 	c->tcl_addr = NULL;
41944c75e3aaSDag-Erling Smørgrav 	c->tcp_keepalive = 0;
4195b7579f77SDag-Erling Smørgrav 	c->tcp_parent = NULL;
4196b7579f77SDag-Erling Smørgrav 	c->max_tcp_count = num;
419709a3aaf3SDag-Erling Smørgrav 	c->cur_tcp_count = 0;
4198b7579f77SDag-Erling Smørgrav 	c->tcp_handlers = (struct comm_point**)calloc((size_t)num,
4199b7579f77SDag-Erling Smørgrav 		sizeof(struct comm_point*));
4200b7579f77SDag-Erling Smørgrav 	if(!c->tcp_handlers) {
4201b7579f77SDag-Erling Smørgrav 		free(c->ev);
4202b7579f77SDag-Erling Smørgrav 		free(c);
4203b7579f77SDag-Erling Smørgrav 		return NULL;
4204b7579f77SDag-Erling Smørgrav 	}
4205b7579f77SDag-Erling Smørgrav 	c->tcp_free = NULL;
4206b7579f77SDag-Erling Smørgrav 	c->type = comm_tcp_accept;
4207b7579f77SDag-Erling Smørgrav 	c->tcp_do_close = 0;
4208b7579f77SDag-Erling Smørgrav 	c->do_not_close = 0;
4209b7579f77SDag-Erling Smørgrav 	c->tcp_do_toggle_rw = 0;
4210b7579f77SDag-Erling Smørgrav 	c->tcp_check_nb_connect = 0;
4211b5663de9SDag-Erling Smørgrav #ifdef USE_MSG_FASTOPEN
4212b5663de9SDag-Erling Smørgrav 	c->tcp_do_fastopen = 0;
4213b5663de9SDag-Erling Smørgrav #endif
421465b390aaSDag-Erling Smørgrav #ifdef USE_DNSCRYPT
421565b390aaSDag-Erling Smørgrav 	c->dnscrypt = 0;
421665b390aaSDag-Erling Smørgrav 	c->dnscrypt_buffer = NULL;
421765b390aaSDag-Erling Smørgrav #endif
4218b7579f77SDag-Erling Smørgrav 	c->callback = NULL;
4219b7579f77SDag-Erling Smørgrav 	c->cb_arg = NULL;
42205469a995SCy Schubert 	c->socket = socket;
4221865f46b2SCy Schubert 	c->pp2_enabled = (port_type==listen_type_http?0:pp2_enabled);
4222865f46b2SCy Schubert 	c->pp2_header_state = pp2_header_none;
4223e2d15004SDag-Erling Smørgrav 	evbits = UB_EV_READ | UB_EV_PERSIST;
4224e2d15004SDag-Erling Smørgrav 	/* ub_event stuff */
4225e2d15004SDag-Erling Smørgrav 	c->ev->ev = ub_event_new(base->eb->base, c->fd, evbits,
4226e2d15004SDag-Erling Smørgrav 		comm_point_tcp_accept_callback, c);
4227e2d15004SDag-Erling Smørgrav 	if(c->ev->ev == NULL) {
4228e2d15004SDag-Erling Smørgrav 		log_err("could not baseset tcpacc event");
4229e2d15004SDag-Erling Smørgrav 		comm_point_delete(c);
4230e2d15004SDag-Erling Smørgrav 		return NULL;
4231e2d15004SDag-Erling Smørgrav 	}
4232e2d15004SDag-Erling Smørgrav 	if (ub_event_add(c->ev->ev, c->timeout) != 0) {
4233b7579f77SDag-Erling Smørgrav 		log_err("could not add tcpacc event");
4234b7579f77SDag-Erling Smørgrav 		comm_point_delete(c);
4235b7579f77SDag-Erling Smørgrav 		return NULL;
4236b7579f77SDag-Erling Smørgrav 	}
4237f44e67d1SCy Schubert 	c->event_added = 1;
4238c0caa2e2SCy Schubert 	/* now prealloc the handlers */
4239b7579f77SDag-Erling Smørgrav 	for(i=0; i<num; i++) {
4240c0caa2e2SCy Schubert 		if(port_type == listen_type_tcp ||
4241c0caa2e2SCy Schubert 			port_type == listen_type_ssl ||
4242c0caa2e2SCy Schubert 			port_type == listen_type_tcp_dnscrypt) {
4243b7579f77SDag-Erling Smørgrav 			c->tcp_handlers[i] = comm_point_create_tcp_handler(base,
42445469a995SCy Schubert 				c, bufsize, spoolbuf, callback, callback_arg, socket);
4245c0caa2e2SCy Schubert 		} else if(port_type == listen_type_http) {
4246c0caa2e2SCy Schubert 			c->tcp_handlers[i] = comm_point_create_http_handler(
4247c0caa2e2SCy Schubert 				base, c, bufsize, harden_large_queries,
4248c0caa2e2SCy Schubert 				http_max_streams, http_endpoint,
42495469a995SCy Schubert 				callback, callback_arg, socket);
4250c0caa2e2SCy Schubert 		}
4251c0caa2e2SCy Schubert 		else {
4252c0caa2e2SCy Schubert 			log_err("could not create tcp handler, unknown listen "
4253c0caa2e2SCy Schubert 				"type");
4254c0caa2e2SCy Schubert 			return NULL;
4255c0caa2e2SCy Schubert 		}
4256b7579f77SDag-Erling Smørgrav 		if(!c->tcp_handlers[i]) {
4257b7579f77SDag-Erling Smørgrav 			comm_point_delete(c);
4258b7579f77SDag-Erling Smørgrav 			return NULL;
4259b7579f77SDag-Erling Smørgrav 		}
4260b7579f77SDag-Erling Smørgrav 	}
4261b7579f77SDag-Erling Smørgrav 
4262b7579f77SDag-Erling Smørgrav 	return c;
4263b7579f77SDag-Erling Smørgrav }
4264b7579f77SDag-Erling Smørgrav 
4265b7579f77SDag-Erling Smørgrav struct comm_point*
4266b7579f77SDag-Erling Smørgrav comm_point_create_tcp_out(struct comm_base *base, size_t bufsize,
42673005e0a3SDag-Erling Smørgrav         comm_point_callback_type* callback, void* callback_arg)
4268b7579f77SDag-Erling Smørgrav {
4269b7579f77SDag-Erling Smørgrav 	struct comm_point* c = (struct comm_point*)calloc(1,
4270b7579f77SDag-Erling Smørgrav 		sizeof(struct comm_point));
4271b7579f77SDag-Erling Smørgrav 	short evbits;
4272b7579f77SDag-Erling Smørgrav 	if(!c)
4273b7579f77SDag-Erling Smørgrav 		return NULL;
4274b7579f77SDag-Erling Smørgrav 	c->ev = (struct internal_event*)calloc(1,
4275b7579f77SDag-Erling Smørgrav 		sizeof(struct internal_event));
4276b7579f77SDag-Erling Smørgrav 	if(!c->ev) {
4277b7579f77SDag-Erling Smørgrav 		free(c);
4278b7579f77SDag-Erling Smørgrav 		return NULL;
4279b7579f77SDag-Erling Smørgrav 	}
4280b7579f77SDag-Erling Smørgrav 	c->ev->base = base;
4281b7579f77SDag-Erling Smørgrav 	c->fd = -1;
428217d15b25SDag-Erling Smørgrav 	c->buffer = sldns_buffer_new(bufsize);
4283b7579f77SDag-Erling Smørgrav 	if(!c->buffer) {
4284b7579f77SDag-Erling Smørgrav 		free(c->ev);
4285b7579f77SDag-Erling Smørgrav 		free(c);
4286b7579f77SDag-Erling Smørgrav 		return NULL;
4287b7579f77SDag-Erling Smørgrav 	}
4288b7579f77SDag-Erling Smørgrav 	c->timeout = NULL;
4289b7579f77SDag-Erling Smørgrav 	c->tcp_is_reading = 0;
4290b7579f77SDag-Erling Smørgrav 	c->tcp_byte_count = 0;
42914c75e3aaSDag-Erling Smørgrav 	c->tcp_timeout_msec = TCP_QUERY_TIMEOUT;
42924c75e3aaSDag-Erling Smørgrav 	c->tcp_conn_limit = NULL;
42934c75e3aaSDag-Erling Smørgrav 	c->tcl_addr = NULL;
42944c75e3aaSDag-Erling Smørgrav 	c->tcp_keepalive = 0;
4295b7579f77SDag-Erling Smørgrav 	c->tcp_parent = NULL;
4296b7579f77SDag-Erling Smørgrav 	c->max_tcp_count = 0;
429709a3aaf3SDag-Erling Smørgrav 	c->cur_tcp_count = 0;
4298b7579f77SDag-Erling Smørgrav 	c->tcp_handlers = NULL;
4299b7579f77SDag-Erling Smørgrav 	c->tcp_free = NULL;
4300b7579f77SDag-Erling Smørgrav 	c->type = comm_tcp;
4301b7579f77SDag-Erling Smørgrav 	c->tcp_do_close = 0;
4302b7579f77SDag-Erling Smørgrav 	c->do_not_close = 0;
4303b7579f77SDag-Erling Smørgrav 	c->tcp_do_toggle_rw = 1;
4304b7579f77SDag-Erling Smørgrav 	c->tcp_check_nb_connect = 1;
4305b5663de9SDag-Erling Smørgrav #ifdef USE_MSG_FASTOPEN
4306b5663de9SDag-Erling Smørgrav 	c->tcp_do_fastopen = 1;
4307b5663de9SDag-Erling Smørgrav #endif
430865b390aaSDag-Erling Smørgrav #ifdef USE_DNSCRYPT
430965b390aaSDag-Erling Smørgrav 	c->dnscrypt = 0;
431065b390aaSDag-Erling Smørgrav 	c->dnscrypt_buffer = c->buffer;
431165b390aaSDag-Erling Smørgrav #endif
4312b7579f77SDag-Erling Smørgrav 	c->repinfo.c = c;
4313b7579f77SDag-Erling Smørgrav 	c->callback = callback;
4314b7579f77SDag-Erling Smørgrav 	c->cb_arg = callback_arg;
4315865f46b2SCy Schubert 	c->pp2_enabled = 0;
4316865f46b2SCy Schubert 	c->pp2_header_state = pp2_header_none;
4317e2d15004SDag-Erling Smørgrav 	evbits = UB_EV_PERSIST | UB_EV_WRITE;
4318e2d15004SDag-Erling Smørgrav 	c->ev->ev = ub_event_new(base->eb->base, c->fd, evbits,
4319e2d15004SDag-Erling Smørgrav 		comm_point_tcp_handle_callback, c);
4320e2d15004SDag-Erling Smørgrav 	if(c->ev->ev == NULL)
4321b7579f77SDag-Erling Smørgrav 	{
4322e2d15004SDag-Erling Smørgrav 		log_err("could not baseset tcpout event");
432317d15b25SDag-Erling Smørgrav 		sldns_buffer_free(c->buffer);
4324b7579f77SDag-Erling Smørgrav 		free(c->ev);
4325b7579f77SDag-Erling Smørgrav 		free(c);
4326b7579f77SDag-Erling Smørgrav 		return NULL;
4327b7579f77SDag-Erling Smørgrav 	}
4328b7579f77SDag-Erling Smørgrav 
4329b7579f77SDag-Erling Smørgrav 	return c;
4330b7579f77SDag-Erling Smørgrav }
4331b7579f77SDag-Erling Smørgrav 
4332b7579f77SDag-Erling Smørgrav struct comm_point*
433357bddd21SDag-Erling Smørgrav comm_point_create_http_out(struct comm_base *base, size_t bufsize,
433457bddd21SDag-Erling Smørgrav         comm_point_callback_type* callback, void* callback_arg,
433557bddd21SDag-Erling Smørgrav 	sldns_buffer* temp)
433657bddd21SDag-Erling Smørgrav {
433757bddd21SDag-Erling Smørgrav 	struct comm_point* c = (struct comm_point*)calloc(1,
433857bddd21SDag-Erling Smørgrav 		sizeof(struct comm_point));
433957bddd21SDag-Erling Smørgrav 	short evbits;
434057bddd21SDag-Erling Smørgrav 	if(!c)
434157bddd21SDag-Erling Smørgrav 		return NULL;
434257bddd21SDag-Erling Smørgrav 	c->ev = (struct internal_event*)calloc(1,
434357bddd21SDag-Erling Smørgrav 		sizeof(struct internal_event));
434457bddd21SDag-Erling Smørgrav 	if(!c->ev) {
434557bddd21SDag-Erling Smørgrav 		free(c);
434657bddd21SDag-Erling Smørgrav 		return NULL;
434757bddd21SDag-Erling Smørgrav 	}
434857bddd21SDag-Erling Smørgrav 	c->ev->base = base;
434957bddd21SDag-Erling Smørgrav 	c->fd = -1;
435057bddd21SDag-Erling Smørgrav 	c->buffer = sldns_buffer_new(bufsize);
435157bddd21SDag-Erling Smørgrav 	if(!c->buffer) {
435257bddd21SDag-Erling Smørgrav 		free(c->ev);
435357bddd21SDag-Erling Smørgrav 		free(c);
435457bddd21SDag-Erling Smørgrav 		return NULL;
435557bddd21SDag-Erling Smørgrav 	}
435657bddd21SDag-Erling Smørgrav 	c->timeout = NULL;
435757bddd21SDag-Erling Smørgrav 	c->tcp_is_reading = 0;
435857bddd21SDag-Erling Smørgrav 	c->tcp_byte_count = 0;
435957bddd21SDag-Erling Smørgrav 	c->tcp_parent = NULL;
436057bddd21SDag-Erling Smørgrav 	c->max_tcp_count = 0;
436157bddd21SDag-Erling Smørgrav 	c->cur_tcp_count = 0;
436257bddd21SDag-Erling Smørgrav 	c->tcp_handlers = NULL;
436357bddd21SDag-Erling Smørgrav 	c->tcp_free = NULL;
436457bddd21SDag-Erling Smørgrav 	c->type = comm_http;
436557bddd21SDag-Erling Smørgrav 	c->tcp_do_close = 0;
436657bddd21SDag-Erling Smørgrav 	c->do_not_close = 0;
436757bddd21SDag-Erling Smørgrav 	c->tcp_do_toggle_rw = 1;
436857bddd21SDag-Erling Smørgrav 	c->tcp_check_nb_connect = 1;
436957bddd21SDag-Erling Smørgrav 	c->http_in_headers = 1;
437057bddd21SDag-Erling Smørgrav 	c->http_in_chunk_headers = 0;
437157bddd21SDag-Erling Smørgrav 	c->http_is_chunked = 0;
437257bddd21SDag-Erling Smørgrav 	c->http_temp = temp;
437357bddd21SDag-Erling Smørgrav #ifdef USE_MSG_FASTOPEN
437457bddd21SDag-Erling Smørgrav 	c->tcp_do_fastopen = 1;
437557bddd21SDag-Erling Smørgrav #endif
437657bddd21SDag-Erling Smørgrav #ifdef USE_DNSCRYPT
437757bddd21SDag-Erling Smørgrav 	c->dnscrypt = 0;
437857bddd21SDag-Erling Smørgrav 	c->dnscrypt_buffer = c->buffer;
437957bddd21SDag-Erling Smørgrav #endif
438057bddd21SDag-Erling Smørgrav 	c->repinfo.c = c;
438157bddd21SDag-Erling Smørgrav 	c->callback = callback;
438257bddd21SDag-Erling Smørgrav 	c->cb_arg = callback_arg;
4383865f46b2SCy Schubert 	c->pp2_enabled = 0;
4384865f46b2SCy Schubert 	c->pp2_header_state = pp2_header_none;
438557bddd21SDag-Erling Smørgrav 	evbits = UB_EV_PERSIST | UB_EV_WRITE;
438657bddd21SDag-Erling Smørgrav 	c->ev->ev = ub_event_new(base->eb->base, c->fd, evbits,
438757bddd21SDag-Erling Smørgrav 		comm_point_http_handle_callback, c);
438857bddd21SDag-Erling Smørgrav 	if(c->ev->ev == NULL)
438957bddd21SDag-Erling Smørgrav 	{
439057bddd21SDag-Erling Smørgrav 		log_err("could not baseset tcpout event");
439157bddd21SDag-Erling Smørgrav #ifdef HAVE_SSL
439257bddd21SDag-Erling Smørgrav 		SSL_free(c->ssl);
439357bddd21SDag-Erling Smørgrav #endif
439457bddd21SDag-Erling Smørgrav 		sldns_buffer_free(c->buffer);
439557bddd21SDag-Erling Smørgrav 		free(c->ev);
439657bddd21SDag-Erling Smørgrav 		free(c);
439757bddd21SDag-Erling Smørgrav 		return NULL;
439857bddd21SDag-Erling Smørgrav 	}
439957bddd21SDag-Erling Smørgrav 
440057bddd21SDag-Erling Smørgrav 	return c;
440157bddd21SDag-Erling Smørgrav }
440257bddd21SDag-Erling Smørgrav 
440357bddd21SDag-Erling Smørgrav struct comm_point*
4404b7579f77SDag-Erling Smørgrav comm_point_create_local(struct comm_base *base, int fd, size_t bufsize,
44053005e0a3SDag-Erling Smørgrav         comm_point_callback_type* callback, void* callback_arg)
4406b7579f77SDag-Erling Smørgrav {
4407b7579f77SDag-Erling Smørgrav 	struct comm_point* c = (struct comm_point*)calloc(1,
4408b7579f77SDag-Erling Smørgrav 		sizeof(struct comm_point));
4409b7579f77SDag-Erling Smørgrav 	short evbits;
4410b7579f77SDag-Erling Smørgrav 	if(!c)
4411b7579f77SDag-Erling Smørgrav 		return NULL;
4412b7579f77SDag-Erling Smørgrav 	c->ev = (struct internal_event*)calloc(1,
4413b7579f77SDag-Erling Smørgrav 		sizeof(struct internal_event));
4414b7579f77SDag-Erling Smørgrav 	if(!c->ev) {
4415b7579f77SDag-Erling Smørgrav 		free(c);
4416b7579f77SDag-Erling Smørgrav 		return NULL;
4417b7579f77SDag-Erling Smørgrav 	}
4418b7579f77SDag-Erling Smørgrav 	c->ev->base = base;
4419b7579f77SDag-Erling Smørgrav 	c->fd = fd;
442017d15b25SDag-Erling Smørgrav 	c->buffer = sldns_buffer_new(bufsize);
4421b7579f77SDag-Erling Smørgrav 	if(!c->buffer) {
4422b7579f77SDag-Erling Smørgrav 		free(c->ev);
4423b7579f77SDag-Erling Smørgrav 		free(c);
4424b7579f77SDag-Erling Smørgrav 		return NULL;
4425b7579f77SDag-Erling Smørgrav 	}
4426b7579f77SDag-Erling Smørgrav 	c->timeout = NULL;
4427b7579f77SDag-Erling Smørgrav 	c->tcp_is_reading = 1;
4428b7579f77SDag-Erling Smørgrav 	c->tcp_byte_count = 0;
4429b7579f77SDag-Erling Smørgrav 	c->tcp_parent = NULL;
4430b7579f77SDag-Erling Smørgrav 	c->max_tcp_count = 0;
443109a3aaf3SDag-Erling Smørgrav 	c->cur_tcp_count = 0;
4432b7579f77SDag-Erling Smørgrav 	c->tcp_handlers = NULL;
4433b7579f77SDag-Erling Smørgrav 	c->tcp_free = NULL;
4434b7579f77SDag-Erling Smørgrav 	c->type = comm_local;
4435b7579f77SDag-Erling Smørgrav 	c->tcp_do_close = 0;
4436b7579f77SDag-Erling Smørgrav 	c->do_not_close = 1;
4437b7579f77SDag-Erling Smørgrav 	c->tcp_do_toggle_rw = 0;
4438b7579f77SDag-Erling Smørgrav 	c->tcp_check_nb_connect = 0;
4439b5663de9SDag-Erling Smørgrav #ifdef USE_MSG_FASTOPEN
4440b5663de9SDag-Erling Smørgrav 	c->tcp_do_fastopen = 0;
4441b5663de9SDag-Erling Smørgrav #endif
444265b390aaSDag-Erling Smørgrav #ifdef USE_DNSCRYPT
444365b390aaSDag-Erling Smørgrav 	c->dnscrypt = 0;
444465b390aaSDag-Erling Smørgrav 	c->dnscrypt_buffer = c->buffer;
444565b390aaSDag-Erling Smørgrav #endif
4446b7579f77SDag-Erling Smørgrav 	c->callback = callback;
4447b7579f77SDag-Erling Smørgrav 	c->cb_arg = callback_arg;
4448865f46b2SCy Schubert 	c->pp2_enabled = 0;
4449865f46b2SCy Schubert 	c->pp2_header_state = pp2_header_none;
4450e2d15004SDag-Erling Smørgrav 	/* ub_event stuff */
4451e2d15004SDag-Erling Smørgrav 	evbits = UB_EV_PERSIST | UB_EV_READ;
4452e2d15004SDag-Erling Smørgrav 	c->ev->ev = ub_event_new(base->eb->base, c->fd, evbits,
4453e2d15004SDag-Erling Smørgrav 		comm_point_local_handle_callback, c);
4454e2d15004SDag-Erling Smørgrav 	if(c->ev->ev == NULL) {
4455e2d15004SDag-Erling Smørgrav 		log_err("could not baseset localhdl event");
4456e2d15004SDag-Erling Smørgrav 		free(c->ev);
4457e2d15004SDag-Erling Smørgrav 		free(c);
4458e2d15004SDag-Erling Smørgrav 		return NULL;
4459e2d15004SDag-Erling Smørgrav 	}
4460e2d15004SDag-Erling Smørgrav 	if (ub_event_add(c->ev->ev, c->timeout) != 0) {
4461b7579f77SDag-Erling Smørgrav 		log_err("could not add localhdl event");
4462e2d15004SDag-Erling Smørgrav 		ub_event_free(c->ev->ev);
4463b7579f77SDag-Erling Smørgrav 		free(c->ev);
4464b7579f77SDag-Erling Smørgrav 		free(c);
4465b7579f77SDag-Erling Smørgrav 		return NULL;
4466b7579f77SDag-Erling Smørgrav 	}
4467f44e67d1SCy Schubert 	c->event_added = 1;
4468b7579f77SDag-Erling Smørgrav 	return c;
4469b7579f77SDag-Erling Smørgrav }
4470b7579f77SDag-Erling Smørgrav 
4471b7579f77SDag-Erling Smørgrav struct comm_point*
4472b7579f77SDag-Erling Smørgrav comm_point_create_raw(struct comm_base* base, int fd, int writing,
44733005e0a3SDag-Erling Smørgrav 	comm_point_callback_type* callback, void* callback_arg)
4474b7579f77SDag-Erling Smørgrav {
4475b7579f77SDag-Erling Smørgrav 	struct comm_point* c = (struct comm_point*)calloc(1,
4476b7579f77SDag-Erling Smørgrav 		sizeof(struct comm_point));
4477b7579f77SDag-Erling Smørgrav 	short evbits;
4478b7579f77SDag-Erling Smørgrav 	if(!c)
4479b7579f77SDag-Erling Smørgrav 		return NULL;
4480b7579f77SDag-Erling Smørgrav 	c->ev = (struct internal_event*)calloc(1,
4481b7579f77SDag-Erling Smørgrav 		sizeof(struct internal_event));
4482b7579f77SDag-Erling Smørgrav 	if(!c->ev) {
4483b7579f77SDag-Erling Smørgrav 		free(c);
4484b7579f77SDag-Erling Smørgrav 		return NULL;
4485b7579f77SDag-Erling Smørgrav 	}
4486b7579f77SDag-Erling Smørgrav 	c->ev->base = base;
4487b7579f77SDag-Erling Smørgrav 	c->fd = fd;
4488b7579f77SDag-Erling Smørgrav 	c->buffer = NULL;
4489b7579f77SDag-Erling Smørgrav 	c->timeout = NULL;
4490b7579f77SDag-Erling Smørgrav 	c->tcp_is_reading = 0;
4491b7579f77SDag-Erling Smørgrav 	c->tcp_byte_count = 0;
4492b7579f77SDag-Erling Smørgrav 	c->tcp_parent = NULL;
4493b7579f77SDag-Erling Smørgrav 	c->max_tcp_count = 0;
449409a3aaf3SDag-Erling Smørgrav 	c->cur_tcp_count = 0;
4495b7579f77SDag-Erling Smørgrav 	c->tcp_handlers = NULL;
4496b7579f77SDag-Erling Smørgrav 	c->tcp_free = NULL;
4497b7579f77SDag-Erling Smørgrav 	c->type = comm_raw;
4498b7579f77SDag-Erling Smørgrav 	c->tcp_do_close = 0;
4499b7579f77SDag-Erling Smørgrav 	c->do_not_close = 1;
4500b7579f77SDag-Erling Smørgrav 	c->tcp_do_toggle_rw = 0;
4501b7579f77SDag-Erling Smørgrav 	c->tcp_check_nb_connect = 0;
4502b5663de9SDag-Erling Smørgrav #ifdef USE_MSG_FASTOPEN
4503b5663de9SDag-Erling Smørgrav 	c->tcp_do_fastopen = 0;
4504b5663de9SDag-Erling Smørgrav #endif
450565b390aaSDag-Erling Smørgrav #ifdef USE_DNSCRYPT
450665b390aaSDag-Erling Smørgrav 	c->dnscrypt = 0;
450765b390aaSDag-Erling Smørgrav 	c->dnscrypt_buffer = c->buffer;
450865b390aaSDag-Erling Smørgrav #endif
4509b7579f77SDag-Erling Smørgrav 	c->callback = callback;
4510b7579f77SDag-Erling Smørgrav 	c->cb_arg = callback_arg;
4511865f46b2SCy Schubert 	c->pp2_enabled = 0;
4512865f46b2SCy Schubert 	c->pp2_header_state = pp2_header_none;
4513e2d15004SDag-Erling Smørgrav 	/* ub_event stuff */
4514b7579f77SDag-Erling Smørgrav 	if(writing)
4515e2d15004SDag-Erling Smørgrav 		evbits = UB_EV_PERSIST | UB_EV_WRITE;
4516e2d15004SDag-Erling Smørgrav 	else 	evbits = UB_EV_PERSIST | UB_EV_READ;
4517e2d15004SDag-Erling Smørgrav 	c->ev->ev = ub_event_new(base->eb->base, c->fd, evbits,
4518e2d15004SDag-Erling Smørgrav 		comm_point_raw_handle_callback, c);
4519e2d15004SDag-Erling Smørgrav 	if(c->ev->ev == NULL) {
4520e2d15004SDag-Erling Smørgrav 		log_err("could not baseset rawhdl event");
4521e2d15004SDag-Erling Smørgrav 		free(c->ev);
4522e2d15004SDag-Erling Smørgrav 		free(c);
4523e2d15004SDag-Erling Smørgrav 		return NULL;
4524e2d15004SDag-Erling Smørgrav 	}
4525e2d15004SDag-Erling Smørgrav 	if (ub_event_add(c->ev->ev, c->timeout) != 0) {
4526b7579f77SDag-Erling Smørgrav 		log_err("could not add rawhdl event");
4527e2d15004SDag-Erling Smørgrav 		ub_event_free(c->ev->ev);
4528b7579f77SDag-Erling Smørgrav 		free(c->ev);
4529b7579f77SDag-Erling Smørgrav 		free(c);
4530b7579f77SDag-Erling Smørgrav 		return NULL;
4531b7579f77SDag-Erling Smørgrav 	}
4532f44e67d1SCy Schubert 	c->event_added = 1;
4533b7579f77SDag-Erling Smørgrav 	return c;
4534b7579f77SDag-Erling Smørgrav }
4535b7579f77SDag-Erling Smørgrav 
4536b7579f77SDag-Erling Smørgrav void
4537b7579f77SDag-Erling Smørgrav comm_point_close(struct comm_point* c)
4538b7579f77SDag-Erling Smørgrav {
4539b7579f77SDag-Erling Smørgrav 	if(!c)
4540b7579f77SDag-Erling Smørgrav 		return;
45413bd4df0aSDag-Erling Smørgrav 	if(c->fd != -1) {
4542369c6923SCy Schubert 		verbose(5, "comm_point_close of %d: event_del", c->fd);
4543f44e67d1SCy Schubert 		if(c->event_added) {
4544e2d15004SDag-Erling Smørgrav 			if(ub_event_del(c->ev->ev) != 0) {
4545b7579f77SDag-Erling Smørgrav 				log_err("could not event_del on close");
4546b7579f77SDag-Erling Smørgrav 			}
4547f44e67d1SCy Schubert 			c->event_added = 0;
4548f44e67d1SCy Schubert 		}
45493bd4df0aSDag-Erling Smørgrav 	}
45504c75e3aaSDag-Erling Smørgrav 	tcl_close_connection(c->tcl_addr);
4551e86b9096SDag-Erling Smørgrav 	if(c->tcp_req_info)
4552e86b9096SDag-Erling Smørgrav 		tcp_req_info_clear(c->tcp_req_info);
4553c0caa2e2SCy Schubert 	if(c->h2_session)
4554c0caa2e2SCy Schubert 		http2_session_server_delete(c->h2_session);
45551838dec3SCy Schubert 	/* stop the comm point from reading or writing after it is closed. */
45561838dec3SCy Schubert 	if(c->tcp_more_read_again && *c->tcp_more_read_again)
45571838dec3SCy Schubert 		*c->tcp_more_read_again = 0;
45581838dec3SCy Schubert 	if(c->tcp_more_write_again && *c->tcp_more_write_again)
45591838dec3SCy Schubert 		*c->tcp_more_write_again = 0;
4560c0caa2e2SCy Schubert 
4561b7579f77SDag-Erling Smørgrav 	/* close fd after removing from event lists, or epoll.. is messed up */
4562b7579f77SDag-Erling Smørgrav 	if(c->fd != -1 && !c->do_not_close) {
45635469a995SCy Schubert #ifdef USE_WINSOCK
45643bd4df0aSDag-Erling Smørgrav 		if(c->type == comm_tcp || c->type == comm_http) {
45653bd4df0aSDag-Erling Smørgrav 			/* delete sticky events for the fd, it gets closed */
45663bd4df0aSDag-Erling Smørgrav 			ub_winsock_tcp_wouldblock(c->ev->ev, UB_EV_READ);
45673bd4df0aSDag-Erling Smørgrav 			ub_winsock_tcp_wouldblock(c->ev->ev, UB_EV_WRITE);
45683bd4df0aSDag-Erling Smørgrav 		}
45695469a995SCy Schubert #endif
4570b7579f77SDag-Erling Smørgrav 		verbose(VERB_ALGO, "close fd %d", c->fd);
4571c0caa2e2SCy Schubert 		sock_close(c->fd);
4572b7579f77SDag-Erling Smørgrav 	}
4573b7579f77SDag-Erling Smørgrav 	c->fd = -1;
4574b7579f77SDag-Erling Smørgrav }
4575b7579f77SDag-Erling Smørgrav 
4576b7579f77SDag-Erling Smørgrav void
4577b7579f77SDag-Erling Smørgrav comm_point_delete(struct comm_point* c)
4578b7579f77SDag-Erling Smørgrav {
4579b7579f77SDag-Erling Smørgrav 	if(!c)
4580b7579f77SDag-Erling Smørgrav 		return;
458157bddd21SDag-Erling Smørgrav 	if((c->type == comm_tcp || c->type == comm_http) && c->ssl) {
45828ed2b524SDag-Erling Smørgrav #ifdef HAVE_SSL
4583b7579f77SDag-Erling Smørgrav 		SSL_shutdown(c->ssl);
4584b7579f77SDag-Erling Smørgrav 		SSL_free(c->ssl);
45858ed2b524SDag-Erling Smørgrav #endif
4586b7579f77SDag-Erling Smørgrav 	}
4587c0caa2e2SCy Schubert 	if(c->type == comm_http && c->http_endpoint) {
4588c0caa2e2SCy Schubert 		free(c->http_endpoint);
4589c0caa2e2SCy Schubert 		c->http_endpoint = NULL;
4590c0caa2e2SCy Schubert 	}
4591b7579f77SDag-Erling Smørgrav 	comm_point_close(c);
4592b7579f77SDag-Erling Smørgrav 	if(c->tcp_handlers) {
4593b7579f77SDag-Erling Smørgrav 		int i;
4594b7579f77SDag-Erling Smørgrav 		for(i=0; i<c->max_tcp_count; i++)
4595b7579f77SDag-Erling Smørgrav 			comm_point_delete(c->tcp_handlers[i]);
4596b7579f77SDag-Erling Smørgrav 		free(c->tcp_handlers);
4597b7579f77SDag-Erling Smørgrav 	}
4598b7579f77SDag-Erling Smørgrav 	free(c->timeout);
459957bddd21SDag-Erling Smørgrav 	if(c->type == comm_tcp || c->type == comm_local || c->type == comm_http) {
460017d15b25SDag-Erling Smørgrav 		sldns_buffer_free(c->buffer);
460165b390aaSDag-Erling Smørgrav #ifdef USE_DNSCRYPT
460265b390aaSDag-Erling Smørgrav 		if(c->dnscrypt && c->dnscrypt_buffer != c->buffer) {
460365b390aaSDag-Erling Smørgrav 			sldns_buffer_free(c->dnscrypt_buffer);
460465b390aaSDag-Erling Smørgrav 		}
460565b390aaSDag-Erling Smørgrav #endif
4606e86b9096SDag-Erling Smørgrav 		if(c->tcp_req_info) {
4607e86b9096SDag-Erling Smørgrav 			tcp_req_info_delete(c->tcp_req_info);
4608e86b9096SDag-Erling Smørgrav 		}
4609c0caa2e2SCy Schubert 		if(c->h2_session) {
4610c0caa2e2SCy Schubert 			http2_session_delete(c->h2_session);
4611c0caa2e2SCy Schubert 		}
461265b390aaSDag-Erling Smørgrav 	}
4613e2d15004SDag-Erling Smørgrav 	ub_event_free(c->ev->ev);
4614b7579f77SDag-Erling Smørgrav 	free(c->ev);
4615b7579f77SDag-Erling Smørgrav 	free(c);
4616b7579f77SDag-Erling Smørgrav }
4617b7579f77SDag-Erling Smørgrav 
4618b7579f77SDag-Erling Smørgrav void
4619b7579f77SDag-Erling Smørgrav comm_point_send_reply(struct comm_reply *repinfo)
4620b7579f77SDag-Erling Smørgrav {
462165b390aaSDag-Erling Smørgrav 	struct sldns_buffer* buffer;
4622b7579f77SDag-Erling Smørgrav 	log_assert(repinfo && repinfo->c);
462365b390aaSDag-Erling Smørgrav #ifdef USE_DNSCRYPT
462465b390aaSDag-Erling Smørgrav 	buffer = repinfo->c->dnscrypt_buffer;
462565b390aaSDag-Erling Smørgrav 	if(!dnsc_handle_uncurved_request(repinfo)) {
462665b390aaSDag-Erling Smørgrav 		return;
462765b390aaSDag-Erling Smørgrav 	}
462865b390aaSDag-Erling Smørgrav #else
462965b390aaSDag-Erling Smørgrav 	buffer = repinfo->c->buffer;
463065b390aaSDag-Erling Smørgrav #endif
4631b7579f77SDag-Erling Smørgrav 	if(repinfo->c->type == comm_udp) {
4632b7579f77SDag-Erling Smørgrav 		if(repinfo->srctype)
4633865f46b2SCy Schubert 			comm_point_send_udp_msg_if(repinfo->c, buffer,
4634865f46b2SCy Schubert 				(struct sockaddr*)&repinfo->remote_addr,
4635865f46b2SCy Schubert 				repinfo->remote_addrlen, repinfo);
4636b7579f77SDag-Erling Smørgrav 		else
463765b390aaSDag-Erling Smørgrav 			comm_point_send_udp_msg(repinfo->c, buffer,
4638865f46b2SCy Schubert 				(struct sockaddr*)&repinfo->remote_addr,
4639865f46b2SCy Schubert 				repinfo->remote_addrlen, 0);
4640ff825849SDag-Erling Smørgrav #ifdef USE_DNSTAP
46415469a995SCy Schubert 		/*
46425469a995SCy Schubert 		 * sending src (client)/dst (local service) addresses over DNSTAP from udp callback
46435469a995SCy Schubert 		 */
46445469a995SCy Schubert 		if(repinfo->c->dtenv != NULL && repinfo->c->dtenv->log_client_response_messages) {
46455469a995SCy Schubert 			log_addr(VERB_ALGO, "from local addr", (void*)repinfo->c->socket->addr->ai_addr, repinfo->c->socket->addr->ai_addrlen);
4646865f46b2SCy Schubert 			log_addr(VERB_ALGO, "response to client", &repinfo->client_addr, repinfo->client_addrlen);
4647865f46b2SCy Schubert 			dt_msg_send_client_response(repinfo->c->dtenv, &repinfo->client_addr, (void*)repinfo->c->socket->addr->ai_addr, repinfo->c->type, repinfo->c->buffer);
46485469a995SCy Schubert 		}
4649ff825849SDag-Erling Smørgrav #endif
4650b7579f77SDag-Erling Smørgrav 	} else {
4651ff825849SDag-Erling Smørgrav #ifdef USE_DNSTAP
46525469a995SCy Schubert 		/*
46535469a995SCy Schubert 		 * sending src (client)/dst (local service) addresses over DNSTAP from TCP callback
46545469a995SCy Schubert 		 */
46555469a995SCy Schubert 		if(repinfo->c->tcp_parent->dtenv != NULL && repinfo->c->tcp_parent->dtenv->log_client_response_messages) {
46565469a995SCy Schubert 			log_addr(VERB_ALGO, "from local addr", (void*)repinfo->c->socket->addr->ai_addr, repinfo->c->socket->addr->ai_addrlen);
4657865f46b2SCy Schubert 			log_addr(VERB_ALGO, "response to client", &repinfo->client_addr, repinfo->client_addrlen);
4658865f46b2SCy Schubert 			dt_msg_send_client_response(repinfo->c->tcp_parent->dtenv, &repinfo->client_addr, (void*)repinfo->c->socket->addr->ai_addr, repinfo->c->type,
46595469a995SCy Schubert 				( repinfo->c->tcp_req_info? repinfo->c->tcp_req_info->spool_buffer: repinfo->c->buffer ));
46605469a995SCy Schubert 		}
4661ff825849SDag-Erling Smørgrav #endif
4662e86b9096SDag-Erling Smørgrav 		if(repinfo->c->tcp_req_info) {
4663e86b9096SDag-Erling Smørgrav 			tcp_req_info_send_reply(repinfo->c->tcp_req_info);
4664c0caa2e2SCy Schubert 		} else if(repinfo->c->use_h2) {
4665c0caa2e2SCy Schubert 			if(!http2_submit_dns_response(repinfo->c->h2_session)) {
4666c0caa2e2SCy Schubert 				comm_point_drop_reply(repinfo);
4667c0caa2e2SCy Schubert 				return;
4668c0caa2e2SCy Schubert 			}
4669c0caa2e2SCy Schubert 			repinfo->c->h2_stream = NULL;
4670c0caa2e2SCy Schubert 			repinfo->c->tcp_is_reading = 0;
4671c0caa2e2SCy Schubert 			comm_point_stop_listening(repinfo->c);
4672c0caa2e2SCy Schubert 			comm_point_start_listening(repinfo->c, -1,
4673f44e67d1SCy Schubert 				adjusted_tcp_timeout(repinfo->c));
4674c0caa2e2SCy Schubert 			return;
4675e86b9096SDag-Erling Smørgrav 		} else {
4676b5663de9SDag-Erling Smørgrav 			comm_point_start_listening(repinfo->c, -1,
4677f44e67d1SCy Schubert 				adjusted_tcp_timeout(repinfo->c));
4678b7579f77SDag-Erling Smørgrav 		}
4679b7579f77SDag-Erling Smørgrav 	}
4680e86b9096SDag-Erling Smørgrav }
4681b7579f77SDag-Erling Smørgrav 
4682b7579f77SDag-Erling Smørgrav void
4683b7579f77SDag-Erling Smørgrav comm_point_drop_reply(struct comm_reply* repinfo)
4684b7579f77SDag-Erling Smørgrav {
4685b7579f77SDag-Erling Smørgrav 	if(!repinfo)
4686b7579f77SDag-Erling Smørgrav 		return;
46870eefd307SCy Schubert 	log_assert(repinfo->c);
4688b7579f77SDag-Erling Smørgrav 	log_assert(repinfo->c->type != comm_tcp_accept);
4689b7579f77SDag-Erling Smørgrav 	if(repinfo->c->type == comm_udp)
4690b7579f77SDag-Erling Smørgrav 		return;
4691e86b9096SDag-Erling Smørgrav 	if(repinfo->c->tcp_req_info)
4692e86b9096SDag-Erling Smørgrav 		repinfo->c->tcp_req_info->is_drop = 1;
4693c0caa2e2SCy Schubert 	if(repinfo->c->type == comm_http) {
4694c0caa2e2SCy Schubert 		if(repinfo->c->h2_session) {
4695c0caa2e2SCy Schubert 			repinfo->c->h2_session->is_drop = 1;
4696c0caa2e2SCy Schubert 			if(!repinfo->c->h2_session->postpone_drop)
4697c0caa2e2SCy Schubert 				reclaim_http_handler(repinfo->c);
4698c0caa2e2SCy Schubert 			return;
4699c0caa2e2SCy Schubert 		}
4700c0caa2e2SCy Schubert 		reclaim_http_handler(repinfo->c);
4701c0caa2e2SCy Schubert 		return;
4702c0caa2e2SCy Schubert 	}
4703b7579f77SDag-Erling Smørgrav 	reclaim_tcp_handler(repinfo->c);
4704b7579f77SDag-Erling Smørgrav }
4705b7579f77SDag-Erling Smørgrav 
4706b7579f77SDag-Erling Smørgrav void
4707b7579f77SDag-Erling Smørgrav comm_point_stop_listening(struct comm_point* c)
4708b7579f77SDag-Erling Smørgrav {
4709b7579f77SDag-Erling Smørgrav 	verbose(VERB_ALGO, "comm point stop listening %d", c->fd);
4710f44e67d1SCy Schubert 	if(c->event_added) {
4711e2d15004SDag-Erling Smørgrav 		if(ub_event_del(c->ev->ev) != 0) {
4712b7579f77SDag-Erling Smørgrav 			log_err("event_del error to stoplisten");
4713b7579f77SDag-Erling Smørgrav 		}
4714f44e67d1SCy Schubert 		c->event_added = 0;
4715f44e67d1SCy Schubert 	}
4716b7579f77SDag-Erling Smørgrav }
4717b7579f77SDag-Erling Smørgrav 
4718b7579f77SDag-Erling Smørgrav void
4719b5663de9SDag-Erling Smørgrav comm_point_start_listening(struct comm_point* c, int newfd, int msec)
4720b7579f77SDag-Erling Smørgrav {
4721e86b9096SDag-Erling Smørgrav 	verbose(VERB_ALGO, "comm point start listening %d (%d msec)",
4722e86b9096SDag-Erling Smørgrav 		c->fd==-1?newfd:c->fd, msec);
4723b7579f77SDag-Erling Smørgrav 	if(c->type == comm_tcp_accept && !c->tcp_free) {
4724b7579f77SDag-Erling Smørgrav 		/* no use to start listening no free slots. */
4725b7579f77SDag-Erling Smørgrav 		return;
4726b7579f77SDag-Erling Smørgrav 	}
4727f44e67d1SCy Schubert 	if(c->event_added) {
4728f44e67d1SCy Schubert 		if(ub_event_del(c->ev->ev) != 0) {
4729f44e67d1SCy Schubert 			log_err("event_del error to startlisten");
4730f44e67d1SCy Schubert 		}
4731f44e67d1SCy Schubert 		c->event_added = 0;
4732f44e67d1SCy Schubert 	}
4733b5663de9SDag-Erling Smørgrav 	if(msec != -1 && msec != 0) {
4734b7579f77SDag-Erling Smørgrav 		if(!c->timeout) {
4735b7579f77SDag-Erling Smørgrav 			c->timeout = (struct timeval*)malloc(sizeof(
4736b7579f77SDag-Erling Smørgrav 				struct timeval));
4737b7579f77SDag-Erling Smørgrav 			if(!c->timeout) {
4738b7579f77SDag-Erling Smørgrav 				log_err("cpsl: malloc failed. No net read.");
4739b7579f77SDag-Erling Smørgrav 				return;
4740b7579f77SDag-Erling Smørgrav 			}
4741b7579f77SDag-Erling Smørgrav 		}
4742e2d15004SDag-Erling Smørgrav 		ub_event_add_bits(c->ev->ev, UB_EV_TIMEOUT);
4743b7579f77SDag-Erling Smørgrav #ifndef S_SPLINT_S /* splint fails on struct timeval. */
4744b5663de9SDag-Erling Smørgrav 		c->timeout->tv_sec = msec/1000;
4745b5663de9SDag-Erling Smørgrav 		c->timeout->tv_usec = (msec%1000)*1000;
4746b7579f77SDag-Erling Smørgrav #endif /* S_SPLINT_S */
474724e36522SCy Schubert 	} else {
474824e36522SCy Schubert 		if(msec == 0 || !c->timeout) {
474924e36522SCy Schubert 			ub_event_del_bits(c->ev->ev, UB_EV_TIMEOUT);
475024e36522SCy Schubert 		}
4751b7579f77SDag-Erling Smørgrav 	}
475257bddd21SDag-Erling Smørgrav 	if(c->type == comm_tcp || c->type == comm_http) {
4753e2d15004SDag-Erling Smørgrav 		ub_event_del_bits(c->ev->ev, UB_EV_READ|UB_EV_WRITE);
4754369c6923SCy Schubert 		if(c->tcp_write_and_read) {
4755369c6923SCy Schubert 			verbose(5, "startlistening %d mode rw", (newfd==-1?c->fd:newfd));
4756369c6923SCy Schubert 			ub_event_add_bits(c->ev->ev, UB_EV_READ|UB_EV_WRITE);
4757369c6923SCy Schubert 		} else if(c->tcp_is_reading) {
4758369c6923SCy Schubert 			verbose(5, "startlistening %d mode r", (newfd==-1?c->fd:newfd));
4759e2d15004SDag-Erling Smørgrav 			ub_event_add_bits(c->ev->ev, UB_EV_READ);
4760369c6923SCy Schubert 		} else	{
4761369c6923SCy Schubert 			verbose(5, "startlistening %d mode w", (newfd==-1?c->fd:newfd));
4762369c6923SCy Schubert 			ub_event_add_bits(c->ev->ev, UB_EV_WRITE);
4763369c6923SCy Schubert 		}
4764b7579f77SDag-Erling Smørgrav 	}
4765b7579f77SDag-Erling Smørgrav 	if(newfd != -1) {
4766369c6923SCy Schubert 		if(c->fd != -1 && c->fd != newfd) {
4767369c6923SCy Schubert 			verbose(5, "cpsl close of fd %d for %d", c->fd, newfd);
4768c0caa2e2SCy Schubert 			sock_close(c->fd);
4769b7579f77SDag-Erling Smørgrav 		}
4770b7579f77SDag-Erling Smørgrav 		c->fd = newfd;
4771e2d15004SDag-Erling Smørgrav 		ub_event_set_fd(c->ev->ev, c->fd);
4772b7579f77SDag-Erling Smørgrav 	}
4773b5663de9SDag-Erling Smørgrav 	if(ub_event_add(c->ev->ev, msec==0?NULL:c->timeout) != 0) {
4774b7579f77SDag-Erling Smørgrav 		log_err("event_add failed. in cpsl.");
477524e36522SCy Schubert 		return;
4776b7579f77SDag-Erling Smørgrav 	}
4777f44e67d1SCy Schubert 	c->event_added = 1;
4778b7579f77SDag-Erling Smørgrav }
4779b7579f77SDag-Erling Smørgrav 
4780b7579f77SDag-Erling Smørgrav void comm_point_listen_for_rw(struct comm_point* c, int rd, int wr)
4781b7579f77SDag-Erling Smørgrav {
4782b7579f77SDag-Erling Smørgrav 	verbose(VERB_ALGO, "comm point listen_for_rw %d %d", c->fd, wr);
4783f44e67d1SCy Schubert 	if(c->event_added) {
4784e2d15004SDag-Erling Smørgrav 		if(ub_event_del(c->ev->ev) != 0) {
4785b7579f77SDag-Erling Smørgrav 			log_err("event_del error to cplf");
4786b7579f77SDag-Erling Smørgrav 		}
4787f44e67d1SCy Schubert 		c->event_added = 0;
4788f44e67d1SCy Schubert 	}
478924e36522SCy Schubert 	if(!c->timeout) {
479024e36522SCy Schubert 		ub_event_del_bits(c->ev->ev, UB_EV_TIMEOUT);
479124e36522SCy Schubert 	}
4792e2d15004SDag-Erling Smørgrav 	ub_event_del_bits(c->ev->ev, UB_EV_READ|UB_EV_WRITE);
4793e2d15004SDag-Erling Smørgrav 	if(rd) ub_event_add_bits(c->ev->ev, UB_EV_READ);
4794e2d15004SDag-Erling Smørgrav 	if(wr) ub_event_add_bits(c->ev->ev, UB_EV_WRITE);
4795e2d15004SDag-Erling Smørgrav 	if(ub_event_add(c->ev->ev, c->timeout) != 0) {
4796b7579f77SDag-Erling Smørgrav 		log_err("event_add failed. in cplf.");
479724e36522SCy Schubert 		return;
4798b7579f77SDag-Erling Smørgrav 	}
4799f44e67d1SCy Schubert 	c->event_added = 1;
4800b7579f77SDag-Erling Smørgrav }
4801b7579f77SDag-Erling Smørgrav 
4802b7579f77SDag-Erling Smørgrav size_t comm_point_get_mem(struct comm_point* c)
4803b7579f77SDag-Erling Smørgrav {
4804b7579f77SDag-Erling Smørgrav 	size_t s;
4805b7579f77SDag-Erling Smørgrav 	if(!c)
4806b7579f77SDag-Erling Smørgrav 		return 0;
4807b7579f77SDag-Erling Smørgrav 	s = sizeof(*c) + sizeof(*c->ev);
4808b7579f77SDag-Erling Smørgrav 	if(c->timeout)
4809b7579f77SDag-Erling Smørgrav 		s += sizeof(*c->timeout);
481065b390aaSDag-Erling Smørgrav 	if(c->type == comm_tcp || c->type == comm_local) {
481117d15b25SDag-Erling Smørgrav 		s += sizeof(*c->buffer) + sldns_buffer_capacity(c->buffer);
481265b390aaSDag-Erling Smørgrav #ifdef USE_DNSCRYPT
481365b390aaSDag-Erling Smørgrav 		s += sizeof(*c->dnscrypt_buffer);
481465b390aaSDag-Erling Smørgrav 		if(c->buffer != c->dnscrypt_buffer) {
481565b390aaSDag-Erling Smørgrav 			s += sldns_buffer_capacity(c->dnscrypt_buffer);
481665b390aaSDag-Erling Smørgrav 		}
481765b390aaSDag-Erling Smørgrav #endif
481865b390aaSDag-Erling Smørgrav 	}
4819b7579f77SDag-Erling Smørgrav 	if(c->type == comm_tcp_accept) {
4820b7579f77SDag-Erling Smørgrav 		int i;
4821b7579f77SDag-Erling Smørgrav 		for(i=0; i<c->max_tcp_count; i++)
4822b7579f77SDag-Erling Smørgrav 			s += comm_point_get_mem(c->tcp_handlers[i]);
4823b7579f77SDag-Erling Smørgrav 	}
4824b7579f77SDag-Erling Smørgrav 	return s;
4825b7579f77SDag-Erling Smørgrav }
4826b7579f77SDag-Erling Smørgrav 
4827b7579f77SDag-Erling Smørgrav struct comm_timer*
4828b7579f77SDag-Erling Smørgrav comm_timer_create(struct comm_base* base, void (*cb)(void*), void* cb_arg)
4829b7579f77SDag-Erling Smørgrav {
4830e2d15004SDag-Erling Smørgrav 	struct internal_timer *tm = (struct internal_timer*)calloc(1,
4831b7579f77SDag-Erling Smørgrav 		sizeof(struct internal_timer));
4832e2d15004SDag-Erling Smørgrav 	if(!tm) {
4833b7579f77SDag-Erling Smørgrav 		log_err("malloc failed");
4834b7579f77SDag-Erling Smørgrav 		return NULL;
4835b7579f77SDag-Erling Smørgrav 	}
4836e2d15004SDag-Erling Smørgrav 	tm->super.ev_timer = tm;
4837e2d15004SDag-Erling Smørgrav 	tm->base = base;
4838e2d15004SDag-Erling Smørgrav 	tm->super.callback = cb;
4839e2d15004SDag-Erling Smørgrav 	tm->super.cb_arg = cb_arg;
4840e2d15004SDag-Erling Smørgrav 	tm->ev = ub_event_new(base->eb->base, -1, UB_EV_TIMEOUT,
4841e2d15004SDag-Erling Smørgrav 		comm_timer_callback, &tm->super);
4842e2d15004SDag-Erling Smørgrav 	if(tm->ev == NULL) {
4843b7579f77SDag-Erling Smørgrav 		log_err("timer_create: event_base_set failed.");
4844b7579f77SDag-Erling Smørgrav 		free(tm);
4845b7579f77SDag-Erling Smørgrav 		return NULL;
4846b7579f77SDag-Erling Smørgrav 	}
4847e2d15004SDag-Erling Smørgrav 	return &tm->super;
4848b7579f77SDag-Erling Smørgrav }
4849b7579f77SDag-Erling Smørgrav 
4850b7579f77SDag-Erling Smørgrav void
4851b7579f77SDag-Erling Smørgrav comm_timer_disable(struct comm_timer* timer)
4852b7579f77SDag-Erling Smørgrav {
4853b7579f77SDag-Erling Smørgrav 	if(!timer)
4854b7579f77SDag-Erling Smørgrav 		return;
4855e2d15004SDag-Erling Smørgrav 	ub_timer_del(timer->ev_timer->ev);
4856b7579f77SDag-Erling Smørgrav 	timer->ev_timer->enabled = 0;
4857b7579f77SDag-Erling Smørgrav }
4858b7579f77SDag-Erling Smørgrav 
4859b7579f77SDag-Erling Smørgrav void
4860b7579f77SDag-Erling Smørgrav comm_timer_set(struct comm_timer* timer, struct timeval* tv)
4861b7579f77SDag-Erling Smørgrav {
4862b7579f77SDag-Erling Smørgrav 	log_assert(tv);
4863b7579f77SDag-Erling Smørgrav 	if(timer->ev_timer->enabled)
4864b7579f77SDag-Erling Smørgrav 		comm_timer_disable(timer);
4865e2d15004SDag-Erling Smørgrav 	if(ub_timer_add(timer->ev_timer->ev, timer->ev_timer->base->eb->base,
4866e2d15004SDag-Erling Smørgrav 		comm_timer_callback, timer, tv) != 0)
4867b7579f77SDag-Erling Smørgrav 		log_err("comm_timer_set: evtimer_add failed.");
4868b7579f77SDag-Erling Smørgrav 	timer->ev_timer->enabled = 1;
4869b7579f77SDag-Erling Smørgrav }
4870b7579f77SDag-Erling Smørgrav 
4871b7579f77SDag-Erling Smørgrav void
4872b7579f77SDag-Erling Smørgrav comm_timer_delete(struct comm_timer* timer)
4873b7579f77SDag-Erling Smørgrav {
4874b7579f77SDag-Erling Smørgrav 	if(!timer)
4875b7579f77SDag-Erling Smørgrav 		return;
4876b7579f77SDag-Erling Smørgrav 	comm_timer_disable(timer);
4877e2d15004SDag-Erling Smørgrav 	/* Free the sub struct timer->ev_timer derived from the super struct timer.
4878e2d15004SDag-Erling Smørgrav 	 * i.e. assert(timer == timer->ev_timer)
4879e2d15004SDag-Erling Smørgrav 	 */
4880e2d15004SDag-Erling Smørgrav 	ub_event_free(timer->ev_timer->ev);
4881b7579f77SDag-Erling Smørgrav 	free(timer->ev_timer);
4882b7579f77SDag-Erling Smørgrav }
4883b7579f77SDag-Erling Smørgrav 
4884b7579f77SDag-Erling Smørgrav void
4885b7579f77SDag-Erling Smørgrav comm_timer_callback(int ATTR_UNUSED(fd), short event, void* arg)
4886b7579f77SDag-Erling Smørgrav {
4887b7579f77SDag-Erling Smørgrav 	struct comm_timer* tm = (struct comm_timer*)arg;
4888e2d15004SDag-Erling Smørgrav 	if(!(event&UB_EV_TIMEOUT))
4889b7579f77SDag-Erling Smørgrav 		return;
4890e2d15004SDag-Erling Smørgrav 	ub_comm_base_now(tm->ev_timer->base);
4891b7579f77SDag-Erling Smørgrav 	tm->ev_timer->enabled = 0;
4892b7579f77SDag-Erling Smørgrav 	fptr_ok(fptr_whitelist_comm_timer(tm->callback));
4893b7579f77SDag-Erling Smørgrav 	(*tm->callback)(tm->cb_arg);
4894b7579f77SDag-Erling Smørgrav }
4895b7579f77SDag-Erling Smørgrav 
4896b7579f77SDag-Erling Smørgrav int
4897b7579f77SDag-Erling Smørgrav comm_timer_is_set(struct comm_timer* timer)
4898b7579f77SDag-Erling Smørgrav {
4899b7579f77SDag-Erling Smørgrav 	return (int)timer->ev_timer->enabled;
4900b7579f77SDag-Erling Smørgrav }
4901b7579f77SDag-Erling Smørgrav 
4902b7579f77SDag-Erling Smørgrav size_t
4903e2d15004SDag-Erling Smørgrav comm_timer_get_mem(struct comm_timer* ATTR_UNUSED(timer))
4904b7579f77SDag-Erling Smørgrav {
4905e2d15004SDag-Erling Smørgrav 	return sizeof(struct internal_timer);
4906b7579f77SDag-Erling Smørgrav }
4907b7579f77SDag-Erling Smørgrav 
4908b7579f77SDag-Erling Smørgrav struct comm_signal*
4909b7579f77SDag-Erling Smørgrav comm_signal_create(struct comm_base* base,
4910b7579f77SDag-Erling Smørgrav         void (*callback)(int, void*), void* cb_arg)
4911b7579f77SDag-Erling Smørgrav {
4912b7579f77SDag-Erling Smørgrav 	struct comm_signal* com = (struct comm_signal*)malloc(
4913b7579f77SDag-Erling Smørgrav 		sizeof(struct comm_signal));
4914b7579f77SDag-Erling Smørgrav 	if(!com) {
4915b7579f77SDag-Erling Smørgrav 		log_err("malloc failed");
4916b7579f77SDag-Erling Smørgrav 		return NULL;
4917b7579f77SDag-Erling Smørgrav 	}
4918b7579f77SDag-Erling Smørgrav 	com->base = base;
4919b7579f77SDag-Erling Smørgrav 	com->callback = callback;
4920b7579f77SDag-Erling Smørgrav 	com->cb_arg = cb_arg;
4921b7579f77SDag-Erling Smørgrav 	com->ev_signal = NULL;
4922b7579f77SDag-Erling Smørgrav 	return com;
4923b7579f77SDag-Erling Smørgrav }
4924b7579f77SDag-Erling Smørgrav 
4925b7579f77SDag-Erling Smørgrav void
4926b7579f77SDag-Erling Smørgrav comm_signal_callback(int sig, short event, void* arg)
4927b7579f77SDag-Erling Smørgrav {
4928b7579f77SDag-Erling Smørgrav 	struct comm_signal* comsig = (struct comm_signal*)arg;
4929e2d15004SDag-Erling Smørgrav 	if(!(event & UB_EV_SIGNAL))
4930b7579f77SDag-Erling Smørgrav 		return;
4931e2d15004SDag-Erling Smørgrav 	ub_comm_base_now(comsig->base);
4932b7579f77SDag-Erling Smørgrav 	fptr_ok(fptr_whitelist_comm_signal(comsig->callback));
4933b7579f77SDag-Erling Smørgrav 	(*comsig->callback)(sig, comsig->cb_arg);
4934b7579f77SDag-Erling Smørgrav }
4935b7579f77SDag-Erling Smørgrav 
4936b7579f77SDag-Erling Smørgrav int
4937b7579f77SDag-Erling Smørgrav comm_signal_bind(struct comm_signal* comsig, int sig)
4938b7579f77SDag-Erling Smørgrav {
4939b7579f77SDag-Erling Smørgrav 	struct internal_signal* entry = (struct internal_signal*)calloc(1,
4940b7579f77SDag-Erling Smørgrav 		sizeof(struct internal_signal));
4941b7579f77SDag-Erling Smørgrav 	if(!entry) {
4942b7579f77SDag-Erling Smørgrav 		log_err("malloc failed");
4943b7579f77SDag-Erling Smørgrav 		return 0;
4944b7579f77SDag-Erling Smørgrav 	}
4945b7579f77SDag-Erling Smørgrav 	log_assert(comsig);
4946b7579f77SDag-Erling Smørgrav 	/* add signal event */
4947e2d15004SDag-Erling Smørgrav 	entry->ev = ub_signal_new(comsig->base->eb->base, sig,
4948e2d15004SDag-Erling Smørgrav 		comm_signal_callback, comsig);
4949e2d15004SDag-Erling Smørgrav 	if(entry->ev == NULL) {
4950e2d15004SDag-Erling Smørgrav 		log_err("Could not create signal event");
4951b7579f77SDag-Erling Smørgrav 		free(entry);
4952b7579f77SDag-Erling Smørgrav 		return 0;
4953b7579f77SDag-Erling Smørgrav 	}
4954e2d15004SDag-Erling Smørgrav 	if(ub_signal_add(entry->ev, NULL) != 0) {
4955b7579f77SDag-Erling Smørgrav 		log_err("Could not add signal handler");
4956e2d15004SDag-Erling Smørgrav 		ub_event_free(entry->ev);
4957b7579f77SDag-Erling Smørgrav 		free(entry);
4958b7579f77SDag-Erling Smørgrav 		return 0;
4959b7579f77SDag-Erling Smørgrav 	}
4960b7579f77SDag-Erling Smørgrav 	/* link into list */
4961b7579f77SDag-Erling Smørgrav 	entry->next = comsig->ev_signal;
4962b7579f77SDag-Erling Smørgrav 	comsig->ev_signal = entry;
4963b7579f77SDag-Erling Smørgrav 	return 1;
4964b7579f77SDag-Erling Smørgrav }
4965b7579f77SDag-Erling Smørgrav 
4966b7579f77SDag-Erling Smørgrav void
4967b7579f77SDag-Erling Smørgrav comm_signal_delete(struct comm_signal* comsig)
4968b7579f77SDag-Erling Smørgrav {
4969b7579f77SDag-Erling Smørgrav 	struct internal_signal* p, *np;
4970b7579f77SDag-Erling Smørgrav 	if(!comsig)
4971b7579f77SDag-Erling Smørgrav 		return;
4972b7579f77SDag-Erling Smørgrav 	p=comsig->ev_signal;
4973b7579f77SDag-Erling Smørgrav 	while(p) {
4974b7579f77SDag-Erling Smørgrav 		np = p->next;
4975e2d15004SDag-Erling Smørgrav 		ub_signal_del(p->ev);
4976e2d15004SDag-Erling Smørgrav 		ub_event_free(p->ev);
4977b7579f77SDag-Erling Smørgrav 		free(p);
4978b7579f77SDag-Erling Smørgrav 		p = np;
4979b7579f77SDag-Erling Smørgrav 	}
4980b7579f77SDag-Erling Smørgrav 	free(comsig);
4981b7579f77SDag-Erling Smørgrav }
4982