#include <usb.h>
#include <stdio.h>
#include <string.h>

#define BSIZE 4000
#define CHUNK_SIZE 50

char rdata[BSIZE];

void perr (char *err, int ecode) {
	fprintf (stderr, err);
	if (ecode) exit (ecode);
}

void strange2string (char n, char *d) {
	unsigned int f = n >> 4;
	sprintf (d, "%i%i", f, n-f*16);
}

static struct usb_device *findDev(uint16_t vendor, uint16_t product) {
  struct usb_bus *bus;
  struct usb_device *dev;
  struct usb_bus *busses;

  usb_init();
  usb_find_busses();
  usb_find_devices();
  busses = usb_get_busses();

	for (bus = busses; bus; bus = bus->next)
		for (dev = bus->devices; dev; dev = dev->next)
			if ((dev->descriptor.idVendor == vendor) && (dev->descriptor.idProduct == product)) {
				return dev;
			}
	return NULL;
}

int sendData (usb_dev_handle *hdl, char *data) {
	usb_control_msg(hdl, 0xc0, 0x04, 0x500d, 0x0000, rdata, 2, 2000);
	usb_control_msg(hdl, 0x40, 0x04, 0x5001, 0x0000, data, 5, 1000);
	int r;
	int rv = 0;
	while (1) {
		r = usb_control_msg(hdl, 0xc0, 0x04, 0x5002, 0x0000, &rdata[rv], CHUNK_SIZE, 5000);
		if (r <= 0) break;
		rv += r;
	}
	return rv;
}

void readLoop (usb_dev_handle *hdl) {
	char s1[5], s2[5], s3[5];

	int rv = sendData (hdl, "\xa1\x01\x00\x03\xa5");

	int loop = 1;

	char tdata[BSIZE];

	while (1) {
		if (rv < 13) break;

		rv = sendData (hdl, "\xa1\x01\x00\x02\xa4");

		strange2string (rdata[5], s1);
		strange2string (rdata[4], s2);

		char tm[6];
		sprintf (tm, "%s:%s", s1, s2);

		int i, j;
		for (i=8; i<rv; i++) {
			tdata[i-8] = rdata[i];
		}
		j = rv-8;

		rv = sendData (hdl, "\xa1\x01\x00\x04\xa6");

		strange2string (rdata[7], s1);
		strange2string (rdata[8], s2);
		strange2string (rdata[9], s3);
		printf ("N%i training started at %s/%s/20%s %s\n", loop, s1, s2, s3, tm);

		for (i=0; i<j; i++) {
			printf ("N%i minute %i, rate %i\n", loop, i+1, (unsigned char)tdata[i]);
		}

		rv = sendData (hdl, "\xa1\x01\x00\x06\xa8");

		loop ++;
	}
}

usb_dev_handle *open_device (struct usb_device *dev) {
	usb_dev_handle *hdl;

	hdl = usb_open (dev);
	int open_status = usb_set_configuration (hdl, 1);
	open_status = usb_claim_interface (hdl, 0);
	open_status = usb_set_altinterface (hdl, 0);

	usb_control_msg(hdl, 0x40, 0x0c, 0x5003, 0x00f0, "\x10", 1, 1000); 
	usb_control_msg(hdl, 0x40, 0x0c, 0x5003, 0xffff, "\xd0", 1, 1000); 

	return hdl;
}

void close_device (usb_dev_handle *hdl) {
	if (usb_close (hdl)) {
		perr ("Cannot close USB device!\n", 1);
	}
}

void readData (usb_dev_handle *hdl) {
	sendData (hdl, "\x91\x01\x00\x01\x93");
	sendData (hdl, "\xa1\x01\x00\x01\xa3");
	readLoop (hdl);
}

void deleteData (usb_dev_handle *hdl) {
	usb_control_msg(hdl, 0xc0, 0x04, 0x500d, 0x0000, rdata, 2, 2000); 
	sendData (hdl, "\xa1\x01\x00\x00\xa2");
}

int main (int argc, char** argv) {
	struct usb_device *dev;

	if (argc < 2) {
		perr ("Should be called with one argument: 'read' or 'delete'\n", 2);
	}

	while (1) {
		if ((dev = findDev(0x0e6a, 0x0101)) == NULL) {
			perr ("Waiting for device...\n", 0);
		} else {
			break;
		}
		usleep (300000);
	}

	usb_dev_handle *hdl = open_device (dev);

	if (!strcmp(argv[1], "read")) {
		readData (hdl);
	} else if (!strcmp(argv[1], "delete")) {
		deleteData (hdl);
	} else {
		perr ("Invalid command line argument.\n", 3);
	}

	close_device (hdl);

	return 0;
}