| /* | 
 |  *  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; | 
 | } | 
 |  |