1 /*
  2  *  Copyright (c) 2021, 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 id=default_gc
 26  * @requires ((os.arch == "amd64" | os.arch == "x86_64") & sun.arch.data.model == "64") | os.arch == "aarch64"
 27  * @library /test/lib
 28  * @build sun.hotspot.WhiteBox
 29  * @run driver jdk.test.lib.helpers.ClassFileInstaller sun.hotspot.WhiteBox
 30  *
 31  * @run main/othervm
 32  *   -Xbootclasspath/a:.
 33  *   -XX:+UnlockDiagnosticVMOptions
 34  *   -XX:+WhiteBoxAPI
 35  *   -Djdk.internal.foreign.ProgrammableInvoker.USE_INTRINSICS=true
 36  *   --enable-native-access=ALL-UNNAMED
 37  *   -Xbatch
 38  *   TestAsyncStackWalk
 39  *
 40  * @run main/othervm
 41  *   -Xbootclasspath/a:.
 42  *   -XX:+UnlockDiagnosticVMOptions
 43  *   -XX:+WhiteBoxAPI
 44  *   -Djdk.internal.foreign.ProgrammableInvoker.USE_INTRINSICS=false
 45  *   --enable-native-access=ALL-UNNAMED
 46  *   -Xbatch
 47  *   TestAsyncStackWalk
 48  */
 49 
 50 /*
 51  * @test id=zgc
 52  * @requires (((os.arch == "amd64" | os.arch == "x86_64") & sun.arch.data.model == "64") | os.arch == "aarch64")
 53  * @requires vm.gc.Z
 54  * @library /test/lib
 55  * @build sun.hotspot.WhiteBox
 56  * @run driver jdk.test.lib.helpers.ClassFileInstaller sun.hotspot.WhiteBox
 57  *
 58  * @run main/othervm
 59  *   -Xbootclasspath/a:.
 60  *   -XX:+UnlockDiagnosticVMOptions
 61  *   -XX:+WhiteBoxAPI
 62  *   -Djdk.internal.foreign.ProgrammableInvoker.USE_INTRINSICS=true
 63  *   --enable-native-access=ALL-UNNAMED
 64  *   -Xbatch
 65  *   -XX:+UseZGC
 66  *   TestAsyncStackWalk
 67  *
 68  * @run main/othervm
 69  *   -Xbootclasspath/a:.
 70  *   -XX:+UnlockDiagnosticVMOptions
 71  *   -XX:+WhiteBoxAPI
 72  *   -Djdk.internal.foreign.ProgrammableInvoker.USE_INTRINSICS=false
 73  *   --enable-native-access=ALL-UNNAMED
 74  *   -Xbatch
 75  *   -XX:+UseZGC
 76  *   TestAsyncStackWalk
 77  */
 78 /*
 79  * @test id=shenandoah
 80  * @requires (((os.arch == "amd64" | os.arch == "x86_64") & sun.arch.data.model == "64") | os.arch == "aarch64")
 81  * @requires vm.gc.Shenandoah
 82  * @library /test/lib
 83  * @build sun.hotspot.WhiteBox
 84  * @run driver jdk.test.lib.helpers.ClassFileInstaller sun.hotspot.WhiteBox
 85  *
 86  * @run main/othervm
 87  *   -Xbootclasspath/a:.
 88  *   -XX:+UnlockDiagnosticVMOptions
 89  *   -XX:+WhiteBoxAPI
 90  *   -Djdk.internal.foreign.ProgrammableInvoker.USE_INTRINSICS=true
 91  *   --enable-native-access=ALL-UNNAMED
 92  *   -Xbatch
 93  *   -XX:+UseShenandoahGC
 94  *   TestAsyncStackWalk
 95  *
 96  * @run main/othervm
 97  *   -Xbootclasspath/a:.
 98  *   -XX:+UnlockDiagnosticVMOptions
 99  *   -XX:+WhiteBoxAPI
100  *   -Djdk.internal.foreign.ProgrammableInvoker.USE_INTRINSICS=false
101  *   --enable-native-access=ALL-UNNAMED
102  *   -Xbatch
103  *   -XX:+UseShenandoahGC
104  *   TestAsyncStackWalk
105  */
106 
107 import jdk.incubator.foreign.CLinker;
108 import jdk.incubator.foreign.FunctionDescriptor;
109 import jdk.incubator.foreign.SymbolLookup;
110 import jdk.incubator.foreign.MemoryAddress;
111 
112 import java.lang.invoke.MethodHandle;
113 import java.lang.invoke.MethodType;
114 
115 import jdk.incubator.foreign.ResourceScope;
116 import sun.hotspot.WhiteBox;
117 
118 import static java.lang.invoke.MethodHandles.lookup;
119 import static jdk.incubator.foreign.CLinker.C_POINTER;
120 import static jdk.test.lib.Asserts.assertTrue;
121 
122 public class TestAsyncStackWalk {
123     static final WhiteBox WB = WhiteBox.getWhiteBox();
124 
125     static final CLinker linker = CLinker.getInstance();
126 
127     static final MethodHandle MH_asyncStackWalk;
128     static final MethodHandle MH_m;
129 
130     static {
131         try {
132             System.loadLibrary("AsyncStackWalk");
133             SymbolLookup lookup = SymbolLookup.loaderLookup();
134             MH_asyncStackWalk = linker.downcallHandle(
135                     lookup.lookup("asyncStackWalk").get(),
136                     MethodType.methodType(void.class, MemoryAddress.class),
137                     FunctionDescriptor.ofVoid(C_POINTER));
138             MH_m = lookup().findStatic(TestAsyncStackWalk.class, "m", MethodType.methodType(void.class));
139         } catch (ReflectiveOperationException e) {
140             throw new RuntimeException(e);
141         }
142     }
143 
144     static int invocations;
145     static boolean didStackWalk;
146 
147     public static void main(String[] args) throws Throwable {
148         try (ResourceScope scope = ResourceScope.newConfinedScope()) {
149             MemoryAddress stub = linker.upcallStub(MH_m, FunctionDescriptor.ofVoid(), scope);
150             MemoryAddress stubAddress = stub.address();
151             invocations = 0;
152             didStackWalk = false;
153             payload(stubAddress);
154             assertTrue(didStackWalk);
155         }
156     }
157 
158     static void payload(MemoryAddress cb) throws Throwable {
159         MH_asyncStackWalk.invokeExact(cb);
160     }
161 
162     static void m() {
163         if (invocations++ >= 20000) { // warmup
164             didStackWalk = true;
165             WB.verifyFrames(/*log=*/true, /*updateRegisterMap=*/true);
166             WB.verifyFrames(/*log=*/true, /*updateRegisterMap=*/false); // triggers different code paths
167         }
168     }
169 
170 }