HardenedBSD/release/sysinstall/ftp_strat.c

/*
 * The new sysinstall program.
 *
 * This is probably the last attempt in the `sysinstall' line, the next
 * generation being slated to essentially a complete rewrite.
 *
 * $Id: ftp_strat.c,v 1.6.2.25 1995/06/07 09:53:14 jkh Exp $
 *
 * Copyright (c) 1995
 *	Jordan Hubbard.  All rights reserved.
 * Copyright (c) 1995
 * 	Gary J Palmer. 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,
 *    verbatim and that no modifications are made prior to this
 *    point in the file.
 * 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 Jordan Hubbard
 *	for the FreeBSD Project.
 * 4. The name of Jordan Hubbard or the FreeBSD project may not be used to
 *    endorse or promote products derived from this software without specific
 *    prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY JORDAN HUBBARD ``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 JORDAN HUBBARD OR HIS PETS 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, LIFE 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 "sysinstall.h"
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/param.h>
#include <sys/wait.h>
#include <netdb.h>
#include "ftp.h"

Boolean ftpInitted;
static FTP_t ftp;
extern int FtpPort;

int
mediaSetFtpUserPass(char *str)
{
    char *user, *pass;

    dialog_clear();
    if ((user = msgGetInput(getenv(FTP_USER), "Please enter the username you wish to login as")) != NULL)
	variable_set2(FTP_USER, user);
    if ((pass = msgGetInput(getenv(FTP_PASS), "Please enter the password for this user.\nWARNING: This password will echo on the screen!")) != NULL)
	variable_set2(FTP_PASS, pass);
    dialog_clear();
    return 0;
}

static Boolean
get_new_host(Device *dev)
{
    Boolean i;
    char *oldTitle = MenuMediaFTP.title;
    Device *netDev = dev->private;

    MenuMediaFTP.title = "Connection timed out - please select another site";
    i = mediaSetFTP(NULL);
    MenuMediaFTP.title = oldTitle;
    if (i) {
	char *cp = getenv(FTP_USER);

	if (cp && *cp)
	    (void)mediaSetFtpUserPass(NULL);
	netDev->flags |= OPT_LEAVE_NETWORK_UP;
	(*dev->shutdown)(dev);
	i = (*dev->init)(dev);
	netDev->flags &= ~OPT_LEAVE_NETWORK_UP;
    }
    return i;
}

static Boolean HasDistsDir;

Boolean
mediaInitFTP(Device *dev)
{
    int i, retries, max_retries = MAX_FTP_RETRIES;
    char *cp, *hostname, *dir;
    char *user, *login_name, password[80], url[BUFSIZ];
    Device *netDevice = (Device *)dev->private;

    if (ftpInitted)
	return TRUE;

    if (!(*netDevice->init)(netDevice))
	return FALSE;

    if ((ftp = FtpInit()) == NULL) {
	msgConfirm("FTP initialisation failed!");
	goto punt;
    }
    if (isDebug())
	msgDebug("Initialized FTP library.\n");

    cp = getenv("ftp");
    if (!cp)
	goto punt;
    if (isDebug())
	msgDebug("Attempting to open connection for: %s\n", cp);
    hostname = getenv(VAR_HOSTNAME);
    if (strncmp("ftp://", cp, 6) != NULL) {
	msgConfirm("Invalid URL: %s\n(A URL must start with `ftp://' here)", cp);
	goto punt;
    }
    strncpy(url, cp, BUFSIZ);
    if (isDebug())
	msgDebug("Using URL `%s'\n", url);
    hostname = url + 6;
    if ((cp = index(hostname, ':')) != NULL) {
	*(cp++) = '\0';
	FtpPort = strtol(cp, 0, 0);
    }
    else
	FtpPort = 21;
    if ((dir = index(cp ? cp : hostname, '/')) != NULL)
	*(dir++) = '\0';
    if (isDebug()) {
	msgDebug("hostname = `%s'\n", hostname);
	msgDebug("dir = `%s'\n", dir ? dir : "/");
	msgDebug("port # = `%d'\n", FtpPort);
    }
    msgNotify("Looking up host %s..", hostname);
    if ((gethostbyname(hostname) == NULL) && (inet_addr(hostname) == INADDR_NONE)) {
	msgConfirm("Cannot resolve hostname `%s'!  Are you sure that your\nname server, gateway and network interface are configured?", hostname);
	goto punt;
    }
    user = getenv(FTP_USER);
    if (!user || !*user) {
	snprintf(password, BUFSIZ, "installer@%s", hostname);
	login_name = "anonymous";
    }
    else {
	login_name = user;
	strcpy(password, getenv(FTP_PASS) ? getenv(FTP_PASS) : login_name);
    }
    retries = i = 0;
    if (OptFlags & (OPT_FTP_RESELECT + OPT_FTP_ABORT))
	max_retries = 0;
retry:
    if (i && ++retries > max_retries) {
	if ((OptFlags & OPT_FTP_ABORT) || !get_new_host(dev))
	    return FALSE;
	retries = 0;
    }
    msgNotify("Logging in as %s..", login_name);
    if ((i = FtpOpen(ftp, hostname, login_name, password)) != 0) {
	if (OptFlags & OPT_NO_CONFIRM)
	    msgNotify("Couldn't open FTP connection to %s\n", hostname);
	else
	    msgConfirm("Couldn't open FTP connection to %s\n", hostname);
	goto retry;
    }

    FtpPassive(ftp, (OptFlags & OPT_FTP_PASSIVE) ? 1 : 0);
    FtpBinary(ftp, 1);
    if (dir && *dir != '\0') {
	msgNotify("CD to distribution in ~ftp/%s", dir);
	if ((i = FtpChdir(ftp, dir)) == -2)
	    goto retry;
    }
    if (!FtpChdir(ftp, "dists")) {
	HasDistsDir = TRUE;
	FtpChdir(ftp, ".."); /* Hope this works! :-( */
    }
    else
	HasDistsDir = FALSE;
    if (isDebug())
	msgDebug("leaving mediaInitFTP!\n");
    ftpInitted = TRUE;
    return TRUE;

punt:
    FtpClose(ftp);
    ftp = NULL;
    (*netDevice->shutdown)(netDevice);
    return FALSE;
}

int
mediaGetFTP(Device *dev, char *file, Attribs *dist_attrs)
{
    int fd;
    int nretries = 0, max_retries = MAX_FTP_RETRIES;
    Boolean inDists = FALSE;

    if (OptFlags & (OPT_FTP_RESELECT + OPT_FTP_ABORT) || dev->flags & OPT_EXPLORATORY_GET)
	max_retries = 1;

    while ((fd = FtpGet(ftp, file)) < 0) {
	/* If a hard fail, try to "bounce" the ftp server to clear it */
	if (fd == -2 || ++nretries > max_retries) {
	    if ((OptFlags & OPT_FTP_ABORT) || (dev->flags & OPT_EXPLORATORY_GET))
		return -1;
	    else if (!get_new_host(dev))
		return -2;
	    nretries = 0;
	    continue;
	}
	if (HasDistsDir) {
	    if (!inDists) {
		FtpChdir(ftp, "dists");
		inDists = TRUE;
	    }
	    else {
		FtpChdir(ftp, "..");
		inDists = FALSE;
	    }
	}
    }
    if (inDists)
	FtpChdir(ftp, "..");
    return fd;
}

Boolean
mediaCloseFTP(Device *dev, int fd)
{
    FtpEOF(ftp);
    if (!close(fd))
	return (TRUE);
    return FALSE;
}

void
mediaShutdownFTP(Device *dev)
{
    Device *netdev = (Device *)dev->private;

    if (!ftpInitted)
	return;

    if (ftp != NULL) {
	FtpClose(ftp);
	ftp = NULL;
    }
    (*netdev->shutdown)(netdev);
    ftpInitted = FALSE;
}