diff options
Diffstat (limited to 'src/usr.bin/nc/netcat.c')
| -rw-r--r-- | src/usr.bin/nc/netcat.c | 761 |
1 files changed, 761 insertions, 0 deletions
diff --git a/src/usr.bin/nc/netcat.c b/src/usr.bin/nc/netcat.c new file mode 100644 index 0000000000..df5c44a8ce --- /dev/null +++ b/src/usr.bin/nc/netcat.c | |||
| @@ -0,0 +1,761 @@ | |||
| 1 | /* $OpenBSD: netcat.c,v 1.64 2003/10/19 22:50:35 deraadt Exp $ */ | ||
| 2 | /* | ||
| 3 | * Copyright (c) 2001 Eric Jackson <ericj@monkey.org> | ||
| 4 | * | ||
| 5 | * Redistribution and use in source and binary forms, with or without | ||
| 6 | * modification, are permitted provided that the following conditions | ||
| 7 | * are met: | ||
| 8 | * | ||
| 9 | * 1. Redistributions of source code must retain the above copyright | ||
| 10 | * notice, this list of conditions and the following disclaimer. | ||
| 11 | * 2. Redistributions in binary form must reproduce the above copyright | ||
| 12 | * notice, this list of conditions and the following disclaimer in the | ||
| 13 | * documentation and/or other materials provided with the distribution. | ||
| 14 | * 3. The name of the author may not be used to endorse or promote products | ||
| 15 | * derived from this software without specific prior written permission. | ||
| 16 | * | ||
| 17 | * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR | ||
| 18 | * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES | ||
| 19 | * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. | ||
| 20 | * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, | ||
| 21 | * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT | ||
| 22 | * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | ||
| 23 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | ||
| 24 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | ||
| 25 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF | ||
| 26 | * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | ||
| 27 | */ | ||
| 28 | |||
| 29 | /* | ||
| 30 | * Re-written nc(1) for OpenBSD. Original implementation by | ||
| 31 | * *Hobbit* <hobbit@avian.org>. | ||
| 32 | */ | ||
| 33 | |||
| 34 | #include <sys/types.h> | ||
| 35 | #include <sys/socket.h> | ||
| 36 | #include <sys/time.h> | ||
| 37 | #include <sys/un.h> | ||
| 38 | |||
| 39 | #include <netinet/in.h> | ||
| 40 | #include <arpa/telnet.h> | ||
| 41 | |||
| 42 | #include <err.h> | ||
| 43 | #include <errno.h> | ||
| 44 | #include <netdb.h> | ||
| 45 | #include <poll.h> | ||
| 46 | #include <stdarg.h> | ||
| 47 | #include <stdio.h> | ||
| 48 | #include <stdlib.h> | ||
| 49 | #include <string.h> | ||
| 50 | #include <unistd.h> | ||
| 51 | #include <fcntl.h> | ||
| 52 | |||
| 53 | #ifndef SUN_LEN | ||
| 54 | #define SUN_LEN(su) \ | ||
| 55 | (sizeof(*(su)) - sizeof((su)->sun_path) + strlen((su)->sun_path)) | ||
| 56 | #endif | ||
| 57 | |||
| 58 | #define PORT_MAX 65535 | ||
| 59 | #define PORT_MAX_LEN 6 | ||
| 60 | |||
| 61 | /* Command Line Options */ | ||
| 62 | int iflag; /* Interval Flag */ | ||
| 63 | int kflag; /* More than one connect */ | ||
| 64 | int lflag; /* Bind to local port */ | ||
| 65 | int nflag; /* Dont do name lookup */ | ||
| 66 | char *pflag; /* Localport flag */ | ||
| 67 | int rflag; /* Random ports flag */ | ||
| 68 | char *sflag; /* Source Address */ | ||
| 69 | int tflag; /* Telnet Emulation */ | ||
| 70 | int uflag; /* UDP - Default to TCP */ | ||
| 71 | int vflag; /* Verbosity */ | ||
| 72 | int xflag; /* Socks proxy */ | ||
| 73 | int zflag; /* Port Scan Flag */ | ||
| 74 | |||
| 75 | int timeout = -1; | ||
| 76 | int family = AF_UNSPEC; | ||
| 77 | char *portlist[PORT_MAX+1]; | ||
| 78 | |||
| 79 | ssize_t atomicio(ssize_t (*)(int, void *, size_t), int, void *, size_t); | ||
| 80 | void atelnet(int, unsigned char *, unsigned int); | ||
| 81 | void build_ports(char *); | ||
| 82 | void help(void); | ||
| 83 | int local_listen(char *, char *, struct addrinfo); | ||
| 84 | void readwrite(int); | ||
| 85 | int remote_connect(char *, char *, struct addrinfo); | ||
| 86 | int socks_connect(char *, char *, struct addrinfo, char *, char *, | ||
| 87 | struct addrinfo, int); | ||
| 88 | int udptest(int); | ||
| 89 | int unix_connect(char *); | ||
| 90 | int unix_listen(char *); | ||
| 91 | void usage(int); | ||
| 92 | |||
| 93 | int | ||
| 94 | main(int argc, char *argv[]) | ||
| 95 | { | ||
| 96 | int ch, s, ret, socksv; | ||
| 97 | char *host, *uport, *endp; | ||
| 98 | struct addrinfo hints; | ||
| 99 | struct servent *sv; | ||
| 100 | socklen_t len; | ||
| 101 | struct sockaddr *cliaddr; | ||
| 102 | char *proxy; | ||
| 103 | char *proxyhost = "", *proxyport = NULL; | ||
| 104 | struct addrinfo proxyhints; | ||
| 105 | |||
| 106 | ret = 1; | ||
| 107 | s = 0; | ||
| 108 | socksv = 5; | ||
| 109 | host = NULL; | ||
| 110 | uport = NULL; | ||
| 111 | endp = NULL; | ||
| 112 | sv = NULL; | ||
| 113 | |||
| 114 | while ((ch = getopt(argc, argv, "46UX:hi:klnp:rs:tuvw:x:z")) != -1) { | ||
| 115 | switch (ch) { | ||
| 116 | case '4': | ||
| 117 | family = AF_INET; | ||
| 118 | break; | ||
| 119 | case '6': | ||
| 120 | family = AF_INET6; | ||
| 121 | break; | ||
| 122 | case 'U': | ||
| 123 | family = AF_UNIX; | ||
| 124 | break; | ||
| 125 | case 'X': | ||
| 126 | socksv = (int)strtoul(optarg, &endp, 10); | ||
| 127 | if ((socksv != 4 && socksv != 5) || *endp != '\0') | ||
| 128 | errx(1, "only SOCKS version 4 and 5 supported"); | ||
| 129 | break; | ||
| 130 | case 'h': | ||
| 131 | help(); | ||
| 132 | break; | ||
| 133 | case 'i': | ||
| 134 | iflag = (int)strtoul(optarg, &endp, 10); | ||
| 135 | if (iflag < 0 || *endp != '\0') | ||
| 136 | errx(1, "interval cannot be negative"); | ||
| 137 | break; | ||
| 138 | case 'k': | ||
| 139 | kflag = 1; | ||
| 140 | break; | ||
| 141 | case 'l': | ||
| 142 | lflag = 1; | ||
| 143 | break; | ||
| 144 | case 'n': | ||
| 145 | nflag = 1; | ||
| 146 | break; | ||
| 147 | case 'p': | ||
| 148 | pflag = optarg; | ||
| 149 | break; | ||
| 150 | case 'r': | ||
| 151 | rflag = 1; | ||
| 152 | break; | ||
| 153 | case 's': | ||
| 154 | sflag = optarg; | ||
| 155 | break; | ||
| 156 | case 't': | ||
| 157 | tflag = 1; | ||
| 158 | break; | ||
| 159 | case 'u': | ||
| 160 | uflag = 1; | ||
| 161 | break; | ||
| 162 | case 'v': | ||
| 163 | vflag = 1; | ||
| 164 | break; | ||
| 165 | case 'w': | ||
| 166 | timeout = (int)strtoul(optarg, &endp, 10); | ||
| 167 | if (timeout < 0 || *endp != '\0') | ||
| 168 | errx(1, "timeout cannot be negative"); | ||
| 169 | if (timeout >= (INT_MAX / 1000)) | ||
| 170 | errx(1, "timeout too large"); | ||
| 171 | timeout *= 1000; | ||
| 172 | break; | ||
| 173 | case 'x': | ||
| 174 | xflag = 1; | ||
| 175 | if ((proxy = strdup(optarg)) == NULL) | ||
| 176 | err(1, NULL); | ||
| 177 | break; | ||
| 178 | case 'z': | ||
| 179 | zflag = 1; | ||
| 180 | break; | ||
| 181 | default: | ||
| 182 | usage(1); | ||
| 183 | } | ||
| 184 | } | ||
| 185 | argc -= optind; | ||
| 186 | argv += optind; | ||
| 187 | |||
| 188 | /* Cruft to make sure options are clean, and used properly. */ | ||
| 189 | if (argv[0] && !argv[1] && family == AF_UNIX) { | ||
| 190 | if (uflag) | ||
| 191 | errx(1, "cannot use -u and -U"); | ||
| 192 | host = argv[0]; | ||
| 193 | uport = NULL; | ||
| 194 | } else if (argv[0] && !argv[1]) { | ||
| 195 | if (!lflag) | ||
| 196 | usage(1); | ||
| 197 | uport = argv[0]; | ||
| 198 | host = NULL; | ||
| 199 | } else if (argv[0] && argv[1]) { | ||
| 200 | host = argv[0]; | ||
| 201 | uport = argv[1]; | ||
| 202 | } else | ||
| 203 | usage(1); | ||
| 204 | |||
| 205 | if (lflag && sflag) | ||
| 206 | errx(1, "cannot use -s and -l"); | ||
| 207 | if (lflag && pflag) | ||
| 208 | errx(1, "cannot use -p and -l"); | ||
| 209 | if (lflag && zflag) | ||
| 210 | errx(1, "cannot use -z and -l"); | ||
| 211 | if (!lflag && kflag) | ||
| 212 | errx(1, "must use -l with -k"); | ||
| 213 | |||
| 214 | /* Initialize addrinfo structure */ | ||
| 215 | if (family != AF_UNIX) { | ||
| 216 | memset(&hints, 0, sizeof(struct addrinfo)); | ||
| 217 | hints.ai_family = family; | ||
| 218 | hints.ai_socktype = uflag ? SOCK_DGRAM : SOCK_STREAM; | ||
| 219 | hints.ai_protocol = uflag ? IPPROTO_UDP : IPPROTO_TCP; | ||
| 220 | if (nflag) | ||
| 221 | hints.ai_flags |= AI_NUMERICHOST; | ||
| 222 | } | ||
| 223 | |||
| 224 | if (xflag) { | ||
| 225 | if (uflag) | ||
| 226 | errx(1, "no proxy support for UDP mode"); | ||
| 227 | |||
| 228 | if (lflag) | ||
| 229 | errx(1, "no proxy support for listen"); | ||
| 230 | |||
| 231 | if (family == AF_UNIX) | ||
| 232 | errx(1, "no proxy support for unix sockets"); | ||
| 233 | |||
| 234 | /* XXX IPv6 transport to proxy would probably work */ | ||
| 235 | if (family == AF_INET6) | ||
| 236 | errx(1, "no proxy support for IPv6"); | ||
| 237 | |||
| 238 | if (sflag) | ||
| 239 | errx(1, "no proxy support for local source address"); | ||
| 240 | |||
| 241 | proxyhost = strsep(&proxy, ":"); | ||
| 242 | proxyport = proxy; | ||
| 243 | |||
| 244 | memset(&proxyhints, 0, sizeof(struct addrinfo)); | ||
| 245 | proxyhints.ai_family = family; | ||
| 246 | proxyhints.ai_socktype = SOCK_STREAM; | ||
| 247 | proxyhints.ai_protocol = IPPROTO_TCP; | ||
| 248 | if (nflag) | ||
| 249 | proxyhints.ai_flags |= AI_NUMERICHOST; | ||
| 250 | } | ||
| 251 | |||
| 252 | if (lflag) { | ||
| 253 | int connfd; | ||
| 254 | ret = 0; | ||
| 255 | |||
| 256 | if (family == AF_UNIX) | ||
| 257 | s = unix_listen(host); | ||
| 258 | |||
| 259 | /* Allow only one connection at a time, but stay alive */ | ||
| 260 | for (;;) { | ||
| 261 | if (family != AF_UNIX) | ||
| 262 | s = local_listen(host, uport, hints); | ||
| 263 | if (s < 0) | ||
| 264 | err(1, NULL); | ||
| 265 | /* | ||
| 266 | * For UDP, we will use recvfrom() initially | ||
| 267 | * to wait for a caller, then use the regular | ||
| 268 | * functions to talk to the caller. | ||
| 269 | */ | ||
| 270 | if (uflag) { | ||
| 271 | int rv; | ||
| 272 | char buf[1024]; | ||
| 273 | struct sockaddr_storage z; | ||
| 274 | |||
| 275 | len = sizeof(z); | ||
| 276 | rv = recvfrom(s, buf, sizeof(buf), MSG_PEEK, | ||
| 277 | (struct sockaddr *)&z, &len); | ||
| 278 | if (rv < 0) | ||
| 279 | err(1, "recvfrom"); | ||
| 280 | |||
| 281 | rv = connect(s, (struct sockaddr *)&z, len); | ||
| 282 | if (rv < 0) | ||
| 283 | err(1, "connect"); | ||
| 284 | |||
| 285 | connfd = s; | ||
| 286 | } else { | ||
| 287 | connfd = accept(s, (struct sockaddr *)&cliaddr, | ||
| 288 | &len); | ||
| 289 | } | ||
| 290 | |||
| 291 | readwrite(connfd); | ||
| 292 | close(connfd); | ||
| 293 | if (family != AF_UNIX) | ||
| 294 | close(s); | ||
| 295 | |||
| 296 | if (!kflag) | ||
| 297 | break; | ||
| 298 | } | ||
| 299 | } else if (family == AF_UNIX) { | ||
| 300 | ret = 0; | ||
| 301 | |||
| 302 | if ((s = unix_connect(host)) > 0 && !zflag) { | ||
| 303 | readwrite(s); | ||
| 304 | close(s); | ||
| 305 | } else | ||
| 306 | ret = 1; | ||
| 307 | |||
| 308 | exit(ret); | ||
| 309 | |||
| 310 | } else { | ||
| 311 | int i = 0; | ||
| 312 | |||
| 313 | /* construct the portlist[] array */ | ||
| 314 | build_ports(uport); | ||
| 315 | |||
| 316 | /* Cycle through portlist, connecting to each port */ | ||
| 317 | for (i = 0; portlist[i] != NULL; i++) { | ||
| 318 | if (s) | ||
| 319 | close(s); | ||
| 320 | |||
| 321 | if (xflag) | ||
| 322 | s = socks_connect(host, portlist[i], hints, | ||
| 323 | proxyhost, proxyport, proxyhints, socksv); | ||
| 324 | else | ||
| 325 | s = remote_connect(host, portlist[i], hints); | ||
| 326 | |||
| 327 | if (s < 0) | ||
| 328 | continue; | ||
| 329 | |||
| 330 | ret = 0; | ||
| 331 | if (vflag || zflag) { | ||
| 332 | /* For UDP, make sure we are connected */ | ||
| 333 | if (uflag) { | ||
| 334 | if (udptest(s) == -1) { | ||
| 335 | ret = 1; | ||
| 336 | continue; | ||
| 337 | } | ||
| 338 | } | ||
| 339 | |||
| 340 | /* Don't lookup port if -n */ | ||
| 341 | if (nflag) | ||
| 342 | sv = NULL; | ||
| 343 | else { | ||
| 344 | sv = getservbyport( | ||
| 345 | ntohs(atoi(portlist[i])), | ||
| 346 | uflag ? "udp" : "tcp"); | ||
| 347 | } | ||
| 348 | |||
| 349 | printf("Connection to %s %s port [%s/%s] succeeded!\n", | ||
| 350 | host, portlist[i], uflag ? "udp" : "tcp", | ||
| 351 | sv ? sv->s_name : "*"); | ||
| 352 | } | ||
| 353 | if (!zflag) | ||
| 354 | readwrite(s); | ||
| 355 | } | ||
| 356 | } | ||
| 357 | |||
| 358 | if (s) | ||
| 359 | close(s); | ||
| 360 | |||
| 361 | exit(ret); | ||
| 362 | } | ||
| 363 | |||
| 364 | /* | ||
| 365 | * unix_connect() | ||
| 366 | * Return's a socket connected to a local unix socket. Return's -1 on failure. | ||
| 367 | */ | ||
| 368 | int | ||
| 369 | unix_connect(char *path) | ||
| 370 | { | ||
| 371 | struct sockaddr_un sun; | ||
| 372 | int s; | ||
| 373 | |||
| 374 | if ((s = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) | ||
| 375 | return (-1); | ||
| 376 | (void)fcntl(s, F_SETFD, 1); | ||
| 377 | |||
| 378 | memset(&sun, 0, sizeof(struct sockaddr_un)); | ||
| 379 | sun.sun_family = AF_UNIX; | ||
| 380 | |||
| 381 | if (strlcpy(sun.sun_path, path, sizeof(sun.sun_path)) >= | ||
| 382 | sizeof(sun.sun_path)) { | ||
| 383 | close(s); | ||
| 384 | errno = ENAMETOOLONG; | ||
| 385 | return (-1); | ||
| 386 | } | ||
| 387 | if (connect(s, (struct sockaddr *)&sun, SUN_LEN(&sun)) < 0) { | ||
| 388 | close(s); | ||
| 389 | return (-1); | ||
| 390 | } | ||
| 391 | return (s); | ||
| 392 | |||
| 393 | } | ||
| 394 | |||
| 395 | /* | ||
| 396 | * unix_listen() | ||
| 397 | * create a unix domain socket, and listen on it. | ||
| 398 | */ | ||
| 399 | int | ||
| 400 | unix_listen(char *path) | ||
| 401 | { | ||
| 402 | struct sockaddr_un sun; | ||
| 403 | int s; | ||
| 404 | |||
| 405 | /* create unix domain socket */ | ||
| 406 | if ((s = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) | ||
| 407 | return (-1); | ||
| 408 | |||
| 409 | memset(&sun, 0, sizeof(struct sockaddr_un)); | ||
| 410 | sun.sun_family = AF_UNIX; | ||
| 411 | |||
| 412 | if (strlcpy(sun.sun_path, path, sizeof(sun.sun_path)) >= | ||
| 413 | sizeof(sun.sun_path)) { | ||
| 414 | close(s); | ||
| 415 | errno = ENAMETOOLONG; | ||
| 416 | return (-1); | ||
| 417 | } | ||
| 418 | |||
| 419 | if (bind(s, (struct sockaddr *)&sun, SUN_LEN(&sun)) < 0) { | ||
| 420 | close(s); | ||
| 421 | return (-1); | ||
| 422 | } | ||
| 423 | |||
| 424 | if (listen(s, 5) < 0) { | ||
| 425 | close(s); | ||
| 426 | return (-1); | ||
| 427 | } | ||
| 428 | return (s); | ||
| 429 | } | ||
| 430 | |||
| 431 | /* | ||
| 432 | * remote_connect() | ||
| 433 | * Return's a socket connected to a remote host. Properly bind's to a local | ||
| 434 | * port or source address if needed. Return's -1 on failure. | ||
| 435 | */ | ||
| 436 | int | ||
| 437 | remote_connect(char *host, char *port, struct addrinfo hints) | ||
| 438 | { | ||
| 439 | struct addrinfo *res, *res0; | ||
| 440 | int s, error; | ||
| 441 | |||
| 442 | if ((error = getaddrinfo(host, port, &hints, &res))) | ||
| 443 | errx(1, "getaddrinfo: %s", gai_strerror(error)); | ||
| 444 | |||
| 445 | res0 = res; | ||
| 446 | do { | ||
| 447 | if ((s = socket(res0->ai_family, res0->ai_socktype, | ||
| 448 | res0->ai_protocol)) < 0) | ||
| 449 | continue; | ||
| 450 | |||
| 451 | /* Bind to a local port or source address if specified */ | ||
| 452 | if (sflag || pflag) { | ||
| 453 | struct addrinfo ahints, *ares; | ||
| 454 | |||
| 455 | if (!(sflag && pflag)) { | ||
| 456 | if (!sflag) | ||
| 457 | sflag = NULL; | ||
| 458 | else | ||
| 459 | pflag = NULL; | ||
| 460 | } | ||
| 461 | |||
| 462 | memset(&ahints, 0, sizeof(struct addrinfo)); | ||
| 463 | ahints.ai_family = res0->ai_family; | ||
| 464 | ahints.ai_socktype = uflag ? SOCK_DGRAM : SOCK_STREAM; | ||
| 465 | ahints.ai_protocol = uflag ? IPPROTO_UDP : IPPROTO_TCP; | ||
| 466 | ahints.ai_flags = AI_PASSIVE; | ||
| 467 | if ((error = getaddrinfo(sflag, pflag, &ahints, &ares))) | ||
| 468 | errx(1, "getaddrinfo: %s", gai_strerror(error)); | ||
| 469 | |||
| 470 | if (bind(s, (struct sockaddr *)ares->ai_addr, | ||
| 471 | ares->ai_addrlen) < 0) | ||
| 472 | errx(1, "bind failed: %s", strerror(errno)); | ||
| 473 | freeaddrinfo(ares); | ||
| 474 | } | ||
| 475 | |||
| 476 | if (connect(s, res0->ai_addr, res0->ai_addrlen) == 0) | ||
| 477 | break; | ||
| 478 | |||
| 479 | close(s); | ||
| 480 | s = -1; | ||
| 481 | } while ((res0 = res0->ai_next) != NULL); | ||
| 482 | |||
| 483 | freeaddrinfo(res); | ||
| 484 | |||
| 485 | return (s); | ||
| 486 | } | ||
| 487 | |||
| 488 | /* | ||
| 489 | * local_listen() | ||
| 490 | * Return's a socket listening on a local port, binds to specified source | ||
| 491 | * address. Return's -1 on failure. | ||
| 492 | */ | ||
| 493 | int | ||
| 494 | local_listen(char *host, char *port, struct addrinfo hints) | ||
| 495 | { | ||
| 496 | struct addrinfo *res, *res0; | ||
| 497 | int s, ret, x = 1; | ||
| 498 | int error; | ||
| 499 | |||
| 500 | /* Allow nodename to be null */ | ||
| 501 | hints.ai_flags |= AI_PASSIVE; | ||
| 502 | |||
| 503 | /* | ||
| 504 | * In the case of binding to a wildcard address | ||
| 505 | * default to binding to an ipv4 address. | ||
| 506 | */ | ||
| 507 | if (host == NULL && hints.ai_family == AF_UNSPEC) | ||
| 508 | hints.ai_family = AF_INET; | ||
| 509 | |||
| 510 | if ((error = getaddrinfo(host, port, &hints, &res))) | ||
| 511 | errx(1, "getaddrinfo: %s", gai_strerror(error)); | ||
| 512 | |||
| 513 | res0 = res; | ||
| 514 | do { | ||
| 515 | if ((s = socket(res0->ai_family, res0->ai_socktype, | ||
| 516 | res0->ai_protocol)) == 0) | ||
| 517 | continue; | ||
| 518 | |||
| 519 | ret = setsockopt(s, SOL_SOCKET, SO_REUSEPORT, &x, sizeof(x)); | ||
| 520 | if (ret == -1) | ||
| 521 | err(1, NULL); | ||
| 522 | |||
| 523 | if (bind(s, (struct sockaddr *)res0->ai_addr, | ||
| 524 | res0->ai_addrlen) == 0) | ||
| 525 | break; | ||
| 526 | |||
| 527 | close(s); | ||
| 528 | s = -1; | ||
| 529 | } while ((res0 = res0->ai_next) != NULL); | ||
| 530 | |||
| 531 | if (!uflag && s != -1) { | ||
| 532 | if (listen(s, 1) < 0) | ||
| 533 | err(1, "listen"); | ||
| 534 | } | ||
| 535 | |||
| 536 | freeaddrinfo(res); | ||
| 537 | |||
| 538 | return (s); | ||
| 539 | } | ||
| 540 | |||
| 541 | /* | ||
| 542 | * readwrite() | ||
| 543 | * Loop that polls on the network file descriptor and stdin. | ||
| 544 | */ | ||
| 545 | void | ||
| 546 | readwrite(int nfd) | ||
| 547 | { | ||
| 548 | struct pollfd pfd[2]; | ||
| 549 | char buf[BUFSIZ]; | ||
| 550 | int wfd = fileno(stdin), n, ret; | ||
| 551 | int lfd = fileno(stdout); | ||
| 552 | |||
| 553 | /* Setup Network FD */ | ||
| 554 | pfd[0].fd = nfd; | ||
| 555 | pfd[0].events = POLLIN; | ||
| 556 | |||
| 557 | /* Setup STDIN FD */ | ||
| 558 | pfd[1].fd = wfd; | ||
| 559 | pfd[1].events = POLLIN; | ||
| 560 | |||
| 561 | while (pfd[0].fd != -1) { | ||
| 562 | if (iflag) | ||
| 563 | sleep(iflag); | ||
| 564 | |||
| 565 | if ((n = poll(pfd, 2, timeout)) < 0) { | ||
| 566 | close(nfd); | ||
| 567 | err(1, "Polling Error"); | ||
| 568 | } | ||
| 569 | |||
| 570 | if (n == 0) | ||
| 571 | return; | ||
| 572 | |||
| 573 | if (pfd[0].revents & POLLIN) { | ||
| 574 | if ((n = read(nfd, buf, sizeof(buf))) < 0) | ||
| 575 | return; | ||
| 576 | else if (n == 0) { | ||
| 577 | shutdown(nfd, SHUT_RD); | ||
| 578 | pfd[0].fd = -1; | ||
| 579 | pfd[0].events = 0; | ||
| 580 | } else { | ||
| 581 | if (tflag) | ||
| 582 | atelnet(nfd, buf, n); | ||
| 583 | if ((ret = atomicio( | ||
| 584 | (ssize_t (*)(int, void *, size_t))write, | ||
| 585 | lfd, buf, n)) != n) | ||
| 586 | return; | ||
| 587 | } | ||
| 588 | } | ||
| 589 | |||
| 590 | if (pfd[1].revents & POLLIN) { | ||
| 591 | if ((n = read(wfd, buf, sizeof(buf))) < 0) | ||
| 592 | return; | ||
| 593 | else if (n == 0) { | ||
| 594 | shutdown(nfd, SHUT_WR); | ||
| 595 | pfd[1].fd = -1; | ||
| 596 | pfd[1].events = 0; | ||
| 597 | } else { | ||
| 598 | if((ret = atomicio( | ||
| 599 | (ssize_t (*)(int, void *, size_t))write, | ||
| 600 | nfd, buf, n)) != n) | ||
| 601 | return; | ||
| 602 | } | ||
| 603 | } | ||
| 604 | } | ||
| 605 | } | ||
| 606 | |||
| 607 | /* Deal with RFC854 WILL/WONT DO/DONT negotiation */ | ||
| 608 | void | ||
| 609 | atelnet(int nfd, unsigned char *buf, unsigned int size) | ||
| 610 | { | ||
| 611 | int ret; | ||
| 612 | unsigned char *p, *end; | ||
| 613 | unsigned char obuf[4]; | ||
| 614 | |||
| 615 | end = buf + size; | ||
| 616 | obuf[0] = '\0'; | ||
| 617 | |||
| 618 | for (p = buf; p < end; p++) { | ||
| 619 | if (*p != IAC) | ||
| 620 | break; | ||
| 621 | |||
| 622 | obuf[0] = IAC; | ||
| 623 | p++; | ||
| 624 | if ((*p == WILL) || (*p == WONT)) | ||
| 625 | obuf[1] = DONT; | ||
| 626 | if ((*p == DO) || (*p == DONT)) | ||
| 627 | obuf[1] = WONT; | ||
| 628 | if (obuf) { | ||
| 629 | p++; | ||
| 630 | obuf[2] = *p; | ||
| 631 | obuf[3] = '\0'; | ||
| 632 | if ((ret = atomicio( | ||
| 633 | (ssize_t (*)(int, void *, size_t))write, | ||
| 634 | nfd, obuf, 3)) != 3) | ||
| 635 | warnx("Write Error!"); | ||
| 636 | obuf[0] = '\0'; | ||
| 637 | } | ||
| 638 | } | ||
| 639 | } | ||
| 640 | |||
| 641 | /* | ||
| 642 | * build_ports() | ||
| 643 | * Build an array or ports in portlist[], listing each port | ||
| 644 | * that we should try to connect too. | ||
| 645 | */ | ||
| 646 | void | ||
| 647 | build_ports(char *p) | ||
| 648 | { | ||
| 649 | char *n, *endp; | ||
| 650 | int hi, lo, cp; | ||
| 651 | int x = 0; | ||
| 652 | |||
| 653 | if ((n = strchr(p, '-')) != NULL) { | ||
| 654 | if (lflag) | ||
| 655 | errx(1, "Cannot use -l with multiple ports!"); | ||
| 656 | |||
| 657 | *n = '\0'; | ||
| 658 | n++; | ||
| 659 | |||
| 660 | /* Make sure the ports are in order: lowest->highest */ | ||
| 661 | hi = (int)strtoul(n, &endp, 10); | ||
| 662 | if (hi <= 0 || hi > PORT_MAX || *endp != '\0') | ||
| 663 | errx(1, "port range not valid"); | ||
| 664 | lo = (int)strtoul(p, &endp, 10); | ||
| 665 | if (lo <= 0 || lo > PORT_MAX || *endp != '\0') | ||
| 666 | errx(1, "port range not valid"); | ||
| 667 | |||
| 668 | if (lo > hi) { | ||
| 669 | cp = hi; | ||
| 670 | hi = lo; | ||
| 671 | lo = cp; | ||
| 672 | } | ||
| 673 | |||
| 674 | /* Load ports sequentially */ | ||
| 675 | for (cp = lo; cp <= hi; cp++) { | ||
| 676 | portlist[x] = calloc(1, PORT_MAX_LEN); | ||
| 677 | if (portlist[x] == NULL) | ||
| 678 | err(1, NULL); | ||
| 679 | snprintf(portlist[x], PORT_MAX_LEN, "%d", cp); | ||
| 680 | x++; | ||
| 681 | } | ||
| 682 | |||
| 683 | /* Randomly swap ports */ | ||
| 684 | if (rflag) { | ||
| 685 | int y; | ||
| 686 | char *c; | ||
| 687 | |||
| 688 | for (x = 0; x <= (hi - lo); x++) { | ||
| 689 | y = (arc4random() & 0xFFFF) % (hi - lo); | ||
| 690 | c = portlist[x]; | ||
| 691 | portlist[x] = portlist[y]; | ||
| 692 | portlist[y] = c; | ||
| 693 | } | ||
| 694 | } | ||
| 695 | } else { | ||
| 696 | hi = (int)strtoul(p, &endp, 10); | ||
| 697 | if (hi <= 0 || hi > PORT_MAX || *endp != '\0') | ||
| 698 | errx(1, "port range not valid"); | ||
| 699 | portlist[0] = calloc(1, PORT_MAX_LEN); | ||
| 700 | if (portlist[0] == NULL) | ||
| 701 | err(1, NULL); | ||
| 702 | portlist[0] = p; | ||
| 703 | } | ||
| 704 | } | ||
| 705 | |||
| 706 | /* | ||
| 707 | * udptest() | ||
| 708 | * Do a few writes to see if the UDP port is there. | ||
| 709 | * XXX - Better way of doing this? Doesn't work for IPv6 | ||
| 710 | * Also fails after around 100 ports checked. | ||
| 711 | */ | ||
| 712 | int | ||
| 713 | udptest(int s) | ||
| 714 | { | ||
| 715 | int i, rv, ret; | ||
| 716 | |||
| 717 | for (i = 0; i <= 3; i++) { | ||
| 718 | if ((rv = write(s, "X", 1)) == 1) | ||
| 719 | ret = 1; | ||
| 720 | else | ||
| 721 | ret = -1; | ||
| 722 | } | ||
| 723 | return (ret); | ||
| 724 | } | ||
| 725 | |||
| 726 | void | ||
| 727 | help(void) | ||
| 728 | { | ||
| 729 | usage(0); | ||
| 730 | fprintf(stderr, "\tCommand Summary:\n\ | ||
| 731 | \t-4 Use IPv4\n\ | ||
| 732 | \t-6 Use IPv6\n\ | ||
| 733 | \t-U Use UNIX domain socket\n\ | ||
| 734 | \t-X vers\t SOCKS version (4 or 5)\n\ | ||
| 735 | \t-h This help text\n\ | ||
| 736 | \t-i secs\t Delay interval for lines sent, ports scanned\n\ | ||
| 737 | \t-k Keep inbound sockets open for multiple connects\n\ | ||
| 738 | \t-l Listen mode, for inbound connects\n\ | ||
| 739 | \t-n Suppress name/port resolutions\n\ | ||
| 740 | \t-p port\t Specify local port for remote connects\n\ | ||
| 741 | \t-r Randomize remote ports\n\ | ||
| 742 | \t-s addr\t Local source address\n\ | ||
| 743 | \t-t Answer TELNET negotiation\n\ | ||
| 744 | \t-u UDP mode\n\ | ||
| 745 | \t-v Verbose\n\ | ||
| 746 | \t-w secs\t Timeout for connects and final net reads\n\ | ||
| 747 | \t-x addr[:port]\tSpecify socks proxy address and port\n\ | ||
| 748 | \t-z Zero-I/O mode [used for scanning]\n\ | ||
| 749 | Port numbers can be individual or ranges: lo-hi [inclusive]\n"); | ||
| 750 | exit(1); | ||
| 751 | } | ||
| 752 | |||
| 753 | void | ||
| 754 | usage(int ret) | ||
| 755 | { | ||
| 756 | fprintf(stderr, "usage: nc [-46Uhklnrtuvz] [-i interval] [-p source port]\n"); | ||
| 757 | fprintf(stderr, "\t [-s ip address] [-w timeout] [-X vers] [-x proxy address [:port]]\n"); | ||
| 758 | fprintf(stderr, "\t [hostname] [port[s...]]\n"); | ||
| 759 | if (ret) | ||
| 760 | exit(1); | ||
| 761 | } | ||
