pvrusb2-dvb.c

Go to the documentation of this file.

/*
 *  pvrusb2-dvb.c - linux-dvb api interface to the pvrusb2 driver.
 *
 *  Copyright (C) 2007, 2008 Michael Krufky <[email protected]>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */

#include <linux/kthread.h>
#include <linux/freezer.h>
#include <linux/slab.h>
#include <linux/mm.h>
#include "dvbdev.h"
#include "pvrusb2-debug.h"
#include "pvrusb2-hdw-internal.h"
#include "pvrusb2-hdw.h"
#include "pvrusb2-io.h"
#include "pvrusb2-dvb.h"

DVB_DEFINE_MOD_OPT_ADAPTER_NR(adapter_nr);

static int pvr2_dvb_feed_func(struct pvr2_dvb_adapter *adap)
{
    int ret;
    unsigned int count;
    struct pvr2_buffer *bp;
    struct pvr2_stream *stream;

    pvr2_trace(PVR2_TRACE_DVB_FEED, "dvb feed thread started");
    set_freezable();

    stream = adap->channel.stream->stream;

    for (;;) {
        if (kthread_should_stop()) break;

        /* Not sure about this... */
        try_to_freeze();

        bp = pvr2_stream_get_ready_buffer(stream);
        if (bp != NULL) {
            count = pvr2_buffer_get_count(bp);
            if (count) {
                dvb_dmx_swfilter(
                    &adap->demux,
                    adap->buffer_storage[
                        pvr2_buffer_get_id(bp)],
                    count);
            } else {
                ret = pvr2_buffer_get_status(bp);
                if (ret < 0) break;
            }
            ret = pvr2_buffer_queue(bp);
            if (ret < 0) break;

            /* Since we know we did something to a buffer,
               just go back and try again.  No point in
               blocking unless we really ran out of
               buffers to process. */
            continue;
        }


        /* Wait until more buffers become available or we're
           told not to wait any longer. */
        ret = wait_event_interruptible(
            adap->buffer_wait_data,
            (pvr2_stream_get_ready_count(stream) > 0) ||
            kthread_should_stop());
        if (ret < 0) break;
    }

    /* If we get here and ret is < 0, then an error has occurred.
       Probably would be a good idea to communicate that to DVB core... */

    pvr2_trace(PVR2_TRACE_DVB_FEED, "dvb feed thread stopped");

    return 0;
}

static int pvr2_dvb_feed_thread(void *data)
{
    int stat = pvr2_dvb_feed_func(data);
    /* from videobuf-dvb.c: */
    while (!kthread_should_stop()) {
        set_current_state(TASK_INTERRUPTIBLE);
        schedule();
    }
    return stat;
}

static void pvr2_dvb_notify(struct pvr2_dvb_adapter *adap)
{
    wake_up(&adap->buffer_wait_data);
}

static void pvr2_dvb_stream_end(struct pvr2_dvb_adapter *adap)
{
    unsigned int idx;
    struct pvr2_stream *stream;

    if (adap->thread) {
        kthread_stop(adap->thread);
        adap->thread = NULL;
    }

    if (adap->channel.stream) {
        stream = adap->channel.stream->stream;
    } else {
        stream = NULL;
    }
    if (stream) {
        pvr2_hdw_set_streaming(adap->channel.hdw, 0);
        pvr2_stream_set_callback(stream, NULL, NULL);
        pvr2_stream_kill(stream);
        pvr2_stream_set_buffer_count(stream, 0);
        pvr2_channel_claim_stream(&adap->channel, NULL);
    }

    if (adap->stream_run) {
        for (idx = 0; idx < PVR2_DVB_BUFFER_COUNT; idx++) {
            if (!(adap->buffer_storage[idx])) continue;
            kfree(adap->buffer_storage[idx]);
            adap->buffer_storage[idx] = NULL;
        }
        adap->stream_run = 0;
    }
}

static int pvr2_dvb_stream_do_start(struct pvr2_dvb_adapter *adap)
{
    struct pvr2_context *pvr = adap->channel.mc_head;
    unsigned int idx;
    int ret;
    struct pvr2_buffer *bp;
    struct pvr2_stream *stream = NULL;

    if (adap->stream_run) return -EIO;

    ret = pvr2_channel_claim_stream(&adap->channel, &pvr->video_stream);
    /* somebody else already has the stream */
    if (ret < 0) return ret;

    stream = adap->channel.stream->stream;

    for (idx = 0; idx < PVR2_DVB_BUFFER_COUNT; idx++) {
        adap->buffer_storage[idx] = kmalloc(PVR2_DVB_BUFFER_SIZE,
                            GFP_KERNEL);
        if (!(adap->buffer_storage[idx])) return -ENOMEM;
    }

    pvr2_stream_set_callback(pvr->video_stream.stream,
                 (pvr2_stream_callback) pvr2_dvb_notify, adap);

    ret = pvr2_stream_set_buffer_count(stream, PVR2_DVB_BUFFER_COUNT);
    if (ret < 0) return ret;

    for (idx = 0; idx < PVR2_DVB_BUFFER_COUNT; idx++) {
        bp = pvr2_stream_get_buffer(stream, idx);
        pvr2_buffer_set_buffer(bp,
                       adap->buffer_storage[idx],
                       PVR2_DVB_BUFFER_SIZE);
    }

    ret = pvr2_hdw_set_streaming(adap->channel.hdw, 1);
    if (ret < 0) return ret;

    while ((bp = pvr2_stream_get_idle_buffer(stream)) != NULL) {
        ret = pvr2_buffer_queue(bp);
        if (ret < 0) return ret;
    }

    adap->thread = kthread_run(pvr2_dvb_feed_thread, adap, "pvrusb2-dvb");

    if (IS_ERR(adap->thread)) {
        ret = PTR_ERR(adap->thread);
        adap->thread = NULL;
        return ret;
    }

    adap->stream_run = !0;

    return 0;
}

static int pvr2_dvb_stream_start(struct pvr2_dvb_adapter *adap)
{
    int ret = pvr2_dvb_stream_do_start(adap);
    if (ret < 0) pvr2_dvb_stream_end(adap);
    return ret;
}

static int pvr2_dvb_ctrl_feed(struct dvb_demux_feed *dvbdmxfeed, int onoff)
{
    struct pvr2_dvb_adapter *adap = dvbdmxfeed->demux->priv;
    int ret = 0;

    if (adap == NULL) return -ENODEV;

    mutex_lock(&adap->lock);
    do {
        if (onoff) {
            if (!adap->feedcount) {
                pvr2_trace(PVR2_TRACE_DVB_FEED,
                       "start feeding demux");
                ret = pvr2_dvb_stream_start(adap);
                if (ret < 0) break;
            }
            (adap->feedcount)++;
        } else if (adap->feedcount > 0) {
            (adap->feedcount)--;
            if (!adap->feedcount) {
                pvr2_trace(PVR2_TRACE_DVB_FEED,
                       "stop feeding demux");
                pvr2_dvb_stream_end(adap);
            }
        }
    } while (0);
    mutex_unlock(&adap->lock);

    return ret;
}

static int pvr2_dvb_start_feed(struct dvb_demux_feed *dvbdmxfeed)
{
    pvr2_trace(PVR2_TRACE_DVB_FEED, "start pid: 0x%04x", dvbdmxfeed->pid);
    return pvr2_dvb_ctrl_feed(dvbdmxfeed, 1);
}

static int pvr2_dvb_stop_feed(struct dvb_demux_feed *dvbdmxfeed)
{
    pvr2_trace(PVR2_TRACE_DVB_FEED, "stop pid: 0x%04x", dvbdmxfeed->pid);
    return pvr2_dvb_ctrl_feed(dvbdmxfeed, 0);
}

static int pvr2_dvb_bus_ctrl(struct dvb_frontend *fe, int acquire)
{
    struct pvr2_dvb_adapter *adap = fe->dvb->priv;
    return pvr2_channel_limit_inputs(
        &adap->channel,
        (acquire ? (1 << PVR2_CVAL_INPUT_DTV) : 0));
}

static int pvr2_dvb_adapter_init(struct pvr2_dvb_adapter *adap)
{
    int ret;

    ret = dvb_register_adapter(&adap->dvb_adap, "pvrusb2-dvb",
                   THIS_MODULE/*&hdw->usb_dev->owner*/,
                   &adap->channel.hdw->usb_dev->dev,
                   adapter_nr);
    if (ret < 0) {
        pvr2_trace(PVR2_TRACE_ERROR_LEGS,
               "dvb_register_adapter failed: error %d", ret);
        goto err;
    }
    adap->dvb_adap.priv = adap;

    adap->demux.dmx.capabilities = DMX_TS_FILTERING |
                       DMX_SECTION_FILTERING |
                       DMX_MEMORY_BASED_FILTERING;
    adap->demux.priv             = adap;
    adap->demux.filternum        = 256;
    adap->demux.feednum          = 256;
    adap->demux.start_feed       = pvr2_dvb_start_feed;
    adap->demux.stop_feed        = pvr2_dvb_stop_feed;
    adap->demux.write_to_decoder = NULL;

    ret = dvb_dmx_init(&adap->demux);
    if (ret < 0) {
        pvr2_trace(PVR2_TRACE_ERROR_LEGS,
               "dvb_dmx_init failed: error %d", ret);
        goto err_dmx;
    }

    adap->dmxdev.filternum       = adap->demux.filternum;
    adap->dmxdev.demux           = &adap->demux.dmx;
    adap->dmxdev.capabilities    = 0;

    ret = dvb_dmxdev_init(&adap->dmxdev, &adap->dvb_adap);
    if (ret < 0) {
        pvr2_trace(PVR2_TRACE_ERROR_LEGS,
               "dvb_dmxdev_init failed: error %d", ret);
        goto err_dmx_dev;
    }

    dvb_net_init(&adap->dvb_adap, &adap->dvb_net, &adap->demux.dmx);

    return 0;

err_dmx_dev:
    dvb_dmx_release(&adap->demux);
err_dmx:
    dvb_unregister_adapter(&adap->dvb_adap);
err:
    return ret;
}

static int pvr2_dvb_adapter_exit(struct pvr2_dvb_adapter *adap)
{
    pvr2_trace(PVR2_TRACE_INFO, "unregistering DVB devices");
    dvb_net_release(&adap->dvb_net);
    adap->demux.dmx.close(&adap->demux.dmx);
    dvb_dmxdev_release(&adap->dmxdev);
    dvb_dmx_release(&adap->demux);
    dvb_unregister_adapter(&adap->dvb_adap);
    return 0;
}

static int pvr2_dvb_frontend_init(struct pvr2_dvb_adapter *adap)
{
    struct pvr2_hdw *hdw = adap->channel.hdw;
    const struct pvr2_dvb_props *dvb_props = hdw->hdw_desc->dvb_props;
    int ret = 0;

    if (dvb_props == NULL) {
        pvr2_trace(PVR2_TRACE_ERROR_LEGS, "fe_props not defined!");
        return -EINVAL;
    }

    ret = pvr2_channel_limit_inputs(
        &adap->channel,
        (1 << PVR2_CVAL_INPUT_DTV));
    if (ret) {
        pvr2_trace(PVR2_TRACE_ERROR_LEGS,
               "failed to grab control of dtv input (code=%d)",
            ret);
        return ret;
    }

    if (dvb_props->frontend_attach == NULL) {
        pvr2_trace(PVR2_TRACE_ERROR_LEGS,
               "frontend_attach not defined!");
        ret = -EINVAL;
        goto done;
    }

    if ((dvb_props->frontend_attach(adap) == 0) && (adap->fe)) {

        if (dvb_register_frontend(&adap->dvb_adap, adap->fe)) {
            pvr2_trace(PVR2_TRACE_ERROR_LEGS,
                   "frontend registration failed!");
            dvb_frontend_detach(adap->fe);
            adap->fe = NULL;
            ret = -ENODEV;
            goto done;
        }

        if (dvb_props->tuner_attach)
            dvb_props->tuner_attach(adap);

        if (adap->fe->ops.analog_ops.standby)
            adap->fe->ops.analog_ops.standby(adap->fe);

        /* Ensure all frontends negotiate bus access */
        adap->fe->ops.ts_bus_ctrl = pvr2_dvb_bus_ctrl;

    } else {
        pvr2_trace(PVR2_TRACE_ERROR_LEGS,
               "no frontend was attached!");
        ret = -ENODEV;
        return ret;
    }

 done:
    pvr2_channel_limit_inputs(&adap->channel, 0);
    return ret;
}

static int pvr2_dvb_frontend_exit(struct pvr2_dvb_adapter *adap)
{
    if (adap->fe != NULL) {
        dvb_unregister_frontend(adap->fe);
        dvb_frontend_detach(adap->fe);
    }
    return 0;
}

static void pvr2_dvb_destroy(struct pvr2_dvb_adapter *adap)
{
    pvr2_dvb_stream_end(adap);
    pvr2_dvb_frontend_exit(adap);
    pvr2_dvb_adapter_exit(adap);
    pvr2_channel_done(&adap->channel);
    kfree(adap);
}

static void pvr2_dvb_internal_check(struct pvr2_channel *chp)
{
    struct pvr2_dvb_adapter *adap;
    adap = container_of(chp, struct pvr2_dvb_adapter, channel);
    if (!adap->channel.mc_head->disconnect_flag) return;
    pvr2_dvb_destroy(adap);
}

struct pvr2_dvb_adapter *pvr2_dvb_create(struct pvr2_context *pvr)
{
    int ret = 0;
    struct pvr2_dvb_adapter *adap;
    if (!pvr->hdw->hdw_desc->dvb_props) {
        /* Device lacks a digital interface so don't set up
           the DVB side of the driver either.  For now. */
        return NULL;
    }
    adap = kzalloc(sizeof(*adap), GFP_KERNEL);
    if (!adap) return adap;
    pvr2_channel_init(&adap->channel, pvr);
    adap->channel.check_func = pvr2_dvb_internal_check;
    init_waitqueue_head(&adap->buffer_wait_data);
    mutex_init(&adap->lock);
    ret = pvr2_dvb_adapter_init(adap);
    if (ret < 0) goto fail1;
    ret = pvr2_dvb_frontend_init(adap);
    if (ret < 0) goto fail2;
    return adap;

fail2:
    pvr2_dvb_adapter_exit(adap);
fail1:
    pvr2_channel_done(&adap->channel);
    return NULL;
}