Mercurial > hg > graal-compiler
view graal/com.oracle.graal.replacements.test/src/com/oracle/graal/replacements/test/UnsafeSubstitutionsTest.java @ 22121:f9ca79a05f72
Quote % in SnippetTemplate names
author | Tom Rodriguez <tom.rodriguez@oracle.com> |
---|---|
date | Mon, 29 Jun 2015 10:16:38 -0700 |
parents | 0e095e2c24e2 |
children | 782cccbfa453 |
line wrap: on
line source
/* * Copyright (c) 2014, 2015, 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 com.oracle.graal.replacements.test; import static jdk.internal.jvmci.common.UnsafeAccess.*; import jdk.internal.jvmci.code.*; import jdk.internal.jvmci.meta.*; import org.junit.*; import sun.misc.*; /** * Tests the VM independent intrinsification of {@link Unsafe} methods. */ public class UnsafeSubstitutionsTest extends MethodSubstitutionTest { public void testSubstitution(String testMethodName, Class<?> holder, String methodName, Class<?>[] parameterTypes, Object receiver, Object[] args1, Object[] args2) { ResolvedJavaMethod testMethod = getResolvedJavaMethod(testMethodName); ResolvedJavaMethod originalMethod = getResolvedJavaMethod(holder, methodName, parameterTypes); // Force compilation InstalledCode code = getCode(testMethod); assert code != null; // Verify that the original method and the substitution produce the same value Object expected = invokeSafe(originalMethod, receiver, args1); Object actual = invokeSafe(testMethod, null, args2); assertDeepEquals(expected, actual); // Verify that the generated code and the original produce the same value expected = invokeSafe(originalMethod, receiver, args1); actual = executeVarargsSafe(code, args2); assertDeepEquals(expected, actual); } static long off(Object o, String name) { try { return unsafe.objectFieldOffset(o.getClass().getDeclaredField(name)); } catch (Exception e) { Assert.fail(e.toString()); return 0L; } } static class Foo { boolean z; byte b; short s; char c; int i; long l; float f; double d; Object o; } @Test public void testUnsafeSubstitutions() throws Exception { test("unsafeCompareAndSwapInt", unsafe, supply(() -> new Foo()), fooOffset("i")); testGraph("unsafeCompareAndSwapInt"); testGraph("unsafeCompareAndSwapLong"); testGraph("unsafeCompareAndSwapObject"); testGraph("unsafeGetBoolean"); testGraph("unsafeGetByte"); testGraph("unsafeGetShort"); testGraph("unsafeGetChar"); testGraph("unsafeGetInt"); testGraph("unsafeGetLong"); testGraph("unsafeGetFloat"); testGraph("unsafeGetDouble"); testGraph("unsafeGetObject"); testGraph("unsafePutBoolean"); testGraph("unsafePutByte"); testGraph("unsafePutShort"); testGraph("unsafePutChar"); testGraph("unsafePutInt"); testGraph("unsafePutLong"); testGraph("unsafePutFloat"); testGraph("unsafePutDouble"); testGraph("unsafePutObject"); testGraph("unsafeGetAddress"); testGraph("unsafePutAddress"); testGraph("unsafeDirectMemoryRead"); testGraph("unsafeDirectMemoryWrite"); long address = unsafe.allocateMemory(8 * Kind.values().length); for (Unsafe unsafeArg : new Unsafe[]{unsafe, null}) { test("unsafeCompareAndSwapInt", unsafeArg, supply(() -> new Foo()), fooOffset("i")); test("unsafeCompareAndSwapLong", unsafeArg, supply(() -> new Foo()), fooOffset("l")); test("unsafeCompareAndSwapObject", unsafeArg, supply(() -> new Foo()), fooOffset("o")); test("unsafeGetBoolean", unsafeArg, supply(() -> new Foo()), fooOffset("z")); test("unsafeGetByte", unsafeArg, supply(() -> new Foo()), fooOffset("b")); test("unsafeGetShort", unsafeArg, supply(() -> new Foo()), fooOffset("s")); test("unsafeGetChar", unsafeArg, supply(() -> new Foo()), fooOffset("c")); test("unsafeGetInt", unsafeArg, supply(() -> new Foo()), fooOffset("i")); test("unsafeGetLong", unsafeArg, supply(() -> new Foo()), fooOffset("l")); test("unsafeGetFloat", unsafeArg, supply(() -> new Foo()), fooOffset("f")); test("unsafeGetDouble", unsafeArg, supply(() -> new Foo()), fooOffset("d")); test("unsafeGetObject", unsafeArg, supply(() -> new Foo()), fooOffset("o")); test("unsafePutBoolean", unsafeArg, supply(() -> new Foo()), fooOffset("z"), true); test("unsafePutByte", unsafeArg, supply(() -> new Foo()), fooOffset("b"), (byte) 87); test("unsafePutShort", unsafeArg, supply(() -> new Foo()), fooOffset("s"), (short) -93); test("unsafePutChar", unsafeArg, supply(() -> new Foo()), fooOffset("c"), 'A'); test("unsafePutInt", unsafeArg, supply(() -> new Foo()), fooOffset("i"), 42); test("unsafePutLong", unsafeArg, supply(() -> new Foo()), fooOffset("l"), 4711L); test("unsafePutFloat", unsafeArg, supply(() -> new Foo()), fooOffset("f"), 58.0F); test("unsafePutDouble", unsafeArg, supply(() -> new Foo()), fooOffset("d"), -28736.243465D); test("unsafePutObject", unsafeArg, supply(() -> new Foo()), fooOffset("i"), "value1", "value2", "value3"); test("unsafeGetAddress", unsafeArg, address); test("unsafePutAddress", unsafeArg, address, 0xDEAD_BEEF_DEAD_BABEL); test("unsafeDirectMemoryRead", unsafeArg, address); test("unsafeDirectMemoryWrite", unsafeArg, address, 0xCAFE_BABE_DEAD_BABEL); } unsafe.freeMemory(address); } private static long fooOffset(String name) { try { return unsafe.objectFieldOffset(Foo.class.getDeclaredField(name)); } catch (NoSuchFieldException | SecurityException e) { throw new AssertionError(e); } } @SuppressWarnings("all") public static boolean unsafeCompareAndSwapInt(Unsafe unsafe, Object obj, long offset) { return unsafe.compareAndSwapInt(obj, offset, 0, 1); } @SuppressWarnings("all") public static boolean unsafeCompareAndSwapLong(Unsafe unsafe, Object obj, long offset) { return unsafe.compareAndSwapLong(obj, offset, 0, 1); } @SuppressWarnings("all") public static boolean unsafeCompareAndSwapObject(Unsafe unsafe, Object obj, long offset) { return unsafe.compareAndSwapObject(obj, offset, null, new Object()); } @SuppressWarnings("all") public static boolean unsafeGetBoolean(Unsafe unsafe, Object obj, long offset) { return unsafe.getBoolean(obj, offset) && unsafe.getBooleanVolatile(obj, offset); } @SuppressWarnings("all") public static int unsafeGetByte(Unsafe unsafe, Object obj, long offset) { return unsafe.getByte(obj, offset) + unsafe.getByteVolatile(obj, offset); } @SuppressWarnings("all") public static int unsafeGetShort(Unsafe unsafe, Object obj, long offset) { return unsafe.getShort(obj, offset) + unsafe.getShortVolatile(obj, offset); } @SuppressWarnings("all") public static int unsafeGetChar(Unsafe unsafe, Object obj, long offset) { return unsafe.getChar(obj, offset) + unsafe.getCharVolatile(obj, offset); } @SuppressWarnings("all") public static int unsafeGetInt(Unsafe unsafe, Object obj, long offset) { return unsafe.getInt(obj, offset) + unsafe.getIntVolatile(obj, offset); } @SuppressWarnings("all") public static long unsafeGetLong(Unsafe unsafe, Object obj, long offset) { return unsafe.getLong(obj, offset) + unsafe.getLongVolatile(obj, offset); } @SuppressWarnings("all") public static float unsafeGetFloat(Unsafe unsafe, Object obj, long offset) { return unsafe.getFloat(obj, offset) + unsafe.getFloatVolatile(obj, offset); } @SuppressWarnings("all") public static double unsafeGetDouble(Unsafe unsafe, Object obj, long offset) { return unsafe.getDouble(obj, offset) + unsafe.getDoubleVolatile(obj, offset); } @SuppressWarnings("all") public static boolean unsafeGetObject(Unsafe unsafe, Object obj, long offset) { return unsafe.getObject(obj, offset) == unsafe.getObjectVolatile(obj, offset); } @SuppressWarnings("all") public static int unsafePutBoolean(Unsafe unsafe, Object obj, long offset, boolean value) { int res = 1; unsafe.putBoolean(obj, offset, value); res += unsafe.getBoolean(obj, offset) ? 3 : 5; unsafe.putBooleanVolatile(obj, offset, value); res += unsafe.getBoolean(obj, offset) ? 7 : 11; return res; } @SuppressWarnings("all") public static int unsafePutByte(Unsafe unsafe, Object obj, long offset, byte value) { int res = 1; unsafe.putByte(obj, offset, (byte) (value + 1)); res += unsafe.getByte(obj, offset); unsafe.putByteVolatile(obj, offset, (byte) (value + 2)); res += unsafe.getByte(obj, offset); return res; } @SuppressWarnings("all") public static int unsafePutShort(Unsafe unsafe, Object obj, long offset, short value) { int res = 1; unsafe.putShort(obj, offset, (short) (value + 1)); res += unsafe.getShort(obj, offset); unsafe.putShortVolatile(obj, offset, (short) (value + 2)); res += unsafe.getShort(obj, offset); return res; } @SuppressWarnings("all") public static int unsafePutChar(Unsafe unsafe, Object obj, long offset, char value) { int res = 1; unsafe.putChar(obj, offset, (char) (value + 1)); res += unsafe.getChar(obj, offset); unsafe.putCharVolatile(obj, offset, (char) (value + 2)); res += unsafe.getChar(obj, offset); return res; } @SuppressWarnings("all") public static int unsafePutInt(Unsafe unsafe, Object obj, long offset, int value) { int res = 1; unsafe.putInt(obj, offset, value); res += unsafe.getInt(obj, offset); unsafe.putIntVolatile(obj, offset, value + 1); res += unsafe.getInt(obj, offset); unsafe.putOrderedInt(obj, offset, value + 2); res += unsafe.getInt(obj, offset); return res; } @SuppressWarnings("all") public static long unsafePutLong(Unsafe unsafe, Object obj, long offset, long value) { long res = 1; unsafe.putLong(obj, offset, value + 1); res += unsafe.getLong(obj, offset); unsafe.putLongVolatile(obj, offset, value + 2); res += unsafe.getLong(obj, offset); unsafe.putOrderedLong(obj, offset, value + 3); res += unsafe.getLong(obj, offset); return res; } @SuppressWarnings("all") public static float unsafePutFloat(Unsafe unsafe, Object obj, long offset, float value) { float res = 1; unsafe.putFloat(obj, offset, value + 1.0F); res += unsafe.getFloat(obj, offset); unsafe.putFloatVolatile(obj, offset, value + 2.0F); res += unsafe.getFloat(obj, offset); return res; } @SuppressWarnings("all") public static double unsafePutDouble(Unsafe unsafe, Object obj, long offset, double value) { double res = 1; unsafe.putDouble(obj, offset, value); res += unsafe.getDouble(obj, offset); unsafe.putDoubleVolatile(obj, offset, value); res += unsafe.getDouble(obj, offset); return res; } @SuppressWarnings("all") public static Object[] unsafePutObject(Unsafe unsafe, Object obj, long offset, Object value1, Object value2, Object value3) { Object[] res = new Object[3]; unsafe.putObject(obj, offset, value1); res[0] = unsafe.getObject(obj, offset); unsafe.putObjectVolatile(obj, offset, value2); res[1] = unsafe.getObject(obj, offset); unsafe.putOrderedObject(obj, offset, value3); res[2] = unsafe.getObject(obj, offset); return res; } @SuppressWarnings("all") public static long unsafeGetAddress(Unsafe unsafe, long offset) { return unsafe.getAddress(offset); } @SuppressWarnings("all") public static long unsafePutAddress(Unsafe unsafe, long offset, long value) { long res = 1; unsafe.putAddress(offset, value); res += unsafe.getAddress(offset); return res; } @SuppressWarnings("all") public static double unsafeDirectMemoryRead(Unsafe unsafe, long address) { // Unsafe.getBoolean(long) and Unsafe.getObject(long) do not exist // @formatter:off return unsafe.getByte(address) + unsafe.getShort(address + 8) + unsafe.getChar(address + 16) + unsafe.getInt(address + 24) + unsafe.getLong(address + 32) + unsafe.getFloat(address + 40) + unsafe.getDouble(address + 48); // @formatter:on } @SuppressWarnings("all") public static double unsafeDirectMemoryWrite(Unsafe unsafe, long address, long value) { // Unsafe.putBoolean(long) and Unsafe.putObject(long) do not exist unsafe.putByte(address + 0, (byte) value); unsafe.putShort(address + 8, (short) value); unsafe.putChar(address + 16, (char) value); unsafe.putInt(address + 24, (int) value); unsafe.putLong(address + 32, value); unsafe.putFloat(address + 40, value); unsafe.putDouble(address + 48, value); return unsafeDirectMemoryRead(unsafe, address); } @Test public void testGetAndAddInt() throws Exception { Foo f1 = new Foo(); Foo f2 = new Foo(); long offset = off(f1, "i"); Class<?>[] parameterTypes = new Class<?>[]{Object.class, long.class, int.class}; for (int delta = Integer.MAX_VALUE - 10; delta < Integer.MAX_VALUE; delta++) { Object[] args1 = new Object[]{f1, offset, delta}; Object[] args2 = new Object[]{f2, offset, delta}; testSubstitution("getAndAddInt", Unsafe.class, "getAndAddInt", parameterTypes, unsafe, args1, args2); } } public static int getAndAddInt(Object obj, long offset, int delta) { return unsafe.getAndAddInt(obj, offset, delta); } @Test public void testGetAndAddLong() throws Exception { Foo f1 = new Foo(); Foo f2 = new Foo(); long offset = off(f1, "l"); Class<?>[] parameterTypes = new Class<?>[]{Object.class, long.class, long.class}; for (long delta = Long.MAX_VALUE - 10; delta < Long.MAX_VALUE; delta++) { Object[] args1 = new Object[]{f1, offset, delta}; Object[] args2 = new Object[]{f2, offset, delta}; testSubstitution("getAndAddLong", Unsafe.class, "getAndAddLong", parameterTypes, unsafe, args1, args2); } } public static long getAndAddLong(Object obj, long offset, long delta) { return unsafe.getAndAddLong(obj, offset, delta); } @Test public void testGetAndSetInt() throws Exception { Foo f1 = new Foo(); Foo f2 = new Foo(); long offset = off(f1, "i"); Class<?>[] parameterTypes = new Class<?>[]{Object.class, long.class, int.class}; for (int delta = Integer.MAX_VALUE - 10; delta < Integer.MAX_VALUE; delta++) { Object[] args1 = new Object[]{f1, offset, delta}; Object[] args2 = new Object[]{f2, offset, delta}; testSubstitution("getAndSetInt", Unsafe.class, "getAndSetInt", parameterTypes, unsafe, args1, args2); } } public static int getAndSetInt(Object obj, long offset, int newValue) { return unsafe.getAndSetInt(obj, offset, newValue); } @Test public void testGetAndSetLong() throws Exception { Foo f1 = new Foo(); Foo f2 = new Foo(); long offset = off(f1, "l"); Class<?>[] parameterTypes = new Class<?>[]{Object.class, long.class, long.class}; for (long newValue = Long.MAX_VALUE - 10; newValue < Long.MAX_VALUE; newValue++) { Object[] args1 = new Object[]{f1, offset, newValue}; Object[] args2 = new Object[]{f2, offset, newValue}; testSubstitution("getAndSetLong", Unsafe.class, "getAndSetLong", parameterTypes, unsafe, args1, args2); } } public static long getAndSetLong(Object obj, long offset, long newValue) { return unsafe.getAndSetLong(obj, offset, newValue); } @Test public void testGetAndSetObject() throws Exception { Foo f1 = new Foo(); Foo f2 = new Foo(); long offset = off(f1, "o"); Class<?>[] parameterTypes = new Class<?>[]{Object.class, long.class, Object.class}; for (long i = 0; i < 10; i++) { Object o = new Object(); Object[] args1 = new Object[]{f1, offset, o}; Object[] args2 = new Object[]{f2, offset, o}; testSubstitution("getAndSetObject", Unsafe.class, "getAndSetObject", parameterTypes, unsafe, args1, args2); System.gc(); } } public static Object getAndSetObject(Object obj, long offset, Object newValue) { return unsafe.getAndSetObject(obj, offset, newValue); } }