1 /* 2 * Copyright (c) 2015, 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 short 28 * @modules java.base/jdk.internal.misc 29 * @run testng/othervm -Diters=100 -Xint JdkInternalMiscUnsafeAccessTestShort 30 * @run testng/othervm -Diters=20000 -XX:TieredStopAtLevel=1 JdkInternalMiscUnsafeAccessTestShort 31 * @run testng/othervm -Diters=20000 -XX:-TieredCompilation JdkInternalMiscUnsafeAccessTestShort 32 * @run testng/othervm -Diters=20000 JdkInternalMiscUnsafeAccessTestShort 33 */ 34 35 import org.testng.annotations.Test; 36 37 import java.lang.reflect.Field; 38 39 import static org.testng.Assert.*; 40 41 public class JdkInternalMiscUnsafeAccessTestShort { 42 static final int ITERS = Integer.getInteger("iters", 1); 43 static final int WEAK_ATTEMPTS = Integer.getInteger("weakAttempts", 10); 44 45 static final jdk.internal.misc.Unsafe UNSAFE; 46 47 static final long V_OFFSET; 48 49 static final Object STATIC_V_BASE; 50 51 static final long STATIC_V_OFFSET; 52 53 static int ARRAY_OFFSET; 54 55 static int ARRAY_SHIFT; 56 57 static { 58 try { 59 Field f = jdk.internal.misc.Unsafe.class.getDeclaredField("theUnsafe"); 60 f.setAccessible(true); 61 UNSAFE = (jdk.internal.misc.Unsafe) f.get(null); 62 } catch (Exception e) { 63 throw new RuntimeException("Unable to get Unsafe instance.", e); 64 } 65 66 try { 67 Field staticVField = JdkInternalMiscUnsafeAccessTestShort.class.getDeclaredField("static_v"); 68 STATIC_V_BASE = UNSAFE.staticFieldBase(staticVField); 69 STATIC_V_OFFSET = UNSAFE.staticFieldOffset(staticVField); 70 } catch (Exception e) { 71 throw new RuntimeException(e); 72 } 73 74 try { 75 Field vField = JdkInternalMiscUnsafeAccessTestShort.class.getDeclaredField("v"); 76 V_OFFSET = UNSAFE.objectFieldOffset(vField); 77 } catch (Exception e) { 78 throw new RuntimeException(e); 79 } 80 81 ARRAY_OFFSET = UNSAFE.arrayBaseOffset(short[].class); 82 int ascale = UNSAFE.arrayIndexScale(short[].class); 83 ARRAY_SHIFT = 31 - Integer.numberOfLeadingZeros(ascale); 84 } 85 86 static short static_v; 87 88 short v; 89 90 @Test 91 public void testFieldInstance() { 92 JdkInternalMiscUnsafeAccessTestShort t = new JdkInternalMiscUnsafeAccessTestShort(); 93 for (int c = 0; c < ITERS; c++) { 94 testAccess(t, V_OFFSET); 95 } 96 } 97 98 @Test 99 public void testFieldStatic() { 100 for (int c = 0; c < ITERS; c++) { 101 testAccess(STATIC_V_BASE, STATIC_V_OFFSET); 102 } 103 } 104 105 @Test 106 public void testArray() { 107 short[] array = new short[10]; 108 for (int c = 0; c < ITERS; c++) { 109 for (int i = 0; i < array.length; i++) { 110 testAccess(array, (((long) i) << ARRAY_SHIFT) + ARRAY_OFFSET); 111 } 112 } 113 } 114 115 @Test 116 public void testArrayOffHeap() { 117 int size = 10; 118 long address = UNSAFE.allocateMemory(size << ARRAY_SHIFT); 119 try { 120 for (int c = 0; c < ITERS; c++) { 121 for (int i = 0; i < size; i++) { 122 testAccess(null, (((long) i) << ARRAY_SHIFT) + address); 123 } 124 } 125 } finally { 126 UNSAFE.freeMemory(address); 127 } 128 } 129 130 @Test 131 public void testArrayOffHeapDirect() { 132 int size = 10; 133 long address = UNSAFE.allocateMemory(size << ARRAY_SHIFT); 134 try { 135 for (int c = 0; c < ITERS; c++) { 136 for (int i = 0; i < size; i++) { 137 testAccess((((long) i) << ARRAY_SHIFT) + address); 138 } 139 } 140 } finally { 141 UNSAFE.freeMemory(address); 142 } 143 } 144 145 static void testAccess(Object base, long offset) { 146 // Plain 147 { 148 UNSAFE.putShort(base, offset, (short)0x0123); 149 short x = UNSAFE.getShort(base, offset); 150 assertEquals(x, (short)0x0123, "set short value"); 151 } 152 153 // Volatile 154 { 155 UNSAFE.putShortVolatile(base, offset, (short)0x4567); 156 short x = UNSAFE.getShortVolatile(base, offset); 157 assertEquals(x, (short)0x4567, "putVolatile short value"); 158 } 159 160 161 // Lazy 162 { 163 UNSAFE.putShortRelease(base, offset, (short)0x0123); 164 short x = UNSAFE.getShortAcquire(base, offset); 165 assertEquals(x, (short)0x0123, "putRelease short value"); 166 } 167 168 // Opaque 169 { 170 UNSAFE.putShortOpaque(base, offset, (short)0x4567); 171 short x = UNSAFE.getShortOpaque(base, offset); 172 assertEquals(x, (short)0x4567, "putOpaque short value"); 173 } 174 175 // Unaligned 176 { 177 UNSAFE.putShortUnaligned(base, offset, (short)0x4567); 178 short x = UNSAFE.getShortUnaligned(base, offset); 179 assertEquals(x, (short)0x4567, "putUnaligned short value"); 180 } 181 182 { 183 UNSAFE.putShortUnaligned(base, offset, (short)0x0123, true); 184 short x = UNSAFE.getShortUnaligned(base, offset, true); 185 assertEquals(x, (short)0x0123, "putUnaligned big endian short value"); 186 } 187 188 { 189 UNSAFE.putShortUnaligned(base, offset, (short)0x4567, false); 190 short x = UNSAFE.getShortUnaligned(base, offset, false); 191 assertEquals(x, (short)0x4567, "putUnaligned little endian short value"); 192 } 193 194 UNSAFE.putShort(base, offset, (short)0x0123); 195 196 // Compare 197 { 198 boolean r = UNSAFE.compareAndSwapShort(base, offset, (short)0x0123, (short)0x4567); 199 assertEquals(r, true, "success compareAndSwap short"); 200 short x = UNSAFE.getShort(base, offset); 201 assertEquals(x, (short)0x4567, "success compareAndSwap short value"); 202 } 203 204 { 205 boolean r = UNSAFE.compareAndSwapShort(base, offset, (short)0x0123, (short)0x89AB); 206 assertEquals(r, false, "failing compareAndSwap short"); 207 short x = UNSAFE.getShort(base, offset); 208 assertEquals(x, (short)0x4567, "failing compareAndSwap short value"); 209 } 210 211 // Advanced compare 212 { 213 short r = UNSAFE.compareAndExchangeShortVolatile(base, offset, (short)0x4567, (short)0x0123); 214 assertEquals(r, (short)0x4567, "success compareAndExchangeVolatile short"); 215 short x = UNSAFE.getShort(base, offset); 216 assertEquals(x, (short)0x0123, "success compareAndExchangeVolatile short value"); 217 } 218 219 { 220 short r = UNSAFE.compareAndExchangeShortVolatile(base, offset, (short)0x4567, (short)0x89AB); 221 assertEquals(r, (short)0x0123, "failing compareAndExchangeVolatile short"); 222 short x = UNSAFE.getShort(base, offset); 223 assertEquals(x, (short)0x0123, "failing compareAndExchangeVolatile short value"); 224 } 225 226 { 227 short r = UNSAFE.compareAndExchangeShortAcquire(base, offset, (short)0x0123, (short)0x4567); 228 assertEquals(r, (short)0x0123, "success compareAndExchangeAcquire short"); 229 short x = UNSAFE.getShort(base, offset); 230 assertEquals(x, (short)0x4567, "success compareAndExchangeAcquire short value"); 231 } 232 233 { 234 short r = UNSAFE.compareAndExchangeShortAcquire(base, offset, (short)0x0123, (short)0x89AB); 235 assertEquals(r, (short)0x4567, "failing compareAndExchangeAcquire short"); 236 short x = UNSAFE.getShort(base, offset); 237 assertEquals(x, (short)0x4567, "failing compareAndExchangeAcquire short value"); 238 } 239 240 { 241 short r = UNSAFE.compareAndExchangeShortRelease(base, offset, (short)0x4567, (short)0x0123); 242 assertEquals(r, (short)0x4567, "success compareAndExchangeRelease short"); 243 short x = UNSAFE.getShort(base, offset); 244 assertEquals(x, (short)0x0123, "success compareAndExchangeRelease short value"); 245 } 246 247 { 248 short r = UNSAFE.compareAndExchangeShortRelease(base, offset, (short)0x4567, (short)0x89AB); 249 assertEquals(r, (short)0x0123, "failing compareAndExchangeRelease short"); 250 short x = UNSAFE.getShort(base, offset); 251 assertEquals(x, (short)0x0123, "failing compareAndExchangeRelease short value"); 252 } 253 254 { 255 boolean success = false; 256 for (int c = 0; c < WEAK_ATTEMPTS && !success; c++) { 257 success = UNSAFE.weakCompareAndSwapShort(base, offset, (short)0x0123, (short)0x4567); 258 } 259 assertEquals(success, true, "weakCompareAndSwap short"); 260 short x = UNSAFE.getShort(base, offset); 261 assertEquals(x, (short)0x4567, "weakCompareAndSwap short value"); 262 } 263 264 { 265 boolean success = false; 266 for (int c = 0; c < WEAK_ATTEMPTS && !success; c++) { 267 success = UNSAFE.weakCompareAndSwapShortAcquire(base, offset, (short)0x4567, (short)0x0123); 268 } 269 assertEquals(success, true, "weakCompareAndSwapAcquire short"); 270 short x = UNSAFE.getShort(base, offset); 271 assertEquals(x, (short)0x0123, "weakCompareAndSwapAcquire short"); 272 } 273 274 { 275 boolean success = false; 276 for (int c = 0; c < WEAK_ATTEMPTS && !success; c++) { 277 success = UNSAFE.weakCompareAndSwapShortRelease(base, offset, (short)0x0123, (short)0x4567); 278 } 279 assertEquals(success, true, "weakCompareAndSwapRelease short"); 280 short x = UNSAFE.getShort(base, offset); 281 assertEquals(x, (short)0x4567, "weakCompareAndSwapRelease short"); 282 } 283 284 { 285 boolean success = false; 286 for (int c = 0; c < WEAK_ATTEMPTS && !success; c++) { 287 success = UNSAFE.weakCompareAndSwapShortVolatile(base, offset, (short)0x4567, (short)0x0123); 288 } 289 assertEquals(success, true, "weakCompareAndSwapVolatile short"); 290 short x = UNSAFE.getShort(base, offset); 291 assertEquals(x, (short)0x0123, "weakCompareAndSwapVolatile short"); 292 } 293 294 UNSAFE.putShort(base, offset, (short)0x4567); 295 296 // Compare set and get 297 { 298 short o = UNSAFE.getAndSetShort(base, offset, (short)0x0123); 299 assertEquals(o, (short)0x4567, "getAndSet short"); 300 short x = UNSAFE.getShort(base, offset); 301 assertEquals(x, (short)0x0123, "getAndSet short value"); 302 } 303 304 UNSAFE.putShort(base, offset, (short)0x0123); 305 306 // get and add, add and get 307 { 308 short o = UNSAFE.getAndAddShort(base, offset, (short)0x4567); 309 assertEquals(o, (short)0x0123, "getAndAdd short"); 310 short x = UNSAFE.getShort(base, offset); 311 assertEquals(x, (short)((short)0x0123 + (short)0x4567), "getAndAdd short"); 312 } 313 } 314 315 static void testAccess(long address) { 316 // Plain 317 { 318 UNSAFE.putShort(address, (short)0x0123); 319 short x = UNSAFE.getShort(address); 320 assertEquals(x, (short)0x0123, "set short value"); 321 } 322 } 323 } 324