/*
 * 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"

extern unsigned eeprom_size;
extern u_short eeprom[256];

char *device_selector, *input_filename;
int detach_num, ft232r_mode;

process_cmdline(argc, argv)
	char **argv;
{
	int c;
	extern int optind;
	extern char *optarg;

	while ((c = getopt(argc, argv, "d:r")) != EOF) {
		switch (c) {
		case 'd':
			detach_num = atoi(optarg);
			switch (detach_num) {
			case 0:
			case 1:
			case 2:
			case 4:
				break;
			default:
				fprintf(stderr, "error: invalid -d argument\n");
				exit(1);
			}
			continue;
		case 'r':
			ft232r_mode = 1;
			detach_num = 1;
			continue;
		default:
			/* error msg already printed */
			exit(1);
		}
	}
	if (argc < optind + 1 || argc > optind + 2) {
		fprintf(stderr,
			"usage: %s [options] device-selector [eeprom-image]\n",
			argv[0]);
		exit(1);
	}
	device_selector = argv[optind];
	input_filename = argv[optind + 1];
}

main(argc, argv)
	char **argv;
{
	struct usb_device *dev;
	usb_dev_handle *usbh;
	unsigned n;

	process_cmdline(argc, argv);
	if (input_filename)
		read_eeprom_from_file(input_filename);
	else
		read_eeprom_from_stdin();
	dev = find_usbdev_by_desc_string(device_selector);
	if (!dev) {
		fprintf(stderr, "error: specified USB device not found\n");
		exit(1);
	}
	usbh = usb_open(dev);
	if (!usbh) {
		fprintf(stderr, "error: usb_open() failed\n");
		exit(1);
	}
	if (detach_num)
		usbwrap_claim_all_ifs(usbh, detach_num);
	if (ft232r_mode)
		ft232r_eeprom_magic(usbh);
	for (n = 0; n < eeprom_size; n++)
		ftmini_write_eeprom_loc(usbh, n, eeprom[n]);
	usb_close(usbh);
	exit(0);
}