MegaGlest/mk/linux/mojosetup/libfetch/ftp.c
2010-04-25 03:01:17 +00:00

1395 lines
31 KiB
C

/*-
* Copyright (c) 1998-2004 Dag-Erling Coïdan Smørgrav
* 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
* in this position and unchanged.
* 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. The name of the author may not be used to endorse or promote products
* derived from this software without specific prior written permission
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 AUTHOR 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 "mojosetup_libfetch.h"
#if SUPPORT_URL_FTP
#if !sun /* __MOJOSETUP__ Solaris support... */
#include <sys/cdefs.h>
__FBSDID("$FreeBSD: src/lib/libfetch/ftp.c,v 1.96 2007/04/22 22:33:29 njl Exp $");
#endif
/*
* Portions of this code were taken from or based on ftpio.c:
*
* ----------------------------------------------------------------------------
* "THE BEER-WARE LICENSE" (Revision 42):
* <phk@FreeBSD.org> wrote this file. As long as you retain this notice you
* can do whatever you want with this stuff. If we meet some day, and you think
* this stuff is worth it, you can buy me a beer in return. Poul-Henning Kamp
* ----------------------------------------------------------------------------
*
* Major Changelog:
*
* Dag-Erling Coïdan Smørgrav
* 9 Jun 1998
*
* Incorporated into libfetch
*
* Jordan K. Hubbard
* 17 Jan 1996
*
* Turned inside out. Now returns xfers as new file ids, not as a special
* `state' of FTP_t
*
* $ftpioId: ftpio.c,v 1.30 1998/04/11 07:28:53 phk Exp $
*
*/
#include <sys/param.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <ctype.h>
#if !sun /* __MOJOSETUP__ Solaris support... */
#include <err.h>
#endif
#include <errno.h>
#include <fcntl.h>
#include <netdb.h>
#include <stdarg.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <unistd.h>
#include "fetch.h"
#include "common.h"
#include "ftperr.h"
#define FTP_ANONYMOUS_USER "anonymous"
#define FTP_CONNECTION_ALREADY_OPEN 125
#define FTP_OPEN_DATA_CONNECTION 150
#define FTP_OK 200
#define FTP_FILE_STATUS 213
#define FTP_SERVICE_READY 220
#define FTP_TRANSFER_COMPLETE 226
#define FTP_PASSIVE_MODE 227
#define FTP_LPASSIVE_MODE 228
#define FTP_EPASSIVE_MODE 229
#define FTP_LOGGED_IN 230
#define FTP_FILE_ACTION_OK 250
#define FTP_DIRECTORY_CREATED 257 /* multiple meanings */
#define FTP_FILE_CREATED 257 /* multiple meanings */
#define FTP_WORKING_DIRECTORY 257 /* multiple meanings */
#define FTP_NEED_PASSWORD 331
#define FTP_NEED_ACCOUNT 332
#define FTP_FILE_OK 350
#define FTP_SYNTAX_ERROR 500
#define FTP_PROTOCOL_ERROR 999
static struct url cached_host;
static conn_t *cached_connection;
#define isftpreply(foo) (isdigit(foo[0]) && isdigit(foo[1]) \
&& isdigit(foo[2]) \
&& (foo[3] == ' ' || foo[3] == '\0'))
#define isftpinfo(foo) (isdigit(foo[0]) && isdigit(foo[1]) \
&& isdigit(foo[2]) && foo[3] == '-')
/*
* Translate IPv4 mapped IPv6 address to IPv4 address
*/
static void
unmappedaddr(struct sockaddr_in6 *sin6)
{
struct sockaddr_in *sin4;
u_int32_t addr;
int port;
if (sin6->sin6_family != AF_INET6 ||
!IN6_IS_ADDR_V4MAPPED(&sin6->sin6_addr))
return;
sin4 = (struct sockaddr_in *)sin6;
addr = *(u_int32_t *)&sin6->sin6_addr.s6_addr[12];
port = sin6->sin6_port;
memset(sin4, 0, sizeof(struct sockaddr_in));
sin4->sin_addr.s_addr = addr;
sin4->sin_port = port;
sin4->sin_family = AF_INET;
#if __MOJOSETUP__
#if FREEBSD
sin4->sin_len = sizeof(struct sockaddr_in);
#endif
#endif
}
/*
* Get server response
*/
static int
_ftp_chkerr(conn_t *conn)
{
if (_fetch_getln(conn) == -1) {
_fetch_syserr();
return (-1);
}
if (isftpinfo(conn->buf)) {
while (conn->buflen && !isftpreply(conn->buf)) {
if (_fetch_getln(conn) == -1) {
_fetch_syserr();
return (-1);
}
}
}
while (conn->buflen && isspace(conn->buf[conn->buflen - 1]))
conn->buflen--;
conn->buf[conn->buflen] = '\0';
if (!isftpreply(conn->buf)) {
_ftp_seterr(FTP_PROTOCOL_ERROR);
return (-1);
}
conn->err = (conn->buf[0] - '0') * 100
+ (conn->buf[1] - '0') * 10
+ (conn->buf[2] - '0');
return (conn->err);
}
/*
* Send a command and check reply
*/
#if __MOJOSETUP__
static int _ftp_cmd(conn_t *conn, const char *fmt, ...) ISPRINTF(2,3);
#endif
static int
_ftp_cmd(conn_t *conn, const char *fmt, ...)
{
va_list ap;
size_t len;
char *msg;
int r;
va_start(ap, fmt);
len = vasprintf(&msg, fmt, ap);
va_end(ap);
if (msg == NULL) {
errno = ENOMEM;
_fetch_syserr();
return (-1);
}
r = _fetch_putln(conn, msg, len);
free(msg);
if (r == -1) {
_fetch_syserr();
return (-1);
}
return (_ftp_chkerr(conn));
}
/*
* Return a pointer to the filename part of a path
*/
static const char *
_ftp_filename(const char *file, int *len, int *type)
{
const char *s;
if ((s = strrchr(file, '/')) == NULL)
s = file;
else
s = s + 1;
*len = strlen(s);
if (*len > 7 && strncmp(s + *len - 7, ";type=", 6) == 0) {
*type = s[*len - 1];
*len -= 7;
} else {
*type = '\0';
}
return (s);
}
/*
* Get current working directory from the reply to a CWD, PWD or CDUP
* command.
*/
static int
_ftp_pwd(conn_t *conn, char *pwd, size_t pwdlen)
{
char *src, *dst, *end;
int q;
if (conn->err != FTP_WORKING_DIRECTORY &&
conn->err != FTP_FILE_ACTION_OK)
return (FTP_PROTOCOL_ERROR);
end = conn->buf + conn->buflen;
src = conn->buf + 4;
if (src >= end || *src++ != '"')
return (FTP_PROTOCOL_ERROR);
for (q = 0, dst = pwd; src < end && pwdlen--; ++src) {
if (!q && *src == '"')
q = 1;
else if (q && *src != '"')
break;
else if (q)
*dst++ = '"', q = 0;
else
*dst++ = *src;
}
if (!pwdlen)
return (FTP_PROTOCOL_ERROR);
*dst = '\0';
#if 0
DEBUG(fprintf(stderr, "pwd: [%s]\n", pwd));
#endif
return (FTP_OK);
}
/*
* Change working directory to the directory that contains the specified
* file.
*/
static int
_ftp_cwd(conn_t *conn, const char *file)
{
const char *beg, *end;
char pwd[PATH_MAX];
int e, i, len;
/* If no slashes in name, no need to change dirs. */
if ((end = strrchr(file, '/')) == NULL)
return (0);
if ((e = _ftp_cmd(conn, "PWD")) != FTP_WORKING_DIRECTORY ||
(e = _ftp_pwd(conn, pwd, sizeof(pwd))) != FTP_OK) {
_ftp_seterr(e);
return (-1);
}
for (;;) {
len = strlen(pwd);
/* Look for a common prefix between PWD and dir to fetch. */
for (i = 0; i <= len && i <= end - file; ++i)
if (pwd[i] != file[i])
break;
#if 0
DEBUG(fprintf(stderr, "have: [%.*s|%s]\n", i, pwd, pwd + i));
DEBUG(fprintf(stderr, "want: [%.*s|%s]\n", i, file, file + i));
#endif
/* Keep going up a dir until we have a matching prefix. */
if (pwd[i] == '\0' && (file[i - 1] == '/' || file[i] == '/'))
break;
if ((e = _ftp_cmd(conn, "CDUP")) != FTP_FILE_ACTION_OK ||
(e = _ftp_cmd(conn, "PWD")) != FTP_WORKING_DIRECTORY ||
(e = _ftp_pwd(conn, pwd, sizeof(pwd))) != FTP_OK) {
_ftp_seterr(e);
return (-1);
}
}
#ifdef FTP_COMBINE_CWDS
/* Skip leading slashes, even "////". */
for (beg = file + i; beg < end && *beg == '/'; ++beg, ++i)
/* nothing */ ;
/* If there is no trailing dir, we're already there. */
if (beg >= end)
return (0);
/* Change to the directory all in one chunk (e.g., foo/bar/baz). */
e = _ftp_cmd(conn, "CWD %.*s", (int)(end - beg), beg);
if (e == FTP_FILE_ACTION_OK)
return (0);
#endif /* FTP_COMBINE_CWDS */
/* That didn't work so go back to legacy behavior (multiple CWDs). */
for (beg = file + i; beg < end; beg = file + i + 1) {
while (*beg == '/')
++beg, ++i;
for (++i; file + i < end && file[i] != '/'; ++i)
/* nothing */ ;
#if __MOJOSETUP__
e = _ftp_cmd(conn, "CWD %.*s", (int) (file + i - beg), beg);
#else
e = _ftp_cmd(conn, "CWD %.*s", file + i - beg, beg);
#endif
if (e != FTP_FILE_ACTION_OK) {
_ftp_seterr(e);
return (-1);
}
}
return (0);
}
/*
* Set transfer mode and data type
*/
static int
_ftp_mode_type(conn_t *conn, int mode, int type)
{
int e;
switch (mode) {
case 0:
case 's':
mode = 'S';
case 'S':
break;
default:
return (FTP_PROTOCOL_ERROR);
}
if ((e = _ftp_cmd(conn, "MODE %c", mode)) != FTP_OK) {
if (mode == 'S') {
/*
* Stream mode is supposed to be the default - so
* much so that some servers not only do not
* support any other mode, but do not support the
* MODE command at all.
*
* If "MODE S" fails, it is unlikely that we
* previously succeeded in setting a different
* mode. Therefore, we simply hope that the
* server is already in the correct mode, and
* silently ignore the failure.
*/
} else {
return (e);
}
}
switch (type) {
case 0:
case 'i':
type = 'I';
case 'I':
break;
case 'a':
type = 'A';
case 'A':
break;
case 'd':
type = 'D';
case 'D':
/* can't handle yet */
default:
return (FTP_PROTOCOL_ERROR);
}
if ((e = _ftp_cmd(conn, "TYPE %c", type)) != FTP_OK)
return (e);
return (FTP_OK);
}
/*
* Request and parse file stats
*/
static int
_ftp_stat(conn_t *conn, const char *file, struct url_stat *us)
{
char *ln;
const char *filename;
int filenamelen, type;
struct tm tm;
time_t t;
int e;
us->size = -1;
us->atime = us->mtime = 0;
filename = _ftp_filename(file, &filenamelen, &type);
if ((e = _ftp_mode_type(conn, 0, type)) != FTP_OK) {
_ftp_seterr(e);
return (-1);
}
e = _ftp_cmd(conn, "SIZE %.*s", filenamelen, filename);
if (e != FTP_FILE_STATUS) {
_ftp_seterr(e);
return (-1);
}
for (ln = conn->buf + 4; *ln && isspace(*ln); ln++)
/* nothing */ ;
for (us->size = 0; *ln && isdigit(*ln); ln++)
us->size = us->size * 10 + *ln - '0';
if (*ln && !isspace(*ln)) {
_ftp_seterr(FTP_PROTOCOL_ERROR);
us->size = -1;
return (-1);
}
if (us->size == 0)
us->size = -1;
DEBUG(fprintf(stderr, "size: [%lld]\n", (long long)us->size));
e = _ftp_cmd(conn, "MDTM %.*s", filenamelen, filename);
if (e != FTP_FILE_STATUS) {
_ftp_seterr(e);
return (-1);
}
for (ln = conn->buf + 4; *ln && isspace(*ln); ln++)
/* nothing */ ;
switch (strspn(ln, "0123456789")) {
case 14:
break;
case 15:
ln++;
ln[0] = '2';
ln[1] = '0';
break;
default:
_ftp_seterr(FTP_PROTOCOL_ERROR);
return (-1);
}
if (sscanf(ln, "%04d%02d%02d%02d%02d%02d",
&tm.tm_year, &tm.tm_mon, &tm.tm_mday,
&tm.tm_hour, &tm.tm_min, &tm.tm_sec) != 6) {
_ftp_seterr(FTP_PROTOCOL_ERROR);
return (-1);
}
tm.tm_mon--;
tm.tm_year -= 1900;
tm.tm_isdst = -1;
t = timegm(&tm);
if (t == (time_t)-1)
t = time(NULL);
us->mtime = t;
us->atime = t;
DEBUG(fprintf(stderr,
"last modified: [%04d-%02d-%02d %02d:%02d:%02d]\n",
tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday,
tm.tm_hour, tm.tm_min, tm.tm_sec));
return (0);
}
/*
* I/O functions for FTP
*/
struct ftpio {
conn_t *cconn; /* Control connection */
conn_t *dconn; /* Data connection */
int dir; /* Direction */
int eof; /* EOF reached */
int err; /* Error code */
#if __MOJOSETUP__
int64 bytes_read;
int64 length;
#endif
};
#if !__MOJOSETUP__
static int _ftp_readfn(void *, char *, int);
static int _ftp_writefn(void *, const char *, int);
static fpos_t _ftp_seekfn(void *, fpos_t, int);
static int _ftp_closefn(void *);
#endif
#if __MOJOSETUP__
static boolean MojoInput_ftp_ready(MojoInput *v)
{
boolean retval = true;
struct ftpio *io = (struct ftpio *)v->opaque;
if (io->dconn != NULL)
{
fd_set rfds;
struct timeval tv;
tv.tv_sec = tv.tv_usec = 0;
FD_ZERO(&rfds);
FD_SET(io->dconn->sd, &rfds);
retval = (select(io->dconn->sd+1, &rfds, NULL, NULL, &tv) > 0);
} // if
return retval;
}
static int64 MojoInput_ftp_tell(MojoInput *v)
{
struct ftpio *io = (struct ftpio *)v->opaque;
return io->bytes_read;
}
static int64 MojoInput_ftp_length(MojoInput *v)
{
struct ftpio *io = (struct ftpio *)v->opaque;
return io->length;
}
static MojoInput* MojoInput_ftp_duplicate(MojoInput *v)
{
return NULL; // !!! FIXME: fatal?
}
#endif
#if __MOJOSETUP__
static int64 MojoInput_ftp_read(MojoInput *v, void *buf, uint32 len)
#else
static int
_ftp_readfn(void *v, char *buf, int len)
#endif
{
struct ftpio *io;
int r;
#if __MOJOSETUP__
io = (struct ftpio *)v->opaque;
#else
io = (struct ftpio *)v;
if (io == NULL) {
errno = EBADF;
return (-1);
}
#endif
if (io->cconn == NULL || io->dconn == NULL || io->dir == O_WRONLY) {
errno = EBADF;
return (-1);
}
if (io->err) {
errno = io->err;
return (-1);
}
if (io->eof)
return (0);
r = _fetch_read(io->dconn, buf, len);
#if __MOJOSETUP__
if (r > 0) {
io->bytes_read += r;
return (r);
}
#else
if (r > 0)
return (r);
#endif
if (r == 0) {
io->eof = 1;
return (0);
}
if (errno != EINTR)
io->err = errno;
return (-1);
}
#if !__MOJOSETUP__
static int
_ftp_writefn(void *v, const char *buf, int len)
{
struct ftpio *io;
int w;
io = (struct ftpio *)v;
if (io == NULL) {
errno = EBADF;
return (-1);
}
if (io->cconn == NULL || io->dconn == NULL || io->dir == O_RDONLY) {
errno = EBADF;
return (-1);
}
if (io->err) {
errno = io->err;
return (-1);
}
w = _fetch_write(io->dconn, buf, len);
if (w >= 0)
return (w);
if (errno != EINTR)
io->err = errno;
return (-1);
}
#endif
#if __MOJOSETUP__
static boolean MojoInput_ftp_seek(MojoInput *v, uint64 pos)
#else
static fpos_t
_ftp_seekfn(void *v, fpos_t pos __unused, int whence __unused)
#endif
{
struct ftpio *io;
#if __MOJOSETUP__
io = (struct ftpio *)v->opaque;
#else
io = (struct ftpio *)v;
if (io == NULL) {
errno = EBADF;
return (-1);
}
#endif
errno = ESPIPE;
return (-1);
}
static int
_ftp_closefn(void *v)
{
struct ftpio *io;
int r;
#if __MOJOSETUP__
io = (struct ftpio *) ((MojoInput *)v)->opaque;
#else
io = (struct ftpio *)v;
if (io == NULL) {
errno = EBADF;
return (-1);
}
#endif
if (io->dir == -1)
return (0);
if (io->cconn == NULL || io->dconn == NULL) {
errno = EBADF;
return (-1);
}
_fetch_close(io->dconn);
io->dir = -1;
io->dconn = NULL;
DEBUG(fprintf(stderr, "Waiting for final status\n"));
r = _ftp_chkerr(io->cconn);
if (io->cconn == cached_connection && io->cconn->ref == 1)
cached_connection = NULL;
_fetch_close(io->cconn);
free(io);
#if __MOJOSETUP__
free(v);
#endif
return (r == FTP_TRANSFER_COMPLETE) ? 0 : -1;
}
#if __MOJOSETUP__
static void MojoInput_ftp_close(MojoInput *v)
{
_ftp_closefn(v);
}
#endif
#if __MOJOSETUP__
static MojoInput *
#else
static FILE *
#endif
_ftp_setup(conn_t *cconn, conn_t *dconn, int mode)
{
struct ftpio *io;
#if __MOJOSETUP__
MojoInput *f = NULL;
#else
FILE *f;
#endif
if (cconn == NULL || dconn == NULL)
return (NULL);
if ((io = malloc(sizeof(*io))) == NULL)
return (NULL);
io->cconn = cconn;
io->dconn = dconn;
io->dir = mode;
io->eof = io->err = 0;
#if __MOJOSETUP__
io->bytes_read = 0;
io->length = -1;
f = (MojoInput *) xmalloc(sizeof (MojoInput));
f->ready = MojoInput_ftp_ready;
f->read = MojoInput_ftp_read;
f->seek = MojoInput_ftp_seek;
f->tell = MojoInput_ftp_tell;
f->length = MojoInput_ftp_length;
f->duplicate = MojoInput_ftp_duplicate;
f->close = MojoInput_ftp_close;
f->opaque = io;
#else
f = funopen(io, _ftp_readfn, _ftp_writefn, _ftp_seekfn, _ftp_closefn);
if (f == NULL)
free(io);
#endif
return (f);
}
/*
* Transfer file
*/
#if __MOJOSETUP__
static MojoInput *
#else
static FILE *
#endif
_ftp_transfer(conn_t *conn, const char *oper, const char *file,
int mode, off_t offset, const char *flags)
{
struct sockaddr_storage sa;
struct sockaddr_in6 *sin6;
struct sockaddr_in *sin4;
const char *bindaddr;
const char *filename;
int filenamelen, type;
int low, pasv, verbose;
int e, sd = -1;
socklen_t l;
char *s;
#if __MOJOSETUP__
MojoInput *df = NULL;
#else
FILE *df;
#endif
/* check flags */
low = CHECK_FLAG('l');
pasv = CHECK_FLAG('p');
verbose = CHECK_FLAG('v');
/* passive mode */
if (!pasv)
pasv = ((s = getenv("FTP_PASSIVE_MODE")) != NULL &&
strncasecmp(s, "no", 2) != 0);
/* isolate filename */
filename = _ftp_filename(file, &filenamelen, &type);
/* set transfer mode and data type */
if ((e = _ftp_mode_type(conn, 0, type)) != FTP_OK)
goto ouch;
/* find our own address, bind, and listen */
l = sizeof(sa);
if (getsockname(conn->sd, (struct sockaddr *)&sa, &l) == -1)
goto sysouch;
if (sa.ss_family == AF_INET6)
unmappedaddr((struct sockaddr_in6 *)&sa);
/* open data socket */
if ((sd = socket(sa.ss_family, SOCK_STREAM, IPPROTO_TCP)) == -1) {
_fetch_syserr();
return (NULL);
}
if (pasv) {
u_char addr[64];
char *ln, *p;
unsigned int i;
int port;
/* send PASV command */
if (verbose)
_fetch_info("setting passive mode");
switch (sa.ss_family) {
case AF_INET:
if ((e = _ftp_cmd(conn, "PASV")) != FTP_PASSIVE_MODE)
goto ouch;
break;
case AF_INET6:
if ((e = _ftp_cmd(conn, "EPSV")) != FTP_EPASSIVE_MODE) {
if (e == -1)
goto ouch;
if ((e = _ftp_cmd(conn, "LPSV")) !=
FTP_LPASSIVE_MODE)
goto ouch;
}
break;
default:
e = FTP_PROTOCOL_ERROR; /* XXX: error code should be prepared */
goto ouch;
}
/*
* Find address and port number. The reply to the PASV command
* is IMHO the one and only weak point in the FTP protocol.
*/
ln = conn->buf;
switch (e) {
case FTP_PASSIVE_MODE:
case FTP_LPASSIVE_MODE:
for (p = ln + 3; *p && !isdigit(*p); p++)
/* nothing */ ;
if (!*p) {
e = FTP_PROTOCOL_ERROR;
goto ouch;
}
l = (e == FTP_PASSIVE_MODE ? 6 : 21);
for (i = 0; *p && i < l; i++, p++)
addr[i] = strtol(p, &p, 10);
if (i < l) {
e = FTP_PROTOCOL_ERROR;
goto ouch;
}
break;
case FTP_EPASSIVE_MODE:
for (p = ln + 3; *p && *p != '('; p++)
/* nothing */ ;
if (!*p) {
e = FTP_PROTOCOL_ERROR;
goto ouch;
}
++p;
if (sscanf(p, "%c%c%c%d%c", &addr[0], &addr[1], &addr[2],
&port, &addr[3]) != 5 ||
addr[0] != addr[1] ||
addr[0] != addr[2] || addr[0] != addr[3]) {
e = FTP_PROTOCOL_ERROR;
goto ouch;
}
break;
}
/* seek to required offset */
if (offset)
if (_ftp_cmd(conn, "REST %lu", (u_long)offset) != FTP_FILE_OK)
goto sysouch;
/* construct sockaddr for data socket */
l = sizeof(sa);
if (getpeername(conn->sd, (struct sockaddr *)&sa, &l) == -1)
goto sysouch;
if (sa.ss_family == AF_INET6)
unmappedaddr((struct sockaddr_in6 *)&sa);
switch (sa.ss_family) {
case AF_INET6:
sin6 = (struct sockaddr_in6 *)&sa;
if (e == FTP_EPASSIVE_MODE)
sin6->sin6_port = htons(port);
else {
#if __MOJOSETUP__
memmove((char *)&sin6->sin6_addr, addr + 2, 16);
memmove((char *)&sin6->sin6_port, addr + 19, 2);
#else
bcopy(addr + 2, (char *)&sin6->sin6_addr, 16);
bcopy(addr + 19, (char *)&sin6->sin6_port, 2);
#endif
}
break;
case AF_INET:
sin4 = (struct sockaddr_in *)&sa;
if (e == FTP_EPASSIVE_MODE)
sin4->sin_port = htons(port);
else {
#if __MOJOSETUP__
memmove((char *)&sin4->sin_addr, addr, 4);
memmove((char *)&sin4->sin_port, addr + 4, 2);
#else
bcopy(addr, (char *)&sin4->sin_addr, 4);
bcopy(addr + 4, (char *)&sin4->sin_port, 2);
#endif
}
break;
default:
e = FTP_PROTOCOL_ERROR; /* XXX: error code should be prepared */
break;
}
/* connect to data port */
if (verbose)
_fetch_info("opening data connection");
bindaddr = getenv("FETCH_BIND_ADDRESS");
if (bindaddr != NULL && *bindaddr != '\0' &&
_fetch_bind(sd, sa.ss_family, bindaddr) != 0)
goto sysouch;
#if __MOJOSETUP__
{
int socklen = sizeof (struct sockaddr_in);
if (sa.ss_family == AF_INET6)
socklen = sizeof (struct sockaddr_in6);
if (connect(sd, (struct sockaddr *)&sa, socklen) == -1)
goto sysouch;
}
#else
if (connect(sd, (struct sockaddr *)&sa, sa.ss_len) == -1)
goto sysouch;
#endif
/* make the server initiate the transfer */
if (verbose)
_fetch_info("initiating transfer");
e = _ftp_cmd(conn, "%s %.*s", oper, filenamelen, filename);
if (e != FTP_CONNECTION_ALREADY_OPEN && e != FTP_OPEN_DATA_CONNECTION)
goto ouch;
} else {
u_int32_t a;
u_short p;
int arg, d;
char *ap;
char hname[INET6_ADDRSTRLEN];
#if __MOJOSETUP__
arg = 0;
#if FREEBSD
switch (sa.ss_family) {
case AF_INET6:
((struct sockaddr_in6 *)&sa)->sin6_port = 0;
#ifdef IPV6_PORTRANGE
arg = low ? IPV6_PORTRANGE_DEFAULT : IPV6_PORTRANGE_HIGH;
if (setsockopt(sd, IPPROTO_IPV6, IPV6_PORTRANGE,
(char *)&arg, sizeof(arg)) == -1)
goto sysouch;
#endif
break;
case AF_INET:
((struct sockaddr_in *)&sa)->sin_port = 0;
arg = low ? IP_PORTRANGE_DEFAULT : IP_PORTRANGE_HIGH;
if (setsockopt(sd, IPPROTO_IP, IP_PORTRANGE,
(char *)&arg, sizeof(arg)) == -1)
goto sysouch;
break;
}
#endif
#endif
if (verbose)
_fetch_info("binding data socket");
#if __MOJOSETUP__
{
int socklen = sizeof (struct sockaddr_in);
if (sa.ss_family == AF_INET6)
socklen = sizeof (struct sockaddr_in6);
if (bind(sd, (struct sockaddr *)&sa, socklen) == -1)
goto sysouch;
}
#else
if (bind(sd, (struct sockaddr *)&sa, sa.ss_len) == -1)
goto sysouch;
#endif
if (listen(sd, 1) == -1)
goto sysouch;
/* find what port we're on and tell the server */
if (getsockname(sd, (struct sockaddr *)&sa, &l) == -1)
goto sysouch;
switch (sa.ss_family) {
case AF_INET:
sin4 = (struct sockaddr_in *)&sa;
a = ntohl(sin4->sin_addr.s_addr);
p = ntohs(sin4->sin_port);
e = _ftp_cmd(conn, "PORT %d,%d,%d,%d,%d,%d",
(a >> 24) & 0xff, (a >> 16) & 0xff,
(a >> 8) & 0xff, a & 0xff,
(p >> 8) & 0xff, p & 0xff);
break;
case AF_INET6:
#define UC(b) (((int)b)&0xff)
e = -1;
sin6 = (struct sockaddr_in6 *)&sa;
sin6->sin6_scope_id = 0;
#if __MOJOSETUP__
if (getnameinfo((struct sockaddr *)&sa, sizeof (struct sockaddr_in6),
#else
if (getnameinfo((struct sockaddr *)&sa, sa.ss_len,
#endif
hname, sizeof(hname),
NULL, 0, NI_NUMERICHOST) == 0) {
e = _ftp_cmd(conn, "EPRT |%d|%s|%d|", 2, hname,
htons(sin6->sin6_port));
if (e == -1)
goto ouch;
}
if (e != FTP_OK) {
ap = (char *)&sin6->sin6_addr;
e = _ftp_cmd(conn,
"LPRT %d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d",
6, 16,
UC(ap[0]), UC(ap[1]), UC(ap[2]), UC(ap[3]),
UC(ap[4]), UC(ap[5]), UC(ap[6]), UC(ap[7]),
UC(ap[8]), UC(ap[9]), UC(ap[10]), UC(ap[11]),
UC(ap[12]), UC(ap[13]), UC(ap[14]), UC(ap[15]),
2,
(ntohs(sin6->sin6_port) >> 8) & 0xff,
ntohs(sin6->sin6_port) & 0xff);
}
break;
default:
e = FTP_PROTOCOL_ERROR; /* XXX: error code should be prepared */
goto ouch;
}
if (e != FTP_OK)
goto ouch;
/* seek to required offset */
if (offset)
if (_ftp_cmd(conn, "REST %ju", (uintmax_t)offset) != FTP_FILE_OK)
goto sysouch;
/* make the server initiate the transfer */
if (verbose)
_fetch_info("initiating transfer");
e = _ftp_cmd(conn, "%s %.*s", oper, filenamelen, filename);
if (e != FTP_CONNECTION_ALREADY_OPEN && e != FTP_OPEN_DATA_CONNECTION)
goto ouch;
/* accept the incoming connection and go to town */
if ((d = accept(sd, NULL, NULL)) == -1)
goto sysouch;
close(sd);
sd = d;
}
if ((df = _ftp_setup(conn, _fetch_reopen(sd), mode)) == NULL)
goto sysouch;
return (df);
sysouch:
_fetch_syserr();
if (sd >= 0)
close(sd);
return (NULL);
ouch:
if (e != -1)
_ftp_seterr(e);
if (sd >= 0)
close(sd);
return (NULL);
}
/*
* Authenticate
*/
static int
_ftp_authenticate(conn_t *conn, struct url *url, struct url *purl)
{
const char *user, *pwd, *logname;
char pbuf[MAXHOSTNAMELEN + MAXLOGNAME + 1];
int e, len;
/* XXX FTP_AUTH, and maybe .netrc */
/* send user name and password */
if (url->user[0] == '\0')
_fetch_netrc_auth(url);
user = url->user;
if (*user == '\0')
user = getenv("FTP_LOGIN");
if (user == NULL || *user == '\0')
user = FTP_ANONYMOUS_USER;
if (purl && url->port == _fetch_default_port(url->scheme))
e = _ftp_cmd(conn, "USER %s@%s", user, url->host);
else if (purl)
e = _ftp_cmd(conn, "USER %s@%s@%d", user, url->host, url->port);
else
e = _ftp_cmd(conn, "USER %s", user);
/* did the server request a password? */
if (e == FTP_NEED_PASSWORD) {
pwd = url->pwd;
if (*pwd == '\0')
pwd = getenv("FTP_PASSWORD");
if (pwd == NULL || *pwd == '\0') {
if ((logname = getlogin()) == 0)
logname = FTP_ANONYMOUS_USER;
if ((len = snprintf(pbuf, MAXLOGNAME + 1, "%s@", logname)) < 0)
len = 0;
else if (len > MAXLOGNAME)
len = MAXLOGNAME;
gethostname(pbuf + len, sizeof(pbuf) - len);
pwd = pbuf;
}
e = _ftp_cmd(conn, "PASS %s", pwd);
}
return (e);
}
/*
* Log on to FTP server
*/
static conn_t *
_ftp_connect(struct url *url, struct url *purl, const char *flags)
{
conn_t *conn;
int e, direct, verbose;
#ifdef INET6
int af = AF_UNSPEC;
#else
int af = AF_INET;
#endif
direct = CHECK_FLAG('d');
verbose = CHECK_FLAG('v');
if (CHECK_FLAG('4'))
af = AF_INET;
else if (CHECK_FLAG('6'))
af = AF_INET6;
if (direct)
purl = NULL;
/* check for proxy */
if (purl) {
/* XXX proxy authentication! */
conn = _fetch_connect(purl->host, purl->port, af, verbose);
} else {
/* no proxy, go straight to target */
conn = _fetch_connect(url->host, url->port, af, verbose);
purl = NULL;
}
/* check connection */
if (conn == NULL)
/* _fetch_connect() has already set an error code */
return (NULL);
/* expect welcome message */
if ((e = _ftp_chkerr(conn)) != FTP_SERVICE_READY)
goto fouch;
/* authenticate */
if ((e = _ftp_authenticate(conn, url, purl)) != FTP_LOGGED_IN)
goto fouch;
/* TODO: Request extended features supported, if any (RFC 3659). */
/* done */
return (conn);
fouch:
if (e != -1)
_ftp_seterr(e);
_fetch_close(conn);
return (NULL);
}
/*
* Disconnect from server
*/
static void
_ftp_disconnect(conn_t *conn)
{
(void)_ftp_cmd(conn, "QUIT");
if (conn == cached_connection && conn->ref == 1)
cached_connection = NULL;
_fetch_close(conn);
}
/*
* Check if we're already connected
*/
static int
_ftp_isconnected(struct url *url)
{
return (cached_connection
&& (strcmp(url->host, cached_host.host) == 0)
&& (strcmp(url->user, cached_host.user) == 0)
&& (strcmp(url->pwd, cached_host.pwd) == 0)
&& (url->port == cached_host.port));
}
/*
* Check the cache, reconnect if no luck
*/
static conn_t *
_ftp_cached_connect(struct url *url, struct url *purl, const char *flags)
{
conn_t *conn;
int e;
/* set default port */
if (!url->port)
url->port = _fetch_default_port(url->scheme);
/* try to use previously cached connection */
if (_ftp_isconnected(url)) {
e = _ftp_cmd(cached_connection, "NOOP");
if (e == FTP_OK || e == FTP_SYNTAX_ERROR)
return (_fetch_ref(cached_connection));
}
/* connect to server */
if ((conn = _ftp_connect(url, purl, flags)) == NULL)
return (NULL);
if (cached_connection)
_ftp_disconnect(cached_connection);
cached_connection = _fetch_ref(conn);
memcpy(&cached_host, url, sizeof(*url));
return (conn);
}
/*
* Check the proxy settings
*/
static struct url *
_ftp_get_proxy(const char *flags)
{
struct url *purl;
char *p;
if (flags != NULL && strchr(flags, 'd') != NULL)
return (NULL);
if (((p = getenv("FTP_PROXY")) || (p = getenv("ftp_proxy")) ||
(p = getenv("HTTP_PROXY")) || (p = getenv("http_proxy"))) &&
*p && (purl = fetchParseURL(p)) != NULL) {
if (!*purl->scheme) {
if (getenv("FTP_PROXY") || getenv("ftp_proxy"))
strcpy(purl->scheme, SCHEME_FTP);
else
strcpy(purl->scheme, SCHEME_HTTP);
}
if (!purl->port)
purl->port = _fetch_default_proxy_port(purl->scheme);
if (strcasecmp(purl->scheme, SCHEME_FTP) == 0 ||
strcasecmp(purl->scheme, SCHEME_HTTP) == 0)
return (purl);
fetchFreeURL(purl);
}
return (NULL);
}
/*
* Process an FTP request
*/
#if __MOJOSETUP__
MojoInput *
#else
FILE *
#endif
_ftp_request(struct url *url, const char *op, struct url_stat *us,
struct url *purl, const char *flags)
{
conn_t *conn;
int oflag;
/* check if we should use HTTP instead */
#if !__MOJOSETUP__
if (purl && strcasecmp(purl->scheme, SCHEME_HTTP) == 0) {
if (strcmp(op, "STAT") == 0)
return (_http_request(url, "HEAD", us, purl, flags));
else if (strcmp(op, "RETR") == 0)
return (_http_request(url, "GET", us, purl, flags));
/*
* Our HTTP code doesn't support PUT requests yet, so try
* a direct connection.
*/
}
#endif
/* connect to server */
conn = _ftp_cached_connect(url, purl, flags);
if (purl)
fetchFreeURL(purl);
if (conn == NULL)
return (NULL);
/* change directory */
if (_ftp_cwd(conn, url->doc) == -1)
return (NULL);
/* stat file */
if (us && _ftp_stat(conn, url->doc, us) == -1
&& fetchLastErrCode != FETCH_PROTO
&& fetchLastErrCode != FETCH_UNAVAIL)
return (NULL);
/* just a stat */
#if __MOJOSETUP__
if (strcmp(op, "STAT") == 0)
return (MojoInput *)1; /* bogus return value */
#else
if (strcmp(op, "STAT") == 0)
return (FILE *)1; /* bogus return value */
#endif
if (strcmp(op, "STOR") == 0 || strcmp(op, "APPE") == 0)
oflag = O_WRONLY;
else
oflag = O_RDONLY;
/* initiate the transfer */
#if __MOJOSETUP__
{
MojoInput *retval = _ftp_transfer(conn, op, url->doc, oflag, url->offset, flags);
if ((retval != NULL) && (us != NULL))
{
struct ftpio *io = (struct ftpio *)retval->opaque;
io->length = us->size;
} // if
return retval;
}
#else
return (_ftp_transfer(conn, op, url->doc, oflag, url->offset, flags));
#endif
}
/*
* Get and stat file
*/
#if __MOJOSETUP__
MojoInput *
#else
FILE *
#endif
fetchXGetFTP(struct url *url, struct url_stat *us, const char *flags)
{
return (_ftp_request(url, "RETR", us, _ftp_get_proxy(flags), flags));
}
/*
* Get file
*/
#if !__MOJOSETUP__
FILE *
fetchGetFTP(struct url *url, const char *flags)
{
return (fetchXGetFTP(url, NULL, flags));
}
/*
* Put file
*/
FILE *
fetchPutFTP(struct url *url, const char *flags)
{
return (_ftp_request(url, CHECK_FLAG('a') ? "APPE" : "STOR", NULL,
_ftp_get_proxy(flags), flags));
}
/*
* Get file stats
*/
int
fetchStatFTP(struct url *url, struct url_stat *us, const char *flags)
{
FILE *f;
f = _ftp_request(url, "STAT", us, _ftp_get_proxy(flags), flags);
if (f == NULL)
return (-1);
return (0);
}
/*
* List a directory
*/
struct url_ent *
fetchListFTP(struct url *url __unused, const char *flags __unused)
{
warnx("fetchListFTP(): not implemented");
return (NULL);
}
#endif
#endif