blob: 857d63570764a315ce8795c8a71b244266a62c5c [file] [log] [blame]
Erik Andersene49d5ec2000-02-08 19:58:47 +00001/* vi: set sw=4 ts=4: */
Erik Andersen7ab9c7e2000-05-12 19:41:47 +00002/* nc: mini-netcat - built from the ground up for LRP
Denis Vlasenko9213a9e2006-09-17 16:28:10 +00003 *
Rob Landley1cca9482006-07-10 19:45:20 +00004 * Copyright (C) 1998, 1999 Charles P. Wright
5 * Copyright (C) 1998 Dave Cinege
Denis Vlasenko9213a9e2006-09-17 16:28:10 +00006 *
Rob Landley1cca9482006-07-10 19:45:20 +00007 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
8 */
Eric Andersencc8ed391999-10-05 16:24:54 +00009
Denis Vlasenkob6adbf12007-05-26 19:00:18 +000010#include "libbb.h"
Eric Andersencc8ed391999-10-05 16:24:54 +000011
Denis Vlasenko29fe7262007-04-05 20:26:28 +000012#if ENABLE_DESKTOP
13#include "nc_bloaty.c"
14#else
15
Denis Vlasenko5d687242007-01-12 20:59:31 +000016/* Lots of small differences in features
17 * when compared to "standard" nc
18 */
19
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +000020static void timeout(int signum UNUSED_PARAM)
Mike Frysinger60a5c382005-05-06 04:45:38 +000021{
Denis Vlasenko13858992006-10-08 12:49:22 +000022 bb_error_msg_and_die("timed out");
Mike Frysinger60a5c382005-05-06 04:45:38 +000023}
24
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +000025int nc_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Erik Andersen7ab9c7e2000-05-12 19:41:47 +000026int nc_main(int argc, char **argv)
Eric Andersencc8ed391999-10-05 16:24:54 +000027{
Denis Vlasenko5d687242007-01-12 20:59:31 +000028 /* sfd sits _here_ only because of "repeat" option (-l -l). */
29 int sfd = sfd; /* for gcc */
Denis Vlasenkod0e70af2006-10-16 01:10:28 +000030 int cfd = 0;
Denis Vlasenko31635552007-01-20 16:54:19 +000031 unsigned lport = 0;
Denis Vlasenko5e34ff22009-04-21 11:09:40 +000032 IF_NOT_NC_SERVER(const) unsigned do_listen = 0;
33 IF_NOT_NC_EXTRA (const) unsigned wsecs = 0;
34 IF_NOT_NC_EXTRA (const) unsigned delay = 0;
35 IF_NOT_NC_EXTRA (const int execparam = 0;)
36 IF_NC_EXTRA (char **execparam = NULL;)
Denis Vlasenko5d687242007-01-12 20:59:31 +000037 len_and_sockaddr *lsa;
Erik Andersene49d5ec2000-02-08 19:58:47 +000038 fd_set readfds, testfds;
Denis Vlasenkod0e70af2006-10-16 01:10:28 +000039 int opt; /* must be signed (getopt returns -1) */
Erik Andersene49d5ec2000-02-08 19:58:47 +000040
Denis Vlasenko9213a9e2006-09-17 16:28:10 +000041 if (ENABLE_NC_SERVER || ENABLE_NC_EXTRA) {
Denis Vlasenkod0e70af2006-10-16 01:10:28 +000042 /* getopt32 is _almost_ usable:
43 ** it cannot handle "... -e prog -prog-opt" */
44 while ((opt = getopt(argc, argv,
Denis Vlasenko5e34ff22009-04-21 11:09:40 +000045 "" IF_NC_SERVER("lp:") IF_NC_EXTRA("w:i:f:e:") )) > 0
Denis Vlasenkod0e70af2006-10-16 01:10:28 +000046 ) {
Denis Vlasenkoc9ca0a32008-02-18 11:08:33 +000047 if (ENABLE_NC_SERVER && opt=='l')
Denis Vlasenko5e34ff22009-04-21 11:09:40 +000048 IF_NC_SERVER(do_listen++);
Denis Vlasenkoc9ca0a32008-02-18 11:08:33 +000049 else if (ENABLE_NC_SERVER && opt=='p')
Denis Vlasenko5e34ff22009-04-21 11:09:40 +000050 IF_NC_SERVER(lport = bb_lookup_port(optarg, "tcp", 0));
Denis Vlasenkoc9ca0a32008-02-18 11:08:33 +000051 else if (ENABLE_NC_EXTRA && opt=='w')
Denis Vlasenko5e34ff22009-04-21 11:09:40 +000052 IF_NC_EXTRA( wsecs = xatou(optarg));
Denis Vlasenkoc9ca0a32008-02-18 11:08:33 +000053 else if (ENABLE_NC_EXTRA && opt=='i')
Denis Vlasenko5e34ff22009-04-21 11:09:40 +000054 IF_NC_EXTRA( delay = xatou(optarg));
Denis Vlasenkoc9ca0a32008-02-18 11:08:33 +000055 else if (ENABLE_NC_EXTRA && opt=='f')
Denis Vlasenko5e34ff22009-04-21 11:09:40 +000056 IF_NC_EXTRA( cfd = xopen(optarg, O_RDWR));
Denis Vlasenkoc9ca0a32008-02-18 11:08:33 +000057 else if (ENABLE_NC_EXTRA && opt=='e' && optind <= argc) {
Denis Vlasenkod0e70af2006-10-16 01:10:28 +000058 /* We cannot just 'break'. We should let getopt finish.
59 ** Or else we won't be able to find where
60 ** 'host' and 'port' params are
61 ** (think "nc -w 60 host port -e prog"). */
Denis Vlasenko5e34ff22009-04-21 11:09:40 +000062 IF_NC_EXTRA(
Denis Vlasenkod0e70af2006-10-16 01:10:28 +000063 char **p;
64 // +2: one for progname (optarg) and one for NULL
65 execparam = xzalloc(sizeof(char*) * (argc - optind + 2));
66 p = execparam;
67 *p++ = optarg;
68 while (optind < argc) {
69 *p++ = argv[optind++];
70 }
71 )
72 /* optind points to argv[arvc] (NULL) now.
73 ** FIXME: we assume that getopt will not count options
74 ** possibly present on "-e prog args" and will not
75 ** include them into final value of optind
76 ** which is to be used ... */
Rob Landley1cca9482006-07-10 19:45:20 +000077 } else bb_show_usage();
Matt Kraai1d702672001-02-07 04:09:23 +000078 }
Denis Vlasenkod0e70af2006-10-16 01:10:28 +000079 argv += optind; /* ... here! */
80 argc -= optind;
81 // -l and -f don't mix
82 if (do_listen && cfd) bb_show_usage();
83 // Listen or file modes need zero arguments, client mode needs 2
Denis Vlasenko5d687242007-01-12 20:59:31 +000084 if (do_listen || cfd) {
85 if (argc) bb_show_usage();
86 } else {
87 if (!argc || argc > 2) bb_show_usage();
88 }
Denis Vlasenkod0e70af2006-10-16 01:10:28 +000089 } else {
90 if (argc != 3) bb_show_usage();
91 argc--;
92 argv++;
93 }
Eric Andersencc8ed391999-10-05 16:24:54 +000094
Mike Frysinger60a5c382005-05-06 04:45:38 +000095 if (wsecs) {
96 signal(SIGALRM, timeout);
97 alarm(wsecs);
98 }
Denis Vlasenko9213a9e2006-09-17 16:28:10 +000099
Denis Vlasenkod0e70af2006-10-16 01:10:28 +0000100 if (!cfd) {
Rob Landley1cca9482006-07-10 19:45:20 +0000101 if (do_listen) {
Denis Vlasenko5d687242007-01-12 20:59:31 +0000102 /* create_and_bind_stream_or_die(NULL, lport)
103 * would've work wonderfully, but we need
104 * to know lsa */
105 sfd = xsocket_stream(&lsa);
106 if (lport)
107 set_nport(lsa, htons(lport));
108 setsockopt_reuseaddr(sfd);
Bernhard Reutner-Fischer8c69afd2008-01-29 10:33:34 +0000109 xbind(sfd, &lsa->u.sa, lsa->len);
Denis Vlasenko5d687242007-01-12 20:59:31 +0000110 xlisten(sfd, do_listen); /* can be > 1 */
111 /* If we didn't specify a port number,
112 * query and print it after listen() */
Rob Landley1cca9482006-07-10 19:45:20 +0000113 if (!lport) {
Denis Vlasenko703e2022007-01-22 14:12:08 +0000114 socklen_t addrlen = lsa->len;
Bernhard Reutner-Fischer8c69afd2008-01-29 10:33:34 +0000115 getsockname(sfd, &lsa->u.sa, &addrlen);
116 lport = get_nport(&lsa->u.sa);
Denis Vlasenko5d687242007-01-12 20:59:31 +0000117 fdprintf(2, "%d\n", ntohs(lport));
Rob Landley1cca9482006-07-10 19:45:20 +0000118 }
Denis Vlasenko96e1b382007-09-30 23:50:48 +0000119 close_on_exec_on(sfd);
Denis Vlasenko5d687242007-01-12 20:59:31 +0000120 accept_again:
Denis Vlasenko703e2022007-01-22 14:12:08 +0000121 cfd = accept(sfd, NULL, 0);
Denis Vlasenko13858992006-10-08 12:49:22 +0000122 if (cfd < 0)
Rob Landley1cca9482006-07-10 19:45:20 +0000123 bb_perror_msg_and_die("accept");
Denis Vlasenko5d687242007-01-12 20:59:31 +0000124 if (!execparam)
125 close(sfd);
Rob Landley1cca9482006-07-10 19:45:20 +0000126 } else {
Denis Vlasenko5d687242007-01-12 20:59:31 +0000127 cfd = create_and_connect_stream_or_die(argv[0],
128 argv[1] ? bb_lookup_port(argv[1], "tcp", 0) : 0);
Rob Landley1cca9482006-07-10 19:45:20 +0000129 }
Matt Kraai1d702672001-02-07 04:09:23 +0000130 }
Eric Andersencc8ed391999-10-05 16:24:54 +0000131
Mike Frysinger60a5c382005-05-06 04:45:38 +0000132 if (wsecs) {
133 alarm(0);
Denis Vlasenko3fa36e22008-11-09 00:15:11 +0000134 /* Non-ignored siganls revert to SIG_DFL on exec anyway */
135 /*signal(SIGALRM, SIG_DFL);*/
Mike Frysinger60a5c382005-05-06 04:45:38 +0000136 }
137
Eric Andersen1323c942002-04-26 23:59:12 +0000138 /* -e given? */
Denis Vlasenkod0e70af2006-10-16 01:10:28 +0000139 if (execparam) {
Denis Vlasenko5d687242007-01-12 20:59:31 +0000140 signal(SIGCHLD, SIG_IGN);
Rob Landley1cca9482006-07-10 19:45:20 +0000141 // With more than one -l, repeatedly act as server.
Denis Vlasenko13858992006-10-08 12:49:22 +0000142 if (do_listen > 1 && vfork()) {
Denis Vlasenko5d687242007-01-12 20:59:31 +0000143 /* parent */
Rob Landley1cca9482006-07-10 19:45:20 +0000144 // This is a bit weird as cleanup goes, since we wind up with no
145 // stdin/stdout/stderr. But it's small and shouldn't hurt anything.
146 // We check for cfd == 0 above.
Denis Vlasenko13858992006-10-08 12:49:22 +0000147 logmode = LOGMODE_NONE;
Rob Landley1cca9482006-07-10 19:45:20 +0000148 close(0);
149 close(1);
150 close(2);
Denis Vlasenko5d687242007-01-12 20:59:31 +0000151 goto accept_again;
Rob Landley1cca9482006-07-10 19:45:20 +0000152 }
Denis Vlasenko5d687242007-01-12 20:59:31 +0000153 /* child (or main thread if no multiple -l) */
Denis Vlasenkofa0b56d2008-07-01 16:09:07 +0000154 xmove_fd(cfd, 0);
155 xdup2(0, 1);
156 xdup2(0, 2);
Denis Vlasenko5e34ff22009-04-21 11:09:40 +0000157 IF_NC_EXTRA(BB_EXECVP(execparam[0], execparam);)
Rob Landley1cca9482006-07-10 19:45:20 +0000158 /* Don't print stuff or it will go over the wire.... */
159 _exit(127);
160 }
161
162 // Select loop copying stdin to cfd, and cfd to stdout.
Denis Vlasenko9213a9e2006-09-17 16:28:10 +0000163
Erik Andersene49d5ec2000-02-08 19:58:47 +0000164 FD_ZERO(&readfds);
Rob Landley1cca9482006-07-10 19:45:20 +0000165 FD_SET(cfd, &readfds);
Matt Kraaibfa79672000-12-15 22:34:34 +0000166 FD_SET(STDIN_FILENO, &readfds);
Eric Andersencc8ed391999-10-05 16:24:54 +0000167
Rob Landley1cca9482006-07-10 19:45:20 +0000168 for (;;) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000169 int fd;
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000170 int ofd;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000171 int nread;
Eric Andersencc8ed391999-10-05 16:24:54 +0000172
Erik Andersene49d5ec2000-02-08 19:58:47 +0000173 testfds = readfds;
Eric Andersencc8ed391999-10-05 16:24:54 +0000174
Matt Kraaibfa79672000-12-15 22:34:34 +0000175 if (select(FD_SETSIZE, &testfds, NULL, NULL, NULL) < 0)
Manuel Novoa III cad53642003-03-19 09:13:01 +0000176 bb_perror_msg_and_die("select");
Eric Andersencc8ed391999-10-05 16:24:54 +0000177
Denis Vlasenko74324c82007-06-04 10:16:52 +0000178#define iobuf bb_common_bufsiz1
Erik Andersene49d5ec2000-02-08 19:58:47 +0000179 for (fd = 0; fd < FD_SETSIZE; fd++) {
180 if (FD_ISSET(fd, &testfds)) {
Denis Vlasenko74324c82007-06-04 10:16:52 +0000181 nread = safe_read(fd, iobuf, sizeof(iobuf));
Rob Landley1cca9482006-07-10 19:45:20 +0000182 if (fd == cfd) {
Denis Vlasenko5d687242007-01-12 20:59:31 +0000183 if (nread < 1)
Bernhard Reutner-Fischer636a1f82008-05-19 09:29:47 +0000184 exit(EXIT_SUCCESS);
Matt Kraaibfa79672000-12-15 22:34:34 +0000185 ofd = STDOUT_FILENO;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000186 } else {
Rob Landley53437472006-07-16 08:14:35 +0000187 if (nread<1) {
Rob Landley1cca9482006-07-10 19:45:20 +0000188 // Close outgoing half-connection so they get EOF, but
189 // leave incoming alone so we can see response.
190 shutdown(cfd, 1);
Paul Fox7b71d742005-07-18 22:23:16 +0000191 FD_CLR(STDIN_FILENO, &readfds);
192 }
Rob Landley1cca9482006-07-10 19:45:20 +0000193 ofd = cfd;
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000194 }
Denis Vlasenko74324c82007-06-04 10:16:52 +0000195 xwrite(ofd, iobuf, nread);
Rob Landley1cca9482006-07-10 19:45:20 +0000196 if (delay > 0) sleep(delay);
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000197 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000198 }
199 }
Eric Andersencc8ed391999-10-05 16:24:54 +0000200}
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000201#endif