1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/other-licenses/7zstub/src/7zip/Common/StreamObjects.h Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,156 @@
1.4 +// StreamObjects.h
1.5 +
1.6 +#ifndef __STREAMOBJECTS_H
1.7 +#define __STREAMOBJECTS_H
1.8 +
1.9 +#include "../../Common/DynamicBuffer.h"
1.10 +#include "../../Common/MyCom.h"
1.11 +#include "../IStream.h"
1.12 +
1.13 +class CSequentialInStreamImp:
1.14 + public ISequentialInStream,
1.15 + public CMyUnknownImp
1.16 +{
1.17 + const Byte *_dataPointer;
1.18 + size_t _size;
1.19 + size_t _pos;
1.20 +
1.21 +public:
1.22 + void Init(const Byte *dataPointer, size_t size)
1.23 + {
1.24 + _dataPointer = dataPointer;
1.25 + _size = size;
1.26 + _pos = 0;
1.27 + }
1.28 +
1.29 + MY_UNKNOWN_IMP
1.30 +
1.31 + STDMETHOD(Read)(void *data, UInt32 size, UInt32 *processedSize);
1.32 +};
1.33 +
1.34 +
1.35 +class CWriteBuffer
1.36 +{
1.37 + CByteDynamicBuffer _buffer;
1.38 + size_t _size;
1.39 +public:
1.40 + CWriteBuffer(): _size(0) {}
1.41 + void Init() { _size = 0; }
1.42 + void Write(const void *data, size_t size);
1.43 + size_t GetSize() const { return _size; }
1.44 + const CByteDynamicBuffer& GetBuffer() const { return _buffer; }
1.45 +};
1.46 +
1.47 +class CSequentialOutStreamImp:
1.48 + public ISequentialOutStream,
1.49 + public CMyUnknownImp
1.50 +{
1.51 + CWriteBuffer _writeBuffer;
1.52 +public:
1.53 + void Init() { _writeBuffer.Init(); }
1.54 + size_t GetSize() const { return _writeBuffer.GetSize(); }
1.55 + const CByteDynamicBuffer& GetBuffer() const { return _writeBuffer.GetBuffer(); }
1.56 +
1.57 + MY_UNKNOWN_IMP
1.58 +
1.59 + STDMETHOD(Write)(const void *data, UInt32 size, UInt32 *processedSize);
1.60 +};
1.61 +
1.62 +class CSequentialOutStreamImp2:
1.63 + public ISequentialOutStream,
1.64 + public CMyUnknownImp
1.65 +{
1.66 + Byte *_buffer;
1.67 +public:
1.68 + size_t _size;
1.69 + size_t _pos;
1.70 +
1.71 + void Init(Byte *buffer, size_t size)
1.72 + {
1.73 + _buffer = buffer;
1.74 + _pos = 0;
1.75 + _size = size;
1.76 + }
1.77 +
1.78 + MY_UNKNOWN_IMP
1.79 +
1.80 + STDMETHOD(Write)(const void *data, UInt32 size, UInt32 *processedSize);
1.81 +};
1.82 +
1.83 +class CSequentialInStreamSizeCount:
1.84 + public ISequentialInStream,
1.85 + public CMyUnknownImp
1.86 +{
1.87 + CMyComPtr<ISequentialInStream> _stream;
1.88 + UInt64 _size;
1.89 +public:
1.90 + void Init(ISequentialInStream *stream)
1.91 + {
1.92 + _stream = stream;
1.93 + _size = 0;
1.94 + }
1.95 + UInt64 GetSize() const { return _size; }
1.96 +
1.97 + MY_UNKNOWN_IMP
1.98 +
1.99 + STDMETHOD(Read)(void *data, UInt32 size, UInt32 *processedSize);
1.100 +};
1.101 +
1.102 +class CSequentialInStreamRollback:
1.103 + public ISequentialInStream,
1.104 + public CMyUnknownImp
1.105 +{
1.106 + CMyComPtr<ISequentialInStream> _stream;
1.107 + Byte *_buffer;
1.108 + size_t _bufferSize;
1.109 + UInt64 _size;
1.110 +
1.111 + size_t _currentSize;
1.112 + size_t _currentPos;
1.113 +public:
1.114 + CSequentialInStreamRollback(size_t bufferSize):
1.115 + _bufferSize(bufferSize),
1.116 + _buffer(0)
1.117 + {
1.118 + _buffer = new Byte[bufferSize];
1.119 + }
1.120 + ~CSequentialInStreamRollback()
1.121 + {
1.122 + delete _buffer;
1.123 + }
1.124 +
1.125 + void Init(ISequentialInStream *stream)
1.126 + {
1.127 + _stream = stream;
1.128 + _size = 0;
1.129 + _currentSize = 0;
1.130 + _currentPos = 0;
1.131 + }
1.132 + UInt64 GetSize() const { return _size; }
1.133 +
1.134 + MY_UNKNOWN_IMP
1.135 +
1.136 + STDMETHOD(Read)(void *data, UInt32 size, UInt32 *processedSize);
1.137 + HRESULT Rollback(size_t rollbackSize);
1.138 +};
1.139 +
1.140 +class CSequentialOutStreamSizeCount:
1.141 + public ISequentialOutStream,
1.142 + public CMyUnknownImp
1.143 +{
1.144 + CMyComPtr<ISequentialOutStream> _stream;
1.145 + UInt64 _size;
1.146 +public:
1.147 + void Init(ISequentialOutStream *stream)
1.148 + {
1.149 + _stream = stream;
1.150 + _size = 0;
1.151 + }
1.152 + UInt64 GetSize() const { return _size; }
1.153 +
1.154 + MY_UNKNOWN_IMP
1.155 +
1.156 + STDMETHOD(Write)(const void *data, UInt32 size, UInt32 *processedSize);
1.157 +};
1.158 +
1.159 +#endif
