descrambler: do not allow to open new pids when flushed

tvheadend · Sep 17, 2014 · a7f80d2 · a7f80d2
1 parent a766245
commit a7f80d2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/descrambler/descrambler.c b/src/descrambler/descrambler.c
@@ -134,6 +134,7 @@ descrambler_service_start ( service_t *t )
     dr->dr_key_index = 0xff;
     dr->dr_last_descramble = dispatch_clock;
   }
+  ((mpegts_service_t *)t)->s_dvb_mux->mm_descrambler_flush = 0;
 }
 
 void
@@ -507,6 +508,8 @@ descrambler_open_pid_( mpegts_mux_t *mux, void *opaque, int pid,
 
   if (mux == NULL)
     return 0;
+  if (mux->mm_descrambler_flush)
+    return 0;
   flags  = pid >> 16;
   pid   &= 0x1fff;
   TAILQ_FOREACH(dt, &mux->mm_descrambler_tables, link) {
@@ -605,6 +608,7 @@ descrambler_flush_tables( mpegts_mux_t *mux )
   cwc_caid_update(mux, 0, 0, -1);
 #endif
   pthread_mutex_lock(&mux->mm_descrambler_lock);
+  mux->mm_descrambler_flush = 1;
   while ((dt = TAILQ_FIRST(&mux->mm_descrambler_tables)) != NULL) {
     while ((ds = TAILQ_FIRST(&dt->sections)) != NULL) {
       TAILQ_REMOVE(&dt->sections, ds, link);
@@ -703,8 +707,11 @@ descrambler_open_emm( mpegts_mux_t *mux, void *opaque, int caid,
   if (mux == NULL)
     return 0;
   pthread_mutex_lock(&mux->mm_descrambler_lock);
+  if (mux->mm_descrambler_flush)
+    goto unlock;
   TAILQ_FOREACH(emm, &mux->mm_descrambler_emms, link) {
     if (emm->caid == caid && emm->opaque == opaque) {
+unlock:
       pthread_mutex_unlock(&mux->mm_descrambler_lock);
       return 0;
     }

diff --git a/src/input/mpegts.h b/src/input/mpegts.h
@@ -388,6 +388,7 @@ struct mpegts_mux
   TAILQ_HEAD(, descrambler_table) mm_descrambler_tables;
   TAILQ_HEAD(, descrambler_emm) mm_descrambler_emms;
   pthread_mutex_t             mm_descrambler_lock;
+  int                         mm_descrambler_flush;
 
   /*
    * Functions