1 /* Based on netcat 1.10 RELEASE 960320 written by hobbit@avian.org.
2  * Released into public domain by the author.
3  *
4  * Copyright (C) 2007 Denys Vlasenko.
5  *
6  * Licensed under GPLv2, see file LICENSE in this source tree.
7  */
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:
39  * - PROG in '-e PROG' can have ARGS (and options).
40  *   Because of this -e option must be last.
41 //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!)
44  * - 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)
50  *   are closed, but we don't exit - we continue to listen/accept.
51  * 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.
57  */
58 
59 /* done in nc.c: #include "libbb.h" */
60 
61 //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"
66 //usage:       "nc [OPTIONS] -l -p PORT [HOST] [PORT]  - listen"
67 //usage:	)
68 //usage:#define nc_full_usage "\n\n"
69 //usage:       "	-e PROG	Run PROG after connect (must be last)"
70 //usage:	IF_NC_SERVER(
71 //usage:     "\n	-l	Listen mode, for inbound connects"
72 //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  */
78 //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"
87 //usage:     "\n	-b	Allow broadcasts"
88 //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 
104 enum {
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 };
111 
112 struct globals {
113 	/* 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 */
122 #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
127 #define SENT_N_RECV_M "sent %u, rcvd %u\n"
128 	unsigned wrote_out;          /* total stdout bytes */
129 	unsigned wrote_net;          /* total net bytes */
130 #endif
131 	char *proggie0saved;
132 	/* ouraddr is never NULL and goes through three states as we progress:
133 	 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 
142 	jmp_buf jbuf;                /* timer crud */
143 
144 	char bigbuf_in[BIGSIZ];      /* data buffers */
145 	char bigbuf_net[BIGSIZ];
146 };
147 
148 #define G (*ptr_to_globals)
149 #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      )
155 #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
164 #define INIT_G() do { \
165 	SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
166 } while (0)
167 
168 
169 /* Must match getopt32 call! */
170 enum {
171 	OPT_n = (1 << 0),
172 	OPT_p = (1 << 1),
173 	OPT_s = (1 << 2),
174 	OPT_u = (1 << 3),
175 	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,
183 };
184 
185 #define o_nflag   (option_mask32 & OPT_n)
186 #define o_udpmode (option_mask32 & OPT_u)
187 #define o_bcmode  (option_mask32 & OPT_b)
188 #if ENABLE_NC_EXTRA
189 #define o_ofile   (option_mask32 & OPT_o)
190 #define o_zero    (option_mask32 & OPT_z)
191 #else
192 #define o_ofile   0
193 #define o_zero    0
194 #endif
195 
196 /* Debug: squirt whatever message and sleep a bit so we can see it go by. */
197 /* Beware: writes to stdOUT... */
198 #if 0
199 #define Debug(...) do { printf(__VA_ARGS__); printf("\n"); fflush_all(); sleep1(); } while (0)
200 #else
201 #define Debug(...) do { } while (0)
202 #endif
203 
204 #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)
206 
207 /* catch: no-brainer interrupt handler */
catch(int sig)208 static void catch(int sig)
209 {
210 	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");
213 	kill_myself_with_sig(sig);
214 }
215 
216 /* unarm  */
unarm(void)217 static void unarm(void)
218 {
219 	signal(SIGALRM, SIG_IGN);
220 	alarm(0);
221 }
222 
223 /* timeout and other signal handling cruft */
tmtravel(int sig UNUSED_PARAM)224 static void tmtravel(int sig UNUSED_PARAM)
225 {
226 	unarm();
227 	longjmp(jbuf, 1);
228 }
229 
230 /* arm: set the timer.  */
arm(unsigned secs)231 static void arm(unsigned secs)
232 {
233 	signal(SIGALRM, tmtravel);
234 	alarm(secs);
235 }
236 
237 /* 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... */
findline(char * buf,unsigned siz)241 static 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! */
254 Debug("findline returning %d", x);
255 			return x;
256 		}
257 		p++;
258 	} /* for */
259 Debug("findline returning whole thing: %d", siz);
260 	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!! */
269 static int doexec(char **proggie) NORETURN;
doexec(char ** proggie)270 static int doexec(char **proggie)
271 {
272 	if (G.proggie0saved)
273 		proggie[0] = G.proggie0saved;
274 	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 */
278 	BB_EXECVP_or_die(proggie);
279 }
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] */
connect_w_timeout(int fd)287 static 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) {
294 		rr = connect(fd, &themaddr->u.sa, themaddr->len);
295 		unarm();
296 	} else { /* setjmp: connect failed... */
297 		rr = -1;
298 		errno = ETIMEDOUT; /* fake it */
299 	}
300 	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... */
dolisten(int is_persistent,char ** proggie)308 static void dolisten(int is_persistent, char **proggie)
309 {
310 	int rr;
311 
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;
326 		getsockname(netfd, &ouraddr->u.sa, &ouraddr->len);
327 		//if (rr < 0)
328 		//	bb_perror_msg_and_die("getsockname after bind");
329 		addr = xmalloc_sockaddr2dotted(&ouraddr->u.sa);
330 		fprintf(stderr, "listening on %s ...\n", addr);
331 		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 */
354 		remend.len = LSA_SIZEOF_SA;
355 		if (themaddr) {
356 			remend = *themaddr;
357 			xconnect(netfd, &themaddr->u.sa, themaddr->len);
358 		}
359 		/* 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*/
365 				&remend.u.sa, &ouraddr->u.sa, ouraddr->len);
366 			if (rr < 0)
367 				bb_simple_perror_msg_and_die("recvfrom");
368 			unarm();
369 		} else
370 			bb_simple_error_msg_and_die("timeout");
371 /* Now we learned *to which IP* peer has connected, and we want to anchor
372 our socket on it, so that our outbound packets will have correct local IP.
373 Unfortunately, bind() on already bound socket will fail now (EINVAL):
374 	xbind(netfd, &ouraddr->u.sa, ouraddr->len);
375 Need to read the packet, save data, close this socket and
376 create new one, and bind() it. TODO */
377 		if (!themaddr)
378 			xconnect(netfd, &remend.u.sa, ouraddr->len);
379 	} else {
380 		/* TCP */
381  another:
382 		arm(o_wait); /* wrap this in a timer, too; 0 = forever */
383 		if (setjmp(jbuf) == 0) {
384  again:
385 			remend.len = LSA_SIZEOF_SA;
386 			rr = accept(netfd, &remend.u.sa, &remend.len);
387 			if (rr < 0)
388 				bb_simple_perror_msg_and_die("accept");
389 			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 */
397 					set_nport(&remend.u.sa, 0); /* blot out remote port# */
398 				}
399 				r = memcmp(&remend.u.sa, &themaddr->u.sa, remend.len);
400 				set_nport(&remend.u.sa, sv_port); /* restore */
401 				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 				}
412 			}
413 			unarm();
414 		} else
415 			bb_simple_error_msg_and_die("timeout");
416 
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 
429 		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. */
434 		getsockname(netfd, &ouraddr->u.sa, &ouraddr->len);
435 		//if (rr < 0)
436 		//	bb_perror_msg_and_die("getsockname after accept");
437 	}
438 
439 	if (o_verbose) {
440 		char *lcladdr, *remaddr, *remhostname;
441 
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. */
448 		char optbuf[40];
449 		socklen_t x = sizeof(optbuf);
450 
451 		rr = getsockopt(netfd, IPPROTO_IP, IP_OPTIONS, optbuf, &x);
452 		if (rr >= 0 && x) {    /* we've got options, lessee em... */
453 			*bin2hex(bigbuf_net, optbuf, x) = '\0';
454 			fprintf(stderr, "IP options: %s\n", bigbuf_net);
455 		}
456 #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
465 	 accept the connection and then reject undesirable ones by closing.
466 	 In other words, we need a TCP MSG_PEEK. */
467 	/* bbox: removed most of it */
468 		lcladdr = xmalloc_sockaddr2dotted(&ouraddr->u.sa);
469 		remaddr = xmalloc_sockaddr2dotted(&remend.u.sa);
470 		remhostname = o_nflag ? remaddr : xmalloc_sockaddr2host(&remend.u.sa);
471 		fprintf(stderr, "connect to %s from %s (%s)\n",
472 				lcladdr, remhostname, remaddr);
473 		free(lcladdr);
474 		free(remaddr);
475 		if (!o_nflag)
476 			free(remhostname);
477 	}
478 
479 	if (proggie)
480 		doexec(proggie);
481 }
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. */
493 #if ENABLE_NC_EXTRA
udptest(void)494 static int udptest(void)
495 {
496 	int rr;
497 
498 	rr = write(netfd, bigbuf_in, 1);
499 	if (rr != 1)
500 		bb_simple_perror_msg("udptest first write");
501 
502 	if (o_wait)
503 		sleep(o_wait); // can be interrupted! while (t) nanosleep(&t)?
504 	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. */
508 	/* Set a temporary connect timeout, so packet filtration doesn't cause
509 	 us to hang forever, and hit it */
510 		o_wait = 5;                     /* enough that we'll notice?? */
511 		rr = xsocket(ouraddr->u.sa.sa_family, SOCK_STREAM, 0);
512 		set_nport(&themaddr->u.sa, htons(SLEAZE_PORT));
513 		connect_w_timeout(rr);
514 		/* don't need to restore themaddr's port, it's not used anymore */
515 		close(rr);
516 		o_wait = 0; /* restore */
517 	}
518 
519 	rr = write(netfd, bigbuf_in, 1);
520 	return (rr != 1); /* if rr == 1, return 0 (success) */
521 }
522 #else
523 int udptest(void);
524 #endif
525 
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... */
536 #if ENABLE_NC_EXTRA
oprint(int direction,unsigned char * p,unsigned bc)537 static void oprint(int direction, unsigned char *p, unsigned bc)
538 {
539 	unsigned obc;           /* current "global" offset */
540 	unsigned x;
541 	unsigned char *op;      /* out hexdump ptr */
542 	unsigned char *ap;      /* out asc-dump ptr */
543 	unsigned char stage[100];
544 
545 	if (bc == 0)
546 		return;
547 
548 	obc = wrote_net; /* use the globals! */
549 	if (direction == '<')
550 		obc = wrote_out;
551 	stage[0] = direction;
552 	stage[59] = '#'; /* preload separator */
553 	stage[60] = ' ';
554 
555 	do {    /* for chunk-o-data ... */
556 		x = 16;
557 		if (bc < 16) {
558 			/* memset(&stage[bc*3 + 11], ' ', 16*3 - bc*3); */
559 			memset(&stage[11], ' ', 16*3);
560 			x = bc;
561 		}
562 		sprintf((char *)&stage[1], " %8.8x ", obc);  /* xxx: still slow? */
563 		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 */
567 
568 		do {  /* for line of dump, however long ... */
569 			*op++ = 0x20 | bb_hexdigits_upcase[*p >> 4];
570 			*op++ = 0x20 | bb_hexdigits_upcase[*p & 0x0f];
571 			*op++ = ' ';
572 			if ((*p > 31) && (*p < 127))
573 				*ap = *p;   /* printing */
574 			else
575 				*ap = '.';  /* nonprinting, loose def */
576 			ap++;
577 			p++;
578 		} while (--x);
579 		*ap++ = '\n';  /* finish the line */
580 		xwrite(ofd, stage, ap - stage);
581 	} while (bc);
582 }
583 #else
584 void oprint(int direction, unsigned char *p, unsigned bc);
585 #endif
586 
587 /* readwrite:
588  handle stdin/stdout/network I/O.  Bwahaha!! -- the i/o loop from hell.
589  In this instance, return what might become our exit status. */
readwrite(void)590 static int readwrite(void)
591 {
592 	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 */
597 	unsigned fds_open;
598 
599 	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;
604 
605 	fds_open = 2;
606 	netretry = 2;
607 	rzleft = rnleft = 0;
608 	if (o_interval)
609 		sleep(o_interval);                /* pause *before* sending stuff, too */
610 
611 	/* and now the big ol' shoveling loop ... */
612 	/* nc 1.10 has "while (FD_ISSET(netfd)" here */
613 	while (fds_open) {
614 		int rr;
615 		int poll_tmout_ms;
616 		unsigned wretry = 8200;               /* net-write sanity counter */
617 
618 		poll_tmout_ms = -1;
619 		if (o_wait) {
620 			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);
625 		if (rr < 0 && errno != EINTR) {                /* might have gotten ^Zed, etc */
626 			holler_perror("poll");
627 			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) {
633 			if (!pfds[0].revents) {
634 				netretry--;                        /* we actually try a coupla times. */
635 				if (!netretry) {
636 					if (o_verbose > 1)         /* normally we don't care */
637 						fprintf(stderr, "net timeout\n");
638 					/*close(netfd); - redundant, exit will do it */
639 					return 0;                  /* not an error! */
640 				}
641 			}
642 		} /* timeout */
643 
644 	/* Ding!!  Something arrived, go check all the incoming hoppers, net first */
645 		if (pfds[1].revents) {                /* net: ding! */
646 			rr = read(netfd, bigbuf_net, BIGSIZ);
647 			if (rr <= 0) {
648 				if (rr < 0 && o_verbose > 1) {
649 					/* nc 1.10 doesn't do this */
650 					bb_simple_perror_msg("net read");
651 				}
652 				pfds[1].fd = -1;                   /* don't poll for netfd anymore */
653 				fds_open--;
654 				rzleft = 0;                        /* can't write anymore: broken pipe */
655 			} else {
656 				rnleft = rr;
657 				np = bigbuf_net;
658 			}
659 Debug("got %d from the net, errno %d", rr, errno);
660 		} /* 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 */
668 		if (pfds[0].revents) {                /* stdin: ding! */
669 			rr = read(STDIN_FILENO, bigbuf_in, BIGSIZ);
670 	/* 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 ... */
673 				pfds[0].fd = -1;              /* disable stdin */
674 				/*close(STDIN_FILENO); - not really necessary */
675 				/* Let peer know we have no more data */
676 				/* nc 1.10 doesn't do this: */
677 				shutdown(netfd, SHUT_WR);
678 				fds_open--;
679 			} 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 
689 		if (rnleft) {
690 			rr = write(STDOUT_FILENO, np, rnleft);
691 			if (rr > 0) {
692 				if (o_ofile) /* log the stdout */
693 					oprint('<', (unsigned char *)np, rr);
694 				np += rr;
695 				rnleft -= rr;
696 				wrote_out += rr; /* global count */
697 			}
698 Debug("wrote %d to stdout, errno %d", rr, errno);
699 		} /* 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) {
707 				if (o_ofile) /* log what got sent */
708 					oprint('>', (unsigned char *)zp, rr);
709 				zp += rr;
710 				rzleft -= rr;
711 				wrote_net += rr; /* global count */
712 			}
713 Debug("wrote %d to net, errno %d", rr, errno);
714 		} /* rzleft */
715 		if (o_interval) {                        /* cycle between slow lines, or ... */
716 			sleep(o_interval);
717 			continue;                        /* ...with hairy loop... */
718 		}
719 		if (rzleft || rnleft) {                  /* shovel that shit till they ain't */
720 			wretry--;                        /* none left, and get another load */
721 	/* 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 			}
726 			goto shovel;
727 		}
728 	} /* while (fds_open) */
729 
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... */
740 int nc_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
nc_main(int argc UNUSED_PARAM,char ** argv)741 int nc_main(int argc UNUSED_PARAM, char **argv)
742 {
743 	char *str_p, *str_s;
744 	IF_NC_EXTRA(char *str_i, *str_o;)
745 	char *themdotted = themdotted; /* for compiler */
746 	char **proggie;
747 	int x;
748 	unsigned cnt_l = 0;
749 	unsigned o_lport = 0;
750 
751 	INIT_G();
752 
753 	/* catch a signal or two for cleanup */
754 	bb_signals(0
755 		+ (1 << SIGINT)
756 		+ (1 << SIGQUIT)
757 		+ (1 << SIGTERM)
758 		, catch);
759 	/* and suppress others... */
760 	bb_signals(0
761 #ifdef SIGURG
762 		+ (1 << SIGURG)
763 #endif
764 		+ (1 << SIGPIPE) /* important! */
765 		, SIG_IGN);
766 
767 	proggie = argv;
768 	while (*++proggie) {
769 		if (strcmp(*proggie, "-e") == 0) {
770 			*proggie = NULL;
771 			proggie++;
772 			goto e_found;
773 		}
774 		/* -<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 */
779 			optpos = optpos + strspn(optpos, "nuv"IF_NC_SERVER("lk")IF_NC_EXTRA("z"));
780 			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 		}
788 	}
789 	proggie = NULL;
790  e_found:
791 
792 	// -g -G -t -r deleted, unimplemented -a deleted too
793 	getopt32(argv, "^"
794 		"np:s:ubvw:+"/* -w N */ IF_NC_SERVER("lk")
795 		IF_NC_EXTRA("i:o:z")
796 			"\0"
797 			"?2:vv"IF_NC_SERVER(":ll"), /* max 2 params; -v and -l are counters */
798 		&str_p, &str_s, &o_wait
799 		IF_NC_EXTRA(, &str_i, &str_o)
800 			, &o_verbose IF_NC_SERVER(, &cnt_l)
801 	);
802 	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
807 #if ENABLE_NC_SERVER
808 	//if (option_mask32 & OPT_l) /* listen mode */
809 	if (option_mask32 & OPT_k) /* persistent server mode */
810 		cnt_l = 2;
811 #endif
812 	//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 	}
817 	//if (option_mask32 & OPT_r) /* randomize various things */
818 	//if (option_mask32 & OPT_u) /* use UDP */
819 	//if (option_mask32 & OPT_v) /* verbose */
820 	//if (option_mask32 & OPT_w) /* wait time */
821 	//if (option_mask32 & OPT_z) /* little or no data xfer */
822 
823 	/* We manage our fd's so that they are never 0,1,2 */
824 	/*bb_sanitize_stdio(); - not needed */
825 
826 	if (argv[0]) {
827 		themaddr = xhost2sockaddr(argv[0],
828 			bb_lookup_port(argv[1], o_udpmode ? "udp" : "tcp", 0)
829 		);
830 	}
831 
832 	/* create & bind network socket */
833 	x = (o_udpmode ? SOCK_DGRAM : SOCK_STREAM);
834 	if (option_mask32 & OPT_s) { /* local address */
835 		/* if o_lport is still 0, then we will use random port */
836 		ouraddr = xhost2sockaddr(str_s, o_lport);
837 #ifdef BLOAT
838 		/* prevent spurious "UDP listen needs !0 port" */
839 		o_lport = get_nport(ouraddr);
840 		o_lport = ntohs(o_lport);
841 #endif
842 		x = xsocket(ouraddr->u.sa.sa_family, x, 0);
843 	} else {
844 		/* We try IPv6, then IPv4, unless addr family is
845 		 * implicitly set by way of remote addr/port spec */
846 		x = xsocket_type(&ouraddr,
847 				(themaddr ? themaddr->u.sa.sa_family : AF_UNSPEC),
848 				x);
849 		if (o_lport)
850 			set_nport(&ouraddr->u.sa, htons(o_lport));
851 	}
852 	xmove_fd(x, netfd);
853 	setsockopt_reuseaddr(netfd);
854 	if (o_udpmode) {
855 		if (o_bcmode)
856 			setsockopt_broadcast(netfd);
857 		socket_want_pktinfo(netfd);
858 	}
859 	if (!ENABLE_FEATURE_UNIX_LOCAL
860 	 || cnt_l != 0 /* listen */
861 	 || ouraddr->u.sa.sa_family != AF_UNIX
862 	) {
863 		xbind(netfd, &ouraddr->u.sa, ouraddr->len);
864 	}
865 #if 0
866 	setsockopt_SOL_SOCKET_int(netfd, SO_RCVBUF, o_rcvbuf);
867 	setsockopt_SOL_SOCKET_int(netfd, SO_SNDBUF, o_sndbuf);
868 #endif
869 
870 #ifdef BLOAT
871 	if (OPT_l && (option_mask32 & (OPT_u|OPT_l)) == (OPT_u|OPT_l)) {
872 		/* apparently UDP can listen ON "port 0",
873 		 but that's not useful */
874 		if (!o_lport)
875 			bb_simple_error_msg_and_die("UDP listen needs nonzero -p port");
876 	}
877 #endif
878 
879 	if (proggie) {
880 		close(STDIN_FILENO); /* won't need stdin */
881 		option_mask32 &= ~OPT_o; /* -o with -e is meaningless! */
882 	}
883 #if ENABLE_NC_EXTRA
884 	if (o_ofile)
885 		xmove_fd(xopen(str_o, O_WRONLY|O_CREAT|O_TRUNC), ofd);
886 #endif
887 
888 	if (cnt_l != 0) {
889 		dolisten((cnt_l - 1), proggie);
890 		/* dolisten does its own connect reporting */
891 		x = readwrite(); /* it even works with UDP! */
892 	} else {
893 		/* Outbound connects.  Now we're more picky about args... */
894 		if (!themaddr)
895 			bb_show_usage();
896 
897 		remend = *themaddr;
898 		if (o_verbose)
899 			themdotted = xmalloc_sockaddr2dotted(&themaddr->u.sa);
900 
901 		x = connect_w_timeout(netfd);
902 		if (o_zero && x == 0 && o_udpmode)        /* if UDP scanning... */
903 			x = udptest();
904 		if (x == 0) {                        /* Yow, are we OPEN YET?! */
905 			if (o_verbose)
906 				fprintf(stderr, "%s (%s) open\n", argv[0], themdotted);
907 			if (proggie)                        /* exec is valid for outbound, too */
908 				doexec(proggie);
909 			if (!o_zero)
910 				x = readwrite();
911 		} else { /* connect or udptest wasn't successful */
912 			x = 1;                                /* exit status */
913 			/* if we're scanning at a "one -v" verbosity level, don't print refusals.
914 			 Give it another -v if you want to see everything. */
915 			if (o_verbose > 1 || (o_verbose && errno != ECONNREFUSED))
916 				bb_perror_msg("%s (%s)", argv[0], themdotted);
917 		}
918 	}
919 	if (o_verbose > 1)                /* normally we don't care */
920 		fprintf(stderr, SENT_N_RECV_M, wrote_net, wrote_out);
921 	return x;
922 }
923