NetBSD/usr.bin/ftp/main.c

696 lines
13 KiB
C
Raw Normal View History

/* $NetBSD: main.c,v 1.15 1996/12/29 04:05:32 lukem Exp $ */
1993-03-21 12:45:37 +03:00
/*
1994-08-25 07:47:50 +04:00
* Copyright (c) 1985, 1989, 1993, 1994
* The Regents of the University of California. All rights reserved.
1993-03-21 12:45:37 +03:00
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software developed by the University of
* California, Berkeley and its contributors.
* 4. Neither the name of the University nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#ifndef lint
1994-08-25 07:47:50 +04:00
static char copyright[] =
"@(#) Copyright (c) 1985, 1989, 1993, 1994\n\
The Regents of the University of California. All rights reserved.\n";
1993-03-21 12:45:37 +03:00
#endif /* not lint */
#ifndef lint
1995-09-08 05:05:59 +04:00
#if 0
static char sccsid[] = "@(#)main.c 8.6 (Berkeley) 10/9/94";
#else
static char rcsid[] = "$NetBSD: main.c,v 1.15 1996/12/29 04:05:32 lukem Exp $";
1995-09-08 05:05:59 +04:00
#endif
1993-03-21 12:45:37 +03:00
#endif /* not lint */
/*
* FTP User Program -- Command Interface.
*/
1994-08-25 07:47:50 +04:00
/*#include <sys/ioctl.h>*/
1993-03-21 12:45:37 +03:00
#include <sys/types.h>
1994-08-25 07:47:50 +04:00
#include <sys/socket.h>
1993-03-21 12:45:37 +03:00
#include <arpa/ftp.h>
#include <ctype.h>
1994-08-25 07:47:50 +04:00
#include <err.h>
1993-03-21 12:45:37 +03:00
#include <netdb.h>
#include <pwd.h>
1994-08-25 07:47:50 +04:00
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
1995-09-14 05:39:52 +04:00
#include <string.h>
1994-08-25 07:47:50 +04:00
#include <unistd.h>
1993-03-21 12:45:37 +03:00
1994-08-25 07:47:50 +04:00
#include "ftp_var.h"
1993-03-21 12:45:37 +03:00
extern char *__progname; /* from crt0.o */
char *portnum; /* alternate port number */
int auto_fetch __P((int, char **));
1994-08-25 07:47:50 +04:00
int
1993-03-21 12:45:37 +03:00
main(argc, argv)
1994-08-25 07:47:50 +04:00
int argc;
1993-03-21 12:45:37 +03:00
char *argv[];
{
1994-08-25 07:47:50 +04:00
int ch, top;
1993-03-21 12:45:37 +03:00
struct passwd *pw = NULL;
1994-08-25 07:47:50 +04:00
char *cp, homedir[MAXPATHLEN];
1993-03-21 12:45:37 +03:00
sp = getservbyname("ftp", "tcp");
1994-08-25 07:47:50 +04:00
if (sp == 0)
errx(1, "ftp/tcp: unknown service");
1993-03-21 12:45:37 +03:00
doglob = 1;
interactive = 1;
autologin = 1;
passivemode = 0;
preserve = 1;
mark = HASHBYTES;
1994-08-25 07:47:50 +04:00
cp = strrchr(argv[0], '/');
cp = (cp == NULL) ? argv[0] : cp + 1;
if (strcmp(cp, "pftp") == 0)
passivemode = 1;
while ((ch = getopt(argc, argv, "adginpP:tv")) != EOF) {
switch (ch) {
case 'a':
anonftp = 1;
break;
1994-08-25 07:47:50 +04:00
case 'd':
options |= SO_DEBUG;
debug++;
break;
1994-08-25 07:47:50 +04:00
case 'g':
doglob = 0;
break;
1993-03-21 12:45:37 +03:00
1994-08-25 07:47:50 +04:00
case 'i':
interactive = 0;
break;
1993-03-21 12:45:37 +03:00
1994-08-25 07:47:50 +04:00
case 'n':
autologin = 0;
break;
1993-03-21 12:45:37 +03:00
case 'p':
passivemode = 1;
break;
case 'P':
portnum = optarg;
break;
1994-08-25 07:47:50 +04:00
case 't':
trace++;
break;
1993-03-21 12:45:37 +03:00
1994-08-25 07:47:50 +04:00
case 'v':
verbose++;
break;
1993-03-21 12:45:37 +03:00
1994-08-25 07:47:50 +04:00
default:
usage();
1994-08-25 07:47:50 +04:00
}
1993-03-21 12:45:37 +03:00
}
1994-08-25 07:47:50 +04:00
argc -= optind;
argv += optind;
1993-03-21 12:45:37 +03:00
fromatty = isatty(fileno(stdin));
if (fromatty)
verbose++;
cpend = 0; /* no pending replies */
proxy = 0; /* proxy not active */
crflag = 1; /* strip c.r. on ascii gets */
sendport = -1; /* not using ports */
/*
* Set up the home directory in case we're globbing.
*/
cp = getlogin();
if (cp != NULL) {
pw = getpwnam(cp);
}
if (pw == NULL)
pw = getpwuid(getuid());
if (pw != NULL) {
home = homedir;
(void) strcpy(home, pw->pw_dir);
}
/* Handle "automatic" transfers. */
if (argc > 0 && strchr(argv[0], ':') != NULL) {
anonftp = 1;
exit(auto_fetch(argc, argv));
}
1993-03-21 12:45:37 +03:00
if (argc > 0) {
1994-08-25 07:47:50 +04:00
char *xargv[5];
1993-03-21 12:45:37 +03:00
if (setjmp(toplevel))
exit(0);
(void) signal(SIGINT, intr);
(void) signal(SIGPIPE, lostpeer);
1994-08-25 07:47:50 +04:00
xargv[0] = __progname;
xargv[1] = argv[0];
xargv[2] = argv[1];
xargv[3] = argv[2];
xargv[4] = NULL;
setpeer(argc+1, xargv);
1993-03-21 12:45:37 +03:00
}
top = setjmp(toplevel) == 0;
if (top) {
(void) signal(SIGINT, intr);
(void) signal(SIGPIPE, lostpeer);
}
for (;;) {
cmdscanner(top);
top = 1;
}
}
void
intr()
{
longjmp(toplevel, 1);
}
void
lostpeer()
{
if (connected) {
if (cout != NULL) {
(void) shutdown(fileno(cout), 1+1);
(void) fclose(cout);
cout = NULL;
}
if (data >= 0) {
(void) shutdown(data, 1+1);
(void) close(data);
data = -1;
}
connected = 0;
}
pswitch(1);
if (connected) {
if (cout != NULL) {
(void) shutdown(fileno(cout), 1+1);
(void) fclose(cout);
cout = NULL;
}
connected = 0;
}
proxflag = 0;
pswitch(0);
}
1994-08-25 07:47:50 +04:00
/*
char *
1993-03-21 12:45:37 +03:00
tail(filename)
char *filename;
{
1994-08-25 07:47:50 +04:00
char *s;
1993-03-21 12:45:37 +03:00
while (*filename) {
1994-08-25 07:47:50 +04:00
s = strrchr(filename, '/');
1993-03-21 12:45:37 +03:00
if (s == NULL)
break;
if (s[1])
return (s + 1);
if (s == filename)
break; XXX
1993-03-21 12:45:37 +03:00
*s = '\0';
}
return (filename);
}
*/
1994-08-25 07:47:50 +04:00
1993-03-21 12:45:37 +03:00
/*
* Command parser.
*/
1994-08-25 07:47:50 +04:00
void
1993-03-21 12:45:37 +03:00
cmdscanner(top)
int top;
{
1994-08-25 07:47:50 +04:00
struct cmd *c;
int l;
1993-03-21 12:45:37 +03:00
if (!top)
(void) putchar('\n');
for (;;) {
if (fromatty) {
printf("ftp> ");
(void) fflush(stdout);
}
if (fgets(line, sizeof line, stdin) == NULL)
1994-08-25 07:47:50 +04:00
quit(0, 0);
1993-03-21 12:45:37 +03:00
l = strlen(line);
if (l == 0)
break;
if (line[--l] == '\n') {
if (l == 0)
break;
line[l] = '\0';
} else if (l == sizeof(line) - 2) {
printf("sorry, input line too long\n");
while ((l = getchar()) != '\n' && l != EOF)
/* void */;
break;
} /* else it was a line without a newline */
makeargv();
if (margc == 0) {
continue;
}
c = getcmd(margv[0]);
if (c == (struct cmd *)-1) {
printf("?Ambiguous command\n");
continue;
}
if (c == 0) {
printf("?Invalid command\n");
continue;
}
if (c->c_conn && !connected) {
printf("Not connected.\n");
continue;
}
confirmrest = 0;
1993-03-21 12:45:37 +03:00
(*c->c_handler)(margc, margv);
if (bell && c->c_bell)
(void) putchar('\007');
if (c->c_handler != help)
break;
}
(void) signal(SIGINT, intr);
(void) signal(SIGPIPE, lostpeer);
}
struct cmd *
getcmd(name)
const char *name;
1993-03-21 12:45:37 +03:00
{
const char *p, *q;
1994-08-25 07:47:50 +04:00
struct cmd *c, *found;
int nmatches, longest;
1993-03-21 12:45:37 +03:00
if (name == NULL)
return (0);
1993-03-21 12:45:37 +03:00
longest = 0;
nmatches = 0;
found = 0;
for (c = cmdtab; (p = c->c_name) != NULL; c++) {
1993-03-21 12:45:37 +03:00
for (q = name; *q == *p++; q++)
if (*q == 0) /* exact match? */
return (c);
if (!*q) { /* the name was a prefix */
if (q - name > longest) {
longest = q - name;
nmatches = 1;
found = c;
} else if (q - name == longest)
nmatches++;
}
}
if (nmatches > 1)
return ((struct cmd *)-1);
return (found);
}
/*
* Slice a string up into argc/argv.
*/
int slrflag;
1994-08-25 07:47:50 +04:00
void
1993-03-21 12:45:37 +03:00
makeargv()
{
char **argp;
argp = margv;
stringbase = line; /* scan from first of buffer */
argbase = argbuf; /* store from first of buffer */
slrflag = 0;
for (margc = 0; ; margc++) {
/* Expand array if necessary */
if (margc == margvlen) {
margv = (margvlen == 0)
? (char **)malloc(20 * sizeof(char *))
: (char **)realloc(margv,
(margvlen + 20)*sizeof(char *));
if (margv == NULL)
errx(1, "cannot realloc argv array");
margvlen += 20;
argp = margv + margc;
}
if ((*argp++ = slurpstring()) == NULL)
break;
}
1993-03-21 12:45:37 +03:00
}
/*
* Parse string into argbuf;
* implemented with FSM to
* handle quoting and strings
*/
char *
slurpstring()
{
int got_one = 0;
1994-08-25 07:47:50 +04:00
char *sb = stringbase;
char *ap = argbase;
1993-03-21 12:45:37 +03:00
char *tmp = argbase; /* will return this if token found */
if (*sb == '!' || *sb == '$') { /* recognize ! as a token for shell */
switch (slrflag) { /* and $ as token for macro invoke */
case 0:
slrflag++;
stringbase++;
return ((*sb == '!') ? "!" : "$");
/* NOTREACHED */
case 1:
slrflag++;
altarg = stringbase;
break;
default:
break;
}
}
S0:
switch (*sb) {
case '\0':
goto OUT;
case ' ':
case '\t':
sb++; goto S0;
default:
switch (slrflag) {
case 0:
slrflag++;
break;
case 1:
slrflag++;
altarg = sb;
break;
default:
break;
}
goto S1;
}
S1:
switch (*sb) {
case ' ':
case '\t':
case '\0':
goto OUT; /* end of token */
case '\\':
sb++; goto S2; /* slurp next character */
case '"':
sb++; goto S3; /* slurp quoted string */
default:
*ap++ = *sb++; /* add character to token */
got_one = 1;
goto S1;
}
S2:
switch (*sb) {
case '\0':
goto OUT;
default:
*ap++ = *sb++;
got_one = 1;
goto S1;
}
S3:
switch (*sb) {
case '\0':
goto OUT;
case '"':
sb++; goto S1;
default:
*ap++ = *sb++;
got_one = 1;
goto S3;
}
OUT:
if (got_one)
*ap++ = '\0';
argbase = ap; /* update storage pointer */
stringbase = sb; /* update scan pointer */
if (got_one) {
1994-08-25 07:47:50 +04:00
return (tmp);
1993-03-21 12:45:37 +03:00
}
switch (slrflag) {
case 0:
slrflag++;
break;
case 1:
slrflag++;
altarg = (char *) 0;
break;
default:
break;
}
1994-08-25 07:47:50 +04:00
return ((char *)0);
1993-03-21 12:45:37 +03:00
}
1994-08-25 07:47:50 +04:00
#define HELPINDENT ((int) sizeof ("directory"))
1993-03-21 12:45:37 +03:00
/*
* Help command.
* Call each command handler with argc == 0 and argv[0] == name.
*/
1994-08-25 07:47:50 +04:00
void
1993-03-21 12:45:37 +03:00
help(argc, argv)
int argc;
char *argv[];
{
1994-08-25 07:47:50 +04:00
struct cmd *c;
1993-03-21 12:45:37 +03:00
if (argc == 1) {
1994-08-25 07:47:50 +04:00
int i, j, w, k;
1993-03-21 12:45:37 +03:00
int columns, width = 0, lines;
printf("Commands may be abbreviated. Commands are:\n\n");
for (c = cmdtab; c < &cmdtab[NCMDS]; c++) {
int len = strlen(c->c_name);
if (len > width)
width = len;
}
width = (width + 8) &~ 7;
columns = 80 / width;
if (columns == 0)
columns = 1;
lines = (NCMDS + columns - 1) / columns;
for (i = 0; i < lines; i++) {
for (j = 0; j < columns; j++) {
c = cmdtab + j * lines + i;
if (c->c_name && (!proxy || c->c_proxy)) {
printf("%s", c->c_name);
}
else if (c->c_name) {
for (k=0; k < strlen(c->c_name); k++) {
(void) putchar(' ');
}
}
if (c + lines >= &cmdtab[NCMDS]) {
printf("\n");
break;
}
w = strlen(c->c_name);
while (w < width) {
w = (w + 8) &~ 7;
(void) putchar('\t');
}
}
}
return;
}
while (--argc > 0) {
1994-08-25 07:47:50 +04:00
char *arg;
1993-03-21 12:45:37 +03:00
arg = *++argv;
c = getcmd(arg);
if (c == (struct cmd *)-1)
printf("?Ambiguous help command %s\n", arg);
else if (c == (struct cmd *)0)
printf("?Invalid help command %s\n", arg);
else
printf("%-*s\t%s\n", HELPINDENT,
c->c_name, c->c_help);
}
}
int
auto_fetch(argc, argv)
int argc;
char *argv[];
{
char *xargv[5];
char *cp, *host, *dir, *file;
int rval, xargc;
size_t urllen;
urllen = strlen(FTPURL);
rval = 0;
if (setjmp(toplevel))
exit(0);
(void) signal(SIGINT, intr);
(void) signal(SIGPIPE, lostpeer);
/*
* Loop through as long as there's files to fetch.
*/
for ( ; argc > 0 && strchr(argv[0], ':') != NULL ; --argc, argv++) {
host = dir = file = portnum = NULL;
/*
* We muck with the string, so we make a copy.
*/
host = strdup(argv[0]);
if (host == NULL)
errx(1, "Can't allocate memory for auto-fetch.");
/*
* Try URL-style arguments first, then host:file.
*/
if (strncmp(host, FTPURL, urllen) == 0) {
host += urllen;
cp = strchr(host, '/');
/* Look for a port number after the nost name. */
portnum = strchr(host, ':');
if (portnum != NULL)
*portnum++ = '\0';
} else
cp = strchr(host, ':');
/*
* If cp is NULL, the file wasn't specified
* (URL looked something like ftp://host)
*/
if (cp == NULL)
usage();
/*
* Extract the file and (if present) directory name.
*/
*cp++ = '\0';
dir = cp;
cp = strrchr(cp, '/');
if (cp != NULL) {
*cp++ = '\0';
file = cp;
} else {
file = dir;
dir = NULL;
}
if (file == NULL || *file == '\0')
usage();
/*
* Set up the connection.
*/
xargv[0] = __progname;
xargv[1] = host;
xargv[2] = NULL;
xargc = 2;
if (portnum != NULL) {
xargv[2] = portnum;
xargv[3] = NULL;
xargc = 3;
}
setpeer(xargc, xargv);
if (connected == 0) {
warnx("Can't connect to host `%s'.", host);
rval = 1;
free(host);
continue;
}
/* Always use binary transfers. */
setbinary(0, NULL);
/* Change directories, if necessary. */
if (dir != NULL) {
xargv[0] = "cd";
xargv[1] = dir;
xargv[2] = NULL;
cd(2, xargv);
/* XXX SHOULD CHECK FOR ERROR CONDITION! */
}
/* Fetch the file. */
xargv[0] = "get";
xargv[1] = file;
xargv[2] = NULL;
get(2, xargv);
disconnect(0, NULL);
free(host);
}
return (rval);
}
void
usage()
{
(void)fprintf(stderr,
"usage: %s [-adginptv] [-P port] [host [port]]\n"
" %s ftp://host[:port]/file\n"
" %s host:file\n",
__progname, __progname, __progname);
exit(1);
}