daemon/net.c

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