zipExampleUsage.cpp

Go to the documentation of this file.

/****************************************************************************
** Filename: main.cpp
** Last updated [dd/mm/yyyy]: 01/02/2007
**
** Test routine for the Zip and UnZip classed.
**
** Copyright (C) 2007 Angius Fabrizio. All rights reserved.
**
** This file is part of the OSDaB project (http://osdab.sourceforge.net/).
**
** 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.
**
**********************************************************************/

#include "zip.h"
#include "unzip.h"

#include <QFile>
#include <QFileInfo>

#include <QString>
#include <QStringList>
#include <QList>

#include <iostream>
#include <iomanip>

void invalidCMD();
bool decompress(const QString& file, const QString& out, const QString& pwd);
bool compress(const QString& zip, const QString& dir, const QString& pwd);
bool listFiles(const QString& file, const QString& pwd);

using namespace std;

int main(int argc, char** argv)
{
    if (argc < 3)
    {
        cout << "Test routine for the OSDaB Project Zip/UnZip classes" << endl << endl;
        cout << "Compression: zip [-p PWD] ZIPFILE DIRECTORY" << endl;
        cout << "List files: zip -l [-p PWD] ZIPFILE" << endl;
        cout << "Decompression: zip -d [-p PWD] ZIPFILE OUTPUT_DIR" << endl << endl;
        cout << "(C) 2007 Angius Fabrizio\nLicensed under the terms of the GNU GPL Version 2 or later" << endl;
        return -1;
    }

    QString fname;
    QString dname;
    QString pwd;

    bool resOK = true;

    if (strlen(argv[1]) == 2 && argv[1][0] == '-')
    {
        switch (argv[1][1])
        {
            case 'd':
            {
                if (argc >= 6)
                {
                    if (strcmp(argv[2], "-p") == 0)
                    {
                        pwd = QString(argv[3]);
                        fname = QString(argv[4]);
                        dname = QString(argv[5]);
                    }
                    else invalidCMD();
                }
                else if (argc >= 4)
                {
                    fname = QString(argv[2]);
                    dname = QString(argv[3]);
                }
                else invalidCMD();

                resOK = decompress(fname, dname, pwd);
            }
            break;
            case 'l':
            {
                if (argc >= 5)
                {
                    if (strcmp(argv[2], "-p") == 0)
                    {
                        pwd = QString(argv[3]);
                        fname = QString(argv[4]);
                    }
                    else invalidCMD();
                }
                else if (argc >= 3)
                {
                    fname = QString(argv[2]);
                }
                else invalidCMD();

                resOK = listFiles(fname, pwd);
            }
            break;
            case 'p':
            {
                if (argc >= 5)
                {
                    pwd = QString(argv[2]);
                    fname = QString(argv[3]);
                    dname = QString(argv[4]);
                }
                else invalidCMD();

                resOK = compress(fname, dname, pwd);
            }
            break;
            default: invalidCMD();
        }
    }
    else
    {
        // no parameters -- compress directly
        resOK = compress(QString(argv[1]), QString(argv[2]), 0);
    }


    if (!resOK)
    {
        cout << "Sorry, some error occurred!" << endl;
        return -1;
    }

    return 0;
}

void invalidCMD()
{
        cout << "Invalid command line. Usage:" << endl;
        cout << "Compression: zip [-p PWD] DIRECTORY" << endl;
        cout << "List files: zip -l [-p PWD] ZIPFILE" << endl;
        cout << "Decompression: zip -d [-p PWD] ZIPFILE OUTPUT_DIR" << endl << endl;
        exit(-1);
}

bool decompress(const QString& file, const QString& out, const QString& pwd)
{

    if (!QFile::exists(file))
    {
        cout << "File does not exist." << endl << endl;
        return false;
    }

    UnZip::ErrorCode ec;
    UnZip uz;

    if (!pwd.isEmpty())
        uz.setPassword(pwd);

    ec = uz.openArchive(file);
    if (ec != UnZip::Ok)
    {
        cout << "Failed to open archive: " << uz.formatError(ec).toAscii().data() << endl << endl;
        return false;
    }

    ec = uz.extractAll(out);
    if (ec != UnZip::Ok)
    {
        cout << "Extraction failed: " << uz.formatError(ec).toAscii().data() << endl << endl;
        uz.closeArchive();
        return false;
    }

    return true;
}

bool compress(const QString& zip, const QString& dir, const QString& pwd)
{
    QFileInfo fi(dir);
    if (!fi.isDir())
    {
        cout << "Directory does not exist." << endl << endl;
        return false;
    }

    Zip::ErrorCode ec;
    Zip uz;

    ec = uz.createArchive(zip);
    if (ec != Zip::Ok)
    {
        cout << "Unable to create archive: " << uz.formatError(ec).toAscii().data() << endl << endl;
        return false;
    }

    uz.setPassword(pwd);
    ec = uz.addDirectory(dir);
    if (ec != Zip::Ok)
    {
        cout << "Unable to add directory: " << uz.formatError(ec).toAscii().data() << endl << endl;
    }

    uz.setArchiveComment("This archive has been created using OSDaB Zip (http://osdab.sourceforge.net/).");

    if (uz.closeArchive() != Zip::Ok)
    {
        cout << "Unable to close the archive: " << uz.formatError(ec).toAscii().data() << endl << endl;
    }

    return ec == Zip::Ok;
}

bool listFiles(const QString& file, const QString& pwd)
{
    if (!QFile::exists(file))
    {
        cout << "File does not exist." << endl << endl;
        return false;
    }

    UnZip::ErrorCode ec;
    UnZip uz;

    if (!pwd.isEmpty())
        uz.setPassword(pwd);

    ec = uz.openArchive(file);
    if (ec != UnZip::Ok)
    {
        cout << "Unable to open archive: " << uz.formatError(ec).toAscii().data() << endl << endl;
        return false;
    }

    QString comment = uz.archiveComment();
    if (!comment.isEmpty())
        cout << "Archive comment: " << comment.toAscii().data() << endl << endl;

    QList<UnZip::ZipEntry> list = uz.entryList();
    if (list.isEmpty())
    {
        cout << "Empty archive.";
    }
    else
    {
        cout.setf(ios::left);
        cout << setw(40) << "Filename";
        cout.unsetf(ios::left);
        cout << setw(10) << "Size" << setw(10) << "Ratio" << setw(10) << "CRC32" << endl;
        cout.setf(ios::left);
        cout << setw(40) << "--------";
        cout.unsetf(ios::left);
        cout << setw(10) << "----" << setw(10) << "-----" << setw(10) << "-----" << endl;

        for (int i = 0; i < list.size(); ++i)
        {
            const UnZip::ZipEntry& entry = list.at(i);

            double ratio = entry.uncompressedSize == 0 ? 0 : 100 - (double) entry.compressedSize * 100 / (double) entry.uncompressedSize;

            QString ratioS = QString::number(ratio, 'f', 2).append("%");
            QString crc;
            crc = crc.sprintf("%X", entry.crc32).rightJustified(8, '0');
            QString file = entry.filename;
            int idx = file.lastIndexOf("/");
            if (idx >= 0 && idx != file.length()-1)
                file = file.right(file.length() - idx - 1);
            file = file.leftJustified(40, ' ', true);

            if (entry.encrypted)
                file.append("*");

            cout << setw(40) << file.toAscii().data() << setw(10) << entry.uncompressedSize << setw(10) << ratioS.toAscii().data() << setw(10) << crc.toAscii().data() << endl;
        }
    }

    uz.closeArchive();
    return true;
}