descrambler: add other odd/even key validity check based on the key u…

…pdate time
tvheadend · Jul 28, 2014 · 4d8aa37 · 4d8aa37
1 parent 3400ee8
commit 4d8aa37
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/descrambler.h b/src/descrambler.h
@@ -63,6 +63,7 @@ typedef struct th_descrambler_runtime {
   uint8_t  dr_key_valid;
   uint8_t  dr_ecm_valid;
   time_t   dr_key_start;
+  time_t   dr_key_timestamp[2];
   time_t   dr_ecm_start;
   time_t   dr_ecm_key_time;
   sbuf_t   dr_buf;

diff --git a/src/descrambler/descrambler.c b/src/descrambler/descrambler.c
@@ -186,13 +186,15 @@ descrambler_keys ( th_descrambler_t *td,
       j++;
       tvhcsa_set_key_even(td->td_csa, even);
       dr->dr_key_valid |= 0x40;
+      dr->dr_key_timestamp[0] = dispatch_clock;
       break;
     }
   for (i = 0; i < 8; i++)
     if (odd[i]) {
       j++;
       tvhcsa_set_key_odd(td->td_csa, odd);
       dr->dr_key_valid |= 0x80;
+      dr->dr_key_timestamp[1] = dispatch_clock;
       break;
     }
 
@@ -260,10 +262,11 @@ descrambler_descramble ( service_t *t,
                          const uint8_t *tsb )
 {
 #define KEY_MASK(k) (((k) & 0x40) + 0x40) /* 0x40 (for even) or 0x80 (for odd) */
+#define KEY_IDX(k)  (((k) & 0x40) >> 6)
   th_descrambler_t *td;
   th_descrambler_runtime_t *dr = t->s_descramble;
   int count, failed, off, size, flush_data = 0;
-  uint8_t *tsb2, ki;
+  uint8_t *tsb2, ki, kidx;
 
   lock_assert(&t->s_stream_mutex);
 
@@ -292,7 +295,9 @@ descrambler_descramble ( service_t *t,
             tvhtrace("descrambler", "stream key changed to %s for service \"%s\"",
                                     (ki & 0x40) ? "odd" : "even",
                                     ((mpegts_service_t *)t)->s_dvb_svcname);
-            if (dr->dr_ecm_key_time +
+            kidx = KEY_IDX(ki);
+            if (dr->dr_key_timestamp[kidx] < dr->dr_key_timestamp[kidx^1] ||
+                dr->dr_ecm_key_time +
                   ((dr->dr_ecm_valid & KEY_MASK(ki)) ? 0 : 2) < dr->dr_key_start) {
               sbuf_cut(&dr->dr_buf, off);
               if (!td->td_ecm_reset(td)) {
@@ -325,7 +330,9 @@ descrambler_descramble ( service_t *t,
         tvhtrace("descrambler", "stream key changed to %s for service \"%s\"",
                                 (ki & 0x40) ? "odd" : "even",
                                 ((mpegts_service_t *)t)->s_dvb_svcname);
-        if (dr->dr_ecm_key_time +
+        kidx = KEY_IDX(ki);
+        if (dr->dr_key_timestamp[kidx] < dr->dr_key_timestamp[kidx^1] ||
+            dr->dr_ecm_key_time +
               ((dr->dr_ecm_valid & KEY_MASK(ki)) ? 0 : 2) < dr->dr_key_start) {
           tvhtrace("descrambler", "ECM late (%ld seconds) for service \"%s\"",
                                   dispatch_clock - dr->dr_ecm_key_time,