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.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