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