undefect. CWE-407 — 63 sites patched across 27 ecosystems
Authors: russell@unturf.com · brackishbert@gmail.com · foxhop.net · TimeHexOn.com Patches, unit tests, benchmarks, whitepaper, and outreach briefs. Public domain — no copyright claimed. Use freely.
This commit is contained in:
commit
0a580b313d
70422 changed files with 17213626 additions and 0 deletions
222
test/jdk/java/lang/invoke/CallSiteTest.java
Normal file
222
test/jdk/java/lang/invoke/CallSiteTest.java
Normal file
|
|
@ -0,0 +1,222 @@
|
|||
/*
|
||||
* 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");
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue