diff options
| author | Denis Vlasenko <vda.linux@googlemail.com> | 2008-03-17 09:17:27 +0000 |
|---|---|---|
| committer | Denis Vlasenko <vda.linux@googlemail.com> | 2008-03-17 09:17:27 +0000 |
| commit | 4866e905d7e1f11d86374fad4e46aa2bd669c2ba (patch) | |
| tree | a6a8e9521de92d658903ba9f54cd01e94a34c262 /networking | |
| parent | 4ee7cd4f6f9f85871c8814bb524d3e691a2992a9 (diff) | |
| download | busybox-w32-4866e905d7e1f11d86374fad4e46aa2bd669c2ba.tar.gz busybox-w32-4866e905d7e1f11d86374fad4e46aa2bd669c2ba.tar.bz2 busybox-w32-4866e905d7e1f11d86374fad4e46aa2bd669c2ba.zip | |
svn add/svn rm to actually move tcp/udpsvd...
Diffstat (limited to 'networking')
| -rw-r--r-- | networking/tcpudp.c | 609 | ||||
| -rw-r--r-- | networking/tcpudp_perhost.c | 65 | ||||
| -rw-r--r-- | networking/tcpudp_perhost.h | 29 |
3 files changed, 703 insertions, 0 deletions
diff --git a/networking/tcpudp.c b/networking/tcpudp.c new file mode 100644 index 000000000..f97787497 --- /dev/null +++ b/networking/tcpudp.c | |||
| @@ -0,0 +1,609 @@ | |||
| 1 | /* 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 | * | ||
| 5 | * Copyright (C) 2007 Denys Vlasenko. | ||
| 6 | * | ||
| 7 | * Licensed under GPLv2, see file LICENSE in this tarball for details. | ||
| 8 | */ | ||
| 9 | |||
| 10 | /* Based on ipsvd-0.12.1. This tcpsvd accepts all options | ||
| 11 | * 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 | * | ||
| 16 | * 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!). | ||
| 20 | * | ||
| 21 | * TCPORIGDSTADDR is Busybox-specific addition. | ||
| 22 | * | ||
| 23 | * udp server is hacked up by reusing TCP code. It has the following | ||
| 24 | * limitation inherent in Unix DGRAM sockets implementation: | ||
| 25 | * - local IP address is retrieved (using recvmsg voodoo) but | ||
| 26 | * child's socket is not bound to it (bind cannot be called on | ||
| 27 | * already bound socket). Thus it still can emit outgoing packets | ||
| 28 | * with wrong source IP... | ||
| 29 | * - don't know how to retrieve ORIGDST for udp. | ||
| 30 | */ | ||
| 31 | |||
| 32 | #include <limits.h> | ||
| 33 | #include <linux/netfilter_ipv4.h> /* wants <limits.h> */ | ||
| 34 | |||
| 35 | #include "libbb.h" | ||
| 36 | // TODO: move into this file: | ||
| 37 | #include "tcpudp_perhost.h" | ||
| 38 | |||
| 39 | #ifdef SSLSVD | ||
| 40 | #include "matrixSsl.h" | ||
| 41 | #include "ssl_io.h" | ||
| 42 | #endif | ||
| 43 | |||
| 44 | struct globals { | ||
| 45 | unsigned verbose; | ||
| 46 | unsigned max_per_host; | ||
| 47 | unsigned cur_per_host; | ||
| 48 | unsigned cnum; | ||
| 49 | unsigned cmax; | ||
| 50 | char **env_cur; | ||
| 51 | char *env_var[1]; /* actually bigger */ | ||
| 52 | }; | ||
| 53 | #define G (*(struct globals*)&bb_common_bufsiz1) | ||
| 54 | #define verbose (G.verbose ) | ||
| 55 | #define max_per_host (G.max_per_host) | ||
| 56 | #define cur_per_host (G.cur_per_host) | ||
| 57 | #define cnum (G.cnum ) | ||
| 58 | #define cmax (G.cmax ) | ||
| 59 | #define env_cur (G.env_cur ) | ||
| 60 | #define env_var (G.env_var ) | ||
| 61 | #define INIT_G() \ | ||
| 62 | do { \ | ||
| 63 | cmax = 30; \ | ||
| 64 | env_cur = &env_var[0]; \ | ||
| 65 | } while (0) | ||
| 66 | |||
| 67 | |||
| 68 | /* We have to be careful about leaking memory in repeated setenv's */ | ||
| 69 | static void xsetenv_plain(const char *n, const char *v) | ||
| 70 | { | ||
| 71 | char *var = xasprintf("%s=%s", n, v); | ||
| 72 | *env_cur++ = var; | ||
| 73 | putenv(var); | ||
| 74 | } | ||
| 75 | |||
| 76 | static void xsetenv_proto(const char *proto, const char *n, const char *v) | ||
| 77 | { | ||
| 78 | char *var = xasprintf("%s%s=%s", proto, n, v); | ||
| 79 | *env_cur++ = var; | ||
| 80 | putenv(var); | ||
| 81 | } | ||
| 82 | |||
| 83 | static void undo_xsetenv(void) | ||
| 84 | { | ||
| 85 | char **pp = env_cur = &env_var[0]; | ||
| 86 | while (*pp) { | ||
| 87 | char *var = *pp; | ||
| 88 | *strchrnul(var, '=') = '\0'; | ||
| 89 | unsetenv(var); | ||
| 90 | free(var); | ||
| 91 | *pp++ = NULL; | ||
| 92 | } | ||
| 93 | } | ||
| 94 | |||
| 95 | static void sig_term_handler(int sig) | ||
| 96 | { | ||
| 97 | if (verbose) | ||
| 98 | bb_error_msg("got signal %u, exit", sig); | ||
| 99 | kill_myself_with_sig(sig); | ||
| 100 | } | ||
| 101 | |||
| 102 | /* Little bloated, but tries to give accurate info how child exited. | ||
| 103 | * Makes easier to spot segfaulting children etc... */ | ||
| 104 | static void print_waitstat(unsigned pid, int wstat) | ||
| 105 | { | ||
| 106 | unsigned e = 0; | ||
| 107 | const char *cause = "?exit"; | ||
| 108 | |||
| 109 | if (WIFEXITED(wstat)) { | ||
| 110 | cause++; | ||
| 111 | e = WEXITSTATUS(wstat); | ||
| 112 | } else if (WIFSIGNALED(wstat)) { | ||
| 113 | cause = "signal"; | ||
| 114 | e = WTERMSIG(wstat); | ||
| 115 | } | ||
| 116 | bb_error_msg("end %d %s %d", pid, cause, e); | ||
| 117 | } | ||
| 118 | |||
| 119 | /* Must match getopt32 in main! */ | ||
| 120 | enum { | ||
| 121 | OPT_c = (1 << 0), | ||
| 122 | OPT_C = (1 << 1), | ||
| 123 | OPT_i = (1 << 2), | ||
| 124 | OPT_x = (1 << 3), | ||
| 125 | OPT_u = (1 << 4), | ||
| 126 | OPT_l = (1 << 5), | ||
| 127 | OPT_E = (1 << 6), | ||
| 128 | OPT_b = (1 << 7), | ||
| 129 | OPT_h = (1 << 8), | ||
| 130 | OPT_p = (1 << 9), | ||
| 131 | OPT_t = (1 << 10), | ||
| 132 | OPT_v = (1 << 11), | ||
| 133 | OPT_V = (1 << 12), | ||
| 134 | OPT_U = (1 << 13), /* from here: sslsvd only */ | ||
| 135 | OPT_slash = (1 << 14), | ||
| 136 | OPT_Z = (1 << 15), | ||
| 137 | OPT_K = (1 << 16), | ||
| 138 | }; | ||
| 139 | |||
| 140 | static void connection_status(void) | ||
| 141 | { | ||
| 142 | /* "only 1 client max" desn't need this */ | ||
| 143 | if (cmax > 1) | ||
| 144 | bb_error_msg("status %u/%u", cnum, cmax); | ||
| 145 | } | ||
| 146 | |||
| 147 | static void sig_child_handler(int sig ATTRIBUTE_UNUSED) | ||
| 148 | { | ||
| 149 | int wstat; | ||
| 150 | int pid; | ||
| 151 | |||
| 152 | while ((pid = wait_any_nohang(&wstat)) > 0) { | ||
| 153 | if (max_per_host) | ||
| 154 | ipsvd_perhost_remove(pid); | ||
| 155 | if (cnum) | ||
| 156 | cnum--; | ||
| 157 | if (verbose) | ||
| 158 | print_waitstat(pid, wstat); | ||
| 159 | } | ||
| 160 | if (verbose) | ||
| 161 | connection_status(); | ||
| 162 | } | ||
| 163 | |||
| 164 | int tcpudpsvd_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; | ||
| 165 | int tcpudpsvd_main(int argc ATTRIBUTE_UNUSED, char **argv) | ||
| 166 | { | ||
| 167 | char *str_C, *str_t; | ||
| 168 | char *user; | ||
| 169 | struct hcc *hccp; | ||
| 170 | const char *instructs; | ||
| 171 | char *msg_per_host = NULL; | ||
| 172 | unsigned len_per_host = len_per_host; /* gcc */ | ||
| 173 | #ifndef SSLSVD | ||
| 174 | struct bb_uidgid_t ugid; | ||
| 175 | #endif | ||
| 176 | bool tcp; | ||
| 177 | uint16_t local_port; | ||
| 178 | char *preset_local_hostname = NULL; | ||
| 179 | char *remote_hostname = remote_hostname; /* for compiler */ | ||
| 180 | char *remote_addr = remote_addr; /* for compiler */ | ||
| 181 | len_and_sockaddr *lsa; | ||
| 182 | len_and_sockaddr local, remote; | ||
| 183 | socklen_t sa_len; | ||
| 184 | int pid; | ||
| 185 | int sock; | ||
| 186 | int conn; | ||
| 187 | unsigned backlog = 20; | ||
| 188 | |||
| 189 | INIT_G(); | ||
| 190 | |||
| 191 | tcp = (applet_name[0] == 't'); | ||
| 192 | |||
| 193 | /* 3+ args, -i at most once, -p implies -h, -v is counter, -b N, -c N */ | ||
| 194 | opt_complementary = "-3:i--i:ph:vv:b+:c+"; | ||
| 195 | #ifdef SSLSVD | ||
| 196 | getopt32(argv, "+c:C:i:x:u:l:Eb:hpt:vU:/:Z:K:", | ||
| 197 | &cmax, &str_C, &instructs, &instructs, &user, &preset_local_hostname, | ||
| 198 | &backlog, &str_t, &ssluser, &root, &cert, &key, &verbose | ||
| 199 | ); | ||
| 200 | #else | ||
| 201 | /* "+": stop on first non-option */ | ||
| 202 | getopt32(argv, "+c:C:i:x:u:l:Eb:hpt:v", | ||
| 203 | &cmax, &str_C, &instructs, &instructs, &user, &preset_local_hostname, | ||
| 204 | &backlog, &str_t, &verbose | ||
| 205 | ); | ||
| 206 | #endif | ||
| 207 | if (option_mask32 & OPT_C) { /* -C n[:message] */ | ||
| 208 | max_per_host = bb_strtou(str_C, &str_C, 10); | ||
| 209 | if (str_C[0]) { | ||
| 210 | if (str_C[0] != ':') | ||
| 211 | bb_show_usage(); | ||
| 212 | msg_per_host = str_C + 1; | ||
| 213 | len_per_host = strlen(msg_per_host); | ||
| 214 | } | ||
| 215 | } | ||
| 216 | if (max_per_host > cmax) | ||
| 217 | max_per_host = cmax; | ||
| 218 | if (option_mask32 & OPT_u) { | ||
| 219 | if (!get_uidgid(&ugid, user, 1)) | ||
| 220 | bb_error_msg_and_die("unknown user/group: %s", user); | ||
| 221 | } | ||
| 222 | #ifdef SSLSVD | ||
| 223 | if (option_mask32 & OPT_U) ssluser = optarg; | ||
| 224 | if (option_mask32 & OPT_slash) root = optarg; | ||
| 225 | if (option_mask32 & OPT_Z) cert = optarg; | ||
| 226 | if (option_mask32 & OPT_K) key = optarg; | ||
| 227 | #endif | ||
| 228 | argv += optind; | ||
| 229 | if (!argv[0][0] || LONE_CHAR(argv[0], '0')) | ||
| 230 | argv[0] = (char*)"0.0.0.0"; | ||
| 231 | |||
| 232 | /* Per-IP flood protection is not thought-out for UDP */ | ||
| 233 | if (!tcp) | ||
| 234 | max_per_host = 0; | ||
| 235 | |||
| 236 | bb_sanitize_stdio(); /* fd# 0,1,2 must be opened */ | ||
| 237 | |||
| 238 | #ifdef SSLSVD | ||
| 239 | sslser = user; | ||
| 240 | client = 0; | ||
| 241 | if ((getuid() == 0) && !(option_mask32 & OPT_u)) { | ||
| 242 | xfunc_exitcode = 100; | ||
| 243 | bb_error_msg_and_die("-U ssluser must be set when running as root"); | ||
| 244 | } | ||
| 245 | if (option_mask32 & OPT_u) | ||
| 246 | if (!uidgid_get(&sslugid, ssluser, 1)) { | ||
| 247 | if (errno) { | ||
| 248 | bb_perror_msg_and_die("fatal: cannot get user/group: %s", ssluser); | ||
| 249 | } | ||
| 250 | bb_error_msg_and_die("unknown user/group '%s'", ssluser); | ||
| 251 | } | ||
| 252 | if (!cert) cert = "./cert.pem"; | ||
| 253 | if (!key) key = cert; | ||
| 254 | if (matrixSslOpen() < 0) | ||
| 255 | fatal("cannot initialize ssl"); | ||
| 256 | if (matrixSslReadKeys(&keys, cert, key, 0, ca) < 0) { | ||
| 257 | if (client) | ||
| 258 | fatal("cannot read cert, key, or ca file"); | ||
| 259 | fatal("cannot read cert or key file"); | ||
| 260 | } | ||
| 261 | if (matrixSslNewSession(&ssl, keys, 0, SSL_FLAGS_SERVER) < 0) | ||
| 262 | fatal("cannot create ssl session"); | ||
| 263 | #endif | ||
| 264 | |||
| 265 | sig_block(SIGCHLD); | ||
| 266 | signal(SIGCHLD, sig_child_handler); | ||
| 267 | bb_signals(BB_SIGS_FATAL, sig_term_handler); | ||
| 268 | signal(SIGPIPE, SIG_IGN); | ||
| 269 | |||
| 270 | if (max_per_host) | ||
| 271 | ipsvd_perhost_init(cmax); | ||
| 272 | |||
| 273 | local_port = bb_lookup_port(argv[1], tcp ? "tcp" : "udp", 0); | ||
| 274 | lsa = xhost2sockaddr(argv[0], local_port); | ||
| 275 | argv += 2; | ||
| 276 | |||
| 277 | sock = xsocket(lsa->u.sa.sa_family, tcp ? SOCK_STREAM : SOCK_DGRAM, 0); | ||
| 278 | setsockopt_reuseaddr(sock); | ||
| 279 | sa_len = lsa->len; /* I presume sockaddr len stays the same */ | ||
| 280 | xbind(sock, &lsa->u.sa, sa_len); | ||
| 281 | if (tcp) | ||
| 282 | xlisten(sock, backlog); | ||
| 283 | else /* udp: needed for recv_from_to to work: */ | ||
| 284 | socket_want_pktinfo(sock); | ||
| 285 | /* ndelay_off(sock); - it is the default I think? */ | ||
| 286 | |||
| 287 | #ifndef SSLSVD | ||
| 288 | if (option_mask32 & OPT_u) { | ||
| 289 | /* drop permissions */ | ||
| 290 | xsetgid(ugid.gid); | ||
| 291 | xsetuid(ugid.uid); | ||
| 292 | } | ||
| 293 | #endif | ||
| 294 | |||
| 295 | if (verbose) { | ||
| 296 | char *addr = xmalloc_sockaddr2dotted(&lsa->u.sa); | ||
| 297 | bb_error_msg("listening on %s, starting", addr); | ||
| 298 | free(addr); | ||
| 299 | #ifndef SSLSVD | ||
| 300 | if (option_mask32 & OPT_u) | ||
| 301 | printf(", uid %u, gid %u", | ||
| 302 | (unsigned)ugid.uid, (unsigned)ugid.gid); | ||
| 303 | #endif | ||
| 304 | } | ||
| 305 | |||
| 306 | /* Main accept() loop */ | ||
| 307 | |||
| 308 | again: | ||
| 309 | hccp = NULL; | ||
| 310 | |||
| 311 | while (cnum >= cmax) | ||
| 312 | wait_for_any_sig(); /* expecting SIGCHLD */ | ||
| 313 | |||
| 314 | /* Accept a connection to fd #0 */ | ||
| 315 | again1: | ||
| 316 | close(0); | ||
| 317 | again2: | ||
| 318 | sig_unblock(SIGCHLD); | ||
| 319 | local.len = remote.len = sa_len; | ||
| 320 | if (tcp) { | ||
| 321 | conn = accept(sock, &remote.u.sa, &remote.len); | ||
| 322 | } else { | ||
| 323 | /* In case recv_from_to won't be able to recover local addr. | ||
| 324 | * Also sets port - recv_from_to is unable to do it. */ | ||
| 325 | local = *lsa; | ||
| 326 | conn = recv_from_to(sock, NULL, 0, MSG_PEEK, | ||
| 327 | &remote.u.sa, &local.u.sa, sa_len); | ||
| 328 | } | ||
| 329 | sig_block(SIGCHLD); | ||
| 330 | if (conn < 0) { | ||
| 331 | if (errno != EINTR) | ||
| 332 | bb_perror_msg(tcp ? "accept" : "recv"); | ||
| 333 | goto again2; | ||
| 334 | } | ||
| 335 | xmove_fd(tcp ? conn : sock, 0); | ||
| 336 | |||
| 337 | if (max_per_host) { | ||
| 338 | /* Drop connection immediately if cur_per_host > max_per_host | ||
| 339 | * (minimizing load under SYN flood) */ | ||
| 340 | remote_addr = xmalloc_sockaddr2dotted_noport(&remote.u.sa); | ||
| 341 | cur_per_host = ipsvd_perhost_add(remote_addr, max_per_host, &hccp); | ||
| 342 | if (cur_per_host > max_per_host) { | ||
| 343 | /* ipsvd_perhost_add detected that max is exceeded | ||
| 344 | * (and did not store ip in connection table) */ | ||
| 345 | free(remote_addr); | ||
| 346 | if (msg_per_host) { | ||
| 347 | /* don't block or test for errors */ | ||
| 348 | send(0, msg_per_host, len_per_host, MSG_DONTWAIT); | ||
| 349 | } | ||
| 350 | goto again1; | ||
| 351 | } | ||
| 352 | /* NB: remote_addr is not leaked, it is stored in conn table */ | ||
| 353 | } | ||
| 354 | |||
| 355 | if (!tcp) { | ||
| 356 | /* Voodoo magic: making udp sockets each receive its own | ||
| 357 | * packets is not trivial, and I still not sure | ||
| 358 | * I do it 100% right. | ||
| 359 | * 1) we have to do it before fork() | ||
| 360 | * 2) order is important - is it right now? */ | ||
| 361 | |||
| 362 | /* Open new non-connected UDP socket for further clients... */ | ||
| 363 | sock = xsocket(lsa->u.sa.sa_family, SOCK_DGRAM, 0); | ||
| 364 | setsockopt_reuseaddr(sock); | ||
| 365 | /* Make plain write/send work for old socket by supplying default | ||
| 366 | * destination address. This also restricts incoming packets | ||
| 367 | * to ones coming from this remote IP. */ | ||
| 368 | xconnect(0, &remote.u.sa, sa_len); | ||
| 369 | /* hole? at this point we have no wildcard udp socket... | ||
| 370 | * can this cause clients to get "port unreachable" icmp? | ||
| 371 | * Yup, time window is very small, but it exists (is it?) */ | ||
| 372 | /* ..."open new socket", continued */ | ||
| 373 | xbind(sock, &lsa->u.sa, sa_len); | ||
| 374 | socket_want_pktinfo(sock); | ||
| 375 | |||
| 376 | /* Doesn't work: | ||
| 377 | * we cannot replace fd #0 - we will lose pending packet | ||
| 378 | * which is already buffered for us! And we cannot use fd #1 | ||
| 379 | * instead - it will "intercept" all following packets, but child | ||
| 380 | * does not expect data coming *from fd #1*! */ | ||
| 381 | #if 0 | ||
| 382 | /* Make it so that local addr is fixed to localp->u.sa | ||
| 383 | * and we don't accidentally accept packets to other local IPs. */ | ||
| 384 | /* NB: we possibly bind to the _very_ same_ address & port as the one | ||
| 385 | * already bound in parent! This seems to work in Linux. | ||
| 386 | * (otherwise we can move socket to fd #0 only if bind succeeds) */ | ||
| 387 | close(0); | ||
| 388 | set_nport(localp, htons(local_port)); | ||
| 389 | xmove_fd(xsocket(localp->u.sa.sa_family, SOCK_DGRAM, 0), 0); | ||
| 390 | setsockopt_reuseaddr(0); /* crucial */ | ||
| 391 | xbind(0, &localp->u.sa, localp->len); | ||
| 392 | #endif | ||
| 393 | } | ||
| 394 | |||
| 395 | pid = vfork(); | ||
| 396 | if (pid == -1) { | ||
| 397 | bb_perror_msg("vfork"); | ||
| 398 | goto again; | ||
| 399 | } | ||
| 400 | |||
| 401 | if (pid != 0) { | ||
| 402 | /* Parent */ | ||
| 403 | cnum++; | ||
| 404 | if (verbose) | ||
| 405 | connection_status(); | ||
| 406 | if (hccp) | ||
| 407 | hccp->pid = pid; | ||
| 408 | /* clean up changes done by vforked child */ | ||
| 409 | undo_xsetenv(); | ||
| 410 | goto again; | ||
| 411 | } | ||
| 412 | |||
| 413 | /* Child: prepare env, log, and exec prog */ | ||
| 414 | |||
| 415 | /* Closing tcp listening socket */ | ||
| 416 | if (tcp) | ||
| 417 | close(sock); | ||
| 418 | |||
| 419 | { /* vfork alert! every xmalloc in this block should be freed! */ | ||
| 420 | char *local_hostname = local_hostname; /* for compiler */ | ||
| 421 | char *local_addr = NULL; | ||
| 422 | char *free_me0 = NULL; | ||
| 423 | char *free_me1 = NULL; | ||
| 424 | char *free_me2 = NULL; | ||
| 425 | |||
| 426 | if (verbose || !(option_mask32 & OPT_E)) { | ||
| 427 | if (!max_per_host) /* remote_addr is not yet known */ | ||
| 428 | free_me0 = remote_addr = xmalloc_sockaddr2dotted(&remote.u.sa); | ||
| 429 | if (option_mask32 & OPT_h) { | ||
| 430 | free_me1 = remote_hostname = xmalloc_sockaddr2host_noport(&remote.u.sa); | ||
| 431 | if (!remote_hostname) { | ||
| 432 | bb_error_msg("cannot look up hostname for %s", remote_addr); | ||
| 433 | remote_hostname = remote_addr; | ||
| 434 | } | ||
| 435 | } | ||
| 436 | /* Find out local IP peer connected to. | ||
| 437 | * Errors ignored (I'm not paranoid enough to imagine kernel | ||
| 438 | * which doesn't know local IP). */ | ||
| 439 | if (tcp) | ||
| 440 | getsockname(0, &local.u.sa, &local.len); | ||
| 441 | /* else: for UDP it is done earlier by parent */ | ||
| 442 | local_addr = xmalloc_sockaddr2dotted(&local.u.sa); | ||
| 443 | if (option_mask32 & OPT_h) { | ||
| 444 | local_hostname = preset_local_hostname; | ||
| 445 | if (!local_hostname) { | ||
| 446 | free_me2 = local_hostname = xmalloc_sockaddr2host_noport(&local.u.sa); | ||
| 447 | if (!local_hostname) | ||
| 448 | bb_error_msg_and_die("cannot look up hostname for %s", local_addr); | ||
| 449 | } | ||
| 450 | /* else: local_hostname is not NULL, but is NOT malloced! */ | ||
| 451 | } | ||
| 452 | } | ||
| 453 | if (verbose) { | ||
| 454 | pid = getpid(); | ||
| 455 | if (max_per_host) { | ||
| 456 | bb_error_msg("concurrency %s %u/%u", | ||
| 457 | remote_addr, | ||
| 458 | cur_per_host, max_per_host); | ||
| 459 | } | ||
| 460 | bb_error_msg((option_mask32 & OPT_h) | ||
| 461 | ? "start %u %s-%s (%s-%s)" | ||
| 462 | : "start %u %s-%s", | ||
| 463 | pid, | ||
| 464 | local_addr, remote_addr, | ||
| 465 | local_hostname, remote_hostname); | ||
| 466 | } | ||
| 467 | |||
| 468 | if (!(option_mask32 & OPT_E)) { | ||
| 469 | /* setup ucspi env */ | ||
| 470 | const char *proto = tcp ? "TCP" : "UDP"; | ||
| 471 | |||
| 472 | /* Extract "original" destination addr:port | ||
| 473 | * from Linux firewall. Useful when you redirect | ||
| 474 | * an outbond connection to local handler, and it needs | ||
| 475 | * to know where it originally tried to connect */ | ||
| 476 | if (tcp && getsockopt(0, SOL_IP, SO_ORIGINAL_DST, &local.u.sa, &local.len) == 0) { | ||
| 477 | char *addr = xmalloc_sockaddr2dotted(&local.u.sa); | ||
| 478 | xsetenv_plain("TCPORIGDSTADDR", addr); | ||
| 479 | free(addr); | ||
| 480 | } | ||
| 481 | xsetenv_plain("PROTO", proto); | ||
| 482 | xsetenv_proto(proto, "LOCALADDR", local_addr); | ||
| 483 | xsetenv_proto(proto, "REMOTEADDR", remote_addr); | ||
| 484 | if (option_mask32 & OPT_h) { | ||
| 485 | xsetenv_proto(proto, "LOCALHOST", local_hostname); | ||
| 486 | xsetenv_proto(proto, "REMOTEHOST", remote_hostname); | ||
| 487 | } | ||
| 488 | //compat? xsetenv_proto(proto, "REMOTEINFO", ""); | ||
| 489 | /* additional */ | ||
| 490 | if (cur_per_host > 0) /* can not be true for udp */ | ||
| 491 | xsetenv_plain("TCPCONCURRENCY", utoa(cur_per_host)); | ||
| 492 | } | ||
| 493 | free(local_addr); | ||
| 494 | free(free_me0); | ||
| 495 | free(free_me1); | ||
| 496 | free(free_me2); | ||
| 497 | } | ||
| 498 | |||
| 499 | xdup2(0, 1); | ||
| 500 | |||
| 501 | signal(SIGTERM, SIG_DFL); | ||
| 502 | signal(SIGPIPE, SIG_DFL); | ||
| 503 | signal(SIGCHLD, SIG_DFL); | ||
| 504 | sig_unblock(SIGCHLD); | ||
| 505 | |||
| 506 | #ifdef SSLSVD | ||
| 507 | strcpy(id, utoa(pid)); | ||
| 508 | ssl_io(0, argv); | ||
| 509 | #else | ||
| 510 | BB_EXECVP(argv[0], argv); | ||
| 511 | #endif | ||
| 512 | bb_perror_msg_and_die("exec '%s'", argv[0]); | ||
| 513 | } | ||
| 514 | |||
| 515 | /* | ||
| 516 | tcpsvd [-hpEvv] [-c n] [-C n:msg] [-b n] [-u user] [-l name] | ||
| 517 | [-i dir|-x cdb] [ -t sec] host port prog | ||
| 518 | |||
| 519 | tcpsvd creates a TCP/IP socket, binds it to the address host:port, | ||
| 520 | and listens on the socket for incoming connections. | ||
| 521 | |||
| 522 | On each incoming connection, tcpsvd conditionally runs a program, | ||
| 523 | with standard input reading from the socket, and standard output | ||
| 524 | writing to the socket, to handle this connection. tcpsvd keeps | ||
| 525 | listening on the socket for new connections, and can handle | ||
| 526 | multiple connections simultaneously. | ||
| 527 | |||
| 528 | tcpsvd optionally checks for special instructions depending | ||
| 529 | on the IP address or hostname of the client that initiated | ||
| 530 | the connection, see ipsvd-instruct(5). | ||
| 531 | |||
| 532 | host | ||
| 533 | host either is a hostname, or a dotted-decimal IP address, | ||
| 534 | or 0. If host is 0, tcpsvd accepts connections to any local | ||
| 535 | IP address. | ||
| 536 | * busybox accepts IPv6 addresses and host:port pairs too | ||
| 537 | In this case second parameter is ignored | ||
| 538 | port | ||
| 539 | tcpsvd accepts connections to host:port. port may be a name | ||
| 540 | from /etc/services or a number. | ||
| 541 | prog | ||
| 542 | prog consists of one or more arguments. For each connection, | ||
| 543 | tcpsvd normally runs prog, with file descriptor 0 reading from | ||
| 544 | the network, and file descriptor 1 writing to the network. | ||
| 545 | By default it also sets up TCP-related environment variables, | ||
| 546 | see tcp-environ(5) | ||
| 547 | -i dir | ||
| 548 | read instructions for handling new connections from the instructions | ||
| 549 | directory dir. See ipsvd-instruct(5) for details. | ||
| 550 | * ignored by busyboxed version | ||
| 551 | -x cdb | ||
| 552 | read instructions for handling new connections from the constant database | ||
| 553 | cdb. The constant database normally is created from an instructions | ||
| 554 | directory by running ipsvd-cdb(8). | ||
| 555 | * ignored by busyboxed version | ||
| 556 | -t sec | ||
| 557 | timeout. This option only takes effect if the -i option is given. | ||
| 558 | While checking the instructions directory, check the time of last access | ||
| 559 | of the file that matches the clients address or hostname if any, discard | ||
| 560 | and remove the file if it wasn't accessed within the last sec seconds; | ||
| 561 | tcpsvd does not discard or remove a file if the user's write permission | ||
| 562 | is not set, for those files the timeout is disabled. Default is 0, | ||
| 563 | which means that the timeout is disabled. | ||
| 564 | * ignored by busyboxed version | ||
| 565 | -l name | ||
| 566 | local hostname. Do not look up the local hostname in DNS, but use name | ||
| 567 | as hostname. This option must be set if tcpsvd listens on port 53 | ||
| 568 | to avoid loops. | ||
| 569 | -u user[:group] | ||
| 570 | drop permissions. Switch user ID to user's UID, and group ID to user's | ||
| 571 | primary GID after creating and binding to the socket. If user is followed | ||
| 572 | by a colon and a group name, the group ID is switched to the GID of group | ||
| 573 | instead. All supplementary groups are removed. | ||
| 574 | -c n | ||
| 575 | concurrency. Handle up to n connections simultaneously. Default is 30. | ||
| 576 | If there are n connections active, tcpsvd defers acceptance of a new | ||
| 577 | connection until an active connection is closed. | ||
| 578 | -C n[:msg] | ||
| 579 | per host concurrency. Allow only up to n connections from the same IP | ||
| 580 | address simultaneously. If there are n active connections from one IP | ||
| 581 | address, new incoming connections from this IP address are closed | ||
| 582 | immediately. If n is followed by :msg, the message msg is written | ||
| 583 | to the client if possible, before closing the connection. By default | ||
| 584 | msg is empty. See ipsvd-instruct(5) for supported escape sequences in msg. | ||
| 585 | |||
| 586 | For each accepted connection, the current per host concurrency is | ||
| 587 | available through the environment variable TCPCONCURRENCY. n and msg | ||
| 588 | can be overwritten by ipsvd(7) instructions, see ipsvd-instruct(5). | ||
| 589 | By default tcpsvd doesn't keep track of connections. | ||
| 590 | -h | ||
| 591 | Look up the client's hostname in DNS. | ||
| 592 | -p | ||
| 593 | paranoid. After looking up the client's hostname in DNS, look up the IP | ||
| 594 | addresses in DNS for that hostname, and forget about the hostname | ||
| 595 | if none of the addresses match the client's IP address. You should | ||
| 596 | set this option if you use hostname based instructions. The -p option | ||
| 597 | implies the -h option. | ||
| 598 | * ignored by busyboxed version | ||
| 599 | -b n | ||
| 600 | backlog. Allow a backlog of approximately n TCP SYNs. On some systems n | ||
| 601 | is silently limited. Default is 20. | ||
| 602 | -E | ||
| 603 | no special environment. Do not set up TCP-related environment variables. | ||
| 604 | -v | ||
| 605 | verbose. Print verbose messsages to standard output. | ||
| 606 | -vv | ||
| 607 | more verbose. Print more verbose messages to standard output. | ||
| 608 | * no difference between -v and -vv in busyboxed version | ||
| 609 | */ | ||
diff --git a/networking/tcpudp_perhost.c b/networking/tcpudp_perhost.c new file mode 100644 index 000000000..3005f12c0 --- /dev/null +++ b/networking/tcpudp_perhost.c | |||
| @@ -0,0 +1,65 @@ | |||
| 1 | /* 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 | * | ||
| 5 | * Copyright (C) 2007 Denys Vlasenko. | ||
| 6 | * | ||
| 7 | * Licensed under GPLv2, see file LICENSE in this tarball for details. | ||
| 8 | */ | ||
| 9 | |||
| 10 | #include "libbb.h" | ||
| 11 | #include "tcpudp_perhost.h" | ||
| 12 | |||
| 13 | static struct hcc *cc; | ||
| 14 | static unsigned cclen; | ||
| 15 | |||
| 16 | /* to be optimized */ | ||
| 17 | |||
| 18 | void ipsvd_perhost_init(unsigned c) | ||
| 19 | { | ||
| 20 | // free(cc); | ||
| 21 | cc = xzalloc(c * sizeof(*cc)); | ||
| 22 | cclen = c; | ||
| 23 | } | ||
| 24 | |||
| 25 | unsigned ipsvd_perhost_add(char *ip, unsigned maxconn, struct hcc **hccpp) | ||
| 26 | { | ||
| 27 | unsigned i; | ||
| 28 | unsigned conn = 1; | ||
| 29 | int freepos = -1; | ||
| 30 | |||
| 31 | for (i = 0; i < cclen; ++i) { | ||
| 32 | if (!cc[i].ip) { | ||
| 33 | freepos = i; | ||
| 34 | continue; | ||
| 35 | } | ||
| 36 | if (strcmp(cc[i].ip, ip) == 0) { | ||
| 37 | conn++; | ||
| 38 | continue; | ||
| 39 | } | ||
| 40 | } | ||
| 41 | if (freepos == -1) return 0; | ||
| 42 | if (conn <= maxconn) { | ||
| 43 | cc[freepos].ip = ip; | ||
| 44 | *hccpp = &cc[freepos]; | ||
| 45 | } | ||
| 46 | return conn; | ||
| 47 | } | ||
| 48 | |||
| 49 | void ipsvd_perhost_remove(int pid) | ||
| 50 | { | ||
| 51 | unsigned i; | ||
| 52 | for (i = 0; i < cclen; ++i) { | ||
| 53 | if (cc[i].pid == pid) { | ||
| 54 | free(cc[i].ip); | ||
| 55 | cc[i].ip = NULL; | ||
| 56 | cc[i].pid = 0; | ||
| 57 | return; | ||
| 58 | } | ||
| 59 | } | ||
| 60 | } | ||
| 61 | |||
| 62 | //void ipsvd_perhost_free(void) | ||
| 63 | //{ | ||
| 64 | // free(cc); | ||
| 65 | //} | ||
diff --git a/networking/tcpudp_perhost.h b/networking/tcpudp_perhost.h new file mode 100644 index 000000000..9fc8cee61 --- /dev/null +++ b/networking/tcpudp_perhost.h | |||
| @@ -0,0 +1,29 @@ | |||
| 1 | /* 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 | * | ||
| 5 | * Copyright (C) 2007 Denys Vlasenko. | ||
| 6 | * | ||
| 7 | * Licensed under GPLv2, see file LICENSE in this tarball for details. | ||
| 8 | */ | ||
| 9 | |||
| 10 | struct hcc { | ||
| 11 | char *ip; | ||
| 12 | int pid; | ||
| 13 | }; | ||
| 14 | |||
| 15 | void ipsvd_perhost_init(unsigned); | ||
| 16 | |||
| 17 | /* Returns number of already opened connects to this ips, including this one. | ||
| 18 | * ip should be a malloc'ed ptr. | ||
| 19 | * If return value is <= maxconn, ip is inserted into the table | ||
| 20 | * and pointer to table entry if stored in *hccpp | ||
| 21 | * (useful for storing pid later). | ||
| 22 | * Else ip is NOT inserted (you must take care of it - free() etc) */ | ||
| 23 | unsigned ipsvd_perhost_add(char *ip, unsigned maxconn, struct hcc **hccpp); | ||
| 24 | |||
| 25 | /* Finds and frees element with pid */ | ||
| 26 | void ipsvd_perhost_remove(int pid); | ||
| 27 | |||
| 28 | //unsigned ipsvd_perhost_setpid(int pid); | ||
| 29 | //void ipsvd_perhost_free(void); | ||
