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