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