/*
 * Copyright (c) 2005, 2019, 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 6348045 6341887 8231770
 * @summary GZipOutputStream/InputStream goes critical(calls JNI_Get*Critical)
 * and causes slowness. This test uses Deflater and Inflater directly.
 * @key randomness
 * @run main/othervm -Xcheck:jni FlaterTest
 */

import java.nio.*;
import java.util.*;
import java.util.zip.*;

/**
 * This test runs Inflater and Defalter in a number of simultaneous threads,
 * validating that the deflated & then inflated data matches the original
 * data.
 */
public class FlaterTest extends Thread {
    private static final int DATA_LEN = 1024 * 128;

    private static ByteBuffer dataDirect;
    private static ByteBuffer dataHeap;

    // If true, print extra info.
    private static final boolean debug = false;

    // Set of Flater threads running.
    private static Set<Flater> flaters =
        Collections.synchronizedSet(new HashSet<>());

    /** Fill in {@code data} with random values. */
    static void createData() {
        ByteBuffer bb = ByteBuffer.allocateDirect(DATA_LEN * 8);
        for (int i = 0; i < DATA_LEN * 8; i += 8) {
            bb.putDouble(i, Math.random());
        }
        dataDirect = bb;
        final ByteBuffer hb = ByteBuffer.allocate(bb.capacity());
        hb.duplicate().put(bb.duplicate());
        dataHeap = hb;
        if (debug) System.out.println("data length is " + bb.capacity());
    }

    /** @return the length of the deflated {@code data}. */
    private static int getDeflatedLength() {
        Deflater deflater = new Deflater();
        deflater.setInput(dataDirect.duplicate());
        deflater.finish();
        byte[] out = new byte[dataDirect.capacity()];
        int rc = deflater.deflate(out);
        deflater.end();
        if (debug) System.out.println("deflatedLength is " + rc);
        return rc;
    }

    /** Compares given bytes with those in {@code data}.
     * @throws Exception if given bytes don't match {@code data}.
     */
    private static void validate(ByteBuffer buf, int offset, int len) throws Exception {
        for (int i = 0; i < len; i++ ) {
            if (buf.get(i) != dataDirect.get(offset+i)) {
                throw new Exception("mismatch at " + (offset + i));
            }
        }
    }

    public static void realMain(String[] args) {
        int numThreads = args.length > 0 ? Integer.parseInt(args[0]) : 5;
        createData();
        for (int srcMode = 0; srcMode <= 2; srcMode ++) {
            for (int dstMode = 0; dstMode <= 2; dstMode ++) {
                new FlaterTest().go(numThreads, srcMode, dstMode);
            }
        }
    }

    private synchronized void go(int numThreads, int srcMode, int dstMode) {
        int deflatedLength = getDeflatedLength();

        long time = System.currentTimeMillis();
        for (int i = 0; i < numThreads; i++) {
            Flater f = new Flater(deflatedLength, srcMode, dstMode);
            flaters.add(f);
            f.start();
        }
        synchronized (flaters) {
            while (flaters.size() != 0) {
                try {
                    flaters.wait();
                } catch (InterruptedException ex) {
                    unexpected(ex);
                }
            }
        }
        time = System.currentTimeMillis() - time;
        System.out.println("Time needed for " + numThreads
                           + " threads to deflate/inflate: " + time + " ms (srcMode="+srcMode+",dstMode="+dstMode+")");
    }

    /** Deflates and inflates data. */
    static class Flater extends Thread {
        private final int deflatedLength;
        private final int srcMode, dstMode;

        private Flater(int length, int srcMode, int dstMode) {
            this.deflatedLength = length;
            this.srcMode = srcMode;
            this.dstMode = dstMode;
        }

        /** Deflates and inflates {@code data}. */
        public void run() {
            if (debug) System.out.println(getName() + " starting run()");
            try {
                ByteBuffer deflated = DeflateData(deflatedLength);
                InflateData(deflated);
            } catch (Throwable t) {
                t.printStackTrace();
                fail(getName() + " failed");
            } finally {
                synchronized (flaters) {
                    flaters.remove(this);
                    if (flaters.isEmpty()) {
                        flaters.notifyAll();
                    }
                }
            }
        }

        /** Returns a copy of {@code data} in deflated form. */
        private ByteBuffer DeflateData(int length) {
            Deflater deflater = new Deflater();
            if (srcMode == 0) {
                deflater.setInput(dataHeap.array());
            } else if (srcMode == 1) {
                deflater.setInput(dataHeap.duplicate());
            } else {
                assert srcMode == 2;
                deflater.setInput(dataDirect.duplicate());
            }
            deflater.finish();
            ByteBuffer out = dstMode == 2 ? ByteBuffer.allocateDirect(length) : ByteBuffer.allocate(length);
            int deflated;
            if (dstMode == 0) {
                deflated = deflater.deflate(out.array(), 0, length);
                out.position(deflated);
            } else {
                deflater.deflate(out);
            }
            out.flip();
            return out;
        }

        /** Inflate a byte array, comparing it with {@code data} during
         * inflation.
         * @throws Exception if inflated bytes don't match {@code data}.
         */
        private void InflateData(ByteBuffer bytes) throws Throwable {
            Inflater inflater = new Inflater();
            if (dstMode == 0) {
                inflater.setInput(bytes.array(), 0, bytes.remaining());
            } else {
                inflater.setInput(bytes);
            }
            if (inflater.getRemaining() == 0) {
                throw new Exception("Nothing to inflate (bytes=" + bytes + ")");
            }
            int len = 1024 * 8;
            int offset = 0;
            ByteBuffer buf = srcMode == 2 ? ByteBuffer.allocateDirect(len) : ByteBuffer.allocate(len);
            while (inflater.getRemaining() > 0) {
                buf.clear();
                int inflated;
                if (srcMode == 0) {
                    inflated = inflater.inflate(buf.array(), 0, buf.remaining());
                } else {
                    inflated = inflater.inflate(buf);
                }
                if (inflated == 0) {
                    throw new Exception("Nothing inflated (dst=" + buf + ",offset=" + offset + ",rem=" + inflater.getRemaining() + ",srcMode="+srcMode+",dstMode="+dstMode+")");
                }
                validate(buf, offset, inflated);
                offset += inflated;
            }
        }
    }

    //--------------------- Infrastructure ---------------------------
    static volatile int passed = 0, failed = 0;
    static void pass() {passed++;}
    static void fail() {failed++; Thread.dumpStack();}
    static void fail(String msg) {System.out.println(msg); fail();}
    static void unexpected(Throwable t) {failed++; t.printStackTrace();}
    static void check(boolean cond) {if (cond) pass(); else fail();}
    static void equal(Object x, Object y) {
        if (x == null ? y == null : x.equals(y)) pass();
        else fail(x + " not equal to " + y);}
    public static void main(String[] args) throws Throwable {
        try {realMain(args);} catch (Throwable t) {unexpected(t);}
        System.out.println("\nPassed = " + passed + " failed = " + failed);
        if (failed > 0) throw new AssertionError("Some tests failed");}
}