blob: 25b95246f4d0f0567e97871d7931e9911e027aea [file] [log] [blame]
Denis Vlasenko29fe7262007-04-05 20:26:28 +00001/* Based on netcat 1.10 RELEASE 960320 written by hobbit@avian.org.
2 * Released into public domain by the author.
3 *
Denis Vlasenko7cff01e2008-02-02 16:23:43 +00004 * Copyright (C) 2007 Denys Vlasenko.
Denis Vlasenko29fe7262007-04-05 20:26:28 +00005 *
Denys Vlasenko0ef64bd2010-08-16 20:14:46 +02006 * Licensed under GPLv2, see file LICENSE in this source tree.
Denis Vlasenko29fe7262007-04-05 20:26:28 +00007 */
8
9/* Author's comments from nc 1.10:
10 * =====================
11 * Netcat is entirely my own creation, although plenty of other code was used as
12 * examples. It is freely given away to the Internet community in the hope that
13 * it will be useful, with no restrictions except giving credit where it is due.
14 * No GPLs, Berkeley copyrights or any of that nonsense. The author assumes NO
15 * responsibility for how anyone uses it. If netcat makes you rich somehow and
16 * you're feeling generous, mail me a check. If you are affiliated in any way
17 * with Microsoft Network, get a life. Always ski in control. Comments,
18 * questions, and patches to hobbit@avian.org.
19 * ...
20 * Netcat and the associated package is a product of Avian Research, and is freely
21 * available in full source form with no restrictions save an obligation to give
22 * credit where due.
23 * ...
24 * A damn useful little "backend" utility begun 950915 or thereabouts,
25 * as *Hobbit*'s first real stab at some sockets programming. Something that
26 * should have and indeed may have existed ten years ago, but never became a
27 * standard Unix utility. IMHO, "nc" could take its place right next to cat,
28 * cp, rm, mv, dd, ls, and all those other cryptic and Unix-like things.
29 * =====================
30 *
31 * Much of author's comments are still retained in the code.
32 *
33 * Functionality removed (rationale):
34 * - miltiple-port ranges, randomized port scanning (use nmap)
35 * - telnet support (use telnet)
36 * - source routing
37 * - multiple DNS checks
38 * Functionalty which is different from nc 1.10:
Denys Vlasenko7d1201c2010-06-28 04:17:06 +020039 * - PROG in '-e PROG' can have ARGS (and options).
Denis Vlasenko29fe7262007-04-05 20:26:28 +000040 * Because of this -e option must be last.
Denys Vlasenko7d1201c2010-06-28 04:17:06 +020041//TODO: remove -e incompatibility?
42 * - we don't redirect stderr to the network socket for the -e PROG.
43 * (PROG can do it itself if needed, but sometimes it is NOT wanted!)
Denis Vlasenko19507f02007-04-06 10:41:05 +000044 * - numeric addresses are printed in (), not [] (IPv6 looks better),
45 * port numbers are inside (): (1.2.3.4:5678)
46 * - network read errors are reported on verbose levels > 1
47 * (nc 1.10 treats them as EOF)
48 * - TCP connects from wrong ip/ports (if peer ip:port is specified
49 * on the command line, but accept() says that it came from different addr)
Denys Vlasenko7d1201c2010-06-28 04:17:06 +020050 * are closed, but we don't exit - we continue to listen/accept.
Denys Vlasenkoffeeb7a2013-07-28 22:23:12 +020051 * Since bbox 1.22:
52 * - nc exits when _both_ stdin and network are closed.
53 * This makes these two commands:
54 * echo "Yes" | nc 127.0.0.1 1234
55 * echo "no" | nc -lp 1234
56 * exchange their data _and exit_ instead of being stuck.
Denis Vlasenko29fe7262007-04-05 20:26:28 +000057 */
58
Denis Vlasenkob6adbf12007-05-26 19:00:18 +000059/* done in nc.c: #include "libbb.h" */
Denis Vlasenko29fe7262007-04-05 20:26:28 +000060
Denys Vlasenko77cc2c52010-06-27 04:22:02 +020061//usage:#if ENABLE_NC_110_COMPAT
62//usage:
63//usage:#define nc_trivial_usage
64//usage: "[OPTIONS] HOST PORT - connect"
65//usage: IF_NC_SERVER("\n"
Denys Vlasenko3b2acb72010-10-09 21:10:32 +020066//usage: "nc [OPTIONS] -l -p PORT [HOST] [PORT] - listen"
67//usage: )
Denys Vlasenko77cc2c52010-06-27 04:22:02 +020068//usage:#define nc_full_usage "\n\n"
Denys Vlasenko66426762011-06-05 03:58:28 +020069//usage: " -e PROG Run PROG after connect (must be last)"
Denys Vlasenko77cc2c52010-06-27 04:22:02 +020070//usage: IF_NC_SERVER(
71//usage: "\n -l Listen mode, for inbound connects"
Denys Vlasenkode6f1482013-02-28 12:20:06 +010072//usage: "\n -lk With -e, provides persistent server"
73/* -ll does the same as -lk, but its our extension, while -k is BSD'd,
74 * presumably more widely known. Therefore we advertise it, not -ll.
75 * I would like to drop -ll support, but our "small" nc supports it,
76 * and Rob uses it.
77 */
Denys Vlasenko77cc2c52010-06-27 04:22:02 +020078//usage: )
79//usage: "\n -p PORT Local port"
80//usage: "\n -s ADDR Local address"
81//usage: "\n -w SEC Timeout for connects and final net reads"
82//usage: IF_NC_EXTRA(
83//usage: "\n -i SEC Delay interval for lines sent" /* ", ports scanned" */
84//usage: )
85//usage: "\n -n Don't do DNS resolution"
86//usage: "\n -u UDP mode"
Norbert Langea16c8ef2020-02-14 22:13:38 +010087//usage: "\n -b Allow broadcasts"
Denys Vlasenko77cc2c52010-06-27 04:22:02 +020088//usage: "\n -v Verbose"
89//usage: IF_NC_EXTRA(
90//usage: "\n -o FILE Hex dump traffic"
91//usage: "\n -z Zero-I/O mode (scanning)"
92//usage: )
93//usage:#endif
94
95/* "\n -r Randomize local and remote ports" */
96/* "\n -g gateway Source-routing hop point[s], up to 8" */
97/* "\n -G num Source-routing pointer: 4, 8, 12, ..." */
98/* "\nport numbers can be individual or ranges: lo-hi [inclusive]" */
99
100/* -e PROG can take ARGS too: "nc ... -e ls -l", but we don't document it
101 * in help text: nc 1.10 does not allow that. We don't want to entice
102 * users to use this incompatibility */
103
Denis Vlasenko19507f02007-04-06 10:41:05 +0000104enum {
105 SLEAZE_PORT = 31337, /* for UDP-scan RTT trick, change if ya want */
106 BIGSIZ = 8192, /* big buffers */
107
108 netfd = 3,
109 ofd = 4,
110};
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000111
112struct globals {
Denis Vlasenko19507f02007-04-06 10:41:05 +0000113 /* global cmd flags: */
114 unsigned o_verbose;
115 unsigned o_wait;
116#if ENABLE_NC_EXTRA
117 unsigned o_interval;
118#endif
119
120 /*int netfd;*/
121 /*int ofd;*/ /* hexdump output fd */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000122#if ENABLE_LFS
123#define SENT_N_RECV_M "sent %llu, rcvd %llu\n"
124 unsigned long long wrote_out; /* total stdout bytes */
125 unsigned long long wrote_net; /* total net bytes */
126#else
Denis Vlasenko19507f02007-04-06 10:41:05 +0000127#define SENT_N_RECV_M "sent %u, rcvd %u\n"
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000128 unsigned wrote_out; /* total stdout bytes */
129 unsigned wrote_net; /* total net bytes */
130#endif
Denys Vlasenko5e896482012-03-19 01:17:36 +0100131 char *proggie0saved;
Bernhard Reutner-Fischera985d302008-02-11 11:44:38 +0000132 /* ouraddr is never NULL and goes through three states as we progress:
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000133 1 - local address before bind (IP/port possibly zero)
134 2 - local address after bind (port is nonzero)
135 3 - local address after connect??/recv/accept (IP and port are nonzero) */
136 struct len_and_sockaddr *ouraddr;
137 /* themaddr is NULL if no peer hostname[:port] specified on command line */
138 struct len_and_sockaddr *themaddr;
139 /* remend is set after connect/recv/accept to the actual ip:port of peer */
140 struct len_and_sockaddr remend;
141
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000142 jmp_buf jbuf; /* timer crud */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000143
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000144 char bigbuf_in[BIGSIZ]; /* data buffers */
145 char bigbuf_net[BIGSIZ];
146};
147
148#define G (*ptr_to_globals)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000149#define wrote_out (G.wrote_out )
150#define wrote_net (G.wrote_net )
151#define ouraddr (G.ouraddr )
152#define themaddr (G.themaddr )
153#define remend (G.remend )
154#define jbuf (G.jbuf )
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000155#define bigbuf_in (G.bigbuf_in )
156#define bigbuf_net (G.bigbuf_net)
157#define o_verbose (G.o_verbose )
158#define o_wait (G.o_wait )
159#if ENABLE_NC_EXTRA
160#define o_interval (G.o_interval)
161#else
162#define o_interval 0
163#endif
Denis Vlasenko574f2f42008-02-27 18:41:59 +0000164#define INIT_G() do { \
165 SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
166} while (0)
167
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000168
169/* Must match getopt32 call! */
170enum {
Denys Vlasenko5e896482012-03-19 01:17:36 +0100171 OPT_n = (1 << 0),
172 OPT_p = (1 << 1),
173 OPT_s = (1 << 2),
174 OPT_u = (1 << 3),
Norbert Langea16c8ef2020-02-14 22:13:38 +0100175 OPT_b = (1 << 4),
176 OPT_v = (1 << 5),
177 OPT_w = (1 << 6),
178 OPT_l = (1 << 7) * ENABLE_NC_SERVER,
179 OPT_k = (1 << 8) * ENABLE_NC_SERVER,
180 OPT_i = (1 << (7+2*ENABLE_NC_SERVER)) * ENABLE_NC_EXTRA,
181 OPT_o = (1 << (8+2*ENABLE_NC_SERVER)) * ENABLE_NC_EXTRA,
182 OPT_z = (1 << (9+2*ENABLE_NC_SERVER)) * ENABLE_NC_EXTRA,
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000183};
184
185#define o_nflag (option_mask32 & OPT_n)
186#define o_udpmode (option_mask32 & OPT_u)
Norbert Langea16c8ef2020-02-14 22:13:38 +0100187#define o_bcmode (option_mask32 & OPT_b)
Denis Vlasenko19507f02007-04-06 10:41:05 +0000188#if ENABLE_NC_EXTRA
189#define o_ofile (option_mask32 & OPT_o)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000190#define o_zero (option_mask32 & OPT_z)
191#else
Denis Vlasenko19507f02007-04-06 10:41:05 +0000192#define o_ofile 0
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000193#define o_zero 0
194#endif
195
Denis Vlasenko19507f02007-04-06 10:41:05 +0000196/* Debug: squirt whatever message and sleep a bit so we can see it go by. */
197/* Beware: writes to stdOUT... */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000198#if 0
Denys Vlasenkoec16c032020-11-29 11:37:34 +0100199#define Debug(...) do { printf(__VA_ARGS__); printf("\n"); fflush_all(); sleep1(); } while (0)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000200#else
Denis Vlasenko51742f42007-04-12 00:32:05 +0000201#define Debug(...) do { } while (0)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000202#endif
203
James Byrne69374872019-07-02 11:35:03 +0200204#define holler_error(msg) do { if (o_verbose) bb_simple_error_msg(msg); } while (0)
205#define holler_perror(msg) do { if (o_verbose) bb_simple_perror_msg(msg); } while (0)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000206
207/* catch: no-brainer interrupt handler */
208static void catch(int sig)
209{
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000210 if (o_verbose > 1) /* normally we don't care */
211 fprintf(stderr, SENT_N_RECV_M, wrote_net, wrote_out);
212 fprintf(stderr, "punt!\n");
Denis Vlasenko400d8bb2008-02-24 13:36:01 +0000213 kill_myself_with_sig(sig);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000214}
215
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000216/* unarm */
217static void unarm(void)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000218{
219 signal(SIGALRM, SIG_IGN);
220 alarm(0);
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000221}
222
223/* timeout and other signal handling cruft */
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000224static void tmtravel(int sig UNUSED_PARAM)
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000225{
226 unarm();
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000227 longjmp(jbuf, 1);
228}
229
230/* arm: set the timer. */
231static void arm(unsigned secs)
232{
233 signal(SIGALRM, tmtravel);
234 alarm(secs);
235}
236
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000237/* findline:
238 find the next newline in a buffer; return inclusive size of that "line",
239 or the entire buffer size, so the caller knows how much to then write().
240 Not distinguishing \n vs \r\n for the nonce; it just works as is... */
241static unsigned findline(char *buf, unsigned siz)
242{
243 char * p;
244 int x;
245 if (!buf) /* various sanity checks... */
246 return 0;
247 if (siz > BIGSIZ)
248 return 0;
249 x = siz;
250 for (p = buf; x > 0; x--) {
251 if (*p == '\n') {
252 x = (int) (p - buf);
253 x++; /* 'sokay if it points just past the end! */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000254Debug("findline returning %d", x);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000255 return x;
256 }
257 p++;
258 } /* for */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000259Debug("findline returning whole thing: %d", siz);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000260 return siz;
261} /* findline */
262
263/* doexec:
264 fiddle all the file descriptors around, and hand off to another prog. Sort
265 of like a one-off "poor man's inetd". This is the only section of code
266 that would be security-critical, which is why it's ifdefed out by default.
267 Use at your own hairy risk; if you leave shells lying around behind open
268 listening ports you deserve to lose!! */
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000269static int doexec(char **proggie) NORETURN;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000270static int doexec(char **proggie)
271{
Denys Vlasenko5e896482012-03-19 01:17:36 +0100272 if (G.proggie0saved)
273 proggie[0] = G.proggie0saved;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000274 xmove_fd(netfd, 0);
275 dup2(0, 1);
276 /* dup2(0, 2); - do we *really* want this? NO!
277 * exec'ed prog can do it yourself, if needed */
Denys Vlasenko1c31e9e2010-11-28 04:34:09 +0100278 BB_EXECVP_or_die(proggie);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000279}
280
281/* connect_w_timeout:
282 return an fd for one of
283 an open outbound TCP connection, a UDP stub-socket thingie, or
284 an unconnected TCP or UDP socket to listen on.
285 Examines various global o_blah flags to figure out what to do.
286 lad can be NULL, then socket is not bound to any local ip[:port] */
287static int connect_w_timeout(int fd)
288{
289 int rr;
290
291 /* wrap connect inside a timer, and hit it */
292 arm(o_wait);
293 if (setjmp(jbuf) == 0) {
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000294 rr = connect(fd, &themaddr->u.sa, themaddr->len);
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000295 unarm();
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000296 } else { /* setjmp: connect failed... */
297 rr = -1;
298 errno = ETIMEDOUT; /* fake it */
299 }
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000300 return rr;
301}
302
303/* dolisten:
304 listens for
305 incoming and returns an open connection *from* someplace. If we were
306 given host/port args, any connections from elsewhere are rejected. This
307 in conjunction with local-address binding should limit things nicely... */
Denys Vlasenkode6f1482013-02-28 12:20:06 +0100308static void dolisten(int is_persistent, char **proggie)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000309{
310 int rr;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000311
312 if (!o_udpmode)
313 xlisten(netfd, 1); /* TCP: gotta listen() before we can get */
314
315 /* Various things that follow temporarily trash bigbuf_net, which might contain
316 a copy of any recvfrom()ed packet, but we'll read() another copy later. */
317
318 /* I can't believe I have to do all this to get my own goddamn bound address
319 and port number. It should just get filled in during bind() or something.
320 All this is only useful if we didn't say -p for listening, since if we
321 said -p we *know* what port we're listening on. At any rate we won't bother
322 with it all unless we wanted to see it, although listening quietly on a
323 random unknown port is probably not very useful without "netstat". */
324 if (o_verbose) {
325 char *addr;
Denis Vlasenkoa771e7c2009-04-21 23:48:38 +0000326 getsockname(netfd, &ouraddr->u.sa, &ouraddr->len);
327 //if (rr < 0)
328 // bb_perror_msg_and_die("getsockname after bind");
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000329 addr = xmalloc_sockaddr2dotted(&ouraddr->u.sa);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000330 fprintf(stderr, "listening on %s ...\n", addr);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000331 free(addr);
332 }
333
334 if (o_udpmode) {
335 /* UDP is a speeeeecial case -- we have to do I/O *and* get the calling
336 party's particulars all at once, listen() and accept() don't apply.
337 At least in the BSD universe, however, recvfrom/PEEK is enough to tell
338 us something came in, and we can set things up so straight read/write
339 actually does work after all. Yow. YMMV on strange platforms! */
340
341 /* I'm not completely clear on how this works -- BSD seems to make UDP
342 just magically work in a connect()ed context, but we'll undoubtedly run
343 into systems this deal doesn't work on. For now, we apparently have to
344 issue a connect() on our just-tickled socket so we can write() back.
345 Again, why the fuck doesn't it just get filled in and taken care of?!
346 This hack is anything but optimal. Basically, if you want your listener
347 to also be able to send data back, you need this connect() line, which
348 also has the side effect that now anything from a different source or even a
349 different port on the other end won't show up and will cause ICMP errors.
350 I guess that's what they meant by "connect".
351 Let's try to remember what the "U" is *really* for, eh? */
352
353 /* If peer address is specified, connect to it */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000354 remend.len = LSA_SIZEOF_SA;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000355 if (themaddr) {
356 remend = *themaddr;
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000357 xconnect(netfd, &themaddr->u.sa, themaddr->len);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000358 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000359 /* peek first packet and remember peer addr */
360 arm(o_wait); /* might as well timeout this, too */
361 if (setjmp(jbuf) == 0) { /* do timeout for initial connect */
362 /* (*ouraddr) is prefilled with "default" address */
363 /* and here we block... */
364 rr = recv_from_to(netfd, NULL, 0, MSG_PEEK, /*was bigbuf_net, BIGSIZ*/
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000365 &remend.u.sa, &ouraddr->u.sa, ouraddr->len);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000366 if (rr < 0)
James Byrne69374872019-07-02 11:35:03 +0200367 bb_simple_perror_msg_and_die("recvfrom");
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000368 unarm();
Denis Vlasenko19507f02007-04-06 10:41:05 +0000369 } else
James Byrne69374872019-07-02 11:35:03 +0200370 bb_simple_error_msg_and_die("timeout");
Denis Vlasenko19507f02007-04-06 10:41:05 +0000371/* Now we learned *to which IP* peer has connected, and we want to anchor
372our socket on it, so that our outbound packets will have correct local IP.
373Unfortunately, bind() on already bound socket will fail now (EINVAL):
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000374 xbind(netfd, &ouraddr->u.sa, ouraddr->len);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000375Need to read the packet, save data, close this socket and
376create new one, and bind() it. TODO */
377 if (!themaddr)
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000378 xconnect(netfd, &remend.u.sa, ouraddr->len);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000379 } else {
380 /* TCP */
Denys Vlasenkode6f1482013-02-28 12:20:06 +0100381 another:
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000382 arm(o_wait); /* wrap this in a timer, too; 0 = forever */
383 if (setjmp(jbuf) == 0) {
Denis Vlasenko19507f02007-04-06 10:41:05 +0000384 again:
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000385 remend.len = LSA_SIZEOF_SA;
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000386 rr = accept(netfd, &remend.u.sa, &remend.len);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000387 if (rr < 0)
James Byrne69374872019-07-02 11:35:03 +0200388 bb_simple_perror_msg_and_die("accept");
Denys Vlasenko866710a2010-01-08 16:09:45 +0100389 if (themaddr) {
390 int sv_port, port, r;
391
392 sv_port = get_nport(&remend.u.sa); /* save */
393 port = get_nport(&themaddr->u.sa);
394 if (port == 0) {
395 /* "nc -nl -p LPORT RHOST" (w/o RPORT!):
396 * we should accept any remote port */
Denys Vlasenkoca183112011-04-07 17:52:20 +0200397 set_nport(&remend.u.sa, 0); /* blot out remote port# */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000398 }
Denys Vlasenko866710a2010-01-08 16:09:45 +0100399 r = memcmp(&remend.u.sa, &themaddr->u.sa, remend.len);
Denys Vlasenkoca183112011-04-07 17:52:20 +0200400 set_nport(&remend.u.sa, sv_port); /* restore */
Denys Vlasenko866710a2010-01-08 16:09:45 +0100401 if (r != 0) {
402 /* nc 1.10 bails out instead, and its error message
403 * is not suppressed by o_verbose */
404 if (o_verbose) {
405 char *remaddr = xmalloc_sockaddr2dotted(&remend.u.sa);
406 bb_error_msg("connect from wrong ip/port %s ignored", remaddr);
407 free(remaddr);
408 }
409 close(rr);
410 goto again;
411 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000412 }
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000413 unarm();
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000414 } else
James Byrne69374872019-07-02 11:35:03 +0200415 bb_simple_error_msg_and_die("timeout");
Denys Vlasenkode6f1482013-02-28 12:20:06 +0100416
417 if (is_persistent && proggie) {
418 /* -l -k -e PROG */
419 signal(SIGCHLD, SIG_IGN); /* no zombies please */
420 if (xvfork() != 0) {
421 /* parent: go back and accept more connections */
422 close(rr);
423 goto another;
424 }
425 /* child */
426 signal(SIGCHLD, SIG_DFL);
427 }
428
Denis Vlasenko19507f02007-04-06 10:41:05 +0000429 xmove_fd(rr, netfd); /* dump the old socket, here's our new one */
430 /* find out what address the connection was *to* on our end, in case we're
431 doing a listen-on-any on a multihomed machine. This allows one to
432 offer different services via different alias addresses, such as the
433 "virtual web site" hack. */
Denis Vlasenkoa771e7c2009-04-21 23:48:38 +0000434 getsockname(netfd, &ouraddr->u.sa, &ouraddr->len);
435 //if (rr < 0)
436 // bb_perror_msg_and_die("getsockname after accept");
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000437 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000438
439 if (o_verbose) {
440 char *lcladdr, *remaddr, *remhostname;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000441
442#if ENABLE_NC_EXTRA && defined(IP_OPTIONS)
443 /* If we can, look for any IP options. Useful for testing the receiving end of
444 such things, and is a good exercise in dealing with it. We do this before
445 the connect message, to ensure that the connect msg is uniformly the LAST
446 thing to emerge after all the intervening crud. Doesn't work for UDP on
447 any machines I've tested, but feel free to surprise me. */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000448 char optbuf[40];
Denis Vlasenkoc4f12f52008-05-12 14:35:56 +0000449 socklen_t x = sizeof(optbuf);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000450
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000451 rr = getsockopt(netfd, IPPROTO_IP, IP_OPTIONS, optbuf, &x);
Denis Vlasenkof6b46852009-04-25 13:16:53 +0000452 if (rr >= 0 && x) { /* we've got options, lessee em... */
Denys Vlasenko2ea73b52011-10-19 22:31:01 +0200453 *bin2hex(bigbuf_net, optbuf, x) = '\0';
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000454 fprintf(stderr, "IP options: %s\n", bigbuf_net);
455 }
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000456#endif
457
458 /* now check out who it is. We don't care about mismatched DNS names here,
459 but any ADDR and PORT we specified had better fucking well match the caller.
460 Converting from addr to inet_ntoa and back again is a bit of a kludge, but
461 gethostpoop wants a string and there's much gnarlier code out there already,
462 so I don't feel bad.
463 The *real* question is why BFD sockets wasn't designed to allow listens for
464 connections *from* specific hosts/ports, instead of requiring the caller to
Denys Vlasenko10ad6222017-04-17 16:13:32 +0200465 accept the connection and then reject undesirable ones by closing.
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000466 In other words, we need a TCP MSG_PEEK. */
467 /* bbox: removed most of it */
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000468 lcladdr = xmalloc_sockaddr2dotted(&ouraddr->u.sa);
469 remaddr = xmalloc_sockaddr2dotted(&remend.u.sa);
470 remhostname = o_nflag ? remaddr : xmalloc_sockaddr2host(&remend.u.sa);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000471 fprintf(stderr, "connect to %s from %s (%s)\n",
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000472 lcladdr, remhostname, remaddr);
473 free(lcladdr);
474 free(remaddr);
475 if (!o_nflag)
476 free(remhostname);
477 }
Denys Vlasenkode6f1482013-02-28 12:20:06 +0100478
479 if (proggie)
480 doexec(proggie);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000481}
482
483/* udptest:
484 fire a couple of packets at a UDP target port, just to see if it's really
485 there. On BSD kernels, ICMP host/port-unreachable errors get delivered to
486 our socket as ECONNREFUSED write errors. On SV kernels, we lose; we'll have
487 to collect and analyze raw ICMP ourselves a la satan's probe_udp_ports
488 backend. Guess where one could swipe the appropriate code from...
489
490 Use the time delay between writes if given, otherwise use the "tcp ping"
491 trick for getting the RTT. [I got that idea from pluvius, and warped it.]
492 Return either the original fd, or clean up and return -1. */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000493#if ENABLE_NC_EXTRA
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000494static int udptest(void)
495{
496 int rr;
497
498 rr = write(netfd, bigbuf_in, 1);
499 if (rr != 1)
James Byrne69374872019-07-02 11:35:03 +0200500 bb_simple_perror_msg("udptest first write");
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000501
502 if (o_wait)
Denis Vlasenko19507f02007-04-06 10:41:05 +0000503 sleep(o_wait); // can be interrupted! while (t) nanosleep(&t)?
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000504 else {
505 /* use the tcp-ping trick: try connecting to a normally refused port, which
506 causes us to block for the time that SYN gets there and RST gets back.
507 Not completely reliable, but it *does* mostly work. */
Denys Vlasenko10ad6222017-04-17 16:13:32 +0200508 /* Set a temporary connect timeout, so packet filtration doesn't cause
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000509 us to hang forever, and hit it */
510 o_wait = 5; /* enough that we'll notice?? */
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000511 rr = xsocket(ouraddr->u.sa.sa_family, SOCK_STREAM, 0);
Denys Vlasenkoca183112011-04-07 17:52:20 +0200512 set_nport(&themaddr->u.sa, htons(SLEAZE_PORT));
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000513 connect_w_timeout(rr);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000514 /* don't need to restore themaddr's port, it's not used anymore */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000515 close(rr);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000516 o_wait = 0; /* restore */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000517 }
518
519 rr = write(netfd, bigbuf_in, 1);
520 return (rr != 1); /* if rr == 1, return 0 (success) */
521}
Denis Vlasenko19507f02007-04-06 10:41:05 +0000522#else
523int udptest(void);
524#endif
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000525
526/* oprint:
527 Hexdump bytes shoveled either way to a running logfile, in the format:
528 D offset - - - - --- 16 bytes --- - - - - # .... ascii .....
529 where "which" sets the direction indicator, D:
530 0 -- sent to network, or ">"
531 1 -- rcvd and printed to stdout, or "<"
532 and "buf" and "n" are data-block and length. If the current block generates
533 a partial line, so be it; we *want* that lockstep indication of who sent
534 what when. Adapted from dgaudet's original example -- but must be ripping
535 *fast*, since we don't want to be too disk-bound... */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000536#if ENABLE_NC_EXTRA
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000537static void oprint(int direction, unsigned char *p, unsigned bc)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000538{
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000539 unsigned obc; /* current "global" offset */
540 unsigned x;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000541 unsigned char *op; /* out hexdump ptr */
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000542 unsigned char *ap; /* out asc-dump ptr */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000543 unsigned char stage[100];
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000544
Denis Vlasenko19507f02007-04-06 10:41:05 +0000545 if (bc == 0)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000546 return;
547
Denis Vlasenko19507f02007-04-06 10:41:05 +0000548 obc = wrote_net; /* use the globals! */
549 if (direction == '<')
550 obc = wrote_out;
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000551 stage[0] = direction;
552 stage[59] = '#'; /* preload separator */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000553 stage[60] = ' ';
554
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000555 do { /* for chunk-o-data ... */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000556 x = 16;
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000557 if (bc < 16) {
558 /* memset(&stage[bc*3 + 11], ' ', 16*3 - bc*3); */
559 memset(&stage[11], ' ', 16*3);
560 x = bc;
561 }
Denis Vlasenkoc4f12f52008-05-12 14:35:56 +0000562 sprintf((char *)&stage[1], " %8.8x ", obc); /* xxx: still slow? */
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000563 bc -= x; /* fix current count */
564 obc += x; /* fix current offset */
565 op = &stage[11]; /* where hex starts */
566 ap = &stage[61]; /* where ascii starts */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000567
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000568 do { /* for line of dump, however long ... */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000569 *op++ = 0x20 | bb_hexdigits_upcase[*p >> 4];
570 *op++ = 0x20 | bb_hexdigits_upcase[*p & 0x0f];
571 *op++ = ' ';
572 if ((*p > 31) && (*p < 127))
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000573 *ap = *p; /* printing */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000574 else
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000575 *ap = '.'; /* nonprinting, loose def */
576 ap++;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000577 p++;
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000578 } while (--x);
579 *ap++ = '\n'; /* finish the line */
580 xwrite(ofd, stage, ap - stage);
581 } while (bc);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000582}
Denis Vlasenko19507f02007-04-06 10:41:05 +0000583#else
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000584void oprint(int direction, unsigned char *p, unsigned bc);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000585#endif
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000586
587/* readwrite:
Denys Vlasenko94dcfd82017-02-17 17:48:59 +0100588 handle stdin/stdout/network I/O. Bwahaha!! -- the i/o loop from hell.
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000589 In this instance, return what might become our exit status. */
590static int readwrite(void)
591{
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000592 char *zp = zp; /* gcc */ /* stdin buf ptr */
593 char *np = np; /* net-in buf ptr */
594 unsigned rzleft;
595 unsigned rnleft;
596 unsigned netretry; /* net-read retry counter */
Denys Vlasenkoffeeb7a2013-07-28 22:23:12 +0200597 unsigned fds_open;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000598
Denys Vlasenko94dcfd82017-02-17 17:48:59 +0100599 struct pollfd pfds[2];
600 pfds[0].fd = STDIN_FILENO;
601 pfds[0].events = POLLIN;
602 pfds[1].fd = netfd;
603 pfds[1].events = POLLIN;
Denys Vlasenkoffeeb7a2013-07-28 22:23:12 +0200604
Denys Vlasenko94dcfd82017-02-17 17:48:59 +0100605 fds_open = 2;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000606 netretry = 2;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000607 rzleft = rnleft = 0;
608 if (o_interval)
609 sleep(o_interval); /* pause *before* sending stuff, too */
610
Denys Vlasenko94dcfd82017-02-17 17:48:59 +0100611 /* and now the big ol' shoveling loop ... */
Denys Vlasenkoffeeb7a2013-07-28 22:23:12 +0200612 /* nc 1.10 has "while (FD_ISSET(netfd)" here */
613 while (fds_open) {
Denys Vlasenko94dcfd82017-02-17 17:48:59 +0100614 int rr;
615 int poll_tmout_ms;
Denys Vlasenkofd3ac962013-07-28 22:24:51 +0200616 unsigned wretry = 8200; /* net-write sanity counter */
617
Denys Vlasenko94dcfd82017-02-17 17:48:59 +0100618 poll_tmout_ms = -1;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000619 if (o_wait) {
Denys Vlasenko94dcfd82017-02-17 17:48:59 +0100620 poll_tmout_ms = INT_MAX;
621 if (o_wait < INT_MAX / 1000)
622 poll_tmout_ms = o_wait * 1000;
623 }
624 rr = poll(pfds, 2, poll_tmout_ms);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000625 if (rr < 0 && errno != EINTR) { /* might have gotten ^Zed, etc */
Denys Vlasenko94dcfd82017-02-17 17:48:59 +0100626 holler_perror("poll");
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000627 close(netfd);
628 return 1;
629 }
630 /* if we have a timeout AND stdin is closed AND we haven't heard anything
631 from the net during that time, assume it's dead and close it too. */
632 if (rr == 0) {
Denys Vlasenko94dcfd82017-02-17 17:48:59 +0100633 if (!pfds[0].revents) {
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000634 netretry--; /* we actually try a coupla times. */
Denys Vlasenkoffeeb7a2013-07-28 22:23:12 +0200635 if (!netretry) {
636 if (o_verbose > 1) /* normally we don't care */
637 fprintf(stderr, "net timeout\n");
Denys Vlasenkofd3ac962013-07-28 22:24:51 +0200638 /*close(netfd); - redundant, exit will do it */
Denys Vlasenkoffeeb7a2013-07-28 22:23:12 +0200639 return 0; /* not an error! */
640 }
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000641 }
Denys Vlasenko94dcfd82017-02-17 17:48:59 +0100642 } /* timeout */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000643
644 /* Ding!! Something arrived, go check all the incoming hoppers, net first */
Denys Vlasenko94dcfd82017-02-17 17:48:59 +0100645 if (pfds[1].revents) { /* net: ding! */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000646 rr = read(netfd, bigbuf_net, BIGSIZ);
647 if (rr <= 0) {
Denis Vlasenko19507f02007-04-06 10:41:05 +0000648 if (rr < 0 && o_verbose > 1) {
649 /* nc 1.10 doesn't do this */
James Byrne69374872019-07-02 11:35:03 +0200650 bb_simple_perror_msg("net read");
Denis Vlasenko19507f02007-04-06 10:41:05 +0000651 }
Denys Vlasenko94dcfd82017-02-17 17:48:59 +0100652 pfds[1].fd = -1; /* don't poll for netfd anymore */
Denys Vlasenkoffeeb7a2013-07-28 22:23:12 +0200653 fds_open--;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000654 rzleft = 0; /* can't write anymore: broken pipe */
655 } else {
656 rnleft = rr;
657 np = bigbuf_net;
658 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000659Debug("got %d from the net, errno %d", rr, errno);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000660 } /* net:ding */
661
662 /* if we're in "slowly" mode there's probably still stuff in the stdin
663 buffer, so don't read unless we really need MORE INPUT! MORE INPUT! */
664 if (rzleft)
665 goto shovel;
666
667 /* okay, suck more stdin */
Denys Vlasenko94dcfd82017-02-17 17:48:59 +0100668 if (pfds[0].revents) { /* stdin: ding! */
Bernhard Reutner-Fischer5e25ddb2008-05-19 09:48:17 +0000669 rr = read(STDIN_FILENO, bigbuf_in, BIGSIZ);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000670 /* Considered making reads here smaller for UDP mode, but 8192-byte
671 mobygrams are kinda fun and exercise the reassembler. */
672 if (rr <= 0) { /* at end, or fukt, or ... */
Denys Vlasenko94dcfd82017-02-17 17:48:59 +0100673 pfds[0].fd = -1; /* disable stdin */
Denys Vlasenkofd3ac962013-07-28 22:24:51 +0200674 /*close(STDIN_FILENO); - not really necessary */
Denys Vlasenkoffeeb7a2013-07-28 22:23:12 +0200675 /* Let peer know we have no more data */
676 /* nc 1.10 doesn't do this: */
677 shutdown(netfd, SHUT_WR);
678 fds_open--;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000679 } else {
680 rzleft = rr;
681 zp = bigbuf_in;
682 }
683 } /* stdin:ding */
684 shovel:
685 /* now that we've dingdonged all our thingdings, send off the results.
686 Geez, why does this look an awful lot like the big loop in "rsh"? ...
687 not sure if the order of this matters, but write net -> stdout first. */
688
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000689 if (rnleft) {
Bernhard Reutner-Fischer5e25ddb2008-05-19 09:48:17 +0000690 rr = write(STDOUT_FILENO, np, rnleft);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000691 if (rr > 0) {
Denis Vlasenkoc4f12f52008-05-12 14:35:56 +0000692 if (o_ofile) /* log the stdout */
693 oprint('<', (unsigned char *)np, rr);
Denys Vlasenko05965cb2013-07-29 14:18:32 +0200694 np += rr;
Denys Vlasenkofd3ac962013-07-28 22:24:51 +0200695 rnleft -= rr;
Denys Vlasenko05965cb2013-07-29 14:18:32 +0200696 wrote_out += rr; /* global count */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000697 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000698Debug("wrote %d to stdout, errno %d", rr, errno);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000699 } /* rnleft */
700 if (rzleft) {
701 if (o_interval) /* in "slowly" mode ?? */
702 rr = findline(zp, rzleft);
703 else
704 rr = rzleft;
705 rr = write(netfd, zp, rr); /* one line, or the whole buffer */
706 if (rr > 0) {
Denis Vlasenkoc4f12f52008-05-12 14:35:56 +0000707 if (o_ofile) /* log what got sent */
708 oprint('>', (unsigned char *)zp, rr);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000709 zp += rr;
710 rzleft -= rr;
Denys Vlasenko05965cb2013-07-29 14:18:32 +0200711 wrote_net += rr; /* global count */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000712 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000713Debug("wrote %d to net, errno %d", rr, errno);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000714 } /* rzleft */
715 if (o_interval) { /* cycle between slow lines, or ... */
716 sleep(o_interval);
Denys Vlasenko94dcfd82017-02-17 17:48:59 +0100717 continue; /* ...with hairy loop... */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000718 }
Denys Vlasenkoffeeb7a2013-07-28 22:23:12 +0200719 if (rzleft || rnleft) { /* shovel that shit till they ain't */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000720 wretry--; /* none left, and get another load */
Denys Vlasenkofd3ac962013-07-28 22:24:51 +0200721 /* net write retries sometimes happen on UDP connections */
722 if (!wretry) { /* is something hung? */
723 holler_error("too many output retries");
724 return 1;
725 }
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000726 goto shovel;
727 }
Denys Vlasenko05965cb2013-07-29 14:18:32 +0200728 } /* while (fds_open) */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000729
730 /* XXX: maybe want a more graceful shutdown() here, or screw around with
731 linger times?? I suspect that I don't need to since I'm always doing
732 blocking reads and writes and my own manual "last ditch" efforts to read
733 the net again after a timeout. I haven't seen any screwups yet, but it's
734 not like my test network is particularly busy... */
735 close(netfd);
736 return 0;
737} /* readwrite */
738
739/* main: now we pull it all together... */
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000740int nc_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denys Vlasenko2ec91ae2010-01-04 14:15:38 +0100741int nc_main(int argc UNUSED_PARAM, char **argv)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000742{
Denis Vlasenko1d426652008-03-17 09:09:09 +0000743 char *str_p, *str_s;
Denis Vlasenko5e34ff22009-04-21 11:09:40 +0000744 IF_NC_EXTRA(char *str_i, *str_o;)
Denys Vlasenko5e896482012-03-19 01:17:36 +0100745 char *themdotted = themdotted; /* for compiler */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000746 char **proggie;
747 int x;
Denys Vlasenkode6f1482013-02-28 12:20:06 +0100748 unsigned cnt_l = 0;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000749 unsigned o_lport = 0;
750
Denis Vlasenko574f2f42008-02-27 18:41:59 +0000751 INIT_G();
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000752
753 /* catch a signal or two for cleanup */
Denis Vlasenko25591c32008-02-16 22:58:56 +0000754 bb_signals(0
755 + (1 << SIGINT)
756 + (1 << SIGQUIT)
757 + (1 << SIGTERM)
758 , catch);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000759 /* and suppress others... */
Denis Vlasenko25591c32008-02-16 22:58:56 +0000760 bb_signals(0
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000761#ifdef SIGURG
Denis Vlasenko25591c32008-02-16 22:58:56 +0000762 + (1 << SIGURG)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000763#endif
Denis Vlasenko25591c32008-02-16 22:58:56 +0000764 + (1 << SIGPIPE) /* important! */
765 , SIG_IGN);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000766
767 proggie = argv;
768 while (*++proggie) {
769 if (strcmp(*proggie, "-e") == 0) {
770 *proggie = NULL;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000771 proggie++;
772 goto e_found;
773 }
Denys Vlasenko5e896482012-03-19 01:17:36 +0100774 /* -<other_opts>e PROG [ARGS] ? */
775 /* (aboriginal linux uses this form) */
776 if (proggie[0][0] == '-') {
777 char *optpos = *proggie + 1;
778 /* Skip all valid opts w/o params */
Denys Vlasenkode6f1482013-02-28 12:20:06 +0100779 optpos = optpos + strspn(optpos, "nuv"IF_NC_SERVER("lk")IF_NC_EXTRA("z"));
Denys Vlasenko5e896482012-03-19 01:17:36 +0100780 if (*optpos == 'e' && !optpos[1]) {
781 *optpos = '\0';
782 proggie++;
783 G.proggie0saved = *proggie;
784 *proggie = NULL; /* terminate argv for getopt32 */
785 goto e_found;
786 }
787 }
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000788 }
789 proggie = NULL;
790 e_found:
Denis Vlasenko4b924f32007-05-30 00:29:55 +0000791
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000792 // -g -G -t -r deleted, unimplemented -a deleted too
Denys Vlasenko22542ec2017-08-08 21:55:02 +0200793 getopt32(argv, "^"
Norbert Langea16c8ef2020-02-14 22:13:38 +0100794 "np:s:ubvw:+"/* -w N */ IF_NC_SERVER("lk")
Denys Vlasenko22542ec2017-08-08 21:55:02 +0200795 IF_NC_EXTRA("i:o:z")
796 "\0"
Denys Vlasenko3f91e662018-05-24 16:38:40 +0200797 "?2:vv"IF_NC_SERVER(":ll"), /* max 2 params; -v and -l are counters */
Denys Vlasenko22542ec2017-08-08 21:55:02 +0200798 &str_p, &str_s, &o_wait
799 IF_NC_EXTRA(, &str_i, &str_o)
800 , &o_verbose IF_NC_SERVER(, &cnt_l)
801 );
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000802 argv += optind;
803#if ENABLE_NC_EXTRA
804 if (option_mask32 & OPT_i) /* line-interval time */
805 o_interval = xatou_range(str_i, 1, 0xffff);
806#endif
Denys Vlasenkode6f1482013-02-28 12:20:06 +0100807#if ENABLE_NC_SERVER
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000808 //if (option_mask32 & OPT_l) /* listen mode */
Denys Vlasenkode6f1482013-02-28 12:20:06 +0100809 if (option_mask32 & OPT_k) /* persistent server mode */
810 cnt_l = 2;
811#endif
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000812 //if (option_mask32 & OPT_n) /* numeric-only, no DNS lookups */
813 //if (option_mask32 & OPT_o) /* hexdump log */
814 if (option_mask32 & OPT_p) { /* local source port */
815 o_lport = bb_lookup_port(str_p, o_udpmode ? "udp" : "tcp", 0);
816 if (!o_lport)
817 bb_error_msg_and_die("bad local port '%s'", str_p);
818 }
819 //if (option_mask32 & OPT_r) /* randomize various things */
820 //if (option_mask32 & OPT_u) /* use UDP */
821 //if (option_mask32 & OPT_v) /* verbose */
Denis Vlasenko1d426652008-03-17 09:09:09 +0000822 //if (option_mask32 & OPT_w) /* wait time */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000823 //if (option_mask32 & OPT_z) /* little or no data xfer */
824
Denis Vlasenko19507f02007-04-06 10:41:05 +0000825 /* We manage our fd's so that they are never 0,1,2 */
826 /*bb_sanitize_stdio(); - not needed */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000827
Denis Vlasenko5c51a7c2007-06-05 20:08:11 +0000828 if (argv[0]) {
829 themaddr = xhost2sockaddr(argv[0],
830 argv[1]
831 ? bb_lookup_port(argv[1], o_udpmode ? "udp" : "tcp", 0)
832 : 0);
833 }
834
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000835 /* create & bind network socket */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000836 x = (o_udpmode ? SOCK_DGRAM : SOCK_STREAM);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000837 if (option_mask32 & OPT_s) { /* local address */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000838 /* if o_lport is still 0, then we will use random port */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000839 ouraddr = xhost2sockaddr(str_s, o_lport);
Denis Vlasenko4cf1d082008-03-12 23:13:50 +0000840#ifdef BLOAT
841 /* prevent spurious "UDP listen needs !0 port" */
842 o_lport = get_nport(ouraddr);
843 o_lport = ntohs(o_lport);
844#endif
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000845 x = xsocket(ouraddr->u.sa.sa_family, x, 0);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000846 } else {
Denis Vlasenko5c51a7c2007-06-05 20:08:11 +0000847 /* We try IPv6, then IPv4, unless addr family is
848 * implicitly set by way of remote addr/port spec */
849 x = xsocket_type(&ouraddr,
Denis Vlasenko4e6d5112008-03-12 22:14:34 +0000850 (themaddr ? themaddr->u.sa.sa_family : AF_UNSPEC),
Denis Vlasenko5c51a7c2007-06-05 20:08:11 +0000851 x);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000852 if (o_lport)
Denys Vlasenkoca183112011-04-07 17:52:20 +0200853 set_nport(&ouraddr->u.sa, htons(o_lport));
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000854 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000855 xmove_fd(x, netfd);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000856 setsockopt_reuseaddr(netfd);
Norbert Langea16c8ef2020-02-14 22:13:38 +0100857 if (o_udpmode) {
858 if (o_bcmode)
859 setsockopt_broadcast(netfd);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000860 socket_want_pktinfo(netfd);
Norbert Langea16c8ef2020-02-14 22:13:38 +0100861 }
Denis Vlasenkof6b46852009-04-25 13:16:53 +0000862 if (!ENABLE_FEATURE_UNIX_LOCAL
Denys Vlasenkode6f1482013-02-28 12:20:06 +0100863 || cnt_l != 0 /* listen */
Denis Vlasenkof6b46852009-04-25 13:16:53 +0000864 || ouraddr->u.sa.sa_family != AF_UNIX
865 ) {
866 xbind(netfd, &ouraddr->u.sa, ouraddr->len);
867 }
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000868#if 0
Denys Vlasenkoc52cbea2015-08-24 19:48:03 +0200869 setsockopt_SOL_SOCKET_int(netfd, SO_RCVBUF, o_rcvbuf);
870 setsockopt_SOL_SOCKET_int(netfd, SO_SNDBUF, o_sndbuf);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000871#endif
872
Denis Vlasenko4cf1d082008-03-12 23:13:50 +0000873#ifdef BLOAT
Denis Vlasenko19507f02007-04-06 10:41:05 +0000874 if (OPT_l && (option_mask32 & (OPT_u|OPT_l)) == (OPT_u|OPT_l)) {
875 /* apparently UDP can listen ON "port 0",
876 but that's not useful */
877 if (!o_lport)
James Byrne69374872019-07-02 11:35:03 +0200878 bb_simple_error_msg_and_die("UDP listen needs nonzero -p port");
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000879 }
Denis Vlasenko4cf1d082008-03-12 23:13:50 +0000880#endif
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000881
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000882 if (proggie) {
Denys Vlasenkoffeeb7a2013-07-28 22:23:12 +0200883 close(STDIN_FILENO); /* won't need stdin */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000884 option_mask32 &= ~OPT_o; /* -o with -e is meaningless! */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000885 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000886#if ENABLE_NC_EXTRA
887 if (o_ofile)
888 xmove_fd(xopen(str_o, O_WRONLY|O_CREAT|O_TRUNC), ofd);
889#endif
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000890
Denys Vlasenkode6f1482013-02-28 12:20:06 +0100891 if (cnt_l != 0) {
892 dolisten((cnt_l - 1), proggie);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000893 /* dolisten does its own connect reporting */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000894 x = readwrite(); /* it even works with UDP! */
895 } else {
896 /* Outbound connects. Now we're more picky about args... */
897 if (!themaddr)
Denys Vlasenkob103fb12010-09-07 18:41:56 +0200898 bb_show_usage();
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000899
900 remend = *themaddr;
901 if (o_verbose)
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000902 themdotted = xmalloc_sockaddr2dotted(&themaddr->u.sa);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000903
904 x = connect_w_timeout(netfd);
905 if (o_zero && x == 0 && o_udpmode) /* if UDP scanning... */
906 x = udptest();
907 if (x == 0) { /* Yow, are we OPEN YET?! */
908 if (o_verbose)
Denis Vlasenko19507f02007-04-06 10:41:05 +0000909 fprintf(stderr, "%s (%s) open\n", argv[0], themdotted);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000910 if (proggie) /* exec is valid for outbound, too */
911 doexec(proggie);
912 if (!o_zero)
913 x = readwrite();
914 } else { /* connect or udptest wasn't successful */
915 x = 1; /* exit status */
916 /* if we're scanning at a "one -v" verbosity level, don't print refusals.
917 Give it another -v if you want to see everything. */
918 if (o_verbose > 1 || (o_verbose && errno != ECONNREFUSED))
Denis Vlasenko19507f02007-04-06 10:41:05 +0000919 bb_perror_msg("%s (%s)", argv[0], themdotted);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000920 }
921 }
922 if (o_verbose > 1) /* normally we don't care */
923 fprintf(stderr, SENT_N_RECV_M, wrote_net, wrote_out);
924 return x;
925}