/*

 * Copyright 2005 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

 * CA 95054 USA or visit www.sun.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();

        }

    }

}