/* Copyright (C) 2005-2014 Sergey A. Tachenov This file is part of QuaZip test suite. QuaZip is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation, either version 2.1 of the License, or (at your option) any later version. QuaZip is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. You should have received a copy of the GNU Lesser General Public License along with QuaZip. If not, see . See COPYING file for the full LGPL text. Original ZIP package is copyrighted by Gilles Vollant and contributors, see quazip/(un)zip.h files for details. Basically it's the zlib license. */ #include "testquazipfile.h" #include "qztest.h" #include #include #include #include #include #include #include #include void TestQuaZipFile::zipUnzip_data() { QTest::addColumn("zipName"); QTest::addColumn("fileNames"); QTest::addColumn("fileNameCodec"); QTest::addColumn("password"); QTest::addColumn("zip64"); QTest::addColumn("utf8"); QTest::addColumn("size"); QTest::newRow("simple") << "simple.zip" << ( QStringList() << "test0.txt" << "testdir1/test1.txt" << "testdir2/test2.txt" << "testdir2/subdir/test2sub.txt") << QByteArray() << QByteArray() << false << false << -1; QTest::newRow("Cyrillic") << "cyrillic.zip" << ( QStringList() << QString::fromUtf8("русское имя файла с пробелами.txt")) << QByteArray("IBM866") << QByteArray() << false << false << -1; QTest::newRow("Unicode") << "unicode.zip" << ( QStringList() << QString::fromUtf8("Українське сало.txt") << QString::fromUtf8("Vin français.txt") << QString::fromUtf8("日本の寿司.txt") << QString::fromUtf8("ქართული ხაჭაპური.txt")) << QByteArray("") << QByteArray() << false << true << -1; QTest::newRow("password") << "password.zip" << ( QStringList() << "test.txt") << QByteArray() << QByteArray("PassPass") << false << false << -1; QTest::newRow("zip64") << "zip64.zip" << ( QStringList() << "test64.txt") << QByteArray() << QByteArray() << true << false << -1; QTest::newRow("large enough to flush") << "flush.zip" << ( QStringList() << "flush.txt") << QByteArray() << QByteArray() << true << false << 65536 * 2; } void TestQuaZipFile::zipUnzip() { QFETCH(QString, zipName); QFETCH(QStringList, fileNames); QFETCH(QByteArray, fileNameCodec); QFETCH(QByteArray, password); QFETCH(bool, zip64); QFETCH(bool, utf8); QFETCH(int, size); QFile testFile(zipName); if (testFile.exists()) { if (!testFile.remove()) { QFAIL("Couldn't remove existing archive to create a new one"); } } if (!createTestFiles(fileNames, size)) { QFAIL("Couldn't create test files for zipping"); } QuaZip testZip(&testFile); testZip.setZip64Enabled(zip64); testZip.setUtf8Enabled(utf8); if (!fileNameCodec.isEmpty()) testZip.setFileNameCodec(fileNameCodec); QVERIFY(testZip.open(QuaZip::mdCreate)); QString comment = utf8 ? "test テスト ჩეკი" : "Test comment"; testZip.setComment(comment); foreach (QString fileName, fileNames) { QFile inFile("tmp/" + fileName); if (!inFile.open(QIODevice::ReadOnly)) { qDebug("File name: %s", fileName.toUtf8().constData()); QFAIL("Couldn't open input file"); } QuaZipFile outFile(&testZip); QVERIFY(outFile.open(QIODevice::WriteOnly, QuaZipNewInfo(fileName, inFile.fileName()), password.isEmpty() ? NULL : password.constData())); for (qint64 pos = 0, len = inFile.size(); pos < len; ) { char buf[4096]; qint64 readSize = qMin(static_cast(4096), len - pos); qint64 l; if ((l = inFile.read(buf, readSize)) != readSize) { qDebug("Reading %ld bytes from %s at %ld returned %ld", static_cast(readSize), fileName.toUtf8().constData(), static_cast(pos), static_cast(l)); QFAIL("Read failure"); } QVERIFY(outFile.write(buf, readSize)); pos += readSize; } inFile.close(); outFile.close(); QCOMPARE(outFile.getZipError(), ZIP_OK); } testZip.close(); QCOMPARE(testZip.getZipError(), ZIP_OK); // now test unzip QuaZip testUnzip(&testFile); if (!fileNameCodec.isEmpty()) testUnzip.setFileNameCodec(fileNameCodec); QVERIFY(testUnzip.open(QuaZip::mdUnzip)); QCOMPARE(testUnzip.getComment(), comment); QVERIFY(testUnzip.goToFirstFile()); foreach (QString fileName, fileNames) { QuaZipFileInfo64 info; QVERIFY(testUnzip.getCurrentFileInfo(&info)); QCOMPARE(info.name, fileName); QCOMPARE(info.isEncrypted(), !password.isEmpty()); QFile original("tmp/" + fileName); QVERIFY(original.open(QIODevice::ReadOnly)); QuaZipFile archived(&testUnzip); QVERIFY(archived.open(QIODevice::ReadOnly, password.isEmpty() ? NULL : password.constData())); QByteArray originalData = original.readAll(); QByteArray archivedData = archived.readAll(); QCOMPARE(archivedData, originalData); testUnzip.goToNextFile(); } if (!password.isEmpty()) { QVERIFY(testUnzip.goToFirstFile()); QuaZipFileInfo64 info; QVERIFY(testUnzip.getCurrentFileInfo(&info)); QFile original("tmp/" + info.name); QVERIFY(original.open(QIODevice::ReadOnly)); QuaZipFile archived(&testUnzip); QVERIFY(archived.open(QIODevice::ReadOnly, "WrongPassword")); QByteArray originalData = original.readAll(); QByteArray archivedData = archived.readAll(); QVERIFY(archivedData != originalData); } testUnzip.close(); QCOMPARE(testUnzip.getZipError(), UNZ_OK); // clean up removeTestFiles(fileNames); testFile.remove(); } void TestQuaZipFile::bytesAvailable_data() { QTest::addColumn("zipName"); QTest::addColumn("fileNames"); QTest::addColumn("size"); QTest::newRow("simple") << "test.zip" << ( QStringList() << "test0.txt" << "testdir1/test1.txt" << "testdir2/test2.txt" << "testdir2/subdir/test2sub.txt") << -1; QTest::newRow("large enough to flush") << "flush.zip" << (QStringList() << "test.txt") << 65536 * 4; } void TestQuaZipFile::bytesAvailable() { QFETCH(QString, zipName); QFETCH(QStringList, fileNames); QFETCH(int, size); QDir curDir; if (!createTestFiles(fileNames, size)) { QFAIL("Couldn't create test files"); } if (!JlCompress::compressDir(zipName, "tmp")) { QFAIL("Couldn't create test archive"); } QuaZip testZip(zipName); QVERIFY(testZip.open(QuaZip::mdUnzip)); foreach (QString fileName, fileNames) { QFileInfo fileInfo("tmp/" + fileName); QVERIFY(testZip.setCurrentFile(fileName)); QuaZipFile zipFile(&testZip); QVERIFY(zipFile.open(QIODevice::ReadOnly)); QCOMPARE(zipFile.bytesAvailable(), fileInfo.size()); QCOMPARE(zipFile.read(1).size(), 1); QCOMPARE(zipFile.bytesAvailable(), fileInfo.size() - 1); QCOMPARE(zipFile.read(fileInfo.size() - 1).size(), static_cast(fileInfo.size() - 1)); QCOMPARE(zipFile.bytesAvailable(), (qint64) 0); } removeTestFiles(fileNames); testZip.close(); curDir.remove(zipName); } void TestQuaZipFile::atEnd_data() { bytesAvailable_data(); } void TestQuaZipFile::atEnd() { QFETCH(QString, zipName); QFETCH(QStringList, fileNames); QFETCH(int, size); QDir curDir; if (!createTestFiles(fileNames, size)) { QFAIL("Couldn't create test files"); } if (!JlCompress::compressDir(zipName, "tmp")) { QFAIL("Couldn't create test archive"); } QuaZip testZip(zipName); QVERIFY(testZip.open(QuaZip::mdUnzip)); foreach (QString fileName, fileNames) { QFileInfo fileInfo("tmp/" + fileName); QVERIFY(testZip.setCurrentFile(fileName)); QuaZipFile zipFile(&testZip); QVERIFY(zipFile.open(QIODevice::ReadOnly)); QCOMPARE(zipFile.atEnd(), false); QCOMPARE(zipFile.read(1).size(), 1); QCOMPARE(zipFile.atEnd(), false); QCOMPARE(zipFile.read(fileInfo.size() - 1).size(), static_cast(fileInfo.size() - 1)); QCOMPARE(zipFile.atEnd(), true); } removeTestFiles(fileNames); testZip.close(); curDir.remove(zipName); } void TestQuaZipFile::posRead_data() { bytesAvailable_data(); } void TestQuaZipFile::posRead() { QFETCH(QString, zipName); QFETCH(QStringList, fileNames); QFETCH(int, size); QDir curDir; if (!createTestFiles(fileNames, size)) { QFAIL("Couldn't create test files"); } if (!JlCompress::compressDir(zipName, "tmp")) { QFAIL("Couldn't create test archive"); } QuaZip testZip(zipName); QVERIFY(testZip.open(QuaZip::mdUnzip)); foreach (QString fileName, fileNames) { QFileInfo fileInfo("tmp/" + fileName); QVERIFY(testZip.setCurrentFile(fileName)); QuaZipFile zipFile(&testZip); QVERIFY(zipFile.open(QIODevice::ReadOnly)); QCOMPARE(zipFile.pos(), (qint64) 0); QCOMPARE(zipFile.read(1).size(), 1); QCOMPARE(zipFile.pos(), (qint64) 1); QCOMPARE(zipFile.read(fileInfo.size() - 1).size(), static_cast(fileInfo.size() - 1)); QCOMPARE(zipFile.pos(), fileInfo.size()); } removeTestFiles(fileNames); testZip.close(); curDir.remove(zipName); } void TestQuaZipFile::posWrite_data() { posRead_data(); } void TestQuaZipFile::posWrite() { QFETCH(QString, zipName); QFETCH(QStringList, fileNames); QFETCH(int, size); if (size == -1) size = 20; QDir curDir; QuaZip testZip(zipName); QVERIFY(testZip.open(QuaZip::mdCreate)); foreach (QString fileName, fileNames) { QuaZipFile zipFile(&testZip); QVERIFY(zipFile.open(QIODevice::WriteOnly, QuaZipNewInfo(fileName))); QCOMPARE(zipFile.pos(), (qint64) 0); zipFile.putChar('0'); QCOMPARE(zipFile.pos(), (qint64) 1); QByteArray buffer(size / 2 - 1, '\0'); for (int i = 0; i < buffer.size(); ++i) buffer[i] = static_cast(i); zipFile.write(buffer); QCOMPARE(zipFile.pos(), qint64(size / 2)); for (int i = 0; i < size - size / 2; ++i) { zipFile.putChar(static_cast(i)); } QCOMPARE(zipFile.pos(), qint64(size)); } testZip.close(); curDir.remove(zipName); } void TestQuaZipFile::getZip() { QuaZip testZip; QuaZipFile f1(&testZip); QCOMPARE(f1.getZip(), &testZip); QuaZipFile f2("doesntexist.zip", "someFile"); QCOMPARE(f2.getZip(), static_cast(NULL)); f2.setZip(&testZip); QCOMPARE(f2.getZip(), &testZip); } void TestQuaZipFile::setZipName() { QString testFileName = "testZipName.txt"; QString testZipName = "testZipName.zip"; QVERIFY(createTestFiles(QStringList() << testFileName)); QVERIFY(createTestArchive(testZipName, QStringList() << testFileName)); QuaZipFile testFile; testFile.setZipName(testZipName); QCOMPARE(testFile.getZipName(), testZipName); testFile.setFileName(testFileName); QVERIFY(testFile.open(QIODevice::ReadOnly)); testFile.close(); removeTestFiles(QStringList() << testFileName); QDir curDir; curDir.remove(testZipName); } void TestQuaZipFile::getFileInfo() { QuaZipFileInfo info32; QuaZipFileInfo64 info64; QString testFileName = "testZipName.txt"; QStringList testFiles; testFiles << testFileName; QString testZipName = "testZipName.zip"; QVERIFY(createTestFiles(testFiles)); QVERIFY(createTestArchive(testZipName, testFiles)); QuaZipFile testFile; testFile.setZipName(testZipName); testFile.setFileName(testFileName); QVERIFY(testFile.open(QIODevice::ReadOnly)); QVERIFY(testFile.getFileInfo(&info32)); QVERIFY(testFile.getFileInfo(&info64)); QCOMPARE(info32.name, info64.name); QCOMPARE(info32.versionCreated, info64.versionCreated); QCOMPARE(info32.versionNeeded, info64.versionNeeded); QCOMPARE(info32.flags, info64.flags); QCOMPARE(info32.method, info64.method); QCOMPARE(info32.dateTime, info64.dateTime); QCOMPARE(info32.crc, info64.crc); QCOMPARE(info32.compressedSize, static_cast(info64.compressedSize)); QCOMPARE(info32.uncompressedSize, static_cast(info64.uncompressedSize)); QCOMPARE(info32.diskNumberStart, info64.diskNumberStart); QCOMPARE(info32.internalAttr, info64.internalAttr); QCOMPARE(info32.externalAttr, info64.externalAttr); QCOMPARE(info32.comment, info64.comment); QCOMPARE(info32.extra, info64.extra); testFile.close(); removeTestFiles(testFiles); QDir curDir; curDir.remove(testZipName); } void TestQuaZipFile::setFileName() { QString testFileName = "testZipName.txt"; QString testZipName = "testZipName.zip"; QVERIFY(createTestFiles(QStringList() << testFileName)); QVERIFY(createTestArchive(testZipName, QStringList() << testFileName)); QuaZipFile testFile(testZipName); testFile.setFileName(testFileName.toUpper()); #ifdef Q_OS_WIN QVERIFY(testFile.open(QIODevice::ReadOnly)); testFile.close(); #else QVERIFY(!testFile.open(QIODevice::ReadOnly)); #endif testFile.setFileName(testFileName.toUpper(), QuaZip::csInsensitive); QCOMPARE(testFile.getCaseSensitivity(), QuaZip::csInsensitive); QVERIFY(testFile.open(QIODevice::ReadOnly)); QCOMPARE(testFile.getActualFileName(), testFileName); testFile.close(); testFile.setFileName(testFileName.toUpper(), QuaZip::csSensitive); QCOMPARE(testFile.getFileName(), testFileName.toUpper()); QCOMPARE(testFile.getActualFileName(), QString()); QVERIFY(!testFile.open(QIODevice::ReadOnly)); testFile.setFileName(testFileName); removeTestFiles(QStringList() << testFileName); QDir curDir; curDir.remove(testZipName); } void TestQuaZipFile::constructorDestructor() { // Just test that all constructors and destructors are available. // (So there are none that are declared but not defined.) QuaZip testZip; QuaZipFile *f1 = new QuaZipFile(); delete f1; // test D0 destructor QObject parent; QuaZipFile f2(&testZip, &parent); QuaZipFile f3(&parent); QuaZipFile f4("zipName.zip"); QuaZipFile f5("zipName.zip", "fileName.txt", QuaZip::csDefault, &parent); } void TestQuaZipFile::setFileAttrs() { QuaZip testZip("setFileAttrs.zip"); QVERIFY(testZip.open(QuaZip::mdCreate)); QuaZipFile zipFile(&testZip); QuaZipNewInfo newInfo("testPerm.txt"); newInfo.setPermissions(QFile::ReadOwner); QVERIFY(zipFile.open(QIODevice::WriteOnly, newInfo)); zipFile.close(); QString fileTestAttr = "testAttr.txt"; QStringList fileNames; fileNames << fileTestAttr; QVERIFY(createTestFiles(fileNames)); newInfo.name = fileTestAttr; newInfo.setFileDateTime("tmp/" + fileTestAttr); newInfo.setFilePermissions("tmp/" + fileTestAttr); QVERIFY(zipFile.open(QIODevice::WriteOnly, newInfo)); zipFile.close(); testZip.close(); QuaZipFileInfo64 info; { QuaZipFile readFilePerm("setFileAttrs.zip", "testPerm.txt"); QVERIFY(readFilePerm.open(QIODevice::ReadOnly)); QVERIFY(readFilePerm.getFileInfo(&info)); QCOMPARE(info.getPermissions(), QFile::ReadOwner); readFilePerm.close(); } { QuaZipFile readFileAttrs("setFileAttrs.zip", "testAttr.txt"); QVERIFY(readFileAttrs.open(QIODevice::ReadOnly)); QVERIFY(readFileAttrs.getFileInfo(&info)); QFileInfo srcInfo("tmp/" + fileTestAttr); QFile::Permissions usedPermissions = QFile::WriteOwner | QFile::ReadOwner | QFile::ExeOwner | QFile::WriteGroup | QFile::ReadGroup | QFile::ExeGroup | QFile::WriteOther | QFile::ReadOther | QFile::ExeOther; QCOMPARE(info.getPermissions() & usedPermissions, srcInfo.permissions() & usedPermissions); qint64 newTime = quazip_to_time64_t(info.dateTime); qint64 oldTime = quazip_to_time64_t(srcInfo.lastModified()); // ZIP uses weird format with 2 second precision QCOMPARE(newTime / 2, oldTime / 2); readFileAttrs.close(); } removeTestFiles(fileNames); QDir().remove(testZip.getZipName()); } void TestQuaZipFile::largeFile() { QDir curDir; QVERIFY(curDir.mkpath("tmp")); QFile fakeLargeFile("tmp/large.zip"); QVERIFY(fakeLargeFile.open(QIODevice::WriteOnly)); QDataStream ds(&fakeLargeFile); ds.setByteOrder(QDataStream::LittleEndian); QList localOffsets; const int numFiles = 2; // name fixed to 5 bytes, so MAX 10 FILES!!! for (int i = 0; i < numFiles; ++i) { localOffsets.append(fakeLargeFile.pos()); QBuffer extra; extra.open(QIODevice::WriteOnly); QDataStream es(&extra); es.setByteOrder(QDataStream::LittleEndian); // prepare extra es << static_cast(0x0001u); // zip64 es << static_cast(16); // extra data size es << static_cast(0); // uncompressed size es << static_cast(0); // compressed size // now the local header ds << static_cast(0x04034b50u); // local magic ds << static_cast(45); // version needed ds << static_cast(0); // flags ds << static_cast(0); // method ds << static_cast(0); // time 00:00:00 ds << static_cast(0x21); // date 1980-01-01 ds << static_cast(0); // CRC-32 ds << static_cast(0xFFFFFFFFu); // compressed size ds << static_cast(0xFFFFFFFFu); // uncompressed size ds << static_cast(5); // name length ds << static_cast(extra.size()); // extra length ds.writeRawData("file", 4); // name ds << static_cast('0' + i); // name (contd.) ds.writeRawData(extra.buffer(), extra.size()); } // central dir: qint64 centralStart = fakeLargeFile.pos(); for (int i = 0; i < numFiles; ++i) { QBuffer extra; extra.open(QIODevice::WriteOnly); QDataStream es(&extra); es.setByteOrder(QDataStream::LittleEndian); // prepare extra es << static_cast(0x0001u); // zip64 es << static_cast(24); // extra data size es << static_cast(0); // uncompressed size es << static_cast(0); // compressed size es << static_cast(localOffsets[i]); // now the central dir header ds << static_cast(0x02014b50u); // central magic ds << static_cast(45); // version made by ds << static_cast(45); // version needed ds << static_cast(0); // flags ds << static_cast(0); // method ds << static_cast(0); // time 00:00:00 ds << static_cast(0x21); // date 1980-01-01 ds << static_cast(0); // CRC-32 ds << static_cast(0xFFFFFFFFu); // compressed size ds << static_cast(0xFFFFFFFFu); // uncompressed size ds << static_cast(5); // name length ds << static_cast(extra.size()); // extra length ds << static_cast(0); // comment length ds << static_cast(0); // disk number ds << static_cast(0); // internal attrs ds << static_cast(0); // external attrs ds << static_cast(0xFFFFFFFFu); // local offset ds.writeRawData("file", 4); // name ds << static_cast('0' + i); // name (contd.) ds.writeRawData(extra.buffer(), extra.size()); } qint64 centralEnd = fakeLargeFile.pos(); // zip64 end ds << static_cast(0x06064b50); // zip64 end magic ds << static_cast(44); // size of the zip64 end ds << static_cast(45); // version made by ds << static_cast(45); // version needed ds << static_cast(0); // disk number ds << static_cast(0); // central dir disk number ds << static_cast(2); // number of entries on disk ds << static_cast(2); // total number of entries ds << static_cast(centralEnd - centralStart); // size ds << static_cast(centralStart); // offset // zip64 locator ds << static_cast(0x07064b50); // zip64 locator magic ds << static_cast(0); // disk number ds << static_cast(centralEnd); // offset ds << static_cast(1); // number of disks // zip32 end ds << static_cast(0x06054b50); // end magic ds << static_cast(0); // disk number ds << static_cast(0); // central dir disk number ds << static_cast(2); // number of entries ds << static_cast(0xFFFFFFFFu); // central dir size ds << static_cast(0xFFFFFFFFu); // central dir offset ds << static_cast(0); // comment length fakeLargeFile.close(); QuaZip fakeLargeZip("tmp/large.zip"); QVERIFY(fakeLargeZip.open(QuaZip::mdUnzip)); QCOMPARE(fakeLargeZip.getFileInfoList().size(), numFiles); QCOMPARE(fakeLargeZip.getFileInfoList()[0].uncompressedSize, static_cast(0)); fakeLargeZip.close(); curDir.remove("tmp/large.zip"); }