1 // InStreamWithCRC.h 2 3 #ifndef __IN_STREAM_WITH_CRC_H 4 #define __IN_STREAM_WITH_CRC_H 5 6 #include "../../../../C/7zCrc.h" 7 8 #include "../../../Common/MyCom.h" 9 10 #include "../../IStream.h" 11 12 class CSequentialInStreamWithCRC: 13 public ISequentialInStream, 14 public CMyUnknownImp 15 { 16 public: 17 MY_UNKNOWN_IMP 18 19 STDMETHOD(Read)(void *data, UInt32 size, UInt32 *processedSize); 20 private: 21 CMyComPtr<ISequentialInStream> _stream; 22 UInt64 _size; 23 UInt32 _crc; 24 bool _wasFinished; 25 public: 26 void SetStream(ISequentialInStream *stream) { _stream = stream; } 27 void Init() 28 { 29 _size = 0; 30 _wasFinished = false; 31 _crc = CRC_INIT_VAL; 32 } 33 void ReleaseStream() { _stream.Release(); } 34 UInt32 GetCRC() const { return CRC_GET_DIGEST(_crc); } 35 UInt64 GetSize() const { return _size; } 36 bool WasFinished() const { return _wasFinished; } 37 }; 38 39 class CInStreamWithCRC: 40 public IInStream, 41 public CMyUnknownImp 42 { 43 public: 44 MY_UNKNOWN_IMP1(IInStream) 45 46 STDMETHOD(Read)(void *data, UInt32 size, UInt32 *processedSize); 47 STDMETHOD(Seek)(Int64 offset, UInt32 seekOrigin, UInt64 *newPosition); 48 private: 49 CMyComPtr<IInStream> _stream; 50 UInt64 _size; 51 UInt32 _crc; 52 // bool _wasFinished; 53 public: 54 void SetStream(IInStream *stream) { _stream = stream; } 55 void Init() 56 { 57 _size = 0; 58 // _wasFinished = false; 59 _crc = CRC_INIT_VAL; 60 } 61 void ReleaseStream() { _stream.Release(); } 62 UInt32 GetCRC() const { return CRC_GET_DIGEST(_crc); } 63 UInt64 GetSize() const { return _size; } 64 // bool WasFinished() const { return _wasFinished; } 65 }; 66 67 #endif 68
