1 /*
   2  * Copyright (c) 2015, Red Hat Inc
   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.net.InetAddress;
  26 import java.net.NetworkInterface;
  27 import java.net.SocketException;
  28 import java.util.ArrayList;
  29 import java.util.Collections;
  30 import java.util.List;
  31 import java.util.concurrent.CountDownLatch;
  32 import java.util.concurrent.TimeUnit;
  33 import java.util.function.Predicate;
  34 
  35 import jdk.testlibrary.ProcessTools;
  36 
  37 /**
  38  * NOTE:
  39  *    This test requires at least a setup similar to the following in
  40  *    /etc/hosts file (or the windows equivalent). I.e. it expects it to
  41  *    be multi-homed and not both being the loop-back interface.
  42  *    For example:
  43  *    ----->8-------- /etc/hosts ----------->8---
  44  *    127.0.0.1   localhost
  45  *    192.168.0.1 localhost
  46  *    ----->8-------- /etc/hosts ----------->8---
  47  *
  48  * @test
  49  * @bug     6425769
  50  * @summary Test JMX agent host address binding. Same ports but different
  51  *          interfaces to bind to (using plain sockets and SSL sockets).
  52  *
  53  * @modules java.management/sun.management
  54  *          java.management/sun.management.jmxremote
  55  * @library /lib/testlibrary
  56  * @build jdk.testlibrary.* JMXAgentInterfaceBinding
  57  * @run main/timeout=5 JMXInterfaceBindingTest
  58  */
  59 public class JMXInterfaceBindingTest {
  60 
  61     public static final int COMMUNICATION_ERROR_EXIT_VAL = 1;
  62     public static final int STOP_PROCESS_EXIT_VAL = 137;
  63     public static final int JMX_PORT = 9111;
  64     public static final int RMI_PORT = 9112;
  65     public static final String READY_MSG = "MainThread: Ready for connections";
  66     public static final String TEST_CLASS = JMXAgentInterfaceBinding.class.getSimpleName();
  67     public static final String KEYSTORE_LOC = System.getProperty("test.src", ".") +
  68                                               File.separator +
  69                                               "ssl" +
  70                                               File.separator +
  71                                               "keystore";
  72     public static final String TRUSTSTORE_LOC = System.getProperty("test.src", ".") +
  73                                                 File.separator +
  74                                                 "ssl" +
  75                                                 File.separator +
  76                                                 "truststore";
  77     public static final String TEST_CLASSPATH = System.getProperty("test.classes", ".");
  78 
  79     public void run(List<InetAddress> addrs) {
  80         System.out.println("DEBUG: Running tests with plain sockets.");
  81         runTests(addrs, false);
  82         System.out.println("DEBUG: Running tests with SSL sockets.");
  83         runTests(addrs, true);
  84     }
  85 
  86     private void runTests(List<InetAddress> addrs, boolean useSSL) {
  87         TestProcessThread[] jvms = new TestProcessThread[addrs.size()];
  88         for (int i = 0; i < addrs.size(); i++) {
  89             String addr = JMXAgentInterfaceBinding.wrapAddress(addrs.get(i).getHostAddress());
  90             System.out.println();
  91             String msg = String.format("DEBUG: Launching java tester for triplet (HOSTNAME,JMX_PORT,RMI_PORT) == (%s,%d,%d)",
  92                     addr,
  93                     JMX_PORT,
  94                     RMI_PORT);
  95             System.out.println(msg);
  96             jvms[i] = runJMXBindingTest(addr, useSSL);
  97             jvms[i].start();
  98             System.out.println("DEBUG: Started " + (i + 1) + " Process(es).");
  99         }
 100         int failedProcesses = 0;
 101         for (TestProcessThread pt: jvms) {
 102             try {
 103                 pt.stopProcess();
 104                 pt.join();
 105             } catch (InterruptedException e) {
 106                 System.err.println("Failed to stop process: " + pt.getName());
 107                 throw new RuntimeException("Test failed", e);
 108             }
 109             int exitValue = pt.getExitValue();
 110             // If there is a communication error (the case we care about)
 111             // we get a exit code of 1
 112             if (exitValue == COMMUNICATION_ERROR_EXIT_VAL) {
 113                 // Failure case since the java processes should still be
 114                 // running.
 115                 System.err.println("Test FAILURE on " + pt.getName());
 116                 failedProcesses++;
 117             } else if (exitValue == STOP_PROCESS_EXIT_VAL) {
 118                 System.out.println("DEBUG: OK. Spawned java process terminated with expected exit code of " + STOP_PROCESS_EXIT_VAL);
 119             } else {
 120                 System.err.println("Test FAILURE on " + pt.getName() + " reason: Unexpected exit code => " + exitValue);
 121                 failedProcesses++;
 122             }
 123         }
 124         if (failedProcesses > 0) {
 125             throw new RuntimeException("Test FAILED. " + failedProcesses + " out of " + addrs.size() + " process(es) failed to start the JMX agent.");
 126         }
 127     }
 128 
 129     private TestProcessThread runJMXBindingTest(String address, boolean useSSL) {
 130         List<String> args = new ArrayList<>();
 131         args.add("-classpath");
 132         args.add(TEST_CLASSPATH);
 133         args.add("-Dcom.sun.management.jmxremote.host=" + address);
 134         args.add("-Dcom.sun.management.jmxremote.port=" + JMX_PORT);
 135         args.add("-Dcom.sun.management.jmxremote.rmi.port=" + RMI_PORT);
 136         args.add("-Dcom.sun.management.jmxremote.authenticate=false");
 137         args.add("-Dcom.sun.management.jmxremote.ssl=" + Boolean.toString(useSSL));
 138         if (useSSL) {
 139             args.add("-Dcom.sun.management.jmxremote.registry.ssl=true");
 140             args.add("-Djavax.net.ssl.keyStore=" + KEYSTORE_LOC);
 141             args.add("-Djavax.net.ssl.trustStore=" + TRUSTSTORE_LOC);
 142             args.add("-Djavax.net.ssl.keyStorePassword=password");
 143             args.add("-Djavax.net.ssl.trustStorePassword=trustword");
 144         }
 145         args.add(TEST_CLASS);
 146         args.add(address);
 147         args.add(Integer.toString(JMX_PORT));
 148         args.add(Integer.toString(RMI_PORT));
 149         args.add(Boolean.toString(useSSL));
 150         try {
 151             ProcessBuilder builder = ProcessTools.createJavaProcessBuilder(args.toArray(new String[] {}));
 152             System.out.println(ProcessTools.getCommandLine(builder));
 153             TestProcessThread jvm = new TestProcessThread("JMX-Tester-" + address, JMXInterfaceBindingTest::isJMXAgentResponseAvailable, builder);
 154             return jvm;
 155         } catch (Exception e) {
 156             throw new RuntimeException("Test failed", e);
 157         }
 158 
 159     }
 160 
 161     private static boolean isJMXAgentResponseAvailable(String line) {
 162         if (line.equals(READY_MSG)) {
 163             System.out.println("DEBUG: Found expected READY_MSG.");
 164             return true;
 165         } else if (line.startsWith("Error:")) {
 166             // Allow for a JVM process that exits with
 167             // "Error: JMX connector server communication error: ..."
 168             // to continue as well since we handle that case elsewhere.
 169             // This has the effect that the test does not timeout and
 170             // fails with an exception in the test.
 171             System.err.println("PROBLEM: JMX agent of target JVM did not start as it should.");
 172             return true;
 173         } else {
 174             return false;
 175         }
 176     }
 177 
 178     public static void main(String[] args) {
 179         List<InetAddress> addrs = getAddressesForLocalHost();
 180         if (addrs.size() < 2) {
 181             System.out.println("Ignoring manual test since no more than one IPs are configured for 'localhost'");
 182             return;
 183         }
 184         JMXInterfaceBindingTest test = new JMXInterfaceBindingTest();
 185         test.run(addrs);
 186         System.out.println("All tests PASSED.");
 187     }
 188 
 189     private static List<InetAddress> getAddressesForLocalHost() {
 190         try {
 191             List<InetAddress> filtered = new ArrayList<>();
 192             for (NetworkInterface iface: Collections.list(NetworkInterface.getNetworkInterfaces())) {
 193                 for (InetAddress addr: Collections.list(iface.getInetAddresses())) {
 194                     if (isNonloopbackLocalhost(addr)) {
 195                         filtered.add(addr);
 196                     }
 197                 }
 198             }
 199             return filtered;
 200         } catch (SocketException e) {
 201             throw new RuntimeException("Test failed", e);
 202         }
 203     }
 204 
 205     // we need 'real' localhost addresses only (eg. not loopback ones)
 206     // so we can bind the remote JMX connector to them
 207     private static boolean isNonloopbackLocalhost(InetAddress i) {
 208         if (!i.isLoopbackAddress()) {
 209             return i.getHostName().toLowerCase().equals("localhost");
 210         }
 211         return false;
 212     }
 213 
 214     private static class TestProcessThread extends Thread {
 215 
 216         private final Predicate<String> predicate;
 217         private final ProcessBuilder pb;
 218         private final CountDownLatch latch;
 219         private Process process;
 220 
 221         public TestProcessThread(String threadName, Predicate<String> predicate, ProcessBuilder pb) {
 222             super(threadName);
 223             this.predicate = predicate;
 224             this.pb = pb;
 225             this.latch = new CountDownLatch(1);
 226         }
 227 
 228         @Override
 229         public void run() {
 230             try {
 231                 process = ProcessTools.startProcess(getName(), pb, predicate, 10, TimeUnit.SECONDS);
 232                 latch.countDown();
 233                 process.waitFor();
 234             } catch (Exception e) {
 235                 throw new RuntimeException("Test failed", e);
 236             }
 237         }
 238 
 239         public void stopProcess() {
 240             try {
 241                 latch.await();
 242             } catch (InterruptedException e1) {
 243                 throw new RuntimeException("Test failed", e1);
 244             }
 245             if (process != null) {
 246                 process.destroyForcibly();
 247                 try {
 248                     process.waitFor();
 249                 } catch (InterruptedException e) {
 250                     throw new RuntimeException("Test failed", e);
 251                 }
 252             }
 253         }
 254 
 255         public int getExitValue() {
 256             return process.exitValue();
 257         }
 258     }
 259 }