1 /*
   2  * Copyright (c) 2015, 2017, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /*
  25  * @test
  26  * @bug 8143628
  27  * @summary Test unsafe access for Object
  28  *
  29  * @modules java.base/jdk.internal.misc:+open
  30  * @run testng/othervm -Diters=100   -Xint                   compiler.unsafe.JdkInternalMiscUnsafeAccessTestObject
  31  * @run testng/othervm -Diters=20000 -XX:TieredStopAtLevel=1 compiler.unsafe.JdkInternalMiscUnsafeAccessTestObject
  32  * @run testng/othervm -Diters=20000 -XX:-TieredCompilation  compiler.unsafe.JdkInternalMiscUnsafeAccessTestObject
  33  * @run testng/othervm -Diters=20000                         compiler.unsafe.JdkInternalMiscUnsafeAccessTestObject
  34  */
  35 
  36 package compiler.unsafe;
  37 
  38 import org.testng.annotations.Test;
  39 
  40 import java.lang.reflect.Field;
  41 
  42 import static org.testng.Assert.*;
  43 
  44 public class JdkInternalMiscUnsafeAccessTestObject {
  45     static final int ITERS = Integer.getInteger("iters", 1);
  46     static final int WEAK_ATTEMPTS = Integer.getInteger("weakAttempts", 10);
  47 
  48     static final jdk.internal.misc.Unsafe UNSAFE;
  49 
  50     static final long V_OFFSET;
  51 
  52     static final Object STATIC_V_BASE;
  53 
  54     static final long STATIC_V_OFFSET;
  55 
  56     static int ARRAY_OFFSET;
  57 
  58     static int ARRAY_SHIFT;
  59 
  60     static {
  61         try {
  62             Field f = jdk.internal.misc.Unsafe.class.getDeclaredField("theUnsafe");
  63             f.setAccessible(true);
  64             UNSAFE = (jdk.internal.misc.Unsafe) f.get(null);
  65         } catch (Exception e) {
  66             throw new RuntimeException("Unable to get Unsafe instance.", e);
  67         }
  68 
  69         try {
  70             Field staticVField = JdkInternalMiscUnsafeAccessTestObject.class.getDeclaredField("static_v");
  71             STATIC_V_BASE = UNSAFE.staticFieldBase(staticVField);
  72             STATIC_V_OFFSET = UNSAFE.staticFieldOffset(staticVField);
  73         } catch (Exception e) {
  74             throw new RuntimeException(e);
  75         }
  76 
  77         try {
  78             Field vField = JdkInternalMiscUnsafeAccessTestObject.class.getDeclaredField("v");
  79             V_OFFSET = UNSAFE.objectFieldOffset(vField);
  80         } catch (Exception e) {
  81             throw new RuntimeException(e);
  82         }
  83 
  84         ARRAY_OFFSET = UNSAFE.arrayBaseOffset(Object[].class);
  85         int ascale = UNSAFE.arrayIndexScale(Object[].class);
  86         ARRAY_SHIFT = 31 - Integer.numberOfLeadingZeros(ascale);
  87     }
  88 
  89     static Object static_v;
  90 
  91     Object v;
  92 
  93     @Test
  94     public void testFieldInstance() {
  95         JdkInternalMiscUnsafeAccessTestObject t = new JdkInternalMiscUnsafeAccessTestObject();
  96         for (int c = 0; c < ITERS; c++) {
  97             testAccess(t, V_OFFSET);
  98         }
  99     }
 100 
 101     @Test
 102     public void testFieldStatic() {
 103         for (int c = 0; c < ITERS; c++) {
 104             testAccess(STATIC_V_BASE, STATIC_V_OFFSET);
 105         }
 106     }
 107 
 108     @Test
 109     public void testArray() {
 110         Object[] array = new Object[10];
 111         for (int c = 0; c < ITERS; c++) {
 112             for (int i = 0; i < array.length; i++) {
 113                 testAccess(array, (((long) i) << ARRAY_SHIFT) + ARRAY_OFFSET);
 114             }
 115         }
 116     }
 117 
 118 
 119     static void testAccess(Object base, long offset) {
 120         // Plain
 121         {
 122             UNSAFE.putReference(base, offset, "foo");
 123             Object x = UNSAFE.getReference(base, offset);
 124             assertEquals(x, "foo", "set Object value");
 125         }
 126 
 127         // Volatile
 128         {
 129             UNSAFE.putReferenceVolatile(base, offset, "bar");
 130             Object x = UNSAFE.getReferenceVolatile(base, offset);
 131             assertEquals(x, "bar", "putVolatile Object value");
 132         }
 133 
 134 
 135         // Lazy
 136         {
 137             UNSAFE.putReferenceRelease(base, offset, "foo");
 138             Object x = UNSAFE.getReferenceAcquire(base, offset);
 139             assertEquals(x, "foo", "putRelease Object value");
 140         }
 141 
 142         // Opaque
 143         {
 144             UNSAFE.putReferenceOpaque(base, offset, "bar");
 145             Object x = UNSAFE.getReferenceOpaque(base, offset);
 146             assertEquals(x, "bar", "putOpaque Object value");
 147         }
 148 
 149 
 150         UNSAFE.putReference(base, offset, "foo");
 151 
 152         // Compare
 153         {
 154             boolean r = UNSAFE.compareAndSetReference(base, offset, "foo", "bar");
 155             assertEquals(r, true, "success compareAndSet Object");
 156             Object x = UNSAFE.getReference(base, offset);
 157             assertEquals(x, "bar", "success compareAndSet Object value");
 158         }
 159 
 160         {
 161             boolean r = UNSAFE.compareAndSetReference(base, offset, "foo", "baz");
 162             assertEquals(r, false, "failing compareAndSet Object");
 163             Object x = UNSAFE.getReference(base, offset);
 164             assertEquals(x, "bar", "failing compareAndSet Object value");
 165         }
 166 
 167         // Advanced compare
 168         {
 169             Object r = UNSAFE.compareAndExchangeReference(base, offset, "bar", "foo");
 170             assertEquals(r, "bar", "success compareAndExchange Object");
 171             Object x = UNSAFE.getReference(base, offset);
 172             assertEquals(x, "foo", "success compareAndExchange Object value");
 173         }
 174 
 175         {
 176             Object r = UNSAFE.compareAndExchangeReference(base, offset, "bar", "baz");
 177             assertEquals(r, "foo", "failing compareAndExchange Object");
 178             Object x = UNSAFE.getReference(base, offset);
 179             assertEquals(x, "foo", "failing compareAndExchange Object value");
 180         }
 181 
 182         {
 183             Object r = UNSAFE.compareAndExchangeReferenceAcquire(base, offset, "foo", "bar");
 184             assertEquals(r, "foo", "success compareAndExchangeAcquire Object");
 185             Object x = UNSAFE.getReference(base, offset);
 186             assertEquals(x, "bar", "success compareAndExchangeAcquire Object value");
 187         }
 188 
 189         {
 190             Object r = UNSAFE.compareAndExchangeReferenceAcquire(base, offset, "foo", "baz");
 191             assertEquals(r, "bar", "failing compareAndExchangeAcquire Object");
 192             Object x = UNSAFE.getReference(base, offset);
 193             assertEquals(x, "bar", "failing compareAndExchangeAcquire Object value");
 194         }
 195 
 196         {
 197             Object r = UNSAFE.compareAndExchangeReferenceRelease(base, offset, "bar", "foo");
 198             assertEquals(r, "bar", "success compareAndExchangeRelease Object");
 199             Object x = UNSAFE.getReference(base, offset);
 200             assertEquals(x, "foo", "success compareAndExchangeRelease Object value");
 201         }
 202 
 203         {
 204             Object r = UNSAFE.compareAndExchangeReferenceRelease(base, offset, "bar", "baz");
 205             assertEquals(r, "foo", "failing compareAndExchangeRelease Object");
 206             Object x = UNSAFE.getReference(base, offset);
 207             assertEquals(x, "foo", "failing compareAndExchangeRelease Object value");
 208         }
 209 
 210         {
 211             boolean success = false;
 212             for (int c = 0; c < WEAK_ATTEMPTS && !success; c++) {
 213                 success = UNSAFE.weakCompareAndSetReferencePlain(base, offset, "foo", "bar");
 214             }
 215             assertEquals(success, true, "weakCompareAndSetPlain Object");
 216             Object x = UNSAFE.getReference(base, offset);
 217             assertEquals(x, "bar", "weakCompareAndSetPlain Object value");
 218         }
 219 
 220         {
 221             boolean success = false;
 222             for (int c = 0; c < WEAK_ATTEMPTS && !success; c++) {
 223                 success = UNSAFE.weakCompareAndSetReferenceAcquire(base, offset, "bar", "foo");
 224             }
 225             assertEquals(success, true, "weakCompareAndSetAcquire Object");
 226             Object x = UNSAFE.getReference(base, offset);
 227             assertEquals(x, "foo", "weakCompareAndSetAcquire Object");
 228         }
 229 
 230         {
 231             boolean success = false;
 232             for (int c = 0; c < WEAK_ATTEMPTS && !success; c++) {
 233                 success = UNSAFE.weakCompareAndSetReferenceRelease(base, offset, "foo", "bar");
 234             }
 235             assertEquals(success, true, "weakCompareAndSetRelease Object");
 236             Object x = UNSAFE.getReference(base, offset);
 237             assertEquals(x, "bar", "weakCompareAndSetRelease Object");
 238         }
 239 
 240         {
 241             boolean success = false;
 242             for (int c = 0; c < WEAK_ATTEMPTS && !success; c++) {
 243                 success = UNSAFE.weakCompareAndSetReference(base, offset, "bar", "foo");
 244             }
 245             assertEquals(success, true, "weakCompareAndSet Object");
 246             Object x = UNSAFE.getReference(base, offset);
 247             assertEquals(x, "foo", "weakCompareAndSet Object");
 248         }
 249 
 250         UNSAFE.putReference(base, offset, "bar");
 251 
 252         // Compare set and get
 253         {
 254             Object o = UNSAFE.getAndSetReference(base, offset, "foo");
 255             assertEquals(o, "bar", "getAndSet Object");
 256             Object x = UNSAFE.getReference(base, offset);
 257             assertEquals(x, "foo", "getAndSet Object value");
 258         }
 259 
 260     }
 261 
 262 }