?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Linux-2.6.12-rc2
[linux-2.6/next.git] / drivers / media / dvb / dvb-core / dvb_demux.c
blobac9889d222882bc6629d5a2c90b7405496d8f740
1 /*
2 * dvb_demux.c - DVB kernel demux API
3 *
4 * Copyright (C) 2000-2001 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/poll.h>
29 #include <linux/string.h>
30 #include <linux/crc32.h>
31 #include <asm/uaccess.h>
32
33 #include "dvb_demux.h"
34
35 #define NOBUFS
36 /*
37 ** #define DVB_DEMUX_SECTION_LOSS_LOG to monitor payload loss in the syslog
38 */
39 // #define DVB_DEMUX_SECTION_LOSS_LOG
40
41
42 static LIST_HEAD(dmx_muxs);
43
44
45 static int dmx_register_demux(struct dmx_demux *demux)
46 {
47 demux->users = 0;
48 list_add(&demux->reg_list, &dmx_muxs);
49 return 0;
50 }
51
52 static int dmx_unregister_demux(struct dmx_demux* demux)
53 {
54 struct list_head *pos, *n, *head=&dmx_muxs;
55
56 list_for_each_safe (pos, n, head) {
57 if (DMX_DIR_ENTRY(pos) == demux) {
58 if (demux->users>0)
59 return -EINVAL;
60 list_del(pos);
61 return 0;
62 }
63 }
64
65 return -ENODEV;
66 }
67
68
69 /******************************************************************************
70 * static inlined helper functions
71 ******************************************************************************/
72
73
74 static inline u16 section_length(const u8 *buf)
75 {
76 return 3+((buf[1]&0x0f)<<8)+buf[2];
77 }
78
79
80 static inline u16 ts_pid(const u8 *buf)
81 {
82 return ((buf[1]&0x1f)<<8)+buf[2];
83 }
84
85
86 static inline u8 payload(const u8 *tsp)
87 {
88 if (!(tsp[3] & 0x10)) // no payload?
89 return 0;
90 if (tsp[3] & 0x20) { // adaptation field?
91 if (tsp[4] > 183) // corrupted data?
92 return 0;
93 else
94 return 184-1-tsp[4];
95 }
96 return 184;
97 }
98
99
100 static u32 dvb_dmx_crc32 (struct dvb_demux_feed *f, const u8 *src, size_t len)
101 {
102 return (f->feed.sec.crc_val = crc32_be (f->feed.sec.crc_val, src, len));
103 }
104
105
106 static void dvb_dmx_memcopy (struct dvb_demux_feed *f, u8 *d, const u8 *s, size_t len)
107 {
108 memcpy (d, s, len);
109 }
110
111
112 /******************************************************************************
113 * Software filter functions
114 ******************************************************************************/
115
116 static inline int dvb_dmx_swfilter_payload (struct dvb_demux_feed *feed, const u8 *buf)
117 {
118 int count = payload(buf);
119 int p;
120 //int ccok;
121 //u8 cc;
122
123 if (count == 0)
124 return -1;
125
126 p = 188-count;
127
128 /*
129 cc=buf[3]&0x0f;
130 ccok=((dvbdmxfeed->cc+1)&0x0f)==cc ? 1 : 0;
131 dvbdmxfeed->cc=cc;
132 if (!ccok)
133 printk("missed packet!\n");
134 */
135
136 if (buf[1] & 0x40) // PUSI ?
137 feed->peslen = 0xfffa;
138
139 feed->peslen += count;
140
141 return feed->cb.ts (&buf[p], count, NULL, 0, &feed->feed.ts, DMX_OK);
142 }
143
144
145 static int dvb_dmx_swfilter_sectionfilter (struct dvb_demux_feed *feed,
146 struct dvb_demux_filter *f)
147 {
148 u8 neq = 0;
149 int i;
150
151 for (i=0; i<DVB_DEMUX_MASK_MAX; i++) {
152 u8 xor = f->filter.filter_value[i] ^ feed->feed.sec.secbuf[i];
153
154 if (f->maskandmode[i] & xor)
155 return 0;
156
157 neq |= f->maskandnotmode[i] & xor;
158 }
159
160 if (f->doneq && !neq)
161 return 0;
162
163 return feed->cb.sec (feed->feed.sec.secbuf, feed->feed.sec.seclen,
164 NULL, 0, &f->filter, DMX_OK);
165 }
166
167
168 static inline int dvb_dmx_swfilter_section_feed (struct dvb_demux_feed *feed)
169 {
170 struct dvb_demux *demux = feed->demux;
171 struct dvb_demux_filter *f = feed->filter;
172 struct dmx_section_feed *sec = &feed->feed.sec;
173 int section_syntax_indicator;
174
175 if (!sec->is_filtering)
176 return 0;
177
178 if (!f)
179 return 0;
180
181 if (sec->check_crc) {
182 section_syntax_indicator = ((sec->secbuf[1] & 0x80) != 0);
183 if (section_syntax_indicator &&
184 demux->check_crc32(feed, sec->secbuf, sec->seclen))
185 return -1;
186 }
187
188 do {
189 if (dvb_dmx_swfilter_sectionfilter(feed, f) < 0)
190 return -1;
191 } while ((f = f->next) && sec->is_filtering);
192
193 sec->seclen = 0;
194
195 return 0;
196 }
197
198
199 static void dvb_dmx_swfilter_section_new(struct dvb_demux_feed *feed)
200 {
201 struct dmx_section_feed *sec = &feed->feed.sec;
202
203 #ifdef DVB_DEMUX_SECTION_LOSS_LOG
204 if(sec->secbufp < sec->tsfeedp)
205 {
206 int i, n = sec->tsfeedp - sec->secbufp;
207
208 /* section padding is done with 0xff bytes entirely.
209 ** due to speed reasons, we won't check all of them
210 ** but just first and last
211 */
212 if(sec->secbuf[0] != 0xff || sec->secbuf[n-1] != 0xff)
213 {
214 printk("dvb_demux.c section ts padding loss: %d/%d\n",
215 n, sec->tsfeedp);
216 printk("dvb_demux.c pad data:");
217 for(i = 0; i < n; i++)
218 printk(" %02x", sec->secbuf[i]);
219 printk("\n");
220 }
221 }
222 #endif
223
224 sec->tsfeedp = sec->secbufp = sec->seclen = 0;
225 sec->secbuf = sec->secbuf_base;
226 }
227
228 /*
229 ** Losless Section Demux 1.4.1 by Emard
230 ** Valsecchi Patrick:
231 ** - middle of section A (no PUSI)
232 ** - end of section A and start of section B
233 ** (with PUSI pointing to the start of the second section)
234 **
235 ** In this case, without feed->pusi_seen you'll receive a garbage section
236 ** consisting of the end of section A. Basically because tsfeedp
237 ** is incemented and the use=0 condition is not raised
238 ** when the second packet arrives.
239 **
240 ** Fix:
241 ** when demux is started, let feed->pusi_seen = 0 to
242 ** prevent initial feeding of garbage from the end of
243 ** previous section. When you for the first time see PUSI=1
244 ** then set feed->pusi_seen = 1
245 */
246 static int dvb_dmx_swfilter_section_copy_dump(struct dvb_demux_feed *feed, const u8 *buf, u8 len)
247 {
248 struct dvb_demux *demux = feed->demux;
249 struct dmx_section_feed *sec = &feed->feed.sec;
250 u16 limit, seclen, n;
251
252 if(sec->tsfeedp >= DMX_MAX_SECFEED_SIZE)
253 return 0;
254
255 if(sec->tsfeedp + len > DMX_MAX_SECFEED_SIZE)
256 {
257 #ifdef DVB_DEMUX_SECTION_LOSS_LOG
258 printk("dvb_demux.c section buffer full loss: %d/%d\n",
259 sec->tsfeedp + len - DMX_MAX_SECFEED_SIZE, DMX_MAX_SECFEED_SIZE);
260 #endif
261 len = DMX_MAX_SECFEED_SIZE - sec->tsfeedp;
262 }
263
264 if(len <= 0)
265 return 0;
266
267 demux->memcopy(feed, sec->secbuf_base + sec->tsfeedp, buf, len);
268 sec->tsfeedp += len;
269
270 /* -----------------------------------------------------
271 ** Dump all the sections we can find in the data (Emard)
272 */
273
274 limit = sec->tsfeedp;
275 if(limit > DMX_MAX_SECFEED_SIZE)
276 return -1; /* internal error should never happen */
277
278 /* to be sure always set secbuf */
279 sec->secbuf = sec->secbuf_base + sec->secbufp;
280
281 for(n = 0; sec->secbufp + 2 < limit; n++)
282 {
283 seclen = section_length(sec->secbuf);
284 if(seclen <= 0 || seclen > DMX_MAX_SECFEED_SIZE
285 || seclen + sec->secbufp > limit)
286 return 0;
287 sec->seclen = seclen;
288 sec->crc_val = ~0;
289 /* dump [secbuf .. secbuf+seclen) */
290 if(feed->pusi_seen)
291 dvb_dmx_swfilter_section_feed(feed);
292 #ifdef DVB_DEMUX_SECTION_LOSS_LOG
293 else
294 printk("dvb_demux.c pusi not seen, discarding section data\n");
295 #endif
296 sec->secbufp += seclen; /* secbufp and secbuf moving together is */
297 sec->secbuf += seclen; /* redundand but saves pointer arithmetic */
298 }
299
300 return 0;
301 }
302
303
304 static int dvb_dmx_swfilter_section_packet(struct dvb_demux_feed *feed, const u8 *buf)
305 {
306 u8 p, count;
307 int ccok, dc_i = 0;
308 u8 cc;
309
310 count = payload(buf);
311
312 if (count == 0) /* count == 0 if no payload or out of range */
313 return -1;
314
315 p = 188 - count; /* payload start */
316
317 cc = buf[3] & 0x0f;
318 ccok = ((feed->cc + 1) & 0x0f) == cc;
319 feed->cc = cc;
320
321 if (buf[3] & 0x20) {
322 /* adaption field present, check for discontinuity_indicator */
323 if ((buf[4] > 0) && (buf[5] & 0x80))
324 dc_i = 1;
325 }
326
327 if (!ccok || dc_i) {
328 #ifdef DVB_DEMUX_SECTION_LOSS_LOG
329 printk("dvb_demux.c discontinuity detected %d bytes lost\n", count);
330 /* those bytes under sume circumstances will again be reported
331 ** in the following dvb_dmx_swfilter_section_new
332 */
333 #endif
334 /* Discontinuity detected. Reset pusi_seen = 0 to
335 ** stop feeding of suspicious data until next PUSI=1 arrives
336 */
337 feed->pusi_seen = 0;
338 dvb_dmx_swfilter_section_new(feed);
339 return 0;
340 }
341
342 if (buf[1] & 0x40) {
343 // PUSI=1 (is set), section boundary is here
344 if (count > 1 && buf[p] < count) {
345 const u8 *before = buf+p+1;
346 u8 before_len = buf[p];
347 const u8 *after = before+before_len;
348 u8 after_len = count-1-before_len;
349
350 dvb_dmx_swfilter_section_copy_dump(feed, before, before_len);
351 /* before start of new section, set pusi_seen = 1 */
352 feed->pusi_seen = 1;
353 dvb_dmx_swfilter_section_new(feed);
354 dvb_dmx_swfilter_section_copy_dump(feed, after, after_len);
355 }
356 #ifdef DVB_DEMUX_SECTION_LOSS_LOG
357 else
358 if (count > 0)
359 printk("dvb_demux.c PUSI=1 but %d bytes lost\n", count);
360 #endif
361 } else {
362 // PUSI=0 (is not set), no section boundary
363 const u8 *entire = buf+p;
364 u8 entire_len = count;
365
366 dvb_dmx_swfilter_section_copy_dump(feed, entire, entire_len);
367 }
368 return 0;
369 }
370
371
372 static inline void dvb_dmx_swfilter_packet_type(struct dvb_demux_feed *feed, const u8 *buf)
373 {
374 switch(feed->type) {
375 case DMX_TYPE_TS:
376 if (!feed->feed.ts.is_filtering)
377 break;
378 if (feed->ts_type & TS_PACKET) {
379 if (feed->ts_type & TS_PAYLOAD_ONLY)
380 dvb_dmx_swfilter_payload(feed, buf);
381 else
382 feed->cb.ts(buf, 188, NULL, 0, &feed->feed.ts, DMX_OK);
383 }
384 if (feed->ts_type & TS_DECODER)
385 if (feed->demux->write_to_decoder)
386 feed->demux->write_to_decoder(feed, buf, 188);
387 break;
388
389 case DMX_TYPE_SEC:
390 if (!feed->feed.sec.is_filtering)
391 break;
392 if (dvb_dmx_swfilter_section_packet(feed, buf) < 0)
393 feed->feed.sec.seclen = feed->feed.sec.secbufp=0;
394 break;
395
396 default:
397 break;
398 }
399 }
400
401 #define DVR_FEED(f) \
402 (((f)->type == DMX_TYPE_TS) && \
403 ((f)->feed.ts.is_filtering) && \
404 (((f)->ts_type & (TS_PACKET|TS_PAYLOAD_ONLY)) == TS_PACKET))
405
406 static void dvb_dmx_swfilter_packet(struct dvb_demux *demux, const u8 *buf)
407 {
408 struct dvb_demux_feed *feed;
409 struct list_head *pos, *head=&demux->feed_list;
410 u16 pid = ts_pid(buf);
411 int dvr_done = 0;
412
413 list_for_each(pos, head) {
414 feed = list_entry(pos, struct dvb_demux_feed, list_head);
415
416 if ((feed->pid != pid) && (feed->pid != 0x2000))
417 continue;
418
419 /* copy each packet only once to the dvr device, even
420 * if a PID is in multiple filters (e.g. video + PCR) */
421 if ((DVR_FEED(feed)) && (dvr_done++))
422 continue;
423
424 if (feed->pid == pid) {
425 dvb_dmx_swfilter_packet_type(feed, buf);
426 if (DVR_FEED(feed))
427 continue;
428 }
429
430 if (feed->pid == 0x2000)
431 feed->cb.ts(buf, 188, NULL, 0, &feed->feed.ts, DMX_OK);
432 }
433 }
434
435 void dvb_dmx_swfilter_packets(struct dvb_demux *demux, const u8 *buf, size_t count)
436 {
437 spin_lock(&demux->lock);
438
439 while (count--) {
440 if(buf[0] == 0x47) {
441 dvb_dmx_swfilter_packet(demux, buf);
442 }
443 buf += 188;
444 }
445
446 spin_unlock(&demux->lock);
447 }
448 EXPORT_SYMBOL(dvb_dmx_swfilter_packets);
449
450
451 void dvb_dmx_swfilter(struct dvb_demux *demux, const u8 *buf, size_t count)
452 {
453 int p = 0, i, j;
454
455 spin_lock(&demux->lock);
456
457 if ((i = demux->tsbufp)) {
458 if (count < (j=188-i)) {
459 memcpy(&demux->tsbuf[i], buf, count);
460 demux->tsbufp += count;
461 goto bailout;
462 }
463 memcpy(&demux->tsbuf[i], buf, j);
464 if (demux->tsbuf[0] == 0x47)
465 dvb_dmx_swfilter_packet(demux, demux->tsbuf);
466 demux->tsbufp = 0;
467 p += j;
468 }
469
470 while (p < count) {
471 if (buf[p] == 0x47) {
472 if (count-p >= 188) {
473 dvb_dmx_swfilter_packet(demux, buf+p);
474 p += 188;
475 } else {
476 i = count-p;
477 memcpy(demux->tsbuf, buf+p, i);
478 demux->tsbufp=i;
479 goto bailout;
480 }
481 } else
482 p++;
483 }
484
485 bailout:
486 spin_unlock(&demux->lock);
487 }
488 EXPORT_SYMBOL(dvb_dmx_swfilter);
489
490 void dvb_dmx_swfilter_204(struct dvb_demux *demux, const u8 *buf, size_t count)
491 {
492 int p = 0,i, j;
493 u8 tmppack[188];
494 spin_lock(&demux->lock);
495
496 if ((i = demux->tsbufp)) {
497 if (count < (j=204-i)) {
498 memcpy(&demux->tsbuf[i], buf, count);
499 demux->tsbufp += count;
500 goto bailout;
501 }
502 memcpy(&demux->tsbuf[i], buf, j);
503 if ((demux->tsbuf[0] == 0x47)|(demux->tsbuf[0]==0xB8)) {
504 memcpy(tmppack, demux->tsbuf, 188);
505 if (tmppack[0] == 0xB8) tmppack[0] = 0x47;
506 dvb_dmx_swfilter_packet(demux, tmppack);
507 }
508 demux->tsbufp = 0;
509 p += j;
510 }
511
512 while (p < count) {
513 if ((buf[p] == 0x47)|(buf[p] == 0xB8)) {
514 if (count-p >= 204) {
515 memcpy(tmppack, buf+p, 188);
516 if (tmppack[0] == 0xB8) tmppack[0] = 0x47;
517 dvb_dmx_swfilter_packet(demux, tmppack);
518 p += 204;
519 } else {
520 i = count-p;
521 memcpy(demux->tsbuf, buf+p, i);
522 demux->tsbufp=i;
523 goto bailout;
524 }
525 } else {
526 p++;
527 }
528 }
529
530 bailout:
531 spin_unlock(&demux->lock);
532 }
533 EXPORT_SYMBOL(dvb_dmx_swfilter_204);
534
535
536 static struct dvb_demux_filter * dvb_dmx_filter_alloc(struct dvb_demux *demux)
537 {
538 int i;
539
540 for (i=0; i<demux->filternum; i++)
541 if (demux->filter[i].state == DMX_STATE_FREE)
542 break;
543
544 if (i == demux->filternum)
545 return NULL;
546
547 demux->filter[i].state = DMX_STATE_ALLOCATED;
548
549 return &demux->filter[i];
550 }
551
552 static struct dvb_demux_feed * dvb_dmx_feed_alloc(struct dvb_demux *demux)
553 {
554 int i;
555
556 for (i=0; i<demux->feednum; i++)
557 if (demux->feed[i].state == DMX_STATE_FREE)
558 break;
559
560 if (i == demux->feednum)
561 return NULL;
562
563 demux->feed[i].state = DMX_STATE_ALLOCATED;
564
565 return &demux->feed[i];
566 }
567
568 static int dvb_demux_feed_find(struct dvb_demux_feed *feed)
569 {
570 struct dvb_demux_feed *entry;
571
572 list_for_each_entry(entry, &feed->demux->feed_list, list_head)
573 if (entry == feed)
574 return 1;
575
576 return 0;
577 }
578
579 static void dvb_demux_feed_add(struct dvb_demux_feed *feed)
580 {
581 spin_lock_irq(&feed->demux->lock);
582 if (dvb_demux_feed_find(feed)) {
583 printk(KERN_ERR "%s: feed already in list (type=%x state=%x pid=%x)\n",
584 __FUNCTION__, feed->type, feed->state, feed->pid);
585 goto out;
586 }
587
588 list_add(&feed->list_head, &feed->demux->feed_list);
589 out:
590 spin_unlock_irq(&feed->demux->lock);
591 }
592
593 static void dvb_demux_feed_del(struct dvb_demux_feed *feed)
594 {
595 spin_lock_irq(&feed->demux->lock);
596 if (!(dvb_demux_feed_find(feed))) {
597 printk(KERN_ERR "%s: feed not in list (type=%x state=%x pid=%x)\n",
598 __FUNCTION__, feed->type, feed->state, feed->pid);
599 goto out;
600 }
601
602 list_del(&feed->list_head);
603 out:
604 spin_unlock_irq(&feed->demux->lock);
605 }
606
607 static int dmx_ts_feed_set (struct dmx_ts_feed* ts_feed, u16 pid, int ts_type,
608 enum dmx_ts_pes pes_type, size_t callback_length,
609 size_t circular_buffer_size, int descramble,
610 struct timespec timeout)
611 {
612 struct dvb_demux_feed *feed = (struct dvb_demux_feed *) ts_feed;
613 struct dvb_demux *demux = feed->demux;
614
615 if (pid > DMX_MAX_PID)
616 return -EINVAL;
617
618 if (down_interruptible (&demux->mutex))
619 return -ERESTARTSYS;
620
621 if (ts_type & TS_DECODER) {
622 if (pes_type >= DMX_TS_PES_OTHER) {
623 up(&demux->mutex);
624 return -EINVAL;
625 }
626
627 if (demux->pesfilter[pes_type] &&
628 demux->pesfilter[pes_type] != feed) {
629 up(&demux->mutex);
630 return -EINVAL;
631 }
632
633 demux->pesfilter[pes_type] = feed;
634 demux->pids[pes_type] = pid;
635 }
636
637 dvb_demux_feed_add(feed);
638
639 feed->pid = pid;
640 feed->buffer_size = circular_buffer_size;
641 feed->descramble = descramble;
642 feed->timeout = timeout;
643 feed->cb_length = callback_length;
644 feed->ts_type = ts_type;
645 feed->pes_type = pes_type;
646
647 if (feed->descramble) {
648 up(&demux->mutex);
649 return -ENOSYS;
650 }
651
652 if (feed->buffer_size) {
653 #ifdef NOBUFS
654 feed->buffer=NULL;
655 #else
656 feed->buffer = vmalloc(feed->buffer_size);
657 if (!feed->buffer) {
658 up(&demux->mutex);
659 return -ENOMEM;
660 }
661 #endif
662 }
663
664 feed->state = DMX_STATE_READY;
665 up(&demux->mutex);
666
667 return 0;
668 }
669
670
671 static int dmx_ts_feed_start_filtering(struct dmx_ts_feed* ts_feed)
672 {
673 struct dvb_demux_feed *feed = (struct dvb_demux_feed *) ts_feed;
674 struct dvb_demux *demux = feed->demux;
675 int ret;
676
677 if (down_interruptible (&demux->mutex))
678 return -ERESTARTSYS;
679
680 if (feed->state != DMX_STATE_READY || feed->type != DMX_TYPE_TS) {
681 up(&demux->mutex);
682 return -EINVAL;
683 }
684
685 if (!demux->start_feed) {
686 up(&demux->mutex);
687 return -ENODEV;
688 }
689
690 if ((ret = demux->start_feed(feed)) < 0) {
691 up(&demux->mutex);
692 return ret;
693 }
694
695 spin_lock_irq(&demux->lock);
696 ts_feed->is_filtering = 1;
697 feed->state = DMX_STATE_GO;
698 spin_unlock_irq(&demux->lock);
699 up(&demux->mutex);
700
701 return 0;
702 }
703
704 static int dmx_ts_feed_stop_filtering(struct dmx_ts_feed* ts_feed)
705 {
706 struct dvb_demux_feed *feed = (struct dvb_demux_feed *) ts_feed;
707 struct dvb_demux *demux = feed->demux;
708 int ret;
709
710 if (down_interruptible (&demux->mutex))
711 return -ERESTARTSYS;
712
713 if (feed->state < DMX_STATE_GO) {
714 up(&demux->mutex);
715 return -EINVAL;
716 }
717
718 if (!demux->stop_feed) {
719 up(&demux->mutex);
720 return -ENODEV;
721 }
722
723 ret = demux->stop_feed(feed);
724
725 spin_lock_irq(&demux->lock);
726 ts_feed->is_filtering = 0;
727 feed->state = DMX_STATE_ALLOCATED;
728 spin_unlock_irq(&demux->lock);
729 up(&demux->mutex);
730
731 return ret;
732 }
733
734 static int dvbdmx_allocate_ts_feed (struct dmx_demux *dmx, struct dmx_ts_feed **ts_feed,
735 dmx_ts_cb callback)
736 {
737 struct dvb_demux *demux = (struct dvb_demux *) dmx;
738 struct dvb_demux_feed *feed;
739
740 if (down_interruptible (&demux->mutex))
741 return -ERESTARTSYS;
742
743 if (!(feed = dvb_dmx_feed_alloc(demux))) {
744 up(&demux->mutex);
745 return -EBUSY;
746 }
747
748 feed->type = DMX_TYPE_TS;
749 feed->cb.ts = callback;
750 feed->demux = demux;
751 feed->pid = 0xffff;
752 feed->peslen = 0xfffa;
753 feed->buffer = NULL;
754
755 (*ts_feed) = &feed->feed.ts;
756 (*ts_feed)->parent = dmx;
757 (*ts_feed)->priv = NULL;
758 (*ts_feed)->is_filtering = 0;
759 (*ts_feed)->start_filtering = dmx_ts_feed_start_filtering;
760 (*ts_feed)->stop_filtering = dmx_ts_feed_stop_filtering;
761 (*ts_feed)->set = dmx_ts_feed_set;
762
763
764 if (!(feed->filter = dvb_dmx_filter_alloc(demux))) {
765 feed->state = DMX_STATE_FREE;
766 up(&demux->mutex);
767 return -EBUSY;
768 }
769
770 feed->filter->type = DMX_TYPE_TS;
771 feed->filter->feed = feed;
772 feed->filter->state = DMX_STATE_READY;
773
774 up(&demux->mutex);
775
776 return 0;
777 }
778
779 static int dvbdmx_release_ts_feed(struct dmx_demux *dmx, struct dmx_ts_feed *ts_feed)
780 {
781 struct dvb_demux *demux = (struct dvb_demux *) dmx;
782 struct dvb_demux_feed *feed = (struct dvb_demux_feed *) ts_feed;
783
784 if (down_interruptible (&demux->mutex))
785 return -ERESTARTSYS;
786
787 if (feed->state == DMX_STATE_FREE) {
788 up(&demux->mutex);
789 return -EINVAL;
790 }
791
792 #ifndef NOBUFS
793 vfree(feed->buffer);
794 feed->buffer=0;
795 #endif
796
797 feed->state = DMX_STATE_FREE;
798 feed->filter->state = DMX_STATE_FREE;
799
800 dvb_demux_feed_del(feed);
801
802 feed->pid = 0xffff;
803
804 if (feed->ts_type & TS_DECODER && feed->pes_type < DMX_TS_PES_OTHER)
805 demux->pesfilter[feed->pes_type] = NULL;
806
807 up(&demux->mutex);
808 return 0;
809 }
810
811
812 /******************************************************************************
813 * dmx_section_feed API calls
814 ******************************************************************************/
815
816 static int dmx_section_feed_allocate_filter(struct dmx_section_feed* feed,
817 struct dmx_section_filter** filter)
818 {
819 struct dvb_demux_feed *dvbdmxfeed = (struct dvb_demux_feed *) feed;
820 struct dvb_demux *dvbdemux = dvbdmxfeed->demux;
821 struct dvb_demux_filter *dvbdmxfilter;
822
823 if (down_interruptible (&dvbdemux->mutex))
824 return -ERESTARTSYS;
825
826 dvbdmxfilter = dvb_dmx_filter_alloc(dvbdemux);
827 if (!dvbdmxfilter) {
828 up(&dvbdemux->mutex);
829 return -EBUSY;
830 }
831
832 spin_lock_irq(&dvbdemux->lock);
833 *filter = &dvbdmxfilter->filter;
834 (*filter)->parent = feed;
835 (*filter)->priv = NULL;
836 dvbdmxfilter->feed = dvbdmxfeed;
837 dvbdmxfilter->type = DMX_TYPE_SEC;
838 dvbdmxfilter->state = DMX_STATE_READY;
839 dvbdmxfilter->next = dvbdmxfeed->filter;
840 dvbdmxfeed->filter = dvbdmxfilter;
841 spin_unlock_irq(&dvbdemux->lock);
842
843 up(&dvbdemux->mutex);
844 return 0;
845 }
846
847
848 static int dmx_section_feed_set(struct dmx_section_feed* feed,
849 u16 pid, size_t circular_buffer_size,
850 int descramble, int check_crc)
851 {
852 struct dvb_demux_feed *dvbdmxfeed = (struct dvb_demux_feed *) feed;
853 struct dvb_demux *dvbdmx = dvbdmxfeed->demux;
854
855 if (pid > 0x1fff)
856 return -EINVAL;
857
858 if (down_interruptible (&dvbdmx->mutex))
859 return -ERESTARTSYS;
860
861 dvb_demux_feed_add(dvbdmxfeed);
862
863 dvbdmxfeed->pid = pid;
864 dvbdmxfeed->buffer_size = circular_buffer_size;
865 dvbdmxfeed->descramble = descramble;
866 if (dvbdmxfeed->descramble) {
867 up(&dvbdmx->mutex);
868 return -ENOSYS;
869 }
870
871 dvbdmxfeed->feed.sec.check_crc = check_crc;
872
873 #ifdef NOBUFS
874 dvbdmxfeed->buffer = NULL;
875 #else
876 dvbdmxfeed->buffer=vmalloc(dvbdmxfeed->buffer_size);
877 if (!dvbdmxfeed->buffer) {
878 up(&dvbdmx->mutex);
879 return -ENOMEM;
880 }
881 #endif
882
883 dvbdmxfeed->state = DMX_STATE_READY;
884 up(&dvbdmx->mutex);
885 return 0;
886 }
887
888
889 static void prepare_secfilters(struct dvb_demux_feed *dvbdmxfeed)
890 {
891 int i;
892 struct dvb_demux_filter *f;
893 struct dmx_section_filter *sf;
894 u8 mask, mode, doneq;
895
896 if (!(f=dvbdmxfeed->filter))
897 return;
898 do {
899 sf = &f->filter;
900 doneq = 0;
901 for (i=0; i<DVB_DEMUX_MASK_MAX; i++) {
902 mode = sf->filter_mode[i];
903 mask = sf->filter_mask[i];
904 f->maskandmode[i] = mask & mode;
905 doneq |= f->maskandnotmode[i] = mask & ~mode;
906 }
907 f->doneq = doneq ? 1 : 0;
908 } while ((f = f->next));
909 }
910
911
912 static int dmx_section_feed_start_filtering(struct dmx_section_feed *feed)
913 {
914 struct dvb_demux_feed *dvbdmxfeed = (struct dvb_demux_feed *) feed;
915 struct dvb_demux *dvbdmx = dvbdmxfeed->demux;
916 int ret;
917
918 if (down_interruptible (&dvbdmx->mutex))
919 return -ERESTARTSYS;
920
921 if (feed->is_filtering) {
922 up(&dvbdmx->mutex);
923 return -EBUSY;
924 }
925
926 if (!dvbdmxfeed->filter) {
927 up(&dvbdmx->mutex);
928 return -EINVAL;
929 }
930
931 dvbdmxfeed->feed.sec.tsfeedp = 0;
932 dvbdmxfeed->feed.sec.secbuf = dvbdmxfeed->feed.sec.secbuf_base;
933 dvbdmxfeed->feed.sec.secbufp = 0;
934 dvbdmxfeed->feed.sec.seclen = 0;
935
936 if (!dvbdmx->start_feed) {
937 up(&dvbdmx->mutex);
938 return -ENODEV;
939 }
940
941 prepare_secfilters(dvbdmxfeed);
942
943 if ((ret = dvbdmx->start_feed(dvbdmxfeed)) < 0) {
944 up(&dvbdmx->mutex);
945 return ret;
946 }
947
948 spin_lock_irq(&dvbdmx->lock);
949 feed->is_filtering = 1;
950 dvbdmxfeed->state = DMX_STATE_GO;
951 spin_unlock_irq(&dvbdmx->lock);
952
953 up(&dvbdmx->mutex);
954 return 0;
955 }
956
957
958 static int dmx_section_feed_stop_filtering(struct dmx_section_feed* feed)
959 {
960 struct dvb_demux_feed *dvbdmxfeed = (struct dvb_demux_feed *) feed;
961 struct dvb_demux *dvbdmx = dvbdmxfeed->demux;
962 int ret;
963
964 if (down_interruptible (&dvbdmx->mutex))
965 return -ERESTARTSYS;
966
967 if (!dvbdmx->stop_feed) {
968 up(&dvbdmx->mutex);
969 return -ENODEV;
970 }
971
972 ret = dvbdmx->stop_feed(dvbdmxfeed);
973
974 spin_lock_irq(&dvbdmx->lock);
975 dvbdmxfeed->state = DMX_STATE_READY;
976 feed->is_filtering = 0;
977 spin_unlock_irq(&dvbdmx->lock);
978
979 up(&dvbdmx->mutex);
980 return ret;
981 }
982
983
984 static int dmx_section_feed_release_filter(struct dmx_section_feed *feed,
985 struct dmx_section_filter* filter)
986 {
987 struct dvb_demux_filter *dvbdmxfilter = (struct dvb_demux_filter *) filter, *f;
988 struct dvb_demux_feed *dvbdmxfeed = (struct dvb_demux_feed *) feed;
989 struct dvb_demux *dvbdmx = dvbdmxfeed->demux;
990
991 if (down_interruptible (&dvbdmx->mutex))
992 return -ERESTARTSYS;
993
994 if (dvbdmxfilter->feed != dvbdmxfeed) {
995 up(&dvbdmx->mutex);
996 return -EINVAL;
997 }
998
999 if (feed->is_filtering)
1000 feed->stop_filtering(feed);
1001
1002 spin_lock_irq(&dvbdmx->lock);
1003 f = dvbdmxfeed->filter;
1004
1005 if (f == dvbdmxfilter) {
1006 dvbdmxfeed->filter = dvbdmxfilter->next;
1007 } else {
1008 while(f->next != dvbdmxfilter)
1009 f = f->next;
1010 f->next = f->next->next;
1011 }
1012
1013 dvbdmxfilter->state = DMX_STATE_FREE;
1014 spin_unlock_irq(&dvbdmx->lock);
1015 up(&dvbdmx->mutex);
1016 return 0;
1017 }
1018
1019 static int dvbdmx_allocate_section_feed(struct dmx_demux *demux,
1020 struct dmx_section_feed **feed,
1021 dmx_section_cb callback)
1022 {
1023 struct dvb_demux *dvbdmx = (struct dvb_demux *) demux;
1024 struct dvb_demux_feed *dvbdmxfeed;
1025
1026 if (down_interruptible (&dvbdmx->mutex))
1027 return -ERESTARTSYS;
1028
1029 if (!(dvbdmxfeed = dvb_dmx_feed_alloc(dvbdmx))) {
1030 up(&dvbdmx->mutex);
1031 return -EBUSY;
1032 }
1033
1034 dvbdmxfeed->type = DMX_TYPE_SEC;
1035 dvbdmxfeed->cb.sec = callback;
1036 dvbdmxfeed->demux = dvbdmx;
1037 dvbdmxfeed->pid = 0xffff;
1038 dvbdmxfeed->feed.sec.secbuf = dvbdmxfeed->feed.sec.secbuf_base;
1039 dvbdmxfeed->feed.sec.secbufp = dvbdmxfeed->feed.sec.seclen = 0;
1040 dvbdmxfeed->feed.sec.tsfeedp = 0;
1041 dvbdmxfeed->filter = NULL;
1042 dvbdmxfeed->buffer = NULL;
1043
1044 (*feed)=&dvbdmxfeed->feed.sec;
1045 (*feed)->is_filtering = 0;
1046 (*feed)->parent = demux;
1047 (*feed)->priv = NULL;
1048
1049 (*feed)->set = dmx_section_feed_set;
1050 (*feed)->allocate_filter = dmx_section_feed_allocate_filter;
1051 (*feed)->start_filtering = dmx_section_feed_start_filtering;
1052 (*feed)->stop_filtering = dmx_section_feed_stop_filtering;
1053 (*feed)->release_filter = dmx_section_feed_release_filter;
1054
1055 up(&dvbdmx->mutex);
1056 return 0;
1057 }
1058
1059 static int dvbdmx_release_section_feed(struct dmx_demux *demux,
1060 struct dmx_section_feed *feed)
1061 {
1062 struct dvb_demux_feed *dvbdmxfeed = (struct dvb_demux_feed *) feed;
1063 struct dvb_demux *dvbdmx = (struct dvb_demux *) demux;
1064
1065 if (down_interruptible (&dvbdmx->mutex))
1066 return -ERESTARTSYS;
1067
1068 if (dvbdmxfeed->state==DMX_STATE_FREE) {
1069 up(&dvbdmx->mutex);
1070 return -EINVAL;
1071 }
1072 #ifndef NOBUFS
1073 vfree(dvbdmxfeed->buffer);
1074 dvbdmxfeed->buffer=0;
1075 #endif
1076 dvbdmxfeed->state=DMX_STATE_FREE;
1077
1078 dvb_demux_feed_del(dvbdmxfeed);
1079
1080 dvbdmxfeed->pid = 0xffff;
1081
1082 up(&dvbdmx->mutex);
1083 return 0;
1084 }
1085
1086
1087 /******************************************************************************
1088 * dvb_demux kernel data API calls
1089 ******************************************************************************/
1090
1091 static int dvbdmx_open(struct dmx_demux *demux)
1092 {
1093 struct dvb_demux *dvbdemux = (struct dvb_demux *) demux;
1094
1095 if (dvbdemux->users >= MAX_DVB_DEMUX_USERS)
1096 return -EUSERS;
1097
1098 dvbdemux->users++;
1099 return 0;
1100 }
1101
1102
1103 static int dvbdmx_close(struct dmx_demux *demux)
1104 {
1105 struct dvb_demux *dvbdemux = (struct dvb_demux *) demux;
1106
1107 if (dvbdemux->users == 0)
1108 return -ENODEV;
1109
1110 dvbdemux->users--;
1111 //FIXME: release any unneeded resources if users==0
1112 return 0;
1113 }
1114
1115
1116 static int dvbdmx_write(struct dmx_demux *demux, const char *buf, size_t count)
1117 {
1118 struct dvb_demux *dvbdemux=(struct dvb_demux *) demux;
1119
1120 if ((!demux->frontend) || (demux->frontend->source != DMX_MEMORY_FE))
1121 return -EINVAL;
1122
1123 if (down_interruptible (&dvbdemux->mutex))
1124 return -ERESTARTSYS;
1125 dvb_dmx_swfilter(dvbdemux, buf, count);
1126 up(&dvbdemux->mutex);
1127
1128 if (signal_pending(current))
1129 return -EINTR;
1130 return count;
1131 }
1132
1133
1134 static int dvbdmx_add_frontend(struct dmx_demux *demux, struct dmx_frontend *frontend)
1135 {
1136 struct dvb_demux *dvbdemux = (struct dvb_demux *) demux;
1137 struct list_head *head = &dvbdemux->frontend_list;
1138
1139 list_add(&(frontend->connectivity_list), head);
1140
1141 return 0;
1142 }
1143
1144
1145 static int dvbdmx_remove_frontend(struct dmx_demux *demux, struct dmx_frontend *frontend)
1146 {
1147 struct dvb_demux *dvbdemux = (struct dvb_demux *) demux;
1148 struct list_head *pos, *n, *head = &dvbdemux->frontend_list;
1149
1150 list_for_each_safe (pos, n, head) {
1151 if (DMX_FE_ENTRY(pos) == frontend) {
1152 list_del(pos);
1153 return 0;
1154 }
1155 }
1156
1157 return -ENODEV;
1158 }
1159
1160
1161 static struct list_head * dvbdmx_get_frontends(struct dmx_demux *demux)
1162 {
1163 struct dvb_demux *dvbdemux = (struct dvb_demux *) demux;
1164
1165 if (list_empty(&dvbdemux->frontend_list))
1166 return NULL;
1167 return &dvbdemux->frontend_list;
1168 }
1169
1170
1171 static int dvbdmx_connect_frontend(struct dmx_demux *demux, struct dmx_frontend *frontend)
1172 {
1173 struct dvb_demux *dvbdemux = (struct dvb_demux *) demux;
1174
1175 if (demux->frontend)
1176 return -EINVAL;
1177
1178 if (down_interruptible (&dvbdemux->mutex))
1179 return -ERESTARTSYS;
1180
1181 demux->frontend = frontend;
1182 up(&dvbdemux->mutex);
1183 return 0;
1184 }
1185
1186
1187 static int dvbdmx_disconnect_frontend(struct dmx_demux *demux)
1188 {
1189 struct dvb_demux *dvbdemux = (struct dvb_demux *) demux;
1190
1191 if (down_interruptible (&dvbdemux->mutex))
1192 return -ERESTARTSYS;
1193
1194 demux->frontend = NULL;
1195 up(&dvbdemux->mutex);
1196 return 0;
1197 }
1198
1199
1200 static int dvbdmx_get_pes_pids(struct dmx_demux *demux, u16 *pids)
1201 {
1202 struct dvb_demux *dvbdemux = (struct dvb_demux *) demux;
1203
1204 memcpy(pids, dvbdemux->pids, 5*sizeof(u16));
1205 return 0;
1206 }
1207
1208
1209 int dvb_dmx_init(struct dvb_demux *dvbdemux)
1210 {
1211 int i, err;
1212 struct dmx_demux *dmx = &dvbdemux->dmx;
1213
1214 dvbdemux->users = 0;
1215 dvbdemux->filter = vmalloc(dvbdemux->filternum*sizeof(struct dvb_demux_filter));
1216
1217 if (!dvbdemux->filter)
1218 return -ENOMEM;
1219
1220 dvbdemux->feed = vmalloc(dvbdemux->feednum*sizeof(struct dvb_demux_feed));
1221 if (!dvbdemux->feed) {
1222 vfree(dvbdemux->filter);
1223 return -ENOMEM;
1224 }
1225 for (i=0; i<dvbdemux->filternum; i++) {
1226 dvbdemux->filter[i].state = DMX_STATE_FREE;
1227 dvbdemux->filter[i].index = i;
1228 }
1229 for (i=0; i<dvbdemux->feednum; i++) {
1230 dvbdemux->feed[i].state = DMX_STATE_FREE;
1231 dvbdemux->feed[i].index = i;
1232 }
1233 dvbdemux->frontend_list.next=
1234 dvbdemux->frontend_list.prev=
1235 &dvbdemux->frontend_list;
1236 for (i=0; i<DMX_TS_PES_OTHER; i++) {
1237 dvbdemux->pesfilter[i] = NULL;
1238 dvbdemux->pids[i] = 0xffff;
1239 }
1240
1241 INIT_LIST_HEAD(&dvbdemux->feed_list);
1242
1243 dvbdemux->playing = 0;
1244 dvbdemux->recording = 0;
1245 dvbdemux->tsbufp = 0;
1246
1247 if (!dvbdemux->check_crc32)
1248 dvbdemux->check_crc32 = dvb_dmx_crc32;
1249
1250 if (!dvbdemux->memcopy)
1251 dvbdemux->memcopy = dvb_dmx_memcopy;
1252
1253 dmx->frontend = NULL;
1254 dmx->reg_list.prev = dmx->reg_list.next = &dmx->reg_list;
1255 dmx->priv = (void *) dvbdemux;
1256 dmx->open = dvbdmx_open;
1257 dmx->close = dvbdmx_close;
1258 dmx->write = dvbdmx_write;
1259 dmx->allocate_ts_feed = dvbdmx_allocate_ts_feed;
1260 dmx->release_ts_feed = dvbdmx_release_ts_feed;
1261 dmx->allocate_section_feed = dvbdmx_allocate_section_feed;
1262 dmx->release_section_feed = dvbdmx_release_section_feed;
1263
1264 dmx->descramble_mac_address = NULL;
1265 dmx->descramble_section_payload = NULL;
1266
1267 dmx->add_frontend = dvbdmx_add_frontend;
1268 dmx->remove_frontend = dvbdmx_remove_frontend;
1269 dmx->get_frontends = dvbdmx_get_frontends;
1270 dmx->connect_frontend = dvbdmx_connect_frontend;
1271 dmx->disconnect_frontend = dvbdmx_disconnect_frontend;
1272 dmx->get_pes_pids = dvbdmx_get_pes_pids;
1273
1274 sema_init(&dvbdemux->mutex, 1);
1275 spin_lock_init(&dvbdemux->lock);
1276
1277 if ((err = dmx_register_demux(dmx)) < 0)
1278 return err;
1279
1280 return 0;
1281 }
1282 EXPORT_SYMBOL(dvb_dmx_init);
1283
1284
1285 int dvb_dmx_release(struct dvb_demux *dvbdemux)
1286 {
1287 struct dmx_demux *dmx = &dvbdemux->dmx;
1288
1289 dmx_unregister_demux(dmx);
1290 vfree(dvbdemux->filter);
1291 vfree(dvbdemux->feed);
1292 return 0;
1293 }
1294 EXPORT_SYMBOL(dvb_dmx_release);
linux-next