CompressionStream.h

Go to the documentation of this file.

//===-- CompressionStream.h --------------------------------------------===//
//
//                     The KLEE Symbolic Virtual Machine
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
 
#ifndef KLEE_COMPRESSIONSTREAM_H
#define KLEE_COMPRESSIONSTREAM_H
 
#include "llvm/Support/raw_ostream.h"
#include "zlib.h"
 
namespace klee {
const size_t BUFSIZE = 128 * 1024;
 
class compressed_fd_ostream : public llvm::raw_ostream {
  int FD;
  uint8_t buffer[BUFSIZE];
  z_stream strm;
  uint64_t pos;
 
  virtual void write_impl(const char *Ptr, size_t Size);
  void write_file(const char *Ptr, size_t Size);
 
  virtual uint64_t current_pos() const { return pos; }
 
  void flush_compressed_data();
  void writeFullCompressedData();
 
public:
  compressed_fd_ostream(const std::string &Filename, std::string &ErrorInfo);
 
  ~compressed_fd_ostream();
};
}
 
#endif /* KLEE_COMPRESSIONSTREAM_H */