1 /* 2 * Hauppauge HD PVR USB driver - video 4 linux 2 interface 3 * 4 * Copyright (C) 2008 Janne Grunau (j@jannau.net) 5 * 6 * This program is free software; you can redistribute it and/or 7 * modify it under the terms of the GNU General Public License as 8 * published by the Free Software Foundation, version 2. 9 * 10 */ 11 12 #include <linux/kernel.h> 13 #include <linux/errno.h> 14 #include <linux/init.h> 15 #include <linux/slab.h> 16 #include <linux/module.h> 17 #include <linux/uaccess.h> 18 #include <linux/usb.h> 19 #include <linux/mutex.h> 20 #include <linux/workqueue.h> 21 22 #include <linux/videodev2.h> 23 #include <media/v4l2-dev.h> 24 #include <media/v4l2-common.h> 25 #include <media/v4l2-ioctl.h> 26 #include "hdpvr.h" 27 28 #define BULK_URB_TIMEOUT 90 /* 0.09 seconds */ 29 30 #define print_buffer_status() { \ 31 v4l2_dbg(MSG_BUFFER, hdpvr_debug, &dev->v4l2_dev, \ 32 "%s:%d buffer stat: %d free, %d proc\n", \ 33 __func__, __LINE__, \ 34 list_size(&dev->free_buff_list), \ 35 list_size(&dev->rec_buff_list)); } 36 37 struct hdpvr_fh { 38 struct hdpvr_device *dev; 39 }; 40 41 static uint list_size(struct list_head *list) 42 { 43 struct list_head *tmp; 44 uint count = 0; 45 46 list_for_each(tmp, list) { 47 count++; 48 } 49 50 return count; 51 } 52 53 /*=========================================================================*/ 54 /* urb callback */ 55 static void hdpvr_read_bulk_callback(struct urb *urb) 56 { 57 struct hdpvr_buffer *buf = (struct hdpvr_buffer *)urb->context; 58 struct hdpvr_device *dev = buf->dev; 59 60 /* marking buffer as received and wake waiting */ 61 buf->status = BUFSTAT_READY; 62 wake_up_interruptible(&dev->wait_data); 63 } 64 65 /*=========================================================================*/ 66 /* bufffer bits */ 67 68 /* function expects dev->io_mutex to be hold by caller */ 69 int hdpvr_cancel_queue(struct hdpvr_device *dev) 70 { 71 struct hdpvr_buffer *buf; 72 73 list_for_each_entry(buf, &dev->rec_buff_list, buff_list) { 74 usb_kill_urb(buf->urb); 75 buf->status = BUFSTAT_AVAILABLE; 76 } 77 78 list_splice_init(&dev->rec_buff_list, dev->free_buff_list.prev); 79 80 return 0; 81 } 82 83 static int hdpvr_free_queue(struct list_head *q) 84 { 85 struct list_head *tmp; 86 struct list_head *p; 87 struct hdpvr_buffer *buf; 88 struct urb *urb; 89 90 for (p = q->next; p != q;) { 91 buf = list_entry(p, struct hdpvr_buffer, buff_list); 92 93 urb = buf->urb; 94 usb_free_coherent(urb->dev, urb->transfer_buffer_length, 95 urb->transfer_buffer, urb->transfer_dma); 96 usb_free_urb(urb); 97 tmp = p->next; 98 list_del(p); 99 kfree(buf); 100 p = tmp; 101 } 102 103 return 0; 104 } 105 106 /* function expects dev->io_mutex to be hold by caller */ 107 int hdpvr_free_buffers(struct hdpvr_device *dev) 108 { 109 hdpvr_cancel_queue(dev); 110 111 hdpvr_free_queue(&dev->free_buff_list); 112 hdpvr_free_queue(&dev->rec_buff_list); 113 114 return 0; 115 } 116 117 /* function expects dev->io_mutex to be hold by caller */ 118 int hdpvr_alloc_buffers(struct hdpvr_device *dev, uint count) 119 { 120 uint i; 121 int retval = -ENOMEM; 122 u8 *mem; 123 struct hdpvr_buffer *buf; 124 struct urb *urb; 125 126 v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev, 127 "allocating %u buffers\n", count); 128 129 for (i = 0; i < count; i++) { 130 131 buf = kzalloc(sizeof(struct hdpvr_buffer), GFP_KERNEL); 132 if (!buf) { 133 v4l2_err(&dev->v4l2_dev, "cannot allocate buffer\n"); 134 goto exit; 135 } 136 buf->dev = dev; 137 138 urb = usb_alloc_urb(0, GFP_KERNEL); 139 if (!urb) { 140 v4l2_err(&dev->v4l2_dev, "cannot allocate urb\n"); 141 goto exit_urb; 142 } 143 buf->urb = urb; 144 145 mem = usb_alloc_coherent(dev->udev, dev->bulk_in_size, GFP_KERNEL, 146 &urb->transfer_dma); 147 if (!mem) { 148 v4l2_err(&dev->v4l2_dev, 149 "cannot allocate usb transfer buffer\n"); 150 goto exit_urb_buffer; 151 } 152 153 usb_fill_bulk_urb(buf->urb, dev->udev, 154 usb_rcvbulkpipe(dev->udev, 155 dev->bulk_in_endpointAddr), 156 mem, dev->bulk_in_size, 157 hdpvr_read_bulk_callback, buf); 158 159 buf->urb->transfer_flags |= URB_NO_TRANSFER_DMA_MAP; 160 buf->status = BUFSTAT_AVAILABLE; 161 list_add_tail(&buf->buff_list, &dev->free_buff_list); 162 } 163 return 0; 164 exit_urb_buffer: 165 usb_free_urb(urb); 166 exit_urb: 167 kfree(buf); 168 exit: 169 hdpvr_free_buffers(dev); 170 return retval; 171 } 172 173 static int hdpvr_submit_buffers(struct hdpvr_device *dev) 174 { 175 struct hdpvr_buffer *buf; 176 struct urb *urb; 177 int ret = 0, err_count = 0; 178 179 mutex_lock(&dev->io_mutex); 180 181 while (dev->status == STATUS_STREAMING && 182 !list_empty(&dev->free_buff_list)) { 183 184 buf = list_entry(dev->free_buff_list.next, struct hdpvr_buffer, 185 buff_list); 186 if (buf->status != BUFSTAT_AVAILABLE) { 187 v4l2_err(&dev->v4l2_dev, 188 "buffer not marked as available\n"); 189 ret = -EFAULT; 190 goto err; 191 } 192 193 urb = buf->urb; 194 urb->status = 0; 195 urb->actual_length = 0; 196 ret = usb_submit_urb(urb, GFP_KERNEL); 197 if (ret) { 198 v4l2_err(&dev->v4l2_dev, 199 "usb_submit_urb in %s returned %d\n", 200 __func__, ret); 201 if (++err_count > 2) 202 break; 203 continue; 204 } 205 buf->status = BUFSTAT_INPROGRESS; 206 list_move_tail(&buf->buff_list, &dev->rec_buff_list); 207 } 208 err: 209 print_buffer_status(); 210 mutex_unlock(&dev->io_mutex); 211 return ret; 212 } 213 214 static struct hdpvr_buffer *hdpvr_get_next_buffer(struct hdpvr_device *dev) 215 { 216 struct hdpvr_buffer *buf; 217 218 mutex_lock(&dev->io_mutex); 219 220 if (list_empty(&dev->rec_buff_list)) { 221 mutex_unlock(&dev->io_mutex); 222 return NULL; 223 } 224 225 buf = list_entry(dev->rec_buff_list.next, struct hdpvr_buffer, 226 buff_list); 227 mutex_unlock(&dev->io_mutex); 228 229 return buf; 230 } 231 232 static void hdpvr_transmit_buffers(struct work_struct *work) 233 { 234 struct hdpvr_device *dev = container_of(work, struct hdpvr_device, 235 worker); 236 237 while (dev->status == STATUS_STREAMING) { 238 239 if (hdpvr_submit_buffers(dev)) { 240 v4l2_err(&dev->v4l2_dev, "couldn't submit buffers\n"); 241 goto error; 242 } 243 if (wait_event_interruptible(dev->wait_buffer, 244 !list_empty(&dev->free_buff_list) || 245 dev->status != STATUS_STREAMING)) 246 goto error; 247 } 248 249 v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev, 250 "transmit worker exited\n"); 251 return; 252 error: 253 v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev, 254 "transmit buffers errored\n"); 255 dev->status = STATUS_ERROR; 256 } 257 258 /* function expects dev->io_mutex to be hold by caller */ 259 static int hdpvr_start_streaming(struct hdpvr_device *dev) 260 { 261 int ret; 262 struct hdpvr_video_info *vidinf; 263 264 if (dev->status == STATUS_STREAMING) 265 return 0; 266 else if (dev->status != STATUS_IDLE) 267 return -EAGAIN; 268 269 vidinf = get_video_info(dev); 270 271 if (vidinf) { 272 v4l2_dbg(MSG_BUFFER, hdpvr_debug, &dev->v4l2_dev, 273 "video signal: %dx%d@%dhz\n", vidinf->width, 274 vidinf->height, vidinf->fps); 275 kfree(vidinf); 276 277 /* start streaming 2 request */ 278 ret = usb_control_msg(dev->udev, 279 usb_sndctrlpipe(dev->udev, 0), 280 0xb8, 0x38, 0x1, 0, NULL, 0, 8000); 281 v4l2_dbg(MSG_BUFFER, hdpvr_debug, &dev->v4l2_dev, 282 "encoder start control request returned %d\n", ret); 283 284 hdpvr_config_call(dev, CTRL_START_STREAMING_VALUE, 0x00); 285 286 dev->status = STATUS_STREAMING; 287 288 INIT_WORK(&dev->worker, hdpvr_transmit_buffers); 289 queue_work(dev->workqueue, &dev->worker); 290 291 v4l2_dbg(MSG_BUFFER, hdpvr_debug, &dev->v4l2_dev, 292 "streaming started\n"); 293 294 return 0; 295 } 296 msleep(250); 297 v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev, 298 "no video signal at input %d\n", dev->options.video_input); 299 return -EAGAIN; 300 } 301 302 303 /* function expects dev->io_mutex to be hold by caller */ 304 static int hdpvr_stop_streaming(struct hdpvr_device *dev) 305 { 306 int actual_length; 307 uint c = 0; 308 u8 *buf; 309 310 if (dev->status == STATUS_IDLE) 311 return 0; 312 else if (dev->status != STATUS_STREAMING) 313 return -EAGAIN; 314 315 buf = kmalloc(dev->bulk_in_size, GFP_KERNEL); 316 if (!buf) 317 v4l2_err(&dev->v4l2_dev, "failed to allocate temporary buffer " 318 "for emptying the internal device buffer. " 319 "Next capture start will be slow\n"); 320 321 dev->status = STATUS_SHUTTING_DOWN; 322 hdpvr_config_call(dev, CTRL_STOP_STREAMING_VALUE, 0x00); 323 mutex_unlock(&dev->io_mutex); 324 325 wake_up_interruptible(&dev->wait_buffer); 326 msleep(50); 327 328 flush_workqueue(dev->workqueue); 329 330 mutex_lock(&dev->io_mutex); 331 /* kill the still outstanding urbs */ 332 hdpvr_cancel_queue(dev); 333 334 /* emptying the device buffer beforeshutting it down */ 335 while (buf && ++c < 500 && 336 !usb_bulk_msg(dev->udev, 337 usb_rcvbulkpipe(dev->udev, 338 dev->bulk_in_endpointAddr), 339 buf, dev->bulk_in_size, &actual_length, 340 BULK_URB_TIMEOUT)) { 341 v4l2_dbg(MSG_BUFFER, hdpvr_debug, &dev->v4l2_dev, 342 "%2d: got %d bytes\n", c, actual_length); 343 } 344 kfree(buf); 345 v4l2_dbg(MSG_BUFFER, hdpvr_debug, &dev->v4l2_dev, 346 "used %d urbs to empty device buffers\n", c-1); 347 msleep(10); 348 349 dev->status = STATUS_IDLE; 350 351 return 0; 352 } 353 354 355 /*=======================================================================*/ 356 /* 357 * video 4 linux 2 file operations 358 */ 359 360 static int hdpvr_open(struct file *file) 361 { 362 struct hdpvr_device *dev; 363 struct hdpvr_fh *fh; 364 int retval = -ENOMEM; 365 366 dev = (struct hdpvr_device *)video_get_drvdata(video_devdata(file)); 367 if (!dev) { 368 pr_err("open failing with with ENODEV\n"); 369 retval = -ENODEV; 370 goto err; 371 } 372 373 fh = kzalloc(sizeof(struct hdpvr_fh), GFP_KERNEL); 374 if (!fh) { 375 v4l2_err(&dev->v4l2_dev, "Out of memory\n"); 376 goto err; 377 } 378 /* lock the device to allow correctly handling errors 379 * in resumption */ 380 mutex_lock(&dev->io_mutex); 381 dev->open_count++; 382 mutex_unlock(&dev->io_mutex); 383 384 fh->dev = dev; 385 386 /* save our object in the file's private structure */ 387 file->private_data = fh; 388 389 retval = 0; 390 err: 391 return retval; 392 } 393 394 static int hdpvr_release(struct file *file) 395 { 396 struct hdpvr_fh *fh = file->private_data; 397 struct hdpvr_device *dev = fh->dev; 398 399 if (!dev) 400 return -ENODEV; 401 402 mutex_lock(&dev->io_mutex); 403 if (!(--dev->open_count) && dev->status == STATUS_STREAMING) 404 hdpvr_stop_streaming(dev); 405 406 mutex_unlock(&dev->io_mutex); 407 408 return 0; 409 } 410 411 /* 412 * hdpvr_v4l2_read() 413 * will allocate buffers when called for the first time 414 */ 415 static ssize_t hdpvr_read(struct file *file, char __user *buffer, size_t count, 416 loff_t *pos) 417 { 418 struct hdpvr_fh *fh = file->private_data; 419 struct hdpvr_device *dev = fh->dev; 420 struct hdpvr_buffer *buf = NULL; 421 struct urb *urb; 422 unsigned int ret = 0; 423 int rem, cnt; 424 425 if (*pos) 426 return -ESPIPE; 427 428 if (!dev) 429 return -ENODEV; 430 431 mutex_lock(&dev->io_mutex); 432 if (dev->status == STATUS_IDLE) { 433 if (hdpvr_start_streaming(dev)) { 434 v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev, 435 "start_streaming failed\n"); 436 ret = -EIO; 437 msleep(200); 438 dev->status = STATUS_IDLE; 439 mutex_unlock(&dev->io_mutex); 440 goto err; 441 } 442 print_buffer_status(); 443 } 444 mutex_unlock(&dev->io_mutex); 445 446 /* wait for the first buffer */ 447 if (!(file->f_flags & O_NONBLOCK)) { 448 if (wait_event_interruptible(dev->wait_data, 449 hdpvr_get_next_buffer(dev))) 450 return -ERESTARTSYS; 451 } 452 453 buf = hdpvr_get_next_buffer(dev); 454 455 while (count > 0 && buf) { 456 457 if (buf->status != BUFSTAT_READY && 458 dev->status != STATUS_DISCONNECTED) { 459 /* return nonblocking */ 460 if (file->f_flags & O_NONBLOCK) { 461 if (!ret) 462 ret = -EAGAIN; 463 goto err; 464 } 465 466 if (wait_event_interruptible(dev->wait_data, 467 buf->status == BUFSTAT_READY)) { 468 ret = -ERESTARTSYS; 469 goto err; 470 } 471 } 472 473 if (buf->status != BUFSTAT_READY) 474 break; 475 476 /* set remaining bytes to copy */ 477 urb = buf->urb; 478 rem = urb->actual_length - buf->pos; 479 cnt = rem > count ? count : rem; 480 481 if (copy_to_user(buffer, urb->transfer_buffer + buf->pos, 482 cnt)) { 483 v4l2_err(&dev->v4l2_dev, "read: copy_to_user failed\n"); 484 if (!ret) 485 ret = -EFAULT; 486 goto err; 487 } 488 489 buf->pos += cnt; 490 count -= cnt; 491 buffer += cnt; 492 ret += cnt; 493 494 /* finished, take next buffer */ 495 if (buf->pos == urb->actual_length) { 496 mutex_lock(&dev->io_mutex); 497 buf->pos = 0; 498 buf->status = BUFSTAT_AVAILABLE; 499 500 list_move_tail(&buf->buff_list, &dev->free_buff_list); 501 502 print_buffer_status(); 503 504 mutex_unlock(&dev->io_mutex); 505 506 wake_up_interruptible(&dev->wait_buffer); 507 508 buf = hdpvr_get_next_buffer(dev); 509 } 510 } 511 err: 512 if (!ret && !buf) 513 ret = -EAGAIN; 514 return ret; 515 } 516 517 static unsigned int hdpvr_poll(struct file *filp, poll_table *wait) 518 { 519 struct hdpvr_buffer *buf = NULL; 520 struct hdpvr_fh *fh = filp->private_data; 521 struct hdpvr_device *dev = fh->dev; 522 unsigned int mask = 0; 523 524 mutex_lock(&dev->io_mutex); 525 526 if (!video_is_registered(dev->video_dev)) { 527 mutex_unlock(&dev->io_mutex); 528 return -EIO; 529 } 530 531 if (dev->status == STATUS_IDLE) { 532 if (hdpvr_start_streaming(dev)) { 533 v4l2_dbg(MSG_BUFFER, hdpvr_debug, &dev->v4l2_dev, 534 "start_streaming failed\n"); 535 dev->status = STATUS_IDLE; 536 } 537 538 print_buffer_status(); 539 } 540 mutex_unlock(&dev->io_mutex); 541 542 buf = hdpvr_get_next_buffer(dev); 543 /* only wait if no data is available */ 544 if (!buf || buf->status != BUFSTAT_READY) { 545 poll_wait(filp, &dev->wait_data, wait); 546 buf = hdpvr_get_next_buffer(dev); 547 } 548 if (buf && buf->status == BUFSTAT_READY) 549 mask |= POLLIN | POLLRDNORM; 550 551 return mask; 552 } 553 554 555 static const struct v4l2_file_operations hdpvr_fops = { 556 .owner = THIS_MODULE, 557 .open = hdpvr_open, 558 .release = hdpvr_release, 559 .read = hdpvr_read, 560 .poll = hdpvr_poll, 561 .unlocked_ioctl = video_ioctl2, 562 }; 563 564 /*=======================================================================*/ 565 /* 566 * V4L2 ioctl handling 567 */ 568 569 static int vidioc_querycap(struct file *file, void *priv, 570 struct v4l2_capability *cap) 571 { 572 struct hdpvr_device *dev = video_drvdata(file); 573 574 strcpy(cap->driver, "hdpvr"); 575 strcpy(cap->card, "Hauppauge HD PVR"); 576 usb_make_path(dev->udev, cap->bus_info, sizeof(cap->bus_info)); 577 cap->capabilities = V4L2_CAP_VIDEO_CAPTURE | 578 V4L2_CAP_AUDIO | 579 V4L2_CAP_READWRITE; 580 return 0; 581 } 582 583 static int vidioc_s_std(struct file *file, void *private_data, 584 v4l2_std_id *std) 585 { 586 struct hdpvr_fh *fh = file->private_data; 587 struct hdpvr_device *dev = fh->dev; 588 u8 std_type = 1; 589 590 if (*std & (V4L2_STD_NTSC | V4L2_STD_PAL_60)) 591 std_type = 0; 592 593 return hdpvr_config_call(dev, CTRL_VIDEO_STD_TYPE, std_type); 594 } 595 596 static const char *iname[] = { 597 [HDPVR_COMPONENT] = "Component", 598 [HDPVR_SVIDEO] = "S-Video", 599 [HDPVR_COMPOSITE] = "Composite", 600 }; 601 602 static int vidioc_enum_input(struct file *file, void *priv, 603 struct v4l2_input *i) 604 { 605 struct hdpvr_fh *fh = file->private_data; 606 struct hdpvr_device *dev = fh->dev; 607 unsigned int n; 608 609 n = i->index; 610 if (n >= HDPVR_VIDEO_INPUTS) 611 return -EINVAL; 612 613 i->type = V4L2_INPUT_TYPE_CAMERA; 614 615 strncpy(i->name, iname[n], sizeof(i->name) - 1); 616 i->name[sizeof(i->name) - 1] = '\0'; 617 618 i->audioset = 1<<HDPVR_RCA_FRONT | 1<<HDPVR_RCA_BACK | 1<<HDPVR_SPDIF; 619 620 i->std = dev->video_dev->tvnorms; 621 622 return 0; 623 } 624 625 static int vidioc_s_input(struct file *file, void *private_data, 626 unsigned int index) 627 { 628 struct hdpvr_fh *fh = file->private_data; 629 struct hdpvr_device *dev = fh->dev; 630 int retval; 631 632 if (index >= HDPVR_VIDEO_INPUTS) 633 return -EINVAL; 634 635 if (dev->status != STATUS_IDLE) 636 return -EAGAIN; 637 638 retval = hdpvr_config_call(dev, CTRL_VIDEO_INPUT_VALUE, index+1); 639 if (!retval) 640 dev->options.video_input = index; 641 642 return retval; 643 } 644 645 static int vidioc_g_input(struct file *file, void *private_data, 646 unsigned int *index) 647 { 648 struct hdpvr_fh *fh = file->private_data; 649 struct hdpvr_device *dev = fh->dev; 650 651 *index = dev->options.video_input; 652 return 0; 653 } 654 655 656 static const char *audio_iname[] = { 657 [HDPVR_RCA_FRONT] = "RCA front", 658 [HDPVR_RCA_BACK] = "RCA back", 659 [HDPVR_SPDIF] = "SPDIF", 660 }; 661 662 static int vidioc_enumaudio(struct file *file, void *priv, 663 struct v4l2_audio *audio) 664 { 665 unsigned int n; 666 667 n = audio->index; 668 if (n >= HDPVR_AUDIO_INPUTS) 669 return -EINVAL; 670 671 audio->capability = V4L2_AUDCAP_STEREO; 672 673 strncpy(audio->name, audio_iname[n], sizeof(audio->name) - 1); 674 audio->name[sizeof(audio->name) - 1] = '\0'; 675 676 return 0; 677 } 678 679 static int vidioc_s_audio(struct file *file, void *private_data, 680 const struct v4l2_audio *audio) 681 { 682 struct hdpvr_fh *fh = file->private_data; 683 struct hdpvr_device *dev = fh->dev; 684 int retval; 685 686 if (audio->index >= HDPVR_AUDIO_INPUTS) 687 return -EINVAL; 688 689 if (dev->status != STATUS_IDLE) 690 return -EAGAIN; 691 692 retval = hdpvr_set_audio(dev, audio->index+1, dev->options.audio_codec); 693 if (!retval) 694 dev->options.audio_input = audio->index; 695 696 return retval; 697 } 698 699 static int vidioc_g_audio(struct file *file, void *private_data, 700 struct v4l2_audio *audio) 701 { 702 struct hdpvr_fh *fh = file->private_data; 703 struct hdpvr_device *dev = fh->dev; 704 705 audio->index = dev->options.audio_input; 706 audio->capability = V4L2_AUDCAP_STEREO; 707 strncpy(audio->name, audio_iname[audio->index], sizeof(audio->name)); 708 audio->name[sizeof(audio->name) - 1] = '\0'; 709 return 0; 710 } 711 712 static const s32 supported_v4l2_ctrls[] = { 713 V4L2_CID_BRIGHTNESS, 714 V4L2_CID_CONTRAST, 715 V4L2_CID_SATURATION, 716 V4L2_CID_HUE, 717 V4L2_CID_SHARPNESS, 718 V4L2_CID_MPEG_AUDIO_ENCODING, 719 V4L2_CID_MPEG_VIDEO_ENCODING, 720 V4L2_CID_MPEG_VIDEO_BITRATE_MODE, 721 V4L2_CID_MPEG_VIDEO_BITRATE, 722 V4L2_CID_MPEG_VIDEO_BITRATE_PEAK, 723 }; 724 725 static int fill_queryctrl(struct hdpvr_options *opt, struct v4l2_queryctrl *qc, 726 int ac3, int fw_ver) 727 { 728 int err; 729 730 if (fw_ver > 0x15) { 731 switch (qc->id) { 732 case V4L2_CID_BRIGHTNESS: 733 return v4l2_ctrl_query_fill(qc, 0x0, 0xff, 1, 0x80); 734 case V4L2_CID_CONTRAST: 735 return v4l2_ctrl_query_fill(qc, 0x0, 0xff, 1, 0x40); 736 case V4L2_CID_SATURATION: 737 return v4l2_ctrl_query_fill(qc, 0x0, 0xff, 1, 0x40); 738 case V4L2_CID_HUE: 739 return v4l2_ctrl_query_fill(qc, 0x0, 0x1e, 1, 0xf); 740 case V4L2_CID_SHARPNESS: 741 return v4l2_ctrl_query_fill(qc, 0x0, 0xff, 1, 0x80); 742 } 743 } else { 744 switch (qc->id) { 745 case V4L2_CID_BRIGHTNESS: 746 return v4l2_ctrl_query_fill(qc, 0x0, 0xff, 1, 0x86); 747 case V4L2_CID_CONTRAST: 748 return v4l2_ctrl_query_fill(qc, 0x0, 0xff, 1, 0x80); 749 case V4L2_CID_SATURATION: 750 return v4l2_ctrl_query_fill(qc, 0x0, 0xff, 1, 0x80); 751 case V4L2_CID_HUE: 752 return v4l2_ctrl_query_fill(qc, 0x0, 0xff, 1, 0x80); 753 case V4L2_CID_SHARPNESS: 754 return v4l2_ctrl_query_fill(qc, 0x0, 0xff, 1, 0x80); 755 } 756 } 757 758 switch (qc->id) { 759 case V4L2_CID_MPEG_AUDIO_ENCODING: 760 return v4l2_ctrl_query_fill( 761 qc, V4L2_MPEG_AUDIO_ENCODING_AAC, 762 ac3 ? V4L2_MPEG_AUDIO_ENCODING_AC3 763 : V4L2_MPEG_AUDIO_ENCODING_AAC, 764 1, V4L2_MPEG_AUDIO_ENCODING_AAC); 765 case V4L2_CID_MPEG_VIDEO_ENCODING: 766 return v4l2_ctrl_query_fill( 767 qc, V4L2_MPEG_VIDEO_ENCODING_MPEG_4_AVC, 768 V4L2_MPEG_VIDEO_ENCODING_MPEG_4_AVC, 1, 769 V4L2_MPEG_VIDEO_ENCODING_MPEG_4_AVC); 770 771 /* case V4L2_CID_MPEG_VIDEO_? maybe keyframe interval: */ 772 /* return v4l2_ctrl_query_fill(qc, 0, 128, 128, 0); */ 773 case V4L2_CID_MPEG_VIDEO_BITRATE_MODE: 774 return v4l2_ctrl_query_fill( 775 qc, V4L2_MPEG_VIDEO_BITRATE_MODE_VBR, 776 V4L2_MPEG_VIDEO_BITRATE_MODE_CBR, 1, 777 V4L2_MPEG_VIDEO_BITRATE_MODE_CBR); 778 779 case V4L2_CID_MPEG_VIDEO_BITRATE: 780 return v4l2_ctrl_query_fill(qc, 1000000, 13500000, 100000, 781 6500000); 782 case V4L2_CID_MPEG_VIDEO_BITRATE_PEAK: 783 err = v4l2_ctrl_query_fill(qc, 1100000, 20200000, 100000, 784 9000000); 785 if (!err && opt->bitrate_mode == HDPVR_CONSTANT) 786 qc->flags |= V4L2_CTRL_FLAG_INACTIVE; 787 return err; 788 default: 789 return -EINVAL; 790 } 791 } 792 793 static int vidioc_queryctrl(struct file *file, void *private_data, 794 struct v4l2_queryctrl *qc) 795 { 796 struct hdpvr_fh *fh = file->private_data; 797 struct hdpvr_device *dev = fh->dev; 798 int i, next; 799 u32 id = qc->id; 800 801 memset(qc, 0, sizeof(*qc)); 802 803 next = !!(id & V4L2_CTRL_FLAG_NEXT_CTRL); 804 qc->id = id & ~V4L2_CTRL_FLAG_NEXT_CTRL; 805 806 for (i = 0; i < ARRAY_SIZE(supported_v4l2_ctrls); i++) { 807 if (next) { 808 if (qc->id < supported_v4l2_ctrls[i]) 809 qc->id = supported_v4l2_ctrls[i]; 810 else 811 continue; 812 } 813 814 if (qc->id == supported_v4l2_ctrls[i]) 815 return fill_queryctrl(&dev->options, qc, 816 dev->flags & HDPVR_FLAG_AC3_CAP, 817 dev->fw_ver); 818 819 if (qc->id < supported_v4l2_ctrls[i]) 820 break; 821 } 822 823 return -EINVAL; 824 } 825 826 static int vidioc_g_ctrl(struct file *file, void *private_data, 827 struct v4l2_control *ctrl) 828 { 829 struct hdpvr_fh *fh = file->private_data; 830 struct hdpvr_device *dev = fh->dev; 831 832 switch (ctrl->id) { 833 case V4L2_CID_BRIGHTNESS: 834 ctrl->value = dev->options.brightness; 835 break; 836 case V4L2_CID_CONTRAST: 837 ctrl->value = dev->options.contrast; 838 break; 839 case V4L2_CID_SATURATION: 840 ctrl->value = dev->options.saturation; 841 break; 842 case V4L2_CID_HUE: 843 ctrl->value = dev->options.hue; 844 break; 845 case V4L2_CID_SHARPNESS: 846 ctrl->value = dev->options.sharpness; 847 break; 848 default: 849 return -EINVAL; 850 } 851 return 0; 852 } 853 854 static int vidioc_s_ctrl(struct file *file, void *private_data, 855 struct v4l2_control *ctrl) 856 { 857 struct hdpvr_fh *fh = file->private_data; 858 struct hdpvr_device *dev = fh->dev; 859 int retval; 860 861 switch (ctrl->id) { 862 case V4L2_CID_BRIGHTNESS: 863 retval = hdpvr_config_call(dev, CTRL_BRIGHTNESS, ctrl->value); 864 if (!retval) 865 dev->options.brightness = ctrl->value; 866 break; 867 case V4L2_CID_CONTRAST: 868 retval = hdpvr_config_call(dev, CTRL_CONTRAST, ctrl->value); 869 if (!retval) 870 dev->options.contrast = ctrl->value; 871 break; 872 case V4L2_CID_SATURATION: 873 retval = hdpvr_config_call(dev, CTRL_SATURATION, ctrl->value); 874 if (!retval) 875 dev->options.saturation = ctrl->value; 876 break; 877 case V4L2_CID_HUE: 878 retval = hdpvr_config_call(dev, CTRL_HUE, ctrl->value); 879 if (!retval) 880 dev->options.hue = ctrl->value; 881 break; 882 case V4L2_CID_SHARPNESS: 883 retval = hdpvr_config_call(dev, CTRL_SHARPNESS, ctrl->value); 884 if (!retval) 885 dev->options.sharpness = ctrl->value; 886 break; 887 default: 888 return -EINVAL; 889 } 890 891 return retval; 892 } 893 894 895 static int hdpvr_get_ctrl(struct hdpvr_options *opt, 896 struct v4l2_ext_control *ctrl) 897 { 898 switch (ctrl->id) { 899 case V4L2_CID_MPEG_AUDIO_ENCODING: 900 ctrl->value = opt->audio_codec; 901 break; 902 case V4L2_CID_MPEG_VIDEO_ENCODING: 903 ctrl->value = V4L2_MPEG_VIDEO_ENCODING_MPEG_4_AVC; 904 break; 905 /* case V4L2_CID_MPEG_VIDEO_B_FRAMES: */ 906 /* ctrl->value = (opt->gop_mode & 0x2) ? 0 : 128; */ 907 /* break; */ 908 case V4L2_CID_MPEG_VIDEO_BITRATE_MODE: 909 ctrl->value = opt->bitrate_mode == HDPVR_CONSTANT 910 ? V4L2_MPEG_VIDEO_BITRATE_MODE_CBR 911 : V4L2_MPEG_VIDEO_BITRATE_MODE_VBR; 912 break; 913 case V4L2_CID_MPEG_VIDEO_BITRATE: 914 ctrl->value = opt->bitrate * 100000; 915 break; 916 case V4L2_CID_MPEG_VIDEO_BITRATE_PEAK: 917 ctrl->value = opt->peak_bitrate * 100000; 918 break; 919 case V4L2_CID_MPEG_STREAM_TYPE: 920 ctrl->value = V4L2_MPEG_STREAM_TYPE_MPEG2_TS; 921 break; 922 default: 923 return -EINVAL; 924 } 925 return 0; 926 } 927 928 static int vidioc_g_ext_ctrls(struct file *file, void *priv, 929 struct v4l2_ext_controls *ctrls) 930 { 931 struct hdpvr_fh *fh = file->private_data; 932 struct hdpvr_device *dev = fh->dev; 933 int i, err = 0; 934 935 if (ctrls->ctrl_class == V4L2_CTRL_CLASS_MPEG) { 936 for (i = 0; i < ctrls->count; i++) { 937 struct v4l2_ext_control *ctrl = ctrls->controls + i; 938 939 err = hdpvr_get_ctrl(&dev->options, ctrl); 940 if (err) { 941 ctrls->error_idx = i; 942 break; 943 } 944 } 945 return err; 946 947 } 948 949 return -EINVAL; 950 } 951 952 953 static int hdpvr_try_ctrl(struct v4l2_ext_control *ctrl, int ac3) 954 { 955 int ret = -EINVAL; 956 957 switch (ctrl->id) { 958 case V4L2_CID_MPEG_AUDIO_ENCODING: 959 if (ctrl->value == V4L2_MPEG_AUDIO_ENCODING_AAC || 960 (ac3 && ctrl->value == V4L2_MPEG_AUDIO_ENCODING_AC3)) 961 ret = 0; 962 break; 963 case V4L2_CID_MPEG_VIDEO_ENCODING: 964 if (ctrl->value == V4L2_MPEG_VIDEO_ENCODING_MPEG_4_AVC) 965 ret = 0; 966 break; 967 /* case V4L2_CID_MPEG_VIDEO_B_FRAMES: */ 968 /* if (ctrl->value == 0 || ctrl->value == 128) */ 969 /* ret = 0; */ 970 /* break; */ 971 case V4L2_CID_MPEG_VIDEO_BITRATE_MODE: 972 if (ctrl->value == V4L2_MPEG_VIDEO_BITRATE_MODE_CBR || 973 ctrl->value == V4L2_MPEG_VIDEO_BITRATE_MODE_VBR) 974 ret = 0; 975 break; 976 case V4L2_CID_MPEG_VIDEO_BITRATE: 977 { 978 uint bitrate = ctrl->value / 100000; 979 if (bitrate >= 10 && bitrate <= 135) 980 ret = 0; 981 break; 982 } 983 case V4L2_CID_MPEG_VIDEO_BITRATE_PEAK: 984 { 985 uint peak_bitrate = ctrl->value / 100000; 986 if (peak_bitrate >= 10 && peak_bitrate <= 202) 987 ret = 0; 988 break; 989 } 990 case V4L2_CID_MPEG_STREAM_TYPE: 991 if (ctrl->value == V4L2_MPEG_STREAM_TYPE_MPEG2_TS) 992 ret = 0; 993 break; 994 default: 995 return -EINVAL; 996 } 997 return ret; 998 } 999 1000 static int vidioc_try_ext_ctrls(struct file *file, void *priv, 1001 struct v4l2_ext_controls *ctrls) 1002 { 1003 struct hdpvr_fh *fh = file->private_data; 1004 struct hdpvr_device *dev = fh->dev; 1005 int i, err = 0; 1006 1007 if (ctrls->ctrl_class == V4L2_CTRL_CLASS_MPEG) { 1008 for (i = 0; i < ctrls->count; i++) { 1009 struct v4l2_ext_control *ctrl = ctrls->controls + i; 1010 1011 err = hdpvr_try_ctrl(ctrl, 1012 dev->flags & HDPVR_FLAG_AC3_CAP); 1013 if (err) { 1014 ctrls->error_idx = i; 1015 break; 1016 } 1017 } 1018 return err; 1019 } 1020 1021 return -EINVAL; 1022 } 1023 1024 1025 static int hdpvr_set_ctrl(struct hdpvr_device *dev, 1026 struct v4l2_ext_control *ctrl) 1027 { 1028 struct hdpvr_options *opt = &dev->options; 1029 int ret = 0; 1030 1031 switch (ctrl->id) { 1032 case V4L2_CID_MPEG_AUDIO_ENCODING: 1033 if (dev->flags & HDPVR_FLAG_AC3_CAP) { 1034 opt->audio_codec = ctrl->value; 1035 ret = hdpvr_set_audio(dev, opt->audio_input, 1036 opt->audio_codec); 1037 } 1038 break; 1039 case V4L2_CID_MPEG_VIDEO_ENCODING: 1040 break; 1041 /* case V4L2_CID_MPEG_VIDEO_B_FRAMES: */ 1042 /* if (ctrl->value == 0 && !(opt->gop_mode & 0x2)) { */ 1043 /* opt->gop_mode |= 0x2; */ 1044 /* hdpvr_config_call(dev, CTRL_GOP_MODE_VALUE, */ 1045 /* opt->gop_mode); */ 1046 /* } */ 1047 /* if (ctrl->value == 128 && opt->gop_mode & 0x2) { */ 1048 /* opt->gop_mode &= ~0x2; */ 1049 /* hdpvr_config_call(dev, CTRL_GOP_MODE_VALUE, */ 1050 /* opt->gop_mode); */ 1051 /* } */ 1052 /* break; */ 1053 case V4L2_CID_MPEG_VIDEO_BITRATE_MODE: 1054 if (ctrl->value == V4L2_MPEG_VIDEO_BITRATE_MODE_CBR && 1055 opt->bitrate_mode != HDPVR_CONSTANT) { 1056 opt->bitrate_mode = HDPVR_CONSTANT; 1057 hdpvr_config_call(dev, CTRL_BITRATE_MODE_VALUE, 1058 opt->bitrate_mode); 1059 } 1060 if (ctrl->value == V4L2_MPEG_VIDEO_BITRATE_MODE_VBR && 1061 opt->bitrate_mode == HDPVR_CONSTANT) { 1062 opt->bitrate_mode = HDPVR_VARIABLE_AVERAGE; 1063 hdpvr_config_call(dev, CTRL_BITRATE_MODE_VALUE, 1064 opt->bitrate_mode); 1065 } 1066 break; 1067 case V4L2_CID_MPEG_VIDEO_BITRATE: { 1068 uint bitrate = ctrl->value / 100000; 1069 1070 opt->bitrate = bitrate; 1071 if (bitrate >= opt->peak_bitrate) 1072 opt->peak_bitrate = bitrate+1; 1073 1074 hdpvr_set_bitrate(dev); 1075 break; 1076 } 1077 case V4L2_CID_MPEG_VIDEO_BITRATE_PEAK: { 1078 uint peak_bitrate = ctrl->value / 100000; 1079 1080 if (opt->bitrate_mode == HDPVR_CONSTANT) 1081 break; 1082 1083 if (opt->bitrate < peak_bitrate) { 1084 opt->peak_bitrate = peak_bitrate; 1085 hdpvr_set_bitrate(dev); 1086 } else 1087 ret = -EINVAL; 1088 break; 1089 } 1090 case V4L2_CID_MPEG_STREAM_TYPE: 1091 break; 1092 default: 1093 return -EINVAL; 1094 } 1095 return ret; 1096 } 1097 1098 static int vidioc_s_ext_ctrls(struct file *file, void *priv, 1099 struct v4l2_ext_controls *ctrls) 1100 { 1101 struct hdpvr_fh *fh = file->private_data; 1102 struct hdpvr_device *dev = fh->dev; 1103 int i, err = 0; 1104 1105 if (ctrls->ctrl_class == V4L2_CTRL_CLASS_MPEG) { 1106 for (i = 0; i < ctrls->count; i++) { 1107 struct v4l2_ext_control *ctrl = ctrls->controls + i; 1108 1109 err = hdpvr_try_ctrl(ctrl, 1110 dev->flags & HDPVR_FLAG_AC3_CAP); 1111 if (err) { 1112 ctrls->error_idx = i; 1113 break; 1114 } 1115 err = hdpvr_set_ctrl(dev, ctrl); 1116 if (err) { 1117 ctrls->error_idx = i; 1118 break; 1119 } 1120 } 1121 return err; 1122 1123 } 1124 1125 return -EINVAL; 1126 } 1127 1128 static int vidioc_enum_fmt_vid_cap(struct file *file, void *private_data, 1129 struct v4l2_fmtdesc *f) 1130 { 1131 1132 if (f->index != 0 || f->type != V4L2_BUF_TYPE_VIDEO_CAPTURE) 1133 return -EINVAL; 1134 1135 f->flags = V4L2_FMT_FLAG_COMPRESSED; 1136 strncpy(f->description, "MPEG2-TS with AVC/AAC streams", 32); 1137 f->pixelformat = V4L2_PIX_FMT_MPEG; 1138 1139 return 0; 1140 } 1141 1142 static int vidioc_g_fmt_vid_cap(struct file *file, void *private_data, 1143 struct v4l2_format *f) 1144 { 1145 struct hdpvr_fh *fh = file->private_data; 1146 struct hdpvr_device *dev = fh->dev; 1147 struct hdpvr_video_info *vid_info; 1148 1149 if (!dev) 1150 return -ENODEV; 1151 1152 vid_info = get_video_info(dev); 1153 if (!vid_info) 1154 return -EFAULT; 1155 1156 f->type = V4L2_BUF_TYPE_VIDEO_CAPTURE; 1157 f->fmt.pix.pixelformat = V4L2_PIX_FMT_MPEG; 1158 f->fmt.pix.width = vid_info->width; 1159 f->fmt.pix.height = vid_info->height; 1160 f->fmt.pix.sizeimage = dev->bulk_in_size; 1161 f->fmt.pix.colorspace = 0; 1162 f->fmt.pix.bytesperline = 0; 1163 f->fmt.pix.field = V4L2_FIELD_ANY; 1164 1165 kfree(vid_info); 1166 return 0; 1167 } 1168 1169 static int vidioc_encoder_cmd(struct file *filp, void *priv, 1170 struct v4l2_encoder_cmd *a) 1171 { 1172 struct hdpvr_fh *fh = filp->private_data; 1173 struct hdpvr_device *dev = fh->dev; 1174 int res; 1175 1176 mutex_lock(&dev->io_mutex); 1177 1178 memset(&a->raw, 0, sizeof(a->raw)); 1179 switch (a->cmd) { 1180 case V4L2_ENC_CMD_START: 1181 a->flags = 0; 1182 res = hdpvr_start_streaming(dev); 1183 break; 1184 case V4L2_ENC_CMD_STOP: 1185 res = hdpvr_stop_streaming(dev); 1186 break; 1187 default: 1188 v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev, 1189 "Unsupported encoder cmd %d\n", a->cmd); 1190 res = -EINVAL; 1191 } 1192 mutex_unlock(&dev->io_mutex); 1193 return res; 1194 } 1195 1196 static int vidioc_try_encoder_cmd(struct file *filp, void *priv, 1197 struct v4l2_encoder_cmd *a) 1198 { 1199 switch (a->cmd) { 1200 case V4L2_ENC_CMD_START: 1201 case V4L2_ENC_CMD_STOP: 1202 return 0; 1203 default: 1204 return -EINVAL; 1205 } 1206 } 1207 1208 static const struct v4l2_ioctl_ops hdpvr_ioctl_ops = { 1209 .vidioc_querycap = vidioc_querycap, 1210 .vidioc_s_std = vidioc_s_std, 1211 .vidioc_enum_input = vidioc_enum_input, 1212 .vidioc_g_input = vidioc_g_input, 1213 .vidioc_s_input = vidioc_s_input, 1214 .vidioc_enumaudio = vidioc_enumaudio, 1215 .vidioc_g_audio = vidioc_g_audio, 1216 .vidioc_s_audio = vidioc_s_audio, 1217 .vidioc_queryctrl = vidioc_queryctrl, 1218 .vidioc_g_ctrl = vidioc_g_ctrl, 1219 .vidioc_s_ctrl = vidioc_s_ctrl, 1220 .vidioc_g_ext_ctrls = vidioc_g_ext_ctrls, 1221 .vidioc_s_ext_ctrls = vidioc_s_ext_ctrls, 1222 .vidioc_try_ext_ctrls = vidioc_try_ext_ctrls, 1223 .vidioc_enum_fmt_vid_cap = vidioc_enum_fmt_vid_cap, 1224 .vidioc_g_fmt_vid_cap = vidioc_g_fmt_vid_cap, 1225 .vidioc_encoder_cmd = vidioc_encoder_cmd, 1226 .vidioc_try_encoder_cmd = vidioc_try_encoder_cmd, 1227 }; 1228 1229 static void hdpvr_device_release(struct video_device *vdev) 1230 { 1231 struct hdpvr_device *dev = video_get_drvdata(vdev); 1232 1233 hdpvr_delete(dev); 1234 mutex_lock(&dev->io_mutex); 1235 destroy_workqueue(dev->workqueue); 1236 mutex_unlock(&dev->io_mutex); 1237 1238 v4l2_device_unregister(&dev->v4l2_dev); 1239 1240 /* deregister I2C adapter */ 1241 #if defined(CONFIG_I2C) || (CONFIG_I2C_MODULE) 1242 mutex_lock(&dev->i2c_mutex); 1243 i2c_del_adapter(&dev->i2c_adapter); 1244 mutex_unlock(&dev->i2c_mutex); 1245 #endif /* CONFIG_I2C */ 1246 1247 kfree(dev->usbc_buf); 1248 kfree(dev); 1249 } 1250 1251 static const struct video_device hdpvr_video_template = { 1252 /* .type = VFL_TYPE_GRABBER, */ 1253 /* .type2 = VID_TYPE_CAPTURE | VID_TYPE_MPEG_ENCODER, */ 1254 .fops = &hdpvr_fops, 1255 .release = hdpvr_device_release, 1256 .ioctl_ops = &hdpvr_ioctl_ops, 1257 .tvnorms = 1258 V4L2_STD_NTSC | V4L2_STD_SECAM | V4L2_STD_PAL_B | 1259 V4L2_STD_PAL_G | V4L2_STD_PAL_H | V4L2_STD_PAL_I | 1260 V4L2_STD_PAL_D | V4L2_STD_PAL_M | V4L2_STD_PAL_N | 1261 V4L2_STD_PAL_60, 1262 .current_norm = V4L2_STD_NTSC | V4L2_STD_PAL_M | 1263 V4L2_STD_PAL_60, 1264 }; 1265 1266 int hdpvr_register_videodev(struct hdpvr_device *dev, struct device *parent, 1267 int devnum) 1268 { 1269 /* setup and register video device */ 1270 dev->video_dev = video_device_alloc(); 1271 if (!dev->video_dev) { 1272 v4l2_err(&dev->v4l2_dev, "video_device_alloc() failed\n"); 1273 goto error; 1274 } 1275 1276 *(dev->video_dev) = hdpvr_video_template; 1277 strcpy(dev->video_dev->name, "Hauppauge HD PVR"); 1278 dev->video_dev->parent = parent; 1279 video_set_drvdata(dev->video_dev, dev); 1280 1281 if (video_register_device(dev->video_dev, VFL_TYPE_GRABBER, devnum)) { 1282 v4l2_err(&dev->v4l2_dev, "video_device registration failed\n"); 1283 goto error; 1284 } 1285 1286 return 0; 1287 error: 1288 return -ENOMEM; 1289 } 1290