/*

 * Copyright (c) 2011, 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 java.util.Random;

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

import org.graalvm.compiler.phases.common.AbstractInliningPhase;

import org.graalvm.compiler.test.ExportingClassLoader;

import org.junit.Assert;

import org.junit.Test;

import org.objectweb.asm.ClassWriter;

import org.objectweb.asm.Label;

import org.objectweb.asm.MethodVisitor;

import org.objectweb.asm.Opcodes;

import jdk.vm.ci.meta.ResolvedJavaMethod;

/**

 * Tests that deoptimization upon exception handling works.

 */

public class DeoptimizeOnExceptionTest extends GraalCompilerTest {

    public DeoptimizeOnExceptionTest() {

        createSuites(getInitialOptions()).getHighTier().findPhase(AbstractInliningPhase.class).remove();

    }

    private static void raiseException(String m1, String m2, String m3, String m4, String m5) {

        throw new RuntimeException(m1 + m2 + m3 + m4 + m5);

    }

    @Test

    public void test1() {

        test("test1Snippet", "m1", "m2", "m3", "m4", "m5");

    }

    // no local exception handler - will deopt

    public static String test1Snippet(String m1, String m2, String m3, String m4, String m5) {

        if (m1 != null) {

            raiseException(m1, m2, m3, m4, m5);

        }

        return m1 + m2 + m3 + m4 + m5;

    }

    @Test

    public void test2() {

        test("test2Snippet");

    }

    public String test2Snippet() throws Exception {

        try {

            ClassLoader testCl = new MyClassLoader();

            @SuppressWarnings("unchecked")

            Class<Runnable> c = (Class<Runnable>) testCl.loadClass(name);

            Runnable r = c.newInstance();

            ct = Long.MAX_VALUE;

            // warmup

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

                r.run();

            }

            // compile

            ResolvedJavaMethod method = getResolvedJavaMethod(c, "run");

            getCode(method);

            ct = 0;

            r.run();

        } catch (Throwable e) {

            e.printStackTrace(System.out);

            Assert.fail();

        }

        return "SUCCESS";

    }

    public static class MyClassLoader extends ExportingClassLoader {

        @Override

        protected Class<?> findClass(String className) throws ClassNotFoundException {

            return defineClass(name.replace('/', '.'), clazz, 0, clazz.length);

        }

    }

    public static void methodB() {

        Random r = new Random(System.currentTimeMillis());

        while (r.nextFloat() > .03f) {

            // Empty

        }

        return;

    }

    public static void methodA() {

        Random r = new Random(System.currentTimeMillis());

        while (r.nextDouble() > .05) {

            // Empty

        }

        return;

    }

    private static Object m = new Object();

    static long ct = Long.MAX_VALUE;

    public static Object getM() {

        if (ct-- > 0) {

            return m;

        } else {

            return null;

        }

    }

    private static String name = "t/TestJSR";

    private static final byte[] clazz = makeClazz();

    private static byte[] makeClazz() {

        // Code generated the class below using asm.

        String clazzName = DeoptimizeOnExceptionTest.class.getName().replace('.', '/');

        final ClassWriter w = new ClassWriter(0);

        w.visit(Opcodes.V1_5, Opcodes.ACC_PUBLIC,

                        "t/TestJSR", null, "java/lang/Object",

                        new String[]{"java/lang/Runnable"});

        MethodVisitor mv = w.visitMethod(Opcodes.ACC_PUBLIC, "<init>", "()V", null, new String[]{});

        mv.visitCode();

        mv.visitVarInsn(Opcodes.ALOAD, 0);

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

        mv.visitInsn(Opcodes.RETURN);

        mv.visitMaxs(10, 10);

        mv.visitEnd();

        mv = w.visitMethod(Opcodes.ACC_PUBLIC, "run", "()V", null, null);

        mv.visitCode();

        mv.visitMethodInsn(Opcodes.INVOKESTATIC, clazzName, "getM", "()Ljava/lang/Object;", false);

        Label l1 = new Label();

        mv.visitJumpInsn(Opcodes.JSR, l1);

        mv.visitInsn(Opcodes.RETURN);

        mv.visitLabel(l1);

        mv.visitVarInsn(Opcodes.ASTORE, 1);

        Label lElse = new Label();

        Label lEnd = new Label();

        mv.visitMethodInsn(Opcodes.INVOKESTATIC, "java/lang/System", "currentTimeMillis", "()J", false);

        mv.visitInsn(Opcodes.POP2);

        mv.visitMethodInsn(Opcodes.INVOKESTATIC, clazzName, "getM", "()Ljava/lang/Object;", false);

        mv.visitInsn(Opcodes.DUP);

        mv.visitJumpInsn(Opcodes.IFNULL, lElse);

        mv.visitMethodInsn(Opcodes.INVOKESTATIC, clazzName, "methodA", "()V", false);

        mv.visitJumpInsn(Opcodes.GOTO, lEnd);

        mv.visitLabel(lElse);

        mv.visitMethodInsn(Opcodes.INVOKESTATIC, clazzName, "methodB", "()V", false);

        mv.visitLabel(lEnd);

        mv.visitVarInsn(Opcodes.RET, 1);

        mv.visitMaxs(10, 10);

        mv.visitEnd();

        return w.toByteArray();

    }

}