jdk-24/test/hotspot/jtreg/runtime/cds/appcds/dynamicArchive/DynamicArchiveTestBase.java
Leo Korinth d52a995f35 8315097: Rename createJavaProcessBuilder
Reviewed-by: lmesnik, dholmes, rriggs, stefank
2023-10-27 08:47:26 +00:00

317 lines
12 KiB
Java

/*
* Copyright (c) 2019, 2023, 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.
*
*/
import java.io.File;
import jdk.test.lib.process.OutputAnalyzer;
import jdk.test.lib.process.ProcessTools;
import jdk.test.lib.cds.CDSOptions;
import jdk.test.lib.cds.CDSTestUtils;
import jdk.test.lib.cds.CDSTestUtils.Result;
import jdk.test.lib.helpers.ClassFileInstaller;
import jdk.test.whitebox.WhiteBox;
/**
* Base class for test cases in test/hotspot/jtreg/runtime/cds/appcds/dynamicArchive/
*/
class DynamicArchiveTestBase {
private static boolean executedIn_run = false;
private static boolean autoMode = false; // -Xshare:auto
private static final WhiteBox WB = WhiteBox.getWhiteBox();
private static String[] baseArchiveOptions = new String[] {};
public static interface DynamicArchiveTest {
public void run() throws Exception;
}
public static interface DynamicArchiveTestWithArgs {
public void run(String args[]) throws Exception;
}
public static void setAutoMode(boolean val) { autoMode = val; }
/*
* Tests for dynamic archives should be written using this pattern:
*
* public class HelloDynamic extends DynamicArchiveTestBase {
* public static void main(String[] args) throws Exception {
* runTest(HelloDynamic::testDefaultBase); // launch one test case
* }
*
* // the body of a test case
* static void testDefaultBase() throws Exception {
* String topArchiveName = getNewArchiveName("top");
* doTest(null, topArchiveName);
* }
* }
*
* The reason for this is so that we can clean up the archive files
* created by prior test cases. Otherwise tests with lots of
* test cases may fill up the scratch directory.
*/
public static void runTest(DynamicArchiveTest t) throws Exception {
executedIn_run = true;
try {
TestCommon.deletePriorArchives();
t.run();
} finally {
executedIn_run = false;
}
}
public static void runTest(DynamicArchiveTestWithArgs t, String... args) throws Exception {
executedIn_run = true;
try {
TestCommon.deletePriorArchives();
t.run(args);
} finally {
executedIn_run = false;
}
}
public static String getNewArchiveName() {
return TestCommon.getNewArchiveName();
}
public static String getNewArchiveName(String stem) {
return TestCommon.getNewArchiveName(stem);
}
public static void setBaseArchiveOptions(String... opts) {
baseArchiveOptions = opts;
}
/**
* Excute a JVM to dump a base archive by
* -Xshare:dump -XX:SharedArchiveFile=baseArchiveName
*/
public static Result dumpBaseArchive(String baseArchiveName, String... cmdLineSuffix)
throws Exception
{
OutputAnalyzer output = TestCommon.dumpBaseArchive(baseArchiveName, cmdLineSuffix);
CDSOptions opts = new CDSOptions();
opts.setXShareMode("dump");
return new Result(opts, output);
}
/**
* Execute a JVM using the base archive (given by baseArchiveName) with the command line
* (given by cmdLineSuffix). At JVM exit, dump all eligible classes into the top archive
* (give by topArchiveName).
*
* If baseArchiveName is null, use the JDK's default archive as the base archive.
*/
public static Result dump2(String baseArchiveName, String topArchiveName, String ... cmdLineSuffix)
throws Exception
{
String[] cmdLine = TestCommon.concat(
"-XX:ArchiveClassesAtExit=" + topArchiveName);
// to allow dynamic archive tests to be run in the "rt-non-cds-mode"
cmdLine = TestCommon.concat(cmdLine, "-Xshare:auto");
if (baseArchiveName == null && isUseSharedSpacesDisabled()) {
baseArchiveName = getTempBaseArchive();
}
if (baseArchiveName != null) {
cmdLine = TestCommon.concat(cmdLine, "-XX:SharedArchiveFile=" + baseArchiveName);
}
cmdLine = TestCommon.concat(cmdLine, cmdLineSuffix);
return execProcess("dump", null, cmdLine);
}
public static Result dump2_WB(String baseArchiveName, String topArchiveName, String ... cmdLineSuffix)
throws Exception
{
if (baseArchiveName == null && isUseSharedSpacesDisabled()) {
baseArchiveName = getTempBaseArchive();
}
return dump2(baseArchiveName, topArchiveName,
TestCommon.concat(wbRuntimeArgs(), cmdLineSuffix));
}
/**
* A convenience method similar to dump2, but always use the JDK's default archive
* as the base archive.
*
* Most dynamicArchive/*.java test cases should be using this method instead of run2.
*/
public static Result dump(String topArchiveName, String ... cmdLineSuffix)
throws Exception
{
return dump2(null, topArchiveName, cmdLineSuffix);
}
/**
* Same as TestCommon.dumpBaseArchive, but also add WhiteBox to the bootcp
*/
public static void dumpBaseArchive_WB(String baseArchiveName, String ... cmdLineSuffix)
throws Exception
{
TestCommon.dumpBaseArchive(baseArchiveName,
TestCommon.concat("-Xbootclasspath/a:" + getWhiteBoxJar(), cmdLineSuffix));
}
private static String getWhiteBoxJar() {
String wbJar = ClassFileInstaller.getJarPath("WhiteBox.jar");
if (!(new File(wbJar)).exists()) {
throw new RuntimeException("Test error: your test must have " +
"'@run driver jdk.test.lib.helpers.ClassFileInstaller -jar WhiteBox.jar jdk.test.whitebox.WhiteBox'");
}
return wbJar;
}
private static String[] wbRuntimeArgs() {
return TestCommon.concat("-Xbootclasspath/a:" + getWhiteBoxJar(),
"-XX:+UnlockDiagnosticVMOptions",
"-XX:+WhiteBoxAPI");
}
/**
* Execute a JVM using the base archive (given by baseArchiveName) and the top archive
* (give by topArchiveName), using the command line (given by cmdLineSuffix).
*
* If baseArchiveName is null, use the JDK's default archive as the base archive.
*/
public static Result run2(String baseArchiveName, String topArchiveName, String ... cmdLineSuffix)
throws Exception {
if (baseArchiveName == null && topArchiveName == null) {
throw new RuntimeException("Both baseArchiveName and topArchiveName cannot be null at the same time.");
}
if (baseArchiveName == null && isUseSharedSpacesDisabled()) {
baseArchiveName = getTempBaseArchive();
}
String archiveFiles = (baseArchiveName == null) ? topArchiveName :
(topArchiveName == null) ? baseArchiveName :
baseArchiveName + File.pathSeparator + topArchiveName;
String[] cmdLine = TestCommon.concat(
autoMode ? "-Xshare:auto" : "-Xshare:on",
"-XX:SharedArchiveFile=" + archiveFiles);
cmdLine = TestCommon.concat(cmdLine, cmdLineSuffix);
return execProcess("exec", null, cmdLine);
}
public static Result runWithRelativePath(String baseArchiveName, String topArchiveName,
String jarDir, String ... cmdLineSuffix)
throws Exception {
if (baseArchiveName == null && topArchiveName == null) {
throw new RuntimeException("Both baseArchiveName and topArchiveName cannot be null at the same time.");
}
if (baseArchiveName == null && isUseSharedSpacesDisabled()) {
baseArchiveName = getTempBaseArchive();
}
String archiveFiles = (baseArchiveName == null) ? topArchiveName :
(topArchiveName == null) ? baseArchiveName :
baseArchiveName + File.pathSeparator + topArchiveName;
String[] cmdLine = TestCommon.concat(
autoMode ? "-Xshare:auto" : "-Xshare:on",
"-XX:SharedArchiveFile=" + archiveFiles);
cmdLine = TestCommon.concat(cmdLine, cmdLineSuffix);
return execProcess("exec", jarDir, cmdLine);
}
public static Result run2_WB(String baseArchiveName, String topArchiveName, String ... cmdLineSuffix)
throws Exception
{
return run2(baseArchiveName, topArchiveName,
TestCommon.concat(wbRuntimeArgs(), cmdLineSuffix));
}
/**
* A convenience method similar to run2, but always use the JDK's default archive
* as the base archive.
*
* Most dynamicArchive/*.java test cases should be using this method instead of run2.
*/
public static Result run(String topArchiveName, String ... cmdLineSuffix)
throws Exception
{
return run2(null, topArchiveName, cmdLineSuffix);
}
private static String getXshareMode(String[] cmdLine) {
for (int i = 0; i <= cmdLine.length - 1; i++) {
int j = cmdLine[i].indexOf("-Xshare:");
if (j != -1) {
return (cmdLine[i].substring(j));
}
}
return null;
}
private static Result execProcess(String mode, String jarDir, String[] cmdLine) throws Exception {
if (!executedIn_run) {
throw new Exception("Test error: dynamic archive tests must be executed via DynamicArchiveTestBase.run()");
}
ProcessBuilder pb = ProcessTools.createTestJavaProcessBuilder(cmdLine);
if (jarDir != null) {
pb.directory(new File(jarDir));
}
OutputAnalyzer output = TestCommon.executeAndLog(pb, mode);
CDSOptions opts = new CDSOptions();
String xShareMode = getXshareMode(cmdLine);
if (xShareMode != null) {
opts.setXShareMode(xShareMode);
}
return new Result(opts, output);
}
/**
* A convenience method for dumping and running, using the default CDS archive from the
* JDK. Both dumping and running should exit normally.
*/
public static void dumpAndRun(String topArchiveName, String ... cmdLineSuffix) throws Exception {
dump(topArchiveName, cmdLineSuffix).assertNormalExit();
run(topArchiveName, cmdLineSuffix).assertNormalExit();
}
private static String tempBaseArchive;
/**
* Return the name of a base archive.
* It will generate one if one doesn't exist.
*/
private static String getTempBaseArchive() throws Exception {
if (tempBaseArchive == null) {
tempBaseArchive = getNewArchiveName("tempBaseArchive");
TestCommon.dumpBaseArchive(tempBaseArchive, baseArchiveOptions);
}
return tempBaseArchive;
}
/**
* Return true if sharing has been disabled.
* By default, the VM will be started with -Xshare:auto.
* Sharing will be disabled by the VM if there's some problem
* in using the default CDS archive. It could happen under some
* situations such as follows:
* - the default CDS archive wasn't generated during build time because
* the JDK was built via cross-compilation on a different platform;
* - the VM under test was started with a different options than the ones
* when the default CDS archive was built. E.g. the VM was started with
* -XX:+UseZGC which implicitly disabled the UseCompressedOoops and the
* UseCompressedClassPointers options. Those "compressed" options were
* enabled when the default CDS archive was built.
*/
public static boolean isUseSharedSpacesDisabled() {
return !WB.isSharingEnabled();
}
}