Subversion Repositories eus100lx

/* $Id: boot.c,v 1.1.1.1 2006-02-04 03:35:00 freza Exp $ */
 
#include <sys/mman.h>
 
#include <errno.h>
#include <fcntl.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
 
 
extern int mcsdecode(FILE *, u_int8_t **, size_t *);
 
static int 		verbose = 0;
static int 		is_ftp = 0;
 
#define VERBOSE(arg) 	if (verbose) warnx arg
#define TRACE(arg) 	if (verbose > 1) warnx arg
 
 
void
warnx(char *fmt, ...)
{
	va_list 		ap;
 
	if (is_ftp)
		fprintf(stdout, "150-");
	fprintf(stdout, "boot: ");
	va_start(ap, fmt);
	vfprintf(stdout, fmt, ap);
	va_end(ap);
	fprintf(stdout, "\r\n");
}
 
void
errx(int ret, char *fmt, ...)
{
	va_list 		ap;
 
	if (is_ftp)
		fprintf(stdout, "510-");
	fprintf(stderr, "boot: ");
	va_start(ap, fmt);
	vfprintf(stderr, fmt, ap);
	va_end(ap);
	fprintf(stderr, "\n");
 
	exit(ret);
}
 
void
err(int ret, char *fmt, ...)
{
	va_list 		ap;
 
	if (is_ftp)
		fprintf(stdout, "510-");
	fprintf(stderr, "boot: ");
	va_start(ap, fmt);
	vfprintf(stderr, fmt, ap);
	va_end(ap);
	fprintf(stderr, ": %s\n", strerror(errno));
 
	exit(ret);
}
 
/*
 * Main and stuff.
 */
#define ARGUMENTS 	"d:hp:v"
 
void
usage()
{
	printf("Usage: boot [options] [file.mcs]\n");
	printf("-d s   Use device 's' [/dev/virtex0]\n");
	printf("-h     Print this help\n");
	printf("-p b   Don't boot, set PROG to 'b' (1 or 0)\n");
	printf("-v     Verbose operation\n");
}
 
int
main(int argc, char *argv[])
{
	FILE 			*file;
	u_int8_t 		*data;
	char 			*path, *device, *str;
	size_t 			count;
	int 			fd, progval;
	int 			c, do_setprog, do_raw_boot;
 
	do_raw_boot = 0; 	/* via kernel by default */
	do_setprog = 0; 	/* boot by default */
	is_ftp = 0;
	progval = 0; 		/* gcc */
	device = "/dev/virtex0";
	path = NULL;
 
	if ((str = getenv("FTP_METHOD")) != NULL) {
		if (strcasecmp(str, "PUT") == 0) {
			is_ftp = 1;
		} else {
			errx(1, "FTP method %s, only PUT supported", str);
		}
	}
 
	while((c = getopt(argc, argv, ARGUMENTS)) != -1) {
		switch (c) {
		case 'd': 		/* Select device 	*/
			device = optarg;
			break;
 
		case 'h': 		/* Show help 		*/
			usage();
			return 0;
 
		case 'p':
			do_setprog = 1;
			switch (optarg[0]) {
			case '0':
				progval = 0;
				break;
			case '1':
				progval = 1;
				break;
			default:
				errx(1, "Argument to -p should be 0 or 1");
				/*UNREACHED*/
			}
			break;
 
		case 'v': 		/* Verbose operation 	*/
			verbose++;
			break;
 
		default:
			errx(1, "unknown argument -%c", optopt);
		}
	}
 
	argc -= optind;
	argv += optind;
 
	if (argc > (do_setprog ? 0 : 1))
		errx(1, "stray arguments");
 
	/* We're going to boot, so read the design. */
	if (do_setprog == 0) {
		if (argc == 1)
			path = argv[0];
 
		if (path != NULL) {
			VERBOSE(("open design file %s", path));
			file = fopen(path, "r");
			if (file == NULL)
				err(1, "could not open %s", path);
		} else {
			/* NOTE: vftpd is broken in that it will blocking
			 *     copy statfd (fd 3) to the user until the end,
			 *     then it will transfer the file, and in the
			 *     end copy messages again. So we have to close
			 *     statfd so that we can proceed reading stdin.
			 */
			if (is_ftp) {
				write(3, "150-boot: design file is stdin\r\n",
				    32);
				(void) close(3);
			} else {
				VERBOSE(("design file is stdin"));
			}
 
			file = stdin;
		}
 
		if (mcsdecode(file, &data, &count) != 0)
			errx(1, "could not decode %s", path);
		VERBOSE(("decoded %d bytes", count));
	} else {
		/* XXX: Control PROG from kernel */
		return (1);
	}
 
	/* Let's do the well behaved boot then. */
	VERBOSE(("open device"));
	fd = open(device, O_RDWR, 0);
	if (fd == -1)
		err(1, "could not open %s", device);
 
	VERBOSE(("write configuration stream"));
	if (write(fd, (const void *) data, count) != count)
		err(1, "failed to boot%s", \
		    errno == EIO ? ": CRC error" : "");
 
	VERBOSE(("acknowledge startup"));
	if (close(fd) == -1)
		err(1, "failed to boot%s", \
		    errno == EIO ? ": DONE timeout" : "");
 
	VERBOSE(("success"));
	return (0);
}