linux/drivers/media/video/pvrusb2/pvrusb2-dvb.c
<<
>>
Prefs
   1/*
   2 *  pvrusb2-dvb.c - linux-dvb api interface to the pvrusb2 driver.
   3 *
   4 *  Copyright (C) 2007, 2008 Michael Krufky <mkrufky@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
   9 *
  10 *  This program is distributed in the hope that it will be useful,
  11 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 *  GNU General Public License for more details.
  14 *
  15 *  You should have received a copy of the GNU General Public License
  16 *  along with this program; if not, write to the Free Software
  17 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18 *
  19 */
  20
  21#include <linux/kthread.h>
  22#include <linux/freezer.h>
  23#include <linux/slab.h>
  24#include <linux/mm.h>
  25#include "dvbdev.h"
  26#include "pvrusb2-debug.h"
  27#include "pvrusb2-hdw-internal.h"
  28#include "pvrusb2-hdw.h"
  29#include "pvrusb2-io.h"
  30#include "pvrusb2-dvb.h"
  31
  32DVB_DEFINE_MOD_OPT_ADAPTER_NR(adapter_nr);
  33
  34static int pvr2_dvb_feed_func(struct pvr2_dvb_adapter *adap)
  35{
  36        int ret;
  37        unsigned int count;
  38        struct pvr2_buffer *bp;
  39        struct pvr2_stream *stream;
  40
  41        pvr2_trace(PVR2_TRACE_DVB_FEED, "dvb feed thread started");
  42        set_freezable();
  43
  44        stream = adap->channel.stream->stream;
  45
  46        for (;;) {
  47                if (kthread_should_stop()) break;
  48
  49                /* Not sure about this... */
  50                try_to_freeze();
  51
  52                bp = pvr2_stream_get_ready_buffer(stream);
  53                if (bp != NULL) {
  54                        count = pvr2_buffer_get_count(bp);
  55                        if (count) {
  56                                dvb_dmx_swfilter(
  57                                        &adap->demux,
  58                                        adap->buffer_storage[
  59                                            pvr2_buffer_get_id(bp)],
  60                                        count);
  61                        } else {
  62                                ret = pvr2_buffer_get_status(bp);
  63                                if (ret < 0) break;
  64                        }
  65                        ret = pvr2_buffer_queue(bp);
  66                        if (ret < 0) break;
  67
  68                        /* Since we know we did something to a buffer,
  69                           just go back and try again.  No point in
  70                           blocking unless we really ran out of
  71                           buffers to process. */
  72                        continue;
  73                }
  74
  75
  76                /* Wait until more buffers become available or we're
  77                   told not to wait any longer. */
  78                ret = wait_event_interruptible(
  79                    adap->buffer_wait_data,
  80                    (pvr2_stream_get_ready_count(stream) > 0) ||
  81                    kthread_should_stop());
  82                if (ret < 0) break;
  83        }
  84
  85        /* If we get here and ret is < 0, then an error has occurred.
  86           Probably would be a good idea to communicate that to DVB core... */
  87
  88        pvr2_trace(PVR2_TRACE_DVB_FEED, "dvb feed thread stopped");
  89
  90        return 0;
  91}
  92
  93static int pvr2_dvb_feed_thread(void *data)
  94{
  95        int stat = pvr2_dvb_feed_func(data);
  96        /* from videobuf-dvb.c: */
  97        while (!kthread_should_stop()) {
  98                set_current_state(TASK_INTERRUPTIBLE);
  99                schedule();
 100        }
 101        return stat;
 102}
 103
 104static void pvr2_dvb_notify(struct pvr2_dvb_adapter *adap)
 105{
 106        wake_up(&adap->buffer_wait_data);
 107}
 108
 109static void pvr2_dvb_stream_end(struct pvr2_dvb_adapter *adap)
 110{
 111        unsigned int idx;
 112        struct pvr2_stream *stream;
 113
 114        if (adap->thread) {
 115                kthread_stop(adap->thread);
 116                adap->thread = NULL;
 117        }
 118
 119        if (adap->channel.stream) {
 120                stream = adap->channel.stream->stream;
 121        } else {
 122                stream = NULL;
 123        }
 124        if (stream) {
 125                pvr2_hdw_set_streaming(adap->channel.hdw, 0);
 126                pvr2_stream_set_callback(stream, NULL, NULL);
 127                pvr2_stream_kill(stream);
 128                pvr2_stream_set_buffer_count(stream, 0);
 129                pvr2_channel_claim_stream(&adap->channel, NULL);
 130        }
 131
 132        if (adap->stream_run) {
 133                for (idx = 0; idx < PVR2_DVB_BUFFER_COUNT; idx++) {
 134                        if (!(adap->buffer_storage[idx])) continue;
 135                        kfree(adap->buffer_storage[idx]);
 136                        adap->buffer_storage[idx] = NULL;
 137                }
 138                adap->stream_run = 0;
 139        }
 140}
 141
 142static int pvr2_dvb_stream_do_start(struct pvr2_dvb_adapter *adap)
 143{
 144        struct pvr2_context *pvr = adap->channel.mc_head;
 145        unsigned int idx;
 146        int ret;
 147        struct pvr2_buffer *bp;
 148        struct pvr2_stream *stream = NULL;
 149
 150        if (adap->stream_run) return -EIO;
 151
 152        ret = pvr2_channel_claim_stream(&adap->channel, &pvr->video_stream);
 153        /* somebody else already has the stream */
 154        if (ret < 0) return ret;
 155
 156        stream = adap->channel.stream->stream;
 157
 158        for (idx = 0; idx < PVR2_DVB_BUFFER_COUNT; idx++) {
 159                adap->buffer_storage[idx] = kmalloc(PVR2_DVB_BUFFER_SIZE,
 160                                                    GFP_KERNEL);
 161                if (!(adap->buffer_storage[idx])) return -ENOMEM;
 162        }
 163
 164        pvr2_stream_set_callback(pvr->video_stream.stream,
 165                                 (pvr2_stream_callback) pvr2_dvb_notify, adap);
 166
 167        ret = pvr2_stream_set_buffer_count(stream, PVR2_DVB_BUFFER_COUNT);
 168        if (ret < 0) return ret;
 169
 170        for (idx = 0; idx < PVR2_DVB_BUFFER_COUNT; idx++) {
 171                bp = pvr2_stream_get_buffer(stream, idx);
 172                pvr2_buffer_set_buffer(bp,
 173                                       adap->buffer_storage[idx],
 174                                       PVR2_DVB_BUFFER_SIZE);
 175        }
 176
 177        ret = pvr2_hdw_set_streaming(adap->channel.hdw, 1);
 178        if (ret < 0) return ret;
 179
 180        while ((bp = pvr2_stream_get_idle_buffer(stream)) != NULL) {
 181                ret = pvr2_buffer_queue(bp);
 182                if (ret < 0) return ret;
 183        }
 184
 185        adap->thread = kthread_run(pvr2_dvb_feed_thread, adap, "pvrusb2-dvb");
 186
 187        if (IS_ERR(adap->thread)) {
 188                ret = PTR_ERR(adap->thread);
 189                adap->thread = NULL;
 190                return ret;
 191        }
 192
 193        adap->stream_run = !0;
 194
 195        return 0;
 196}
 197
 198static int pvr2_dvb_stream_start(struct pvr2_dvb_adapter *adap)
 199{
 200        int ret = pvr2_dvb_stream_do_start(adap);
 201        if (ret < 0) pvr2_dvb_stream_end(adap);
 202        return ret;
 203}
 204
 205static int pvr2_dvb_ctrl_feed(struct dvb_demux_feed *dvbdmxfeed, int onoff)
 206{
 207        struct pvr2_dvb_adapter *adap = dvbdmxfeed->demux->priv;
 208        int ret = 0;
 209
 210        if (adap == NULL) return -ENODEV;
 211
 212        mutex_lock(&adap->lock);
 213        do {
 214                if (onoff) {
 215                        if (!adap->feedcount) {
 216                                pvr2_trace(PVR2_TRACE_DVB_FEED,
 217                                           "start feeding demux");
 218                                ret = pvr2_dvb_stream_start(adap);
 219                                if (ret < 0) break;
 220                        }
 221                        (adap->feedcount)++;
 222                } else if (adap->feedcount > 0) {
 223                        (adap->feedcount)--;
 224                        if (!adap->feedcount) {
 225                                pvr2_trace(PVR2_TRACE_DVB_FEED,
 226                                           "stop feeding demux");
 227                                pvr2_dvb_stream_end(adap);
 228                        }
 229                }
 230        } while (0);
 231        mutex_unlock(&adap->lock);
 232
 233        return ret;
 234}
 235
 236static int pvr2_dvb_start_feed(struct dvb_demux_feed *dvbdmxfeed)
 237{
 238        pvr2_trace(PVR2_TRACE_DVB_FEED, "start pid: 0x%04x", dvbdmxfeed->pid);
 239        return pvr2_dvb_ctrl_feed(dvbdmxfeed, 1);
 240}
 241
 242static int pvr2_dvb_stop_feed(struct dvb_demux_feed *dvbdmxfeed)
 243{
 244        pvr2_trace(PVR2_TRACE_DVB_FEED, "stop pid: 0x%04x", dvbdmxfeed->pid);
 245        return pvr2_dvb_ctrl_feed(dvbdmxfeed, 0);
 246}
 247
 248static int pvr2_dvb_bus_ctrl(struct dvb_frontend *fe, int acquire)
 249{
 250        struct pvr2_dvb_adapter *adap = fe->dvb->priv;
 251        return pvr2_channel_limit_inputs(
 252            &adap->channel,
 253            (acquire ? (1 << PVR2_CVAL_INPUT_DTV) : 0));
 254}
 255
 256static int pvr2_dvb_adapter_init(struct pvr2_dvb_adapter *adap)
 257{
 258        int ret;
 259
 260        ret = dvb_register_adapter(&adap->dvb_adap, "pvrusb2-dvb",
 261                                   THIS_MODULE/*&hdw->usb_dev->owner*/,
 262                                   &adap->channel.hdw->usb_dev->dev,
 263                                   adapter_nr);
 264        if (ret < 0) {
 265                pvr2_trace(PVR2_TRACE_ERROR_LEGS,
 266                           "dvb_register_adapter failed: error %d", ret);
 267                goto err;
 268        }
 269        adap->dvb_adap.priv = adap;
 270
 271        adap->demux.dmx.capabilities = DMX_TS_FILTERING |
 272                                       DMX_SECTION_FILTERING |
 273                                       DMX_MEMORY_BASED_FILTERING;
 274        adap->demux.priv             = adap;
 275        adap->demux.filternum        = 256;
 276        adap->demux.feednum          = 256;
 277        adap->demux.start_feed       = pvr2_dvb_start_feed;
 278        adap->demux.stop_feed        = pvr2_dvb_stop_feed;
 279        adap->demux.write_to_decoder = NULL;
 280
 281        ret = dvb_dmx_init(&adap->demux);
 282        if (ret < 0) {
 283                pvr2_trace(PVR2_TRACE_ERROR_LEGS,
 284                           "dvb_dmx_init failed: error %d", ret);
 285                goto err_dmx;
 286        }
 287
 288        adap->dmxdev.filternum       = adap->demux.filternum;
 289        adap->dmxdev.demux           = &adap->demux.dmx;
 290        adap->dmxdev.capabilities    = 0;
 291
 292        ret = dvb_dmxdev_init(&adap->dmxdev, &adap->dvb_adap);
 293        if (ret < 0) {
 294                pvr2_trace(PVR2_TRACE_ERROR_LEGS,
 295                           "dvb_dmxdev_init failed: error %d", ret);
 296                goto err_dmx_dev;
 297        }
 298
 299        dvb_net_init(&adap->dvb_adap, &adap->dvb_net, &adap->demux.dmx);
 300
 301        return 0;
 302
 303err_dmx_dev:
 304        dvb_dmx_release(&adap->demux);
 305err_dmx:
 306        dvb_unregister_adapter(&adap->dvb_adap);
 307err:
 308        return ret;
 309}
 310
 311static int pvr2_dvb_adapter_exit(struct pvr2_dvb_adapter *adap)
 312{
 313        pvr2_trace(PVR2_TRACE_INFO, "unregistering DVB devices");
 314        dvb_net_release(&adap->dvb_net);
 315        adap->demux.dmx.close(&adap->demux.dmx);
 316        dvb_dmxdev_release(&adap->dmxdev);
 317        dvb_dmx_release(&adap->demux);
 318        dvb_unregister_adapter(&adap->dvb_adap);
 319        return 0;
 320}
 321
 322static int pvr2_dvb_frontend_init(struct pvr2_dvb_adapter *adap)
 323{
 324        struct pvr2_hdw *hdw = adap->channel.hdw;
 325        const struct pvr2_dvb_props *dvb_props = hdw->hdw_desc->dvb_props;
 326        int ret = 0;
 327
 328        if (dvb_props == NULL) {
 329                pvr2_trace(PVR2_TRACE_ERROR_LEGS, "fe_props not defined!");
 330                return -EINVAL;
 331        }
 332
 333        ret = pvr2_channel_limit_inputs(
 334            &adap->channel,
 335            (1 << PVR2_CVAL_INPUT_DTV));
 336        if (ret) {
 337                pvr2_trace(PVR2_TRACE_ERROR_LEGS,
 338                           "failed to grab control of dtv input (code=%d)",
 339                    ret);
 340                return ret;
 341        }
 342
 343        if (dvb_props->frontend_attach == NULL) {
 344                pvr2_trace(PVR2_TRACE_ERROR_LEGS,
 345                           "frontend_attach not defined!");
 346                ret = -EINVAL;
 347                goto done;
 348        }
 349
 350        if ((dvb_props->frontend_attach(adap) == 0) && (adap->fe)) {
 351
 352                if (dvb_register_frontend(&adap->dvb_adap, adap->fe)) {
 353                        pvr2_trace(PVR2_TRACE_ERROR_LEGS,
 354                                   "frontend registration failed!");
 355                        dvb_frontend_detach(adap->fe);
 356                        adap->fe = NULL;
 357                        ret = -ENODEV;
 358                        goto done;
 359                }
 360
 361                if (dvb_props->tuner_attach)
 362                        dvb_props->tuner_attach(adap);
 363
 364                if (adap->fe->ops.analog_ops.standby)
 365                        adap->fe->ops.analog_ops.standby(adap->fe);
 366
 367                /* Ensure all frontends negotiate bus access */
 368                adap->fe->ops.ts_bus_ctrl = pvr2_dvb_bus_ctrl;
 369
 370        } else {
 371                pvr2_trace(PVR2_TRACE_ERROR_LEGS,
 372                           "no frontend was attached!");
 373                ret = -ENODEV;
 374                return ret;
 375        }
 376
 377 done:
 378        pvr2_channel_limit_inputs(&adap->channel, 0);
 379        return ret;
 380}
 381
 382static int pvr2_dvb_frontend_exit(struct pvr2_dvb_adapter *adap)
 383{
 384        if (adap->fe != NULL) {
 385                dvb_unregister_frontend(adap->fe);
 386                dvb_frontend_detach(adap->fe);
 387        }
 388        return 0;
 389}
 390
 391static void pvr2_dvb_destroy(struct pvr2_dvb_adapter *adap)
 392{
 393        pvr2_dvb_stream_end(adap);
 394        pvr2_dvb_frontend_exit(adap);
 395        pvr2_dvb_adapter_exit(adap);
 396        pvr2_channel_done(&adap->channel);
 397        kfree(adap);
 398}
 399
 400static void pvr2_dvb_internal_check(struct pvr2_channel *chp)
 401{
 402        struct pvr2_dvb_adapter *adap;
 403        adap = container_of(chp, struct pvr2_dvb_adapter, channel);
 404        if (!adap->channel.mc_head->disconnect_flag) return;
 405        pvr2_dvb_destroy(adap);
 406}
 407
 408struct pvr2_dvb_adapter *pvr2_dvb_create(struct pvr2_context *pvr)
 409{
 410        int ret = 0;
 411        struct pvr2_dvb_adapter *adap;
 412        if (!pvr->hdw->hdw_desc->dvb_props) {
 413                /* Device lacks a digital interface so don't set up
 414                   the DVB side of the driver either.  For now. */
 415                return NULL;
 416        }
 417        adap = kzalloc(sizeof(*adap), GFP_KERNEL);
 418        if (!adap) return adap;
 419        pvr2_channel_init(&adap->channel, pvr);
 420        adap->channel.check_func = pvr2_dvb_internal_check;
 421        init_waitqueue_head(&adap->buffer_wait_data);
 422        mutex_init(&adap->lock);
 423        ret = pvr2_dvb_adapter_init(adap);
 424        if (ret < 0) goto fail1;
 425        ret = pvr2_dvb_frontend_init(adap);
 426        if (ret < 0) goto fail2;
 427        return adap;
 428
 429fail2:
 430        pvr2_dvb_adapter_exit(adap);
 431fail1:
 432        pvr2_channel_done(&adap->channel);
 433        return NULL;
 434}
 435
 436