zip_p.h

/****************************************************************************
** Filename: zip_p.h
** Last updated [dd/mm/yyyy]: 27/03/2011
**
** pkzip 2.0 file compression.
**
** Some of the code has been inspired by other open source projects,
** (mainly Info-Zip and Gilles Vollant's minizip).
** Compression and decompression actually uses the zlib library.
**
** Copyright (C) 2007-2012 Angius Fabrizio. All rights reserved.
**
** This file is part of the OSDaB project (http://osdab.42cows.org/).
**
** This file may be distributed and/or modified under the terms of the
** GNU General Public License version 2 as published by the Free Software
** Foundation and appearing in the file LICENSE.GPL included in the
** packaging of this file.
**
** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
**
** See the file LICENSE.GPL that came with this software distribution or
** visit http://www.gnu.org/copyleft/gpl.html for GPL licensing information.
**
**********************************************************************/

//
//  W A R N I N G
//  -------------
//
// This file is not part of the Zip/UnZip API.  It exists purely as an
// implementation detail. This header file may change from version to
// version without notice, or even be removed.
//
// We mean it.
//

#ifndef OSDAB_ZIP_P__H
#define OSDAB_ZIP_P__H

#include "zip.h"
#include "zipentry_p.h"

#include <QtCore/QFileInfo>
#include <QtCore/QObject>
#include <QtCore/QtGlobal>

#include <zlib.h>
//#include <zlib/zconf.h>

#define ZIP_READ_BUFFER (256*1024)

OSDAB_BEGIN_NAMESPACE(Zip)

class ZipPrivate : public QObject
{
    Q_OBJECT

public:
    typedef z_crc_t crc_t;
    ZipPrivate();
    virtual ~ZipPrivate();

    QMap<QString,ZipEntryP*>* headers;

    QIODevice* device;
    QFile* file;

    char buffer1[ZIP_READ_BUFFER];
    char buffer2[ZIP_READ_BUFFER];

    unsigned char* uBuffer;

    const crc_t* crcTable;

    QString comment;
    QString password;

    Zip::ErrorCode createArchive(QIODevice* device);
    Zip::ErrorCode closeArchive();
    void reset();

    bool zLibInit();

    bool containsEntry(const QFileInfo& info) const;

    Zip::ErrorCode addDirectory(const QString& path, const QString& root,
        Zip::CompressionOptions options, Zip::CompressionLevel level,
        int hierarchyLevel, int* addedFiles = 0);
    Zip::ErrorCode addFiles(const QStringList& paths, const QString& root,
        Zip::CompressionOptions options, Zip::CompressionLevel level,
        int* addedFiles);

    Zip::ErrorCode createEntry(const QFileInfo& file, const QString& root,
        Zip::CompressionLevel level);
    Zip::CompressionLevel detectCompressionByMime(const QString& ext);

    inline quint32 updateChecksum(const quint32& crc, const quint32& val) const;

    inline void encryptBytes(quint32* keys, char* buffer, qint64 read);

    inline void setULong(quint32 v, char* buffer, unsigned int offset);
    inline void updateKeys(quint32* keys, int c) const;
    inline void initKeys(quint32* keys) const;
    inline int decryptByte(quint32 key2) const;

    inline QString extractRoot(const QString& p, Zip::CompressionOptions o);

private slots:
    void deviceDestroyed(QObject*);

private:
    int compressionStrategy(const QString& path, QIODevice& file) const;
    Zip::ErrorCode deflateFile(const QFileInfo& fileInfo,
        quint32& crc, qint64& written, const Zip::CompressionLevel& level, quint32** keys);
    Zip::ErrorCode storeFile(const QString& path, QIODevice& file,
        quint32& crc, qint64& written, quint32** keys);
    Zip::ErrorCode compressFile(const QString& path, QIODevice& file,
        quint32& crc, qint64& written, const Zip::CompressionLevel& level, quint32** keys);
    Zip::ErrorCode do_closeArchive();
    Zip::ErrorCode writeEntry(const QString& fileName, const ZipEntryP* h, quint32& szCentralDir);
    Zip::ErrorCode writeCentralDir(quint32 offCentralDir, quint32 szCentralDir);
};

OSDAB_END_NAMESPACE

#endif // OSDAB_ZIP_P__H