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