4.54 beta

2 files changed, 17 insertions, 10 deletions

diff --git a/CPP/7zip/Archive/7z/7zIn.cpp b/CPP/7zip/Archive/7z/7zIn.cpp
index e04313be..a4292545 100755
--- a/CPP/7zip/Archive/7z/7zIn.cpp
+++ b/CPP/7zip/Archive/7z/7zIn.cpp
@@ -607,7 +607,9 @@ void CInArchive::ReadStreamsInfo(
   for (;;)
   {
     UInt64 type = ReadID();
-    switch(type)
+    if (type > ((UInt32)1 << 30))
+      ThrowIncorrect();
+    switch((UInt32)type)
     {
       case NID::kEnd:
         return;
@@ -627,6 +629,8 @@ void CInArchive::ReadStreamsInfo(
             unPackSizes, digestsDefined, digests);
         break;
       }
+      default:
+        ThrowIncorrect();
     }
   }
 }
@@ -664,7 +668,7 @@ void CInArchive::ReadBoolVector2(int numItems, CBoolVector &v)
 }
 
 void CInArchive::ReadTime(const CObjectVector<CByteBuffer> &dataVector,
-    CObjectVector<CFileItem> &files, UInt64 type)
+    CObjectVector<CFileItem> &files, UInt32 type)
 {
   CBoolVector boolVector;
   ReadBoolVector2(files.Size(), boolVector);
@@ -881,8 +885,10 @@ HRESULT CInArchive::ReadHeader(
     if (type == NID::kEnd)
       break;
     UInt64 size = ReadNumber();
-    database.ArchiveInfo.FileInfoPopIDs.Add(type);
-    switch(type)
+    bool isKnownType = true;
+    if (type > ((UInt32)1 << 30))
+      isKnownType = false;
+    else switch((UInt32)type)
     {
       case NID::kName:
       {
@@ -951,15 +957,16 @@ HRESULT CInArchive::ReadHeader(
       case NID::kLastWriteTime:
       case NID::kLastAccessTime:
       {
-        ReadTime(dataVector, database.Files, type);
+        ReadTime(dataVector, database.Files, (UInt32)type);
         break;
       }
       default:
-      {
-        database.ArchiveInfo.FileInfoPopIDs.DeleteBack();
-        SkeepData(size);
-      }
+        isKnownType = false;
     }
+    if (isKnownType)
+      database.ArchiveInfo.FileInfoPopIDs.Add(type);
+    else
+      SkeepData(size);
   }
 
   CNum emptyFileIndex = 0;
diff --git a/CPP/7zip/Archive/7z/7zIn.h b/CPP/7zip/Archive/7z/7zIn.h
index 47cd3b35..aae43506 100755
--- a/CPP/7zip/Archive/7z/7zIn.h
+++ b/CPP/7zip/Archive/7z/7zIn.h
@@ -194,7 +194,7 @@ private:
   void ReadBoolVector(int numItems, CBoolVector &v);
   void ReadBoolVector2(int numItems, CBoolVector &v);
   void ReadTime(const CObjectVector<CByteBuffer> &dataVector,
-      CObjectVector<CFileItem> &files, UInt64 type);
+      CObjectVector<CFileItem> &files, UInt32 type);
   HRESULT ReadAndDecodePackedStreams(
       DECL_EXTERNAL_CODECS_LOC_VARS
       UInt64 baseOffset, UInt64 &dataOffset,