First attempt at replacing sockblock with autoblocking.
[doldaconnect.git] / daemon / net.c
CommitLineData
d3372da9 1/*
2 * Dolda Connect - Modular multiuser Direct Connect-style client
302a2600 3 * Copyright (C) 2004 Fredrik Tolf <fredrik@dolda2000.com>
d3372da9 4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18*/
19/* XXX: Implement SOCKS proxyability */
20
21#ifdef HAVE_CONFIG_H
22#include <config.h>
23#endif
24#include <string.h>
25#include <stdlib.h>
26#include <unistd.h>
27#include <fcntl.h>
28#include <sys/ioctl.h>
29#include <sys/socket.h>
30#include <sys/un.h>
72a7835e 31#include <sys/select.h>
d3372da9 32#include <arpa/inet.h>
33#include <netinet/in.h>
34#include <netdb.h>
35#include <sys/signal.h>
7ab36fbd 36#include <sys/stat.h> /* For rebindunix() */
d3372da9 37#ifdef HAVE_LINUX_SOCKIOS_H
38#include <linux/sockios.h>
39#endif
40#include <errno.h>
41#include <net/if.h>
42
43#include "conf.h"
44#include "net.h"
45#include "module.h"
46#include "log.h"
47#include "utils.h"
48#include "sysevents.h"
49
50static struct configvar myvars[] =
51{
d9f89ef5 52 /** The network mode to use. Currently supported values are 0 for
53 * active mode and 1 for passive mode. In the future, SOCKS5 proxy
54 * support may be added. */
d3372da9 55 {CONF_VAR_INT, "mode", {.num = 0}},
d9f89ef5 56 /** Set the SO_REUSEADDR socket option on listening sockets, so
57 * that dead TCP connections waiting for timeout are ignored. */
347d6d76 58 {CONF_VAR_BOOL, "reuseaddr", {.num = 0}},
d9f89ef5 59 /** Overrides the IPv4 address reported to other clients in active
60 * mode. Useful for servers behind NAT routers. If both this and
61 * net.publicif are unspecified the address of the hub connection
62 * is used. */
d3372da9 63 {CONF_VAR_IPV4, "visibleipv4", {.ipv4 = {0}}},
d9f89ef5 64 /** Specifies an interface name from which to fetch the IPv4
65 * address reported to other clients in active mode. If both this
66 * and net.visibleipv4 are unspecified the address of the hub
67 * connection is used. */
d3372da9 68 {CONF_VAR_STRING, "publicif", {.str = L""}},
b020fb3d 69 /* Diffserv should be supported on IPv4, too, but I don't know the
70 * API to do that. */
d9f89ef5 71 /** The Diffserv value to use on IPv6 connections when the
72 * minimize cost TOS value is used (see the TOS VALUES
73 * section). */
b020fb3d 74 {CONF_VAR_INT, "diffserv-mincost", {.num = 0}},
d9f89ef5 75 /** The Diffserv value to use on IPv6 connections when the
76 * maximize reliability TOS value is used (see the TOS VALUES
77 * section). */
b020fb3d 78 {CONF_VAR_INT, "diffserv-maxrel", {.num = 0}},
d9f89ef5 79 /** The Diffserv value to use on IPv6 connections when the
80 * maximize throughput TOS value is used (see the TOS VALUES
81 * section). */
b020fb3d 82 {CONF_VAR_INT, "diffserv-maxtp", {.num = 0}},
d9f89ef5 83 /** The Diffserv value to use on IPv6 connections when the
84 * minimize delay TOS value is used (see the TOS VALUES
85 * section). */
b020fb3d 86 {CONF_VAR_INT, "diffserv-mindelay", {.num = 0}},
d3372da9 87 {CONF_VAR_END}
88};
89
02a705ad
FT
90#define UFD_SOCK 0
91#define UFD_PIPE 1
92#define UFD_LISTEN 2
93
94struct scons {
95 struct scons *n, *p;
96 struct socket *s;
97};
98
99struct ufd {
100 struct ufd *next, *prev;
101 int fd;
102 int type;
103 int ignread;
104 struct socket *sk;
105 union {
106 struct {
107 int family;
108 int type;
109 struct sockaddr *remote;
110 socklen_t remotelen;
111 struct {
112 uid_t uid;
113 gid_t gid;
114 } ucred;
115 } s;
116 struct {
117 struct lport *lp;
118 int family;
119 } l;
120 } d;
121};
122
40c21c4f
FT
123static int getlocalname(int fd, struct sockaddr **namebuf, socklen_t *lenbuf);
124
02a705ad
FT
125static struct ufd *ufds = NULL;
126static struct scons *rbatch, *wbatch, *cbatch;
d3372da9 127int numsocks = 0;
128
129/* XXX: Get autoconf for all this... */
130int getpublicaddr(int af, struct sockaddr **addr, socklen_t *lenbuf)
131{
132 struct sockaddr_in *ipv4;
133 struct configvar *var;
134 void *bufend;
135 int sock;
136 struct ifconf conf;
137 struct ifreq *ifr, req;
138 char *pif;
139
140 if(af == AF_INET)
141 {
142 var = confgetvar("net", "visibleipv4");
143 if(var->val.ipv4.s_addr != 0)
144 {
145 ipv4 = smalloc(sizeof(*ipv4));
146 ipv4->sin_family = AF_INET;
147 ipv4->sin_addr.s_addr = var->val.ipv4.s_addr;
148 *addr = (struct sockaddr *)ipv4;
149 *lenbuf = sizeof(*ipv4);
150 return(0);
151 }
bcb73bb3 152 if((pif = icswcstombs(confgetstr("net", "publicif"), NULL, NULL)) == NULL)
d3372da9 153 {
154 flog(LOG_ERR, "could not convert net.publicif into local charset: %s", strerror(errno));
155 return(-1);
156 }
6fb0c6ac 157 if(!strcmp(pif, ""))
158 return(1);
d3372da9 159 if((sock = socket(AF_INET, SOCK_DGRAM, 0)) < 0)
160 return(-1);
161 conf.ifc_buf = smalloc(conf.ifc_len = 65536);
162 if(ioctl(sock, SIOCGIFCONF, &conf) < 0)
163 {
164 free(conf.ifc_buf);
165 close(sock);
166 return(-1);
167 }
168 bufend = ((char *)conf.ifc_buf) + conf.ifc_len;
169 ipv4 = NULL;
170 for(ifr = conf.ifc_ifcu.ifcu_req; (void *)ifr < bufend; ifr++)
171 {
1315d02a 172 if(strcmp(ifr->ifr_name, pif))
6fb0c6ac 173 continue;
d3372da9 174 memset(&req, 0, sizeof(req));
175 memcpy(req.ifr_name, ifr->ifr_name, sizeof(ifr->ifr_name));
176 if(ioctl(sock, SIOCGIFFLAGS, &req) < 0)
6fb0c6ac 177 break;
178 if(!(req.ifr_flags & IFF_UP))
d3372da9 179 {
6fb0c6ac 180 flog(LOG_WARNING, "public interface is down");
181 break;
d3372da9 182 }
6fb0c6ac 183 if(ifr->ifr_addr.sa_family != AF_INET)
d3372da9 184 {
6fb0c6ac 185 flog(LOG_WARNING, "address of the public interface is not AF_INET");
186 break;
d3372da9 187 }
6fb0c6ac 188 ipv4 = smalloc(sizeof(*ipv4));
189 memcpy(ipv4, &ifr->ifr_addr, sizeof(ifr->ifr_addr));
190 break;
d3372da9 191 }
bcb73bb3 192 free(conf.ifc_buf);
d3372da9 193 close(sock);
194 if(ipv4 != NULL)
195 {
196 *addr = (struct sockaddr *)ipv4;
197 *lenbuf = sizeof(*ipv4);
198 return(0);
199 }
200 errno = ENETDOWN;
201 return(-1);
202 }
6fb0c6ac 203 return(1);
d3372da9 204}
205
02a705ad 206static struct socket *newsock1(int dgram)
d3372da9 207{
208 struct socket *new;
209
02a705ad
FT
210 new = memset(smalloc(sizeof(*new)), 0, sizeof(*new));
211 new->refcount = 1;
d3372da9 212 new->state = -1;
02a705ad 213 new->dgram = dgram;
81286a22 214 new->maxbuf = 65536;
d3372da9 215 numsocks++;
216 return(new);
217}
218
02a705ad
FT
219static struct socket *sockpair(int dgram)
220{
221 struct socket *s1, *s2;
222
223 s1 = newsock1(dgram);
224 s2 = newsock1(dgram);
225 s1->back = s2;
226 s2->back = s1;
227 putsock(s2);
228 return(s1);
229}
230
231static void sksetstate(struct socket *sk, int state)
232{
233 sk->state = state;
234 sk->back->state = state;
235}
236
237static void closeufd(struct ufd *ufd)
238{
239 if(ufd->fd != -1)
240 close(ufd->fd);
241 ufd->fd = -1;
242}
243
244static void freeufd(struct ufd *ufd)
245{
246 if(ufd->next != NULL)
247 ufd->next->prev = ufd->prev;
248 if(ufd->prev != NULL)
249 ufd->prev->next = ufd->next;
250 if(ufd == ufds)
251 ufds = ufd->next;
252 closeufd(ufd);
5afb0d14
FT
253 if(ufd->sk != NULL)
254 putsock(ufd->sk);
02a705ad
FT
255 if(ufd->type == UFD_SOCK) {
256 if(ufd->d.s.remote != NULL)
257 free(ufd->d.s.remote);
258 }
259 free(ufd);
260}
261
262static struct ufd *mkufd(int fd, int type, struct socket *sk)
263{
264 struct ufd *ufd;
265
266 ufd = memset(smalloc(sizeof(*ufd)), 0, sizeof(*ufd));
267 ufd->fd = fd;
5afb0d14 268 ufd->type = type;
02a705ad
FT
269 if(sk != NULL) {
270 getsock(ufd->sk = sk);
271 sk->ufd = ufd;
272 }
273 if(type == UFD_SOCK) {
274 ufd->d.s.ucred.uid = -1;
275 ufd->d.s.ucred.gid = -1;
276 }
277 ufd->next = ufds;
278 if(ufds)
279 ufds->prev = ufd;
280 ufds = ufd;
281 return(ufd);
282}
283
284static struct ufd *dupufd(struct ufd *ufd)
285{
286 struct ufd *nufd;
287 struct socket *nsk;
288
289 if(ufd->sk != NULL)
290 nsk = sockpair(ufd->sk->dgram);
291 else
292 nsk = NULL;
293 nufd = mkufd(ufd->fd, ufd->type, nsk);
294 if(nsk != NULL)
295 putsock(nsk);
296 if((nufd->fd = dup(ufd->fd)) < 0)
297 {
298 flog(LOG_WARNING, "could not dup() fd: %s", strerror(errno));
299 freeufd(nufd);
300 return(NULL);
301 }
5afb0d14 302 sksetstate(nsk, SOCK_EST);
02a705ad
FT
303 if(ufd->type == UFD_SOCK) {
304 nufd->d.s.family = ufd->d.s.family;
305 nufd->d.s.type = ufd->d.s.type;
306 nufd->d.s.ucred.uid = ufd->d.s.ucred.uid;
307 nufd->d.s.ucred.gid = ufd->d.s.ucred.gid;
308 if(ufd->d.s.remote != NULL)
309 nufd->d.s.remote = memcpy(smalloc(ufd->d.s.remotelen), ufd->d.s.remote, nufd->d.s.remotelen = ufd->d.s.remotelen);
310 } else if(ufd->type == UFD_LISTEN) {
311 nufd->d.l.family = ufd->d.l.family;
312 }
313 return(nufd);
314}
315
d3372da9 316static struct socket *mksock(int domain, int type)
317{
318 int fd;
02a705ad
FT
319 struct socket *sk;
320 struct ufd *ufd;
d3372da9 321
322 if((fd = socket(domain, type, 0)) < 0)
323 {
324 flog(LOG_CRIT, "could not create socket: %s", strerror(errno));
325 return(NULL);
326 }
02a705ad
FT
327 sk = sockpair(type == SOCK_DGRAM);
328 ufd = mkufd(fd, UFD_SOCK, sk);
329 ufd->d.s.family = domain;
330 ufd->d.s.type = type;
d3372da9 331 fcntl(fd, F_SETFL, fcntl(fd, F_GETFL) | O_NONBLOCK);
02a705ad 332 return(sk);
d3372da9 333}
334
335struct socket *wrapsock(int fd)
336{
02a705ad
FT
337 struct socket *sk;
338 struct ufd *ufd;
d3372da9 339
02a705ad
FT
340 sk = sockpair(0);
341 ufd = mkufd(fd, UFD_PIPE, sk->back);
342 sksetstate(sk, SOCK_EST);
d3372da9 343 fcntl(fd, F_SETFL, fcntl(fd, F_GETFL) | O_NONBLOCK);
02a705ad 344 return(sk);
d3372da9 345}
346
347void getsock(struct socket *sk)
348{
349 sk->refcount++;
350}
351
02a705ad 352static void freesock(struct socket *sk)
d3372da9 353{
354 struct dgrambuf *buf;
355
02a705ad
FT
356 if(sk->dgram) {
357 while((buf = sk->buf.d.f) != NULL) {
358 sk->buf.d.f = buf->next;
359 freedgbuf(buf);
d3372da9 360 }
02a705ad
FT
361 } else {
362 if(sk->buf.s.buf != NULL)
363 free(sk->buf.s.buf);
364 }
365 free(sk);
366 numsocks--;
367}
368
369void putsock(struct socket *sk)
370{
7dddb551
FT
371 struct socket *back;
372
02a705ad
FT
373 if(--(sk->refcount) < 0) {
374 flog(LOG_CRIT, "BUG: socket refcount < 0");
375 abort();
376 }
377 if((sk->refcount == 0) && (sk->back->refcount == 0)) {
7dddb551 378 back = sk->back;
02a705ad 379 freesock(sk);
7dddb551 380 freesock(back);
02a705ad
FT
381 }
382}
383
384static void linksock(struct scons **list, struct socket *sk)
385{
386 struct scons *sc;
387
388 for(sc = *list; sc != NULL; sc = sc->n) {
389 if(sc->s == sk)
390 return;
d3372da9 391 }
02a705ad
FT
392 sc = smalloc(sizeof(*sc));
393 getsock(sc->s = sk);
394 sc->n = *list;
395 sc->p = NULL;
396 if(*list)
397 (*list)->p = sc;
398 *list = sc;
d3372da9 399}
400
336539c2 401void sockpushdata(struct socket *sk, void *buf, size_t size)
402{
02a705ad
FT
403 if(size == 0)
404 return;
405 if(sk->dgram) {
336539c2 406 /* XXX */
02a705ad
FT
407 } else {
408 sizebuf(&sk->buf.s.buf, &sk->buf.s.bufsize, sk->buf.s.datasize + size, 1, 1);
409 memmove(sk->buf.s.buf + size, sk->buf.s.buf, sk->buf.s.datasize);
410 memcpy(sk->buf.s.buf, buf, size);
411 sk->buf.s.datasize += size;
412 linksock(&rbatch, sk);
336539c2 413 }
02a705ad
FT
414}
415
416/* Read as the preterite of `read' */
417void sockread(struct socket *sk)
418{
5afb0d14
FT
419 if((sockgetdatalen(sk) == 0) && (sk->eos == 1))
420 linksock(&rbatch, sk);
02a705ad
FT
421 linksock(&wbatch, sk->back);
422}
423
424void freedgbuf(struct dgrambuf *dg)
425{
426 if(dg->data != NULL)
427 free(dg->data);
428 if(dg->addr != NULL)
429 free(dg->addr);
430 free(dg);
431}
432
433struct dgrambuf *sockgetdgbuf(struct socket *sk)
434{
435 struct dgrambuf *dbuf;
436
437 if((dbuf = sk->buf.d.f) == NULL)
438 return(NULL);
439 sk->buf.d.f = dbuf->next;
440 if(dbuf->next == NULL)
441 sk->buf.d.l = NULL;
442 dbuf->next = NULL;
443 sockread(sk);
444 return(dbuf);
336539c2 445}
446
d3372da9 447void *sockgetinbuf(struct socket *sk, size_t *size)
448{
449 void *buf;
450 struct dgrambuf *dbuf;
451
02a705ad
FT
452 if(sk->dgram) {
453 dbuf = sockgetdgbuf(sk);
d3372da9 454 buf = dbuf->data;
455 *size = dbuf->size;
456 free(dbuf->addr);
457 free(dbuf);
02a705ad
FT
458 } else {
459 if((sk->buf.s.buf == NULL) || (sk->buf.s.datasize == 0))
460 {
461 *size = 0;
462 return(NULL);
463 }
464 buf = sk->buf.s.buf;
465 *size = sk->buf.s.datasize;
466 sk->buf.s.buf = NULL;
467 sk->buf.s.bufsize = sk->buf.s.datasize = 0;
468 sockread(sk);
d3372da9 469 }
02a705ad
FT
470 return(buf);
471}
472
473void sockqueue(struct socket *sk, void *data, size_t size)
474{
475 struct dgrambuf *new;
476 struct sockaddr *remote;
477 socklen_t remotelen;
478
479 if(size == 0)
480 return;
481 if(sk->state == SOCK_STL)
482 return;
483 if(sk->dgram) {
484 if(sockpeeraddr(sk, &remote, &remotelen))
485 return;
486 new = smalloc(sizeof(*new));
487 new->next = NULL;
488 memcpy(new->data = smalloc(size), data, new->size = size);
489 new->addr = remote;
490 new->addrlen = remotelen;
491 if(sk->back->buf.d.l == NULL)
492 {
493 sk->back->buf.d.l = sk->back->buf.d.f = new;
494 } else {
495 sk->back->buf.d.l->next = new;
496 sk->back->buf.d.l = new;
497 }
498 } else {
499 sizebuf(&(sk->back->buf.s.buf), &(sk->back->buf.s.bufsize), sk->back->buf.s.datasize + size, 1, 1);
500 memcpy(sk->back->buf.s.buf + sk->back->buf.s.datasize, data, size);
501 sk->back->buf.s.datasize += size;
502 }
503 linksock(&rbatch, sk->back);
504}
505
506void sockqueuedg(struct socket *sk, struct dgrambuf *dg)
507{
508 if(sk->state == SOCK_STL) {
509 freedgbuf(dg);
510 return;
511 }
512 if(!sk->dgram) {
513 flog(LOG_ERR, "BUG: sockqueuedg called on non-dgram socket");
514 freedgbuf(dg);
515 return;
516 }
517 dg->next = NULL;
518 if(sk->back->buf.d.l == NULL)
519 {
520 sk->back->buf.d.l = sk->back->buf.d.f = dg;
521 } else {
522 sk->back->buf.d.l->next = dg;
523 sk->back->buf.d.l = dg;
524 }
525 linksock(&rbatch, sk->back);
526}
527
40c21c4f 528void sockerror(struct socket *sk, int en)
02a705ad
FT
529{
530 sksetstate(sk, SOCK_STL);
531 if(sk->back->errcb != NULL)
532 sk->back->errcb(sk->back, en, sk->back->data);
d3372da9 533}
534
02a705ad 535static void recvcmsg(struct ufd *ufd, struct msghdr *msg)
9e5f2b29 536{
537 struct cmsghdr *cmsg;
9e5f2b29 538
539 for(cmsg = CMSG_FIRSTHDR(msg); cmsg != NULL; cmsg = CMSG_NXTHDR(msg, cmsg))
540 {
a8c5ada6 541#if UNIX_AUTH_STYLE == 1
9e5f2b29 542 if((cmsg->cmsg_level == SOL_SOCKET) && (cmsg->cmsg_type == SCM_CREDENTIALS))
543 {
a8c5ada6 544 struct ucred *cred;
02a705ad 545 if(ufd->d.s.ucred.uid == -1)
9e5f2b29 546 {
547 cred = (struct ucred *)CMSG_DATA(cmsg);
02a705ad
FT
548 ufd->d.s.ucred.uid = cred->uid;
549 ufd->d.s.ucred.gid = cred->gid;
9e5f2b29 550 }
551 }
a8c5ada6 552#endif
9e5f2b29 553 }
554}
555
02a705ad
FT
556static int ufddgram(struct ufd *ufd)
557{
558 int dgram;
559
560 if(ufd->type == UFD_SOCK) {
561 dgram = ufd->d.s.type == SOCK_DGRAM;
562 } else if(ufd->type == UFD_PIPE) {
563 dgram = 0;
564 } else {
565 flog(LOG_ERR, "BUG: calling ufddgram on ufd of bad type %i", ufd->type);
566 return(-1);
567 }
568 if(ufd->sk == NULL) {
569 flog(LOG_ERR, "BUG: calling ufddgram on socketless ufd (type %i)", ufd->type);
570 return(-1);
571 }
572 if(dgram != ufd->sk->dgram) {
573 flog(LOG_ERR, "BUG: ufd/socket dgram value mismatch");
574 return(-1);
575 }
576 return(dgram);
577}
578
579static void sockrecv(struct ufd *ufd)
d3372da9 580{
581 int ret, inq;
02a705ad 582 int dgram;
d3372da9 583 struct dgrambuf *dbuf;
9e5f2b29 584 struct msghdr msg;
585 char cbuf[65536];
586 struct iovec bufvec;
02a705ad 587 void *buf;
d3372da9 588
9e5f2b29 589 memset(&msg, 0, sizeof(msg));
590 msg.msg_iov = &bufvec;
591 msg.msg_iovlen = 1;
592 msg.msg_control = cbuf;
593 msg.msg_controllen = sizeof(cbuf);
02a705ad
FT
594 if((dgram = ufddgram(ufd)) < 0)
595 return;
596 if(dgram) {
9e5f2b29 597#if defined(HAVE_LINUX_SOCKIOS_H) && defined(SIOCINQ)
02a705ad 598 if(ioctl(ufd->fd, SIOCINQ, &inq))
d3372da9 599 {
600 /* I don't really know what could go wrong here, so let's
601 * assume it's transient. */
02a705ad 602 flog(LOG_WARNING, "SIOCINQ return %s on socket %i", strerror(errno), ufd->fd);
d3372da9 603 return;
604 }
9e5f2b29 605#else
606 inq = 65536;
607#endif
d3372da9 608 dbuf = smalloc(sizeof(*dbuf));
609 dbuf->data = smalloc(inq);
610 dbuf->addr = smalloc(dbuf->addrlen = sizeof(struct sockaddr_storage));
9e5f2b29 611 msg.msg_name = dbuf->addr;
612 msg.msg_namelen = dbuf->addrlen;
613 bufvec.iov_base = dbuf->data;
614 bufvec.iov_len = inq;
02a705ad 615 ret = recvmsg(ufd->fd, &msg, 0);
9e5f2b29 616 dbuf->addrlen = msg.msg_namelen;
d3372da9 617 if(ret < 0)
618 {
02a705ad 619 freedgbuf(dbuf);
d3372da9 620 if((errno == EINTR) || (errno == EAGAIN))
621 return;
02a705ad
FT
622 closeufd(ufd);
623 sockerror(ufd->sk, errno);
d3372da9 624 return;
625 }
9e5f2b29 626 if(msg.msg_flags & MSG_CTRUNC)
627 flog(LOG_DEBUG, "ancillary data was truncated");
628 else
02a705ad 629 recvcmsg(ufd, &msg);
d3372da9 630 /* On UDP/IPv[46], ret == 0 doesn't mean EOF (since UDP can't
631 * have EOF), but rather an empty packet. I don't know if any
632 * other potential DGRAM protocols might have an EOF
633 * condition, so let's play safe. */
634 if(ret == 0)
635 {
02a705ad
FT
636 freedgbuf(dbuf);
637 if((ufd->type != UFD_SOCK) || !((ufd->d.s.family == AF_INET) || (ufd->d.s.family == AF_INET6)))
d3372da9 638 {
40c21c4f 639 closesock(ufd->sk);
02a705ad 640 closeufd(ufd);
d3372da9 641 }
642 return;
643 }
644 dbuf->addr = srealloc(dbuf->addr, dbuf->addrlen);
645 dbuf->data = srealloc(dbuf->data, dbuf->size = ret);
646 dbuf->next = NULL;
02a705ad
FT
647 sockqueuedg(ufd->sk, dbuf);
648 } else {
649#if defined(HAVE_LINUX_SOCKIOS_H) && defined(SIOCINQ)
650 /* SIOCINQ is Linux-specific AFAIK, but I really have no idea
651 * how to read the inqueue size on other OSs */
652 if(ufd->type == UFD_SOCK) {
653 if(ioctl(ufd->fd, SIOCINQ, &inq))
654 {
655 /* I don't really know what could go wrong here, so let's
656 * assume it's transient. */
657 flog(LOG_WARNING, "SIOCINQ return %s on socket %i, falling back to 2048 bytes", strerror(errno), ufd->fd);
658 inq = 2048;
659 }
660 } else {
661 /* There are perils when trying to use SIOCINQ on files >2GiB... */
662 inq = 65536;
663 }
664#else
665 inq = 2048;
666#endif
667 if(inq > 65536)
668 inq = 65536;
669 /* This part could be optimized by telling the kernel to read
670 * directly into ufd->sk->back->buf, but that would be uglier
671 * by not using the socket function interface. */
672 buf = smalloc(inq);
673 if(ufd->type == UFD_SOCK)
674 {
675 bufvec.iov_base = buf;
676 bufvec.iov_len = inq;
677 ret = recvmsg(ufd->fd, &msg, 0);
678 } else {
679 ret = read(ufd->fd, buf, inq);
680 msg.msg_controllen = 0;
681 msg.msg_flags = 0;
682 }
683 if(ret < 0)
684 {
685 free(buf);
686 if((errno == EINTR) || (errno == EAGAIN))
687 return;
688 closeufd(ufd);
689 sockerror(ufd->sk, errno);
690 return;
691 }
692 if(msg.msg_flags & MSG_CTRUNC)
693 flog(LOG_DEBUG, "ancillary data was truncated");
d3372da9 694 else
02a705ad
FT
695 recvcmsg(ufd, &msg);
696 if(ret == 0)
697 {
698 free(buf);
699 closeufd(ufd);
40c21c4f 700 closesock(ufd->sk);
02a705ad
FT
701 return;
702 }
703 sockqueue(ufd->sk, buf, ret);
704 free(buf);
d3372da9 705 }
706}
707
89d650f1 708static int sockflush(struct ufd *ufd)
d3372da9 709{
710 int ret;
711 struct dgrambuf *dbuf;
02a705ad 712 int dgram;
d3372da9 713
89d650f1
FT
714 if((dgram = ufddgram(ufd)) < 0) {
715 errno = EBADFD;
716 return(-1);
717 }
02a705ad
FT
718 if(dgram) {
719 dbuf = sockgetdgbuf(ufd->sk);
720 sendto(ufd->fd, dbuf->data, dbuf->size, MSG_DONTWAIT | MSG_NOSIGNAL, dbuf->addr, dbuf->addrlen);
721 freedgbuf(dbuf);
722 } else {
723 if(ufd->type == UFD_SOCK)
724 ret = send(ufd->fd, ufd->sk->buf.s.buf, ufd->sk->buf.s.datasize, MSG_DONTWAIT | MSG_NOSIGNAL);
d3372da9 725 else
02a705ad 726 ret = write(ufd->fd, ufd->sk->buf.s.buf, ufd->sk->buf.s.datasize);
89d650f1
FT
727 if(ret < 0)
728 return(-1);
02a705ad
FT
729 if(ret > 0) {
730 memmove(ufd->sk->buf.s.buf, ((char *)ufd->sk->buf.s.buf) + ret, ufd->sk->buf.s.datasize -= ret);
731 sockread(ufd->sk);
d3372da9 732 }
d3372da9 733 }
89d650f1 734 return(0);
d3372da9 735}
736
737void closesock(struct socket *sk)
738{
02a705ad 739 sksetstate(sk, SOCK_STL);
40c21c4f
FT
740 if(sk->back->eos == 0)
741 sk->back->eos = 1;
742 linksock(&rbatch, sk->back);
d3372da9 743}
744
745size_t sockgetdatalen(struct socket *sk)
746{
747 struct dgrambuf *b;
748 size_t ret;
749
02a705ad 750 if(sk->dgram) {
d3372da9 751 ret = 0;
02a705ad 752 for(b = sk->buf.d.f; b != NULL; b = b->next)
d3372da9 753 ret += b->size;
02a705ad
FT
754 } else {
755 ret = sk->buf.s.datasize;
d3372da9 756 }
757 return(ret);
758}
759
81286a22
FT
760/* size_t sockqueuesize(struct socket *sk) */
761/* { */
762/* return(sockgetdatalen(sk->back)); */
763/* } */
764
765ssize_t sockqueueleft(struct socket *sk)
d3372da9 766{
81286a22 767 return(sk->back->maxbuf - sockgetdatalen(sk->back));
d3372da9 768}
769
d3372da9 770/*
7ab36fbd 771 * Seriously, I don't know if it's naughty or not to remove
772 * pre-existing Unix sockets.
773 */
02a705ad 774static int rebindunix(struct ufd *ufd, struct sockaddr *name, socklen_t namelen)
7ab36fbd 775{
776 struct sockaddr_un *un;
777 struct stat sb;
778
02a705ad 779 if((ufd->d.l.family != AF_UNIX) || (name->sa_family != PF_UNIX))
7ab36fbd 780 return(-1);
781 un = (struct sockaddr_un *)name;
782 if(stat(un->sun_path, &sb))
783 return(-1);
784 if(!S_ISSOCK(sb.st_mode))
785 return(-1);
786 if(unlink(un->sun_path))
787 return(-1);
02a705ad 788 if(bind(ufd->fd, name, namelen) < 0)
7ab36fbd 789 return(-1);
790 return(0);
791}
792
f96d0914
FT
793void closelport(struct lport *lp)
794{
40c21c4f
FT
795 struct ufd *ufd;
796 struct sockaddr_un *un;
797
798 ufd = lp->ufd;
799 if((ufd->d.l.family == AF_UNIX) && !getlocalname(ufd->fd, (struct sockaddr **)(void *)&un, NULL) && (un->sun_family == PF_UNIX) && strchr(un->sun_path, '/')) {
800 if(unlink(un->sun_path))
801 flog(LOG_WARNING, "could not unlink Unix socket %s: %s", un->sun_path, strerror(errno));
802 }
f96d0914
FT
803 freeufd(lp->ufd);
804}
805
7ab36fbd 806/*
d3372da9 807 * The difference between netcslisten() and netcslistenlocal() is that
808 * netcslistenlocal() always listens on the local host, instead of
809 * following proxy/passive mode directions. It is suitable for eg. the
810 * UI channel, while the file sharing networks should, naturally, use
811 * netcslisten() instead.
812*/
813
02a705ad 814struct lport *netcslistenlocal(int type, struct sockaddr *name, socklen_t namelen, void (*func)(struct lport *, struct socket *, void *), void *data)
d3372da9 815{
02a705ad
FT
816 struct lport *lp;
817 struct ufd *ufd;
818 int fd;
66c517d2 819 int intbuf;
d3372da9 820
821 /* I don't know if this is actually correct (it probably isn't),
822 * but since, at on least Linux systems, PF_* are specifically
823 * #define'd to their AF_* counterparts, it allows for a severely
824 * smoother implementation. If it breaks something on your
825 * platform, please tell me so.
826 */
02a705ad 827 if((fd = socket(name->sa_family, type, 0)) < 0)
d3372da9 828 return(NULL);
02a705ad 829 if(confgetint("net", "reuseaddr")) {
687b2ee2 830 intbuf = 1;
02a705ad 831 setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &intbuf, sizeof(intbuf));
687b2ee2 832 }
02a705ad 833 ufd = mkufd(fd, UFD_LISTEN, NULL);
02a705ad 834 ufd->d.l.family = name->sa_family;
f96d0914
FT
835 lp = memset(smalloc(sizeof(*lp)), 0, sizeof(*lp));
836 lp->ufd = ufd;
837 ufd->d.l.lp = lp;
02a705ad
FT
838 if((bind(fd, name, namelen) < 0) && ((errno != EADDRINUSE) || (rebindunix(ufd, name, namelen) < 0))) {
839 freeufd(ufd);
d3372da9 840 return(NULL);
841 }
02a705ad 842 if(listen(fd, 16) < 0)
d3372da9 843 {
02a705ad 844 freeufd(ufd);
d3372da9 845 return(NULL);
846 }
02a705ad
FT
847 lp->acceptcb = func;
848 lp->data = data;
849 return(lp);
d3372da9 850}
851
02a705ad 852struct lport *netcslisten(int type, struct sockaddr *name, socklen_t namelen, void (*func)(struct lport *, struct socket *, void *), void *data)
c23acc61 853{
854 if(confgetint("net", "mode") == 1)
855 {
856 errno = EOPNOTSUPP;
857 return(NULL);
858 }
c23acc61 859 if(confgetint("net", "mode") == 0)
860 return(netcslistenlocal(type, name, namelen, func, data));
861 errno = EOPNOTSUPP;
862 return(NULL);
863}
864
02a705ad 865struct lport *netcstcplisten(int port, int local, void (*func)(struct lport *, struct socket *, void *), void *data)
0a1bc5b1 866{
867 struct sockaddr_in addr;
868#ifdef HAVE_IPV6
869 struct sockaddr_in6 addr6;
870#endif
02a705ad
FT
871 struct lport *(*csfunc)(int, struct sockaddr *, socklen_t, void (*)(struct lport *, struct socket *, void *), void *);
872 struct lport *ret;
0a1bc5b1 873
874 if(local)
875 csfunc = netcslistenlocal;
876 else
877 csfunc = netcslisten;
878#ifdef HAVE_IPV6
879 memset(&addr6, 0, sizeof(addr6));
880 addr6.sin6_family = AF_INET6;
881 addr6.sin6_port = htons(port);
882 addr6.sin6_addr = in6addr_any;
883 if((ret = csfunc(SOCK_STREAM, (struct sockaddr *)&addr6, sizeof(addr6), func, data)) != NULL)
884 return(ret);
885 if((ret == NULL) && (errno != EAFNOSUPPORT))
886 return(NULL);
887#endif
888 memset(&addr, 0, sizeof(addr));
889 addr.sin_family = AF_INET;
890 addr.sin_port = htons(port);
891 return(csfunc(SOCK_STREAM, (struct sockaddr *)&addr, sizeof(addr), func, data));
892}
893
d3372da9 894struct socket *netcsdgram(struct sockaddr *name, socklen_t namelen)
895{
896 struct socket *sk;
897 int mode;
898
899 mode = confgetint("net", "mode");
900 if((mode == 0) || (mode == 1))
901 {
902 if((sk = mksock(name->sa_family, SOCK_DGRAM)) == NULL)
903 return(NULL);
02a705ad 904 if(bind(sk->ufd->fd, name, namelen) < 0)
d3372da9 905 {
906 putsock(sk);
907 return(NULL);
908 }
02a705ad
FT
909 sksetstate(sk, SOCK_EST);
910 return(sk->back);
d3372da9 911 }
912 errno = EOPNOTSUPP;
913 return(NULL);
914}
915
02a705ad 916struct socket *netdgramconn(struct socket *sk, struct sockaddr *addr, socklen_t addrlen)
d3372da9 917{
02a705ad 918 struct ufd *nufd;
d3372da9 919
02a705ad 920 nufd = dupufd(sk->back->ufd);
5afb0d14 921 getsock(sk = nufd->sk->back);
02a705ad
FT
922 memcpy(nufd->d.s.remote = smalloc(addrlen), addr, nufd->d.s.remotelen = addrlen);
923 nufd->ignread = 1;
924 return(sk);
d3372da9 925}
926
cab0b442 927struct socket *netcsconn(struct sockaddr *addr, socklen_t addrlen, void (*func)(struct socket *, int, void *), void *data)
d3372da9 928{
929 struct socket *sk;
930 int mode;
931
932 mode = confgetint("net", "mode");
933 if((mode == 0) || (mode == 1))
934 {
935 if((sk = mksock(addr->sa_family, SOCK_STREAM)) == NULL)
936 return(NULL);
02a705ad
FT
937 memcpy(sk->ufd->d.s.remote = smalloc(addrlen), addr, sk->ufd->d.s.remotelen = addrlen);
938 sk->back->conncb = func;
939 sk->back->data = data;
940 if(!connect(sk->ufd->fd, addr, addrlen))
d3372da9 941 {
02a705ad
FT
942 sksetstate(sk, SOCK_EST);
943 linksock(&cbatch, sk->back);
944 return(sk->back);
d3372da9 945 }
946 if(errno == EINPROGRESS)
947 {
02a705ad
FT
948 sksetstate(sk, SOCK_SYN);
949 return(sk->back);
d3372da9 950 }
951 putsock(sk);
952 return(NULL);
953 }
954 errno = EOPNOTSUPP;
955 return(NULL);
956}
957
02a705ad 958static void acceptunix(struct ufd *ufd)
9e5f2b29 959{
960 int buf;
961
962 buf = 1;
a8c5ada6 963#if UNIX_AUTH_STYLE == 1
02a705ad
FT
964 if(setsockopt(ufd->fd, SOL_SOCKET, SO_PASSCRED, &buf, sizeof(buf)) < 0)
965 flog(LOG_WARNING, "could not enable SO_PASSCRED on Unix socket %i: %s", ufd->fd, strerror(errno));
a8c5ada6 966#elif UNIX_AUTH_STYLE == 2
02a705ad 967 if(getpeereid(ufd->fd, &ufd->d.s.ucred.uid, &ufd->d.s.ucred.gid) < 0)
a8c5ada6 968 {
02a705ad
FT
969 flog(LOG_WARNING, "could not get peer creds on Unix socket %i: %s", ufd->fd, strerror(errno));
970 ufd->d.s.ucred.uid = -1;
971 ufd->d.s.ucred.gid = -1;
a8c5ada6 972 }
973#endif
9e5f2b29 974}
975
f96d0914
FT
976static void runbatches(void)
977{
978 struct scons *sc, *nsc;
979
980 for(sc = cbatch, cbatch = NULL; sc; sc = nsc) {
981 nsc = sc->n;
5afb0d14
FT
982 if(sc->s->conncb != NULL)
983 sc->s->conncb(sc->s, 0, sc->s->data);
834b0aa6 984 putsock(sc->s);
f96d0914
FT
985 free(sc);
986 }
987 for(sc = rbatch, rbatch = NULL; sc; sc = nsc) {
988 nsc = sc->n;
5afb0d14
FT
989 if(sc->s->readcb != NULL)
990 sc->s->readcb(sc->s, sc->s->data);
991 if((sockgetdatalen(sc->s) == 0) && (sc->s->eos == 1)) {
992 if(sc->s->errcb != NULL)
993 sc->s->errcb(sc->s, 0, sc->s->data);
994 sc->s->eos = 2;
995 }
834b0aa6 996 putsock(sc->s);
f96d0914
FT
997 free(sc);
998 }
999 for(sc = wbatch, wbatch = NULL; sc; sc = nsc) {
1000 nsc = sc->n;
5afb0d14
FT
1001 if(sc->s->writecb != NULL)
1002 sc->s->writecb(sc->s, sc->s->data);
834b0aa6 1003 putsock(sc->s);
f96d0914
FT
1004 free(sc);
1005 }
1006}
1007
5afb0d14
FT
1008static void cleansocks(void)
1009{
1010 struct ufd *ufd, *next;
1011
1012 for(ufd = ufds; ufd != NULL; ufd = next) {
1013 next = ufd->next;
834b0aa6 1014 if(ufd->sk && ((ufd->fd < 0) || (sockgetdatalen(ufd->sk) == 0))) {
5afb0d14
FT
1015 if(ufd->sk->eos == 1) {
1016 ufd->sk->eos = 2;
1017 closeufd(ufd);
1018 closesock(ufd->sk);
1019 }
1020 if((ufd->sk->refcount == 1) && (ufd->sk->back->refcount == 0)) {
1021 freeufd(ufd);
1022 continue;
1023 }
1024 }
1025 }
1026}
1027
d3372da9 1028int pollsocks(int timeout)
1029{
02a705ad 1030 int ret;
eaa35963 1031 socklen_t retlen;
72a7835e
FT
1032 int newfd, maxfd;
1033 fd_set rfds, wfds, efds;
5afb0d14 1034 struct ufd *ufd, *nufd;
02a705ad 1035 struct socket *nsk;
d3372da9 1036 struct sockaddr_storage ss;
1037 socklen_t sslen;
72a7835e 1038 struct timeval tv;
d3372da9 1039
5afb0d14 1040 cleansocks();
72a7835e
FT
1041 FD_ZERO(&rfds);
1042 FD_ZERO(&wfds);
1043 FD_ZERO(&efds);
02a705ad
FT
1044 for(maxfd = 0, ufd = ufds; ufd != NULL; ufd = ufd->next) {
1045 if(ufd->fd < 0)
d3372da9 1046 continue;
81286a22 1047 if(!ufd->ignread && ((ufd->sk == NULL) || (sockqueueleft(ufd->sk) > 0)))
02a705ad
FT
1048 FD_SET(ufd->fd, &rfds);
1049 if(ufd->sk != NULL) {
1050 if(sockgetdatalen(ufd->sk) > 0)
1051 FD_SET(ufd->fd, &wfds);
1052 else if(ufd->sk->state == SOCK_SYN)
1053 FD_SET(ufd->fd, &wfds);
1054 }
1055 FD_SET(ufd->fd, &efds);
1056 if(ufd->fd > maxfd)
1057 maxfd = ufd->fd;
d3372da9 1058 }
5afb0d14
FT
1059 if(rbatch || wbatch || cbatch)
1060 timeout = 0;
72a7835e
FT
1061 tv.tv_sec = timeout / 1000;
1062 tv.tv_usec = (timeout % 1000) * 1000;
1063 ret = select(maxfd + 1, &rfds, &wfds, &efds, (timeout < 0)?NULL:&tv);
02a705ad
FT
1064 if(ret < 0) {
1065 if(errno != EINTR) {
72a7835e 1066 flog(LOG_CRIT, "pollsocks: select errored out: %s", strerror(errno));
d3372da9 1067 /* To avoid CPU hogging in case it's bad, which it
1068 * probably is. */
1069 sleep(1);
1070 }
d3372da9 1071 return(1);
1072 }
5afb0d14 1073 for(ufd = ufds; ufd != NULL; ufd = ufd->next) {
02a705ad
FT
1074 if(ufd->sk < 0)
1075 continue;
1076 if(ufd->type == UFD_LISTEN) {
1077 if(FD_ISSET(ufd->fd, &rfds)) {
d3372da9 1078 sslen = sizeof(ss);
02a705ad
FT
1079 if((newfd = accept(ufd->fd, (struct sockaddr *)&ss, &sslen)) < 0) {
1080 if(ufd->d.l.lp->errcb != NULL)
1081 ufd->d.l.lp->errcb(ufd->d.l.lp, errno, ufd->d.l.lp->data);
cab0b442 1082 }
02a705ad
FT
1083 nsk = sockpair(0);
1084 nufd = mkufd(newfd, UFD_SOCK, nsk);
1085 nufd->d.s.family = ufd->d.l.family;
1086 sksetstate(nsk, SOCK_EST);
1087 memcpy(nufd->d.s.remote = smalloc(sslen), &ss, sslen);
1088 nufd->d.s.remotelen = sslen;
9e5f2b29 1089 if(ss.ss_family == PF_UNIX)
02a705ad
FT
1090 acceptunix(nufd);
1091 if(ufd->d.l.lp->acceptcb != NULL)
5afb0d14 1092 ufd->d.l.lp->acceptcb(ufd->d.l.lp, nsk->back, ufd->d.l.lp->data);
02a705ad 1093 putsock(nsk);
d3372da9 1094 }
02a705ad 1095 if(FD_ISSET(ufd->fd, &efds)) {
d3372da9 1096 retlen = sizeof(ret);
02a705ad
FT
1097 getsockopt(ufd->fd, SOL_SOCKET, SO_ERROR, &ret, &retlen);
1098 if(ufd->d.l.lp->errcb != NULL)
1099 ufd->d.l.lp->errcb(ufd->d.l.lp, ret, ufd->d.l.lp->data);
d3372da9 1100 continue;
1101 }
02a705ad
FT
1102 } else {
1103 if(ufd->sk->state == SOCK_SYN) {
1104 if(FD_ISSET(ufd->fd, &efds)) {
1105 retlen = sizeof(ret);
1106 getsockopt(ufd->fd, SOL_SOCKET, SO_ERROR, &ret, &retlen);
1107 if(ufd->sk->back->conncb != NULL)
1108 ufd->sk->back->conncb(ufd->sk->back, ret, ufd->sk->back->data);
1109 closeufd(ufd);
1110 continue;
1111 }
1112 if(FD_ISSET(ufd->fd, &rfds) || FD_ISSET(ufd->fd, &wfds)) {
1113 sksetstate(ufd->sk, SOCK_EST);
1114 linksock(&cbatch, ufd->sk->back);
1115 }
1116 } else if(ufd->sk->state == SOCK_EST) {
1117 if(FD_ISSET(ufd->fd, &efds)) {
1118 retlen = sizeof(ret);
1119 getsockopt(ufd->fd, SOL_SOCKET, SO_ERROR, &ret, &retlen);
1120 sockerror(ufd->sk, ret);
1121 closeufd(ufd);
1122 continue;
1123 }
1124 if(FD_ISSET(ufd->fd, &rfds))
1125 sockrecv(ufd);
5afb0d14
FT
1126 if(ufd->fd == -1)
1127 continue;
89d650f1
FT
1128 if(FD_ISSET(ufd->fd, &wfds)) {
1129 if(sockflush(ufd)) {
1130 sockerror(ufd->sk, errno);
1131 closeufd(ufd);
1132 continue;
1133 }
1134 }
d3372da9 1135 }
d3372da9 1136 }
d3372da9 1137 }
f96d0914 1138 runbatches();
5afb0d14 1139 cleansocks();
d3372da9 1140 return(1);
1141}
1142
f96d0914
FT
1143static struct ufd *getskufd(struct socket *sk)
1144{
1145 while(1) {
1146 if(sk->back->ufd != NULL)
1147 return(sk->back->ufd);
1148 if((sk = sk->back->pnext) == NULL)
1149 break;
1150 }
1151 return(NULL);
1152}
1153
d3372da9 1154int socksettos(struct socket *sk, int tos)
1155{
b020fb3d 1156 int buf;
f96d0914 1157 struct ufd *ufd;
b020fb3d 1158
f96d0914
FT
1159 ufd = getskufd(sk);
1160 if(ufd->type != UFD_SOCK) {
1161 errno = EOPNOTSUPP;
1162 return(-1);
1163 }
1164 if(ufd->d.s.family == AF_UNIX)
7ab36fbd 1165 return(0); /* Unix sockets are always perfect. :) */
f96d0914 1166 if(ufd->d.s.family == AF_INET)
d3372da9 1167 {
b020fb3d 1168 switch(tos)
1169 {
b198bed6 1170 case 0:
1171 buf = 0;
1172 break;
b020fb3d 1173 case SOCK_TOS_MINCOST:
18c1ae1d 1174 buf = 0x02;
b020fb3d 1175 break;
1176 case SOCK_TOS_MAXREL:
18c1ae1d 1177 buf = 0x04;
b020fb3d 1178 break;
1179 case SOCK_TOS_MAXTP:
18c1ae1d 1180 buf = 0x08;
b020fb3d 1181 break;
1182 case SOCK_TOS_MINDELAY:
18c1ae1d 1183 buf = 0x10;
b020fb3d 1184 break;
1185 default:
1186 flog(LOG_WARNING, "attempted to set unknown TOS value %i to IPv4 sock", tos);
1187 return(-1);
1188 }
f96d0914 1189 if(setsockopt(ufd->fd, IPPROTO_IP, IP_TOS, &buf, sizeof(buf)) < 0)
d3372da9 1190 {
1191 flog(LOG_WARNING, "could not set sock TOS to %i: %s", tos, strerror(errno));
1192 return(-1);
1193 }
1194 return(0);
1195 }
f96d0914 1196 if(ufd->d.s.family == AF_INET6)
b020fb3d 1197 {
1198 switch(tos)
1199 {
b198bed6 1200 case 0:
1201 buf = 0;
b020fb3d 1202 case SOCK_TOS_MINCOST:
1203 buf = confgetint("net", "diffserv-mincost");
1204 break;
1205 case SOCK_TOS_MAXREL:
1206 buf = confgetint("net", "diffserv-maxrel");
1207 break;
1208 case SOCK_TOS_MAXTP:
1209 buf = confgetint("net", "diffserv-maxtp");
1210 break;
1211 case SOCK_TOS_MINDELAY:
1212 buf = confgetint("net", "diffserv-mindelay");
1213 break;
1214 default:
1215 flog(LOG_WARNING, "attempted to set unknown TOS value %i to IPv4 sock", tos);
1216 return(-1);
1217 }
1218 /*
1219 On Linux, the API IPv6 flow label management doesn't seem to
1220 be entirely complete, so I guess this will have to wait.
1221
1222 if(setsockopt(...) < 0)
1223 {
1224 flog(LOG_WARNING, "could not set sock traffic class to %i: %s", tos, strerror(errno));
1225 return(-1);
1226 }
1227 */
1228 return(0);
1229 }
f96d0914 1230 flog(LOG_WARNING, "could not set TOS on sock of family %i", ufd->d.s.family);
d3372da9 1231 return(1);
1232}
1233
1234struct resolvedata
1235{
1236 int fd;
1237 void (*callback)(struct sockaddr *addr, int addrlen, void *data);
1238 void *data;
1239 struct sockaddr_storage addr;
1240 int addrlen;
1241};
1242
1243static void resolvecb(pid_t pid, int status, struct resolvedata *data)
1244{
1245 static char buf[80];
1246 int ret;
1247 struct sockaddr_in *ipv4;
1248
1249 if(!status)
1250 {
1251 if((ret = read(data->fd, buf, sizeof(buf))) != 4)
1252 {
b4d16e90 1253 errno = ENOENT;
d3372da9 1254 data->callback(NULL, 0, data->data);
1255 } else {
1256 ipv4 = (struct sockaddr_in *)&data->addr;
1257 memcpy(&ipv4->sin_addr, buf, 4);
1258 data->callback((struct sockaddr *)ipv4, sizeof(*ipv4), data->data);
1259 }
1260 } else {
b4d16e90 1261 errno = ENOENT;
d3372da9 1262 data->callback(NULL, 0, data->data);
1263 }
1264 close(data->fd);
1265 free(data);
1266}
1267
1268int netresolve(char *addr, void (*callback)(struct sockaddr *addr, int addrlen, void *data), void *data)
1269{
1270 int i;
1271 char *p;
1272 int port;
1273 int pfd[2];
1274 pid_t child;
1275 struct resolvedata *rdata;
1276 struct sockaddr_in ipv4;
1277 struct hostent *he;
1278 sigset_t sigset;
1279
1280 /* IPv4 */
1281 port = -1;
1282 if((p = strchr(addr, ':')) != NULL)
1283 {
1284 *p = 0;
1285 port = atoi(p + 1);
1286 }
1287 ipv4.sin_family = AF_INET;
1288 ipv4.sin_port = htons(port);
1289 if(inet_aton(addr, &ipv4.sin_addr))
1290 {
1291 callback((struct sockaddr *)&ipv4, sizeof(ipv4), data);
1292 } else {
1293 sigemptyset(&sigset);
1294 sigaddset(&sigset, SIGCHLD);
1295 sigprocmask(SIG_BLOCK, &sigset, NULL);
1296 if((pipe(pfd) < 0) || ((child = fork()) < 0))
1297 {
1298 sigprocmask(SIG_UNBLOCK, &sigset, NULL);
1299 return(-1);
1300 }
1301 if(child == 0)
1302 {
1303 sigprocmask(SIG_UNBLOCK, &sigset, NULL);
1304 for(i = 3; i < FD_SETSIZE; i++)
1305 {
1306 if(i != pfd[1])
1307 close(i);
1308 }
1309 signal(SIGALRM, SIG_DFL);
1310 alarm(30);
1311 if((he = gethostbyname(addr)) == NULL)
1312 exit(1);
1313 write(pfd[1], he->h_addr_list[0], 4);
1314 exit(0);
1315 } else {
1316 close(pfd[1]);
1317 fcntl(pfd[0], F_SETFL, fcntl(pfd[0], F_GETFL) | O_NONBLOCK);
1318 rdata = smalloc(sizeof(*rdata));
1319 rdata->fd = pfd[0];
1320 rdata->callback = callback;
1321 rdata->data = data;
1322 memcpy(&rdata->addr, &ipv4, rdata->addrlen = sizeof(ipv4));
1323 childcallback(child, (void (*)(pid_t, int, void *))resolvecb, rdata);
1324 sigprocmask(SIG_UNBLOCK, &sigset, NULL);
1325 return(1);
1326 }
1327 }
1328 return(0);
1329}
1330
f96d0914 1331static int getlocalname(int fd, struct sockaddr **namebuf, socklen_t *lenbuf)
d3372da9 1332{
1333 socklen_t len;
1334 struct sockaddr_storage name;
1335
1336 *namebuf = NULL;
f96d0914 1337 if(fd < 0)
d3372da9 1338 return(-1);
1339 len = sizeof(name);
f96d0914 1340 if(getsockname(fd, (struct sockaddr *)&name, &len) < 0)
d3372da9 1341 {
6fb0c6ac 1342 flog(LOG_ERR, "BUG: alive socket with dead fd in sockgetlocalname (%s)", strerror(errno));
d3372da9 1343 return(-1);
1344 }
1345 *namebuf = memcpy(smalloc(len), &name, len);
b870805e 1346 if(lenbuf != NULL)
1347 *lenbuf = len;
d3372da9 1348 return(0);
1349}
1350
f96d0914
FT
1351int lstgetlocalname(struct lport *lp, struct sockaddr **namebuf, socklen_t *lenbuf)
1352{
1353 struct ufd *ufd;
1354
1355 ufd = lp->ufd;
1356 return(getlocalname(ufd->fd, namebuf, lenbuf));
1357}
1358
1359int sockgetlocalname(struct socket *sk, struct sockaddr **namebuf, socklen_t *lenbuf)
1360{
1361 struct ufd *ufd;
1362
1363 ufd = getskufd(sk);
1364 if(ufd->type != UFD_SOCK) {
1365 errno = EOPNOTSUPP;
1366 return(-1);
1367 }
1368 return(getlocalname(ufd->fd, namebuf, lenbuf));
1369}
1370
eaa35963 1371static void sethostaddr(struct sockaddr *dst, struct sockaddr *src)
1372{
1373 if(dst->sa_family != src->sa_family)
1374 {
1375 flog(LOG_ERR, "BUG: non-matching socket families in sethostaddr (%i -> %i)", src->sa_family, dst->sa_family);
1376 return;
1377 }
1378 switch(src->sa_family)
1379 {
1380 case AF_INET:
1381 ((struct sockaddr_in *)dst)->sin_addr = ((struct sockaddr_in *)src)->sin_addr;
1382 break;
1383 case AF_INET6:
1384 ((struct sockaddr_in6 *)dst)->sin6_addr = ((struct sockaddr_in6 *)src)->sin6_addr;
1385 break;
1386 default:
1387 flog(LOG_WARNING, "sethostaddr unimplemented for family %i", src->sa_family);
1388 break;
1389 }
1390}
1391
1392static int makepublic(struct sockaddr *addr)
d3372da9 1393{
6fb0c6ac 1394 int ret;
eaa35963 1395 socklen_t plen;
1396 struct sockaddr *pname;
1397
1398 if((ret = getpublicaddr(addr->sa_family, &pname, &plen)) < 0)
1399 {
1400 flog(LOG_ERR, "could not get public address: %s", strerror(errno));
1401 return(-1);
1402 }
1403 if(ret)
1404 return(0);
1405 sethostaddr(addr, pname);
1406 free(pname);
1407 return(0);
1408}
1409
f96d0914 1410static int getremotename(int fd, struct sockaddr **namebuf, socklen_t *lenbuf)
eaa35963 1411{
d3372da9 1412 socklen_t len;
6fb0c6ac 1413 struct sockaddr *name;
f96d0914
FT
1414
1415 switch(confgetint("net", "mode")) {
d3372da9 1416 case 0:
1417 *namebuf = NULL;
f96d0914 1418 if(!getlocalname(fd, &name, &len)) {
6fb0c6ac 1419 *namebuf = name;
1420 *lenbuf = len;
eaa35963 1421 makepublic(name);
6fb0c6ac 1422 return(0);
1423 }
1424 flog(LOG_ERR, "could not get remotely accessible name by any means");
1425 return(-1);
d3372da9 1426 case 1:
1427 errno = EOPNOTSUPP;
1428 return(-1);
1429 default:
1430 flog(LOG_CRIT, "unknown net mode %i active", confgetint("net", "mode"));
1431 errno = EOPNOTSUPP;
1432 return(-1);
1433 }
1434}
1435
f96d0914
FT
1436int sockgetremotename(struct socket *sk, struct sockaddr **namebuf, socklen_t *lenbuf)
1437{
1438 struct ufd *ufd;
1439
1440 ufd = getskufd(sk);
1441 if(ufd->type != UFD_SOCK) {
1442 errno = EOPNOTSUPP;
1443 return(-1);
1444 }
1445 if(ufd->fd < 0) {
1446 errno = EBADF;
1447 return(-1);
1448 }
1449 return(getremotename(ufd->fd, namebuf, lenbuf));
1450}
1451
1452int lstgetremotename(struct lport *lp, struct sockaddr **namebuf, socklen_t *lenbuf)
1453{
1454 struct ufd *ufd;
1455
1456 ufd = lp->ufd;
1457 return(getremotename(ufd->fd, namebuf, lenbuf));
1458}
1459
d29853b1 1460int sockgetremotename2(struct socket *sk, struct socket *sk2, struct sockaddr **namebuf, socklen_t *lenbuf)
1461{
1462 struct sockaddr *name1, *name2;
1463 socklen_t len1, len2;
f96d0914
FT
1464 struct ufd *ufd1, *ufd2;
1465
1466 ufd1 = getskufd(sk);
1467 ufd2 = getskufd(sk2);
1468 if((ufd1->type != UFD_SOCK) || (ufd2->type != UFD_SOCK)) {
1469 errno = EOPNOTSUPP;
1470 return(-1);
1471 }
1472 if(ufd1->d.s.family != ufd2->d.s.family)
1473 {
1474 flog(LOG_ERR, "using sockgetremotename2 with sockets of differing family: %i %i", ufd1->d.s.family, ufd2->d.s.family);
1475 return(-1);
1476 }
1477 if(getremotename(ufd1->fd, &name1, &len1))
1478 return(-1);
1479 if(getremotename(ufd2->fd, &name2, &len2)) {
1480 free(name1);
1481 return(-1);
1482 }
1483 sethostaddr(name1, name2);
1484 free(name2);
1485 *namebuf = name1;
1486 *lenbuf = len1;
1487 return(0);
1488}
1489
1490int lstgetremotename2(struct lport *lp, struct socket *sk2, struct sockaddr **namebuf, socklen_t *lenbuf)
1491{
1492 struct sockaddr *name1, *name2;
1493 socklen_t len1, len2;
1494 struct ufd *ufd1, *ufd2;
d29853b1 1495
f96d0914
FT
1496 ufd1 = lp->ufd;
1497 ufd2 = getskufd(sk2);
1498 if(ufd2->type != UFD_SOCK) {
1499 errno = EOPNOTSUPP;
1500 return(-1);
1501 }
d334fbb5 1502 if(ufd1->d.l.family != ufd2->d.s.family)
d29853b1 1503 {
d334fbb5 1504 flog(LOG_ERR, "using lstgetremotename2 with sockets of differing family: %i %i", ufd1->d.l.family, ufd2->d.s.family);
d29853b1 1505 return(-1);
1506 }
f96d0914 1507 if(getremotename(ufd1->fd, &name1, &len1))
d29853b1 1508 return(-1);
f96d0914 1509 if(getremotename(ufd2->fd, &name2, &len2)) {
d29853b1 1510 free(name1);
1511 return(-1);
1512 }
1513 sethostaddr(name1, name2);
1514 free(name2);
1515 *namebuf = name1;
1516 *lenbuf = len1;
1517 return(0);
1518}
1519
f96d0914
FT
1520int getucred(struct socket *sk, uid_t *uid, gid_t *gid)
1521{
1522 struct ufd *ufd;
1523
1524 ufd = getskufd(sk);
1525 if(ufd->type != UFD_SOCK) {
1526 errno = EOPNOTSUPP;
1527 return(-1);
1528 }
1529 if(ufd->d.s.family != AF_UNIX) {
1530 errno = EOPNOTSUPP;
1531 return(-1);
1532 }
1533 *uid = ufd->d.s.ucred.uid;
1534 *gid = ufd->d.s.ucred.gid;
1535 return(0);
1536}
1537
81286a22
FT
1538/* void sockblock(struct socket *sk, int block) */
1539/* { */
1540/* struct ufd *ufd; */
f96d0914 1541
81286a22
FT
1542/* ufd = getskufd(sk); */
1543/* ufd->ignread = block; */
1544/* } */
f96d0914
FT
1545
1546int sockfamily(struct socket *sk)
1547{
1548 struct ufd *ufd;
1549
1550 ufd = getskufd(sk);
1551 if(ufd->type != UFD_SOCK) {
1552 errno = EOPNOTSUPP;
1553 return(-1);
1554 }
1555 return(ufd->d.s.family);
02a705ad
FT
1556}
1557
1558int sockpeeraddr(struct socket *sk, struct sockaddr **namebuf, socklen_t *lenbuf)
1559{
f96d0914
FT
1560 struct ufd *ufd;
1561
1562 ufd = getskufd(sk);
1563 if(ufd->type != UFD_SOCK) {
1564 errno = EOPNOTSUPP;
1565 return(-1);
1566 }
1567 if(ufd->d.s.remote == NULL)
1568 return(-1);
1569 *namebuf = memcpy(smalloc(ufd->d.s.remotelen), ufd->d.s.remote, ufd->d.s.remotelen);
1570 if(lenbuf != NULL)
1571 *lenbuf = ufd->d.s.remotelen;
1572 return(0);
02a705ad
FT
1573}
1574
1575char *formatsockpeer(struct socket *sk)
1576{
1577 struct sockaddr *name;
1578 socklen_t nlen;
1579 char *ret;
1580
1581 if(sockpeeraddr(sk, &name, &nlen))
1582 return(NULL);
1583 ret = formataddress(name, nlen);
1584 free(name);
1585 return(ret);
1586}
1587
99a28d47 1588int addreq(struct sockaddr *x, struct sockaddr *y)
1589{
1590 struct sockaddr_un *u1, *u2;
1591 struct sockaddr_in *n1, *n2;
1592#ifdef HAVE_IPV6
1593 struct sockaddr_in6 *s1, *s2;
1594#endif
1595
1596 if(x->sa_family != y->sa_family)
1597 return(0);
1598 switch(x->sa_family) {
1599 case AF_UNIX:
1600 u1 = (struct sockaddr_un *)x; u2 = (struct sockaddr_un *)y;
1601 if(strncmp(u1->sun_path, u2->sun_path, sizeof(u1->sun_path)))
1602 return(0);
1603 break;
1604 case AF_INET:
1605 n1 = (struct sockaddr_in *)x; n2 = (struct sockaddr_in *)y;
1606 if(n1->sin_port != n2->sin_port)
1607 return(0);
1608 if(n1->sin_addr.s_addr != n2->sin_addr.s_addr)
1609 return(0);
1610 break;
c8a729d7 1611#ifdef HAVE_IPV6
99a28d47 1612 case AF_INET6:
1613 s1 = (struct sockaddr_in6 *)x; s2 = (struct sockaddr_in6 *)y;
1614 if(s1->sin6_port != s2->sin6_port)
1615 return(0);
1616 if(memcmp(s1->sin6_addr.s6_addr, s2->sin6_addr.s6_addr, sizeof(s1->sin6_addr.s6_addr)))
1617 return(0);
1618 break;
c8a729d7 1619#endif
99a28d47 1620 }
1621 return(1);
1622}
1623
d3372da9 1624char *formataddress(struct sockaddr *arg, socklen_t arglen)
1625{
d3372da9 1626 struct sockaddr_in *ipv4;
1627#ifdef HAVE_IPV6
1628 struct sockaddr_in6 *ipv6;
1629#endif
1630 static char *ret = NULL;
1631 char buf[1024];
1632
1633 if(ret != NULL)
1634 free(ret);
1635 ret = NULL;
1636 switch(arg->sa_family)
1637 {
1638 case AF_UNIX:
6a97462e 1639 ret = sstrdup("Unix socket");
d3372da9 1640 break;
1641 case AF_INET:
1642 ipv4 = (struct sockaddr_in *)arg;
1643 if(inet_ntop(AF_INET, &ipv4->sin_addr, buf, sizeof(buf)) == NULL)
1644 return(NULL);
1645 ret = sprintf2("%s:%i", buf, (int)ntohs(ipv4->sin_port));
1646 break;
1647#ifdef HAVE_IPV6
1648 case AF_INET6:
1649 ipv6 = (struct sockaddr_in6 *)arg;
1650 if(inet_ntop(AF_INET6, &ipv6->sin6_addr, buf, sizeof(buf)) == NULL)
1651 return(NULL);
fc7f7735 1652 ret = sprintf2("[%s]:%i", buf, (int)ntohs(ipv6->sin6_port));
d3372da9 1653 break;
1654#endif
1655 default:
1656 errno = EPFNOSUPPORT;
1657 break;
1658 }
1659 return(ret);
1660}
1661
1662#if 0
1663
1664/*
1665 * It was very nice to use this, but it seems
1666 * to mess things up, so I guess it has to go... :-(
1667 */
1668
1669static int formataddress(FILE *stream, const struct printf_info *info, const void *const *args)
1670{
1671 struct sockaddr *arg;
1672 socklen_t arglen;
1673 struct sockaddr_un *UNIX; /* Some wise guy has #defined unix with
1674 * lowercase letters to 1, so I do this
1675 * instead. */
1676 struct sockaddr_in *ipv4;
1677 int ret;
1678
1679 arg = *(struct sockaddr **)(args[0]);
1680 arglen = *(socklen_t *)(args[1]);
1681 switch(arg->sa_family)
1682 {
1683 case AF_UNIX:
1684 UNIX = (struct sockaddr_un *)arg;
1685 ret = fprintf(stream, "%s", UNIX->sun_path);
1686 break;
1687 case AF_INET:
1688 ipv4 = (struct sockaddr_in *)arg;
1689 ret = fprintf(stream, "%s:%i", inet_ntoa(ipv4->sin_addr), (int)ntohs(ipv4->sin_port));
1690 break;
1691 default:
1692 ret = -1;
1693 errno = EPFNOSUPPORT;
1694 break;
1695 }
1696 return(ret);
1697}
1698
1699static int formataddress_arginfo(const struct printf_info *info, size_t n, int *argtypes)
1700{
1701 if(n > 0)
1702 argtypes[0] = PA_POINTER;
1703 if(n > 1)
1704 argtypes[1] = PA_INT; /* Sources tell me that socklen_t _must_
1705 * be an int, so I guess this should be
1706 * safe. */
1707 return(2);
1708}
1709#endif
1710
1711static int init(int hup)
1712{
1713 if(!hup)
1714 {
1715 /*
1716 if(register_printf_function('N', formataddress, formataddress_arginfo))
1717 {
1718 flog(LOG_CRIT, "could not register printf handler %%N: %s", strerror(errno));
1719 return(1);
1720 }
1721 */
1722 }
1723 return(0);
1724}
1725
1726static void terminate(void)
1727{
f96d0914
FT
1728 /*
1729 while(ufds != NULL)
1730 freeufd(ufds);
1731 */
d3372da9 1732}
1733
1734static struct module me =
1735{
1736 .name = "net",
1737 .conf =
1738 {
1739 .vars = myvars
1740 },
1741 .init = init,
1742 .terminate = terminate
1743};
1744
1745MODULE(me)