#include <sys/types.h>
#include <sys/ioctl.h>
#include <dev/usb/usb.h>
#include <sys/errno.h>
#include <sys/uio.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>

/* Return codes for OWFirst()/OWNext() */
#define OW_BADWIRE		-3
#define OW_BADCRC		-2
#define OW_NOPRESENCE		-1
#define OW_NOMODULES		0
#define OW_FOUND		1

void	docmd(int fd, uint8_t *buffer, int rlen, int wlen);

int
main(int argc, char **argv) {
    int		endptfd, i;
    char	*endptname = "/dev/ugen0.2";
    uint8_t	buffer[9];
    
    if (argc < 2) {
	fprintf(stderr,
		"Bad usage:\n"
		"\t%s cmd\n", argv[0]);
	exit(1);
    }
    
    if ((endptfd = open(endptname, O_RDWR)) == -1) {
	fprintf(stderr, "Unable to open %s: %s\n", endptname, strerror(errno));
	exit(1);
    }

    i = 1;
    if (ioctl(endptfd, USB_SET_SHORT_XFER, &i) == -1) {
	fprintf(stderr, "Unable to set short xfer on end point 1: %s\n", strerror(errno));
	exit(1);
    }
    
    bzero(buffer, 9);

    if (!strcmp(argv[1], "reset")) {
	buffer[0] = 0x00;
	docmd(endptfd, buffer, 1, 1);
	printf("OWTouchReset() reported %d\n", buffer[0]);
    } else if (!strcmp(argv[1], "first")) {
	buffer[0] = 0x01;
	docmd(endptfd, buffer, 1, 9);
	printf("OWFirst() = %d\n", buffer[0]);
	if (buffer[0] == OW_FOUND)
	    printf("Found module %02x:%02x:%02x:%02x:%02x:%02x:%02x:%02x\n",
		   buffer[1], buffer[2], buffer[3], buffer[4],
		   buffer[5], buffer[6], buffer[7], buffer[8]);
    } else if (!strcmp(argv[1], "next")) {
	if (scanf(argv[2], "%hh:%hh:%hh:%hh:%hh:%hh:%hh:%hh", &buffer[1], &buffer[2], &buffer[3],
		  &buffer[4], &buffer[5], &buffer[6], &buffer[7], &buffer[8]) != 8) {
	    fprintf(stderr, "Unable to parse ROM ID\n");
	    exit(1);
	}
	buffer[0] = 0x02;
	fprintf(stderr, "About to OWNext()\n");
	docmd(endptfd, buffer, 9, 9);
	fprintf(stderr, "OWNext() = %d\n", buffer[0]);
	if (buffer[0] == OW_FOUND)
	    printf("Found module %02x:%02x:%02x:%02x:%02x:%02x:%02x:%02x\n",
		   buffer[1], buffer[2], buffer[3], buffer[4],
		   buffer[5], buffer[6], buffer[7], buffer[8]);
	
    } else if (!strcmp(argv[1], "scan")) {
	buffer[0] = 0x00;
	fprintf(stderr, "About to reset\n");
	docmd(endptfd, buffer, 1, 1);
	printf("OWReset() reported %d\n", buffer[0]);
	if (buffer[0] != 0)
	    exit(1);
	
	bzero(buffer, 9);
	buffer[0] = 0x01;
	fprintf(stderr, "About to OWFirst()\n");
	docmd(endptfd, buffer, 1, 9);
	printf("OWFirst() = %d\n", buffer[0]);

	do {
	    switch (buffer[0]) {
		case OW_BADWIRE:
		    printf("Bad wiring, either bus is held low, or a presence pulse was detected but no module found\n");
		    break;
		    
		case OW_NOPRESENCE:
		    printf("No presence pulse detected\n");
		    break;
		    
		case OW_BADCRC:
		    printf("Bad CRC\n");
		    break;
		    
		case OW_NOMODULES:
		case OW_FOUND:
		    break;
		    
		default:
		    printf("Unknown return code %d\n", buffer[0]);
		    break;
	    }
	    
	    if (buffer[0] != OW_FOUND) {
		printf("No more modules\n");
		break;
	    } else {
		printf("Found module %02x:%02x:%02x:%02x:%02x:%02x:%02x:%02x\n",
		       buffer[1], buffer[2], buffer[3], buffer[4],
		       buffer[5], buffer[6], buffer[7], buffer[8]);
	    }
	    
	    buffer[0] = 0x02;
	    fprintf(stderr, "About to OWNext()\n");
	    docmd(endptfd, buffer, 9, 9);
	} while(1);
    } else {
	printf("Unknown command: %s\n", argv[1]);
	exit(1);
    }
    
    
    exit(0);
}

void
docmd(int fd, uint8_t *buffer, int wlen, int rlen) {
    char blah[20];
    int	i;

    fprintf(stderr, "About to write\n");
    gets(blah);

    if ((i = write(fd, buffer, wlen)) == -1) {
	fprintf(stderr, "Unable to write end point: %s\n", strerror(errno));
	exit(1);
    }
    if (i != wlen)
	printf("Warning, only wrote %d of %d bytes\n", i, wlen);
    
    fprintf(stderr, "About to read\n");
    gets(blah);

    if ((i = read(fd, buffer, rlen)) == -1) {
	fprintf(stderr, "Unable to read from end point: %s\n", strerror(errno));
	exit(1);
    }
    if (i != rlen)
	printf("Warning, only read %d of %d bytes\n", i, wlen);
}