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 import java.io.File;
25 import java.io.IOException;
26 import java.lang.reflect.Method;
27 import java.lang.reflect.Modifier;
28 import java.nio.file.FileSystem;
29 import java.nio.file.FileSystems;
30 import java.nio.file.Files;
31 import java.nio.file.Path;
32 import java.util.ArrayList;
33 import java.util.List;
34 import java.util.concurrent.TimeUnit;
35 import java.util.concurrent.atomic.AtomicReference;
36
37 /**
38 * @test
39 * @library /lib/testlibrary
40 * @bug 5016507 6173612 6319776 6342019 6484550 8004926
41 * @summary Start a managed VM and test that a management tool can connect
42 * without connection or username/password details.
43 * TestManager will attempt a connection to the address obtained from
44 * both agent properties and jvmstat buffer.
45 * @build jdk.testlibrary.ProcessTools
46 * @build jdk.testlibrary.Utils
47 * @build TestManager TestApplication
48 * @run main/othervm/timeout=300 -XX:+UsePerfData LocalManagementTest
49 */
50
51 import jdk.testlibrary.ProcessTools;
52 import jdk.testlibrary.Utils;
53
54 public class LocalManagementTest {
55 private static final String TEST_CLASSPATH = System.getProperty("test.class.path");
56 private static final String TEST_JDK = System.getProperty("test.jdk");
57 private static int MAX_GET_FREE_PORT_TRIES = 10;
58
59 public static void main(String[] args) throws Exception {
60 try {
61 MAX_GET_FREE_PORT_TRIES = Integer.parseInt(System.getProperty("test.getfreeport.max.tries", "10"));
62 } catch (NumberFormatException ex) {
63 }
64
65 int failures = 0;
66 for(Method m : LocalManagementTest.class.getDeclaredMethods()) {
67 if (Modifier.isStatic(m.getModifiers()) &&
68 m.getName().startsWith("test")) {
69 m.setAccessible(true);
70 try {
71 System.out.println(m.getName());
72 System.out.println("==========");
73 Boolean rslt = (Boolean)m.invoke(null);
74 if (!rslt) {
75 System.err.println(m.getName() + " failed");
76 failures++;
77 }
78 } catch (Exception e) {
79 e.printStackTrace();
80 failures++;
81 }
82 }
83 }
84 if (failures > 0) {
85 throw new Error("Test failed");
86 }
87 }
88
89 private static boolean test1() throws Exception {
90 return doTest("1", "-Dcom.sun.management.jmxremote");
91 }
92
93 private static boolean test2() throws Exception {
94 Path agentPath = findAgent();
95 if (agentPath != null) {
96 String agent = agentPath.toString();
97 return doTest("2", "-javaagent:" + agent);
98 } else {
99 return false;
100 }
101 }
102
103 /**
104 * no args (blank) - manager should attach and start agent
105 */
106 private static boolean test3() throws Exception {
107 return doTest("3", null);
108 }
109
110 /**
111 * sanity check arguments to management-agent.jar
112 */
113 private static boolean test4() throws Exception {
114 Path agentPath = findAgent();
115 if (agentPath != null) {
116
117 for (int i = 0; i < MAX_GET_FREE_PORT_TRIES; ++i) {
118 ProcessBuilder builder = ProcessTools.createJavaProcessBuilder(
119 "-javaagent:" + agentPath.toString() +
120 "=com.sun.management.jmxremote.port=" + Utils.getFreePort() + "," +
121 "com.sun.management.jmxremote.authenticate=false," +
122 "com.sun.management.jmxremote.ssl=false",
123 "-cp",
124 TEST_CLASSPATH,
125 "TestApplication",
126 "-exit"
127 );
128
129 Process prc = null;
130 final AtomicReference<Boolean> isBindExceptionThrown = new AtomicReference<>();
131 isBindExceptionThrown.set(new Boolean(false));
132 try {
133 prc = ProcessTools.startProcess(
134 "TestApplication",
135 builder,
136 (String line) -> {
137 if (line.contains("Exception thrown by the agent : " +
138 "java.rmi.server.ExportException: Port already in use")) {
139 isBindExceptionThrown.set(new Boolean(true));
140 }
141 });
142
143 prc.waitFor();
144
145 if (prc.exitValue() == 0) {
146 return true;
147 }
148
149 if (isBindExceptionThrown.get().booleanValue()) {
150 System.out.println("'Port already in use' error detected. Try again");
151 } else {
152 return false;
153 }
154 } finally {
155 if (prc != null) {
156 prc.destroy();
157 prc.waitFor();
158 }
159 }
160 }
161 }
162 return false;
163 }
164
165 /**
166 * use DNS-only name service
167 */
168 private static boolean test5() throws Exception {
169 return doTest("5", "-Dsun.net.spi.namservice.provider.1=\"dns,sun\"");
170 }
171
172 private static Path findAgent() {
173 FileSystem FS = FileSystems.getDefault();
174 Path agentPath = FS.getPath(
175 TEST_JDK, "jre", "lib", "management-agent.jar"
176 );
177 if (!isFileOk(agentPath)) {
178 agentPath = FS.getPath(
179 TEST_JDK, "lib", "management-agent.jar"
180 );
181 }
182 if (!isFileOk(agentPath)) {
183 System.err.println("Can not locate management-agent.jar");
184 return null;
185 }
186 return agentPath;
187 }
188
189 private static boolean isFileOk(Path path) {
190 return Files.isRegularFile(path) && Files.isReadable(path);
191 }
192
193 private static boolean doTest(String testId, String arg) throws Exception {
194 List<String> args = new ArrayList<>();
195 args.add("-cp");
196 args.add(TEST_CLASSPATH);
197
198 if (arg != null) {
199 args.add(arg);
200 }
201 args.add("TestApplication");
202 ProcessBuilder server = ProcessTools.createJavaProcessBuilder(
203 args.toArray(new String[args.size()])
204 );
205
206 Process serverPrc = null, clientPrc = null;
207 try {
208 final AtomicReference<String> port = new AtomicReference<>();
209 final AtomicReference<String> pid = new AtomicReference<>();
210
211 serverPrc = ProcessTools.startProcess(
212 "TestApplication(" + testId + ")",
213 server,
214 (String line) -> {
215 if (line.startsWith("port:")) {
216 port.set(line.split("\\:")[1]);
217 } else if (line.startsWith("pid:")) {
218 pid.set(line.split("\\:")[1]);
219 } else if (line.startsWith("waiting")) {
220 return true;
221 }
222 return false;
223 },
224 5,
225 TimeUnit.SECONDS
226 );
227
228 System.out.println("Attaching test manager:");
229 System.out.println("=========================");
230 System.out.println(" PID : " + pid.get());
231 System.out.println(" shutdown port : " + port.get());
232
233 ProcessBuilder client = ProcessTools.createJavaProcessBuilder(
234 "-cp",
235 TEST_CLASSPATH +
236 File.pathSeparator +
237 TEST_JDK +
238 File.separator +
239 "lib" +
240 File.separator +
241 "tools.jar",
242 "TestManager",
243 pid.get(),
244 port.get(),
245 "true"
246 );
247
248 clientPrc = ProcessTools.startProcess(
249 "TestManager",
250 client,
251 (String line) -> line.startsWith("Starting TestManager for PID"),
252 10,
253 TimeUnit.SECONDS
254 );
255
256 int clientExitCode = clientPrc.waitFor();
257 int serverExitCode = serverPrc.waitFor();
258 return clientExitCode == 0 && serverExitCode == 0;
259 } finally {
260 if (clientPrc != null) {
261 System.out.println("Stopping process " + clientPrc);
262 clientPrc.destroy();
263 clientPrc.waitFor();
264 }
265 if (serverPrc != null) {
266 System.out.println("Stopping process " + serverPrc);
267 serverPrc.destroy();
268 serverPrc.waitFor();
269 }
270 }
271 }
272 }