main.c revision 1.104 1 /* $NetBSD: main.c,v 1.104 2007/04/17 05:52:03 lukem Exp $ */
2
3 /*-
4 * Copyright (c) 1996-2005 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by Luke Mewburn.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 * 3. All advertising materials mentioning features or use of this software
19 * must display the following acknowledgement:
20 * This product includes software developed by the NetBSD
21 * Foundation, Inc. and its contributors.
22 * 4. Neither the name of The NetBSD Foundation nor the names of its
23 * contributors may be used to endorse or promote products derived
24 * from this software without specific prior written permission.
25 *
26 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
27 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
28 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
30 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
31 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
32 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
33 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
34 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
35 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
36 * POSSIBILITY OF SUCH DAMAGE.
37 */
38
39 /*
40 * Copyright (c) 1985, 1989, 1993, 1994
41 * The Regents of the University of California. All rights reserved.
42 *
43 * Redistribution and use in source and binary forms, with or without
44 * modification, are permitted provided that the following conditions
45 * are met:
46 * 1. Redistributions of source code must retain the above copyright
47 * notice, this list of conditions and the following disclaimer.
48 * 2. Redistributions in binary form must reproduce the above copyright
49 * notice, this list of conditions and the following disclaimer in the
50 * documentation and/or other materials provided with the distribution.
51 * 3. Neither the name of the University nor the names of its contributors
52 * may be used to endorse or promote products derived from this software
53 * without specific prior written permission.
54 *
55 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
56 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
57 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
58 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
59 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
60 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
61 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
62 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
63 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
64 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
65 * SUCH DAMAGE.
66 */
67
68 /*
69 * Copyright (C) 1997 and 1998 WIDE Project.
70 * All rights reserved.
71 *
72 * Redistribution and use in source and binary forms, with or without
73 * modification, are permitted provided that the following conditions
74 * are met:
75 * 1. Redistributions of source code must retain the above copyright
76 * notice, this list of conditions and the following disclaimer.
77 * 2. Redistributions in binary form must reproduce the above copyright
78 * notice, this list of conditions and the following disclaimer in the
79 * documentation and/or other materials provided with the distribution.
80 * 3. Neither the name of the project nor the names of its contributors
81 * may be used to endorse or promote products derived from this software
82 * without specific prior written permission.
83 *
84 * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
85 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
86 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
87 * ARE DISCLAIMED. IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
88 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
89 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
90 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
91 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
92 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
93 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
94 * SUCH DAMAGE.
95 */
96
97 #include <sys/cdefs.h>
98 #ifndef lint
99 __COPYRIGHT("@(#) Copyright (c) 1985, 1989, 1993, 1994\n\
100 The Regents of the University of California. All rights reserved.\n");
101 #endif /* not lint */
102
103 #ifndef lint
104 #if 0
105 static char sccsid[] = "@(#)main.c 8.6 (Berkeley) 10/9/94";
106 #else
107 __RCSID("$NetBSD: main.c,v 1.104 2007/04/17 05:52:03 lukem Exp $");
108 #endif
109 #endif /* not lint */
110
111 /*
112 * FTP User Program -- Command Interface.
113 */
114 #include <sys/types.h>
115 #include <sys/socket.h>
116
117 #include <err.h>
118 #include <errno.h>
119 #include <netdb.h>
120 #include <paths.h>
121 #include <pwd.h>
122 #include <signal.h>
123 #include <stdio.h>
124 #include <stdlib.h>
125 #include <string.h>
126 #include <unistd.h>
127 #include <locale.h>
128
129 #define GLOBAL /* force GLOBAL decls in ftp_var.h to be declared */
130 #include "ftp_var.h"
131
132 #define FTP_PROXY "ftp_proxy" /* env var with FTP proxy location */
133 #define HTTP_PROXY "http_proxy" /* env var with HTTP proxy location */
134 #define NO_PROXY "no_proxy" /* env var with list of non-proxied
135 * hosts, comma or space separated */
136
137 static void setupoption(char *, char *, char *);
138 int main(int, char *[]);
139
140 int
141 main(int volatile argc, char **volatile argv)
142 {
143 int ch, rval;
144 struct passwd *pw;
145 char *cp, *ep, *anonuser, *anonpass, *upload_path, *src_addr;
146 int dumbterm, s, isupload;
147 size_t len;
148 socklen_t slen;
149
150 setlocale(LC_ALL, "");
151 setprogname(argv[0]);
152
153 sigint_raised = 0;
154
155 ftpport = "ftp";
156 httpport = "http";
157 gateport = NULL;
158 cp = getenv("FTPSERVERPORT");
159 if (cp != NULL)
160 gateport = cp;
161 else
162 gateport = "ftpgate";
163 doglob = 1;
164 interactive = 1;
165 autologin = 1;
166 passivemode = 1;
167 activefallback = 1;
168 preserve = 1;
169 verbose = 0;
170 progress = 0;
171 gatemode = 0;
172 data = -1;
173 outfile = NULL;
174 restartautofetch = 0;
175 #ifndef NO_EDITCOMPLETE
176 editing = 0;
177 el = NULL;
178 hist = NULL;
179 #endif
180 bytes = 0;
181 mark = HASHBYTES;
182 rate_get = 0;
183 rate_get_incr = DEFAULTINCR;
184 rate_put = 0;
185 rate_put_incr = DEFAULTINCR;
186 #ifdef INET6
187 epsv4 = 1;
188 #else
189 epsv4 = 0;
190 #endif
191 epsv4bad = 0;
192 src_addr = NULL;
193 upload_path = NULL;
194 isupload = 0;
195 reply_callback = NULL;
196 family = AF_UNSPEC;
197
198 netrc[0] = '\0';
199 cp = getenv("NETRC");
200 if (cp != NULL && strlcpy(netrc, cp, sizeof(netrc)) >= sizeof(netrc))
201 errx(1, "$NETRC `%s': %s", cp, strerror(ENAMETOOLONG));
202
203 /*
204 * Get the default socket buffer sizes if we don't already have them.
205 * It doesn't matter which socket we do this to, because on the first
206 * call no socket buffer sizes will have been modified, so we are
207 * guaranteed to get the system defaults.
208 */
209 s = socket(AF_INET, SOCK_STREAM, 0);
210 if (s == -1)
211 err(1, "Can't create socket to determine default socket sizes");
212 slen = sizeof(rcvbuf_size);
213 if (getsockopt(s, SOL_SOCKET, SO_RCVBUF,
214 (void *)&rcvbuf_size, &slen) == -1)
215 err(1, "Unable to get default rcvbuf size");
216 slen = sizeof(sndbuf_size);
217 if (getsockopt(s, SOL_SOCKET, SO_SNDBUF,
218 (void *)&sndbuf_size, &slen) == -1)
219 err(1, "Unable to get default sndbuf size");
220 (void)close(s);
221 /* sanity check returned buffer sizes */
222 if (rcvbuf_size <= 0)
223 rcvbuf_size = 8 * 1024;
224 if (sndbuf_size <= 0)
225 sndbuf_size = 8 * 1024;
226
227 if (sndbuf_size > 8 * 1024 * 1024)
228 sndbuf_size = 8 * 1024 * 1024;
229 if (rcvbuf_size > 8 * 1024 * 1024)
230 rcvbuf_size = 8 * 1024 * 1024;
231
232 marg_sl = ftp_sl_init();
233 if ((tmpdir = getenv("TMPDIR")) == NULL)
234 tmpdir = _PATH_TMP;
235
236 /* Set default operation mode based on FTPMODE environment variable */
237 if ((cp = getenv("FTPMODE")) != NULL) {
238 if (strcasecmp(cp, "passive") == 0) {
239 passivemode = 1;
240 activefallback = 0;
241 } else if (strcasecmp(cp, "active") == 0) {
242 passivemode = 0;
243 activefallback = 0;
244 } else if (strcasecmp(cp, "gate") == 0) {
245 gatemode = 1;
246 } else if (strcasecmp(cp, "auto") == 0) {
247 passivemode = 1;
248 activefallback = 1;
249 } else
250 warnx("Unknown $FTPMODE `%s'; using defaults", cp);
251 }
252
253 if (strcmp(getprogname(), "pftp") == 0) {
254 passivemode = 1;
255 activefallback = 0;
256 } else if (strcmp(getprogname(), "gate-ftp") == 0)
257 gatemode = 1;
258
259 gateserver = getenv("FTPSERVER");
260 if (gateserver == NULL || *gateserver == '\0')
261 gateserver = GATE_SERVER;
262 if (gatemode) {
263 if (*gateserver == '\0') {
264 warnx(
265 "Neither $FTPSERVER nor GATE_SERVER is defined; disabling gate-ftp");
266 gatemode = 0;
267 }
268 }
269
270 cp = getenv("TERM");
271 if (cp == NULL || strcmp(cp, "dumb") == 0)
272 dumbterm = 1;
273 else
274 dumbterm = 0;
275 fromatty = isatty(fileno(stdin));
276 ttyout = stdout;
277 if (isatty(fileno(ttyout))) {
278 verbose = 1; /* verbose if to a tty */
279 if (! dumbterm) {
280 #ifndef NO_EDITCOMPLETE
281 if (fromatty) /* editing mode on if tty is usable */
282 editing = 1;
283 #endif
284 #ifndef NO_PROGRESS
285 if (foregroundproc())
286 progress = 1; /* progress bar on if fg */
287 #endif
288 }
289 }
290
291 while ((ch = getopt(argc, argv, "46AadefginN:o:pP:q:r:Rs:tT:u:vV")) != -1) {
292 switch (ch) {
293 case '4':
294 family = AF_INET;
295 break;
296
297 case '6':
298 #ifdef INET6
299 family = AF_INET6;
300 #else
301 warnx("INET6 support is not available; ignoring -6");
302 #endif
303 break;
304
305 case 'A':
306 activefallback = 0;
307 passivemode = 0;
308 break;
309
310 case 'a':
311 anonftp = 1;
312 break;
313
314 case 'd':
315 options |= SO_DEBUG;
316 ftp_debug++;
317 break;
318
319 case 'e':
320 #ifndef NO_EDITCOMPLETE
321 editing = 0;
322 #endif
323 break;
324
325 case 'f':
326 flushcache = 1;
327 break;
328
329 case 'g':
330 doglob = 0;
331 break;
332
333 case 'i':
334 interactive = 0;
335 break;
336
337 case 'n':
338 autologin = 0;
339 break;
340
341 case 'N':
342 if (strlcpy(netrc, optarg, sizeof(netrc))
343 >= sizeof(netrc))
344 errx(1, "%s: %s", optarg,
345 strerror(ENAMETOOLONG));
346 break;
347
348 case 'o':
349 outfile = optarg;
350 if (strcmp(outfile, "-") == 0)
351 ttyout = stderr;
352 break;
353
354 case 'p':
355 passivemode = 1;
356 activefallback = 0;
357 break;
358
359 case 'P':
360 ftpport = optarg;
361 break;
362
363 case 'q':
364 quit_time = strtol(optarg, &ep, 10);
365 if (quit_time < 1 || *ep != '\0')
366 errx(1, "Bad quit value: %s", optarg);
367 break;
368
369 case 'r':
370 retry_connect = strtol(optarg, &ep, 10);
371 if (retry_connect < 1 || *ep != '\0')
372 errx(1, "Bad retry value: %s", optarg);
373 break;
374
375 case 'R':
376 restartautofetch = 1;
377 break;
378
379 case 's':
380 src_addr = optarg;
381 break;
382
383 case 't':
384 trace = 1;
385 break;
386
387 case 'T':
388 {
389 int targc;
390 char *targv[6], *oac;
391
392 /* look for `dir,max[,incr]' */
393 targc = 0;
394 targv[targc++] = "-T";
395 oac = ftp_strdup(optarg);
396
397 while ((cp = strsep(&oac, ",")) != NULL) {
398 if (*cp == '\0') {
399 warnx("Bad throttle value `%s'",
400 optarg);
401 usage();
402 /* NOTREACHED */
403 }
404 targv[targc++] = cp;
405 if (targc >= 5)
406 break;
407 }
408 if (parserate(targc, targv, 1) == -1)
409 usage();
410 free(oac);
411 break;
412 }
413
414 case 'u':
415 {
416 isupload = 1;
417 interactive = 0;
418 upload_path = ftp_strdup(optarg);
419
420 break;
421 }
422
423 case 'v':
424 progress = verbose = 1;
425 break;
426
427 case 'V':
428 progress = verbose = 0;
429 break;
430
431 default:
432 usage();
433 }
434 }
435 /* set line buffering on ttyout */
436 setvbuf(ttyout, NULL, _IOLBF, 0);
437 argc -= optind;
438 argv += optind;
439
440 cpend = 0; /* no pending replies */
441 proxy = 0; /* proxy not active */
442 crflag = 1; /* strip c.r. on ascii gets */
443 sendport = -1; /* not using ports */
444
445 if (src_addr != NULL) {
446 struct addrinfo hints;
447 int error;
448
449 memset(&hints, 0, sizeof(hints));
450 hints.ai_family = family;
451 hints.ai_socktype = SOCK_STREAM;
452 hints.ai_flags = AI_PASSIVE;
453 error = getaddrinfo(src_addr, NULL, &hints, &bindai);
454 if (error) {
455 errx(1, "Can't lookup `%s': %s", src_addr,
456 (error == EAI_SYSTEM) ? strerror(errno)
457 : gai_strerror(error));
458 }
459 }
460
461 /*
462 * Cache the user name and home directory.
463 */
464 localhome = NULL;
465 localname = NULL;
466 anonuser = "anonymous";
467 cp = getenv("HOME");
468 if (! EMPTYSTRING(cp))
469 localhome = ftp_strdup(cp);
470 pw = NULL;
471 cp = getlogin();
472 if (cp != NULL)
473 pw = getpwnam(cp);
474 if (pw == NULL)
475 pw = getpwuid(getuid());
476 if (pw != NULL) {
477 if (localhome == NULL && !EMPTYSTRING(pw->pw_dir))
478 localhome = ftp_strdup(pw->pw_dir);
479 localname = ftp_strdup(pw->pw_name);
480 anonuser = localname;
481 }
482 if (netrc[0] == '\0' && localhome != NULL) {
483 if (strlcpy(netrc, localhome, sizeof(netrc)) >= sizeof(netrc) ||
484 strlcat(netrc, "/.netrc", sizeof(netrc)) >= sizeof(netrc)) {
485 warnx("%s/.netrc: %s", localhome,
486 strerror(ENAMETOOLONG));
487 netrc[0] = '\0';
488 }
489 }
490 if (localhome == NULL)
491 localhome = ftp_strdup("/");
492
493 /*
494 * Every anonymous FTP server I've encountered will accept the
495 * string "username@", and will append the hostname itself. We
496 * do this by default since many servers are picky about not
497 * having a FQDN in the anonymous password.
498 * - thorpej (at) NetBSD.org
499 */
500 len = strlen(anonuser) + 2;
501 anonpass = ftp_malloc(len);
502 (void)strlcpy(anonpass, anonuser, len);
503 (void)strlcat(anonpass, "@", len);
504
505 /*
506 * set all the defaults for options defined in
507 * struct option optiontab[] declared in cmdtab.c
508 */
509 setupoption("anonpass", getenv("FTPANONPASS"), anonpass);
510 setupoption("ftp_proxy", getenv(FTP_PROXY), "");
511 setupoption("http_proxy", getenv(HTTP_PROXY), "");
512 setupoption("no_proxy", getenv(NO_PROXY), "");
513 setupoption("pager", getenv("PAGER"), DEFAULTPAGER);
514 setupoption("prompt", getenv("FTPPROMPT"), DEFAULTPROMPT);
515 setupoption("rprompt", getenv("FTPRPROMPT"), DEFAULTRPROMPT);
516
517 free(anonpass);
518
519 setttywidth(0);
520 #ifdef SIGINFO
521 (void)xsignal(SIGINFO, psummary);
522 #endif
523 (void)xsignal(SIGQUIT, psummary);
524 (void)xsignal(SIGUSR1, crankrate);
525 (void)xsignal(SIGUSR2, crankrate);
526 (void)xsignal(SIGWINCH, setttywidth);
527
528 if (argc > 0) {
529 if (isupload) {
530 rval = auto_put(argc, argv, upload_path);
531 sigint_or_rval_exit:
532 if (sigint_raised) {
533 (void)xsignal(SIGINT, SIG_DFL);
534 raise(SIGINT);
535 }
536 exit(rval);
537 } else if (strchr(argv[0], ':') != NULL
538 && ! isipv6addr(argv[0])) {
539 rval = auto_fetch(argc, argv);
540 if (rval >= 0) /* -1 == connected and cd-ed */
541 goto sigint_or_rval_exit;
542 } else {
543 char *xargv[4], *user, *host;
544
545 if ((rval = sigsetjmp(toplevel, 1)))
546 goto sigint_or_rval_exit;
547 (void)xsignal(SIGINT, intr);
548 (void)xsignal(SIGPIPE, lostpeer);
549 user = NULL;
550 host = argv[0];
551 cp = strchr(host, '@');
552 if (cp) {
553 *cp = '\0';
554 user = host;
555 host = cp + 1;
556 }
557 /* XXX discards const */
558 xargv[0] = (char *)getprogname();
559 xargv[1] = host;
560 xargv[2] = argv[1];
561 xargv[3] = NULL;
562 do {
563 int oautologin;
564
565 oautologin = autologin;
566 if (user != NULL) {
567 anonftp = 0;
568 autologin = 0;
569 }
570 setpeer(argc+1, xargv);
571 autologin = oautologin;
572 if (connected == 1 && user != NULL)
573 (void)ftp_login(host, user, NULL);
574 if (!retry_connect)
575 break;
576 if (!connected) {
577 macnum = 0;
578 fprintf(ttyout,
579 "Retrying in %d seconds...\n",
580 retry_connect);
581 sleep(retry_connect);
582 }
583 } while (!connected);
584 retry_connect = 0; /* connected, stop hiding msgs */
585 }
586 }
587 if (isupload)
588 usage();
589
590 #ifndef NO_EDITCOMPLETE
591 controlediting();
592 #endif /* !NO_EDITCOMPLETE */
593
594 (void)sigsetjmp(toplevel, 1);
595 (void)xsignal(SIGINT, intr);
596 (void)xsignal(SIGPIPE, lostpeer);
597 for (;;)
598 cmdscanner();
599 }
600
601 /*
602 * Generate a prompt
603 */
604 char *
605 prompt(void)
606 {
607 static char **prompt;
608 static char buf[MAXPATHLEN];
609
610 if (prompt == NULL) {
611 struct option *o;
612
613 o = getoption("prompt");
614 if (o == NULL)
615 errx(1, "prompt: no such option `prompt'");
616 prompt = &(o->value);
617 }
618 formatbuf(buf, sizeof(buf), *prompt ? *prompt : DEFAULTPROMPT);
619 return (buf);
620 }
621
622 /*
623 * Generate an rprompt
624 */
625 char *
626 rprompt(void)
627 {
628 static char **rprompt;
629 static char buf[MAXPATHLEN];
630
631 if (rprompt == NULL) {
632 struct option *o;
633
634 o = getoption("rprompt");
635 if (o == NULL)
636 errx(1, "rprompt: no such option `rprompt'");
637 rprompt = &(o->value);
638 }
639 formatbuf(buf, sizeof(buf), *rprompt ? *rprompt : DEFAULTRPROMPT);
640 return (buf);
641 }
642
643 /*
644 * Command parser.
645 */
646 void
647 cmdscanner(void)
648 {
649 struct cmd *c;
650 char *p;
651 #ifndef NO_EDITCOMPLETE
652 int ch;
653 #endif
654 size_t num;
655
656 for (;;) {
657 #ifndef NO_EDITCOMPLETE
658 if (!editing) {
659 #endif /* !NO_EDITCOMPLETE */
660 if (fromatty) {
661 fputs(prompt(), ttyout);
662 p = rprompt();
663 if (*p)
664 fprintf(ttyout, "%s ", p);
665 }
666 (void)fflush(ttyout);
667 num = getline(stdin, line, sizeof(line), NULL);
668 switch (num) {
669 case -1: /* EOF */
670 case -2: /* error */
671 if (fromatty)
672 putc('\n', ttyout);
673 quit(0, NULL);
674 /* NOTREACHED */
675 case -3: /* too long; try again */
676 fputs("Sorry, input line is too long.\n",
677 ttyout);
678 continue;
679 case 0: /* empty; try again */
680 continue;
681 default: /* all ok */
682 break;
683 }
684 #ifndef NO_EDITCOMPLETE
685 } else {
686 const char *buf;
687 HistEvent ev;
688 cursor_pos = NULL;
689
690 buf = el_gets(el, &ch);
691 num = ch;
692 if (buf == NULL || num == 0) {
693 if (fromatty)
694 putc('\n', ttyout);
695 quit(0, NULL);
696 }
697 if (num >= sizeof(line)) {
698 fputs("Sorry, input line is too long.\n",
699 ttyout);
700 break;
701 }
702 memcpy(line, buf, num);
703 if (line[--num] == '\n') {
704 line[num] = '\0';
705 if (num == 0)
706 break;
707 }
708 history(hist, &ev, H_ENTER, buf);
709 }
710 #endif /* !NO_EDITCOMPLETE */
711
712 makeargv();
713 if (margc == 0)
714 continue;
715 c = getcmd(margv[0]);
716 if (c == (struct cmd *)-1) {
717 fputs("?Ambiguous command.\n", ttyout);
718 continue;
719 }
720 if (c == NULL) {
721 #if !defined(NO_EDITCOMPLETE)
722 /*
723 * attempt to el_parse() unknown commands.
724 * any command containing a ':' would be parsed
725 * as "[prog:]cmd ...", and will result in a
726 * false positive if prog != "ftp", so treat
727 * such commands as invalid.
728 */
729 if (strchr(margv[0], ':') != NULL ||
730 el_parse(el, margc, (const char **)margv) != 0)
731 #endif /* !NO_EDITCOMPLETE */
732 fputs("?Invalid command.\n", ttyout);
733 continue;
734 }
735 if (c->c_conn && !connected) {
736 fputs("Not connected.\n", ttyout);
737 continue;
738 }
739 confirmrest = 0;
740 margv[0] = c->c_name;
741 (*c->c_handler)(margc, margv);
742 if (bell && c->c_bell)
743 (void)putc('\007', ttyout);
744 if (c->c_handler != help)
745 break;
746 }
747 (void)xsignal(SIGINT, intr);
748 (void)xsignal(SIGPIPE, lostpeer);
749 }
750
751 struct cmd *
752 getcmd(const char *name)
753 {
754 const char *p, *q;
755 struct cmd *c, *found;
756 int nmatches, longest;
757
758 if (name == NULL)
759 return (0);
760
761 longest = 0;
762 nmatches = 0;
763 found = 0;
764 for (c = cmdtab; (p = c->c_name) != NULL; c++) {
765 for (q = name; *q == *p++; q++)
766 if (*q == 0) /* exact match? */
767 return (c);
768 if (!*q) { /* the name was a prefix */
769 if (q - name > longest) {
770 longest = q - name;
771 nmatches = 1;
772 found = c;
773 } else if (q - name == longest)
774 nmatches++;
775 }
776 }
777 if (nmatches > 1)
778 return ((struct cmd *)-1);
779 return (found);
780 }
781
782 /*
783 * Slice a string up into argc/argv.
784 */
785
786 int slrflag;
787
788 void
789 makeargv(void)
790 {
791 char *argp;
792
793 stringbase = line; /* scan from first of buffer */
794 argbase = argbuf; /* store from first of buffer */
795 slrflag = 0;
796 marg_sl->sl_cur = 0; /* reset to start of marg_sl */
797 for (margc = 0; ; margc++) {
798 argp = slurpstring();
799 ftp_sl_add(marg_sl, argp);
800 if (argp == NULL)
801 break;
802 }
803 #ifndef NO_EDITCOMPLETE
804 if (cursor_pos == line) {
805 cursor_argc = 0;
806 cursor_argo = 0;
807 } else if (cursor_pos != NULL) {
808 cursor_argc = margc;
809 cursor_argo = strlen(margv[margc-1]);
810 }
811 #endif /* !NO_EDITCOMPLETE */
812 }
813
814 #ifdef NO_EDITCOMPLETE
815 #define INC_CHKCURSOR(x) (x)++
816 #else /* !NO_EDITCOMPLETE */
817 #define INC_CHKCURSOR(x) { (x)++ ; \
818 if (x == cursor_pos) { \
819 cursor_argc = margc; \
820 cursor_argo = ap-argbase; \
821 cursor_pos = NULL; \
822 } }
823
824 #endif /* !NO_EDITCOMPLETE */
825
826 /*
827 * Parse string into argbuf;
828 * implemented with FSM to
829 * handle quoting and strings
830 */
831 char *
832 slurpstring(void)
833 {
834 int got_one = 0;
835 char *sb = stringbase;
836 char *ap = argbase;
837 char *tmp = argbase; /* will return this if token found */
838
839 if (*sb == '!' || *sb == '$') { /* recognize ! as a token for shell */
840 switch (slrflag) { /* and $ as token for macro invoke */
841 case 0:
842 slrflag++;
843 INC_CHKCURSOR(stringbase);
844 return ((*sb == '!') ? "!" : "$");
845 /* NOTREACHED */
846 case 1:
847 slrflag++;
848 altarg = stringbase;
849 break;
850 default:
851 break;
852 }
853 }
854
855 S0:
856 switch (*sb) {
857
858 case '\0':
859 goto OUT;
860
861 case ' ':
862 case '\t':
863 INC_CHKCURSOR(sb);
864 goto S0;
865
866 default:
867 switch (slrflag) {
868 case 0:
869 slrflag++;
870 break;
871 case 1:
872 slrflag++;
873 altarg = sb;
874 break;
875 default:
876 break;
877 }
878 goto S1;
879 }
880
881 S1:
882 switch (*sb) {
883
884 case ' ':
885 case '\t':
886 case '\0':
887 goto OUT; /* end of token */
888
889 case '\\':
890 INC_CHKCURSOR(sb);
891 goto S2; /* slurp next character */
892
893 case '"':
894 INC_CHKCURSOR(sb);
895 goto S3; /* slurp quoted string */
896
897 default:
898 *ap = *sb; /* add character to token */
899 ap++;
900 INC_CHKCURSOR(sb);
901 got_one = 1;
902 goto S1;
903 }
904
905 S2:
906 switch (*sb) {
907
908 case '\0':
909 goto OUT;
910
911 default:
912 *ap = *sb;
913 ap++;
914 INC_CHKCURSOR(sb);
915 got_one = 1;
916 goto S1;
917 }
918
919 S3:
920 switch (*sb) {
921
922 case '\0':
923 goto OUT;
924
925 case '"':
926 INC_CHKCURSOR(sb);
927 goto S1;
928
929 default:
930 *ap = *sb;
931 ap++;
932 INC_CHKCURSOR(sb);
933 got_one = 1;
934 goto S3;
935 }
936
937 OUT:
938 if (got_one)
939 *ap++ = '\0';
940 argbase = ap; /* update storage pointer */
941 stringbase = sb; /* update scan pointer */
942 if (got_one) {
943 return (tmp);
944 }
945 switch (slrflag) {
946 case 0:
947 slrflag++;
948 break;
949 case 1:
950 slrflag++;
951 altarg = NULL;
952 break;
953 default:
954 break;
955 }
956 return (NULL);
957 }
958
959 /*
960 * Help/usage command.
961 * Call each command handler with argc == 0 and argv[0] == name.
962 */
963 void
964 help(int argc, char *argv[])
965 {
966 struct cmd *c;
967 char *nargv[1], *p, *cmd;
968 int isusage;
969
970 cmd = argv[0];
971 isusage = (strcmp(cmd, "usage") == 0);
972 if (argc == 0 || (isusage && argc == 1)) {
973 UPRINTF("usage: %s [command [...]]\n", cmd);
974 return;
975 }
976 if (argc == 1) {
977 StringList *buf;
978
979 buf = ftp_sl_init();
980 fprintf(ttyout,
981 "%sommands may be abbreviated. Commands are:\n\n",
982 proxy ? "Proxy c" : "C");
983 for (c = cmdtab; (p = c->c_name) != NULL; c++)
984 if (!proxy || c->c_proxy)
985 ftp_sl_add(buf, p);
986 list_vertical(buf);
987 sl_free(buf, 0);
988 return;
989 }
990
991 #define HELPINDENT ((int) sizeof("disconnect"))
992
993 while (--argc > 0) {
994 char *arg;
995
996 arg = *++argv;
997 c = getcmd(arg);
998 if (c == (struct cmd *)-1)
999 fprintf(ttyout, "?Ambiguous %s command `%s'\n",
1000 cmd, arg);
1001 else if (c == NULL)
1002 fprintf(ttyout, "?Invalid %s command `%s'\n",
1003 cmd, arg);
1004 else {
1005 if (isusage) {
1006 nargv[0] = c->c_name;
1007 (*c->c_handler)(0, nargv);
1008 } else
1009 fprintf(ttyout, "%-*s\t%s\n", HELPINDENT,
1010 c->c_name, c->c_help);
1011 }
1012 }
1013 }
1014
1015 struct option *
1016 getoption(const char *name)
1017 {
1018 const char *p;
1019 struct option *c;
1020
1021 if (name == NULL)
1022 return (NULL);
1023 for (c = optiontab; (p = c->name) != NULL; c++) {
1024 if (strcasecmp(p, name) == 0)
1025 return (c);
1026 }
1027 return (NULL);
1028 }
1029
1030 char *
1031 getoptionvalue(const char *name)
1032 {
1033 struct option *c;
1034
1035 if (name == NULL)
1036 errx(1, "getoptionvalue: invoked with NULL name");
1037 c = getoption(name);
1038 if (c != NULL)
1039 return (c->value);
1040 errx(1, "getoptionvalue: invoked with unknown option `%s'", name);
1041 /* NOTREACHED */
1042 }
1043
1044 static void
1045 setupoption(char *name, char *value, char *defaultvalue)
1046 {
1047 char *nargv[3];
1048 int overbose;
1049
1050 nargv[0] = "setupoption()";
1051 nargv[1] = name;
1052 nargv[2] = (value ? value : defaultvalue);
1053 overbose = verbose;
1054 verbose = 0;
1055 setoption(3, nargv);
1056 verbose = overbose;
1057 }
1058
1059 void
1060 usage(void)
1061 {
1062 const char *progname = getprogname();
1063
1064 (void)fprintf(stderr,
1065 "usage: %s [-46AadefginpRtvV] [-N netrc] [-o outfile] [-P port] [-q quittime]\n"
1066 " [-r retry] [-s srcaddr] [-T dir,max[,inc]]\n"
1067 " [[user@]host [port]] [host:path[/]] [file:///file]\n"
1068 " [ftp://[user[:pass]@]host[:port]/path[/]]\n"
1069 " [http://[user[:pass]@]host[:port]/path] [...]\n"
1070 " %s -u URL file [...]\n", progname, progname);
1071 exit(1);
1072 }
1073