dmxdev.c 26 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075
  1. /*
  2. * dmxdev.c - DVB demultiplexer device
  3. *
  4. * Copyright (C) 2000 Ralph Metzler & Marcus Metzler
  5. * for convergence integrated media GmbH
  6. *
  7. * This program is free software; you can redistribute it and/or
  8. * modify it under the terms of the GNU Lesser General Public License
  9. * as published by the Free Software Foundation; either version 2.1
  10. * of the License, or (at your option) any later version.
  11. *
  12. * This program is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15. * GNU General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public License
  18. * along with this program; if not, write to the Free Software
  19. * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  20. *
  21. */
  22. #include <linux/spinlock.h>
  23. #include <linux/slab.h>
  24. #include <linux/vmalloc.h>
  25. #include <linux/module.h>
  26. #include <linux/moduleparam.h>
  27. #include <linux/poll.h>
  28. #include <linux/ioctl.h>
  29. #include <linux/wait.h>
  30. #include <asm/uaccess.h>
  31. #include <asm/system.h>
  32. #include "dmxdev.h"
  33. static int debug;
  34. module_param(debug, int, 0644);
  35. MODULE_PARM_DESC(debug, "Turn on/off debugging (default:off).");
  36. #define dprintk if (debug) printk
  37. static int dvb_dmxdev_buffer_write(struct dvb_ringbuffer *buf,
  38. const u8 *src, size_t len)
  39. {
  40. ssize_t free;
  41. if (!len)
  42. return 0;
  43. if (!buf->data)
  44. return 0;
  45. free = dvb_ringbuffer_free(buf);
  46. if (len > free) {
  47. dprintk("dmxdev: buffer overflow\n");
  48. return -EOVERFLOW;
  49. }
  50. return dvb_ringbuffer_write(buf, src, len);
  51. }
  52. static ssize_t dvb_dmxdev_buffer_read(struct dvb_ringbuffer *src,
  53. int non_blocking, char __user *buf,
  54. size_t count, loff_t *ppos)
  55. {
  56. size_t todo;
  57. ssize_t avail;
  58. ssize_t ret = 0;
  59. if (!src->data)
  60. return 0;
  61. if (src->error) {
  62. ret = src->error;
  63. dvb_ringbuffer_flush(src);
  64. return ret;
  65. }
  66. for (todo = count; todo > 0; todo -= ret) {
  67. if (non_blocking && dvb_ringbuffer_empty(src)) {
  68. ret = -EWOULDBLOCK;
  69. break;
  70. }
  71. ret = wait_event_interruptible(src->queue,
  72. !dvb_ringbuffer_empty(src) ||
  73. (src->error != 0));
  74. if (ret < 0)
  75. break;
  76. if (src->error) {
  77. ret = src->error;
  78. dvb_ringbuffer_flush(src);
  79. break;
  80. }
  81. avail = dvb_ringbuffer_avail(src);
  82. if (avail > todo)
  83. avail = todo;
  84. ret = dvb_ringbuffer_read(src, buf, avail, 1);
  85. if (ret < 0)
  86. break;
  87. buf += ret;
  88. }
  89. return (count - todo) ? (count - todo) : ret;
  90. }
  91. static struct dmx_frontend *get_fe(struct dmx_demux *demux, int type)
  92. {
  93. struct list_head *head, *pos;
  94. head = demux->get_frontends(demux);
  95. if (!head)
  96. return NULL;
  97. list_for_each(pos, head)
  98. if (DMX_FE_ENTRY(pos)->source == type)
  99. return DMX_FE_ENTRY(pos);
  100. return NULL;
  101. }
  102. static int dvb_dvr_open(struct inode *inode, struct file *file)
  103. {
  104. struct dvb_device *dvbdev = file->private_data;
  105. struct dmxdev *dmxdev = dvbdev->priv;
  106. struct dmx_frontend *front;
  107. dprintk("function : %s\n", __FUNCTION__);
  108. if (mutex_lock_interruptible(&dmxdev->mutex))
  109. return -ERESTARTSYS;
  110. if ((file->f_flags & O_ACCMODE) == O_RDWR) {
  111. if (!(dmxdev->capabilities & DMXDEV_CAP_DUPLEX)) {
  112. mutex_unlock(&dmxdev->mutex);
  113. return -EOPNOTSUPP;
  114. }
  115. }
  116. if ((file->f_flags & O_ACCMODE) == O_RDONLY) {
  117. void *mem;
  118. if (!dvbdev->readers) {
  119. mutex_unlock(&dmxdev->mutex);
  120. return -EBUSY;
  121. }
  122. mem = vmalloc(DVR_BUFFER_SIZE);
  123. if (!mem) {
  124. mutex_unlock(&dmxdev->mutex);
  125. return -ENOMEM;
  126. }
  127. dvb_ringbuffer_init(&dmxdev->dvr_buffer, mem, DVR_BUFFER_SIZE);
  128. dvbdev->readers--;
  129. }
  130. if ((file->f_flags & O_ACCMODE) == O_WRONLY) {
  131. dmxdev->dvr_orig_fe = dmxdev->demux->frontend;
  132. if (!dmxdev->demux->write) {
  133. mutex_unlock(&dmxdev->mutex);
  134. return -EOPNOTSUPP;
  135. }
  136. front = get_fe(dmxdev->demux, DMX_MEMORY_FE);
  137. if (!front) {
  138. mutex_unlock(&dmxdev->mutex);
  139. return -EINVAL;
  140. }
  141. dmxdev->demux->disconnect_frontend(dmxdev->demux);
  142. dmxdev->demux->connect_frontend(dmxdev->demux, front);
  143. }
  144. mutex_unlock(&dmxdev->mutex);
  145. return 0;
  146. }
  147. static int dvb_dvr_release(struct inode *inode, struct file *file)
  148. {
  149. struct dvb_device *dvbdev = file->private_data;
  150. struct dmxdev *dmxdev = dvbdev->priv;
  151. mutex_lock(&dmxdev->mutex);
  152. if ((file->f_flags & O_ACCMODE) == O_WRONLY) {
  153. dmxdev->demux->disconnect_frontend(dmxdev->demux);
  154. dmxdev->demux->connect_frontend(dmxdev->demux,
  155. dmxdev->dvr_orig_fe);
  156. }
  157. if ((file->f_flags & O_ACCMODE) == O_RDONLY) {
  158. dvbdev->readers++;
  159. if (dmxdev->dvr_buffer.data) {
  160. void *mem = dmxdev->dvr_buffer.data;
  161. mb();
  162. spin_lock_irq(&dmxdev->lock);
  163. dmxdev->dvr_buffer.data = NULL;
  164. spin_unlock_irq(&dmxdev->lock);
  165. vfree(mem);
  166. }
  167. }
  168. mutex_unlock(&dmxdev->mutex);
  169. return 0;
  170. }
  171. static ssize_t dvb_dvr_write(struct file *file, const char __user *buf,
  172. size_t count, loff_t *ppos)
  173. {
  174. struct dvb_device *dvbdev = file->private_data;
  175. struct dmxdev *dmxdev = dvbdev->priv;
  176. int ret;
  177. if (!dmxdev->demux->write)
  178. return -EOPNOTSUPP;
  179. if ((file->f_flags & O_ACCMODE) != O_WRONLY)
  180. return -EINVAL;
  181. if (mutex_lock_interruptible(&dmxdev->mutex))
  182. return -ERESTARTSYS;
  183. ret = dmxdev->demux->write(dmxdev->demux, buf, count);
  184. mutex_unlock(&dmxdev->mutex);
  185. return ret;
  186. }
  187. static ssize_t dvb_dvr_read(struct file *file, char __user *buf, size_t count,
  188. loff_t *ppos)
  189. {
  190. struct dvb_device *dvbdev = file->private_data;
  191. struct dmxdev *dmxdev = dvbdev->priv;
  192. int ret;
  193. //mutex_lock(&dmxdev->mutex);
  194. ret = dvb_dmxdev_buffer_read(&dmxdev->dvr_buffer,
  195. file->f_flags & O_NONBLOCK,
  196. buf, count, ppos);
  197. //mutex_unlock(&dmxdev->mutex);
  198. return ret;
  199. }
  200. static inline void dvb_dmxdev_filter_state_set(struct dmxdev_filter
  201. *dmxdevfilter, int state)
  202. {
  203. spin_lock_irq(&dmxdevfilter->dev->lock);
  204. dmxdevfilter->state = state;
  205. spin_unlock_irq(&dmxdevfilter->dev->lock);
  206. }
  207. static int dvb_dmxdev_set_buffer_size(struct dmxdev_filter *dmxdevfilter,
  208. unsigned long size)
  209. {
  210. struct dvb_ringbuffer *buf = &dmxdevfilter->buffer;
  211. void *mem;
  212. if (buf->size == size)
  213. return 0;
  214. if (dmxdevfilter->state >= DMXDEV_STATE_GO)
  215. return -EBUSY;
  216. spin_lock_irq(&dmxdevfilter->dev->lock);
  217. mem = buf->data;
  218. buf->data = NULL;
  219. buf->size = size;
  220. dvb_ringbuffer_flush(buf);
  221. spin_unlock_irq(&dmxdevfilter->dev->lock);
  222. vfree(mem);
  223. if (buf->size) {
  224. mem = vmalloc(dmxdevfilter->buffer.size);
  225. if (!mem)
  226. return -ENOMEM;
  227. spin_lock_irq(&dmxdevfilter->dev->lock);
  228. buf->data = mem;
  229. spin_unlock_irq(&dmxdevfilter->dev->lock);
  230. }
  231. return 0;
  232. }
  233. static void dvb_dmxdev_filter_timeout(unsigned long data)
  234. {
  235. struct dmxdev_filter *dmxdevfilter = (struct dmxdev_filter *)data;
  236. dmxdevfilter->buffer.error = -ETIMEDOUT;
  237. spin_lock_irq(&dmxdevfilter->dev->lock);
  238. dmxdevfilter->state = DMXDEV_STATE_TIMEDOUT;
  239. spin_unlock_irq(&dmxdevfilter->dev->lock);
  240. wake_up(&dmxdevfilter->buffer.queue);
  241. }
  242. static void dvb_dmxdev_filter_timer(struct dmxdev_filter *dmxdevfilter)
  243. {
  244. struct dmx_sct_filter_params *para = &dmxdevfilter->params.sec;
  245. del_timer(&dmxdevfilter->timer);
  246. if (para->timeout) {
  247. dmxdevfilter->timer.function = dvb_dmxdev_filter_timeout;
  248. dmxdevfilter->timer.data = (unsigned long)dmxdevfilter;
  249. dmxdevfilter->timer.expires =
  250. jiffies + 1 + (HZ / 2 + HZ * para->timeout) / 1000;
  251. add_timer(&dmxdevfilter->timer);
  252. }
  253. }
  254. static int dvb_dmxdev_section_callback(const u8 *buffer1, size_t buffer1_len,
  255. const u8 *buffer2, size_t buffer2_len,
  256. struct dmx_section_filter *filter,
  257. enum dmx_success success)
  258. {
  259. struct dmxdev_filter *dmxdevfilter = filter->priv;
  260. int ret;
  261. if (dmxdevfilter->buffer.error) {
  262. wake_up(&dmxdevfilter->buffer.queue);
  263. return 0;
  264. }
  265. spin_lock(&dmxdevfilter->dev->lock);
  266. if (dmxdevfilter->state != DMXDEV_STATE_GO) {
  267. spin_unlock(&dmxdevfilter->dev->lock);
  268. return 0;
  269. }
  270. del_timer(&dmxdevfilter->timer);
  271. dprintk("dmxdev: section callback %02x %02x %02x %02x %02x %02x\n",
  272. buffer1[0], buffer1[1],
  273. buffer1[2], buffer1[3], buffer1[4], buffer1[5]);
  274. ret = dvb_dmxdev_buffer_write(&dmxdevfilter->buffer, buffer1,
  275. buffer1_len);
  276. if (ret == buffer1_len) {
  277. ret = dvb_dmxdev_buffer_write(&dmxdevfilter->buffer, buffer2,
  278. buffer2_len);
  279. }
  280. if (ret < 0) {
  281. dvb_ringbuffer_flush(&dmxdevfilter->buffer);
  282. dmxdevfilter->buffer.error = ret;
  283. }
  284. if (dmxdevfilter->params.sec.flags & DMX_ONESHOT)
  285. dmxdevfilter->state = DMXDEV_STATE_DONE;
  286. spin_unlock(&dmxdevfilter->dev->lock);
  287. wake_up(&dmxdevfilter->buffer.queue);
  288. return 0;
  289. }
  290. static int dvb_dmxdev_ts_callback(const u8 *buffer1, size_t buffer1_len,
  291. const u8 *buffer2, size_t buffer2_len,
  292. struct dmx_ts_feed *feed,
  293. enum dmx_success success)
  294. {
  295. struct dmxdev_filter *dmxdevfilter = feed->priv;
  296. struct dvb_ringbuffer *buffer;
  297. int ret;
  298. spin_lock(&dmxdevfilter->dev->lock);
  299. if (dmxdevfilter->params.pes.output == DMX_OUT_DECODER) {
  300. spin_unlock(&dmxdevfilter->dev->lock);
  301. return 0;
  302. }
  303. if (dmxdevfilter->params.pes.output == DMX_OUT_TAP)
  304. buffer = &dmxdevfilter->buffer;
  305. else
  306. buffer = &dmxdevfilter->dev->dvr_buffer;
  307. if (buffer->error) {
  308. spin_unlock(&dmxdevfilter->dev->lock);
  309. wake_up(&buffer->queue);
  310. return 0;
  311. }
  312. ret = dvb_dmxdev_buffer_write(buffer, buffer1, buffer1_len);
  313. if (ret == buffer1_len)
  314. ret = dvb_dmxdev_buffer_write(buffer, buffer2, buffer2_len);
  315. if (ret < 0) {
  316. dvb_ringbuffer_flush(buffer);
  317. buffer->error = ret;
  318. }
  319. spin_unlock(&dmxdevfilter->dev->lock);
  320. wake_up(&buffer->queue);
  321. return 0;
  322. }
  323. /* stop feed but only mark the specified filter as stopped (state set) */
  324. static int dvb_dmxdev_feed_stop(struct dmxdev_filter *dmxdevfilter)
  325. {
  326. dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_SET);
  327. switch (dmxdevfilter->type) {
  328. case DMXDEV_TYPE_SEC:
  329. del_timer(&dmxdevfilter->timer);
  330. dmxdevfilter->feed.sec->stop_filtering(dmxdevfilter->feed.sec);
  331. break;
  332. case DMXDEV_TYPE_PES:
  333. dmxdevfilter->feed.ts->stop_filtering(dmxdevfilter->feed.ts);
  334. break;
  335. default:
  336. return -EINVAL;
  337. }
  338. return 0;
  339. }
  340. /* start feed associated with the specified filter */
  341. static int dvb_dmxdev_feed_start(struct dmxdev_filter *filter)
  342. {
  343. dvb_dmxdev_filter_state_set(filter, DMXDEV_STATE_GO);
  344. switch (filter->type) {
  345. case DMXDEV_TYPE_SEC:
  346. return filter->feed.sec->start_filtering(filter->feed.sec);
  347. case DMXDEV_TYPE_PES:
  348. return filter->feed.ts->start_filtering(filter->feed.ts);
  349. default:
  350. return -EINVAL;
  351. }
  352. return 0;
  353. }
  354. /* restart section feed if it has filters left associated with it,
  355. otherwise release the feed */
  356. static int dvb_dmxdev_feed_restart(struct dmxdev_filter *filter)
  357. {
  358. int i;
  359. struct dmxdev *dmxdev = filter->dev;
  360. u16 pid = filter->params.sec.pid;
  361. for (i = 0; i < dmxdev->filternum; i++)
  362. if (dmxdev->filter[i].state >= DMXDEV_STATE_GO &&
  363. dmxdev->filter[i].type == DMXDEV_TYPE_SEC &&
  364. dmxdev->filter[i].params.sec.pid == pid) {
  365. dvb_dmxdev_feed_start(&dmxdev->filter[i]);
  366. return 0;
  367. }
  368. filter->dev->demux->release_section_feed(dmxdev->demux,
  369. filter->feed.sec);
  370. return 0;
  371. }
  372. static int dvb_dmxdev_filter_stop(struct dmxdev_filter *dmxdevfilter)
  373. {
  374. if (dmxdevfilter->state < DMXDEV_STATE_GO)
  375. return 0;
  376. switch (dmxdevfilter->type) {
  377. case DMXDEV_TYPE_SEC:
  378. if (!dmxdevfilter->feed.sec)
  379. break;
  380. dvb_dmxdev_feed_stop(dmxdevfilter);
  381. if (dmxdevfilter->filter.sec)
  382. dmxdevfilter->feed.sec->
  383. release_filter(dmxdevfilter->feed.sec,
  384. dmxdevfilter->filter.sec);
  385. dvb_dmxdev_feed_restart(dmxdevfilter);
  386. dmxdevfilter->feed.sec = NULL;
  387. break;
  388. case DMXDEV_TYPE_PES:
  389. if (!dmxdevfilter->feed.ts)
  390. break;
  391. dvb_dmxdev_feed_stop(dmxdevfilter);
  392. dmxdevfilter->dev->demux->
  393. release_ts_feed(dmxdevfilter->dev->demux,
  394. dmxdevfilter->feed.ts);
  395. dmxdevfilter->feed.ts = NULL;
  396. break;
  397. default:
  398. if (dmxdevfilter->state == DMXDEV_STATE_ALLOCATED)
  399. return 0;
  400. return -EINVAL;
  401. }
  402. dvb_ringbuffer_flush(&dmxdevfilter->buffer);
  403. return 0;
  404. }
  405. static inline int dvb_dmxdev_filter_reset(struct dmxdev_filter *dmxdevfilter)
  406. {
  407. if (dmxdevfilter->state < DMXDEV_STATE_SET)
  408. return 0;
  409. dmxdevfilter->type = DMXDEV_TYPE_NONE;
  410. dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_ALLOCATED);
  411. return 0;
  412. }
  413. static int dvb_dmxdev_filter_start(struct dmxdev_filter *filter)
  414. {
  415. struct dmxdev *dmxdev = filter->dev;
  416. void *mem;
  417. int ret, i;
  418. if (filter->state < DMXDEV_STATE_SET)
  419. return -EINVAL;
  420. if (filter->state >= DMXDEV_STATE_GO)
  421. dvb_dmxdev_filter_stop(filter);
  422. if (!filter->buffer.data) {
  423. mem = vmalloc(filter->buffer.size);
  424. if (!mem)
  425. return -ENOMEM;
  426. spin_lock_irq(&filter->dev->lock);
  427. filter->buffer.data = mem;
  428. spin_unlock_irq(&filter->dev->lock);
  429. }
  430. dvb_ringbuffer_flush(&filter->buffer);
  431. switch (filter->type) {
  432. case DMXDEV_TYPE_SEC:
  433. {
  434. struct dmx_sct_filter_params *para = &filter->params.sec;
  435. struct dmx_section_filter **secfilter = &filter->filter.sec;
  436. struct dmx_section_feed **secfeed = &filter->feed.sec;
  437. *secfilter = NULL;
  438. *secfeed = NULL;
  439. /* find active filter/feed with same PID */
  440. for (i = 0; i < dmxdev->filternum; i++) {
  441. if (dmxdev->filter[i].state >= DMXDEV_STATE_GO &&
  442. dmxdev->filter[i].type == DMXDEV_TYPE_SEC &&
  443. dmxdev->filter[i].params.sec.pid == para->pid) {
  444. *secfeed = dmxdev->filter[i].feed.sec;
  445. break;
  446. }
  447. }
  448. /* if no feed found, try to allocate new one */
  449. if (!*secfeed) {
  450. ret = dmxdev->demux->allocate_section_feed(dmxdev->demux,
  451. secfeed,
  452. dvb_dmxdev_section_callback);
  453. if (ret < 0) {
  454. printk("DVB (%s): could not alloc feed\n",
  455. __FUNCTION__);
  456. return ret;
  457. }
  458. ret = (*secfeed)->set(*secfeed, para->pid, 32768,
  459. (para->flags & DMX_CHECK_CRC) ? 1 : 0);
  460. if (ret < 0) {
  461. printk("DVB (%s): could not set feed\n",
  462. __FUNCTION__);
  463. dvb_dmxdev_feed_restart(filter);
  464. return ret;
  465. }
  466. } else {
  467. dvb_dmxdev_feed_stop(filter);
  468. }
  469. ret = (*secfeed)->allocate_filter(*secfeed, secfilter);
  470. if (ret < 0) {
  471. dvb_dmxdev_feed_restart(filter);
  472. filter->feed.sec->start_filtering(*secfeed);
  473. dprintk("could not get filter\n");
  474. return ret;
  475. }
  476. (*secfilter)->priv = filter;
  477. memcpy(&((*secfilter)->filter_value[3]),
  478. &(para->filter.filter[1]), DMX_FILTER_SIZE - 1);
  479. memcpy(&(*secfilter)->filter_mask[3],
  480. &para->filter.mask[1], DMX_FILTER_SIZE - 1);
  481. memcpy(&(*secfilter)->filter_mode[3],
  482. &para->filter.mode[1], DMX_FILTER_SIZE - 1);
  483. (*secfilter)->filter_value[0] = para->filter.filter[0];
  484. (*secfilter)->filter_mask[0] = para->filter.mask[0];
  485. (*secfilter)->filter_mode[0] = para->filter.mode[0];
  486. (*secfilter)->filter_mask[1] = 0;
  487. (*secfilter)->filter_mask[2] = 0;
  488. filter->todo = 0;
  489. ret = filter->feed.sec->start_filtering(filter->feed.sec);
  490. if (ret < 0)
  491. return ret;
  492. dvb_dmxdev_filter_timer(filter);
  493. break;
  494. }
  495. case DMXDEV_TYPE_PES:
  496. {
  497. struct timespec timeout = { 0 };
  498. struct dmx_pes_filter_params *para = &filter->params.pes;
  499. dmx_output_t otype;
  500. int ret;
  501. int ts_type;
  502. enum dmx_ts_pes ts_pes;
  503. struct dmx_ts_feed **tsfeed = &filter->feed.ts;
  504. filter->feed.ts = NULL;
  505. otype = para->output;
  506. ts_pes = (enum dmx_ts_pes)para->pes_type;
  507. if (ts_pes < DMX_PES_OTHER)
  508. ts_type = TS_DECODER;
  509. else
  510. ts_type = 0;
  511. if (otype == DMX_OUT_TS_TAP)
  512. ts_type |= TS_PACKET;
  513. if (otype == DMX_OUT_TAP)
  514. ts_type |= TS_PAYLOAD_ONLY | TS_PACKET;
  515. ret = dmxdev->demux->allocate_ts_feed(dmxdev->demux,
  516. tsfeed,
  517. dvb_dmxdev_ts_callback);
  518. if (ret < 0)
  519. return ret;
  520. (*tsfeed)->priv = filter;
  521. ret = (*tsfeed)->set(*tsfeed, para->pid, ts_type, ts_pes,
  522. 32768, timeout);
  523. if (ret < 0) {
  524. dmxdev->demux->release_ts_feed(dmxdev->demux,
  525. *tsfeed);
  526. return ret;
  527. }
  528. ret = filter->feed.ts->start_filtering(filter->feed.ts);
  529. if (ret < 0) {
  530. dmxdev->demux->release_ts_feed(dmxdev->demux,
  531. *tsfeed);
  532. return ret;
  533. }
  534. break;
  535. }
  536. default:
  537. return -EINVAL;
  538. }
  539. dvb_dmxdev_filter_state_set(filter, DMXDEV_STATE_GO);
  540. return 0;
  541. }
  542. static int dvb_demux_open(struct inode *inode, struct file *file)
  543. {
  544. struct dvb_device *dvbdev = file->private_data;
  545. struct dmxdev *dmxdev = dvbdev->priv;
  546. int i;
  547. struct dmxdev_filter *dmxdevfilter;
  548. if (!dmxdev->filter)
  549. return -EINVAL;
  550. if (mutex_lock_interruptible(&dmxdev->mutex))
  551. return -ERESTARTSYS;
  552. for (i = 0; i < dmxdev->filternum; i++)
  553. if (dmxdev->filter[i].state == DMXDEV_STATE_FREE)
  554. break;
  555. if (i == dmxdev->filternum) {
  556. mutex_unlock(&dmxdev->mutex);
  557. return -EMFILE;
  558. }
  559. dmxdevfilter = &dmxdev->filter[i];
  560. mutex_init(&dmxdevfilter->mutex);
  561. file->private_data = dmxdevfilter;
  562. dvb_ringbuffer_init(&dmxdevfilter->buffer, NULL, 8192);
  563. dmxdevfilter->type = DMXDEV_TYPE_NONE;
  564. dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_ALLOCATED);
  565. dmxdevfilter->feed.ts = NULL;
  566. init_timer(&dmxdevfilter->timer);
  567. mutex_unlock(&dmxdev->mutex);
  568. return 0;
  569. }
  570. static int dvb_dmxdev_filter_free(struct dmxdev *dmxdev,
  571. struct dmxdev_filter *dmxdevfilter)
  572. {
  573. mutex_lock(&dmxdev->mutex);
  574. mutex_lock(&dmxdevfilter->mutex);
  575. dvb_dmxdev_filter_stop(dmxdevfilter);
  576. dvb_dmxdev_filter_reset(dmxdevfilter);
  577. if (dmxdevfilter->buffer.data) {
  578. void *mem = dmxdevfilter->buffer.data;
  579. spin_lock_irq(&dmxdev->lock);
  580. dmxdevfilter->buffer.data = NULL;
  581. spin_unlock_irq(&dmxdev->lock);
  582. vfree(mem);
  583. }
  584. dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_FREE);
  585. wake_up(&dmxdevfilter->buffer.queue);
  586. mutex_unlock(&dmxdevfilter->mutex);
  587. mutex_unlock(&dmxdev->mutex);
  588. return 0;
  589. }
  590. static inline void invert_mode(dmx_filter_t *filter)
  591. {
  592. int i;
  593. for (i = 0; i < DMX_FILTER_SIZE; i++)
  594. filter->mode[i] ^= 0xff;
  595. }
  596. static int dvb_dmxdev_filter_set(struct dmxdev *dmxdev,
  597. struct dmxdev_filter *dmxdevfilter,
  598. struct dmx_sct_filter_params *params)
  599. {
  600. dprintk("function : %s\n", __FUNCTION__);
  601. dvb_dmxdev_filter_stop(dmxdevfilter);
  602. dmxdevfilter->type = DMXDEV_TYPE_SEC;
  603. memcpy(&dmxdevfilter->params.sec,
  604. params, sizeof(struct dmx_sct_filter_params));
  605. invert_mode(&dmxdevfilter->params.sec.filter);
  606. dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_SET);
  607. if (params->flags & DMX_IMMEDIATE_START)
  608. return dvb_dmxdev_filter_start(dmxdevfilter);
  609. return 0;
  610. }
  611. static int dvb_dmxdev_pes_filter_set(struct dmxdev *dmxdev,
  612. struct dmxdev_filter *dmxdevfilter,
  613. struct dmx_pes_filter_params *params)
  614. {
  615. dvb_dmxdev_filter_stop(dmxdevfilter);
  616. if (params->pes_type > DMX_PES_OTHER || params->pes_type < 0)
  617. return -EINVAL;
  618. dmxdevfilter->type = DMXDEV_TYPE_PES;
  619. memcpy(&dmxdevfilter->params, params,
  620. sizeof(struct dmx_pes_filter_params));
  621. dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_SET);
  622. if (params->flags & DMX_IMMEDIATE_START)
  623. return dvb_dmxdev_filter_start(dmxdevfilter);
  624. return 0;
  625. }
  626. static ssize_t dvb_dmxdev_read_sec(struct dmxdev_filter *dfil,
  627. struct file *file, char __user *buf,
  628. size_t count, loff_t *ppos)
  629. {
  630. int result, hcount;
  631. int done = 0;
  632. if (dfil->todo <= 0) {
  633. hcount = 3 + dfil->todo;
  634. if (hcount > count)
  635. hcount = count;
  636. result = dvb_dmxdev_buffer_read(&dfil->buffer,
  637. file->f_flags & O_NONBLOCK,
  638. buf, hcount, ppos);
  639. if (result < 0) {
  640. dfil->todo = 0;
  641. return result;
  642. }
  643. if (copy_from_user(dfil->secheader - dfil->todo, buf, result))
  644. return -EFAULT;
  645. buf += result;
  646. done = result;
  647. count -= result;
  648. dfil->todo -= result;
  649. if (dfil->todo > -3)
  650. return done;
  651. dfil->todo = ((dfil->secheader[1] << 8) | dfil->secheader[2]) & 0xfff;
  652. if (!count)
  653. return done;
  654. }
  655. if (count > dfil->todo)
  656. count = dfil->todo;
  657. result = dvb_dmxdev_buffer_read(&dfil->buffer,
  658. file->f_flags & O_NONBLOCK,
  659. buf, count, ppos);
  660. if (result < 0)
  661. return result;
  662. dfil->todo -= result;
  663. return (result + done);
  664. }
  665. static ssize_t
  666. dvb_demux_read(struct file *file, char __user *buf, size_t count,
  667. loff_t *ppos)
  668. {
  669. struct dmxdev_filter *dmxdevfilter = file->private_data;
  670. int ret;
  671. if (mutex_lock_interruptible(&dmxdevfilter->mutex))
  672. return -ERESTARTSYS;
  673. if (dmxdevfilter->type == DMXDEV_TYPE_SEC)
  674. ret = dvb_dmxdev_read_sec(dmxdevfilter, file, buf, count, ppos);
  675. else
  676. ret = dvb_dmxdev_buffer_read(&dmxdevfilter->buffer,
  677. file->f_flags & O_NONBLOCK,
  678. buf, count, ppos);
  679. mutex_unlock(&dmxdevfilter->mutex);
  680. return ret;
  681. }
  682. static int dvb_demux_do_ioctl(struct inode *inode, struct file *file,
  683. unsigned int cmd, void *parg)
  684. {
  685. struct dmxdev_filter *dmxdevfilter = file->private_data;
  686. struct dmxdev *dmxdev = dmxdevfilter->dev;
  687. unsigned long arg = (unsigned long)parg;
  688. int ret = 0;
  689. if (mutex_lock_interruptible(&dmxdev->mutex))
  690. return -ERESTARTSYS;
  691. switch (cmd) {
  692. case DMX_START:
  693. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  694. mutex_unlock(&dmxdev->mutex);
  695. return -ERESTARTSYS;
  696. }
  697. if (dmxdevfilter->state < DMXDEV_STATE_SET)
  698. ret = -EINVAL;
  699. else
  700. ret = dvb_dmxdev_filter_start(dmxdevfilter);
  701. mutex_unlock(&dmxdevfilter->mutex);
  702. break;
  703. case DMX_STOP:
  704. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  705. mutex_unlock(&dmxdev->mutex);
  706. return -ERESTARTSYS;
  707. }
  708. ret = dvb_dmxdev_filter_stop(dmxdevfilter);
  709. mutex_unlock(&dmxdevfilter->mutex);
  710. break;
  711. case DMX_SET_FILTER:
  712. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  713. mutex_unlock(&dmxdev->mutex);
  714. return -ERESTARTSYS;
  715. }
  716. ret = dvb_dmxdev_filter_set(dmxdev, dmxdevfilter, parg);
  717. mutex_unlock(&dmxdevfilter->mutex);
  718. break;
  719. case DMX_SET_PES_FILTER:
  720. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  721. mutex_unlock(&dmxdev->mutex);
  722. return -ERESTARTSYS;
  723. }
  724. ret = dvb_dmxdev_pes_filter_set(dmxdev, dmxdevfilter, parg);
  725. mutex_unlock(&dmxdevfilter->mutex);
  726. break;
  727. case DMX_SET_BUFFER_SIZE:
  728. if (mutex_lock_interruptible(&dmxdevfilter->mutex)) {
  729. mutex_unlock(&dmxdev->mutex);
  730. return -ERESTARTSYS;
  731. }
  732. ret = dvb_dmxdev_set_buffer_size(dmxdevfilter, arg);
  733. mutex_unlock(&dmxdevfilter->mutex);
  734. break;
  735. case DMX_GET_PES_PIDS:
  736. if (!dmxdev->demux->get_pes_pids) {
  737. ret = -EINVAL;
  738. break;
  739. }
  740. dmxdev->demux->get_pes_pids(dmxdev->demux, parg);
  741. break;
  742. case DMX_GET_CAPS:
  743. if (!dmxdev->demux->get_caps) {
  744. ret = -EINVAL;
  745. break;
  746. }
  747. ret = dmxdev->demux->get_caps(dmxdev->demux, parg);
  748. break;
  749. case DMX_SET_SOURCE:
  750. if (!dmxdev->demux->set_source) {
  751. ret = -EINVAL;
  752. break;
  753. }
  754. ret = dmxdev->demux->set_source(dmxdev->demux, parg);
  755. break;
  756. case DMX_GET_STC:
  757. if (!dmxdev->demux->get_stc) {
  758. ret = -EINVAL;
  759. break;
  760. }
  761. ret = dmxdev->demux->get_stc(dmxdev->demux,
  762. ((struct dmx_stc *)parg)->num,
  763. &((struct dmx_stc *)parg)->stc,
  764. &((struct dmx_stc *)parg)->base);
  765. break;
  766. default:
  767. ret = -EINVAL;
  768. break;
  769. }
  770. mutex_unlock(&dmxdev->mutex);
  771. return ret;
  772. }
  773. static int dvb_demux_ioctl(struct inode *inode, struct file *file,
  774. unsigned int cmd, unsigned long arg)
  775. {
  776. return dvb_usercopy(inode, file, cmd, arg, dvb_demux_do_ioctl);
  777. }
  778. static unsigned int dvb_demux_poll(struct file *file, poll_table *wait)
  779. {
  780. struct dmxdev_filter *dmxdevfilter = file->private_data;
  781. unsigned int mask = 0;
  782. if (!dmxdevfilter)
  783. return -EINVAL;
  784. poll_wait(file, &dmxdevfilter->buffer.queue, wait);
  785. if (dmxdevfilter->state != DMXDEV_STATE_GO &&
  786. dmxdevfilter->state != DMXDEV_STATE_DONE &&
  787. dmxdevfilter->state != DMXDEV_STATE_TIMEDOUT)
  788. return 0;
  789. if (dmxdevfilter->buffer.error)
  790. mask |= (POLLIN | POLLRDNORM | POLLPRI | POLLERR);
  791. if (!dvb_ringbuffer_empty(&dmxdevfilter->buffer))
  792. mask |= (POLLIN | POLLRDNORM | POLLPRI);
  793. return mask;
  794. }
  795. static int dvb_demux_release(struct inode *inode, struct file *file)
  796. {
  797. struct dmxdev_filter *dmxdevfilter = file->private_data;
  798. struct dmxdev *dmxdev = dmxdevfilter->dev;
  799. return dvb_dmxdev_filter_free(dmxdev, dmxdevfilter);
  800. }
  801. static struct file_operations dvb_demux_fops = {
  802. .owner = THIS_MODULE,
  803. .read = dvb_demux_read,
  804. .ioctl = dvb_demux_ioctl,
  805. .open = dvb_demux_open,
  806. .release = dvb_demux_release,
  807. .poll = dvb_demux_poll,
  808. };
  809. static struct dvb_device dvbdev_demux = {
  810. .priv = NULL,
  811. .users = 1,
  812. .writers = 1,
  813. .fops = &dvb_demux_fops
  814. };
  815. static int dvb_dvr_do_ioctl(struct inode *inode, struct file *file,
  816. unsigned int cmd, void *parg)
  817. {
  818. struct dvb_device *dvbdev = file->private_data;
  819. struct dmxdev *dmxdev = dvbdev->priv;
  820. int ret;
  821. if (mutex_lock_interruptible(&dmxdev->mutex))
  822. return -ERESTARTSYS;
  823. switch (cmd) {
  824. case DMX_SET_BUFFER_SIZE:
  825. // FIXME: implement
  826. ret = 0;
  827. break;
  828. default:
  829. ret = -EINVAL;
  830. break;
  831. }
  832. mutex_unlock(&dmxdev->mutex);
  833. return ret;
  834. }
  835. static int dvb_dvr_ioctl(struct inode *inode, struct file *file,
  836. unsigned int cmd, unsigned long arg)
  837. {
  838. return dvb_usercopy(inode, file, cmd, arg, dvb_dvr_do_ioctl);
  839. }
  840. static unsigned int dvb_dvr_poll(struct file *file, poll_table *wait)
  841. {
  842. struct dvb_device *dvbdev = file->private_data;
  843. struct dmxdev *dmxdev = dvbdev->priv;
  844. unsigned int mask = 0;
  845. dprintk("function : %s\n", __FUNCTION__);
  846. poll_wait(file, &dmxdev->dvr_buffer.queue, wait);
  847. if ((file->f_flags & O_ACCMODE) == O_RDONLY) {
  848. if (dmxdev->dvr_buffer.error)
  849. mask |= (POLLIN | POLLRDNORM | POLLPRI | POLLERR);
  850. if (!dvb_ringbuffer_empty(&dmxdev->dvr_buffer))
  851. mask |= (POLLIN | POLLRDNORM | POLLPRI);
  852. } else
  853. mask |= (POLLOUT | POLLWRNORM | POLLPRI);
  854. return mask;
  855. }
  856. static struct file_operations dvb_dvr_fops = {
  857. .owner = THIS_MODULE,
  858. .read = dvb_dvr_read,
  859. .write = dvb_dvr_write,
  860. .ioctl = dvb_dvr_ioctl,
  861. .open = dvb_dvr_open,
  862. .release = dvb_dvr_release,
  863. .poll = dvb_dvr_poll,
  864. };
  865. static struct dvb_device dvbdev_dvr = {
  866. .priv = NULL,
  867. .readers = 1,
  868. .fops = &dvb_dvr_fops
  869. };
  870. int dvb_dmxdev_init(struct dmxdev *dmxdev, struct dvb_adapter *dvb_adapter)
  871. {
  872. int i;
  873. if (dmxdev->demux->open(dmxdev->demux) < 0)
  874. return -EUSERS;
  875. dmxdev->filter = vmalloc(dmxdev->filternum * sizeof(struct dmxdev_filter));
  876. if (!dmxdev->filter)
  877. return -ENOMEM;
  878. mutex_init(&dmxdev->mutex);
  879. spin_lock_init(&dmxdev->lock);
  880. for (i = 0; i < dmxdev->filternum; i++) {
  881. dmxdev->filter[i].dev = dmxdev;
  882. dmxdev->filter[i].buffer.data = NULL;
  883. dvb_dmxdev_filter_state_set(&dmxdev->filter[i],
  884. DMXDEV_STATE_FREE);
  885. }
  886. dvb_register_device(dvb_adapter, &dmxdev->dvbdev, &dvbdev_demux, dmxdev,
  887. DVB_DEVICE_DEMUX);
  888. dvb_register_device(dvb_adapter, &dmxdev->dvr_dvbdev, &dvbdev_dvr,
  889. dmxdev, DVB_DEVICE_DVR);
  890. dvb_ringbuffer_init(&dmxdev->dvr_buffer, NULL, 8192);
  891. return 0;
  892. }
  893. EXPORT_SYMBOL(dvb_dmxdev_init);
  894. void dvb_dmxdev_release(struct dmxdev *dmxdev)
  895. {
  896. dvb_unregister_device(dmxdev->dvbdev);
  897. dvb_unregister_device(dmxdev->dvr_dvbdev);
  898. vfree(dmxdev->filter);
  899. dmxdev->filter = NULL;
  900. dmxdev->demux->close(dmxdev->demux);
  901. }
  902. EXPORT_SYMBOL(dvb_dmxdev_release);