drivers/media/dvb/dvb-core/dvb_demux.c

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