]> git.ipfire.org Git - people/ms/dnsmasq.git/blame - src/forward.c
import of dnsmasq-2.59.tar.gz
[people/ms/dnsmasq.git] / src / forward.c
CommitLineData
28866e95 1/* dnsmasq is Copyright (c) 2000-2011 Simon Kelley
9e4abcb5
SK
2
3 This program is free software; you can redistribute it and/or modify
4 it under the terms of the GNU General Public License as published by
824af85b
SK
5 the Free Software Foundation; version 2 dated June, 1991, or
6 (at your option) version 3 dated 29 June, 2007.
7
9e4abcb5
SK
8 This program is distributed in the hope that it will be useful,
9 but WITHOUT ANY WARRANTY; without even the implied warranty of
10 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 GNU General Public License for more details.
824af85b 12
73a08a24
SK
13 You should have received a copy of the GNU General Public License
14 along with this program. If not, see <http://www.gnu.org/licenses/>.
9e4abcb5
SK
15*/
16
9e4abcb5
SK
17#include "dnsmasq.h"
18
832af0ba 19static struct frec *lookup_frec(unsigned short id, unsigned int crc);
9e4abcb5 20static struct frec *lookup_frec_by_sender(unsigned short id,
fd9fa481
SK
21 union mysockaddr *addr,
22 unsigned int crc);
316e2730 23static unsigned short get_id(unsigned int crc);
1a6bca81
SK
24static void free_frec(struct frec *f);
25static struct randfd *allocate_rfd(int family);
9e4abcb5 26
824af85b 27/* Send a UDP packet with its source address set as "source"
44a2a316 28 unless nowild is true, when we just send it with the kernel default */
cdeda28f 29static void send_from(int fd, int nowild, char *packet, size_t len,
dfa666f2
SK
30 union mysockaddr *to, struct all_addr *source,
31 unsigned int iface)
9e4abcb5 32{
44a2a316
SK
33 struct msghdr msg;
34 struct iovec iov[1];
44a2a316
SK
35 union {
36 struct cmsghdr align; /* this ensures alignment */
5e9e0efb 37#if defined(HAVE_LINUX_NETWORK)
44a2a316
SK
38 char control[CMSG_SPACE(sizeof(struct in_pktinfo))];
39#elif defined(IP_SENDSRCADDR)
40 char control[CMSG_SPACE(sizeof(struct in_addr))];
41#endif
42#ifdef HAVE_IPV6
43 char control6[CMSG_SPACE(sizeof(struct in6_pktinfo))];
44#endif
45 } control_u;
feba5c1d 46
44a2a316
SK
47 iov[0].iov_base = packet;
48 iov[0].iov_len = len;
49
feba5c1d
SK
50 msg.msg_control = NULL;
51 msg.msg_controllen = 0;
44a2a316
SK
52 msg.msg_flags = 0;
53 msg.msg_name = to;
54 msg.msg_namelen = sa_len(to);
55 msg.msg_iov = iov;
56 msg.msg_iovlen = 1;
feba5c1d 57
26128d27 58 if (!nowild)
44a2a316 59 {
26128d27 60 struct cmsghdr *cmptr;
feba5c1d
SK
61 msg.msg_control = &control_u;
62 msg.msg_controllen = sizeof(control_u);
26128d27
SK
63 cmptr = CMSG_FIRSTHDR(&msg);
64
65 if (to->sa.sa_family == AF_INET)
66 {
5e9e0efb 67#if defined(HAVE_LINUX_NETWORK)
8ef5ada2
SK
68 struct in_pktinfo p;
69 p.ipi_ifindex = 0;
70 p.ipi_spec_dst = source->addr.addr4;
71 memcpy(CMSG_DATA(cmptr), &p, sizeof(p));
26128d27
SK
72 msg.msg_controllen = cmptr->cmsg_len = CMSG_LEN(sizeof(struct in_pktinfo));
73 cmptr->cmsg_level = SOL_IP;
74 cmptr->cmsg_type = IP_PKTINFO;
44a2a316 75#elif defined(IP_SENDSRCADDR)
8ef5ada2 76 memcpy(CMSG_DATA(cmptr), &(source->addr.addr4), sizeof(source->addr.addr4));
26128d27
SK
77 msg.msg_controllen = cmptr->cmsg_len = CMSG_LEN(sizeof(struct in_addr));
78 cmptr->cmsg_level = IPPROTO_IP;
79 cmptr->cmsg_type = IP_SENDSRCADDR;
44a2a316 80#endif
26128d27 81 }
26128d27 82 else
b8187c80 83#ifdef HAVE_IPV6
26128d27 84 {
8ef5ada2
SK
85 struct in6_pktinfo p;
86 p.ipi6_ifindex = iface; /* Need iface for IPv6 to handle link-local addrs */
87 p.ipi6_addr = source->addr.addr6;
88 memcpy(CMSG_DATA(cmptr), &p, sizeof(p));
26128d27 89 msg.msg_controllen = cmptr->cmsg_len = CMSG_LEN(sizeof(struct in6_pktinfo));
316e2730 90 cmptr->cmsg_type = daemon->v6pktinfo;
26128d27
SK
91 cmptr->cmsg_level = IPV6_LEVEL;
92 }
3d8df260
SK
93#else
94 iface = 0; /* eliminate warning */
44a2a316 95#endif
26128d27 96 }
feba5c1d 97
fd9fa481
SK
98 retry:
99 if (sendmsg(fd, &msg, 0) == -1)
feba5c1d 100 {
fd9fa481
SK
101 /* certain Linux kernels seem to object to setting the source address in the IPv6 stack
102 by returning EINVAL from sendmsg. In that case, try again without setting the
103 source address, since it will nearly alway be correct anyway. IPv6 stinks. */
104 if (errno == EINVAL && msg.msg_controllen)
105 {
106 msg.msg_controllen = 0;
107 goto retry;
108 }
109 if (retry_send())
110 goto retry;
feba5c1d 111 }
9e4abcb5 112}
44a2a316 113
28866e95
SK
114static unsigned int search_servers(time_t now, struct all_addr **addrpp,
115 unsigned int qtype, char *qdomain, int *type, char **domain, int *norebind)
feba5c1d
SK
116
117{
118 /* If the query ends in the domain in one of our servers, set
119 domain to point to that name. We find the largest match to allow both
120 domain.org and sub.domain.org to exist. */
121
122 unsigned int namelen = strlen(qdomain);
123 unsigned int matchlen = 0;
124 struct server *serv;
28866e95 125 unsigned int flags = 0;
feba5c1d 126
3be34541 127 for (serv = daemon->servers; serv; serv=serv->next)
feba5c1d 128 /* domain matches take priority over NODOTS matches */
3d8df260 129 if ((serv->flags & SERV_FOR_NODOTS) && *type != SERV_HAS_DOMAIN && !strchr(qdomain, '.') && namelen != 0)
feba5c1d 130 {
28866e95 131 unsigned int sflag = serv->addr.sa.sa_family == AF_INET ? F_IPV4 : F_IPV6;
feba5c1d 132 *type = SERV_FOR_NODOTS;
feba5c1d 133 if (serv->flags & SERV_NO_ADDR)
36717eee
SK
134 flags = F_NXDOMAIN;
135 else if (serv->flags & SERV_LITERAL_ADDRESS)
136 {
137 if (sflag & qtype)
138 {
139 flags = sflag;
140 if (serv->addr.sa.sa_family == AF_INET)
141 *addrpp = (struct all_addr *)&serv->addr.in.sin_addr;
feba5c1d 142#ifdef HAVE_IPV6
36717eee
SK
143 else
144 *addrpp = (struct all_addr *)&serv->addr.in6.sin6_addr;
feba5c1d 145#endif
36717eee 146 }
824af85b 147 else if (!flags || (flags & F_NXDOMAIN))
36717eee
SK
148 flags = F_NOERR;
149 }
feba5c1d
SK
150 }
151 else if (serv->flags & SERV_HAS_DOMAIN)
152 {
153 unsigned int domainlen = strlen(serv->domain);
b8187c80 154 char *matchstart = qdomain + namelen - domainlen;
feba5c1d 155 if (namelen >= domainlen &&
b8187c80 156 hostname_isequal(matchstart, serv->domain) &&
8ef5ada2 157 (domainlen == 0 || namelen == domainlen || *(matchstart-1) == '.' ))
feba5c1d 158 {
8ef5ada2
SK
159 if (serv->flags & SERV_NO_REBIND)
160 *norebind = 1;
28866e95 161 else
feba5c1d 162 {
28866e95
SK
163 unsigned int sflag = serv->addr.sa.sa_family == AF_INET ? F_IPV4 : F_IPV6;
164 /* implement priority rules for --address and --server for same domain.
165 --address wins if the address is for the correct AF
166 --server wins otherwise. */
167 if (domainlen != 0 && domainlen == matchlen)
36717eee 168 {
28866e95 169 if ((serv->flags & SERV_LITERAL_ADDRESS))
8ef5ada2 170 {
28866e95
SK
171 if (!(sflag & qtype) && flags == 0)
172 continue;
173 }
174 else
175 {
176 if (flags & (F_IPV4 | F_IPV6))
177 continue;
178 }
179 }
180
181 if (domainlen >= matchlen)
182 {
183 *type = serv->flags & (SERV_HAS_DOMAIN | SERV_USE_RESOLV | SERV_NO_REBIND);
184 *domain = serv->domain;
185 matchlen = domainlen;
186 if (serv->flags & SERV_NO_ADDR)
187 flags = F_NXDOMAIN;
188 else if (serv->flags & SERV_LITERAL_ADDRESS)
189 {
190 if (sflag & qtype)
191 {
192 flags = sflag;
193 if (serv->addr.sa.sa_family == AF_INET)
194 *addrpp = (struct all_addr *)&serv->addr.in.sin_addr;
feba5c1d 195#ifdef HAVE_IPV6
28866e95
SK
196 else
197 *addrpp = (struct all_addr *)&serv->addr.in6.sin6_addr;
feba5c1d 198#endif
28866e95
SK
199 }
200 else if (!flags || (flags & F_NXDOMAIN))
201 flags = F_NOERR;
8ef5ada2 202 }
28866e95
SK
203 else
204 flags = 0;
205 }
206 }
8ef5ada2 207 }
feba5c1d 208 }
8ef5ada2 209
7de060b0 210 if (flags == 0 && !(qtype & F_QUERY) &&
28866e95 211 option_bool(OPT_NODOTS_LOCAL) && !strchr(qdomain, '.') && namelen != 0)
7de060b0
SK
212 /* don't forward A or AAAA queries for simple names, except the empty name */
213 flags = F_NOERR;
8ef5ada2 214
5aabfc78 215 if (flags == F_NXDOMAIN && check_for_local_domain(qdomain, now))
c1bb8504 216 flags = F_NOERR;
feba5c1d 217
824af85b
SK
218 if (flags)
219 {
220 int logflags = 0;
221
222 if (flags == F_NXDOMAIN || flags == F_NOERR)
223 logflags = F_NEG | qtype;
224
1a6bca81 225 log_query(logflags | flags | F_CONFIG | F_FORWARD, qdomain, *addrpp, NULL);
824af85b 226 }
8ef5ada2
SK
227 else if ((*type) & SERV_USE_RESOLV)
228 {
229 *type = 0; /* use normal servers for this domain */
230 *domain = NULL;
231 }
feba5c1d
SK
232 return flags;
233}
44a2a316 234
824af85b
SK
235static int forward_query(int udpfd, union mysockaddr *udpaddr,
236 struct all_addr *dst_addr, unsigned int dst_iface,
572b41eb 237 struct dns_header *header, size_t plen, time_t now, struct frec *forward)
9e4abcb5 238{
9e4abcb5 239 char *domain = NULL;
8ef5ada2 240 int type = 0, norebind = 0;
9e4abcb5 241 struct all_addr *addrp = NULL;
cdeda28f 242 unsigned int crc = questions_crc(header, plen, daemon->namebuff);
28866e95
SK
243 unsigned int flags = 0;
244 unsigned int gotname = extract_request(header, plen, daemon->namebuff, NULL);
de37951c 245 struct server *start = NULL;
7de060b0 246
28866e95 247 /* RFC 4035: sect 4.6 para 2 */
572b41eb
SK
248 header->hb4 &= ~HB4_AD;
249
3d8df260
SK
250 /* may be no servers available. */
251 if (!daemon->servers)
9e4abcb5 252 forward = NULL;
b8187c80 253 else if (forward || (forward = lookup_frec_by_sender(ntohs(header->id), udpaddr, crc)))
9e4abcb5 254 {
de37951c 255 /* retry on existing query, send to all available servers */
9e4abcb5 256 domain = forward->sentto->domain;
824af85b 257 forward->sentto->failed_queries++;
28866e95 258 if (!option_bool(OPT_ORDER))
de37951c 259 {
0a852541 260 forward->forwardall = 1;
3be34541 261 daemon->last_server = NULL;
de37951c 262 }
9e4abcb5 263 type = forward->sentto->flags & SERV_TYPE;
de37951c 264 if (!(start = forward->sentto->next))
3be34541 265 start = daemon->servers; /* at end of list, recycle */
9e4abcb5
SK
266 header->id = htons(forward->new_id);
267 }
268 else
269 {
270 if (gotname)
8ef5ada2 271 flags = search_servers(now, &addrp, gotname, daemon->namebuff, &type, &domain, &norebind);
9e4abcb5 272
5aabfc78 273 if (!flags && !(forward = get_new_frec(now, NULL)))
feba5c1d
SK
274 /* table full - server failure. */
275 flags = F_NEG;
9e4abcb5
SK
276
277 if (forward)
278 {
0a852541
SK
279 forward->source = *udpaddr;
280 forward->dest = *dst_addr;
281 forward->iface = dst_iface;
0a852541 282 forward->orig_id = ntohs(header->id);
316e2730 283 forward->new_id = get_id(crc);
832af0ba 284 forward->fd = udpfd;
0a852541
SK
285 forward->crc = crc;
286 forward->forwardall = 0;
28866e95
SK
287 if (norebind)
288 forward->flags |= FREC_NOREBIND;
572b41eb 289 if (header->hb4 & HB4_CD)
28866e95 290 forward->flags |= FREC_CHECKING_DISABLED;
0a852541 291
28866e95
SK
292 header->id = htons(forward->new_id);
293
8ef5ada2
SK
294 /* In strict_order mode, always try servers in the order
295 specified in resolv.conf, if a domain is given
296 always try all the available servers,
9e4abcb5
SK
297 otherwise, use the one last known to work. */
298
8ef5ada2
SK
299 if (type == 0)
300 {
28866e95 301 if (option_bool(OPT_ORDER))
8ef5ada2
SK
302 start = daemon->servers;
303 else if (!(start = daemon->last_server) ||
304 daemon->forwardcount++ > FORWARD_TEST ||
305 difftime(now, daemon->forwardtime) > FORWARD_TIME)
306 {
307 start = daemon->servers;
308 forward->forwardall = 1;
309 daemon->forwardcount = 0;
310 daemon->forwardtime = now;
311 }
312 }
313 else
de37951c 314 {
3be34541 315 start = daemon->servers;
28866e95 316 if (!option_bool(OPT_ORDER))
8ef5ada2 317 forward->forwardall = 1;
de37951c 318 }
9e4abcb5
SK
319 }
320 }
feba5c1d 321
9e4abcb5
SK
322 /* check for send errors here (no route to host)
323 if we fail to send to all nameservers, send back an error
324 packet straight away (helps modem users when offline) */
325
326 if (!flags && forward)
327 {
de37951c
SK
328 struct server *firstsentto = start;
329 int forwarded = 0;
28866e95
SK
330
331 if (udpaddr && option_bool(OPT_ADD_MAC))
332 plen = add_mac(header, plen, ((char *) header) + PACKETSZ, udpaddr);
333
9e4abcb5
SK
334 while (1)
335 {
9e4abcb5
SK
336 /* only send to servers dealing with our domain.
337 domain may be NULL, in which case server->domain
338 must be NULL also. */
339
de37951c 340 if (type == (start->flags & SERV_TYPE) &&
fd9fa481
SK
341 (type != SERV_HAS_DOMAIN || hostname_isequal(domain, start->domain)) &&
342 !(start->flags & SERV_LITERAL_ADDRESS))
9e4abcb5 343 {
1a6bca81
SK
344 int fd;
345
346 /* find server socket to use, may need to get random one. */
347 if (start->sfd)
348 fd = start->sfd->fd;
349 else
350 {
351#ifdef HAVE_IPV6
352 if (start->addr.sa.sa_family == AF_INET6)
353 {
354 if (!forward->rfd6 &&
355 !(forward->rfd6 = allocate_rfd(AF_INET6)))
356 break;
3927da46 357 daemon->rfd_save = forward->rfd6;
1a6bca81
SK
358 fd = forward->rfd6->fd;
359 }
360 else
361#endif
362 {
363 if (!forward->rfd4 &&
364 !(forward->rfd4 = allocate_rfd(AF_INET)))
365 break;
3927da46 366 daemon->rfd_save = forward->rfd4;
1a6bca81
SK
367 fd = forward->rfd4->fd;
368 }
7de060b0
SK
369
370#ifdef HAVE_CONNTRACK
371 /* Copy connection mark of incoming query to outgoing connection. */
372 if (option_bool(OPT_CONNTRACK))
373 {
374 unsigned int mark;
375 if (get_incoming_mark(udpaddr, dst_addr, 0, &mark))
376 setsockopt(fd, SOL_SOCKET, SO_MARK, &mark, sizeof(unsigned int));
377 }
378#endif
1a6bca81
SK
379 }
380
381 if (sendto(fd, (char *)header, plen, 0,
feba5c1d 382 &start->addr.sa,
fd9fa481
SK
383 sa_len(&start->addr)) == -1)
384 {
385 if (retry_send())
386 continue;
387 }
388 else
9e4abcb5 389 {
cdeda28f
SK
390 /* Keep info in case we want to re-send this packet */
391 daemon->srv_save = start;
392 daemon->packet_len = plen;
393
de37951c 394 if (!gotname)
3be34541 395 strcpy(daemon->namebuff, "query");
de37951c 396 if (start->addr.sa.sa_family == AF_INET)
3be34541 397 log_query(F_SERVER | F_IPV4 | F_FORWARD, daemon->namebuff,
1a6bca81 398 (struct all_addr *)&start->addr.in.sin_addr, NULL);
de37951c
SK
399#ifdef HAVE_IPV6
400 else
3be34541 401 log_query(F_SERVER | F_IPV6 | F_FORWARD, daemon->namebuff,
1a6bca81 402 (struct all_addr *)&start->addr.in6.sin6_addr, NULL);
de37951c 403#endif
824af85b 404 start->queries++;
de37951c
SK
405 forwarded = 1;
406 forward->sentto = start;
0a852541 407 if (!forward->forwardall)
de37951c 408 break;
0a852541 409 forward->forwardall++;
9e4abcb5
SK
410 }
411 }
412
de37951c 413 if (!(start = start->next))
3be34541 414 start = daemon->servers;
9e4abcb5 415
de37951c 416 if (start == firstsentto)
9e4abcb5
SK
417 break;
418 }
419
de37951c 420 if (forwarded)
824af85b 421 return 1;
de37951c 422
9e4abcb5
SK
423 /* could not send on, prepare to return */
424 header->id = htons(forward->orig_id);
1a6bca81 425 free_frec(forward); /* cancel */
9e4abcb5
SK
426 }
427
428 /* could not send on, return empty answer or address if known for whole domain */
b8187c80
SK
429 if (udpfd != -1)
430 {
cdeda28f 431 plen = setup_reply(header, plen, addrp, flags, daemon->local_ttl);
28866e95 432 send_from(udpfd, option_bool(OPT_NOWILD), (char *)header, plen, udpaddr, dst_addr, dst_iface);
b8187c80
SK
433 }
434
824af85b 435 return 0;
9e4abcb5
SK
436}
437
572b41eb 438static size_t process_reply(struct dns_header *header, time_t now,
28866e95 439 struct server *server, size_t n, int check_rebind, int checking_disabled)
feba5c1d 440{
36717eee 441 unsigned char *pheader, *sizep;
832af0ba 442 int munged = 0, is_sign;
cdeda28f
SK
443 size_t plen;
444
feba5c1d 445 /* If upstream is advertising a larger UDP packet size
9009d746
SK
446 than we allow, trim it so that we don't get overlarge
447 requests for the client. We can't do this for signed packets. */
feba5c1d 448
832af0ba 449 if ((pheader = find_pseudoheader(header, n, &plen, &sizep, &is_sign)) && !is_sign)
feba5c1d
SK
450 {
451 unsigned short udpsz;
36717eee 452 unsigned char *psave = sizep;
feba5c1d 453
36717eee 454 GETSHORT(udpsz, sizep);
3be34541
SK
455 if (udpsz > daemon->edns_pktsz)
456 PUTSHORT(daemon->edns_pktsz, psave);
feba5c1d
SK
457 }
458
28866e95
SK
459 /* RFC 4035 sect 4.6 para 3 */
460 if (!is_sign && !option_bool(OPT_DNSSEC))
572b41eb 461 header->hb4 &= ~HB4_AD;
28866e95 462
572b41eb 463 if (OPCODE(header) != QUERY || (RCODE(header) != NOERROR && RCODE(header) != NXDOMAIN))
0a852541
SK
464 return n;
465
feba5c1d 466 /* Complain loudly if the upstream server is non-recursive. */
572b41eb 467 if (!(header->hb4 & HB4_RA) && RCODE(header) == NOERROR && ntohs(header->ancount) == 0 &&
0a852541 468 server && !(server->flags & SERV_WARNED_RECURSIVE))
feba5c1d 469 {
3d8df260 470 prettyprint_addr(&server->addr, daemon->namebuff);
f2621c7f 471 my_syslog(LOG_WARNING, _("nameserver %s refused to do a recursive query"), daemon->namebuff);
28866e95 472 if (!option_bool(OPT_LOG))
0a852541
SK
473 server->flags |= SERV_WARNED_RECURSIVE;
474 }
475
572b41eb 476 if (daemon->bogus_addr && RCODE(header) != NXDOMAIN &&
fd9fa481 477 check_for_bogus_wildcard(header, n, daemon->namebuff, daemon->bogus_addr, now))
feba5c1d 478 {
fd9fa481 479 munged = 1;
572b41eb
SK
480 SET_RCODE(header, NXDOMAIN);
481 header->hb3 &= ~HB3_AA;
36717eee 482 }
fd9fa481 483 else
36717eee 484 {
572b41eb 485 if (RCODE(header) == NXDOMAIN &&
fd9fa481 486 extract_request(header, n, daemon->namebuff, NULL) &&
5aabfc78 487 check_for_local_domain(daemon->namebuff, now))
36717eee
SK
488 {
489 /* if we forwarded a query for a locally known name (because it was for
490 an unknown type) and the answer is NXDOMAIN, convert that to NODATA,
491 since we know that the domain exists, even if upstream doesn't */
fd9fa481 492 munged = 1;
572b41eb
SK
493 header->hb3 |= HB3_AA;
494 SET_RCODE(header, NOERROR);
feba5c1d 495 }
832af0ba 496
28866e95 497 if (extract_addresses(header, n, daemon->namebuff, now, is_sign, check_rebind, checking_disabled))
824af85b 498 {
8ef5ada2 499 my_syslog(LOG_WARNING, _("possible DNS-rebind attack detected: %s"), daemon->namebuff);
824af85b
SK
500 munged = 1;
501 }
feba5c1d 502 }
fd9fa481
SK
503
504 /* do this after extract_addresses. Ensure NODATA reply and remove
505 nameserver info. */
506
507 if (munged)
508 {
509 header->ancount = htons(0);
510 header->nscount = htons(0);
511 header->arcount = htons(0);
512 }
513
36717eee
SK
514 /* the bogus-nxdomain stuff, doctor and NXDOMAIN->NODATA munging can all elide
515 sections of the packet. Find the new length here and put back pseudoheader
516 if it was removed. */
517 return resize_packet(header, n, pheader, plen);
feba5c1d
SK
518}
519
3be34541 520/* sets new last_server */
1a6bca81 521void reply_query(int fd, int family, time_t now)
9e4abcb5
SK
522{
523 /* packet from peer server, extract data for cache, and send to
524 original requester */
572b41eb 525 struct dns_header *header;
de37951c 526 union mysockaddr serveraddr;
832af0ba 527 struct frec *forward;
de37951c 528 socklen_t addrlen = sizeof(serveraddr);
1a6bca81 529 ssize_t n = recvfrom(fd, daemon->packet, daemon->edns_pktsz, 0, &serveraddr.sa, &addrlen);
cdeda28f 530 size_t nn;
1a6bca81
SK
531 struct server *server;
532
cdeda28f
SK
533 /* packet buffer overwritten */
534 daemon->srv_save = NULL;
832af0ba 535
de37951c 536 /* Determine the address of the server replying so that we can mark that as good */
1a6bca81 537 serveraddr.sa.sa_family = family;
de37951c
SK
538#ifdef HAVE_IPV6
539 if (serveraddr.sa.sa_family == AF_INET6)
5e9e0efb 540 serveraddr.in6.sin6_flowinfo = 0;
de37951c 541#endif
9e4abcb5 542
1a6bca81
SK
543 /* spoof check: answer must come from known server, */
544 for (server = daemon->servers; server; server = server->next)
545 if (!(server->flags & (SERV_LITERAL_ADDRESS | SERV_NO_ADDR)) &&
546 sockaddr_isequal(&server->addr, &serveraddr))
547 break;
548
572b41eb 549 header = (struct dns_header *)daemon->packet;
fd9fa481 550
1a6bca81 551 if (!server ||
572b41eb 552 n < (int)sizeof(struct dns_header) || !(header->hb3 & HB3_QR) ||
1a6bca81
SK
553 !(forward = lookup_frec(ntohs(header->id), questions_crc(header, n, daemon->namebuff))))
554 return;
555
556 server = forward->sentto;
557
572b41eb 558 if ((RCODE(header) == SERVFAIL || RCODE(header) == REFUSED) &&
28866e95 559 !option_bool(OPT_ORDER) &&
1a6bca81
SK
560 forward->forwardall == 0)
561 /* for broken servers, attempt to send to another one. */
9e4abcb5 562 {
1a6bca81
SK
563 unsigned char *pheader;
564 size_t plen;
565 int is_sign;
832af0ba 566
1a6bca81
SK
567 /* recreate query from reply */
568 pheader = find_pseudoheader(header, (size_t)n, &plen, NULL, &is_sign);
569 if (!is_sign)
832af0ba 570 {
1a6bca81
SK
571 header->ancount = htons(0);
572 header->nscount = htons(0);
573 header->arcount = htons(0);
574 if ((nn = resize_packet(header, (size_t)n, pheader, plen)))
832af0ba 575 {
572b41eb 576 header->hb3 &= ~(HB3_QR | HB3_TC);
1a6bca81
SK
577 forward_query(-1, NULL, NULL, 0, header, nn, now, forward);
578 return;
832af0ba 579 }
832af0ba 580 }
1a6bca81
SK
581 }
582
583 if ((forward->sentto->flags & SERV_TYPE) == 0)
584 {
572b41eb 585 if (RCODE(header) == SERVFAIL || RCODE(header) == REFUSED)
1a6bca81
SK
586 server = NULL;
587 else
b8187c80 588 {
1a6bca81
SK
589 struct server *last_server;
590
591 /* find good server by address if possible, otherwise assume the last one we sent to */
592 for (last_server = daemon->servers; last_server; last_server = last_server->next)
593 if (!(last_server->flags & (SERV_LITERAL_ADDRESS | SERV_HAS_DOMAIN | SERV_FOR_NODOTS | SERV_NO_ADDR)) &&
594 sockaddr_isequal(&last_server->addr, &serveraddr))
595 {
596 server = last_server;
597 break;
598 }
599 }
28866e95 600 if (!option_bool(OPT_ALL_SERVERS))
1a6bca81
SK
601 daemon->last_server = server;
602 }
603
604 /* If the answer is an error, keep the forward record in place in case
605 we get a good reply from another server. Kill it when we've
606 had replies from all to avoid filling the forwarding table when
607 everything is broken */
608 if (forward->forwardall == 0 || --forward->forwardall == 1 ||
572b41eb 609 (RCODE(header) != REFUSED && RCODE(header) != SERVFAIL))
1a6bca81 610 {
28866e95 611 int check_rebind = !(forward->flags & FREC_NOREBIND);
8ef5ada2 612
28866e95 613 if (!option_bool(OPT_NO_REBIND))
8ef5ada2
SK
614 check_rebind = 0;
615
28866e95 616 if ((nn = process_reply(header, now, server, (size_t)n, check_rebind, forward->flags & FREC_CHECKING_DISABLED)))
1a6bca81
SK
617 {
618 header->id = htons(forward->orig_id);
572b41eb 619 header->hb4 |= HB4_RA; /* recursion if available */
28866e95 620 send_from(forward->fd, option_bool(OPT_NOWILD), daemon->packet, nn,
1a6bca81 621 &forward->source, &forward->dest, forward->iface);
b8187c80 622 }
1a6bca81 623 free_frec(forward); /* cancel */
9e4abcb5 624 }
9e4abcb5 625}
44a2a316 626
1a6bca81 627
5aabfc78 628void receive_query(struct listener *listen, time_t now)
44a2a316 629{
572b41eb 630 struct dns_header *header = (struct dns_header *)daemon->packet;
44a2a316 631 union mysockaddr source_addr;
c1bb8504 632 unsigned short type;
44a2a316 633 struct all_addr dst_addr;
f6b7dc47 634 struct in_addr netmask, dst_addr_4;
cdeda28f
SK
635 size_t m;
636 ssize_t n;
637 int if_index = 0;
44a2a316
SK
638 struct iovec iov[1];
639 struct msghdr msg;
640 struct cmsghdr *cmptr;
44a2a316
SK
641 union {
642 struct cmsghdr align; /* this ensures alignment */
643#ifdef HAVE_IPV6
644 char control6[CMSG_SPACE(sizeof(struct in6_pktinfo))];
645#endif
5e9e0efb 646#if defined(HAVE_LINUX_NETWORK)
44a2a316 647 char control[CMSG_SPACE(sizeof(struct in_pktinfo))];
824af85b
SK
648#elif defined(IP_RECVDSTADDR) && defined(HAVE_SOLARIS_NETWORK)
649 char control[CMSG_SPACE(sizeof(struct in_addr)) +
650 CMSG_SPACE(sizeof(unsigned int))];
44a2a316
SK
651#elif defined(IP_RECVDSTADDR)
652 char control[CMSG_SPACE(sizeof(struct in_addr)) +
653 CMSG_SPACE(sizeof(struct sockaddr_dl))];
654#endif
655 } control_u;
656
cdeda28f
SK
657 /* packet buffer overwritten */
658 daemon->srv_save = NULL;
659
28866e95 660 if (listen->family == AF_INET && option_bool(OPT_NOWILD))
f6b7dc47
SK
661 {
662 dst_addr_4 = listen->iface->addr.in.sin_addr;
663 netmask = listen->iface->netmask;
664 }
665 else
3d8df260
SK
666 {
667 dst_addr_4.s_addr = 0;
668 netmask.s_addr = 0;
669 }
f6b7dc47 670
3be34541
SK
671 iov[0].iov_base = daemon->packet;
672 iov[0].iov_len = daemon->edns_pktsz;
44a2a316
SK
673
674 msg.msg_control = control_u.control;
675 msg.msg_controllen = sizeof(control_u);
676 msg.msg_flags = 0;
677 msg.msg_name = &source_addr;
678 msg.msg_namelen = sizeof(source_addr);
679 msg.msg_iov = iov;
680 msg.msg_iovlen = 1;
681
de37951c 682 if ((n = recvmsg(listen->fd, &msg, 0)) == -1)
3be34541 683 return;
44a2a316 684
572b41eb 685 if (n < (int)sizeof(struct dns_header) ||
5e9e0efb 686 (msg.msg_flags & MSG_TRUNC) ||
572b41eb 687 (header->hb3 & HB3_QR))
26128d27
SK
688 return;
689
44a2a316
SK
690 source_addr.sa.sa_family = listen->family;
691#ifdef HAVE_IPV6
692 if (listen->family == AF_INET6)
5e9e0efb 693 source_addr.in6.sin6_flowinfo = 0;
44a2a316 694#endif
28866e95
SK
695
696 if (!option_bool(OPT_NOWILD))
26128d27
SK
697 {
698 struct ifreq ifr;
699
700 if (msg.msg_controllen < sizeof(struct cmsghdr))
701 return;
44a2a316 702
5e9e0efb 703#if defined(HAVE_LINUX_NETWORK)
26128d27
SK
704 if (listen->family == AF_INET)
705 for (cmptr = CMSG_FIRSTHDR(&msg); cmptr; cmptr = CMSG_NXTHDR(&msg, cmptr))
706 if (cmptr->cmsg_level == SOL_IP && cmptr->cmsg_type == IP_PKTINFO)
707 {
8ef5ada2
SK
708 union {
709 unsigned char *c;
710 struct in_pktinfo *p;
711 } p;
712 p.c = CMSG_DATA(cmptr);
713 dst_addr_4 = dst_addr.addr.addr4 = p.p->ipi_spec_dst;
714 if_index = p.p->ipi_ifindex;
26128d27
SK
715 }
716#elif defined(IP_RECVDSTADDR) && defined(IP_RECVIF)
717 if (listen->family == AF_INET)
44a2a316 718 {
26128d27 719 for (cmptr = CMSG_FIRSTHDR(&msg); cmptr; cmptr = CMSG_NXTHDR(&msg, cmptr))
8ef5ada2
SK
720 {
721 union {
722 unsigned char *c;
723 unsigned int *i;
724 struct in_addr *a;
725#ifndef HAVE_SOLARIS_NETWORK
726 struct sockaddr_dl *s;
727#endif
728 } p;
729 p.c = CMSG_DATA(cmptr);
730 if (cmptr->cmsg_level == IPPROTO_IP && cmptr->cmsg_type == IP_RECVDSTADDR)
731 dst_addr_4 = dst_addr.addr.addr4 = *(p.a);
732 else if (cmptr->cmsg_level == IPPROTO_IP && cmptr->cmsg_type == IP_RECVIF)
824af85b 733#ifdef HAVE_SOLARIS_NETWORK
8ef5ada2 734 if_index = *(p.i);
824af85b 735#else
8ef5ada2 736 if_index = p.s->sdl_index;
824af85b 737#endif
8ef5ada2 738 }
44a2a316 739 }
44a2a316 740#endif
26128d27 741
44a2a316 742#ifdef HAVE_IPV6
26128d27
SK
743 if (listen->family == AF_INET6)
744 {
745 for (cmptr = CMSG_FIRSTHDR(&msg); cmptr; cmptr = CMSG_NXTHDR(&msg, cmptr))
316e2730 746 if (cmptr->cmsg_level == IPV6_LEVEL && cmptr->cmsg_type == daemon->v6pktinfo)
26128d27 747 {
8ef5ada2
SK
748 union {
749 unsigned char *c;
750 struct in6_pktinfo *p;
751 } p;
752 p.c = CMSG_DATA(cmptr);
753
754 dst_addr.addr.addr6 = p.p->ipi6_addr;
755 if_index = p.p->ipi6_ifindex;
26128d27
SK
756 }
757 }
44a2a316 758#endif
26128d27
SK
759
760 /* enforce available interface configuration */
761
7622fc06
SK
762 if (!indextoname(listen->fd, if_index, ifr.ifr_name) ||
763 !iface_check(listen->family, &dst_addr, ifr.ifr_name, &if_index))
5e9e0efb 764 return;
832af0ba
SK
765
766 if (listen->family == AF_INET &&
28866e95 767 option_bool(OPT_LOCALISE) &&
832af0ba
SK
768 ioctl(listen->fd, SIOCGIFNETMASK, &ifr) == -1)
769 return;
770
771 netmask = ((struct sockaddr_in *) &ifr.ifr_addr)->sin_addr;
44a2a316
SK
772 }
773
cdeda28f 774 if (extract_request(header, (size_t)n, daemon->namebuff, &type))
44a2a316 775 {
1a6bca81
SK
776 char types[20];
777
778 querystr(types, type);
779
44a2a316 780 if (listen->family == AF_INET)
3be34541 781 log_query(F_QUERY | F_IPV4 | F_FORWARD, daemon->namebuff,
1a6bca81 782 (struct all_addr *)&source_addr.in.sin_addr, types);
44a2a316
SK
783#ifdef HAVE_IPV6
784 else
3be34541 785 log_query(F_QUERY | F_IPV6 | F_FORWARD, daemon->namebuff,
1a6bca81 786 (struct all_addr *)&source_addr.in6.sin6_addr, types);
44a2a316
SK
787#endif
788 }
789
5aabfc78 790 m = answer_request (header, ((char *) header) + PACKETSZ, (size_t)n,
f6b7dc47 791 dst_addr_4, netmask, now);
44a2a316 792 if (m >= 1)
824af85b 793 {
28866e95 794 send_from(listen->fd, option_bool(OPT_NOWILD), (char *)header,
824af85b
SK
795 m, &source_addr, &dst_addr, if_index);
796 daemon->local_answer++;
797 }
798 else if (forward_query(listen->fd, &source_addr, &dst_addr, if_index,
799 header, (size_t)n, now, NULL))
800 daemon->queries_forwarded++;
44a2a316 801 else
824af85b 802 daemon->local_answer++;
44a2a316
SK
803}
804
feba5c1d
SK
805/* The daemon forks before calling this: it should deal with one connection,
806 blocking as neccessary, and then return. Note, need to be a bit careful
807 about resources for debug mode, when the fork is suppressed: that's
808 done by the caller. */
5aabfc78 809unsigned char *tcp_request(int confd, time_t now,
7de060b0 810 union mysockaddr *local_addr, struct in_addr netmask)
feba5c1d 811{
28866e95
SK
812 size_t size = 0;
813 int norebind = 0;
814 int checking_disabled;
cdeda28f 815 size_t m;
c1bb8504 816 unsigned short qtype, gotname;
feba5c1d
SK
817 unsigned char c1, c2;
818 /* Max TCP packet + slop */
5aabfc78 819 unsigned char *packet = whine_malloc(65536 + MAXDNAME + RRFIXEDSZ);
572b41eb 820 struct dns_header *header;
3be34541 821 struct server *last_server;
7de060b0
SK
822 struct in_addr dst_addr_4;
823 union mysockaddr peer_addr;
824 socklen_t peer_len = sizeof(union mysockaddr);
3be34541 825
7de060b0
SK
826 if (getpeername(confd, (struct sockaddr *)&peer_addr, &peer_len) == -1)
827 return packet;
828
feba5c1d
SK
829 while (1)
830 {
831 if (!packet ||
832 !read_write(confd, &c1, 1, 1) || !read_write(confd, &c2, 1, 1) ||
833 !(size = c1 << 8 | c2) ||
834 !read_write(confd, packet, size, 1))
835 return packet;
836
572b41eb 837 if (size < (int)sizeof(struct dns_header))
feba5c1d
SK
838 continue;
839
572b41eb 840 header = (struct dns_header *)packet;
28866e95
SK
841
842 /* save state of "cd" flag in query */
572b41eb 843 checking_disabled = header->hb4 & HB4_CD;
28866e95
SK
844
845 /* RFC 4035: sect 4.6 para 2 */
572b41eb 846 header->hb4 &= ~HB4_AD;
feba5c1d 847
3be34541 848 if ((gotname = extract_request(header, (unsigned int)size, daemon->namebuff, &qtype)))
feba5c1d 849 {
7de060b0 850 char types[20];
feba5c1d 851
7de060b0
SK
852 querystr(types, qtype);
853
854 if (peer_addr.sa.sa_family == AF_INET)
855 log_query(F_QUERY | F_IPV4 | F_FORWARD, daemon->namebuff,
856 (struct all_addr *)&peer_addr.in.sin_addr, types);
feba5c1d 857#ifdef HAVE_IPV6
7de060b0
SK
858 else
859 log_query(F_QUERY | F_IPV6 | F_FORWARD, daemon->namebuff,
860 (struct all_addr *)&peer_addr.in6.sin6_addr, types);
feba5c1d 861#endif
feba5c1d
SK
862 }
863
7de060b0
SK
864 if (local_addr->sa.sa_family == AF_INET)
865 dst_addr_4 = local_addr->in.sin_addr;
866 else
867 dst_addr_4.s_addr = 0;
868
feba5c1d 869 /* m > 0 if answered from cache */
5aabfc78 870 m = answer_request(header, ((char *) header) + 65536, (unsigned int)size,
7de060b0 871 dst_addr_4, netmask, now);
5aabfc78
SK
872
873 /* Do this by steam now we're not in the select() loop */
874 check_log_writer(NULL);
feba5c1d
SK
875
876 if (m == 0)
877 {
28866e95 878 unsigned int flags = 0;
feba5c1d
SK
879 struct all_addr *addrp = NULL;
880 int type = 0;
881 char *domain = NULL;
28866e95
SK
882
883 if (option_bool(OPT_ADD_MAC))
7de060b0
SK
884 size = add_mac(header, size, ((char *) header) + 65536, &peer_addr);
885
feba5c1d 886 if (gotname)
8ef5ada2 887 flags = search_servers(now, &addrp, gotname, daemon->namebuff, &type, &domain, &norebind);
feba5c1d 888
28866e95 889 if (type != 0 || option_bool(OPT_ORDER) || !daemon->last_server)
3be34541
SK
890 last_server = daemon->servers;
891 else
892 last_server = daemon->last_server;
feba5c1d
SK
893
894 if (!flags && last_server)
895 {
896 struct server *firstsendto = NULL;
0a852541
SK
897 unsigned int crc = questions_crc(header, (unsigned int)size, daemon->namebuff);
898
feba5c1d
SK
899 /* Loop round available servers until we succeed in connecting to one.
900 Note that this code subtley ensures that consecutive queries on this connection
901 which can go to the same server, do so. */
902 while (1)
8ef5ada2 903 {
feba5c1d
SK
904 if (!firstsendto)
905 firstsendto = last_server;
906 else
907 {
908 if (!(last_server = last_server->next))
3be34541 909 last_server = daemon->servers;
feba5c1d
SK
910
911 if (last_server == firstsendto)
912 break;
913 }
914
915 /* server for wrong domain */
916 if (type != (last_server->flags & SERV_TYPE) ||
917 (type == SERV_HAS_DOMAIN && !hostname_isequal(domain, last_server->domain)))
918 continue;
7de060b0
SK
919
920 if (last_server->tcpfd == -1)
feba5c1d 921 {
7de060b0
SK
922 if ((last_server->tcpfd = socket(last_server->addr.sa.sa_family, SOCK_STREAM, 0)) == -1)
923 continue;
924
925 if ((!local_bind(last_server->tcpfd, &last_server->source_addr, last_server->interface, 1) ||
926 connect(last_server->tcpfd, &last_server->addr.sa, sa_len(&last_server->addr)) == -1))
927 {
928 close(last_server->tcpfd);
929 last_server->tcpfd = -1;
930 continue;
931 }
932
933#ifdef HAVE_CONNTRACK
934 /* Copy connection mark of incoming query to outgoing connection. */
935 if (option_bool(OPT_CONNTRACK))
936 {
937 unsigned int mark;
938 struct all_addr local;
939#ifdef HAVE_IPV6
940 if (local_addr->sa.sa_family == AF_INET6)
941 local.addr.addr6 = local_addr->in6.sin6_addr;
942 else
943#endif
944 local.addr.addr4 = local_addr->in.sin_addr;
945
946 if (get_incoming_mark(&peer_addr, &local, 1, &mark))
947 setsockopt(last_server->tcpfd, SOL_SOCKET, SO_MARK, &mark, sizeof(unsigned int));
948 }
949#endif
feba5c1d 950 }
824af85b 951
feba5c1d
SK
952 c1 = size >> 8;
953 c2 = size;
954
955 if (!read_write(last_server->tcpfd, &c1, 1, 0) ||
956 !read_write(last_server->tcpfd, &c2, 1, 0) ||
957 !read_write(last_server->tcpfd, packet, size, 0) ||
958 !read_write(last_server->tcpfd, &c1, 1, 1) ||
959 !read_write(last_server->tcpfd, &c2, 1, 1))
960 {
961 close(last_server->tcpfd);
962 last_server->tcpfd = -1;
963 continue;
964 }
824af85b 965
feba5c1d
SK
966 m = (c1 << 8) | c2;
967 if (!read_write(last_server->tcpfd, packet, m, 1))
968 return packet;
969
970 if (!gotname)
3be34541 971 strcpy(daemon->namebuff, "query");
feba5c1d 972 if (last_server->addr.sa.sa_family == AF_INET)
3be34541 973 log_query(F_SERVER | F_IPV4 | F_FORWARD, daemon->namebuff,
1a6bca81 974 (struct all_addr *)&last_server->addr.in.sin_addr, NULL);
feba5c1d
SK
975#ifdef HAVE_IPV6
976 else
3be34541 977 log_query(F_SERVER | F_IPV6 | F_FORWARD, daemon->namebuff,
1a6bca81 978 (struct all_addr *)&last_server->addr.in6.sin6_addr, NULL);
feba5c1d
SK
979#endif
980
981 /* There's no point in updating the cache, since this process will exit and
832af0ba 982 lose the information after a few queries. We make this call for the alias and
feba5c1d 983 bogus-nxdomain side-effects. */
832af0ba
SK
984 /* If the crc of the question section doesn't match the crc we sent, then
985 someone might be attempting to insert bogus values into the cache by
986 sending replies containing questions and bogus answers. */
987 if (crc == questions_crc(header, (unsigned int)m, daemon->namebuff))
28866e95
SK
988 m = process_reply(header, now, last_server, (unsigned int)m,
989 option_bool(OPT_NO_REBIND) && !norebind, checking_disabled);
feba5c1d
SK
990
991 break;
992 }
993 }
994
995 /* In case of local answer or no connections made. */
996 if (m == 0)
3be34541 997 m = setup_reply(header, (unsigned int)size, addrp, flags, daemon->local_ttl);
feba5c1d 998 }
5aabfc78
SK
999
1000 check_log_writer(NULL);
feba5c1d
SK
1001
1002 c1 = m>>8;
1003 c2 = m;
1004 if (!read_write(confd, &c1, 1, 0) ||
1005 !read_write(confd, &c2, 1, 0) ||
1006 !read_write(confd, packet, m, 0))
1007 return packet;
1008 }
1009}
1010
1697269c 1011static struct frec *allocate_frec(time_t now)
9e4abcb5 1012{
1697269c
SK
1013 struct frec *f;
1014
5aabfc78 1015 if ((f = (struct frec *)whine_malloc(sizeof(struct frec))))
9e4abcb5 1016 {
1a6bca81 1017 f->next = daemon->frec_list;
1697269c 1018 f->time = now;
832af0ba 1019 f->sentto = NULL;
1a6bca81 1020 f->rfd4 = NULL;
28866e95 1021 f->flags = 0;
1a6bca81
SK
1022#ifdef HAVE_IPV6
1023 f->rfd6 = NULL;
1024#endif
1025 daemon->frec_list = f;
1697269c 1026 }
9e4abcb5 1027
1697269c
SK
1028 return f;
1029}
9e4abcb5 1030
1a6bca81
SK
1031static struct randfd *allocate_rfd(int family)
1032{
1033 static int finger = 0;
1034 int i;
1035
1036 /* limit the number of sockets we have open to avoid starvation of
1037 (eg) TFTP. Once we have a reasonable number, randomness should be OK */
1038
1039 for (i = 0; i < RANDOM_SOCKS; i++)
9009d746 1040 if (daemon->randomsocks[i].refcount == 0)
1a6bca81 1041 {
9009d746
SK
1042 if ((daemon->randomsocks[i].fd = random_sock(family)) == -1)
1043 break;
1044
1a6bca81
SK
1045 daemon->randomsocks[i].refcount = 1;
1046 daemon->randomsocks[i].family = family;
1047 return &daemon->randomsocks[i];
1048 }
1049
9009d746 1050 /* No free ones or cannot get new socket, grab an existing one */
1a6bca81
SK
1051 for (i = 0; i < RANDOM_SOCKS; i++)
1052 {
1053 int j = (i+finger) % RANDOM_SOCKS;
9009d746
SK
1054 if (daemon->randomsocks[j].refcount != 0 &&
1055 daemon->randomsocks[j].family == family &&
1056 daemon->randomsocks[j].refcount != 0xffff)
1a6bca81
SK
1057 {
1058 finger = j;
1059 daemon->randomsocks[j].refcount++;
1060 return &daemon->randomsocks[j];
1061 }
1062 }
1063
1064 return NULL; /* doom */
1065}
1066
1067static void free_frec(struct frec *f)
1068{
1069 if (f->rfd4 && --(f->rfd4->refcount) == 0)
1070 close(f->rfd4->fd);
1071
1072 f->rfd4 = NULL;
1073 f->sentto = NULL;
28866e95 1074 f->flags = 0;
1a6bca81
SK
1075
1076#ifdef HAVE_IPV6
1077 if (f->rfd6 && --(f->rfd6->refcount) == 0)
1078 close(f->rfd6->fd);
1079
1080 f->rfd6 = NULL;
1081#endif
1082}
1083
1697269c
SK
1084/* if wait==NULL return a free or older than TIMEOUT record.
1085 else return *wait zero if one available, or *wait is delay to
1a6bca81
SK
1086 when the oldest in-use record will expire. Impose an absolute
1087 limit of 4*TIMEOUT before we wipe things (for random sockets) */
5aabfc78 1088struct frec *get_new_frec(time_t now, int *wait)
1697269c 1089{
1a6bca81 1090 struct frec *f, *oldest, *target;
1697269c
SK
1091 int count;
1092
1093 if (wait)
1094 *wait = 0;
1095
1a6bca81 1096 for (f = daemon->frec_list, oldest = NULL, target = NULL, count = 0; f; f = f->next, count++)
832af0ba 1097 if (!f->sentto)
1a6bca81
SK
1098 target = f;
1099 else
1697269c 1100 {
1a6bca81
SK
1101 if (difftime(now, f->time) >= 4*TIMEOUT)
1102 {
1103 free_frec(f);
1104 target = f;
1105 }
1106
1107 if (!oldest || difftime(f->time, oldest->time) <= 0)
1108 oldest = f;
1697269c 1109 }
1a6bca81
SK
1110
1111 if (target)
1112 {
1113 target->time = now;
1114 return target;
1115 }
9e4abcb5
SK
1116
1117 /* can't find empty one, use oldest if there is one
1118 and it's older than timeout */
1697269c 1119 if (oldest && ((int)difftime(now, oldest->time)) >= TIMEOUT)
9e4abcb5 1120 {
1697269c
SK
1121 /* keep stuff for twice timeout if we can by allocating a new
1122 record instead */
1123 if (difftime(now, oldest->time) < 2*TIMEOUT &&
1124 count <= daemon->ftabsize &&
1125 (f = allocate_frec(now)))
1126 return f;
1127
1128 if (!wait)
1129 {
1a6bca81 1130 free_frec(oldest);
1697269c
SK
1131 oldest->time = now;
1132 }
9e4abcb5
SK
1133 return oldest;
1134 }
1135
1697269c 1136 /* none available, calculate time 'till oldest record expires */
208b65c5 1137 if (count > daemon->ftabsize)
1697269c
SK
1138 {
1139 if (oldest && wait)
1140 *wait = oldest->time + (time_t)TIMEOUT - now;
9e4abcb5
SK
1141 return NULL;
1142 }
1697269c
SK
1143
1144 if (!(f = allocate_frec(now)) && wait)
1145 /* wait one second on malloc failure */
1146 *wait = 1;
9e4abcb5 1147
9e4abcb5
SK
1148 return f; /* OK if malloc fails and this is NULL */
1149}
1150
832af0ba
SK
1151/* crc is all-ones if not known. */
1152static struct frec *lookup_frec(unsigned short id, unsigned int crc)
9e4abcb5
SK
1153{
1154 struct frec *f;
1155
1a6bca81 1156 for(f = daemon->frec_list; f; f = f->next)
832af0ba
SK
1157 if (f->sentto && f->new_id == id &&
1158 (f->crc == crc || crc == 0xffffffff))
9e4abcb5
SK
1159 return f;
1160
1161 return NULL;
1162}
1163
1164static struct frec *lookup_frec_by_sender(unsigned short id,
fd9fa481
SK
1165 union mysockaddr *addr,
1166 unsigned int crc)
9e4abcb5 1167{
feba5c1d
SK
1168 struct frec *f;
1169
1a6bca81 1170 for(f = daemon->frec_list; f; f = f->next)
832af0ba 1171 if (f->sentto &&
9e4abcb5 1172 f->orig_id == id &&
fd9fa481 1173 f->crc == crc &&
9e4abcb5
SK
1174 sockaddr_isequal(&f->source, addr))
1175 return f;
1176
1177 return NULL;
1178}
1179
849a8357 1180/* A server record is going away, remove references to it */
5aabfc78 1181void server_gone(struct server *server)
849a8357
SK
1182{
1183 struct frec *f;
1184
1a6bca81 1185 for (f = daemon->frec_list; f; f = f->next)
832af0ba 1186 if (f->sentto && f->sentto == server)
1a6bca81 1187 free_frec(f);
849a8357
SK
1188
1189 if (daemon->last_server == server)
1190 daemon->last_server = NULL;
1191
1192 if (daemon->srv_save == server)
1193 daemon->srv_save = NULL;
1194}
9e4abcb5 1195
316e2730
SK
1196/* return unique random ids. */
1197static unsigned short get_id(unsigned int crc)
9e4abcb5
SK
1198{
1199 unsigned short ret = 0;
832af0ba 1200
316e2730 1201 do
832af0ba
SK
1202 ret = rand16();
1203 while (lookup_frec(ret, crc));
1204
9e4abcb5
SK
1205 return ret;
1206}
1207
1208
1209
1210
1211