main.c revision 1.30.2.1
1 /* $NetBSD: main.c,v 1.30.2.1 2012/07/20 23:14:23 riz Exp $ */ 2 3 /* 4 * Copyright (c) 1983, 1993 5 * The Regents of the University of California. All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 3. Neither the name of the University nor the names of its contributors 16 * may be used to endorse or promote products derived from this software 17 * without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 22 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 29 * SUCH DAMAGE. 30 */ 31 32 #include <sys/cdefs.h> 33 #ifndef lint 34 __COPYRIGHT("@(#) Copyright (c) 1983, 1993\ 35 The Regents of the University of California. All rights reserved."); 36 #if 0 37 static char sccsid[] = "@(#)main.c 8.1 (Berkeley) 6/6/93"; 38 #else 39 __RCSID("$NetBSD: main.c,v 1.30.2.1 2012/07/20 23:14:23 riz Exp $"); 40 #endif 41 #endif /* not lint */ 42 43 /* Many bug fixes are from Jim Guyton <guyton@rand-unix> */ 44 45 /* 46 * TFTP User Program -- Command Interface. 47 */ 48 #include <sys/types.h> 49 #include <sys/param.h> 50 #include <sys/socket.h> 51 52 #include <netinet/in.h> 53 54 #include <arpa/inet.h> 55 #include <arpa/tftp.h> 56 57 #include <ctype.h> 58 #include <fcntl.h> 59 #include <err.h> 60 #include <errno.h> 61 #include <netdb.h> 62 #include <setjmp.h> 63 #include <signal.h> 64 #include <stdio.h> 65 #include <stdlib.h> 66 #include <string.h> 67 #include <unistd.h> 68 69 #include "extern.h" 70 71 #define TIMEOUT 5 /* secs between rexmt's */ 72 #define LBUFLEN 200 /* size of input buffer */ 73 74 struct sockaddr_storage peeraddr; 75 int f; 76 int mf; 77 int trace; 78 int verbose; 79 int tsize=0; 80 int tout=0; 81 size_t def_blksize = SEGSIZE; 82 size_t blksize = SEGSIZE; 83 in_addr_t mcaddr = INADDR_NONE; 84 uint16_t mcport; 85 u_int def_rexmtval = TIMEOUT; 86 u_int rexmtval = TIMEOUT; 87 ushort mcmasterslave; 88 int maxtimeout = 5 * TIMEOUT; 89 90 jmp_buf toplevel; 91 92 static int connected; 93 static char mode[32]; 94 static char line[LBUFLEN]; 95 static int margc; 96 static char *margv[20]; 97 static const char *prompt = "tftp"; 98 static char hostname[MAXHOSTNAMELEN]; 99 100 static void get(int, char **); 101 static void help(int, char **); 102 static void modecmd(int, char **); 103 static void put(int, char **); 104 static __dead void quit(int, char **); 105 static void setascii(int, char **); 106 static void setbinary(int, char **); 107 static void setpeer0(const char *, const char *); 108 static void setpeer(int, char **); 109 static void setrexmt(int, char **); 110 static void settimeout(int, char **); 111 static void settrace(int, char **); 112 static void setverbose(int, char **); 113 static void setblksize(int, char **); 114 static void settsize(int, char **); 115 static void settimeoutopt(int, char **); 116 static void status(int, char **); 117 static char *tail(char *); 118 static __dead void intr(int); 119 static const struct cmd *getcmd(const char *); 120 121 static __dead void command(void); 122 123 static void getUsage(char *); 124 static void makeargv(void); 125 static void putUsage(const char *); 126 static void settftpmode(const char *); 127 128 #define HELPINDENT sizeof("connect") 129 130 struct cmd { 131 const char *name; 132 const char *help; 133 void (*handler)(int, char **); 134 }; 135 136 static const char vhelp[] = "toggle verbose mode"; 137 static const char thelp[] = "toggle packet tracing"; 138 static const char tshelp[] = "toggle extended tsize option"; 139 static const char tohelp[] = "toggle extended timeout option"; 140 static const char blhelp[] = "set an alternative blocksize (def. 512)"; 141 static const char chelp[] = "connect to remote tftp"; 142 static const char qhelp[] = "exit tftp"; 143 static const char hhelp[] = "print help information"; 144 static const char shelp[] = "send file"; 145 static const char rhelp[] = "receive file"; 146 static const char mhelp[] = "set file transfer mode"; 147 static const char sthelp[] = "show current status"; 148 static const char xhelp[] = "set per-packet retransmission timeout"; 149 static const char ihelp[] = "set total retransmission timeout"; 150 static const char ashelp[] = "set mode to netascii"; 151 static const char bnhelp[] = "set mode to octet"; 152 153 static const struct cmd cmdtab[] = { 154 { "connect", chelp, setpeer }, 155 { "mode", mhelp, modecmd }, 156 { "put", shelp, put }, 157 { "get", rhelp, get }, 158 { "quit", qhelp, quit }, 159 { "verbose", vhelp, setverbose }, 160 { "blksize", blhelp, setblksize }, 161 { "tsize", tshelp, settsize }, 162 { "trace", thelp, settrace }, 163 { "status", sthelp, status }, 164 { "binary", bnhelp, setbinary }, 165 { "ascii", ashelp, setascii }, 166 { "rexmt", xhelp, setrexmt }, 167 { "timeout", ihelp, settimeout }, 168 { "tout", tohelp, settimeoutopt }, 169 { "?", hhelp, help }, 170 { .name = NULL } 171 }; 172 173 static struct modes { 174 const char *m_name; 175 const char *m_mode; 176 } modes[] = { 177 { "ascii", "netascii" }, 178 { "netascii", "netascii" }, 179 { "binary", "octet" }, 180 { "image", "octet" }, 181 { "octet", "octet" }, 182 /* { "mail", "mail" }, */ 183 { 0, 0 } 184 }; 185 186 int 187 main(int argc, char *argv[]) 188 { 189 int c; 190 191 f = mf = -1; 192 (void)strlcpy(mode, "netascii", sizeof(mode)); 193 (void)signal(SIGINT, intr); 194 195 setprogname(argv[0]); 196 while ((c = getopt(argc, argv, "e")) != -1) { 197 switch (c) { 198 case 'e': 199 blksize = MAXSEGSIZE; 200 (void)strlcpy(mode, "octet", sizeof(mode)); 201 tsize = 1; 202 tout = 1; 203 break; 204 default: 205 (void)fprintf(stderr, 206 "Usage: %s [-e] host-name [port]\n", getprogname()); 207 exit(1); 208 } 209 } 210 argc -= optind; 211 argv += optind; 212 213 if (argc >= 1) { 214 if (setjmp(toplevel) != 0) 215 exit(0); 216 argc++; 217 argv--; 218 setpeer(argc, argv); 219 } 220 if (setjmp(toplevel) != 0) 221 (void)putchar('\n'); 222 command(); 223 return 0; 224 } 225 226 static void 227 getmore(const char *cmd, const char *prm) 228 { 229 (void)strlcpy(line, cmd, sizeof(line)); 230 (void)printf("%s", prm); 231 (void)fgets(&line[strlen(line)], (int)(LBUFLEN-strlen(line)), stdin); 232 makeargv(); 233 } 234 235 static void 236 setpeer0(const char *host, const char *port) 237 { 238 struct addrinfo hints, *res0, *res; 239 int error, soopt; 240 struct sockaddr_storage ss; 241 const char *cause = "unknown"; 242 243 if (connected) { 244 (void)close(f); 245 f = -1; 246 } 247 connected = 0; 248 249 (void)memset(&hints, 0, sizeof(hints)); 250 hints.ai_family = PF_UNSPEC; 251 hints.ai_socktype = SOCK_DGRAM; 252 hints.ai_protocol = IPPROTO_UDP; 253 hints.ai_flags = AI_CANONNAME; 254 if (!port) 255 port = "tftp"; 256 error = getaddrinfo(host, port, &hints, &res0); 257 if (error) { 258 warnx("%s", gai_strerror(error)); 259 return; 260 } 261 262 for (res = res0; res; res = res->ai_next) { 263 if (res->ai_addrlen > sizeof(peeraddr)) 264 continue; 265 f = socket(res->ai_family, res->ai_socktype, res->ai_protocol); 266 if (f == -1) { 267 cause = "socket"; 268 continue; 269 } 270 271 (void)memset(&ss, 0, sizeof(ss)); 272 ss.ss_family = res->ai_family; 273 ss.ss_len = res->ai_addrlen; 274 if (bind(f, (struct sockaddr *)(void *)&ss, 275 (socklen_t)ss.ss_len) == -1) { 276 cause = "bind"; 277 (void)close(f); 278 f = -1; 279 continue; 280 } 281 282 break; 283 } 284 285 if (f >= 0) { 286 soopt = 65536; 287 if (setsockopt(f, SOL_SOCKET, SO_SNDBUF, &soopt, sizeof(soopt)) 288 == -1) { 289 (void)close(f); 290 f = -1; 291 cause = "setsockopt SNDBUF"; 292 } 293 else if (setsockopt(f, SOL_SOCKET, SO_RCVBUF, &soopt, 294 sizeof(soopt)) == -1) { 295 (void)close(f); 296 f = -1; 297 cause = "setsockopt RCVBUF"; 298 } 299 } 300 301 if (f == -1 || res == NULL) 302 warn("%s", cause); 303 else { 304 /* res->ai_addr <= sizeof(peeraddr) is guaranteed */ 305 (void)memcpy(&peeraddr, res->ai_addr, res->ai_addrlen); 306 if (res->ai_canonname) { 307 (void)strlcpy(hostname, res->ai_canonname, 308 sizeof(hostname)); 309 } else 310 (void)strlcpy(hostname, host, sizeof(hostname)); 311 connected = 1; 312 } 313 314 freeaddrinfo(res0); 315 } 316 317 static void 318 setpeer(int argc, char *argv[]) 319 { 320 321 if (argc < 2) { 322 getmore("Connect ", "(to) "); 323 argc = margc; 324 argv = margv; 325 } 326 if (argc < 2 || argc > 3) { 327 (void)printf("Usage: %s [-e] host-name [port]\n", 328 getprogname()); 329 return; 330 } 331 if (argc == 2) 332 setpeer0(argv[1], NULL); 333 else 334 setpeer0(argv[1], argv[2]); 335 } 336 337 static void 338 modecmd(int argc, char *argv[]) 339 { 340 struct modes *p; 341 const char *sep; 342 343 if (argc < 2) { 344 (void)printf("Using %s mode to transfer files.\n", mode); 345 return; 346 } 347 if (argc == 2) { 348 for (p = modes; p->m_name; p++) 349 if (strcmp(argv[1], p->m_name) == 0) 350 break; 351 if (p->m_name) { 352 settftpmode(p->m_mode); 353 return; 354 } 355 (void)printf("%s: unknown mode\n", argv[1]); 356 /* drop through and print Usage message */ 357 } 358 359 (void)printf("Usage: %s [", argv[0]); 360 sep = " "; 361 for (p = modes; p->m_name; p++) { 362 (void)printf("%s%s", sep, p->m_name); 363 if (*sep == ' ') 364 sep = " | "; 365 } 366 (void)printf(" ]\n"); 367 return; 368 } 369 370 static void 371 /*ARGSUSED*/ 372 setbinary(int argc, char *argv[]) 373 { 374 375 settftpmode("octet"); 376 } 377 378 static void 379 /*ARGSUSED*/ 380 setascii(int argc, char *argv[]) 381 { 382 383 settftpmode("netascii"); 384 } 385 386 static void 387 settftpmode(const char *newmode) 388 { 389 (void)strlcpy(mode, newmode, sizeof(mode)); 390 if (verbose) 391 (void)printf("mode set to %s\n", mode); 392 } 393 394 395 /* 396 * Send file(s). 397 */ 398 static void 399 put(int argc, char *argv[]) 400 { 401 int fd; 402 int n; 403 char *targ, *p; 404 405 if (argc < 2) { 406 getmore("send ", "(file) "); 407 argc = margc; 408 argv = margv; 409 } 410 if (argc < 2) { 411 putUsage(argv[0]); 412 return; 413 } 414 targ = argv[argc - 1]; 415 if (strrchr(argv[argc - 1], ':')) { 416 char *cp; 417 418 for (n = 1; n < argc - 1; n++) 419 if (strchr(argv[n], ':')) { 420 putUsage(argv[0]); 421 return; 422 } 423 cp = argv[argc - 1]; 424 targ = strrchr(cp, ':'); 425 *targ++ = 0; 426 if (cp[0] == '[' && cp[strlen(cp) - 1] == ']') { 427 cp[strlen(cp) - 1] = '\0'; 428 cp++; 429 } 430 setpeer0(cp, NULL); 431 } 432 if (!connected) { 433 (void)printf("No target machine specified.\n"); 434 return; 435 } 436 if (argc < 4) { 437 char *cp = argc == 2 ? tail(targ) : argv[1]; 438 fd = open(cp, O_RDONLY); 439 if (fd == -1) { 440 warn("%s", cp); 441 return; 442 } 443 if (verbose) 444 (void)printf("putting %s to %s:%s [%s]\n", 445 cp, hostname, targ, mode); 446 sendfile(fd, targ, mode); 447 return; 448 } 449 /* this assumes the target is a directory */ 450 /* on a remote unix system. hmmmm. */ 451 p = strchr(targ, '\0'); 452 *p++ = '/'; 453 for (n = 1; n < argc - 1; n++) { 454 (void)strcpy(p, tail(argv[n])); 455 fd = open(argv[n], O_RDONLY); 456 if (fd == -1) { 457 warn("%s", argv[n]); 458 continue; 459 } 460 if (verbose) 461 (void)printf("putting %s to %s:%s [%s]\n", 462 argv[n], hostname, targ, mode); 463 sendfile(fd, targ, mode); 464 } 465 } 466 467 static void 468 putUsage(const char *s) 469 { 470 (void)printf("Usage: %s file ... host:target, or\n", s); 471 (void)printf(" %s file ... target (when already connected)\n", s); 472 } 473 474 /* 475 * Receive file(s). 476 */ 477 static void 478 get(int argc, char *argv[]) 479 { 480 int fd; 481 int n; 482 char *p; 483 char *src; 484 485 if (argc < 2) { 486 getmore("get ", "(files) "); 487 argc = margc; 488 argv = margv; 489 } 490 if (argc < 2) { 491 getUsage(argv[0]); 492 return; 493 } 494 if (!connected) { 495 for (n = 1; n < argc ; n++) 496 if (strrchr(argv[n], ':') == 0) { 497 getUsage(argv[0]); 498 return; 499 } 500 } 501 for (n = 1; n < argc ; n++) { 502 src = strrchr(argv[n], ':'); 503 if (src == NULL) 504 src = argv[n]; 505 else { 506 char *cp; 507 *src++ = 0; 508 cp = argv[n]; 509 if (cp[0] == '[' && cp[strlen(cp) - 1] == ']') { 510 cp[strlen(cp) - 1] = '\0'; 511 cp++; 512 } 513 setpeer0(cp, NULL); 514 if (!connected) 515 continue; 516 } 517 if (argc < 4) { 518 char *cp = argc == 3 ? argv[2] : tail(src); 519 fd = creat(cp, 0644); 520 if (fd == -1) { 521 warn("%s", cp); 522 return; 523 } 524 if (verbose) 525 (void)printf("getting from %s:%s to %s [%s]\n", 526 hostname, src, cp, mode); 527 recvfile(fd, src, mode); 528 break; 529 } 530 p = tail(src); /* new .. jdg */ 531 fd = creat(p, 0644); 532 if (fd == -1) { 533 warn("%s", p); 534 continue; 535 } 536 if (verbose) 537 (void)printf("getting from %s:%s to %s [%s]\n", 538 hostname, src, p, mode); 539 recvfile(fd, src, mode); 540 } 541 } 542 543 static void 544 getUsage(s) 545 char *s; 546 { 547 (void)printf("Usage: %s host:file host:file ... file, or\n", s); 548 (void)printf(" %s file file ... file if connected\n", s); 549 } 550 551 void 552 setblksize(argc, argv) 553 int argc; 554 char *argv[]; 555 { 556 int t; 557 558 if (argc < 2) { 559 getmore("blksize ", "(blksize) "); 560 argc = margc; 561 argv = margv; 562 } 563 if (argc != 2) { 564 (void)printf("Usage: %s value\n", argv[0]); 565 return; 566 } 567 t = atoi(argv[1]); 568 if (t < 8 || t > 65464) 569 (void)printf("%s: bad value\n", argv[1]); 570 else 571 blksize = t; 572 } 573 574 static void 575 setrexmt(int argc, char *argv[]) 576 { 577 int t; 578 579 if (argc < 2) { 580 getmore("Rexmt-timeout ", "(value) "); 581 argc = margc; 582 argv = margv; 583 } 584 if (argc != 2) { 585 (void)printf("Usage: %s value\n", argv[0]); 586 return; 587 } 588 t = atoi(argv[1]); 589 if (t < 0) 590 (void)printf("%s: bad value\n", argv[1]); 591 else 592 rexmtval = t; 593 } 594 595 static void 596 settimeout(int argc, char *argv[]) 597 { 598 int t; 599 600 if (argc < 2) { 601 getmore("Maximum-timeout ", "(value) "); 602 argc = margc; 603 argv = margv; 604 } 605 if (argc != 2) { 606 (void)printf("Usage: %s value\n", argv[0]); 607 return; 608 } 609 t = atoi(argv[1]); 610 if (t < 0) 611 (void)printf("%s: bad value\n", argv[1]); 612 else 613 maxtimeout = t; 614 } 615 616 static void 617 /*ARGSUSED*/ 618 status(int argc, char *argv[]) 619 { 620 if (connected) 621 (void)printf("Connected to %s.\n", hostname); 622 else 623 (void)printf("Not connected.\n"); 624 (void)printf("Mode: %s Verbose: %s Tracing: %s\n", mode, 625 verbose ? "on" : "off", trace ? "on" : "off"); 626 (void)printf("Rexmt-interval: %d seconds, Max-timeout: %d seconds\n", 627 rexmtval, maxtimeout); 628 } 629 630 static void 631 /*ARGSUSED*/ 632 intr(int dummy) 633 { 634 635 (void)signal(SIGALRM, SIG_IGN); 636 (void)alarm(0); 637 longjmp(toplevel, -1); 638 } 639 640 static char * 641 tail(char *filename) 642 { 643 char *s; 644 645 while (*filename) { 646 s = strrchr(filename, '/'); 647 if (s == NULL) 648 break; 649 if (s[1]) 650 return s + 1; 651 *s = '\0'; 652 } 653 return filename; 654 } 655 656 /* 657 * Command parser. 658 */ 659 static __dead void 660 command(void) 661 { 662 const struct cmd *c; 663 664 for (;;) { 665 (void)printf("%s> ", prompt); 666 if (fgets(line, LBUFLEN, stdin) == NULL) { 667 if (feof(stdin)) { 668 exit(0); 669 } else { 670 continue; 671 } 672 } 673 if (line[0] == '\0' || line[0] == '\n') 674 continue; 675 makeargv(); 676 if (margc == 0) 677 continue; 678 c = getcmd(margv[0]); 679 if (c == (struct cmd *)-1) { 680 (void)printf("?Ambiguous command\n"); 681 continue; 682 } 683 if (c == 0) { 684 (void)printf("?Invalid command\n"); 685 continue; 686 } 687 (*c->handler)(margc, margv); 688 } 689 } 690 691 static const struct cmd * 692 getcmd(const char *name) 693 { 694 const char *p, *q; 695 const struct cmd *c, *found; 696 int nmatches, longest; 697 698 longest = 0; 699 nmatches = 0; 700 found = 0; 701 for (c = cmdtab; (p = c->name) != NULL; c++) { 702 for (q = name; *q == *p++; q++) 703 if (*q == 0) /* exact match? */ 704 return c; 705 if (!*q) { /* the name was a prefix */ 706 if (q - name > longest) { 707 longest = q - name; 708 nmatches = 1; 709 found = c; 710 } else if (q - name == longest) 711 nmatches++; 712 } 713 } 714 if (nmatches > 1) 715 return (struct cmd *)-1; 716 return found; 717 } 718 719 /* 720 * Slice a string up into argc/argv. 721 */ 722 static void 723 makeargv(void) 724 { 725 char *cp; 726 char **argp = margv; 727 728 margc = 0; 729 for (cp = line; *cp;) { 730 while (isspace((unsigned char)*cp)) 731 cp++; 732 if (*cp == '\0') 733 break; 734 *argp++ = cp; 735 margc += 1; 736 while (*cp != '\0' && !isspace((unsigned char)*cp)) 737 cp++; 738 if (*cp == '\0') 739 break; 740 *cp++ = '\0'; 741 } 742 *argp++ = NULL; 743 } 744 745 static void 746 /*ARGSUSED*/ 747 quit(int argc, char *argv[]) 748 { 749 750 exit(0); 751 } 752 753 /* 754 * Help command. 755 */ 756 static void 757 help(int argc, char *argv[]) 758 { 759 const struct cmd *c; 760 761 if (argc == 1) { 762 (void)printf("Commands may be abbreviated. Commands are:\n\n"); 763 for (c = cmdtab; c->name; c++) 764 (void)printf("%-*s\t%s\n", (int)HELPINDENT, c->name, 765 c->help); 766 return; 767 } 768 while (--argc > 0) { 769 char *arg; 770 arg = *++argv; 771 c = getcmd(arg); 772 if (c == (struct cmd *)-1) 773 (void)printf("?Ambiguous help command %s\n", arg); 774 else if (c == NULL) 775 (void)printf("?Invalid help command %s\n", arg); 776 else 777 (void)printf("%s\n", c->help); 778 } 779 } 780 781 static void 782 /*ARGSUSED*/ 783 settrace(int argc, char **argv) 784 { 785 trace = !trace; 786 (void)printf("Packet tracing %s.\n", trace ? "on" : "off"); 787 } 788 789 static void 790 /*ARGSUSED*/ 791 setverbose(int argc, char **argv) 792 { 793 verbose = !verbose; 794 (void)printf("Verbose mode %s.\n", verbose ? "on" : "off"); 795 } 796 797 static void 798 /*ARGSUSED*/ 799 settsize(int argc, char **argv) 800 { 801 tsize = !tsize; 802 (void)printf("Tsize mode %s.\n", tsize ? "on" : "off"); 803 } 804 805 static void 806 /*ARGSUSED*/ 807 settimeoutopt(int argc, char **argv) 808 { 809 tout = !tout; 810 (void)printf("Timeout option %s.\n", tout ? "on" : "off"); 811 } 812
Indexes created Sun Oct 12 09:09:55 GMT 2025