test/java/util/zip/ZipFile/ManyEntries.java

/*
 * Copyright (c) 2005, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code 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 General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

/* @test
 * @bug 4770745 4828461
 * @summary open zip files with more than 64k entries
 * @run main/timeout=600 ManyEntries
 * @author Martin Buchholz
 */

import java.util.*;
import java.util.zip.*;
import java.io.*;

public class ManyEntries {
    public static void main(String[] args) throws Exception {
        boolean runMoreTests = args.length > 0;
        int[] sizes = (runMoreTests ?
                       new int[]{1, 2, 42, 65534, 65535, 65536, 65537, 68000} :
                       new int[]{1, 2, 42, 68000});
        int[] methods = {ZipEntry.STORED, ZipEntry.DEFLATED};

        for (int size : sizes)
            for (int method : methods)
                test(size, method);
    }

    // JDK doesn't like having zip file contents changed at runtime
    static int uniquifier = 42;

    static void test(int N, int method) throws Exception {
        System.out.println("N="+N+", method="+method);
        long time = System.currentTimeMillis() / 2000 * 2000;
        CRC32 crc32 = new CRC32();
        File zipFile = new File(++uniquifier + ".zip");
        try {
            zipFile.delete();
            ZipOutputStream zos =
                new ZipOutputStream(
                    new BufferedOutputStream(
                        new FileOutputStream(zipFile)));
            for (int i = 0; i < N; i++) {
                ZipEntry e = new ZipEntry("DIR/"+i);
                e.setMethod(method);
                e.setTime(time);
                if (method == ZipEntry.STORED) {
                    e.setSize(1);
                    crc32.reset();
                    crc32.update((byte)i);
                    e.setCrc(crc32.getValue());
                } else {
                    e.setSize(0);
                    e.setCrc(0);
                }
                zos.putNextEntry(e);
                zos.write(i);
            }
            zos.close();
            zos = null;

            ZipFile zip = new ZipFile(zipFile);
            if (! (zip.size() == N))
                throw new Exception("Bad ZipFile size: " + zip.size());
            Enumeration entries = zip.entries();

            for (int i = 0; i < N; i++) {
                if (i % 1000 == 0) {System.gc(); System.runFinalization();}
                if (! (entries.hasMoreElements()))
                    throw new Exception("only " + i + " elements");
                ZipEntry e = (ZipEntry)entries.nextElement();
                if (! (e.getSize() == 1))
                    throw new Exception("bad size: " + e.getSize());
                if (! (e.getName().equals("DIR/" + i)))
                    throw new Exception("bad name: " + i);
                if (! (e.getMethod() == method))
                    throw new Exception("getMethod="+e.getMethod()+", method=" + method);
                if (! (e.getTime() == time))
                    throw new Exception("getTime="+e.getTime()+", time=" + time);
                if (! (zip.getInputStream(e).read() == (i & 0xff)))
                    throw new Exception("Bad file contents: " + i);
            }
            if (entries.hasMoreElements())
                throw new Exception("too many elements");
        }
        finally {
            zipFile.delete();
        }
    }
}