2025-10-30 14:23:31 +00:00

103 lines
3.6 KiB
Java

/*
* Copyright (c) 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.
*/
import com.sun.tools.attach.VirtualMachine;
import com.sun.tools.attach.AgentLoadException;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.AfterAll;
import java.io.File;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.concurrent.TimeUnit;
import jdk.test.lib.dcmd.PidJcmdExecutor;
import jdk.test.lib.process.OutputAnalyzer;
import jdk.test.lib.process.ProcessTools;
import jdk.test.lib.Utils;
/*
* @test EarlyDynamicLoad
* @summary Test that dynamic attach fails gracefully when the JVM is not in live phase.
* @requires vm.jvmti
* @library /test/lib
* @run junit EarlyDynamicLoad
*/
public class EarlyDynamicLoad {
private static final String EXPECTED_MESSAGE = "Dynamic agent loading is only permitted in the live phase";
private static Process child;
@BeforeAll
static void startAndWaitChild() throws Exception {
child = ProcessTools.createTestJavaProcessBuilder(
"-XX:+StartAttachListener",
"-agentpath:" + Utils.TEST_NATIVE_PATH + File.separator + System.mapLibraryName("EarlyDynamicLoad"),
"--version").start();
// Wait until the process enters VMStartCallback
try (InputStream is = child.getInputStream()) {
is.read();
}
}
@AfterAll
static void stopChild() throws Exception {
try (OutputStream os = child.getOutputStream()) {
os.write(0);
}
if (!child.waitFor(5, TimeUnit.SECONDS)) {
child.destroyForcibly();
throw new AssertionError("Timed out while waiting child process to complete");
}
OutputAnalyzer analyzer = new OutputAnalyzer(child);
analyzer.shouldHaveExitValue(0);
}
@Test
public void virtualMachine() throws Exception {
try {
VirtualMachine vm = VirtualMachine.attach(String.valueOf(child.pid()));
vm.loadAgent("some.jar");
vm.detach();
throw new AssertionError("Should have failed with AgentLoadException");
} catch(AgentLoadException exception) {
if (!exception.getMessage().contains(EXPECTED_MESSAGE)) {
throw new AssertionError("Unexpected error message", exception);
}
}
}
@Test
public void jcmd() throws Exception {
PidJcmdExecutor executor = new PidJcmdExecutor(String.valueOf(child.pid()));
OutputAnalyzer out = executor.execute("JVMTI.agent_load some.jar");
out.shouldHaveExitValue(0);
out.stdoutShouldContain(EXPECTED_MESSAGE);
}
}