[linux.git] / tools / usb / testusb.c

/* $(CROSS_COMPILE)cc -Wall -Wextra -g -lpthread -o testusb testusb.c */

/*
 * Copyright (c) 2002 by David Brownell
 * Copyright (c) 2010 by Samsung Electronics
 * Author: Michal Nazarewicz <[email protected]>
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation; either version 2 of the License, or (at your
 * option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

/*
 * This program issues ioctls to perform the tests implemented by the
 * kernel driver.  It can generate a variety of transfer patterns; you
 * should make sure to test both regular streaming and mixes of
 * transfer sizes (including short transfers).
 *
 * For more information on how this can be used and on USB testing
 * refer to <URL:http://www.linux-usb.org/usbtest/>.
 */

#include <stdio.h>
#include <string.h>
#include <ftw.h>
#include <stdlib.h>
#include <pthread.h>
#include <unistd.h>
#include <errno.h>
#include <limits.h>

#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include <sys/ioctl.h>
#include <linux/usbdevice_fs.h>

/*-------------------------------------------------------------------------*/

#define	TEST_CASES	30

// FIXME make these public somewhere; usbdevfs.h?

struct usbtest_param {
	// inputs
	unsigned		test_num;	/* 0..(TEST_CASES-1) */
	unsigned		iterations;
	unsigned		length;
	unsigned		vary;
	unsigned		sglen;

	// outputs
	struct timeval		duration;
};
#define USBTEST_REQUEST	_IOWR('U', 100, struct usbtest_param)

/*-------------------------------------------------------------------------*/

/* #include <linux/usb_ch9.h> */

#define USB_DT_DEVICE			0x01
#define USB_DT_INTERFACE		0x04

#define USB_CLASS_PER_INTERFACE		0	/* for DeviceClass */
#define USB_CLASS_VENDOR_SPEC		0xff


struct usb_device_descriptor {
	__u8  bLength;
	__u8  bDescriptorType;
	__u16 bcdUSB;
	__u8  bDeviceClass;
	__u8  bDeviceSubClass;
	__u8  bDeviceProtocol;
	__u8  bMaxPacketSize0;
	__u16 idVendor;
	__u16 idProduct;
	__u16 bcdDevice;
	__u8  iManufacturer;
	__u8  iProduct;
	__u8  iSerialNumber;
	__u8  bNumConfigurations;
} __attribute__ ((packed));

struct usb_interface_descriptor {
	__u8  bLength;
	__u8  bDescriptorType;

	__u8  bInterfaceNumber;
	__u8  bAlternateSetting;
	__u8  bNumEndpoints;
	__u8  bInterfaceClass;
	__u8  bInterfaceSubClass;
	__u8  bInterfaceProtocol;
	__u8  iInterface;
} __attribute__ ((packed));

enum usb_device_speed {
	USB_SPEED_UNKNOWN = 0,			/* enumerating */
	USB_SPEED_LOW, USB_SPEED_FULL,		/* usb 1.1 */
	USB_SPEED_HIGH				/* usb 2.0 */
};

/*-------------------------------------------------------------------------*/

static char *speed (enum usb_device_speed s)
{
	switch (s) {
	case USB_SPEED_UNKNOWN:	return "unknown";
	case USB_SPEED_LOW:	return "low";
	case USB_SPEED_FULL:	return "full";
	case USB_SPEED_HIGH:	return "high";
	default:		return "??";
	}
}

struct testdev {
	struct testdev		*next;
	char			*name;
	pthread_t		thread;
	enum usb_device_speed	speed;
	unsigned		ifnum : 8;
	unsigned		forever : 1;
	int			test;

	struct usbtest_param	param;
};
static struct testdev		*testdevs;

static int testdev_ffs_ifnum(FILE *fd)
{
	union {
		char buf[255];
		struct usb_interface_descriptor intf;
	} u;

	for (;;) {
		if (fread(u.buf, 1, 1, fd) != 1)
			return -1;
		if (fread(u.buf + 1, (unsigned char)u.buf[0] - 1, 1, fd) != 1)
			return -1;

		if (u.intf.bLength == sizeof u.intf
		 && u.intf.bDescriptorType == USB_DT_INTERFACE
		 && u.intf.bNumEndpoints == 2
		 && u.intf.bInterfaceClass == USB_CLASS_VENDOR_SPEC
		 && u.intf.bInterfaceSubClass == 0
		 && u.intf.bInterfaceProtocol == 0)
			return (unsigned char)u.intf.bInterfaceNumber;
	}
}

static int testdev_ifnum(FILE *fd)
{
	struct usb_device_descriptor dev;

	if (fread(&dev, sizeof dev, 1, fd) != 1)
		return -1;

	if (dev.bLength != sizeof dev || dev.bDescriptorType != USB_DT_DEVICE)
		return -1;

	/* FX2 with (tweaked) bulksrc firmware */
	if (dev.idVendor == 0x0547 && dev.idProduct == 0x1002)
		return 0;

	/*----------------------------------------------------*/

	/* devices that start up using the EZ-USB default device and
	 * which we can use after loading simple firmware.  hotplug
	 * can fxload it, and then run this test driver.
	 *
	 * we return false positives in two cases:
	 * - the device has a "real" driver (maybe usb-serial) that
	 *   renumerates.  the device should vanish quickly.
	 * - the device doesn't have the test firmware installed.
	 */

	/* generic EZ-USB FX controller */
	if (dev.idVendor == 0x0547 && dev.idProduct == 0x2235)
		return 0;

	/* generic EZ-USB FX2 controller */
	if (dev.idVendor == 0x04b4 && dev.idProduct == 0x8613)
		return 0;

	/* CY3671 development board with EZ-USB FX */
	if (dev.idVendor == 0x0547 && dev.idProduct == 0x0080)
		return 0;

	/* Keyspan 19Qi uses an21xx (original EZ-USB) */
	if (dev.idVendor == 0x06cd && dev.idProduct == 0x010b)
		return 0;

	/*----------------------------------------------------*/

	/* "gadget zero", Linux-USB test software */
	if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4a0)
		return 0;

	/* user mode subset of that */
	if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4a4)
		return testdev_ffs_ifnum(fd);
		/* return 0; */

	/* iso version of usermode code */
	if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4a3)
		return 0;

	/* some GPL'd test firmware uses these IDs */

	if (dev.idVendor == 0xfff0 && dev.idProduct == 0xfff0)
		return 0;

	/*----------------------------------------------------*/

	/* iBOT2 high speed webcam */
	if (dev.idVendor == 0x0b62 && dev.idProduct == 0x0059)
		return 0;

	/*----------------------------------------------------*/

	/* the FunctionFS gadget can have the source/sink interface
	 * anywhere.  We look for an interface descriptor that match
	 * what we expect.  We ignore configuratiens thou. */

	if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4ac
	 && (dev.bDeviceClass == USB_CLASS_PER_INTERFACE
	  || dev.bDeviceClass == USB_CLASS_VENDOR_SPEC))
		return testdev_ffs_ifnum(fd);

	return -1;
}

static int find_testdev(const char *name, const struct stat *sb, int flag)
{
	FILE				*fd;
	int				ifnum;
	struct testdev			*entry;

	(void)sb; /* unused */

	if (flag != FTW_F)
		return 0;
	/* ignore /proc/bus/usb/{devices,drivers} */
	if (strrchr(name, '/')[1] == 'd')
		return 0;

	fd = fopen(name, "rb");
	if (!fd) {
		perror(name);
		return 0;
	}

	ifnum = testdev_ifnum(fd);
	fclose(fd);
	if (ifnum < 0)
		return 0;

	entry = calloc(1, sizeof *entry);
	if (!entry)
		goto nomem;

	entry->name = strdup(name);
	if (!entry->name) {
		free(entry);
nomem:
		perror("malloc");
		return 0;
	}

	entry->ifnum = ifnum;

	/* FIXME ask usbfs what speed; update USBDEVFS_CONNECTINFO so
	 * it tells about high speed etc */

	fprintf(stderr, "%s speed\t%s\t%u\n",
		speed(entry->speed), entry->name, entry->ifnum);

	entry->next = testdevs;
	testdevs = entry;
	return 0;
}

static int
usbdev_ioctl (int fd, int ifno, unsigned request, void *param)
{
	struct usbdevfs_ioctl	wrapper;

	wrapper.ifno = ifno;
	wrapper.ioctl_code = request;
	wrapper.data = param;

	return ioctl (fd, USBDEVFS_IOCTL, &wrapper);
}

static void *handle_testdev (void *arg)
{
	struct testdev		*dev = arg;
	int			fd, i;
	int			status;

	if ((fd = open (dev->name, O_RDWR)) < 0) {
		perror ("can't open dev file r/w");
		return 0;
	}

restart:
	for (i = 0; i < TEST_CASES; i++) {
		if (dev->test != -1 && dev->test != i)
			continue;
		dev->param.test_num = i;

		status = usbdev_ioctl (fd, dev->ifnum,
				USBTEST_REQUEST, &dev->param);
		if (status < 0 && errno == EOPNOTSUPP)
			continue;

		/* FIXME need a "syslog it" option for background testing */

		/* NOTE: each thread emits complete lines; no fragments! */
		if (status < 0) {
			char	buf [80];
			int	err = errno;

			if (strerror_r (errno, buf, sizeof buf)) {
				snprintf (buf, sizeof buf, "error %d", err);
				errno = err;
			}
			printf ("%s test %d --> %d (%s)\n",
				dev->name, i, errno, buf);
		} else
			printf ("%s test %d, %4d.%.06d secs\n", dev->name, i,
				(int) dev->param.duration.tv_sec,
				(int) dev->param.duration.tv_usec);

		fflush (stdout);
	}
	if (dev->forever)
		goto restart;

	close (fd);
	return arg;
}

static const char *usbfs_dir_find(void)
{
	static char usbfs_path_0[] = "/dev/usb/devices";
	static char usbfs_path_1[] = "/proc/bus/usb/devices";

	static char *const usbfs_paths[] = {
		usbfs_path_0, usbfs_path_1
	};

	static char *const *
		end = usbfs_paths + sizeof usbfs_paths / sizeof *usbfs_paths;

	char *const *it = usbfs_paths;
	do {
		int fd = open(*it, O_RDONLY);
		close(fd);
		if (fd >= 0) {
			strrchr(*it, '/')[0] = '\0';
			return *it;
		}
	} while (++it != end);

	return NULL;
}

static int parse_num(unsigned *num, const char *str)
{
	unsigned long val;
	char *end;

	errno = 0;
	val = strtoul(str, &end, 0);
	if (errno || *end || val > UINT_MAX)
		return -1;
	*num = val;
	return 0;
}

int main (int argc, char **argv)
{

	int			c;
	struct testdev		*entry;
	char			*device;
	const char		*usbfs_dir = NULL;
	int			all = 0, forever = 0, not = 0;
	int			test = -1 /* all */;
	struct usbtest_param	param;

	/* pick defaults that works with all speeds, without short packets.
	 *
	 * Best per-frame data rates:
	 *     high speed, bulk       512 * 13 * 8 = 53248
	 *                 interrupt 1024 *  3 * 8 = 24576
	 *     full speed, bulk/intr   64 * 19     =  1216
	 *                 interrupt   64 *  1     =    64
	 *      low speed, interrupt    8 *  1     =     8
	 */
	param.iterations = 1000;
	param.length = 512;
	param.vary = 512;
	param.sglen = 32;

	/* for easy use when hotplugging */
	device = getenv ("DEVICE");

	while ((c = getopt (argc, argv, "D:aA:c:g:hns:t:v:")) != EOF)
	switch (c) {
	case 'D':	/* device, if only one */
		device = optarg;
		continue;
	case 'A':	/* use all devices with specified usbfs dir */
		usbfs_dir = optarg;
		/* FALL THROUGH */
	case 'a':	/* use all devices */
		device = NULL;
		all = 1;
		continue;
	case 'c':	/* count iterations */
		if (parse_num(&param.iterations, optarg))
			goto usage;
		continue;
	case 'g':	/* scatter/gather entries */
		if (parse_num(&param.sglen, optarg))
			goto usage;
		continue;
	case 'l':	/* loop forever */
		forever = 1;
		continue;
	case 'n':	/* no test running! */
		not = 1;
		continue;
	case 's':	/* size of packet */
		if (parse_num(&param.length, optarg))
			goto usage;
		continue;
	case 't':	/* run just one test */
		test = atoi (optarg);
		if (test < 0)
			goto usage;
		continue;
	case 'v':	/* vary packet size by ... */
		if (parse_num(&param.vary, optarg))
			goto usage;
		continue;
	case '?':
	case 'h':
	default:
usage:
		fprintf (stderr, "usage: %s [-n] [-D dev | -a | -A usbfs-dir]\n"
			"\t[-c iterations]  [-t testnum]\n"
			"\t[-s packetsize] [-g sglen] [-v vary]\n",
			argv [0]);
		return 1;
	}
	if (optind != argc)
		goto usage;
	if (!all && !device) {
		fprintf (stderr, "must specify '-a' or '-D dev', "
			"or DEVICE=/proc/bus/usb/BBB/DDD in env\n");
		goto usage;
	}

	/* Find usbfs mount point */
	if (!usbfs_dir) {
		usbfs_dir = usbfs_dir_find();
		if (!usbfs_dir) {
			fputs ("usbfs files are missing\n", stderr);
			return -1;
		}
	}

	/* collect and list the test devices */
	if (ftw (usbfs_dir, find_testdev, 3) != 0) {
		fputs ("ftw failed; is usbfs missing?\n", stderr);
		return -1;
	}

	/* quit, run single test, or create test threads */
	if (!testdevs && !device) {
		fputs ("no test devices recognized\n", stderr);
		return -1;
	}
	if (not)
		return 0;
	if (testdevs && testdevs->next == 0 && !device)
		device = testdevs->name;
	for (entry = testdevs; entry; entry = entry->next) {
		int	status;

		entry->param = param;
		entry->forever = forever;
		entry->test = test;

		if (device) {
			if (strcmp (entry->name, device))
				continue;
			return handle_testdev (entry) != entry;
		}
		status = pthread_create (&entry->thread, 0, handle_testdev, entry);
		if (status) {
			perror ("pthread_create");
			continue;
		}
	}
	if (device) {
		struct testdev		dev;

		/* kernel can recognize test devices we don't */
		fprintf (stderr, "%s: %s may see only control tests\n",
				argv [0], device);

		memset (&dev, 0, sizeof dev);
		dev.name = device;
		dev.param = param;
		dev.forever = forever;
		dev.test = test;
		return handle_testdev (&dev) != &dev;
	}

	/* wait for tests to complete */
	for (entry = testdevs; entry; entry = entry->next) {
		void	*retval;

		if (pthread_join (entry->thread, &retval))
			perror ("pthread_join");
		/* testing errors discarded! */
	}

	return 0;
}
Commit	Line	Data
5bc9661c	1	/* $(CROSS_COMPILE)cc -Wall -Wextra -g -lpthread -o testusb testusb.c */
2201d6b1 DB	2
	3	/*
	4	* Copyright (c) 2002 by David Brownell
5bc9661c MN	5	* Copyright (c) 2010 by Samsung Electronics
5bc9661c MN	6	* Author: Michal Nazarewicz <[email protected]>
2201d6b1 DB	7	*
	8	* This program is free software; you can redistribute it and/or modify it
	9	* under the terms of the GNU General Public License as published by the
	10	* Free Software Foundation; either version 2 of the License, or (at your
	11	* option) any later version.
	12	*
	13	* This program is distributed in the hope that it will be useful, but
	14	* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
	15	* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
	16	* for more details.
	17	*
	18	* You should have received a copy of the GNU General Public License
	19	* along with this program; if not, write to the Free Software Foundation,
	20	* Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
	21	*/
	22
5bc9661c MN	23	/*
	24	* This program issues ioctls to perform the tests implemented by the
	25	* kernel driver. It can generate a variety of transfer patterns; you
	26	* should make sure to test both regular streaming and mixes of
	27	* transfer sizes (including short transfers).
	28	*
	29	* For more information on how this can be used and on USB testing
	30	* refer to <URL:http://www.linux-usb.org/usbtest/>.
	31	*/
	32
2201d6b1 DB	33	#include <stdio.h>
	34	#include <string.h>
	35	#include <ftw.h>
	36	#include <stdlib.h>
	37	#include <pthread.h>
	38	#include <unistd.h>
	39	#include <errno.h>
5bc9661c	40	#include <limits.h>
2201d6b1 DB	41
	42	#include <sys/types.h>
	43	#include <sys/stat.h>
	44	#include <fcntl.h>
	45
	46	#include <sys/ioctl.h>
	47	#include <linux/usbdevice_fs.h>
	48
	49	/-------------------------------------------------------------------------/
	50
	51	#define TEST_CASES 30
	52
	53	// FIXME make these public somewhere; usbdevfs.h?
	54
	55	struct usbtest_param {
	56	// inputs
	57	unsigned test_num; /* 0..(TEST_CASES-1) */
	58	unsigned iterations;
	59	unsigned length;
	60	unsigned vary;
	61	unsigned sglen;
	62
	63	// outputs
	64	struct timeval duration;
	65	};
	66	#define USBTEST_REQUEST _IOWR('U', 100, struct usbtest_param)
	67
	68	/-------------------------------------------------------------------------/
	69
	70	/* #include <linux/usb_ch9.h> */
	71
5bc9661c MN	72	#define USB_DT_DEVICE 0x01
	73	#define USB_DT_INTERFACE 0x04
	74
	75	#define USB_CLASS_PER_INTERFACE 0 /* for DeviceClass */
	76	#define USB_CLASS_VENDOR_SPEC 0xff
	77
	78
2201d6b1 DB	79	struct usb_device_descriptor {
	80	__u8 bLength;
	81	__u8 bDescriptorType;
	82	__u16 bcdUSB;
	83	__u8 bDeviceClass;
	84	__u8 bDeviceSubClass;
	85	__u8 bDeviceProtocol;
	86	__u8 bMaxPacketSize0;
	87	__u16 idVendor;
	88	__u16 idProduct;
	89	__u16 bcdDevice;
	90	__u8 iManufacturer;
	91	__u8 iProduct;
	92	__u8 iSerialNumber;
	93	__u8 bNumConfigurations;
	94	} __attribute__ ((packed));
	95
5bc9661c MN	96	struct usb_interface_descriptor {
	97	__u8 bLength;
	98	__u8 bDescriptorType;
	99
	100	__u8 bInterfaceNumber;
	101	__u8 bAlternateSetting;
	102	__u8 bNumEndpoints;
	103	__u8 bInterfaceClass;
	104	__u8 bInterfaceSubClass;
	105	__u8 bInterfaceProtocol;
	106	__u8 iInterface;
	107	} __attribute__ ((packed));
	108
2201d6b1 DB	109	enum usb_device_speed {
	110	USB_SPEED_UNKNOWN = 0, /* enumerating */
	111	USB_SPEED_LOW, USB_SPEED_FULL, /* usb 1.1 */
	112	USB_SPEED_HIGH /* usb 2.0 */
	113	};
	114
	115	/-------------------------------------------------------------------------/
	116
	117	static char *speed (enum usb_device_speed s)
	118	{
	119	switch (s) {
	120	case USB_SPEED_UNKNOWN: return "unknown";
	121	case USB_SPEED_LOW: return "low";
	122	case USB_SPEED_FULL: return "full";
	123	case USB_SPEED_HIGH: return "high";
	124	default: return "??";
	125	}
	126	}
	127
	128	struct testdev {
	129	struct testdev *next;
	130	char *name;
	131	pthread_t thread;
	132	enum usb_device_speed speed;
	133	unsigned ifnum : 8;
	134	unsigned forever : 1;
	135	int test;
	136
	137	struct usbtest_param param;
	138	};
	139	static struct testdev *testdevs;
	140
5bc9661c	141	static int testdev_ffs_ifnum(FILE *fd)
2201d6b1	142	{
5bc9661c MN	143	union {
	144	char buf[255];
	145	struct usb_interface_descriptor intf;
	146	} u;
	147
	148	for (;;) {
	149	if (fread(u.buf, 1, 1, fd) != 1)
	150	return -1;
	151	if (fread(u.buf + 1, (unsigned char)u.buf[0] - 1, 1, fd) != 1)
	152	return -1;
	153
	154	if (u.intf.bLength == sizeof u.intf
	155	&& u.intf.bDescriptorType == USB_DT_INTERFACE
	156	&& u.intf.bNumEndpoints == 2
	157	&& u.intf.bInterfaceClass == USB_CLASS_VENDOR_SPEC
	158	&& u.intf.bInterfaceSubClass == 0
	159	&& u.intf.bInterfaceProtocol == 0)
	160	return (unsigned char)u.intf.bInterfaceNumber;
	161	}
	162	}
	163
	164	static int testdev_ifnum(FILE *fd)
	165	{
	166	struct usb_device_descriptor dev;
	167
	168	if (fread(&dev, sizeof dev, 1, fd) != 1)
	169	return -1;
	170
	171	if (dev.bLength != sizeof dev \|\| dev.bDescriptorType != USB_DT_DEVICE)
	172	return -1;
	173
2201d6b1	174	/* FX2 with (tweaked) bulksrc firmware */
5bc9661c MN	175	if (dev.idVendor == 0x0547 && dev.idProduct == 0x1002)
5bc9661c MN	176	return 0;
2201d6b1 DB	177
	178	/----------------------------------------------------/
	179
	180	/* devices that start up using the EZ-USB default device and
	181	* which we can use after loading simple firmware. hotplug
	182	* can fxload it, and then run this test driver.
	183	*
	184	* we return false positives in two cases:
	185	* - the device has a "real" driver (maybe usb-serial) that
	186	* renumerates. the device should vanish quickly.
	187	* - the device doesn't have the test firmware installed.
	188	*/
	189
	190	/* generic EZ-USB FX controller */
5bc9661c MN	191	if (dev.idVendor == 0x0547 && dev.idProduct == 0x2235)
5bc9661c MN	192	return 0;
2201d6b1 DB	193
2201d6b1 DB	194	/* generic EZ-USB FX2 controller */
5bc9661c MN	195	if (dev.idVendor == 0x04b4 && dev.idProduct == 0x8613)
5bc9661c MN	196	return 0;
2201d6b1 DB	197
2201d6b1 DB	198	/* CY3671 development board with EZ-USB FX */
5bc9661c MN	199	if (dev.idVendor == 0x0547 && dev.idProduct == 0x0080)
5bc9661c MN	200	return 0;
2201d6b1 DB	201
2201d6b1 DB	202	/* Keyspan 19Qi uses an21xx (original EZ-USB) */
5bc9661c MN	203	if (dev.idVendor == 0x06cd && dev.idProduct == 0x010b)
5bc9661c MN	204	return 0;
2201d6b1 DB	205
	206	/----------------------------------------------------/
	207
	208	/* "gadget zero", Linux-USB test software */
5bc9661c MN	209	if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4a0)
5bc9661c MN	210	return 0;
2201d6b1 DB	211
2201d6b1 DB	212	/* user mode subset of that */
5bc9661c MN	213	if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4a4)
	214	return testdev_ffs_ifnum(fd);
	215	/* return 0; */
2201d6b1 DB	216
2201d6b1 DB	217	/* iso version of usermode code */
5bc9661c MN	218	if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4a3)
5bc9661c MN	219	return 0;
2201d6b1 DB	220
	221	/* some GPL'd test firmware uses these IDs */
	222
5bc9661c MN	223	if (dev.idVendor == 0xfff0 && dev.idProduct == 0xfff0)
5bc9661c MN	224	return 0;
2201d6b1 DB	225
	226	/----------------------------------------------------/
	227
	228	/* iBOT2 high speed webcam */
5bc9661c MN	229	if (dev.idVendor == 0x0b62 && dev.idProduct == 0x0059)
5bc9661c MN	230	return 0;
2201d6b1	231
5bc9661c MN	232	/----------------------------------------------------/
	233
	234	/* the FunctionFS gadget can have the source/sink interface
	235	* anywhere. We look for an interface descriptor that match
	236	* what we expect. We ignore configuratiens thou. */
	237
	238	if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4ac
	239	&& (dev.bDeviceClass == USB_CLASS_PER_INTERFACE
	240	\|\| dev.bDeviceClass == USB_CLASS_VENDOR_SPEC))
	241	return testdev_ffs_ifnum(fd);
	242
	243	return -1;
2201d6b1 DB	244	}
2201d6b1 DB	245
5bc9661c	246	static int find_testdev(const char name, const struct stat sb, int flag)
2201d6b1	247	{
5bc9661c MN	248	FILE *fd;
	249	int ifnum;
	250	struct testdev *entry;
	251
	252	(void)sb; /* unused */
2201d6b1 DB	253
	254	if (flag != FTW_F)
	255	return 0;
	256	/* ignore /proc/bus/usb/{devices,drivers} */
5bc9661c	257	if (strrchr(name, '/')[1] == 'd')
2201d6b1 DB	258	return 0;
2201d6b1 DB	259
5bc9661c MN	260	fd = fopen(name, "rb");
	261	if (!fd) {
	262	perror(name);
2201d6b1 DB	263	return 0;
2201d6b1 DB	264	}
2201d6b1	265
5bc9661c MN	266	ifnum = testdev_ifnum(fd);
	267	fclose(fd);
	268	if (ifnum < 0)
	269	return 0;
2201d6b1	270
5bc9661c MN	271	entry = calloc(1, sizeof *entry);
	272	if (!entry)
	273	goto nomem;
2201d6b1	274
5bc9661c MN	275	entry->name = strdup(name);
	276	if (!entry->name) {
	277	free(entry);
	278	nomem:
	279	perror("malloc");
	280	return 0;
2201d6b1 DB	281	}
2201d6b1 DB	282
5bc9661c MN	283	entry->ifnum = ifnum;
	284
	285	/* FIXME ask usbfs what speed; update USBDEVFS_CONNECTINFO so
	286	* it tells about high speed etc */
	287
	288	fprintf(stderr, "%s speed\t%s\t%u\n",
	289	speed(entry->speed), entry->name, entry->ifnum);
	290
	291	entry->next = testdevs;
	292	testdevs = entry;
2201d6b1 DB	293	return 0;
	294	}
	295
	296	static int
	297	usbdev_ioctl (int fd, int ifno, unsigned request, void *param)
	298	{
	299	struct usbdevfs_ioctl wrapper;
	300
	301	wrapper.ifno = ifno;
	302	wrapper.ioctl_code = request;
	303	wrapper.data = param;
	304
	305	return ioctl (fd, USBDEVFS_IOCTL, &wrapper);
	306	}
	307
	308	static void handle_testdev (void arg)
	309	{
	310	struct testdev *dev = arg;
	311	int fd, i;
	312	int status;
	313
	314	if ((fd = open (dev->name, O_RDWR)) < 0) {
	315	perror ("can't open dev file r/w");
	316	return 0;
	317	}
	318
	319	restart:
	320	for (i = 0; i < TEST_CASES; i++) {
	321	if (dev->test != -1 && dev->test != i)
	322	continue;
	323	dev->param.test_num = i;
	324
	325	status = usbdev_ioctl (fd, dev->ifnum,
	326	USBTEST_REQUEST, &dev->param);
	327	if (status < 0 && errno == EOPNOTSUPP)
	328	continue;
	329
	330	/* FIXME need a "syslog it" option for background testing */
	331
	332	/* NOTE: each thread emits complete lines; no fragments! */
	333	if (status < 0) {
	334	char buf [80];
	335	int err = errno;
	336
	337	if (strerror_r (errno, buf, sizeof buf)) {
	338	snprintf (buf, sizeof buf, "error %d", err);
	339	errno = err;
	340	}
	341	printf ("%s test %d --> %d (%s)\n",
	342	dev->name, i, errno, buf);
	343	} else
	344	printf ("%s test %d, %4d.%.06d secs\n", dev->name, i,
	345	(int) dev->param.duration.tv_sec,
	346	(int) dev->param.duration.tv_usec);
	347
	348	fflush (stdout);
	349	}
	350	if (dev->forever)
	351	goto restart;
	352
	353	close (fd);
	354	return arg;
	355	}
	356
5bc9661c MN	357	static const char *usbfs_dir_find(void)
	358	{
	359	static char usbfs_path_0[] = "/dev/usb/devices";
	360	static char usbfs_path_1[] = "/proc/bus/usb/devices";
	361
	362	static char *const usbfs_paths[] = {
	363	usbfs_path_0, usbfs_path_1
	364	};
	365
	366	static char const
	367	end = usbfs_paths + sizeof usbfs_paths / sizeof *usbfs_paths;
	368
	369	char const it = usbfs_paths;
	370	do {
	371	int fd = open(*it, O_RDONLY);
	372	close(fd);
	373	if (fd >= 0) {
	374	strrchr(*it, '/')[0] = '\0';
	375	return *it;
	376	}
	377	} while (++it != end);
	378
	379	return NULL;
	380	}
	381
	382	static int parse_num(unsigned num, const char str)
	383	{
	384	unsigned long val;
	385	char *end;
	386
	387	errno = 0;
	388	val = strtoul(str, &end, 0);
	389	if (errno \|\| *end \|\| val > UINT_MAX)
	390	return -1;
	391	*num = val;
	392	return 0;
	393	}
	394
2201d6b1 DB	395	int main (int argc, char **argv)
2201d6b1 DB	396	{
5bc9661c	397
2201d6b1 DB	398	int c;
	399	struct testdev *entry;
	400	char *device;
5bc9661c	401	const char *usbfs_dir = NULL;
2201d6b1 DB	402	int all = 0, forever = 0, not = 0;
	403	int test = -1 /* all */;
	404	struct usbtest_param param;
	405
	406	/* pick defaults that works with all speeds, without short packets.
	407	*
	408	* Best per-frame data rates:
	409	* high speed, bulk 512 * 13 * 8 = 53248
	410	* interrupt 1024 * 3 * 8 = 24576
	411	* full speed, bulk/intr 64 * 19 = 1216
	412	* interrupt 64 * 1 = 64
	413	* low speed, interrupt 8 * 1 = 8
	414	*/
	415	param.iterations = 1000;
	416	param.length = 512;
	417	param.vary = 512;
	418	param.sglen = 32;
	419
	420	/* for easy use when hotplugging */
	421	device = getenv ("DEVICE");
	422
5bc9661c	423	while ((c = getopt (argc, argv, "D:aA:c:g:hns:t:v:")) != EOF)
2201d6b1 DB	424	switch (c) {
	425	case 'D': /* device, if only one */
	426	device = optarg;
	427	continue;
5bc9661c MN	428	case 'A': /* use all devices with specified usbfs dir */
	429	usbfs_dir = optarg;
	430	/* FALL THROUGH */
2201d6b1	431	case 'a': /* use all devices */
5bc9661c	432	device = NULL;
2201d6b1 DB	433	all = 1;
	434	continue;
	435	case 'c': /* count iterations */
5bc9661c	436	if (parse_num(&param.iterations, optarg))
2201d6b1 DB	437	goto usage;
	438	continue;
	439	case 'g': /* scatter/gather entries */
5bc9661c	440	if (parse_num(&param.sglen, optarg))
2201d6b1 DB	441	goto usage;
	442	continue;
	443	case 'l': /* loop forever */
	444	forever = 1;
	445	continue;
	446	case 'n': /* no test running! */
	447	not = 1;
	448	continue;
	449	case 's': /* size of packet */
5bc9661c	450	if (parse_num(&param.length, optarg))
2201d6b1 DB	451	goto usage;
	452	continue;
	453	case 't': /* run just one test */
	454	test = atoi (optarg);
	455	if (test < 0)
	456	goto usage;
	457	continue;
	458	case 'v': /* vary packet size by ... */
5bc9661c	459	if (parse_num(&param.vary, optarg))
2201d6b1 DB	460	goto usage;
	461	continue;
	462	case '?':
	463	case 'h':
	464	default:
	465	usage:
5bc9661c	466	fprintf (stderr, "usage: %s [-n] [-D dev \| -a \| -A usbfs-dir]\n"
2201d6b1 DB	467	"\t[-c iterations] [-t testnum]\n"
	468	"\t[-s packetsize] [-g sglen] [-v vary]\n",
	469	argv [0]);
	470	return 1;
	471	}
	472	if (optind != argc)
	473	goto usage;
	474	if (!all && !device) {
	475	fprintf (stderr, "must specify '-a' or '-D dev', "
	476	"or DEVICE=/proc/bus/usb/BBB/DDD in env\n");
	477	goto usage;
	478	}
	479
5bc9661c MN	480	/* Find usbfs mount point */
	481	if (!usbfs_dir) {
	482	usbfs_dir = usbfs_dir_find();
	483	if (!usbfs_dir) {
	484	fputs ("usbfs files are missing\n", stderr);
	485	return -1;
	486	}
2201d6b1 DB	487	}
	488
	489	/* collect and list the test devices */
5bc9661c	490	if (ftw (usbfs_dir, find_testdev, 3) != 0) {
2201d6b1 DB	491	fputs ("ftw failed; is usbfs missing?\n", stderr);
	492	return -1;
	493	}
	494
	495	/* quit, run single test, or create test threads */
	496	if (!testdevs && !device) {
	497	fputs ("no test devices recognized\n", stderr);
	498	return -1;
	499	}
	500	if (not)
	501	return 0;
	502	if (testdevs && testdevs->next == 0 && !device)
	503	device = testdevs->name;
	504	for (entry = testdevs; entry; entry = entry->next) {
	505	int status;
	506
	507	entry->param = param;
	508	entry->forever = forever;
	509	entry->test = test;
	510
	511	if (device) {
	512	if (strcmp (entry->name, device))
	513	continue;
	514	return handle_testdev (entry) != entry;
	515	}
	516	status = pthread_create (&entry->thread, 0, handle_testdev, entry);
	517	if (status) {
	518	perror ("pthread_create");
	519	continue;
	520	}
	521	}
	522	if (device) {
	523	struct testdev dev;
	524
	525	/* kernel can recognize test devices we don't */
	526	fprintf (stderr, "%s: %s may see only control tests\n",
	527	argv [0], device);
	528
	529	memset (&dev, 0, sizeof dev);
	530	dev.name = device;
	531	dev.param = param;
	532	dev.forever = forever;
	533	dev.test = test;
	534	return handle_testdev (&dev) != &dev;
	535	}
	536
	537	/* wait for tests to complete */
	538	for (entry = testdevs; entry; entry = entry->next) {
	539	void *retval;
	540
	541	if (pthread_join (entry->thread, &retval))
	542	perror ("pthread_join");
	543	/* testing errors discarded! */
	544	}
	545
	546	return 0;
	547	}