// // DeflateStream.h // OpenSpades // // Created by yvt on 7/16/13. // Copyright (c) 2013 yvt.jp. All rights reserved. // #pragma once #include "IStream.h" #include #include namespace spades { class DeflateStream: public IStream { IStream *baseStream; CompressMode mode; z_stream zstream; bool autoClose; bool valid; uint64_t position; // when compressing: // input buffer // when decompressing: // output buffer std::vector buffer; // input buffer that wasn't used in FillBuffer // is stored here std::vector nextbuffer; // decompression only bool reachedEOF; size_t bufferPos; void CompressBuffer(); void FillBuffer(); public: DeflateStream(IStream *stream, CompressMode mode, bool autoClose = false); virtual ~DeflateStream(); virtual int ReadByte(); virtual size_t Read(void *, size_t bytes); virtual void WriteByte(int); virtual void Write(const void *, size_t bytes); virtual uint64_t GetPosition(); virtual void SetPosition(uint64_t); virtual uint64_t GetLength(); virtual void SetLength(uint64_t); /** Must be called when all data was written, in case of compressing, or output will be corrupted */ void DeflateEnd(); }; }