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