/*
 * This program connects to an FTDI chip via libusb and programs its EEPROM
 * with a raw hex image read from stdin or from a backup file.
 *
 * The present version has been converted to use our local libraries
 * (libftmini and libuwrap) instead of libftdi - thus the external dependency
 * is only libusb.
 */

#include <sys/types.h>
#include <ctype.h>
#include <string.h>
#include <strings.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <usb.h>
#include "../libuwrap/find_dev.h"
#include "../libuwrap/open_close.h"
#include "../libftmini/eeprom_func.h"

unsigned eeprom_size;
u_short eeprom[256];

read_eeprom_common(inf, filename_for_errs)
	FILE *inf;
	char *filename_for_errs;
{
	char linebuf[1024], *cp;
	int lineno, rc;
	unsigned ptr = 0, input_off;

	for (lineno = 1; fgets(linebuf, sizeof linebuf, inf); lineno++) {
		for (cp = linebuf; isspace(*cp); cp++)
			;
		if (*cp == '\0' || *cp == '#')
			continue;
		if (ptr >= 256) {
			fprintf(stderr,
				"%s line %d: maximum EEPROM size exceeded\n",
				filename_for_errs, lineno);
			exit(1);
		}
		rc = sscanf(linebuf, "%x: %hx %hx %hx %hx %hx %hx %hx %hx",
			    &input_off, eeprom + ptr, eeprom + ptr + 1,
			    eeprom + ptr + 2, eeprom + ptr + 3,
			    eeprom + ptr + 4, eeprom + ptr + 5,
			    eeprom + ptr + 6, eeprom + ptr + 7);
		if (rc != 9 || input_off != ptr * 2) {
			fprintf(stderr, "%s line %d: invalid input\n",
				filename_for_errs, lineno);
			exit(1);
		}
		ptr += 8;
	}
	if (ptr != 64 && ptr != 128 && ptr != 256) {
		fprintf(stderr, "%s: not an EEPROM image of recognized size\n",
			filename_for_errs);
		exit(1);
	}
	eeprom_size = ptr;
}

read_eeprom_from_file(filename)
	char *filename;
{
	FILE *inf;

	inf = fopen(filename, "r");
	if (!inf) {
		perror(filename);
		exit(1);
	}
	read_eeprom_common(inf, filename);
	fclose(inf);
}

read_eeprom_from_stdin()
{
	read_eeprom_common(stdin, "stdin");
}

main(argc, argv)
	char **argv;
{
	char **ap, *devsel;
	struct usb_device *dev;
	usb_dev_handle *usbh;
	int ft232r_mode = 0;
	unsigned n;

	if (argc < 2) {
usage:		fprintf(stderr,
			"usage: %s [-r] device-selector [eeprom-image]\n",
			argv[0]);
		exit(1);
	}
	ap = argv + 1;
	if (!strcmp(*ap, "-r")) {
		ft232r_mode = 1;
		ap++;
	}
	if (!*ap)
		goto usage;
	devsel = *ap++;
	if (*ap) {
		if (ap[1])
			goto usage;
		read_eeprom_from_file(*ap);
	} else
		read_eeprom_from_stdin();
	dev = find_usbdev_by_desc_string(devsel);
	if (!dev) {
		fprintf(stderr, "error: specified USB device not found\n");
		exit(1);
	}
	usbh = usbwrap_open_dev(dev, 1);
	if (ft232r_mode)
		ft232r_eeprom_magic(usbh);
	for (n = 0; n < eeprom_size; n++)
		ftmini_write_eeprom_loc(usbh, n, eeprom[n]);
	usbwrap_close_dev(usbh);
	exit(0);
}