/home/cypheon/dev/limiro/trunk/src/MiniZip.h

// -*- C++ -*-
/* Copyright (C) 2007-2008 Johann Rudloff
 * License: GNU GPLv2 or later, see file "COPYING". No warranty of any kind! */

#ifndef _MINI_ZIP_H_
#define _MINI_ZIP_H_

#include <zlib.h>

#include <string>
#include <ctime>
#include <iostream>
#include <vector>

namespace MiniZip
{

struct _LocalFileHeader
{
  long locSig;

  short minVer;
  short flag;
  short compr;

  unsigned short modTime;
  unsigned short modDate;

  long crc32;

  long szCompr;
  long szUncompr;
  short lenFilename;
  short lenExtra;
} __attribute__((__packed__));

struct _CentralFileHeader
{
  long ctrSig;

  short verMade;
  short verNeeded;
  short flag;
  short compr;

  unsigned short modTime;
  unsigned short modDate;

  long crc32;

  long szCompr;
  long szUncompr;

  short lenFilename;
  short lenExtra;
  short lenComment;

  short numDiskStart;

  short attrInternal;
  long attrExternal;

  long offLocalHeader;
} __attribute__((__packed__));

struct _EndOfCDir
{
  long endSig;

  short numDisk;
  short numDiskStart;
  short numEntriesDisk;
  short numEntries;

  long szCentr;
  long offCentr;

  short lenComment;
} __attribute__((__packed__));


inline unsigned long crc32(const std::string& data)
{
  unsigned long crc = ::crc32(0, NULL, 0);
  return ::crc32(crc, (const Bytef *)data.c_str(), data.length());
};


class ZipInfo
{
public:
  ZipInfo(const std::string& filename, const std::tm *time);

  void setData(const std::string& data, bool compress);
  const std::string& getData() const {return m_data;};

  void setOffset(long offset) {m_offset = offset;};

  std::string localFileHeader() const;
  std::string centralFileHeader() const;

private:
  std::string m_filename;
  std::tm m_time;
  unsigned short m_flags;
  unsigned short m_compression;
  std::string m_extra;

  long m_offset;
  long m_origSize;
  long m_crc;
  std::string m_data;
};

class ZipFile
{
public:
  ZipFile(std::ostream& out);

  bool addString(const std::string& filename, const std::string& data, bool compress=true);

  bool finalize();

private:
  void writeCentralDirectory();

  std::ostream& m_out;
  bool m_finalized;
  std::vector<ZipInfo> m_files;
};

}

#endif // _MINI_ZIP_H_

---