[mirror_ubuntu-bionic-kernel.git] / drivers / usb / misc / yurex.c

// SPDX-License-Identifier: GPL-2.0
/*
 * Driver for Meywa-Denki & KAYAC YUREX
 *
 * Copyright (C) 2010 Tomoki Sekiyama (tomoki.sekiyama@gmail.com)
 */

#include <linux/kernel.h>
#include <linux/errno.h>
#include <linux/slab.h>
#include <linux/module.h>
#include <linux/kref.h>
#include <linux/mutex.h>
#include <linux/uaccess.h>
#include <linux/usb.h>
#include <linux/hid.h>

#define DRIVER_AUTHOR "Tomoki Sekiyama"
#define DRIVER_DESC "Driver for Meywa-Denki & KAYAC YUREX"

#define YUREX_VENDOR_ID		0x0c45
#define YUREX_PRODUCT_ID	0x1010

#define CMD_ACK		'!'
#define CMD_ANIMATE	'A'
#define CMD_COUNT	'C'
#define CMD_LED		'L'
#define CMD_READ	'R'
#define CMD_SET		'S'
#define CMD_VERSION	'V'
#define CMD_EOF		0x0d
#define CMD_PADDING	0xff

#define YUREX_BUF_SIZE		8
#define YUREX_WRITE_TIMEOUT	(HZ*2)

/* table of devices that work with this driver */
static struct usb_device_id yurex_table[] = {
	{ USB_DEVICE(YUREX_VENDOR_ID, YUREX_PRODUCT_ID) },
	{ }					/* Terminating entry */
};
MODULE_DEVICE_TABLE(usb, yurex_table);

#ifdef CONFIG_USB_DYNAMIC_MINORS
#define YUREX_MINOR_BASE	0
#else
#define YUREX_MINOR_BASE	192
#endif

/* Structure to hold all of our device specific stuff */
struct usb_yurex {
	struct usb_device	*udev;
	struct usb_interface	*interface;
	__u8			int_in_endpointAddr;
	struct urb		*urb;		/* URB for interrupt in */
	unsigned char           *int_buffer;	/* buffer for intterupt in */
	struct urb		*cntl_urb;	/* URB for control msg */
	struct usb_ctrlrequest	*cntl_req;	/* req for control msg */
	unsigned char		*cntl_buffer;	/* buffer for control msg */

	struct kref		kref;
	struct mutex		io_mutex;
	unsigned long		disconnected:1;
	struct fasync_struct	*async_queue;
	wait_queue_head_t	waitq;

	spinlock_t		lock;
	__s64			bbu;		/* BBU from device */
};
#define to_yurex_dev(d) container_of(d, struct usb_yurex, kref)

static struct usb_driver yurex_driver;
static const struct file_operations yurex_fops;


static void yurex_control_callback(struct urb *urb)
{
	struct usb_yurex *dev = urb->context;
	int status = urb->status;

	if (status) {
		dev_err(&urb->dev->dev, "%s - control failed: %d\n",
			__func__, status);
		wake_up_interruptible(&dev->waitq);
		return;
	}
	/* on success, sender woken up by CMD_ACK int in, or timeout */
}

static void yurex_delete(struct kref *kref)
{
	struct usb_yurex *dev = to_yurex_dev(kref);

	dev_dbg(&dev->interface->dev, "%s\n", __func__);

	if (dev->cntl_urb) {
		usb_kill_urb(dev->cntl_urb);
		kfree(dev->cntl_req);
		if (dev->cntl_buffer)
			usb_free_coherent(dev->udev, YUREX_BUF_SIZE,
				dev->cntl_buffer, dev->cntl_urb->transfer_dma);
		usb_free_urb(dev->cntl_urb);
	}
	if (dev->urb) {
		usb_kill_urb(dev->urb);
		if (dev->int_buffer)
			usb_free_coherent(dev->udev, YUREX_BUF_SIZE,
				dev->int_buffer, dev->urb->transfer_dma);
		usb_free_urb(dev->urb);
	}
	usb_put_intf(dev->interface);
	usb_put_dev(dev->udev);
	kfree(dev);
}

/*
 * usb class driver info in order to get a minor number from the usb core,
 * and to have the device registered with the driver core
 */
static struct usb_class_driver yurex_class = {
	.name =		"yurex%d",
	.fops =		&yurex_fops,
	.minor_base =	YUREX_MINOR_BASE,
};

static void yurex_interrupt(struct urb *urb)
{
	struct usb_yurex *dev = urb->context;
	unsigned char *buf = dev->int_buffer;
	int status = urb->status;
	unsigned long flags;
	int retval, i;

	switch (status) {
	case 0: /*success*/
		break;
	/* The device is terminated or messed up, give up */
	case -EOVERFLOW:
		dev_err(&dev->interface->dev,
			"%s - overflow with length %d, actual length is %d\n",
			__func__, YUREX_BUF_SIZE, dev->urb->actual_length);
	case -ECONNRESET:
	case -ENOENT:
	case -ESHUTDOWN:
	case -EILSEQ:
	case -EPROTO:
	case -ETIME:
		return;
	default:
		dev_err(&dev->interface->dev,
			"%s - unknown status received: %d\n", __func__, status);
		return;
	}

	/* handle received message */
	switch (buf[0]) {
	case CMD_COUNT:
	case CMD_READ:
		if (buf[6] == CMD_EOF) {
			spin_lock_irqsave(&dev->lock, flags);
			dev->bbu = 0;
			for (i = 1; i < 6; i++) {
				dev->bbu += buf[i];
				if (i != 5)
					dev->bbu <<= 8;
			}
			dev_dbg(&dev->interface->dev, "%s count: %lld\n",
				__func__, dev->bbu);
			spin_unlock_irqrestore(&dev->lock, flags);

			kill_fasync(&dev->async_queue, SIGIO, POLL_IN);
		}
		else
			dev_dbg(&dev->interface->dev,
				"data format error - no EOF\n");
		break;
	case CMD_ACK:
		dev_dbg(&dev->interface->dev, "%s ack: %c\n",
			__func__, buf[1]);
		wake_up_interruptible(&dev->waitq);
		break;
	}

	retval = usb_submit_urb(dev->urb, GFP_ATOMIC);
	if (retval) {
		dev_err(&dev->interface->dev, "%s - usb_submit_urb failed: %d\n",
			__func__, retval);
	}
}

static int yurex_probe(struct usb_interface *interface, const struct usb_device_id *id)
{
	struct usb_yurex *dev;
	struct usb_host_interface *iface_desc;
	struct usb_endpoint_descriptor *endpoint;
	int retval = -ENOMEM;
	DEFINE_WAIT(wait);
	int res;

	/* allocate memory for our device state and initialize it */
	dev = kzalloc(sizeof(*dev), GFP_KERNEL);
	if (!dev)
		goto error;
	kref_init(&dev->kref);
	mutex_init(&dev->io_mutex);
	spin_lock_init(&dev->lock);
	init_waitqueue_head(&dev->waitq);

	dev->udev = usb_get_dev(interface_to_usbdev(interface));
	dev->interface = usb_get_intf(interface);

	/* set up the endpoint information */
	iface_desc = interface->cur_altsetting;
	res = usb_find_int_in_endpoint(iface_desc, &endpoint);
	if (res) {
		dev_err(&interface->dev, "Could not find endpoints\n");
		retval = res;
		goto error;
	}

	dev->int_in_endpointAddr = endpoint->bEndpointAddress;

	/* allocate control URB */
	dev->cntl_urb = usb_alloc_urb(0, GFP_KERNEL);
	if (!dev->cntl_urb)
		goto error;

	/* allocate buffer for control req */
	dev->cntl_req = kmalloc(YUREX_BUF_SIZE, GFP_KERNEL);
	if (!dev->cntl_req)
		goto error;

	/* allocate buffer for control msg */
	dev->cntl_buffer = usb_alloc_coherent(dev->udev, YUREX_BUF_SIZE,
					      GFP_KERNEL,
					      &dev->cntl_urb->transfer_dma);
	if (!dev->cntl_buffer) {
		dev_err(&interface->dev, "Could not allocate cntl_buffer\n");
		goto error;
	}

	/* configure control URB */
	dev->cntl_req->bRequestType = USB_DIR_OUT | USB_TYPE_CLASS |
				      USB_RECIP_INTERFACE;
	dev->cntl_req->bRequest	= HID_REQ_SET_REPORT;
	dev->cntl_req->wValue	= cpu_to_le16((HID_OUTPUT_REPORT + 1) << 8);
	dev->cntl_req->wIndex	= cpu_to_le16(iface_desc->desc.bInterfaceNumber);
	dev->cntl_req->wLength	= cpu_to_le16(YUREX_BUF_SIZE);

	usb_fill_control_urb(dev->cntl_urb, dev->udev,
			     usb_sndctrlpipe(dev->udev, 0),
			     (void *)dev->cntl_req, dev->cntl_buffer,
			     YUREX_BUF_SIZE, yurex_control_callback, dev);
	dev->cntl_urb->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;


	/* allocate interrupt URB */
	dev->urb = usb_alloc_urb(0, GFP_KERNEL);
	if (!dev->urb)
		goto error;

	/* allocate buffer for interrupt in */
	dev->int_buffer = usb_alloc_coherent(dev->udev, YUREX_BUF_SIZE,
					GFP_KERNEL, &dev->urb->transfer_dma);
	if (!dev->int_buffer) {
		dev_err(&interface->dev, "Could not allocate int_buffer\n");
		goto error;
	}

	/* configure interrupt URB */
	usb_fill_int_urb(dev->urb, dev->udev,
			 usb_rcvintpipe(dev->udev, dev->int_in_endpointAddr),
			 dev->int_buffer, YUREX_BUF_SIZE, yurex_interrupt,
			 dev, 1);
	dev->urb->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
	if (usb_submit_urb(dev->urb, GFP_KERNEL)) {
		retval = -EIO;
		dev_err(&interface->dev, "Could not submitting URB\n");
		goto error;
	}

	/* save our data pointer in this interface device */
	usb_set_intfdata(interface, dev);
	dev->bbu = -1;

	/* we can register the device now, as it is ready */
	retval = usb_register_dev(interface, &yurex_class);
	if (retval) {
		dev_err(&interface->dev,
			"Not able to get a minor for this device.\n");
		usb_set_intfdata(interface, NULL);
		goto error;
	}

	dev_info(&interface->dev,
		 "USB YUREX device now attached to Yurex #%d\n",
		 interface->minor);

	return 0;

error:
	if (dev)
		/* this frees allocated memory */
		kref_put(&dev->kref, yurex_delete);
	return retval;
}

static void yurex_disconnect(struct usb_interface *interface)
{
	struct usb_yurex *dev;
	int minor = interface->minor;

	dev = usb_get_intfdata(interface);
	usb_set_intfdata(interface, NULL);

	/* give back our minor */
	usb_deregister_dev(interface, &yurex_class);

	/* prevent more I/O from starting */
	usb_poison_urb(dev->urb);
	usb_poison_urb(dev->cntl_urb);
	mutex_lock(&dev->io_mutex);
	dev->disconnected = 1;
	mutex_unlock(&dev->io_mutex);

	/* wakeup waiters */
	kill_fasync(&dev->async_queue, SIGIO, POLL_IN);
	wake_up_interruptible(&dev->waitq);

	/* decrement our usage count */
	kref_put(&dev->kref, yurex_delete);

	dev_info(&interface->dev, "USB YUREX #%d now disconnected\n", minor);
}

static struct usb_driver yurex_driver = {
	.name =		"yurex",
	.probe =	yurex_probe,
	.disconnect =	yurex_disconnect,
	.id_table =	yurex_table,
};


static int yurex_fasync(int fd, struct file *file, int on)
{
	struct usb_yurex *dev;

	dev = file->private_data;
	return fasync_helper(fd, file, on, &dev->async_queue);
}

static int yurex_open(struct inode *inode, struct file *file)
{
	struct usb_yurex *dev;
	struct usb_interface *interface;
	int subminor;
	int retval = 0;

	subminor = iminor(inode);

	interface = usb_find_interface(&yurex_driver, subminor);
	if (!interface) {
		printk(KERN_ERR "%s - error, can't find device for minor %d",
		       __func__, subminor);
		retval = -ENODEV;
		goto exit;
	}

	dev = usb_get_intfdata(interface);
	if (!dev) {
		retval = -ENODEV;
		goto exit;
	}

	/* increment our usage count for the device */
	kref_get(&dev->kref);

	/* save our object in the file's private structure */
	mutex_lock(&dev->io_mutex);
	file->private_data = dev;
	mutex_unlock(&dev->io_mutex);

exit:
	return retval;
}

static int yurex_release(struct inode *inode, struct file *file)
{
	struct usb_yurex *dev;

	dev = file->private_data;
	if (dev == NULL)
		return -ENODEV;

	/* decrement the count on our device */
	kref_put(&dev->kref, yurex_delete);
	return 0;
}

static ssize_t yurex_read(struct file *file, char __user *buffer, size_t count,
			  loff_t *ppos)
{
	struct usb_yurex *dev;
	int len = 0;
	char in_buffer[20];
	unsigned long flags;

	dev = file->private_data;

	mutex_lock(&dev->io_mutex);
	if (dev->disconnected) {		/* already disconnected */
		mutex_unlock(&dev->io_mutex);
		return -ENODEV;
	}

	spin_lock_irqsave(&dev->lock, flags);
	len = snprintf(in_buffer, 20, "%lld\n", dev->bbu);
	spin_unlock_irqrestore(&dev->lock, flags);
	mutex_unlock(&dev->io_mutex);

	if (WARN_ON_ONCE(len >= sizeof(in_buffer)))
		return -EIO;

	return simple_read_from_buffer(buffer, count, ppos, in_buffer, len);
}

static ssize_t yurex_write(struct file *file, const char __user *user_buffer,
			   size_t count, loff_t *ppos)
{
	struct usb_yurex *dev;
	int i, set = 0, retval = 0;
	char buffer[16 + 1];
	char *data = buffer;
	unsigned long long c, c2 = 0;
	signed long timeout = 0;
	DEFINE_WAIT(wait);

	count = min(sizeof(buffer) - 1, count);
	dev = file->private_data;

	/* verify that we actually have some data to write */
	if (count == 0)
		goto error;

	mutex_lock(&dev->io_mutex);
	if (dev->disconnected) {		/* already disconnected */
		mutex_unlock(&dev->io_mutex);
		retval = -ENODEV;
		goto error;
	}

	if (copy_from_user(buffer, user_buffer, count)) {
		mutex_unlock(&dev->io_mutex);
		retval = -EFAULT;
		goto error;
	}
	buffer[count] = 0;
	memset(dev->cntl_buffer, CMD_PADDING, YUREX_BUF_SIZE);

	switch (buffer[0]) {
	case CMD_ANIMATE:
	case CMD_LED:
		dev->cntl_buffer[0] = buffer[0];
		dev->cntl_buffer[1] = buffer[1];
		dev->cntl_buffer[2] = CMD_EOF;
		break;
	case CMD_READ:
	case CMD_VERSION:
		dev->cntl_buffer[0] = buffer[0];
		dev->cntl_buffer[1] = 0x00;
		dev->cntl_buffer[2] = CMD_EOF;
		break;
	case CMD_SET:
		data++;
		/* FALL THROUGH */
	case '0' ... '9':
		set = 1;
		c = c2 = simple_strtoull(data, NULL, 0);
		dev->cntl_buffer[0] = CMD_SET;
		for (i = 1; i < 6; i++) {
			dev->cntl_buffer[i] = (c>>32) & 0xff;
			c <<= 8;
		}
		buffer[6] = CMD_EOF;
		break;
	default:
		mutex_unlock(&dev->io_mutex);
		return -EINVAL;
	}

	/* send the data as the control msg */
	prepare_to_wait(&dev->waitq, &wait, TASK_INTERRUPTIBLE);
	dev_dbg(&dev->interface->dev, "%s - submit %c\n", __func__,
		dev->cntl_buffer[0]);
	retval = usb_submit_urb(dev->cntl_urb, GFP_KERNEL);
	if (retval >= 0)
		timeout = schedule_timeout(YUREX_WRITE_TIMEOUT);
	finish_wait(&dev->waitq, &wait);

	mutex_unlock(&dev->io_mutex);

	if (retval < 0) {
		dev_err(&dev->interface->dev,
			"%s - failed to send bulk msg, error %d\n",
			__func__, retval);
		goto error;
	}
	if (set && timeout)
		dev->bbu = c2;
	return timeout ? count : -EIO;

error:
	return retval;
}

static const struct file_operations yurex_fops = {
	.owner =	THIS_MODULE,
	.read =		yurex_read,
	.write =	yurex_write,
	.open =		yurex_open,
	.release =	yurex_release,
	.fasync	=	yurex_fasync,
	.llseek =	default_llseek,
};

module_usb_driver(yurex_driver);

MODULE_LICENSE("GPL");
Commit	Line	Data
5fd54ace	1	// SPDX-License-Identifier: GPL-2.0
6bc235a2 TS	2	/*
	3	* Driver for Meywa-Denki & KAYAC YUREX
	4	*
	5	* Copyright (C) 2010 Tomoki Sekiyama (tomoki.sekiyama@gmail.com)
6bc235a2 TS	6	*/
	7
	8	#include <linux/kernel.h>
	9	#include <linux/errno.h>
6bc235a2 TS	10	#include <linux/slab.h>
	11	#include <linux/module.h>
	12	#include <linux/kref.h>
	13	#include <linux/mutex.h>
	14	#include <linux/uaccess.h>
	15	#include <linux/usb.h>
	16	#include <linux/hid.h>
	17
	18	#define DRIVER_AUTHOR "Tomoki Sekiyama"
	19	#define DRIVER_DESC "Driver for Meywa-Denki & KAYAC YUREX"
	20
	21	#define YUREX_VENDOR_ID 0x0c45
	22	#define YUREX_PRODUCT_ID 0x1010
	23
	24	#define CMD_ACK '!'
	25	#define CMD_ANIMATE 'A'
	26	#define CMD_COUNT 'C'
	27	#define CMD_LED 'L'
	28	#define CMD_READ 'R'
	29	#define CMD_SET 'S'
	30	#define CMD_VERSION 'V'
	31	#define CMD_EOF 0x0d
	32	#define CMD_PADDING 0xff
	33
	34	#define YUREX_BUF_SIZE 8
e06ea97f	35	#define YUREX_WRITE_TIMEOUT (HZ*2)
6bc235a2 TS	36
	37	/* table of devices that work with this driver */
	38	static struct usb_device_id yurex_table[] = {
	39	{ USB_DEVICE(YUREX_VENDOR_ID, YUREX_PRODUCT_ID) },
	40	{ } /* Terminating entry */
	41	};
	42	MODULE_DEVICE_TABLE(usb, yurex_table);
	43
	44	#ifdef CONFIG_USB_DYNAMIC_MINORS
1b62d258	45	#define YUREX_MINOR_BASE 0
6bc235a2	46	#else
1b62d258	47	#define YUREX_MINOR_BASE 192
6bc235a2 TS	48	#endif
	49
	50	/* Structure to hold all of our device specific stuff */
	51	struct usb_yurex {
	52	struct usb_device *udev;
	53	struct usb_interface *interface;
	54	__u8 int_in_endpointAddr;
	55	struct urb urb; / URB for interrupt in */
	56	unsigned char int_buffer; / buffer for intterupt in */
	57	struct urb cntl_urb; / URB for control msg */
	58	struct usb_ctrlrequest cntl_req; / req for control msg */
	59	unsigned char cntl_buffer; / buffer for control msg */
	60
	61	struct kref kref;
	62	struct mutex io_mutex;
3c00ef59	63	unsigned long disconnected:1;
6bc235a2 TS	64	struct fasync_struct *async_queue;
	65	wait_queue_head_t waitq;
	66
	67	spinlock_t lock;
	68	__s64 bbu; /* BBU from device */
	69	};
	70	#define to_yurex_dev(d) container_of(d, struct usb_yurex, kref)
	71
	72	static struct usb_driver yurex_driver;
	73	static const struct file_operations yurex_fops;
	74
	75
	76	static void yurex_control_callback(struct urb *urb)
	77	{
	78	struct usb_yurex *dev = urb->context;
	79	int status = urb->status;
	80
	81	if (status) {
45714104 GKH	82	dev_err(&urb->dev->dev, "%s - control failed: %d\n",
45714104 GKH	83	__func__, status);
6bc235a2 TS	84	wake_up_interruptible(&dev->waitq);
	85	return;
	86	}
	87	/* on success, sender woken up by CMD_ACK int in, or timeout */
	88	}
	89
	90	static void yurex_delete(struct kref *kref)
	91	{
	92	struct usb_yurex *dev = to_yurex_dev(kref);
	93
aadd6472	94	dev_dbg(&dev->interface->dev, "%s\n", __func__);
6bc235a2	95
e06ea97f TS	96	if (dev->cntl_urb) {
e06ea97f TS	97	usb_kill_urb(dev->cntl_urb);
523fc5c1	98	kfree(dev->cntl_req);
e06ea97f TS	99	if (dev->cntl_buffer)
	100	usb_free_coherent(dev->udev, YUREX_BUF_SIZE,
	101	dev->cntl_buffer, dev->cntl_urb->transfer_dma);
	102	usb_free_urb(dev->cntl_urb);
	103	}
6bc235a2 TS	104	if (dev->urb) {
	105	usb_kill_urb(dev->urb);
	106	if (dev->int_buffer)
	107	usb_free_coherent(dev->udev, YUREX_BUF_SIZE,
	108	dev->int_buffer, dev->urb->transfer_dma);
	109	usb_free_urb(dev->urb);
	110	}
3c00ef59	111	usb_put_intf(dev->interface);
88d76fc0	112	usb_put_dev(dev->udev);
6bc235a2 TS	113	kfree(dev);
	114	}
	115
	116	/*
	117	* usb class driver info in order to get a minor number from the usb core,
	118	* and to have the device registered with the driver core
	119	*/
	120	static struct usb_class_driver yurex_class = {
	121	.name = "yurex%d",
	122	.fops = &yurex_fops,
	123	.minor_base = YUREX_MINOR_BASE,
	124	};
	125
	126	static void yurex_interrupt(struct urb *urb)
	127	{
	128	struct usb_yurex *dev = urb->context;
	129	unsigned char *buf = dev->int_buffer;
	130	int status = urb->status;
	131	unsigned long flags;
	132	int retval, i;
	133
	134	switch (status) {
	135	case 0: /success/
	136	break;
ed57add3	137	/* The device is terminated or messed up, give up */
6bc235a2	138	case -EOVERFLOW:
45714104 GKH	139	dev_err(&dev->interface->dev,
	140	"%s - overflow with length %d, actual length is %d\n",
	141	__func__, YUREX_BUF_SIZE, dev->urb->actual_length);
6bc235a2 TS	142	case -ECONNRESET:
	143	case -ENOENT:
	144	case -ESHUTDOWN:
	145	case -EILSEQ:
ed57add3 AS	146	case -EPROTO:
ed57add3 AS	147	case -ETIME:
6bc235a2 TS	148	return;
6bc235a2 TS	149	default:
45714104 GKH	150	dev_err(&dev->interface->dev,
45714104 GKH	151	"%s - unknown status received: %d\n", __func__, status);
ed57add3	152	return;
6bc235a2 TS	153	}
	154
	155	/* handle received message */
	156	switch (buf[0]) {
	157	case CMD_COUNT:
	158	case CMD_READ:
	159	if (buf[6] == CMD_EOF) {
	160	spin_lock_irqsave(&dev->lock, flags);
	161	dev->bbu = 0;
	162	for (i = 1; i < 6; i++) {
	163	dev->bbu += buf[i];
	164	if (i != 5)
	165	dev->bbu <<= 8;
	166	}
aadd6472 GKH	167	dev_dbg(&dev->interface->dev, "%s count: %lld\n",
aadd6472 GKH	168	__func__, dev->bbu);
6bc235a2 TS	169	spin_unlock_irqrestore(&dev->lock, flags);
	170
	171	kill_fasync(&dev->async_queue, SIGIO, POLL_IN);
	172	}
	173	else
aadd6472 GKH	174	dev_dbg(&dev->interface->dev,
aadd6472 GKH	175	"data format error - no EOF\n");
6bc235a2 TS	176	break;
6bc235a2 TS	177	case CMD_ACK:
aadd6472 GKH	178	dev_dbg(&dev->interface->dev, "%s ack: %c\n",
aadd6472 GKH	179	__func__, buf[1]);
6bc235a2 TS	180	wake_up_interruptible(&dev->waitq);
	181	break;
	182	}
	183
6bc235a2 TS	184	retval = usb_submit_urb(dev->urb, GFP_ATOMIC);
6bc235a2 TS	185	if (retval) {
45714104	186	dev_err(&dev->interface->dev, "%s - usb_submit_urb failed: %d\n",
6bc235a2 TS	187	__func__, retval);
	188	}
	189	}
	190
	191	static int yurex_probe(struct usb_interface interface, const struct usb_device_id id)
	192	{
	193	struct usb_yurex *dev;
	194	struct usb_host_interface *iface_desc;
	195	struct usb_endpoint_descriptor *endpoint;
	196	int retval = -ENOMEM;
6bc235a2	197	DEFINE_WAIT(wait);
499841e6	198	int res;
6bc235a2 TS	199
	200	/* allocate memory for our device state and initialize it */
	201	dev = kzalloc(sizeof(*dev), GFP_KERNEL);
0c2bc5c2	202	if (!dev)
6bc235a2	203	goto error;
6bc235a2 TS	204	kref_init(&dev->kref);
	205	mutex_init(&dev->io_mutex);
	206	spin_lock_init(&dev->lock);
	207	init_waitqueue_head(&dev->waitq);
	208
	209	dev->udev = usb_get_dev(interface_to_usbdev(interface));
3c00ef59	210	dev->interface = usb_get_intf(interface);
6bc235a2 TS	211
	212	/* set up the endpoint information */
	213	iface_desc = interface->cur_altsetting;
499841e6 JH	214	res = usb_find_int_in_endpoint(iface_desc, &endpoint);
499841e6 JH	215	if (res) {
45714104	216	dev_err(&interface->dev, "Could not find endpoints\n");
499841e6	217	retval = res;
6bc235a2 TS	218	goto error;
	219	}
	220
499841e6	221	dev->int_in_endpointAddr = endpoint->bEndpointAddress;
6bc235a2 TS	222
	223	/* allocate control URB */
	224	dev->cntl_urb = usb_alloc_urb(0, GFP_KERNEL);
0450ba40	225	if (!dev->cntl_urb)
6bc235a2	226	goto error;
6bc235a2 TS	227
6bc235a2 TS	228	/* allocate buffer for control req */
523fc5c1	229	dev->cntl_req = kmalloc(YUREX_BUF_SIZE, GFP_KERNEL);
0c2bc5c2	230	if (!dev->cntl_req)
6bc235a2	231	goto error;
6bc235a2 TS	232
	233	/* allocate buffer for control msg */
	234	dev->cntl_buffer = usb_alloc_coherent(dev->udev, YUREX_BUF_SIZE,
	235	GFP_KERNEL,
	236	&dev->cntl_urb->transfer_dma);
	237	if (!dev->cntl_buffer) {
45714104	238	dev_err(&interface->dev, "Could not allocate cntl_buffer\n");
6bc235a2 TS	239	goto error;
	240	}
	241
	242	/* configure control URB */
	243	dev->cntl_req->bRequestType = USB_DIR_OUT \| USB_TYPE_CLASS \|
	244	USB_RECIP_INTERFACE;
	245	dev->cntl_req->bRequest = HID_REQ_SET_REPORT;
	246	dev->cntl_req->wValue = cpu_to_le16((HID_OUTPUT_REPORT + 1) << 8);
	247	dev->cntl_req->wIndex = cpu_to_le16(iface_desc->desc.bInterfaceNumber);
	248	dev->cntl_req->wLength = cpu_to_le16(YUREX_BUF_SIZE);
	249
	250	usb_fill_control_urb(dev->cntl_urb, dev->udev,
	251	usb_sndctrlpipe(dev->udev, 0),
	252	(void *)dev->cntl_req, dev->cntl_buffer,
	253	YUREX_BUF_SIZE, yurex_control_callback, dev);
e06ea97f	254	dev->cntl_urb->transfer_flags \|= URB_NO_TRANSFER_DMA_MAP;
6bc235a2 TS	255
	256
	257	/* allocate interrupt URB */
	258	dev->urb = usb_alloc_urb(0, GFP_KERNEL);
0450ba40	259	if (!dev->urb)
6bc235a2	260	goto error;
6bc235a2 TS	261
	262	/* allocate buffer for interrupt in */
	263	dev->int_buffer = usb_alloc_coherent(dev->udev, YUREX_BUF_SIZE,
	264	GFP_KERNEL, &dev->urb->transfer_dma);
	265	if (!dev->int_buffer) {
45714104	266	dev_err(&interface->dev, "Could not allocate int_buffer\n");
6bc235a2 TS	267	goto error;
	268	}
	269
	270	/* configure interrupt URB */
	271	usb_fill_int_urb(dev->urb, dev->udev,
	272	usb_rcvintpipe(dev->udev, dev->int_in_endpointAddr),
	273	dev->int_buffer, YUREX_BUF_SIZE, yurex_interrupt,
	274	dev, 1);
532f17b5	275	dev->urb->transfer_flags \|= URB_NO_TRANSFER_DMA_MAP;
6bc235a2 TS	276	if (usb_submit_urb(dev->urb, GFP_KERNEL)) {
6bc235a2 TS	277	retval = -EIO;
45714104	278	dev_err(&interface->dev, "Could not submitting URB\n");
6bc235a2 TS	279	goto error;
	280	}
	281
	282	/* save our data pointer in this interface device */
	283	usb_set_intfdata(interface, dev);
c78d1ecf	284	dev->bbu = -1;
6bc235a2 TS	285
	286	/* we can register the device now, as it is ready */
	287	retval = usb_register_dev(interface, &yurex_class);
	288	if (retval) {
45714104 GKH	289	dev_err(&interface->dev,
45714104 GKH	290	"Not able to get a minor for this device.\n");
6bc235a2 TS	291	usb_set_intfdata(interface, NULL);
	292	goto error;
	293	}
	294
6bc235a2	295	dev_info(&interface->dev,
e06ea97f	296	"USB YUREX device now attached to Yurex #%d\n",
6bc235a2 TS	297	interface->minor);
	298
	299	return 0;
	300
	301	error:
	302	if (dev)
	303	/* this frees allocated memory */
	304	kref_put(&dev->kref, yurex_delete);
	305	return retval;
	306	}
	307
	308	static void yurex_disconnect(struct usb_interface *interface)
	309	{
	310	struct usb_yurex *dev;
	311	int minor = interface->minor;
	312
	313	dev = usb_get_intfdata(interface);
	314	usb_set_intfdata(interface, NULL);
	315
	316	/* give back our minor */
	317	usb_deregister_dev(interface, &yurex_class);
	318
	319	/* prevent more I/O from starting */
99923880	320	usb_poison_urb(dev->urb);
3c00ef59	321	usb_poison_urb(dev->cntl_urb);
6bc235a2	322	mutex_lock(&dev->io_mutex);
3c00ef59	323	dev->disconnected = 1;
6bc235a2 TS	324	mutex_unlock(&dev->io_mutex);
	325
	326	/* wakeup waiters */
	327	kill_fasync(&dev->async_queue, SIGIO, POLL_IN);
	328	wake_up_interruptible(&dev->waitq);
	329
	330	/* decrement our usage count */
	331	kref_put(&dev->kref, yurex_delete);
	332
e06ea97f	333	dev_info(&interface->dev, "USB YUREX #%d now disconnected\n", minor);
6bc235a2 TS	334	}
	335
	336	static struct usb_driver yurex_driver = {
	337	.name = "yurex",
	338	.probe = yurex_probe,
	339	.disconnect = yurex_disconnect,
	340	.id_table = yurex_table,
	341	};
	342
	343
	344	static int yurex_fasync(int fd, struct file *file, int on)
	345	{
	346	struct usb_yurex *dev;
	347
113ad911	348	dev = file->private_data;
6bc235a2 TS	349	return fasync_helper(fd, file, on, &dev->async_queue);
	350	}
	351
	352	static int yurex_open(struct inode inode, struct file file)
	353	{
	354	struct usb_yurex *dev;
	355	struct usb_interface *interface;
	356	int subminor;
	357	int retval = 0;
	358
	359	subminor = iminor(inode);
	360
	361	interface = usb_find_interface(&yurex_driver, subminor);
	362	if (!interface) {
45714104 GKH	363	printk(KERN_ERR "%s - error, can't find device for minor %d",
45714104 GKH	364	__func__, subminor);
6bc235a2 TS	365	retval = -ENODEV;
	366	goto exit;
	367	}
	368
	369	dev = usb_get_intfdata(interface);
	370	if (!dev) {
	371	retval = -ENODEV;
	372	goto exit;
	373	}
	374
	375	/* increment our usage count for the device */
	376	kref_get(&dev->kref);
	377
	378	/* save our object in the file's private structure */
	379	mutex_lock(&dev->io_mutex);
	380	file->private_data = dev;
	381	mutex_unlock(&dev->io_mutex);
	382
	383	exit:
	384	return retval;
	385	}
	386
	387	static int yurex_release(struct inode inode, struct file file)
	388	{
	389	struct usb_yurex *dev;
	390
113ad911	391	dev = file->private_data;
6bc235a2 TS	392	if (dev == NULL)
	393	return -ENODEV;
	394
6bc235a2 TS	395	/* decrement the count on our device */
	396	kref_put(&dev->kref, yurex_delete);
	397	return 0;
	398	}
	399
1cc373c6 SM	400	static ssize_t yurex_read(struct file file, char __user buffer, size_t count,
1cc373c6 SM	401	loff_t *ppos)
6bc235a2 TS	402	{
6bc235a2 TS	403	struct usb_yurex *dev;
76cc4343	404	int len = 0;
6bc235a2 TS	405	char in_buffer[20];
	406	unsigned long flags;
	407
113ad911	408	dev = file->private_data;
6bc235a2 TS	409
6bc235a2 TS	410	mutex_lock(&dev->io_mutex);
3c00ef59	411	if (dev->disconnected) { /* already disconnected */
76cc4343 JH	412	mutex_unlock(&dev->io_mutex);
76cc4343 JH	413	return -ENODEV;
6bc235a2 TS	414	}
	415
	416	spin_lock_irqsave(&dev->lock, flags);
76cc4343	417	len = snprintf(in_buffer, 20, "%lld\n", dev->bbu);
6bc235a2	418	spin_unlock_irqrestore(&dev->lock, flags);
6bc235a2	419	mutex_unlock(&dev->io_mutex);
76cc4343	420
daaf4950 BH	421	if (WARN_ON_ONCE(len >= sizeof(in_buffer)))
	422	return -EIO;
	423
76cc4343	424	return simple_read_from_buffer(buffer, count, ppos, in_buffer, len);
6bc235a2 TS	425	}
6bc235a2 TS	426
1cc373c6 SM	427	static ssize_t yurex_write(struct file file, const char __user user_buffer,
1cc373c6 SM	428	size_t count, loff_t *ppos)
6bc235a2 TS	429	{
	430	struct usb_yurex *dev;
	431	int i, set = 0, retval = 0;
822091d4	432	char buffer[16 + 1];
6bc235a2 TS	433	char *data = buffer;
	434	unsigned long long c, c2 = 0;
	435	signed long timeout = 0;
	436	DEFINE_WAIT(wait);
	437
822091d4	438	count = min(sizeof(buffer) - 1, count);
113ad911	439	dev = file->private_data;
6bc235a2 TS	440
	441	/* verify that we actually have some data to write */
	442	if (count == 0)
	443	goto error;
	444
	445	mutex_lock(&dev->io_mutex);
3c00ef59	446	if (dev->disconnected) { /* already disconnected */
6bc235a2 TS	447	mutex_unlock(&dev->io_mutex);
	448	retval = -ENODEV;
	449	goto error;
	450	}
	451
	452	if (copy_from_user(buffer, user_buffer, count)) {
	453	mutex_unlock(&dev->io_mutex);
	454	retval = -EFAULT;
	455	goto error;
	456	}
822091d4	457	buffer[count] = 0;
6bc235a2 TS	458	memset(dev->cntl_buffer, CMD_PADDING, YUREX_BUF_SIZE);
	459
	460	switch (buffer[0]) {
	461	case CMD_ANIMATE:
	462	case CMD_LED:
	463	dev->cntl_buffer[0] = buffer[0];
	464	dev->cntl_buffer[1] = buffer[1];
	465	dev->cntl_buffer[2] = CMD_EOF;
	466	break;
	467	case CMD_READ:
	468	case CMD_VERSION:
	469	dev->cntl_buffer[0] = buffer[0];
	470	dev->cntl_buffer[1] = 0x00;
	471	dev->cntl_buffer[2] = CMD_EOF;
	472	break;
	473	case CMD_SET:
	474	data++;
	475	/* FALL THROUGH */
	476	case '0' ... '9':
	477	set = 1;
	478	c = c2 = simple_strtoull(data, NULL, 0);
	479	dev->cntl_buffer[0] = CMD_SET;
	480	for (i = 1; i < 6; i++) {
	481	dev->cntl_buffer[i] = (c>>32) & 0xff;
	482	c <<= 8;
	483	}
	484	buffer[6] = CMD_EOF;
	485	break;
	486	default:
	487	mutex_unlock(&dev->io_mutex);
	488	return -EINVAL;
	489	}
	490
	491	/* send the data as the control msg */
	492	prepare_to_wait(&dev->waitq, &wait, TASK_INTERRUPTIBLE);
aadd6472 GKH	493	dev_dbg(&dev->interface->dev, "%s - submit %c\n", __func__,
aadd6472 GKH	494	dev->cntl_buffer[0]);
6bc235a2 TS	495	retval = usb_submit_urb(dev->cntl_urb, GFP_KERNEL);
	496	if (retval >= 0)
	497	timeout = schedule_timeout(YUREX_WRITE_TIMEOUT);
	498	finish_wait(&dev->waitq, &wait);
	499
	500	mutex_unlock(&dev->io_mutex);
	501
	502	if (retval < 0) {
45714104 GKH	503	dev_err(&dev->interface->dev,
	504	"%s - failed to send bulk msg, error %d\n",
	505	__func__, retval);
6bc235a2 TS	506	goto error;
	507	}
	508	if (set && timeout)
	509	dev->bbu = c2;
	510	return timeout ? count : -EIO;
	511
	512	error:
	513	return retval;
	514	}
	515
	516	static const struct file_operations yurex_fops = {
	517	.owner = THIS_MODULE,
	518	.read = yurex_read,
	519	.write = yurex_write,
	520	.open = yurex_open,
	521	.release = yurex_release,
	522	.fasync = yurex_fasync,
27f485b5	523	.llseek = default_llseek,
6bc235a2 TS	524	};
6bc235a2 TS	525
65db4305	526	module_usb_driver(yurex_driver);
6bc235a2 TS	527
6bc235a2 TS	528	MODULE_LICENSE("GPL");