Obtained from: https://anonscm.debian.org/cgit/users/robert/p7zip-rar.git/tree/debian/patches/06-CVE-2018-5996.patch diff -pruN '--exclude=*.orig' p7zip_16.02~/CPP/7zip/Compress/Rar1Decoder.cpp p7zip_16.02/CPP/7zip/Compress/Rar1Decoder.cpp --- p7zip_16.02~/CPP/7zip/Compress/Rar1Decoder.cpp 2015-09-01 18:04:52.000000000 +0000 +++ p7zip_16.02/CPP/7zip/Compress/Rar1Decoder.cpp 2018-05-02 15:06:58.034027678 +0000 @@ -29,7 +29,7 @@ public: }; */ -CDecoder::CDecoder(): m_IsSolid(false) { } +CDecoder::CDecoder(): m_IsSolid(false), _errorMode(false) { } void CDecoder::InitStructures() { @@ -406,9 +406,14 @@ HRESULT CDecoder::CodeReal(ISequentialIn InitData(); if (!m_IsSolid) { + _errorMode = false; InitStructures(); InitHuff(); } + + if (_errorMode) + return S_FALSE; + if (m_UnpackSize > 0) { GetFlagsBuf(); @@ -477,9 +482,9 @@ STDMETHODIMP CDecoder::Code(ISequentialI const UInt64 *inSize, const UInt64 *outSize, ICompressProgressInfo *progress) { try { return CodeReal(inStream, outStream, inSize, outSize, progress); } - catch(const CInBufferException &e) { return e.ErrorCode; } - catch(const CLzOutWindowException &e) { return e.ErrorCode; } - catch(...) { return S_FALSE; } + catch(const CInBufferException &e) { _errorMode = true; return e.ErrorCode; } + catch(const CLzOutWindowException &e) { _errorMode = true; return e.ErrorCode; } + catch(...) { _errorMode = true; return S_FALSE; } } STDMETHODIMP CDecoder::SetDecoderProperties2(const Byte *data, UInt32 size) diff -pruN '--exclude=*.orig' p7zip_16.02~/CPP/7zip/Compress/Rar1Decoder.h p7zip_16.02/CPP/7zip/Compress/Rar1Decoder.h --- p7zip_16.02~/CPP/7zip/Compress/Rar1Decoder.h 2014-12-21 12:44:00.000000000 +0000 +++ p7zip_16.02/CPP/7zip/Compress/Rar1Decoder.h 2018-05-02 15:06:58.034273736 +0000 @@ -39,6 +39,7 @@ public: Int64 m_UnpackSize; bool m_IsSolid; + bool _errorMode; UInt32 ReadBits(int numBits); HRESULT CopyBlock(UInt32 distance, UInt32 len); diff -pruN '--exclude=*.orig' p7zip_16.02~/CPP/7zip/Compress/Rar2Decoder.cpp p7zip_16.02/CPP/7zip/Compress/Rar2Decoder.cpp --- p7zip_16.02~/CPP/7zip/Compress/Rar2Decoder.cpp 2015-10-03 08:49:14.000000000 +0000 +++ p7zip_16.02/CPP/7zip/Compress/Rar2Decoder.cpp 2018-05-02 15:06:58.034541629 +0000 @@ -80,7 +80,8 @@ static const UInt32 kHistorySize = 1 << static const UInt32 kWindowReservSize = (1 << 22) + 256; CDecoder::CDecoder(): - m_IsSolid(false) + m_IsSolid(false), + m_TablesOK(false) { } @@ -100,6 +101,8 @@ UInt32 CDecoder::ReadBits(unsigned numBi bool CDecoder::ReadTables(void) { + m_TablesOK = false; + Byte levelLevels[kLevelTableSize]; Byte newLevels[kMaxTableSize]; m_AudioMode = (ReadBits(1) == 1); @@ -170,6 +173,8 @@ bool CDecoder::ReadTables(void) } memcpy(m_LastLevels, newLevels, kMaxTableSize); + m_TablesOK = true; + return true; } @@ -344,6 +349,9 @@ HRESULT CDecoder::CodeReal(ISequentialIn return S_FALSE; } + if (!m_TablesOK) + return S_FALSE; + UInt64 startPos = m_OutWindowStream.GetProcessedSize(); while (pos < unPackSize) { diff -pruN '--exclude=*.orig' p7zip_16.02~/CPP/7zip/Compress/Rar2Decoder.h p7zip_16.02/CPP/7zip/Compress/Rar2Decoder.h --- p7zip_16.02~/CPP/7zip/Compress/Rar2Decoder.h 2015-06-19 10:52:06.000000000 +0000 +++ p7zip_16.02/CPP/7zip/Compress/Rar2Decoder.h 2018-05-02 15:06:58.034736908 +0000 @@ -139,6 +139,7 @@ class CDecoder : UInt64 m_PackSize; bool m_IsSolid; + bool m_TablesOK; void InitStructures(); UInt32 ReadBits(unsigned numBits); diff -pruN '--exclude=*.orig' p7zip_16.02~/CPP/7zip/Compress/Rar3Decoder.cpp p7zip_16.02/CPP/7zip/Compress/Rar3Decoder.cpp --- p7zip_16.02~/CPP/7zip/Compress/Rar3Decoder.cpp 2016-05-20 08:20:03.000000000 +0000 +++ p7zip_16.02/CPP/7zip/Compress/Rar3Decoder.cpp 2018-05-02 15:06:58.035151069 +0000 @@ -92,7 +92,8 @@ CDecoder::CDecoder(): _writtenFileSize(0), _vmData(0), _vmCode(0), - m_IsSolid(false) + m_IsSolid(false), + _errorMode(false) { Ppmd7_Construct(&_ppmd); } @@ -545,6 +546,9 @@ HRESULT CDecoder::ReadTables(bool &keepD return InitPPM(); } + TablesRead = false; + TablesOK = false; + _lzMode = true; PrevAlignBits = 0; PrevAlignCount = 0; @@ -606,6 +610,9 @@ HRESULT CDecoder::ReadTables(bool &keepD } } } + if (InputEofError()) + return S_FALSE; + TablesRead = true; // original code has check here: @@ -623,6 +630,9 @@ HRESULT CDecoder::ReadTables(bool &keepD RIF(m_LenDecoder.Build(&newLevels[kMainTableSize + kDistTableSize + kAlignTableSize])); memcpy(m_LastLevels, newLevels, kTablesSizesSum); + + TablesOK = true; + return S_OK; } @@ -824,7 +834,12 @@ HRESULT CDecoder::CodeReal(ICompressProg PpmEscChar = 2; PpmError = true; InitFilters(); + _errorMode = false; } + + if (_errorMode) + return S_FALSE; + if (!m_IsSolid || !TablesRead) { bool keepDecompressing; @@ -838,6 +853,8 @@ HRESULT CDecoder::CodeReal(ICompressProg bool keepDecompressing; if (_lzMode) { + if (!TablesOK) + return S_FALSE; RINOK(DecodeLZ(keepDecompressing)) } else @@ -901,8 +918,8 @@ STDMETHODIMP CDecoder::Code(ISequentialI _unpackSize = outSize ? *outSize : (UInt64)(Int64)-1; return CodeReal(progress); } - catch(const CInBufferException &e) { return e.ErrorCode; } - catch(...) { return S_FALSE; } + catch(const CInBufferException &e) { _errorMode = true; return e.ErrorCode; } + catch(...) { _errorMode = true; return S_FALSE; } // CNewException is possible here. But probably CNewException is caused // by error in data stream. } diff -pruN '--exclude=*.orig' p7zip_16.02~/CPP/7zip/Compress/Rar3Decoder.h p7zip_16.02/CPP/7zip/Compress/Rar3Decoder.h --- p7zip_16.02~/CPP/7zip/Compress/Rar3Decoder.h 2015-10-03 08:49:12.000000000 +0000 +++ p7zip_16.02/CPP/7zip/Compress/Rar3Decoder.h 2018-05-02 15:06:58.035371032 +0000 @@ -192,6 +192,7 @@ class CDecoder: UInt32 _lastFilter; bool m_IsSolid; + bool _errorMode; bool _lzMode; bool _unsupportedFilter; @@ -200,6 +201,7 @@ class CDecoder: UInt32 PrevAlignCount; bool TablesRead; + bool TablesOK; CPpmd7 _ppmd; int PpmEscChar;