NetBSD/usr.bin/ftp/util.c
lukem b918fc0803 Forbid filenames returned from mget that aren't in (or below) the
current directory.
The previous behaviour (of trusting the remote server's response when
retrieving the list of files to mget with prompting disabled) has been
in ftp ~forever, and has been a "known issue" for a long time.
Recently an advisory was published by D.J. Bernstein on behalf of
Yosef Klein warning of the problems with the previous behaviour, so
to alleviate concern I've fixed this with a sledgehammer.

Remember the local cwd after any operation which may change it.
Use "remotecwd" instead of "remotepwd".
2005-01-03 09:50:09 +00:00

1335 lines
29 KiB
C

/* $NetBSD: util.c,v 1.117 2005/01/03 09:50:09 lukem Exp $ */
/*-
* Copyright (c) 1997-2005 The NetBSD Foundation, Inc.
* All rights reserved.
*
* This code is derived from software contributed to The NetBSD Foundation
* by Luke Mewburn.
*
* This code is derived from software contributed to The NetBSD Foundation
* by Jason R. Thorpe of the Numerical Aerospace Simulation Facility,
* NASA Ames Research Center.
*
* 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 NetBSD
* Foundation, Inc. and its contributors.
* 4. Neither the name of The NetBSD Foundation 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 NETBSD FOUNDATION, INC. 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 FOUNDATION 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.
*/
/*
* Copyright (c) 1985, 1989, 1993, 1994
* The Regents of the University of California. All rights reserved.
*
* 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. 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.
*/
#include <sys/cdefs.h>
#ifndef lint
__RCSID("$NetBSD: util.c,v 1.117 2005/01/03 09:50:09 lukem Exp $");
#endif /* not lint */
/*
* FTP User Program -- Misc support routines
*/
#include <sys/param.h>
#include <sys/socket.h>
#include <sys/ioctl.h>
#include <sys/time.h>
#include <netinet/in.h>
#include <arpa/ftp.h>
#include <ctype.h>
#include <err.h>
#include <errno.h>
#include <fcntl.h>
#include <glob.h>
#include <signal.h>
#include <limits.h>
#include <netdb.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <termios.h>
#include <time.h>
#include <tzfile.h>
#include <unistd.h>
#include "ftp_var.h"
/*
* Connect to peer server and auto-login, if possible.
*/
void
setpeer(int argc, char *argv[])
{
char *host;
char *port;
if (argc == 0)
goto usage;
if (connected) {
fprintf(ttyout, "Already connected to %s, use close first.\n",
hostname);
code = -1;
return;
}
if (argc < 2)
(void)another(&argc, &argv, "to");
if (argc < 2 || argc > 3) {
usage:
fprintf(ttyout, "usage: %s host-name [port]\n", argv[0]);
code = -1;
return;
}
if (gatemode)
port = gateport;
else
port = ftpport;
if (argc > 2)
port = argv[2];
if (gatemode) {
if (gateserver == NULL || *gateserver == '\0')
errx(1, "gateserver not defined (shouldn't happen)");
host = hookup(gateserver, port);
} else
host = hookup(argv[1], port);
if (host) {
if (gatemode && verbose) {
fprintf(ttyout,
"Connecting via pass-through server %s\n",
gateserver);
}
connected = 1;
/*
* Set up defaults for FTP.
*/
(void)strlcpy(typename, "ascii", sizeof(typename));
type = TYPE_A;
curtype = TYPE_A;
(void)strlcpy(formname, "non-print", sizeof(formname));
form = FORM_N;
(void)strlcpy(modename, "stream", sizeof(modename));
mode = MODE_S;
(void)strlcpy(structname, "file", sizeof(structname));
stru = STRU_F;
(void)strlcpy(bytename, "8", sizeof(bytename));
bytesize = 8;
if (autologin)
(void)ftp_login(argv[1], NULL, NULL);
}
}
static void
parse_feat(const char *line)
{
/*
* work-around broken ProFTPd servers that can't
* even obey RFC 2389.
*/
while (*line && isspace((int)*line))
line++;
if (strcasecmp(line, "MDTM") == 0)
features[FEAT_MDTM] = 1;
else if (strncasecmp(line, "MLST", sizeof("MLST") - 1) == 0) {
features[FEAT_MLST] = 1;
} else if (strcasecmp(line, "REST STREAM") == 0)
features[FEAT_REST_STREAM] = 1;
else if (strcasecmp(line, "SIZE") == 0)
features[FEAT_SIZE] = 1;
else if (strcasecmp(line, "TVFS") == 0)
features[FEAT_TVFS] = 1;
}
/*
* Determine the remote system type (SYST) and features (FEAT).
* Call after a successful login (i.e, connected = -1)
*/
void
getremoteinfo(void)
{
int overbose, i;
overbose = verbose;
if (debug == 0)
verbose = -1;
/* determine remote system type */
if (command("SYST") == COMPLETE) {
if (overbose) {
char *cp, c;
c = 0;
cp = strchr(reply_string + 4, ' ');
if (cp == NULL)
cp = strchr(reply_string + 4, '\r');
if (cp) {
if (cp[-1] == '.')
cp--;
c = *cp;
*cp = '\0';
}
fprintf(ttyout, "Remote system type is %s.\n",
reply_string + 4);
if (cp)
*cp = c;
}
if (!strncmp(reply_string, "215 UNIX Type: L8", 17)) {
if (proxy)
unix_proxy = 1;
else
unix_server = 1;
/*
* Set type to 0 (not specified by user),
* meaning binary by default, but don't bother
* telling server. We can use binary
* for text files unless changed by the user.
*/
type = 0;
(void)strlcpy(typename, "binary", sizeof(typename));
if (overbose)
fprintf(ttyout,
"Using %s mode to transfer files.\n",
typename);
} else {
if (proxy)
unix_proxy = 0;
else
unix_server = 0;
if (overbose &&
!strncmp(reply_string, "215 TOPS20", 10))
fputs(
"Remember to set tenex mode when transferring binary files from this machine.\n",
ttyout);
}
}
/* determine features (if any) */
for (i = 0; i < FEAT_max; i++)
features[i] = -1;
reply_callback = parse_feat;
if (command("FEAT") == COMPLETE) {
for (i = 0; i < FEAT_max; i++) {
if (features[i] == -1)
features[i] = 0;
}
features[FEAT_FEAT] = 1;
} else
features[FEAT_FEAT] = 0;
if (debug) {
#define DEBUG_FEAT(x) fprintf(ttyout, "features[" #x "] = %d\n", features[(x)])
DEBUG_FEAT(FEAT_FEAT);
DEBUG_FEAT(FEAT_MDTM);
DEBUG_FEAT(FEAT_MLST);
DEBUG_FEAT(FEAT_REST_STREAM);
DEBUG_FEAT(FEAT_SIZE);
DEBUG_FEAT(FEAT_TVFS);
#undef DEBUG_FEAT
}
reply_callback = NULL;
verbose = overbose;
}
/*
* Reset the various variables that indicate connection state back to
* disconnected settings.
* The caller is responsible for issuing any commands to the remote server
* to perform a clean shutdown before this is invoked.
*/
void
cleanuppeer(void)
{
if (cout)
(void)fclose(cout);
cout = NULL;
connected = 0;
unix_server = 0;
unix_proxy = 0;
/*
* determine if anonftp was specifically set with -a
* (1), or implicitly set by auto_fetch() (2). in the
* latter case, disable after the current xfer
*/
if (anonftp == 2)
anonftp = 0;
data = -1;
epsv4bad = 0;
if (username)
free(username);
username = NULL;
if (!proxy)
macnum = 0;
}
/*
* Top-level signal handler for interrupted commands.
*/
void
intr(int signo)
{
sigint_raised = 1;
alarmtimer(0);
if (fromatty)
write(fileno(ttyout), "\n", 1);
siglongjmp(toplevel, 1);
}
/*
* Signal handler for lost connections; cleanup various elements of
* the connection state, and call cleanuppeer() to finish it off.
*/
void
lostpeer(int dummy)
{
int oerrno = errno;
alarmtimer(0);
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);
cleanuppeer();
errno = oerrno;
}
/*
* Login to remote host, using given username & password if supplied.
* Return non-zero if successful.
*/
int
ftp_login(const char *host, const char *user, const char *pass)
{
char tmp[80];
const char *acct;
int n, aflag, rval, freeuser, freepass, freeacct;
acct = NULL;
aflag = rval = freeuser = freepass = freeacct = 0;
if (debug)
fprintf(ttyout, "ftp_login: user `%s' pass `%s' host `%s'\n",
user ? user : "<null>", pass ? pass : "<null>",
host ? host : "<null>");
/*
* Set up arguments for an anonymous FTP session, if necessary.
*/
if (anonftp) {
user = "anonymous"; /* as per RFC 1635 */
pass = getoptionvalue("anonpass");
}
if (user == NULL)
freeuser = 1;
if (pass == NULL)
freepass = 1;
freeacct = 1;
if (ruserpass(host, &user, &pass, &acct) < 0) {
code = -1;
goto cleanup_ftp_login;
}
while (user == NULL) {
if (localname)
fprintf(ttyout, "Name (%s:%s): ", host, localname);
else
fprintf(ttyout, "Name (%s): ", host);
*tmp = '\0';
if (fgets(tmp, sizeof(tmp) - 1, stdin) == NULL) {
fprintf(ttyout, "\nEOF received; login aborted.\n");
clearerr(stdin);
code = -1;
goto cleanup_ftp_login;
}
tmp[strlen(tmp) - 1] = '\0';
freeuser = 0;
if (*tmp == '\0')
user = localname;
else
user = tmp;
}
if (gatemode) {
char *nuser;
int len;
len = strlen(user) + 1 + strlen(host) + 1;
nuser = xmalloc(len);
(void)strlcpy(nuser, user, len);
(void)strlcat(nuser, "@", len);
(void)strlcat(nuser, host, len);
freeuser = 1;
user = nuser;
}
n = command("USER %s", user);
if (n == CONTINUE) {
if (pass == NULL) {
freepass = 0;
pass = getpass("Password:");
}
n = command("PASS %s", pass);
}
if (n == CONTINUE) {
aflag++;
if (acct == NULL) {
freeacct = 0;
acct = getpass("Account:");
}
if (acct[0] == '\0') {
warnx("Login failed.");
goto cleanup_ftp_login;
}
n = command("ACCT %s", acct);
}
if ((n != COMPLETE) ||
(!aflag && acct != NULL && command("ACCT %s", acct) != COMPLETE)) {
warnx("Login failed.");
goto cleanup_ftp_login;
}
rval = 1;
username = xstrdup(user);
if (proxy)
goto cleanup_ftp_login;
connected = -1;
getremoteinfo();
for (n = 0; n < macnum; ++n) {
if (!strcmp("init", macros[n].mac_name)) {
(void)strlcpy(line, "$init", sizeof(line));
makeargv();
domacro(margc, margv);
break;
}
}
updatelocalcwd();
updateremotecwd();
cleanup_ftp_login:
if (user != NULL && freeuser)
free((char *)user);
if (pass != NULL && freepass)
free((char *)pass);
if (acct != NULL && freeacct)
free((char *)acct);
return (rval);
}
/*
* `another' gets another argument, and stores the new argc and argv.
* It reverts to the top level (via intr()) on EOF/error.
*
* Returns false if no new arguments have been added.
*/
int
another(int *pargc, char ***pargv, const char *prompt)
{
int len = strlen(line), ret;
if (len >= sizeof(line) - 3) {
fputs("sorry, arguments too long.\n", ttyout);
intr(0);
}
fprintf(ttyout, "(%s) ", prompt);
line[len++] = ' ';
if (fgets(&line[len], sizeof(line) - len, stdin) == NULL) {
clearerr(stdin);
intr(0);
}
len += strlen(&line[len]);
if (len > 0 && line[len - 1] == '\n')
line[len - 1] = '\0';
makeargv();
ret = margc > *pargc;
*pargc = margc;
*pargv = margv;
return (ret);
}
/*
* glob files given in argv[] from the remote server.
* if errbuf isn't NULL, store error messages there instead
* of writing to the screen.
*/
char *
remglob(char *argv[], int doswitch, char **errbuf)
{
char temp[MAXPATHLEN];
static char buf[MAXPATHLEN];
static FILE *ftemp = NULL;
static char **args;
int oldverbose, oldhash, oldprogress, fd, len;
char *cp, *mode;
if (!mflag || !connected) {
if (!doglob)
args = NULL;
else {
if (ftemp) {
(void)fclose(ftemp);
ftemp = NULL;
}
}
return (NULL);
}
if (!doglob) {
if (args == NULL)
args = argv;
if ((cp = *++args) == NULL)
args = NULL;
return (cp);
}
if (ftemp == NULL) {
len = strlcpy(temp, tmpdir, sizeof(temp));
if (temp[len - 1] != '/')
(void)strlcat(temp, "/", sizeof(temp));
(void)strlcat(temp, TMPFILE, sizeof(temp));
if ((fd = mkstemp(temp)) < 0) {
warn("unable to create temporary file %s", temp);
return (NULL);
}
close(fd);
oldverbose = verbose;
verbose = (errbuf != NULL) ? -1 : 0;
oldhash = hash;
oldprogress = progress;
hash = 0;
progress = 0;
if (doswitch)
pswitch(!proxy);
for (mode = "w"; *++argv != NULL; mode = "a")
recvrequest("NLST", temp, *argv, mode, 0, 0);
if ((code / 100) != COMPLETE) {
if (errbuf != NULL)
*errbuf = reply_string;
}
if (doswitch)
pswitch(!proxy);
verbose = oldverbose;
hash = oldhash;
progress = oldprogress;
ftemp = fopen(temp, "r");
(void)unlink(temp);
if (ftemp == NULL) {
if (errbuf == NULL)
fputs(
"can't find list of remote files, oops.\n",
ttyout);
else
*errbuf =
"can't find list of remote files, oops.";
return (NULL);
}
}
if (fgets(buf, sizeof(buf), ftemp) == NULL) {
(void)fclose(ftemp);
ftemp = NULL;
return (NULL);
}
if ((cp = strchr(buf, '\n')) != NULL)
*cp = '\0';
return (buf);
}
/*
* Glob a local file name specification with the expectation of a single
* return value. Can't control multiple values being expanded from the
* expression, we return only the first.
* Returns NULL on error, or a pointer to a buffer containing the filename
* that's the caller's responsiblity to free(3) when finished with.
*/
char *
globulize(const char *pattern)
{
glob_t gl;
int flags;
char *p;
if (!doglob)
return (xstrdup(pattern));
flags = GLOB_BRACE|GLOB_NOCHECK|GLOB_TILDE;
memset(&gl, 0, sizeof(gl));
if (glob(pattern, flags, NULL, &gl) || gl.gl_pathc == 0) {
warnx("%s: not found", pattern);
globfree(&gl);
return (NULL);
}
p = xstrdup(gl.gl_pathv[0]);
globfree(&gl);
return (p);
}
/*
* determine size of remote file
*/
off_t
remotesize(const char *file, int noisy)
{
int overbose, r;
off_t size;
overbose = verbose;
size = -1;
if (debug == 0)
verbose = -1;
if (! features[FEAT_SIZE]) {
if (noisy)
fprintf(ttyout,
"SIZE is not supported by remote server.\n");
goto cleanup_remotesize;
}
r = command("SIZE %s", file);
if (r == COMPLETE) {
char *cp, *ep;
cp = strchr(reply_string, ' ');
if (cp != NULL) {
cp++;
size = STRTOLL(cp, &ep, 10);
if (*ep != '\0' && !isspace((unsigned char)*ep))
size = -1;
}
} else {
if (r == ERROR && code == 500 && features[FEAT_SIZE] == -1)
features[FEAT_SIZE] = 0;
if (noisy && debug == 0) {
fputs(reply_string, ttyout);
putc('\n', ttyout);
}
}
cleanup_remotesize:
verbose = overbose;
return (size);
}
/*
* determine last modification time (in GMT) of remote file
*/
time_t
remotemodtime(const char *file, int noisy)
{
int overbose, ocode, r;
time_t rtime;
overbose = verbose;
ocode = code;
rtime = -1;
if (debug == 0)
verbose = -1;
if (! features[FEAT_MDTM]) {
if (noisy)
fprintf(ttyout,
"MDTM is not supported by remote server.\n");
goto cleanup_parse_time;
}
r = command("MDTM %s", file);
if (r == COMPLETE) {
struct tm timebuf;
char *timestr, *frac;
int yy, mo, day, hour, min, sec;
/*
* time-val = 14DIGIT [ "." 1*DIGIT ]
* YYYYMMDDHHMMSS[.sss]
* mdtm-response = "213" SP time-val CRLF / error-response
*/
timestr = reply_string + 4;
/*
* parse fraction.
* XXX: ignored for now
*/
frac = strchr(timestr, '\r');
if (frac != NULL)
*frac = '\0';
frac = strchr(timestr, '.');
if (frac != NULL)
*frac++ = '\0';
if (strlen(timestr) == 15 && strncmp(timestr, "191", 3) == 0) {
/*
* XXX: Workaround for lame ftpd's that return
* `19100' instead of `2000'
*/
fprintf(ttyout,
"Y2K warning! Incorrect time-val `%s' received from server.\n",
timestr);
timestr++;
timestr[0] = '2';
timestr[1] = '0';
fprintf(ttyout, "Converted to `%s'\n", timestr);
}
if (strlen(timestr) != 14 ||
sscanf(timestr, "%04d%02d%02d%02d%02d%02d",
&yy, &mo, &day, &hour, &min, &sec) != 6) {
bad_parse_time:
fprintf(ttyout, "Can't parse time `%s'.\n", timestr);
goto cleanup_parse_time;
}
memset(&timebuf, 0, sizeof(timebuf));
timebuf.tm_sec = sec;
timebuf.tm_min = min;
timebuf.tm_hour = hour;
timebuf.tm_mday = day;
timebuf.tm_mon = mo - 1;
timebuf.tm_year = yy - TM_YEAR_BASE;
timebuf.tm_isdst = -1;
rtime = timegm(&timebuf);
if (rtime == -1) {
if (noisy || debug != 0)
goto bad_parse_time;
else
goto cleanup_parse_time;
} else if (debug)
fprintf(ttyout, "parsed date as: %s", ctime(&rtime));
} else {
if (r == ERROR && code == 500 && features[FEAT_MDTM] == -1)
features[FEAT_MDTM] = 0;
if (noisy && debug == 0) {
fputs(reply_string, ttyout);
putc('\n', ttyout);
}
}
cleanup_parse_time:
verbose = overbose;
if (rtime == -1)
code = ocode;
return (rtime);
}
/*
* Update global `localcwd', which contains the state of the local cwd
*/
void
updatelocalcwd(void)
{
if (getcwd(localcwd, sizeof(localcwd)) == NULL)
localcwd[0] = '\0';
if (debug)
fprintf(ttyout, "got localcwd as `%s'\n", localcwd);
}
/*
* Update global `remotecwd', which contains the state of the remote cwd
*/
void
updateremotecwd(void)
{
int overbose, ocode, i;
char *cp;
overbose = verbose;
ocode = code;
if (debug == 0)
verbose = -1;
if (command("PWD") != COMPLETE)
goto badremotecwd;
cp = strchr(reply_string, ' ');
if (cp == NULL || cp[0] == '\0' || cp[1] != '"')
goto badremotecwd;
cp += 2;
for (i = 0; *cp && i < sizeof(remotecwd) - 1; i++, cp++) {
if (cp[0] == '"') {
if (cp[1] == '"')
cp++;
else
break;
}
remotecwd[i] = *cp;
}
remotecwd[i] = '\0';
if (debug)
fprintf(ttyout, "got remotecwd as `%s'\n", remotecwd);
goto cleanupremotecwd;
badremotecwd:
remotecwd[0]='\0';
cleanupremotecwd:
verbose = overbose;
code = ocode;
}
/*
* Ensure file is in or under dir.
* Returns 1 if so, 0 if not (or an error occurred).
*/
int
fileindir(const char *file, const char *dir)
{
char realfile[PATH_MAX+1];
size_t dirlen;
if (realpath(file, realfile) == NULL) {
warn("Unable to determine real path of `%s'", file);
return 0;
}
if (realfile[0] != '/') /* relative result */
return 1;
dirlen = strlen(dir);
#if 0
printf("file %s realfile %s dir %s [%d]\n", file, realfile, dir, dirlen);
#endif
if (strncmp(realfile, dir, dirlen) == 0 && realfile[dirlen] == '/')
return 1;
return 0;
}
/*
* List words in stringlist, vertically arranged
*/
void
list_vertical(StringList *sl)
{
int i, j, w;
int columns, width, lines;
char *p;
width = 0;
for (i = 0 ; i < sl->sl_cur ; i++) {
w = strlen(sl->sl_str[i]);
if (w > width)
width = w;
}
width = (width + 8) &~ 7;
columns = ttywidth / width;
if (columns == 0)
columns = 1;
lines = (sl->sl_cur + columns - 1) / columns;
for (i = 0; i < lines; i++) {
for (j = 0; j < columns; j++) {
p = sl->sl_str[j * lines + i];
if (p)
fputs(p, ttyout);
if (j * lines + i + lines >= sl->sl_cur) {
putc('\n', ttyout);
break;
}
w = strlen(p);
while (w < width) {
w = (w + 8) &~ 7;
(void)putc('\t', ttyout);
}
}
}
}
/*
* Update the global ttywidth value, using TIOCGWINSZ.
*/
void
setttywidth(int a)
{
struct winsize winsize;
int oerrno = errno;
if (ioctl(fileno(ttyout), TIOCGWINSZ, &winsize) != -1 &&
winsize.ws_col != 0)
ttywidth = winsize.ws_col;
else
ttywidth = 80;
errno = oerrno;
}
/*
* Change the rate limit up (SIGUSR1) or down (SIGUSR2)
*/
void
crankrate(int sig)
{
switch (sig) {
case SIGUSR1:
if (rate_get)
rate_get += rate_get_incr;
if (rate_put)
rate_put += rate_put_incr;
break;
case SIGUSR2:
if (rate_get && rate_get > rate_get_incr)
rate_get -= rate_get_incr;
if (rate_put && rate_put > rate_put_incr)
rate_put -= rate_put_incr;
break;
default:
err(1, "crankrate invoked with unknown signal: %d", sig);
}
}
/*
* Setup or cleanup EditLine structures
*/
#ifndef NO_EDITCOMPLETE
void
controlediting(void)
{
if (editing && el == NULL && hist == NULL) {
HistEvent ev;
int editmode;
el = el_init(getprogname(), stdin, ttyout, stderr);
/* init editline */
hist = history_init(); /* init the builtin history */
history(hist, &ev, H_SETSIZE, 100);/* remember 100 events */
el_set(el, EL_HIST, history, hist); /* use history */
el_set(el, EL_EDITOR, "emacs"); /* default editor is emacs */
el_set(el, EL_PROMPT, prompt); /* set the prompt functions */
el_set(el, EL_RPROMPT, rprompt);
/* add local file completion, bind to TAB */
el_set(el, EL_ADDFN, "ftp-complete",
"Context sensitive argument completion",
complete);
el_set(el, EL_BIND, "^I", "ftp-complete", NULL);
el_source(el, NULL); /* read ~/.editrc */
if ((el_get(el, EL_EDITMODE, &editmode) != -1) && editmode == 0)
editing = 0; /* the user doesn't want editing,
* so disable, and let statement
* below cleanup */
else
el_set(el, EL_SIGNAL, 1);
}
if (!editing) {
if (hist) {
history_end(hist);
hist = NULL;
}
if (el) {
el_end(el);
el = NULL;
}
}
}
#endif /* !NO_EDITCOMPLETE */
/*
* Convert the string `arg' to an int, which may have an optional SI suffix
* (`b', `k', `m', `g'). Returns the number for success, -1 otherwise.
*/
int
strsuftoi(const char *arg)
{
char *cp;
long val;
if (!isdigit((unsigned char)arg[0]))
return (-1);
val = strtol(arg, &cp, 10);
if (cp != NULL) {
if (cp[0] != '\0' && cp[1] != '\0')
return (-1);
switch (tolower((unsigned char)cp[0])) {
case '\0':
case 'b':
break;
case 'k':
val <<= 10;
break;
case 'm':
val <<= 20;
break;
case 'g':
val <<= 30;
break;
default:
return (-1);
}
}
if (val < 0 || val > INT_MAX)
return (-1);
return (val);
}
/*
* Set up socket buffer sizes before a connection is made.
*/
void
setupsockbufsize(int sock)
{
if (setsockopt(sock, SOL_SOCKET, SO_SNDBUF, (void *) &sndbuf_size,
sizeof(rcvbuf_size)) < 0)
warn("unable to set sndbuf size %d", sndbuf_size);
if (setsockopt(sock, SOL_SOCKET, SO_RCVBUF, (void *) &rcvbuf_size,
sizeof(rcvbuf_size)) < 0)
warn("unable to set rcvbuf size %d", rcvbuf_size);
}
/*
* Copy characters from src into dst, \ quoting characters that require it
*/
void
ftpvis(char *dst, size_t dstlen, const char *src, size_t srclen)
{
int di, si;
for (di = si = 0;
src[si] != '\0' && di < dstlen && si < srclen;
di++, si++) {
switch (src[si]) {
case '\\':
case ' ':
case '\t':
case '\r':
case '\n':
case '"':
dst[di++] = '\\';
if (di >= dstlen)
break;
/* FALLTHROUGH */
default:
dst[di] = src[si];
}
}
dst[di] = '\0';
}
/*
* Copy src into buf (which is len bytes long), expanding % sequences.
*/
void
formatbuf(char *buf, size_t len, const char *src)
{
const char *p;
char *p2, *q;
int i, op, updirs, pdirs;
#define ADDBUF(x) do { \
if (i >= len - 1) \
goto endbuf; \
buf[i++] = (x); \
} while (0)
p = src;
for (i = 0; *p; p++) {
if (*p != '%') {
ADDBUF(*p);
continue;
}
p++;
switch (op = *p) {
case '/':
case '.':
case 'c':
p2 = connected ? remotecwd : "";
updirs = pdirs = 0;
/* option to determine fixed # of dirs from path */
if (op == '.' || op == 'c') {
int skip;
q = p2;
while (*p2) /* calc # of /'s */
if (*p2++ == '/')
updirs++;
if (p[1] == '0') { /* print <x> or ... */
pdirs = 1;
p++;
}
if (p[1] >= '1' && p[1] <= '9') {
/* calc # to skip */
skip = p[1] - '0';
p++;
} else
skip = 1;
updirs -= skip;
while (skip-- > 0) {
while ((p2 > q) && (*p2 != '/'))
p2--; /* back up */
if (skip && p2 > q)
p2--;
}
if (*p2 == '/' && p2 != q)
p2++;
}
if (updirs > 0 && pdirs) {
if (i >= len - 5)
break;
if (op == '.') {
ADDBUF('.');
ADDBUF('.');
ADDBUF('.');
} else {
ADDBUF('/');
ADDBUF('<');
if (updirs > 9) {
ADDBUF('9');
ADDBUF('+');
} else
ADDBUF('0' + updirs);
ADDBUF('>');
}
}
for (; *p2; p2++)
ADDBUF(*p2);
break;
case 'M':
case 'm':
for (p2 = connected && username ? username : "-";
*p2 ; p2++) {
if (op == 'm' && *p2 == '.')
break;
ADDBUF(*p2);
}
break;
case 'n':
for (p2 = connected ? username : "-"; *p2 ; p2++)
ADDBUF(*p2);
break;
case '%':
ADDBUF('%');
break;
default: /* display unknown codes literally */
ADDBUF('%');
ADDBUF(op);
break;
}
}
endbuf:
buf[i] = '\0';
}
/*
* Parse `port' into a TCP port number, defaulting to `defport' if `port' is
* an unknown service name. If defport != -1, print a warning upon bad parse.
*/
int
parseport(const char *port, int defport)
{
int rv;
long nport;
char *p, *ep;
p = xstrdup(port);
nport = strtol(p, &ep, 10);
if (*ep != '\0' && ep == p) {
struct servent *svp;
svp = getservbyname(port, "tcp");
if (svp == NULL) {
badparseport:
if (defport != -1)
warnx("Unknown port `%s', using port %d",
port, defport);
rv = defport;
} else
rv = ntohs(svp->s_port);
} else if (nport < 1 || nport > MAX_IN_PORT_T || *ep != '\0')
goto badparseport;
else
rv = nport;
free(p);
return (rv);
}
/*
* Determine if given string is an IPv6 address or not.
* Return 1 for yes, 0 for no
*/
int
isipv6addr(const char *addr)
{
int rv = 0;
#ifdef INET6
struct addrinfo hints, *res;
memset(&hints, 0, sizeof(hints));
hints.ai_family = PF_INET6;
hints.ai_socktype = SOCK_DGRAM; /*dummy*/
hints.ai_flags = AI_NUMERICHOST;
if (getaddrinfo(addr, "0", &hints, &res) != 0)
rv = 0;
else {
rv = 1;
freeaddrinfo(res);
}
if (debug)
fprintf(ttyout, "isipv6addr: got %d for %s\n", rv, addr);
#endif
return (rv == 1) ? 1 : 0;
}
/*
* Internal version of connect(2); sets socket buffer sizes first and
* handles the syscall being interrupted.
* Returns -1 upon failure (with errno set to the problem), or 0 on success.
*/
int
xconnect(int sock, const struct sockaddr *name, int namelen)
{
int rv;
setupsockbufsize(sock);
rv = connect(sock, name, namelen);
if (rv == -1 && errno == EINTR) {
fd_set connfd;
FD_ZERO(&connfd);
FD_SET(sock, &connfd);
do {
rv = select(sock + 1, NULL, &connfd, NULL, NULL);
} while (rv == -1 && errno == EINTR);
if (rv > 0)
rv = 0;
}
return (rv);
}
/*
* Internal version of listen(2); sets socket buffer sizes first.
*/
int
xlisten(int sock, int backlog)
{
setupsockbufsize(sock);
return (listen(sock, backlog));
}
/*
* malloc() with inbuilt error checking
*/
void *
xmalloc(size_t size)
{
void *p;
p = malloc(size);
if (p == NULL)
err(1, "Unable to allocate %ld bytes of memory", (long)size);
return (p);
}
/*
* sl_init() with inbuilt error checking
*/
StringList *
xsl_init(void)
{
StringList *p;
p = sl_init();
if (p == NULL)
err(1, "Unable to allocate memory for stringlist");
return (p);
}
/*
* sl_add() with inbuilt error checking
*/
void
xsl_add(StringList *sl, char *i)
{
if (sl_add(sl, i) == -1)
err(1, "Unable to add `%s' to stringlist", i);
}
/*
* strdup() with inbuilt error checking
*/
char *
xstrdup(const char *str)
{
char *s;
if (str == NULL)
errx(1, "xstrdup() called with NULL argument");
s = strdup(str);
if (s == NULL)
err(1, "Unable to allocate memory for string copy");
return (s);
}