1/*
2 *  STV0680 USB Camera Driver, by Kevin Sisson (kjsisson@bellsouth.net)
3 *
4 * Thanks to STMicroelectronics for information on the usb commands, and
5 * to Steve Miller at STM for his help and encouragement while I was
6 * writing this driver.
7 *
8 * This driver is based heavily on the
9 * Endpoints (formerly known as AOX) se401 USB Camera Driver
10 * Copyright (c) 2000 Jeroen B. Vreeken (pe1rxq@amsat.org)
11 *
12 * Still somewhat based on the Linux ov511 driver.
13 *
14 * This program is free software; you can redistribute it and/or modify it
15 * under the terms of the GNU General Public License as published by the
16 * Free Software Foundation; either version 2 of the License, or (at your
17 * option) any later version.
18 *
19 * This program is distributed in the hope that it will be useful, but
20 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
21 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
22 * for more details.
23 *
24 * You should have received a copy of the GNU General Public License
25 * along with this program; if not, write to the Free Software Foundation,
26 * Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
27 *
28 * History:
29 * ver 0.1 October, 2001. Initial attempt.
30 *
31 * ver 0.2 November, 2001. Fixed asbility to resize, added brightness
32 *                         function, made more stable (?)
33 *
34 * ver 0.21 Nov, 2001.     Added gamma correction and white balance,
35 *                         due to Alexander Schwartz. Still trying to
36 *                         improve stablility. Moved stuff into stv680.h
37 *
38 * ver 0.22 Nov, 2001.	   Added sharpen function (by Michael Sweet,
39 *                         mike@easysw.com) from GIMP, also used in pencam.
40 *                         Simple, fast, good integer math routine.
41 *
42 * ver 0.23 Dec, 2001 (gkh)
43 * 			   Took out sharpen function, ran code through
44 * 			   Lindent, and did other minor tweaks to get
45 * 			   things to work properly with 2.5.1
46 *
47 * ver 0.24 Jan, 2002 (kjs)
48 *                         Fixed the problem with webcam crashing after
49 *                         two pictures. Changed the way pic is halved to
50 *                         improve quality. Got rid of green line around
51 *                         frame. Fix brightness reset when changing size
52 *                         bug. Adjusted gamma filters slightly.
53 *
54 * ver 0.25 Jan, 2002 (kjs)
55 *			   Fixed a bug in which the driver sometimes attempted
56 *			   to set to a non-supported size. This allowed
57 *			   gnomemeeting to work.
58 *			   Fixed proc entry removal bug.
59 */
60
61#include <linux/module.h>
62#include <linux/init.h>
63#include <linux/vmalloc.h>
64#include <linux/slab.h>
65#include <linux/pagemap.h>
66#include <linux/errno.h>
67#include <linux/videodev.h>
68#include <media/v4l2-common.h>
69#include <linux/usb.h>
70#include <linux/mutex.h>
71
72#include "stv680.h"
73
74static int video_nr = -1;
75static int swapRGB = 0;   /* default for auto sleect */
76static int swapRGB_on = 0; /* default to allow auto select; -1=swap never, +1= swap always */
77
78static unsigned int debug = 0;
79
80#define PDEBUG(level, fmt, args...) \
81	do { \
82	if (debug >= level)	\
83		info("[%s:%d] " fmt, __FUNCTION__, __LINE__ , ## args);	\
84	} while (0)
85
86
87/*
88 * Version Information
89 */
90#define DRIVER_VERSION "v0.25"
91#define DRIVER_AUTHOR "Kevin Sisson <kjsisson@bellsouth.net>"
92#define DRIVER_DESC "STV0680 USB Camera Driver"
93
94MODULE_AUTHOR (DRIVER_AUTHOR);
95MODULE_DESCRIPTION (DRIVER_DESC);
96MODULE_LICENSE ("GPL");
97module_param(debug, int, S_IRUGO | S_IWUSR);
98MODULE_PARM_DESC (debug, "Debug enabled or not");
99module_param(swapRGB_on, int, 0);
100MODULE_PARM_DESC (swapRGB_on, "Red/blue swap: 1=always, 0=auto, -1=never");
101module_param(video_nr, int, 0);
102
103/********************************************************************
104 *
105 * Memory management
106 *
107 * This is a shameless copy from the USB-cpia driver (linux kernel
108 * version 2.3.29 or so, I have no idea what this code actually does ;).
109 * Actually it seems to be a copy of a shameless copy of the bttv-driver.
110 * Or that is a copy of a shameless copy of ... (To the powers: is there
111 * no generic kernel-function to do this sort of stuff?)
112 *
113 * Yes, it was a shameless copy from the bttv-driver. IIRC, Alan says
114 * there will be one, but apparentely not yet -jerdfelt
115 *
116 * So I copied it again for the ov511 driver -claudio
117 *
118 * Same for the se401 driver -Jeroen
119 *
120 * And the STV0680 driver - Kevin
121 ********************************************************************/
122static void *rvmalloc (unsigned long size)
123{
124	void *mem;
125	unsigned long adr;
126
127	size = PAGE_ALIGN(size);
128	mem = vmalloc_32 (size);
129	if (!mem)
130		return NULL;
131
132	memset (mem, 0, size);	/* Clear the ram out, no junk to the user */
133	adr = (unsigned long) mem;
134	while (size > 0) {
135		SetPageReserved(vmalloc_to_page((void *)adr));
136		adr += PAGE_SIZE;
137		size -= PAGE_SIZE;
138	}
139	return mem;
140}
141
142static void rvfree (void *mem, unsigned long size)
143{
144	unsigned long adr;
145
146	if (!mem)
147		return;
148
149	adr = (unsigned long) mem;
150	while ((long) size > 0) {
151		ClearPageReserved(vmalloc_to_page((void *)adr));
152		adr += PAGE_SIZE;
153		size -= PAGE_SIZE;
154	}
155	vfree (mem);
156}
157
158
159/*********************************************************************
160 * pencam read/write functions
161 ********************************************************************/
162
163static int stv_sndctrl (int set, struct usb_stv *stv680, unsigned short req, unsigned short value, unsigned char *buffer, int size)
164{
165	int ret = -1;
166
167	switch (set) {
168	case 0:		/*  0xc1  */
169		ret = usb_control_msg (stv680->udev,
170				       usb_rcvctrlpipe (stv680->udev, 0),
171				       req,
172				       (USB_DIR_IN | USB_TYPE_VENDOR | USB_RECIP_ENDPOINT),
173				       value, 0, buffer, size, PENCAM_TIMEOUT);
174		break;
175
176	case 1:		/*  0x41  */
177		ret = usb_control_msg (stv680->udev,
178				       usb_sndctrlpipe (stv680->udev, 0),
179				       req,
180				       (USB_DIR_OUT | USB_TYPE_VENDOR | USB_RECIP_ENDPOINT),
181				       value, 0, buffer, size, PENCAM_TIMEOUT);
182		break;
183
184	case 2:		/*  0x80  */
185		ret = usb_control_msg (stv680->udev,
186				       usb_rcvctrlpipe (stv680->udev, 0),
187				       req,
188				       (USB_DIR_IN | USB_RECIP_DEVICE),
189				       value, 0, buffer, size, PENCAM_TIMEOUT);
190		break;
191
192	case 3:		/*  0x40  */
193		ret = usb_control_msg (stv680->udev,
194				       usb_sndctrlpipe (stv680->udev, 0),
195				       req,
196				       (USB_DIR_OUT | USB_TYPE_VENDOR | USB_RECIP_DEVICE),
197				       value, 0, buffer, size, PENCAM_TIMEOUT);
198		break;
199
200	}
201	if ((ret < 0) && (req != 0x0a)) {
202		PDEBUG (1, "STV(e): usb_control_msg error %i, request = 0x%x, error = %i", set, req, ret);
203	}
204	return ret;
205}
206
207static int stv_set_config (struct usb_stv *dev, int configuration, int interface, int alternate)
208{
209
210	if (configuration != dev->udev->actconfig->desc.bConfigurationValue
211			|| usb_reset_configuration (dev->udev) < 0) {
212		PDEBUG (1, "STV(e): FAILED to reset configuration %i", configuration);
213		return -1;
214	}
215	if (usb_set_interface (dev->udev, interface, alternate) < 0) {
216		PDEBUG (1, "STV(e): FAILED to set alternate interface %i", alternate);
217		return -1;
218	}
219	return 0;
220}
221
222static int stv_stop_video (struct usb_stv *dev)
223{
224	int i;
225	unsigned char *buf;
226
227	buf = kmalloc (40, GFP_KERNEL);
228	if (buf == NULL) {
229		PDEBUG (0, "STV(e): Out of (small buf) memory");
230		return -1;
231	}
232
233	/* this is a high priority command; it stops all lower order commands */
234	if ((i = stv_sndctrl (1, dev, 0x04, 0x0000, buf, 0x0)) < 0) {
235		i = stv_sndctrl (0, dev, 0x80, 0, buf, 0x02);	/* Get Last Error; 2 = busy */
236		PDEBUG (1, "STV(i): last error: %i,  command = 0x%x", buf[0], buf[1]);
237	} else {
238		PDEBUG (1, "STV(i): Camera reset to idle mode.");
239	}
240
241	if ((i = stv_set_config (dev, 1, 0, 0)) < 0)
242		PDEBUG (1, "STV(e): Reset config during exit failed");
243
244	/*  get current mode  */
245	buf[0] = 0xf0;
246	if ((i = stv_sndctrl (0, dev, 0x87, 0, buf, 0x08)) != 0x08)	/* get mode */
247		PDEBUG (0, "STV(e): Stop_video: problem setting original mode");
248	if (dev->origMode != buf[0]) {
249		memset (buf, 0, 8);
250		buf[0] = (unsigned char) dev->origMode;
251		if ((i = stv_sndctrl (3, dev, 0x07, 0x0100, buf, 0x08)) != 0x08) {
252			PDEBUG (0, "STV(e): Stop_video: Set_Camera_Mode failed");
253			i = -1;
254		}
255		buf[0] = 0xf0;
256		i = stv_sndctrl (0, dev, 0x87, 0, buf, 0x08);
257		if ((i != 0x08) || (buf[0] != dev->origMode)) {
258			PDEBUG (0, "STV(e): camera NOT set to original resolution.");
259			i = -1;
260		} else
261			PDEBUG (0, "STV(i): Camera set to original resolution");
262	}
263	/* origMode */
264	kfree(buf);
265	return i;
266}
267
268static int stv_set_video_mode (struct usb_stv *dev)
269{
270	int i, stop_video = 1;
271	unsigned char *buf;
272
273	buf = kmalloc (40, GFP_KERNEL);
274	if (buf == NULL) {
275		PDEBUG (0, "STV(e): Out of (small buf) memory");
276		return -1;
277	}
278
279	if ((i = stv_set_config (dev, 1, 0, 0)) < 0) {
280		kfree(buf);
281		return i;
282	}
283
284	i = stv_sndctrl (2, dev, 0x06, 0x0100, buf, 0x12);
285	if (!(i > 0) && (buf[8] == 0x53) && (buf[9] == 0x05)) {
286		PDEBUG (1, "STV(e): Could not get descriptor 0100.");
287		goto error;
288	}
289
290	/*  set alternate interface 1 */
291	if ((i = stv_set_config (dev, 1, 0, 1)) < 0)
292		goto error;
293
294	if ((i = stv_sndctrl (0, dev, 0x85, 0, buf, 0x10)) != 0x10)
295		goto error;
296	PDEBUG (1, "STV(i): Setting video mode.");
297	/*  Switch to Video mode: 0x0100 = VGA (640x480), 0x0000 = CIF (352x288) 0x0300 = QVGA (320x240)  */
298	if ((i = stv_sndctrl (1, dev, 0x09, dev->VideoMode, buf, 0x0)) < 0) {
299		stop_video = 0;
300		goto error;
301	}
302	goto exit;
303
304error:
305	kfree(buf);
306	if (stop_video == 1)
307		stv_stop_video (dev);
308	return -1;
309
310exit:
311	kfree(buf);
312	return 0;
313}
314
315static int stv_init (struct usb_stv *stv680)
316{
317	int i = 0;
318	unsigned char *buffer;
319	unsigned long int bufsize;
320
321	buffer = kzalloc (40, GFP_KERNEL);
322	if (buffer == NULL) {
323		PDEBUG (0, "STV(e): Out of (small buf) memory");
324		return -1;
325	}
326	udelay (100);
327
328	/* set config 1, interface 0, alternate 0 */
329	if ((i = stv_set_config (stv680, 1, 0, 0)) < 0) {
330		kfree(buffer);
331		PDEBUG (0, "STV(e): set config 1,0,0 failed");
332		return -1;
333	}
334	/* ping camera to be sure STV0680 is present */
335	if ((i = stv_sndctrl (0, stv680, 0x88, 0x5678, buffer, 0x02)) != 0x02)
336		goto error;
337	if ((buffer[0] != 0x56) || (buffer[1] != 0x78)) {
338		PDEBUG (1, "STV(e): camera ping failed!!");
339		goto error;
340	}
341
342	/* get camera descriptor */
343	if ((i = stv_sndctrl (2, stv680, 0x06, 0x0200, buffer, 0x09)) != 0x09)
344		goto error;
345	i = stv_sndctrl (2, stv680, 0x06, 0x0200, buffer, 0x22);
346	if (!(i >= 0) && (buffer[7] == 0xa0) && (buffer[8] == 0x23)) {
347		PDEBUG (1, "STV(e): Could not get descriptor 0200.");
348		goto error;
349	}
350	if ((i = stv_sndctrl (0, stv680, 0x8a, 0, buffer, 0x02)) != 0x02)
351		goto error;
352	if ((i = stv_sndctrl (0, stv680, 0x8b, 0, buffer, 0x24)) != 0x24)
353		goto error;
354	if ((i = stv_sndctrl (0, stv680, 0x85, 0, buffer, 0x10)) != 0x10)
355		goto error;
356
357	stv680->SupportedModes = buffer[7];
358	i = stv680->SupportedModes;
359	stv680->CIF = 0;
360	stv680->VGA = 0;
361	stv680->QVGA = 0;
362	if (i & 1)
363		stv680->CIF = 1;
364	if (i & 2)
365		stv680->VGA = 1;
366	if (i & 8)
367		stv680->QVGA = 1;
368	if (stv680->SupportedModes == 0) {
369		PDEBUG (0, "STV(e): There are NO supported STV680 modes!!");
370		i = -1;
371		goto error;
372	} else {
373		if (stv680->CIF)
374			PDEBUG (0, "STV(i): CIF is supported");
375		if (stv680->QVGA)
376			PDEBUG (0, "STV(i): QVGA is supported");
377	}
378	/* FW rev, ASIC rev, sensor ID  */
379	PDEBUG (1, "STV(i): Firmware rev is %i.%i", buffer[0], buffer[1]);
380	PDEBUG (1, "STV(i): ASIC rev is %i.%i", buffer[2], buffer[3]);
381	PDEBUG (1, "STV(i): Sensor ID is %i", (buffer[4]*16) + (buffer[5]>>4));
382
383	/*  set alternate interface 1 */
384	if ((i = stv_set_config (stv680, 1, 0, 1)) < 0)
385		goto error;
386
387	if ((i = stv_sndctrl (0, stv680, 0x85, 0, buffer, 0x10)) != 0x10)
388		goto error;
389	if ((i = stv_sndctrl (0, stv680, 0x8d, 0, buffer, 0x08)) != 0x08)
390		goto error;
391	i = buffer[3];
392	PDEBUG (0, "STV(i): Camera has %i pictures.", i);
393
394	/*  get current mode */
395	if ((i = stv_sndctrl (0, stv680, 0x87, 0, buffer, 0x08)) != 0x08)
396		goto error;
397	stv680->origMode = buffer[0];	/* 01 = VGA, 03 = QVGA, 00 = CIF */
398
399	/* This will attemp CIF mode, if supported. If not, set to QVGA  */
400	memset (buffer, 0, 8);
401	if (stv680->CIF)
402		buffer[0] = 0x00;
403	else if (stv680->QVGA)
404		buffer[0] = 0x03;
405	if ((i = stv_sndctrl (3, stv680, 0x07, 0x0100, buffer, 0x08)) != 0x08) {
406		PDEBUG (0, "STV(i): Set_Camera_Mode failed");
407		i = -1;
408		goto error;
409	}
410	buffer[0] = 0xf0;
411	stv_sndctrl (0, stv680, 0x87, 0, buffer, 0x08);
412	if (((stv680->CIF == 1) && (buffer[0] != 0x00)) || ((stv680->QVGA == 1) && (buffer[0] != 0x03))) {
413		PDEBUG (0, "STV(e): Error setting camera video mode!");
414		i = -1;
415		goto error;
416	} else {
417		if (buffer[0] == 0) {
418			stv680->VideoMode = 0x0000;
419			PDEBUG (0, "STV(i): Video Mode set to CIF");
420		}
421		if (buffer[0] == 0x03) {
422			stv680->VideoMode = 0x0300;
423			PDEBUG (0, "STV(i): Video Mode set to QVGA");
424		}
425	}
426	if ((i = stv_sndctrl (0, stv680, 0x8f, 0, buffer, 0x10)) != 0x10)
427		goto error;
428	bufsize = (buffer[0] << 24) | (buffer[1] << 16) | (buffer[2] << 8) | (buffer[3]);
429	stv680->cwidth = (buffer[4] << 8) | (buffer[5]);	/* ->camera = 322, 356, 644  */
430	stv680->cheight = (buffer[6] << 8) | (buffer[7]);	/* ->camera = 242, 292, 484  */
431	stv680->origGain = buffer[12];
432
433	goto exit;
434
435error:
436	i = stv_sndctrl (0, stv680, 0x80, 0, buffer, 0x02);	/* Get Last Error */
437	PDEBUG (1, "STV(i): last error: %i,  command = 0x%x", buffer[0], buffer[1]);
438	kfree(buffer);
439	return -1;
440
441exit:
442	kfree(buffer);
443
444	/* video = 320x240, 352x288 */
445	if (stv680->CIF == 1) {
446		stv680->maxwidth = 352;
447		stv680->maxheight = 288;
448		stv680->vwidth = 352;
449		stv680->vheight = 288;
450	}
451	if (stv680->QVGA == 1) {
452		stv680->maxwidth = 320;
453		stv680->maxheight = 240;
454		stv680->vwidth = 320;
455		stv680->vheight = 240;
456	}
457
458	stv680->rawbufsize = bufsize;	/* must be ./. by 8 */
459	stv680->maxframesize = bufsize * 3;	/* RGB size */
460	PDEBUG (2, "STV(i): cwidth = %i, cheight = %i", stv680->cwidth, stv680->cheight);
461	PDEBUG (1, "STV(i): width = %i, height = %i, rawbufsize = %li", stv680->vwidth, stv680->vheight, stv680->rawbufsize);
462
463	/* some default values */
464	stv680->bulk_in_endpointAddr = 0x82;
465	stv680->dropped = 0;
466	stv680->error = 0;
467	stv680->framecount = 0;
468	stv680->readcount = 0;
469	stv680->streaming = 0;
470	/* bright, white, colour, hue, contrast are set by software, not in stv0680 */
471	stv680->brightness = 32767;
472	stv680->chgbright = 0;
473	stv680->whiteness = 0;	/* only for greyscale */
474	stv680->colour = 32767;
475	stv680->contrast = 32767;
476	stv680->hue = 32767;
477	stv680->palette = STV_VIDEO_PALETTE;
478	stv680->depth = 24;	/* rgb24 bits */
479	if ((swapRGB_on == 0) && (swapRGB == 0))
480		PDEBUG (1, "STV(i): swapRGB is (auto) OFF");
481	else if ((swapRGB_on == 0) && (swapRGB == 1))
482		PDEBUG (1, "STV(i): swapRGB is (auto) ON");
483	else if (swapRGB_on == 1)
484		PDEBUG (1, "STV(i): swapRGB is (forced) ON");
485	else if (swapRGB_on == -1)
486		PDEBUG (1, "STV(i): swapRGB is (forced) OFF");
487
488	if (stv_set_video_mode (stv680) < 0) {
489		PDEBUG (0, "STV(e): Could not set video mode in stv_init");
490		return -1;
491	}
492
493	return 0;
494}
495
496/***************** last of pencam  routines  *******************/
497
498/****************************************************************************
499 *  sysfs
500 ***************************************************************************/
501#define stv680_file(name, variable, field)				\
502static ssize_t show_##name(struct class_device *class_dev, char *buf)	\
503{									\
504	struct video_device *vdev = to_video_device(class_dev);		\
505	struct usb_stv *stv = video_get_drvdata(vdev);			\
506	return sprintf(buf, field, stv->variable);			\
507}									\
508static CLASS_DEVICE_ATTR(name, S_IRUGO, show_##name, NULL);
509
510stv680_file(model, camera_name, "%s\n");
511stv680_file(in_use, user, "%d\n");
512stv680_file(streaming, streaming, "%d\n");
513stv680_file(palette, palette, "%i\n");
514stv680_file(frames_total, readcount, "%d\n");
515stv680_file(frames_read, framecount, "%d\n");
516stv680_file(packets_dropped, dropped, "%d\n");
517stv680_file(decoding_errors, error, "%d\n");
518
519static int stv680_create_sysfs_files(struct video_device *vdev)
520{
521	int rc;
522
523	rc = video_device_create_file(vdev, &class_device_attr_model);
524	if (rc) goto err;
525	rc = video_device_create_file(vdev, &class_device_attr_in_use);
526	if (rc) goto err_model;
527	rc = video_device_create_file(vdev, &class_device_attr_streaming);
528	if (rc) goto err_inuse;
529	rc = video_device_create_file(vdev, &class_device_attr_palette);
530	if (rc) goto err_stream;
531	rc = video_device_create_file(vdev, &class_device_attr_frames_total);
532	if (rc) goto err_pal;
533	rc = video_device_create_file(vdev, &class_device_attr_frames_read);
534	if (rc) goto err_framtot;
535	rc = video_device_create_file(vdev, &class_device_attr_packets_dropped);
536	if (rc) goto err_framread;
537	rc = video_device_create_file(vdev, &class_device_attr_decoding_errors);
538	if (rc) goto err_dropped;
539
540	return 0;
541
542err_dropped:
543	video_device_remove_file(vdev, &class_device_attr_packets_dropped);
544err_framread:
545	video_device_remove_file(vdev, &class_device_attr_frames_read);
546err_framtot:
547	video_device_remove_file(vdev, &class_device_attr_frames_total);
548err_pal:
549	video_device_remove_file(vdev, &class_device_attr_palette);
550err_stream:
551	video_device_remove_file(vdev, &class_device_attr_streaming);
552err_inuse:
553	video_device_remove_file(vdev, &class_device_attr_in_use);
554err_model:
555	video_device_remove_file(vdev, &class_device_attr_model);
556err:
557	return rc;
558}
559
560static void stv680_remove_sysfs_files(struct video_device *vdev)
561{
562	video_device_remove_file(vdev, &class_device_attr_model);
563	video_device_remove_file(vdev, &class_device_attr_in_use);
564	video_device_remove_file(vdev, &class_device_attr_streaming);
565	video_device_remove_file(vdev, &class_device_attr_palette);
566	video_device_remove_file(vdev, &class_device_attr_frames_total);
567	video_device_remove_file(vdev, &class_device_attr_frames_read);
568	video_device_remove_file(vdev, &class_device_attr_packets_dropped);
569	video_device_remove_file(vdev, &class_device_attr_decoding_errors);
570}
571
572/********************************************************************
573 * Camera control
574 *******************************************************************/
575
576static int stv680_get_pict (struct usb_stv *stv680, struct video_picture *p)
577{
578	/* This sets values for v4l interface. max/min = 65535/0  */
579
580	p->brightness = stv680->brightness;
581	p->whiteness = stv680->whiteness;	/* greyscale */
582	p->colour = stv680->colour;
583	p->contrast = stv680->contrast;
584	p->hue = stv680->hue;
585	p->palette = stv680->palette;
586	p->depth = stv680->depth;
587	return 0;
588}
589
590static int stv680_set_pict (struct usb_stv *stv680, struct video_picture *p)
591{
592	/* See above stv680_get_pict  */
593
594	if (p->palette != STV_VIDEO_PALETTE) {
595		PDEBUG (2, "STV(e): Palette set error in _set_pic");
596		return 1;
597	}
598
599	if (stv680->brightness != p->brightness) {
600		stv680->chgbright = 1;
601		stv680->brightness = p->brightness;
602	}
603
604	stv680->whiteness = p->whiteness;	/* greyscale */
605	stv680->colour = p->colour;
606	stv680->contrast = p->contrast;
607	stv680->hue = p->hue;
608	stv680->palette = p->palette;
609	stv680->depth = p->depth;
610
611	return 0;
612}
613
614static void stv680_video_irq (struct urb *urb)
615{
616	struct usb_stv *stv680 = urb->context;
617	int length = urb->actual_length;
618
619	if (length < stv680->rawbufsize)
620		PDEBUG (2, "STV(i): Lost data in transfer: exp %li, got %i", stv680->rawbufsize, length);
621
622	/* ohoh... */
623	if (!stv680->streaming)
624		return;
625
626	if (!stv680->udev) {
627		PDEBUG (0, "STV(e): device vapourished in video_irq");
628		return;
629	}
630
631	/* 0 sized packets happen if we are to fast, but sometimes the camera
632	   keeps sending them forever...
633	 */
634	if (length && !urb->status) {
635		stv680->nullpackets = 0;
636		switch (stv680->scratch[stv680->scratch_next].state) {
637		case BUFFER_READY:
638		case BUFFER_BUSY:
639			stv680->dropped++;
640			break;
641
642		case BUFFER_UNUSED:
643			memcpy (stv680->scratch[stv680->scratch_next].data,
644				(unsigned char *) urb->transfer_buffer, length);
645			stv680->scratch[stv680->scratch_next].state = BUFFER_READY;
646			stv680->scratch[stv680->scratch_next].length = length;
647			if (waitqueue_active (&stv680->wq)) {
648				wake_up_interruptible (&stv680->wq);
649			}
650			stv680->scratch_overflow = 0;
651			stv680->scratch_next++;
652			if (stv680->scratch_next >= STV680_NUMSCRATCH)
653				stv680->scratch_next = 0;
654			break;
655		}		/* switch  */
656	} else {
657		stv680->nullpackets++;
658		if (stv680->nullpackets > STV680_MAX_NULLPACKETS) {
659			if (waitqueue_active (&stv680->wq)) {
660				wake_up_interruptible (&stv680->wq);
661			}
662		}
663	}			/*  if - else */
664
665	/* Resubmit urb for new data */
666	urb->status = 0;
667	urb->dev = stv680->udev;
668	if (usb_submit_urb (urb, GFP_ATOMIC))
669		PDEBUG (0, "STV(e): urb burned down in video irq");
670	return;
671}				/*  _video_irq  */
672
673static int stv680_start_stream (struct usb_stv *stv680)
674{
675	struct urb *urb;
676	int err = 0, i;
677
678	stv680->streaming = 1;
679
680	/* Do some memory allocation */
681	for (i = 0; i < STV680_NUMFRAMES; i++) {
682		stv680->frame[i].data = stv680->fbuf + i * stv680->maxframesize;
683		stv680->frame[i].curpix = 0;
684	}
685	/* packet size = 4096  */
686	for (i = 0; i < STV680_NUMSBUF; i++) {
687		stv680->sbuf[i].data = kmalloc (stv680->rawbufsize, GFP_KERNEL);
688		if (stv680->sbuf[i].data == NULL) {
689			PDEBUG (0, "STV(e): Could not kmalloc raw data buffer %i", i);
690			goto nomem_err;
691		}
692	}
693
694	stv680->scratch_next = 0;
695	stv680->scratch_use = 0;
696	stv680->scratch_overflow = 0;
697	for (i = 0; i < STV680_NUMSCRATCH; i++) {
698		stv680->scratch[i].data = kmalloc (stv680->rawbufsize, GFP_KERNEL);
699		if (stv680->scratch[i].data == NULL) {
700			PDEBUG (0, "STV(e): Could not kmalloc raw scratch buffer %i", i);
701			goto nomem_err;
702		}
703		stv680->scratch[i].state = BUFFER_UNUSED;
704	}
705
706	for (i = 0; i < STV680_NUMSBUF; i++) {
707		urb = usb_alloc_urb (0, GFP_KERNEL);
708		if (!urb)
709			goto nomem_err;
710
711		/* sbuf is urb->transfer_buffer, later gets memcpyed to scratch */
712		usb_fill_bulk_urb (urb, stv680->udev,
713				   usb_rcvbulkpipe (stv680->udev, stv680->bulk_in_endpointAddr),
714				   stv680->sbuf[i].data, stv680->rawbufsize,
715				   stv680_video_irq, stv680);
716		stv680->urb[i] = urb;
717		err = usb_submit_urb (stv680->urb[i], GFP_KERNEL);
718		if (err)
719			PDEBUG (0, "STV(e): urb burned down in start stream");
720	}			/* i STV680_NUMSBUF */
721
722	stv680->framecount = 0;
723	return 0;
724
725 nomem_err:
726	for (i = 0; i < STV680_NUMSCRATCH; i++) {
727		kfree(stv680->scratch[i].data);
728		stv680->scratch[i].data = NULL;
729	}
730	for (i = 0; i < STV680_NUMSBUF; i++) {
731		usb_kill_urb(stv680->urb[i]);
732		usb_free_urb(stv680->urb[i]);
733		stv680->urb[i] = NULL;
734		kfree(stv680->sbuf[i].data);
735		stv680->sbuf[i].data = NULL;
736	}
737	return -ENOMEM;
738
739}
740
741static int stv680_stop_stream (struct usb_stv *stv680)
742{
743	int i;
744
745	if (!stv680->streaming || !stv680->udev)
746		return 1;
747
748	stv680->streaming = 0;
749
750	for (i = 0; i < STV680_NUMSBUF; i++)
751		if (stv680->urb[i]) {
752			usb_kill_urb (stv680->urb[i]);
753			usb_free_urb (stv680->urb[i]);
754			stv680->urb[i] = NULL;
755			kfree(stv680->sbuf[i].data);
756		}
757	for (i = 0; i < STV680_NUMSCRATCH; i++) {
758		kfree(stv680->scratch[i].data);
759		stv680->scratch[i].data = NULL;
760	}
761
762	return 0;
763}
764
765static int stv680_set_size (struct usb_stv *stv680, int width, int height)
766{
767	int wasstreaming = stv680->streaming;
768
769	/* Check to see if we need to change */
770	if ((stv680->vwidth == width) && (stv680->vheight == height))
771		return 0;
772
773	PDEBUG (1, "STV(i): size request for %i x %i", width, height);
774	/* Check for a valid mode */
775	if ((!width || !height) || ((width & 1) || (height & 1))) {
776		PDEBUG (1, "STV(e): set_size error: request: v.width = %i, v.height = %i  actual: stv.width = %i, stv.height = %i", width, height, stv680->vwidth, stv680->vheight);
777		return 1;
778	}
779
780	if ((width < (stv680->maxwidth / 2)) || (height < (stv680->maxheight / 2))) {
781		width = stv680->maxwidth / 2;
782		height = stv680->maxheight / 2;
783	} else if ((width >= 158) && (width <= 166) && (stv680->QVGA == 1)) {
784		width = 160;
785		height = 120;
786	} else if ((width >= 172) && (width <= 180) && (stv680->CIF == 1)) {
787		width = 176;
788		height = 144;
789	} else if ((width >= 318) && (width <= 350) && (stv680->QVGA == 1)) {
790		width = 320;
791		height = 240;
792	} else if ((width >= 350) && (width <= 358) && (stv680->CIF == 1)) {
793		width = 352;
794		height = 288;
795	} else {
796		PDEBUG (1, "STV(e): request for non-supported size: request: v.width = %i, v.height = %i  actual: stv.width = %i, stv.height = %i", width, height, stv680->vwidth, stv680->vheight);
797		return 1;
798	}
799
800	/* Stop a current stream and start it again at the new size */
801	if (wasstreaming)
802		stv680_stop_stream (stv680);
803	stv680->vwidth = width;
804	stv680->vheight = height;
805	PDEBUG (1, "STV(i): size set to %i x %i", stv680->vwidth, stv680->vheight);
806	if (wasstreaming)
807		stv680_start_stream (stv680);
808
809	return 0;
810}
811
812/**********************************************************************
813 * Video Decoding
814 **********************************************************************/
815
816/*******  routines from the pencam program; hey, they work!  ********/
817
818/*
819 * STV0680 Vision Camera Chipset Driver
820 * Copyright (C) 2000 Adam Harrison <adam@antispin.org>
821*/
822
823#define RED 0
824#define GREEN 1
825#define BLUE 2
826#define AD(x, y, w) (((y)*(w)+(x))*3)
827
828static void bayer_unshuffle (struct usb_stv *stv680, struct stv680_scratch *buffer)
829{
830	int x, y, i;
831	int w = stv680->cwidth;
832	int vw = stv680->cwidth, vh = stv680->cheight;
833	unsigned int p = 0;
834	int colour = 0, bayer = 0;
835	unsigned char *raw = buffer->data;
836	struct stv680_frame *frame = &stv680->frame[stv680->curframe];
837	unsigned char *output = frame->data;
838	unsigned char *temp = frame->data;
839	int offset = buffer->offset;
840
841	if (frame->curpix == 0) {
842		if (frame->grabstate == FRAME_READY) {
843			frame->grabstate = FRAME_GRABBING;
844		}
845	}
846	if (offset != frame->curpix) {	/* Regard frame as lost :( */
847		frame->curpix = 0;
848		stv680->error++;
849		return;
850	}
851
852	if ((stv680->vwidth == 320) || (stv680->vwidth == 160)) {
853		vw = 320;
854		vh = 240;
855	}
856	if ((stv680->vwidth == 352) || (stv680->vwidth == 176)) {
857		vw = 352;
858		vh = 288;
859	}
860
861	memset (output, 0, 3 * vw * vh);	/* clear output matrix. */
862
863	for (y = 0; y < vh; y++) {
864		for (x = 0; x < vw; x++) {
865			if (x & 1)
866				p = *(raw + y * w + (x >> 1));
867			else
868				p = *(raw + y * w + (x >> 1) + (w >> 1));
869
870			if (y & 1)
871				bayer = 2;
872			else
873				bayer = 0;
874			if (x & 1)
875				bayer++;
876
877			switch (bayer) {
878			case 0:
879			case 3:
880				colour = 1;
881				break;
882			case 1:
883				colour = 0;
884				break;
885			case 2:
886				colour = 2;
887				break;
888			}
889			i = (y * vw + x) * 3;
890			*(output + i + colour) = (unsigned char) p;
891		}		/* for x */
892
893	}			/* for y */
894
895	/****** gamma correction plus hardcoded white balance */
896	/* Thanks to Alexander Schwartx <alexander.schwartx@gmx.net> for this code.
897	   Correction values red[], green[], blue[], are generated by
898	   (pow(i/256.0, GAMMA)*255.0)*white balanceRGB where GAMMA=0.55, 1<i<255.
899	   White balance (RGB)= 1.0, 1.17, 1.48. Values are calculated as double float and
900	   converted to unsigned char. Values are in stv680.h  */
901
902	for (y = 0; y < vh; y++) {
903		for (x = 0; x < vw; x++) {
904			i = (y * vw + x) * 3;
905			*(output + i) = red[*(output + i)];
906			*(output + i + 1) = green[*(output + i + 1)];
907			*(output + i + 2) = blue[*(output + i + 2)];
908		}
909	}
910
911	/******  bayer demosaic  ******/
912	for (y = 1; y < (vh - 1); y++) {
913		for (x = 1; x < (vw - 1); x++) {	/* work out pixel type */
914			if (y & 1)
915				bayer = 0;
916			else
917				bayer = 2;
918			if (!(x & 1))
919				bayer++;
920
921			switch (bayer) {
922			case 0:	/* green. blue lr, red tb */
923				*(output + AD (x, y, vw) + BLUE) = ((int) *(output + AD (x - 1, y, vw) + BLUE) + (int) *(output + AD (x + 1, y, vw) + BLUE)) >> 1;
924				*(output + AD (x, y, vw) + RED) = ((int) *(output + AD (x, y - 1, vw) + RED) + (int) *(output + AD (x, y + 1, vw) + RED)) >> 1;
925				break;
926
927			case 1:	/* blue. green lrtb, red diagonals */
928				*(output + AD (x, y, vw) + GREEN) = ((int) *(output + AD (x - 1, y, vw) + GREEN) + (int) *(output + AD (x + 1, y, vw) + GREEN) + (int) *(output + AD (x, y - 1, vw) + GREEN) + (int) *(output + AD (x, y + 1, vw) + GREEN)) >> 2;
929				*(output + AD (x, y, vw) + RED) = ((int) *(output + AD (x - 1, y - 1, vw) + RED) + (int) *(output + AD (x - 1, y + 1, vw) + RED) + (int) *(output + AD (x + 1, y - 1, vw) + RED) + (int) *(output + AD (x + 1, y + 1, vw) + RED)) >> 2;
930				break;
931
932			case 2:	/* red. green lrtb, blue diagonals */
933				*(output + AD (x, y, vw) + GREEN) = ((int) *(output + AD (x - 1, y, vw) + GREEN) + (int) *(output + AD (x + 1, y, vw) + GREEN) + (int) *(output + AD (x, y - 1, vw) + GREEN) + (int) *(output + AD (x, y + 1, vw) + GREEN)) >> 2;
934				*(output + AD (x, y, vw) + BLUE) = ((int) *(output + AD (x - 1, y - 1, vw) + BLUE) + (int) *(output + AD (x + 1, y - 1, vw) + BLUE) + (int) *(output + AD (x - 1, y + 1, vw) + BLUE) + (int) *(output + AD (x + 1, y + 1, vw) + BLUE)) >> 2;
935				break;
936
937			case 3:	/* green. red lr, blue tb */
938				*(output + AD (x, y, vw) + RED) = ((int) *(output + AD (x - 1, y, vw) + RED) + (int) *(output + AD (x + 1, y, vw) + RED)) >> 1;
939				*(output + AD (x, y, vw) + BLUE) = ((int) *(output + AD (x, y - 1, vw) + BLUE) + (int) *(output + AD (x, y + 1, vw) + BLUE)) >> 1;
940				break;
941			}	/* switch */
942		}		/* for x */
943	}			/* for y  - end demosaic  */
944
945	/* fix top and bottom row, left and right side */
946	i = vw * 3;
947	memcpy (output, (output + i), i);
948	memcpy ((output + (vh * i)), (output + ((vh - 1) * i)), i);
949	for (y = 0; y < vh; y++) {
950		i = y * vw * 3;
951		memcpy ((output + i), (output + i + 3), 3);
952		memcpy ((output + i + (vw * 3)), (output + i + (vw - 1) * 3), 3);
953	}
954
955	/*  process all raw data, then trim to size if necessary */
956	if ((stv680->vwidth == 160) || (stv680->vwidth == 176))  {
957		i = 0;
958		for (y = 0; y < vh; y++) {
959			if (!(y & 1)) {
960				for (x = 0; x < vw; x++) {
961					p = (y * vw + x) * 3;
962					if (!(x & 1)) {
963						*(output + i) = *(output + p);
964						*(output + i + 1) = *(output + p + 1);
965						*(output + i + 2) = *(output + p + 2);
966						i += 3;
967					}
968				}  /* for x */
969			}
970		}  /* for y */
971	}
972	/* reset to proper width */
973	if ((stv680->vwidth == 160)) {
974		vw = 160;
975		vh = 120;
976	}
977	if ((stv680->vwidth == 176)) {
978		vw = 176;
979		vh = 144;
980	}
981
982	/* output is RGB; some programs want BGR  */
983	/* swapRGB_on=0 -> program decides;  swapRGB_on=1, always swap */
984	/* swapRGB_on=-1, never swap */
985	if (((swapRGB == 1) && (swapRGB_on != -1)) || (swapRGB_on == 1)) {
986		for (y = 0; y < vh; y++) {
987			for (x = 0; x < vw; x++) {
988				i = (y * vw + x) * 3;
989				*(temp) = *(output + i);
990				*(output + i) = *(output + i + 2);
991				*(output + i + 2) = *(temp);
992			}
993		}
994	}
995	/* brightness */
996	if (stv680->chgbright == 1) {
997		if (stv680->brightness >= 32767) {
998			p = (stv680->brightness - 32767) / 256;
999			for (x = 0; x < (vw * vh * 3); x++) {
1000				if ((*(output + x) + (unsigned char) p) > 255)
1001					*(output + x) = 255;
1002				else
1003					*(output + x) += (unsigned char) p;
1004			}	/* for */
1005		} else {
1006			p = (32767 - stv680->brightness) / 256;
1007			for (x = 0; x < (vw * vh * 3); x++) {
1008				if ((unsigned char) p > *(output + x))
1009					*(output + x) = 0;
1010				else
1011					*(output + x) -= (unsigned char) p;
1012			}	/* for */
1013		}		/* else */
1014	}
1015	/* if */
1016	frame->curpix = 0;
1017	frame->curlinepix = 0;
1018	frame->grabstate = FRAME_DONE;
1019	stv680->framecount++;
1020	stv680->readcount++;
1021	if (stv680->frame[(stv680->curframe + 1) & (STV680_NUMFRAMES - 1)].grabstate == FRAME_READY) {
1022		stv680->curframe = (stv680->curframe + 1) & (STV680_NUMFRAMES - 1);
1023	}
1024
1025}				/* bayer_unshuffle */
1026
1027/*******  end routines from the pencam program  *********/
1028
1029static int stv680_newframe (struct usb_stv *stv680, int framenr)
1030{
1031	int errors = 0;
1032
1033	while (stv680->streaming && (stv680->frame[framenr].grabstate == FRAME_READY || stv680->frame[framenr].grabstate == FRAME_GRABBING)) {
1034		if (!stv680->frame[framenr].curpix) {
1035			errors++;
1036		}
1037		wait_event_interruptible (stv680->wq, (stv680->scratch[stv680->scratch_use].state == BUFFER_READY));
1038
1039		if (stv680->nullpackets > STV680_MAX_NULLPACKETS) {
1040			stv680->nullpackets = 0;
1041			PDEBUG (2, "STV(i): too many null length packets, restarting capture");
1042			stv680_stop_stream (stv680);
1043			stv680_start_stream (stv680);
1044		} else {
1045			if (stv680->scratch[stv680->scratch_use].state != BUFFER_READY) {
1046				stv680->frame[framenr].grabstate = FRAME_ERROR;
1047				PDEBUG (2, "STV(e): FRAME_ERROR in _newframe");
1048				return -EIO;
1049			}
1050			stv680->scratch[stv680->scratch_use].state = BUFFER_BUSY;
1051
1052			bayer_unshuffle (stv680, &stv680->scratch[stv680->scratch_use]);
1053
1054			stv680->scratch[stv680->scratch_use].state = BUFFER_UNUSED;
1055			stv680->scratch_use++;
1056			if (stv680->scratch_use >= STV680_NUMSCRATCH)
1057				stv680->scratch_use = 0;
1058			if (errors > STV680_MAX_ERRORS) {
1059				errors = 0;
1060				PDEBUG (2, "STV(i): too many errors, restarting capture");
1061				stv680_stop_stream (stv680);
1062				stv680_start_stream (stv680);
1063			}
1064		}		/* else */
1065	}			/* while */
1066	return 0;
1067}
1068
1069/*********************************************************************
1070 * Video4Linux
1071 *********************************************************************/
1072
1073static int stv_open (struct inode *inode, struct file *file)
1074{
1075	struct video_device *dev = video_devdata(file);
1076	struct usb_stv *stv680 = video_get_drvdata(dev);
1077	int err = 0;
1078
1079	/* we are called with the BKL held */
1080	stv680->user = 1;
1081	err = stv_init (stv680);	/* main initialization routine for camera */
1082
1083	if (err >= 0) {
1084		stv680->fbuf = rvmalloc (stv680->maxframesize * STV680_NUMFRAMES);
1085		if (!stv680->fbuf) {
1086			PDEBUG (0, "STV(e): Could not rvmalloc frame bufer");
1087			err = -ENOMEM;
1088		}
1089		file->private_data = dev;
1090	}
1091	if (err)
1092		stv680->user = 0;
1093
1094	return err;
1095}
1096
1097static int stv_close (struct inode *inode, struct file *file)
1098{
1099	struct video_device *dev = file->private_data;
1100	struct usb_stv *stv680 = video_get_drvdata(dev);
1101	int i;
1102
1103	for (i = 0; i < STV680_NUMFRAMES; i++)
1104		stv680->frame[i].grabstate = FRAME_UNUSED;
1105	if (stv680->streaming)
1106		stv680_stop_stream (stv680);
1107
1108	if ((i = stv_stop_video (stv680)) < 0)
1109		PDEBUG (1, "STV(e): stop_video failed in stv_close");
1110
1111	rvfree (stv680->fbuf, stv680->maxframesize * STV680_NUMFRAMES);
1112	stv680->user = 0;
1113
1114	if (stv680->removed) {
1115		kfree(stv680);
1116		stv680 = NULL;
1117		PDEBUG (0, "STV(i): device unregistered");
1118	}
1119	file->private_data = NULL;
1120	return 0;
1121}
1122
1123static int stv680_do_ioctl (struct inode *inode, struct file *file,
1124			    unsigned int cmd, void *arg)
1125{
1126	struct video_device *vdev = file->private_data;
1127	struct usb_stv *stv680 = video_get_drvdata(vdev);
1128
1129	if (!stv680->udev)
1130		return -EIO;
1131
1132	switch (cmd) {
1133	case VIDIOCGCAP:{
1134			struct video_capability *b = arg;
1135
1136			strcpy (b->name, stv680->camera_name);
1137			b->type = VID_TYPE_CAPTURE;
1138			b->channels = 1;
1139			b->audios = 0;
1140			b->maxwidth = stv680->maxwidth;
1141			b->maxheight = stv680->maxheight;
1142			b->minwidth = stv680->maxwidth / 2;
1143			b->minheight = stv680->maxheight / 2;
1144			return 0;
1145		}
1146	case VIDIOCGCHAN:{
1147			struct video_channel *v = arg;
1148
1149			if (v->channel != 0)
1150				return -EINVAL;
1151			v->flags = 0;
1152			v->tuners = 0;
1153			v->type = VIDEO_TYPE_CAMERA;
1154			strcpy (v->name, "STV Camera");
1155			return 0;
1156		}
1157	case VIDIOCSCHAN:{
1158			struct video_channel *v = arg;
1159			if (v->channel != 0)
1160				return -EINVAL;
1161			return 0;
1162		}
1163	case VIDIOCGPICT:{
1164			struct video_picture *p = arg;
1165
1166			stv680_get_pict (stv680, p);
1167			return 0;
1168		}
1169	case VIDIOCSPICT:{
1170			struct video_picture *p = arg;
1171
1172			if (stv680_set_pict (stv680, p))
1173				return -EINVAL;
1174			return 0;
1175		}
1176	case VIDIOCSWIN:{
1177			struct video_window *vw = arg;
1178
1179			if (vw->flags)
1180				return -EINVAL;
1181			if (vw->clipcount)
1182				return -EINVAL;
1183			if (vw->width != stv680->vwidth) {
1184				if (stv680_set_size (stv680, vw->width, vw->height)) {
1185					PDEBUG (2, "STV(e): failed (from user) set size in VIDIOCSWIN");
1186					return -EINVAL;
1187				}
1188			}
1189			return 0;
1190		}
1191	case VIDIOCGWIN:{
1192			struct video_window *vw = arg;
1193
1194			vw->x = 0;
1195			vw->y = 0;
1196			vw->chromakey = 0;
1197			vw->flags = 0;
1198			vw->clipcount = 0;
1199			vw->width = stv680->vwidth;
1200			vw->height = stv680->vheight;
1201			return 0;
1202		}
1203	case VIDIOCGMBUF:{
1204			struct video_mbuf *vm = arg;
1205			int i;
1206
1207			memset (vm, 0, sizeof (*vm));
1208			vm->size = STV680_NUMFRAMES * stv680->maxframesize;
1209			vm->frames = STV680_NUMFRAMES;
1210			for (i = 0; i < STV680_NUMFRAMES; i++)
1211				vm->offsets[i] = stv680->maxframesize * i;
1212			return 0;
1213		}
1214	case VIDIOCMCAPTURE:{
1215			struct video_mmap *vm = arg;
1216
1217			if (vm->format != STV_VIDEO_PALETTE) {
1218				PDEBUG (2, "STV(i): VIDIOCMCAPTURE vm.format (%i) != VIDEO_PALETTE (%i)",
1219					vm->format, STV_VIDEO_PALETTE);
1220				if ((vm->format == 3) && (swapRGB_on == 0))  {
1221					PDEBUG (2, "STV(i): VIDIOCMCAPTURE swapRGB is (auto) ON");
1222					/* this may fix those apps (e.g., xawtv) that want BGR */
1223					swapRGB = 1;
1224				}
1225				return -EINVAL;
1226			}
1227			if (vm->frame >= STV680_NUMFRAMES) {
1228				PDEBUG (2, "STV(e): VIDIOCMCAPTURE vm.frame > NUMFRAMES");
1229				return -EINVAL;
1230			}
1231			if ((stv680->frame[vm->frame].grabstate == FRAME_ERROR)
1232			    || (stv680->frame[vm->frame].grabstate == FRAME_GRABBING)) {
1233				PDEBUG (2, "STV(e): VIDIOCMCAPTURE grabstate (%i) error",
1234					stv680->frame[vm->frame].grabstate);
1235				return -EBUSY;
1236			}
1237			/* Is this according to the v4l spec??? */
1238			if (stv680->vwidth != vm->width) {
1239				if (stv680_set_size (stv680, vm->width, vm->height)) {
1240					PDEBUG (2, "STV(e): VIDIOCMCAPTURE set_size failed");
1241					return -EINVAL;
1242				}
1243			}
1244			stv680->frame[vm->frame].grabstate = FRAME_READY;
1245
1246			if (!stv680->streaming)
1247				stv680_start_stream (stv680);
1248
1249			return 0;
1250		}
1251	case VIDIOCSYNC:{
1252			int *frame = arg;
1253			int ret = 0;
1254
1255			if (*frame < 0 || *frame >= STV680_NUMFRAMES) {
1256				PDEBUG (2, "STV(e): Bad frame # in VIDIOCSYNC");
1257				return -EINVAL;
1258			}
1259			ret = stv680_newframe (stv680, *frame);
1260			stv680->frame[*frame].grabstate = FRAME_UNUSED;
1261			return ret;
1262		}
1263	case VIDIOCGFBUF:{
1264			struct video_buffer *vb = arg;
1265
1266			memset (vb, 0, sizeof (*vb));
1267			return 0;
1268		}
1269	case VIDIOCKEY:
1270		return 0;
1271	case VIDIOCCAPTURE:
1272		{
1273			PDEBUG (2, "STV(e): VIDIOCCAPTURE failed");
1274			return -EINVAL;
1275		}
1276	case VIDIOCSFBUF:
1277	case VIDIOCGTUNER:
1278	case VIDIOCSTUNER:
1279	case VIDIOCGFREQ:
1280	case VIDIOCSFREQ:
1281	case VIDIOCGAUDIO:
1282	case VIDIOCSAUDIO:
1283		return -EINVAL;
1284	default:
1285		return -ENOIOCTLCMD;
1286	}			/* end switch */
1287
1288	return 0;
1289}
1290
1291static int stv680_ioctl(struct inode *inode, struct file *file,
1292			unsigned int cmd, unsigned long arg)
1293{
1294	return video_usercopy(inode, file, cmd, arg, stv680_do_ioctl);
1295}
1296
1297static int stv680_mmap (struct file *file, struct vm_area_struct *vma)
1298{
1299	struct video_device *dev = file->private_data;
1300	struct usb_stv *stv680 = video_get_drvdata(dev);
1301	unsigned long start = vma->vm_start;
1302	unsigned long size  = vma->vm_end-vma->vm_start;
1303	unsigned long page, pos;
1304
1305	mutex_lock(&stv680->lock);
1306
1307	if (stv680->udev == NULL) {
1308		mutex_unlock(&stv680->lock);
1309		return -EIO;
1310	}
1311	if (size > (((STV680_NUMFRAMES * stv680->maxframesize) + PAGE_SIZE - 1)
1312		    & ~(PAGE_SIZE - 1))) {
1313		mutex_unlock(&stv680->lock);
1314		return -EINVAL;
1315	}
1316	pos = (unsigned long) stv680->fbuf;
1317	while (size > 0) {
1318		page = vmalloc_to_pfn((void *)pos);
1319		if (remap_pfn_range(vma, start, page, PAGE_SIZE, PAGE_SHARED)) {
1320			mutex_unlock(&stv680->lock);
1321			return -EAGAIN;
1322		}
1323		start += PAGE_SIZE;
1324		pos += PAGE_SIZE;
1325		if (size > PAGE_SIZE)
1326			size -= PAGE_SIZE;
1327		else
1328			size = 0;
1329	}
1330	mutex_unlock(&stv680->lock);
1331
1332	return 0;
1333}
1334
1335static ssize_t stv680_read (struct file *file, char __user *buf,
1336			size_t count, loff_t *ppos)
1337{
1338	struct video_device *dev = file->private_data;
1339	unsigned long int realcount = count;
1340	int ret = 0;
1341	struct usb_stv *stv680 = video_get_drvdata(dev);
1342	unsigned long int i;
1343
1344	if (STV680_NUMFRAMES != 2) {
1345		PDEBUG (0, "STV(e): STV680_NUMFRAMES needs to be 2!");
1346		return -1;
1347	}
1348	if (stv680->udev == NULL)
1349		return -EIO;
1350	if (realcount > (stv680->vwidth * stv680->vheight * 3))
1351		realcount = stv680->vwidth * stv680->vheight * 3;
1352
1353	/* Shouldn't happen: */
1354	if (stv680->frame[0].grabstate == FRAME_GRABBING) {
1355		PDEBUG (2, "STV(e): FRAME_GRABBING in stv680_read");
1356		return -EBUSY;
1357	}
1358	stv680->frame[0].grabstate = FRAME_READY;
1359	stv680->frame[1].grabstate = FRAME_UNUSED;
1360	stv680->curframe = 0;
1361
1362	if (!stv680->streaming)
1363		stv680_start_stream (stv680);
1364
1365	if (!stv680->streaming) {
1366		ret = stv680_newframe (stv680, 0);	/* ret should = 0 */
1367	}
1368
1369	ret = stv680_newframe (stv680, 0);
1370
1371	if (!ret) {
1372		if ((i = copy_to_user (buf, stv680->frame[0].data, realcount)) != 0) {
1373			PDEBUG (2, "STV(e): copy_to_user frame 0 failed, ret count = %li", i);
1374			return -EFAULT;
1375		}
1376	} else {
1377		realcount = ret;
1378	}
1379	stv680->frame[0].grabstate = FRAME_UNUSED;
1380	return realcount;
1381}				/* stv680_read */
1382
1383static const struct file_operations stv680_fops = {
1384	.owner =	THIS_MODULE,
1385	.open =		stv_open,
1386	.release =     	stv_close,
1387	.read =		stv680_read,
1388	.mmap =		stv680_mmap,
1389	.ioctl =        stv680_ioctl,
1390	.compat_ioctl = v4l_compat_ioctl32,
1391	.llseek =       no_llseek,
1392};
1393static struct video_device stv680_template = {
1394	.owner =	THIS_MODULE,
1395	.name =		"STV0680 USB camera",
1396	.type =		VID_TYPE_CAPTURE,
1397	.hardware =	VID_HARDWARE_SE401,
1398	.fops =         &stv680_fops,
1399	.release =	video_device_release,
1400	.minor = 	-1,
1401};
1402
1403static int stv680_probe (struct usb_interface *intf, const struct usb_device_id *id)
1404{
1405	struct usb_device *dev = interface_to_usbdev(intf);
1406	struct usb_host_interface *interface;
1407	struct usb_stv *stv680 = NULL;
1408	char *camera_name = NULL;
1409	int retval = 0;
1410
1411	/* We don't handle multi-config cameras */
1412	if (dev->descriptor.bNumConfigurations != 1) {
1413		PDEBUG (0, "STV(e): Number of Configurations != 1");
1414		return -ENODEV;
1415	}
1416
1417	interface = &intf->altsetting[0];
1418	/* Is it a STV680? */
1419	if ((le16_to_cpu(dev->descriptor.idVendor) == USB_PENCAM_VENDOR_ID) &&
1420	    (le16_to_cpu(dev->descriptor.idProduct) == USB_PENCAM_PRODUCT_ID)) {
1421		camera_name = "STV0680";
1422		PDEBUG (0, "STV(i): STV0680 camera found.");
1423	} else if ((le16_to_cpu(dev->descriptor.idVendor) == USB_CREATIVEGOMINI_VENDOR_ID) &&
1424		   (le16_to_cpu(dev->descriptor.idProduct) == USB_CREATIVEGOMINI_PRODUCT_ID)) {
1425		camera_name = "Creative WebCam Go Mini";
1426		PDEBUG (0, "STV(i): Creative WebCam Go Mini found.");
1427	} else {
1428		PDEBUG (0, "STV(e): Vendor/Product ID do not match STV0680 or Creative WebCam Go Mini values.");
1429		PDEBUG (0, "STV(e): Check that the STV0680 or Creative WebCam Go Mini camera is connected to the computer.");
1430		retval = -ENODEV;
1431		goto error;
1432	}
1433	/* We found one */
1434	if ((stv680 = kzalloc (sizeof (*stv680), GFP_KERNEL)) == NULL) {
1435		PDEBUG (0, "STV(e): couldn't kmalloc stv680 struct.");
1436		retval = -ENOMEM;
1437		goto error;
1438	}
1439
1440	stv680->udev = dev;
1441	stv680->camera_name = camera_name;
1442
1443	stv680->vdev = video_device_alloc();
1444	if (!stv680->vdev) {
1445		retval = -ENOMEM;
1446		goto error;
1447	}
1448	memcpy(stv680->vdev, &stv680_template, sizeof(stv680_template));
1449	stv680->vdev->dev = &intf->dev;
1450	video_set_drvdata(stv680->vdev, stv680);
1451
1452	memcpy (stv680->vdev->name, stv680->camera_name, strlen (stv680->camera_name));
1453	init_waitqueue_head (&stv680->wq);
1454	mutex_init (&stv680->lock);
1455	wmb ();
1456
1457	if (video_register_device (stv680->vdev, VFL_TYPE_GRABBER, video_nr) == -1) {
1458		PDEBUG (0, "STV(e): video_register_device failed");
1459		retval = -EIO;
1460		goto error_vdev;
1461	}
1462	PDEBUG (0, "STV(i): registered new video device: video%d", stv680->vdev->minor);
1463
1464	usb_set_intfdata (intf, stv680);
1465	retval = stv680_create_sysfs_files(stv680->vdev);
1466	if (retval)
1467		goto error_unreg;
1468	return 0;
1469
1470error_unreg:
1471	video_unregister_device(stv680->vdev);
1472error_vdev:
1473	video_device_release(stv680->vdev);
1474error:
1475	kfree(stv680);
1476	return retval;
1477}
1478
1479static inline void usb_stv680_remove_disconnected (struct usb_stv *stv680)
1480{
1481	int i;
1482
1483	stv680->udev = NULL;
1484	stv680->frame[0].grabstate = FRAME_ERROR;
1485	stv680->frame[1].grabstate = FRAME_ERROR;
1486	stv680->streaming = 0;
1487
1488	wake_up_interruptible (&stv680->wq);
1489
1490	for (i = 0; i < STV680_NUMSBUF; i++)
1491		if (stv680->urb[i]) {
1492			usb_kill_urb (stv680->urb[i]);
1493			usb_free_urb (stv680->urb[i]);
1494			stv680->urb[i] = NULL;
1495			kfree(stv680->sbuf[i].data);
1496		}
1497	for (i = 0; i < STV680_NUMSCRATCH; i++)
1498		kfree(stv680->scratch[i].data);
1499	PDEBUG (0, "STV(i): %s disconnected", stv680->camera_name);
1500
1501	/* Free the memory */
1502	kfree(stv680);
1503}
1504
1505static void stv680_disconnect (struct usb_interface *intf)
1506{
1507	struct usb_stv *stv680 = usb_get_intfdata (intf);
1508
1509	usb_set_intfdata (intf, NULL);
1510
1511	if (stv680) {
1512		/* We don't want people trying to open up the device */
1513		if (stv680->vdev) {
1514			stv680_remove_sysfs_files(stv680->vdev);
1515			video_unregister_device(stv680->vdev);
1516			stv680->vdev = NULL;
1517		}
1518		if (!stv680->user) {
1519			usb_stv680_remove_disconnected (stv680);
1520		} else {
1521			stv680->removed = 1;
1522		}
1523	}
1524}
1525
1526static struct usb_driver stv680_driver = {
1527	.name =		"stv680",
1528	.probe =	stv680_probe,
1529	.disconnect =	stv680_disconnect,
1530	.id_table =	device_table
1531};
1532
1533/********************************************************************
1534 *  Module routines
1535 ********************************************************************/
1536
1537static int __init usb_stv680_init (void)
1538{
1539	if (usb_register (&stv680_driver) < 0) {
1540		PDEBUG (0, "STV(e): Could not setup STV0680 driver");
1541		return -1;
1542	}
1543	PDEBUG (0, "STV(i): usb camera driver version %s registering", DRIVER_VERSION);
1544
1545	info(DRIVER_DESC " " DRIVER_VERSION);
1546	return 0;
1547}
1548
1549static void __exit usb_stv680_exit (void)
1550{
1551	usb_deregister (&stv680_driver);
1552	PDEBUG (0, "STV(i): driver deregistered");
1553}
1554
1555module_init (usb_stv680_init);
1556module_exit (usb_stv680_exit);
1557