equal
deleted
inserted
replaced
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 void ReadBytes(void *data, UInt32 size, UInt32 &processedSize) |
|
60 { |
|
61 for(processedSize = 0; processedSize < size; processedSize++) |
|
62 if (!ReadByte(((Byte *)data)[processedSize])) |
|
63 return; |
|
64 } |
|
65 bool ReadBytes(void *data, UInt32 size) |
|
66 { |
|
67 UInt32 processedSize; |
|
68 ReadBytes(data, size, processedSize); |
|
69 return (processedSize == size); |
|
70 } |
|
71 UInt64 GetProcessedSize() const { return _processedSize + (_buffer - _bufferBase); } |
|
72 bool WasFinished() const { return _wasFinished; } |
|
73 }; |
|
74 |
|
75 #endif |
|