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