/*

 * Copyright (c) 2014, 2016, 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.

 */

import java.io.File;

import java.io.FileOutputStream;

import jdk.internal.org.objectweb.asm.ClassWriter;

import jdk.internal.org.objectweb.asm.MethodVisitor;

import static jdk.internal.org.objectweb.asm.Opcodes.*;

import jdk.test.lib.process.ProcessTools;

import jdk.test.lib.process.OutputAnalyzer;

/*

 * @test

 * @summary Test that anewarray bytecode is valid only if it specifies 255 or fewer dimensions.

 * @library /test/lib

 * @modules java.base/jdk.internal.org.objectweb.asm

 *          java.base/jdk.internal.misc

 *          java.management

 * @compile -XDignore.symbol.file TestANewArray.java

 * @run main/othervm TestANewArray 49

 * @run main/othervm TestANewArray 50

 * @run main/othervm TestANewArray 51

 * @run main/othervm TestANewArray 52

 */

/*

 * Testing anewarray instruction with 254, 255 & 264 dimensions to verify JVMS 8,

 * Section 4.9.1, Static Constraints that states the following:

 *

 * "No anewarray instruction may be used to create an array of more than 255 dimensions."

 *

 */

public class TestANewArray {

    static String classCName = null; // the generated class name

    static final int test_Dimension_254 = 254; // should always pass

    static final int test_Dimension_255 = 255; // should always pass, except for cfv 49

    static final int test_Dimension_264 = 264; // should always fail

    static final String array_Dimension_254 = genArrayDim(test_Dimension_254);

    static final String array_Dimension_255 = genArrayDim(test_Dimension_255);

    static final String array_Dimension_264 = genArrayDim(test_Dimension_264);

    public static void main(String... args) throws Exception {

        int cfv = Integer.parseInt(args[0]);

        // 254 array dimensions

        byte[] classFile_254 = dumpClassFile(cfv, test_Dimension_254, array_Dimension_254);

        writeClassFileFromByteArray(classFile_254);

        System.err.println("Running with cfv: " + cfv + ", test_Dimension_254");

        ProcessBuilder pb = ProcessTools.createJavaProcessBuilder(true, "-verify", "-cp", ".",  classCName);

        OutputAnalyzer output = new OutputAnalyzer(pb.start());

        output.shouldNotContain("java.lang.VerifyError");

        output.shouldHaveExitValue(0);

        // 255 array dimensions

        byte[] classFile_255 = dumpClassFile(cfv, test_Dimension_255, array_Dimension_255);

        writeClassFileFromByteArray(classFile_255);

        System.err.println("Running with cfv: " + cfv + ", test_Dimension_255");

        pb = ProcessTools.createJavaProcessBuilder(true, "-verify", "-cp", ".",  classCName);

        output = new OutputAnalyzer(pb.start());

        if (cfv == 49) {

            // The type-inferencing verifier used for <=49.0 ClassFiles detects an anewarray instruction

            // with exactly 255 dimensions and incorrectly issues the "Array with too many dimensions" VerifyError.

            output.shouldContain("Array with too many dimensions");

            output.shouldHaveExitValue(1);

        } else {

            // 255 dimensions should always pass, except for cfv 49

            output.shouldNotContain("java.lang.VerifyError");

            output.shouldNotContain("java.lang.ClassFormatError");

            output.shouldHaveExitValue(0);

        }

        // 264 array dimensions

        byte[] classFile_264 = dumpClassFile(cfv, test_Dimension_264, array_Dimension_264);

        writeClassFileFromByteArray(classFile_264);

        System.err.println("Running with cfv: " + cfv + ", test_Dimension_264");

        pb = ProcessTools.createJavaProcessBuilder(true, "-verify", "-cp", ".",  classCName);

        output = new OutputAnalyzer(pb.start());

        output.shouldContain("java.lang.ClassFormatError");

        output.shouldHaveExitValue(1);

    }

    public static byte[] dumpClassFile(int cfv, int testDimension264, String arrayDim) throws Exception {

        ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES);

        MethodVisitor mv;

        classCName = "classCName_" + cfv + "_" + testDimension264;

        cw.visit(cfv, ACC_PUBLIC + ACC_SUPER, classCName, null, "java/lang/Object", null);

        {

            mv = cw.visitMethod(ACC_PUBLIC, "<init>", "()V", null, null);

            mv.visitCode();

            mv.visitVarInsn(ALOAD, 0);

            mv.visitMethodInsn(INVOKESPECIAL, "java/lang/Object", "<init>", "()V", false);

            mv.visitInsn(RETURN);

            mv.visitMaxs(1, 1);

            mv.visitEnd();

        }

        {   // classCName main method

            mv = cw.visitMethod(ACC_PUBLIC + ACC_STATIC, "main", "([Ljava/lang/String;)V", null, null);

            mv.visitCode();

            mv.visitIntInsn(BIPUSH, 1);

            mv.visitTypeInsn(ANEWARRAY, arrayDim); // Test ANEWARRAY bytecode with various dimensions

            mv.visitInsn(RETURN);

            mv.visitMaxs(2, 2);

            mv.visitEnd();

        }

        cw.visitEnd();

        return cw.toByteArray();

    }

    public static FileOutputStream writeClassFileFromByteArray(byte[] classFileByteArray) throws Exception {

        FileOutputStream fos = new FileOutputStream(new File(classCName + ".class"));

        fos.write(classFileByteArray);

        fos.close();

        return fos;

    }

    private static String genArrayDim(int testDim) {

        StringBuilder array_Dimension = new StringBuilder();

        for (int i = 0; i < testDim; i++)

        {

            array_Dimension.append("[");

        }

        return array_Dimension.append("Ljava/lang/Object;").toString();

    }

}