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