httpd: do not set alarm() timeout if we read cached header
[oweals/busybox.git] / networking / nc.c
index 126bec906488193736e52400aaab5a8702ffa38c..b208f46c638f78f04e6d27592398d00766b16a2a 100644 (file)
@@ -1,47 +1,60 @@
 /* vi: set sw=4 ts=4: */
-/* nc: mini-netcat - built from the ground up for LRP
+/*
+ * nc: mini-netcat - built from the ground up for LRP
  *
  * Copyright (C) 1998, 1999  Charles P. Wright
  * Copyright (C) 1998  Dave Cinege
  *
  * Licensed under GPLv2 or later, see file LICENSE in this source tree.
  */
-
-#include "libbb.h"
-
 //config:config NC
-//config:      bool "nc"
+//config:      bool "nc (11 kb)"
 //config:      default y
 //config:      help
-//config:        A simple Unix utility which reads and writes data across network
-//config:        connections.
+//config:      A simple Unix utility which reads and writes data across network
+//config:      connections.
+//config:
+//config:config NETCAT
+//config:      bool "netcat (11 kb)"
+//config:      default n
+//config:      help
+//config:      Alias to nc.
 //config:
 //config:config NC_SERVER
 //config:      bool "Netcat server options (-l)"
 //config:      default y
-//config:      depends on NC
+//config:      depends on NC || NETCAT
 //config:      help
-//config:        Allow netcat to act as a server.
+//config:      Allow netcat to act as a server.
 //config:
 //config:config NC_EXTRA
 //config:      bool "Netcat extensions (-eiw and -f FILE)"
 //config:      default y
-//config:      depends on NC
+//config:      depends on NC || NETCAT
 //config:      help
-//config:        Add -e (support for executing the rest of the command line after
-//config:        making or receiving a successful connection), -i (delay interval for
-//config:        lines sent), -w (timeout for initial connection).
+//config:      Add -e (support for executing the rest of the command line after
+//config:      making or receiving a successful connection), -i (delay interval for
+//config:      lines sent), -w (timeout for initial connection).
 //config:
 //config:config NC_110_COMPAT
 //config:      bool "Netcat 1.10 compatibility (+2.5k)"
-//config:      default n  # off specially for Rob
-//config:      depends on NC
+//config:      default y
+//config:      depends on NC || NETCAT
 //config:      help
-//config:        This option makes nc closely follow original nc-1.10.
-//config:        The code is about 2.5k bigger. It enables
-//config:        -s ADDR, -n, -u, -v, -o FILE, -z options, but loses
-//config:        busybox-specific extensions: -f FILE.
+//config:      This option makes nc closely follow original nc-1.10.
+//config:      The code is about 2.5k bigger. It enables
+//config:      -s ADDR, -n, -u, -v, -o FILE, -z options, but loses
+//config:      busybox-specific extensions: -f FILE.
 
+//applet:IF_NC(APPLET(nc, BB_DIR_USR_BIN, BB_SUID_DROP))
+//                 APPLET_ODDNAME:name    main location        suid_type     help
+//applet:IF_NETCAT(APPLET_ODDNAME(netcat, nc,  BB_DIR_USR_BIN, BB_SUID_DROP, nc))
+
+//kbuild:lib-$(CONFIG_NC) += nc.o
+//kbuild:lib-$(CONFIG_NETCAT) += nc.o
+
+#include "libbb.h"
+#include "common_bufsiz.h"
 #if ENABLE_NC_110_COMPAT
 # include "nc_bloaty.c"
 #else
@@ -114,7 +127,7 @@ int nc_main(int argc, char **argv)
        IF_NOT_NC_EXTRA (const) unsigned delay = 0;
        IF_NOT_NC_EXTRA (const int execparam = 0;)
        IF_NC_EXTRA     (char **execparam = NULL;)
-       fd_set readfds, testfds;
+       struct pollfd pfds[2];
        int opt; /* must be signed (getopt returns -1) */
 
        if (ENABLE_NC_SERVER || ENABLE_NC_EXTRA) {
@@ -232,28 +245,28 @@ int nc_main(int argc, char **argv)
                IF_NC_EXTRA(bb_perror_msg_and_die("can't execute '%s'", execparam[0]);)
        }
 
-       /* Select loop copying stdin to cfd, and cfd to stdout */
+       /* loop copying stdin to cfd, and cfd to stdout */
 
-       FD_ZERO(&readfds);
-       FD_SET(cfd, &readfds);
-       FD_SET(STDIN_FILENO, &readfds);
+       pfds[0].fd = STDIN_FILENO;
+       pfds[0].events = POLLIN;
+       pfds[1].fd = cfd;
+       pfds[1].events = POLLIN;
 
+#define iobuf bb_common_bufsiz1
+       setup_common_bufsiz();
        for (;;) {
-               int fd;
+               int fdidx;
                int ofd;
                int nread;
 
-               testfds = readfds;
-
-               if (select(cfd + 1, &testfds, NULL, NULL, NULL) < 0)
-                       bb_perror_msg_and_die("select");
+               if (safe_poll(pfds, 2, -1) < 0)
+                       bb_perror_msg_and_die("poll");
 
-#define iobuf bb_common_bufsiz1
-               fd = STDIN_FILENO;
+               fdidx = 0;
                while (1) {
-                       if (FD_ISSET(fd, &testfds)) {
-                               nread = safe_read(fd, iobuf, sizeof(iobuf));
-                               if (fd == cfd) {
+                       if (pfds[fdidx].revents) {
+                               nread = safe_read(pfds[fdidx].fd, iobuf, COMMON_BUFSIZE);
+                               if (fdidx != 0) {
                                        if (nread < 1)
                                                exit(EXIT_SUCCESS);
                                        ofd = STDOUT_FILENO;
@@ -261,8 +274,8 @@ int nc_main(int argc, char **argv)
                                        if (nread < 1) {
                                                /* Close outgoing half-connection so they get EOF,
                                                 * but leave incoming alone so we can see response */
-                                               shutdown(cfd, 1);
-                                               FD_CLR(STDIN_FILENO, &readfds);
+                                               shutdown(cfd, SHUT_WR);
+                                               pfds[0].fd = -1;
                                        }
                                        ofd = cfd;
                                }
@@ -270,9 +283,9 @@ int nc_main(int argc, char **argv)
                                if (delay > 0)
                                        sleep(delay);
                        }
-                       if (fd == cfd)
+                       if (fdidx == 1)
                                break;
-                       fd = cfd;
+                       fdidx++;
                }
        }
 }