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