1 /*
2 * Copyright (c) 2025, 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 /*
26 * @test id=aot
27 * @requires vm.cds.supports.aot.class.linking
28 * @comment work around JDK-8345635
29 * @requires !vm.jvmci.enabled
30 * @library /test/jdk/lib/testlibrary /test/lib
31 * @build EndTrainingOnMethodEntry
32 * @run driver jdk.test.lib.helpers.ClassFileInstaller -jar app.jar MyTestApp ShouldBeCached ShouldNotBeCached
33 * @run driver EndTrainingOnMethodEntry AOT
34 */
35
36 /*
37 * @test id=leyden
38 * @requires vm.cds.supports.aot.class.linking
39 * @comment work around JDK-8345635
40 * @requires !vm.jvmci.enabled
41 * @library /test/jdk/lib/testlibrary /test/lib
42 * @build EndTrainingOnMethodEntry
43 * @run driver jdk.test.lib.helpers.ClassFileInstaller -jar app.jar MyTestApp ShouldBeCached ShouldNotBeCached
44 * @run driver EndTrainingOnMethodEntry LEYDEN
45 */
46
47 import jdk.test.lib.cds.CDSAppTester;
48 import jdk.test.lib.helpers.ClassFileInstaller;
49 import jdk.test.lib.process.OutputAnalyzer;
50
51 public class EndTrainingOnMethodEntry {
52 static final String appJar = ClassFileInstaller.getJarPath("app.jar");
53 static final String mainClass = "MyTestApp";
54
55 public static void main(String[] args) throws Exception {
56 // We want to test the entry count implementation in both interpreter and compiler.
57 (new Tester(1)).run(args);
58 (new Tester(10)).run(args); // the loop will probably be interpreted
59 (new Tester(10000)).run(args); // the loop will probably be compiled.
60 }
61
62 static class Tester extends CDSAppTester {
63 int count;
64
65 public Tester(int count) {
66 super(mainClass);
67 this.count = count;
68 }
69
70 @Override
71 public String classpath(RunMode runMode) {
72 return appJar;
73 }
74
75 public String[] vmArgs(RunMode runMode) {
76 String stop = count > 1 ? ("stopTrainingOnMeWithCount,count=" + count) : "stopTrainingOnMe";
77 return new String[] {
78 "-Xlog:aot+class=debug",
79 "-Xlog:cds+class=debug",
80 "-XX:AOTEndTrainingOnMethodEntry=MyTestApp." + stop,
81 };
82 }
83
84 @Override
85 public String[] appCommandLine(RunMode runMode) {
86 return new String[] {
87 mainClass, runMode.name(), Integer.toString(count),
88 };
89 }
90
91 @Override
92 public void checkExecution(OutputAnalyzer out, RunMode runMode) {
93 if (runMode.isApplicationExecuted()) {
94 out.shouldContain("Hello Leyden " + runMode.name());
95 out.shouldContain("ShouldBeCached.dummy()");
96 out.shouldContain("ShouldNotBeCached.dummy()");
97 }
98 if (isDumping(runMode)) {
99 if (isAOTWorkflow()) {
100 out.shouldMatch("aot,class.* ShouldBeCached");
101 out.shouldNotMatch("aot,class.* ShouldNotBeCached");
102 } else {
103 out.shouldMatch("cds,class.* ShouldBeCached");
104 out.shouldNotMatch("cds,class.* ShouldNotBeCached");
105 }
106 }
107 }
108 }
109 }
110
111 class MyTestApp {
112 public static int COUNT;
113 public static void main(String args[]) throws Exception {
114 System.out.println("Hello Leyden " + args[0] + ", count = " + args[1]);
115 COUNT = Integer.parseInt(args[1]);
116 if (COUNT > 1) {
117 int max = COUNT + 10;
118 for (int i = 0; i < max; i++) {
119 stopTrainingOnMeWithCount(i);
120 }
121 } else {
122 ShouldBeCached.dummy();
123 stopTrainingOnMe();
124 }
125 }
126
127 static void stopTrainingOnMe() {
128 // The AOT configuration file should have been recorded before the body
129 // of this method is executed, so the ShouldNotBeCached class should not be
130 // recorded in the config.
131 ShouldNotBeCached.dummy();
132 }
133
134 static void stopTrainingOnMeWithCount(int i) {
135 if (i >= COUNT - 2) {
136 ShouldBeCached.dummy();
137 }
138 if (i >= COUNT) {
139 // The AOT configuration file should have been recorded before this block is entered,
140 // so the ShouldNotBeCached class should not be recorded in the config.
141 ShouldNotBeCached.dummy();
142 }
143 }
144 }
145
146 class ShouldBeCached {
147 static void dummy() {
148 System.out.println("ShouldBeCached.dummy()");
149 }
150 }
151
152 class ShouldNotBeCached {
153 static void dummy() {
154 System.out.println("ShouldNotBeCached.dummy()");
155 }
156 }