jdk-24/test/hotspot/jtreg/vmTestbase/nsk/jdi/VirtualMachine/redefineClasses/redefineclasses035.java
Leonid Mesnik 771f6a2f8e 8327704: Update nsk/jdi tests to use driver instead of othervm
Reviewed-by: sspitsyn, cjplummer
2024-03-21 22:35:25 +00:00

339 lines
12 KiB
Java

/*
* Copyright (c) 2002, 2024, 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.
*/
package nsk.jdi.VirtualMachine.redefineClasses;
import nsk.share.*;
import nsk.share.jpda.*;
import nsk.share.jdi.*;
import com.sun.jdi.*;
import com.sun.jdi.request.*;
import com.sun.jdi.event.*;
import java.util.*;
import java.io.*;
/**
* The test checks up <code>com.sun.jdi.VirtualMachine.redefineClasses()</code>
* in case when redefined class loaded by a custom loader.
*
* The test perform next steps: <br>
* 1. Debugee loads a tested class by the custom loader. <br>
* 2. When getting <code>ClassPrepareEvent</code>, debugger checks class
* loader of tested class to be the custom class loader and creates
* <code>MethodEntryRequest</code> for tested class and waits
* a corresponding event. <br>
* 3. Debugee calls a method of the tested class. <br>
* 4. After getting <code>MethodEntryEvent</code>, debugger redefines
* the tested class and checks the redefinition to be successful. <br>
*
* The above steps are performed for two cases, when the custom loader
* extends <code>java.net.URLClassLoader</code> and when the custom loader directly
* extends <code>java.lang.ClassLoader</code>.
*/
public class redefineclasses035 {
public final static String UNEXPECTED_STRING = "***Unexpected exception ";
private final static String prefix = "nsk.jdi.VirtualMachine.redefineClasses.";
private final static String debuggerName = prefix + "redefineclasses035";
public final static String debugeeName = debuggerName + "a";
public final static int TESTCASE_NUMBER = 2;
public final static String [] testedClassName = {
debuggerName + "b",
debuggerName + "c"
};
private final static String [] expectedClsLdrs = {
debugeeName + "ClassLoaderB",
debugeeName + "ClassLoaderC"
};
static final String redefinedMethodName = "justMethod";
private final static String PATH = "newclass" + File.separator
+ prefix.replace('.',File.separatorChar);
private final static String [] newClassFile = {
PATH + "redefineclasses035b.class",
PATH + "redefineclasses035c.class"
};
private static int exitStatus;
private static Log log;
private static Debugee debugee;
private static long waitTime;
private static String classDir;
private int expectedEventCount = 4;
private int currentCase = -1;
private ReferenceType testedClass, debugeeClass;
private static void display(String msg) {
log.display(msg);
}
private static void complain(String msg) {
log.complain(msg + "\n");
}
public static void main(String argv[]) {
int result = run(argv,System.out);
if (result != 0) {
throw new RuntimeException("TEST FAILED with result " + result);
}
}
public static int run(String argv[], PrintStream out) {
exitStatus = Consts.TEST_PASSED;
redefineclasses035 thisTest = new redefineclasses035();
ArgumentHandler argHandler = new ArgumentHandler(argv);
log = new Log(out, argHandler);
classDir = argv[0];
waitTime = argHandler.getWaitTime() * 60000;
Binder binder = new Binder(argHandler, log);
debugee = binder.bindToDebugee(debugeeName);
debugee.redirectOutput(log);
try {
thisTest.execTest();
} catch (Throwable e) {
exitStatus = Consts.TEST_FAILED;
e.printStackTrace();
} finally {
debugee.endDebugee();
}
display("Test finished. exitStatus = " + exitStatus);
return exitStatus;
}
private void execTest() throws Failure {
if (!debugee.VM().canRedefineClasses()) {
display("\n>>>canRedefineClasses() is false<<< test canceled.\n");
return;
}
display("\nTEST BEGINS");
display("===========");
EventSet eventSet = null;
EventIterator eventIterator = null;
Event event;
long totalTime = waitTime;
long tmp, begin = System.currentTimeMillis(),
delta = 0;
boolean exit = false;
int eventCount = 0;
EventRequestManager evm = debugee.getEventRequestManager();
ClassPrepareRequest req = evm.createClassPrepareRequest();
req.addClassFilter(debuggerName + "*");
req.addClassExclusionFilter(debugeeName + "Class*");
req.addClassExclusionFilter(debuggerName);
req.enable();
debugee.resume();
while (totalTime > 0 && !exit) {
if (eventIterator == null || !eventIterator.hasNext()) {
try {
eventSet = debugee.VM().eventQueue().remove(totalTime);
} catch (InterruptedException e) {
new Failure(e);
}
if (eventSet != null) {
eventIterator = eventSet.eventIterator();
} else {
eventIterator = null;
}
}
if (eventIterator != null) {
while (eventIterator.hasNext()) {
event = eventIterator.nextEvent();
display("\nevent ===>>> " + event);
if (event instanceof ClassPrepareEvent) {
testedClass = ((ClassPrepareEvent )event).referenceType();
String className = testedClass.name();
if (className.compareTo(debugeeName) == 0) {
prepareTestCases();
} else {
createMethodEntryRequest(currentCase);
}
debugee.resume();
} else if (event instanceof BreakpointEvent) {
currentCase++;
display("\ntestcase " + (currentCase + 1));
display("===========");
debugee.resume();
} else if (event instanceof MethodEntryEvent) {
checkCase((MethodEntryEvent )event);
eventCount++;
debugee.resume();
} else if (event instanceof VMDeathEvent) {
exit = true;
break;
} else if (event instanceof VMDisconnectEvent) {
exit = true;
break;
} // if
} // while
} // if
tmp = System.currentTimeMillis();
delta = tmp - begin;
totalTime -= delta;
begin = tmp;
}
if (totalTime <= 0) {
complain("out of wait time...");
exitStatus = Consts.TEST_FAILED;
}
if (eventCount != expectedEventCount) {
complain("expecting " + expectedEventCount
+ " events, but "
+ eventCount + " events arrived.");
exitStatus = Consts.TEST_FAILED;
}
display("=============");
display("TEST FINISHES\n");
}
private void checkCase(MethodEntryEvent event) {
Method mthd = getCurrentMethod(event.thread());
if (mthd.isConstructor()) {
display("it is constructor (skipped)");
return;
}
display("\nredefining...");
redefineDebugee(newClassFile[currentCase]);
mthd = getCurrentMethod(event.thread());
if (mthd.isObsolete()) {
display("!!!method is obsolete!!!");
display("redefinition is OK");
event.request().disable();
display("MethodEntryEvent is being disabled");
} else {
complain("method is not obsolete");
complain("redefinition was not OK");
exitStatus = Consts.TEST_FAILED;
}
display("");
return;
}
private MethodEntryRequest createMethodEntryRequest(int index) {
display("\nclasses info on debugger side:");
display("~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~");
display("testedClass\t:" + testedClassName[currentCase]);
String clsLdrName = testedClass.classLoader().referenceType().name();
String clsLdrName4Debugee = debugeeClass.classLoader().referenceType().name();
display("class loader\t:" + clsLdrName);
display("~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~");
if (clsLdrName.compareTo(expectedClsLdrs[currentCase]) != 0) {
complain("wrong class loader \"" + clsLdrName
+ "\"\n\tfor \"" + testedClassName[currentCase] + "\"");
exitStatus = Consts.TEST_FAILED;
}
EventRequestManager evm = debugee.getEventRequestManager();
MethodEntryRequest request = evm.createMethodEntryRequest();
request.addClassFilter(testedClassName[index]);
request.enable();
return request;
}
private void redefineDebugee(String classFileName) {
Map<? extends com.sun.jdi.ReferenceType,byte[]> mapBytes;
boolean alreadyComplained = false;
mapBytes = mapClassToBytes(classDir + File.separator + classFileName);
try {
debugee.VM().redefineClasses(mapBytes);
} catch (Exception e) {
throw new Failure(UNEXPECTED_STRING + e);
}
}
private Map<? extends com.sun.jdi.ReferenceType,byte[]> mapClassToBytes(String fileName) {
display("class-file\t:" + fileName);
File fileToBeRedefined = new File(fileName);
int fileToRedefineLength = (int )fileToBeRedefined.length();
byte[] arrayToRedefine = new byte[fileToRedefineLength];
FileInputStream inputFile;
try {
inputFile = new FileInputStream(fileToBeRedefined);
} catch (FileNotFoundException e) {
throw new Failure(UNEXPECTED_STRING + e);
}
try {
inputFile.read(arrayToRedefine);
inputFile.close();
} catch (IOException e) {
throw new Failure(UNEXPECTED_STRING + e);
}
HashMap<com.sun.jdi.ReferenceType,byte[]> mapForClass = new HashMap<com.sun.jdi.ReferenceType,byte[]>();
mapForClass.put(testedClass, arrayToRedefine);
return mapForClass;
}
private Method getCurrentMethod(ThreadReference thread) {
try {
Method mthd = thread.frame(0).location().method();
display("current method\t:" + mthd);
return mthd;
} catch (IncompatibleThreadStateException e) {
throw new Failure(UNEXPECTED_STRING + e);
}
}
private void prepareTestCases() {
debugeeClass = debugee.classByName(debugeeName);
display("debugeeClass\t\t:" + debugeeClass.name());
display("default class loader\t:"
+ debugeeClass.classLoader().referenceType().name());
display("setting breakpoint...");
debugee.setBreakpoint(testedClass,
redefineclasses035a.brkpMethodName,
redefineclasses035a.brkpLineNumber);
}
}