/*

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

 */

package org.graalvm.compiler.replacements.test;

import static org.graalvm.compiler.test.SubprocessUtil.getVMCommandLine;

import static org.graalvm.compiler.test.SubprocessUtil.withoutDebuggerArguments;

import java.nio.file.Files;

import java.nio.file.Path;

import java.nio.file.Paths;

import java.util.List;

import org.graalvm.compiler.core.test.CustomizedBytecodePatternTest;

import org.graalvm.compiler.serviceprovider.JavaVersionUtil;

import org.graalvm.compiler.test.SubprocessUtil;

import org.graalvm.compiler.test.SubprocessUtil.Subprocess;

import org.junit.Test;

import org.objectweb.asm.ClassWriter;

import org.objectweb.asm.MethodVisitor;

import org.objectweb.asm.Type;

public class InvokerSignatureMismatchTest extends CustomizedBytecodePatternTest {

    @SuppressWarnings("try")

    @Test

    public void test() throws Throwable {

        List<String> args = withoutDebuggerArguments(getVMCommandLine());

        try (TemporaryDirectory temp = new TemporaryDirectory(null, getClass().getSimpleName())) {

            if (JavaVersionUtil.JAVA_SPEC > 8) {

                args.add("--class-path=" + temp);

                args.add("--patch-module=java.base=" + temp);

            } else {

                args.add("-Xbootclasspath/a:" + temp);

            }

            args.add("-XX:-TieredCompilation");

            args.add("-XX:+UnlockExperimentalVMOptions");

            args.add("-XX:+EnableJVMCI");

            args.add("-XX:+UseJVMCICompiler");

            Path invokeDir = Files.createDirectories(temp.path.resolve(Paths.get("java", "lang", "invoke")));

            Files.write(temp.path.resolve("ISMTest.class"), generateClass("ISMTest"));

            Files.write(invokeDir.resolve("MethodHandleHelper.class"), generateClass("java/lang/invoke/MethodHandleHelper"));

            args.add("ISMTest");

            Subprocess proc = SubprocessUtil.java(args);

            if (proc.exitCode != 0) {

                throw new AssertionError(proc.toString());

            }

        }

    }

    @Override

    protected byte[] generateClass(String className) {

        String[] exceptions = new String[]{"java/lang/Throwable"};

        ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES);

        cw.visit(52, ACC_SUPER | ACC_PUBLIC, className, null, "java/lang/Object", null);

        if (className.equals("java/lang/invoke/MethodHandleHelper")) {

            MethodVisitor internalMemberName = cw.visitMethod(ACC_PUBLIC | ACC_STATIC, "internalMemberName", "(Ljava/lang/invoke/MethodHandle;)Ljava/lang/Object;", null, exceptions);

            internalMemberName.visitCode();

            internalMemberName.visitVarInsn(ALOAD, 0);

            internalMemberName.visitMethodInsn(INVOKEVIRTUAL, "java/lang/invoke/MethodHandle", "internalMemberName", "()Ljava/lang/invoke/MemberName;", false);

            internalMemberName.visitInsn(ARETURN);

            internalMemberName.visitMaxs(1, 1);

            internalMemberName.visitEnd();

            MethodVisitor linkToStatic = cw.visitMethod(ACC_PUBLIC | ACC_STATIC, "linkToStatic", "(FLjava/lang/Object;)I", null, exceptions);

            linkToStatic.visitCode();

            linkToStatic.visitVarInsn(FLOAD, 0);

            linkToStatic.visitVarInsn(ALOAD, 1);

            linkToStatic.visitMethodInsn(INVOKESTATIC, "java/lang/invoke/MethodHandle", "linkToStatic", "(FLjava/lang/Object;)I", false);

            linkToStatic.visitInsn(IRETURN);

            linkToStatic.visitMaxs(1, 1);

            linkToStatic.visitEnd();

            MethodVisitor invokeBasicI = cw.visitMethod(ACC_PUBLIC | ACC_STATIC, "invokeBasicI", "(Ljava/lang/invoke/MethodHandle;F)I", null, exceptions);

            invokeBasicI.visitCode();

            invokeBasicI.visitVarInsn(ALOAD, 0);

            invokeBasicI.visitVarInsn(FLOAD, 1);

            invokeBasicI.visitMethodInsn(INVOKEVIRTUAL, "java/lang/invoke/MethodHandle", "invokeBasic", "(F)I", false);

            invokeBasicI.visitInsn(IRETURN);

            invokeBasicI.visitMaxs(1, 1);

            invokeBasicI.visitEnd();

        } else {

            assert className.equals("ISMTest") : className;

            cw.visitField(ACC_FINAL | ACC_STATIC, "INT_MH", "Ljava/lang/invoke/MethodHandle;", null, null).visitAnnotation("Ljava/lang/invoke/Stable.class;", true).visitEnd();

            MethodVisitor clinit = cw.visitMethod(ACC_STATIC, "<clinit>", "()V", null, exceptions);

            clinit.visitCode();

            clinit.visitInsn(ACONST_NULL);

            clinit.visitVarInsn(ASTORE, 0);

            clinit.visitMethodInsn(INVOKESTATIC, "java/lang/invoke/MethodHandles", "lookup", "()Ljava/lang/invoke/MethodHandles$Lookup;", false);

            clinit.visitLdcInsn(Type.getObjectType(className));

            clinit.visitLdcInsn("bodyI");

            clinit.visitFieldInsn(GETSTATIC, "java/lang/Integer", "TYPE", "Ljava/lang/Class;");

            clinit.visitFieldInsn(GETSTATIC, "java/lang/Integer", "TYPE", "Ljava/lang/Class;");

            clinit.visitMethodInsn(INVOKESTATIC, "java/lang/invoke/MethodType", "methodType", "(Ljava/lang/Class;Ljava/lang/Class;)Ljava/lang/invoke/MethodType;", false);

            clinit.visitMethodInsn(INVOKEVIRTUAL, "java/lang/invoke/MethodHandles$Lookup", "findStatic",

                            "(Ljava/lang/Class;Ljava/lang/String;Ljava/lang/invoke/MethodType;)Ljava/lang/invoke/MethodHandle;", false);

            clinit.visitFieldInsn(PUTSTATIC, className, "INT_MH", "Ljava/lang/invoke/MethodHandle;");

            clinit.visitInsn(RETURN);

            clinit.visitMaxs(1, 1);

            clinit.visitEnd();

            MethodVisitor mainLink = cw.visitMethod(ACC_PUBLIC | ACC_STATIC, "mainLink", "(I)I", null, exceptions);

            mainLink.visitCode();

            mainLink.visitFieldInsn(GETSTATIC, className, "INT_MH", "Ljava/lang/invoke/MethodHandle;");

            mainLink.visitMethodInsn(INVOKESTATIC, "java/lang/invoke/MethodHandleHelper", "internalMemberName", "(Ljava/lang/invoke/MethodHandle;)Ljava/lang/Object;", false);

            mainLink.visitVarInsn(ASTORE, 1);

            mainLink.visitVarInsn(ILOAD, 0);

            mainLink.visitInsn(I2F);

            mainLink.visitVarInsn(ALOAD, 1);

            mainLink.visitMethodInsn(INVOKESTATIC, "java/lang/invoke/MethodHandleHelper", "linkToStatic", "(FLjava/lang/Object;)I", false);

            mainLink.visitInsn(IRETURN);

            mainLink.visitMaxs(1, 1);

            mainLink.visitEnd();

            MethodVisitor mainInvoke = cw.visitMethod(ACC_PUBLIC | ACC_STATIC, "mainInvoke", "(I)I", null, exceptions);

            mainInvoke.visitCode();

            mainInvoke.visitFieldInsn(GETSTATIC, className, "INT_MH", "Ljava/lang/invoke/MethodHandle;");

            mainInvoke.visitVarInsn(ILOAD, 0);

            mainInvoke.visitInsn(I2F);

            mainInvoke.visitMethodInsn(INVOKESTATIC, "java/lang/invoke/MethodHandleHelper", "invokeBasicI", "(Ljava/lang/invoke/MethodHandle;F)I", false);

            mainInvoke.visitInsn(IRETURN);

            mainInvoke.visitMaxs(1, 1);

            mainInvoke.visitEnd();

            MethodVisitor bodyI = cw.visitMethod(ACC_PUBLIC | ACC_STATIC, "bodyI", "(I)I", null, null);

            bodyI.visitCode();

            bodyI.visitVarInsn(ILOAD, 0);

            bodyI.visitIntInsn(SIPUSH, 1023);

            bodyI.visitInsn(IAND);

            bodyI.visitInsn(IRETURN);

            bodyI.visitMaxs(1, 1);

            bodyI.visitEnd();

            MethodVisitor main = cw.visitMethod(ACC_PUBLIC | ACC_STATIC, "main", "([Ljava/lang/String;)V", null, exceptions);

            main.visitCode();

            main.visitIntInsn(SIPUSH, 100);

            main.visitMethodInsn(INVOKESTATIC, "ISMTest", "mainLink", "(I)I", false);

            main.visitInsn(POP);

            main.visitIntInsn(SIPUSH, 100);

            main.visitMethodInsn(INVOKESTATIC, "ISMTest", "mainInvoke", "(I)I", false);

            main.visitInsn(POP);

            main.visitInsn(RETURN);

            main.visitMaxs(1, 1);

            main.visitEnd();

        }

        cw.visitEnd();

        return cw.toByteArray();

    }

}