include/library/zlibstream.hpp

   1 #ifndef _library__zlibstream__hpp__included__
   2 #define _library__zlibstream__hpp__included__
   3
   4 #include <vector>
   5 #include <list>
   6 #include <cstdint>
   7 #include <zlib.h>
   8 #define ZLIB_PAGE_STORAGE 65500
   9
  10 /**
  11  * Zlib stream.
  12  */
  13 struct zlibstream
  14 {
  15 /**
  16  * Create new zlib stream.
  17  *
  18  * Parameter compression: The compression level.
  19  */
  20         zlibstream(unsigned compression);
  21 /**
  22  * Destroy zlib stream.
  23  */
  24         ~zlibstream();
  25 /**
  26  * Reset the stream, adding some initial data.
  27  *
  28  * Parameter data: The data to write (may be NULL if datalen=0).
  29  * Parameter datalen: The length of data.
  30  */
  31         void reset(uint8_t* data, size_t datalen);
  32 /**
  33  * Write data into stream.
  34  *
  35  * Parameter data: The data to write.
  36  * Parameter datalen: The length of data.
  37  */
  38         void write(uint8_t* data, size_t datalen);
  39 /**
  40  * Read the data as buffer.
  41  *
  42  * Parameter out: The buffer to write to.
  43  */
  44         void read(std::vector<char>& out);
  45 /**
  46  * Read the data so far as buffer and empty compressed data.
  47  *
  48  * This performs a sync flush.
  49  *
  50  * Parameter out: The buffer to write to.
  51  */
  52         void readsync(std::vector<char>& out);
  53 /**
  54  * Clear the output stream and add some uncompressed data.
  55  *
  56  * This is useful after doing readsync().
  57  *
  58  * Parameter data: The data to write (may be NULL if datalen=0).
  59  * Parameter datalen: The length of data.
  60  */
  61         void adddata(uint8_t* data, size_t datalen);
  62 /**
  63  * Set flag.
  64  *
  65  * Parameter f: New flag value.
  66  */
  67         void set_flag(bool f);
  68 /**
  69  * Get flag.
  70  *
  71  * Returns: Flag value.
  72  */
  73         bool get_flag();
  74 private:
  75         struct page
  76         {
  77                 uint8_t data[ZLIB_PAGE_STORAGE];
  78                 size_t used;
  79         };
  80
  81         void _read(std::vector<char>& out, int mode);
  82         void _reset(uint8_t* data, size_t datalen, bool doactually);
  83         void flushpage(uint8_t* data, size_t datalen, int mode);
  84         zlibstream(zlibstream&);
  85         zlibstream& operator=(zlibstream&);
  86         z_stream z;
  87         size_t streamsize;
  88         std::list<page> storage;
  89         bool flag;
  90 };
  91
  92 #endif