3
0

inetd.c 46 KB

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