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

   1 /*
   2  * dmxdev.c - DVB demultiplexer device
   3  *
   4  * Copyright (C) 2000 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/moduleparam.h>
  29 #include <linux/sched.h>
  30 #include <linux/poll.h>
  31 #include <linux/ioctl.h>
  32 #include <linux/wait.h>
  33 #include <asm/uaccess.h>
  34 #include <asm/system.h>
  35
  36 #include "dmxdev.h"
  37
  38 static int debug;
  39
  40 module_param(debug, int, 0644);
  41 MODULE_PARM_DESC(debug, "Turn on/off debugging (default:off).");
  42
  43 #define dprintk if (debug) printk
  44
  45 static inline struct dmxdev_filter *
  46 dvb_dmxdev_file_to_filter(struct file *file)
  47 {
  48         return (struct dmxdev_filter *) file->private_data;
  49 }
  50
  51 static inline void dvb_dmxdev_buffer_init(struct dmxdev_buffer *buffer)
  52 {
  53         buffer->data=NULL;
  54         buffer->size=8192;
  55         buffer->pread=0;
  56         buffer->pwrite=0;
  57         buffer->error=0;
  58         init_waitqueue_head(&buffer->queue);
  59 }
  60
  61 static inline int dvb_dmxdev_buffer_write(struct dmxdev_buffer *buf, const u8 *src, int len)
  62 {
  63         int split;
  64         int free;
  65         int todo;
  66
  67         if (!len)
  68                 return 0;
  69         if (!buf->data)
  70                 return 0;
  71
  72         free=buf->pread-buf->pwrite;
  73         split=0;
  74         if (free<=0) {
  75                 free+=buf->size;
  76                 split=buf->size-buf->pwrite;
  77         }
  78         if (len>=free) {
  79                 dprintk("dmxdev: buffer overflow\n");
  80                 return -1;
  81         }
  82         if (split>=len)
  83                 split=0;
  84         todo=len;
  85         if (split) {
  86                 memcpy(buf->data + buf->pwrite, src, split);
  87                 todo-=split;
  88                 buf->pwrite=0;
  89         }
  90         memcpy(buf->data + buf->pwrite, src+split, todo);
  91         buf->pwrite=(buf->pwrite+todo)%buf->size;
  92         return len;
  93 }
  94
  95 static ssize_t dvb_dmxdev_buffer_read(struct dmxdev_buffer *src,
  96                 int non_blocking, char __user *buf, size_t count, loff_t *ppos)
  97 {
  98         unsigned long todo=count;
  99         int split, avail, error;
 100
 101         if (!src->data)
 102                 return 0;
 103
 104         if ((error=src->error)) {
 105                 src->pwrite=src->pread;
 106                 src->error=0;
 107                 return error;
 108         }
 109
 110         if (non_blocking && (src->pwrite==src->pread))
 111                 return -EWOULDBLOCK;
 112
 113         while (todo>0) {
 114                 if (non_blocking && (src->pwrite==src->pread))
 115                         return (count-todo) ? (count-todo) : -EWOULDBLOCK;
 116
 117                 if (wait_event_interruptible(src->queue,
 118                                              (src->pread!=src->pwrite) ||
 119                                              (src->error))<0)
 120                         return count-todo;
 121
 122                 if ((error=src->error)) {
 123                         src->pwrite=src->pread;
 124                         src->error=0;
 125                         return error;
 126                 }
 127
 128                 split=src->size;
 129                 avail=src->pwrite - src->pread;
 130                 if (avail<0) {
 131                         avail+=src->size;
 132                         split=src->size - src->pread;
 133                 }
 134                 if (avail>todo)
 135                         avail=todo;
 136                 if (split<avail) {
 137                         if (copy_to_user(buf, src->data+src->pread, split))
 138                                   return -EFAULT;
 139                         buf+=split;
 140                         src->pread=0;
 141                         todo-=split;
 142                         avail-=split;
 143                 }
 144                 if (avail) {
 145                         if (copy_to_user(buf, src->data+src->pread, avail))
 146                                 return -EFAULT;
 147                         src->pread = (src->pread + avail) % src->size;
 148                         todo-=avail;
 149                         buf+=avail;
 150                 }
 151         }
 152         return count;
 153 }
 154
 155 static struct dmx_frontend * get_fe(struct dmx_demux *demux, int type)
 156 {
 157         struct list_head *head, *pos;
 158
 159         head=demux->get_frontends(demux);
 160         if (!head)
 161                 return NULL;
 162         list_for_each(pos, head)
 163                 if (DMX_FE_ENTRY(pos)->source==type)
 164                         return DMX_FE_ENTRY(pos);
 165
 166         return NULL;
 167 }
 168
 169 static inline void dvb_dmxdev_dvr_state_set(struct dmxdev_dvr *dmxdevdvr, int state)
 170 {
 171         spin_lock_irq(&dmxdevdvr->dev->lock);
 172         dmxdevdvr->state=state;
 173         spin_unlock_irq(&dmxdevdvr->dev->lock);
 174 }
 175
 176 static int dvb_dvr_open(struct inode *inode, struct file *file)
 177 {
 178         struct dvb_device *dvbdev=(struct dvb_device *) file->private_data;
 179         struct dmxdev *dmxdev=(struct dmxdev *) dvbdev->priv;
 180         struct dmx_frontend *front;
 181
 182         dprintk ("function : %s\n", __FUNCTION__);
 183
 184         if (down_interruptible (&dmxdev->mutex))
 185                 return -ERESTARTSYS;
 186
 187         if ((file->f_flags&O_ACCMODE)==O_RDWR) {
 188                 if (!(dmxdev->capabilities&DMXDEV_CAP_DUPLEX)) {
 189                         up(&dmxdev->mutex);
 190                         return -EOPNOTSUPP;
 191                 }
 192         }
 193
 194         if ((file->f_flags&O_ACCMODE)==O_RDONLY) {
 195               dvb_dmxdev_buffer_init(&dmxdev->dvr_buffer);
 196               dmxdev->dvr_buffer.size=DVR_BUFFER_SIZE;
 197               dmxdev->dvr_buffer.data=vmalloc(DVR_BUFFER_SIZE);
 198               if (!dmxdev->dvr_buffer.data) {
 199                       up(&dmxdev->mutex);
 200                       return -ENOMEM;
 201               }
 202         }
 203
 204         if ((file->f_flags&O_ACCMODE)==O_WRONLY) {
 205                 dmxdev->dvr_orig_fe=dmxdev->demux->frontend;
 206
 207                 if (!dmxdev->demux->write) {
 208                         up(&dmxdev->mutex);
 209                         return -EOPNOTSUPP;
 210                 }
 211
 212                 front=get_fe(dmxdev->demux, DMX_MEMORY_FE);
 213
 214                 if (!front) {
 215                         up(&dmxdev->mutex);
 216                         return -EINVAL;
 217                 }
 218                 dmxdev->demux->disconnect_frontend(dmxdev->demux);
 219                 dmxdev->demux->connect_frontend(dmxdev->demux, front);
 220         }
 221         up(&dmxdev->mutex);
 222         return 0;
 223 }
 224
 225 static int dvb_dvr_release(struct inode *inode, struct file *file)
 226 {
 227         struct dvb_device *dvbdev=(struct dvb_device *) file->private_data;
 228         struct dmxdev *dmxdev=(struct dmxdev *) dvbdev->priv;
 229
 230         if (down_interruptible (&dmxdev->mutex))
 231                 return -ERESTARTSYS;
 232
 233         if ((file->f_flags&O_ACCMODE)==O_WRONLY) {
 234                 dmxdev->demux->disconnect_frontend(dmxdev->demux);
 235                 dmxdev->demux->connect_frontend(dmxdev->demux,
 236                                                 dmxdev->dvr_orig_fe);
 237         }
 238         if ((file->f_flags&O_ACCMODE)==O_RDONLY) {
 239                 if (dmxdev->dvr_buffer.data) {
 240                         void *mem=dmxdev->dvr_buffer.data;
 241                         mb();
 242                         spin_lock_irq(&dmxdev->lock);
 243                         dmxdev->dvr_buffer.data=NULL;
 244                         spin_unlock_irq(&dmxdev->lock);
 245                         vfree(mem);
 246                 }
 247         }
 248         up(&dmxdev->mutex);
 249         return 0;
 250 }
 251
 252 static ssize_t dvb_dvr_write(struct file *file, const char __user *buf,
 253                 size_t count, loff_t *ppos)
 254 {
 255         struct dvb_device *dvbdev=(struct dvb_device *) file->private_data;
 256         struct dmxdev *dmxdev=(struct dmxdev *) dvbdev->priv;
 257         int ret;
 258
 259         if (!dmxdev->demux->write)
 260                 return -EOPNOTSUPP;
 261         if ((file->f_flags&O_ACCMODE)!=O_WRONLY)
 262                 return -EINVAL;
 263         if (down_interruptible (&dmxdev->mutex))
 264                 return -ERESTARTSYS;
 265         ret=dmxdev->demux->write(dmxdev->demux, buf, count);
 266         up(&dmxdev->mutex);
 267         return ret;
 268 }
 269
 270 static ssize_t dvb_dvr_read(struct file *file, char __user *buf, size_t count,
 271                 loff_t *ppos)
 272 {
 273         struct dvb_device *dvbdev=(struct dvb_device *) file->private_data;
 274         struct dmxdev *dmxdev=(struct dmxdev *) dvbdev->priv;
 275         int ret;
 276
 277         //down(&dmxdev->mutex);
 278         ret= dvb_dmxdev_buffer_read(&dmxdev->dvr_buffer,
 279                               file->f_flags&O_NONBLOCK,
 280                               buf, count, ppos);
 281         //up(&dmxdev->mutex);
 282         return ret;
 283 }
 284
 285 static inline void dvb_dmxdev_filter_state_set(struct dmxdev_filter *dmxdevfilter, int state)
 286 {
 287         spin_lock_irq(&dmxdevfilter->dev->lock);
 288         dmxdevfilter->state=state;
 289         spin_unlock_irq(&dmxdevfilter->dev->lock);
 290 }
 291
 292 static int dvb_dmxdev_set_buffer_size(struct dmxdev_filter *dmxdevfilter, unsigned long size)
 293 {
 294         struct dmxdev_buffer *buf=&dmxdevfilter->buffer;
 295         void *mem;
 296
 297         if (buf->size==size)
 298                 return 0;
 299         if (dmxdevfilter->state>=DMXDEV_STATE_GO)
 300                 return -EBUSY;
 301         spin_lock_irq(&dmxdevfilter->dev->lock);
 302         mem=buf->data;
 303         buf->data=NULL;
 304         buf->size=size;
 305         buf->pwrite=buf->pread=0;
 306         spin_unlock_irq(&dmxdevfilter->dev->lock);
 307         if (mem)
 308                 vfree(mem);
 309
 310         if (buf->size) {
 311                 mem=vmalloc(dmxdevfilter->buffer.size);
 312                 if (!mem)
 313                         return -ENOMEM;
 314                 spin_lock_irq(&dmxdevfilter->dev->lock);
 315                 buf->data=mem;
 316                 spin_unlock_irq(&dmxdevfilter->dev->lock);
 317         }
 318         return 0;
 319 }
 320
 321 static void dvb_dmxdev_filter_timeout(unsigned long data)
 322 {
 323         struct dmxdev_filter *dmxdevfilter=(struct dmxdev_filter *)data;
 324
 325         dmxdevfilter->buffer.error=-ETIMEDOUT;
 326         spin_lock_irq(&dmxdevfilter->dev->lock);
 327         dmxdevfilter->state=DMXDEV_STATE_TIMEDOUT;
 328         spin_unlock_irq(&dmxdevfilter->dev->lock);
 329         wake_up(&dmxdevfilter->buffer.queue);
 330 }
 331
 332 static void dvb_dmxdev_filter_timer(struct dmxdev_filter *dmxdevfilter)
 333 {
 334         struct dmx_sct_filter_params *para=&dmxdevfilter->params.sec;
 335
 336         del_timer(&dmxdevfilter->timer);
 337         if (para->timeout) {
 338                 dmxdevfilter->timer.function=dvb_dmxdev_filter_timeout;
 339                 dmxdevfilter->timer.data=(unsigned long) dmxdevfilter;
 340                 dmxdevfilter->timer.expires=jiffies+1+(HZ/2+HZ*para->timeout)/1000;
 341                 add_timer(&dmxdevfilter->timer);
 342         }
 343 }
 344
 345 static int dvb_dmxdev_section_callback(const u8 *buffer1, size_t buffer1_len,
 346                             const u8 *buffer2, size_t buffer2_len,
 347                             struct dmx_section_filter *filter, enum dmx_success success)
 348 {
 349         struct dmxdev_filter *dmxdevfilter=(struct dmxdev_filter *) filter->priv;
 350         int ret;
 351
 352         if (dmxdevfilter->buffer.error) {
 353                 wake_up(&dmxdevfilter->buffer.queue);
 354                 return 0;
 355         }
 356         spin_lock(&dmxdevfilter->dev->lock);
 357         if (dmxdevfilter->state!=DMXDEV_STATE_GO) {
 358                 spin_unlock(&dmxdevfilter->dev->lock);
 359                 return 0;
 360         }
 361         del_timer(&dmxdevfilter->timer);
 362         dprintk("dmxdev: section callback %02x %02x %02x %02x %02x %02x\n",
 363                 buffer1[0], buffer1[1],
 364                 buffer1[2], buffer1[3],
 365                 buffer1[4], buffer1[5]);
 366         ret=dvb_dmxdev_buffer_write(&dmxdevfilter->buffer, buffer1, buffer1_len);
 367         if (ret==buffer1_len) {
 368                 ret=dvb_dmxdev_buffer_write(&dmxdevfilter->buffer, buffer2, buffer2_len);
 369         }
 370         if (ret<0) {
 371                 dmxdevfilter->buffer.pwrite=dmxdevfilter->buffer.pread;
 372                 dmxdevfilter->buffer.error=-EOVERFLOW;
 373         }
 374         if (dmxdevfilter->params.sec.flags&DMX_ONESHOT)
 375                 dmxdevfilter->state=DMXDEV_STATE_DONE;
 376         spin_unlock(&dmxdevfilter->dev->lock);
 377         wake_up(&dmxdevfilter->buffer.queue);
 378         return 0;
 379 }
 380
 381 static int dvb_dmxdev_ts_callback(const u8 *buffer1, size_t buffer1_len,
 382                        const u8 *buffer2, size_t buffer2_len,
 383                        struct dmx_ts_feed *feed, enum dmx_success success)
 384 {
 385         struct dmxdev_filter *dmxdevfilter=(struct dmxdev_filter *) feed->priv;
 386         struct dmxdev_buffer *buffer;
 387         int ret;
 388
 389         spin_lock(&dmxdevfilter->dev->lock);
 390         if (dmxdevfilter->params.pes.output==DMX_OUT_DECODER) {
 391                 spin_unlock(&dmxdevfilter->dev->lock);
 392                 return 0;
 393         }
 394
 395         if (dmxdevfilter->params.pes.output==DMX_OUT_TAP)
 396                 buffer=&dmxdevfilter->buffer;
 397         else
 398                 buffer=&dmxdevfilter->dev->dvr_buffer;
 399         if (buffer->error) {
 400                 spin_unlock(&dmxdevfilter->dev->lock);
 401                 wake_up(&buffer->queue);
 402                 return 0;
 403         }
 404         ret=dvb_dmxdev_buffer_write(buffer, buffer1, buffer1_len);
 405         if (ret==buffer1_len)
 406                 ret=dvb_dmxdev_buffer_write(buffer, buffer2, buffer2_len);
 407         if (ret<0) {
 408                 buffer->pwrite=buffer->pread;
 409                 buffer->error=-EOVERFLOW;
 410         }
 411         spin_unlock(&dmxdevfilter->dev->lock);
 412         wake_up(&buffer->queue);
 413         return 0;
 414 }
 415
 416
 417 /* stop feed but only mark the specified filter as stopped (state set) */
 418
 419 static int dvb_dmxdev_feed_stop(struct dmxdev_filter *dmxdevfilter)
 420 {
 421         dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_SET);
 422
 423         switch (dmxdevfilter->type) {
 424         case DMXDEV_TYPE_SEC:
 425                 del_timer(&dmxdevfilter->timer);
 426                 dmxdevfilter->feed.sec->stop_filtering(dmxdevfilter->feed.sec);
 427                 break;
 428         case DMXDEV_TYPE_PES:
 429                 dmxdevfilter->feed.ts->stop_filtering(dmxdevfilter->feed.ts);
 430                 break;
 431         default:
 432                 return -EINVAL;
 433         }
 434         return 0;
 435 }
 436
 437
 438 /* start feed associated with the specified filter */
 439
 440 static int dvb_dmxdev_feed_start(struct dmxdev_filter *filter)
 441 {
 442         dvb_dmxdev_filter_state_set (filter, DMXDEV_STATE_GO);
 443
 444         switch (filter->type) {
 445         case DMXDEV_TYPE_SEC:
 446                 return filter->feed.sec->start_filtering(filter->feed.sec);
 447                 break;
 448         case DMXDEV_TYPE_PES:
 449                 return filter->feed.ts->start_filtering(filter->feed.ts);
 450                 break;
 451         default:
 452                 return -EINVAL;
 453         }
 454
 455         return 0;
 456 }
 457
 458
 459 /* restart section feed if it has filters left associated with it,
 460    otherwise release the feed */
 461
 462 static int dvb_dmxdev_feed_restart(struct dmxdev_filter *filter)
 463 {
 464         int i;
 465         struct dmxdev *dmxdev = filter->dev;
 466         u16 pid = filter->params.sec.pid;
 467
 468         for (i=0; i<dmxdev->filternum; i++)
 469                 if (dmxdev->filter[i].state>=DMXDEV_STATE_GO &&
 470                     dmxdev->filter[i].type==DMXDEV_TYPE_SEC &&
 471                     dmxdev->filter[i].pid==pid) {
 472                         dvb_dmxdev_feed_start(&dmxdev->filter[i]);
 473                         return 0;
 474                 }
 475
 476         filter->dev->demux->release_section_feed(dmxdev->demux, filter->feed.sec);
 477
 478         return 0;
 479 }
 480
 481 static int dvb_dmxdev_filter_stop(struct dmxdev_filter *dmxdevfilter)
 482 {
 483         if (dmxdevfilter->state<DMXDEV_STATE_GO)
 484                 return 0;
 485
 486         switch (dmxdevfilter->type) {
 487         case DMXDEV_TYPE_SEC:
 488                 if (!dmxdevfilter->feed.sec)
 489                         break;
 490                 dvb_dmxdev_feed_stop(dmxdevfilter);
 491                 if (dmxdevfilter->filter.sec)
 492                         dmxdevfilter->feed.sec->
 493                                 release_filter(dmxdevfilter->feed.sec,
 494                                                dmxdevfilter->filter.sec);
 495                 dvb_dmxdev_feed_restart(dmxdevfilter);
 496                 dmxdevfilter->feed.sec=NULL;
 497                 break;
 498         case DMXDEV_TYPE_PES:
 499                 if (!dmxdevfilter->feed.ts)
 500                         break;
 501                 dvb_dmxdev_feed_stop(dmxdevfilter);
 502                 dmxdevfilter->dev->demux->
 503                         release_ts_feed(dmxdevfilter->dev->demux,
 504                                         dmxdevfilter->feed.ts);
 505                 dmxdevfilter->feed.ts=NULL;
 506                 break;
 507         default:
 508                 if (dmxdevfilter->state==DMXDEV_STATE_ALLOCATED)
 509                         return 0;
 510                 return -EINVAL;
 511         }
 512         dmxdevfilter->buffer.pwrite=dmxdevfilter->buffer.pread=0;
 513         return 0;
 514 }
 515
 516 static inline int dvb_dmxdev_filter_reset(struct dmxdev_filter *dmxdevfilter)
 517 {
 518         if (dmxdevfilter->state<DMXDEV_STATE_SET)
 519                 return 0;
 520
 521         dmxdevfilter->type=DMXDEV_TYPE_NONE;
 522         dmxdevfilter->pid=0xffff;
 523         dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_ALLOCATED);
 524         return 0;
 525 }
 526
 527 static int dvb_dmxdev_filter_start(struct dmxdev_filter *filter)
 528 {
 529         struct dmxdev *dmxdev = filter->dev;
 530         void *mem;
 531         int ret, i;
 532
 533         if (filter->state < DMXDEV_STATE_SET)
 534                 return -EINVAL;
 535
 536         if (filter->state >= DMXDEV_STATE_GO)
 537                 dvb_dmxdev_filter_stop(filter);
 538
 539         if (!(mem = filter->buffer.data)) {
 540                 mem = vmalloc(filter->buffer.size);
 541                 spin_lock_irq(&filter->dev->lock);
 542                 filter->buffer.data=mem;
 543                 spin_unlock_irq(&filter->dev->lock);
 544                 if (!filter->buffer.data)
 545                         return -ENOMEM;
 546         }
 547
 548         filter->buffer.pwrite = filter->buffer.pread = 0;
 549
 550         switch (filter->type) {
 551         case DMXDEV_TYPE_SEC:
 552         {
 553                 struct dmx_sct_filter_params *para=&filter->params.sec;
 554                 struct dmx_section_filter **secfilter=&filter->filter.sec;
 555                 struct dmx_section_feed **secfeed=&filter->feed.sec;
 556
 557                 *secfilter=NULL;
 558                 *secfeed=NULL;
 559
 560                 /* find active filter/feed with same PID */
 561                 for (i=0; i<dmxdev->filternum; i++) {
 562                         if (dmxdev->filter[i].state >= DMXDEV_STATE_GO &&
 563                             dmxdev->filter[i].pid == para->pid &&
 564                             dmxdev->filter[i].type == DMXDEV_TYPE_SEC) {
 565                                 *secfeed = dmxdev->filter[i].feed.sec;
 566                                 break;
 567                         }
 568                 }
 569
 570                 /* if no feed found, try to allocate new one */
 571                 if (!*secfeed) {
 572                         ret=dmxdev->demux->allocate_section_feed(dmxdev->demux,
 573                                                                  secfeed,
 574                                                    dvb_dmxdev_section_callback);
 575                         if (ret<0) {
 576                                 printk ("DVB (%s): could not alloc feed\n",
 577                                         __FUNCTION__);
 578                                 return ret;
 579                         }
 580
 581                         ret=(*secfeed)->set(*secfeed, para->pid, 32768, 0,
 582                                             (para->flags & DMX_CHECK_CRC) ? 1 : 0);
 583
 584                         if (ret<0) {
 585                                 printk ("DVB (%s): could not set feed\n",
 586                                         __FUNCTION__);
 587                                 dvb_dmxdev_feed_restart(filter);
 588                                 return ret;
 589                         }
 590                 } else {
 591                         dvb_dmxdev_feed_stop(filter);
 592                 }
 593
 594                 ret=(*secfeed)->allocate_filter(*secfeed, secfilter);
 595
 596                 if (ret < 0) {
 597                         dvb_dmxdev_feed_restart(filter);
 598                         filter->feed.sec->start_filtering(*secfeed);
 599                         dprintk ("could not get filter\n");
 600                         return ret;
 601                 }
 602
 603                 (*secfilter)->priv = filter;
 604
 605                 memcpy(&((*secfilter)->filter_value[3]),
 606                        &(para->filter.filter[1]), DMX_FILTER_SIZE-1);
 607                 memcpy(&(*secfilter)->filter_mask[3],
 608                        &para->filter.mask[1], DMX_FILTER_SIZE-1);
 609                 memcpy(&(*secfilter)->filter_mode[3],
 610                        &para->filter.mode[1], DMX_FILTER_SIZE-1);
 611
 612                 (*secfilter)->filter_value[0]=para->filter.filter[0];
 613                 (*secfilter)->filter_mask[0]=para->filter.mask[0];
 614                 (*secfilter)->filter_mode[0]=para->filter.mode[0];
 615                 (*secfilter)->filter_mask[1]=0;
 616                 (*secfilter)->filter_mask[2]=0;
 617
 618                 filter->todo = 0;
 619
 620                 ret = filter->feed.sec->start_filtering (filter->feed.sec);
 621
 622                 if (ret < 0)
 623                         return ret;
 624
 625                 dvb_dmxdev_filter_timer(filter);
 626                 break;
 627         }
 628
 629         case DMXDEV_TYPE_PES:
 630         {
 631                 struct timespec timeout = { 0 };
 632                 struct dmx_pes_filter_params *para = &filter->params.pes;
 633                 dmx_output_t otype;
 634                 int ret;
 635                 int ts_type;
 636                 enum dmx_ts_pes ts_pes;
 637                 struct dmx_ts_feed **tsfeed = &filter->feed.ts;
 638
 639                 filter->feed.ts = NULL;
 640                 otype=para->output;
 641
 642                 ts_pes=(enum dmx_ts_pes) para->pes_type;
 643
 644                 if (ts_pes<DMX_PES_OTHER)
 645                         ts_type=TS_DECODER;
 646                 else
 647                         ts_type=0;
 648
 649                 if (otype == DMX_OUT_TS_TAP)
 650                         ts_type |= TS_PACKET;
 651
 652                 if (otype == DMX_OUT_TAP)
 653                         ts_type |= TS_PAYLOAD_ONLY|TS_PACKET;
 654
 655                 ret=dmxdev->demux->allocate_ts_feed(dmxdev->demux,
 656                                                     tsfeed,
 657                                                     dvb_dmxdev_ts_callback);
 658                 if (ret<0)
 659                         return ret;
 660
 661                 (*tsfeed)->priv = (void *) filter;
 662
 663                 ret = (*tsfeed)->set(*tsfeed, para->pid, ts_type, ts_pes,
 664                                      188, 32768, 0, timeout);
 665
 666                 if (ret < 0) {
 667                         dmxdev->demux->release_ts_feed(dmxdev->demux, *tsfeed);
 668                         return ret;
 669                 }
 670
 671                 ret = filter->feed.ts->start_filtering(filter->feed.ts);
 672
 673                 if (ret < 0)
 674                         return ret;
 675
 676                 break;
 677         }
 678         default:
 679                 return -EINVAL;
 680         }
 681
 682         dvb_dmxdev_filter_state_set(filter, DMXDEV_STATE_GO);
 683         return 0;
 684 }
 685
 686 static int dvb_demux_open(struct inode *inode, struct file *file)
 687 {
 688         struct dvb_device *dvbdev=(struct dvb_device *) file->private_data;
 689         struct dmxdev *dmxdev=(struct dmxdev *) dvbdev->priv;
 690         int i;
 691         struct dmxdev_filter *dmxdevfilter;
 692
 693         if (!dmxdev->filter)
 694                 return -EINVAL;
 695
 696         if (down_interruptible(&dmxdev->mutex))
 697                 return -ERESTARTSYS;
 698
 699         for (i=0; i<dmxdev->filternum; i++)
 700                 if (dmxdev->filter[i].state==DMXDEV_STATE_FREE)
 701                         break;
 702
 703         if (i==dmxdev->filternum) {
 704                 up(&dmxdev->mutex);
 705                 return -EMFILE;
 706         }
 707
 708         dmxdevfilter=&dmxdev->filter[i];
 709         sema_init(&dmxdevfilter->mutex, 1);
 710         dmxdevfilter->dvbdev=dmxdev->dvbdev;
 711         file->private_data=dmxdevfilter;
 712
 713         dvb_dmxdev_buffer_init(&dmxdevfilter->buffer);
 714         dmxdevfilter->type=DMXDEV_TYPE_NONE;
 715         dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_ALLOCATED);
 716         dmxdevfilter->feed.ts=NULL;
 717         init_timer(&dmxdevfilter->timer);
 718
 719         up(&dmxdev->mutex);
 720         return 0;
 721 }
 722
 723
 724 static int dvb_dmxdev_filter_free(struct dmxdev *dmxdev, struct dmxdev_filter *dmxdevfilter)
 725 {
 726         if (down_interruptible(&dmxdev->mutex))
 727                 return -ERESTARTSYS;
 728
 729         if (down_interruptible(&dmxdevfilter->mutex)) {
 730                 up(&dmxdev->mutex);
 731                 return -ERESTARTSYS;
 732         }
 733
 734         dvb_dmxdev_filter_stop(dmxdevfilter);
 735         dvb_dmxdev_filter_reset(dmxdevfilter);
 736
 737         if (dmxdevfilter->buffer.data) {
 738                 void *mem=dmxdevfilter->buffer.data;
 739
 740                 spin_lock_irq(&dmxdev->lock);
 741                 dmxdevfilter->buffer.data=NULL;
 742                 spin_unlock_irq(&dmxdev->lock);
 743                 vfree(mem);
 744         }
 745
 746         dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_FREE);
 747         wake_up(&dmxdevfilter->buffer.queue);
 748         up(&dmxdevfilter->mutex);
 749         up(&dmxdev->mutex);
 750         return 0;
 751 }
 752
 753 static inline void invert_mode(dmx_filter_t *filter)
 754 {
 755         int i;
 756
 757         for (i=0; i<DMX_FILTER_SIZE; i++)
 758                 filter->mode[i]^=0xff;
 759 }
 760
 761
 762 static int dvb_dmxdev_filter_set(struct dmxdev *dmxdev,
 763                 struct dmxdev_filter *dmxdevfilter,
 764                 struct dmx_sct_filter_params *params)
 765 {
 766         dprintk ("function : %s\n", __FUNCTION__);
 767
 768         dvb_dmxdev_filter_stop(dmxdevfilter);
 769
 770         dmxdevfilter->type=DMXDEV_TYPE_SEC;
 771         dmxdevfilter->pid=params->pid;
 772         memcpy(&dmxdevfilter->params.sec,
 773                params, sizeof(struct dmx_sct_filter_params));
 774         invert_mode(&dmxdevfilter->params.sec.filter);
 775         dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_SET);
 776
 777         if (params->flags&DMX_IMMEDIATE_START)
 778                 return dvb_dmxdev_filter_start(dmxdevfilter);
 779
 780         return 0;
 781 }
 782
 783 static int dvb_dmxdev_pes_filter_set(struct dmxdev *dmxdev,
 784                    struct dmxdev_filter *dmxdevfilter,
 785                    struct dmx_pes_filter_params *params)
 786 {
 787         dvb_dmxdev_filter_stop(dmxdevfilter);
 788
 789         if (params->pes_type>DMX_PES_OTHER || params->pes_type<0)
 790                 return -EINVAL;
 791
 792         dmxdevfilter->type=DMXDEV_TYPE_PES;
 793         dmxdevfilter->pid=params->pid;
 794         memcpy(&dmxdevfilter->params, params, sizeof(struct dmx_pes_filter_params));
 795
 796         dvb_dmxdev_filter_state_set(dmxdevfilter, DMXDEV_STATE_SET);
 797
 798         if (params->flags&DMX_IMMEDIATE_START)
 799                 return dvb_dmxdev_filter_start(dmxdevfilter);
 800
 801         return 0;
 802 }
 803
 804 static ssize_t dvb_dmxdev_read_sec(struct dmxdev_filter *dfil,
 805                 struct file *file, char __user *buf, size_t count, loff_t *ppos)
 806 {
 807         int result, hcount;
 808         int done=0;
 809
 810         if (dfil->todo<=0) {
 811                 hcount=3+dfil->todo;
 812                 if (hcount>count)
 813                         hcount=count;
 814                 result=dvb_dmxdev_buffer_read(&dfil->buffer, file->f_flags&O_NONBLOCK,
 815                                         buf, hcount, ppos);
 816                 if (result<0) {
 817                         dfil->todo=0;
 818                         return result;
 819                 }
 820                 if (copy_from_user(dfil->secheader-dfil->todo, buf, result))
 821                         return -EFAULT;
 822                 buf+=result;
 823                 done=result;
 824                 count-=result;
 825                 dfil->todo-=result;
 826                 if (dfil->todo>-3)
 827                         return done;
 828                 dfil->todo=((dfil->secheader[1]<<8)|dfil->secheader[2])&0xfff;
 829                 if (!count)
 830                         return done;
 831         }
 832         if (count>dfil->todo)
 833                 count=dfil->todo;
 834         result=dvb_dmxdev_buffer_read(&dfil->buffer, file->f_flags&O_NONBLOCK,
 835                                 buf, count, ppos);
 836         if (result<0)
 837                 return result;
 838         dfil->todo-=result;
 839         return (result+done);
 840 }
 841
 842
 843 static ssize_t
 844 dvb_demux_read(struct file *file, char __user *buf, size_t count, loff_t *ppos)
 845 {
 846         struct dmxdev_filter *dmxdevfilter=dvb_dmxdev_file_to_filter(file);
 847         int ret=0;
 848
 849         if (down_interruptible(&dmxdevfilter->mutex))
 850                 return -ERESTARTSYS;
 851
 852         if (dmxdevfilter->type==DMXDEV_TYPE_SEC)
 853                 ret=dvb_dmxdev_read_sec(dmxdevfilter, file, buf, count, ppos);
 854         else
 855                 ret=dvb_dmxdev_buffer_read(&dmxdevfilter->buffer,
 856                                      file->f_flags&O_NONBLOCK,
 857                                      buf, count, ppos);
 858
 859         up(&dmxdevfilter->mutex);
 860         return ret;
 861 }
 862
 863
 864 static int dvb_demux_do_ioctl(struct inode *inode, struct file *file,
 865                               unsigned int cmd, void *parg)
 866 {
 867         struct dmxdev_filter *dmxdevfilter=dvb_dmxdev_file_to_filter(file);
 868         struct dmxdev *dmxdev=dmxdevfilter->dev;
 869         unsigned long arg=(unsigned long) parg;
 870         int ret=0;
 871
 872         if (down_interruptible (&dmxdev->mutex))
 873                 return -ERESTARTSYS;
 874
 875         switch (cmd) {
 876         case DMX_START:
 877                 if (down_interruptible(&dmxdevfilter->mutex)) {
 878                         up(&dmxdev->mutex);
 879                         return -ERESTARTSYS;
 880                 }
 881                 if (dmxdevfilter->state<DMXDEV_STATE_SET)
 882                         ret = -EINVAL;
 883                 else
 884                         ret = dvb_dmxdev_filter_start(dmxdevfilter);
 885                 up(&dmxdevfilter->mutex);
 886                 break;
 887
 888         case DMX_STOP:
 889                 if (down_interruptible(&dmxdevfilter->mutex)) {
 890                         up(&dmxdev->mutex);
 891                         return -ERESTARTSYS;
 892                 }
 893                 ret=dvb_dmxdev_filter_stop(dmxdevfilter);
 894                 up(&dmxdevfilter->mutex);
 895                 break;
 896
 897         case DMX_SET_FILTER:
 898                 if (down_interruptible(&dmxdevfilter->mutex)) {
 899                         up(&dmxdev->mutex);
 900                         return -ERESTARTSYS;
 901                 }
 902                 ret = dvb_dmxdev_filter_set(dmxdev, dmxdevfilter,
 903                                     (struct dmx_sct_filter_params *)parg);
 904                 up(&dmxdevfilter->mutex);
 905                 break;
 906
 907         case DMX_SET_PES_FILTER:
 908                 if (down_interruptible(&dmxdevfilter->mutex)) {
 909                         up(&dmxdev->mutex);
 910                         return -ERESTARTSYS;
 911                 }
 912                 ret=dvb_dmxdev_pes_filter_set(dmxdev, dmxdevfilter,
 913                                                (struct dmx_pes_filter_params *)parg);
 914                 up(&dmxdevfilter->mutex);
 915                 break;
 916
 917         case DMX_SET_BUFFER_SIZE:
 918                 if (down_interruptible(&dmxdevfilter->mutex)) {
 919                         up(&dmxdev->mutex);
 920                         return -ERESTARTSYS;
 921                 }
 922                 ret=dvb_dmxdev_set_buffer_size(dmxdevfilter, arg);
 923                 up(&dmxdevfilter->mutex);
 924                 break;
 925
 926         case DMX_GET_EVENT:
 927                 break;
 928
 929         case DMX_GET_PES_PIDS:
 930                 if (!dmxdev->demux->get_pes_pids) {
 931                         ret=-EINVAL;
 932                         break;
 933                 }
 934                 dmxdev->demux->get_pes_pids(dmxdev->demux, (u16 *)parg);
 935                 break;
 936
 937         case DMX_GET_STC:
 938                 if (!dmxdev->demux->get_stc) {
 939                         ret=-EINVAL;
 940                         break;
 941                 }
 942                 ret = dmxdev->demux->get_stc(dmxdev->demux,
 943                                 ((struct dmx_stc *)parg)->num,
 944                                 &((struct dmx_stc *)parg)->stc,
 945                                 &((struct dmx_stc *)parg)->base);
 946                 break;
 947
 948         default:
 949                 ret=-EINVAL;
 950         }
 951         up(&dmxdev->mutex);
 952         return ret;
 953 }
 954
 955 static int dvb_demux_ioctl(struct inode *inode, struct file *file,
 956                            unsigned int cmd, unsigned long arg)
 957 {
 958         return dvb_usercopy(inode, file, cmd, arg, dvb_demux_do_ioctl);
 959 }
 960
 961
 962 static unsigned int dvb_demux_poll (struct file *file, poll_table *wait)
 963 {
 964         struct dmxdev_filter *dmxdevfilter = dvb_dmxdev_file_to_filter(file);
 965         unsigned int mask = 0;
 966
 967         if (!dmxdevfilter)
 968                 return -EINVAL;
 969
 970         poll_wait(file, &dmxdevfilter->buffer.queue, wait);
 971
 972         if (dmxdevfilter->state != DMXDEV_STATE_GO &&
 973             dmxdevfilter->state != DMXDEV_STATE_DONE &&
 974             dmxdevfilter->state != DMXDEV_STATE_TIMEDOUT)
 975                 return 0;
 976
 977         if (dmxdevfilter->buffer.error)
 978                 mask |= (POLLIN | POLLRDNORM | POLLPRI | POLLERR);
 979
 980         if (dmxdevfilter->buffer.pread != dmxdevfilter->buffer.pwrite)
 981                 mask |= (POLLIN | POLLRDNORM | POLLPRI);
 982
 983         return mask;
 984 }
 985
 986
 987 static int dvb_demux_release(struct inode *inode, struct file *file)
 988 {
 989         struct dmxdev_filter *dmxdevfilter = dvb_dmxdev_file_to_filter(file);
 990         struct dmxdev *dmxdev = dmxdevfilter->dev;
 991
 992         return dvb_dmxdev_filter_free(dmxdev, dmxdevfilter);
 993 }
 994
 995 static struct file_operations dvb_demux_fops = {
 996         .owner          = THIS_MODULE,
 997         .read           = dvb_demux_read,
 998         .ioctl          = dvb_demux_ioctl,
 999         .open           = dvb_demux_open,
1000         .release        = dvb_demux_release,
1001         .poll           = dvb_demux_poll,
1002 };
1003
1004 static struct dvb_device dvbdev_demux = {
1005         .priv           = NULL,
1006         .users          = 1,
1007         .writers        = 1,
1008         .fops           = &dvb_demux_fops
1009 };
1010
1011 static int dvb_dvr_do_ioctl(struct inode *inode, struct file *file,
1012                             unsigned int cmd, void *parg)
1013 {
1014         struct dvb_device *dvbdev=(struct dvb_device *) file->private_data;
1015         struct dmxdev *dmxdev=(struct dmxdev *) dvbdev->priv;
1016
1017         int ret=0;
1018
1019         if (down_interruptible (&dmxdev->mutex))
1020                 return -ERESTARTSYS;
1021
1022         switch (cmd) {
1023         case DMX_SET_BUFFER_SIZE:
1024                 // FIXME: implement
1025                 ret=0;
1026                 break;
1027
1028         default:
1029                 ret=-EINVAL;
1030         }
1031         up(&dmxdev->mutex);
1032         return ret;
1033 }
1034
1035 static int dvb_dvr_ioctl(struct inode *inode, struct file *file,
1036                          unsigned int cmd, unsigned long arg)
1037 {
1038         return dvb_usercopy(inode, file, cmd, arg, dvb_dvr_do_ioctl);
1039 }
1040
1041
1042 static unsigned int dvb_dvr_poll (struct file *file, poll_table *wait)
1043 {
1044         struct dvb_device *dvbdev = (struct dvb_device *) file->private_data;
1045         struct dmxdev *dmxdev = (struct dmxdev *) dvbdev->priv;
1046         unsigned int mask = 0;
1047
1048         dprintk ("function : %s\n", __FUNCTION__);
1049
1050         poll_wait(file, &dmxdev->dvr_buffer.queue, wait);
1051
1052         if ((file->f_flags&O_ACCMODE) == O_RDONLY) {
1053                 if (dmxdev->dvr_buffer.error)
1054                         mask |= (POLLIN | POLLRDNORM | POLLPRI | POLLERR);
1055
1056                 if (dmxdev->dvr_buffer.pread!=dmxdev->dvr_buffer.pwrite)
1057                         mask |= (POLLIN | POLLRDNORM | POLLPRI);
1058         } else
1059                 mask |= (POLLOUT | POLLWRNORM | POLLPRI);
1060
1061         return mask;
1062 }
1063
1064
1065 static struct file_operations dvb_dvr_fops = {
1066         .owner          = THIS_MODULE,
1067         .read           = dvb_dvr_read,
1068         .write          = dvb_dvr_write,
1069         .ioctl          = dvb_dvr_ioctl,
1070         .open           = dvb_dvr_open,
1071         .release        = dvb_dvr_release,
1072         .poll           = dvb_dvr_poll,
1073 };
1074
1075 static struct dvb_device dvbdev_dvr = {
1076         .priv           = NULL,
1077         .users          = 1,
1078         .writers        = 1,
1079         .fops           = &dvb_dvr_fops
1080 };
1081
1082 int
1083 dvb_dmxdev_init(struct dmxdev *dmxdev, struct dvb_adapter *dvb_adapter)
1084 {
1085         int i;
1086
1087         if (dmxdev->demux->open(dmxdev->demux)<0)
1088                 return -EUSERS;
1089
1090         dmxdev->filter = vmalloc(dmxdev->filternum*sizeof(struct dmxdev_filter));
1091         if (!dmxdev->filter)
1092                 return -ENOMEM;
1093
1094         dmxdev->dvr = vmalloc(dmxdev->filternum*sizeof(struct dmxdev_dvr));
1095         if (!dmxdev->dvr) {
1096                 vfree(dmxdev->filter);
1097                 dmxdev->filter = NULL;
1098                 return -ENOMEM;
1099         }
1100
1101         sema_init(&dmxdev->mutex, 1);
1102         spin_lock_init(&dmxdev->lock);
1103         for (i=0; i<dmxdev->filternum; i++) {
1104                 dmxdev->filter[i].dev=dmxdev;
1105                 dmxdev->filter[i].buffer.data=NULL;
1106                 dvb_dmxdev_filter_state_set(&dmxdev->filter[i], DMXDEV_STATE_FREE);
1107                 dmxdev->dvr[i].dev=dmxdev;
1108                 dmxdev->dvr[i].buffer.data=NULL;
1109                 dvb_dmxdev_filter_state_set(&dmxdev->filter[i], DMXDEV_STATE_FREE);
1110                 dvb_dmxdev_dvr_state_set(&dmxdev->dvr[i], DMXDEV_STATE_FREE);
1111         }
1112         dvb_register_device(dvb_adapter, &dmxdev->dvbdev, &dvbdev_demux, dmxdev, DVB_DEVICE_DEMUX);
1113         dvb_register_device(dvb_adapter, &dmxdev->dvr_dvbdev, &dvbdev_dvr, dmxdev, DVB_DEVICE_DVR);
1114
1115         dvb_dmxdev_buffer_init(&dmxdev->dvr_buffer);
1116
1117         return 0;
1118 }
1119 EXPORT_SYMBOL(dvb_dmxdev_init);
1120
1121 void
1122 dvb_dmxdev_release(struct dmxdev *dmxdev)
1123 {
1124         dvb_unregister_device(dmxdev->dvbdev);
1125         dvb_unregister_device(dmxdev->dvr_dvbdev);
1126         if (dmxdev->filter) {
1127                 vfree(dmxdev->filter);
1128                 dmxdev->filter=NULL;
1129         }
1130         if (dmxdev->dvr) {
1131                 vfree(dmxdev->dvr);
1132                 dmxdev->dvr=NULL;
1133         }
1134         dmxdev->demux->close(dmxdev->demux);
1135 }
1136 EXPORT_SYMBOL(dvb_dmxdev_release);
1137