Fix race condition between vxlan_sock_add and vxlan_sock_release
[cascardo/linux.git] / drivers / media / usb / au0828 / au0828-dvb.c
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 "au0828.h"
23
24 #include <linux/module.h>
25 #include <linux/slab.h>
26 #include <linux/init.h>
27 #include <linux/device.h>
28 #include <media/v4l2-common.h>
29 #include <media/tuner.h>
30
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: %d\n", __func__, purb->actual_length);
118
119         if (!dev) {
120                 dprintk(2, "%s: no dev!\n", __func__);
121                 return;
122         }
123
124         if (!dev->urb_streaming) {
125                 dprintk(2, "%s: not streaming!\n", __func__);
126                 return;
127         }
128
129         if (ptype != PIPE_BULK) {
130                 pr_err("%s: Unsupported URB type %d\n",
131                        __func__, ptype);
132                 return;
133         }
134
135         /* See if the stream is corrupted (to work around a hardware
136            bug where the stream gets misaligned */
137         ptr = purb->transfer_buffer;
138         if (purb->actual_length > 0 && ptr[0] != 0x47) {
139                 dprintk(1, "Need to restart streaming %02x len=%d!\n",
140                         ptr[0], purb->actual_length);
141                 schedule_work(&dev->restart_streaming);
142                 return;
143         }
144
145         /* Feed the transport payload into the kernel demux */
146         dvb_dmx_swfilter_packets(&dev->dvb.demux,
147                 purb->transfer_buffer, purb->actual_length / 188);
148
149         /* Clean the buffer before we requeue */
150         memset(purb->transfer_buffer, 0, URB_BUFSIZE);
151
152         /* Requeue URB */
153         usb_submit_urb(purb, GFP_ATOMIC);
154 }
155
156 static int stop_urb_transfer(struct au0828_dev *dev)
157 {
158         int i;
159
160         dprintk(2, "%s()\n", __func__);
161
162         if (!dev->urb_streaming)
163                 return 0;
164
165         dev->urb_streaming = false;
166         for (i = 0; i < URB_COUNT; i++) {
167                 if (dev->urbs[i]) {
168                         usb_kill_urb(dev->urbs[i]);
169                         if (!preallocate_big_buffers)
170                                 kfree(dev->urbs[i]->transfer_buffer);
171
172                         usb_free_urb(dev->urbs[i]);
173                 }
174         }
175
176         return 0;
177 }
178
179 static int start_urb_transfer(struct au0828_dev *dev)
180 {
181         struct urb *purb;
182         int i, ret = -ENOMEM;
183
184         dprintk(2, "%s()\n", __func__);
185
186         if (dev->urb_streaming) {
187                 dprintk(2, "%s: bulk xfer already running!\n", __func__);
188                 return 0;
189         }
190
191         for (i = 0; i < URB_COUNT; i++) {
192
193                 dev->urbs[i] = usb_alloc_urb(0, GFP_KERNEL);
194                 if (!dev->urbs[i])
195                         goto err;
196
197                 purb = dev->urbs[i];
198
199                 if (preallocate_big_buffers)
200                         purb->transfer_buffer = dev->dig_transfer_buffer[i];
201                 else
202                         purb->transfer_buffer = kzalloc(URB_BUFSIZE,
203                                         GFP_KERNEL);
204
205                 if (!purb->transfer_buffer) {
206                         usb_free_urb(purb);
207                         dev->urbs[i] = NULL;
208                         pr_err("%s: failed big buffer allocation, err = %d\n",
209                                __func__, ret);
210                         goto err;
211                 }
212
213                 purb->status = -EINPROGRESS;
214                 usb_fill_bulk_urb(purb,
215                                   dev->usbdev,
216                                   usb_rcvbulkpipe(dev->usbdev,
217                                         _AU0828_BULKPIPE),
218                                   purb->transfer_buffer,
219                                   URB_BUFSIZE,
220                                   urb_completion,
221                                   dev);
222
223         }
224
225         for (i = 0; i < URB_COUNT; i++) {
226                 ret = usb_submit_urb(dev->urbs[i], GFP_ATOMIC);
227                 if (ret != 0) {
228                         stop_urb_transfer(dev);
229                         pr_err("%s: failed urb submission, err = %d\n",
230                                __func__, ret);
231                         return ret;
232                 }
233         }
234
235         dev->urb_streaming = true;
236         ret = 0;
237
238 err:
239         return ret;
240 }
241
242 static void au0828_start_transport(struct au0828_dev *dev)
243 {
244         au0828_write(dev, 0x608, 0x90);
245         au0828_write(dev, 0x609, 0x72);
246         au0828_write(dev, 0x60a, 0x71);
247         au0828_write(dev, 0x60b, 0x01);
248
249 }
250
251 static void au0828_stop_transport(struct au0828_dev *dev, int full_stop)
252 {
253         if (full_stop) {
254                 au0828_write(dev, 0x608, 0x00);
255                 au0828_write(dev, 0x609, 0x00);
256                 au0828_write(dev, 0x60a, 0x00);
257         }
258         au0828_write(dev, 0x60b, 0x00);
259 }
260
261 static int au0828_dvb_start_feed(struct dvb_demux_feed *feed)
262 {
263         struct dvb_demux *demux = feed->demux;
264         struct au0828_dev *dev = (struct au0828_dev *) demux->priv;
265         struct au0828_dvb *dvb = &dev->dvb;
266         int ret = 0;
267
268         dprintk(1, "%s()\n", __func__);
269
270         if (!demux->dmx.frontend)
271                 return -EINVAL;
272
273         if (dvb->frontend) {
274                 mutex_lock(&dvb->lock);
275                 dvb->start_count++;
276                 dprintk(1, "%s(), start_count: %d, stop_count: %d\n", __func__,
277                         dvb->start_count, dvb->stop_count);
278                 if (dvb->feeding++ == 0) {
279                         /* Start transport */
280                         au0828_start_transport(dev);
281                         ret = start_urb_transfer(dev);
282                         if (ret < 0) {
283                                 au0828_stop_transport(dev, 0);
284                                 dvb->feeding--; /* We ran out of memory... */
285                         }
286                 }
287                 mutex_unlock(&dvb->lock);
288         }
289
290         return ret;
291 }
292
293 static int au0828_dvb_stop_feed(struct dvb_demux_feed *feed)
294 {
295         struct dvb_demux *demux = feed->demux;
296         struct au0828_dev *dev = (struct au0828_dev *) demux->priv;
297         struct au0828_dvb *dvb = &dev->dvb;
298         int ret = 0;
299
300         dprintk(1, "%s()\n", __func__);
301
302         if (dvb->frontend) {
303                 cancel_work_sync(&dev->restart_streaming);
304
305                 mutex_lock(&dvb->lock);
306                 dvb->stop_count++;
307                 dprintk(1, "%s(), start_count: %d, stop_count: %d\n", __func__,
308                         dvb->start_count, dvb->stop_count);
309                 if (dvb->feeding > 0) {
310                         dvb->feeding--;
311                         if (dvb->feeding == 0) {
312                                 /* Stop transport */
313                                 ret = stop_urb_transfer(dev);
314                                 au0828_stop_transport(dev, 0);
315                         }
316                 }
317                 mutex_unlock(&dvb->lock);
318         }
319
320         return ret;
321 }
322
323 static void au0828_restart_dvb_streaming(struct work_struct *work)
324 {
325         struct au0828_dev *dev = container_of(work, struct au0828_dev,
326                                               restart_streaming);
327         struct au0828_dvb *dvb = &dev->dvb;
328
329         if (!dev->urb_streaming)
330                 return;
331
332         dprintk(1, "Restarting streaming...!\n");
333
334         mutex_lock(&dvb->lock);
335
336         /* Stop transport */
337         stop_urb_transfer(dev);
338         au0828_stop_transport(dev, 1);
339
340         /* Start transport */
341         au0828_start_transport(dev);
342         start_urb_transfer(dev);
343
344         mutex_unlock(&dvb->lock);
345 }
346
347 static int au0828_set_frontend(struct dvb_frontend *fe)
348 {
349         struct au0828_dev *dev = fe->dvb->priv;
350         struct au0828_dvb *dvb = &dev->dvb;
351         int ret, was_streaming;
352
353         mutex_lock(&dvb->lock);
354         was_streaming = dev->urb_streaming;
355         if (was_streaming) {
356                 au0828_stop_transport(dev, 1);
357
358                 /*
359                  * We can't hold a mutex here, as the restart_streaming
360                  * kthread may also hold it.
361                  */
362                 mutex_unlock(&dvb->lock);
363                 cancel_work_sync(&dev->restart_streaming);
364                 mutex_lock(&dvb->lock);
365
366                 stop_urb_transfer(dev);
367         }
368         mutex_unlock(&dvb->lock);
369
370         ret = dvb->set_frontend(fe);
371
372         if (was_streaming) {
373                 mutex_lock(&dvb->lock);
374                 au0828_start_transport(dev);
375                 start_urb_transfer(dev);
376                 mutex_unlock(&dvb->lock);
377         }
378
379         return ret;
380 }
381
382 static int dvb_register(struct au0828_dev *dev)
383 {
384         struct au0828_dvb *dvb = &dev->dvb;
385         int result;
386
387         dprintk(1, "%s()\n", __func__);
388
389         if (preallocate_big_buffers) {
390                 int i;
391                 for (i = 0; i < URB_COUNT; i++) {
392                         dev->dig_transfer_buffer[i] = kzalloc(URB_BUFSIZE,
393                                         GFP_KERNEL);
394
395                         if (!dev->dig_transfer_buffer[i]) {
396                                 result = -ENOMEM;
397
398                                 pr_err("failed buffer allocation (errno = %d)\n",
399                                        result);
400                                 goto fail_adapter;
401                         }
402                 }
403         }
404
405         INIT_WORK(&dev->restart_streaming, au0828_restart_dvb_streaming);
406
407         /* register adapter */
408         result = dvb_register_adapter(&dvb->adapter,
409                                       KBUILD_MODNAME, THIS_MODULE,
410                                       &dev->usbdev->dev, adapter_nr);
411         if (result < 0) {
412                 pr_err("dvb_register_adapter failed (errno = %d)\n",
413                        result);
414                 goto fail_adapter;
415         }
416         dvb->adapter.priv = dev;
417
418         /* register frontend */
419         result = dvb_register_frontend(&dvb->adapter, dvb->frontend);
420         if (result < 0) {
421                 pr_err("dvb_register_frontend failed (errno = %d)\n",
422                        result);
423                 goto fail_frontend;
424         }
425
426         /* Hook dvb frontend */
427         dvb->set_frontend = dvb->frontend->ops.set_frontend;
428         dvb->frontend->ops.set_frontend = au0828_set_frontend;
429
430         /* register demux stuff */
431         dvb->demux.dmx.capabilities =
432                 DMX_TS_FILTERING | DMX_SECTION_FILTERING |
433                 DMX_MEMORY_BASED_FILTERING;
434         dvb->demux.priv       = dev;
435         dvb->demux.filternum  = 256;
436         dvb->demux.feednum    = 256;
437         dvb->demux.start_feed = au0828_dvb_start_feed;
438         dvb->demux.stop_feed  = au0828_dvb_stop_feed;
439         result = dvb_dmx_init(&dvb->demux);
440         if (result < 0) {
441                 pr_err("dvb_dmx_init failed (errno = %d)\n", result);
442                 goto fail_dmx;
443         }
444
445         dvb->dmxdev.filternum    = 256;
446         dvb->dmxdev.demux        = &dvb->demux.dmx;
447         dvb->dmxdev.capabilities = 0;
448         result = dvb_dmxdev_init(&dvb->dmxdev, &dvb->adapter);
449         if (result < 0) {
450                 pr_err("dvb_dmxdev_init failed (errno = %d)\n", result);
451                 goto fail_dmxdev;
452         }
453
454         dvb->fe_hw.source = DMX_FRONTEND_0;
455         result = dvb->demux.dmx.add_frontend(&dvb->demux.dmx, &dvb->fe_hw);
456         if (result < 0) {
457                 pr_err("add_frontend failed (DMX_FRONTEND_0, errno = %d)\n",
458                        result);
459                 goto fail_fe_hw;
460         }
461
462         dvb->fe_mem.source = DMX_MEMORY_FE;
463         result = dvb->demux.dmx.add_frontend(&dvb->demux.dmx, &dvb->fe_mem);
464         if (result < 0) {
465                 pr_err("add_frontend failed (DMX_MEMORY_FE, errno = %d)\n",
466                        result);
467                 goto fail_fe_mem;
468         }
469
470         result = dvb->demux.dmx.connect_frontend(&dvb->demux.dmx, &dvb->fe_hw);
471         if (result < 0) {
472                 pr_err("connect_frontend failed (errno = %d)\n", result);
473                 goto fail_fe_conn;
474         }
475
476         /* register network adapter */
477         dvb_net_init(&dvb->adapter, &dvb->net, &dvb->demux.dmx);
478
479         dvb->start_count = 0;
480         dvb->stop_count = 0;
481         return 0;
482
483 fail_fe_conn:
484         dvb->demux.dmx.remove_frontend(&dvb->demux.dmx, &dvb->fe_mem);
485 fail_fe_mem:
486         dvb->demux.dmx.remove_frontend(&dvb->demux.dmx, &dvb->fe_hw);
487 fail_fe_hw:
488         dvb_dmxdev_release(&dvb->dmxdev);
489 fail_dmxdev:
490         dvb_dmx_release(&dvb->demux);
491 fail_dmx:
492         dvb_unregister_frontend(dvb->frontend);
493 fail_frontend:
494         dvb_frontend_detach(dvb->frontend);
495         dvb_unregister_adapter(&dvb->adapter);
496 fail_adapter:
497
498         if (preallocate_big_buffers) {
499                 int i;
500                 for (i = 0; i < URB_COUNT; i++)
501                         kfree(dev->dig_transfer_buffer[i]);
502         }
503
504         return result;
505 }
506
507 void au0828_dvb_unregister(struct au0828_dev *dev)
508 {
509         struct au0828_dvb *dvb = &dev->dvb;
510
511         dprintk(1, "%s()\n", __func__);
512
513         if (dvb->frontend == NULL)
514                 return;
515
516         cancel_work_sync(&dev->restart_streaming);
517
518         dvb_net_release(&dvb->net);
519         dvb->demux.dmx.remove_frontend(&dvb->demux.dmx, &dvb->fe_mem);
520         dvb->demux.dmx.remove_frontend(&dvb->demux.dmx, &dvb->fe_hw);
521         dvb_dmxdev_release(&dvb->dmxdev);
522         dvb_dmx_release(&dvb->demux);
523         dvb_unregister_frontend(dvb->frontend);
524         dvb_frontend_detach(dvb->frontend);
525         dvb_unregister_adapter(&dvb->adapter);
526
527         if (preallocate_big_buffers) {
528                 int i;
529                 for (i = 0; i < URB_COUNT; i++)
530                         kfree(dev->dig_transfer_buffer[i]);
531         }
532         dvb->frontend = NULL;
533 }
534
535 /* All the DVB attach calls go here, this function get's modified
536  * for each new card. No other function in this file needs
537  * to change.
538  */
539 int au0828_dvb_register(struct au0828_dev *dev)
540 {
541         struct au0828_dvb *dvb = &dev->dvb;
542         int ret;
543
544         dprintk(1, "%s()\n", __func__);
545
546         /* init frontend */
547         switch (dev->boardnr) {
548         case AU0828_BOARD_HAUPPAUGE_HVR850:
549         case AU0828_BOARD_HAUPPAUGE_HVR950Q:
550                 dvb->frontend = dvb_attach(au8522_attach,
551                                 &hauppauge_hvr950q_config,
552                                 &dev->i2c_adap);
553                 if (dvb->frontend != NULL)
554                         switch (dev->board.tuner_type) {
555                         default:
556                         case TUNER_XC5000:
557                                 dvb_attach(xc5000_attach, dvb->frontend,
558                                            &dev->i2c_adap,
559                                            &hauppauge_xc5000a_config);
560                                 break;
561                         case TUNER_XC5000C:
562                                 dvb_attach(xc5000_attach, dvb->frontend,
563                                            &dev->i2c_adap,
564                                            &hauppauge_xc5000c_config);
565                                 break;
566                         }
567                 break;
568         case AU0828_BOARD_HAUPPAUGE_HVR950Q_MXL:
569                 dvb->frontend = dvb_attach(au8522_attach,
570                                 &hauppauge_hvr950q_config,
571                                 &dev->i2c_adap);
572                 if (dvb->frontend != NULL)
573                         dvb_attach(mxl5007t_attach, dvb->frontend,
574                                    &dev->i2c_adap, 0x60,
575                                    &mxl5007t_hvr950q_config);
576                 break;
577         case AU0828_BOARD_HAUPPAUGE_WOODBURY:
578                 dvb->frontend = dvb_attach(au8522_attach,
579                                 &hauppauge_woodbury_config,
580                                 &dev->i2c_adap);
581                 if (dvb->frontend != NULL)
582                         dvb_attach(tda18271_attach, dvb->frontend,
583                                    0x60, &dev->i2c_adap,
584                                    &hauppauge_woodbury_tunerconfig);
585                 break;
586         case AU0828_BOARD_DVICO_FUSIONHDTV7:
587                 dvb->frontend = dvb_attach(au8522_attach,
588                                 &fusionhdtv7usb_config,
589                                 &dev->i2c_adap);
590                 if (dvb->frontend != NULL) {
591                         dvb_attach(xc5000_attach, dvb->frontend,
592                                 &dev->i2c_adap,
593                                 &hauppauge_xc5000a_config);
594                 }
595                 break;
596         default:
597                 pr_warn("The frontend of your DVB/ATSC card isn't supported yet\n");
598                 break;
599         }
600         if (NULL == dvb->frontend) {
601                 pr_err("%s() Frontend initialization failed\n",
602                        __func__);
603                 return -1;
604         }
605         /* define general-purpose callback pointer */
606         dvb->frontend->callback = au0828_tuner_callback;
607
608         /* register everything */
609         ret = dvb_register(dev);
610         if (ret < 0) {
611                 if (dvb->frontend->ops.release)
612                         dvb->frontend->ops.release(dvb->frontend);
613                 dvb->frontend = NULL;
614                 return ret;
615         }
616
617         return 0;
618 }
619
620 void au0828_dvb_suspend(struct au0828_dev *dev)
621 {
622         struct au0828_dvb *dvb = &dev->dvb;
623         int rc;
624
625         if (dvb->frontend) {
626                 if (dev->urb_streaming) {
627                         cancel_work_sync(&dev->restart_streaming);
628                         /* Stop transport */
629                         mutex_lock(&dvb->lock);
630                         stop_urb_transfer(dev);
631                         au0828_stop_transport(dev, 1);
632                         mutex_unlock(&dvb->lock);
633                         dev->need_urb_start = true;
634                 }
635                 /* suspend frontend - does tuner and fe to sleep */
636                 rc = dvb_frontend_suspend(dvb->frontend);
637                 pr_info("au0828_dvb_suspend(): Suspending DVB fe %d\n", rc);
638         }
639 }
640
641 void au0828_dvb_resume(struct au0828_dev *dev)
642 {
643         struct au0828_dvb *dvb = &dev->dvb;
644         int rc;
645
646         if (dvb->frontend) {
647                 /* resume frontend - does fe and tuner init */
648                 rc = dvb_frontend_resume(dvb->frontend);
649                 pr_info("au0828_dvb_resume(): Resuming DVB fe %d\n", rc);
650                 if (dev->need_urb_start) {
651                         /* Start transport */
652                         mutex_lock(&dvb->lock);
653                         au0828_start_transport(dev);
654                         start_urb_transfer(dev);
655                         mutex_unlock(&dvb->lock);
656                 }
657         }
658 }