test/jdk/java/lang/invoke/CallSiteTest.java - platform/libcore - Git at Google

 /*
  * Copyright (c) 2011, 2012, 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
  * @summary smoke tests for CallSite
  * @library /test/lib
  *
  * @build indify.Indify
  * @compile CallSiteTest.java
  * @run main/othervm/timeout=3600 -XX:+IgnoreUnrecognizedVMOptions -XX:-VerifyDependencies -Xbatch
  *      indify.Indify
  *      --expand-properties --classpath ${test.classes}
  *      --java test.java.lang.invoke.CallSiteTest
  */

 package test.java.lang.invoke;

 import java.io.*;

 import java.lang.invoke.*;
 import static java.lang.invoke.MethodHandles.*;
 import static java.lang.invoke.MethodType.*;
 import static jdk.test.lib.Asserts.*;

 public class CallSiteTest {
     private static final Class<?> CLASS = CallSiteTest.class;

     private static CallSite mcs;
     private static CallSite vcs;
     private static MethodHandle mh_foo;
     private static MethodHandle mh_bar;

     static {
         try {

             mh_foo = lookup().findStatic(CLASS, "foo", methodType(int.class, int.class, int.class));
             mh_bar = lookup().findStatic(CLASS, "bar", methodType(int.class, int.class, int.class));
             mcs = new MutableCallSite(mh_foo);
             vcs = new VolatileCallSite(mh_foo);
         } catch (Exception e) {
             e.printStackTrace();
             throw new Error(e);
         }
     }

     public static void main(String... av) throws Throwable {
         testConstantCallSite();
         testMutableCallSite();
         testVolatileCallSite();
     }

     private static final int N = Integer.MAX_VALUE / 100;
     private static final int RESULT1 = 762786192;
     private static final int RESULT2 = -21474836;

     static final CallSite     MCS         = new MutableCallSite(methodType(void.class));
     static final MethodHandle MCS_INVOKER = MCS.dynamicInvoker();

     static void test(boolean shouldThrow) {
         try {
             MCS_INVOKER.invokeExact();
             if (shouldThrow) {
                 throw new AssertionError("should throw");
             }
         } catch (IllegalStateException ise) {
             if (!shouldThrow) {
                 throw new AssertionError("should not throw", ise);
             }
         } catch (Throwable e) {
             throw new Error(e);
         }
     }

     static class MyCCS extends ConstantCallSite {
         public MyCCS(MethodType targetType, MethodHandle createTargetHook) throws Throwable {
             super(targetType, createTargetHook);
         }
     }

     private static MethodHandle testConstantCallSiteHandler(CallSite cs, CallSite[] holder) throws Throwable {
         holder[0] = cs; // capture call site instance for subsequent checks

         MethodType csType = cs.type(); // should not throw on partially constructed instance

         // Truly dynamic invoker for constant call site
         MethodHandle getTarget = lookup().findVirtual(CallSite.class, "getTarget", MethodType.methodType(MethodHandle.class))
                                          .bindTo(cs);
         MethodHandle invoker = MethodHandles.exactInvoker(csType);
         MethodHandle target = MethodHandles.foldArguments(invoker, getTarget);

         MCS.setTarget(target);
         // warmup
         for (int i = 0; i < 20_000; i++) {
             test(true); // should throw IllegalStateException
         }

         return MethodHandles.empty(csType); // initialize cs with an empty method handle
     }

     private static void testConstantCallSite() throws Throwable {
         CallSite[] holder = new CallSite[1];
         MethodHandle handler = lookup().findStatic(CLASS, "testConstantCallSiteHandler", MethodType.methodType(MethodHandle.class, CallSite.class, CallSite[].class));
         handler = MethodHandles.insertArguments(handler, 1, new Object[] { holder } );

         CallSite ccs = new MyCCS(MCS.type(), handler); // trigger call to handler

         if (ccs != holder[0]) {
             throw new AssertionError("different call site instances");
         }
         for (int i = 0; i < 20_000; i++) {
             test(false); // should not throw
         }
     }

     private static void testMutableCallSite() throws Throwable {
         // warm-up
         for (int i = 0; i < 20000; i++) {
             mcs.setTarget(mh_foo);
         }
         // run
         for (int n = 0; n < 2; n++) {
             mcs.setTarget(mh_foo);
             for (int i = 0; i < 5; i++) {
                 assertEQ(RESULT1, runMutableCallSite());
             }
             mcs.setTarget(mh_bar);
             for (int i = 0; i < 5; i++) {
                 assertEQ(RESULT2, runMutableCallSite());
             }
         }
     }
     private static void testVolatileCallSite() throws Throwable {
         // warm-up
         for (int i = 0; i < 20000; i++) {
             vcs.setTarget(mh_foo);
         }
         // run
         for (int n = 0; n < 2; n++) {
             vcs.setTarget(mh_foo);
             for (int i = 0; i < 5; i++) {
                 assertEQ(RESULT1, runVolatileCallSite());
             }
             vcs.setTarget(mh_bar);
             for (int i = 0; i < 5; i++) {
                 assertEQ(RESULT2, runVolatileCallSite());
             }
         }
     }

     private static int runMutableCallSite() throws Throwable {
         int sum = 0;
         for (int i = 0; i < N; i++) {
             sum += (int) INDY_mcs().invokeExact(i, i+1);
         }
         return sum;
     }
     private static int runVolatileCallSite() throws Throwable {
         int sum = 0;
         for (int i = 0; i < N; i++) {
             sum += (int) INDY_vcs().invokeExact(i, i+1);
         }
         return sum;
     }

     static int foo(int a, int b) { return a + b; }
     static int bar(int a, int b) { return a - b; }

     private static MethodType MT_bsm() {
         shouldNotCallThis();
         return methodType(CallSite.class, Lookup.class, String.class, MethodType.class);
     }

     private static CallSite bsm_mcs(Lookup caller, String name, MethodType type) throws ReflectiveOperationException {
         return mcs;
     }
     private static MethodHandle MH_bsm_mcs() throws ReflectiveOperationException {
         shouldNotCallThis();
         return lookup().findStatic(lookup().lookupClass(), "bsm_mcs", MT_bsm());
     }
     private static MethodHandle INDY_mcs() throws Throwable {
         shouldNotCallThis();
         return ((CallSite) MH_bsm_mcs().invoke(lookup(), "foo", methodType(int.class, int.class, int.class))).dynamicInvoker();
     }

     private static CallSite bsm_vcs(Lookup caller, String name, MethodType type) throws ReflectiveOperationException {
         return vcs;
     }
     private static MethodHandle MH_bsm_vcs() throws ReflectiveOperationException {
         shouldNotCallThis();
         return lookup().findStatic(lookup().lookupClass(), "bsm_vcs", MT_bsm());
     }
     private static MethodHandle INDY_vcs() throws Throwable {
         shouldNotCallThis();
         return ((CallSite) MH_bsm_vcs().invoke(lookup(), "foo", methodType(int.class, int.class, int.class))).dynamicInvoker();
     }

     private static void shouldNotCallThis() {
         // if this gets called, the transformation has not taken place
         throw new AssertionError("this code should be statically transformed away by Indify");
     }
 }
	/*
	* Copyright (c) 2011, 2012, 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
	* @summary smoke tests for CallSite
	* @library /test/lib
	*
	* @build indify.Indify
	* @compile CallSiteTest.java
	* @run main/othervm/timeout=3600 -XX:+IgnoreUnrecognizedVMOptions -XX:-VerifyDependencies -Xbatch
	* indify.Indify
	* --expand-properties --classpath ${test.classes}
	* --java test.java.lang.invoke.CallSiteTest
	*/

	package test.java.lang.invoke;

	import java.io.*;

	import java.lang.invoke.*;
	import static java.lang.invoke.MethodHandles.*;
	import static java.lang.invoke.MethodType.*;
	import static jdk.test.lib.Asserts.*;

	public class CallSiteTest {
	private static final Class<?> CLASS = CallSiteTest.class;

	private static CallSite mcs;
	private static CallSite vcs;
	private static MethodHandle mh_foo;
	private static MethodHandle mh_bar;

	static {
	try {

	mh_foo = lookup().findStatic(CLASS, "foo", methodType(int.class, int.class, int.class));
	mh_bar = lookup().findStatic(CLASS, "bar", methodType(int.class, int.class, int.class));
	mcs = new MutableCallSite(mh_foo);
	vcs = new VolatileCallSite(mh_foo);
	} catch (Exception e) {
	e.printStackTrace();
	throw new Error(e);
	}
	}

	public static void main(String... av) throws Throwable {
	testConstantCallSite();
	testMutableCallSite();
	testVolatileCallSite();
	}

	private static final int N = Integer.MAX_VALUE / 100;
	private static final int RESULT1 = 762786192;
	private static final int RESULT2 = -21474836;

	static final CallSite MCS = new MutableCallSite(methodType(void.class));
	static final MethodHandle MCS_INVOKER = MCS.dynamicInvoker();

	static void test(boolean shouldThrow) {
	try {
	MCS_INVOKER.invokeExact();
	if (shouldThrow) {
	throw new AssertionError("should throw");
	}
	} catch (IllegalStateException ise) {
	if (!shouldThrow) {
	throw new AssertionError("should not throw", ise);
	}
	} catch (Throwable e) {
	throw new Error(e);
	}
	}

	static class MyCCS extends ConstantCallSite {
	public MyCCS(MethodType targetType, MethodHandle createTargetHook) throws Throwable {
	super(targetType, createTargetHook);
	}
	}

	private static MethodHandle testConstantCallSiteHandler(CallSite cs, CallSite[] holder) throws Throwable {
	holder[0] = cs; // capture call site instance for subsequent checks

	MethodType csType = cs.type(); // should not throw on partially constructed instance

	// Truly dynamic invoker for constant call site
	MethodHandle getTarget = lookup().findVirtual(CallSite.class, "getTarget", MethodType.methodType(MethodHandle.class))
	.bindTo(cs);
	MethodHandle invoker = MethodHandles.exactInvoker(csType);
	MethodHandle target = MethodHandles.foldArguments(invoker, getTarget);

	MCS.setTarget(target);
	// warmup
	for (int i = 0; i < 20_000; i++) {
	test(true); // should throw IllegalStateException
	}

	return MethodHandles.empty(csType); // initialize cs with an empty method handle
	}

	private static void testConstantCallSite() throws Throwable {
	CallSite[] holder = new CallSite[1];
	MethodHandle handler = lookup().findStatic(CLASS, "testConstantCallSiteHandler", MethodType.methodType(MethodHandle.class, CallSite.class, CallSite[].class));
	handler = MethodHandles.insertArguments(handler, 1, new Object[] { holder } );

	CallSite ccs = new MyCCS(MCS.type(), handler); // trigger call to handler

	if (ccs != holder[0]) {
	throw new AssertionError("different call site instances");
	}
	for (int i = 0; i < 20_000; i++) {
	test(false); // should not throw
	}
	}

	private static void testMutableCallSite() throws Throwable {
	// warm-up
	for (int i = 0; i < 20000; i++) {
	mcs.setTarget(mh_foo);
	}
	// run
	for (int n = 0; n < 2; n++) {
	mcs.setTarget(mh_foo);
	for (int i = 0; i < 5; i++) {
	assertEQ(RESULT1, runMutableCallSite());
	}
	mcs.setTarget(mh_bar);
	for (int i = 0; i < 5; i++) {
	assertEQ(RESULT2, runMutableCallSite());
	}
	}
	}
	private static void testVolatileCallSite() throws Throwable {
	// warm-up
	for (int i = 0; i < 20000; i++) {
	vcs.setTarget(mh_foo);
	}
	// run
	for (int n = 0; n < 2; n++) {
	vcs.setTarget(mh_foo);
	for (int i = 0; i < 5; i++) {
	assertEQ(RESULT1, runVolatileCallSite());
	}
	vcs.setTarget(mh_bar);
	for (int i = 0; i < 5; i++) {
	assertEQ(RESULT2, runVolatileCallSite());
	}
	}
	}

	private static int runMutableCallSite() throws Throwable {
	int sum = 0;
	for (int i = 0; i < N; i++) {
	sum += (int) INDY_mcs().invokeExact(i, i+1);
	}
	return sum;
	}
	private static int runVolatileCallSite() throws Throwable {
	int sum = 0;
	for (int i = 0; i < N; i++) {
	sum += (int) INDY_vcs().invokeExact(i, i+1);
	}
	return sum;
	}

	static int foo(int a, int b) { return a + b; }
	static int bar(int a, int b) { return a - b; }

	private static MethodType MT_bsm() {
	shouldNotCallThis();
	return methodType(CallSite.class, Lookup.class, String.class, MethodType.class);
	}

	private static CallSite bsm_mcs(Lookup caller, String name, MethodType type) throws ReflectiveOperationException {
	return mcs;
	}
	private static MethodHandle MH_bsm_mcs() throws ReflectiveOperationException {
	shouldNotCallThis();
	return lookup().findStatic(lookup().lookupClass(), "bsm_mcs", MT_bsm());
	}
	private static MethodHandle INDY_mcs() throws Throwable {
	shouldNotCallThis();
	return ((CallSite) MH_bsm_mcs().invoke(lookup(), "foo", methodType(int.class, int.class, int.class))).dynamicInvoker();
	}

	private static CallSite bsm_vcs(Lookup caller, String name, MethodType type) throws ReflectiveOperationException {
	return vcs;
	}
	private static MethodHandle MH_bsm_vcs() throws ReflectiveOperationException {
	shouldNotCallThis();
	return lookup().findStatic(lookup().lookupClass(), "bsm_vcs", MT_bsm());
	}
	private static MethodHandle INDY_vcs() throws Throwable {
	shouldNotCallThis();
	return ((CallSite) MH_bsm_vcs().invoke(lookup(), "foo", methodType(int.class, int.class, int.class))).dynamicInvoker();
	}

	private static void shouldNotCallThis() {
	// if this gets called, the transformation has not taken place
	throw new AssertionError("this code should be statically transformed away by Indify");
	}
	}