tftplib.c   [plain text]


/*
 * Copyright (c) 2003 Apple Inc. All rights reserved.
 *
 * @APPLE_LICENSE_HEADER_START@
 * 
 * This file contains Original Code and/or Modifications of Original Code
 * as defined in and that are subject to the Apple Public Source License
 * Version 2.0 (the 'License'). You may not use this file except in
 * compliance with the License. Please obtain a copy of the License at
 * http://www.opensource.apple.com/apsl/ and read it before using this
 * file.
 * 
 * The Original Code and all software distributed under the License are
 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
 * Please see the License for the specific language governing rights and
 * limitations under the License.
 * 
 * @APPLE_LICENSE_HEADER_END@
 */

/*
 * tftp.c
 * - fork/exec tftp to retrieve a remote file
 */
/*
 * Modification History
 *
 * December 14, 1999	Dieter Siegmund (dieter@apple)
 * - created
 */
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/param.h>
#include <mach/boolean.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <fcntl.h>
#include <sys/errno.h>
#include <string.h>

#define TFTP	"/usr/bin/tftp > /dev/null 2>&1"
 
/*
 * Function: tftp_get
 *
 * Purpose:
 *   Reads the file from the given server and places it in a temporary
 *   file in /tmp. 
 * Returns:
 *   Malloc'd pathname to temporary file if successful,
 *   NULL otherwise.
 */
char *
tftp_get(char * host, char * remote_file, off_t * len_p, int t)
{
    struct stat sb;
    char 	filename[128] = "/tmp/tftp_getXXXXXX";
    FILE * 	f;

    *len_p = 0;

    if (mktemp(filename) == NULL) {
#ifdef DEBUG
	perror("mktemp");
#endif /* DEBUG */
	return (NULL);
    }
    f = popen(TFTP, "w");
    if (f == NULL)
	return (NULL);

    fprintf(f, "binary\n");
    fprintf(f, "timeout %d\n", t);
    fprintf(f, "connect %s\n", host);
    fprintf(f, "get %s %s\n", remote_file, filename);
    pclose(f);
    if (stat(filename, &sb) < 0) {
#ifdef DEBUG
	fprintf(stderr, TFTP " %s:%s -> %s: download unsuccessful", 
		host, remote_file, filename);
#endif /* DEBUG */
	return (NULL);
    }
    *len_p = sb.st_size;
    return (strdup(filename));
}

#ifdef TESTING
int 
main(int argc, char * argv[])
{
    char * 	filename = NULL;
    off_t 	len = 0;

    if (argc < 3) {
	fprintf(stderr, "usage: %s <host> <remote>\n", argv[0]);
	exit(2);
    }
    filename = tftp_get(argv[1], argv[2], &len);
    if (filename) {
	printf("read %s:%s -> %s %qd\n", argv[1], argv[2], filename, len);
	free(filename);
	exit(0);
    }
    else {
	fprintf(stderr, "tftp %s:%s failed\n", argv[1], argv[2]);
	exit(1);
    }
    return (0);
}
#endif /* TESTING */