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

// SPDX-License-Identifier: GPL-2.0-or-later
/* $(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 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;

	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 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 *usb_dir_find(void)
{
	static char udev_usb_path[] = "/dev/bus/usb";

	if (access(udev_usb_path, F_OK) == 0)
		return udev_usb_path;

	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		*usb_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:
	 *     super speed,bulk      1024 * 16 * 8 = 131072
	 *                 interrupt 1024 *  3 * 8 =  24576
	 *     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 = 1024;
	param.vary = 1024;
	param.sglen = 32;

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

	while ((c = getopt (argc, argv, "D:aA:c:g:hlns:t:v:")) != EOF)
	switch (c) {
	case 'D':	/* device, if only one */
		device = optarg;
		continue;
	case 'A':	/* use all devices with specified USB dir */
		usb_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 [options]\n"
			"Options:\n"
			"\t-D dev		only test specific device\n"
			"\t-A usb-dir\n"
			"\t-a		test all recognized devices\n"
			"\t-l		loop forever(for stress test)\n"
			"\t-t testnum	only run specified case\n"
			"\t-n		no test running, show devices to be tested\n"
			"Case arguments:\n"
			"\t-c iterations		default 1000\n"
			"\t-s transfer length	default 1024\n"
			"\t-g sglen		default 32\n"
			"\t-v vary			default 1024\n",
			argv[0]);
		return 1;
	}
	if (optind != argc)
		goto usage;
	if (!all && !device) {
		fprintf (stderr, "must specify '-a' or '-D dev', "
			"or DEVICE=/dev/bus/usb/BBB/DDD in env\n");
		goto usage;
	}

	/* Find usb device subdirectory */
	if (!usb_dir) {
		usb_dir = usb_dir_find();
		if (!usb_dir) {
			fputs ("USB device files are missing\n", stderr);
			return -1;
		}
	}

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