blob: 4569e872cd1780a54d4485bbb64c814c270a99e4 [file] [log] [blame]
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*/
package org.apache.commons.compress.archivers.zip;
import static org.apache.commons.compress.AbstractTestCase.getFile;
import static org.junit.Assert.*;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Enumeration;
import java.util.TreeMap;
import java.util.zip.ZipEntry;
import org.apache.commons.compress.utils.IOUtils;
import org.junit.After;
import org.junit.Test;
public class ZipFileTest {
private ZipFile zf = null;
@After
public void tearDown() {
ZipFile.closeQuietly(zf);
}
@Test
public void testCDOrder() throws Exception {
readOrderTest();
final ArrayList<ZipArchiveEntry> l = Collections.list(zf.getEntries());
assertEntryName(l, 0, "AbstractUnicodeExtraField");
assertEntryName(l, 1, "AsiExtraField");
assertEntryName(l, 2, "ExtraFieldUtils");
assertEntryName(l, 3, "FallbackZipEncoding");
assertEntryName(l, 4, "GeneralPurposeBit");
assertEntryName(l, 5, "JarMarker");
assertEntryName(l, 6, "NioZipEncoding");
assertEntryName(l, 7, "Simple8BitZipEncoding");
assertEntryName(l, 8, "UnicodeCommentExtraField");
assertEntryName(l, 9, "UnicodePathExtraField");
assertEntryName(l, 10, "UnixStat");
assertEntryName(l, 11, "UnparseableExtraFieldData");
assertEntryName(l, 12, "UnrecognizedExtraField");
assertEntryName(l, 13, "ZipArchiveEntry");
assertEntryName(l, 14, "ZipArchiveInputStream");
assertEntryName(l, 15, "ZipArchiveOutputStream");
assertEntryName(l, 16, "ZipEncoding");
assertEntryName(l, 17, "ZipEncodingHelper");
assertEntryName(l, 18, "ZipExtraField");
assertEntryName(l, 19, "ZipUtil");
assertEntryName(l, 20, "ZipLong");
assertEntryName(l, 21, "ZipShort");
assertEntryName(l, 22, "ZipFile");
}
@Test
public void testPhysicalOrder() throws Exception {
readOrderTest();
final ArrayList<ZipArchiveEntry> l = Collections.list(zf.getEntriesInPhysicalOrder());
assertEntryName(l, 0, "AbstractUnicodeExtraField");
assertEntryName(l, 1, "AsiExtraField");
assertEntryName(l, 2, "ExtraFieldUtils");
assertEntryName(l, 3, "FallbackZipEncoding");
assertEntryName(l, 4, "GeneralPurposeBit");
assertEntryName(l, 5, "JarMarker");
assertEntryName(l, 6, "NioZipEncoding");
assertEntryName(l, 7, "Simple8BitZipEncoding");
assertEntryName(l, 8, "UnicodeCommentExtraField");
assertEntryName(l, 9, "UnicodePathExtraField");
assertEntryName(l, 10, "UnixStat");
assertEntryName(l, 11, "UnparseableExtraFieldData");
assertEntryName(l, 12, "UnrecognizedExtraField");
assertEntryName(l, 13, "ZipArchiveEntry");
assertEntryName(l, 14, "ZipArchiveInputStream");
assertEntryName(l, 15, "ZipArchiveOutputStream");
assertEntryName(l, 16, "ZipEncoding");
assertEntryName(l, 17, "ZipEncodingHelper");
assertEntryName(l, 18, "ZipExtraField");
assertEntryName(l, 19, "ZipFile");
assertEntryName(l, 20, "ZipLong");
assertEntryName(l, 21, "ZipShort");
assertEntryName(l, 22, "ZipUtil");
}
@Test
public void testDoubleClose() throws Exception {
readOrderTest();
zf.close();
try {
zf.close();
} catch (final Exception ex) {
fail("Caught exception of second close");
}
}
@Test
public void testReadingOfStoredEntry() throws Exception {
final File f = File.createTempFile("commons-compress-zipfiletest", ".zip");
f.deleteOnExit();
OutputStream o = null;
InputStream i = null;
try {
o = new FileOutputStream(f);
final ZipArchiveOutputStream zo = new ZipArchiveOutputStream(o);
ZipArchiveEntry ze = new ZipArchiveEntry("foo");
ze.setMethod(ZipEntry.STORED);
ze.setSize(4);
ze.setCrc(0xb63cfbcdl);
zo.putArchiveEntry(ze);
zo.write(new byte[] { 1, 2, 3, 4 });
zo.closeArchiveEntry();
zo.close();
o.close();
o = null;
zf = new ZipFile(f);
ze = zf.getEntry("foo");
assertNotNull(ze);
i = zf.getInputStream(ze);
final byte[] b = new byte[4];
assertEquals(4, i.read(b));
assertEquals(-1, i.read());
} finally {
if (o != null) {
o.close();
}
if (i != null) {
i.close();
}
f.delete();
}
}
/**
* @see "https://issues.apache.org/jira/browse/COMPRESS-176"
*/
@Test
public void testWinzipBackSlashWorkaround() throws Exception {
final File archive = getFile("test-winzip.zip");
zf = new ZipFile(archive);
assertNull(zf.getEntry("\u00e4\\\u00fc.txt"));
assertNotNull(zf.getEntry("\u00e4/\u00fc.txt"));
}
/**
* Test case for
* <a href="https://issues.apache.org/jira/browse/COMPRESS-208"
* >COMPRESS-208</a>.
*/
@Test
public void testSkipsPK00Prefix() throws Exception {
final File archive = getFile("COMPRESS-208.zip");
zf = new ZipFile(archive);
assertNotNull(zf.getEntry("test1.xml"));
assertNotNull(zf.getEntry("test2.xml"));
}
@Test
public void testUnixSymlinkSampleFile() throws Exception {
final String entryPrefix = "COMPRESS-214_unix_symlinks/";
final TreeMap<String, String> expectedVals = new TreeMap<>();
// I threw in some Japanese characters to keep things interesting.
expectedVals.put(entryPrefix + "link1", "../COMPRESS-214_unix_symlinks/./a/b/c/../../../\uF999");
expectedVals.put(entryPrefix + "link2", "../COMPRESS-214_unix_symlinks/./a/b/c/../../../g");
expectedVals.put(entryPrefix + "link3", "../COMPRESS-214_unix_symlinks/././a/b/c/../../../\u76F4\u6A39");
expectedVals.put(entryPrefix + "link4", "\u82B1\u5B50/\u745B\u5B50");
expectedVals.put(entryPrefix + "\uF999", "./\u82B1\u5B50/\u745B\u5B50/\u5897\u8C37/\uF999");
expectedVals.put(entryPrefix + "g", "./a/b/c/d/e/f/g");
expectedVals.put(entryPrefix + "\u76F4\u6A39", "./g");
// Notice how a directory link might contain a trailing slash, or it might not.
// Also note: symlinks are always stored as files, even if they link to directories.
expectedVals.put(entryPrefix + "link5", "../COMPRESS-214_unix_symlinks/././a/b");
expectedVals.put(entryPrefix + "link6", "../COMPRESS-214_unix_symlinks/././a/b/");
// I looked into creating a test with hard links, but zip does not appear to
// support hard links, so nevermind.
final File archive = getFile("COMPRESS-214_unix_symlinks.zip");
zf = new ZipFile(archive);
final Enumeration<ZipArchiveEntry> en = zf.getEntries();
while (en.hasMoreElements()) {
final ZipArchiveEntry zae = en.nextElement();
final String link = zf.getUnixSymlink(zae);
if (zae.isUnixSymlink()) {
final String name = zae.getName();
final String expected = expectedVals.get(name);
assertEquals(expected, link);
} else {
// Should be null if it's not a symlink!
assertNull(link);
}
}
}
/**
* @see "https://issues.apache.org/jira/browse/COMPRESS-227"
*/
@Test
public void testDuplicateEntry() throws Exception {
final File archive = getFile("COMPRESS-227.zip");
zf = new ZipFile(archive);
final ZipArchiveEntry ze = zf.getEntry("test1.txt");
assertNotNull(ze);
assertNotNull(zf.getInputStream(ze));
int numberOfEntries = 0;
for (final ZipArchiveEntry entry : zf.getEntries("test1.txt")) {
numberOfEntries++;
assertNotNull(zf.getInputStream(entry));
}
assertEquals(2, numberOfEntries);
}
/**
* @see "https://issues.apache.org/jira/browse/COMPRESS-228"
*/
@Test
public void testExcessDataInZip64ExtraField() throws Exception {
final File archive = getFile("COMPRESS-228.zip");
zf = new ZipFile(archive);
// actually, if we get here, the test already has passed
final ZipArchiveEntry ze = zf.getEntry("src/main/java/org/apache/commons/compress/archivers/zip/ZipFile.java");
assertEquals(26101, ze.getSize());
}
@Test
public void testUnshrinking() throws Exception {
zf = new ZipFile(getFile("SHRUNK.ZIP"));
ZipArchiveEntry test = zf.getEntry("TEST1.XML");
FileInputStream original = new FileInputStream(getFile("test1.xml"));
try {
assertArrayEquals(IOUtils.toByteArray(original),
IOUtils.toByteArray(zf.getInputStream(test)));
} finally {
original.close();
}
test = zf.getEntry("TEST2.XML");
original = new FileInputStream(getFile("test2.xml"));
try {
assertArrayEquals(IOUtils.toByteArray(original),
IOUtils.toByteArray(zf.getInputStream(test)));
} finally {
original.close();
}
}
/**
* Test case for
* <a href="https://issues.apache.org/jira/browse/COMPRESS-264"
* >COMPRESS-264</a>.
*/
@Test
public void testReadingOfFirstStoredEntry() throws Exception {
final File archive = getFile("COMPRESS-264.zip");
zf = new ZipFile(archive);
final ZipArchiveEntry ze = zf.getEntry("test.txt");
assertEquals(5, ze.getSize());
assertArrayEquals(new byte[] {'d', 'a', 't', 'a', '\n'},
IOUtils.toByteArray(zf.getInputStream(ze)));
}
@Test
public void testUnzipBZip2CompressedEntry() throws Exception {
final File archive = getFile("bzip2-zip.zip");
zf = new ZipFile(archive);
final ZipArchiveEntry ze = zf.getEntry("lots-of-as");
assertEquals(42, ze.getSize());
final byte[] expected = new byte[42];
Arrays.fill(expected , (byte)'a');
assertArrayEquals(expected, IOUtils.toByteArray(zf.getInputStream(ze)));
}
/*
* ordertest.zip has been handcrafted.
*
* It contains enough files so any random coincidence of
* entries.keySet() and central directory order would be unlikely
* - in fact testCDOrder fails in svn revision 920284.
*
* The central directory has ZipFile and ZipUtil swapped so
* central directory order is different from entry data order.
*/
private void readOrderTest() throws Exception {
final File archive = getFile("ordertest.zip");
zf = new ZipFile(archive);
}
private static void assertEntryName(final ArrayList<ZipArchiveEntry> entries,
final int index,
final String expectedName) {
final ZipArchiveEntry ze = entries.get(index);
assertEquals("src/main/java/org/apache/commons/compress/archivers/zip/"
+ expectedName + ".java",
ze.getName());
}
}