1 /*
2  *  Driver for the Auvitek USB bridge
3  *
4  *  Copyright (c) 2008 Steven Toth <stoth@linuxtv.org>
5  *
6  *  This program is free software; you can redistribute it and/or modify
7  *  it under the terms of the GNU General Public License as published by
8  *  the Free Software Foundation; either version 2 of the License, or
9  *  (at your option) any later version.
10  *
11  *  This program is distributed in the hope that it will be useful,
12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  *
15  *  GNU General Public License for more details.
16  *
17  *  You should have received a copy of the GNU General Public License
18  *  along with this program; if not, write to the Free Software
19  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20  */
21 
22 #include <linux/module.h>
23 #include <linux/slab.h>
24 #include <linux/init.h>
25 #include <linux/device.h>
26 #include <linux/suspend.h>
27 #include <media/v4l2-common.h>
28 #include <media/tuner.h>
29 
30 #include "au0828.h"
31 #include "au8522.h"
32 #include "xc5000.h"
33 #include "mxl5007t.h"
34 #include "tda18271.h"
35 
36 static int preallocate_big_buffers;
37 module_param_named(preallocate_big_buffers, preallocate_big_buffers, int, 0644);
38 MODULE_PARM_DESC(preallocate_big_buffers, "Preallocate the larger transfer buffers at module load time");
39 
40 DVB_DEFINE_MOD_OPT_ADAPTER_NR(adapter_nr);
41 
42 #define _AU0828_BULKPIPE 0x83
43 #define _BULKPIPESIZE 0xe522
44 
45 static u8 hauppauge_hvr950q_led_states[] = {
46 	0x00, /* off */
47 	0x02, /* yellow */
48 	0x04, /* green */
49 };
50 
51 static struct au8522_led_config hauppauge_hvr950q_led_cfg = {
52 	.gpio_output = 0x00e0,
53 	.gpio_output_enable  = 0x6006,
54 	.gpio_output_disable = 0x0660,
55 
56 	.gpio_leds = 0x00e2,
57 	.led_states  = hauppauge_hvr950q_led_states,
58 	.num_led_states = sizeof(hauppauge_hvr950q_led_states),
59 
60 	.vsb8_strong   = 20 /* dB */ * 10,
61 	.qam64_strong  = 25 /* dB */ * 10,
62 	.qam256_strong = 32 /* dB */ * 10,
63 };
64 
65 static struct au8522_config hauppauge_hvr950q_config = {
66 	.demod_address = 0x8e >> 1,
67 	.status_mode   = AU8522_DEMODLOCKING,
68 	.qam_if        = AU8522_IF_6MHZ,
69 	.vsb_if        = AU8522_IF_6MHZ,
70 	.led_cfg       = &hauppauge_hvr950q_led_cfg,
71 };
72 
73 static struct au8522_config fusionhdtv7usb_config = {
74 	.demod_address = 0x8e >> 1,
75 	.status_mode   = AU8522_DEMODLOCKING,
76 	.qam_if        = AU8522_IF_6MHZ,
77 	.vsb_if        = AU8522_IF_6MHZ,
78 };
79 
80 static struct au8522_config hauppauge_woodbury_config = {
81 	.demod_address = 0x8e >> 1,
82 	.status_mode   = AU8522_DEMODLOCKING,
83 	.qam_if        = AU8522_IF_4MHZ,
84 	.vsb_if        = AU8522_IF_3_25MHZ,
85 };
86 
87 static struct xc5000_config hauppauge_xc5000a_config = {
88 	.i2c_address      = 0x61,
89 	.if_khz           = 6000,
90 	.chip_id          = XC5000A,
91 };
92 
93 static struct xc5000_config hauppauge_xc5000c_config = {
94 	.i2c_address      = 0x61,
95 	.if_khz           = 6000,
96 	.chip_id          = XC5000C,
97 };
98 
99 static struct mxl5007t_config mxl5007t_hvr950q_config = {
100 	.xtal_freq_hz = MxL_XTAL_24_MHZ,
101 	.if_freq_hz = MxL_IF_6_MHZ,
102 };
103 
104 static struct tda18271_config hauppauge_woodbury_tunerconfig = {
105 	.gate    = TDA18271_GATE_DIGITAL,
106 };
107 
108 static void au0828_restart_dvb_streaming(struct work_struct *work);
109 
110 /*-------------------------------------------------------------------*/
111 static void urb_completion(struct urb *purb)
112 {
113 	struct au0828_dev *dev = purb->context;
114 	int ptype = usb_pipetype(purb->pipe);
115 	unsigned char *ptr;
116 
117 	dprintk(2, "%s()\n", __func__);
118 
119 	if (!dev)
120 		return;
121 
122 	if (dev->urb_streaming == 0)
123 		return;
124 
125 	if (ptype != PIPE_BULK) {
126 		printk(KERN_ERR "%s() Unsupported URB type %d\n",
127 		       __func__, ptype);
128 		return;
129 	}
130 
131 	/* See if the stream is corrupted (to work around a hardware
132 	   bug where the stream gets misaligned */
133 	ptr = purb->transfer_buffer;
134 	if (purb->actual_length > 0 && ptr[0] != 0x47) {
135 		dprintk(1, "Need to restart streaming %02x len=%d!\n",
136 			ptr[0], purb->actual_length);
137 		schedule_work(&dev->restart_streaming);
138 		return;
139 	}
140 
141 	/* Feed the transport payload into the kernel demux */
142 	dvb_dmx_swfilter_packets(&dev->dvb.demux,
143 		purb->transfer_buffer, purb->actual_length / 188);
144 
145 	/* Clean the buffer before we requeue */
146 	memset(purb->transfer_buffer, 0, URB_BUFSIZE);
147 
148 	/* Requeue URB */
149 	usb_submit_urb(purb, GFP_ATOMIC);
150 }
151 
152 static int stop_urb_transfer(struct au0828_dev *dev)
153 {
154 	int i;
155 
156 	dprintk(2, "%s()\n", __func__);
157 
158 	dev->urb_streaming = 0;
159 	for (i = 0; i < URB_COUNT; i++) {
160 		if (dev->urbs[i]) {
161 			usb_kill_urb(dev->urbs[i]);
162 			if (!preallocate_big_buffers)
163 				kfree(dev->urbs[i]->transfer_buffer);
164 
165 			usb_free_urb(dev->urbs[i]);
166 		}
167 	}
168 
169 	return 0;
170 }
171 
172 static int start_urb_transfer(struct au0828_dev *dev)
173 {
174 	struct urb *purb;
175 	int i, ret = -ENOMEM;
176 
177 	dprintk(2, "%s()\n", __func__);
178 
179 	if (dev->urb_streaming) {
180 		dprintk(2, "%s: bulk xfer already running!\n", __func__);
181 		return 0;
182 	}
183 
184 	for (i = 0; i < URB_COUNT; i++) {
185 
186 		dev->urbs[i] = usb_alloc_urb(0, GFP_KERNEL);
187 		if (!dev->urbs[i])
188 			goto err;
189 
190 		purb = dev->urbs[i];
191 
192 		if (preallocate_big_buffers)
193 			purb->transfer_buffer = dev->dig_transfer_buffer[i];
194 		else
195 			purb->transfer_buffer = kzalloc(URB_BUFSIZE,
196 					GFP_KERNEL);
197 
198 		if (!purb->transfer_buffer) {
199 			usb_free_urb(purb);
200 			dev->urbs[i] = NULL;
201 			printk(KERN_ERR
202 			       "%s: failed big buffer allocation, err = %d\n",
203 			       __func__, ret);
204 			goto err;
205 		}
206 
207 		purb->status = -EINPROGRESS;
208 		usb_fill_bulk_urb(purb,
209 				  dev->usbdev,
210 				  usb_rcvbulkpipe(dev->usbdev,
211 					_AU0828_BULKPIPE),
212 				  purb->transfer_buffer,
213 				  URB_BUFSIZE,
214 				  urb_completion,
215 				  dev);
216 
217 	}
218 
219 	for (i = 0; i < URB_COUNT; i++) {
220 		ret = usb_submit_urb(dev->urbs[i], GFP_ATOMIC);
221 		if (ret != 0) {
222 			stop_urb_transfer(dev);
223 			printk(KERN_ERR "%s: failed urb submission, "
224 			       "err = %d\n", __func__, ret);
225 			return ret;
226 		}
227 	}
228 
229 	dev->urb_streaming = 1;
230 	ret = 0;
231 
232 err:
233 	return ret;
234 }
235 
236 static void au0828_start_transport(struct au0828_dev *dev)
237 {
238 	au0828_write(dev, 0x608, 0x90);
239 	au0828_write(dev, 0x609, 0x72);
240 	au0828_write(dev, 0x60a, 0x71);
241 	au0828_write(dev, 0x60b, 0x01);
242 
243 }
244 
245 static void au0828_stop_transport(struct au0828_dev *dev, int full_stop)
246 {
247 	if (full_stop) {
248 		au0828_write(dev, 0x608, 0x00);
249 		au0828_write(dev, 0x609, 0x00);
250 		au0828_write(dev, 0x60a, 0x00);
251 	}
252 	au0828_write(dev, 0x60b, 0x00);
253 }
254 
255 
256 
257 static int au0828_dvb_start_feed(struct dvb_demux_feed *feed)
258 {
259 	struct dvb_demux *demux = feed->demux;
260 	struct au0828_dev *dev = (struct au0828_dev *) demux->priv;
261 	struct au0828_dvb *dvb = &dev->dvb;
262 	int ret = 0;
263 
264 	dprintk(1, "%s()\n", __func__);
265 
266 	if (!demux->dmx.frontend)
267 		return -EINVAL;
268 
269 	if (dvb) {
270 		mutex_lock(&dvb->lock);
271 		dvb->start_count++;
272 		dprintk(1, "%s(), start_count: %d, stop_count: %d\n", __func__,
273 			dvb->start_count, dvb->stop_count);
274 		if (dvb->feeding++ == 0) {
275 			/* Start transport */
276 			au0828_start_transport(dev);
277 			ret = start_urb_transfer(dev);
278 			if (ret < 0) {
279 				au0828_stop_transport(dev, 0);
280 				dvb->feeding--;	/* We ran out of memory... */
281 			}
282 		}
283 		mutex_unlock(&dvb->lock);
284 	}
285 
286 	return ret;
287 }
288 
289 static int au0828_dvb_stop_feed(struct dvb_demux_feed *feed)
290 {
291 	struct dvb_demux *demux = feed->demux;
292 	struct au0828_dev *dev = (struct au0828_dev *) demux->priv;
293 	struct au0828_dvb *dvb = &dev->dvb;
294 	int ret = 0;
295 
296 	dprintk(1, "%s()\n", __func__);
297 
298 	if (dvb) {
299 		mutex_lock(&dvb->lock);
300 		dvb->stop_count++;
301 		dprintk(1, "%s(), start_count: %d, stop_count: %d\n", __func__,
302 			dvb->start_count, dvb->stop_count);
303 		if (dvb->feeding > 0) {
304 			dvb->feeding--;
305 			if (dvb->feeding == 0) {
306 				/* Stop transport */
307 				ret = stop_urb_transfer(dev);
308 				au0828_stop_transport(dev, 0);
309 			}
310 		}
311 		mutex_unlock(&dvb->lock);
312 	}
313 
314 	return ret;
315 }
316 
317 static void au0828_restart_dvb_streaming(struct work_struct *work)
318 {
319 	struct au0828_dev *dev = container_of(work, struct au0828_dev,
320 					      restart_streaming);
321 	struct au0828_dvb *dvb = &dev->dvb;
322 
323 	if (dev->urb_streaming == 0)
324 		return;
325 
326 	dprintk(1, "Restarting streaming...!\n");
327 
328 	mutex_lock(&dvb->lock);
329 
330 	/* Stop transport */
331 	stop_urb_transfer(dev);
332 	au0828_stop_transport(dev, 1);
333 
334 	/* Start transport */
335 	au0828_start_transport(dev);
336 	start_urb_transfer(dev);
337 
338 	mutex_unlock(&dvb->lock);
339 }
340 
341 static int dvb_register(struct au0828_dev *dev)
342 {
343 	struct au0828_dvb *dvb = &dev->dvb;
344 	int result;
345 
346 	dprintk(1, "%s()\n", __func__);
347 
348 	if (preallocate_big_buffers) {
349 		int i;
350 		for (i = 0; i < URB_COUNT; i++) {
351 			dev->dig_transfer_buffer[i] = kzalloc(URB_BUFSIZE,
352 					GFP_KERNEL);
353 
354 			if (!dev->dig_transfer_buffer[i]) {
355 				result = -ENOMEM;
356 
357 				printk(KERN_ERR
358 				       "%s: failed buffer allocation (errno = %d)\n",
359 				       DRIVER_NAME, result);
360 				goto fail_adapter;
361 			}
362 		}
363 	}
364 
365 	INIT_WORK(&dev->restart_streaming, au0828_restart_dvb_streaming);
366 
367 	/* register adapter */
368 	result = dvb_register_adapter(&dvb->adapter, DRIVER_NAME, THIS_MODULE,
369 				      &dev->usbdev->dev, adapter_nr);
370 	if (result < 0) {
371 		printk(KERN_ERR "%s: dvb_register_adapter failed "
372 		       "(errno = %d)\n", DRIVER_NAME, result);
373 		goto fail_adapter;
374 	}
375 	dvb->adapter.priv = dev;
376 
377 	/* register frontend */
378 	result = dvb_register_frontend(&dvb->adapter, dvb->frontend);
379 	if (result < 0) {
380 		printk(KERN_ERR "%s: dvb_register_frontend failed "
381 		       "(errno = %d)\n", DRIVER_NAME, result);
382 		goto fail_frontend;
383 	}
384 
385 	/* register demux stuff */
386 	dvb->demux.dmx.capabilities =
387 		DMX_TS_FILTERING | DMX_SECTION_FILTERING |
388 		DMX_MEMORY_BASED_FILTERING;
389 	dvb->demux.priv       = dev;
390 	dvb->demux.filternum  = 256;
391 	dvb->demux.feednum    = 256;
392 	dvb->demux.start_feed = au0828_dvb_start_feed;
393 	dvb->demux.stop_feed  = au0828_dvb_stop_feed;
394 	result = dvb_dmx_init(&dvb->demux);
395 	if (result < 0) {
396 		printk(KERN_ERR "%s: dvb_dmx_init failed (errno = %d)\n",
397 		       DRIVER_NAME, result);
398 		goto fail_dmx;
399 	}
400 
401 	dvb->dmxdev.filternum    = 256;
402 	dvb->dmxdev.demux        = &dvb->demux.dmx;
403 	dvb->dmxdev.capabilities = 0;
404 	result = dvb_dmxdev_init(&dvb->dmxdev, &dvb->adapter);
405 	if (result < 0) {
406 		printk(KERN_ERR "%s: dvb_dmxdev_init failed (errno = %d)\n",
407 		       DRIVER_NAME, result);
408 		goto fail_dmxdev;
409 	}
410 
411 	dvb->fe_hw.source = DMX_FRONTEND_0;
412 	result = dvb->demux.dmx.add_frontend(&dvb->demux.dmx, &dvb->fe_hw);
413 	if (result < 0) {
414 		printk(KERN_ERR "%s: add_frontend failed "
415 		       "(DMX_FRONTEND_0, errno = %d)\n", DRIVER_NAME, result);
416 		goto fail_fe_hw;
417 	}
418 
419 	dvb->fe_mem.source = DMX_MEMORY_FE;
420 	result = dvb->demux.dmx.add_frontend(&dvb->demux.dmx, &dvb->fe_mem);
421 	if (result < 0) {
422 		printk(KERN_ERR "%s: add_frontend failed "
423 		       "(DMX_MEMORY_FE, errno = %d)\n", DRIVER_NAME, result);
424 		goto fail_fe_mem;
425 	}
426 
427 	result = dvb->demux.dmx.connect_frontend(&dvb->demux.dmx, &dvb->fe_hw);
428 	if (result < 0) {
429 		printk(KERN_ERR "%s: connect_frontend failed (errno = %d)\n",
430 		       DRIVER_NAME, result);
431 		goto fail_fe_conn;
432 	}
433 
434 	/* register network adapter */
435 	dvb_net_init(&dvb->adapter, &dvb->net, &dvb->demux.dmx);
436 
437 	dvb->start_count = 0;
438 	dvb->stop_count = 0;
439 	return 0;
440 
441 fail_fe_conn:
442 	dvb->demux.dmx.remove_frontend(&dvb->demux.dmx, &dvb->fe_mem);
443 fail_fe_mem:
444 	dvb->demux.dmx.remove_frontend(&dvb->demux.dmx, &dvb->fe_hw);
445 fail_fe_hw:
446 	dvb_dmxdev_release(&dvb->dmxdev);
447 fail_dmxdev:
448 	dvb_dmx_release(&dvb->demux);
449 fail_dmx:
450 	dvb_unregister_frontend(dvb->frontend);
451 fail_frontend:
452 	dvb_frontend_detach(dvb->frontend);
453 	dvb_unregister_adapter(&dvb->adapter);
454 fail_adapter:
455 
456 	if (preallocate_big_buffers) {
457 		int i;
458 		for (i = 0; i < URB_COUNT; i++)
459 			kfree(dev->dig_transfer_buffer[i]);
460 	}
461 
462 	return result;
463 }
464 
465 void au0828_dvb_unregister(struct au0828_dev *dev)
466 {
467 	struct au0828_dvb *dvb = &dev->dvb;
468 
469 	dprintk(1, "%s()\n", __func__);
470 
471 	if (dvb->frontend == NULL)
472 		return;
473 
474 	cancel_work_sync(&dev->restart_streaming);
475 
476 	dvb_net_release(&dvb->net);
477 	dvb->demux.dmx.remove_frontend(&dvb->demux.dmx, &dvb->fe_mem);
478 	dvb->demux.dmx.remove_frontend(&dvb->demux.dmx, &dvb->fe_hw);
479 	dvb_dmxdev_release(&dvb->dmxdev);
480 	dvb_dmx_release(&dvb->demux);
481 	dvb_unregister_frontend(dvb->frontend);
482 	dvb_frontend_detach(dvb->frontend);
483 	dvb_unregister_adapter(&dvb->adapter);
484 
485 	if (preallocate_big_buffers) {
486 		int i;
487 		for (i = 0; i < URB_COUNT; i++)
488 			kfree(dev->dig_transfer_buffer[i]);
489 	}
490 
491 
492 }
493 
494 /* All the DVB attach calls go here, this function get's modified
495  * for each new card. No other function in this file needs
496  * to change.
497  */
498 int au0828_dvb_register(struct au0828_dev *dev)
499 {
500 	struct au0828_dvb *dvb = &dev->dvb;
501 	int ret;
502 
503 	dprintk(1, "%s()\n", __func__);
504 
505 	/* init frontend */
506 	switch (dev->boardnr) {
507 	case AU0828_BOARD_HAUPPAUGE_HVR850:
508 	case AU0828_BOARD_HAUPPAUGE_HVR950Q:
509 		dvb->frontend = dvb_attach(au8522_attach,
510 				&hauppauge_hvr950q_config,
511 				&dev->i2c_adap);
512 		if (dvb->frontend != NULL)
513 			switch (dev->board.tuner_type) {
514 			default:
515 			case TUNER_XC5000:
516 				dvb_attach(xc5000_attach, dvb->frontend,
517 					   &dev->i2c_adap,
518 					   &hauppauge_xc5000a_config);
519 				break;
520 			case TUNER_XC5000C:
521 				dvb_attach(xc5000_attach, dvb->frontend,
522 					   &dev->i2c_adap,
523 					   &hauppauge_xc5000c_config);
524 				break;
525 			}
526 		break;
527 	case AU0828_BOARD_HAUPPAUGE_HVR950Q_MXL:
528 		dvb->frontend = dvb_attach(au8522_attach,
529 				&hauppauge_hvr950q_config,
530 				&dev->i2c_adap);
531 		if (dvb->frontend != NULL)
532 			dvb_attach(mxl5007t_attach, dvb->frontend,
533 				   &dev->i2c_adap, 0x60,
534 				   &mxl5007t_hvr950q_config);
535 		break;
536 	case AU0828_BOARD_HAUPPAUGE_WOODBURY:
537 		dvb->frontend = dvb_attach(au8522_attach,
538 				&hauppauge_woodbury_config,
539 				&dev->i2c_adap);
540 		if (dvb->frontend != NULL)
541 			dvb_attach(tda18271_attach, dvb->frontend,
542 				   0x60, &dev->i2c_adap,
543 				   &hauppauge_woodbury_tunerconfig);
544 		break;
545 	case AU0828_BOARD_DVICO_FUSIONHDTV7:
546 		dvb->frontend = dvb_attach(au8522_attach,
547 				&fusionhdtv7usb_config,
548 				&dev->i2c_adap);
549 		if (dvb->frontend != NULL) {
550 			dvb_attach(xc5000_attach, dvb->frontend,
551 				&dev->i2c_adap,
552 				&hauppauge_xc5000a_config);
553 		}
554 		break;
555 	default:
556 		printk(KERN_WARNING "The frontend of your DVB/ATSC card "
557 		       "isn't supported yet\n");
558 		break;
559 	}
560 	if (NULL == dvb->frontend) {
561 		printk(KERN_ERR "%s() Frontend initialization failed\n",
562 		       __func__);
563 		return -1;
564 	}
565 	/* define general-purpose callback pointer */
566 	dvb->frontend->callback = au0828_tuner_callback;
567 
568 	/* register everything */
569 	ret = dvb_register(dev);
570 	if (ret < 0) {
571 		if (dvb->frontend->ops.release)
572 			dvb->frontend->ops.release(dvb->frontend);
573 		return ret;
574 	}
575 
576 	return 0;
577 }
578