]>
Commit | Line | Data |
---|---|---|
f3d27f34 LR |
1 | /* |
2 | * Fushicai USBTV007 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 <linux/init.h> | |
32 | #include <linux/list.h> | |
33 | #include <linux/module.h> | |
34 | #include <linux/slab.h> | |
35 | #include <linux/usb.h> | |
36 | #include <linux/version.h> | |
37 | #include <linux/videodev2.h> | |
38 | ||
39 | #include <media/v4l2-device.h> | |
40 | #include <media/v4l2-ioctl.h> | |
41 | #include <media/videobuf2-core.h> | |
42 | #include <media/videobuf2-vmalloc.h> | |
43 | ||
44 | /* Hardware. */ | |
45 | #define USBTV_VIDEO_ENDP 0x81 | |
46 | #define USBTV_BASE 0xc000 | |
47 | #define USBTV_REQUEST_REG 12 | |
48 | ||
49 | /* Number of concurrent isochronous urbs submitted. | |
50 | * Higher numbers was seen to overly saturate the USB bus. */ | |
51 | #define USBTV_ISOC_TRANSFERS 16 | |
52 | #define USBTV_ISOC_PACKETS 8 | |
53 | ||
54 | #define USBTV_WIDTH 720 | |
55 | #define USBTV_HEIGHT 480 | |
56 | ||
57 | #define USBTV_CHUNK_SIZE 256 | |
58 | #define USBTV_CHUNK 240 | |
59 | #define USBTV_CHUNKS (USBTV_WIDTH * USBTV_HEIGHT \ | |
a34cacab | 60 | / 4 / USBTV_CHUNK) |
f3d27f34 LR |
61 | |
62 | /* Chunk header. */ | |
63 | #define USBTV_MAGIC_OK(chunk) ((be32_to_cpu(chunk[0]) & 0xff000000) \ | |
64 | == 0x88000000) | |
65 | #define USBTV_FRAME_ID(chunk) ((be32_to_cpu(chunk[0]) & 0x00ff0000) >> 16) | |
66 | #define USBTV_ODD(chunk) ((be32_to_cpu(chunk[0]) & 0x0000f000) >> 15) | |
67 | #define USBTV_CHUNK_NO(chunk) (be32_to_cpu(chunk[0]) & 0x00000fff) | |
68 | ||
69 | /* A single videobuf2 frame buffer. */ | |
70 | struct usbtv_buf { | |
71 | struct vb2_buffer vb; | |
72 | struct list_head list; | |
73 | }; | |
74 | ||
75 | /* Per-device structure. */ | |
76 | struct usbtv { | |
77 | struct device *dev; | |
78 | struct usb_device *udev; | |
79 | struct v4l2_device v4l2_dev; | |
80 | struct video_device vdev; | |
81 | struct vb2_queue vb2q; | |
82 | struct mutex v4l2_lock; | |
83 | struct mutex vb2q_lock; | |
84 | ||
85 | /* List of videobuf2 buffers protected by a lock. */ | |
86 | spinlock_t buflock; | |
87 | struct list_head bufs; | |
88 | ||
89 | /* Number of currently processed frame, useful find | |
90 | * out when a new one begins. */ | |
91 | u32 frame_id; | |
f52dc448 | 92 | int chunks_done; |
f3d27f34 LR |
93 | |
94 | int iso_size; | |
95 | unsigned int sequence; | |
96 | struct urb *isoc_urbs[USBTV_ISOC_TRANSFERS]; | |
97 | }; | |
98 | ||
99 | static int usbtv_setup_capture(struct usbtv *usbtv) | |
100 | { | |
101 | int ret; | |
102 | int pipe = usb_rcvctrlpipe(usbtv->udev, 0); | |
103 | int i; | |
104 | static const u16 protoregs[][2] = { | |
105 | /* These seem to enable the device. */ | |
106 | { USBTV_BASE + 0x0008, 0x0001 }, | |
107 | { USBTV_BASE + 0x01d0, 0x00ff }, | |
108 | { USBTV_BASE + 0x01d9, 0x0002 }, | |
109 | ||
110 | /* These seem to influence color parameters, such as | |
111 | * brightness, etc. */ | |
112 | { USBTV_BASE + 0x0239, 0x0040 }, | |
113 | { USBTV_BASE + 0x0240, 0x0000 }, | |
114 | { USBTV_BASE + 0x0241, 0x0000 }, | |
115 | { USBTV_BASE + 0x0242, 0x0002 }, | |
116 | { USBTV_BASE + 0x0243, 0x0080 }, | |
117 | { USBTV_BASE + 0x0244, 0x0012 }, | |
118 | { USBTV_BASE + 0x0245, 0x0090 }, | |
119 | { USBTV_BASE + 0x0246, 0x0000 }, | |
120 | ||
121 | { USBTV_BASE + 0x0278, 0x002d }, | |
122 | { USBTV_BASE + 0x0279, 0x000a }, | |
123 | { USBTV_BASE + 0x027a, 0x0032 }, | |
124 | { 0xf890, 0x000c }, | |
125 | { 0xf894, 0x0086 }, | |
126 | ||
127 | { USBTV_BASE + 0x00ac, 0x00c0 }, | |
128 | { USBTV_BASE + 0x00ad, 0x0000 }, | |
129 | { USBTV_BASE + 0x00a2, 0x0012 }, | |
130 | { USBTV_BASE + 0x00a3, 0x00e0 }, | |
131 | { USBTV_BASE + 0x00a4, 0x0028 }, | |
132 | { USBTV_BASE + 0x00a5, 0x0082 }, | |
133 | { USBTV_BASE + 0x00a7, 0x0080 }, | |
134 | { USBTV_BASE + 0x0000, 0x0014 }, | |
135 | { USBTV_BASE + 0x0006, 0x0003 }, | |
136 | { USBTV_BASE + 0x0090, 0x0099 }, | |
137 | { USBTV_BASE + 0x0091, 0x0090 }, | |
138 | { USBTV_BASE + 0x0094, 0x0068 }, | |
139 | { USBTV_BASE + 0x0095, 0x0070 }, | |
140 | { USBTV_BASE + 0x009c, 0x0030 }, | |
141 | { USBTV_BASE + 0x009d, 0x00c0 }, | |
142 | { USBTV_BASE + 0x009e, 0x00e0 }, | |
143 | { USBTV_BASE + 0x0019, 0x0006 }, | |
144 | { USBTV_BASE + 0x008c, 0x00ba }, | |
145 | { USBTV_BASE + 0x0101, 0x00ff }, | |
146 | { USBTV_BASE + 0x010c, 0x00b3 }, | |
147 | { USBTV_BASE + 0x01b2, 0x0080 }, | |
148 | { USBTV_BASE + 0x01b4, 0x00a0 }, | |
149 | { USBTV_BASE + 0x014c, 0x00ff }, | |
150 | { USBTV_BASE + 0x014d, 0x00ca }, | |
151 | { USBTV_BASE + 0x0113, 0x0053 }, | |
152 | { USBTV_BASE + 0x0119, 0x008a }, | |
153 | { USBTV_BASE + 0x013c, 0x0003 }, | |
154 | { USBTV_BASE + 0x0150, 0x009c }, | |
155 | { USBTV_BASE + 0x0151, 0x0071 }, | |
156 | { USBTV_BASE + 0x0152, 0x00c6 }, | |
157 | { USBTV_BASE + 0x0153, 0x0084 }, | |
158 | { USBTV_BASE + 0x0154, 0x00bc }, | |
159 | { USBTV_BASE + 0x0155, 0x00a0 }, | |
160 | { USBTV_BASE + 0x0156, 0x00a0 }, | |
161 | { USBTV_BASE + 0x0157, 0x009c }, | |
162 | { USBTV_BASE + 0x0158, 0x001f }, | |
163 | { USBTV_BASE + 0x0159, 0x0006 }, | |
164 | { USBTV_BASE + 0x015d, 0x0000 }, | |
165 | ||
166 | { USBTV_BASE + 0x0284, 0x0088 }, | |
167 | { USBTV_BASE + 0x0003, 0x0004 }, | |
168 | { USBTV_BASE + 0x001a, 0x0079 }, | |
169 | { USBTV_BASE + 0x0100, 0x00d3 }, | |
170 | { USBTV_BASE + 0x010e, 0x0068 }, | |
171 | { USBTV_BASE + 0x010f, 0x009c }, | |
172 | { USBTV_BASE + 0x0112, 0x00f0 }, | |
173 | { USBTV_BASE + 0x0115, 0x0015 }, | |
174 | { USBTV_BASE + 0x0117, 0x0000 }, | |
175 | { USBTV_BASE + 0x0118, 0x00fc }, | |
176 | { USBTV_BASE + 0x012d, 0x0004 }, | |
177 | { USBTV_BASE + 0x012f, 0x0008 }, | |
178 | { USBTV_BASE + 0x0220, 0x002e }, | |
179 | { USBTV_BASE + 0x0225, 0x0008 }, | |
180 | { USBTV_BASE + 0x024e, 0x0002 }, | |
181 | { USBTV_BASE + 0x024f, 0x0001 }, | |
182 | { USBTV_BASE + 0x0254, 0x005f }, | |
183 | { USBTV_BASE + 0x025a, 0x0012 }, | |
184 | { USBTV_BASE + 0x025b, 0x0001 }, | |
185 | { USBTV_BASE + 0x0263, 0x001c }, | |
186 | { USBTV_BASE + 0x0266, 0x0011 }, | |
187 | { USBTV_BASE + 0x0267, 0x0005 }, | |
188 | { USBTV_BASE + 0x024e, 0x0002 }, | |
189 | { USBTV_BASE + 0x024f, 0x0002 }, | |
190 | }; | |
191 | ||
192 | for (i = 0; i < ARRAY_SIZE(protoregs); i++) { | |
193 | u16 index = protoregs[i][0]; | |
194 | u16 value = protoregs[i][1]; | |
195 | ||
196 | ret = usb_control_msg(usbtv->udev, pipe, USBTV_REQUEST_REG, | |
197 | USB_DIR_OUT | USB_TYPE_VENDOR | USB_RECIP_DEVICE, | |
198 | value, index, NULL, 0, 0); | |
199 | if (ret < 0) | |
200 | return ret; | |
201 | } | |
202 | ||
203 | return 0; | |
204 | } | |
205 | ||
a34cacab LR |
206 | /* Copy data from chunk into a frame buffer, deinterlacing the data |
207 | * into every second line. Unfortunately, they don't align nicely into | |
208 | * 720 pixel lines, as the chunk is 240 words long, which is 480 pixels. | |
209 | * Therefore, we break down the chunk into two halves before copyting, | |
210 | * so that we can interleave a line if needed. */ | |
211 | static void usbtv_chunk_to_vbuf(u32 *frame, u32 *src, int chunk_no, int odd) | |
212 | { | |
213 | int half; | |
214 | ||
215 | for (half = 0; half < 2; half++) { | |
216 | int part_no = chunk_no * 2 + half; | |
217 | int line = part_no / 3; | |
218 | int part_index = (line * 2 + !odd) * 3 + (part_no % 3); | |
219 | ||
220 | u32 *dst = &frame[part_index * USBTV_CHUNK/2]; | |
221 | memcpy(dst, src, USBTV_CHUNK/2 * sizeof(*src)); | |
222 | src += USBTV_CHUNK/2; | |
223 | } | |
224 | } | |
225 | ||
f3d27f34 LR |
226 | /* Called for each 256-byte image chunk. |
227 | * First word identifies the chunk, followed by 240 words of image | |
228 | * data and padding. */ | |
229 | static void usbtv_image_chunk(struct usbtv *usbtv, u32 *chunk) | |
230 | { | |
231 | int frame_id, odd, chunk_no; | |
232 | u32 *frame; | |
233 | struct usbtv_buf *buf; | |
234 | unsigned long flags; | |
235 | ||
236 | /* Ignore corrupted lines. */ | |
237 | if (!USBTV_MAGIC_OK(chunk)) | |
238 | return; | |
239 | frame_id = USBTV_FRAME_ID(chunk); | |
240 | odd = USBTV_ODD(chunk); | |
241 | chunk_no = USBTV_CHUNK_NO(chunk); | |
f3d27f34 LR |
242 | if (chunk_no >= USBTV_CHUNKS) |
243 | return; | |
244 | ||
245 | /* Beginning of a frame. */ | |
f52dc448 | 246 | if (chunk_no == 0) { |
f3d27f34 | 247 | usbtv->frame_id = frame_id; |
f52dc448 LR |
248 | usbtv->chunks_done = 0; |
249 | } | |
250 | ||
251 | if (usbtv->frame_id != frame_id) | |
252 | return; | |
f3d27f34 LR |
253 | |
254 | spin_lock_irqsave(&usbtv->buflock, flags); | |
255 | if (list_empty(&usbtv->bufs)) { | |
256 | /* No free buffers. Userspace likely too slow. */ | |
257 | spin_unlock_irqrestore(&usbtv->buflock, flags); | |
258 | return; | |
259 | } | |
260 | ||
261 | /* First available buffer. */ | |
262 | buf = list_first_entry(&usbtv->bufs, struct usbtv_buf, list); | |
263 | frame = vb2_plane_vaddr(&buf->vb, 0); | |
264 | ||
a34cacab LR |
265 | /* Copy the chunk data. */ |
266 | usbtv_chunk_to_vbuf(frame, &chunk[1], chunk_no, odd); | |
f52dc448 | 267 | usbtv->chunks_done++; |
f3d27f34 LR |
268 | |
269 | /* Last chunk in a frame, signalling an end */ | |
a34cacab | 270 | if (odd && chunk_no == USBTV_CHUNKS-1) { |
f3d27f34 | 271 | int size = vb2_plane_size(&buf->vb, 0); |
f52dc448 LR |
272 | enum vb2_buffer_state state = usbtv->chunks_done == |
273 | USBTV_CHUNKS ? | |
274 | VB2_BUF_STATE_DONE : | |
275 | VB2_BUF_STATE_ERROR; | |
f3d27f34 LR |
276 | |
277 | buf->vb.v4l2_buf.field = V4L2_FIELD_INTERLACED; | |
278 | buf->vb.v4l2_buf.sequence = usbtv->sequence++; | |
279 | v4l2_get_timestamp(&buf->vb.v4l2_buf.timestamp); | |
280 | vb2_set_plane_payload(&buf->vb, 0, size); | |
f52dc448 | 281 | vb2_buffer_done(&buf->vb, state); |
f3d27f34 LR |
282 | list_del(&buf->list); |
283 | } | |
284 | ||
285 | spin_unlock_irqrestore(&usbtv->buflock, flags); | |
286 | } | |
287 | ||
288 | /* Got image data. Each packet contains a number of 256-word chunks we | |
289 | * compose the image from. */ | |
290 | static void usbtv_iso_cb(struct urb *ip) | |
291 | { | |
292 | int ret; | |
293 | int i; | |
294 | struct usbtv *usbtv = (struct usbtv *)ip->context; | |
295 | ||
296 | switch (ip->status) { | |
297 | /* All fine. */ | |
298 | case 0: | |
299 | break; | |
300 | /* Device disconnected or capture stopped? */ | |
301 | case -ENODEV: | |
302 | case -ENOENT: | |
303 | case -ECONNRESET: | |
304 | case -ESHUTDOWN: | |
305 | return; | |
306 | /* Unknown error. Retry. */ | |
307 | default: | |
308 | dev_warn(usbtv->dev, "Bad response for ISO request.\n"); | |
309 | goto resubmit; | |
310 | } | |
311 | ||
312 | for (i = 0; i < ip->number_of_packets; i++) { | |
313 | int size = ip->iso_frame_desc[i].actual_length; | |
314 | unsigned char *data = ip->transfer_buffer + | |
315 | ip->iso_frame_desc[i].offset; | |
316 | int offset; | |
317 | ||
318 | for (offset = 0; USBTV_CHUNK_SIZE * offset < size; offset++) | |
319 | usbtv_image_chunk(usbtv, | |
320 | (u32 *)&data[USBTV_CHUNK_SIZE * offset]); | |
321 | } | |
322 | ||
323 | resubmit: | |
324 | ret = usb_submit_urb(ip, GFP_ATOMIC); | |
325 | if (ret < 0) | |
326 | dev_warn(usbtv->dev, "Could not resubmit ISO URB\n"); | |
327 | } | |
328 | ||
329 | static struct urb *usbtv_setup_iso_transfer(struct usbtv *usbtv) | |
330 | { | |
331 | struct urb *ip; | |
332 | int size = usbtv->iso_size; | |
333 | int i; | |
334 | ||
335 | ip = usb_alloc_urb(USBTV_ISOC_PACKETS, GFP_KERNEL); | |
336 | if (ip == NULL) | |
337 | return NULL; | |
338 | ||
339 | ip->dev = usbtv->udev; | |
340 | ip->context = usbtv; | |
341 | ip->pipe = usb_rcvisocpipe(usbtv->udev, USBTV_VIDEO_ENDP); | |
342 | ip->interval = 1; | |
343 | ip->transfer_flags = URB_ISO_ASAP; | |
344 | ip->transfer_buffer = kzalloc(size * USBTV_ISOC_PACKETS, | |
345 | GFP_KERNEL); | |
346 | ip->complete = usbtv_iso_cb; | |
347 | ip->number_of_packets = USBTV_ISOC_PACKETS; | |
348 | ip->transfer_buffer_length = size * USBTV_ISOC_PACKETS; | |
349 | for (i = 0; i < USBTV_ISOC_PACKETS; i++) { | |
350 | ip->iso_frame_desc[i].offset = size * i; | |
351 | ip->iso_frame_desc[i].length = size; | |
352 | } | |
353 | ||
354 | return ip; | |
355 | } | |
356 | ||
357 | static void usbtv_stop(struct usbtv *usbtv) | |
358 | { | |
359 | int i; | |
360 | unsigned long flags; | |
361 | ||
362 | /* Cancel running transfers. */ | |
363 | for (i = 0; i < USBTV_ISOC_TRANSFERS; i++) { | |
364 | struct urb *ip = usbtv->isoc_urbs[i]; | |
365 | if (ip == NULL) | |
366 | continue; | |
367 | usb_kill_urb(ip); | |
368 | kfree(ip->transfer_buffer); | |
369 | usb_free_urb(ip); | |
370 | usbtv->isoc_urbs[i] = NULL; | |
371 | } | |
372 | ||
373 | /* Return buffers to userspace. */ | |
374 | spin_lock_irqsave(&usbtv->buflock, flags); | |
375 | while (!list_empty(&usbtv->bufs)) { | |
376 | struct usbtv_buf *buf = list_first_entry(&usbtv->bufs, | |
377 | struct usbtv_buf, list); | |
378 | vb2_buffer_done(&buf->vb, VB2_BUF_STATE_ERROR); | |
379 | list_del(&buf->list); | |
380 | } | |
381 | spin_unlock_irqrestore(&usbtv->buflock, flags); | |
382 | } | |
383 | ||
384 | static int usbtv_start(struct usbtv *usbtv) | |
385 | { | |
386 | int i; | |
387 | int ret; | |
388 | ||
389 | ret = usb_set_interface(usbtv->udev, 0, 0); | |
390 | if (ret < 0) | |
391 | return ret; | |
392 | ||
393 | ret = usbtv_setup_capture(usbtv); | |
394 | if (ret < 0) | |
395 | return ret; | |
396 | ||
397 | ret = usb_set_interface(usbtv->udev, 0, 1); | |
398 | if (ret < 0) | |
399 | return ret; | |
400 | ||
401 | for (i = 0; i < USBTV_ISOC_TRANSFERS; i++) { | |
402 | struct urb *ip; | |
403 | ||
404 | ip = usbtv_setup_iso_transfer(usbtv); | |
405 | if (ip == NULL) { | |
406 | ret = -ENOMEM; | |
407 | goto start_fail; | |
408 | } | |
409 | usbtv->isoc_urbs[i] = ip; | |
410 | ||
411 | ret = usb_submit_urb(ip, GFP_KERNEL); | |
412 | if (ret < 0) | |
413 | goto start_fail; | |
414 | } | |
415 | ||
416 | return 0; | |
417 | ||
418 | start_fail: | |
419 | usbtv_stop(usbtv); | |
420 | return ret; | |
421 | } | |
422 | ||
423 | struct usb_device_id usbtv_id_table[] = { | |
424 | { USB_DEVICE(0x1b71, 0x3002) }, | |
425 | {} | |
426 | }; | |
427 | MODULE_DEVICE_TABLE(usb, usbtv_id_table); | |
428 | ||
429 | static int usbtv_querycap(struct file *file, void *priv, | |
430 | struct v4l2_capability *cap) | |
431 | { | |
432 | struct usbtv *dev = video_drvdata(file); | |
433 | ||
434 | strlcpy(cap->driver, "usbtv", sizeof(cap->driver)); | |
435 | strlcpy(cap->card, "usbtv", sizeof(cap->card)); | |
436 | usb_make_path(dev->udev, cap->bus_info, sizeof(cap->bus_info)); | |
437 | cap->device_caps = V4L2_CAP_VIDEO_CAPTURE; | |
438 | cap->device_caps |= V4L2_CAP_READWRITE | V4L2_CAP_STREAMING; | |
439 | cap->capabilities = cap->device_caps | V4L2_CAP_DEVICE_CAPS; | |
440 | return 0; | |
441 | } | |
442 | ||
443 | static int usbtv_enum_input(struct file *file, void *priv, | |
444 | struct v4l2_input *i) | |
445 | { | |
446 | if (i->index > 0) | |
447 | return -EINVAL; | |
448 | ||
449 | strlcpy(i->name, "Composite", sizeof(i->name)); | |
450 | i->type = V4L2_INPUT_TYPE_CAMERA; | |
451 | i->std = V4L2_STD_525_60; | |
452 | return 0; | |
453 | } | |
454 | ||
455 | static int usbtv_enum_fmt_vid_cap(struct file *file, void *priv, | |
456 | struct v4l2_fmtdesc *f) | |
457 | { | |
458 | if (f->index > 0) | |
459 | return -EINVAL; | |
460 | ||
461 | strlcpy(f->description, "16 bpp YUY2, 4:2:2, packed", | |
462 | sizeof(f->description)); | |
463 | f->pixelformat = V4L2_PIX_FMT_YUYV; | |
464 | return 0; | |
465 | } | |
466 | ||
467 | static int usbtv_fmt_vid_cap(struct file *file, void *priv, | |
468 | struct v4l2_format *f) | |
469 | { | |
470 | f->fmt.pix.width = USBTV_WIDTH; | |
471 | f->fmt.pix.height = USBTV_HEIGHT; | |
472 | f->fmt.pix.pixelformat = V4L2_PIX_FMT_YUYV; | |
473 | f->fmt.pix.field = V4L2_FIELD_INTERLACED; | |
474 | f->fmt.pix.bytesperline = USBTV_WIDTH * 2; | |
475 | f->fmt.pix.sizeimage = (f->fmt.pix.bytesperline * f->fmt.pix.height); | |
476 | f->fmt.pix.colorspace = V4L2_COLORSPACE_SMPTE170M; | |
477 | f->fmt.pix.priv = 0; | |
478 | return 0; | |
479 | } | |
480 | ||
481 | static int usbtv_g_std(struct file *file, void *priv, v4l2_std_id *norm) | |
482 | { | |
483 | *norm = V4L2_STD_525_60; | |
484 | return 0; | |
485 | } | |
486 | ||
487 | static int usbtv_g_input(struct file *file, void *priv, unsigned int *i) | |
488 | { | |
489 | *i = 0; | |
490 | return 0; | |
491 | } | |
492 | ||
493 | static int usbtv_s_input(struct file *file, void *priv, unsigned int i) | |
494 | { | |
495 | if (i > 0) | |
496 | return -EINVAL; | |
497 | return 0; | |
498 | } | |
499 | ||
500 | static int usbtv_s_std(struct file *file, void *priv, v4l2_std_id norm) | |
501 | { | |
502 | if (norm & V4L2_STD_525_60) | |
503 | return 0; | |
504 | return -EINVAL; | |
505 | } | |
506 | ||
507 | struct v4l2_ioctl_ops usbtv_ioctl_ops = { | |
508 | .vidioc_querycap = usbtv_querycap, | |
509 | .vidioc_enum_input = usbtv_enum_input, | |
510 | .vidioc_enum_fmt_vid_cap = usbtv_enum_fmt_vid_cap, | |
511 | .vidioc_g_fmt_vid_cap = usbtv_fmt_vid_cap, | |
512 | .vidioc_try_fmt_vid_cap = usbtv_fmt_vid_cap, | |
513 | .vidioc_s_fmt_vid_cap = usbtv_fmt_vid_cap, | |
514 | .vidioc_g_std = usbtv_g_std, | |
515 | .vidioc_s_std = usbtv_s_std, | |
516 | .vidioc_g_input = usbtv_g_input, | |
517 | .vidioc_s_input = usbtv_s_input, | |
518 | ||
519 | .vidioc_reqbufs = vb2_ioctl_reqbufs, | |
520 | .vidioc_prepare_buf = vb2_ioctl_prepare_buf, | |
521 | .vidioc_querybuf = vb2_ioctl_querybuf, | |
522 | .vidioc_create_bufs = vb2_ioctl_create_bufs, | |
523 | .vidioc_qbuf = vb2_ioctl_qbuf, | |
524 | .vidioc_dqbuf = vb2_ioctl_dqbuf, | |
525 | .vidioc_streamon = vb2_ioctl_streamon, | |
526 | .vidioc_streamoff = vb2_ioctl_streamoff, | |
527 | }; | |
528 | ||
529 | struct v4l2_file_operations usbtv_fops = { | |
530 | .owner = THIS_MODULE, | |
531 | .unlocked_ioctl = video_ioctl2, | |
532 | .mmap = vb2_fop_mmap, | |
533 | .open = v4l2_fh_open, | |
534 | .release = vb2_fop_release, | |
535 | .read = vb2_fop_read, | |
536 | .poll = vb2_fop_poll, | |
537 | }; | |
538 | ||
539 | static int usbtv_queue_setup(struct vb2_queue *vq, | |
540 | const struct v4l2_format *v4l_fmt, unsigned int *nbuffers, | |
541 | unsigned int *nplanes, unsigned int sizes[], void *alloc_ctxs[]) | |
542 | { | |
543 | if (*nbuffers < 2) | |
544 | *nbuffers = 2; | |
545 | *nplanes = 1; | |
a34cacab | 546 | sizes[0] = USBTV_WIDTH * USBTV_HEIGHT / 2 * sizeof(u32); |
f3d27f34 LR |
547 | |
548 | return 0; | |
549 | } | |
550 | ||
551 | static void usbtv_buf_queue(struct vb2_buffer *vb) | |
552 | { | |
553 | struct usbtv *usbtv = vb2_get_drv_priv(vb->vb2_queue); | |
554 | struct usbtv_buf *buf = container_of(vb, struct usbtv_buf, vb); | |
555 | unsigned long flags; | |
556 | ||
557 | if (usbtv->udev == NULL) { | |
558 | vb2_buffer_done(vb, VB2_BUF_STATE_ERROR); | |
559 | return; | |
560 | } | |
561 | ||
562 | spin_lock_irqsave(&usbtv->buflock, flags); | |
563 | list_add_tail(&buf->list, &usbtv->bufs); | |
564 | spin_unlock_irqrestore(&usbtv->buflock, flags); | |
565 | } | |
566 | ||
567 | static int usbtv_start_streaming(struct vb2_queue *vq, unsigned int count) | |
568 | { | |
569 | struct usbtv *usbtv = vb2_get_drv_priv(vq); | |
570 | ||
571 | if (usbtv->udev == NULL) | |
572 | return -ENODEV; | |
573 | ||
574 | return usbtv_start(usbtv); | |
575 | } | |
576 | ||
577 | static int usbtv_stop_streaming(struct vb2_queue *vq) | |
578 | { | |
579 | struct usbtv *usbtv = vb2_get_drv_priv(vq); | |
580 | ||
581 | if (usbtv->udev == NULL) | |
582 | return -ENODEV; | |
583 | ||
584 | usbtv_stop(usbtv); | |
585 | return 0; | |
586 | } | |
587 | ||
588 | struct vb2_ops usbtv_vb2_ops = { | |
589 | .queue_setup = usbtv_queue_setup, | |
590 | .buf_queue = usbtv_buf_queue, | |
591 | .start_streaming = usbtv_start_streaming, | |
592 | .stop_streaming = usbtv_stop_streaming, | |
593 | }; | |
594 | ||
595 | static void usbtv_release(struct v4l2_device *v4l2_dev) | |
596 | { | |
597 | struct usbtv *usbtv = container_of(v4l2_dev, struct usbtv, v4l2_dev); | |
598 | ||
599 | v4l2_device_unregister(&usbtv->v4l2_dev); | |
600 | vb2_queue_release(&usbtv->vb2q); | |
601 | kfree(usbtv); | |
602 | } | |
603 | ||
604 | static int usbtv_probe(struct usb_interface *intf, | |
605 | const struct usb_device_id *id) | |
606 | { | |
607 | int ret; | |
608 | int size; | |
609 | struct device *dev = &intf->dev; | |
610 | struct usbtv *usbtv; | |
611 | ||
612 | /* Checks that the device is what we think it is. */ | |
613 | if (intf->num_altsetting != 2) | |
614 | return -ENODEV; | |
615 | if (intf->altsetting[1].desc.bNumEndpoints != 4) | |
616 | return -ENODEV; | |
617 | ||
618 | /* Packet size is split into 11 bits of base size and count of | |
619 | * extra multiplies of it.*/ | |
620 | size = usb_endpoint_maxp(&intf->altsetting[1].endpoint[0].desc); | |
621 | size = (size & 0x07ff) * (((size & 0x1800) >> 11) + 1); | |
622 | ||
623 | /* Device structure */ | |
624 | usbtv = kzalloc(sizeof(struct usbtv), GFP_KERNEL); | |
625 | if (usbtv == NULL) | |
626 | return -ENOMEM; | |
627 | usbtv->dev = dev; | |
628 | usbtv->udev = usb_get_dev(interface_to_usbdev(intf)); | |
629 | usbtv->iso_size = size; | |
630 | spin_lock_init(&usbtv->buflock); | |
631 | mutex_init(&usbtv->v4l2_lock); | |
632 | mutex_init(&usbtv->vb2q_lock); | |
633 | INIT_LIST_HEAD(&usbtv->bufs); | |
634 | ||
635 | /* videobuf2 structure */ | |
636 | usbtv->vb2q.type = V4L2_BUF_TYPE_VIDEO_CAPTURE; | |
637 | usbtv->vb2q.io_modes = VB2_MMAP | VB2_USERPTR | VB2_READ; | |
638 | usbtv->vb2q.drv_priv = usbtv; | |
639 | usbtv->vb2q.buf_struct_size = sizeof(struct usbtv_buf); | |
640 | usbtv->vb2q.ops = &usbtv_vb2_ops; | |
641 | usbtv->vb2q.mem_ops = &vb2_vmalloc_memops; | |
642 | usbtv->vb2q.timestamp_type = V4L2_BUF_FLAG_TIMESTAMP_MONOTONIC; | |
643 | usbtv->vb2q.lock = &usbtv->vb2q_lock; | |
644 | ret = vb2_queue_init(&usbtv->vb2q); | |
645 | if (ret < 0) { | |
646 | dev_warn(dev, "Could not initialize videobuf2 queue\n"); | |
647 | goto usbtv_fail; | |
648 | } | |
649 | ||
650 | /* v4l2 structure */ | |
651 | usbtv->v4l2_dev.release = usbtv_release; | |
652 | ret = v4l2_device_register(dev, &usbtv->v4l2_dev); | |
653 | if (ret < 0) { | |
654 | dev_warn(dev, "Could not register v4l2 device\n"); | |
655 | goto v4l2_fail; | |
656 | } | |
657 | ||
658 | usb_set_intfdata(intf, usbtv); | |
659 | ||
660 | /* Video structure */ | |
661 | strlcpy(usbtv->vdev.name, "usbtv", sizeof(usbtv->vdev.name)); | |
662 | usbtv->vdev.v4l2_dev = &usbtv->v4l2_dev; | |
663 | usbtv->vdev.release = video_device_release_empty; | |
664 | usbtv->vdev.fops = &usbtv_fops; | |
665 | usbtv->vdev.ioctl_ops = &usbtv_ioctl_ops; | |
666 | usbtv->vdev.tvnorms = V4L2_STD_525_60; | |
667 | usbtv->vdev.queue = &usbtv->vb2q; | |
668 | usbtv->vdev.lock = &usbtv->v4l2_lock; | |
669 | set_bit(V4L2_FL_USE_FH_PRIO, &usbtv->vdev.flags); | |
670 | video_set_drvdata(&usbtv->vdev, usbtv); | |
671 | ret = video_register_device(&usbtv->vdev, VFL_TYPE_GRABBER, -1); | |
672 | if (ret < 0) { | |
673 | dev_warn(dev, "Could not register video device\n"); | |
674 | goto vdev_fail; | |
675 | } | |
676 | ||
677 | dev_info(dev, "Fushicai USBTV007 Video Grabber\n"); | |
678 | return 0; | |
679 | ||
680 | vdev_fail: | |
681 | v4l2_device_unregister(&usbtv->v4l2_dev); | |
682 | v4l2_fail: | |
683 | vb2_queue_release(&usbtv->vb2q); | |
684 | usbtv_fail: | |
685 | kfree(usbtv); | |
686 | ||
687 | return ret; | |
688 | } | |
689 | ||
690 | static void usbtv_disconnect(struct usb_interface *intf) | |
691 | { | |
692 | struct usbtv *usbtv = usb_get_intfdata(intf); | |
693 | ||
694 | mutex_lock(&usbtv->vb2q_lock); | |
695 | mutex_lock(&usbtv->v4l2_lock); | |
696 | ||
697 | usbtv_stop(usbtv); | |
698 | usb_set_intfdata(intf, NULL); | |
699 | video_unregister_device(&usbtv->vdev); | |
700 | v4l2_device_disconnect(&usbtv->v4l2_dev); | |
701 | usb_put_dev(usbtv->udev); | |
702 | usbtv->udev = NULL; | |
703 | ||
704 | mutex_unlock(&usbtv->v4l2_lock); | |
705 | mutex_unlock(&usbtv->vb2q_lock); | |
706 | ||
707 | v4l2_device_put(&usbtv->v4l2_dev); | |
708 | } | |
709 | ||
710 | MODULE_AUTHOR("Lubomir Rintel"); | |
711 | MODULE_DESCRIPTION("Fushicai USBTV007 Video Grabber Driver"); | |
712 | MODULE_LICENSE("Dual BSD/GPL"); | |
713 | ||
714 | struct usb_driver usbtv_usb_driver = { | |
715 | .name = "usbtv", | |
716 | .id_table = usbtv_id_table, | |
717 | .probe = usbtv_probe, | |
718 | .disconnect = usbtv_disconnect, | |
719 | }; | |
720 | ||
721 | module_usb_driver(usbtv_usb_driver); |