blob: d4c69e0f76a270b76d2e62fd66ebfc4059ef276f [file] [log] [blame]
Denis Vlasenkob933ac12007-04-03 12:09:46 +00001/* Based on ipsvd utilities written by Gerrit Pape <pape@smarden.org>
2 * which are released into public domain by the author.
3 * Homepage: http://smarden.sunsite.dk/ipsvd/
4 *
Denis Vlasenkod18f52b2008-03-02 12:53:15 +00005 * Copyright (C) 2007 Denys Vlasenko.
Denis Vlasenkob933ac12007-04-03 12:09:46 +00006 *
Denys Vlasenko0ef64bd2010-08-16 20:14:46 +02007 * Licensed under GPLv2, see file LICENSE in this source tree.
Denis Vlasenkob933ac12007-04-03 12:09:46 +00008 */
9
Denis Vlasenko4866e902008-03-17 09:17:27 +000010/* Based on ipsvd-0.12.1. This tcpsvd accepts all options
Denis Vlasenko02fd6682007-04-03 23:23:10 +000011 * which are supported by one from ipsvd-0.12.1, but not all are
12 * functional. See help text at the end of this file for details.
13 *
14 * Code inside "#ifdef SSLSVD" is for sslsvd and is currently unused.
15 *
Denis Vlasenkoaefed942008-03-17 08:35:44 +000016 * Busybox version exports TCPLOCALADDR instead of
17 * TCPLOCALIP + TCPLOCALPORT pair. ADDR more closely matches reality
18 * (which is "struct sockaddr_XXX". Port is not a separate entity,
19 * it's just a part of (AF_INET[6]) sockaddr!).
Denis Vlasenko02fd6682007-04-03 23:23:10 +000020 *
Denis Vlasenkoaefed942008-03-17 08:35:44 +000021 * TCPORIGDSTADDR is Busybox-specific addition.
Denis Vlasenko02fd6682007-04-03 23:23:10 +000022 *
23 * udp server is hacked up by reusing TCP code. It has the following
24 * limitation inherent in Unix DGRAM sockets implementation:
Denis Vlasenko64a15122007-04-04 10:16:15 +000025 * - local IP address is retrieved (using recvmsg voodoo) but
Denis Vlasenko02fd6682007-04-03 23:23:10 +000026 * child's socket is not bound to it (bind cannot be called on
Denis Vlasenko64a15122007-04-04 10:16:15 +000027 * already bound socket). Thus it still can emit outgoing packets
Denis Vlasenkobeaca812007-04-13 16:32:26 +000028 * with wrong source IP...
Denis Vlasenko02fd6682007-04-03 23:23:10 +000029 * - don't know how to retrieve ORIGDST for udp.
30 */
Denys Vlasenko47367e12016-11-23 09:05:14 +010031//config:config TCPSVD
Denys Vlasenko4eed2c62017-07-18 22:01:24 +020032//config: bool "tcpsvd (13 kb)"
Denys Vlasenko47367e12016-11-23 09:05:14 +010033//config: default y
34//config: help
Denys Vlasenko72089cf2017-07-21 09:50:55 +020035//config: tcpsvd listens on a TCP port and runs a program for each new
36//config: connection.
Denys Vlasenko47367e12016-11-23 09:05:14 +010037//config:
38//config:config UDPSVD
Denys Vlasenko4eed2c62017-07-18 22:01:24 +020039//config: bool "udpsvd (13 kb)"
Denys Vlasenko47367e12016-11-23 09:05:14 +010040//config: default y
41//config: help
Denys Vlasenko72089cf2017-07-21 09:50:55 +020042//config: udpsvd listens on an UDP port and runs a program for each new
43//config: connection.
Denys Vlasenko47367e12016-11-23 09:05:14 +010044
45//applet:IF_TCPSVD(APPLET_ODDNAME(tcpsvd, tcpudpsvd, BB_DIR_USR_BIN, BB_SUID_DROP, tcpsvd))
46//applet:IF_UDPSVD(APPLET_ODDNAME(udpsvd, tcpudpsvd, BB_DIR_USR_BIN, BB_SUID_DROP, udpsvd))
47
48//kbuild:lib-$(CONFIG_TCPSVD) += tcpudp.o tcpudp_perhost.o
49//kbuild:lib-$(CONFIG_UDPSVD) += tcpudp.o tcpudp_perhost.o
Denis Vlasenko02fd6682007-04-03 23:23:10 +000050
Pere Orga5bc8c002011-04-11 03:29:49 +020051//usage:#define tcpsvd_trivial_usage
52//usage: "[-hEv] [-c N] [-C N[:MSG]] [-b N] [-u USER] [-l NAME] IP PORT PROG"
53/* with not-implemented options: */
54/* //usage: "[-hpEvv] [-c N] [-C N[:MSG]] [-b N] [-u USER] [-l NAME] [-i DIR|-x CDB] [-t SEC] IP PORT PROG" */
55//usage:#define tcpsvd_full_usage "\n\n"
Denys Vlasenkoec1ea162016-10-07 15:56:47 +020056//usage: "Create TCP socket, bind to IP:PORT and listen for incoming connections.\n"
57//usage: "Run PROG for each connection.\n"
58//usage: "\n IP PORT IP:PORT to listen on"
Pere Orga5bc8c002011-04-11 03:29:49 +020059//usage: "\n PROG ARGS Program to run"
Pere Orga5bc8c002011-04-11 03:29:49 +020060//usage: "\n -u USER[:GRP] Change to user/group after bind"
Denys Vlasenkoec1ea162016-10-07 15:56:47 +020061//usage: "\n -c N Up to N connections simultaneously (default 30)"
62//usage: "\n -b N Allow backlog of approximately N TCP SYNs (default 20)"
63//usage: "\n -C N[:MSG] Allow only up to N connections from the same IP:"
64//usage: "\n new connections from this IP address are closed"
65//usage: "\n immediately, MSG is written to the peer before close"
66//usage: "\n -E Don't set up environment"
Pere Orga5bc8c002011-04-11 03:29:49 +020067//usage: "\n -h Look up peer's hostname"
Denys Vlasenkoec1ea162016-10-07 15:56:47 +020068//usage: "\n -l NAME Local hostname (else look up local hostname in DNS)"
Pere Orga5bc8c002011-04-11 03:29:49 +020069//usage: "\n -v Verbose"
Denys Vlasenkoec1ea162016-10-07 15:56:47 +020070//usage: "\n"
71//usage: "\nEnvironment if no -E:"
72//usage: "\nPROTO='TCP'"
73//usage: "\nTCPREMOTEADDR='ip:port'" IF_FEATURE_IPV6(" ('[ip]:port' for IPv6)")
74//usage: "\nTCPLOCALADDR='ip:port'"
75//usage: "\nTCPORIGDSTADDR='ip:port' of destination before firewall"
76//usage: "\n Useful for REDIRECTed-to-local connections:"
77//usage: "\n iptables -t nat -A PREROUTING -p tcp --dport 80 -j REDIRECT --to 8080"
78//usage: "\nTCPCONCURRENCY=num_of_connects_from_this_ip"
79//usage: "\nIf -h:"
80//usage: "\nTCPLOCALHOST='hostname' (-l NAME is used if specified)"
81//usage: "\nTCPREMOTEHOST='hostname'"
82
Pere Orga5bc8c002011-04-11 03:29:49 +020083//usage:
84//usage:#define udpsvd_trivial_usage
85//usage: "[-hEv] [-c N] [-u USER] [-l NAME] IP PORT PROG"
86//usage:#define udpsvd_full_usage "\n\n"
Denys Vlasenkoec1ea162016-10-07 15:56:47 +020087//usage: "Create UDP socket, bind to IP:PORT and wait for incoming packets.\n"
88//usage: "Run PROG for each packet, redirecting all further packets with same\n"
89//usage: "peer ip:port to it.\n"
90//usage: "\n IP PORT IP:PORT to listen on"
Pere Orga5bc8c002011-04-11 03:29:49 +020091//usage: "\n PROG ARGS Program to run"
Pere Orga5bc8c002011-04-11 03:29:49 +020092//usage: "\n -u USER[:GRP] Change to user/group after bind"
Denys Vlasenkoec1ea162016-10-07 15:56:47 +020093//usage: "\n -c N Up to N connections simultaneously (default 30)"
94//usage: "\n -E Don't set up environment"
Pere Orga5bc8c002011-04-11 03:29:49 +020095//usage: "\n -h Look up peer's hostname"
Denys Vlasenkoec1ea162016-10-07 15:56:47 +020096//usage: "\n -l NAME Local hostname (else look up local hostname in DNS)"
Pere Orga5bc8c002011-04-11 03:29:49 +020097//usage: "\n -v Verbose"
Denys Vlasenkoec1ea162016-10-07 15:56:47 +020098//usage: "\n"
99//usage: "\nEnvironment if no -E:"
100//usage: "\nPROTO='UDP'"
101//usage: "\nUDPREMOTEADDR='ip:port'" IF_FEATURE_IPV6(" ('[ip]:port' for IPv6)")
102//usage: "\nUDPLOCALADDR='ip:port'"
103//usage: "\nIf -h:"
104//usage: "\nUDPLOCALHOST='hostname' (-l NAME is used if specified)"
105//usage: "\nUDPREMOTEHOST='hostname'"
Pere Orga5bc8c002011-04-11 03:29:49 +0200106
Denis Vlasenkob6adbf12007-05-26 19:00:18 +0000107#include "libbb.h"
Denys Vlasenkoe6a2f4c2016-04-21 16:26:30 +0200108#include "common_bufsiz.h"
Jeremie Koenig2ea12d82010-05-27 15:46:25 +0200109
Jeremie Koenig2ea12d82010-05-27 15:46:25 +0200110#ifdef __linux__
Denys Vlasenko78900352017-01-02 10:46:08 +0100111/* from linux/netfilter_ipv4.h: */
112# undef SO_ORIGINAL_DST
113# define SO_ORIGINAL_DST 80
Jeremie Koenig2ea12d82010-05-27 15:46:25 +0200114#endif
Denis Vlasenko165f5b32008-03-31 20:30:38 +0000115
Denis Vlasenko4866e902008-03-17 09:17:27 +0000116// TODO: move into this file:
117#include "tcpudp_perhost.h"
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000118
119#ifdef SSLSVD
120#include "matrixSsl.h"
121#include "ssl_io.h"
122#endif
123
Denis Vlasenkob9256052007-09-28 10:29:17 +0000124struct globals {
125 unsigned verbose;
126 unsigned max_per_host;
127 unsigned cur_per_host;
128 unsigned cnum;
Denis Vlasenko3f5fdc72007-10-14 04:55:59 +0000129 unsigned cmax;
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000130 char **env_cur;
131 char *env_var[1]; /* actually bigger */
Denys Vlasenko98a4c7c2010-02-04 15:00:15 +0100132} FIX_ALIASING;
Denys Vlasenkoe6a2f4c2016-04-21 16:26:30 +0200133#define G (*(struct globals*)bb_common_bufsiz1)
Denis Vlasenkob9256052007-09-28 10:29:17 +0000134#define verbose (G.verbose )
135#define max_per_host (G.max_per_host)
136#define cur_per_host (G.cur_per_host)
137#define cnum (G.cnum )
138#define cmax (G.cmax )
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000139#define env_cur (G.env_cur )
140#define env_var (G.env_var )
Denis Vlasenko7049ff82008-06-25 09:53:17 +0000141#define INIT_G() do { \
Denys Vlasenko47cfbf32016-04-21 18:18:48 +0200142 setup_common_bufsiz(); \
Denis Vlasenko7049ff82008-06-25 09:53:17 +0000143 cmax = 30; \
144 env_cur = &env_var[0]; \
145} while (0)
Denis Vlasenkob9256052007-09-28 10:29:17 +0000146
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000147
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000148/* We have to be careful about leaking memory in repeated setenv's */
149static void xsetenv_plain(const char *n, const char *v)
150{
151 char *var = xasprintf("%s=%s", n, v);
152 *env_cur++ = var;
153 putenv(var);
154}
155
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000156static void xsetenv_proto(const char *proto, const char *n, const char *v)
157{
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000158 char *var = xasprintf("%s%s=%s", proto, n, v);
159 *env_cur++ = var;
160 putenv(var);
161}
162
163static void undo_xsetenv(void)
164{
165 char **pp = env_cur = &env_var[0];
166 while (*pp) {
167 char *var = *pp;
Denys Vlasenkodd8adde2010-06-24 05:00:50 +0200168 bb_unsetenv_and_free(var);
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000169 *pp++ = NULL;
170 }
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000171}
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000172
173static void sig_term_handler(int sig)
174{
175 if (verbose)
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000176 bb_error_msg("got signal %u, exit", sig);
Denis Vlasenko400d8bb2008-02-24 13:36:01 +0000177 kill_myself_with_sig(sig);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000178}
179
180/* Little bloated, but tries to give accurate info how child exited.
181 * Makes easier to spot segfaulting children etc... */
182static void print_waitstat(unsigned pid, int wstat)
183{
184 unsigned e = 0;
185 const char *cause = "?exit";
186
187 if (WIFEXITED(wstat)) {
188 cause++;
189 e = WEXITSTATUS(wstat);
190 } else if (WIFSIGNALED(wstat)) {
191 cause = "signal";
192 e = WTERMSIG(wstat);
193 }
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000194 bb_error_msg("end %d %s %d", pid, cause, e);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000195}
196
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000197/* Must match getopt32 in main! */
198enum {
199 OPT_c = (1 << 0),
200 OPT_C = (1 << 1),
201 OPT_i = (1 << 2),
202 OPT_x = (1 << 3),
203 OPT_u = (1 << 4),
204 OPT_l = (1 << 5),
205 OPT_E = (1 << 6),
206 OPT_b = (1 << 7),
207 OPT_h = (1 << 8),
208 OPT_p = (1 << 9),
209 OPT_t = (1 << 10),
210 OPT_v = (1 << 11),
211 OPT_V = (1 << 12),
212 OPT_U = (1 << 13), /* from here: sslsvd only */
213 OPT_slash = (1 << 14),
214 OPT_Z = (1 << 15),
215 OPT_K = (1 << 16),
216};
217
218static void connection_status(void)
219{
Denis Vlasenkof87f4952007-08-24 10:27:41 +0000220 /* "only 1 client max" desn't need this */
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000221 if (cmax > 1)
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000222 bb_error_msg("status %u/%u", cnum, cmax);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000223}
224
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000225static void sig_child_handler(int sig UNUSED_PARAM)
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000226{
227 int wstat;
Denis Vlasenko3854c5d2008-11-06 22:39:57 +0000228 pid_t pid;
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000229
Denis Vlasenkofb0eba72008-01-02 19:55:04 +0000230 while ((pid = wait_any_nohang(&wstat)) > 0) {
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000231 if (max_per_host)
232 ipsvd_perhost_remove(pid);
233 if (cnum)
234 cnum--;
235 if (verbose)
236 print_waitstat(pid, wstat);
237 }
238 if (verbose)
239 connection_status();
240}
241
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000242int tcpudpsvd_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000243int tcpudpsvd_main(int argc UNUSED_PARAM, char **argv)
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000244{
Denis Vlasenko1d426652008-03-17 09:09:09 +0000245 char *str_C, *str_t;
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000246 char *user;
247 struct hcc *hccp;
248 const char *instructs;
249 char *msg_per_host = NULL;
250 unsigned len_per_host = len_per_host; /* gcc */
Denis Vlasenko64a15122007-04-04 10:16:15 +0000251#ifndef SSLSVD
252 struct bb_uidgid_t ugid;
253#endif
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000254 bool tcp;
Denis Vlasenko64a15122007-04-04 10:16:15 +0000255 uint16_t local_port;
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000256 char *preset_local_hostname = NULL;
257 char *remote_hostname = remote_hostname; /* for compiler */
258 char *remote_addr = remote_addr; /* for compiler */
Denis Vlasenko64a15122007-04-04 10:16:15 +0000259 len_and_sockaddr *lsa;
260 len_and_sockaddr local, remote;
261 socklen_t sa_len;
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000262 int pid;
263 int sock;
264 int conn;
265 unsigned backlog = 20;
Denys Vlasenkod6513cf2009-07-19 23:07:13 +0200266 unsigned opts;
Denis Vlasenko64a15122007-04-04 10:16:15 +0000267
Denis Vlasenkob9256052007-09-28 10:29:17 +0000268 INIT_G();
269
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000270 tcp = (applet_name[0] == 't');
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000271
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000272#ifdef SSLSVD
Denys Vlasenko22542ec2017-08-08 21:55:02 +0200273 opts = getopt32(argv, "^+"
274 "c:+C:i:x:u:l:Eb:+hpt:vU:/:Z:K:" /* -c NUM, -b NUM */
275 /* 3+ args, -i at most once, -p implies -h, -v is a counter */
276 "\0" "-3:i--i:ph:vv",
Denis Vlasenko1d426652008-03-17 09:09:09 +0000277 &cmax, &str_C, &instructs, &instructs, &user, &preset_local_hostname,
278 &backlog, &str_t, &ssluser, &root, &cert, &key, &verbose
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000279 );
280#else
Denis Vlasenko1d426652008-03-17 09:09:09 +0000281 /* "+": stop on first non-option */
Denys Vlasenkoec1ea162016-10-07 15:56:47 +0200282 opts = getopt32(argv, "+c:+C:i:x:u:l:Eb:hpt:v",
Denis Vlasenko1d426652008-03-17 09:09:09 +0000283 &cmax, &str_C, &instructs, &instructs, &user, &preset_local_hostname,
284 &backlog, &str_t, &verbose
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000285 );
286#endif
Denys Vlasenkod6513cf2009-07-19 23:07:13 +0200287 if (opts & OPT_C) { /* -C n[:message] */
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000288 max_per_host = bb_strtou(str_C, &str_C, 10);
289 if (str_C[0]) {
290 if (str_C[0] != ':')
291 bb_show_usage();
292 msg_per_host = str_C + 1;
293 len_per_host = strlen(msg_per_host);
294 }
295 }
296 if (max_per_host > cmax)
297 max_per_host = cmax;
Denys Vlasenkod6513cf2009-07-19 23:07:13 +0200298 if (opts & OPT_u) {
Bernhard Reutner-Fischerd73cbd32008-07-21 14:41:33 +0000299 xget_uidgid(&ugid, user);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000300 }
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000301#ifdef SSLSVD
Denys Vlasenkod6513cf2009-07-19 23:07:13 +0200302 if (opts & OPT_U) ssluser = optarg;
303 if (opts & OPT_slash) root = optarg;
304 if (opts & OPT_Z) cert = optarg;
305 if (opts & OPT_K) key = optarg;
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000306#endif
307 argv += optind;
308 if (!argv[0][0] || LONE_CHAR(argv[0], '0'))
309 argv[0] = (char*)"0.0.0.0";
310
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000311 /* Per-IP flood protection is not thought-out for UDP */
312 if (!tcp)
313 max_per_host = 0;
314
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000315 bb_sanitize_stdio(); /* fd# 0,1,2 must be opened */
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000316
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000317#ifdef SSLSVD
318 sslser = user;
319 client = 0;
Denys Vlasenkod6513cf2009-07-19 23:07:13 +0200320 if ((getuid() == 0) && !(opts & OPT_u)) {
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000321 xfunc_exitcode = 100;
Denys Vlasenkob2e5fc32009-11-25 14:52:47 +0100322 bb_error_msg_and_die(bb_msg_you_must_be_root);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000323 }
Denys Vlasenkod6513cf2009-07-19 23:07:13 +0200324 if (opts & OPT_u)
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000325 if (!uidgid_get(&sslugid, ssluser, 1)) {
326 if (errno) {
Bernhard Reutner-Fischerd73cbd32008-07-21 14:41:33 +0000327 bb_perror_msg_and_die("can't get user/group: %s", ssluser);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000328 }
Bernhard Reutner-Fischerd73cbd32008-07-21 14:41:33 +0000329 bb_error_msg_and_die("unknown user/group %s", ssluser);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000330 }
331 if (!cert) cert = "./cert.pem";
332 if (!key) key = cert;
333 if (matrixSslOpen() < 0)
Denys Vlasenko6331cf02009-11-13 09:08:27 +0100334 fatal("can't initialize ssl");
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000335 if (matrixSslReadKeys(&keys, cert, key, 0, ca) < 0) {
336 if (client)
Denys Vlasenko6331cf02009-11-13 09:08:27 +0100337 fatal("can't read cert, key, or ca file");
338 fatal("can't read cert or key file");
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000339 }
340 if (matrixSslNewSession(&ssl, keys, 0, SSL_FLAGS_SERVER) < 0)
Denys Vlasenko6331cf02009-11-13 09:08:27 +0100341 fatal("can't create ssl session");
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000342#endif
343
344 sig_block(SIGCHLD);
345 signal(SIGCHLD, sig_child_handler);
Denis Vlasenkocf7cf622008-03-19 19:38:46 +0000346 bb_signals(BB_FATAL_SIGS, sig_term_handler);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000347 signal(SIGPIPE, SIG_IGN);
348
349 if (max_per_host)
350 ipsvd_perhost_init(cmax);
351
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000352 local_port = bb_lookup_port(argv[1], tcp ? "tcp" : "udp", 0);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000353 lsa = xhost2sockaddr(argv[0], local_port);
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000354 argv += 2;
355
Bernhard Reutner-Fischer8c69afd2008-01-29 10:33:34 +0000356 sock = xsocket(lsa->u.sa.sa_family, tcp ? SOCK_STREAM : SOCK_DGRAM, 0);
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000357 setsockopt_reuseaddr(sock);
Denis Vlasenko64a15122007-04-04 10:16:15 +0000358 sa_len = lsa->len; /* I presume sockaddr len stays the same */
Bernhard Reutner-Fischer8c69afd2008-01-29 10:33:34 +0000359 xbind(sock, &lsa->u.sa, sa_len);
Denys Vlasenko16635cc2009-06-13 22:49:08 +0200360 if (tcp) {
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000361 xlisten(sock, backlog);
Denys Vlasenko16635cc2009-06-13 22:49:08 +0200362 close_on_exec_on(sock);
363 } else { /* udp: needed for recv_from_to to work: */
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000364 socket_want_pktinfo(sock);
Denys Vlasenko16635cc2009-06-13 22:49:08 +0200365 }
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000366 /* ndelay_off(sock); - it is the default I think? */
367
368#ifndef SSLSVD
Denys Vlasenkod6513cf2009-07-19 23:07:13 +0200369 if (opts & OPT_u) {
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000370 /* drop permissions */
371 xsetgid(ugid.gid);
372 xsetuid(ugid.uid);
373 }
374#endif
375
376 if (verbose) {
Bernhard Reutner-Fischer8c69afd2008-01-29 10:33:34 +0000377 char *addr = xmalloc_sockaddr2dotted(&lsa->u.sa);
Denys Vlasenkod6513cf2009-07-19 23:07:13 +0200378 if (opts & OPT_u)
379 bb_error_msg("listening on %s, starting, uid %u, gid %u", addr,
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000380 (unsigned)ugid.uid, (unsigned)ugid.gid);
Denys Vlasenkod6513cf2009-07-19 23:07:13 +0200381 else
382 bb_error_msg("listening on %s, starting", addr);
383 free(addr);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000384 }
385
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000386 /* Main accept() loop */
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000387
388 again:
389 hccp = NULL;
390
Denys Vlasenkoec1ea162016-10-07 15:56:47 +0200391 again1:
392 close(0);
393 /* It's important to close(0) _before_ wait loop:
394 * fd#0 can be a shared connection fd.
395 * If kept open by us, peer can't detect PROG closing it.
396 */
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000397 while (cnum >= cmax)
Denis Vlasenko3f165fa2008-03-17 08:29:08 +0000398 wait_for_any_sig(); /* expecting SIGCHLD */
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000399
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000400 again2:
401 sig_unblock(SIGCHLD);
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000402 local.len = remote.len = sa_len;
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000403 if (tcp) {
Denys Vlasenkoec1ea162016-10-07 15:56:47 +0200404 /* Accept a connection to fd #0 */
Bernhard Reutner-Fischer8c69afd2008-01-29 10:33:34 +0000405 conn = accept(sock, &remote.u.sa, &remote.len);
Denis Vlasenko64a15122007-04-04 10:16:15 +0000406 } else {
Denis Vlasenko1ca332b2007-04-04 17:49:47 +0000407 /* In case recv_from_to won't be able to recover local addr.
Denis Vlasenko64a15122007-04-04 10:16:15 +0000408 * Also sets port - recv_from_to is unable to do it. */
409 local = *lsa;
Denis Vlasenkoaa9b1822008-03-17 09:10:39 +0000410 conn = recv_from_to(sock, NULL, 0, MSG_PEEK,
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000411 &remote.u.sa, &local.u.sa, sa_len);
Denis Vlasenko64a15122007-04-04 10:16:15 +0000412 }
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000413 sig_block(SIGCHLD);
414 if (conn < 0) {
415 if (errno != EINTR)
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000416 bb_perror_msg(tcp ? "accept" : "recv");
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000417 goto again2;
418 }
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000419 xmove_fd(tcp ? conn : sock, 0);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000420
421 if (max_per_host) {
422 /* Drop connection immediately if cur_per_host > max_per_host
423 * (minimizing load under SYN flood) */
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000424 remote_addr = xmalloc_sockaddr2dotted_noport(&remote.u.sa);
425 cur_per_host = ipsvd_perhost_add(remote_addr, max_per_host, &hccp);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000426 if (cur_per_host > max_per_host) {
427 /* ipsvd_perhost_add detected that max is exceeded
428 * (and did not store ip in connection table) */
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000429 free(remote_addr);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000430 if (msg_per_host) {
431 /* don't block or test for errors */
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000432 send(0, msg_per_host, len_per_host, MSG_DONTWAIT);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000433 }
434 goto again1;
435 }
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000436 /* NB: remote_addr is not leaked, it is stored in conn table */
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000437 }
438
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000439 if (!tcp) {
440 /* Voodoo magic: making udp sockets each receive its own
Denis Vlasenko64a15122007-04-04 10:16:15 +0000441 * packets is not trivial, and I still not sure
442 * I do it 100% right.
443 * 1) we have to do it before fork()
444 * 2) order is important - is it right now? */
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000445
Denis Vlasenko4cf1d082008-03-12 23:13:50 +0000446 /* Open new non-connected UDP socket for further clients... */
447 sock = xsocket(lsa->u.sa.sa_family, SOCK_DGRAM, 0);
448 setsockopt_reuseaddr(sock);
449 /* Make plain write/send work for old socket by supplying default
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000450 * destination address. This also restricts incoming packets
451 * to ones coming from this remote IP. */
Bernhard Reutner-Fischer8c69afd2008-01-29 10:33:34 +0000452 xconnect(0, &remote.u.sa, sa_len);
Denis Vlasenko64a15122007-04-04 10:16:15 +0000453 /* hole? at this point we have no wildcard udp socket...
Denis Vlasenko79468792007-04-04 11:02:55 +0000454 * can this cause clients to get "port unreachable" icmp?
455 * Yup, time window is very small, but it exists (is it?) */
Denis Vlasenko4cf1d082008-03-12 23:13:50 +0000456 /* ..."open new socket", continued */
Bernhard Reutner-Fischer8c69afd2008-01-29 10:33:34 +0000457 xbind(sock, &lsa->u.sa, sa_len);
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000458 socket_want_pktinfo(sock);
Denis Vlasenko79468792007-04-04 11:02:55 +0000459
460 /* Doesn't work:
461 * we cannot replace fd #0 - we will lose pending packet
462 * which is already buffered for us! And we cannot use fd #1
463 * instead - it will "intercept" all following packets, but child
Denis Vlasenkof87f4952007-08-24 10:27:41 +0000464 * does not expect data coming *from fd #1*! */
Denis Vlasenko79468792007-04-04 11:02:55 +0000465#if 0
Bernhard Reutner-Fischer8c69afd2008-01-29 10:33:34 +0000466 /* Make it so that local addr is fixed to localp->u.sa
Denis Vlasenko79468792007-04-04 11:02:55 +0000467 * and we don't accidentally accept packets to other local IPs. */
468 /* NB: we possibly bind to the _very_ same_ address & port as the one
469 * already bound in parent! This seems to work in Linux.
470 * (otherwise we can move socket to fd #0 only if bind succeeds) */
471 close(0);
Denys Vlasenkoca183112011-04-07 17:52:20 +0200472 set_nport(&localp->u.sa, htons(local_port));
Bernhard Reutner-Fischer8c69afd2008-01-29 10:33:34 +0000473 xmove_fd(xsocket(localp->u.sa.sa_family, SOCK_DGRAM, 0), 0);
Denis Vlasenko79468792007-04-04 11:02:55 +0000474 setsockopt_reuseaddr(0); /* crucial */
Bernhard Reutner-Fischer8c69afd2008-01-29 10:33:34 +0000475 xbind(0, &localp->u.sa, localp->len);
Denis Vlasenko79468792007-04-04 11:02:55 +0000476#endif
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000477 }
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000478
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000479 pid = vfork();
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000480 if (pid == -1) {
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000481 bb_perror_msg("vfork");
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000482 goto again;
483 }
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000484
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000485 if (pid != 0) {
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000486 /* Parent */
Denis Vlasenko02fd6682007-04-03 23:23:10 +0000487 cnum++;
488 if (verbose)
489 connection_status();
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000490 if (hccp)
491 hccp->pid = pid;
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000492 /* clean up changes done by vforked child */
493 undo_xsetenv();
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000494 goto again;
495 }
496
497 /* Child: prepare env, log, and exec prog */
498
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000499 { /* vfork alert! every xmalloc in this block should be freed! */
500 char *local_hostname = local_hostname; /* for compiler */
501 char *local_addr = NULL;
502 char *free_me0 = NULL;
503 char *free_me1 = NULL;
504 char *free_me2 = NULL;
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000505
Denys Vlasenkod6513cf2009-07-19 23:07:13 +0200506 if (verbose || !(opts & OPT_E)) {
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000507 if (!max_per_host) /* remote_addr is not yet known */
508 free_me0 = remote_addr = xmalloc_sockaddr2dotted(&remote.u.sa);
Denys Vlasenkod6513cf2009-07-19 23:07:13 +0200509 if (opts & OPT_h) {
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000510 free_me1 = remote_hostname = xmalloc_sockaddr2host_noport(&remote.u.sa);
511 if (!remote_hostname) {
Denys Vlasenko6331cf02009-11-13 09:08:27 +0100512 bb_error_msg("can't look up hostname for %s", remote_addr);
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000513 remote_hostname = remote_addr;
514 }
515 }
516 /* Find out local IP peer connected to.
517 * Errors ignored (I'm not paranoid enough to imagine kernel
518 * which doesn't know local IP). */
519 if (tcp)
520 getsockname(0, &local.u.sa, &local.len);
521 /* else: for UDP it is done earlier by parent */
522 local_addr = xmalloc_sockaddr2dotted(&local.u.sa);
Denys Vlasenkod6513cf2009-07-19 23:07:13 +0200523 if (opts & OPT_h) {
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000524 local_hostname = preset_local_hostname;
525 if (!local_hostname) {
526 free_me2 = local_hostname = xmalloc_sockaddr2host_noport(&local.u.sa);
527 if (!local_hostname)
Denys Vlasenko6331cf02009-11-13 09:08:27 +0100528 bb_error_msg_and_die("can't look up hostname for %s", local_addr);
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000529 }
530 /* else: local_hostname is not NULL, but is NOT malloced! */
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000531 }
532 }
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000533 if (verbose) {
534 pid = getpid();
535 if (max_per_host) {
536 bb_error_msg("concurrency %s %u/%u",
537 remote_addr,
538 cur_per_host, max_per_host);
539 }
Denys Vlasenkod6513cf2009-07-19 23:07:13 +0200540 bb_error_msg((opts & OPT_h)
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000541 ? "start %u %s-%s (%s-%s)"
542 : "start %u %s-%s",
543 pid,
544 local_addr, remote_addr,
545 local_hostname, remote_hostname);
Denis Vlasenko64a15122007-04-04 10:16:15 +0000546 }
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000547
Denys Vlasenkod6513cf2009-07-19 23:07:13 +0200548 if (!(opts & OPT_E)) {
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000549 /* setup ucspi env */
550 const char *proto = tcp ? "TCP" : "UDP";
551
Jeremie Koenig2ea12d82010-05-27 15:46:25 +0200552#ifdef SO_ORIGINAL_DST
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000553 /* Extract "original" destination addr:port
554 * from Linux firewall. Useful when you redirect
555 * an outbond connection to local handler, and it needs
556 * to know where it originally tried to connect */
557 if (tcp && getsockopt(0, SOL_IP, SO_ORIGINAL_DST, &local.u.sa, &local.len) == 0) {
558 char *addr = xmalloc_sockaddr2dotted(&local.u.sa);
559 xsetenv_plain("TCPORIGDSTADDR", addr);
560 free(addr);
561 }
Jeremie Koenig2ea12d82010-05-27 15:46:25 +0200562#endif
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000563 xsetenv_plain("PROTO", proto);
564 xsetenv_proto(proto, "LOCALADDR", local_addr);
565 xsetenv_proto(proto, "REMOTEADDR", remote_addr);
Denys Vlasenkod6513cf2009-07-19 23:07:13 +0200566 if (opts & OPT_h) {
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000567 xsetenv_proto(proto, "LOCALHOST", local_hostname);
568 xsetenv_proto(proto, "REMOTEHOST", remote_hostname);
569 }
570 //compat? xsetenv_proto(proto, "REMOTEINFO", "");
571 /* additional */
572 if (cur_per_host > 0) /* can not be true for udp */
573 xsetenv_plain("TCPCONCURRENCY", utoa(cur_per_host));
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000574 }
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000575 free(local_addr);
576 free(free_me0);
577 free(free_me1);
578 free(free_me2);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000579 }
580
Denis Vlasenkoaefed942008-03-17 08:35:44 +0000581 xdup2(0, 1);
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000582
Denis Vlasenko3fa36e22008-11-09 00:15:11 +0000583 signal(SIGPIPE, SIG_DFL); /* this one was SIG_IGNed */
584 /* Non-ignored signals revert to SIG_DFL on exec anyway */
585 /*signal(SIGCHLD, SIG_DFL);*/
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000586 sig_unblock(SIGCHLD);
587
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000588#ifdef SSLSVD
Denis Vlasenko4cf1d082008-03-12 23:13:50 +0000589 strcpy(id, utoa(pid));
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000590 ssl_io(0, argv);
Denys Vlasenko41ddd9f2010-06-25 01:46:53 +0200591 bb_perror_msg_and_die("can't execute '%s'", argv[0]);
Pascal Bellard21e8e8d2010-07-04 00:57:03 +0200592#else
593 BB_EXECVP_or_die(argv);
594#endif
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000595}
596
597/*
598tcpsvd [-hpEvv] [-c n] [-C n:msg] [-b n] [-u user] [-l name]
599 [-i dir|-x cdb] [ -t sec] host port prog
600
601tcpsvd creates a TCP/IP socket, binds it to the address host:port,
602and listens on the socket for incoming connections.
603
604On each incoming connection, tcpsvd conditionally runs a program,
605with standard input reading from the socket, and standard output
606writing to the socket, to handle this connection. tcpsvd keeps
607listening on the socket for new connections, and can handle
608multiple connections simultaneously.
609
610tcpsvd optionally checks for special instructions depending
611on the IP address or hostname of the client that initiated
612the connection, see ipsvd-instruct(5).
613
614host
615 host either is a hostname, or a dotted-decimal IP address,
616 or 0. If host is 0, tcpsvd accepts connections to any local
617 IP address.
618 * busybox accepts IPv6 addresses and host:port pairs too
619 In this case second parameter is ignored
620port
621 tcpsvd accepts connections to host:port. port may be a name
622 from /etc/services or a number.
623prog
624 prog consists of one or more arguments. For each connection,
625 tcpsvd normally runs prog, with file descriptor 0 reading from
626 the network, and file descriptor 1 writing to the network.
627 By default it also sets up TCP-related environment variables,
628 see tcp-environ(5)
629-i dir
630 read instructions for handling new connections from the instructions
631 directory dir. See ipsvd-instruct(5) for details.
632 * ignored by busyboxed version
633-x cdb
634 read instructions for handling new connections from the constant database
635 cdb. The constant database normally is created from an instructions
636 directory by running ipsvd-cdb(8).
637 * ignored by busyboxed version
638-t sec
639 timeout. This option only takes effect if the -i option is given.
640 While checking the instructions directory, check the time of last access
641 of the file that matches the clients address or hostname if any, discard
642 and remove the file if it wasn't accessed within the last sec seconds;
643 tcpsvd does not discard or remove a file if the user's write permission
644 is not set, for those files the timeout is disabled. Default is 0,
645 which means that the timeout is disabled.
646 * ignored by busyboxed version
647-l name
648 local hostname. Do not look up the local hostname in DNS, but use name
649 as hostname. This option must be set if tcpsvd listens on port 53
650 to avoid loops.
651-u user[:group]
652 drop permissions. Switch user ID to user's UID, and group ID to user's
653 primary GID after creating and binding to the socket. If user is followed
654 by a colon and a group name, the group ID is switched to the GID of group
655 instead. All supplementary groups are removed.
656-c n
657 concurrency. Handle up to n connections simultaneously. Default is 30.
658 If there are n connections active, tcpsvd defers acceptance of a new
659 connection until an active connection is closed.
660-C n[:msg]
661 per host concurrency. Allow only up to n connections from the same IP
662 address simultaneously. If there are n active connections from one IP
663 address, new incoming connections from this IP address are closed
664 immediately. If n is followed by :msg, the message msg is written
665 to the client if possible, before closing the connection. By default
666 msg is empty. See ipsvd-instruct(5) for supported escape sequences in msg.
667
668 For each accepted connection, the current per host concurrency is
669 available through the environment variable TCPCONCURRENCY. n and msg
670 can be overwritten by ipsvd(7) instructions, see ipsvd-instruct(5).
671 By default tcpsvd doesn't keep track of connections.
672-h
673 Look up the client's hostname in DNS.
674-p
675 paranoid. After looking up the client's hostname in DNS, look up the IP
676 addresses in DNS for that hostname, and forget about the hostname
677 if none of the addresses match the client's IP address. You should
678 set this option if you use hostname based instructions. The -p option
679 implies the -h option.
680 * ignored by busyboxed version
681-b n
682 backlog. Allow a backlog of approximately n TCP SYNs. On some systems n
683 is silently limited. Default is 20.
684-E
685 no special environment. Do not set up TCP-related environment variables.
686-v
Denys Vlasenko10ad6222017-04-17 16:13:32 +0200687 verbose. Print verbose messages to standard output.
Denis Vlasenkob933ac12007-04-03 12:09:46 +0000688-vv
689 more verbose. Print more verbose messages to standard output.
690 * no difference between -v and -vv in busyboxed version
691*/