The Tor Browser / file comparison
comparison: other-licenses/7zstub/src/7zip/Common/InBuffer.h
other-licenses/7zstub/src/7zip/Common/InBuffer.h
- branch
- TOR_BUG_9701
- changeset 8
- 97036ab72558
equal
deleted
inserted
replaced
| -1:000000000000 | 0:d519c1e4a70e |
|---|---|
| 1 // InBuffer.h | |
| 2 | |
| 3 #ifndef __INBUFFER_H | |
| 4 #define __INBUFFER_H | |
| 5 | |
| 6 #include "../IStream.h" | |
| 7 #include "../../Common/MyCom.h" | |
| 8 | |
| 9 #ifndef _NO_EXCEPTIONS | |
| 10 class CInBufferException | |
| 11 { | |
| 12 public: | |
| 13 HRESULT ErrorCode; | |
| 14 CInBufferException(HRESULT errorCode): ErrorCode(errorCode) {} | |
| 15 }; | |
| 16 #endif | |
| 17 | |
| 18 class CInBuffer | |
| 19 { | |
| 20 Byte *_buffer; | |
| 21 Byte *_bufferLimit; | |
| 22 Byte *_bufferBase; | |
| 23 CMyComPtr<ISequentialInStream> _stream; | |
| 24 UInt64 _processedSize; | |
| 25 UInt32 _bufferSize; | |
| 26 bool _wasFinished; | |
| 27 | |
| 28 bool ReadBlock(); | |
| 29 Byte ReadBlock2(); | |
| 30 | |
| 31 public: | |
| 32 #ifdef _NO_EXCEPTIONS | |
| 33 HRESULT ErrorCode; | |
| 34 #endif | |
| 35 | |
| 36 CInBuffer(); | |
| 37 ~CInBuffer() { Free(); } | |
| 38 | |
| 39 bool Create(UInt32 bufferSize); | |
| 40 void Free(); | |
| 41 | |
| 42 void SetStream(ISequentialInStream *stream); | |
| 43 void Init(); | |
| 44 void ReleaseStream() { _stream.Release(); } | |
| 45 | |
| 46 bool ReadByte(Byte &b) | |
| 47 { | |
| 48 if(_buffer >= _bufferLimit) | |
| 49 if(!ReadBlock()) | |
| 50 return false; | |
| 51 b = *_buffer++; | |
| 52 return true; | |
| 53 } | |
| 54 Byte ReadByte() | |
| 55 { | |
| 56 if(_buffer >= _bufferLimit) | |
| 57 return ReadBlock2(); | |
| 58 return *_buffer++; | |
| 59 } | |
| 60 void ReadBytes(void *data, UInt32 size, UInt32 &processedSize) | |
| 61 { | |
| 62 for(processedSize = 0; processedSize < size; processedSize++) | |
| 63 if (!ReadByte(((Byte *)data)[processedSize])) | |
| 64 return; | |
| 65 } | |
| 66 bool ReadBytes(void *data, UInt32 size) | |
| 67 { | |
| 68 UInt32 processedSize; | |
| 69 ReadBytes(data, size, processedSize); | |
| 70 return (processedSize == size); | |
| 71 } | |
| 72 UInt64 GetProcessedSize() const { return _processedSize + (_buffer - _bufferBase); } | |
| 73 bool WasFinished() const { return _wasFinished; } | |
| 74 }; | |
| 75 | |
| 76 #endif |
