freebsd-skq/lib/libfetch/ftp.c

/*-
 * Copyright (c) 1998 Dag-Erling Co<EFBFBD>dan Sm<EFBFBD>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.
 *
 *	$Id: ftp.c,v 1.1.1.1 1998/07/09 16:52:42 des Exp $
 */

/*
 * Portions of this code were taken from or based on ftpio.c:
 *
 * ----------------------------------------------------------------------------
 * "THE BEER-WARE LICENSE" (Revision 42):
 * <phk@login.dknet.dk> 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<EFBFBD>dan Sm<EFBFBD>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/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <sys/errno.h>

#include <ctype.h>
#include <errno.h>
#include <netdb.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "fetch.h"
#include "ftperr.c"

#define FTP_DEFAULT_TO_ANONYMOUS
#define FTP_ANONYMOUS_USER	"ftp"
#define FTP_ANONYMOUS_PASSWORD	"ftp"
#define FTP_DEFAULT_PORT 21

#define FTP_OPEN_DATA_CONNECTION	150
#define FTP_OK				200
#define FTP_PASSIVE_MODE		227
#define FTP_LOGGED_IN			230
#define FTP_FILE_ACTION_OK		250
#define FTP_NEED_PASSWORD		331
#define FTP_NEED_ACCOUNT		332

#define ENDL "\r\n"

static url_t cached_host;
static FILE *cached_socket;

static char *_ftp_last_reply;

/*
 * Map error code to string
 */
static const char *
_ftp_errstring(int e)
{
    struct ftperr *p = _ftp_errlist;

    while ((p->num != -1) && (p->num != e))
	p++;
    
    return p->string;
}

/*
 * Set error code
 */
static void
_ftp_seterr(int e)
{
    fetchLastErrCode = e;
    fetchLastErrText = _ftp_errstring(e);
}

/*
 * Set error code according to errno
 */
static void
_ftp_syserr(void)
{
    fetchLastErrCode = errno;
    fetchLastErrText = strerror(errno);
}

/*
 * Get server response, check that first digit is a '2'
 */
static int
_ftp_chkerr(FILE *s, int *e)
{
    char *line;
    size_t len;

    if (e)
	*e = 0;
    
    do {
	if (((line = fgetln(s, &len)) == NULL) || (len < 4)) {
	    _ftp_syserr();
	    return -1;
	}
    } while (line[3] == '-');

    _ftp_last_reply = line;
    
#ifndef NDEBUG
    fprintf(stderr, "\033[1m<<< ");
    fprintf(stderr, "%*.*s", (int)len, (int)len, line);
    fprintf(stderr, "\033[m");
#endif
    
    if (!isdigit(line[1]) || !isdigit(line[1])
	|| !isdigit(line[2]) || (line[3] != ' ')) {
	_ftp_seterr(-1);
	return -1;
    }

    _ftp_seterr((line[0] - '0') * 100 + (line[1] - '0') * 10 + (line[2] - '0'));

    if (e)
	*e = fetchLastErrCode;

    return (line[0] == '2') - 1;
}

/*
 * Send a command and check reply
 */
static int
_ftp_cmd(FILE *f, char *fmt, ...)
{
    va_list ap;
    int e;

    va_start(ap, fmt);
    vfprintf(f, fmt, ap);
#ifndef NDEBUG
    fprintf(stderr, "\033[1m>>> ");
    vfprintf(stderr, fmt, ap);
    fprintf(stderr, "\033[m");
#endif
    va_end(ap);
    
    _ftp_chkerr(f, &e);
    return e;
}

/*
 * Retrieve file
 */
static FILE *
_ftp_retrieve(FILE *cf, char *file, int pasv)
{
    struct sockaddr_in sin;
    int sd = -1, l;
    char *s;
    FILE *df;
    
    /* change directory */
    if (((s = strrchr(file, '/')) != NULL) && (s != file)) {
	*s = 0;
	if (_ftp_cmd(cf, "CWD %s" ENDL, file) != FTP_FILE_ACTION_OK) {
	    *s = '/';
	    return NULL;
	}
	*s++ = '/';
    } else {
	if (_ftp_cmd(cf, "CWD /" ENDL) != FTP_FILE_ACTION_OK)
	    return NULL;
    }

    /* s now points to file name */

    /* open data socket */
    if ((sd = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0) {
	_ftp_syserr();
	return NULL;
    }
    
    if (pasv) {
	u_char addr[6];
	char *ln, *p;
	int i;
	
	/* send PASV command */
	if (_ftp_cmd(cf, "PASV" ENDL) != FTP_PASSIVE_MODE)
	    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 = _ftp_last_reply;
	for (p = ln + 3; !isdigit(*p); p++)
	    /* nothing */ ;
	for (p--, i = 0; i < 6; i++) {
	    p++; /* skip the comma */
	    addr[i] = strtol(p, &p, 10);
	}

	/* construct sockaddr for data socket */
	l = sizeof(sin);
	if (getpeername(fileno(cf), (struct sockaddr *)&sin, &l) < 0)
	    goto sysouch;
	bcopy(addr, (char *)&sin.sin_addr, 4);
	bcopy(addr + 4, (char *)&sin.sin_port, 2);	

	/* connect to data port */
	if (connect(sd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
	    goto sysouch;
	
	/* make the server initiate the transfer */
	if (_ftp_cmd(cf, "RETR %s" ENDL, s) != FTP_OPEN_DATA_CONNECTION)
	    goto ouch;
	
    } else {
	u_int32_t a;
	u_short p;
	int d;
	
	/* find our own address, bind, and listen */
	l = sizeof(sin);
	if (getsockname(fileno(cf), (struct sockaddr *)&sin, &l) < 0)
	    goto sysouch;
	sin.sin_port = 0;
	if (bind(sd, (struct sockaddr *)&sin, l) < 0)
	    goto sysouch;
	if (listen(sd, 1) < 0)
	    goto sysouch;

	/* find what port we're on and tell the server */
	if (getsockname(sd, (struct sockaddr *)&sin, &l) < 0)
	    goto sysouch;
	a = ntohl(sin.sin_addr.s_addr);
	p = ntohs(sin.sin_port);
	if (_ftp_cmd(cf, "PORT %d,%d,%d,%d,%d,%d" ENDL,
		     (a >> 24) & 0xff, (a >> 16) & 0xff, (a >> 8) & 0xff, a & 0xff,
		     (p >> 8) & 0xff, p & 0xff) != FTP_OK)
	    goto ouch;

	/* make the server initiate the transfer */
	if (_ftp_cmd(cf, "RETR %s" ENDL, s) != FTP_OPEN_DATA_CONNECTION)
	    goto ouch;
	
	/* accept the incoming connection and go to town */
	if ((d = accept(sd, NULL, NULL)) < 0)
	    goto sysouch;
	close(sd);
	sd = d;
    }

    if ((df = fdopen(sd, "r")) == NULL)
	goto sysouch;
    return df;

sysouch:
    _ftp_syserr();
ouch:
    close(sd);
    return NULL;
}

/*
 * Store file
 */
static FILE *
_ftp_store(FILE *cf, char *file, int pasv)
{
    fprintf(stderr, "_ftp_store: not implemented yet.\n");
    
    cf = cf;
    file = file;
    pasv = pasv;
    return NULL;
}

/*
 * Log on to FTP server
 */
static FILE *
_ftp_connect(char *host, int port, char *user, char *pwd)
{
    int sd, e;
    FILE *f;

    /* establish control connection */
    if ((sd = fetchConnect(host, port)) < 0) {
	_ftp_syserr();
	return NULL;
    }
    if ((f = fdopen(sd, "r+")) == NULL) {
	_ftp_syserr();
	goto ouch;
    }

    /* expect welcome message */
    if (_ftp_chkerr(f, NULL) < 0)
	goto fouch;
    
    /* send user name and password */
    e = _ftp_cmd(f, "USER %s" ENDL, user);
    if (e == FTP_NEED_PASSWORD)	/* server requested a password */
	e = _ftp_cmd(f, "PASS %s" ENDL, pwd);
    if (e == FTP_NEED_ACCOUNT) /* server requested an account */
	/* help! */ ;
    if (e != FTP_LOGGED_IN) /* won't let us near the WaReZ */
	goto fouch;

    /* might as well select mode and type at once */
#ifdef FTP_FORCE_STREAM_MODE
    if (_ftp_cmd(f, "MODE S" ENDL) != FTP_OK)
	goto ouch;
#endif
    if (_ftp_cmd(f, "TYPE I" ENDL) != FTP_OK)
	goto ouch;

    /* done */
    return f;
    
ouch:
    close(sd);
    return NULL;
fouch:
    fclose(f);
    return NULL;
}

/*
 * Disconnect from server
 */
static void
_ftp_disconnect(FILE *f)
{
    _ftp_cmd(f, "QUIT" ENDL);
    fclose(f);
}

/*
 * Check if we're already connected
 */
static int
_ftp_isconnected(url_t *url)
{
    return (cached_socket
	    && (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));
}

FILE *
fetchGetFTP(url_t *url, char *flags)
{
    FILE *cf = NULL;
    int e;

#ifdef DEFAULT_TO_ANONYMOUS
    if (!url->user[0]) {
	strcpy(url->user, FTP_ANONYMOUS_USER);
	strcpy(url->pwd, FTP_ANONYMOUS_PASSWORD);
    }
#endif

    /* set default port */
    if (!url->port)
	url->port = FTP_DEFAULT_PORT;
    
    /* try to use previously cached connection */
    if (_ftp_isconnected(url)) {
	fprintf(cached_socket, "PWD" ENDL);
	_ftp_chkerr(cached_socket, &e);
	if (e > 0)
	    cf = cached_socket;
    }

    /* connect to server */
    if (!cf) {
	cf = _ftp_connect(url->host, url->port, url->user, url->pwd);
	if (!cf)
	    return NULL;
	if (cached_socket)
	    _ftp_disconnect(cached_socket);
	cached_socket = cf;
	memcpy(&cached_host, url, sizeof(url_t));
    }

    /* initiate the transfer */
    return _ftp_retrieve(cf, url->doc, (flags && strchr(flags, 'p')));
}

/*
 * Upload a file.
 * Hmmm, that's almost an exact duplicate of the above...
 */
FILE *
fetchPutFTP(url_t *url, char *flags)
{
    FILE *cf = NULL;
    int e;
   
#ifdef DEFAULT_TO_ANONYMOUS
    if (!url->user[0]) {
	strcpy(url->user, FTP_ANONYMOUS_USER);
	strcpy(url->pwd, FTP_ANONYMOUS_PASSWORD);
    }
#endif

    /* set default port */
    if (!url->port)
	url->port = htons(FTP_DEFAULT_PORT);
    
    /* try to use previously cached connection */
    if (_ftp_isconnected(url)) {
	fprintf(cached_socket, "PWD" ENDL);
	_ftp_chkerr(cached_socket, &e);
	if (e > 0)
	    cf = cached_socket;
    }

    /* connect to server */
    if (!cf) {
	cf = _ftp_connect(url->host, url->port, url->user, url->pwd);
	if (!cf)
	    return NULL;
	if (cached_socket)
	    _ftp_disconnect(cached_socket);
	cached_socket = cf;
	memcpy(&cached_host, url, sizeof(url_t));
    }


    /* initiate the transfer */
    return _ftp_store(cf, url->doc, (flags && strchr(flags, 'p')));
}