cp: -r and -R imply -d (coreutils compat)
[oweals/busybox.git] / ipsvd / tcpudp.c
index 419551d8dd70fa48367d76fe33e6bbe9c57487fd..6a11539c7a3c3a26c8515a23f53c58a8c773759b 100644 (file)
@@ -7,12 +7,49 @@
  * Licensed under GPLv2, see file LICENSE in this tarball for details.
  */
 
-/* TCP and UDP server are using a lot of same string constants
- * We reuse them by keeping both in one source file */
+/* Based on ipsvd ipsvd-0.12.1. This tcpsvd accepts all options
+ * which are supported by one from ipsvd-0.12.1, but not all are
+ * functional. See help text at the end of this file for details.
+ *
+ * Code inside "#ifdef SSLSVD" is for sslsvd and is currently unused.
+ *
+ * Output of verbose mode matches original (modulo bugs and
+ * unimplemented stuff). Unnatural splitting of IP and PORT
+ * is retained (personally I prefer one-value "IP:PORT" notation -
+ * it is a natural string representation of struct sockaddr_XX).
+ *
+ * TCPORIGDST{IP,PORT} is busybox-specific addition
+ *
+ * udp server is hacked up by reusing TCP code. It has the following
+ * limitation inherent in Unix DGRAM sockets implementation:
+ * - local IP address is retrieved (using recvmsg voodoo) but
+ *   child's socket is not bound to it (bind cannot be called on
+ *   already bound socket). Thus it still can emit outgoing packets
+ *   with wrong source IP...
+ * - don't know how to retrieve ORIGDST for udp.
+ */
+
+#include <limits.h>
+#include <linux/netfilter_ipv4.h> /* wants <limits.h> */
+
+#include "libbb.h"
+#include "ipsvd_perhost.h"
 
-#include "busybox.h"
+#ifdef SSLSVD
+#include "matrixSsl.h"
+#include "ssl_io.h"
+#endif
 
 static unsigned verbose;
+static unsigned max_per_host;
+static unsigned cur_per_host;
+static unsigned cnum;
+static unsigned cmax = 30;
+
+static void xsetenv_proto(const char *proto, const char *n, const char *v)
+{
+       putenv(xasprintf("%s%s=%s", proto, n, v));
+}
 
 static void sig_term_handler(int sig)
 {
@@ -38,287 +75,6 @@ static void print_waitstat(unsigned pid, int wstat)
        printf("%s: info: end %d %s %d\n", applet_name, pid, cause, e);
 }
 
-
-#if ENABLE_UDPSVD
-/* Based on ipsvd ipsvd-0.12.1. This udpsvd accepts all options
- * which are supported by one from ipsvd-0.12.1, but not all are
- * functional. See help text at the end of this file for details.
- *
- * Output of verbose mode matches original (modulo bugs and
- * unimplemented stuff). Unnatural splitting of IP and PORT
- * is retained (personally I prefer one-value "IP:PORT" notation -
- * it is a natural string representation of struct sockaddr_XX).
- */
-
-#include "udp_io.c"
-
-int udpsvd_main(int argc, char **argv);
-int udpsvd_main(int argc, char **argv)
-{
-       const char *instructs;
-       char *str_t, *user;
-       unsigned opt;
-
-       char *remote_hostname = (char*)""; /* used if no -h */
-       char *local_hostname = NULL;
-       char *remote_ip;
-       char *local_ip;// = local_ip; /* gcc */
-       uint16_t local_port, remote_port;
-       len_and_sockaddr remote;
-       len_and_sockaddr *localp;
-       int wstat;
-       unsigned pid;
-       struct bb_uidgid_t ugid;
-
-       enum {
-               OPT_v = (1 << 0),
-               OPT_u = (1 << 1),
-               OPT_l = (1 << 2),
-               OPT_h = (1 << 3),
-               OPT_p = (1 << 4),
-               OPT_i = (1 << 5),
-               OPT_x = (1 << 6),
-               OPT_t = (1 << 7),
-       };
-
-       opt_complementary = "-3:ph:vv";
-       opt = getopt32(argc, argv, "vu:l:hpi:x:t:",
-                       &user, &local_hostname, &instructs, &instructs, &str_t, &verbose);
-       if (opt & OPT_u) {
-               if (!get_uidgid(&ugid, user, 1))
-                       bb_error_msg_and_die("unknown user/group: %s", user);
-       }
-       argv += optind;
-       if (!argv[0][0] || LONE_CHAR(argv[0], '0'))
-               argv[0] = (char*)"0.0.0.0";
-
-       /* stdout is used for logging, don't buffer */
-       setlinebuf(stdout);
-       bb_sanitize_stdio(); /* fd# 1,2 must be opened */
-
-       signal(SIGTERM, sig_term_handler);
-       signal(SIGPIPE, SIG_IGN);
-
-       local_port = bb_lookup_port(argv[1], "udp", 0);
-       localp = xhost2sockaddr(argv[0], local_port);
-       /* fd #0 is the open UDP socket */
-       xmove_fd(xsocket(localp->sa.sa_family, SOCK_DGRAM, 0), 0);
-       setsockopt_reuseaddr(0); /* crucial */
-       xbind(0, &localp->sa, localp->len);
-       socket_want_pktinfo(0); /* needed for recv_from_to to work */
-
-       if (opt & OPT_u) { /* drop permissions */
-               xsetgid(ugid.gid);
-               xsetuid(ugid.uid);
-       }
-
-       if (verbose) {
-               /* we do it only for ":port" cosmetics... oh well */
-               char *addr = xmalloc_sockaddr2dotted(&localp->sa, localp->len);
-
-               printf("%s: info: listening on %s", applet_name, addr);
-               free(addr);
-               if (option_mask32 & OPT_u)
-                       printf(", uid %u, gid %u",
-                               (unsigned)ugid.uid, (unsigned)ugid.gid);
-               puts(", starting");
-       }
-
- again:
-       /* if (recvfrom(0, NULL, 0, MSG_PEEK, &remote.sa, &localp->len) < 0) { */
-       if (recv_from_to(0, NULL, 0, MSG_PEEK, &remote.sa, &localp->sa, localp->len) < 0) {
-               bb_perror_msg("recvfrom");
-               goto again;
-       }
-
-       while ((pid = fork()) < 0) {
-               bb_perror_msg("fork failed, sleeping");
-               sleep(5);
-       }
-       if (pid > 0) { /* parent */
-               while (wait_pid(&wstat, pid) < 0)
-                       bb_perror_msg("error waiting for child");
-               if (verbose)
-                       print_waitstat(pid, wstat);
-               goto again;
-       }
-
-       /* Child */
-
-       if (verbose) {
-               remote_ip = xmalloc_sockaddr2dotted_noport(&remote.sa, localp->len);
-               local_ip = xmalloc_sockaddr2dotted_noport(&localp->sa, localp->len);
-
-               pid = getpid();
-               printf("%s: info: pid %u from %s\n", applet_name, pid, remote_ip);
-
-               if (!local_hostname) {
-                       local_hostname = xmalloc_sockaddr2host_noport(&localp->sa, localp->len);
-                       if (!local_hostname)
-                               bb_error_msg_and_die("cannot look up local hostname for %s", local_ip);
-               }
-               if (opt & OPT_h) {
-                       remote_hostname = xmalloc_sockaddr2host(&remote.sa, localp->len);
-                       if (!remote_hostname) {
-                               bb_error_msg("warning: cannot look up hostname for %s", remote_ip);
-                               remote_hostname = (char*)"";
-                       }
-               }
-
-               remote_port = get_nport(&remote.sa);
-               remote_port = ntohs(remote_port);
-               printf("%s: info: %u %s:%s :%s:%s:%u\n",
-                               applet_name, pid, local_hostname, local_ip,
-                               remote_hostname, remote_ip, remote_port);
-       }
-
-       /* Doesn't work:
-        * we cannot replace fd #0 - we will lose pending packet
-        * which is already buffered for us! And we cannot use fd #1
-        * instead - it will "intercept" all following packets, but child
-        * do not expect data coming *from fd #1*! */
-#if 0
-       /* Make it so that local addr is fixed to localp->sa
-        * and we don't accidentally accept packets to other local IPs. */
-       /* NB: we possibly bind to the _very_ same_ address & port as the one
-        * already bound in parent! This seems to work in Linux.
-        * (otherwise we can move socket to fd #0 only if bind succeeds) */
-       close(0);
-       set_nport(localp, htons(local_port));
-       xmove_fd(xsocket(localp->sa.sa_family, SOCK_DGRAM, 0), 0);
-       setsockopt_reuseaddr(0); /* crucial */
-       xbind(0, &localp->sa, localp->len);
-#endif
-
-       /* Make plain write to fd #1 work for the child by supplying default
-        * destination address. This also restricts incoming packets
-        * to ones coming from this remote IP. */
-       xconnect(0, &remote.sa, localp->len);
-       dup2(0 ,1);
-
-       signal(SIGTERM, SIG_DFL);
-       signal(SIGPIPE, SIG_DFL);
-
-       argv += 2;
-       BB_EXECVP(argv[0], argv);
-       bb_perror_msg_and_die("exec '%s'", argv[0]);
-}
-
-
-/*
-udpsvd [-hpvv] [-u user] [-l name] [-i dir|-x cdb] [-t sec] host port prog
-
-udpsvd creates an UDP/IP socket, binds it to the address host:port,
-and listens on the socket for incoming datagrams.
-
-If a datagram is available on the socket, udpsvd conditionally starts
-a program, with standard input reading from the socket, and standard
-output redirected to standard error, to handle this, and possibly
-more datagrams. udpsvd does not start the program if another program
-that it has started before still is running. If the program exits,
-udpsvd again listens to the socket until a new datagram is available.
-If there are still datagrams available on the socket, the program
-is restarted immediately.
-
-udpsvd optionally checks for special intructions depending on
-the IP address or hostname of the client sending the datagram which
-not yet was handled by a running program, see ipsvd-instruct(5)
-for details.
-
-Attention:
-UDP is a connectionless protocol. Most programs that handle user datagrams,
-such as talkd(8), keep running after receiving a datagram, and process
-subsequent datagrams sent to the socket until a timeout is reached.
-udpsvd only checks special instructions for a datagram that causes a startup
-of the program; not if a program handling datagrams already is running.
-It doesn't make much sense to restrict access through special instructions
-when using such a program.
-
-On the other hand, it makes perfectly sense with programs like tftpd(8),
-that fork to establish a separate connection to the client when receiving
-the datagram. In general it's adequate to set up special instructions for
-programs that support being run by tcpwrapper.
-Options
-
-host
-    host either is a hostname, or a dotted-decimal IP address, or 0.
-    If host is 0, udpsvd accepts datagrams to any local IP address.
-port
-    udpsvd accepts datagrams to host:port. port may be a name from
-    /etc/services or a number.
-prog
-    prog consists of one or more arguments. udpsvd normally runs prog
-    to handle a datagram, and possibly more, that is sent to the socket,
-    if there is no program that was started before by udpsvd still running
-    and handling datagrams.
--i dir
-    read instructions for handling new connections from the instructions
-    directory dir. See ipsvd-instruct(5) for details.
--x cdb
-    read instructions for handling new connections from the constant
-    database cdb. The constant database normally is created from
-    an instructions directory by running ipsvd-cdb(8).
--t sec
-    timeout. This option only takes effect if the -i option is given.
-    While checking the instructions directory, check the time of last
-    access of the file that matches the clients address or hostname if any,
-    discard and remove the file if it wasn't accessed within the last
-    sec seconds; udpsvd does not discard or remove a file if the user's
-    write permission is not set, for those files the timeout is disabled.
-    Default is 0, which means that the timeout is disabled.
--l name
-    local hostname. Do not look up the local hostname in DNS, but use name
-    as hostname. By default udpsvd looks up the local hostname once at startup.
--u user[:group]
-    drop permissions. Switch user ID to user's UID, and group ID to user's
-    primary GID after creating and binding to the socket. If user
-    is followed by a colon and a group name, the group ID is switched
-    to the GID of group instead. All supplementary groups are removed.
--h
-    Look up the client's hostname in DNS.
--p
-    paranoid. After looking up the client's hostname in DNS, look up
-    the IP addresses in DNS for that hostname, and forget the hostname
-    if none of the addresses match the client's IP address. You should
-    set this option if you use hostname based instructions. The -p option
-    implies the -h option.
--v
-    verbose. Print verbose messages to standard output.
--vv
-    more verbose. Print more verbose messages to standard output.
-*/
-#endif
-
-
-#if ENABLE_TCPSVD
-/* Based on ipsvd ipsvd-0.12.1. This tcpsvd accepts all options
- * which are supported by one from ipsvd-0.12.1, but not all are
- * functional. See help text at the end of this file for details.
- *
- * Code inside "#ifdef SSLSVD" is for sslsvd and is currently unused.
- *
- * Output of verbose mode matches original (modulo bugs and
- * unimplemented stuff). Unnatural splitting of IP and PORT
- * is retained (personally I prefer one-value "IP:PORT" notation -
- * it is a natural string representation of struct sockaddr_XX).
- *
- * TCPORIGDST{IP,PORT} is busybox-specific addition
- */
-
-#include <limits.h>
-#include <linux/netfilter_ipv4.h> /* wants <limits.h> */
-#include "ipsvd_perhost.h"
-
-#ifdef SSLSVD
-#include "matrixSsl.h"
-#include "ssl_io.h"
-#endif
-
-static unsigned max_per_host; /* originally in ipsvd_check.c */
-static unsigned cur_per_host;
-static unsigned cnum;
-static unsigned cmax = 30;
-
 /* Must match getopt32 in main! */
 enum {
        OPT_c = (1 << 0),
@@ -342,7 +98,9 @@ enum {
 
 static void connection_status(void)
 {
-       printf("%s: info: status %u/%u\n", applet_name, cnum, cmax);
+       /* "only 1 client max" desn't need this */
+       if (cmax > 1)
+               printf("%s: info: status %u/%u\n", applet_name, cnum, cmax);
 }
 
 static void sig_child_handler(int sig)
@@ -362,8 +120,8 @@ static void sig_child_handler(int sig)
                connection_status();
 }
 
-int tcpsvd_main(int argc, char **argv);
-int tcpsvd_main(int argc, char **argv)
+int tcpudpsvd_main(int argc, char **argv);
+int tcpudpsvd_main(int argc, char **argv)
 {
        char *str_c, *str_C, *str_b, *str_t;
        char *user;
@@ -371,31 +129,35 @@ int tcpsvd_main(int argc, char **argv)
        const char *instructs;
        char *msg_per_host = NULL;
        unsigned len_per_host = len_per_host; /* gcc */
-       int need_hostnames, need_remote_ip;
+#ifndef SSLSVD
+       struct bb_uidgid_t ugid;
+#endif
+       bool need_hostnames, need_remote_ip, tcp;
+       uint16_t local_port;
+       char *local_hostname = NULL;
+       char *remote_hostname = (char*)""; /* "" used if no -h */
+       char *local_addr = local_addr; /* gcc */
+       char *remote_addr = remote_addr; /* gcc */
+       char *remote_ip = remote_addr; /* gcc */
+       len_and_sockaddr *lsa;
+       len_and_sockaddr local, remote;
+       socklen_t sa_len;
        int pid;
        int sock;
        int conn;
        unsigned backlog = 20;
-       len_and_sockaddr *lsa;
-       uint16_t local_port;
-       uint16_t remote_port = remote_port; /* gcc */
-       char *local_hostname = NULL;
-       char *remote_hostname = (char*)""; /* "" used if no -h */
-       char *local_ip = local_ip; /* gcc */
-       char *remote_ip = remote_ip; /* gcc */
-#ifndef SSLSVD
-       struct bb_uidgid_t ugid;
-#endif
+
+       tcp = (applet_name[0] == 't');
 
        /* 3+ args, -i at most once, -p implies -h, -v is counter */
-       opt_complementary = "-3:?:i--i:ph:vv";
+       opt_complementary = "-3:i--i:ph:vv";
 #ifdef SSLSVD
-       getopt32(argc, argv, "c:C:i:x:u:l:Eb:hpt:vU:/:Z:K:",
+       getopt32(argv, "+c:C:i:x:u:l:Eb:hpt:vU:/:Z:K:",
                &str_c, &str_C, &instructs, &instructs, &user, &local_hostname,
                &str_b, &str_t, &ssluser, &root, &cert, &key, &verbose
        );
 #else
-       getopt32(argc, argv, "c:C:i:x:u:l:Eb:hpt:v",
+       getopt32(argv, "+c:C:i:x:u:l:Eb:hpt:v",
                &str_c, &str_C, &instructs, &instructs, &user, &local_hostname,
                &str_b, &str_t, &verbose
        );
@@ -420,18 +182,22 @@ int tcpsvd_main(int argc, char **argv)
        if (option_mask32 & OPT_b)
                backlog = xatou(str_b);
 #ifdef SSLSVD
-       if (option_mask32 & OPT_U) ssluser = (char*)optarg; break;
-       if (option_mask32 & OPT_slash) root = (char*)optarg; break;
-       if (option_mask32 & OPT_Z) cert = (char*)optarg; break;
-       if (option_mask32 & OPT_K) key = (char*)optarg; break;
+       if (option_mask32 & OPT_U) ssluser = optarg;
+       if (option_mask32 & OPT_slash) root = optarg;
+       if (option_mask32 & OPT_Z) cert = optarg;
+       if (option_mask32 & OPT_K) key = optarg;
 #endif
        argv += optind;
        if (!argv[0][0] || LONE_CHAR(argv[0], '0'))
                argv[0] = (char*)"0.0.0.0";
 
+       /* Per-IP flood protection is not thought-out for UDP */
+       if (!tcp)
+               max_per_host = 0;
+
        /* stdout is used for logging, don't buffer */
        setlinebuf(stdout);
-       bb_sanitize_stdio(); /* fd# 1,2 must be opened */
+       bb_sanitize_stdio(); /* fd# 0,1,2 must be opened */
 
        need_hostnames = verbose || !(option_mask32 & OPT_E);
        need_remote_ip = max_per_host || need_hostnames;
@@ -446,10 +212,8 @@ int tcpsvd_main(int argc, char **argv)
        if (option_mask32 & OPT_u)
                if (!uidgid_get(&sslugid, ssluser, 1)) {
                        if (errno) {
-                               xfunc_exitcode = 100;
                                bb_perror_msg_and_die("fatal: cannot get user/group: %s", ssluser);
                        }
-                       xfunc_exitcode = 111;
                        bb_error_msg_and_die("fatal: unknown user/group '%s'", ssluser);
                }
        if (!cert) cert = "./cert.pem";
@@ -473,12 +237,16 @@ int tcpsvd_main(int argc, char **argv)
        if (max_per_host)
                ipsvd_perhost_init(cmax);
 
-       local_port = bb_lookup_port(argv[1], "tcp", 0);
+       local_port = bb_lookup_port(argv[1], tcp ? "tcp" : "udp", 0);
        lsa = xhost2sockaddr(argv[0], local_port);
-       sock = xsocket(lsa->sa.sa_family, SOCK_STREAM, 0);
-       setsockopt_reuseaddr(sock); /* desirable */
-       xbind(sock, &lsa->sa, lsa->len);
-       xlisten(sock, backlog);
+       sock = xsocket(lsa->sa.sa_family, tcp ? SOCK_STREAM : SOCK_DGRAM, 0);
+       setsockopt_reuseaddr(sock);
+       sa_len = lsa->len; /* I presume sockaddr len stays the same */
+       xbind(sock, &lsa->sa, sa_len);
+       if (tcp)
+               xlisten(sock, backlog);
+       else /* udp: needed for recv_from_to to work: */
+               socket_want_pktinfo(sock);
        /* ndelay_off(sock); - it is the default I think? */
 
 #ifndef SSLSVD
@@ -490,9 +258,7 @@ int tcpsvd_main(int argc, char **argv)
 #endif
 
        if (verbose) {
-               /* we do it only for ":port" cosmetics... oh well */
-               char *addr = xmalloc_sockaddr2dotted(&lsa->sa, lsa->len);
-
+               char *addr = xmalloc_sockaddr2dotted(&lsa->sa);
                printf("%s: info: listening on %s", applet_name, addr);
                free(addr);
 #ifndef SSLSVD
@@ -503,7 +269,7 @@ int tcpsvd_main(int argc, char **argv)
                puts(", starting");
        }
 
-       /* The rest is a main accept() loop */
+       /* Main accept() loop */
 
  again:
        hccp = NULL;
@@ -516,19 +282,27 @@ int tcpsvd_main(int argc, char **argv)
        close(0);
  again2:
        sig_unblock(SIGCHLD);
-       conn = accept(sock, &lsa->sa, &lsa->len);
+       if (tcp) {
+               remote.len = sa_len;
+               conn = accept(sock, &remote.sa, &remote.len);
+       } else {
+               /* In case recv_from_to won't be able to recover local addr.
+                * Also sets port - recv_from_to is unable to do it. */
+               local = *lsa;
+               conn = recv_from_to(sock, NULL, 0, MSG_PEEK, &remote.sa, &local.sa, sa_len);
+       }
        sig_block(SIGCHLD);
        if (conn < 0) {
                if (errno != EINTR)
-                       bb_perror_msg("accept");
+                       bb_perror_msg(tcp ? "accept" : "recv");
                goto again2;
        }
-       xmove_fd(conn, 0);
+       xmove_fd(tcp ? conn : sock, 0);
 
        if (max_per_host) {
                /* Drop connection immediately if cur_per_host > max_per_host
                 * (minimizing load under SYN flood) */
-               remote_ip = xmalloc_sockaddr2dotted_noport(&lsa->sa, lsa->len);
+               remote_ip = xmalloc_sockaddr2dotted_noport(&remote.sa);
                cur_per_host = ipsvd_perhost_add(remote_ip, max_per_host, &hccp);
                if (cur_per_host > max_per_host) {
                        /* ipsvd_perhost_add detected that max is exceeded
@@ -543,17 +317,57 @@ int tcpsvd_main(int argc, char **argv)
                }
        }
 
-       cnum++;
-       if (verbose)
-               connection_status();
+       if (!tcp) {
+               /* Voodoo magic: making udp sockets each receive its own
+                * packets is not trivial, and I still not sure
+                * I do it 100% right.
+                * 1) we have to do it before fork()
+                * 2) order is important - is it right now? */
+
+               /* Make plain write/send work for this socket by supplying default
+                * destination address. This also restricts incoming packets
+                * to ones coming from this remote IP. */
+               xconnect(0, &remote.sa, sa_len);
+       /* hole? at this point we have no wildcard udp socket...
+        * can this cause clients to get "port unreachable" icmp?
+        * Yup, time window is very small, but it exists (is it?) */
+               /* Open new non-connected UDP socket for further clients */
+               sock = xsocket(lsa->sa.sa_family, tcp ? SOCK_STREAM : SOCK_DGRAM, 0);
+               setsockopt_reuseaddr(sock);
+               xbind(sock, &lsa->sa, sa_len);
+               socket_want_pktinfo(sock);
+
+               /* Doesn't work:
+                * we cannot replace fd #0 - we will lose pending packet
+                * which is already buffered for us! And we cannot use fd #1
+                * instead - it will "intercept" all following packets, but child
+                * does not expect data coming *from fd #1*! */
+#if 0
+               /* Make it so that local addr is fixed to localp->sa
+                * and we don't accidentally accept packets to other local IPs. */
+               /* NB: we possibly bind to the _very_ same_ address & port as the one
+                * already bound in parent! This seems to work in Linux.
+                * (otherwise we can move socket to fd #0 only if bind succeeds) */
+               close(0);
+               set_nport(localp, htons(local_port));
+               xmove_fd(xsocket(localp->sa.sa_family, SOCK_DGRAM, 0), 0);
+               setsockopt_reuseaddr(0); /* crucial */
+               xbind(0, &localp->sa, localp->len);
+#endif
+       }
 
        pid = fork();
        if (pid == -1) {
                bb_perror_msg("fork");
                goto again;
        }
+
+
        if (pid != 0) {
                /* parent */
+               cnum++;
+               if (verbose)
+                       connection_status();
                if (hccp)
                        hccp->pid = pid;
                goto again;
@@ -561,77 +375,71 @@ int tcpsvd_main(int argc, char **argv)
 
        /* Child: prepare env, log, and exec prog */
 
-       close(sock); /* listening socket */
-       /* Find out local IP peer connected to.
-        * Errors ignored (I'm not paranoid enough to imagine kernel
-        * which doesn't know local IP). */
-       getsockname(0, &lsa->sa, &lsa->len);
-
-       if (need_remote_ip) {
-               if (!max_per_host)
-                       remote_ip = xmalloc_sockaddr2dotted_noport(&lsa->sa, lsa->len);
-               /* else it is already done */
-               remote_port = get_nport(&lsa->sa);
-               remote_port = ntohs(remote_port);
-       }
+       /* Closing tcp listening socket */
+       if (tcp)
+               close(sock);
+
+       if (need_remote_ip)
+               remote_addr = xmalloc_sockaddr2dotted(&remote.sa);
 
        if (need_hostnames) {
                if (option_mask32 & OPT_h) {
-                       remote_hostname = xmalloc_sockaddr2host(&lsa->sa, lsa->len);
+                       remote_hostname = xmalloc_sockaddr2host_noport(&remote.sa);
                        if (!remote_hostname) {
-                               bb_error_msg("warning: cannot look up hostname for %s", remote_ip);
+                               bb_error_msg("warning: cannot look up hostname for %s", remote_addr);
                                remote_hostname = (char*)"";
                        }
                }
-               local_ip = xmalloc_sockaddr2dotted_noport(&lsa->sa, lsa->len);
-               local_port = get_nport(&lsa->sa);
-               local_port = ntohs(local_port);
+               /* Find out local IP peer connected to.
+                * Errors ignored (I'm not paranoid enough to imagine kernel
+                * which doesn't know local IP). */
+               if (tcp) {
+                       local.len = sa_len;
+                       getsockname(0, &local.sa, &local.len);
+               }
+               local_addr = xmalloc_sockaddr2dotted(&local.sa);
                if (!local_hostname) {
-                       local_hostname = xmalloc_sockaddr2host_noport(&lsa->sa, lsa->len);
+                       local_hostname = xmalloc_sockaddr2host_noport(&local.sa);
                        if (!local_hostname)
-                               bb_error_msg_and_die("cannot look up local hostname for %s", local_ip);
+                               bb_error_msg_and_die("warning: cannot look up hostname for %s"+9, local_addr);
                }
        }
 
        if (verbose) {
                pid = getpid();
-               printf("%s: info: pid %u from %s\n", applet_name, pid, remote_ip);
+               printf("%s: info: pid %u from %s\n", applet_name, pid, remote_addr);
                if (max_per_host)
                        printf("%s: info: concurrency %u %s %u/%u\n",
                                applet_name, pid, remote_ip, cur_per_host, max_per_host);
-               printf("%s: info: start %u %s:%s :%s:%s:%u\n",
+               printf("%s: info: start %u %s:%s :%s:%s\n",
                        applet_name, pid,
-                       local_hostname, local_ip,
-                       remote_hostname, remote_ip, (unsigned)remote_port);
+                       local_hostname, local_addr,
+                       remote_hostname, remote_addr);
        }
 
        if (!(option_mask32 & OPT_E)) {
                /* setup ucspi env */
+               const char *proto = tcp ? "TCP" : "UDP";
 
                /* Extract "original" destination addr:port
                 * from Linux firewall. Useful when you redirect
                 * an outbond connection to local handler, and it needs
                 * to know where it originally tried to connect */
-               if (getsockopt(0, SOL_IP, SO_ORIGINAL_DST, &lsa->sa, &lsa->len) == 0) {
-                       char *ip = xmalloc_sockaddr2dotted_noport(&lsa->sa, lsa->len);
-                       unsigned port = get_nport(&lsa->sa);
-                       port = ntohs(port);
-                       xsetenv("TCPORIGDSTIP", ip);
-                       xsetenv("TCPORIGDSTPORT", utoa(port));
-                       free(ip);
+               if (tcp && getsockopt(0, SOL_IP, SO_ORIGINAL_DST, &lsa->sa, &lsa->len) == 0) {
+                       char *addr = xmalloc_sockaddr2dotted(&lsa->sa);
+                       xsetenv("TCPORIGDSTADDR", addr);
+                       free(addr);
                }
-               xsetenv("PROTO", "TCP");
-               xsetenv("TCPLOCALIP", local_ip);
-               xsetenv("TCPLOCALPORT", utoa(local_port));
-               xsetenv("TCPLOCALHOST", local_hostname);
-               xsetenv("TCPREMOTEIP", remote_ip);
-               xsetenv("TCPREMOTEPORT", utoa(remote_port));
+               xsetenv("PROTO", proto);
+               xsetenv_proto(proto, "LOCALADDR", local_addr);
+               xsetenv_proto(proto, "LOCALHOST", local_hostname);
+               xsetenv_proto(proto, "REMOTEADDR", remote_addr);
                if (option_mask32 & OPT_h) {
-                       xsetenv("TCPREMOTEHOST", remote_hostname);
+                       xsetenv_proto(proto, "REMOTEHOST", remote_hostname);
                }
-               xsetenv("TCPREMOTEINFO", "");
+               xsetenv_proto(proto, "REMOTEINFO", "");
                /* additional */
-               if (cur_per_host > 0)
+               if (cur_per_host > 0) /* can not be true for udp */
                        xsetenv("TCPCONCURRENCY", utoa(cur_per_host));
        }
 
@@ -747,4 +555,3 @@ prog
     more verbose. Print more verbose messages to standard output.
     * no difference between -v and -vv in busyboxed version
 */
-#endif