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