Add new parameter to enable/disable all filetype recovery: photorec /cmd device ...,fileopts,everything,disable"

MPG recovery: check size defined by video sequence start code

1 files changed, 3 insertions, 1 deletions

diff --git a/src/file_mpg.c b/src/file_mpg.c
index 9cf6d913..c16aabf6 100644
--- a/src/file_mpg.c
+++ b/src/file_mpg.c
@@ -72,7 +72,9 @@ static int header_check_mpg(const unsigned char *buffer, const unsigned int buff
    * 0x000001B5 visual object start code				*/
 
   if(buffer[0]==0x00 && buffer[1]==0x00 && buffer[2]==0x01 &&
-      (buffer[3]==0xB0 || buffer[3]==0xB3 || buffer[3]==0xB5 || buffer[3]==0xBA || buffer[3]==0xBB))
+      (buffer[3]==0xB0 ||
+       (buffer[3]==0xB3 && ((buffer[4]<<8)+(buffer[5]>>8)>0) && ((buffer[5]<<8)+buffer[6]>0)) ||
+       buffer[3]==0xB5 || buffer[3]==0xBA || buffer[3]==0xBB))
   {
     reset_file_recovery(file_recovery_new);
     file_recovery_new->extension=file_hint_mpg.extension;