busybox/networking/inetd.c
<<
>>
Prefs
   1/* vi: set sw=4 ts=4: */
   2/*      $Slackware: inetd.c 1.79s 2001/02/06 13:18:00 volkerdi Exp $    */
   3/*      $OpenBSD: inetd.c,v 1.79 2001/01/30 08:30:57 deraadt Exp $      */
   4/*      $NetBSD: inetd.c,v 1.11 1996/02/22 11:14:41 mycroft Exp $       */
   5/* Busybox port by Vladimir Oleynik (C) 2001-2005 <dzo@simtreas.ru>     */
   6/* IPv6 support, many bug fixes by Denys Vlasenko (c) 2008 */
   7/*
   8 * Copyright (c) 1983,1991 The Regents of the University of California.
   9 * All rights reserved.
  10 *
  11 * Redistribution and use in source and binary forms, with or without
  12 * modification, are permitted provided that the following conditions
  13 * are met:
  14 * 1. Redistributions of source code must retain the above copyright
  15 *    notice, this list of conditions and the following disclaimer.
  16 * 2. Redistributions in binary form must reproduce the above copyright
  17 *    notice, this list of conditions and the following disclaimer in the
  18 *    documentation and/or other materials provided with the distribution.
  19 * 3. All advertising materials mentioning features or use of this software
  20 *    must display the following acknowledgement:
  21 *      This product includes software developed by the University of
  22 *      California, Berkeley and its contributors.
  23 * 4. Neither the name of the University nor the names of its contributors
  24 *    may be used to endorse or promote products derived from this software
  25 *    without specific prior written permission.
  26 *
  27 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS "AS IS" AND
  28 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  29 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  30 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  31 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  32 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  33 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  34 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  35 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  36 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  37 * SUCH DAMAGE.
  38 */
  39
  40/* Inetd - Internet super-server
  41 *
  42 * This program invokes configured services when a connection
  43 * from a peer is established or a datagram arrives.
  44 * Connection-oriented services are invoked each time a
  45 * connection is made, by creating a process.  This process
  46 * is passed the connection as file descriptor 0 and is
  47 * expected to do a getpeername to find out peer's host
  48 * and port.
  49 * Datagram oriented services are invoked when a datagram
  50 * arrives; a process is created and passed a pending message
  51 * on file descriptor 0. peer's address can be obtained
  52 * using recvfrom.
  53 *
  54 * Inetd uses a configuration file which is read at startup
  55 * and, possibly, at some later time in response to a hangup signal.
  56 * The configuration file is "free format" with fields given in the
  57 * order shown below.  Continuation lines for an entry must begin with
  58 * a space or tab.  All fields must be present in each entry.
  59 *
  60 *      service_name                    must be in /etc/services
  61 *      socket_type                     stream/dgram/raw/rdm/seqpacket
  62 *      protocol                        must be in /etc/protocols
  63 *                                      (usually "tcp" or "udp")
  64 *      wait/nowait[.max]               single-threaded/multi-threaded, max #
  65 *      user[.group] or user[:group]    user/group to run daemon as
  66 *      server_program                  full path name
  67 *      server_program_arguments        maximum of MAXARGS (20)
  68 *
  69 * For RPC services
  70 *      service_name/version            must be in /etc/rpc
  71 *      socket_type                     stream/dgram/raw/rdm/seqpacket
  72 *      rpc/protocol                    "rpc/tcp" etc
  73 *      wait/nowait[.max]               single-threaded/multi-threaded
  74 *      user[.group] or user[:group]    user to run daemon as
  75 *      server_program                  full path name
  76 *      server_program_arguments        maximum of MAXARGS (20)
  77 *
  78 * For non-RPC services, the "service name" can be of the form
  79 * hostaddress:servicename, in which case the hostaddress is used
  80 * as the host portion of the address to listen on.  If hostaddress
  81 * consists of a single '*' character, INADDR_ANY is used.
  82 *
  83 * A line can also consist of just
  84 *      hostaddress:
  85 * where hostaddress is as in the preceding paragraph.  Such a line must
  86 * have no further fields; the specified hostaddress is remembered and
  87 * used for all further lines that have no hostaddress specified,
  88 * until the next such line (or EOF).  (This is why * is provided to
  89 * allow explicit specification of INADDR_ANY.)  A line
  90 *      *:
  91 * is implicitly in effect at the beginning of the file.
  92 *
  93 * The hostaddress specifier may (and often will) contain dots;
  94 * the service name must not.
  95 *
  96 * For RPC services, host-address specifiers are accepted and will
  97 * work to some extent; however, because of limitations in the
  98 * portmapper interface, it will not work to try to give more than
  99 * one line for any given RPC service, even if the host-address
 100 * specifiers are different.
 101 *
 102 * Comment lines are indicated by a '#' in column 1.
 103 */
 104
 105/* inetd rules for passing file descriptors to children
 106 * (http://www.freebsd.org/cgi/man.cgi?query=inetd):
 107 *
 108 * The wait/nowait entry specifies whether the server that is invoked by
 109 * inetd will take over the socket associated with the service access point,
 110 * and thus whether inetd should wait for the server to exit before listen-
 111 * ing for new service requests.  Datagram servers must use "wait", as
 112 * they are always invoked with the original datagram socket bound to the
 113 * specified service address.  These servers must read at least one datagram
 114 * from the socket before exiting.  If a datagram server connects to its
 115 * peer, freeing the socket so inetd can receive further messages on the
 116 * socket, it is said to be a "multi-threaded" server; it should read one
 117 * datagram from the socket and create a new socket connected to the peer.
 118 * It should fork, and the parent should then exit to allow inetd to check
 119 * for new service requests to spawn new servers.  Datagram servers which
 120 * process all incoming datagrams on a socket and eventually time out are
 121 * said to be "single-threaded".  The comsat(8), biff(1) and talkd(8)
 122 * utilities are both examples of the latter type of datagram server.  The
 123 * tftpd(8) utility is an example of a multi-threaded datagram server.
 124 *
 125 * Servers using stream sockets generally are multi-threaded and use the
 126 * "nowait" entry. Connection requests for these services are accepted by
 127 * inetd, and the server is given only the newly-accepted socket connected
 128 * to a client of the service.  Most stream-based services operate in this
 129 * manner.  Stream-based servers that use "wait" are started with the lis-
 130 * tening service socket, and must accept at least one connection request
 131 * before exiting.  Such a server would normally accept and process incoming
 132 * connection requests until a timeout.
 133 */
 134
 135/* Despite of above doc saying that dgram services must use "wait",
 136 * "udp nowait" servers are implemented in busyboxed inetd.
 137 * IPv6 addresses are also implemented. However, they may look ugly -
 138 * ":::service..." means "address '::' (IPv6 wildcard addr)":"service"...
 139 * You have to put "tcp6"/"udp6" in protocol field to select IPv6.
 140 */
 141
 142/* Here's the scoop concerning the user[:group] feature:
 143 * 1) group is not specified:
 144 *      a) user = root: NO setuid() or setgid() is done
 145 *      b) other:       initgroups(name, primary group)
 146 *                      setgid(primary group as found in passwd)
 147 *                      setuid()
 148 * 2) group is specified:
 149 *      a) user = root: setgid(specified group)
 150 *                      NO initgroups()
 151 *                      NO setuid()
 152 *      b) other:       initgroups(name, specified group)
 153 *                      setgid(specified group)
 154 *                      setuid()
 155 */
 156
 157#include <syslog.h>
 158#include <sys/un.h>
 159
 160#include "libbb.h"
 161
 162#if ENABLE_FEATURE_INETD_RPC
 163#include <rpc/rpc.h>
 164#include <rpc/pmap_clnt.h>
 165#endif
 166
 167#if !BB_MMU
 168/* stream version of chargen is forking but not execing,
 169 * can't do that (easily) on NOMMU */
 170#undef  ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_CHARGEN
 171#define ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_CHARGEN 0
 172#endif
 173
 174#define _PATH_INETDPID  "/var/run/inetd.pid"
 175
 176#define CNT_INTERVAL    60      /* servers in CNT_INTERVAL sec. */
 177#define RETRYTIME       60      /* retry after bind or server fail */
 178
 179// TODO: explain, or get rid of setrlimit games
 180
 181#ifndef RLIMIT_NOFILE
 182#define RLIMIT_NOFILE   RLIMIT_OFILE
 183#endif
 184
 185#ifndef OPEN_MAX
 186#define OPEN_MAX        64
 187#endif
 188
 189/* Reserve some descriptors, 3 stdio + at least: 1 log, 1 conf. file */
 190#define FD_MARGIN       8
 191
 192#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_DISCARD \
 193 || ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_ECHO    \
 194 || ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_CHARGEN \
 195 || ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_TIME    \
 196 || ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_DAYTIME
 197# define INETD_BUILTINS_ENABLED
 198#endif
 199
 200typedef struct servtab_t {
 201        /* The most frequently referenced one: */
 202        int se_fd;                            /* open descriptor */
 203        /* NB: 'biggest fields last' saves on code size (~250 bytes) */
 204        /* [addr:]service socktype proto wait user[:group] prog [args] */
 205        char *se_local_hostname;              /* addr to listen on */
 206        char *se_service;                     /* "80" or "www" or "mount/2[-3]" */
 207        /* socktype is in se_socktype */      /* "stream" "dgram" "raw" "rdm" "seqpacket" */
 208        char *se_proto;                       /* "unix" or "[rpc/]tcp[6]" */
 209#if ENABLE_FEATURE_INETD_RPC
 210        int se_rpcprog;                       /* rpc program number */
 211        int se_rpcver_lo;                     /* rpc program lowest version */
 212        int se_rpcver_hi;                     /* rpc program highest version */
 213#define is_rpc_service(sep)       ((sep)->se_rpcver_lo != 0)
 214#else
 215#define is_rpc_service(sep)       0
 216#endif
 217        pid_t se_wait;                        /* 0:"nowait", 1:"wait", >1:"wait" */
 218                                              /* and waiting for this pid */
 219        socktype_t se_socktype;               /* SOCK_STREAM/DGRAM/RDM/... */
 220        family_t se_family;                   /* AF_UNIX/INET[6] */
 221        /* se_proto_no is used by RPC code only... hmm */
 222        smallint se_proto_no;                 /* IPPROTO_TCP/UDP, n/a for AF_UNIX */
 223        smallint se_checked;                  /* looked at during merge */
 224        unsigned se_max;                      /* allowed instances per minute */
 225        unsigned se_count;                    /* number started since se_time */
 226        unsigned se_time;                     /* when we started counting */
 227        char *se_user;                        /* user name to run as */
 228        char *se_group;                       /* group name to run as, can be NULL */
 229#ifdef INETD_BUILTINS_ENABLED
 230        const struct builtin *se_builtin;     /* if built-in, description */
 231#endif
 232        struct servtab_t *se_next;
 233        len_and_sockaddr *se_lsa;
 234        char *se_program;                     /* server program */
 235#define MAXARGV 20
 236        char *se_argv[MAXARGV + 1];           /* program arguments */
 237} servtab_t;
 238
 239#ifdef INETD_BUILTINS_ENABLED
 240/* Echo received data */
 241#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_ECHO
 242static void echo_stream(int, servtab_t *);
 243static void echo_dg(int, servtab_t *);
 244#endif
 245/* Internet /dev/null */
 246#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_DISCARD
 247static void discard_stream(int, servtab_t *);
 248static void discard_dg(int, servtab_t *);
 249#endif
 250/* Return 32 bit time since 1900 */
 251#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_TIME
 252static void machtime_stream(int, servtab_t *);
 253static void machtime_dg(int, servtab_t *);
 254#endif
 255/* Return human-readable time */
 256#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_DAYTIME
 257static void daytime_stream(int, servtab_t *);
 258static void daytime_dg(int, servtab_t *);
 259#endif
 260/* Familiar character generator */
 261#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_CHARGEN
 262static void chargen_stream(int, servtab_t *);
 263static void chargen_dg(int, servtab_t *);
 264#endif
 265
 266struct builtin {
 267        /* NB: not necessarily NUL terminated */
 268        char bi_service7[7];      /* internally provided service name */
 269        uint8_t bi_fork;          /* 1 if stream fn should run in child */
 270        void (*bi_stream_fn)(int, servtab_t *);
 271        void (*bi_dgram_fn)(int, servtab_t *);
 272};
 273
 274static const struct builtin builtins[] = {
 275#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_ECHO
 276        { "echo", 1, echo_stream, echo_dg },
 277#endif
 278#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_DISCARD
 279        { "discard", 1, discard_stream, discard_dg },
 280#endif
 281#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_CHARGEN
 282        { "chargen", 1, chargen_stream, chargen_dg },
 283#endif
 284#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_TIME
 285        { "time", 0, machtime_stream, machtime_dg },
 286#endif
 287#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_DAYTIME
 288        { "daytime", 0, daytime_stream, daytime_dg },
 289#endif
 290};
 291#endif /* INETD_BUILTINS_ENABLED */
 292
 293struct globals {
 294        rlim_t rlim_ofile_cur;
 295        struct rlimit rlim_ofile;
 296        servtab_t *serv_list;
 297        int global_queuelen;
 298        int maxsock;            /* max fd# in allsock, -1: unknown */
 299        /* whenever maxsock grows, prev_maxsock is set to new maxsock,
 300         * but if maxsock is set to -1, prev_maxsock is not changed */
 301        int prev_maxsock;
 302        unsigned max_concurrency;
 303        smallint alarm_armed;
 304        uid_t real_uid; /* user ID who ran us */
 305        const char *config_filename;
 306        parser_t *parser;
 307        char *default_local_hostname;
 308#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_CHARGEN
 309        char *end_ring;
 310        char *ring_pos;
 311        char ring[128];
 312#endif
 313        fd_set allsock;
 314        /* Used in next_line(), and as scratch read buffer */
 315        char line[256];          /* _at least_ 256, see LINE_SIZE */
 316};
 317#define G (*(struct globals*)&bb_common_bufsiz1)
 318enum { LINE_SIZE = COMMON_BUFSIZE - offsetof(struct globals, line) };
 319struct BUG_G_too_big {
 320        char BUG_G_too_big[sizeof(G) <= COMMON_BUFSIZE ? 1 : -1];
 321};
 322#define rlim_ofile_cur  (G.rlim_ofile_cur )
 323#define rlim_ofile      (G.rlim_ofile     )
 324#define serv_list       (G.serv_list      )
 325#define global_queuelen (G.global_queuelen)
 326#define maxsock         (G.maxsock        )
 327#define prev_maxsock    (G.prev_maxsock   )
 328#define max_concurrency (G.max_concurrency)
 329#define alarm_armed     (G.alarm_armed    )
 330#define real_uid        (G.real_uid       )
 331#define config_filename (G.config_filename)
 332#define parser          (G.parser         )
 333#define default_local_hostname (G.default_local_hostname)
 334#define first_ps_byte   (G.first_ps_byte  )
 335#define last_ps_byte    (G.last_ps_byte   )
 336#define end_ring        (G.end_ring       )
 337#define ring_pos        (G.ring_pos       )
 338#define ring            (G.ring           )
 339#define allsock         (G.allsock        )
 340#define line            (G.line           )
 341#define INIT_G() do { \
 342        rlim_ofile_cur = OPEN_MAX; \
 343        global_queuelen = 128; \
 344        config_filename = "/etc/inetd.conf"; \
 345} while (0)
 346
 347static void maybe_close(int fd)
 348{
 349        if (fd >= 0)
 350                close(fd);
 351}
 352
 353// TODO: move to libbb?
 354static len_and_sockaddr *xzalloc_lsa(int family)
 355{
 356        len_and_sockaddr *lsa;
 357        int sz;
 358
 359        sz = sizeof(struct sockaddr_in);
 360        if (family == AF_UNIX)
 361                sz = sizeof(struct sockaddr_un);
 362#if ENABLE_FEATURE_IPV6
 363        if (family == AF_INET6)
 364                sz = sizeof(struct sockaddr_in6);
 365#endif
 366        lsa = xzalloc(LSA_LEN_SIZE + sz);
 367        lsa->len = sz;
 368        lsa->u.sa.sa_family = family;
 369        return lsa;
 370}
 371
 372static void rearm_alarm(void)
 373{
 374        if (!alarm_armed) {
 375                alarm_armed = 1;
 376                alarm(RETRYTIME);
 377        }
 378}
 379
 380static void block_CHLD_HUP_ALRM(sigset_t *m)
 381{
 382        sigemptyset(m);
 383        sigaddset(m, SIGCHLD);
 384        sigaddset(m, SIGHUP);
 385        sigaddset(m, SIGALRM);
 386        sigprocmask(SIG_BLOCK, m, m); /* old sigmask is stored in m */
 387}
 388
 389static void restore_sigmask(sigset_t *m)
 390{
 391        sigprocmask(SIG_SETMASK, m, NULL);
 392}
 393
 394#if ENABLE_FEATURE_INETD_RPC
 395static void register_rpc(servtab_t *sep)
 396{
 397        int n;
 398        struct sockaddr_in ir_sin;
 399        socklen_t size;
 400
 401        size = sizeof(ir_sin);
 402        if (getsockname(sep->se_fd, (struct sockaddr *) &ir_sin, &size) < 0) {
 403                bb_perror_msg("getsockname");
 404                return;
 405        }
 406
 407        for (n = sep->se_rpcver_lo; n <= sep->se_rpcver_hi; n++) {
 408                pmap_unset(sep->se_rpcprog, n);
 409                if (!pmap_set(sep->se_rpcprog, n, sep->se_proto_no, ntohs(ir_sin.sin_port)))
 410                        bb_perror_msg("%s %s: pmap_set(%u,%u,%u,%u)",
 411                                sep->se_service, sep->se_proto,
 412                                sep->se_rpcprog, n, sep->se_proto_no, ntohs(ir_sin.sin_port));
 413        }
 414}
 415
 416static void unregister_rpc(servtab_t *sep)
 417{
 418        int n;
 419
 420        for (n = sep->se_rpcver_lo; n <= sep->se_rpcver_hi; n++) {
 421                if (!pmap_unset(sep->se_rpcprog, n))
 422                        bb_perror_msg("pmap_unset(%u,%u)", sep->se_rpcprog, n);
 423        }
 424}
 425#endif /* FEATURE_INETD_RPC */
 426
 427static void bump_nofile(void)
 428{
 429        enum { FD_CHUNK = 32 };
 430        struct rlimit rl;
 431
 432        /* Never fails under Linux (except if you pass it bad arguments) */
 433        getrlimit(RLIMIT_NOFILE, &rl);
 434        rl.rlim_cur = MIN(rl.rlim_max, rl.rlim_cur + FD_CHUNK);
 435        rl.rlim_cur = MIN(FD_SETSIZE, rl.rlim_cur + FD_CHUNK);
 436        if (rl.rlim_cur <= rlim_ofile_cur) {
 437                bb_error_msg("can't extend file limit, max = %d",
 438                                                (int) rl.rlim_cur);
 439                return;
 440        }
 441
 442        if (setrlimit(RLIMIT_NOFILE, &rl) < 0) {
 443                bb_perror_msg("setrlimit");
 444                return;
 445        }
 446
 447        rlim_ofile_cur = rl.rlim_cur;
 448}
 449
 450static void remove_fd_from_set(int fd)
 451{
 452        if (fd >= 0) {
 453                FD_CLR(fd, &allsock);
 454                maxsock = -1;
 455        }
 456}
 457
 458static void add_fd_to_set(int fd)
 459{
 460        if (fd >= 0) {
 461                FD_SET(fd, &allsock);
 462                if (maxsock >= 0 && fd > maxsock) {
 463                        prev_maxsock = maxsock = fd;
 464                        if ((rlim_t)fd > rlim_ofile_cur - FD_MARGIN)
 465                                bump_nofile();
 466                }
 467        }
 468}
 469
 470static void recalculate_maxsock(void)
 471{
 472        int fd = 0;
 473
 474        /* We may have no services, in this case maxsock should still be >= 0
 475         * (code elsewhere is not happy with maxsock == -1) */
 476        maxsock = 0;
 477        while (fd <= prev_maxsock) {
 478                if (FD_ISSET(fd, &allsock))
 479                        maxsock = fd;
 480                fd++;
 481        }
 482        prev_maxsock = maxsock;
 483        if ((rlim_t)maxsock > rlim_ofile_cur - FD_MARGIN)
 484                bump_nofile();
 485}
 486
 487static void prepare_socket_fd(servtab_t *sep)
 488{
 489        int r, fd;
 490
 491        fd = socket(sep->se_family, sep->se_socktype, 0);
 492        if (fd < 0) {
 493                bb_perror_msg("socket");
 494                return;
 495        }
 496        setsockopt_reuseaddr(fd);
 497
 498#if ENABLE_FEATURE_INETD_RPC
 499        if (is_rpc_service(sep)) {
 500                struct passwd *pwd;
 501
 502                /* zero out the port for all RPC services; let bind()
 503                 * find one. */
 504                set_nport(sep->se_lsa, 0);
 505
 506                /* for RPC services, attempt to use a reserved port
 507                 * if they are going to be running as root. */
 508                if (real_uid == 0 && sep->se_family == AF_INET
 509                 && (pwd = getpwnam(sep->se_user)) != NULL
 510                 && pwd->pw_uid == 0
 511                ) {
 512                        r = bindresvport(fd, &sep->se_lsa->u.sin);
 513                } else {
 514                        r = bind(fd, &sep->se_lsa->u.sa, sep->se_lsa->len);
 515                }
 516                if (r == 0) {
 517                        int saveerrno = errno;
 518                        /* update lsa with port# */
 519                        getsockname(fd, &sep->se_lsa->u.sa, &sep->se_lsa->len);
 520                        errno = saveerrno;
 521                }
 522        } else
 523#endif
 524        {
 525                if (sep->se_family == AF_UNIX) {
 526                        struct sockaddr_un *sun;
 527                        sun = (struct sockaddr_un*)&(sep->se_lsa->u.sa);
 528                        unlink(sun->sun_path);
 529                }
 530                r = bind(fd, &sep->se_lsa->u.sa, sep->se_lsa->len);
 531        }
 532        if (r < 0) {
 533                bb_perror_msg("%s/%s: bind",
 534                                sep->se_service, sep->se_proto);
 535                close(fd);
 536                rearm_alarm();
 537                return;
 538        }
 539        if (sep->se_socktype == SOCK_STREAM)
 540                listen(fd, global_queuelen);
 541
 542        add_fd_to_set(fd);
 543        sep->se_fd = fd;
 544}
 545
 546static int reopen_config_file(void)
 547{
 548        free(default_local_hostname);
 549        default_local_hostname = xstrdup("*");
 550        if (parser != NULL)
 551                config_close(parser);
 552        parser = config_open(config_filename);
 553        return (parser != NULL);
 554}
 555
 556static void close_config_file(void)
 557{
 558        if (parser) {
 559                config_close(parser);
 560                parser = NULL;
 561        }
 562}
 563
 564static void free_servtab_strings(servtab_t *cp)
 565{
 566        int i;
 567
 568        free(cp->se_local_hostname);
 569        free(cp->se_service);
 570        free(cp->se_proto);
 571        free(cp->se_user);
 572        free(cp->se_group);
 573        free(cp->se_lsa); /* not a string in fact */
 574        free(cp->se_program);
 575        for (i = 0; i < MAXARGV; i++)
 576                free(cp->se_argv[i]);
 577}
 578
 579static servtab_t *new_servtab(void)
 580{
 581        servtab_t *newtab = xzalloc(sizeof(servtab_t));
 582        newtab->se_fd = -1; /* paranoia */
 583        return newtab;
 584}
 585
 586static servtab_t *dup_servtab(servtab_t *sep)
 587{
 588        servtab_t *newtab;
 589        int argc;
 590
 591        newtab = new_servtab();
 592        *newtab = *sep; /* struct copy */
 593        /* deep-copying strings */
 594        newtab->se_service = xstrdup(newtab->se_service);
 595        newtab->se_proto = xstrdup(newtab->se_proto);
 596        newtab->se_user = xstrdup(newtab->se_user);
 597        newtab->se_group = xstrdup(newtab->se_group);
 598        newtab->se_program = xstrdup(newtab->se_program);
 599        for (argc = 0; argc <= MAXARGV; argc++)
 600                newtab->se_argv[argc] = xstrdup(newtab->se_argv[argc]);
 601        /* NB: se_fd, se_hostaddr and se_next are always
 602         * overwrittend by callers, so we don't bother resetting them
 603         * to NULL/0/-1 etc */
 604
 605        return newtab;
 606}
 607
 608/* gcc generates much more code if this is inlined */
 609static servtab_t *parse_one_line(void)
 610{
 611        int argc;
 612        char *token[6+MAXARGV];
 613        char *p, *arg;
 614        char *hostdelim;
 615        servtab_t *sep;
 616        servtab_t *nsep;
 617 new:
 618        sep = new_servtab();
 619 more:
 620        argc = config_read(parser, token, 6+MAXARGV, 1, "# \t", PARSE_NORMAL);
 621        if (!argc) {
 622                free(sep);
 623                return NULL;
 624        }
 625
 626        /* [host:]service socktype proto wait user[:group] prog [args] */
 627        /* Check for "host:...." line */
 628        arg = token[0];
 629        hostdelim = strrchr(arg, ':');
 630        if (hostdelim) {
 631                *hostdelim = '\0';
 632                sep->se_local_hostname = xstrdup(arg);
 633                arg = hostdelim + 1;
 634                if (*arg == '\0' && argc == 1) {
 635                        /* Line has just "host:", change the
 636                         * default host for the following lines. */
 637                        free(default_local_hostname);
 638                        default_local_hostname = sep->se_local_hostname;
 639                        goto more;
 640                }
 641        } else
 642                sep->se_local_hostname = xstrdup(default_local_hostname);
 643
 644        /* service socktype proto wait user[:group] prog [args] */
 645        sep->se_service = xstrdup(arg);
 646
 647        /* socktype proto wait user[:group] prog [args] */
 648        if (argc < 6) {
 649 parse_err:
 650                bb_error_msg("parse error on line %u, line is ignored",
 651                                parser->lineno);
 652                free_servtab_strings(sep);
 653                /* Just "goto more" can make sep to carry over e.g.
 654                 * "rpc"-ness (by having se_rpcver_lo != 0).
 655                 * We will be more paranoid: */
 656                free(sep);
 657                goto new;
 658        }
 659
 660        {
 661                static int8_t SOCK_xxx[] ALIGN1 = {
 662                        -1,
 663                        SOCK_STREAM, SOCK_DGRAM, SOCK_RDM,
 664                        SOCK_SEQPACKET, SOCK_RAW
 665                };
 666                sep->se_socktype = SOCK_xxx[1 + index_in_strings(
 667                        "stream""\0" "dgram""\0" "rdm""\0"
 668                        "seqpacket""\0" "raw""\0"
 669                        , token[1])];
 670        }
 671
 672        /* {unix,[rpc/]{tcp,udp}[6]} wait user[:group] prog [args] */
 673        sep->se_proto = arg = xstrdup(token[2]);
 674        if (strcmp(arg, "unix") == 0) {
 675                sep->se_family = AF_UNIX;
 676        } else {
 677                char *six;
 678                sep->se_family = AF_INET;
 679                six = last_char_is(arg, '6');
 680                if (six) {
 681#if ENABLE_FEATURE_IPV6
 682                        *six = '\0';
 683                        sep->se_family = AF_INET6;
 684#else
 685                        bb_error_msg("%s: no support for IPv6", sep->se_proto);
 686                        goto parse_err;
 687#endif
 688                }
 689                if (strncmp(arg, "rpc/", 4) == 0) {
 690#if ENABLE_FEATURE_INETD_RPC
 691                        unsigned n;
 692                        arg += 4;
 693                        p = strchr(sep->se_service, '/');
 694                        if (p == NULL) {
 695                                bb_error_msg("no rpc version: '%s'", sep->se_service);
 696                                goto parse_err;
 697                        }
 698                        *p++ = '\0';
 699                        n = bb_strtou(p, &p, 10);
 700                        if (n > INT_MAX) {
 701 bad_ver_spec:
 702                                bb_error_msg("bad rpc version");
 703                                goto parse_err;
 704                        }
 705                        sep->se_rpcver_lo = sep->se_rpcver_hi = n;
 706                        if (*p == '-') {
 707                                p++;
 708                                n = bb_strtou(p, &p, 10);
 709                                if (n > INT_MAX || (int)n < sep->se_rpcver_lo)
 710                                        goto bad_ver_spec;
 711                                sep->se_rpcver_hi = n;
 712                        }
 713                        if (*p != '\0')
 714                                goto bad_ver_spec;
 715#else
 716                        bb_error_msg("no support for rpc services");
 717                        goto parse_err;
 718#endif
 719                }
 720                /* we don't really need getprotobyname()! */
 721                if (strcmp(arg, "tcp") == 0)
 722                        sep->se_proto_no = IPPROTO_TCP; /* = 6 */
 723                if (strcmp(arg, "udp") == 0)
 724                        sep->se_proto_no = IPPROTO_UDP; /* = 17 */
 725                if (six)
 726                        *six = '6';
 727                if (!sep->se_proto_no) /* not tcp/udp?? */
 728                        goto parse_err;
 729        }
 730
 731        /* [no]wait[.max] user[:group] prog [args] */
 732        arg = token[3];
 733        sep->se_max = max_concurrency;
 734        p = strchr(arg, '.');
 735        if (p) {
 736                *p++ = '\0';
 737                sep->se_max = bb_strtou(p, NULL, 10);
 738                if (errno)
 739                        goto parse_err;
 740        }
 741        sep->se_wait = (arg[0] != 'n' || arg[1] != 'o');
 742        if (!sep->se_wait) /* "no" seen */
 743                arg += 2;
 744        if (strcmp(arg, "wait") != 0)
 745                goto parse_err;
 746
 747        /* user[:group] prog [args] */
 748        sep->se_user = xstrdup(token[4]);
 749        arg = strchr(sep->se_user, '.');
 750        if (arg == NULL)
 751                arg = strchr(sep->se_user, ':');
 752        if (arg) {
 753                *arg++ = '\0';
 754                sep->se_group = xstrdup(arg);
 755        }
 756
 757        /* prog [args] */
 758        sep->se_program = xstrdup(token[5]);
 759#ifdef INETD_BUILTINS_ENABLED
 760        if (strcmp(sep->se_program, "internal") == 0
 761         && strlen(sep->se_service) <= 7
 762         && (sep->se_socktype == SOCK_STREAM
 763             || sep->se_socktype == SOCK_DGRAM)
 764        ) {
 765                unsigned i;
 766                for (i = 0; i < ARRAY_SIZE(builtins); i++)
 767                        if (strncmp(builtins[i].bi_service7, sep->se_service, 7) == 0)
 768                                goto found_bi;
 769                bb_error_msg("unknown internal service %s", sep->se_service);
 770                goto parse_err;
 771 found_bi:
 772                sep->se_builtin = &builtins[i];
 773                /* stream builtins must be "nowait", dgram must be "wait" */
 774                if (sep->se_wait != (sep->se_socktype == SOCK_DGRAM))
 775                        goto parse_err;
 776        }
 777#endif
 778        argc = 0;
 779        while ((arg = token[6+argc]) != NULL && argc < MAXARGV)
 780                sep->se_argv[argc++] = xstrdup(arg);
 781
 782        /* catch mixups. "<service> stream udp ..." == wtf */
 783        if (sep->se_socktype == SOCK_STREAM) {
 784                if (sep->se_proto_no == IPPROTO_UDP)
 785                        goto parse_err;
 786        }
 787        if (sep->se_socktype == SOCK_DGRAM) {
 788                if (sep->se_proto_no == IPPROTO_TCP)
 789                        goto parse_err;
 790        }
 791
 792//      bb_info_msg(
 793//              "ENTRY[%s][%s][%s][%d][%d][%d][%d][%d][%s][%s][%s]",
 794//              sep->se_local_hostname, sep->se_service, sep->se_proto, sep->se_wait, sep->se_proto_no,
 795//              sep->se_max, sep->se_count, sep->se_time, sep->se_user, sep->se_group, sep->se_program);
 796
 797        /* check if the hostname specifier is a comma separated list
 798         * of hostnames. we'll make new entries for each address. */
 799        while ((hostdelim = strrchr(sep->se_local_hostname, ',')) != NULL) {
 800                nsep = dup_servtab(sep);
 801                /* NUL terminate the hostname field of the existing entry,
 802                 * and make a dup for the new entry. */
 803                *hostdelim++ = '\0';
 804                nsep->se_local_hostname = xstrdup(hostdelim);
 805                nsep->se_next = sep->se_next;
 806                sep->se_next = nsep;
 807        }
 808
 809        /* was doing it here: */
 810        /* DNS resolution, create copies for each IP address */
 811        /* IPv6-ization destroyed it :( */
 812
 813        return sep;
 814}
 815
 816static servtab_t *insert_in_servlist(servtab_t *cp)
 817{
 818        servtab_t *sep;
 819        sigset_t omask;
 820
 821        sep = new_servtab();
 822        *sep = *cp; /* struct copy */
 823        sep->se_fd = -1;
 824#if ENABLE_FEATURE_INETD_RPC
 825        sep->se_rpcprog = -1;
 826#endif
 827        block_CHLD_HUP_ALRM(&omask);
 828        sep->se_next = serv_list;
 829        serv_list = sep;
 830        restore_sigmask(&omask);
 831        return sep;
 832}
 833
 834static int same_serv_addr_proto(servtab_t *old, servtab_t *new)
 835{
 836        if (strcmp(old->se_local_hostname, new->se_local_hostname) != 0)
 837                return 0;
 838        if (strcmp(old->se_service, new->se_service) != 0)
 839                return 0;
 840        if (strcmp(old->se_proto, new->se_proto) != 0)
 841                return 0;
 842        return 1;
 843}
 844
 845static void reread_config_file(int sig UNUSED_PARAM)
 846{
 847        servtab_t *sep, *cp, **sepp;
 848        len_and_sockaddr *lsa;
 849        sigset_t omask;
 850        unsigned n;
 851        uint16_t port;
 852        int save_errno = errno;
 853
 854        if (!reopen_config_file())
 855                goto ret;
 856        for (sep = serv_list; sep; sep = sep->se_next)
 857                sep->se_checked = 0;
 858
 859        goto first_line;
 860        while (1) {
 861                if (cp == NULL) {
 862 first_line:
 863                        cp = parse_one_line();
 864                        if (cp == NULL)
 865                                break;
 866                }
 867                for (sep = serv_list; sep; sep = sep->se_next)
 868                        if (same_serv_addr_proto(sep, cp))
 869                                goto equal_servtab;
 870                /* not an "equal" servtab */
 871                sep = insert_in_servlist(cp);
 872                goto after_check;
 873 equal_servtab:
 874                {
 875                        int i;
 876
 877                        block_CHLD_HUP_ALRM(&omask);
 878#if ENABLE_FEATURE_INETD_RPC
 879                        if (is_rpc_service(sep))
 880                                unregister_rpc(sep);
 881                        sep->se_rpcver_lo = cp->se_rpcver_lo;
 882                        sep->se_rpcver_hi = cp->se_rpcver_hi;
 883#endif
 884                        if (cp->se_wait == 0) {
 885                                /* New config says "nowait". If old one
 886                                 * was "wait", we currently may be waiting
 887                                 * for a child (and not accepting connects).
 888                                 * Stop waiting, start listening again.
 889                                 * (if it's not true, this op is harmless) */
 890                                add_fd_to_set(sep->se_fd);
 891                        }
 892                        sep->se_wait = cp->se_wait;
 893                        sep->se_max = cp->se_max;
 894                        /* string fields need more love - we don't want to leak them */
 895#define SWAP(type, a, b) do { type c = (type)a; a = (type)b; b = (type)c; } while (0)
 896                        SWAP(char*, sep->se_user, cp->se_user);
 897                        SWAP(char*, sep->se_group, cp->se_group);
 898                        SWAP(char*, sep->se_program, cp->se_program);
 899                        for (i = 0; i < MAXARGV; i++)
 900                                SWAP(char*, sep->se_argv[i], cp->se_argv[i]);
 901#undef SWAP
 902                        restore_sigmask(&omask);
 903                        free_servtab_strings(cp);
 904                }
 905 after_check:
 906                /* cp->string_fields are consumed by insert_in_servlist()
 907                 * or freed at this point, cp itself is not yet freed. */
 908                sep->se_checked = 1;
 909
 910                /* create new len_and_sockaddr */
 911                switch (sep->se_family) {
 912                        struct sockaddr_un *sun;
 913                case AF_UNIX:
 914                        lsa = xzalloc_lsa(AF_UNIX);
 915                        sun = (struct sockaddr_un*)&lsa->u.sa;
 916                        safe_strncpy(sun->sun_path, sep->se_service, sizeof(sun->sun_path));
 917                        break;
 918
 919                default: /* case AF_INET, case AF_INET6 */
 920                        n = bb_strtou(sep->se_service, NULL, 10);
 921#if ENABLE_FEATURE_INETD_RPC
 922                        if (is_rpc_service(sep)) {
 923                                sep->se_rpcprog = n;
 924                                if (errno) { /* se_service is not numeric */
 925                                        struct rpcent *rp = getrpcbyname(sep->se_service);
 926                                        if (rp == NULL) {
 927                                                bb_error_msg("%s: unknown rpc service", sep->se_service);
 928                                                goto next_cp;
 929                                        }
 930                                        sep->se_rpcprog = rp->r_number;
 931                                }
 932                                if (sep->se_fd == -1)
 933                                        prepare_socket_fd(sep);
 934                                if (sep->se_fd != -1)
 935                                        register_rpc(sep);
 936                                goto next_cp;
 937                        }
 938#endif
 939                        /* what port to listen on? */
 940                        port = htons(n);
 941                        if (errno || n > 0xffff) { /* se_service is not numeric */
 942                                char protoname[4];
 943                                struct servent *sp;
 944                                /* can result only in "tcp" or "udp": */
 945                                safe_strncpy(protoname, sep->se_proto, 4);
 946                                sp = getservbyname(sep->se_service, protoname);
 947                                if (sp == NULL) {
 948                                        bb_error_msg("%s/%s: unknown service",
 949                                                        sep->se_service, sep->se_proto);
 950                                        goto next_cp;
 951                                }
 952                                port = sp->s_port;
 953                        }
 954                        if (LONE_CHAR(sep->se_local_hostname, '*')) {
 955                                lsa = xzalloc_lsa(sep->se_family);
 956                                set_nport(lsa, port);
 957                        } else {
 958                                lsa = host_and_af2sockaddr(sep->se_local_hostname,
 959                                                ntohs(port), sep->se_family);
 960                                if (!lsa) {
 961                                        bb_error_msg("%s/%s: unknown host '%s'",
 962                                                sep->se_service, sep->se_proto,
 963                                                sep->se_local_hostname);
 964                                        goto next_cp;
 965                                }
 966                        }
 967                        break;
 968                } /* end of "switch (sep->se_family)" */
 969
 970                /* did lsa change? Then close/open */
 971                if (sep->se_lsa == NULL
 972                 || lsa->len != sep->se_lsa->len
 973                 || memcmp(&lsa->u.sa, &sep->se_lsa->u.sa, lsa->len) != 0
 974                ) {
 975                        remove_fd_from_set(sep->se_fd);
 976                        maybe_close(sep->se_fd);
 977                        free(sep->se_lsa);
 978                        sep->se_lsa = lsa;
 979                        sep->se_fd = -1;
 980                } else {
 981                        free(lsa);
 982                }
 983                if (sep->se_fd == -1)
 984                        prepare_socket_fd(sep);
 985 next_cp:
 986                sep = cp->se_next;
 987                free(cp);
 988                cp = sep;
 989        } /* end of "while (1) parse lines" */
 990        close_config_file();
 991
 992        /* Purge anything not looked at above - these are stale entries,
 993         * new config file doesnt have them. */
 994        block_CHLD_HUP_ALRM(&omask);
 995        sepp = &serv_list;
 996        while ((sep = *sepp)) {
 997                if (sep->se_checked) {
 998                        sepp = &sep->se_next;
 999                        continue;
1000                }
1001                *sepp = sep->se_next;
1002                remove_fd_from_set(sep->se_fd);
1003                maybe_close(sep->se_fd);
1004#if ENABLE_FEATURE_INETD_RPC
1005                if (is_rpc_service(sep))
1006                        unregister_rpc(sep);
1007#endif
1008                if (sep->se_family == AF_UNIX)
1009                        unlink(sep->se_service);
1010                free_servtab_strings(sep);
1011                free(sep);
1012        }
1013        restore_sigmask(&omask);
1014 ret:
1015        errno = save_errno;
1016}
1017
1018static void reap_child(int sig UNUSED_PARAM)
1019{
1020        pid_t pid;
1021        int status;
1022        servtab_t *sep;
1023        int save_errno = errno;
1024
1025        for (;;) {
1026                pid = wait_any_nohang(&status);
1027                if (pid <= 0)
1028                        break;
1029                for (sep = serv_list; sep; sep = sep->se_next) {
1030                        if (sep->se_wait != pid)
1031                                continue;
1032                        /* One of our "wait" services */
1033                        if (WIFEXITED(status) && WEXITSTATUS(status))
1034                                bb_error_msg("%s: exit status 0x%x",
1035                                                sep->se_program, WEXITSTATUS(status));
1036                        else if (WIFSIGNALED(status))
1037                                bb_error_msg("%s: exit signal 0x%x",
1038                                                sep->se_program, WTERMSIG(status));
1039                        sep->se_wait = 1;
1040                        add_fd_to_set(sep->se_fd);
1041                        break;
1042                }
1043        }
1044        errno = save_errno;
1045}
1046
1047static void retry_network_setup(int sig UNUSED_PARAM)
1048{
1049        int save_errno = errno;
1050        servtab_t *sep;
1051
1052        alarm_armed = 0;
1053        for (sep = serv_list; sep; sep = sep->se_next) {
1054                if (sep->se_fd == -1) {
1055                        prepare_socket_fd(sep);
1056#if ENABLE_FEATURE_INETD_RPC
1057                        if (sep->se_fd != -1 && is_rpc_service(sep))
1058                                register_rpc(sep);
1059#endif
1060                }
1061        }
1062        errno = save_errno;
1063}
1064
1065static void clean_up_and_exit(int sig UNUSED_PARAM)
1066{
1067        servtab_t *sep;
1068
1069        /* XXX signal race walking sep list */
1070        for (sep = serv_list; sep; sep = sep->se_next) {
1071                if (sep->se_fd == -1)
1072                        continue;
1073
1074                switch (sep->se_family) {
1075                case AF_UNIX:
1076                        unlink(sep->se_service);
1077                        break;
1078                default: /* case AF_INET, AF_INET6 */
1079#if ENABLE_FEATURE_INETD_RPC
1080                        if (sep->se_wait == 1 && is_rpc_service(sep))
1081                                unregister_rpc(sep);   /* XXX signal race */
1082#endif
1083                        break;
1084                }
1085                if (ENABLE_FEATURE_CLEAN_UP)
1086                        close(sep->se_fd);
1087        }
1088        remove_pidfile(_PATH_INETDPID);
1089        exit(EXIT_SUCCESS);
1090}
1091
1092int inetd_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
1093int inetd_main(int argc UNUSED_PARAM, char **argv)
1094{
1095        struct sigaction sa, saved_pipe_handler;
1096        servtab_t *sep, *sep2;
1097        struct passwd *pwd;
1098        struct group *grp = grp; /* for compiler */
1099        int opt;
1100        pid_t pid;
1101        sigset_t omask;
1102
1103        INIT_G();
1104
1105        real_uid = getuid();
1106        if (real_uid != 0) /* run by non-root user */
1107                config_filename = NULL;
1108
1109        opt_complementary = "R+:q+"; /* -q N, -R N */
1110        opt = getopt32(argv, "R:feq:", &max_concurrency, &global_queuelen);
1111        argv += optind;
1112        //argc -= optind;
1113        if (argv[0])
1114                config_filename = argv[0];
1115        if (config_filename == NULL)
1116                bb_error_msg_and_die("non-root must specify config file");
1117        if (!(opt & 2))
1118                bb_daemonize_or_rexec(0, argv - optind);
1119        else
1120                bb_sanitize_stdio();
1121        if (!(opt & 4)) {
1122                openlog(applet_name, LOG_PID | LOG_NOWAIT, LOG_DAEMON);
1123                logmode = LOGMODE_SYSLOG;
1124        }
1125
1126        if (real_uid == 0) {
1127                /* run by root, ensure groups vector gets trashed */
1128                gid_t gid = getgid();
1129                setgroups(1, &gid);
1130        }
1131
1132        write_pidfile(_PATH_INETDPID);
1133
1134        /* never fails under Linux (except if you pass it bad arguments) */
1135        getrlimit(RLIMIT_NOFILE, &rlim_ofile);
1136        rlim_ofile_cur = rlim_ofile.rlim_cur;
1137        if (rlim_ofile_cur == RLIM_INFINITY)    /* ! */
1138                rlim_ofile_cur = OPEN_MAX;
1139
1140        memset(&sa, 0, sizeof(sa));
1141        /*sigemptyset(&sa.sa_mask); - memset did it */
1142        sigaddset(&sa.sa_mask, SIGALRM);
1143        sigaddset(&sa.sa_mask, SIGCHLD);
1144        sigaddset(&sa.sa_mask, SIGHUP);
1145        sa.sa_handler = retry_network_setup;
1146        sigaction_set(SIGALRM, &sa);
1147        sa.sa_handler = reread_config_file;
1148        sigaction_set(SIGHUP, &sa);
1149        sa.sa_handler = reap_child;
1150        sigaction_set(SIGCHLD, &sa);
1151        sa.sa_handler = clean_up_and_exit;
1152        sigaction_set(SIGTERM, &sa);
1153        sa.sa_handler = clean_up_and_exit;
1154        sigaction_set(SIGINT, &sa);
1155        sa.sa_handler = SIG_IGN;
1156        sigaction(SIGPIPE, &sa, &saved_pipe_handler);
1157
1158        reread_config_file(SIGHUP); /* load config from file */
1159
1160        for (;;) {
1161                int ready_fd_cnt;
1162                int ctrl, accepted_fd, new_udp_fd;
1163                fd_set readable;
1164
1165                if (maxsock < 0)
1166                        recalculate_maxsock();
1167
1168                readable = allsock; /* struct copy */
1169                /* if there are no fds to wait on, we will block
1170                 * until signal wakes us up (maxsock == 0, but readable
1171                 * never contains fds 0 and 1...) */
1172                ready_fd_cnt = select(maxsock + 1, &readable, NULL, NULL, NULL);
1173                if (ready_fd_cnt < 0) {
1174                        if (errno != EINTR) {
1175                                bb_perror_msg("select");
1176                                sleep(1);
1177                        }
1178                        continue;
1179                }
1180
1181                for (sep = serv_list; ready_fd_cnt && sep; sep = sep->se_next) {
1182                        if (sep->se_fd == -1 || !FD_ISSET(sep->se_fd, &readable))
1183                                continue;
1184
1185                        ready_fd_cnt--;
1186                        ctrl = sep->se_fd;
1187                        accepted_fd = -1;
1188                        new_udp_fd = -1;
1189                        if (!sep->se_wait) {
1190                                if (sep->se_socktype == SOCK_STREAM) {
1191                                        ctrl = accepted_fd = accept(sep->se_fd, NULL, NULL);
1192                                        if (ctrl < 0) {
1193                                                if (errno != EINTR)
1194                                                        bb_perror_msg("accept (for %s)", sep->se_service);
1195                                                continue;
1196                                        }
1197                                }
1198                                /* "nowait" udp */
1199                                if (sep->se_socktype == SOCK_DGRAM
1200                                 && sep->se_family != AF_UNIX
1201                                ) {
1202/* How udp "nowait" works:
1203 * child peeks at (received and buffered by kernel) UDP packet,
1204 * performs connect() on the socket so that it is linked only
1205 * to this peer. But this also affects parent, because descriptors
1206 * are shared after fork() a-la dup(). When parent performs
1207 * select(), it will see this descriptor connected to the peer (!)
1208 * and still readable, will act on it and mess things up
1209 * (can create many copies of same child, etc).
1210 * Parent must create and use new socket instead. */
1211                                        new_udp_fd = socket(sep->se_family, SOCK_DGRAM, 0);
1212                                        if (new_udp_fd < 0) { /* error: eat packet, forget about it */
1213 udp_err:
1214                                                recv(sep->se_fd, line, LINE_SIZE, MSG_DONTWAIT);
1215                                                continue;
1216                                        }
1217                                        setsockopt_reuseaddr(new_udp_fd);
1218                                        /* TODO: better do bind after vfork in parent,
1219                                         * so that we don't have two wildcard bound sockets
1220                                         * even for a brief moment? */
1221                                        if (bind(new_udp_fd, &sep->se_lsa->u.sa, sep->se_lsa->len) < 0) {
1222                                                close(new_udp_fd);
1223                                                goto udp_err;
1224                                        }
1225                                }
1226                        }
1227
1228                        block_CHLD_HUP_ALRM(&omask);
1229                        pid = 0;
1230#ifdef INETD_BUILTINS_ENABLED
1231                        /* do we need to fork? */
1232                        if (sep->se_builtin == NULL
1233                         || (sep->se_socktype == SOCK_STREAM
1234                             && sep->se_builtin->bi_fork))
1235#endif
1236                        {
1237                                if (sep->se_max != 0) {
1238                                        if (++sep->se_count == 1)
1239                                                sep->se_time = monotonic_sec();
1240                                        else if (sep->se_count >= sep->se_max) {
1241                                                unsigned now = monotonic_sec();
1242                                                /* did we accumulate se_max connects too quickly? */
1243                                                if (now - sep->se_time <= CNT_INTERVAL) {
1244                                                        bb_error_msg("%s/%s: too many connections, pausing",
1245                                                                        sep->se_service, sep->se_proto);
1246                                                        remove_fd_from_set(sep->se_fd);
1247                                                        close(sep->se_fd);
1248                                                        sep->se_fd = -1;
1249                                                        sep->se_count = 0;
1250                                                        rearm_alarm(); /* will revive it in RETRYTIME sec */
1251                                                        restore_sigmask(&omask);
1252                                                        maybe_close(accepted_fd);
1253                                                        continue; /* -> check next fd in fd set */
1254                                                }
1255                                                sep->se_count = 0;
1256                                        }
1257                                }
1258                                /* on NOMMU, streamed chargen
1259                                 * builtin wouldn't work, but it is
1260                                 * not allowed on NOMMU (ifdefed out) */
1261#ifdef INETD_BUILTINS_ENABLED
1262                                if (BB_MMU && sep->se_builtin)
1263                                        pid = fork();
1264                                else
1265#endif
1266                                        pid = vfork();
1267
1268                                if (pid < 0) { /* fork error */
1269                                        bb_perror_msg("fork");
1270                                        sleep(1);
1271                                        restore_sigmask(&omask);
1272                                        maybe_close(accepted_fd);
1273                                        continue; /* -> check next fd in fd set */
1274                                }
1275                                if (pid == 0)
1276                                        pid--; /* -1: "we did fork and we are child" */
1277                        }
1278                        /* if pid == 0 here, we never forked */
1279
1280                        if (pid > 0) { /* parent */
1281                                if (sep->se_wait) {
1282                                        /* tcp wait: we passed listening socket to child,
1283                                         * will wait for child to terminate */
1284                                        sep->se_wait = pid;
1285                                        remove_fd_from_set(sep->se_fd);
1286                                }
1287                                if (new_udp_fd >= 0) {
1288                                        /* udp nowait: child connected the socket,
1289                                         * we created and will use new, unconnected one */
1290                                        xmove_fd(new_udp_fd, sep->se_fd);
1291                                }
1292                                restore_sigmask(&omask);
1293                                maybe_close(accepted_fd);
1294                                continue; /* -> check next fd in fd set */
1295                        }
1296
1297                        /* we are either child or didn't vfork at all */
1298#ifdef INETD_BUILTINS_ENABLED
1299                        if (sep->se_builtin) {
1300                                if (pid) { /* "pid" is -1: we did vfork */
1301                                        close(sep->se_fd); /* listening socket */
1302                                        logmode = 0; /* make xwrite etc silent */
1303                                }
1304                                restore_sigmask(&omask);
1305                                if (sep->se_socktype == SOCK_STREAM)
1306                                        sep->se_builtin->bi_stream_fn(ctrl, sep);
1307                                else
1308                                        sep->se_builtin->bi_dgram_fn(ctrl, sep);
1309                                if (pid) /* we did vfork */
1310                                        _exit(EXIT_FAILURE);
1311                                maybe_close(accepted_fd);
1312                                continue; /* -> check next fd in fd set */
1313                        }
1314#endif
1315                        /* child */
1316                        setsid();
1317                        /* "nowait" udp */
1318                        if (new_udp_fd >= 0) {
1319                                len_and_sockaddr *lsa = xzalloc_lsa(sep->se_family);
1320                                /* peek at the packet and remember peer addr */
1321                                int r = recvfrom(ctrl, NULL, 0, MSG_PEEK|MSG_DONTWAIT,
1322                                        &lsa->u.sa, &lsa->len);
1323                                if (r < 0)
1324                                        goto do_exit1;
1325                                /* make this socket "connected" to peer addr:
1326                                 * only packets from this peer will be recv'ed,
1327                                 * and bare write()/send() will work on it */
1328                                connect(ctrl, &lsa->u.sa, lsa->len);
1329                                free(lsa);
1330                        }
1331                        /* prepare env and exec program */
1332                        pwd = getpwnam(sep->se_user);
1333                        if (pwd == NULL) {
1334                                bb_error_msg("%s: no such %s", sep->se_user, "user");
1335                                goto do_exit1;
1336                        }
1337                        if (sep->se_group && (grp = getgrnam(sep->se_group)) == NULL) {
1338                                bb_error_msg("%s: no such %s", sep->se_group, "group");
1339                                goto do_exit1;
1340                        }
1341                        if (real_uid != 0 && real_uid != pwd->pw_uid) {
1342                                /* a user running private inetd */
1343                                bb_error_msg("non-root must run services as himself");
1344                                goto do_exit1;
1345                        }
1346                        if (pwd->pw_uid) {
1347                                if (sep->se_group)
1348                                        pwd->pw_gid = grp->gr_gid;
1349                                /* initgroups, setgid, setuid: */
1350                                change_identity(pwd);
1351                        } else if (sep->se_group) {
1352                                xsetgid(grp->gr_gid);
1353                                setgroups(1, &grp->gr_gid);
1354                        }
1355                        if (rlim_ofile.rlim_cur != rlim_ofile_cur)
1356                                if (setrlimit(RLIMIT_NOFILE, &rlim_ofile) < 0)
1357                                        bb_perror_msg("setrlimit");
1358                        closelog();
1359                        xmove_fd(ctrl, 0);
1360                        xdup2(0, 1);
1361                        xdup2(0, 2);
1362                        /* NB: among others, this loop closes listening socket
1363                         * for nowait stream children */
1364                        for (sep2 = serv_list; sep2; sep2 = sep2->se_next)
1365                                maybe_close(sep2->se_fd);
1366                        sigaction_set(SIGPIPE, &saved_pipe_handler);
1367                        restore_sigmask(&omask);
1368                        BB_EXECVP(sep->se_program, sep->se_argv);
1369                        bb_perror_msg("exec %s", sep->se_program);
1370 do_exit1:
1371                        /* eat packet in udp case */
1372                        if (sep->se_socktype != SOCK_STREAM)
1373                                recv(0, line, LINE_SIZE, MSG_DONTWAIT);
1374                        _exit(EXIT_FAILURE);
1375                } /* for (sep = servtab...) */
1376        } /* for (;;) */
1377}
1378
1379#if !BB_MMU
1380static const char *const cat_args[] = { "cat", NULL };
1381#endif
1382
1383/*
1384 * Internet services provided internally by inetd:
1385 */
1386#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_ECHO
1387/* Echo service -- echo data back. */
1388/* ARGSUSED */
1389static void echo_stream(int s, servtab_t *sep UNUSED_PARAM)
1390{
1391#if BB_MMU
1392        while (1) {
1393                ssize_t sz = safe_read(s, line, LINE_SIZE);
1394                if (sz <= 0)
1395                        break;
1396                xwrite(s, line, sz);
1397        }
1398#else
1399        /* We are after vfork here! */
1400        /* move network socket to stdin/stdout */
1401        xmove_fd(s, STDIN_FILENO);
1402        xdup2(STDIN_FILENO, STDOUT_FILENO);
1403        /* no error messages please... */
1404        close(STDERR_FILENO);
1405        xopen(bb_dev_null, O_WRONLY);
1406        BB_EXECVP("cat", (char**)cat_args);
1407        /* on failure we return to main, which does exit(EXIT_FAILURE) */
1408#endif
1409}
1410static void echo_dg(int s, servtab_t *sep)
1411{
1412        enum { BUFSIZE = 12*1024 }; /* for jumbo sized packets! :) */
1413        char *buf = xmalloc(BUFSIZE); /* too big for stack */
1414        int sz;
1415        len_and_sockaddr *lsa = alloca(LSA_LEN_SIZE + sep->se_lsa->len);
1416
1417        lsa->len = sep->se_lsa->len;
1418        /* dgram builtins are non-forking - DONT BLOCK! */
1419        sz = recvfrom(s, buf, BUFSIZE, MSG_DONTWAIT, &lsa->u.sa, &lsa->len);
1420        if (sz > 0)
1421                sendto(s, buf, sz, 0, &lsa->u.sa, lsa->len);
1422        free(buf);
1423}
1424#endif  /* FEATURE_INETD_SUPPORT_BUILTIN_ECHO */
1425
1426
1427#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_DISCARD
1428/* Discard service -- ignore data. */
1429/* ARGSUSED */
1430static void discard_stream(int s, servtab_t *sep UNUSED_PARAM)
1431{
1432#if BB_MMU
1433        while (safe_read(s, line, LINE_SIZE) > 0)
1434                continue;
1435#else
1436        /* We are after vfork here! */
1437        /* move network socket to stdin */
1438        xmove_fd(s, STDIN_FILENO);
1439        /* discard output */
1440        close(STDOUT_FILENO);
1441        xopen(bb_dev_null, O_WRONLY);
1442        /* no error messages please... */
1443        xdup2(STDOUT_FILENO, STDERR_FILENO);
1444        BB_EXECVP("cat", (char**)cat_args);
1445        /* on failure we return to main, which does exit(EXIT_FAILURE) */
1446#endif
1447}
1448/* ARGSUSED */
1449static void discard_dg(int s, servtab_t *sep UNUSED_PARAM)
1450{
1451        /* dgram builtins are non-forking - DONT BLOCK! */
1452        recv(s, line, LINE_SIZE, MSG_DONTWAIT);
1453}
1454#endif /* FEATURE_INETD_SUPPORT_BUILTIN_DISCARD */
1455
1456
1457#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_CHARGEN
1458#define LINESIZ 72
1459static void init_ring(void)
1460{
1461        int i;
1462
1463        end_ring = ring;
1464        for (i = 0; i <= 128; ++i)
1465                if (isprint(i))
1466                        *end_ring++ = i;
1467}
1468/* Character generator. MMU arches only. */
1469/* ARGSUSED */
1470static void chargen_stream(int s, servtab_t *sep UNUSED_PARAM)
1471{
1472        char *rs;
1473        int len;
1474        char text[LINESIZ + 2];
1475
1476        if (!end_ring) {
1477                init_ring();
1478                rs = ring;
1479        }
1480
1481        text[LINESIZ] = '\r';
1482        text[LINESIZ + 1] = '\n';
1483        rs = ring;
1484        for (;;) {
1485                len = end_ring - rs;
1486                if (len >= LINESIZ)
1487                        memmove(text, rs, LINESIZ);
1488                else {
1489                        memmove(text, rs, len);
1490                        memmove(text + len, ring, LINESIZ - len);
1491                }
1492                if (++rs == end_ring)
1493                        rs = ring;
1494                xwrite(s, text, sizeof(text));
1495        }
1496}
1497/* ARGSUSED */
1498static void chargen_dg(int s, servtab_t *sep)
1499{
1500        int len;
1501        char text[LINESIZ + 2];
1502        len_and_sockaddr *lsa = alloca(LSA_LEN_SIZE + sep->se_lsa->len);
1503
1504        /* Eat UDP packet which started it all */
1505        /* dgram builtins are non-forking - DONT BLOCK! */
1506        lsa->len = sep->se_lsa->len;
1507        if (recvfrom(s, text, sizeof(text), MSG_DONTWAIT, &lsa->u.sa, &lsa->len) < 0)
1508                return;
1509
1510        if (!end_ring) {
1511                init_ring();
1512                ring_pos = ring;
1513        }
1514
1515        len = end_ring - ring_pos;
1516        if (len >= LINESIZ)
1517                memmove(text, ring_pos, LINESIZ);
1518        else {
1519                memmove(text, ring_pos, len);
1520                memmove(text + len, ring, LINESIZ - len);
1521        }
1522        if (++ring_pos == end_ring)
1523                ring_pos = ring;
1524        text[LINESIZ] = '\r';
1525        text[LINESIZ + 1] = '\n';
1526        sendto(s, text, sizeof(text), 0, &lsa->u.sa, lsa->len);
1527}
1528#endif /* FEATURE_INETD_SUPPORT_BUILTIN_CHARGEN */
1529
1530
1531#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_TIME
1532/*
1533 * Return a machine readable date and time, in the form of the
1534 * number of seconds since midnight, Jan 1, 1900.  Since gettimeofday
1535 * returns the number of seconds since midnight, Jan 1, 1970,
1536 * we must add 2208988800 seconds to this figure to make up for
1537 * some seventy years Bell Labs was asleep.
1538 */
1539static uint32_t machtime(void)
1540{
1541        struct timeval tv;
1542
1543        gettimeofday(&tv, NULL);
1544        return htonl((uint32_t)(tv.tv_sec + 2208988800));
1545}
1546/* ARGSUSED */
1547static void machtime_stream(int s, servtab_t *sep UNUSED_PARAM)
1548{
1549        uint32_t result;
1550
1551        result = machtime();
1552        full_write(s, &result, sizeof(result));
1553}
1554static void machtime_dg(int s, servtab_t *sep)
1555{
1556        uint32_t result;
1557        len_and_sockaddr *lsa = alloca(LSA_LEN_SIZE + sep->se_lsa->len);
1558
1559        lsa->len = sep->se_lsa->len;
1560        if (recvfrom(s, line, LINE_SIZE, MSG_DONTWAIT, &lsa->u.sa, &lsa->len) < 0)
1561                return;
1562
1563        result = machtime();
1564        sendto(s, &result, sizeof(result), 0, &lsa->u.sa, lsa->len);
1565}
1566#endif /* FEATURE_INETD_SUPPORT_BUILTIN_TIME */
1567
1568
1569#if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_DAYTIME
1570/* Return human-readable time of day */
1571/* ARGSUSED */
1572static void daytime_stream(int s, servtab_t *sep UNUSED_PARAM)
1573{
1574        time_t t;
1575
1576        t = time(NULL);
1577        fdprintf(s, "%.24s\r\n", ctime(&t));
1578}
1579static void daytime_dg(int s, servtab_t *sep)
1580{
1581        time_t t;
1582        len_and_sockaddr *lsa = alloca(LSA_LEN_SIZE + sep->se_lsa->len);
1583
1584        lsa->len = sep->se_lsa->len;
1585        if (recvfrom(s, line, LINE_SIZE, MSG_DONTWAIT, &lsa->u.sa, &lsa->len) < 0)
1586                return;
1587
1588        t = time(NULL);
1589        sprintf(line, "%.24s\r\n", ctime(&t));
1590        sendto(s, line, strlen(line), 0, &lsa->u.sa, lsa->len);
1591}
1592#endif /* FEATURE_INETD_SUPPORT_BUILTIN_DAYTIME */
1593