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