package org.graalvm.compiler.hotspot.test;
import static org.graalvm.compiler.hotspot.test.HotSpotGraalManagementTest.JunitShield.findAttributeInfo;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import java.io.File;
import java.io.IOException;
import java.lang.management.ManagementFactory;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import javax.management.Attribute;
import javax.management.AttributeList;
import javax.management.AttributeNotFoundException;
import javax.management.InvalidAttributeValueException;
import javax.management.MBeanAttributeInfo;
import javax.management.MBeanInfo;
import javax.management.MBeanOperationInfo;
import javax.management.MBeanServer;
import javax.management.MBeanServerFactory;
import javax.management.ObjectInstance;
import javax.management.ObjectName;
import org.graalvm.compiler.api.test.Graal;
import org.graalvm.compiler.hotspot.HotSpotGraalManagementRegistration;
import org.graalvm.compiler.hotspot.HotSpotGraalRuntime;
import org.graalvm.compiler.options.EnumOptionKey;
import org.graalvm.compiler.options.NestedBooleanOptionKey;
import org.graalvm.compiler.options.OptionDescriptor;
import org.graalvm.compiler.options.OptionDescriptors;
import org.graalvm.compiler.options.OptionKey;
import org.graalvm.compiler.options.OptionsParser;
import org.junit.Assert;
import org.junit.AssumptionViolatedException;
import org.junit.Test;
public class HotSpotGraalManagementTest {
private static final boolean DEBUG = Boolean.getBoolean(HotSpotGraalManagementTest.class.getSimpleName() + ".debug");
public HotSpotGraalManagementTest() {
try {
MBeanServerFactory.findMBeanServer(null);
} catch (NoClassDefFoundError e) {
throw new AssumptionViolatedException("Management classes/module(s) not available: " + e);
}
}
@Test
public void registration() throws Exception {
HotSpotGraalRuntime runtime = (HotSpotGraalRuntime) Graal.getRuntime();
HotSpotGraalManagementRegistration management = runtime.getManagement();
if (management == null) {
return;
}
MBeanServer server = ManagementFactory.getPlatformMBeanServer();
ObjectName name;
assertNotNull("Now the bean thinks it is registered", name = (ObjectName) management.poll(true));
assertNotNull("And the bean is found", server.getObjectInstance(name));
}
@Test
public void readBeanInfo() throws Exception {
assertNotNull("Server is started", ManagementFactory.getPlatformMBeanServer());
HotSpotGraalRuntime runtime = (HotSpotGraalRuntime) Graal.getRuntime();
HotSpotGraalManagementRegistration management = runtime.getManagement();
if (management == null) {
return;
}
ObjectName mbeanName;
assertNotNull("Bean is registered", mbeanName = (ObjectName) management.poll(true));
final MBeanServer server = ManagementFactory.getPlatformMBeanServer();
ObjectInstance bean = server.getObjectInstance(mbeanName);
assertNotNull("Bean is registered", bean);
MBeanInfo info = server.getMBeanInfo(mbeanName);
assertNotNull("Info is found", info);
AttributeList originalValues = new AttributeList();
AttributeList newValues = new AttributeList();
for (OptionDescriptors set : OptionsParser.getOptionsLoader()) {
for (OptionDescriptor option : set) {
JunitShield.testOption(info, mbeanName, server, runtime, option, newValues, originalValues);
}
}
String[] attributeNames = new String[originalValues.size()];
for (int i = 0; i < attributeNames.length; i++) {
attributeNames[i] = ((Attribute) originalValues.get(i)).getName();
}
AttributeList actualValues = server.getAttributes(mbeanName, attributeNames);
assertEquals(originalValues.size(), actualValues.size());
for (int i = 0; i < attributeNames.length; i++) {
Object expect = String.valueOf(((Attribute) originalValues.get(i)).getValue());
Object actual = String.valueOf(((Attribute) actualValues.get(i)).getValue());
assertEquals(attributeNames[i], expect, actual);
}
try {
server.setAttributes(mbeanName, newValues);
} finally {
server.setAttributes(mbeanName, originalValues);
}
}
static class JunitShield {
static void testOption(MBeanInfo mbeanInfo,
ObjectName mbeanName,
MBeanServer server,
HotSpotGraalRuntime runtime,
OptionDescriptor option,
AttributeList newValues,
AttributeList originalValues) throws Exception {
OptionKey<?> optionKey = option.getOptionKey();
Object currentValue = optionKey.getValue(runtime.getOptions());
Class<?> optionType = option.getOptionValueType();
String name = option.getName();
if (DEBUG) {
System.out.println("Testing option " + name);
}
MBeanAttributeInfo attrInfo = findAttributeInfo(name, mbeanInfo);
assertNotNull("Attribute not found for option " + name, attrInfo);
String expectAttributeValue = stringValue(currentValue, option.getOptionValueType() == String.class);
Object actualAttributeValue = server.getAttribute(mbeanName, name);
assertEquals(expectAttributeValue, actualAttributeValue);
Map<String, String> legalValues = new HashMap<>();
List<String> illegalValues = new ArrayList<>();
if (optionKey instanceof EnumOptionKey) {
EnumOptionKey<?> enumOptionKey = (EnumOptionKey<?>) optionKey;
for (Object obj : enumOptionKey.getAllValues()) {
if (obj != currentValue) {
legalValues.put(obj.toString(), obj.toString());
}
}
illegalValues.add(String.valueOf(42));
} else if (optionType == Boolean.class) {
Object defaultValue;
if (optionKey instanceof NestedBooleanOptionKey) {
NestedBooleanOptionKey nbok = (NestedBooleanOptionKey) optionKey;
defaultValue = nbok.getMasterOption().getValue(runtime.getOptions());
} else {
defaultValue = optionKey.getDefaultValue();
}
legalValues.put("", unquotedStringValue(defaultValue));
illegalValues.add(String.valueOf(42));
illegalValues.add("true");
illegalValues.add("false");
} else if (optionType == String.class) {
legalValues.put("", quotedStringValue(optionKey.getDefaultValue()));
legalValues.put("\"" + currentValue + "Prime\"", "\"" + currentValue + "Prime\"");
legalValues.put("\"quoted string\"", "\"quoted string\"");
illegalValues.add("\"unbalanced quotes");
illegalValues.add("\"");
illegalValues.add("non quoted string");
} else if (optionType == Float.class) {
legalValues.put("", unquotedStringValue(optionKey.getDefaultValue()));
String value = unquotedStringValue(currentValue == null ? 33F : ((float) currentValue) + 11F);
legalValues.put(value, value);
illegalValues.add("string");
} else if (optionType == Double.class) {
legalValues.put("", unquotedStringValue(optionKey.getDefaultValue()));
String value = unquotedStringValue(currentValue == null ? 33D : ((double) currentValue) + 11D);
legalValues.put(value, value);
illegalValues.add("string");
} else if (optionType == Integer.class) {
legalValues.put("", unquotedStringValue(optionKey.getDefaultValue()));
String value = unquotedStringValue(currentValue == null ? 33 : ((int) currentValue) + 11);
legalValues.put(value, value);
illegalValues.add("42.42");
illegalValues.add("string");
} else if (optionType == Long.class) {
legalValues.put("", unquotedStringValue(optionKey.getDefaultValue()));
String value = unquotedStringValue(currentValue == null ? 33L : ((long) currentValue) + 11L);
legalValues.put(value, value);
illegalValues.add("42.42");
illegalValues.add("string");
}
Attribute originalAttributeValue = new Attribute(name, expectAttributeValue);
try {
for (Map.Entry<String, String> e : legalValues.entrySet()) {
String legalValue = e.getKey();
if (DEBUG) {
System.out.printf("Changing %s from %s to %s%n", name, currentValue, legalValue);
}
Attribute newAttributeValue = new Attribute(name, legalValue);
newValues.add(newAttributeValue);
server.setAttribute(mbeanName, newAttributeValue);
Object actual = optionKey.getValue(runtime.getOptions());
actual = server.getAttribute(mbeanName, name);
String expectValue = e.getValue();
if (option.getOptionValueType() == String.class && expectValue == null) {
expectValue = "";
} else if (option.getOptionKey() instanceof NestedBooleanOptionKey && null == expectValue) {
NestedBooleanOptionKey nbok = (NestedBooleanOptionKey) option.getOptionKey();
expectValue = String.valueOf(nbok.getValue(runtime.getOptions()));
actual = server.getAttribute(mbeanName, name);
}
assertEquals(expectValue, actual);
}
} finally {
if (DEBUG) {
System.out.printf("Resetting %s to %s%n", name, currentValue);
}
originalValues.add(originalAttributeValue);
server.setAttribute(mbeanName, originalAttributeValue);
}
try {
for (Object illegalValue : illegalValues) {
if (DEBUG) {
System.out.printf("Changing %s from %s to illegal value %s%n", name, currentValue, illegalValue);
}
server.setAttribute(mbeanName, new Attribute(name, illegalValue));
Assert.fail("Expected setting " + name + " to " + illegalValue + " to fail");
}
} catch (InvalidAttributeValueException e) {
} finally {
if (DEBUG) {
System.out.printf("Resetting %s to %s%n", name, currentValue);
}
server.setAttribute(mbeanName, originalAttributeValue);
}
try {
String unknownOptionName = "definitely not an option name";
server.setAttribute(mbeanName, new Attribute(unknownOptionName, ""));
Assert.fail("Expected setting option with name \"" + unknownOptionName + "\" to fail");
} catch (AttributeNotFoundException e) {
}
}
static MBeanAttributeInfo findAttributeInfo(String attrName, MBeanInfo info) {
for (MBeanAttributeInfo attr : info.getAttributes()) {
if (attr.getName().equals(attrName)) {
assertTrue("Readable", attr.isReadable());
assertTrue("Writable", attr.isWritable());
return attr;
}
}
return null;
}
}
private static String quotedStringValue(Object optionValue) {
return stringValue(optionValue, true);
}
private static String unquotedStringValue(Object optionValue) {
return stringValue(optionValue, false);
}
private static String stringValue(Object optionValue, boolean withQuoting) {
if (optionValue == null) {
return "";
}
if (withQuoting) {
return "\"" + optionValue + "\"";
}
return String.valueOf(optionValue);
}
private static String quoted(Object s) {
return "\"" + s + "\"";
}
@Test
public void publicJmxApiOfGraalDumpOperation() throws Exception {
assertNotNull("Server is started", ManagementFactory.getPlatformMBeanServer());
HotSpotGraalRuntime runtime = (HotSpotGraalRuntime) Graal.getRuntime();
HotSpotGraalManagementRegistration management = runtime.getManagement();
if (management == null) {
return;
}
ObjectName mbeanName;
assertNotNull("Bean is registered", mbeanName = (ObjectName) management.poll(true));
final MBeanServer server = ManagementFactory.getPlatformMBeanServer();
assertEquals("Domain name is used to lookup the beans by VisualVM", "org.graalvm.compiler.hotspot", mbeanName.getDomain());
assertEquals("type can be used to identify the Graal bean", "HotSpotGraalRuntime_VM", mbeanName.getKeyProperty("type"));
ObjectInstance bean = server.getObjectInstance(mbeanName);
assertNotNull("Bean is registered", bean);
MBeanInfo info = server.getMBeanInfo(mbeanName);
assertNotNull("Info is found", info);
final MBeanOperationInfo[] arr = info.getOperations();
assertEquals("Currently three overloads", 3, arr.length);
MBeanOperationInfo dumpOp = null;
for (int i = 0; i < arr.length; i++) {
assertEquals("dumpMethod", arr[i].getName());
if (arr[i].getSignature().length == 3) {
dumpOp = arr[i];
}
}
assertNotNull("three args variant (as used by VisualVM) found", dumpOp);
MBeanAttributeInfo dumpPath = findAttributeInfo("DumpPath", info);
MBeanAttributeInfo printGraphFile = findAttributeInfo("PrintGraphFile", info);
MBeanAttributeInfo showDumpFiles = findAttributeInfo("ShowDumpFiles", info);
Object originalDumpPath = server.getAttribute(mbeanName, dumpPath.getName());
Object originalPrintGraphFile = server.getAttribute(mbeanName, printGraphFile.getName());
Object originalShowDumpFiles = server.getAttribute(mbeanName, showDumpFiles.getName());
final File tmpDir = new File(HotSpotGraalManagementTest.class.getSimpleName() + "_" + System.currentTimeMillis()).getAbsoluteFile();
server.setAttribute(mbeanName, new Attribute(dumpPath.getName(), quoted(tmpDir)));
server.setAttribute(mbeanName, new Attribute(printGraphFile.getName(), true));
server.setAttribute(mbeanName, new Attribute(showDumpFiles.getName(), false));
Object[] params = {"java.util.Arrays", "asList", ":3"};
try {
server.invoke(mbeanName, "dumpMethod", params, null);
boolean found = false;
String expectedIgvDumpSuffix = "[Arrays.asList(Object[])List].bgv";
List<String> dumpPathEntries = Arrays.asList(tmpDir.list());
for (String entry : dumpPathEntries) {
if (entry.endsWith(expectedIgvDumpSuffix)) {
found = true;
}
}
if (!found) {
Assert.fail(String.format("Expected file ending with \"%s\" in %s but only found:%n%s", expectedIgvDumpSuffix, tmpDir,
dumpPathEntries.stream().collect(Collectors.joining(System.lineSeparator()))));
}
} finally {
deleteDirectory(tmpDir.toPath());
server.setAttribute(mbeanName, new Attribute(dumpPath.getName(), originalDumpPath));
server.setAttribute(mbeanName, new Attribute(printGraphFile.getName(), originalPrintGraphFile));
server.setAttribute(mbeanName, new Attribute(showDumpFiles.getName(), originalShowDumpFiles));
}
}
static void deleteDirectory(Path toDelete) throws IOException {
Files.walk(toDelete).sorted(Comparator.reverseOrder()).map(Path::toFile).forEach(File::delete);
}
}