| OLD | NEW |
|---|
| (Empty) |
| 1 // InBuffer.h | |
| 2 | |
| 3 #ifndef __INBUFFER_H | |
| 4 #define __INBUFFER_H | |
| 5 | |
| 6 #include "../IStream.h" | |
| 7 #include "../../Common/MyCom.h" | |
| 8 #include "../../Common/MyException.h" | |
| 9 | |
| 10 #ifndef _NO_EXCEPTIONS | |
| 11 struct CInBufferException: public CSystemException | |
| 12 { | |
| 13 CInBufferException(HRESULT errorCode): CSystemException(errorCode) {} | |
| 14 }; | |
| 15 #endif | |
| 16 | |
| 17 class CInBuffer | |
| 18 { | |
| 19 Byte *_buffer; | |
| 20 Byte *_bufferLimit; | |
| 21 Byte *_bufferBase; | |
| 22 CMyComPtr<ISequentialInStream> _stream; | |
| 23 UInt64 _processedSize; | |
| 24 UInt32 _bufferSize; | |
| 25 bool _wasFinished; | |
| 26 | |
| 27 bool ReadBlock(); | |
| 28 Byte ReadBlock2(); | |
| 29 | |
| 30 public: | |
| 31 #ifdef _NO_EXCEPTIONS | |
| 32 HRESULT ErrorCode; | |
| 33 #endif | |
| 34 | |
| 35 CInBuffer(); | |
| 36 ~CInBuffer() { Free(); } | |
| 37 | |
| 38 bool Create(UInt32 bufferSize); | |
| 39 void Free(); | |
| 40 | |
| 41 void SetStream(ISequentialInStream *stream); | |
| 42 void Init(); | |
| 43 void ReleaseStream() { _stream.Release(); } | |
| 44 | |
| 45 bool ReadByte(Byte &b) | |
| 46 { | |
| 47 if (_buffer >= _bufferLimit) | |
| 48 if (!ReadBlock()) | |
| 49 return false; | |
| 50 b = *_buffer++; | |
| 51 return true; | |
| 52 } | |
| 53 Byte ReadByte() | |
| 54 { | |
| 55 if (_buffer >= _bufferLimit) | |
| 56 return ReadBlock2(); | |
| 57 return *_buffer++; | |
| 58 } | |
| 59 UInt32 ReadBytes(Byte *buf, UInt32 size) | |
| 60 { | |
| 61 if ((UInt32)(_bufferLimit - _buffer) >= size) | |
| 62 { | |
| 63 for (UInt32 i = 0; i < size; i++) | |
| 64 buf[i] = _buffer[i]; | |
| 65 _buffer += size; | |
| 66 return size; | |
| 67 } | |
| 68 for (UInt32 i = 0; i < size; i++) | |
| 69 { | |
| 70 if (_buffer >= _bufferLimit) | |
| 71 if (!ReadBlock()) | |
| 72 return i; | |
| 73 buf[i] = *_buffer++; | |
| 74 } | |
| 75 return size; | |
| 76 } | |
| 77 UInt64 GetProcessedSize() const { return _processedSize + (_buffer - _bufferBa
se); } | |
| 78 bool WasFinished() const { return _wasFinished; } | |
| 79 }; | |
| 80 | |
| 81 #endif | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 624713003:
Keep only base/extractor.[cc|h]. (Closed)
Base URL: https://chromium.googlesource.com/external/omaha.git@master