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