/*
 * Replacement for ftpXfer.
 *
 * Author: Curt McDowell (csm@broadcom.com)
 *
 *
 * $Id: ftpXfer2.c,v 1.2 1999/10/25 18:22:05 borkhuis Exp $
 *
 */


#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <selectLib.h>
#include <errnoLib.h>
#include <ftpLib.h>

#include "config.h"
#include "sysLib.h"
#include "netinet/in.h"
#include "sockLib.h"

#define FTP_DATA_CONN_TIMEOUT		20	/* sec */

/*
 * ftpXfer2
 *
 * This routine replaces VxWorks ftpXfer and doesn't have a bug that can
 * cause very small file transfers to fail.  In particular, it calls
 * select() on both the data and control sockets, and if they become
 * ready at the SAME TIME (as opposed to just the data socket), vxWorks
 * falsely assumes an error.
 */

extern BOOL ftpVerbose;

STATUS ftpXfer2(char *host, char *user, char *passwd, char *acct,
		char *cmd, char *dirname, char *filename,
		int *pCtrlSock, int *pDataSock)
{
    int			ctrlSock = ERROR, dataSock = ERROR;
    struct fd_set 	rfd;
    int			result;
    struct timeval	tmo;
    char		*errmsg = NULL;

    if (ftpVerbose)
	printf("ftpXfer2: hookup host=%s file=%s\n", host, filename);

    if ((ctrlSock = ftpHookup(host)) == ERROR) {
	errmsg = "server unreachable";
	goto error;
    }

    *pCtrlSock = ctrlSock;

    if (ftpVerbose)
	printf("ftpXfer2: login user=%s\n", user);

    if (ftpLogin(ctrlSock, user, passwd, acct) != OK) {
	errmsg = "authentication failed";
	goto error;
    }

    if (ftpVerbose)
	printf("ftpXfer2: set binary\n");

    if (ftpCommand(ctrlSock, "TYPE I", 0, 0, 0, 0, 0, 0) != FTP_COMPLETE) {
	errmsg = "set binary mode failed";
	goto error;
    }

    if (dirname[0]) {
	if (ftpVerbose)
	    printf("ftpXfer2: cd %s\n", dirname);

	if (ftpCommand(ctrlSock, "CWD %s",
		       (int) dirname, 0, 0, 0, 0, 0) != FTP_COMPLETE) {
	    errmsg = "change directory failed";
	    goto error;
	}
    }

    /*
     * Retry loop for transient errors, such as the remote host being
     * unable to assign the requested port number.
     */

 retry_transient_error:

    /*
     * If this is a transfer command requiring a data connection,
     * first establish socket for server to connect back to.
     */

    if (pDataSock) {
	int			len;
	short			port;
	struct sockaddr_in 	ownAddr;
	struct sockaddr_in 	dataAddr;

	if (ftpVerbose)
	    printf("ftpXfer2: set binary\n");

	/* Find out our own address */

	len = sizeof (ownAddr);
	if (getsockname(ctrlSock, (struct sockaddr *) &ownAddr, &len) < 0) {
	    errmsg = "FTP could not get own addr";
	    goto error;
	}

	if ((dataSock = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
	    errmsg = "FTP could not create data socket";
	    goto error;
	}

	dataAddr = ownAddr;
	dataAddr.sin_port = htons(0);

	if (bind(dataSock,
		 (struct sockaddr *) &dataAddr,
		 sizeof (dataAddr)) != OK) {
	    close(dataSock);
	    errmsg = "FTP could not bind data socket";
	    goto error;
	}

	/* Read back to find out what port was bound */

	len = sizeof (dataAddr);
	if (getsockname(dataSock, (struct sockaddr *) &dataAddr, &len) < 0) {
	    errmsg = "FTP could not get data addr";
	    goto error;
	}

	port = ntohs(dataAddr.sin_port);

	if (listen(dataSock, 1) < 0) {
	    close(dataSock);
	    errmsg = "FTP could not listen on data socket";
	    goto error;
	}

	/* Use PORT command to inform server of data socket address */

	if (ftpCommand(ctrlSock,
		       "PORT %d,%d,%d,%d,%d,%d",
		       (int) ((UINT8 *) &dataAddr.sin_addr)[0],
		       (int) ((UINT8 *) &dataAddr.sin_addr)[1],
		       (int) ((UINT8 *) &dataAddr.sin_addr)[2],
		       (int) ((UINT8 *) &dataAddr.sin_addr)[3],
		       (int) (port >> 8),
		       (int) (port & 0xff)) != FTP_COMPLETE) {
	    close(dataSock);
	    errmsg = "FTP could not send PORT command";
	    goto error;
	}
    }

    /*
     * Send the FTP command.
     */

    if (ftpVerbose)
	printf("ftpXfer2: command %s\n", cmd);

    result = ftpCommand(ctrlSock, cmd, (int) filename, 0, 0, 0, 0, 0);

    if (ftpVerbose)
	printf("ftpXfer2: result %d\n", result);

    switch (result) {
    case FTP_TRANSIENT:
	if (pDataSock)
	    close(dataSock);
	goto retry_transient_error;
    case FTP_COMPLETE:
    case FTP_CONTINUE:
	if (pDataSock) {
	    close(dataSock);
	    errmsg = "server returned COMPLETE or CONTINUE instead of PRELIM";
	    goto error;
	}
	return OK;	/* Non-transfer command succeeded */
    case FTP_PRELIM:
	if (! pDataSock) {
	    errmsg = "server returned PRELIM for non-transfer command";
	    goto error;
	}
	break;		/* Continue below to start transfer */
    default:
	if (pDataSock)
	    close(dataSock);
	errmsg = "command failed";
	goto error;
    }

    /*
     * Wait for server to connect back on data socket.
     * Use select to provide a timeout.
     */

    FD_ZERO(&rfd);
    FD_SET(dataSock, &rfd);

    tmo.tv_sec = FTP_DATA_CONN_TIMEOUT;
    tmo.tv_usec = 0;

    if (ftpVerbose)
	printf("ftpXfer2: wait for data\n");

    if (select(FD_SETSIZE, &rfd, NULL, NULL, &tmo) < 0) {
	if (pDataSock)
	    close(dataSock);
	errmsg = "data conn failed or timed out";
	goto error;
    }

    if (ftpVerbose)
	printf("ftpXfer2: get data conn\n");

    if ((dataSock = ftpDataConnGet(dataSock)) == ERROR) {
	errmsg = "failed to accept server connection";
	goto error;
    }

    *pDataSock = dataSock;

    if (ftpVerbose)
	printf("ftpXfer2: return OK\n");

    return OK;

 error:

    if (errmsg && ftpVerbose)
	printErr("FTP ERROR: %s (errno=%d)\n", errmsg, errnoGet());

    if (ctrlSock != ERROR) {
	(void) ftpCommand(ctrlSock, "QUIT", 0, 0, 0, 0, 0, 0);
	close(ctrlSock);
    }

    if (ftpVerbose)
	printf("ftpXfer2: return ERROR\n");

    return ERROR;
}