#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);

 }