123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691 |
- #include "libbb.h"
- #include "common_bufsiz.h"
- #ifdef __linux__
- # undef SO_ORIGINAL_DST
- # define SO_ORIGINAL_DST 80
- #endif
- #include "tcpudp_perhost.h"
- #ifdef SSLSVD
- #include "matrixSsl.h"
- #include "ssl_io.h"
- #endif
- struct globals {
- unsigned verbose;
- unsigned max_per_host;
- unsigned cur_per_host;
- unsigned cnum;
- unsigned cmax;
- char **env_cur;
- char *env_var[1];
- } FIX_ALIASING;
- #define G (*(struct globals*)bb_common_bufsiz1)
- #define verbose (G.verbose )
- #define max_per_host (G.max_per_host)
- #define cur_per_host (G.cur_per_host)
- #define cnum (G.cnum )
- #define cmax (G.cmax )
- #define env_cur (G.env_cur )
- #define env_var (G.env_var )
- #define INIT_G() do { \
- setup_common_bufsiz(); \
- cmax = 30; \
- env_cur = &env_var[0]; \
- } while (0)
- static void xsetenv_plain(const char *n, const char *v)
- {
- char *var = xasprintf("%s=%s", n, v);
- *env_cur++ = var;
- putenv(var);
- }
- static void xsetenv_proto(const char *proto, const char *n, const char *v)
- {
- char *var = xasprintf("%s%s=%s", proto, n, v);
- *env_cur++ = var;
- putenv(var);
- }
- static void undo_xsetenv(void)
- {
- char **pp = env_cur = &env_var[0];
- while (*pp) {
- char *var = *pp;
- bb_unsetenv_and_free(var);
- *pp++ = NULL;
- }
- }
- static void sig_term_handler(int sig)
- {
- if (verbose)
- bb_error_msg("got signal %u, exit", sig);
- kill_myself_with_sig(sig);
- }
- static void print_waitstat(unsigned pid, int wstat)
- {
- unsigned e = 0;
- const char *cause = "?exit";
- if (WIFEXITED(wstat)) {
- cause++;
- e = WEXITSTATUS(wstat);
- } else if (WIFSIGNALED(wstat)) {
- cause = "signal";
- e = WTERMSIG(wstat);
- }
- bb_error_msg("end %d %s %d", pid, cause, e);
- }
- enum {
- OPT_c = (1 << 0),
- OPT_C = (1 << 1),
- OPT_i = (1 << 2),
- OPT_x = (1 << 3),
- OPT_u = (1 << 4),
- OPT_l = (1 << 5),
- OPT_E = (1 << 6),
- OPT_b = (1 << 7),
- OPT_h = (1 << 8),
- OPT_p = (1 << 9),
- OPT_t = (1 << 10),
- OPT_v = (1 << 11),
- OPT_V = (1 << 12),
- OPT_U = (1 << 13),
- OPT_slash = (1 << 14),
- OPT_Z = (1 << 15),
- OPT_K = (1 << 16),
- };
- static void connection_status(void)
- {
-
- if (cmax > 1)
- bb_error_msg("status %u/%u", cnum, cmax);
- }
- static void sig_child_handler(int sig UNUSED_PARAM)
- {
- int wstat;
- pid_t pid;
- while ((pid = wait_any_nohang(&wstat)) > 0) {
- if (max_per_host)
- ipsvd_perhost_remove(pid);
- if (cnum)
- cnum--;
- if (verbose)
- print_waitstat(pid, wstat);
- }
- if (verbose)
- connection_status();
- }
- int tcpudpsvd_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
- int tcpudpsvd_main(int argc UNUSED_PARAM, char **argv)
- {
- char *str_C, *str_t;
- char *user;
- struct hcc *hccp;
- const char *instructs;
- char *msg_per_host = NULL;
- unsigned len_per_host = len_per_host;
- #ifndef SSLSVD
- struct bb_uidgid_t ugid;
- #endif
- bool tcp;
- uint16_t local_port;
- char *preset_local_hostname = NULL;
- char *remote_hostname = remote_hostname;
- char *remote_addr = remote_addr;
- len_and_sockaddr *lsa;
- len_and_sockaddr local, remote;
- socklen_t sa_len;
- int pid;
- int sock;
- int conn;
- unsigned backlog = 20;
- unsigned opts;
- INIT_G();
- tcp = (applet_name[0] == 't');
- #ifdef SSLSVD
- opts = getopt32(argv, "^+"
- "c:+C:i:x:u:l:Eb:+hpt:vU:/:Z:K:"
-
- "\0" "-3:i--i:ph:vv",
- &cmax, &str_C, &instructs, &instructs, &user, &preset_local_hostname,
- &backlog, &str_t, &ssluser, &root, &cert, &key, &verbose
- );
- #else
-
- opts = getopt32(argv, "+c:+C:i:x:u:l:Eb:hpt:v",
- &cmax, &str_C, &instructs, &instructs, &user, &preset_local_hostname,
- &backlog, &str_t, &verbose
- );
- #endif
- if (opts & OPT_C) {
- max_per_host = bb_strtou(str_C, &str_C, 10);
- if (str_C[0]) {
- if (str_C[0] != ':')
- bb_show_usage();
- msg_per_host = str_C + 1;
- len_per_host = strlen(msg_per_host);
- }
- }
- if (max_per_host > cmax)
- max_per_host = cmax;
- if (opts & OPT_u) {
- xget_uidgid(&ugid, user);
- }
- #ifdef SSLSVD
- if (opts & OPT_U) ssluser = optarg;
- if (opts & OPT_slash) root = optarg;
- if (opts & OPT_Z) cert = optarg;
- if (opts & OPT_K) key = optarg;
- #endif
- argv += optind;
- if (!argv[0][0] || LONE_CHAR(argv[0], '0'))
- argv[0] = (char*)"0.0.0.0";
-
- if (!tcp)
- max_per_host = 0;
- bb_sanitize_stdio();
- #ifdef SSLSVD
- sslser = user;
- client = 0;
- if ((getuid() == 0) && !(opts & OPT_u)) {
- xfunc_exitcode = 100;
- bb_error_msg_and_die(bb_msg_you_must_be_root);
- }
- if (opts & OPT_u)
- if (!uidgid_get(&sslugid, ssluser, 1)) {
- if (errno) {
- bb_perror_msg_and_die("can't get user/group: %s", ssluser);
- }
- bb_error_msg_and_die("unknown user/group %s", ssluser);
- }
- if (!cert) cert = "./cert.pem";
- if (!key) key = cert;
- if (matrixSslOpen() < 0)
- fatal("can't initialize ssl");
- if (matrixSslReadKeys(&keys, cert, key, 0, ca) < 0) {
- if (client)
- fatal("can't read cert, key, or ca file");
- fatal("can't read cert or key file");
- }
- if (matrixSslNewSession(&ssl, keys, 0, SSL_FLAGS_SERVER) < 0)
- fatal("can't create ssl session");
- #endif
- sig_block(SIGCHLD);
- signal(SIGCHLD, sig_child_handler);
- bb_signals(BB_FATAL_SIGS, sig_term_handler);
- signal(SIGPIPE, SIG_IGN);
- if (max_per_host)
- ipsvd_perhost_init(cmax);
- local_port = bb_lookup_port(argv[1], tcp ? "tcp" : "udp", 0);
- lsa = xhost2sockaddr(argv[0], local_port);
- argv += 2;
- sock = xsocket(lsa->u.sa.sa_family, tcp ? SOCK_STREAM : SOCK_DGRAM, 0);
- setsockopt_reuseaddr(sock);
- sa_len = lsa->len;
- xbind(sock, &lsa->u.sa, sa_len);
- if (tcp) {
- xlisten(sock, backlog);
- close_on_exec_on(sock);
- } else {
- socket_want_pktinfo(sock);
- }
-
- #ifndef SSLSVD
- if (opts & OPT_u) {
-
- xsetgid(ugid.gid);
- xsetuid(ugid.uid);
- }
- #endif
- if (verbose) {
- char *addr = xmalloc_sockaddr2dotted(&lsa->u.sa);
- if (opts & OPT_u)
- bb_error_msg("listening on %s, starting, uid %u, gid %u", addr,
- (unsigned)ugid.uid, (unsigned)ugid.gid);
- else
- bb_error_msg("listening on %s, starting", addr);
- free(addr);
- }
-
- again:
- hccp = NULL;
- again1:
- close(0);
-
- while (cnum >= cmax)
- wait_for_any_sig();
- again2:
- sig_unblock(SIGCHLD);
- local.len = remote.len = sa_len;
- if (tcp) {
-
- conn = accept(sock, &remote.u.sa, &remote.len);
- } else {
-
- local = *lsa;
- conn = recv_from_to(sock, NULL, 0, MSG_PEEK,
- &remote.u.sa, &local.u.sa, sa_len);
- }
- sig_block(SIGCHLD);
- if (conn < 0) {
- if (errno != EINTR)
- bb_perror_msg(tcp ? "accept" : "recv");
- goto again2;
- }
- xmove_fd(tcp ? conn : sock, 0);
- if (max_per_host) {
-
- remote_addr = xmalloc_sockaddr2dotted_noport(&remote.u.sa);
- cur_per_host = ipsvd_perhost_add(remote_addr, max_per_host, &hccp);
- if (cur_per_host > max_per_host) {
-
- free(remote_addr);
- if (msg_per_host) {
-
- send(0, msg_per_host, len_per_host, MSG_DONTWAIT);
- }
- goto again1;
- }
-
- }
- if (!tcp) {
-
-
- sock = xsocket(lsa->u.sa.sa_family, SOCK_DGRAM, 0);
- setsockopt_reuseaddr(sock);
-
- xconnect(0, &remote.u.sa, sa_len);
-
-
- xbind(sock, &lsa->u.sa, sa_len);
- socket_want_pktinfo(sock);
-
- #if 0
-
-
- close(0);
- set_nport(&localp->u.sa, htons(local_port));
- xmove_fd(xsocket(localp->u.sa.sa_family, SOCK_DGRAM, 0), 0);
- setsockopt_reuseaddr(0);
- xbind(0, &localp->u.sa, localp->len);
- #endif
- }
- pid = vfork();
- if (pid == -1) {
- bb_perror_msg("vfork");
- goto again;
- }
- if (pid != 0) {
-
- cnum++;
- if (verbose)
- connection_status();
- if (hccp)
- hccp->pid = pid;
-
- undo_xsetenv();
- goto again;
- }
-
- {
- char *local_hostname = local_hostname;
- char *local_addr = NULL;
- char *free_me0 = NULL;
- char *free_me1 = NULL;
- char *free_me2 = NULL;
- if (verbose || !(opts & OPT_E)) {
- if (!max_per_host)
- free_me0 = remote_addr = xmalloc_sockaddr2dotted(&remote.u.sa);
- if (opts & OPT_h) {
- free_me1 = remote_hostname = xmalloc_sockaddr2host_noport(&remote.u.sa);
- if (!remote_hostname) {
- bb_error_msg("can't look up hostname for %s", remote_addr);
- remote_hostname = remote_addr;
- }
- }
-
- if (tcp)
- getsockname(0, &local.u.sa, &local.len);
-
- local_addr = xmalloc_sockaddr2dotted(&local.u.sa);
- if (opts & OPT_h) {
- local_hostname = preset_local_hostname;
- if (!local_hostname) {
- free_me2 = local_hostname = xmalloc_sockaddr2host_noport(&local.u.sa);
- if (!local_hostname)
- bb_error_msg_and_die("can't look up hostname for %s", local_addr);
- }
-
- }
- }
- if (verbose) {
- pid = getpid();
- if (max_per_host) {
- bb_error_msg("concurrency %s %u/%u",
- remote_addr,
- cur_per_host, max_per_host);
- }
- bb_error_msg((opts & OPT_h)
- ? "start %u %s-%s (%s-%s)"
- : "start %u %s-%s",
- pid,
- local_addr, remote_addr,
- local_hostname, remote_hostname);
- }
- if (!(opts & OPT_E)) {
-
- const char *proto = tcp ? "TCP" : "UDP";
- #ifdef SO_ORIGINAL_DST
-
- if (tcp && getsockopt(0, SOL_IP, SO_ORIGINAL_DST, &local.u.sa, &local.len) == 0) {
- char *addr = xmalloc_sockaddr2dotted(&local.u.sa);
- xsetenv_plain("TCPORIGDSTADDR", addr);
- free(addr);
- }
- #endif
- xsetenv_plain("PROTO", proto);
- xsetenv_proto(proto, "LOCALADDR", local_addr);
- xsetenv_proto(proto, "REMOTEADDR", remote_addr);
- if (opts & OPT_h) {
- xsetenv_proto(proto, "LOCALHOST", local_hostname);
- xsetenv_proto(proto, "REMOTEHOST", remote_hostname);
- }
-
-
- if (cur_per_host > 0)
- xsetenv_plain("TCPCONCURRENCY", utoa(cur_per_host));
- }
- free(local_addr);
- free(free_me0);
- free(free_me1);
- free(free_me2);
- }
- xdup2(0, 1);
- signal(SIGPIPE, SIG_DFL);
-
-
- sig_unblock(SIGCHLD);
- #ifdef SSLSVD
- strcpy(id, utoa(pid));
- ssl_io(0, argv);
- bb_perror_msg_and_die("can't execute '%s'", argv[0]);
- #else
- BB_EXECVP_or_die(argv);
- #endif
- }
|