blob: d184f689b57479f26e8ca33e5cfc0fc4475597fc [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.
Denis Vlasenko29fe7262007-04-05 20:26:28 +000051 */
52
Denis Vlasenkob6adbf12007-05-26 19:00:18 +000053/* done in nc.c: #include "libbb.h" */
Denis Vlasenko29fe7262007-04-05 20:26:28 +000054
Denys Vlasenko77cc2c52010-06-27 04:22:02 +020055//usage:#if ENABLE_NC_110_COMPAT
56//usage:
57//usage:#define nc_trivial_usage
58//usage: "[OPTIONS] HOST PORT - connect"
59//usage: IF_NC_SERVER("\n"
Denys Vlasenko3b2acb72010-10-09 21:10:32 +020060//usage: "nc [OPTIONS] -l -p PORT [HOST] [PORT] - listen"
61//usage: )
Denys Vlasenko77cc2c52010-06-27 04:22:02 +020062//usage:#define nc_full_usage "\n\n"
Denys Vlasenko66426762011-06-05 03:58:28 +020063//usage: " -e PROG Run PROG after connect (must be last)"
Denys Vlasenko77cc2c52010-06-27 04:22:02 +020064//usage: IF_NC_SERVER(
65//usage: "\n -l Listen mode, for inbound connects"
66//usage: )
67//usage: "\n -p PORT Local port"
68//usage: "\n -s ADDR Local address"
69//usage: "\n -w SEC Timeout for connects and final net reads"
70//usage: IF_NC_EXTRA(
71//usage: "\n -i SEC Delay interval for lines sent" /* ", ports scanned" */
72//usage: )
73//usage: "\n -n Don't do DNS resolution"
74//usage: "\n -u UDP mode"
75//usage: "\n -v Verbose"
76//usage: IF_NC_EXTRA(
77//usage: "\n -o FILE Hex dump traffic"
78//usage: "\n -z Zero-I/O mode (scanning)"
79//usage: )
80//usage:#endif
81
82/* "\n -r Randomize local and remote ports" */
83/* "\n -g gateway Source-routing hop point[s], up to 8" */
84/* "\n -G num Source-routing pointer: 4, 8, 12, ..." */
85/* "\nport numbers can be individual or ranges: lo-hi [inclusive]" */
86
87/* -e PROG can take ARGS too: "nc ... -e ls -l", but we don't document it
88 * in help text: nc 1.10 does not allow that. We don't want to entice
89 * users to use this incompatibility */
90
Denis Vlasenko19507f02007-04-06 10:41:05 +000091enum {
92 SLEAZE_PORT = 31337, /* for UDP-scan RTT trick, change if ya want */
93 BIGSIZ = 8192, /* big buffers */
94
95 netfd = 3,
96 ofd = 4,
97};
Denis Vlasenko29fe7262007-04-05 20:26:28 +000098
99struct globals {
Denis Vlasenko19507f02007-04-06 10:41:05 +0000100 /* global cmd flags: */
101 unsigned o_verbose;
102 unsigned o_wait;
103#if ENABLE_NC_EXTRA
104 unsigned o_interval;
105#endif
106
107 /*int netfd;*/
108 /*int ofd;*/ /* hexdump output fd */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000109#if ENABLE_LFS
110#define SENT_N_RECV_M "sent %llu, rcvd %llu\n"
111 unsigned long long wrote_out; /* total stdout bytes */
112 unsigned long long wrote_net; /* total net bytes */
113#else
Denis Vlasenko19507f02007-04-06 10:41:05 +0000114#define SENT_N_RECV_M "sent %u, rcvd %u\n"
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000115 unsigned wrote_out; /* total stdout bytes */
116 unsigned wrote_net; /* total net bytes */
117#endif
Bernhard Reutner-Fischera985d302008-02-11 11:44:38 +0000118 /* ouraddr is never NULL and goes through three states as we progress:
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000119 1 - local address before bind (IP/port possibly zero)
120 2 - local address after bind (port is nonzero)
121 3 - local address after connect??/recv/accept (IP and port are nonzero) */
122 struct len_and_sockaddr *ouraddr;
123 /* themaddr is NULL if no peer hostname[:port] specified on command line */
124 struct len_and_sockaddr *themaddr;
125 /* remend is set after connect/recv/accept to the actual ip:port of peer */
126 struct len_and_sockaddr remend;
127
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000128 jmp_buf jbuf; /* timer crud */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000129
130 /* will malloc up the following globals: */
131 fd_set ding1; /* for select loop */
132 fd_set ding2;
133 char bigbuf_in[BIGSIZ]; /* data buffers */
134 char bigbuf_net[BIGSIZ];
135};
136
137#define G (*ptr_to_globals)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000138#define wrote_out (G.wrote_out )
139#define wrote_net (G.wrote_net )
140#define ouraddr (G.ouraddr )
141#define themaddr (G.themaddr )
142#define remend (G.remend )
143#define jbuf (G.jbuf )
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000144#define ding1 (G.ding1 )
145#define ding2 (G.ding2 )
146#define bigbuf_in (G.bigbuf_in )
147#define bigbuf_net (G.bigbuf_net)
148#define o_verbose (G.o_verbose )
149#define o_wait (G.o_wait )
150#if ENABLE_NC_EXTRA
151#define o_interval (G.o_interval)
152#else
153#define o_interval 0
154#endif
Denis Vlasenko574f2f42008-02-27 18:41:59 +0000155#define INIT_G() do { \
156 SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
157} while (0)
158
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000159
160/* Must match getopt32 call! */
161enum {
162 OPT_h = (1 << 0),
163 OPT_n = (1 << 1),
164 OPT_p = (1 << 2),
165 OPT_s = (1 << 3),
166 OPT_u = (1 << 4),
167 OPT_v = (1 << 5),
168 OPT_w = (1 << 6),
169 OPT_l = (1 << 7) * ENABLE_NC_SERVER,
170 OPT_i = (1 << (7+ENABLE_NC_SERVER)) * ENABLE_NC_EXTRA,
171 OPT_o = (1 << (8+ENABLE_NC_SERVER)) * ENABLE_NC_EXTRA,
172 OPT_z = (1 << (9+ENABLE_NC_SERVER)) * ENABLE_NC_EXTRA,
173};
174
175#define o_nflag (option_mask32 & OPT_n)
176#define o_udpmode (option_mask32 & OPT_u)
Denis Vlasenko19507f02007-04-06 10:41:05 +0000177#if ENABLE_NC_SERVER
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000178#define o_listen (option_mask32 & OPT_l)
Denis Vlasenko19507f02007-04-06 10:41:05 +0000179#else
180#define o_listen 0
181#endif
182#if ENABLE_NC_EXTRA
183#define o_ofile (option_mask32 & OPT_o)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000184#define o_zero (option_mask32 & OPT_z)
185#else
Denis Vlasenko19507f02007-04-06 10:41:05 +0000186#define o_ofile 0
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000187#define o_zero 0
188#endif
189
Denis Vlasenko19507f02007-04-06 10:41:05 +0000190/* Debug: squirt whatever message and sleep a bit so we can see it go by. */
191/* Beware: writes to stdOUT... */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000192#if 0
Denys Vlasenko8131eea2009-11-02 14:19:51 +0100193#define Debug(...) do { printf(__VA_ARGS__); printf("\n"); fflush_all(); sleep(1); } while (0)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000194#else
Denis Vlasenko51742f42007-04-12 00:32:05 +0000195#define Debug(...) do { } while (0)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000196#endif
197
Denis Vlasenko51742f42007-04-12 00:32:05 +0000198#define holler_error(...) do { if (o_verbose) bb_error_msg(__VA_ARGS__); } while (0)
199#define holler_perror(...) do { if (o_verbose) bb_perror_msg(__VA_ARGS__); } while (0)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000200
201/* catch: no-brainer interrupt handler */
202static void catch(int sig)
203{
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000204 if (o_verbose > 1) /* normally we don't care */
205 fprintf(stderr, SENT_N_RECV_M, wrote_net, wrote_out);
206 fprintf(stderr, "punt!\n");
Denis Vlasenko400d8bb2008-02-24 13:36:01 +0000207 kill_myself_with_sig(sig);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000208}
209
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000210/* unarm */
211static void unarm(void)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000212{
213 signal(SIGALRM, SIG_IGN);
214 alarm(0);
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000215}
216
217/* timeout and other signal handling cruft */
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000218static void tmtravel(int sig UNUSED_PARAM)
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000219{
220 unarm();
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000221 longjmp(jbuf, 1);
222}
223
224/* arm: set the timer. */
225static void arm(unsigned secs)
226{
227 signal(SIGALRM, tmtravel);
228 alarm(secs);
229}
230
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000231/* findline:
232 find the next newline in a buffer; return inclusive size of that "line",
233 or the entire buffer size, so the caller knows how much to then write().
234 Not distinguishing \n vs \r\n for the nonce; it just works as is... */
235static unsigned findline(char *buf, unsigned siz)
236{
237 char * p;
238 int x;
239 if (!buf) /* various sanity checks... */
240 return 0;
241 if (siz > BIGSIZ)
242 return 0;
243 x = siz;
244 for (p = buf; x > 0; x--) {
245 if (*p == '\n') {
246 x = (int) (p - buf);
247 x++; /* 'sokay if it points just past the end! */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000248Debug("findline returning %d", x);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000249 return x;
250 }
251 p++;
252 } /* for */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000253Debug("findline returning whole thing: %d", siz);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000254 return siz;
255} /* findline */
256
257/* doexec:
258 fiddle all the file descriptors around, and hand off to another prog. Sort
259 of like a one-off "poor man's inetd". This is the only section of code
260 that would be security-critical, which is why it's ifdefed out by default.
261 Use at your own hairy risk; if you leave shells lying around behind open
262 listening ports you deserve to lose!! */
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000263static int doexec(char **proggie) NORETURN;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000264static int doexec(char **proggie)
265{
266 xmove_fd(netfd, 0);
267 dup2(0, 1);
268 /* dup2(0, 2); - do we *really* want this? NO!
269 * exec'ed prog can do it yourself, if needed */
Denys Vlasenko1c31e9e2010-11-28 04:34:09 +0100270 BB_EXECVP_or_die(proggie);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000271}
272
273/* connect_w_timeout:
274 return an fd for one of
275 an open outbound TCP connection, a UDP stub-socket thingie, or
276 an unconnected TCP or UDP socket to listen on.
277 Examines various global o_blah flags to figure out what to do.
278 lad can be NULL, then socket is not bound to any local ip[:port] */
279static int connect_w_timeout(int fd)
280{
281 int rr;
282
283 /* wrap connect inside a timer, and hit it */
284 arm(o_wait);
285 if (setjmp(jbuf) == 0) {
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000286 rr = connect(fd, &themaddr->u.sa, themaddr->len);
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000287 unarm();
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000288 } else { /* setjmp: connect failed... */
289 rr = -1;
290 errno = ETIMEDOUT; /* fake it */
291 }
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000292 return rr;
293}
294
295/* dolisten:
296 listens for
297 incoming and returns an open connection *from* someplace. If we were
298 given host/port args, any connections from elsewhere are rejected. This
299 in conjunction with local-address binding should limit things nicely... */
300static void dolisten(void)
301{
302 int rr;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000303
304 if (!o_udpmode)
305 xlisten(netfd, 1); /* TCP: gotta listen() before we can get */
306
307 /* Various things that follow temporarily trash bigbuf_net, which might contain
308 a copy of any recvfrom()ed packet, but we'll read() another copy later. */
309
310 /* I can't believe I have to do all this to get my own goddamn bound address
311 and port number. It should just get filled in during bind() or something.
312 All this is only useful if we didn't say -p for listening, since if we
313 said -p we *know* what port we're listening on. At any rate we won't bother
314 with it all unless we wanted to see it, although listening quietly on a
315 random unknown port is probably not very useful without "netstat". */
316 if (o_verbose) {
317 char *addr;
Denis Vlasenkoa771e7c2009-04-21 23:48:38 +0000318 getsockname(netfd, &ouraddr->u.sa, &ouraddr->len);
319 //if (rr < 0)
320 // bb_perror_msg_and_die("getsockname after bind");
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000321 addr = xmalloc_sockaddr2dotted(&ouraddr->u.sa);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000322 fprintf(stderr, "listening on %s ...\n", addr);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000323 free(addr);
324 }
325
326 if (o_udpmode) {
327 /* UDP is a speeeeecial case -- we have to do I/O *and* get the calling
328 party's particulars all at once, listen() and accept() don't apply.
329 At least in the BSD universe, however, recvfrom/PEEK is enough to tell
330 us something came in, and we can set things up so straight read/write
331 actually does work after all. Yow. YMMV on strange platforms! */
332
333 /* I'm not completely clear on how this works -- BSD seems to make UDP
334 just magically work in a connect()ed context, but we'll undoubtedly run
335 into systems this deal doesn't work on. For now, we apparently have to
336 issue a connect() on our just-tickled socket so we can write() back.
337 Again, why the fuck doesn't it just get filled in and taken care of?!
338 This hack is anything but optimal. Basically, if you want your listener
339 to also be able to send data back, you need this connect() line, which
340 also has the side effect that now anything from a different source or even a
341 different port on the other end won't show up and will cause ICMP errors.
342 I guess that's what they meant by "connect".
343 Let's try to remember what the "U" is *really* for, eh? */
344
345 /* If peer address is specified, connect to it */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000346 remend.len = LSA_SIZEOF_SA;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000347 if (themaddr) {
348 remend = *themaddr;
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000349 xconnect(netfd, &themaddr->u.sa, themaddr->len);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000350 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000351 /* peek first packet and remember peer addr */
352 arm(o_wait); /* might as well timeout this, too */
353 if (setjmp(jbuf) == 0) { /* do timeout for initial connect */
354 /* (*ouraddr) is prefilled with "default" address */
355 /* and here we block... */
356 rr = recv_from_to(netfd, NULL, 0, MSG_PEEK, /*was bigbuf_net, BIGSIZ*/
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000357 &remend.u.sa, &ouraddr->u.sa, ouraddr->len);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000358 if (rr < 0)
359 bb_perror_msg_and_die("recvfrom");
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000360 unarm();
Denis Vlasenko19507f02007-04-06 10:41:05 +0000361 } else
362 bb_error_msg_and_die("timeout");
Denis Vlasenko19507f02007-04-06 10:41:05 +0000363/* Now we learned *to which IP* peer has connected, and we want to anchor
364our socket on it, so that our outbound packets will have correct local IP.
365Unfortunately, bind() on already bound socket will fail now (EINVAL):
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000366 xbind(netfd, &ouraddr->u.sa, ouraddr->len);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000367Need to read the packet, save data, close this socket and
368create new one, and bind() it. TODO */
369 if (!themaddr)
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000370 xconnect(netfd, &remend.u.sa, ouraddr->len);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000371 } else {
372 /* TCP */
373 arm(o_wait); /* wrap this in a timer, too; 0 = forever */
374 if (setjmp(jbuf) == 0) {
Denis Vlasenko19507f02007-04-06 10:41:05 +0000375 again:
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000376 remend.len = LSA_SIZEOF_SA;
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000377 rr = accept(netfd, &remend.u.sa, &remend.len);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000378 if (rr < 0)
379 bb_perror_msg_and_die("accept");
Denys Vlasenko866710a2010-01-08 16:09:45 +0100380 if (themaddr) {
381 int sv_port, port, r;
382
383 sv_port = get_nport(&remend.u.sa); /* save */
384 port = get_nport(&themaddr->u.sa);
385 if (port == 0) {
386 /* "nc -nl -p LPORT RHOST" (w/o RPORT!):
387 * we should accept any remote port */
Denys Vlasenkoca183112011-04-07 17:52:20 +0200388 set_nport(&remend.u.sa, 0); /* blot out remote port# */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000389 }
Denys Vlasenko866710a2010-01-08 16:09:45 +0100390 r = memcmp(&remend.u.sa, &themaddr->u.sa, remend.len);
Denys Vlasenkoca183112011-04-07 17:52:20 +0200391 set_nport(&remend.u.sa, sv_port); /* restore */
Denys Vlasenko866710a2010-01-08 16:09:45 +0100392 if (r != 0) {
393 /* nc 1.10 bails out instead, and its error message
394 * is not suppressed by o_verbose */
395 if (o_verbose) {
396 char *remaddr = xmalloc_sockaddr2dotted(&remend.u.sa);
397 bb_error_msg("connect from wrong ip/port %s ignored", remaddr);
398 free(remaddr);
399 }
400 close(rr);
401 goto again;
402 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000403 }
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000404 unarm();
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000405 } else
406 bb_error_msg_and_die("timeout");
Denis Vlasenko19507f02007-04-06 10:41:05 +0000407 xmove_fd(rr, netfd); /* dump the old socket, here's our new one */
408 /* find out what address the connection was *to* on our end, in case we're
409 doing a listen-on-any on a multihomed machine. This allows one to
410 offer different services via different alias addresses, such as the
411 "virtual web site" hack. */
Denis Vlasenkoa771e7c2009-04-21 23:48:38 +0000412 getsockname(netfd, &ouraddr->u.sa, &ouraddr->len);
413 //if (rr < 0)
414 // bb_perror_msg_and_die("getsockname after accept");
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000415 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000416
417 if (o_verbose) {
418 char *lcladdr, *remaddr, *remhostname;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000419
420#if ENABLE_NC_EXTRA && defined(IP_OPTIONS)
421 /* If we can, look for any IP options. Useful for testing the receiving end of
422 such things, and is a good exercise in dealing with it. We do this before
423 the connect message, to ensure that the connect msg is uniformly the LAST
424 thing to emerge after all the intervening crud. Doesn't work for UDP on
425 any machines I've tested, but feel free to surprise me. */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000426 char optbuf[40];
Denis Vlasenkoc4f12f52008-05-12 14:35:56 +0000427 socklen_t x = sizeof(optbuf);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000428
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000429 rr = getsockopt(netfd, IPPROTO_IP, IP_OPTIONS, optbuf, &x);
Denis Vlasenkof6b46852009-04-25 13:16:53 +0000430 if (rr >= 0 && x) { /* we've got options, lessee em... */
Denys Vlasenko2ea73b52011-10-19 22:31:01 +0200431 *bin2hex(bigbuf_net, optbuf, x) = '\0';
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000432 fprintf(stderr, "IP options: %s\n", bigbuf_net);
433 }
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000434#endif
435
436 /* now check out who it is. We don't care about mismatched DNS names here,
437 but any ADDR and PORT we specified had better fucking well match the caller.
438 Converting from addr to inet_ntoa and back again is a bit of a kludge, but
439 gethostpoop wants a string and there's much gnarlier code out there already,
440 so I don't feel bad.
441 The *real* question is why BFD sockets wasn't designed to allow listens for
442 connections *from* specific hosts/ports, instead of requiring the caller to
443 accept the connection and then reject undesireable ones by closing.
444 In other words, we need a TCP MSG_PEEK. */
445 /* bbox: removed most of it */
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000446 lcladdr = xmalloc_sockaddr2dotted(&ouraddr->u.sa);
447 remaddr = xmalloc_sockaddr2dotted(&remend.u.sa);
448 remhostname = o_nflag ? remaddr : xmalloc_sockaddr2host(&remend.u.sa);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000449 fprintf(stderr, "connect to %s from %s (%s)\n",
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000450 lcladdr, remhostname, remaddr);
451 free(lcladdr);
452 free(remaddr);
453 if (!o_nflag)
454 free(remhostname);
455 }
456}
457
458/* udptest:
459 fire a couple of packets at a UDP target port, just to see if it's really
460 there. On BSD kernels, ICMP host/port-unreachable errors get delivered to
461 our socket as ECONNREFUSED write errors. On SV kernels, we lose; we'll have
462 to collect and analyze raw ICMP ourselves a la satan's probe_udp_ports
463 backend. Guess where one could swipe the appropriate code from...
464
465 Use the time delay between writes if given, otherwise use the "tcp ping"
466 trick for getting the RTT. [I got that idea from pluvius, and warped it.]
467 Return either the original fd, or clean up and return -1. */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000468#if ENABLE_NC_EXTRA
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000469static int udptest(void)
470{
471 int rr;
472
473 rr = write(netfd, bigbuf_in, 1);
474 if (rr != 1)
475 bb_perror_msg("udptest first write");
476
477 if (o_wait)
Denis Vlasenko19507f02007-04-06 10:41:05 +0000478 sleep(o_wait); // can be interrupted! while (t) nanosleep(&t)?
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000479 else {
480 /* use the tcp-ping trick: try connecting to a normally refused port, which
481 causes us to block for the time that SYN gets there and RST gets back.
482 Not completely reliable, but it *does* mostly work. */
483 /* Set a temporary connect timeout, so packet filtration doesnt cause
484 us to hang forever, and hit it */
485 o_wait = 5; /* enough that we'll notice?? */
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000486 rr = xsocket(ouraddr->u.sa.sa_family, SOCK_STREAM, 0);
Denys Vlasenkoca183112011-04-07 17:52:20 +0200487 set_nport(&themaddr->u.sa, htons(SLEAZE_PORT));
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000488 connect_w_timeout(rr);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000489 /* don't need to restore themaddr's port, it's not used anymore */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000490 close(rr);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000491 o_wait = 0; /* restore */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000492 }
493
494 rr = write(netfd, bigbuf_in, 1);
495 return (rr != 1); /* if rr == 1, return 0 (success) */
496}
Denis Vlasenko19507f02007-04-06 10:41:05 +0000497#else
498int udptest(void);
499#endif
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000500
501/* oprint:
502 Hexdump bytes shoveled either way to a running logfile, in the format:
503 D offset - - - - --- 16 bytes --- - - - - # .... ascii .....
504 where "which" sets the direction indicator, D:
505 0 -- sent to network, or ">"
506 1 -- rcvd and printed to stdout, or "<"
507 and "buf" and "n" are data-block and length. If the current block generates
508 a partial line, so be it; we *want* that lockstep indication of who sent
509 what when. Adapted from dgaudet's original example -- but must be ripping
510 *fast*, since we don't want to be too disk-bound... */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000511#if ENABLE_NC_EXTRA
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000512static void oprint(int direction, unsigned char *p, unsigned bc)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000513{
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000514 unsigned obc; /* current "global" offset */
515 unsigned x;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000516 unsigned char *op; /* out hexdump ptr */
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000517 unsigned char *ap; /* out asc-dump ptr */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000518 unsigned char stage[100];
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000519
Denis Vlasenko19507f02007-04-06 10:41:05 +0000520 if (bc == 0)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000521 return;
522
Denis Vlasenko19507f02007-04-06 10:41:05 +0000523 obc = wrote_net; /* use the globals! */
524 if (direction == '<')
525 obc = wrote_out;
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000526 stage[0] = direction;
527 stage[59] = '#'; /* preload separator */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000528 stage[60] = ' ';
529
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000530 do { /* for chunk-o-data ... */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000531 x = 16;
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000532 if (bc < 16) {
533 /* memset(&stage[bc*3 + 11], ' ', 16*3 - bc*3); */
534 memset(&stage[11], ' ', 16*3);
535 x = bc;
536 }
Denis Vlasenkoc4f12f52008-05-12 14:35:56 +0000537 sprintf((char *)&stage[1], " %8.8x ", obc); /* xxx: still slow? */
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000538 bc -= x; /* fix current count */
539 obc += x; /* fix current offset */
540 op = &stage[11]; /* where hex starts */
541 ap = &stage[61]; /* where ascii starts */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000542
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000543 do { /* for line of dump, however long ... */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000544 *op++ = 0x20 | bb_hexdigits_upcase[*p >> 4];
545 *op++ = 0x20 | bb_hexdigits_upcase[*p & 0x0f];
546 *op++ = ' ';
547 if ((*p > 31) && (*p < 127))
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000548 *ap = *p; /* printing */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000549 else
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000550 *ap = '.'; /* nonprinting, loose def */
551 ap++;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000552 p++;
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000553 } while (--x);
554 *ap++ = '\n'; /* finish the line */
555 xwrite(ofd, stage, ap - stage);
556 } while (bc);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000557}
Denis Vlasenko19507f02007-04-06 10:41:05 +0000558#else
Denis Vlasenko828e2a92007-07-13 12:37:31 +0000559void oprint(int direction, unsigned char *p, unsigned bc);
Denis Vlasenko19507f02007-04-06 10:41:05 +0000560#endif
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000561
562/* readwrite:
563 handle stdin/stdout/network I/O. Bwahaha!! -- the select loop from hell.
564 In this instance, return what might become our exit status. */
565static int readwrite(void)
566{
567 int rr;
568 char *zp = zp; /* gcc */ /* stdin buf ptr */
569 char *np = np; /* net-in buf ptr */
570 unsigned rzleft;
571 unsigned rnleft;
572 unsigned netretry; /* net-read retry counter */
573 unsigned wretry; /* net-write sanity counter */
574 unsigned wfirst; /* one-shot flag to skip first net read */
575
576 /* if you don't have all this FD_* macro hair in sys/types.h, you'll have to
577 either find it or do your own bit-bashing: *ding1 |= (1 << fd), etc... */
578 FD_SET(netfd, &ding1); /* global: the net is open */
579 netretry = 2;
580 wfirst = 0;
581 rzleft = rnleft = 0;
582 if (o_interval)
583 sleep(o_interval); /* pause *before* sending stuff, too */
584
585 errno = 0; /* clear from sleep, close, whatever */
586 /* and now the big ol' select shoveling loop ... */
587 while (FD_ISSET(netfd, &ding1)) { /* i.e. till the *net* closes! */
588 wretry = 8200; /* more than we'll ever hafta write */
589 if (wfirst) { /* any saved stdin buffer? */
590 wfirst = 0; /* clear flag for the duration */
591 goto shovel; /* and go handle it first */
592 }
593 ding2 = ding1; /* FD_COPY ain't portable... */
594 /* some systems, notably linux, crap into their select timers on return, so
595 we create a expendable copy and give *that* to select. */
596 if (o_wait) {
597 struct timeval tmp_timer;
598 tmp_timer.tv_sec = o_wait;
599 tmp_timer.tv_usec = 0;
Denis Vlasenko19507f02007-04-06 10:41:05 +0000600 /* highest possible fd is netfd (3) */
601 rr = select(netfd+1, &ding2, NULL, NULL, &tmp_timer);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000602 } else
Denis Vlasenko19507f02007-04-06 10:41:05 +0000603 rr = select(netfd+1, &ding2, NULL, NULL, NULL);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000604 if (rr < 0 && errno != EINTR) { /* might have gotten ^Zed, etc */
605 holler_perror("select");
606 close(netfd);
607 return 1;
608 }
609 /* if we have a timeout AND stdin is closed AND we haven't heard anything
610 from the net during that time, assume it's dead and close it too. */
611 if (rr == 0) {
Bernhard Reutner-Fischer5e25ddb2008-05-19 09:48:17 +0000612 if (!FD_ISSET(STDIN_FILENO, &ding1))
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000613 netretry--; /* we actually try a coupla times. */
614 if (!netretry) {
615 if (o_verbose > 1) /* normally we don't care */
616 fprintf(stderr, "net timeout\n");
617 close(netfd);
618 return 0; /* not an error! */
619 }
620 } /* select timeout */
621 /* xxx: should we check the exception fds too? The read fds seem to give
622 us the right info, and none of the examples I found bothered. */
623
624 /* Ding!! Something arrived, go check all the incoming hoppers, net first */
625 if (FD_ISSET(netfd, &ding2)) { /* net: ding! */
626 rr = read(netfd, bigbuf_net, BIGSIZ);
627 if (rr <= 0) {
Denis Vlasenko19507f02007-04-06 10:41:05 +0000628 if (rr < 0 && o_verbose > 1) {
629 /* nc 1.10 doesn't do this */
630 bb_perror_msg("net read");
631 }
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000632 FD_CLR(netfd, &ding1); /* net closed, we'll finish up... */
633 rzleft = 0; /* can't write anymore: broken pipe */
634 } else {
635 rnleft = rr;
636 np = bigbuf_net;
637 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000638Debug("got %d from the net, errno %d", rr, errno);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000639 } /* net:ding */
640
641 /* if we're in "slowly" mode there's probably still stuff in the stdin
642 buffer, so don't read unless we really need MORE INPUT! MORE INPUT! */
643 if (rzleft)
644 goto shovel;
645
646 /* okay, suck more stdin */
Bernhard Reutner-Fischer5e25ddb2008-05-19 09:48:17 +0000647 if (FD_ISSET(STDIN_FILENO, &ding2)) { /* stdin: ding! */
648 rr = read(STDIN_FILENO, bigbuf_in, BIGSIZ);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000649 /* Considered making reads here smaller for UDP mode, but 8192-byte
650 mobygrams are kinda fun and exercise the reassembler. */
651 if (rr <= 0) { /* at end, or fukt, or ... */
Bernhard Reutner-Fischer5e25ddb2008-05-19 09:48:17 +0000652 FD_CLR(STDIN_FILENO, &ding1); /* disable and close stdin */
Denis Vlasenkof6b46852009-04-25 13:16:53 +0000653 close(STDIN_FILENO);
654// Does it make sense to shutdown(net_fd, SHUT_WR)
655// to let other side know that we won't write anything anymore?
656// (and what about keeping compat if we do that?)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000657 } else {
658 rzleft = rr;
659 zp = bigbuf_in;
660 }
661 } /* stdin:ding */
662 shovel:
663 /* now that we've dingdonged all our thingdings, send off the results.
664 Geez, why does this look an awful lot like the big loop in "rsh"? ...
665 not sure if the order of this matters, but write net -> stdout first. */
666
667 /* sanity check. Works because they're both unsigned... */
668 if ((rzleft > 8200) || (rnleft > 8200)) {
669 holler_error("bogus buffers: %u, %u", rzleft, rnleft);
670 rzleft = rnleft = 0;
671 }
672 /* net write retries sometimes happen on UDP connections */
673 if (!wretry) { /* is something hung? */
674 holler_error("too many output retries");
675 return 1;
676 }
677 if (rnleft) {
Bernhard Reutner-Fischer5e25ddb2008-05-19 09:48:17 +0000678 rr = write(STDOUT_FILENO, np, rnleft);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000679 if (rr > 0) {
Denis Vlasenkoc4f12f52008-05-12 14:35:56 +0000680 if (o_ofile) /* log the stdout */
681 oprint('<', (unsigned char *)np, rr);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000682 np += rr; /* fix up ptrs and whatnot */
683 rnleft -= rr; /* will get sanity-checked above */
684 wrote_out += rr; /* global count */
685 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000686Debug("wrote %d to stdout, errno %d", rr, errno);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000687 } /* rnleft */
688 if (rzleft) {
689 if (o_interval) /* in "slowly" mode ?? */
690 rr = findline(zp, rzleft);
691 else
692 rr = rzleft;
693 rr = write(netfd, zp, rr); /* one line, or the whole buffer */
694 if (rr > 0) {
Denis Vlasenkoc4f12f52008-05-12 14:35:56 +0000695 if (o_ofile) /* log what got sent */
696 oprint('>', (unsigned char *)zp, rr);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000697 zp += rr;
698 rzleft -= rr;
699 wrote_net += rr; /* global count */
700 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000701Debug("wrote %d to net, errno %d", rr, errno);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000702 } /* rzleft */
703 if (o_interval) { /* cycle between slow lines, or ... */
704 sleep(o_interval);
705 errno = 0; /* clear from sleep */
706 continue; /* ...with hairy select loop... */
707 }
708 if ((rzleft) || (rnleft)) { /* shovel that shit till they ain't */
709 wretry--; /* none left, and get another load */
710 goto shovel;
711 }
712 } /* while ding1:netfd is open */
713
714 /* XXX: maybe want a more graceful shutdown() here, or screw around with
715 linger times?? I suspect that I don't need to since I'm always doing
716 blocking reads and writes and my own manual "last ditch" efforts to read
717 the net again after a timeout. I haven't seen any screwups yet, but it's
718 not like my test network is particularly busy... */
719 close(netfd);
720 return 0;
721} /* readwrite */
722
723/* main: now we pull it all together... */
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000724int nc_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denys Vlasenko2ec91ae2010-01-04 14:15:38 +0100725int nc_main(int argc UNUSED_PARAM, char **argv)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000726{
Denis Vlasenko1d426652008-03-17 09:09:09 +0000727 char *str_p, *str_s;
Denis Vlasenko5e34ff22009-04-21 11:09:40 +0000728 IF_NC_EXTRA(char *str_i, *str_o;)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000729 char *themdotted = themdotted; /* gcc */
730 char **proggie;
731 int x;
732 unsigned o_lport = 0;
733
Denis Vlasenko574f2f42008-02-27 18:41:59 +0000734 INIT_G();
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000735
736 /* catch a signal or two for cleanup */
Denis Vlasenko25591c32008-02-16 22:58:56 +0000737 bb_signals(0
738 + (1 << SIGINT)
739 + (1 << SIGQUIT)
740 + (1 << SIGTERM)
741 , catch);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000742 /* and suppress others... */
Denis Vlasenko25591c32008-02-16 22:58:56 +0000743 bb_signals(0
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000744#ifdef SIGURG
Denis Vlasenko25591c32008-02-16 22:58:56 +0000745 + (1 << SIGURG)
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000746#endif
Denis Vlasenko25591c32008-02-16 22:58:56 +0000747 + (1 << SIGPIPE) /* important! */
748 , SIG_IGN);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000749
750 proggie = argv;
751 while (*++proggie) {
752 if (strcmp(*proggie, "-e") == 0) {
753 *proggie = NULL;
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000754 proggie++;
755 goto e_found;
756 }
757 }
758 proggie = NULL;
759 e_found:
Denis Vlasenko4b924f32007-05-30 00:29:55 +0000760
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000761 // -g -G -t -r deleted, unimplemented -a deleted too
Denis Vlasenko1d426652008-03-17 09:09:09 +0000762 opt_complementary = "?2:vv:w+"; /* max 2 params; -v is a counter; -w N */
Denis Vlasenko5e34ff22009-04-21 11:09:40 +0000763 getopt32(argv, "hnp:s:uvw:" IF_NC_SERVER("l")
764 IF_NC_EXTRA("i:o:z"),
Denis Vlasenko1d426652008-03-17 09:09:09 +0000765 &str_p, &str_s, &o_wait
Denys Vlasenko1e23f322010-11-22 05:30:54 +0100766 IF_NC_EXTRA(, &str_i, &str_o), &o_verbose);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000767 argv += optind;
768#if ENABLE_NC_EXTRA
769 if (option_mask32 & OPT_i) /* line-interval time */
770 o_interval = xatou_range(str_i, 1, 0xffff);
771#endif
772 //if (option_mask32 & OPT_l) /* listen mode */
773 //if (option_mask32 & OPT_n) /* numeric-only, no DNS lookups */
774 //if (option_mask32 & OPT_o) /* hexdump log */
775 if (option_mask32 & OPT_p) { /* local source port */
776 o_lport = bb_lookup_port(str_p, o_udpmode ? "udp" : "tcp", 0);
777 if (!o_lport)
778 bb_error_msg_and_die("bad local port '%s'", str_p);
779 }
780 //if (option_mask32 & OPT_r) /* randomize various things */
781 //if (option_mask32 & OPT_u) /* use UDP */
782 //if (option_mask32 & OPT_v) /* verbose */
Denis Vlasenko1d426652008-03-17 09:09:09 +0000783 //if (option_mask32 & OPT_w) /* wait time */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000784 //if (option_mask32 & OPT_z) /* little or no data xfer */
785
Denis Vlasenko19507f02007-04-06 10:41:05 +0000786 /* We manage our fd's so that they are never 0,1,2 */
787 /*bb_sanitize_stdio(); - not needed */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000788
Denis Vlasenko5c51a7c2007-06-05 20:08:11 +0000789 if (argv[0]) {
790 themaddr = xhost2sockaddr(argv[0],
791 argv[1]
792 ? bb_lookup_port(argv[1], o_udpmode ? "udp" : "tcp", 0)
793 : 0);
794 }
795
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000796 /* create & bind network socket */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000797 x = (o_udpmode ? SOCK_DGRAM : SOCK_STREAM);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000798 if (option_mask32 & OPT_s) { /* local address */
Denis Vlasenko19507f02007-04-06 10:41:05 +0000799 /* if o_lport is still 0, then we will use random port */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000800 ouraddr = xhost2sockaddr(str_s, o_lport);
Denis Vlasenko4cf1d082008-03-12 23:13:50 +0000801#ifdef BLOAT
802 /* prevent spurious "UDP listen needs !0 port" */
803 o_lport = get_nport(ouraddr);
804 o_lport = ntohs(o_lport);
805#endif
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000806 x = xsocket(ouraddr->u.sa.sa_family, x, 0);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000807 } else {
Denis Vlasenko5c51a7c2007-06-05 20:08:11 +0000808 /* We try IPv6, then IPv4, unless addr family is
809 * implicitly set by way of remote addr/port spec */
810 x = xsocket_type(&ouraddr,
Denis Vlasenko4e6d5112008-03-12 22:14:34 +0000811 (themaddr ? themaddr->u.sa.sa_family : AF_UNSPEC),
Denis Vlasenko5c51a7c2007-06-05 20:08:11 +0000812 x);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000813 if (o_lport)
Denys Vlasenkoca183112011-04-07 17:52:20 +0200814 set_nport(&ouraddr->u.sa, htons(o_lport));
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000815 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000816 xmove_fd(x, netfd);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000817 setsockopt_reuseaddr(netfd);
818 if (o_udpmode)
819 socket_want_pktinfo(netfd);
Denis Vlasenkof6b46852009-04-25 13:16:53 +0000820 if (!ENABLE_FEATURE_UNIX_LOCAL
821 || o_listen
822 || ouraddr->u.sa.sa_family != AF_UNIX
823 ) {
824 xbind(netfd, &ouraddr->u.sa, ouraddr->len);
825 }
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000826#if 0
827 setsockopt(netfd, SOL_SOCKET, SO_RCVBUF, &o_rcvbuf, sizeof o_rcvbuf);
828 setsockopt(netfd, SOL_SOCKET, SO_SNDBUF, &o_sndbuf, sizeof o_sndbuf);
829#endif
830
Denis Vlasenko4cf1d082008-03-12 23:13:50 +0000831#ifdef BLOAT
Denis Vlasenko19507f02007-04-06 10:41:05 +0000832 if (OPT_l && (option_mask32 & (OPT_u|OPT_l)) == (OPT_u|OPT_l)) {
833 /* apparently UDP can listen ON "port 0",
834 but that's not useful */
835 if (!o_lport)
836 bb_error_msg_and_die("UDP listen needs nonzero -p port");
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000837 }
Denis Vlasenko4cf1d082008-03-12 23:13:50 +0000838#endif
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000839
Bernhard Reutner-Fischer5e25ddb2008-05-19 09:48:17 +0000840 FD_SET(STDIN_FILENO, &ding1); /* stdin *is* initially open */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000841 if (proggie) {
842 close(0); /* won't need stdin */
843 option_mask32 &= ~OPT_o; /* -o with -e is meaningless! */
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000844 }
Denis Vlasenko19507f02007-04-06 10:41:05 +0000845#if ENABLE_NC_EXTRA
846 if (o_ofile)
847 xmove_fd(xopen(str_o, O_WRONLY|O_CREAT|O_TRUNC), ofd);
848#endif
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000849
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000850 if (o_listen) {
851 dolisten();
852 /* dolisten does its own connect reporting */
853 if (proggie) /* -e given? */
854 doexec(proggie);
855 x = readwrite(); /* it even works with UDP! */
856 } else {
857 /* Outbound connects. Now we're more picky about args... */
858 if (!themaddr)
Denys Vlasenkob103fb12010-09-07 18:41:56 +0200859 bb_show_usage();
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000860
861 remend = *themaddr;
862 if (o_verbose)
Denis Vlasenko7cff01e2008-02-02 16:23:43 +0000863 themdotted = xmalloc_sockaddr2dotted(&themaddr->u.sa);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000864
865 x = connect_w_timeout(netfd);
866 if (o_zero && x == 0 && o_udpmode) /* if UDP scanning... */
867 x = udptest();
868 if (x == 0) { /* Yow, are we OPEN YET?! */
869 if (o_verbose)
Denis Vlasenko19507f02007-04-06 10:41:05 +0000870 fprintf(stderr, "%s (%s) open\n", argv[0], themdotted);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000871 if (proggie) /* exec is valid for outbound, too */
872 doexec(proggie);
873 if (!o_zero)
874 x = readwrite();
875 } else { /* connect or udptest wasn't successful */
876 x = 1; /* exit status */
877 /* if we're scanning at a "one -v" verbosity level, don't print refusals.
878 Give it another -v if you want to see everything. */
879 if (o_verbose > 1 || (o_verbose && errno != ECONNREFUSED))
Denis Vlasenko19507f02007-04-06 10:41:05 +0000880 bb_perror_msg("%s (%s)", argv[0], themdotted);
Denis Vlasenko29fe7262007-04-05 20:26:28 +0000881 }
882 }
883 if (o_verbose > 1) /* normally we don't care */
884 fprintf(stderr, SENT_N_RECV_M, wrote_net, wrote_out);
885 return x;
886}