1 /*
2  * Fushicai USBTV007 Audio-Video Grabber Driver
3  *
4  * Product web site:
5  * http://www.fushicai.com/products_detail/&productId=d05449ee-b690-42f9-a661-aa7353894bed.html
6  *
7  * Following LWN articles were very useful in construction of this driver:
8  * Video4Linux2 API series: http://lwn.net/Articles/203924/
9  * videobuf2 API explanation: http://lwn.net/Articles/447435/
10  * Thanks go to Jonathan Corbet for providing this quality documentation.
11  * He is awesome.
12  *
13  * Copyright (c) 2013 Lubomir Rintel
14  * All rights reserved.
15  * No physical hardware was harmed running Windows during the
16  * reverse-engineering activity
17  *
18  * Redistribution and use in source and binary forms, with or without
19  * modification, are permitted provided that the following conditions
20  * are met:
21  * 1. Redistributions of source code must retain the above copyright
22  *    notice, this list of conditions, and the following disclaimer,
23  *    without modification.
24  * 2. The name of the author may not be used to endorse or promote products
25  *    derived from this software without specific prior written permission.
26  *
27  * Alternatively, this software may be distributed under the terms of the
28  * GNU General Public License ("GPL").
29  */
30 
31 #include <media/v4l2-ioctl.h>
32 #include <media/videobuf2-v4l2.h>
33 
34 #include "usbtv.h"
35 
36 static struct usbtv_norm_params norm_params[] = {
37 	{
38 		.norm = V4L2_STD_525_60,
39 		.cap_width = 720,
40 		.cap_height = 480,
41 	},
42 	{
43 		.norm = V4L2_STD_PAL,
44 		.cap_width = 720,
45 		.cap_height = 576,
46 	}
47 };
48 
49 static int usbtv_configure_for_norm(struct usbtv *usbtv, v4l2_std_id norm)
50 {
51 	int i, ret = 0;
52 	struct usbtv_norm_params *params = NULL;
53 
54 	for (i = 0; i < ARRAY_SIZE(norm_params); i++) {
55 		if (norm_params[i].norm & norm) {
56 			params = &norm_params[i];
57 			break;
58 		}
59 	}
60 
61 	if (params) {
62 		usbtv->width = params->cap_width;
63 		usbtv->height = params->cap_height;
64 		usbtv->n_chunks = usbtv->width * usbtv->height
65 						/ 4 / USBTV_CHUNK;
66 		usbtv->norm = params->norm;
67 	} else
68 		ret = -EINVAL;
69 
70 	return ret;
71 }
72 
73 static int usbtv_select_input(struct usbtv *usbtv, int input)
74 {
75 	int ret;
76 
77 	static const u16 composite[][2] = {
78 		{ USBTV_BASE + 0x0105, 0x0060 },
79 		{ USBTV_BASE + 0x011f, 0x00f2 },
80 		{ USBTV_BASE + 0x0127, 0x0060 },
81 		{ USBTV_BASE + 0x00ae, 0x0010 },
82 		{ USBTV_BASE + 0x0239, 0x0060 },
83 	};
84 
85 	static const u16 svideo[][2] = {
86 		{ USBTV_BASE + 0x0105, 0x0010 },
87 		{ USBTV_BASE + 0x011f, 0x00ff },
88 		{ USBTV_BASE + 0x0127, 0x0060 },
89 		{ USBTV_BASE + 0x00ae, 0x0030 },
90 		{ USBTV_BASE + 0x0239, 0x0060 },
91 	};
92 
93 	switch (input) {
94 	case USBTV_COMPOSITE_INPUT:
95 		ret = usbtv_set_regs(usbtv, composite, ARRAY_SIZE(composite));
96 		break;
97 	case USBTV_SVIDEO_INPUT:
98 		ret = usbtv_set_regs(usbtv, svideo, ARRAY_SIZE(svideo));
99 		break;
100 	default:
101 		ret = -EINVAL;
102 	}
103 
104 	if (!ret)
105 		usbtv->input = input;
106 
107 	return ret;
108 }
109 
110 static int usbtv_select_norm(struct usbtv *usbtv, v4l2_std_id norm)
111 {
112 	int ret;
113 	static const u16 pal[][2] = {
114 		{ USBTV_BASE + 0x001a, 0x0068 },
115 		{ USBTV_BASE + 0x010e, 0x0072 },
116 		{ USBTV_BASE + 0x010f, 0x00a2 },
117 		{ USBTV_BASE + 0x0112, 0x00b0 },
118 		{ USBTV_BASE + 0x0117, 0x0001 },
119 		{ USBTV_BASE + 0x0118, 0x002c },
120 		{ USBTV_BASE + 0x012d, 0x0010 },
121 		{ USBTV_BASE + 0x012f, 0x0020 },
122 		{ USBTV_BASE + 0x024f, 0x0002 },
123 		{ USBTV_BASE + 0x0254, 0x0059 },
124 		{ USBTV_BASE + 0x025a, 0x0016 },
125 		{ USBTV_BASE + 0x025b, 0x0035 },
126 		{ USBTV_BASE + 0x0263, 0x0017 },
127 		{ USBTV_BASE + 0x0266, 0x0016 },
128 		{ USBTV_BASE + 0x0267, 0x0036 }
129 	};
130 
131 	static const u16 ntsc[][2] = {
132 		{ USBTV_BASE + 0x001a, 0x0079 },
133 		{ USBTV_BASE + 0x010e, 0x0068 },
134 		{ USBTV_BASE + 0x010f, 0x009c },
135 		{ USBTV_BASE + 0x0112, 0x00f0 },
136 		{ USBTV_BASE + 0x0117, 0x0000 },
137 		{ USBTV_BASE + 0x0118, 0x00fc },
138 		{ USBTV_BASE + 0x012d, 0x0004 },
139 		{ USBTV_BASE + 0x012f, 0x0008 },
140 		{ USBTV_BASE + 0x024f, 0x0001 },
141 		{ USBTV_BASE + 0x0254, 0x005f },
142 		{ USBTV_BASE + 0x025a, 0x0012 },
143 		{ USBTV_BASE + 0x025b, 0x0001 },
144 		{ USBTV_BASE + 0x0263, 0x001c },
145 		{ USBTV_BASE + 0x0266, 0x0011 },
146 		{ USBTV_BASE + 0x0267, 0x0005 }
147 	};
148 
149 	ret = usbtv_configure_for_norm(usbtv, norm);
150 
151 	if (!ret) {
152 		if (norm & V4L2_STD_525_60)
153 			ret = usbtv_set_regs(usbtv, ntsc, ARRAY_SIZE(ntsc));
154 		else if (norm & V4L2_STD_PAL)
155 			ret = usbtv_set_regs(usbtv, pal, ARRAY_SIZE(pal));
156 	}
157 
158 	return ret;
159 }
160 
161 static int usbtv_setup_capture(struct usbtv *usbtv)
162 {
163 	int ret;
164 	static const u16 setup[][2] = {
165 		/* These seem to enable the device. */
166 		{ USBTV_BASE + 0x0008, 0x0001 },
167 		{ USBTV_BASE + 0x01d0, 0x00ff },
168 		{ USBTV_BASE + 0x01d9, 0x0002 },
169 
170 		/* These seem to influence color parameters, such as
171 		 * brightness, etc. */
172 		{ USBTV_BASE + 0x0239, 0x0040 },
173 		{ USBTV_BASE + 0x0240, 0x0000 },
174 		{ USBTV_BASE + 0x0241, 0x0000 },
175 		{ USBTV_BASE + 0x0242, 0x0002 },
176 		{ USBTV_BASE + 0x0243, 0x0080 },
177 		{ USBTV_BASE + 0x0244, 0x0012 },
178 		{ USBTV_BASE + 0x0245, 0x0090 },
179 		{ USBTV_BASE + 0x0246, 0x0000 },
180 
181 		{ USBTV_BASE + 0x0278, 0x002d },
182 		{ USBTV_BASE + 0x0279, 0x000a },
183 		{ USBTV_BASE + 0x027a, 0x0032 },
184 		{ 0xf890, 0x000c },
185 		{ 0xf894, 0x0086 },
186 
187 		{ USBTV_BASE + 0x00ac, 0x00c0 },
188 		{ USBTV_BASE + 0x00ad, 0x0000 },
189 		{ USBTV_BASE + 0x00a2, 0x0012 },
190 		{ USBTV_BASE + 0x00a3, 0x00e0 },
191 		{ USBTV_BASE + 0x00a4, 0x0028 },
192 		{ USBTV_BASE + 0x00a5, 0x0082 },
193 		{ USBTV_BASE + 0x00a7, 0x0080 },
194 		{ USBTV_BASE + 0x0000, 0x0014 },
195 		{ USBTV_BASE + 0x0006, 0x0003 },
196 		{ USBTV_BASE + 0x0090, 0x0099 },
197 		{ USBTV_BASE + 0x0091, 0x0090 },
198 		{ USBTV_BASE + 0x0094, 0x0068 },
199 		{ USBTV_BASE + 0x0095, 0x0070 },
200 		{ USBTV_BASE + 0x009c, 0x0030 },
201 		{ USBTV_BASE + 0x009d, 0x00c0 },
202 		{ USBTV_BASE + 0x009e, 0x00e0 },
203 		{ USBTV_BASE + 0x0019, 0x0006 },
204 		{ USBTV_BASE + 0x008c, 0x00ba },
205 		{ USBTV_BASE + 0x0101, 0x00ff },
206 		{ USBTV_BASE + 0x010c, 0x00b3 },
207 		{ USBTV_BASE + 0x01b2, 0x0080 },
208 		{ USBTV_BASE + 0x01b4, 0x00a0 },
209 		{ USBTV_BASE + 0x014c, 0x00ff },
210 		{ USBTV_BASE + 0x014d, 0x00ca },
211 		{ USBTV_BASE + 0x0113, 0x0053 },
212 		{ USBTV_BASE + 0x0119, 0x008a },
213 		{ USBTV_BASE + 0x013c, 0x0003 },
214 		{ USBTV_BASE + 0x0150, 0x009c },
215 		{ USBTV_BASE + 0x0151, 0x0071 },
216 		{ USBTV_BASE + 0x0152, 0x00c6 },
217 		{ USBTV_BASE + 0x0153, 0x0084 },
218 		{ USBTV_BASE + 0x0154, 0x00bc },
219 		{ USBTV_BASE + 0x0155, 0x00a0 },
220 		{ USBTV_BASE + 0x0156, 0x00a0 },
221 		{ USBTV_BASE + 0x0157, 0x009c },
222 		{ USBTV_BASE + 0x0158, 0x001f },
223 		{ USBTV_BASE + 0x0159, 0x0006 },
224 		{ USBTV_BASE + 0x015d, 0x0000 },
225 
226 		{ USBTV_BASE + 0x0003, 0x0004 },
227 		{ USBTV_BASE + 0x0100, 0x00d3 },
228 		{ USBTV_BASE + 0x0115, 0x0015 },
229 		{ USBTV_BASE + 0x0220, 0x002e },
230 		{ USBTV_BASE + 0x0225, 0x0008 },
231 		{ USBTV_BASE + 0x024e, 0x0002 },
232 		{ USBTV_BASE + 0x024e, 0x0002 },
233 		{ USBTV_BASE + 0x024f, 0x0002 },
234 	};
235 
236 	ret = usbtv_set_regs(usbtv, setup, ARRAY_SIZE(setup));
237 	if (ret)
238 		return ret;
239 
240 	ret = usbtv_select_norm(usbtv, usbtv->norm);
241 	if (ret)
242 		return ret;
243 
244 	ret = usbtv_select_input(usbtv, usbtv->input);
245 	if (ret)
246 		return ret;
247 
248 	return 0;
249 }
250 
251 /* Copy data from chunk into a frame buffer, deinterlacing the data
252  * into every second line. Unfortunately, they don't align nicely into
253  * 720 pixel lines, as the chunk is 240 words long, which is 480 pixels.
254  * Therefore, we break down the chunk into two halves before copyting,
255  * so that we can interleave a line if needed. */
256 static void usbtv_chunk_to_vbuf(u32 *frame, __be32 *src, int chunk_no, int odd)
257 {
258 	int half;
259 
260 	for (half = 0; half < 2; half++) {
261 		int part_no = chunk_no * 2 + half;
262 		int line = part_no / 3;
263 		int part_index = (line * 2 + !odd) * 3 + (part_no % 3);
264 
265 		u32 *dst = &frame[part_index * USBTV_CHUNK/2];
266 
267 		memcpy(dst, src, USBTV_CHUNK/2 * sizeof(*src));
268 		src += USBTV_CHUNK/2;
269 	}
270 }
271 
272 /* Called for each 256-byte image chunk.
273  * First word identifies the chunk, followed by 240 words of image
274  * data and padding. */
275 static void usbtv_image_chunk(struct usbtv *usbtv, __be32 *chunk)
276 {
277 	int frame_id, odd, chunk_no;
278 	u32 *frame;
279 	struct usbtv_buf *buf;
280 	unsigned long flags;
281 
282 	/* Ignore corrupted lines. */
283 	if (!USBTV_MAGIC_OK(chunk))
284 		return;
285 	frame_id = USBTV_FRAME_ID(chunk);
286 	odd = USBTV_ODD(chunk);
287 	chunk_no = USBTV_CHUNK_NO(chunk);
288 	if (chunk_no >= usbtv->n_chunks)
289 		return;
290 
291 	/* Beginning of a frame. */
292 	if (chunk_no == 0) {
293 		usbtv->frame_id = frame_id;
294 		usbtv->chunks_done = 0;
295 	}
296 
297 	if (usbtv->frame_id != frame_id)
298 		return;
299 
300 	spin_lock_irqsave(&usbtv->buflock, flags);
301 	if (list_empty(&usbtv->bufs)) {
302 		/* No free buffers. Userspace likely too slow. */
303 		spin_unlock_irqrestore(&usbtv->buflock, flags);
304 		return;
305 	}
306 
307 	/* First available buffer. */
308 	buf = list_first_entry(&usbtv->bufs, struct usbtv_buf, list);
309 	frame = vb2_plane_vaddr(&buf->vb.vb2_buf, 0);
310 
311 	/* Copy the chunk data. */
312 	usbtv_chunk_to_vbuf(frame, &chunk[1], chunk_no, odd);
313 	usbtv->chunks_done++;
314 
315 	/* Last chunk in a frame, signalling an end */
316 	if (odd && chunk_no == usbtv->n_chunks-1) {
317 		int size = vb2_plane_size(&buf->vb.vb2_buf, 0);
318 		enum vb2_buffer_state state = usbtv->chunks_done ==
319 						usbtv->n_chunks ?
320 						VB2_BUF_STATE_DONE :
321 						VB2_BUF_STATE_ERROR;
322 
323 		buf->vb.field = V4L2_FIELD_INTERLACED;
324 		buf->vb.sequence = usbtv->sequence++;
325 		v4l2_get_timestamp(&buf->vb.timestamp);
326 		vb2_set_plane_payload(&buf->vb.vb2_buf, 0, size);
327 		vb2_buffer_done(&buf->vb.vb2_buf, state);
328 		list_del(&buf->list);
329 	}
330 
331 	spin_unlock_irqrestore(&usbtv->buflock, flags);
332 }
333 
334 /* Got image data. Each packet contains a number of 256-word chunks we
335  * compose the image from. */
336 static void usbtv_iso_cb(struct urb *ip)
337 {
338 	int ret;
339 	int i;
340 	struct usbtv *usbtv = (struct usbtv *)ip->context;
341 
342 	switch (ip->status) {
343 	/* All fine. */
344 	case 0:
345 		break;
346 	/* Device disconnected or capture stopped? */
347 	case -ENODEV:
348 	case -ENOENT:
349 	case -ECONNRESET:
350 	case -ESHUTDOWN:
351 		return;
352 	/* Unknown error. Retry. */
353 	default:
354 		dev_warn(usbtv->dev, "Bad response for ISO request.\n");
355 		goto resubmit;
356 	}
357 
358 	for (i = 0; i < ip->number_of_packets; i++) {
359 		int size = ip->iso_frame_desc[i].actual_length;
360 		unsigned char *data = ip->transfer_buffer +
361 				ip->iso_frame_desc[i].offset;
362 		int offset;
363 
364 		for (offset = 0; USBTV_CHUNK_SIZE * offset < size; offset++)
365 			usbtv_image_chunk(usbtv,
366 				(__be32 *)&data[USBTV_CHUNK_SIZE * offset]);
367 	}
368 
369 resubmit:
370 	ret = usb_submit_urb(ip, GFP_ATOMIC);
371 	if (ret < 0)
372 		dev_warn(usbtv->dev, "Could not resubmit ISO URB\n");
373 }
374 
375 static struct urb *usbtv_setup_iso_transfer(struct usbtv *usbtv)
376 {
377 	struct urb *ip;
378 	int size = usbtv->iso_size;
379 	int i;
380 
381 	ip = usb_alloc_urb(USBTV_ISOC_PACKETS, GFP_KERNEL);
382 	if (ip == NULL)
383 		return NULL;
384 
385 	ip->dev = usbtv->udev;
386 	ip->context = usbtv;
387 	ip->pipe = usb_rcvisocpipe(usbtv->udev, USBTV_VIDEO_ENDP);
388 	ip->interval = 1;
389 	ip->transfer_flags = URB_ISO_ASAP;
390 	ip->transfer_buffer = kzalloc(size * USBTV_ISOC_PACKETS,
391 						GFP_KERNEL);
392 	ip->complete = usbtv_iso_cb;
393 	ip->number_of_packets = USBTV_ISOC_PACKETS;
394 	ip->transfer_buffer_length = size * USBTV_ISOC_PACKETS;
395 	for (i = 0; i < USBTV_ISOC_PACKETS; i++) {
396 		ip->iso_frame_desc[i].offset = size * i;
397 		ip->iso_frame_desc[i].length = size;
398 	}
399 
400 	return ip;
401 }
402 
403 static void usbtv_stop(struct usbtv *usbtv)
404 {
405 	int i;
406 	unsigned long flags;
407 
408 	/* Cancel running transfers. */
409 	for (i = 0; i < USBTV_ISOC_TRANSFERS; i++) {
410 		struct urb *ip = usbtv->isoc_urbs[i];
411 
412 		if (ip == NULL)
413 			continue;
414 		usb_kill_urb(ip);
415 		kfree(ip->transfer_buffer);
416 		usb_free_urb(ip);
417 		usbtv->isoc_urbs[i] = NULL;
418 	}
419 
420 	/* Return buffers to userspace. */
421 	spin_lock_irqsave(&usbtv->buflock, flags);
422 	while (!list_empty(&usbtv->bufs)) {
423 		struct usbtv_buf *buf = list_first_entry(&usbtv->bufs,
424 						struct usbtv_buf, list);
425 		vb2_buffer_done(&buf->vb.vb2_buf, VB2_BUF_STATE_ERROR);
426 		list_del(&buf->list);
427 	}
428 	spin_unlock_irqrestore(&usbtv->buflock, flags);
429 }
430 
431 static int usbtv_start(struct usbtv *usbtv)
432 {
433 	int i;
434 	int ret;
435 
436 	usbtv_audio_suspend(usbtv);
437 
438 	ret = usb_set_interface(usbtv->udev, 0, 0);
439 	if (ret < 0)
440 		return ret;
441 
442 	ret = usbtv_setup_capture(usbtv);
443 	if (ret < 0)
444 		return ret;
445 
446 	ret = usb_set_interface(usbtv->udev, 0, 1);
447 	if (ret < 0)
448 		return ret;
449 
450 	usbtv_audio_resume(usbtv);
451 
452 	for (i = 0; i < USBTV_ISOC_TRANSFERS; i++) {
453 		struct urb *ip;
454 
455 		ip = usbtv_setup_iso_transfer(usbtv);
456 		if (ip == NULL) {
457 			ret = -ENOMEM;
458 			goto start_fail;
459 		}
460 		usbtv->isoc_urbs[i] = ip;
461 
462 		ret = usb_submit_urb(ip, GFP_KERNEL);
463 		if (ret < 0)
464 			goto start_fail;
465 	}
466 
467 	return 0;
468 
469 start_fail:
470 	usbtv_stop(usbtv);
471 	return ret;
472 }
473 
474 static int usbtv_querycap(struct file *file, void *priv,
475 				struct v4l2_capability *cap)
476 {
477 	struct usbtv *dev = video_drvdata(file);
478 
479 	strlcpy(cap->driver, "usbtv", sizeof(cap->driver));
480 	strlcpy(cap->card, "usbtv", sizeof(cap->card));
481 	usb_make_path(dev->udev, cap->bus_info, sizeof(cap->bus_info));
482 	cap->device_caps = V4L2_CAP_VIDEO_CAPTURE;
483 	cap->device_caps |= V4L2_CAP_READWRITE | V4L2_CAP_STREAMING;
484 	cap->capabilities = cap->device_caps | V4L2_CAP_DEVICE_CAPS;
485 	return 0;
486 }
487 
488 static int usbtv_enum_input(struct file *file, void *priv,
489 					struct v4l2_input *i)
490 {
491 	struct usbtv *dev = video_drvdata(file);
492 
493 	switch (i->index) {
494 	case USBTV_COMPOSITE_INPUT:
495 		strlcpy(i->name, "Composite", sizeof(i->name));
496 		break;
497 	case USBTV_SVIDEO_INPUT:
498 		strlcpy(i->name, "S-Video", sizeof(i->name));
499 		break;
500 	default:
501 		return -EINVAL;
502 	}
503 
504 	i->type = V4L2_INPUT_TYPE_CAMERA;
505 	i->std = dev->vdev.tvnorms;
506 	return 0;
507 }
508 
509 static int usbtv_enum_fmt_vid_cap(struct file *file, void  *priv,
510 					struct v4l2_fmtdesc *f)
511 {
512 	if (f->index > 0)
513 		return -EINVAL;
514 
515 	strlcpy(f->description, "16 bpp YUY2, 4:2:2, packed",
516 					sizeof(f->description));
517 	f->pixelformat = V4L2_PIX_FMT_YUYV;
518 	return 0;
519 }
520 
521 static int usbtv_fmt_vid_cap(struct file *file, void *priv,
522 					struct v4l2_format *f)
523 {
524 	struct usbtv *usbtv = video_drvdata(file);
525 
526 	f->fmt.pix.width = usbtv->width;
527 	f->fmt.pix.height = usbtv->height;
528 	f->fmt.pix.pixelformat = V4L2_PIX_FMT_YUYV;
529 	f->fmt.pix.field = V4L2_FIELD_INTERLACED;
530 	f->fmt.pix.bytesperline = usbtv->width * 2;
531 	f->fmt.pix.sizeimage = (f->fmt.pix.bytesperline * f->fmt.pix.height);
532 	f->fmt.pix.colorspace = V4L2_COLORSPACE_SMPTE170M;
533 
534 	return 0;
535 }
536 
537 static int usbtv_g_std(struct file *file, void *priv, v4l2_std_id *norm)
538 {
539 	struct usbtv *usbtv = video_drvdata(file);
540 	*norm = usbtv->norm;
541 	return 0;
542 }
543 
544 static int usbtv_s_std(struct file *file, void *priv, v4l2_std_id norm)
545 {
546 	int ret = -EINVAL;
547 	struct usbtv *usbtv = video_drvdata(file);
548 
549 	if ((norm & V4L2_STD_525_60) || (norm & V4L2_STD_PAL))
550 		ret = usbtv_select_norm(usbtv, norm);
551 
552 	return ret;
553 }
554 
555 static int usbtv_g_input(struct file *file, void *priv, unsigned int *i)
556 {
557 	struct usbtv *usbtv = video_drvdata(file);
558 	*i = usbtv->input;
559 	return 0;
560 }
561 
562 static int usbtv_s_input(struct file *file, void *priv, unsigned int i)
563 {
564 	struct usbtv *usbtv = video_drvdata(file);
565 
566 	return usbtv_select_input(usbtv, i);
567 }
568 
569 static struct v4l2_ioctl_ops usbtv_ioctl_ops = {
570 	.vidioc_querycap = usbtv_querycap,
571 	.vidioc_enum_input = usbtv_enum_input,
572 	.vidioc_enum_fmt_vid_cap = usbtv_enum_fmt_vid_cap,
573 	.vidioc_g_fmt_vid_cap = usbtv_fmt_vid_cap,
574 	.vidioc_try_fmt_vid_cap = usbtv_fmt_vid_cap,
575 	.vidioc_s_fmt_vid_cap = usbtv_fmt_vid_cap,
576 	.vidioc_g_std = usbtv_g_std,
577 	.vidioc_s_std = usbtv_s_std,
578 	.vidioc_g_input = usbtv_g_input,
579 	.vidioc_s_input = usbtv_s_input,
580 
581 	.vidioc_reqbufs = vb2_ioctl_reqbufs,
582 	.vidioc_prepare_buf = vb2_ioctl_prepare_buf,
583 	.vidioc_querybuf = vb2_ioctl_querybuf,
584 	.vidioc_create_bufs = vb2_ioctl_create_bufs,
585 	.vidioc_qbuf = vb2_ioctl_qbuf,
586 	.vidioc_dqbuf = vb2_ioctl_dqbuf,
587 	.vidioc_streamon = vb2_ioctl_streamon,
588 	.vidioc_streamoff = vb2_ioctl_streamoff,
589 };
590 
591 static struct v4l2_file_operations usbtv_fops = {
592 	.owner = THIS_MODULE,
593 	.unlocked_ioctl = video_ioctl2,
594 	.mmap = vb2_fop_mmap,
595 	.open = v4l2_fh_open,
596 	.release = vb2_fop_release,
597 	.read = vb2_fop_read,
598 	.poll = vb2_fop_poll,
599 };
600 
601 static int usbtv_queue_setup(struct vb2_queue *vq,
602 	const void *parg, unsigned int *nbuffers,
603 	unsigned int *nplanes, unsigned int sizes[], void *alloc_ctxs[])
604 {
605 	const struct v4l2_format *fmt = parg;
606 	struct usbtv *usbtv = vb2_get_drv_priv(vq);
607 	unsigned size = USBTV_CHUNK * usbtv->n_chunks * 2 * sizeof(u32);
608 
609 	if (vq->num_buffers + *nbuffers < 2)
610 		*nbuffers = 2 - vq->num_buffers;
611 	*nplanes = 1;
612 	if (fmt && fmt->fmt.pix.sizeimage < size)
613 		return -EINVAL;
614 	sizes[0] = fmt ? fmt->fmt.pix.sizeimage : size;
615 
616 	return 0;
617 }
618 
619 static void usbtv_buf_queue(struct vb2_buffer *vb)
620 {
621 	struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
622 	struct usbtv *usbtv = vb2_get_drv_priv(vb->vb2_queue);
623 	struct usbtv_buf *buf = container_of(vbuf, struct usbtv_buf, vb);
624 	unsigned long flags;
625 
626 	if (usbtv->udev == NULL) {
627 		vb2_buffer_done(vb, VB2_BUF_STATE_ERROR);
628 		return;
629 	}
630 
631 	spin_lock_irqsave(&usbtv->buflock, flags);
632 	list_add_tail(&buf->list, &usbtv->bufs);
633 	spin_unlock_irqrestore(&usbtv->buflock, flags);
634 }
635 
636 static int usbtv_start_streaming(struct vb2_queue *vq, unsigned int count)
637 {
638 	struct usbtv *usbtv = vb2_get_drv_priv(vq);
639 
640 	if (usbtv->udev == NULL)
641 		return -ENODEV;
642 
643 	usbtv->sequence = 0;
644 	return usbtv_start(usbtv);
645 }
646 
647 static void usbtv_stop_streaming(struct vb2_queue *vq)
648 {
649 	struct usbtv *usbtv = vb2_get_drv_priv(vq);
650 
651 	if (usbtv->udev)
652 		usbtv_stop(usbtv);
653 }
654 
655 static struct vb2_ops usbtv_vb2_ops = {
656 	.queue_setup = usbtv_queue_setup,
657 	.buf_queue = usbtv_buf_queue,
658 	.start_streaming = usbtv_start_streaming,
659 	.stop_streaming = usbtv_stop_streaming,
660 };
661 
662 static void usbtv_release(struct v4l2_device *v4l2_dev)
663 {
664 	struct usbtv *usbtv = container_of(v4l2_dev, struct usbtv, v4l2_dev);
665 
666 	v4l2_device_unregister(&usbtv->v4l2_dev);
667 	vb2_queue_release(&usbtv->vb2q);
668 	kfree(usbtv);
669 }
670 
671 int usbtv_video_init(struct usbtv *usbtv)
672 {
673 	int ret;
674 
675 	(void)usbtv_configure_for_norm(usbtv, V4L2_STD_525_60);
676 
677 	spin_lock_init(&usbtv->buflock);
678 	mutex_init(&usbtv->v4l2_lock);
679 	mutex_init(&usbtv->vb2q_lock);
680 	INIT_LIST_HEAD(&usbtv->bufs);
681 
682 	/* videobuf2 structure */
683 	usbtv->vb2q.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
684 	usbtv->vb2q.io_modes = VB2_MMAP | VB2_USERPTR | VB2_READ;
685 	usbtv->vb2q.drv_priv = usbtv;
686 	usbtv->vb2q.buf_struct_size = sizeof(struct usbtv_buf);
687 	usbtv->vb2q.ops = &usbtv_vb2_ops;
688 	usbtv->vb2q.mem_ops = &vb2_vmalloc_memops;
689 	usbtv->vb2q.timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_MONOTONIC;
690 	usbtv->vb2q.lock = &usbtv->vb2q_lock;
691 	ret = vb2_queue_init(&usbtv->vb2q);
692 	if (ret < 0) {
693 		dev_warn(usbtv->dev, "Could not initialize videobuf2 queue\n");
694 		return ret;
695 	}
696 
697 	/* v4l2 structure */
698 	usbtv->v4l2_dev.release = usbtv_release;
699 	ret = v4l2_device_register(usbtv->dev, &usbtv->v4l2_dev);
700 	if (ret < 0) {
701 		dev_warn(usbtv->dev, "Could not register v4l2 device\n");
702 		goto v4l2_fail;
703 	}
704 
705 	/* Video structure */
706 	strlcpy(usbtv->vdev.name, "usbtv", sizeof(usbtv->vdev.name));
707 	usbtv->vdev.v4l2_dev = &usbtv->v4l2_dev;
708 	usbtv->vdev.release = video_device_release_empty;
709 	usbtv->vdev.fops = &usbtv_fops;
710 	usbtv->vdev.ioctl_ops = &usbtv_ioctl_ops;
711 	usbtv->vdev.tvnorms = USBTV_TV_STD;
712 	usbtv->vdev.queue = &usbtv->vb2q;
713 	usbtv->vdev.lock = &usbtv->v4l2_lock;
714 	video_set_drvdata(&usbtv->vdev, usbtv);
715 	ret = video_register_device(&usbtv->vdev, VFL_TYPE_GRABBER, -1);
716 	if (ret < 0) {
717 		dev_warn(usbtv->dev, "Could not register video device\n");
718 		goto vdev_fail;
719 	}
720 
721 	return 0;
722 
723 vdev_fail:
724 	v4l2_device_unregister(&usbtv->v4l2_dev);
725 v4l2_fail:
726 	vb2_queue_release(&usbtv->vb2q);
727 
728 	return ret;
729 }
730 
731 void usbtv_video_free(struct usbtv *usbtv)
732 {
733 	mutex_lock(&usbtv->vb2q_lock);
734 	mutex_lock(&usbtv->v4l2_lock);
735 
736 	usbtv_stop(usbtv);
737 	video_unregister_device(&usbtv->vdev);
738 	v4l2_device_disconnect(&usbtv->v4l2_dev);
739 
740 	mutex_unlock(&usbtv->v4l2_lock);
741 	mutex_unlock(&usbtv->vb2q_lock);
742 
743 	v4l2_device_put(&usbtv->v4l2_dev);
744 }
745