freebsd-nq/release/sysinstall/ftp_strat.c
Jordan K. Hubbard fc3d6da119 If the FTP install bombs out due to not being able to resolve the host
name given, go back to the TCP setup screen for their interface and give
them a chance to review the values.
1996-03-19 12:23:54 +00:00

311 lines
8.3 KiB
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.9 1996/03/02 07:31:53 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 = FALSE;
static FTP_t ftp;
extern int FtpPort;
static char *lastRequest;
static Boolean
get_new_host(Device *dev, Boolean probe)
{
Boolean i;
int j;
char *oldTitle = MenuMediaFTP.title;
char *cp = variable_get(VAR_FTP_ONERROR);
if (probe || (cp && strcmp(cp, "reselect")))
i = TRUE;
else {
i = FALSE;
dialog_clear();
msgConfirm("The %s file failed to load from the FTP site you\n"
"selected. Please select another one from the FTP menu.", lastRequest ? lastRequest : "requested");
MenuMediaFTP.title = "Request failed - please select another site";
j = mediaSetFTP(NULL);
MenuMediaFTP.title = oldTitle;
if (j == RET_SUCCESS) {
/* Bounce the link if necessary */
if (ftpInitted) {
msgDebug("Bouncing FTP connection before reselecting new host.\n");
dev->shutdown(dev);
i = dev->init(dev);
}
}
else {
msgDebug("User elected not to reselect, shutting down open connection.\n");
dev->shutdown(dev);
}
}
return i;
}
/* Should we throw in the towel? */
static Boolean
ftpShouldAbort(Device *dev, int retries)
{
char *cp, *cp2;
int maxretries, rval = FALSE;
cp = variable_get(VAR_FTP_ONERROR);
cp2 = variable_get(VAR_FTP_RETRIES);
maxretries = atoi(cp2);
if (retries > maxretries || (cp && !strcmp(cp, "abort"))) {
rval = TRUE;
if (isDebug())
msgDebug("Aborting FTP connection.\n");
dev->shutdown(dev);
}
return rval;
}
Boolean
mediaInitFTP(Device *dev)
{
int i, retries;
char *cp, *hostname, *dir;
char *user, *login_name, password[80], url[BUFSIZ];
Device *netDevice = (Device *)dev->private;
if (ftpInitted)
return TRUE;
if (isDebug())
msgDebug("Init routine for FTP called. Net device is %x\n", netDevice);
if (!netDevice->init(netDevice))
return FALSE;
if (!ftp && (ftp = FtpInit()) == NULL) {
dialog_clear();
msgConfirm("FTP initialisation failed!");
return FALSE;
}
cp = variable_get(VAR_FTP_PATH);
if (!cp) {
dialog_clear();
msgConfirm("%s is not set!", VAR_FTP_PATH);
return FALSE;
}
if (isDebug())
msgDebug("Attempting to open connection for URL: %s\n", cp);
hostname = variable_get(VAR_HOSTNAME);
if (strncmp("ftp://", cp, 6) != NULL) {
dialog_clear();
msgConfirm("Invalid URL: %s\n(A URL must start with `ftp://' here)", cp);
return FALSE;
}
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)) {
dialog_clear();
msgConfirm("Cannot resolve hostname `%s'! Are you sure that your\n"
"name server, gateway and network interface are correctly configured?", hostname);
netDevice->shutdown(netDevice);
tcpOpenDialog(netDevice);
return FALSE;
}
user = variable_get(VAR_FTP_USER);
if (!user || !*user) {
snprintf(password, BUFSIZ, "installer@%s", variable_get(VAR_HOSTNAME));
login_name = "anonymous";
}
else {
login_name = user;
strcpy(password, variable_get(VAR_FTP_PASS) ? variable_get(VAR_FTP_PASS) : login_name);
}
retries = 0;
retry:
msgNotify("Logging in as %s..", login_name);
if (FtpOpen(ftp, hostname, login_name, password) != 0) {
if (variable_get(VAR_NO_CONFIRM))
msgNotify("Couldn't open FTP connection to %s", hostname);
else {
dialog_clear();
msgConfirm("Couldn't open FTP connection to %s", hostname);
}
if (ftpShouldAbort(dev, ++retries) || !get_new_host(dev, FALSE))
return FALSE;
goto retry;
}
FtpPassive(ftp, !strcmp(variable_get(VAR_FTP_STATE), "passive"));
FtpBinary(ftp, 1);
if (dir && *dir != '\0') {
msgNotify("Attempt to chdir to distribution in %s..", dir);
if ((i = FtpChdir(ftp, dir)) != 0) {
if (i == -2 || ftpShouldAbort(dev, ++retries))
goto punt;
else if (get_new_host(dev, FALSE))
retries = 0;
goto retry;
}
}
/* Give it a shot - can't hurt to try and zoom in if we can, unless we get a hard error back that is! */
if (FtpChdir(ftp, getenv(VAR_RELNAME)) == -2)
goto punt;
if (isDebug())
msgDebug("mediaInitFTP was successful (logged in and chdir'd)\n");
ftpInitted = TRUE;
return TRUE;
punt:
if (ftp != NULL) {
FtpClose(ftp);
ftp = NULL;
}
return FALSE;
}
int
mediaGetFTP(Device *dev, char *file, Boolean probe)
{
int fd;
int nretries;
char *fp;
char buf[PATH_MAX];
fp = file;
nretries = 0;
lastRequest = file;
while ((fd = FtpGet(ftp, fp)) < 0) {
/* If a hard fail, try to "bounce" the ftp server to clear it */
if (fd == -2 && ++nretries < atoi(variable_get(VAR_FTP_RETRIES))) {
dev->shutdown(dev);
/* If we can't re-initialize, just forget it */
if (!dev->init(dev))
return -2;
}
else if (probe || ftpShouldAbort(dev, ++nretries))
return -1;
else {
/* Try some alternatives */
switch (nretries) {
case 1:
sprintf(buf, "dists/%s", file);
fp = buf;
break;
case 2:
sprintf(buf, "%s/%s", variable_get(VAR_RELNAME), file);
fp = buf;
break;
case 3:
sprintf(buf, "%s/dists/%s", variable_get(VAR_RELNAME), file);
fp = buf;
break;
case 4:
fp = file;
if (get_new_host(dev, probe)) {
nretries = 0;
continue;
}
else
break;
}
}
}
return fd;
}
Boolean
mediaCloseFTP(Device *dev, int fd)
{
if (isDebug())
msgDebug("FTP Close called\n");
if (ftp)
FtpEOF(ftp);
return FALSE;
}
void
mediaShutdownFTP(Device *dev)
{
/* Device *netdev = (Device *)dev->private; */
if (!ftpInitted)
return;
if (isDebug())
msgDebug("FTP shutdown called. FTP = %x\n", ftp);
if (ftp != NULL) {
FtpClose(ftp);
ftp = NULL;
}
/* (*netdev->shutdown)(netdev); */
ftpInitted = FALSE;
}