/*
 * 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;
{
	struct usb_device *dev;
	usb_dev_handle *usbh;
	unsigned n;

	if (argc < 2 || argc > 3) {
		fprintf(stderr, "usage: %s device-selector [eeprom-image]\n",
			argv[0]);
		exit(1);
	}
	if (argv[2])
		read_eeprom_from_file(argv[2]);
	else
		read_eeprom_from_stdin();
	dev = find_usbdev_by_desc_string(argv[1]);
	if (!dev) {
		fprintf(stderr, "error: specified USB device not found\n");
		exit(1);
	}
	usbh = usbwrap_open_dev(dev, 1);
	/* FIXME: reimplement FT232R magic sequence */
	for (n = 0; n < eeprom_size; n++)
		ftmini_write_eeprom_loc(usbh, n, eeprom[n]);
	usbwrap_close_dev(usbh);
	exit(0);
}