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