1 /*
2 * Copyright (c) 2013, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24 /*
25 * @test
26 * @bug 8028994
27 * @author Staffan Larsen
28 * @library /lib/testlibrary
29 * @run main CheckOrigin
30 */
31
32 import com.sun.management.HotSpotDiagnosticMXBean;
33 import com.sun.management.VMOption;
34 import com.sun.management.VMOption.Origin;
35 import com.sun.tools.attach.VirtualMachine;
36 import java.io.File;
37 import java.io.FileWriter;
38 import java.io.InputStream;
39 import java.io.PrintWriter;
40 import java.lang.management.ManagementFactory;
41 import java.util.Map;
42 import jdk.testlibrary.ProcessTools;
43 import sun.tools.attach.HotSpotVirtualMachine;
44
45 public class CheckOrigin {
46
47 private static HotSpotDiagnosticMXBean mbean;
48
49 public static void main(String... args) throws Exception {
50 if (args.length == 0) {
51 // start a process that has options set in a number of different ways
52
53 File flagsFile = File.createTempFile("CheckOriginFlags", null);
54 try (PrintWriter pw =
55 new PrintWriter(new FileWriter(flagsFile))) {
56 pw.println("+PrintSafepointStatistics");
57 }
58
59 ProcessBuilder pb = ProcessTools.
60 createJavaProcessBuilder(
61 "-XX:+UseConcMarkSweepGC", // this will cause UseParNewGC to be FLAG_SET_ERGO
62 "-XX:+PrintGCDetails",
63 "-XX:Flags=" + flagsFile.getAbsolutePath(),
64 "-cp", System.getProperty("test.class.path") + File.pathSeparator + getToolsJarPath(),
65 "CheckOrigin",
66 "-runtests");
67
68 Map<String, String> env = pb.environment();
69 env.put("_JAVA_OPTIONS", "-XX:+PrintOopAddress");
70
71 pb.redirectOutput(ProcessBuilder.Redirect.INHERIT);
72 pb.redirectError(ProcessBuilder.Redirect.INHERIT);
73 Process p = pb.start();
74 int exit = p.waitFor();
75 System.out.println("sub process exit == " + exit);
76 if (exit != 0) {
77 throw new Exception("Unexpected exit code from subprocess == " + exit);
78 }
79 } else {
80 mbean =
81 ManagementFactory.getPlatformMXBean(HotSpotDiagnosticMXBean.class);
82
83 // set a few more options
84 mbean.setVMOption("HeapDumpOnOutOfMemoryError", "true");
85 setOptionUsingAttach("HeapDumpPath", "/a/sample/path");
86
87 // check the origin field for all the options we set
88
89 // Not set, so should be default
90 checkOrigin("ManagementServer", Origin.DEFAULT);
91 // Set on the command line
92 checkOrigin("PrintGCDetails", Origin.VM_CREATION);
93 // Set in _JAVA_OPTIONS
94 checkOrigin("PrintOopAddress", Origin.ENVIRON_VAR);
95 // Set in -XX:Flags file
96 checkOrigin("PrintSafepointStatistics", Origin.CONFIG_FILE);
97 // Set through j.l.m
98 checkOrigin("HeapDumpOnOutOfMemoryError", Origin.MANAGEMENT);
99 // Should be set by the VM, when we set UseConcMarkSweepGC
100 checkOrigin("UseParNewGC", Origin.ERGONOMIC);
101 // Set using attach
102 checkOrigin("HeapDumpPath", Origin.ATTACH_ON_DEMAND);
103 }
104 }
105
106 private static void checkOrigin(String option, Origin origin) throws Exception
107 {
108 Origin o = mbean.getVMOption(option).getOrigin();
109 if (!o.equals(origin)) {
110 throw new Exception("Option '" + option + "' should have origin '" + origin + "' but had '" + o + "'");
111 }
112 System.out.println("Option '" + option + "' verified origin = '" + origin + "'");
113 }
114
115 // use attach to set a manageable vm option
116 private static void setOptionUsingAttach(String option, String value) throws Exception {
117 HotSpotVirtualMachine vm = (HotSpotVirtualMachine) VirtualMachine.attach(ProcessTools.getProcessId()+"");
118 InputStream in = vm.setFlag(option, value);
119 System.out.println("Result from setting '" + option + "' to '" + value + "' using attach:");
120 drain(vm, in);
121 System.out.println("-- end -- ");
122 }
123
124 // Read the stream from the target VM until EOF, print to output, then detach
125 private static void drain(VirtualMachine vm, InputStream in) throws Exception {
126 byte b[] = new byte[256];
127 int n;
128 do {
129 n = in.read(b);
130 if (n > 0) {
131 String s = new String(b, 0, n, "UTF-8");
132 System.out.print(s);
133 }
134 } while (n > 0);
135 in.close();
136 vm.detach();
137 }
138
139 private static String getToolsJarPath() {
140 return System.getProperty("java.home") +
141 "/../lib/tools.jar".replace("/", File.separator);
142 }
143 }