drivers/media/usb/hdpvr/hdpvr-core.c

   1 /*
   2  * Hauppauge HD PVR USB driver
   3  *
   4  * Copyright (C) 2001-2004 Greg Kroah-Hartman (greg@kroah.com)
   5  * Copyright (C) 2008      Janne Grunau (j@jannau.net)
   6  * Copyright (C) 2008      John Poet
   7  *
   8  *      This program is free software; you can redistribute it and/or
   9  *      modify it under the terms of the GNU General Public License as
  10  *      published by the Free Software Foundation, version 2.
  11  *
  12  */
  13
  14 #include <linux/kernel.h>
  15 #include <linux/errno.h>
  16 #include <linux/init.h>
  17 #include <linux/slab.h>
  18 #include <linux/module.h>
  19 #include <linux/uaccess.h>
  20 #include <linux/atomic.h>
  21 #include <linux/usb.h>
  22 #include <linux/mutex.h>
  23 #include <linux/i2c.h>
  24
  25 #include <linux/videodev2.h>
  26 #include <media/v4l2-dev.h>
  27 #include <media/v4l2-common.h>
  28
  29 #include "hdpvr.h"
  30
  31 static int video_nr[HDPVR_MAX] = {[0 ... (HDPVR_MAX - 1)] = UNSET};
  32 module_param_array(video_nr, int, NULL, 0);
  33 MODULE_PARM_DESC(video_nr, "video device number (-1=Auto)");
  34
  35 /* holds the number of currently registered devices */
  36 static atomic_t dev_nr = ATOMIC_INIT(-1);
  37
  38 int hdpvr_debug;
  39 module_param(hdpvr_debug, int, S_IRUGO|S_IWUSR);
  40 MODULE_PARM_DESC(hdpvr_debug, "enable debugging output");
  41
  42 static uint default_video_input = HDPVR_VIDEO_INPUTS;
  43 module_param(default_video_input, uint, S_IRUGO|S_IWUSR);
  44 MODULE_PARM_DESC(default_video_input, "default video input: 0=Component / "
  45                  "1=S-Video / 2=Composite");
  46
  47 static uint default_audio_input = HDPVR_AUDIO_INPUTS;
  48 module_param(default_audio_input, uint, S_IRUGO|S_IWUSR);
  49 MODULE_PARM_DESC(default_audio_input, "default audio input: 0=RCA back / "
  50                  "1=RCA front / 2=S/PDIF");
  51
  52 static bool boost_audio;
  53 module_param(boost_audio, bool, S_IRUGO|S_IWUSR);
  54 MODULE_PARM_DESC(boost_audio, "boost the audio signal");
  55
  56
  57 /* table of devices that work with this driver */
  58 static struct usb_device_id hdpvr_table[] = {
  59         { USB_DEVICE(HD_PVR_VENDOR_ID, HD_PVR_PRODUCT_ID) },
  60         { USB_DEVICE(HD_PVR_VENDOR_ID, HD_PVR_PRODUCT_ID1) },
  61         { USB_DEVICE(HD_PVR_VENDOR_ID, HD_PVR_PRODUCT_ID2) },
  62         { USB_DEVICE(HD_PVR_VENDOR_ID, HD_PVR_PRODUCT_ID3) },
  63         { USB_DEVICE(HD_PVR_VENDOR_ID, HD_PVR_PRODUCT_ID4) },
  64         { }                                     /* Terminating entry */
  65 };
  66 MODULE_DEVICE_TABLE(usb, hdpvr_table);
  67
  68
  69 void hdpvr_delete(struct hdpvr_device *dev)
  70 {
  71         hdpvr_free_buffers(dev);
  72
  73         if (dev->video_dev)
  74                 video_device_release(dev->video_dev);
  75
  76         usb_put_dev(dev->udev);
  77 }
  78
  79 static void challenge(u8 *bytes)
  80 {
  81         u64 *i64P, tmp64;
  82         uint i, idx;
  83
  84         for (idx = 0; idx < 32; ++idx) {
  85
  86                 if (idx & 0x3)
  87                         bytes[(idx >> 3) + 3] = bytes[(idx >> 2) & 0x3];
  88
  89                 switch (idx & 0x3) {
  90                 case 0x3:
  91                         bytes[2] += bytes[3] * 4 + bytes[4] + bytes[5];
  92                         bytes[4] += bytes[(idx & 0x1) * 2] * 9 + 9;
  93                         break;
  94                 case 0x1:
  95                         bytes[0] *= 8;
  96                         bytes[0] += 7*idx + 4;
  97                         bytes[6] += bytes[3] * 3;
  98                         break;
  99                 case 0x0:
 100                         bytes[3 - (idx >> 3)] = bytes[idx >> 2];
 101                         bytes[5] += bytes[6] * 3;
 102                         for (i = 0; i < 3; i++)
 103                                 bytes[3] *= bytes[3] + 1;
 104                         break;
 105                 case 0x2:
 106                         for (i = 0; i < 3; i++)
 107                                 bytes[1] *= bytes[6] + 1;
 108                         for (i = 0; i < 3; i++) {
 109                                 i64P = (u64 *)bytes;
 110                                 tmp64 = le64_to_cpup(i64P);
 111                                 tmp64 <<= bytes[7] & 0x0f;
 112                                 *i64P += cpu_to_le64(tmp64);
 113                         }
 114                         break;
 115                 }
 116         }
 117 }
 118
 119 /* try to init the device like the windows driver */
 120 static int device_authorization(struct hdpvr_device *dev)
 121 {
 122
 123         int ret, retval = -ENOMEM;
 124         char request_type = 0x38, rcv_request = 0x81;
 125         char *response;
 126 #ifdef HDPVR_DEBUG
 127         size_t buf_size = 46;
 128         char *print_buf = kzalloc(5*buf_size+1, GFP_KERNEL);
 129         if (!print_buf) {
 130                 v4l2_err(&dev->v4l2_dev, "Out of memory\n");
 131                 return retval;
 132         }
 133 #endif
 134
 135         mutex_lock(&dev->usbc_mutex);
 136         ret = usb_control_msg(dev->udev,
 137                               usb_rcvctrlpipe(dev->udev, 0),
 138                               rcv_request, 0x80 | request_type,
 139                               0x0400, 0x0003,
 140                               dev->usbc_buf, 46,
 141                               10000);
 142         if (ret != 46) {
 143                 v4l2_err(&dev->v4l2_dev,
 144                          "unexpected answer of status request, len %d\n", ret);
 145                 goto unlock;
 146         }
 147 #ifdef HDPVR_DEBUG
 148         else {
 149                 hex_dump_to_buffer(dev->usbc_buf, 46, 16, 1, print_buf,
 150                                    5*buf_size+1, 0);
 151                 v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
 152                          "Status request returned, len %d: %s\n",
 153                          ret, print_buf);
 154         }
 155 #endif
 156
 157         dev->fw_ver = dev->usbc_buf[1];
 158
 159         v4l2_info(&dev->v4l2_dev, "firmware version 0x%x dated %s\n",
 160                           dev->fw_ver, &dev->usbc_buf[2]);
 161
 162         if (dev->fw_ver > 0x15) {
 163                 dev->options.brightness = 0x80;
 164                 dev->options.contrast   = 0x40;
 165                 dev->options.hue        = 0xf;
 166                 dev->options.saturation = 0x40;
 167                 dev->options.sharpness  = 0x80;
 168         }
 169
 170         switch (dev->fw_ver) {
 171         case HDPVR_FIRMWARE_VERSION:
 172                 dev->flags &= ~HDPVR_FLAG_AC3_CAP;
 173                 break;
 174         case HDPVR_FIRMWARE_VERSION_AC3:
 175         case HDPVR_FIRMWARE_VERSION_0X12:
 176         case HDPVR_FIRMWARE_VERSION_0X15:
 177         case HDPVR_FIRMWARE_VERSION_0X1E:
 178                 dev->flags |= HDPVR_FLAG_AC3_CAP;
 179                 break;
 180         default:
 181                 v4l2_info(&dev->v4l2_dev, "untested firmware, the driver might"
 182                           " not work.\n");
 183                 if (dev->fw_ver >= HDPVR_FIRMWARE_VERSION_AC3)
 184                         dev->flags |= HDPVR_FLAG_AC3_CAP;
 185                 else
 186                         dev->flags &= ~HDPVR_FLAG_AC3_CAP;
 187         }
 188
 189         response = dev->usbc_buf+38;
 190 #ifdef HDPVR_DEBUG
 191         hex_dump_to_buffer(response, 8, 16, 1, print_buf, 5*buf_size+1, 0);
 192         v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev, "challenge: %s\n",
 193                  print_buf);
 194 #endif
 195         challenge(response);
 196 #ifdef HDPVR_DEBUG
 197         hex_dump_to_buffer(response, 8, 16, 1, print_buf, 5*buf_size+1, 0);
 198         v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev, " response: %s\n",
 199                  print_buf);
 200         kfree(print_buf);
 201 #endif
 202
 203         msleep(100);
 204         ret = usb_control_msg(dev->udev,
 205                               usb_sndctrlpipe(dev->udev, 0),
 206                               0xd1, 0x00 | request_type,
 207                               0x0000, 0x0000,
 208                               response, 8,
 209                               10000);
 210         v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
 211                  "magic request returned %d\n", ret);
 212
 213         retval = ret != 8;
 214 unlock:
 215         mutex_unlock(&dev->usbc_mutex);
 216         return retval;
 217 }
 218
 219 static int hdpvr_device_init(struct hdpvr_device *dev)
 220 {
 221         int ret;
 222         u8 *buf;
 223         struct hdpvr_video_info *vidinf;
 224
 225         if (device_authorization(dev))
 226                 return -EACCES;
 227
 228         /* default options for init */
 229         hdpvr_set_options(dev);
 230
 231         /* set filter options */
 232         mutex_lock(&dev->usbc_mutex);
 233         buf = dev->usbc_buf;
 234         buf[0] = 0x03; buf[1] = 0x03; buf[2] = 0x00; buf[3] = 0x00;
 235         ret = usb_control_msg(dev->udev,
 236                               usb_sndctrlpipe(dev->udev, 0),
 237                               0x01, 0x38,
 238                               CTRL_LOW_PASS_FILTER_VALUE, CTRL_DEFAULT_INDEX,
 239                               buf, 4,
 240                               1000);
 241         v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
 242                  "control request returned %d\n", ret);
 243         mutex_unlock(&dev->usbc_mutex);
 244
 245         vidinf = get_video_info(dev);
 246         if (!vidinf)
 247                 v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
 248                         "no valid video signal or device init failed\n");
 249         else
 250                 kfree(vidinf);
 251
 252         /* enable fan and bling leds */
 253         mutex_lock(&dev->usbc_mutex);
 254         buf[0] = 0x1;
 255         ret = usb_control_msg(dev->udev,
 256                               usb_sndctrlpipe(dev->udev, 0),
 257                               0xd4, 0x38, 0, 0, buf, 1,
 258                               1000);
 259         v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
 260                  "control request returned %d\n", ret);
 261
 262         /* boost analog audio */
 263         buf[0] = boost_audio;
 264         ret = usb_control_msg(dev->udev,
 265                               usb_sndctrlpipe(dev->udev, 0),
 266                               0xd5, 0x38, 0, 0, buf, 1,
 267                               1000);
 268         v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
 269                  "control request returned %d\n", ret);
 270         mutex_unlock(&dev->usbc_mutex);
 271
 272         dev->status = STATUS_IDLE;
 273         return 0;
 274 }
 275
 276 static const struct hdpvr_options hdpvr_default_options = {
 277         .video_std      = HDPVR_60HZ,
 278         .video_input    = HDPVR_COMPONENT,
 279         .audio_input    = HDPVR_RCA_BACK,
 280         .bitrate        = 65, /* 6 mbps */
 281         .peak_bitrate   = 90, /* 9 mbps */
 282         .bitrate_mode   = HDPVR_CONSTANT,
 283         .gop_mode       = HDPVR_SIMPLE_IDR_GOP,
 284         .audio_codec    = V4L2_MPEG_AUDIO_ENCODING_AAC,
 285         /* original picture controls for firmware version <= 0x15 */
 286         /* updated in device_authorization() for newer firmware */
 287         .brightness     = 0x86,
 288         .contrast       = 0x80,
 289         .hue            = 0x80,
 290         .saturation     = 0x80,
 291         .sharpness      = 0x80,
 292 };
 293
 294 static int hdpvr_probe(struct usb_interface *interface,
 295                        const struct usb_device_id *id)
 296 {
 297         struct hdpvr_device *dev;
 298         struct usb_host_interface *iface_desc;
 299         struct usb_endpoint_descriptor *endpoint;
 300         struct i2c_client *client;
 301         size_t buffer_size;
 302         int i;
 303         int retval = -ENOMEM;
 304
 305         /* allocate memory for our device state and initialize it */
 306         dev = kzalloc(sizeof(*dev), GFP_KERNEL);
 307         if (!dev) {
 308                 dev_err(&interface->dev, "Out of memory\n");
 309                 goto error;
 310         }
 311
 312         dev->workqueue = 0;
 313
 314         /* register v4l2_device early so it can be used for printks */
 315         if (v4l2_device_register(&interface->dev, &dev->v4l2_dev)) {
 316                 dev_err(&interface->dev, "v4l2_device_register failed\n");
 317                 goto error;
 318         }
 319
 320         mutex_init(&dev->io_mutex);
 321         mutex_init(&dev->i2c_mutex);
 322         mutex_init(&dev->usbc_mutex);
 323         dev->usbc_buf = kmalloc(64, GFP_KERNEL);
 324         if (!dev->usbc_buf) {
 325                 v4l2_err(&dev->v4l2_dev, "Out of memory\n");
 326                 goto error;
 327         }
 328
 329         init_waitqueue_head(&dev->wait_buffer);
 330         init_waitqueue_head(&dev->wait_data);
 331
 332         dev->workqueue = create_singlethread_workqueue("hdpvr_buffer");
 333         if (!dev->workqueue)
 334                 goto error;
 335
 336         /* init video transfer queues */
 337         INIT_LIST_HEAD(&dev->free_buff_list);
 338         INIT_LIST_HEAD(&dev->rec_buff_list);
 339
 340         dev->options = hdpvr_default_options;
 341
 342         if (default_video_input < HDPVR_VIDEO_INPUTS)
 343                 dev->options.video_input = default_video_input;
 344
 345         if (default_audio_input < HDPVR_AUDIO_INPUTS) {
 346                 dev->options.audio_input = default_audio_input;
 347                 if (default_audio_input == HDPVR_SPDIF)
 348                         dev->options.audio_codec =
 349                                 V4L2_MPEG_AUDIO_ENCODING_AC3;
 350         }
 351
 352         dev->udev = usb_get_dev(interface_to_usbdev(interface));
 353
 354         /* set up the endpoint information */
 355         /* use only the first bulk-in and bulk-out endpoints */
 356         iface_desc = interface->cur_altsetting;
 357         for (i = 0; i < iface_desc->desc.bNumEndpoints; ++i) {
 358                 endpoint = &iface_desc->endpoint[i].desc;
 359
 360                 if (!dev->bulk_in_endpointAddr &&
 361                     usb_endpoint_is_bulk_in(endpoint)) {
 362                         /* USB interface description is buggy, reported max
 363                          * packet size is 512 bytes, windows driver uses 8192 */
 364                         buffer_size = 8192;
 365                         dev->bulk_in_size = buffer_size;
 366                         dev->bulk_in_endpointAddr = endpoint->bEndpointAddress;
 367                 }
 368
 369         }
 370         if (!dev->bulk_in_endpointAddr) {
 371                 v4l2_err(&dev->v4l2_dev, "Could not find bulk-in endpoint\n");
 372                 goto error;
 373         }
 374
 375         /* init the device */
 376         if (hdpvr_device_init(dev)) {
 377                 v4l2_err(&dev->v4l2_dev, "device init failed\n");
 378                 goto error;
 379         }
 380
 381         mutex_lock(&dev->io_mutex);
 382         if (hdpvr_alloc_buffers(dev, NUM_BUFFERS)) {
 383                 mutex_unlock(&dev->io_mutex);
 384                 v4l2_err(&dev->v4l2_dev,
 385                          "allocating transfer buffers failed\n");
 386                 goto error;
 387         }
 388         mutex_unlock(&dev->io_mutex);
 389
 390 #if IS_ENABLED(CONFIG_I2C)
 391         retval = hdpvr_register_i2c_adapter(dev);
 392         if (retval < 0) {
 393                 v4l2_err(&dev->v4l2_dev, "i2c adapter register failed\n");
 394                 goto error;
 395         }
 396
 397         client = hdpvr_register_ir_rx_i2c(dev);
 398         if (!client) {
 399                 v4l2_err(&dev->v4l2_dev, "i2c IR RX device register failed\n");
 400                 retval = -ENODEV;
 401                 goto reg_fail;
 402         }
 403
 404         client = hdpvr_register_ir_tx_i2c(dev);
 405         if (!client) {
 406                 v4l2_err(&dev->v4l2_dev, "i2c IR TX device register failed\n");
 407                 retval = -ENODEV;
 408                 goto reg_fail;
 409         }
 410 #endif
 411
 412         retval = hdpvr_register_videodev(dev, &interface->dev,
 413                                     video_nr[atomic_inc_return(&dev_nr)]);
 414         if (retval < 0) {
 415                 v4l2_err(&dev->v4l2_dev, "registering videodev failed\n");
 416                 goto error;
 417         }
 418
 419         /* let the user know what node this device is now attached to */
 420         v4l2_info(&dev->v4l2_dev, "device now attached to %s\n",
 421                   video_device_node_name(dev->video_dev));
 422         return 0;
 423
 424 reg_fail:
 425 #if IS_ENABLED(CONFIG_I2C)
 426         i2c_del_adapter(&dev->i2c_adapter);
 427 #endif
 428 error:
 429         if (dev) {
 430                 /* Destroy single thread */
 431                 if (dev->workqueue)
 432                         destroy_workqueue(dev->workqueue);
 433                 /* this frees allocated memory */
 434                 hdpvr_delete(dev);
 435         }
 436         return retval;
 437 }
 438
 439 static void hdpvr_disconnect(struct usb_interface *interface)
 440 {
 441         struct hdpvr_device *dev = to_hdpvr_dev(usb_get_intfdata(interface));
 442
 443         v4l2_info(&dev->v4l2_dev, "device %s disconnected\n",
 444                   video_device_node_name(dev->video_dev));
 445         /* prevent more I/O from starting and stop any ongoing */
 446         mutex_lock(&dev->io_mutex);
 447         dev->status = STATUS_DISCONNECTED;
 448         wake_up_interruptible(&dev->wait_data);
 449         wake_up_interruptible(&dev->wait_buffer);
 450         mutex_unlock(&dev->io_mutex);
 451         v4l2_device_disconnect(&dev->v4l2_dev);
 452         msleep(100);
 453         flush_workqueue(dev->workqueue);
 454         mutex_lock(&dev->io_mutex);
 455         hdpvr_cancel_queue(dev);
 456         mutex_unlock(&dev->io_mutex);
 457 #if IS_ENABLED(CONFIG_I2C)
 458         i2c_del_adapter(&dev->i2c_adapter);
 459 #endif
 460         video_unregister_device(dev->video_dev);
 461         atomic_dec(&dev_nr);
 462 }
 463
 464
 465 static struct usb_driver hdpvr_usb_driver = {
 466         .name =         "hdpvr",
 467         .probe =        hdpvr_probe,
 468         .disconnect =   hdpvr_disconnect,
 469         .id_table =     hdpvr_table,
 470 };
 471
 472 module_usb_driver(hdpvr_usb_driver);
 473
 474 MODULE_LICENSE("GPL");
 475 MODULE_VERSION("0.2.1");
 476 MODULE_AUTHOR("Janne Grunau");
 477 MODULE_DESCRIPTION("Hauppauge HD PVR driver");