0001
0002
0003
0004
0005
0006
0007
0008 #include <linux/kernel.h>
0009 #include <linux/errno.h>
0010 #include <linux/slab.h>
0011 #include <linux/module.h>
0012 #include <linux/kref.h>
0013 #include <linux/mutex.h>
0014 #include <linux/uaccess.h>
0015 #include <linux/usb.h>
0016 #include <linux/hid.h>
0017
0018 #define DRIVER_AUTHOR "Tomoki Sekiyama"
0019 #define DRIVER_DESC "Driver for Meywa-Denki & KAYAC YUREX"
0020
0021 #define YUREX_VENDOR_ID 0x0c45
0022 #define YUREX_PRODUCT_ID 0x1010
0023
0024 #define CMD_ACK '!'
0025 #define CMD_ANIMATE 'A'
0026 #define CMD_COUNT 'C'
0027 #define CMD_LED 'L'
0028 #define CMD_READ 'R'
0029 #define CMD_SET 'S'
0030 #define CMD_VERSION 'V'
0031 #define CMD_EOF 0x0d
0032 #define CMD_PADDING 0xff
0033
0034 #define YUREX_BUF_SIZE 8
0035 #define YUREX_WRITE_TIMEOUT (HZ*2)
0036
0037
0038 static struct usb_device_id yurex_table[] = {
0039 { USB_DEVICE(YUREX_VENDOR_ID, YUREX_PRODUCT_ID) },
0040 { }
0041 };
0042 MODULE_DEVICE_TABLE(usb, yurex_table);
0043
0044 #ifdef CONFIG_USB_DYNAMIC_MINORS
0045 #define YUREX_MINOR_BASE 0
0046 #else
0047 #define YUREX_MINOR_BASE 192
0048 #endif
0049
0050
0051 struct usb_yurex {
0052 struct usb_device *udev;
0053 struct usb_interface *interface;
0054 __u8 int_in_endpointAddr;
0055 struct urb *urb;
0056 unsigned char *int_buffer;
0057 struct urb *cntl_urb;
0058 struct usb_ctrlrequest *cntl_req;
0059 unsigned char *cntl_buffer;
0060
0061 struct kref kref;
0062 struct mutex io_mutex;
0063 unsigned long disconnected:1;
0064 struct fasync_struct *async_queue;
0065 wait_queue_head_t waitq;
0066
0067 spinlock_t lock;
0068 __s64 bbu;
0069 };
0070 #define to_yurex_dev(d) container_of(d, struct usb_yurex, kref)
0071
0072 static struct usb_driver yurex_driver;
0073 static const struct file_operations yurex_fops;
0074
0075
0076 static void yurex_control_callback(struct urb *urb)
0077 {
0078 struct usb_yurex *dev = urb->context;
0079 int status = urb->status;
0080
0081 if (status) {
0082 dev_err(&urb->dev->dev, "%s - control failed: %d\n",
0083 __func__, status);
0084 wake_up_interruptible(&dev->waitq);
0085 return;
0086 }
0087
0088 }
0089
0090 static void yurex_delete(struct kref *kref)
0091 {
0092 struct usb_yurex *dev = to_yurex_dev(kref);
0093
0094 dev_dbg(&dev->interface->dev, "%s\n", __func__);
0095
0096 if (dev->cntl_urb) {
0097 usb_kill_urb(dev->cntl_urb);
0098 kfree(dev->cntl_req);
0099 usb_free_coherent(dev->udev, YUREX_BUF_SIZE,
0100 dev->cntl_buffer, dev->cntl_urb->transfer_dma);
0101 usb_free_urb(dev->cntl_urb);
0102 }
0103 if (dev->urb) {
0104 usb_kill_urb(dev->urb);
0105 usb_free_coherent(dev->udev, YUREX_BUF_SIZE,
0106 dev->int_buffer, dev->urb->transfer_dma);
0107 usb_free_urb(dev->urb);
0108 }
0109 usb_put_intf(dev->interface);
0110 usb_put_dev(dev->udev);
0111 kfree(dev);
0112 }
0113
0114
0115
0116
0117
0118 static struct usb_class_driver yurex_class = {
0119 .name = "yurex%d",
0120 .fops = &yurex_fops,
0121 .minor_base = YUREX_MINOR_BASE,
0122 };
0123
0124 static void yurex_interrupt(struct urb *urb)
0125 {
0126 struct usb_yurex *dev = urb->context;
0127 unsigned char *buf = dev->int_buffer;
0128 int status = urb->status;
0129 unsigned long flags;
0130 int retval, i;
0131
0132 switch (status) {
0133 case 0:
0134 break;
0135
0136 case -EOVERFLOW:
0137 dev_err(&dev->interface->dev,
0138 "%s - overflow with length %d, actual length is %d\n",
0139 __func__, YUREX_BUF_SIZE, dev->urb->actual_length);
0140 return;
0141 case -ECONNRESET:
0142 case -ENOENT:
0143 case -ESHUTDOWN:
0144 case -EILSEQ:
0145 case -EPROTO:
0146 case -ETIME:
0147 return;
0148 default:
0149 dev_err(&dev->interface->dev,
0150 "%s - unknown status received: %d\n", __func__, status);
0151 return;
0152 }
0153
0154
0155 switch (buf[0]) {
0156 case CMD_COUNT:
0157 case CMD_READ:
0158 if (buf[6] == CMD_EOF) {
0159 spin_lock_irqsave(&dev->lock, flags);
0160 dev->bbu = 0;
0161 for (i = 1; i < 6; i++) {
0162 dev->bbu += buf[i];
0163 if (i != 5)
0164 dev->bbu <<= 8;
0165 }
0166 dev_dbg(&dev->interface->dev, "%s count: %lld\n",
0167 __func__, dev->bbu);
0168 spin_unlock_irqrestore(&dev->lock, flags);
0169
0170 kill_fasync(&dev->async_queue, SIGIO, POLL_IN);
0171 }
0172 else
0173 dev_dbg(&dev->interface->dev,
0174 "data format error - no EOF\n");
0175 break;
0176 case CMD_ACK:
0177 dev_dbg(&dev->interface->dev, "%s ack: %c\n",
0178 __func__, buf[1]);
0179 wake_up_interruptible(&dev->waitq);
0180 break;
0181 }
0182
0183 retval = usb_submit_urb(dev->urb, GFP_ATOMIC);
0184 if (retval) {
0185 dev_err(&dev->interface->dev, "%s - usb_submit_urb failed: %d\n",
0186 __func__, retval);
0187 }
0188 }
0189
0190 static int yurex_probe(struct usb_interface *interface, const struct usb_device_id *id)
0191 {
0192 struct usb_yurex *dev;
0193 struct usb_host_interface *iface_desc;
0194 struct usb_endpoint_descriptor *endpoint;
0195 int retval = -ENOMEM;
0196 DEFINE_WAIT(wait);
0197 int res;
0198
0199
0200 dev = kzalloc(sizeof(*dev), GFP_KERNEL);
0201 if (!dev)
0202 goto error;
0203 kref_init(&dev->kref);
0204 mutex_init(&dev->io_mutex);
0205 spin_lock_init(&dev->lock);
0206 init_waitqueue_head(&dev->waitq);
0207
0208 dev->udev = usb_get_dev(interface_to_usbdev(interface));
0209 dev->interface = usb_get_intf(interface);
0210
0211
0212 iface_desc = interface->cur_altsetting;
0213 res = usb_find_int_in_endpoint(iface_desc, &endpoint);
0214 if (res) {
0215 dev_err(&interface->dev, "Could not find endpoints\n");
0216 retval = res;
0217 goto error;
0218 }
0219
0220 dev->int_in_endpointAddr = endpoint->bEndpointAddress;
0221
0222
0223 dev->cntl_urb = usb_alloc_urb(0, GFP_KERNEL);
0224 if (!dev->cntl_urb)
0225 goto error;
0226
0227
0228 dev->cntl_req = kmalloc(YUREX_BUF_SIZE, GFP_KERNEL);
0229 if (!dev->cntl_req)
0230 goto error;
0231
0232
0233 dev->cntl_buffer = usb_alloc_coherent(dev->udev, YUREX_BUF_SIZE,
0234 GFP_KERNEL,
0235 &dev->cntl_urb->transfer_dma);
0236 if (!dev->cntl_buffer) {
0237 dev_err(&interface->dev, "Could not allocate cntl_buffer\n");
0238 goto error;
0239 }
0240
0241
0242 dev->cntl_req->bRequestType = USB_DIR_OUT | USB_TYPE_CLASS |
0243 USB_RECIP_INTERFACE;
0244 dev->cntl_req->bRequest = HID_REQ_SET_REPORT;
0245 dev->cntl_req->wValue = cpu_to_le16((HID_OUTPUT_REPORT + 1) << 8);
0246 dev->cntl_req->wIndex = cpu_to_le16(iface_desc->desc.bInterfaceNumber);
0247 dev->cntl_req->wLength = cpu_to_le16(YUREX_BUF_SIZE);
0248
0249 usb_fill_control_urb(dev->cntl_urb, dev->udev,
0250 usb_sndctrlpipe(dev->udev, 0),
0251 (void *)dev->cntl_req, dev->cntl_buffer,
0252 YUREX_BUF_SIZE, yurex_control_callback, dev);
0253 dev->cntl_urb->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
0254
0255
0256
0257 dev->urb = usb_alloc_urb(0, GFP_KERNEL);
0258 if (!dev->urb)
0259 goto error;
0260
0261
0262 dev->int_buffer = usb_alloc_coherent(dev->udev, YUREX_BUF_SIZE,
0263 GFP_KERNEL, &dev->urb->transfer_dma);
0264 if (!dev->int_buffer) {
0265 dev_err(&interface->dev, "Could not allocate int_buffer\n");
0266 goto error;
0267 }
0268
0269
0270 usb_fill_int_urb(dev->urb, dev->udev,
0271 usb_rcvintpipe(dev->udev, dev->int_in_endpointAddr),
0272 dev->int_buffer, YUREX_BUF_SIZE, yurex_interrupt,
0273 dev, 1);
0274 dev->urb->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
0275 if (usb_submit_urb(dev->urb, GFP_KERNEL)) {
0276 retval = -EIO;
0277 dev_err(&interface->dev, "Could not submitting URB\n");
0278 goto error;
0279 }
0280
0281
0282 usb_set_intfdata(interface, dev);
0283 dev->bbu = -1;
0284
0285
0286 retval = usb_register_dev(interface, &yurex_class);
0287 if (retval) {
0288 dev_err(&interface->dev,
0289 "Not able to get a minor for this device.\n");
0290 usb_set_intfdata(interface, NULL);
0291 goto error;
0292 }
0293
0294 dev_info(&interface->dev,
0295 "USB YUREX device now attached to Yurex #%d\n",
0296 interface->minor);
0297
0298 return 0;
0299
0300 error:
0301 if (dev)
0302
0303 kref_put(&dev->kref, yurex_delete);
0304 return retval;
0305 }
0306
0307 static void yurex_disconnect(struct usb_interface *interface)
0308 {
0309 struct usb_yurex *dev;
0310 int minor = interface->minor;
0311
0312 dev = usb_get_intfdata(interface);
0313 usb_set_intfdata(interface, NULL);
0314
0315
0316 usb_deregister_dev(interface, &yurex_class);
0317
0318
0319 usb_poison_urb(dev->urb);
0320 usb_poison_urb(dev->cntl_urb);
0321 mutex_lock(&dev->io_mutex);
0322 dev->disconnected = 1;
0323 mutex_unlock(&dev->io_mutex);
0324
0325
0326 kill_fasync(&dev->async_queue, SIGIO, POLL_IN);
0327 wake_up_interruptible(&dev->waitq);
0328
0329
0330 kref_put(&dev->kref, yurex_delete);
0331
0332 dev_info(&interface->dev, "USB YUREX #%d now disconnected\n", minor);
0333 }
0334
0335 static struct usb_driver yurex_driver = {
0336 .name = "yurex",
0337 .probe = yurex_probe,
0338 .disconnect = yurex_disconnect,
0339 .id_table = yurex_table,
0340 };
0341
0342
0343 static int yurex_fasync(int fd, struct file *file, int on)
0344 {
0345 struct usb_yurex *dev;
0346
0347 dev = file->private_data;
0348 return fasync_helper(fd, file, on, &dev->async_queue);
0349 }
0350
0351 static int yurex_open(struct inode *inode, struct file *file)
0352 {
0353 struct usb_yurex *dev;
0354 struct usb_interface *interface;
0355 int subminor;
0356 int retval = 0;
0357
0358 subminor = iminor(inode);
0359
0360 interface = usb_find_interface(&yurex_driver, subminor);
0361 if (!interface) {
0362 printk(KERN_ERR "%s - error, can't find device for minor %d",
0363 __func__, subminor);
0364 retval = -ENODEV;
0365 goto exit;
0366 }
0367
0368 dev = usb_get_intfdata(interface);
0369 if (!dev) {
0370 retval = -ENODEV;
0371 goto exit;
0372 }
0373
0374
0375 kref_get(&dev->kref);
0376
0377
0378 mutex_lock(&dev->io_mutex);
0379 file->private_data = dev;
0380 mutex_unlock(&dev->io_mutex);
0381
0382 exit:
0383 return retval;
0384 }
0385
0386 static int yurex_release(struct inode *inode, struct file *file)
0387 {
0388 struct usb_yurex *dev;
0389
0390 dev = file->private_data;
0391 if (dev == NULL)
0392 return -ENODEV;
0393
0394
0395 kref_put(&dev->kref, yurex_delete);
0396 return 0;
0397 }
0398
0399 static ssize_t yurex_read(struct file *file, char __user *buffer, size_t count,
0400 loff_t *ppos)
0401 {
0402 struct usb_yurex *dev;
0403 int len = 0;
0404 char in_buffer[20];
0405 unsigned long flags;
0406
0407 dev = file->private_data;
0408
0409 mutex_lock(&dev->io_mutex);
0410 if (dev->disconnected) {
0411 mutex_unlock(&dev->io_mutex);
0412 return -ENODEV;
0413 }
0414
0415 spin_lock_irqsave(&dev->lock, flags);
0416 len = snprintf(in_buffer, 20, "%lld\n", dev->bbu);
0417 spin_unlock_irqrestore(&dev->lock, flags);
0418 mutex_unlock(&dev->io_mutex);
0419
0420 if (WARN_ON_ONCE(len >= sizeof(in_buffer)))
0421 return -EIO;
0422
0423 return simple_read_from_buffer(buffer, count, ppos, in_buffer, len);
0424 }
0425
0426 static ssize_t yurex_write(struct file *file, const char __user *user_buffer,
0427 size_t count, loff_t *ppos)
0428 {
0429 struct usb_yurex *dev;
0430 int i, set = 0, retval = 0;
0431 char buffer[16 + 1];
0432 char *data = buffer;
0433 unsigned long long c, c2 = 0;
0434 signed long timeout = 0;
0435 DEFINE_WAIT(wait);
0436
0437 count = min(sizeof(buffer) - 1, count);
0438 dev = file->private_data;
0439
0440
0441 if (count == 0)
0442 goto error;
0443
0444 mutex_lock(&dev->io_mutex);
0445 if (dev->disconnected) {
0446 mutex_unlock(&dev->io_mutex);
0447 retval = -ENODEV;
0448 goto error;
0449 }
0450
0451 if (copy_from_user(buffer, user_buffer, count)) {
0452 mutex_unlock(&dev->io_mutex);
0453 retval = -EFAULT;
0454 goto error;
0455 }
0456 buffer[count] = 0;
0457 memset(dev->cntl_buffer, CMD_PADDING, YUREX_BUF_SIZE);
0458
0459 switch (buffer[0]) {
0460 case CMD_ANIMATE:
0461 case CMD_LED:
0462 dev->cntl_buffer[0] = buffer[0];
0463 dev->cntl_buffer[1] = buffer[1];
0464 dev->cntl_buffer[2] = CMD_EOF;
0465 break;
0466 case CMD_READ:
0467 case CMD_VERSION:
0468 dev->cntl_buffer[0] = buffer[0];
0469 dev->cntl_buffer[1] = 0x00;
0470 dev->cntl_buffer[2] = CMD_EOF;
0471 break;
0472 case CMD_SET:
0473 data++;
0474 fallthrough;
0475 case '0' ... '9':
0476 set = 1;
0477 c = c2 = simple_strtoull(data, NULL, 0);
0478 dev->cntl_buffer[0] = CMD_SET;
0479 for (i = 1; i < 6; i++) {
0480 dev->cntl_buffer[i] = (c>>32) & 0xff;
0481 c <<= 8;
0482 }
0483 buffer[6] = CMD_EOF;
0484 break;
0485 default:
0486 mutex_unlock(&dev->io_mutex);
0487 return -EINVAL;
0488 }
0489
0490
0491 prepare_to_wait(&dev->waitq, &wait, TASK_INTERRUPTIBLE);
0492 dev_dbg(&dev->interface->dev, "%s - submit %c\n", __func__,
0493 dev->cntl_buffer[0]);
0494 retval = usb_submit_urb(dev->cntl_urb, GFP_ATOMIC);
0495 if (retval >= 0)
0496 timeout = schedule_timeout(YUREX_WRITE_TIMEOUT);
0497 finish_wait(&dev->waitq, &wait);
0498
0499
0500 usb_kill_urb(dev->cntl_urb);
0501
0502 mutex_unlock(&dev->io_mutex);
0503
0504 if (retval < 0) {
0505 dev_err(&dev->interface->dev,
0506 "%s - failed to send bulk msg, error %d\n",
0507 __func__, retval);
0508 goto error;
0509 }
0510 if (set && timeout)
0511 dev->bbu = c2;
0512 return timeout ? count : -EIO;
0513
0514 error:
0515 return retval;
0516 }
0517
0518 static const struct file_operations yurex_fops = {
0519 .owner = THIS_MODULE,
0520 .read = yurex_read,
0521 .write = yurex_write,
0522 .open = yurex_open,
0523 .release = yurex_release,
0524 .fasync = yurex_fasync,
0525 .llseek = default_llseek,
0526 };
0527
0528 module_usb_driver(yurex_driver);
0529
0530 MODULE_LICENSE("GPL");