dmxdev.c 35 KB

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