From d25fba8a1407a6f32d68fdbcd1401f893a959d3f Mon Sep 17 00:00:00 2001
From: Adam Sutton <dev@adamsutton.me.uk>
Date: Wed, 27 Nov 2013 17:50:24 +0000
Subject: [PATCH] mpegts: stop NIT processing from hanging

Was stupidly returning error (instead of marking complete) when unwanted
NID is detected. This causes real problems on DVB-C, and possibly DVB-S.

Might need to revisit some SDT stuff, which I think was blocking, and
now isn't waiting, for NIT.
---
 src/input/mpegts.h         |  1 +
 src/input/mpegts/dvb_psi.c | 14 +++++++++-----
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/input/mpegts.h b/src/input/mpegts.h
index 6ac7de1d..ef01a3e2 100644
--- a/src/input/mpegts.h
+++ b/src/input/mpegts.h
@@ -142,6 +142,7 @@ struct mpegts_table
   RB_HEAD(,mpegts_table_state) mt_state;
   int mt_complete;
   int mt_incomplete;
+  int mt_finished;
 
   int mt_count;
 
diff --git a/src/input/mpegts/dvb_psi.c b/src/input/mpegts/dvb_psi.c
index 92085a31..88450adc 100644
--- a/src/input/mpegts/dvb_psi.c
+++ b/src/input/mpegts/dvb_psi.c
@@ -421,6 +421,7 @@ mpegts_table_state_reset
     mt->mt_complete--;
     mt->mt_incomplete++;
   }
+  mt->mt_finished = 0;
   st->complete = 0;
   memset(st->sections, 0, sizeof(st->sections));
   for (i = 0; i < last / 32; i++)
@@ -465,7 +466,9 @@ dvb_table_complete
              mt->mt_incomplete, mt->mt_complete, total);
     return -1;
   }
-  tvhdebug(mt->mt_name, "subtable completed");
+  if (!mt->mt_finished)
+    tvhdebug(mt->mt_name, "completed pid %d table %08X / %08x", mt->mt_pid, mt->mt_table, mt->mt_mask);
+  mt->mt_finished = 1;
   return 0;
 }
 
@@ -745,17 +748,18 @@ dvb_nit_callback
   r = dvb_table_begin(mt, ptr, len, tableid, nbid, 7, &st, &sect, &last, &ver);
   if (r != 1) return r;
 
-  /* BAT */
+  /* NIT */
   if (tableid != 0x4A) {
 
     /* Specific NID */
     if (mn->mn_nid) {
-      if (mn->mn_nid != nbid)
-        return -1;
+      if (mn->mn_nid != nbid) {
+        return dvb_table_end(mt, st, sect);
+      }
   
     /* Only use "this" network */
     } else if (tableid != 0x40) {
-      return -1;
+      return dvb_table_end(mt, st, sect);
     }
   }