git.maemo.org Git - busybox4maemo/blob - networking/inetd.c

   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
 200 typedef 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;                     /* whem 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
 242 static void echo_stream(int, servtab_t *);
 243 static void echo_dg(int, servtab_t *);
 244 #endif
 245 /* Internet /dev/null */
 246 #if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_DISCARD
 247 static void discard_stream(int, servtab_t *);
 248 static void discard_dg(int, servtab_t *);
 249 #endif
 250 /* Return 32 bit time since 1900 */
 251 #if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_TIME
 252 static void machtime_stream(int, servtab_t *);
 253 static void machtime_dg(int, servtab_t *);
 254 #endif
 255 /* Return human-readable time */
 256 #if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_DAYTIME
 257 static void daytime_stream(int, servtab_t *);
 258 static void daytime_dg(int, servtab_t *);
 259 #endif
 260 /* Familiar character generator */
 261 #if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_CHARGEN
 262 static void chargen_stream(int, servtab_t *);
 263 static void chargen_dg(int, servtab_t *);
 264 #endif
 265
 266 struct 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
 274 static 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
 293 struct globals {
 294         rlim_t rlim_ofile_cur;
 295         struct rlimit rlim_ofile;
 296         servtab_t *serv_list;
 297         int global_queuelen;
 298         int prev_maxsock;
 299         int maxsock;
 300         unsigned max_concurrency;
 301         smallint alarm_armed;
 302         uid_t real_uid; /* user ID who ran us */
 303         unsigned config_lineno;
 304         const char *config_filename;
 305         FILE *fconfig;
 306         char *default_local_hostname;
 307 #if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_CHARGEN
 308         char *end_ring;
 309         char *ring_pos;
 310         char ring[128];
 311 #endif
 312         fd_set allsock;
 313         /* Used in next_line(), and as scratch read buffer */
 314         char line[256];          /* _at least_ 256, see LINE_SIZE */
 315 };
 316 #define G (*(struct globals*)&bb_common_bufsiz1)
 317 enum { LINE_SIZE = COMMON_BUFSIZE - offsetof(struct globals, line) };
 318 struct BUG_G_too_big {
 319         char BUG_G_too_big[sizeof(G) <= COMMON_BUFSIZE ? 1 : -1];
 320 };
 321 #define rlim_ofile_cur  (G.rlim_ofile_cur )
 322 #define rlim_ofile      (G.rlim_ofile     )
 323 #define serv_list       (G.serv_list      )
 324 #define global_queuelen (G.global_queuelen)
 325 #define prev_maxsock    (G.prev_maxsock   )
 326 #define maxsock         (G.maxsock        )
 327 #define max_concurrency (G.max_concurrency)
 328 #define alarm_armed     (G.alarm_armed    )
 329 #define real_uid        (G.real_uid       )
 330 #define config_lineno   (G.config_lineno  )
 331 #define config_filename (G.config_filename)
 332 #define fconfig         (G.fconfig        )
 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
 347 static void maybe_close(int fd)
 348 {
 349         if (fd >= 0)
 350                 close(fd);
 351 }
 352
 353 // TODO: move to libbb?
 354 static 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
 372 static void rearm_alarm(void)
 373 {
 374         if (!alarm_armed) {
 375                 alarm_armed = 1;
 376                 alarm(RETRYTIME);
 377         }
 378 }
 379
 380 static 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
 389 static void restore_sigmask(sigset_t *m)
 390 {
 391         sigprocmask(SIG_SETMASK, m, NULL);
 392 }
 393
 394 #if ENABLE_FEATURE_INETD_RPC
 395 static 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
 416 static 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
 427 static 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
 450 static void remove_fd_from_set(int fd)
 451 {
 452         if (fd >= 0) {
 453                 FD_CLR(fd, &allsock);
 454                 maxsock = -1;
 455         }
 456 }
 457
 458 static 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)maxsock > rlim_ofile_cur - FD_MARGIN)
 465                                 bump_nofile();
 466                 }
 467         }
 468 }
 469
 470 static void recalculate_maxsock(void)
 471 {
 472         int fd = 0;
 473         while (fd <= prev_maxsock) {
 474                 if (FD_ISSET(fd, &allsock))
 475                         maxsock = fd;
 476                 fd++;
 477         }
 478         prev_maxsock = maxsock;
 479         if ((rlim_t)maxsock > rlim_ofile_cur - FD_MARGIN)
 480                 bump_nofile();
 481 }
 482
 483 static void prepare_socket_fd(servtab_t *sep)
 484 {
 485         int r, fd;
 486
 487         fd = socket(sep->se_family, sep->se_socktype, 0);
 488         if (fd < 0) {
 489                 bb_perror_msg("socket");
 490                 return;
 491         }
 492         setsockopt_reuseaddr(fd);
 493
 494 #if ENABLE_FEATURE_INETD_RPC
 495         if (is_rpc_service(sep)) {
 496                 struct passwd *pwd;
 497
 498                 /* zero out the port for all RPC services; let bind()
 499                  * find one. */
 500                 set_nport(sep->se_lsa, 0);
 501
 502                 /* for RPC services, attempt to use a reserved port
 503                  * if they are going to be running as root. */
 504                 if (real_uid == 0 && sep->se_family == AF_INET
 505                  && (pwd = getpwnam(sep->se_user)) != NULL
 506                  && pwd->pw_uid == 0
 507                 ) {
 508                         r = bindresvport(fd, &sep->se_lsa->u.sin);
 509                 } else {
 510                         r = bind(fd, &sep->se_lsa->u.sa, sep->se_lsa->len);
 511                 }
 512                 if (r == 0) {
 513                         int saveerrno = errno;
 514                         /* update lsa with port# */
 515                         getsockname(fd, &sep->se_lsa->u.sa, &sep->se_lsa->len);
 516                         errno = saveerrno;
 517                 }
 518         } else
 519 #endif
 520         {
 521                 if (sep->se_family == AF_UNIX) {
 522                         struct sockaddr_un *sun;
 523                         sun = (struct sockaddr_un*)&(sep->se_lsa->u.sa);
 524                         unlink(sun->sun_path);
 525                 }
 526                 r = bind(fd, &sep->se_lsa->u.sa, sep->se_lsa->len);
 527         }
 528         if (r < 0) {
 529                 bb_perror_msg("%s/%s: bind",
 530                                 sep->se_service, sep->se_proto);
 531                 close(fd);
 532                 rearm_alarm();
 533                 return;
 534         }
 535         if (sep->se_socktype == SOCK_STREAM)
 536                 listen(fd, global_queuelen);
 537
 538         add_fd_to_set(fd);
 539         sep->se_fd = fd;
 540 }
 541
 542 static int reopen_config_file(void)
 543 {
 544         free(default_local_hostname);
 545         default_local_hostname = xstrdup("*");
 546         if (fconfig != NULL)
 547                 fclose(fconfig);
 548         config_lineno = 0;
 549         fconfig = fopen_or_warn(config_filename, "r");
 550         return (fconfig != NULL);
 551 }
 552
 553 static void close_config_file(void)
 554 {
 555         if (fconfig) {
 556                 fclose(fconfig);
 557                 fconfig = NULL;
 558         }
 559 }
 560
 561 static char *next_line(void)
 562 {
 563         if (fgets(line, LINE_SIZE, fconfig) == NULL)
 564                 return NULL;
 565         config_lineno++;
 566         *strchrnul(line, '\n') = '\0';
 567         return line;
 568 }
 569
 570 static char *next_word(char **cpp)
 571 {
 572         char *start;
 573         char *cp = *cpp;
 574
 575         if (cp == NULL)
 576                 return NULL;
 577  again:
 578         while (*cp == ' ' || *cp == '\t')
 579                 cp++;
 580         if (*cp == '\0') {
 581                 int c = getc(fconfig);
 582                 ungetc(c, fconfig);
 583                 if (c == ' ' || c == '\t') {
 584                         cp = next_line();
 585                         if (cp)
 586                                 goto again;
 587                 }
 588                 *cpp = NULL;
 589                 return NULL;
 590         }
 591         start = cp;
 592         while (*cp && *cp != ' ' && *cp != '\t')
 593                 cp++;
 594         if (*cp != '\0')
 595                 *cp++ = '\0';
 596
 597         *cpp = cp;
 598         return start;
 599 }
 600
 601 static void free_servtab_strings(servtab_t *cp)
 602 {
 603         int i;
 604
 605         free(cp->se_local_hostname);
 606         free(cp->se_service);
 607         free(cp->se_proto);
 608         free(cp->se_user);
 609         free(cp->se_group);
 610         free(cp->se_lsa); /* not a string in fact */
 611         free(cp->se_program);
 612         for (i = 0; i < MAXARGV; i++)
 613                 free(cp->se_argv[i]);
 614 }
 615
 616 static servtab_t *new_servtab(void)
 617 {
 618         servtab_t *newtab = xzalloc(sizeof(servtab_t));
 619         newtab->se_fd = -1; /* paranoia */
 620         return newtab;
 621 }
 622
 623 static servtab_t *dup_servtab(servtab_t *sep)
 624 {
 625         servtab_t *newtab;
 626         int argc;
 627
 628         newtab = new_servtab();
 629         *newtab = *sep; /* struct copy */
 630         /* deep-copying strings */
 631         newtab->se_service = xstrdup(newtab->se_service);
 632         newtab->se_proto = xstrdup(newtab->se_proto);
 633         newtab->se_user = xstrdup(newtab->se_user);
 634         newtab->se_group = xstrdup(newtab->se_group);
 635         newtab->se_program = xstrdup(newtab->se_program);
 636         for (argc = 0; argc <= MAXARGV; argc++)
 637                 newtab->se_argv[argc] = xstrdup(newtab->se_argv[argc]);
 638         /* NB: se_fd, se_hostaddr and se_next are always
 639          * overwrittend by callers, so we don't bother resetting them
 640          * to NULL/0/-1 etc */
 641
 642         return newtab;
 643 }
 644
 645 /* gcc generates much more code if this is inlined */
 646 static NOINLINE servtab_t *parse_one_line(void)
 647 {
 648         int argc;
 649         char *p, *cp, *arg;
 650         char *hostdelim;
 651         servtab_t *sep;
 652         servtab_t *nsep;
 653  new:
 654         sep = new_servtab();
 655  more:
 656         while ((cp = next_line()) && *cp == '#')
 657                 continue; /* skip comment lines */
 658         if (cp == NULL) {
 659                 free(sep);
 660                 return NULL;
 661         }
 662
 663         arg = next_word(&cp);
 664         if (arg == NULL) /* a blank line. */
 665                 goto more;
 666
 667         /* [host:]service socktype proto wait user[:group] prog [args] */
 668         /* Check for "host:...." line */
 669         hostdelim = strrchr(arg, ':');
 670         if (hostdelim) {
 671                 *hostdelim = '\0';
 672                 sep->se_local_hostname = xstrdup(arg);
 673                 arg = hostdelim + 1;
 674                 if (*arg == '\0') {
 675                         arg = next_word(&cp);
 676                         if (arg == NULL) {
 677                                 /* Line has just "host:", change the
 678                                  * default host for the following lines. */
 679                                 free(default_local_hostname);
 680                                 default_local_hostname = sep->se_local_hostname;
 681                                 goto more;
 682                         }
 683                 }
 684         } else
 685                 sep->se_local_hostname = xstrdup(default_local_hostname);
 686
 687         /* service socktype proto wait user[:group] prog [args] */
 688         sep->se_service = xstrdup(arg);
 689         /* socktype proto wait user[:group] prog [args] */
 690         arg = next_word(&cp);
 691         if (arg == NULL) {
 692  parse_err:
 693                 bb_error_msg("parse error on line %u, line is ignored",
 694                                 config_lineno);
 695                 free_servtab_strings(sep);
 696                 /* Just "goto more" can make sep to carry over e.g.
 697                  * "rpc"-ness (by having se_rpcver_lo != 0).
 698                  * We will be more paranoid: */
 699                 free(sep);
 700                 goto new;
 701         }
 702         {
 703                 static int8_t SOCK_xxx[] ALIGN1 = {
 704                         -1,
 705                         SOCK_STREAM, SOCK_DGRAM, SOCK_RDM,
 706                         SOCK_SEQPACKET, SOCK_RAW
 707                 };
 708                 sep->se_socktype = SOCK_xxx[1 + index_in_strings(
 709                         "stream""\0" "dgram""\0" "rdm""\0"
 710                         "seqpacket""\0" "raw""\0"
 711                         , arg)];
 712         }
 713
 714         /* {unix,[rpc/]{tcp,udp}[6]} wait user[:group] prog [args] */
 715         sep->se_proto = arg = xstrdup(next_word(&cp));
 716         if (arg == NULL)
 717                 goto parse_err;
 718         if (strcmp(arg, "unix") == 0) {
 719                 sep->se_family = AF_UNIX;
 720         } else {
 721                 char *six;
 722                 sep->se_family = AF_INET;
 723                 six = last_char_is(arg, '6');
 724                 if (six) {
 725 #if ENABLE_FEATURE_IPV6
 726                         *six = '\0';
 727                         sep->se_family = AF_INET6;
 728 #else
 729                         bb_error_msg("%s: no support for IPv6", sep->se_proto);
 730                         goto parse_err;
 731 #endif
 732                 }
 733                 if (strncmp(arg, "rpc/", 4) == 0) {
 734 #if ENABLE_FEATURE_INETD_RPC
 735                         unsigned n;
 736                         arg += 4;
 737                         p = strchr(sep->se_service, '/');
 738                         if (p == NULL) {
 739                                 bb_error_msg("no rpc version: '%s'", sep->se_service);
 740                                 goto parse_err;
 741                         }
 742                         *p++ = '\0';
 743                         n = bb_strtou(p, &p, 10);
 744                         if (n > INT_MAX) {
 745  bad_ver_spec:
 746                                 bb_error_msg("bad rpc version");
 747                                 goto parse_err;
 748                         }
 749                         sep->se_rpcver_lo = sep->se_rpcver_hi = n;
 750                         if (*p == '-') {
 751                                 p++;
 752                                 n = bb_strtou(p, &p, 10);
 753                                 if (n > INT_MAX || n < sep->se_rpcver_lo)
 754                                         goto bad_ver_spec;
 755                                 sep->se_rpcver_hi = n;
 756                         }
 757                         if (*p != '\0')
 758                                 goto bad_ver_spec;
 759 #else
 760                         bb_error_msg("no support for rpc services");
 761                         goto parse_err;
 762 #endif
 763                 }
 764                 /* we don't really need getprotobyname()! */
 765                 if (strcmp(arg, "tcp") == 0)
 766                         sep->se_proto_no = IPPROTO_TCP; /* = 6 */
 767                 if (strcmp(arg, "udp") == 0)
 768                         sep->se_proto_no = IPPROTO_UDP; /* = 17 */
 769                 if (six)
 770                         *six = '6';
 771                 if (!sep->se_proto_no) /* not tcp/udp?? */
 772                         goto parse_err;
 773         }
 774
 775         /* [no]wait[.max] user[:group] prog [args] */
 776         arg = next_word(&cp);
 777         if (arg == NULL)
 778                 goto parse_err;
 779         sep->se_max = max_concurrency;
 780         p = strchr(arg, '.');
 781         if (p) {
 782                 *p++ = '\0';
 783                 sep->se_max = bb_strtou(p, NULL, 10);
 784                 if (errno)
 785                         goto parse_err;
 786         }
 787         sep->se_wait = (arg[0] != 'n' || arg[1] != 'o');
 788         if (!sep->se_wait) /* "no" seen */
 789                 arg += 2;
 790         if (strcmp(arg, "wait") != 0)
 791                 goto parse_err;
 792
 793         /* user[:group] prog [args] */
 794         sep->se_user = xstrdup(next_word(&cp));
 795         if (sep->se_user == NULL)
 796                 goto parse_err;
 797         arg = strchr(sep->se_user, '.');
 798         if (arg == NULL)
 799                 arg = strchr(sep->se_user, ':');
 800         if (arg) {
 801                 *arg++ = '\0';
 802                 sep->se_group = xstrdup(arg);
 803         }
 804
 805         /* prog [args] */
 806         sep->se_program = xstrdup(next_word(&cp));
 807         if (sep->se_program == NULL)
 808                 goto parse_err;
 809 #ifdef INETD_BUILTINS_ENABLED
 810         if (strcmp(sep->se_program, "internal") == 0
 811          && strlen(sep->se_service) <= 7
 812          && (sep->se_socktype == SOCK_STREAM
 813              || sep->se_socktype == SOCK_DGRAM)
 814         ) {
 815                 int i;
 816                 for (i = 0; i < ARRAY_SIZE(builtins); i++)
 817                         if (strncmp(builtins[i].bi_service7, sep->se_service, 7) == 0)
 818                                 goto found_bi;
 819                 bb_error_msg("unknown internal service %s", sep->se_service);
 820                 goto parse_err;
 821  found_bi:
 822                 sep->se_builtin = &builtins[i];
 823                 /* stream builtins must be "nowait", dgram must be "wait" */
 824                 if (sep->se_wait != (sep->se_socktype == SOCK_DGRAM))
 825                         goto parse_err;
 826         }
 827 #endif
 828         argc = 0;
 829         while ((arg = next_word(&cp)) != NULL && argc < MAXARGV)
 830                 sep->se_argv[argc++] = xstrdup(arg);
 831
 832         /* catch mixups. "<service> stream udp ..." == wtf */
 833         if (sep->se_socktype == SOCK_STREAM) {
 834                 if (sep->se_proto_no == IPPROTO_UDP)
 835                         goto parse_err;
 836         }
 837         if (sep->se_socktype == SOCK_DGRAM) {
 838                 if (sep->se_proto_no == IPPROTO_TCP)
 839                         goto parse_err;
 840         }
 841
 842         /* check if the hostname specifier is a comma separated list
 843          * of hostnames. we'll make new entries for each address. */
 844         while ((hostdelim = strrchr(sep->se_local_hostname, ',')) != NULL) {
 845                 nsep = dup_servtab(sep);
 846                 /* NUL terminate the hostname field of the existing entry,
 847                  * and make a dup for the new entry. */
 848                 *hostdelim++ = '\0';
 849                 nsep->se_local_hostname = xstrdup(hostdelim);
 850                 nsep->se_next = sep->se_next;
 851                 sep->se_next = nsep;
 852         }
 853
 854         /* was doing it here: */
 855         /* DNS resolution, create copies for each IP address */
 856         /* IPv6-ization destroyed it :( */
 857
 858         return sep;
 859 }
 860
 861 static servtab_t *insert_in_servlist(servtab_t *cp)
 862 {
 863         servtab_t *sep;
 864         sigset_t omask;
 865
 866         sep = new_servtab();
 867         *sep = *cp; /* struct copy */
 868         sep->se_fd = -1;
 869 #if ENABLE_FEATURE_INETD_RPC
 870         sep->se_rpcprog = -1;
 871 #endif
 872         block_CHLD_HUP_ALRM(&omask);
 873         sep->se_next = serv_list;
 874         serv_list = sep;
 875         restore_sigmask(&omask);
 876         return sep;
 877 }
 878
 879 static int same_serv_addr_proto(servtab_t *old, servtab_t *new)
 880 {
 881         if (strcmp(old->se_local_hostname, new->se_local_hostname) != 0)
 882                 return 0;
 883         if (strcmp(old->se_service, new->se_service) != 0)
 884                 return 0;
 885         if (strcmp(old->se_proto, new->se_proto) != 0)
 886                 return 0;
 887         return 1;
 888 }
 889
 890 static void reread_config_file(int sig ATTRIBUTE_UNUSED)
 891 {
 892         servtab_t *sep, *cp, **sepp;
 893         len_and_sockaddr *lsa;
 894         sigset_t omask;
 895         unsigned n;
 896         uint16_t port;
 897
 898         if (!reopen_config_file())
 899                 return;
 900         for (sep = serv_list; sep; sep = sep->se_next)
 901                 sep->se_checked = 0;
 902
 903         goto first_line;
 904         while (1) {
 905                 if (cp == NULL) {
 906  first_line:
 907                         cp = parse_one_line();
 908                         if (cp == NULL)
 909                                 break;
 910                 }
 911                 for (sep = serv_list; sep; sep = sep->se_next)
 912                         if (same_serv_addr_proto(sep, cp))
 913                                 goto equal_servtab;
 914                 /* not an "equal" servtab */
 915                 sep = insert_in_servlist(cp);
 916                 goto after_check;
 917  equal_servtab:
 918                 {
 919                         int i;
 920
 921                         block_CHLD_HUP_ALRM(&omask);
 922 #if ENABLE_FEATURE_INETD_RPC
 923                         if (is_rpc_service(sep))
 924                                 unregister_rpc(sep);
 925                         sep->se_rpcver_lo = cp->se_rpcver_lo;
 926                         sep->se_rpcver_hi = cp->se_rpcver_hi;
 927 #endif
 928                         if (cp->se_wait == 0) {
 929                                 /* New config says "nowait". If old one
 930                                  * was "wait", we currently may be waiting
 931                                  * for a child (and not accepting connects).
 932                                  * Stop waiting, start listening again.
 933                                  * (if it's not true, this op is harmless) */
 934                                 add_fd_to_set(sep->se_fd);
 935                         }
 936                         sep->se_wait = cp->se_wait;
 937                         sep->se_max = cp->se_max;
 938                         /* string fields need more love - we don't want to leak them */
 939 #define SWAP(type, a, b) do { type c = (type)a; a = (type)b; b = (type)c; } while (0)
 940                         SWAP(char*, sep->se_user, cp->se_user);
 941                         SWAP(char*, sep->se_group, cp->se_group);
 942                         SWAP(char*, sep->se_program, cp->se_program);
 943                         for (i = 0; i < MAXARGV; i++)
 944                                 SWAP(char*, sep->se_argv[i], cp->se_argv[i]);
 945 #undef SWAP
 946                         restore_sigmask(&omask);
 947                         free_servtab_strings(cp);
 948                 }
 949  after_check:
 950                 /* cp->string_fields are consumed by insert_in_servlist()
 951                  * or freed at this point, cp itself is not yet freed. */
 952                 sep->se_checked = 1;
 953
 954                 /* create new len_and_sockaddr */
 955                 switch (sep->se_family) {
 956                         struct sockaddr_un *sun;
 957                 case AF_UNIX:
 958                         lsa = xzalloc_lsa(AF_UNIX);
 959                         sun = (struct sockaddr_un*)&lsa->u.sa;
 960                         safe_strncpy(sun->sun_path, sep->se_service, sizeof(sun->sun_path));
 961                         break;
 962
 963                 default: /* case AF_INET, case AF_INET6 */
 964                         n = bb_strtou(sep->se_service, NULL, 10);
 965 #if ENABLE_FEATURE_INETD_RPC
 966                         if (is_rpc_service(sep)) {
 967                                 sep->se_rpcprog = n;
 968                                 if (errno) { /* se_service is not numeric */
 969                                         struct rpcent *rp = getrpcbyname(sep->se_service);
 970                                         if (rp == NULL) {
 971                                                 bb_error_msg("%s: unknown rpc service", sep->se_service);
 972                                                 goto next_cp;
 973                                         }
 974                                         sep->se_rpcprog = rp->r_number;
 975                                 }
 976                                 if (sep->se_fd == -1)
 977                                         prepare_socket_fd(sep);
 978                                 if (sep->se_fd != -1)
 979                                         register_rpc(sep);
 980                                 goto next_cp;
 981                         }
 982 #endif
 983                         /* what port to listen on? */
 984                         port = htons(n);
 985                         if (errno || n > 0xffff) { /* se_service is not numeric */
 986                                 char protoname[4];
 987                                 struct servent *sp;
 988                                 /* can result only in "tcp" or "udp": */
 989                                 safe_strncpy(protoname, sep->se_proto, 4);
 990                                 sp = getservbyname(sep->se_service, protoname);
 991                                 if (sp == NULL) {
 992                                         bb_error_msg("%s/%s: unknown service",
 993                                                         sep->se_service, sep->se_proto);
 994                                         goto next_cp;
 995                                 }
 996                                 port = sp->s_port;
 997                         }
 998                         if (LONE_CHAR(sep->se_local_hostname, '*')) {
 999                                 lsa = xzalloc_lsa(sep->se_family);
1000                                 set_nport(lsa, port);
1001                         } else {
1002                                 lsa = host_and_af2sockaddr(sep->se_local_hostname,
1003                                                 ntohs(port), sep->se_family);
1004                                 if (!lsa) {
1005                                         bb_error_msg("%s/%s: unknown host '%s'",
1006                                                 sep->se_service, sep->se_proto,
1007                                                 sep->se_local_hostname);
1008                                         goto next_cp;
1009                                 }
1010                         }
1011                         break;
1012                 } /* end of "switch (sep->se_family)" */
1013
1014                 /* did lsa change? Then close/open */
1015                 if (sep->se_lsa == NULL
1016                  || lsa->len != sep->se_lsa->len
1017                  || memcmp(&lsa->u.sa, &sep->se_lsa->u.sa, lsa->len) != 0
1018                 ) {
1019                         remove_fd_from_set(sep->se_fd);
1020                         maybe_close(sep->se_fd);
1021                         free(sep->se_lsa);
1022                         sep->se_lsa = lsa;
1023                         sep->se_fd = -1;
1024                 } else {
1025                         free(lsa);
1026                 }
1027                 if (sep->se_fd == -1)
1028                         prepare_socket_fd(sep);
1029  next_cp:
1030                 sep = cp->se_next;
1031                 free(cp);
1032                 cp = sep;
1033         } /* end of "while (1) parse lines" */
1034         close_config_file();
1035
1036         /* Purge anything not looked at above - these are stale entries,
1037          * new config file doesnt have them. */
1038         block_CHLD_HUP_ALRM(&omask);
1039         sepp = &serv_list;
1040         while ((sep = *sepp)) {
1041                 if (sep->se_checked) {
1042                         sepp = &sep->se_next;
1043                         continue;
1044                 }
1045                 *sepp = sep->se_next;
1046                 remove_fd_from_set(sep->se_fd);
1047                 maybe_close(sep->se_fd);
1048 #if ENABLE_FEATURE_INETD_RPC
1049                 if (is_rpc_service(sep))
1050                         unregister_rpc(sep);
1051 #endif
1052                 if (sep->se_family == AF_UNIX)
1053                         unlink(sep->se_service);
1054                 free_servtab_strings(sep);
1055                 free(sep);
1056         }
1057         restore_sigmask(&omask);
1058 }
1059
1060 static void reap_child(int sig ATTRIBUTE_UNUSED)
1061 {
1062         pid_t pid;
1063         int status;
1064         servtab_t *sep;
1065         int save_errno = errno;
1066
1067         for (;;) {
1068                 pid = wait_any_nohang(&status);
1069                 if (pid <= 0)
1070                         break;
1071                 for (sep = serv_list; sep; sep = sep->se_next)
1072                         if (sep->se_wait == pid) {
1073                                 /* One of our "wait" services */
1074                                 if (WIFEXITED(status) && WEXITSTATUS(status))
1075                                         bb_error_msg("%s: exit status 0x%x",
1076                                                         sep->se_program, WEXITSTATUS(status));
1077                                 else if (WIFSIGNALED(status))
1078                                         bb_error_msg("%s: exit signal 0x%x",
1079                                                         sep->se_program, WTERMSIG(status));
1080                                 sep->se_wait = 1;
1081                                 add_fd_to_set(sep->se_fd);
1082                         }
1083         }
1084         errno = save_errno;
1085 }
1086
1087 static void retry_network_setup(int sig ATTRIBUTE_UNUSED)
1088 {
1089         servtab_t *sep;
1090
1091         alarm_armed = 0;
1092         for (sep = serv_list; sep; sep = sep->se_next) {
1093                 if (sep->se_fd == -1) {
1094                         prepare_socket_fd(sep);
1095 #if ENABLE_FEATURE_INETD_RPC
1096                         if (sep->se_fd != -1 && is_rpc_service(sep))
1097                                 register_rpc(sep);
1098 #endif
1099                 }
1100         }
1101 }
1102
1103 static void clean_up_and_exit(int sig ATTRIBUTE_UNUSED)
1104 {
1105         servtab_t *sep;
1106
1107         /* XXX signal race walking sep list */
1108         for (sep = serv_list; sep; sep = sep->se_next) {
1109                 if (sep->se_fd == -1)
1110                         continue;
1111
1112                 switch (sep->se_family) {
1113                 case AF_UNIX:
1114                         unlink(sep->se_service);
1115                         break;
1116                 default: /* case AF_INET, AF_INET6 */
1117 #if ENABLE_FEATURE_INETD_RPC
1118                         if (sep->se_wait == 1 && is_rpc_service(sep))
1119                                 unregister_rpc(sep);   /* XXX signal race */
1120 #endif
1121                         break;
1122                 }
1123                 if (ENABLE_FEATURE_CLEAN_UP)
1124                         close(sep->se_fd);
1125         }
1126         remove_pidfile(_PATH_INETDPID);
1127         exit(0);
1128 }
1129
1130 int inetd_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
1131 int inetd_main(int argc ATTRIBUTE_UNUSED, char **argv)
1132 {
1133         struct sigaction sa, saved_pipe_handler;
1134         servtab_t *sep, *sep2;
1135         struct passwd *pwd;
1136         struct group *grp = grp; /* for compiler */
1137         int opt;
1138         pid_t pid;
1139         sigset_t omask;
1140
1141         INIT_G();
1142
1143         real_uid = getuid();
1144         if (real_uid != 0) /* run by non-root user */
1145                 config_filename = NULL;
1146
1147         opt_complementary = "R+:q+"; /* -q N, -R N */
1148         opt = getopt32(argv, "R:feq:", &max_concurrency, &global_queuelen);
1149         argv += optind;
1150         //argc -= optind;
1151         if (argv[0])
1152                 config_filename = argv[0];
1153         if (config_filename == NULL)
1154                 bb_error_msg_and_die("non-root must specify config file");
1155         if (!(opt & 2))
1156                 bb_daemonize_or_rexec(0, argv - optind);
1157         else
1158                 bb_sanitize_stdio();
1159         if (!(opt & 4)) {
1160                 openlog(applet_name, LOG_PID | LOG_NOWAIT, LOG_DAEMON);
1161                 logmode = LOGMODE_SYSLOG;
1162         }
1163
1164         if (real_uid == 0) {
1165                 /* run by root, ensure groups vector gets trashed */
1166                 gid_t gid = getgid();
1167                 setgroups(1, &gid);
1168         }
1169
1170         write_pidfile(_PATH_INETDPID);
1171
1172         /* never fails under Linux (except if you pass it bad arguments) */
1173         getrlimit(RLIMIT_NOFILE, &rlim_ofile);
1174         rlim_ofile_cur = rlim_ofile.rlim_cur;
1175         if (rlim_ofile_cur == RLIM_INFINITY)    /* ! */
1176                 rlim_ofile_cur = OPEN_MAX;
1177
1178         memset(&sa, 0, sizeof(sa));
1179         /*sigemptyset(&sa.sa_mask); - memset did it */
1180         sigaddset(&sa.sa_mask, SIGALRM);
1181         sigaddset(&sa.sa_mask, SIGCHLD);
1182         sigaddset(&sa.sa_mask, SIGHUP);
1183         sa.sa_handler = retry_network_setup;
1184         sigaction_set(SIGALRM, &sa);
1185         sa.sa_handler = reread_config_file;
1186         sigaction_set(SIGHUP, &sa);
1187         sa.sa_handler = reap_child;
1188         sigaction_set(SIGCHLD, &sa);
1189         sa.sa_handler = clean_up_and_exit;
1190         sigaction_set(SIGTERM, &sa);
1191         sa.sa_handler = clean_up_and_exit;
1192         sigaction_set(SIGINT, &sa);
1193         sa.sa_handler = SIG_IGN;
1194         sigaction(SIGPIPE, &sa, &saved_pipe_handler);
1195
1196         reread_config_file(SIGHUP); /* load config from file */
1197
1198         for (;;) {
1199                 int ready_fd_cnt;
1200                 int ctrl, accepted_fd, new_udp_fd;
1201                 fd_set readable;
1202
1203                 if (maxsock < 0)
1204                         recalculate_maxsock();
1205
1206                 readable = allsock; /* struct copy */
1207                 /* if there are no fds to wait on, we will block
1208                  * until signal wakes us up */
1209                 ready_fd_cnt = select(maxsock + 1, &readable, NULL, NULL, NULL);
1210                 if (ready_fd_cnt < 0) {
1211                         if (errno != EINTR) {
1212                                 bb_perror_msg("select");
1213                                 sleep(1);
1214                         }
1215                         continue;
1216                 }
1217
1218                 for (sep = serv_list; ready_fd_cnt && sep; sep = sep->se_next) {
1219                         if (sep->se_fd == -1 || !FD_ISSET(sep->se_fd, &readable))
1220                                 continue;
1221
1222                         ready_fd_cnt--;
1223                         ctrl = sep->se_fd;
1224                         accepted_fd = -1;
1225                         new_udp_fd = -1;
1226                         if (!sep->se_wait) {
1227                                 if (sep->se_socktype == SOCK_STREAM) {
1228                                         ctrl = accepted_fd = accept(sep->se_fd, NULL, NULL);
1229                                         if (ctrl < 0) {
1230                                                 if (errno != EINTR)
1231                                                         bb_perror_msg("accept (for %s)", sep->se_service);
1232                                                 continue;
1233                                         }
1234                                 }
1235                                 /* "nowait" udp */
1236                                 if (sep->se_socktype == SOCK_DGRAM
1237                                  && sep->se_family != AF_UNIX
1238                                 ) {
1239 /* How udp "nowait" works:
1240  * child peeks at (received and buffered by kernel) UDP packet,
1241  * performs connect() on the socket so that it is linked only
1242  * to this peer. But this also affects parent, because descriptors
1243  * are shared after fork() a-la dup(). When parent performs
1244  * select(), it will see this descriptor connected to the peer (!)
1245  * and still readable, will act on it and mess things up
1246  * (can create many copies of same child, etc).
1247  * Parent must create and use new socket instead. */
1248                                         new_udp_fd = socket(sep->se_family, SOCK_DGRAM, 0);
1249                                         if (new_udp_fd < 0) { /* error: eat packet, forget about it */
1250  udp_err:
1251                                                 recv(sep->se_fd, line, LINE_SIZE, MSG_DONTWAIT);
1252                                                 continue;
1253                                         }
1254                                         setsockopt_reuseaddr(new_udp_fd);
1255                                         /* TODO: better do bind after vfork in parent,
1256                                          * so that we don't have two wildcard bound sockets
1257                                          * even for a brief moment? */
1258                                         if (bind(new_udp_fd, &sep->se_lsa->u.sa, sep->se_lsa->len) < 0) {
1259                                                 close(new_udp_fd);
1260                                                 goto udp_err;
1261                                         }
1262                                 }
1263                         }
1264
1265                         block_CHLD_HUP_ALRM(&omask);
1266                         pid = 0;
1267 #ifdef INETD_BUILTINS_ENABLED
1268                         /* do we need to fork? */
1269                         if (sep->se_builtin == NULL
1270                          || (sep->se_socktype == SOCK_STREAM
1271                              && sep->se_builtin->bi_fork))
1272 #endif
1273                         {
1274                                 if (sep->se_max != 0) {
1275                                         if (++sep->se_count == 1)
1276                                                 sep->se_time = monotonic_sec();
1277                                         else if (sep->se_count >= sep->se_max) {
1278                                                 unsigned now = monotonic_sec();
1279                                                 /* did we accumulate se_max connects too quickly? */
1280                                                 if (now - sep->se_time <= CNT_INTERVAL) {
1281                                                         bb_error_msg("%s/%s: too many connections, pausing",
1282                                                                         sep->se_service, sep->se_proto);
1283                                                         remove_fd_from_set(sep->se_fd);
1284                                                         close(sep->se_fd);
1285                                                         sep->se_fd = -1;
1286                                                         sep->se_count = 0;
1287                                                         rearm_alarm(); /* will revive it in RETRYTIME sec */
1288                                                         restore_sigmask(&omask);
1289                                                         maybe_close(accepted_fd);
1290                                                         continue; /* -> check next fd in fd set */
1291                                                 }
1292                                                 sep->se_count = 0;
1293                                         }
1294                                 }
1295                                 /* on NOMMU, streamed chargen
1296                                  * builtin wouldn't work, but it is
1297                                  * not allowed on NOMMU (ifdefed out) */
1298 #ifdef INETD_BUILTINS_ENABLED
1299                                 if (BB_MMU && sep->se_builtin)
1300                                         pid = fork();
1301                                 else
1302 #endif
1303                                         pid = vfork();
1304
1305                                 if (pid < 0) { /* fork error */
1306                                         bb_perror_msg("fork");
1307                                         sleep(1);
1308                                         restore_sigmask(&omask);
1309                                         maybe_close(accepted_fd);
1310                                         continue; /* -> check next fd in fd set */
1311                                 }
1312                                 if (pid == 0)
1313                                         pid--; /* -1: "we did fork and we are child" */
1314                         }
1315                         /* if pid == 0 here, we never forked */
1316
1317                         if (pid > 0) { /* parent */
1318                                 if (sep->se_wait) {
1319                                         /* tcp wait: we passed listening socket to child,
1320                                          * will wait for child to terminate */
1321                                         sep->se_wait = pid;
1322                                         remove_fd_from_set(sep->se_fd);
1323                                 }
1324                                 if (new_udp_fd >= 0) {
1325                                         /* udp nowait: child connected the socket,
1326                                          * we created and will use new, unconnected one */
1327                                         xmove_fd(new_udp_fd, sep->se_fd);
1328                                 }
1329                                 restore_sigmask(&omask);
1330                                 maybe_close(accepted_fd);
1331                                 continue; /* -> check next fd in fd set */
1332                         }
1333
1334                         /* we are either child or didn't vfork at all */
1335 #ifdef INETD_BUILTINS_ENABLED
1336                         if (sep->se_builtin) {
1337                                 if (pid) { /* "pid" is -1: we did vfork */
1338                                         close(sep->se_fd); /* listening socket */
1339                                         logmode = 0; /* make xwrite etc silent */
1340                                 }
1341                                 restore_sigmask(&omask);
1342                                 if (sep->se_socktype == SOCK_STREAM)
1343                                         sep->se_builtin->bi_stream_fn(ctrl, sep);
1344                                 else
1345                                         sep->se_builtin->bi_dgram_fn(ctrl, sep);
1346                                 if (pid) /* we did vfork */
1347                                         _exit(1);
1348                                 maybe_close(accepted_fd);
1349                                 continue; /* -> check next fd in fd set */
1350                         }
1351 #endif
1352                         /* child */
1353                         setsid();
1354                         /* "nowait" udp */
1355                         if (new_udp_fd >= 0) {
1356                                 len_and_sockaddr *lsa = xzalloc_lsa(sep->se_family);
1357                                 /* peek at the packet and remember peer addr */
1358                                 int r = recvfrom(ctrl, NULL, 0, MSG_PEEK|MSG_DONTWAIT,
1359                                         &lsa->u.sa, &lsa->len);
1360                                 if (r < 0)
1361                                         goto do_exit1;
1362                                 /* make this socket "connected" to peer addr:
1363                                  * only packets from this peer will be recv'ed,
1364                                  * and bare write()/send() will work on it */
1365                                 connect(ctrl, &lsa->u.sa, lsa->len);
1366                                 free(lsa);
1367                         }
1368                         /* prepare env and exec program */
1369                         pwd = getpwnam(sep->se_user);
1370                         if (pwd == NULL) {
1371                                 bb_error_msg("%s: no such user", sep->se_user);
1372                                 goto do_exit1;
1373                         }
1374                         if (sep->se_group && (grp = getgrnam(sep->se_group)) == NULL) {
1375                                 bb_error_msg("%s: no such group", sep->se_group);
1376                                 goto do_exit1;
1377                         }
1378                         if (real_uid != 0 && real_uid != pwd->pw_uid) {
1379                                 /* a user running private inetd */
1380                                 bb_error_msg("non-root must run services as himself");
1381                                 goto do_exit1;
1382                         }
1383                         if (pwd->pw_uid) {
1384                                 if (sep->se_group)
1385                                         pwd->pw_gid = grp->gr_gid;
1386                                 /* initgroups, setgid, setuid: */
1387                                 change_identity(pwd);
1388                         } else if (sep->se_group) {
1389                                 xsetgid(grp->gr_gid);
1390                                 setgroups(1, &grp->gr_gid);
1391                         }
1392                         if (rlim_ofile.rlim_cur != rlim_ofile_cur)
1393                                 if (setrlimit(RLIMIT_NOFILE, &rlim_ofile) < 0)
1394                                         bb_perror_msg("setrlimit");
1395                         closelog();
1396                         xmove_fd(ctrl, 0);
1397                         xdup2(0, 1);
1398                         xdup2(0, 2);
1399                         /* NB: among others, this loop closes listening socket
1400                          * for nowait stream children */
1401                         for (sep2 = serv_list; sep2; sep2 = sep2->se_next)
1402                                 maybe_close(sep2->se_fd);
1403                         sigaction_set(SIGPIPE, &saved_pipe_handler);
1404                         restore_sigmask(&omask);
1405                         BB_EXECVP(sep->se_program, sep->se_argv);
1406                         bb_perror_msg("exec %s", sep->se_program);
1407  do_exit1:
1408                         /* eat packet in udp case */
1409                         if (sep->se_socktype != SOCK_STREAM)
1410                                 recv(0, line, LINE_SIZE, MSG_DONTWAIT);
1411                         _exit(1);
1412                 } /* for (sep = servtab...) */
1413         } /* for (;;) */
1414 }
1415
1416 /*
1417  * Internet services provided internally by inetd:
1418  */
1419 #if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_ECHO
1420 /* Echo service -- echo data back. */
1421 /* ARGSUSED */
1422 static void echo_stream(int s, servtab_t *sep ATTRIBUTE_UNUSED)
1423 {
1424 #if BB_MMU
1425         while (1) {
1426                 ssize_t sz = safe_read(s, line, LINE_SIZE);
1427                 if (sz <= 0)
1428                         break;
1429                 xwrite(s, line, sz);
1430         }
1431 #else
1432         /* We are after vfork here! */
1433         static const char *const args[] = { "cat", NULL };
1434         /* move network socket to stdin */
1435         xmove_fd(s, STDIN_FILENO);
1436         xdup2(STDIN_FILENO, STDOUT_FILENO);
1437         /* no error messages please... */
1438         xmove_fd(xopen("/dev/null", O_WRONLY), STDERR_FILENO);
1439         BB_EXECVP("cat", (char**)args);
1440         /* on failure we return to main, which does exit(1) */
1441 #endif
1442 }
1443 static void echo_dg(int s, servtab_t *sep)
1444 {
1445         enum { BUFSIZE = 12*1024 }; /* for jumbo sized packets! :) */
1446         char *buf = xmalloc(BUFSIZE); /* too big for stack */
1447         int sz;
1448         len_and_sockaddr *lsa = alloca(LSA_LEN_SIZE + sep->se_lsa->len);
1449
1450         lsa->len = sep->se_lsa->len;
1451         /* dgram builtins are non-forking - DONT BLOCK! */
1452         sz = recvfrom(s, buf, BUFSIZE, MSG_DONTWAIT, &lsa->u.sa, &lsa->len);
1453         if (sz > 0)
1454                 sendto(s, buf, sz, 0, &lsa->u.sa, lsa->len);
1455         free(buf);
1456 }
1457 #endif  /* FEATURE_INETD_SUPPORT_BUILTIN_ECHO */
1458
1459
1460 #if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_DISCARD
1461 /* Discard service -- ignore data. MMU arches only. */
1462 /* ARGSUSED */
1463 static void discard_stream(int s, servtab_t *sep ATTRIBUTE_UNUSED)
1464 {
1465 #if BB_MMU
1466         while (safe_read(s, line, LINE_SIZE) > 0)
1467                 continue;
1468 #else
1469         /* We are after vfork here! */
1470         static const char *const args[] = { "dd", "of=/dev/null", NULL };
1471         /* move network socket to stdin */
1472         xmove_fd(s, STDIN_FILENO);
1473         xdup2(STDIN_FILENO, STDOUT_FILENO);
1474         /* no error messages */
1475         xmove_fd(xopen("/dev/null", O_WRONLY), STDERR_FILENO);
1476         BB_EXECVP("dd", (char**)args);
1477         /* on failure we return to main, which does exit(1) */
1478 #endif
1479 }
1480 /* ARGSUSED */
1481 static void discard_dg(int s, servtab_t *sep ATTRIBUTE_UNUSED)
1482 {
1483         /* dgram builtins are non-forking - DONT BLOCK! */
1484         recv(s, line, LINE_SIZE, MSG_DONTWAIT);
1485 }
1486 #endif /* FEATURE_INETD_SUPPORT_BUILTIN_DISCARD */
1487
1488
1489 #if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_CHARGEN
1490 #define LINESIZ 72
1491 static void init_ring(void)
1492 {
1493         int i;
1494
1495         end_ring = ring;
1496         for (i = 0; i <= 128; ++i)
1497                 if (isprint(i))
1498                         *end_ring++ = i;
1499 }
1500 /* Character generator. MMU arches only. */
1501 /* ARGSUSED */
1502 static void chargen_stream(int s, servtab_t *sep ATTRIBUTE_UNUSED)
1503 {
1504         char *rs;
1505         int len;
1506         char text[LINESIZ + 2];
1507
1508         if (!end_ring) {
1509                 init_ring();
1510                 rs = ring;
1511         }
1512
1513         text[LINESIZ] = '\r';
1514         text[LINESIZ + 1] = '\n';
1515         rs = ring;
1516         for (;;) {
1517                 len = end_ring - rs;
1518                 if (len >= LINESIZ)
1519                         memmove(text, rs, LINESIZ);
1520                 else {
1521                         memmove(text, rs, len);
1522                         memmove(text + len, ring, LINESIZ - len);
1523                 }
1524                 if (++rs == end_ring)
1525                         rs = ring;
1526                 xwrite(s, text, sizeof(text));
1527         }
1528 }
1529 /* ARGSUSED */
1530 static void chargen_dg(int s, servtab_t *sep)
1531 {
1532         int len;
1533         char text[LINESIZ + 2];
1534         len_and_sockaddr *lsa = alloca(LSA_LEN_SIZE + sep->se_lsa->len);
1535
1536         /* Eat UDP packet which started it all */
1537         /* dgram builtins are non-forking - DONT BLOCK! */
1538         lsa->len = sep->se_lsa->len;
1539         if (recvfrom(s, text, sizeof(text), MSG_DONTWAIT, &lsa->u.sa, &lsa->len) < 0)
1540                 return;
1541
1542         if (!end_ring) {
1543                 init_ring();
1544                 ring_pos = ring;
1545         }
1546
1547         len = end_ring - ring_pos;
1548         if (len >= LINESIZ)
1549                 memmove(text, ring_pos, LINESIZ);
1550         else {
1551                 memmove(text, ring_pos, len);
1552                 memmove(text + len, ring, LINESIZ - len);
1553         }
1554         if (++ring_pos == end_ring)
1555                 ring_pos = ring;
1556         text[LINESIZ] = '\r';
1557         text[LINESIZ + 1] = '\n';
1558         sendto(s, text, sizeof(text), 0, &lsa->u.sa, lsa->len);
1559 }
1560 #endif /* FEATURE_INETD_SUPPORT_BUILTIN_CHARGEN */
1561
1562
1563 #if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_TIME
1564 /*
1565  * Return a machine readable date and time, in the form of the
1566  * number of seconds since midnight, Jan 1, 1900.  Since gettimeofday
1567  * returns the number of seconds since midnight, Jan 1, 1970,
1568  * we must add 2208988800 seconds to this figure to make up for
1569  * some seventy years Bell Labs was asleep.
1570  */
1571 static uint32_t machtime(void)
1572 {
1573         struct timeval tv;
1574
1575         gettimeofday(&tv, NULL);
1576         return htonl((uint32_t)(tv.tv_sec + 2208988800));
1577 }
1578 /* ARGSUSED */
1579 static void machtime_stream(int s, servtab_t *sep ATTRIBUTE_UNUSED)
1580 {
1581         uint32_t result;
1582
1583         result = machtime();
1584         full_write(s, &result, sizeof(result));
1585 }
1586 static void machtime_dg(int s, servtab_t *sep)
1587 {
1588         uint32_t result;
1589         len_and_sockaddr *lsa = alloca(LSA_LEN_SIZE + sep->se_lsa->len);
1590
1591         lsa->len = sep->se_lsa->len;
1592         if (recvfrom(s, line, LINE_SIZE, MSG_DONTWAIT, &lsa->u.sa, &lsa->len) < 0)
1593                 return;
1594
1595         result = machtime();
1596         sendto(s, &result, sizeof(result), 0, &lsa->u.sa, lsa->len);
1597 }
1598 #endif /* FEATURE_INETD_SUPPORT_BUILTIN_TIME */
1599
1600
1601 #if ENABLE_FEATURE_INETD_SUPPORT_BUILTIN_DAYTIME
1602 /* Return human-readable time of day */
1603 /* ARGSUSED */
1604 static void daytime_stream(int s, servtab_t *sep ATTRIBUTE_UNUSED)
1605 {
1606         time_t t;
1607
1608         t = time(NULL);
1609         fdprintf(s, "%.24s\r\n", ctime(&t));
1610 }
1611 static void daytime_dg(int s, servtab_t *sep)
1612 {
1613         time_t t;
1614         len_and_sockaddr *lsa = alloca(LSA_LEN_SIZE + sep->se_lsa->len);
1615
1616         lsa->len = sep->se_lsa->len;
1617         if (recvfrom(s, line, LINE_SIZE, MSG_DONTWAIT, &lsa->u.sa, &lsa->len) < 0)
1618                 return;
1619
1620         t = time(NULL);
1621         sprintf(line, "%.24s\r\n", ctime(&t));
1622         sendto(s, line, strlen(line), 0, &lsa->u.sa, lsa->len);
1623 }
1624 #endif /* FEATURE_INETD_SUPPORT_BUILTIN_DAYTIME */