mirror of
https://git.FreeBSD.org/src.git
synced 2024-12-14 10:09:48 +00:00
301cba219f
is shorter than previously announced by the server. Tested by asami. Approved by: jkh
522 lines
13 KiB
C
522 lines
13 KiB
C
/*-
|
|
* Copyright 1997 Massachusetts Institute of Technology
|
|
*
|
|
* Permission to use, copy, modify, and distribute this software and
|
|
* its documentation for any purpose and without fee is hereby
|
|
* granted, provided that both the above copyright notice and this
|
|
* permission notice appear in all copies, that both the above
|
|
* copyright notice and this permission notice appear in all
|
|
* supporting documentation, and that the name of M.I.T. not be used
|
|
* in advertising or publicity pertaining to distribution of the
|
|
* software without specific, written prior permission. M.I.T. makes
|
|
* no representations about the suitability of this software for any
|
|
* purpose. It is provided "as is" without express or implied
|
|
* warranty.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY M.I.T. ``AS IS''. M.I.T. DISCLAIMS
|
|
* ALL EXPRESS OR IMPLIED WARRANTIES WITH REGARD TO THIS SOFTWARE,
|
|
* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
|
|
* MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. IN NO EVENT
|
|
* SHALL M.I.T. 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.
|
|
*
|
|
* $FreeBSD$
|
|
*/
|
|
|
|
#include <sys/types.h>
|
|
|
|
#include <err.h>
|
|
#include <errno.h>
|
|
#include <ftpio.h>
|
|
#include <limits.h>
|
|
#include <netdb.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sysexits.h>
|
|
#include <unistd.h>
|
|
|
|
#include <sys/param.h>
|
|
#include <sys/stat.h>
|
|
|
|
#include "fetch.h"
|
|
|
|
struct ftp_state {
|
|
char *ftp_hostname;
|
|
char *ftp_user;
|
|
char *ftp_password;
|
|
char *ftp_remote_file;
|
|
char **ftp_remote_dirs;
|
|
int ftp_remote_ndirs;
|
|
char *ftp_remote_path;
|
|
char *ftp_type;
|
|
unsigned ftp_port;
|
|
};
|
|
|
|
static int ftp_close(struct fetch_state *fs);
|
|
static int ftp_retrieve(struct fetch_state *fs);
|
|
static int ftp_parse(struct fetch_state *fs, const char *uri);
|
|
static int ftp_proxy_parse(struct fetch_state *fs, const char *uri);
|
|
|
|
struct uri_scheme ftp_scheme =
|
|
{ "ftp", ftp_parse, ftp_proxy_parse, "FTP_PROXY", "ftp,http" };
|
|
|
|
static int
|
|
ftp_parse(struct fetch_state *fs, const char *uri)
|
|
{
|
|
const char *p, *slash, *q;
|
|
char *hostname, *atsign, *colon, *path, *r, *s, **dp;
|
|
unsigned port;
|
|
struct ftp_state *ftps;
|
|
|
|
p = uri + 4;
|
|
port = 0;
|
|
|
|
if (p[0] != '/' || p[1] != '/') {
|
|
warnx("`%s': invalid `ftp' URL", uri);
|
|
return EX_USAGE;
|
|
}
|
|
|
|
p += 2;
|
|
slash = strchr(p, '/');
|
|
if (slash == 0) {
|
|
warnx("`%s': malformed `ftp' URL", uri);
|
|
return EX_USAGE;
|
|
}
|
|
hostname = alloca(slash - p + 1);
|
|
hostname[0] = '\0';
|
|
strncat(hostname, p, slash - p);
|
|
|
|
if ((atsign = strrchr(hostname, '@')) == 0)
|
|
q = hostname;
|
|
else
|
|
q = atsign + 1;
|
|
|
|
if ((colon = strchr(q, ':')) != 0)
|
|
*colon = '\0';
|
|
|
|
if (colon && *(colon + 1)) {
|
|
unsigned long ul;
|
|
char *ep;
|
|
|
|
errno = 0;
|
|
ul = strtoul(colon + 1, &ep, 10);
|
|
if (*ep || errno != 0 || ul < 1 || ul > 65534) {
|
|
if (errno)
|
|
warn("`%s': invalid port in URL", uri);
|
|
else
|
|
warnx("`%s': invalid port in URL", uri);
|
|
return EX_USAGE;
|
|
}
|
|
|
|
port = ul;
|
|
} else {
|
|
port = 21;
|
|
}
|
|
|
|
p = slash + 1;
|
|
|
|
ftps = safe_malloc(sizeof *ftps);
|
|
ftps->ftp_password = 0;
|
|
ftps->ftp_user = 0;
|
|
|
|
/*
|
|
* Now, we have a copy of the hostname in hostname, the specified port
|
|
* (or the default value) in port, and p points to the filename part
|
|
* of the URI. We just need to check for a user in the hostname,
|
|
* and then save all the bits in our state.
|
|
*/
|
|
if (atsign) {
|
|
if (atsign[1] == '\0') {
|
|
warnx("`%s': malformed `ftp' hostname", hostname);
|
|
free(ftps);
|
|
return EX_USAGE;
|
|
}
|
|
|
|
*atsign = '\0';
|
|
if ((colon = strchr(hostname, ':')) != 0)
|
|
*colon = '\0';
|
|
if (hostname[0] == '\0') {
|
|
warnx("`%s': malformed `ftp' user", atsign + 1);
|
|
free(ftps);
|
|
return EX_USAGE;
|
|
}
|
|
if (colon != 0)
|
|
ftps->ftp_password = percent_decode(colon + 1);
|
|
ftps->ftp_user = percent_decode(hostname);
|
|
ftps->ftp_hostname = safe_strdup(atsign + 1);
|
|
} else
|
|
ftps->ftp_hostname = safe_strdup(hostname);
|
|
ftps->ftp_port = port;
|
|
|
|
/* Save the full path for error messages. */
|
|
ftps->ftp_remote_path = percent_decode(p);
|
|
|
|
/* Build a list of directory components plus the filename. */
|
|
ftps->ftp_remote_ndirs = 0;
|
|
q = p;
|
|
while ((q = strchr(q, '/')) != 0) {
|
|
q++;
|
|
ftps->ftp_remote_ndirs++;
|
|
}
|
|
path = safe_strdup(p);
|
|
if (ftps->ftp_remote_ndirs != 0) {
|
|
ftps->ftp_remote_dirs = safe_malloc(ftps->ftp_remote_ndirs *
|
|
sizeof(char *));
|
|
r = s = path = safe_strdup(p);
|
|
dp = ftps->ftp_remote_dirs;
|
|
while ((s = strchr(s, '/')) != 0) {
|
|
*s++ = '\0';
|
|
/*
|
|
* Skip double-slashes. According to RFC1738,
|
|
* double-slashes mean "send 'CWD '", which is
|
|
* a syntax error to most FTP servers. Instead,
|
|
* we just pretend that multiple slashes are a
|
|
* single slash.
|
|
*/
|
|
if (*r == '\0') {
|
|
warnx("skipping double slash in FTP URL; see man page or RFC1738.");
|
|
ftps->ftp_remote_ndirs--;
|
|
} else
|
|
*dp++ = percent_decode(r);
|
|
r = s;
|
|
}
|
|
} else {
|
|
ftps->ftp_remote_dirs = 0;
|
|
r = path;
|
|
}
|
|
if ((s = strchr(r, ';')) != 0 && strncmp(s, ";type=", 6) == 0) {
|
|
*s = '\0';
|
|
ftps->ftp_type = percent_decode(s+6);
|
|
} else
|
|
ftps->ftp_type = 0;
|
|
ftps->ftp_remote_file = percent_decode(r);
|
|
free(path);
|
|
|
|
if (fs->fs_outputfile == 0) {
|
|
fs->fs_outputfile = ftps->ftp_remote_file;
|
|
}
|
|
|
|
if (ftps->ftp_password == 0)
|
|
ftps->ftp_password = getenv("FTP_PASSWORD");
|
|
if (ftps->ftp_password != 0) {
|
|
ftps->ftp_password = safe_strdup(ftps->ftp_password);
|
|
} else {
|
|
char *pw;
|
|
const char *logname;
|
|
char localhost[MAXHOSTNAMELEN];
|
|
|
|
logname = getlogin();
|
|
if (logname == 0)
|
|
logname = "root";
|
|
gethostname(localhost, sizeof localhost);
|
|
pw = safe_malloc(strlen(logname) + 1 + strlen(localhost) + 1);
|
|
strcpy(pw, logname);
|
|
strcat(pw, "@");
|
|
strcat(pw, localhost);
|
|
ftps->ftp_password = pw;
|
|
setenv("FTP_PASSWORD", pw, 0); /* cache the result */
|
|
}
|
|
|
|
if (ftps->ftp_user == 0)
|
|
ftps->ftp_user = getenv("FTP_LOGIN");
|
|
if (ftps->ftp_user != 0)
|
|
ftps->ftp_user = safe_strdup(ftps->ftp_user);
|
|
|
|
fs->fs_proto = ftps;
|
|
fs->fs_close = ftp_close;
|
|
fs->fs_retrieve = ftp_retrieve;
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* The only URIs we can handle in the FTP proxy are FTP URLs.
|
|
* This makes it possible to take a few short cuts.
|
|
*/
|
|
static int
|
|
ftp_proxy_parse(struct fetch_state *fs, const char *uri)
|
|
{
|
|
int rv;
|
|
char *hostname;
|
|
char *port;
|
|
const char *user;
|
|
char *newuser;
|
|
unsigned portno;
|
|
struct ftp_state *ftps;
|
|
|
|
hostname = getenv("FTP_PROXY");
|
|
port = strchr(hostname, ':');
|
|
if (port == 0) {
|
|
portno = 21;
|
|
} else {
|
|
unsigned long ul;
|
|
char *ep;
|
|
|
|
/* All this to avoid modifying the environment. */
|
|
ep = alloca(strlen(hostname) + 1);
|
|
strcpy(ep, hostname);
|
|
port = ep + (port - hostname);
|
|
hostname = ep;
|
|
|
|
*port++ = '\0';
|
|
errno = 0;
|
|
ul = strtoul(port, &ep, 0);
|
|
if (*ep || !*port || errno != 0 || ul < 1 || ul > 65534) {
|
|
warnx("`%s': invalid port specification for FTP proxy",
|
|
port);
|
|
return EX_USAGE;
|
|
}
|
|
portno = ul;
|
|
}
|
|
|
|
/* ftp_parse() does most of the work; we can just fix things up */
|
|
rv = ftp_parse(fs, uri);
|
|
if (rv)
|
|
return rv;
|
|
/* Oops.. it got turned into a file: */
|
|
if (fs->fs_retrieve != ftp_retrieve) {
|
|
return 0;
|
|
}
|
|
|
|
ftps = fs->fs_proto;
|
|
|
|
user = ftps->ftp_user ? ftps->ftp_user : "anonymous";
|
|
/* user @ hostname [ @port ] \0 */
|
|
newuser = safe_malloc(strlen(user) + 1 + strlen(ftps->ftp_hostname)
|
|
+ ((ftps->ftp_port != 21) ? 6 : 0) + 1);
|
|
|
|
strcpy(newuser, user);
|
|
strcat(newuser, "@");
|
|
strcat(newuser, ftps->ftp_hostname);
|
|
if (ftps->ftp_port != 21) {
|
|
char numbuf[6];
|
|
|
|
snprintf(numbuf, sizeof(numbuf), "%d", ftps->ftp_port);
|
|
numbuf[sizeof(numbuf)-1] = '\0';
|
|
strcat(newuser, "@");
|
|
strcat(newuser, numbuf);
|
|
}
|
|
|
|
ftps->ftp_port = portno;
|
|
free(ftps->ftp_hostname);
|
|
ftps->ftp_hostname = safe_strdup(hostname);
|
|
free(ftps->ftp_user);
|
|
ftps->ftp_user = newuser;
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
ftp_close(struct fetch_state *fs)
|
|
{
|
|
struct ftp_state *ftps = fs->fs_proto;
|
|
int i;
|
|
char **dp;
|
|
|
|
if (ftps->ftp_user)
|
|
free(ftps->ftp_user);
|
|
free(ftps->ftp_hostname);
|
|
free(ftps->ftp_password);
|
|
free(ftps->ftp_remote_file);
|
|
for (i = 0, dp = ftps->ftp_remote_dirs; i < ftps->ftp_remote_ndirs; i++, dp++)
|
|
free(*dp);
|
|
if (ftps->ftp_remote_dirs)
|
|
free(ftps->ftp_remote_dirs);
|
|
free(ftps->ftp_remote_path);
|
|
if (ftps->ftp_type)
|
|
free(ftps->ftp_type);
|
|
free(ftps);
|
|
fs->fs_proto = 0;
|
|
fs->fs_outputfile = 0;
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
ftp_retrieve(struct fetch_state *fs)
|
|
{
|
|
struct ftp_state *ftps = fs->fs_proto;
|
|
FILE *ftp, *remote, *local;
|
|
char **dp;
|
|
int i, status;
|
|
off_t size;
|
|
off_t seekloc, wehave;
|
|
time_t modtime;
|
|
size_t readresult, writeresult;
|
|
|
|
fs->fs_status = "logging in to FTP server";
|
|
ftp = ftpLogin(ftps->ftp_hostname,
|
|
(char *)(ftps->ftp_user ? ftps->ftp_user : "anonymous"),
|
|
/* XXX ^^^^ bad API */
|
|
ftps->ftp_password, ftps->ftp_port, fs->fs_verbose > 1,
|
|
&status);
|
|
if (ftp == 0) {
|
|
warnx("%s: %s", ftps->ftp_hostname,
|
|
status ? ftpErrString(status) : hstrerror(h_errno));
|
|
return EX_IOERR;
|
|
}
|
|
fs->fs_status = "preparing for FTP transfer";
|
|
if (ftps->ftp_type && strcasecmp(ftps->ftp_type, "i") != 0) {
|
|
if (strcasecmp(ftps->ftp_type, "a") == 0)
|
|
ftpAscii(ftp);
|
|
else {
|
|
warnx("unknown or unsupported type %s", ftps->ftp_type);
|
|
return EX_USAGE;
|
|
}
|
|
} else
|
|
ftpBinary(ftp);
|
|
ftpPassive(ftp, fs->fs_passive_mode);
|
|
for (i = 0, dp = ftps->ftp_remote_dirs; i < ftps->ftp_remote_ndirs; i++, dp++) {
|
|
if ((status = ftpChdir(ftp, *dp)) != 0) {
|
|
warnx("%s: %s: %s", ftps->ftp_hostname,
|
|
*dp, ftpErrString(status));
|
|
return EX_IOERR;
|
|
}
|
|
}
|
|
size = ftpGetSize(ftp, ftps->ftp_remote_file);
|
|
|
|
if (fs->fs_reportsize) {
|
|
fclose(ftp);
|
|
if (size == -1) {
|
|
warnx("%s: size not known\n", fs->fs_outputfile);
|
|
printf("Unknown\n");
|
|
return 1;
|
|
}
|
|
else {
|
|
printf("%qd\n", (quad_t)size);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
if (size > 0 && fs->fs_expectedsize != -1 && size != fs->fs_expectedsize) {
|
|
warnx("%s: size mismatch, expected=%lu / actual=%lu",
|
|
ftps->ftp_remote_path,
|
|
(unsigned long)fs->fs_expectedsize,
|
|
(unsigned long)size);
|
|
return EX_DATAERR;
|
|
}
|
|
modtime = ftpGetModtime(ftp, ftps->ftp_remote_file);
|
|
if (modtime <= 0) { /* xxx */
|
|
warnx("%s: cannot get remote modification time",
|
|
ftps->ftp_remote_path);
|
|
modtime = -1;
|
|
}
|
|
fs->fs_modtime = modtime;
|
|
seekloc = wehave = 0;
|
|
if (fs->fs_restart || fs->fs_mirror) {
|
|
struct stat stab;
|
|
|
|
if (fs->fs_outputfile[0] == '-'
|
|
&& fs->fs_outputfile[1] == '\0')
|
|
status = fstat(STDOUT_FILENO, &stab);
|
|
else
|
|
status = stat(fs->fs_outputfile, &stab);
|
|
if (status < 0) {
|
|
stab.st_mtime = -1;
|
|
stab.st_size = 0;
|
|
}
|
|
if (status == 0 && !S_ISREG(stab.st_mode)) {
|
|
fs->fs_restart = 0;
|
|
fs->fs_mirror = 0;
|
|
}
|
|
if (fs->fs_mirror && stab.st_size == size
|
|
&& modtime <= stab.st_mtime) {
|
|
fclose(ftp);
|
|
return 0;
|
|
}
|
|
if (fs->fs_restart) {
|
|
if (stab.st_size != 0 && stab.st_size < size)
|
|
seekloc = wehave = stab.st_size;
|
|
}
|
|
}
|
|
|
|
fs->fs_status = "retrieving file from FTP server";
|
|
remote = ftpGet(ftp, ftps->ftp_remote_file, &seekloc);
|
|
if (remote == 0) {
|
|
if (ftpErrno(ftp)) {
|
|
warnx("ftp://%s/%s: FTP error:",
|
|
ftps->ftp_hostname, ftps->ftp_remote_path);
|
|
warnx("%s", ftpErrString(ftpErrno(ftp)));
|
|
fclose(ftp);
|
|
return EX_IOERR;
|
|
} else {
|
|
warn("ftpGet");
|
|
return EX_OSERR;
|
|
}
|
|
}
|
|
|
|
if (fs->fs_outputfile[0] == '-' && fs->fs_outputfile[1] == '\0')
|
|
local = fopen("/dev/stdout", wehave ? "a" : "w");
|
|
else
|
|
local = fopen(fs->fs_outputfile, wehave ? "a" : "w");
|
|
if (local == 0) {
|
|
warn("%s", fs->fs_outputfile);
|
|
fclose(remote);
|
|
fclose(ftp);
|
|
return EX_OSERR;
|
|
}
|
|
|
|
if (fs->fs_timeout) {
|
|
char buf[sizeof("18446744073709551616")]; /* 2**64 */
|
|
snprintf(buf, sizeof buf, "%d", fs->fs_timeout);
|
|
setenv("FTP_TIMEOUT", buf, 1);
|
|
} else {
|
|
char *env = getenv("FTP_TIMEOUT");
|
|
char *ep;
|
|
unsigned long ul;
|
|
|
|
if (env) {
|
|
errno = 0;
|
|
ul = strtoul(env, &ep, 0);
|
|
if (*env && *ep == '\0' && errno == 0 && ul <= INT_MAX)
|
|
fs->fs_timeout = ul;
|
|
else
|
|
warnx("`%s': invalid FTP timeout", env);
|
|
}
|
|
}
|
|
|
|
display(fs, size, wehave);
|
|
setup_sigalrm();
|
|
|
|
do {
|
|
char buf[BUFFER_SIZE];
|
|
|
|
alarm(fs->fs_timeout);
|
|
readresult = fread(buf, 1, sizeof buf, remote);
|
|
alarm(0);
|
|
if (readresult == 0)
|
|
break;
|
|
display(fs, size, readresult);
|
|
writeresult = fwrite(buf, 1, readresult, local);
|
|
} while (writeresult == readresult);
|
|
unsetup_sigalrm();
|
|
|
|
if (ferror(remote)) {
|
|
warn("reading remote file from %s", ftps->ftp_hostname);
|
|
fclose(local);
|
|
fclose(remote);
|
|
fclose(ftp);
|
|
rm(fs);
|
|
return EX_IOERR;
|
|
} else if(ferror(local)) {
|
|
warn("%s", fs->fs_outputfile);
|
|
fclose(local);
|
|
fclose(remote);
|
|
fclose(ftp);
|
|
rm(fs);
|
|
return EX_IOERR;
|
|
}
|
|
|
|
fclose(local);
|
|
fclose(remote);
|
|
fclose(ftp);
|
|
if (display(fs, size, -1) != 0)
|
|
return EX_PROTOCOL;
|
|
adjmodtime(fs);
|
|
return 0;
|
|
}
|