/* $(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 <mina86@mina86.com>
 *
 * 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;

        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:
         *     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: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 packetsize        default 512\n"
                        "\t-g sglen     default 32\n"
                        "\t-v vary              default 512\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;
}