dmxdev.c 35 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474
  1. // SPDX-License-Identifier: LGPL-2.1-or-later
  2. /*
  3. * dmxdev.c - DVB demultiplexer device
  4. *
  5. * Copyright (C) 2000 Ralph Metzler & Marcus Metzler
  6. * for convergence integrated media GmbH
  7. */
  8. #define pr_fmt(fmt) "dmxdev: " fmt
  9. #include <linux/sched.h>
  10. #include <linux/spinlock.h>
  11. #include <linux/slab.h>
  12. #include <linux/vmalloc.h>
  13. #include <linux/module.h>
  14. #include <linux/poll.h>
  15. #include <linux/ioctl.h>
  16. #include <linux/wait.h>
  17. #include <linux/uaccess.h>
  18. #include <media/dmxdev.h>
  19. #include <media/dvb_vb2.h>
  20. static int debug;
  21. module_param(debug, int, 0644);
  22. MODULE_PARM_DESC(debug, "Turn on/off debugging (default:off).");
  23. #define dprintk(fmt, arg...) do { \
  24. if (debug) \
  25. printk(KERN_DEBUG pr_fmt("%s: " fmt), \
  26. __func__, ##arg); \
  27. } while (0)
  28. static int dvb_dmxdev_buffer_write(struct dvb_ringbuffer *buf,
  29. const u8 *src, size_t len)
  30. {
  31. ssize_t free;
  32. if (!len)
  33. return 0;
  34. if (!buf->data)
  35. return 0;
  36. free = dvb_ringbuffer_free(buf);
  37. if (len > free) {
  38. dprintk("buffer overflow\n");
  39. return -EOVERFLOW;
  40. }
  41. return dvb_ringbuffer_write(buf, src, len);
  42. }
  43. static ssize_t dvb_dmxdev_buffer_read(struct dvb_ringbuffer *src,
  44. int non_blocking, char __user *buf,
  45. size_t count, loff_t *ppos)
  46. {
  47. size_t todo;
  48. ssize_t avail;
  49. ssize_t ret = 0;
  50. if (!src->data)
  51. return 0;
  52. if (src->error) {
  53. ret = src->error;
  54. dvb_ringbuffer_flush(src);
  55. return ret;
  56. }
  57. for (todo = count; todo > 0; todo -= ret) {
  58. if (non_blocking && dvb_ringbuffer_empty(src)) {
  59. ret = -EWOULDBLOCK;
  60. break;
  61. }
  62. ret = wait_event_interruptible(src->queue,
  63. !dvb_ringbuffer_empty(src) ||
  64. (src->error != 0));
  65. if (ret < 0)
  66. break;
  67. if (src->error) {
  68. ret = src->error;
  69. dvb_ringbuffer_flush(src);
  70. break;
  71. }
  72. avail = dvb_ringbuffer_avail(src);
  73. if (avail > todo)
  74. avail = todo;
  75. ret = dvb_ringbuffer_read_user(src, buf, avail);
  76. if (ret < 0)
  77. break;
  78. buf += ret;
  79. }
  80. return (count - todo) ? (count - todo) : ret;
  81. }
  82. static struct dmx_frontend *get_fe(struct dmx_demux *demux, int type)
  83. {
  84. struct list_head *head, *pos;
  85. head = demux->get_frontends(demux);
  86. if (!head)
  87. return NULL;
  88. list_for_each(pos, head)
  89. if (DMX_FE_ENTRY(pos)->source == type)
  90. return DMX_FE_ENTRY(pos);
  91. return NULL;
  92. }
  93. static int dvb_dvr_open(struct inode *inode, struct file *file)
  94. {
  95. struct dvb_device *dvbdev = file->private_data;
  96. struct dmxdev *dmxdev = dvbdev->priv;
  97. struct dmx_frontend *front;
  98. bool need_ringbuffer = false;
  99. dprintk("%s\n", __func__);
  100. if (mutex_lock_interruptible(&dmxdev->mutex))
  101. return -ERESTARTSYS;
  102. if (dmxdev->exit) {
  103. mutex_unlock(&dmxdev->mutex);
  104. return -ENODEV;
  105. }
  106. dmxdev->may_do_mmap = 0;
  107. /*
  108. * The logic here is a little tricky due to the ifdef.
  109. *
  110. * The ringbuffer is used for both read and mmap.
  111. *
  112. * It is not needed, however, on two situations:
  113. * - Write devices (access with O_WRONLY);
  114. * - For duplex device nodes, opened with O_RDWR.
  115. */
  116. if ((file->f_flags & O_ACCMODE) == O_RDONLY)
  117. need_ringbuffer = true;
  118. else if ((file->f_flags & O_ACCMODE) == O_RDWR) {
  119. if (!(dmxdev->capabilities & DMXDEV_CAP_DUPLEX)) {
  120. #ifdef CONFIG_DVB_MMAP
  121. dmxdev->may_do_mmap = 1;
  122. need_ringbuffer = true;
  123. #else
  124. mutex_unlock(&dmxdev->mutex);
  125. return -EOPNOTSUPP;
  126. #endif
  127. }
  128. }
  129. if (need_ringbuffer) {
  130. void *mem;
  131. if (!dvbdev->readers) {
  132. mutex_unlock(&dmxdev->mutex);
  133. return -EBUSY;
  134. }
  135. mem = vmalloc(DVR_BUFFER_SIZE);
  136. if (!mem) {
  137. mutex_unlock(&dmxdev->mutex);
  138. return -ENOMEM;
  139. }
  140. dmxdev->dvr_buffer.data = mem;
  141. dmxdev->dvr_buffer.size = DVR_BUFFER_SIZE;
  142. dvb_ringbuffer_reset(&dmxdev->dvr_buffer);
  143. if (dmxdev->may_do_mmap)
  144. dvb_vb2_init(&dmxdev->dvr_vb2_ctx, "dvr",
  145. &dmxdev->mutex,
  146. file->f_flags & O_NONBLOCK);
  147. dvbdev->readers--;
  148. }
  149. if ((file->f_flags & O_ACCMODE) == O_WRONLY) {
  150. dmxdev->dvr_orig_fe = dmxdev->demux->frontend;
  151. if (!dmxdev->demux->write) {
  152. mutex_unlock(&dmxdev->mutex);
  153. return -EOPNOTSUPP;
  154. }
  155. front = get_fe(dmxdev->demux, DMX_MEMORY_FE);
  156. if (!front) {
  157. mutex_unlock(&dmxdev->mutex);
  158. return -EINVAL;
  159. }
  160. dmxdev->demux->disconnect_frontend(dmxdev->demux);
  161. dmxdev->demux->connect_frontend(dmxdev->demux, front);
  162. }
  163. dvbdev->users++;
  164. mutex_unlock(&dmxdev->mutex);
  165. return 0;
  166. }
  167. static int dvb_dvr_release(struct inode *inode, struct file *file)
  168. {
  169. struct dvb_device *dvbdev = file->private_data;
  170. struct dmxdev *dmxdev = dvbdev->priv;
  171. mutex_lock(&dmxdev->mutex);
  172. if ((file->f_flags & O_ACCMODE) == O_WRONLY) {
  173. dmxdev->demux->disconnect_frontend(dmxdev->demux);
  174. dmxdev->demux->connect_frontend(dmxdev->demux,
  175. dmxdev->dvr_orig_fe);
  176. }
  177. if (((file->f_flags & O_ACCMODE) == O_RDONLY) ||
  178. dmxdev->may_do_mmap) {
  179. if (dmxdev->may_do_mmap) {
  180. if (dvb_vb2_is_streaming(&dmxdev->dvr_vb2_ctx))
  181. dvb_vb2_stream_off(&dmxdev->dvr_vb2_ctx);
  182. dvb_vb2_release(&dmxdev->dvr_vb2_ctx);
  183. }
  184. dvbdev->readers++;
  185. if (dmxdev->dvr_buffer.data) {
  186. void *mem = dmxdev->dvr_buffer.data;
  187. /*memory barrier*/
  188. mb();
  189. spin_lock_irq(&dmxdev->lock);
  190. dmxdev->dvr_buffer.data = NULL;
  191. spin_unlock_irq(&dmxdev->lock);
  192. vfree(mem);
  193. }
  194. }
  195. /* TODO */
  196. dvbdev->users--;
  197. if (dvbdev->users == 1 && dmxdev->exit == 1) {
  198. mutex_unlock(&dmxdev->mutex);
  199. wake_up(&dvbdev->wait_queue);
  200. } else
  201. mutex_unlock(&dmxdev->mutex);
  202. return 0;
  203. }
  204. static ssize_t dvb_dvr_write(struct file *file, const char __user *buf,
  205. size_t count, loff_t *ppos)
  206. {
  207. struct dvb_device *dvbdev = file->private_data;
  208. struct dmxdev *dmxdev = dvbdev->priv;
  209. int ret;
  210. if (!dmxdev->demux->write)
  211. return -EOPNOTSUPP;
  212. if ((file->f_flags & O_ACCMODE) != O_WRONLY)
  213. return -EINVAL;
  214. if (mutex_lock_interruptible(&dmxdev->mutex))
  215. return -ERESTARTSYS;
  216. if (dmxdev->exit) {
  217. mutex_unlock(&dmxdev->mutex);
  218. return -ENODEV;
  219. }
  220. ret = dmxdev->demux->write(dmxdev->demux, buf, count);
  221. mutex_unlock(&dmxdev->mutex);
  222. return ret;
  223. }
  224. static ssize_t dvb_dvr_read(struct file *file, char __user *buf, size_t count,
  225. loff_t *ppos)
  226. {
  227. struct dvb_device *dvbdev = file->private_data;
  228. struct dmxdev *dmxdev = dvbdev->priv;
  229. if (dmxdev->exit)
  230. return -ENODEV;
  231. return dvb_dmxdev_buffer_read(&dmxdev->dvr_buffer,
  232. file->f_flags & O_NONBLOCK,
  233. buf, count, ppos);
  234. }
  235. static int dvb_dvr_set_buffer_size(struct dmxdev *dmxdev,
  236. unsigned long size)
  237. {
  238. struct dvb_ringbuffer *buf = &dmxdev->dvr_buffer;
  239. void *newmem;
  240. void *oldmem;
  241. dprintk("%s\n", __func__);
  242. if (buf->size == size)
  243. return 0;
  244. if (!size)
  245. return -EINVAL;
  246. newmem = vmalloc(size);
  247. if (!newmem)
  248. return -ENOMEM;
  249. oldmem = buf->data;
  250. spin_lock_irq(&dmxdev->lock);
  251. buf->data = newmem;
  252. buf->size = size;
  253. /* reset and not flush in case the buffer shrinks */
  254. dvb_ringbuffer_reset(buf);
  255. spin_unlock_irq(&dmxdev->lock);
  256. vfree(oldmem);
  257. return 0;
  258. }
  259. static inline void dvb_dmxdev_filter_state_set(struct dmxdev_filter
  260. *dmxdevfilter, int state)
  261. {
  262. spin_lock_irq(&dmxdevfilter->dev->lock);
  263. dmxdevfilter->state = state;
  264. spin_unlock_irq(&dmxdevfilter->dev->lock);
  265. }
  266. static int dvb_dmxdev_set_buffer_size(struct dmxdev_filter *dmxdevfilter,
  267. unsigned long size)
  268. {
  269. struct dvb_ringbuffer *buf = &dmxdevfilter->buffer;
  270. void *newmem;
  271. void *oldmem;
  272. if (buf->size == size)
  273. return 0;
  274. if (!size)
  275. return -EINVAL;
  276. if (dmxdevfilter->state >= DMXDEV_STATE_GO)
  277. return -EBUSY;
  278. newmem = vmalloc(size);
  279. if (!newmem)
  280. return -ENOMEM;
  281. oldmem = buf->data;
  282. spin_lock_irq(&dmxdevfilter->dev->lock);
  283. buf->data = newmem;
  284. buf->size = size;
  285. /* reset and not flush in case the buffer shrinks */
  286. dvb_ringbuffer_reset(buf);
  287. spin_unlock_irq(&dmxdevfilter->dev->lock);
  288. vfree(oldmem);
  289. return 0;
  290. }
  291. static void dvb_dmxdev_filter_timeout(struct timer_list *t)
  292. {
  293. struct dmxdev_filter *dmxdevfilter = timer_container_of(dmxdevfilter,
  294. t, timer);
  295. dmxdevfilter->buffer.error = -ETIMEDOUT;
  296. spin_lock_irq(&dmxdevfilter->dev->lock);
  297. dmxdevfilter->state = DMXDEV_STATE_TIMEDOUT;
  298. spin_unlock_irq(&dmxdevfilter->dev->lock);
  299. wake_up(&dmxdevfilter->buffer.queue);
  300. }
  301. static void dvb_dmxdev_filter_timer(struct dmxdev_filter *dmxdevfilter)
  302. {
  303. struct dmx_sct_filter_params *para = &dmxdevfilter->params.sec;
  304. timer_delete(&dmxdevfilter->timer);
  305. if (para->timeout) {
  306. dmxdevfilter->timer.expires =
  307. jiffies + 1 + (HZ / 2 + HZ * para->timeout) / 1000;
  308. add_timer(&dmxdevfilter->timer);
  309. }
  310. }
  311. static int dvb_dmxdev_section_callback(const u8 *buffer1, size_t buffer1_len,
  312. const u8 *buffer2, size_t buffer2_len,
  313. struct dmx_section_filter *filter,
  314. u32 *buffer_flags)
  315. {
  316. struct dmxdev_filter *dmxdevfilter = filter->priv;
  317. int ret;
  318. if (!dvb_vb2_is_streaming(&dmxdevfilter->vb2_ctx) &&
  319. dmxdevfilter->buffer.error) {
  320. wake_up(&dmxdevfilter->buffer.queue);
  321. return 0;
  322. }
  323. spin_lock(&dmxdevfilter->dev->lock);
  324. if (dmxdevfilter->state != DMXDEV_STATE_GO) {
  325. spin_unlock(&dmxdevfilter->dev->lock);
  326. return 0;
  327. }
  328. timer_delete(&dmxdevfilter->timer);
  329. dprintk("section callback %*ph\n", 6, buffer1);
  330. if (dvb_vb2_is_streaming(&dmxdevfilter->vb2_ctx)) {
  331. ret = dvb_vb2_fill_buffer(&dmxdevfilter->vb2_ctx,
  332. buffer1, buffer1_len,
  333. buffer_flags, true);
  334. if (ret == buffer1_len)
  335. ret = dvb_vb2_fill_buffer(&dmxdevfilter->vb2_ctx,
  336. buffer2, buffer2_len,
  337. buffer_flags, true);
  338. } else {
  339. ret = dvb_dmxdev_buffer_write(&dmxdevfilter->buffer,
  340. buffer1, buffer1_len);
  341. if (ret == buffer1_len) {
  342. ret = dvb_dmxdev_buffer_write(&dmxdevfilter->buffer,
  343. buffer2, buffer2_len);
  344. }
  345. }
  346. if (ret < 0)
  347. dmxdevfilter->buffer.error = ret;
  348. if (dmxdevfilter->params.sec.flags & DMX_ONESHOT)
  349. dmxdevfilter->state = DMXDEV_STATE_DONE;
  350. spin_unlock(&dmxdevfilter->dev->lock);
  351. wake_up(&dmxdevfilter->buffer.queue);
  352. return 0;
  353. }
  354. static int dvb_dmxdev_ts_callback(const u8 *buffer1, size_t buffer1_len,
  355. const u8 *buffer2, size_t buffer2_len,
  356. struct dmx_ts_feed *feed,
  357. u32 *buffer_flags)
  358. {
  359. struct dmxdev_filter *dmxdevfilter = feed->priv;
  360. struct dvb_ringbuffer *buffer;
  361. #ifdef CONFIG_DVB_MMAP
  362. struct dvb_vb2_ctx *ctx;
  363. #endif
  364. int ret;
  365. spin_lock(&dmxdevfilter->dev->lock);
  366. if (dmxdevfilter->params.pes.output == DMX_OUT_DECODER) {
  367. spin_unlock(&dmxdevfilter->dev->lock);
  368. return 0;
  369. }
  370. if (dmxdevfilter->params.pes.output == DMX_OUT_TAP ||
  371. dmxdevfilter->params.pes.output == DMX_OUT_TSDEMUX_TAP) {
  372. buffer = &dmxdevfilter->buffer;
  373. #ifdef CONFIG_DVB_MMAP
  374. ctx = &dmxdevfilter->vb2_ctx;
  375. #endif
  376. } else {
  377. buffer = &dmxdevfilter->dev->dvr_buffer;
  378. #ifdef CONFIG_DVB_MMAP
  379. ctx = &dmxdevfilter->dev->dvr_vb2_ctx;
  380. #endif
  381. }
  382. if (dvb_vb2_is_streaming(ctx)) {
  383. ret = dvb_vb2_fill_buffer(ctx, buffer1, buffer1_len,
  384. buffer_flags, false);
  385. if (ret == buffer1_len)
  386. ret = dvb_vb2_fill_buffer(ctx, buffer2, buffer2_len,
  387. buffer_flags, false);
  388. } else {
  389. if (buffer->error) {
  390. spin_unlock(&dmxdevfilter->dev->lock);
  391. wake_up(&buffer->queue);
  392. return 0;
  393. }
  394. ret = dvb_dmxdev_buffer_write(buffer, buffer1, buffer1_len);
  395. if (ret == buffer1_len)
  396. ret = dvb_dmxdev_buffer_write(buffer,
  397. buffer2, buffer2_len);
  398. }
  399. if (ret < 0)
  400. buffer->error = ret;
  401. spin_unlock(&dmxdevfilter->dev->lock);
  402. wake_up(&buffer->queue);
  403. return 0;
  404. }
  405. /* stop feed but only mark the specified filter as stopped (state set) */
  406. static int dvb_dmxdev_feed_stop(struct dmxdev_filter *dmxdevfilter)
  407. {
  408. struct dmxdev_feed *feed;
  409. dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_SET);
  410. switch (dmxdevfilter->type) {
  411. case DMXDEV_TYPE_SEC:
  412. timer_delete(&dmxdevfilter->timer);
  413. dmxdevfilter->feed.sec->stop_filtering(dmxdevfilter->feed.sec);
  414. break;
  415. case DMXDEV_TYPE_PES:
  416. list_for_each_entry(feed, &dmxdevfilter->feed.ts, next)
  417. feed->ts->stop_filtering(feed->ts);
  418. break;
  419. default:
  420. return -EINVAL;
  421. }
  422. return 0;
  423. }
  424. /* start feed associated with the specified filter */
  425. static int dvb_dmxdev_feed_start(struct dmxdev_filter *filter)
  426. {
  427. struct dmxdev_feed *feed;
  428. int ret;
  429. dvb_dmxdev_filter_state_set(filter, DMXDEV_STATE_GO);
  430. switch (filter->type) {
  431. case DMXDEV_TYPE_SEC:
  432. return filter->feed.sec->start_filtering(filter->feed.sec);
  433. case DMXDEV_TYPE_PES:
  434. list_for_each_entry(feed, &filter->feed.ts, next) {
  435. ret = feed->ts->start_filtering(feed->ts);
  436. if (ret < 0) {
  437. dvb_dmxdev_feed_stop(filter);
  438. return ret;
  439. }
  440. }
  441. break;
  442. default:
  443. return -EINVAL;
  444. }
  445. return 0;
  446. }
  447. /* restart section feed if it has filters left associated with it,
  448. otherwise release the feed */
  449. static int dvb_dmxdev_feed_restart(struct dmxdev_filter *filter)
  450. {
  451. int i;
  452. struct dmxdev *dmxdev = filter->dev;
  453. u16 pid = filter->params.sec.pid;
  454. for (i = 0; i < dmxdev->filternum; i++)
  455. if (dmxdev->filter[i].state >= DMXDEV_STATE_GO &&
  456. dmxdev->filter[i].type == DMXDEV_TYPE_SEC &&
  457. dmxdev->filter[i].params.sec.pid == pid) {
  458. dvb_dmxdev_feed_start(&dmxdev->filter[i]);
  459. return 0;
  460. }
  461. filter->dev->demux->release_section_feed(dmxdev->demux,
  462. filter->feed.sec);
  463. return 0;
  464. }
  465. static int dvb_dmxdev_filter_stop(struct dmxdev_filter *dmxdevfilter)
  466. {
  467. struct dmxdev_feed *feed;
  468. struct dmx_demux *demux;
  469. if (dmxdevfilter->state < DMXDEV_STATE_GO)
  470. return 0;
  471. switch (dmxdevfilter->type) {
  472. case DMXDEV_TYPE_SEC:
  473. if (!dmxdevfilter->feed.sec)
  474. break;
  475. dvb_dmxdev_feed_stop(dmxdevfilter);
  476. if (dmxdevfilter->filter.sec)
  477. dmxdevfilter->feed.sec->
  478. release_filter(dmxdevfilter->feed.sec,
  479. dmxdevfilter->filter.sec);
  480. dvb_dmxdev_feed_restart(dmxdevfilter);
  481. dmxdevfilter->feed.sec = NULL;
  482. break;
  483. case DMXDEV_TYPE_PES:
  484. dvb_dmxdev_feed_stop(dmxdevfilter);
  485. demux = dmxdevfilter->dev->demux;
  486. list_for_each_entry(feed, &dmxdevfilter->feed.ts, next) {
  487. demux->release_ts_feed(demux, feed->ts);
  488. feed->ts = NULL;
  489. }
  490. break;
  491. default:
  492. if (dmxdevfilter->state == DMXDEV_STATE_ALLOCATED)
  493. return 0;
  494. return -EINVAL;
  495. }
  496. dvb_ringbuffer_flush(&dmxdevfilter->buffer);
  497. return 0;
  498. }
  499. static void dvb_dmxdev_delete_pids(struct dmxdev_filter *dmxdevfilter)
  500. {
  501. struct dmxdev_feed *feed, *tmp;
  502. /* delete all PIDs */
  503. list_for_each_entry_safe(feed, tmp, &dmxdevfilter->feed.ts, next) {
  504. list_del(&feed->next);
  505. kfree(feed);
  506. }
  507. BUG_ON(!list_empty(&dmxdevfilter->feed.ts));
  508. }
  509. static inline int dvb_dmxdev_filter_reset(struct dmxdev_filter *dmxdevfilter)
  510. {
  511. if (dmxdevfilter->state < DMXDEV_STATE_SET)
  512. return 0;
  513. if (dmxdevfilter->type == DMXDEV_TYPE_PES)
  514. dvb_dmxdev_delete_pids(dmxdevfilter);
  515. dmxdevfilter->type = DMXDEV_TYPE_NONE;
  516. dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_ALLOCATED);
  517. return 0;
  518. }
  519. static int dvb_dmxdev_start_feed(struct dmxdev *dmxdev,
  520. struct dmxdev_filter *filter,
  521. struct dmxdev_feed *feed)
  522. {
  523. ktime_t timeout = ktime_set(0, 0);
  524. struct dmx_pes_filter_params *para = &filter->params.pes;
  525. enum dmx_output otype;
  526. int ret;
  527. int ts_type;
  528. enum dmx_ts_pes ts_pes;
  529. struct dmx_ts_feed *tsfeed;
  530. feed->ts = NULL;
  531. otype = para->output;
  532. ts_pes = para->pes_type;
  533. if (ts_pes < DMX_PES_OTHER)
  534. ts_type = TS_DECODER;
  535. else
  536. ts_type = 0;
  537. if (otype == DMX_OUT_TS_TAP)
  538. ts_type |= TS_PACKET;
  539. else if (otype == DMX_OUT_TSDEMUX_TAP)
  540. ts_type |= TS_PACKET | TS_DEMUX;
  541. else if (otype == DMX_OUT_TAP)
  542. ts_type |= TS_PACKET | TS_DEMUX | TS_PAYLOAD_ONLY;
  543. ret = dmxdev->demux->allocate_ts_feed(dmxdev->demux, &feed->ts,
  544. dvb_dmxdev_ts_callback);
  545. if (ret < 0)
  546. return ret;
  547. tsfeed = feed->ts;
  548. tsfeed->priv = filter;
  549. ret = tsfeed->set(tsfeed, feed->pid, ts_type, ts_pes, timeout);
  550. if (ret < 0) {
  551. dmxdev->demux->release_ts_feed(dmxdev->demux, tsfeed);
  552. return ret;
  553. }
  554. ret = tsfeed->start_filtering(tsfeed);
  555. if (ret < 0) {
  556. dmxdev->demux->release_ts_feed(dmxdev->demux, tsfeed);
  557. return ret;
  558. }
  559. return 0;
  560. }
  561. static int dvb_dmxdev_filter_start(struct dmxdev_filter *filter)
  562. {
  563. struct dmxdev *dmxdev = filter->dev;
  564. struct dmxdev_feed *feed;
  565. void *mem;
  566. int ret, i;
  567. if (filter->state < DMXDEV_STATE_SET)
  568. return -EINVAL;
  569. if (filter->state >= DMXDEV_STATE_GO)
  570. dvb_dmxdev_filter_stop(filter);
  571. if (!filter->buffer.data) {
  572. mem = vmalloc(filter->buffer.size);
  573. if (!mem)
  574. return -ENOMEM;
  575. spin_lock_irq(&filter->dev->lock);
  576. filter->buffer.data = mem;
  577. spin_unlock_irq(&filter->dev->lock);
  578. }
  579. dvb_ringbuffer_flush(&filter->buffer);
  580. switch (filter->type) {
  581. case DMXDEV_TYPE_SEC:
  582. {
  583. struct dmx_sct_filter_params *para = &filter->params.sec;
  584. struct dmx_section_filter **secfilter = &filter->filter.sec;
  585. struct dmx_section_feed **secfeed = &filter->feed.sec;
  586. *secfilter = NULL;
  587. *secfeed = NULL;
  588. /* find active filter/feed with same PID */
  589. for (i = 0; i < dmxdev->filternum; i++) {
  590. if (dmxdev->filter[i].state >= DMXDEV_STATE_GO &&
  591. dmxdev->filter[i].type == DMXDEV_TYPE_SEC &&
  592. dmxdev->filter[i].params.sec.pid == para->pid) {
  593. *secfeed = dmxdev->filter[i].feed.sec;
  594. break;
  595. }
  596. }
  597. /* if no feed found, try to allocate new one */
  598. if (!*secfeed) {
  599. ret = dmxdev->demux->allocate_section_feed(dmxdev->demux,
  600. secfeed,
  601. dvb_dmxdev_section_callback);
  602. if (!*secfeed) {
  603. pr_err("DVB (%s): could not alloc feed\n",
  604. __func__);
  605. return ret;
  606. }
  607. ret = (*secfeed)->set(*secfeed, para->pid,
  608. (para->flags & DMX_CHECK_CRC) ? 1 : 0);
  609. if (ret < 0) {
  610. pr_err("DVB (%s): could not set feed\n",
  611. __func__);
  612. dvb_dmxdev_feed_restart(filter);
  613. return ret;
  614. }
  615. } else {
  616. dvb_dmxdev_feed_stop(filter);
  617. }
  618. ret = (*secfeed)->allocate_filter(*secfeed, secfilter);
  619. if (ret < 0) {
  620. dvb_dmxdev_feed_restart(filter);
  621. *secfeed = NULL;
  622. dprintk("could not get filter\n");
  623. return ret;
  624. }
  625. (*secfilter)->priv = filter;
  626. memcpy(&((*secfilter)->filter_value[3]),
  627. &(para->filter.filter[1]), DMX_FILTER_SIZE - 1);
  628. memcpy(&(*secfilter)->filter_mask[3],
  629. &para->filter.mask[1], DMX_FILTER_SIZE - 1);
  630. memcpy(&(*secfilter)->filter_mode[3],
  631. &para->filter.mode[1], DMX_FILTER_SIZE - 1);
  632. (*secfilter)->filter_value[0] = para->filter.filter[0];
  633. (*secfilter)->filter_mask[0] = para->filter.mask[0];
  634. (*secfilter)->filter_mode[0] = para->filter.mode[0];
  635. (*secfilter)->filter_mask[1] = 0;
  636. (*secfilter)->filter_mask[2] = 0;
  637. filter->todo = 0;
  638. ret = filter->feed.sec->start_filtering(filter->feed.sec);
  639. if (ret < 0)
  640. return ret;
  641. dvb_dmxdev_filter_timer(filter);
  642. break;
  643. }
  644. case DMXDEV_TYPE_PES:
  645. list_for_each_entry(feed, &filter->feed.ts, next) {
  646. ret = dvb_dmxdev_start_feed(dmxdev, filter, feed);
  647. if (ret < 0) {
  648. dvb_dmxdev_filter_stop(filter);
  649. return ret;
  650. }
  651. }
  652. break;
  653. default:
  654. return -EINVAL;
  655. }
  656. dvb_dmxdev_filter_state_set(filter, DMXDEV_STATE_GO);
  657. return 0;
  658. }
  659. static int dvb_demux_open(struct inode *inode, struct file *file)
  660. {
  661. struct dvb_device *dvbdev = file->private_data;
  662. struct dmxdev *dmxdev = dvbdev->priv;
  663. int i;
  664. struct dmxdev_filter *dmxdevfilter;
  665. if (!dmxdev->filter)
  666. return -EINVAL;
  667. if (mutex_lock_interruptible(&dmxdev->mutex))
  668. return -ERESTARTSYS;
  669. if (dmxdev->exit) {
  670. mutex_unlock(&dmxdev->mutex);
  671. return -ENODEV;
  672. }
  673. for (i = 0; i < dmxdev->filternum; i++)
  674. if (dmxdev->filter[i].state == DMXDEV_STATE_FREE)
  675. break;
  676. if (i == dmxdev->filternum) {
  677. mutex_unlock(&dmxdev->mutex);
  678. return -EMFILE;
  679. }
  680. dmxdevfilter = &dmxdev->filter[i];
  681. mutex_init(&dmxdevfilter->mutex);
  682. file->private_data = dmxdevfilter;
  683. #ifdef CONFIG_DVB_MMAP
  684. dmxdev->may_do_mmap = 1;
  685. #else
  686. dmxdev->may_do_mmap = 0;
  687. #endif
  688. /*
  689. * The mutex passed to dvb_vb2_init is unlocked when a buffer
  690. * is in a blocking wait. However, dmxdevfilter has two mutexes:
  691. * dmxdevfilter->mutex and dmxdev->mutex. So this will not work.
  692. * The solution would be to support unlocking two mutexes in vb2,
  693. * but since this problem has been here since the beginning and
  694. * nobody ever complained, we leave it as-is rather than adding
  695. * that second mutex pointer to vb2.
  696. *
  697. * In the unlikely event that someone complains about this, then
  698. * this comment will hopefully help.
  699. */
  700. dvb_ringbuffer_init(&dmxdevfilter->buffer, NULL, 8192);
  701. dvb_vb2_init(&dmxdevfilter->vb2_ctx, "demux_filter",
  702. &dmxdevfilter->mutex, file->f_flags & O_NONBLOCK);
  703. dmxdevfilter->type = DMXDEV_TYPE_NONE;
  704. dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_ALLOCATED);
  705. timer_setup(&dmxdevfilter->timer, dvb_dmxdev_filter_timeout, 0);
  706. dvbdev->users++;
  707. mutex_unlock(&dmxdev->mutex);
  708. return 0;
  709. }
  710. static int dvb_dmxdev_filter_free(struct dmxdev *dmxdev,
  711. struct dmxdev_filter *dmxdevfilter)
  712. {
  713. mutex_lock(&dmxdev->mutex);
  714. mutex_lock(&dmxdevfilter->mutex);
  715. if (dvb_vb2_is_streaming(&dmxdevfilter->vb2_ctx))
  716. dvb_vb2_stream_off(&dmxdevfilter->vb2_ctx);
  717. dvb_vb2_release(&dmxdevfilter->vb2_ctx);
  718. dvb_dmxdev_filter_stop(dmxdevfilter);
  719. dvb_dmxdev_filter_reset(dmxdevfilter);
  720. if (dmxdevfilter->buffer.data) {
  721. void *mem = dmxdevfilter->buffer.data;
  722. spin_lock_irq(&dmxdev->lock);
  723. dmxdevfilter->buffer.data = NULL;
  724. spin_unlock_irq(&dmxdev->lock);
  725. vfree(mem);
  726. }
  727. dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_FREE);
  728. wake_up(&dmxdevfilter->buffer.queue);
  729. mutex_unlock(&dmxdevfilter->mutex);
  730. mutex_unlock(&dmxdev->mutex);
  731. return 0;
  732. }
  733. static inline void invert_mode(struct dmx_filter *filter)
  734. {
  735. int i;
  736. for (i = 0; i < DMX_FILTER_SIZE; i++)
  737. filter->mode[i] ^= 0xff;
  738. }
  739. static int dvb_dmxdev_add_pid(struct dmxdev *dmxdev,
  740. struct dmxdev_filter *filter, u16 pid)
  741. {
  742. struct dmxdev_feed *feed;
  743. if ((filter->type != DMXDEV_TYPE_PES) ||
  744. (filter->state < DMXDEV_STATE_SET))
  745. return -EINVAL;
  746. /* only TS packet filters may have multiple PIDs */
  747. if ((filter->params.pes.output != DMX_OUT_TSDEMUX_TAP) &&
  748. (!list_empty(&filter->feed.ts)))
  749. return -EINVAL;
  750. feed = kzalloc_obj(struct dmxdev_feed);
  751. if (feed == NULL)
  752. return -ENOMEM;
  753. feed->pid = pid;
  754. list_add(&feed->next, &filter->feed.ts);
  755. if (filter->state >= DMXDEV_STATE_GO)
  756. return dvb_dmxdev_start_feed(dmxdev, filter, feed);
  757. return 0;
  758. }
  759. static int dvb_dmxdev_remove_pid(struct dmxdev *dmxdev,
  760. struct dmxdev_filter *filter, u16 pid)
  761. {
  762. struct dmxdev_feed *feed, *tmp;
  763. if ((filter->type != DMXDEV_TYPE_PES) ||
  764. (filter->state < DMXDEV_STATE_SET))
  765. return -EINVAL;
  766. list_for_each_entry_safe(feed, tmp, &filter->feed.ts, next) {
  767. if ((feed->pid == pid) && (feed->ts != NULL)) {
  768. feed->ts->stop_filtering(feed->ts);
  769. filter->dev->demux->release_ts_feed(filter->dev->demux,
  770. feed->ts);
  771. list_del(&feed->next);
  772. kfree(feed);
  773. }
  774. }
  775. return 0;
  776. }
  777. static int dvb_dmxdev_filter_set(struct dmxdev *dmxdev,
  778. struct dmxdev_filter *dmxdevfilter,
  779. struct dmx_sct_filter_params *params)
  780. {
  781. dprintk("%s: PID=0x%04x, flags=%02x, timeout=%d\n",
  782. __func__, params->pid, params->flags, params->timeout);
  783. dvb_dmxdev_filter_stop(dmxdevfilter);
  784. dmxdevfilter->type = DMXDEV_TYPE_SEC;
  785. memcpy(&dmxdevfilter->params.sec,
  786. params, sizeof(struct dmx_sct_filter_params));
  787. invert_mode(&dmxdevfilter->params.sec.filter);
  788. dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_SET);
  789. if (params->flags & DMX_IMMEDIATE_START)
  790. return dvb_dmxdev_filter_start(dmxdevfilter);
  791. return 0;
  792. }
  793. static int dvb_dmxdev_pes_filter_set(struct dmxdev *dmxdev,
  794. struct dmxdev_filter *dmxdevfilter,
  795. struct dmx_pes_filter_params *params)
  796. {
  797. int ret;
  798. dvb_dmxdev_filter_stop(dmxdevfilter);
  799. dvb_dmxdev_filter_reset(dmxdevfilter);
  800. if ((unsigned int)params->pes_type > DMX_PES_OTHER)
  801. return -EINVAL;
  802. dmxdevfilter->type = DMXDEV_TYPE_PES;
  803. memcpy(&dmxdevfilter->params, params,
  804. sizeof(struct dmx_pes_filter_params));
  805. INIT_LIST_HEAD(&dmxdevfilter->feed.ts);
  806. dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_SET);
  807. ret = dvb_dmxdev_add_pid(dmxdev, dmxdevfilter,
  808. dmxdevfilter->params.pes.pid);
  809. if (ret < 0)
  810. return ret;
  811. if (params->flags & DMX_IMMEDIATE_START)
  812. return dvb_dmxdev_filter_start(dmxdevfilter);
  813. return 0;
  814. }
  815. static ssize_t dvb_dmxdev_read_sec(struct dmxdev_filter *dfil,
  816. struct file *file, char __user *buf,
  817. size_t count, loff_t *ppos)
  818. {
  819. int result, hcount;
  820. int done = 0;
  821. if (dfil->todo <= 0) {
  822. hcount = 3 + dfil->todo;
  823. if (hcount > count)
  824. hcount = count;
  825. result = dvb_dmxdev_buffer_read(&dfil->buffer,
  826. file->f_flags & O_NONBLOCK,
  827. buf, hcount, ppos);
  828. if (result < 0) {
  829. dfil->todo = 0;
  830. return result;
  831. }
  832. if (copy_from_user(dfil->secheader - dfil->todo, buf, result))
  833. return -EFAULT;
  834. buf += result;
  835. done = result;
  836. count -= result;
  837. dfil->todo -= result;
  838. if (dfil->todo > -3)
  839. return done;
  840. dfil->todo = ((dfil->secheader[1] << 8) | dfil->secheader[2]) & 0xfff;
  841. if (!count)
  842. return done;
  843. }
  844. if (count > dfil->todo)
  845. count = dfil->todo;
  846. result = dvb_dmxdev_buffer_read(&dfil->buffer,
  847. file->f_flags & O_NONBLOCK,
  848. buf, count, ppos);
  849. if (result < 0)
  850. return result;
  851. dfil->todo -= result;
  852. return (result + done);
  853. }
  854. static ssize_t
  855. dvb_demux_read(struct file *file, char __user *buf, size_t count,
  856. loff_t *ppos)
  857. {
  858. struct dmxdev_filter *dmxdevfilter = file->private_data;
  859. int ret;
  860. if (mutex_lock_interruptible(&dmxdevfilter->mutex))
  861. return -ERESTARTSYS;
  862. if (dmxdevfilter->type == DMXDEV_TYPE_SEC)
  863. ret = dvb_dmxdev_read_sec(dmxdevfilter, file, buf, count, ppos);
  864. else
  865. ret = dvb_dmxdev_buffer_read(&dmxdevfilter->buffer,
  866. file->f_flags & O_NONBLOCK,
  867. buf, count, ppos);
  868. mutex_unlock(&dmxdevfilter->mutex);
  869. return ret;
  870. }
  871. static int dvb_demux_do_ioctl(struct file *file,
  872. unsigned int cmd, void *parg)
  873. {
  874. struct dmxdev_filter *dmxdevfilter = file->private_data;
  875. struct dmxdev *dmxdev = dmxdevfilter->dev;
  876. unsigned long arg = (unsigned long)parg;
  877. int ret = 0;
  878. if (mutex_lock_interruptible(&dmxdev->mutex))
  879. return -ERESTARTSYS;
  880. switch (cmd) {
  881. case DMX_START:
  882. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  883. mutex_unlock(&dmxdev->mutex);
  884. return -ERESTARTSYS;
  885. }
  886. if (dmxdevfilter->state < DMXDEV_STATE_SET)
  887. ret = -EINVAL;
  888. else
  889. ret = dvb_dmxdev_filter_start(dmxdevfilter);
  890. mutex_unlock(&dmxdevfilter->mutex);
  891. break;
  892. case DMX_STOP:
  893. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  894. mutex_unlock(&dmxdev->mutex);
  895. return -ERESTARTSYS;
  896. }
  897. ret = dvb_dmxdev_filter_stop(dmxdevfilter);
  898. mutex_unlock(&dmxdevfilter->mutex);
  899. break;
  900. case DMX_SET_FILTER:
  901. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  902. mutex_unlock(&dmxdev->mutex);
  903. return -ERESTARTSYS;
  904. }
  905. ret = dvb_dmxdev_filter_set(dmxdev, dmxdevfilter, parg);
  906. mutex_unlock(&dmxdevfilter->mutex);
  907. break;
  908. case DMX_SET_PES_FILTER:
  909. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  910. mutex_unlock(&dmxdev->mutex);
  911. return -ERESTARTSYS;
  912. }
  913. ret = dvb_dmxdev_pes_filter_set(dmxdev, dmxdevfilter, parg);
  914. mutex_unlock(&dmxdevfilter->mutex);
  915. break;
  916. case DMX_SET_BUFFER_SIZE:
  917. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  918. mutex_unlock(&dmxdev->mutex);
  919. return -ERESTARTSYS;
  920. }
  921. ret = dvb_dmxdev_set_buffer_size(dmxdevfilter, arg);
  922. mutex_unlock(&dmxdevfilter->mutex);
  923. break;
  924. case DMX_GET_PES_PIDS:
  925. if (!dmxdev->demux->get_pes_pids) {
  926. ret = -EINVAL;
  927. break;
  928. }
  929. dmxdev->demux->get_pes_pids(dmxdev->demux, parg);
  930. break;
  931. case DMX_GET_STC:
  932. if (!dmxdev->demux->get_stc) {
  933. ret = -EINVAL;
  934. break;
  935. }
  936. ret = dmxdev->demux->get_stc(dmxdev->demux,
  937. ((struct dmx_stc *)parg)->num,
  938. &((struct dmx_stc *)parg)->stc,
  939. &((struct dmx_stc *)parg)->base);
  940. break;
  941. case DMX_ADD_PID:
  942. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  943. ret = -ERESTARTSYS;
  944. break;
  945. }
  946. ret = dvb_dmxdev_add_pid(dmxdev, dmxdevfilter, *(u16 *)parg);
  947. mutex_unlock(&dmxdevfilter->mutex);
  948. break;
  949. case DMX_REMOVE_PID:
  950. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  951. ret = -ERESTARTSYS;
  952. break;
  953. }
  954. ret = dvb_dmxdev_remove_pid(dmxdev, dmxdevfilter, *(u16 *)parg);
  955. mutex_unlock(&dmxdevfilter->mutex);
  956. break;
  957. #ifdef CONFIG_DVB_MMAP
  958. case DMX_REQBUFS:
  959. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  960. mutex_unlock(&dmxdev->mutex);
  961. return -ERESTARTSYS;
  962. }
  963. ret = dvb_vb2_reqbufs(&dmxdevfilter->vb2_ctx, parg);
  964. mutex_unlock(&dmxdevfilter->mutex);
  965. break;
  966. case DMX_QUERYBUF:
  967. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  968. mutex_unlock(&dmxdev->mutex);
  969. return -ERESTARTSYS;
  970. }
  971. ret = dvb_vb2_querybuf(&dmxdevfilter->vb2_ctx, parg);
  972. mutex_unlock(&dmxdevfilter->mutex);
  973. break;
  974. case DMX_EXPBUF:
  975. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  976. mutex_unlock(&dmxdev->mutex);
  977. return -ERESTARTSYS;
  978. }
  979. ret = dvb_vb2_expbuf(&dmxdevfilter->vb2_ctx, parg);
  980. mutex_unlock(&dmxdevfilter->mutex);
  981. break;
  982. case DMX_QBUF:
  983. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  984. mutex_unlock(&dmxdev->mutex);
  985. return -ERESTARTSYS;
  986. }
  987. ret = dvb_vb2_qbuf(&dmxdevfilter->vb2_ctx, parg);
  988. if (ret == 0 && !dvb_vb2_is_streaming(&dmxdevfilter->vb2_ctx))
  989. ret = dvb_vb2_stream_on(&dmxdevfilter->vb2_ctx);
  990. mutex_unlock(&dmxdevfilter->mutex);
  991. break;
  992. case DMX_DQBUF:
  993. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  994. mutex_unlock(&dmxdev->mutex);
  995. return -ERESTARTSYS;
  996. }
  997. ret = dvb_vb2_dqbuf(&dmxdevfilter->vb2_ctx, parg);
  998. mutex_unlock(&dmxdevfilter->mutex);
  999. break;
  1000. #endif
  1001. default:
  1002. ret = -ENOTTY;
  1003. break;
  1004. }
  1005. mutex_unlock(&dmxdev->mutex);
  1006. return ret;
  1007. }
  1008. static long dvb_demux_ioctl(struct file *file, unsigned int cmd,
  1009. unsigned long arg)
  1010. {
  1011. return dvb_usercopy(file, cmd, arg, dvb_demux_do_ioctl);
  1012. }
  1013. static __poll_t dvb_demux_poll(struct file *file, poll_table *wait)
  1014. {
  1015. struct dmxdev_filter *dmxdevfilter = file->private_data;
  1016. __poll_t mask = 0;
  1017. poll_wait(file, &dmxdevfilter->buffer.queue, wait);
  1018. if ((!dmxdevfilter) || dmxdevfilter->dev->exit)
  1019. return EPOLLERR;
  1020. if (dvb_vb2_is_streaming(&dmxdevfilter->vb2_ctx))
  1021. return dvb_vb2_poll(&dmxdevfilter->vb2_ctx, file, wait);
  1022. if (dmxdevfilter->state != DMXDEV_STATE_GO &&
  1023. dmxdevfilter->state != DMXDEV_STATE_DONE &&
  1024. dmxdevfilter->state != DMXDEV_STATE_TIMEDOUT)
  1025. return 0;
  1026. if (dmxdevfilter->buffer.error)
  1027. mask |= (EPOLLIN | EPOLLRDNORM | EPOLLPRI | EPOLLERR);
  1028. if (!dvb_ringbuffer_empty(&dmxdevfilter->buffer))
  1029. mask |= (EPOLLIN | EPOLLRDNORM | EPOLLPRI);
  1030. return mask;
  1031. }
  1032. #ifdef CONFIG_DVB_MMAP
  1033. static int dvb_demux_mmap(struct file *file, struct vm_area_struct *vma)
  1034. {
  1035. struct dmxdev_filter *dmxdevfilter = file->private_data;
  1036. struct dmxdev *dmxdev = dmxdevfilter->dev;
  1037. if (!dmxdev->may_do_mmap)
  1038. return -ENOTTY;
  1039. return dvb_vb2_mmap(&dmxdevfilter->vb2_ctx, vma);
  1040. }
  1041. #endif
  1042. static int dvb_demux_release(struct inode *inode, struct file *file)
  1043. {
  1044. struct dmxdev_filter *dmxdevfilter = file->private_data;
  1045. struct dmxdev *dmxdev = dmxdevfilter->dev;
  1046. int ret;
  1047. ret = dvb_dmxdev_filter_free(dmxdev, dmxdevfilter);
  1048. mutex_lock(&dmxdev->mutex);
  1049. dmxdev->dvbdev->users--;
  1050. if (dmxdev->dvbdev->users == 1 && dmxdev->exit == 1) {
  1051. mutex_unlock(&dmxdev->mutex);
  1052. wake_up(&dmxdev->dvbdev->wait_queue);
  1053. } else
  1054. mutex_unlock(&dmxdev->mutex);
  1055. return ret;
  1056. }
  1057. static const struct file_operations dvb_demux_fops = {
  1058. .owner = THIS_MODULE,
  1059. .read = dvb_demux_read,
  1060. .unlocked_ioctl = dvb_demux_ioctl,
  1061. .compat_ioctl = dvb_demux_ioctl,
  1062. .open = dvb_demux_open,
  1063. .release = dvb_demux_release,
  1064. .poll = dvb_demux_poll,
  1065. .llseek = default_llseek,
  1066. #ifdef CONFIG_DVB_MMAP
  1067. .mmap = dvb_demux_mmap,
  1068. #endif
  1069. };
  1070. static const struct dvb_device dvbdev_demux = {
  1071. .priv = NULL,
  1072. .users = 1,
  1073. .writers = 1,
  1074. #if defined(CONFIG_MEDIA_CONTROLLER_DVB)
  1075. .name = "dvb-demux",
  1076. #endif
  1077. .fops = &dvb_demux_fops
  1078. };
  1079. static int dvb_dvr_do_ioctl(struct file *file,
  1080. unsigned int cmd, void *parg)
  1081. {
  1082. struct dvb_device *dvbdev = file->private_data;
  1083. struct dmxdev *dmxdev = dvbdev->priv;
  1084. unsigned long arg = (unsigned long)parg;
  1085. int ret;
  1086. if (mutex_lock_interruptible(&dmxdev->mutex))
  1087. return -ERESTARTSYS;
  1088. switch (cmd) {
  1089. case DMX_SET_BUFFER_SIZE:
  1090. ret = dvb_dvr_set_buffer_size(dmxdev, arg);
  1091. break;
  1092. #ifdef CONFIG_DVB_MMAP
  1093. case DMX_REQBUFS:
  1094. ret = dvb_vb2_reqbufs(&dmxdev->dvr_vb2_ctx, parg);
  1095. break;
  1096. case DMX_QUERYBUF:
  1097. ret = dvb_vb2_querybuf(&dmxdev->dvr_vb2_ctx, parg);
  1098. break;
  1099. case DMX_EXPBUF:
  1100. ret = dvb_vb2_expbuf(&dmxdev->dvr_vb2_ctx, parg);
  1101. break;
  1102. case DMX_QBUF:
  1103. ret = dvb_vb2_qbuf(&dmxdev->dvr_vb2_ctx, parg);
  1104. if (ret == 0 && !dvb_vb2_is_streaming(&dmxdev->dvr_vb2_ctx))
  1105. ret = dvb_vb2_stream_on(&dmxdev->dvr_vb2_ctx);
  1106. break;
  1107. case DMX_DQBUF:
  1108. ret = dvb_vb2_dqbuf(&dmxdev->dvr_vb2_ctx, parg);
  1109. break;
  1110. #endif
  1111. default:
  1112. ret = -ENOTTY;
  1113. break;
  1114. }
  1115. mutex_unlock(&dmxdev->mutex);
  1116. return ret;
  1117. }
  1118. static long dvb_dvr_ioctl(struct file *file,
  1119. unsigned int cmd, unsigned long arg)
  1120. {
  1121. return dvb_usercopy(file, cmd, arg, dvb_dvr_do_ioctl);
  1122. }
  1123. static __poll_t dvb_dvr_poll(struct file *file, poll_table *wait)
  1124. {
  1125. struct dvb_device *dvbdev = file->private_data;
  1126. struct dmxdev *dmxdev = dvbdev->priv;
  1127. __poll_t mask = 0;
  1128. dprintk("%s\n", __func__);
  1129. poll_wait(file, &dmxdev->dvr_buffer.queue, wait);
  1130. if (dmxdev->exit)
  1131. return EPOLLERR;
  1132. if (dvb_vb2_is_streaming(&dmxdev->dvr_vb2_ctx))
  1133. return dvb_vb2_poll(&dmxdev->dvr_vb2_ctx, file, wait);
  1134. if (((file->f_flags & O_ACCMODE) == O_RDONLY) ||
  1135. dmxdev->may_do_mmap) {
  1136. if (dmxdev->dvr_buffer.error)
  1137. mask |= (EPOLLIN | EPOLLRDNORM | EPOLLPRI | EPOLLERR);
  1138. if (!dvb_ringbuffer_empty(&dmxdev->dvr_buffer))
  1139. mask |= (EPOLLIN | EPOLLRDNORM | EPOLLPRI);
  1140. } else
  1141. mask |= (EPOLLOUT | EPOLLWRNORM | EPOLLPRI);
  1142. return mask;
  1143. }
  1144. #ifdef CONFIG_DVB_MMAP
  1145. static int dvb_dvr_mmap(struct file *file, struct vm_area_struct *vma)
  1146. {
  1147. struct dvb_device *dvbdev = file->private_data;
  1148. struct dmxdev *dmxdev = dvbdev->priv;
  1149. if (!dmxdev->may_do_mmap)
  1150. return -ENOTTY;
  1151. if (dmxdev->exit)
  1152. return -ENODEV;
  1153. return dvb_vb2_mmap(&dmxdev->dvr_vb2_ctx, vma);
  1154. }
  1155. #endif
  1156. static const struct file_operations dvb_dvr_fops = {
  1157. .owner = THIS_MODULE,
  1158. .read = dvb_dvr_read,
  1159. .write = dvb_dvr_write,
  1160. .unlocked_ioctl = dvb_dvr_ioctl,
  1161. .open = dvb_dvr_open,
  1162. .release = dvb_dvr_release,
  1163. .poll = dvb_dvr_poll,
  1164. .llseek = default_llseek,
  1165. #ifdef CONFIG_DVB_MMAP
  1166. .mmap = dvb_dvr_mmap,
  1167. #endif
  1168. };
  1169. static const struct dvb_device dvbdev_dvr = {
  1170. .priv = NULL,
  1171. .readers = 1,
  1172. .users = 1,
  1173. #if defined(CONFIG_MEDIA_CONTROLLER_DVB)
  1174. .name = "dvb-dvr",
  1175. #endif
  1176. .fops = &dvb_dvr_fops
  1177. };
  1178. int dvb_dmxdev_init(struct dmxdev *dmxdev, struct dvb_adapter *dvb_adapter)
  1179. {
  1180. int i, ret;
  1181. if (dmxdev->demux->open(dmxdev->demux) < 0)
  1182. return -EUSERS;
  1183. dmxdev->filter = vmalloc_array(dmxdev->filternum,
  1184. sizeof(struct dmxdev_filter));
  1185. if (!dmxdev->filter)
  1186. return -ENOMEM;
  1187. mutex_init(&dmxdev->mutex);
  1188. spin_lock_init(&dmxdev->lock);
  1189. for (i = 0; i < dmxdev->filternum; i++) {
  1190. dmxdev->filter[i].dev = dmxdev;
  1191. dmxdev->filter[i].buffer.data = NULL;
  1192. dvb_dmxdev_filter_state_set(&dmxdev->filter[i],
  1193. DMXDEV_STATE_FREE);
  1194. }
  1195. ret = dvb_register_device(dvb_adapter, &dmxdev->dvbdev, &dvbdev_demux, dmxdev,
  1196. DVB_DEVICE_DEMUX, dmxdev->filternum);
  1197. if (ret < 0)
  1198. goto err_register_dvbdev;
  1199. ret = dvb_register_device(dvb_adapter, &dmxdev->dvr_dvbdev, &dvbdev_dvr,
  1200. dmxdev, DVB_DEVICE_DVR, dmxdev->filternum);
  1201. if (ret < 0)
  1202. goto err_register_dvr_dvbdev;
  1203. dvb_ringbuffer_init(&dmxdev->dvr_buffer, NULL, 8192);
  1204. return 0;
  1205. err_register_dvr_dvbdev:
  1206. dvb_unregister_device(dmxdev->dvbdev);
  1207. err_register_dvbdev:
  1208. vfree(dmxdev->filter);
  1209. dmxdev->filter = NULL;
  1210. return ret;
  1211. }
  1212. EXPORT_SYMBOL(dvb_dmxdev_init);
  1213. void dvb_dmxdev_release(struct dmxdev *dmxdev)
  1214. {
  1215. mutex_lock(&dmxdev->mutex);
  1216. dmxdev->exit = 1;
  1217. mutex_unlock(&dmxdev->mutex);
  1218. if (dmxdev->dvbdev->users > 1) {
  1219. wait_event(dmxdev->dvbdev->wait_queue,
  1220. dmxdev->dvbdev->users == 1);
  1221. }
  1222. if (dmxdev->dvr_dvbdev->users > 1) {
  1223. wait_event(dmxdev->dvr_dvbdev->wait_queue,
  1224. dmxdev->dvr_dvbdev->users == 1);
  1225. }
  1226. dvb_unregister_device(dmxdev->dvbdev);
  1227. dvb_unregister_device(dmxdev->dvr_dvbdev);
  1228. vfree(dmxdev->filter);
  1229. dmxdev->filter = NULL;
  1230. dmxdev->demux->close(dmxdev->demux);
  1231. }
  1232. EXPORT_SYMBOL(dvb_dmxdev_release);