/* * pvrusb2-dvb.c - linux-dvb api interface to the pvrusb2 driver. * * Copyright (C) 2007, 2008 Michael Krufky <mkrufky@linuxtv.org> * * 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 "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; 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; }