[net-next-2.6.git] / drivers / media / video / pvrusb2 / pvrusb2-dvb.c

/*
 *  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 <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;
}
Commit	Line	Data
04910bdc MK	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
d8abe97d	21	#include <linux/kthread.h>
bb8ce9d9	22	#include <linux/freezer.h>
5a0e3ad6	23	#include <linux/slab.h>
27ac792c	24	#include <linux/mm.h>
04910bdc	25	#include "dvbdev.h"
485fcaed	26	#include "pvrusb2-debug.h"
04910bdc MK	27	#include "pvrusb2-hdw-internal.h"
04910bdc MK	28	#include "pvrusb2-hdw.h"
bb8ce9d9	29	#include "pvrusb2-io.h"
04910bdc MK	30	#include "pvrusb2-dvb.h"
	31
	32	DVB_DEFINE_MOD_OPT_ADAPTER_NR(adapter_nr);
	33
ceb4340d	34	static int pvr2_dvb_feed_func(struct pvr2_dvb_adapter *adap)
bb8ce9d9	35	{
bb8ce9d9 MI	36	int ret;
	37	unsigned int count;
	38	struct pvr2_buffer *bp;
a36416d0	39	struct pvr2_stream *stream;
bb8ce9d9	40
485fcaed	41	pvr2_trace(PVR2_TRACE_DVB_FEED, "dvb feed thread started");
bb8ce9d9 MI	42	set_freezable();
bb8ce9d9 MI	43
a36416d0	44	stream = adap->channel.stream->stream;
bb8ce9d9 MI	45
bb8ce9d9 MI	46	for (;;) {
a36416d0	47	if (kthread_should_stop()) break;
bb8ce9d9 MI	48
	49	/* Not sure about this... */
	50	try_to_freeze();
	51
a36416d0	52	bp = pvr2_stream_get_ready_buffer(stream);
bb8ce9d9 MI	53	if (bp != NULL) {
	54	count = pvr2_buffer_get_count(bp);
	55	if (count) {
	56	dvb_dmx_swfilter(
	57	&adap->demux,
a36416d0 MI	58	adap->buffer_storage[
a36416d0 MI	59	pvr2_buffer_get_id(bp)],
bb8ce9d9 MI	60	count);
	61	} else {
	62	ret = pvr2_buffer_get_status(bp);
a36416d0	63	if (ret < 0) break;
bb8ce9d9 MI	64	}
bb8ce9d9 MI	65	ret = pvr2_buffer_queue(bp);
a36416d0	66	if (ret < 0) break;
bb8ce9d9 MI	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
07b80264 MI	76	/* Wait until more buffers become available or we're
07b80264 MI	77	told not to wait any longer. */
bb8ce9d9	78	ret = wait_event_interruptible(
a36416d0	79	adap->buffer_wait_data,
07b80264	80	(pvr2_stream_get_ready_count(stream) > 0) \|\|
13e027a8	81	kthread_should_stop());
a36416d0	82	if (ret < 0) break;
bb8ce9d9 MI	83	}
bb8ce9d9 MI	84
bb8ce9d9 MI	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
485fcaed	88	pvr2_trace(PVR2_TRACE_DVB_FEED, "dvb feed thread stopped");
bb8ce9d9	89
ceb4340d MI	90	return 0;
	91	}
	92
	93	static int pvr2_dvb_feed_thread(void *data)
	94	{
	95	int stat = pvr2_dvb_feed_func(data);
bb8ce9d9 MI	96	/* from videobuf-dvb.c: */
	97	while (!kthread_should_stop()) {
	98	set_current_state(TASK_INTERRUPTIBLE);
	99	schedule();
	100	}
ceb4340d	101	return stat;
bb8ce9d9 MI	102	}
bb8ce9d9 MI	103
a36416d0	104	static void pvr2_dvb_notify(struct pvr2_dvb_adapter *adap)
d8abe97d	105	{
a36416d0 MI	106	wake_up(&adap->buffer_wait_data);
a36416d0 MI	107	}
d8abe97d	108
a36416d0 MI	109	static void pvr2_dvb_stream_end(struct pvr2_dvb_adapter *adap)
	110	{
	111	unsigned int idx;
	112	struct pvr2_stream *stream;
d8abe97d	113
a36416d0 MI	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]);
a6a3a17b	136	adap->buffer_storage[idx] = NULL;
a36416d0 MI	137	}
	138	adap->stream_run = 0;
	139	}
	140	}
	141
	142	static 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;
a6a3a17b	148	struct pvr2_stream *stream = NULL;
a36416d0 MI	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;
d8abe97d	169
a36416d0 MI	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
a6a3a17b	180	while ((bp = pvr2_stream_get_idle_buffer(stream)) != NULL) {
a36416d0 MI	181	ret = pvr2_buffer_queue(bp);
	182	if (ret < 0) return ret;
	183	}
d8abe97d	184
a36416d0 MI	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);
d8abe97d	189	adap->thread = NULL;
a36416d0	190	return ret;
d8abe97d MK	191	}
d8abe97d MK	192
a36416d0	193	adap->stream_run = !0;
d8abe97d	194
a36416d0 MI	195	return 0;
a36416d0 MI	196	}
d8abe97d	197
a36416d0 MI	198	static 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	}
d8abe97d	204
a36416d0 MI	205	static 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) {
485fcaed MK	216	pvr2_trace(PVR2_TRACE_DVB_FEED,
485fcaed MK	217	"start feeding demux");
a36416d0 MI	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) {
485fcaed MK	225	pvr2_trace(PVR2_TRACE_DVB_FEED,
485fcaed MK	226	"stop feeding demux");
a36416d0 MI	227	pvr2_dvb_stream_end(adap);
a36416d0 MI	228	}
d8abe97d	229	}
a36416d0	230	} while (0);
d8abe97d MK	231	mutex_unlock(&adap->lock);
	232
	233	return ret;
	234	}
	235
04910bdc MK	236	static int pvr2_dvb_start_feed(struct dvb_demux_feed *dvbdmxfeed)
04910bdc MK	237	{
485fcaed	238	pvr2_trace(PVR2_TRACE_DVB_FEED, "start pid: 0x%04x", dvbdmxfeed->pid);
d8abe97d	239	return pvr2_dvb_ctrl_feed(dvbdmxfeed, 1);
04910bdc MK	240	}
	241
	242	static int pvr2_dvb_stop_feed(struct dvb_demux_feed *dvbdmxfeed)
	243	{
485fcaed	244	pvr2_trace(PVR2_TRACE_DVB_FEED, "stop pid: 0x%04x", dvbdmxfeed->pid);
d8abe97d	245	return pvr2_dvb_ctrl_feed(dvbdmxfeed, 0);
04910bdc MK	246	}
04910bdc MK	247
99443ae0 MK	248	static int pvr2_dvb_bus_ctrl(struct dvb_frontend *fe, int acquire)
99443ae0 MK	249	{
1cb03b76 MI	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));
99443ae0 MK	254	}
99443ae0 MK	255
04910bdc MK	256	static 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/,
a36416d0	262	&adap->channel.hdw->usb_dev->dev,
04910bdc MK	263	adapter_nr);
04910bdc MK	264	if (ret < 0) {
485fcaed MK	265	pvr2_trace(PVR2_TRACE_ERROR_LEGS,
485fcaed MK	266	"dvb_register_adapter failed: error %d", ret);
04910bdc MK	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) {
485fcaed MK	283	pvr2_trace(PVR2_TRACE_ERROR_LEGS,
485fcaed MK	284	"dvb_dmx_init failed: error %d", ret);
04910bdc MK	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) {
485fcaed MK	294	pvr2_trace(PVR2_TRACE_ERROR_LEGS,
485fcaed MK	295	"dvb_dmxdev_init failed: error %d", ret);
04910bdc MK	296	goto err_dmx_dev;
	297	}
	298
	299	dvb_net_init(&adap->dvb_adap, &adap->dvb_net, &adap->demux.dmx);
	300
04910bdc MK	301	return 0;
	302
	303	err_dmx_dev:
	304	dvb_dmx_release(&adap->demux);
	305	err_dmx:
	306	dvb_unregister_adapter(&adap->dvb_adap);
	307	err:
	308	return ret;
	309	}
	310
	311	static int pvr2_dvb_adapter_exit(struct pvr2_dvb_adapter *adap)
	312	{
485fcaed	313	pvr2_trace(PVR2_TRACE_INFO, "unregistering DVB devices");
7dcc48fb MI	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);
04910bdc MK	319	return 0;
	320	}
	321
	322	static int pvr2_dvb_frontend_init(struct pvr2_dvb_adapter *adap)
	323	{
a36416d0	324	struct pvr2_hdw *hdw = adap->channel.hdw;
69ea3c1c	325	const struct pvr2_dvb_props *dvb_props = hdw->hdw_desc->dvb_props;
1cb03b76	326	int ret = 0;
04910bdc MK	327
04910bdc MK	328	if (dvb_props == NULL) {
485fcaed	329	pvr2_trace(PVR2_TRACE_ERROR_LEGS, "fe_props not defined!");
04910bdc MK	330	return -EINVAL;
	331	}
	332
1cb03b76 MI	333	ret = pvr2_channel_limit_inputs(
	334	&adap->channel,
	335	(1 << PVR2_CVAL_INPUT_DTV));
	336	if (ret) {
485fcaed MK	337	pvr2_trace(PVR2_TRACE_ERROR_LEGS,
485fcaed MK	338	"failed to grab control of dtv input (code=%d)",
1cb03b76	339	ret);
bde316a4	340	return ret;
1cb03b76	341	}
bde316a4	342
04910bdc	343	if (dvb_props->frontend_attach == NULL) {
485fcaed MK	344	pvr2_trace(PVR2_TRACE_ERROR_LEGS,
485fcaed MK	345	"frontend_attach not defined!");
1cb03b76 MI	346	ret = -EINVAL;
1cb03b76 MI	347	goto done;
04910bdc MK	348	}
	349
	350	if ((dvb_props->frontend_attach(adap) == 0) && (adap->fe)) {
	351
	352	if (dvb_register_frontend(&adap->dvb_adap, adap->fe)) {
485fcaed MK	353	pvr2_trace(PVR2_TRACE_ERROR_LEGS,
485fcaed MK	354	"frontend registration failed!");
04910bdc MK	355	dvb_frontend_detach(adap->fe);
04910bdc MK	356	adap->fe = NULL;
1cb03b76 MI	357	ret = -ENODEV;
1cb03b76 MI	358	goto done;
04910bdc MK	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
99443ae0 MK	367	/* Ensure all frontends negotiate bus access */
	368	adap->fe->ops.ts_bus_ctrl = pvr2_dvb_bus_ctrl;
	369
04910bdc	370	} else {
485fcaed MK	371	pvr2_trace(PVR2_TRACE_ERROR_LEGS,
485fcaed MK	372	"no frontend was attached!");
1cb03b76 MI	373	ret = -ENODEV;
1cb03b76 MI	374	return ret;
04910bdc MK	375	}
04910bdc MK	376
1cb03b76 MI	377	done:
	378	pvr2_channel_limit_inputs(&adap->channel, 0);
	379	return ret;
04910bdc MK	380	}
	381
	382	static 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
c5317b17	391	static void pvr2_dvb_destroy(struct pvr2_dvb_adapter *adap)
891d99ef MI	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);
c5317b17	397	kfree(adap);
891d99ef MI	398	}
	399
	400	static 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;
c5317b17	405	pvr2_dvb_destroy(adap);
891d99ef MI	406	}
891d99ef MI	407
c5317b17	408	struct pvr2_dvb_adapter pvr2_dvb_create(struct pvr2_context pvr)
04910bdc MK	409	{
04910bdc MK	410	int ret = 0;
a36416d0	411	struct pvr2_dvb_adapter *adap;
129a2f5e MI	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. */
e6d11865	415	return NULL;
129a2f5e	416	}
c5317b17 MI	417	adap = kzalloc(sizeof(*adap), GFP_KERNEL);
c5317b17 MI	418	if (!adap) return adap;
a36416d0	419	pvr2_channel_init(&adap->channel, pvr);
891d99ef	420	adap->channel.check_func = pvr2_dvb_internal_check;
a36416d0	421	init_waitqueue_head(&adap->buffer_wait_data);
c5317b17 MI	422	mutex_init(&adap->lock);
c5317b17 MI	423	ret = pvr2_dvb_adapter_init(adap);
129a2f5e	424	if (ret < 0) goto fail1;
c5317b17	425	ret = pvr2_dvb_frontend_init(adap);
129a2f5e	426	if (ret < 0) goto fail2;
c5317b17	427	return adap;
129a2f5e MI	428
	429	fail2:
	430	pvr2_dvb_adapter_exit(adap);
	431	fail1:
891d99ef	432	pvr2_channel_done(&adap->channel);
c5317b17	433	return NULL;
04910bdc MK	434	}
04910bdc MK	435