jdk/test/hotspot/jtreg/runtime/cds/appcds/aotClassLinking/AOTClassLinkingVerification.java
2025-09-10 19:21:00 +00:00

295 lines
11 KiB
Java

/*
* Copyright (c) 2023, 2025, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation.
*
* This code is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*
*/
/*
* @test
* @bug 8317269
* @requires vm.cds
* @requires vm.cds.supports.aot.class.linking
* @summary Test for verification of classes that are aot-linked
* @library /test/jdk/lib/testlibrary
* /test/lib
* /test/hotspot/jtreg/runtime/cds/appcds
* /test/hotspot/jtreg/runtime/cds/appcds/test-classes
* @build GoodOldClass
* BadOldClass BadOldClass2 BadOldClass3 BadOldClass4
* BadNewClass BadNewClass2 BadNewClass3 BadNewClass4
* @build AOTClassLinkingVerification
* @run driver jdk.test.lib.helpers.ClassFileInstaller -jar WhiteBox.jar jdk.test.whitebox.WhiteBox
* @run driver jdk.test.lib.helpers.ClassFileInstaller -jar app1.jar
* AOTClassLinkingVerificationApp
* Unlinked UnlinkedSuper
* BadOldClass
* BadOldClass2
* BadOldClass3
* BadOldClass4
* BadNewClass
* BadNewClass2
* BadNewClass3
* BadNewClass4
* GoodOldClass Vehicle Car
* Util
* @run driver jdk.test.lib.helpers.ClassFileInstaller -jar app2.jar
* Foo NotFoo
* UnlinkedSub
* @run driver AOTClassLinkingVerification
*/
import java.io.File;
import java.lang.invoke.MethodHandles;
import jdk.test.lib.cds.CDSAppTester;
import jdk.test.lib.helpers.ClassFileInstaller;
import jdk.test.lib.process.OutputAnalyzer;
import jdk.test.whitebox.WhiteBox;
public class AOTClassLinkingVerification {
static final String app1Jar = ClassFileInstaller.getJarPath("app1.jar");
static final String app2Jar = ClassFileInstaller.getJarPath("app2.jar");
static final String wbJar = TestCommon.getTestJar("WhiteBox.jar");
static final String bootAppendWhiteBox = "-Xbootclasspath/a:" + wbJar;
static final String mainClass = AOTClassLinkingVerificationApp.class.getName();
static class Tester extends CDSAppTester {
public Tester(String testName) {
super(testName);
}
@Override
public String[] vmArgs(RunMode runMode) {
if (runMode == RunMode.TRAINING ||
runMode == RunMode.ASSEMBLY) {
return new String[] {
"-XX:+AOTClassLinking", "-Xlog:cds+class=debug", bootAppendWhiteBox,
};
} else {
return new String[] {
"-XX:+UnlockDiagnosticVMOptions", "-XX:+WhiteBoxAPI", bootAppendWhiteBox,
};
}
}
@Override
public String classpath(RunMode runMode) {
if (runMode == RunMode.TRAINING ||
runMode == RunMode.ASSEMBLY) {
return app1Jar;
} else {
return app1Jar + File.pathSeparator + app2Jar;
}
}
@Override
public String[] appCommandLine(RunMode runMode) {
if (runMode == RunMode.TRAINING ||
runMode == RunMode.ASSEMBLY) {
return new String[] {
"AOTClassLinkingVerificationApp", app1Jar, "ASSEMBLY"
};
} else {
return new String[] {
"AOTClassLinkingVerificationApp", app1Jar, "PRODUCTION"
};
}
}
@Override
public void checkExecution(OutputAnalyzer out, RunMode runMode) throws Exception {
if (runMode == RunMode.TRAINING) {
out.shouldContain("Preload Warning: Verification failed for BadNewClass");
out.shouldContain("Preload Warning: Verification failed for BadNewClass2");
out.shouldContain("Preload Warning: Verification failed for BadNewClass3");
out.shouldContain("Preload Warning: Verification failed for BadNewClass4");
out.shouldContain("Preload Warning: Verification failed for BadOldClass");
out.shouldContain("Preload Warning: Verification failed for BadOldClass2");
out.shouldContain("Preload Warning: Verification failed for BadOldClass3");
out.shouldContain("Preload Warning: Verification failed for BadOldClass4");
out.shouldContain("Preload Warning: Verification failed for Unlinked");
}
}
}
public static void main(String[] args) throws Exception {
// Dump without app2.jar so:
// - Unlinked can be resolved, but UnlinkedSuper UnlinkedSub cannot be resolved,
// so Unlinked cannot be verified at dump time.
// - BadOldClass2 can be resolved, but Foo and NotFoo cannot be resolved,
// so BadOldClass2 cannot be verified at dump time.
// - BadNewClass2 can be resolved, but Foo and NotFoo cannot be resolved,
// so BadNewClass2 cannot be verified at dump time.
Tester t1 = new Tester("verification-aot-linked-classes");
t1.run("AOT");
}
}
class AOTClassLinkingVerificationApp {
static WhiteBox wb = WhiteBox.getWhiteBox();
static ClassLoader classLoader = AOTClassLinkingVerificationApp.class.getClassLoader();
static File app1Jar;
static boolean isProduction;
public static void main(String[] args) throws Exception {
app1Jar = new File(args[0]);
isProduction = args[1].equals("PRODUCTION");
if (isProduction) {
assertNotShared(UnlinkedSub.class);
assertShared(UnlinkedSuper.class);
assertNotShared(Unlinked.class); // failed verification during dump time
assertNotShared(Foo.class);
assertNotShared(NotFoo.class);
}
String s = null;
try {
s = Unlinked.doit();
} catch (NoClassDefFoundError ncdfe) {
// UnlinkedSub is in app2Jar but only app1Jar is used during training
// and assembly phases. So NoClassDefFoundError is expected during
// during training and assembly phases.
if (isProduction) {
throw ncdfe;
}
}
if (isProduction && !s.equals("heyhey")) {
throw new RuntimeException("Unlinked.doit() returns wrong result: " + s);
}
// ===============================================================================
checkSimpleBadClass("BadOldClass");
Class cls_BadOldClass2 = Class.forName("BadOldClass2", false, classLoader);
if (isProduction) {
assertNotShared(cls_BadOldClass2); // failed verification during dump time
}
try {
cls_BadOldClass2.newInstance();
throw new RuntimeException("BadOldClass2 cannot be verified");
} catch (NoClassDefFoundError ncdfe) {
// BadOldClass2 loads Foo and NotFoo which is in app2Jar which is used
// only in production run.
if (isProduction) {
throw ncdfe;
}
} catch (VerifyError expected) {}
checkSimpleBadClass("BadOldClass3");
checkSimpleBadClass("BadOldClass4");
// ===============================================================================
checkSimpleBadClass("BadNewClass");
Class cls_BadNewClass2 = Class.forName("BadNewClass2", false, classLoader);
if (isProduction) {
assertNotShared(cls_BadNewClass2); // failed verification during dump time
}
try {
cls_BadNewClass2.newInstance();
throw new RuntimeException("BadNewClass2 cannot be verified");
} catch (NoClassDefFoundError ncdfe) {
// BadNewClass2 loads Foo and NotFoo which is in app2Jar which is used
// only in production run.
if (isProduction) {
throw ncdfe;
}
} catch (VerifyError expected) {}
checkSimpleBadClass("BadNewClass3");
checkSimpleBadClass("BadNewClass4");
// ===============================================================================
if (isProduction) {
assertAlreadyLoaded("Vehicle");
assertAlreadyLoaded("Car");
assertAlreadyLoaded("GoodOldClass");
assertShared(GoodOldClass.class);
assertShared(Vehicle.class);
assertShared(Car.class);
}
GoodOldClass.doit(); // Should not fail
}
static void checkSimpleBadClass(String className) throws Exception {
Class cls = Class.forName(className, false, classLoader);
if (isProduction) {
assertNotShared(cls); // failed verification during dump time
}
try {
cls.newInstance();
throw new RuntimeException(className + " should not pass verification");
} catch (VerifyError expected) {}
}
static void assertShared(Class c) {
if (!wb.isSharedClass(c)) {
throw new RuntimeException("wb.isSharedClass(" + c.getName() + ") should be true");
}
}
static void assertNotShared(Class c) {
if (wb.isSharedClass(c)) {
throw new RuntimeException("wb.isSharedClass(" + c.getName() + ") should be false");
}
}
static void assertAlreadyLoaded(String className) throws Exception {
byte[] data = Util.getClassFileFromJar(app1Jar, className);
try {
MethodHandles.lookup().defineClass(data);
} catch (LinkageError e) {
if (e.getMessage().contains("duplicate class definition for " + className)) {
return;
} else {
throw e;
}
}
throw new RuntimeException(className + " must have already been loaded");
}
}
class Unlinked {
static String doit() {
UnlinkedSuper sup = new UnlinkedSub();
return sup.doit();
}
}
abstract class UnlinkedSuper {
abstract String doit();
}
class UnlinkedSub extends UnlinkedSuper {
String doit() {
return "heyhey";
}
}
class Foo {}
class NotFoo {}
class Vehicle {}
class Car extends Vehicle {}