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" |
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(); |
43 | ||
a36416d0 | 44 | stream = adap->channel.stream->stream; |
bb8ce9d9 MI |
45 | |
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[ |
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 | } |
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 |
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 | } |
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 | } |
103 | ||
a36416d0 | 104 | static void pvr2_dvb_notify(struct pvr2_dvb_adapter *adap) |
d8abe97d | 105 | { |
a36416d0 MI |
106 | wake_up(&adap->buffer_wait_data); |
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 | } |
192 | ||
a36416d0 | 193 | adap->stream_run = !0; |
d8abe97d | 194 | |
a36416d0 MI |
195 | return 0; |
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, |
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, |
226 | "stop feeding demux"); | |
a36416d0 MI |
227 | pvr2_dvb_stream_end(adap); |
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) |
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 | } |
247 | ||
99443ae0 MK |
248 | static int pvr2_dvb_bus_ctrl(struct dvb_frontend *fe, int acquire) |
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 | } |
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); |
264 | if (ret < 0) { | |
485fcaed MK |
265 | pvr2_trace(PVR2_TRACE_ERROR_LEGS, |
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, |
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, |
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 | |
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, |
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, |
345 | "frontend_attach not defined!"); | |
1cb03b76 MI |
346 | ret = -EINVAL; |
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, |
354 | "frontend registration failed!"); | |
04910bdc MK |
355 | dvb_frontend_detach(adap->fe); |
356 | adap->fe = NULL; | |
1cb03b76 MI |
357 | ret = -ENODEV; |
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, |
372 | "no frontend was attached!"); | |
1cb03b76 MI |
373 | ret = -ENODEV; |
374 | return ret; | |
04910bdc MK |
375 | } |
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 | } |
407 | ||
c5317b17 | 408 | struct pvr2_dvb_adapter *pvr2_dvb_create(struct pvr2_context *pvr) |
04910bdc MK |
409 | { |
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); |
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); |
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 | } |
435 |