1 // 7zFolderOutStream.h 2 3 #ifndef __7Z_FOLDER_OUT_STREAM_H 4 #define __7Z_FOLDER_OUT_STREAM_H 5 6 #include "../../IStream.h" 7 #include "../IArchive.h" 8 #include "../Common/OutStreamWithCRC.h" 9 10 #include "7zIn.h" 11 12 namespace NArchive { 13 namespace N7z { 14 15 class CFolderOutStream: 16 public ISequentialOutStream, 17 public ICompressGetSubStreamSize, 18 public CMyUnknownImp 19 { 20 COutStreamWithCRC *_crcStreamSpec; 21 CMyComPtr<ISequentialOutStream> _crcStream; 22 const CDbEx *_db; 23 const CBoolVector *_extractStatuses; 24 CMyComPtr<IArchiveExtractCallback> _extractCallback; 25 UInt32 _ref2Offset; 26 UInt32 _startIndex; 27 unsigned _currentIndex; 28 bool _testMode; 29 bool _checkCrc; 30 bool _fileIsOpen; 31 UInt64 _rem; 32 33 HRESULT OpenFile(); 34 HRESULT CloseFileAndSetResult(Int32 res); 35 HRESULT CloseFileAndSetResult(); 36 HRESULT ProcessEmptyFiles(); 37 public: 38 MY_UNKNOWN_IMP1(ICompressGetSubStreamSize) 39 40 CFolderOutStream(); 41 42 STDMETHOD(Write)(const void *data, UInt32 size, UInt32 *processedSize); 43 STDMETHOD(GetSubStreamSize)(UInt64 subStream, UInt64 *value); 44 45 HRESULT Init( 46 const CDbEx *db, 47 UInt32 ref2Offset, UInt32 startIndex, 48 const CBoolVector *extractStatuses, 49 IArchiveExtractCallback *extractCallback, 50 bool testMode, bool checkCrc); 51 HRESULT FlushCorrupted(Int32 resultEOperationResult); 52 HRESULT WasWritingFinished() const 53 { return (_currentIndex == _extractStatuses->Size()) ? S_OK: E_FAIL; } 54 }; 55 56 }} 57 58 #endif 59