Fixed TS packet processing (buffer overflows)

Added packet injection

1 files changed, 10 insertions, 4 deletions

diff --git a/markad-standalone.h b/markad-standalone.h
index b1dc0ce..145a9ea 100644
--- a/markad-standalone.h
+++ b/markad-standalone.h
@@ -131,15 +131,16 @@ unsigned reserved2:
         4;
 unsigned ES_info_length_L:
         8;
-};
+    };
 #pragma pack()
 
-struct ES_DESCRIPTOR {
+    struct ES_DESCRIPTOR
+    {
 unsigned Descriptor_Tag:
         8;
 unsigned Descriptor_Length:
         8;
-};
+    };
 
 
     cMarkAdDemux *video_demux;
@@ -156,13 +157,18 @@ unsigned Descriptor_Length:
     bool isTS;
     int MaxFiles;
     int framecnt;
+    bool abort;
 
     void AddMark(MarkAdMark *Mark);
-    void CheckPATPMT(const char *Directory);
+    bool CheckPATPMT(const char *Directory);
     bool CheckTS(const char *Directory);
     bool ProcessFile(const char *Directory, int Number);
 
 public:
+    void SetAbort()
+    {
+        abort=true;
+    }
     void Process(const char *Directory);
     cMarkAdStandalone(const char *Directory);
     ~cMarkAdStandalone();