/*

 * Copyright (c) 2013 Google 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

/**

 * @test

 * @summary Test java.util.zip behavior with ~64k entries

 * @run main/othervm EntryCount64k

 * @run main/othervm -Djdk.util.zip.inhibitZip64=true EntryCount64k

 * @run main/othervm -Djdk.util.zip.inhibitZip64=false EntryCount64k

 */

import java.io.*;

import java.util.*;

import java.util.zip.*;

public class EntryCount64k {

    public static void main(String[] args) throws Exception {

        for (int i = (1 << 16) - 2; i < (1 << 16) + 2; i++)

            test(i);

    }

    static void test(int entryCount) throws Exception {

        File zipFile = new File("EntryCount64k-tmp.zip");

        zipFile.delete();

        try (ZipOutputStream zos =

             new ZipOutputStream(

                new BufferedOutputStream(

                    new FileOutputStream(zipFile)))) {

            for (int i = 0; i < entryCount; i++) {

                ZipEntry e = new ZipEntry(Integer.toString(i));

                zos.putNextEntry(e);

                zos.closeEntry();

            }

        }

        String p = System.getProperty("jdk.util.zip.inhibitZip64");

        boolean tooManyEntries = entryCount >= (1 << 16) - 1;

        boolean shouldUseZip64 = tooManyEntries & !("true".equals(p));

        boolean usesZip64 = usesZip64(zipFile);

        String details = String.format

            ("entryCount=%d shouldUseZip64=%s usesZip64=%s zipSize=%d%n",

             entryCount, shouldUseZip64, usesZip64, zipFile.length());

        System.err.println(details);

        checkCanRead(zipFile, entryCount);

        if (shouldUseZip64 != usesZip64)

            throw new Error(details);

        zipFile.delete();

    }

    static boolean usesZip64(File zipFile) throws Exception {

        RandomAccessFile raf = new RandomAccessFile(zipFile, "r");

        byte[] buf = new byte[4096];

        raf.seek(raf.length() - buf.length);

        raf.read(buf);

        for (int i = 0; i < buf.length - 4; i++) {

            // Look for ZIP64 End Header Signature

            // Phil Katz: yes, we will always remember you

            if (buf[i+0] == 'P' &&

                buf[i+1] == 'K' &&

                buf[i+2] == 6   &&

                buf[i+3] == 6)

                return true;

        }

        return false;

    }

    static void checkCanRead(File zipFile, int entryCount) throws Exception {

        // Check ZipInputStream API

        try (ZipInputStream zis =

             new ZipInputStream(

                 new BufferedInputStream(

                     new FileInputStream(zipFile)))) {

            for (int i = 0; i < entryCount; i++) {

                ZipEntry e = zis.getNextEntry();

                if (Integer.parseInt(e.getName()) != i)

                    throw new AssertionError();

            }

            if (zis.getNextEntry() != null)

                throw new AssertionError();

        }

        // Check ZipFile API

        try (ZipFile zf = new ZipFile(zipFile)) {

            Enumeration<? extends ZipEntry> en = zf.entries();

            for (int i = 0; i < entryCount; i++) {

                ZipEntry e = en.nextElement();

                if (Integer.parseInt(e.getName()) != i)

                    throw new AssertionError();

            }

            if (en.hasMoreElements()

                || (zf.size() != entryCount)

                || (zf.getEntry(Integer.toString(entryCount - 1)) == null)

                || (zf.getEntry(Integer.toString(entryCount)) != null))

                throw new AssertionError();

        }

    }

}