1 /*

     2  * Copyright (c) 2014 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 com.sun.tools.attach.*;

    25 

    26 import java.nio.file.Files;

    27 import java.nio.file.Path;

    28 import java.util.Properties;

    29 import java.util.List;

    30 import java.io.File;

    31 

    32 import jdk.testlibrary.OutputAnalyzer;

    33 import jdk.testlibrary.ProcessTools;

    34 import jdk.testlibrary.ProcessThread;

    35 

    36 /*

    37  * @test

    38  * @bug 8033104

    39  * @summary Test to make sure attach and jvmstat works correctly when java.io.tmpdir is set

    40  * @library /lib/testlibrary

    41  * @run build Application Shutdown RunnerUtil

    42  * @run main/timeout=10 TempDirTest

    43  */

    44 

    45 public class TempDirTest {

    46 

    47     public static void main(String args[]) throws Throwable {

    48 

    49         Path clientTmpDir = Files.createTempDirectory("TempDirTest-client");

    50         clientTmpDir.toFile().deleteOnExit();

    51         Path targetTmpDir = Files.createTempDirectory("TempDirTest-target");

    52         targetTmpDir.toFile().deleteOnExit();

    53 

    54         // run the test with all possible combinations of setting java.io.tmpdir

    55         runExperiment(null, null);

    56         runExperiment(clientTmpDir, null);

    57         runExperiment(clientTmpDir, targetTmpDir);

    58         runExperiment(null, targetTmpDir);

    59 

    60     }

    61 

    62     private static int counter = 0;

    63 

    64     /*

    65      * The actual test is in the nested class TestMain.

    66      * The responsibility of this class is to:

    67      * 1. Start the Application class in a separate process.

    68      * 2. Find the pid and shutdown port of the running Application.

    69      * 3. Launches the tests in nested class TestMain that will attach to the Application.

    70      * 4. Shut down the Application.

    71      */

    72     public static void runExperiment(Path clientTmpDir, Path targetTmpDir) throws Throwable {

    73 

    74         System.out.print("### Running tests with overridden tmpdir for");

    75         System.out.print(" client: " + (clientTmpDir == null ? "no" : "yes"));

    76         System.out.print(" target: " + (targetTmpDir == null ? "no" : "yes"));

    77         System.out.println(" ###");

    78 

    79         final String pidFile = "TempDirTest.Application.pid-" + counter++;

    80         ProcessThread processThread = null;

    81         RunnerUtil.ProcessInfo info = null;

    82         try {

    83             String[] tmpDirArg = null;

    84             if (targetTmpDir != null) {

    85                 tmpDirArg = new String[] {"-Djava.io.tmpdir=" + targetTmpDir};

    86             }

    87             processThread = RunnerUtil.startApplication(pidFile, tmpDirArg);

    88             info = RunnerUtil.readProcessInfo(pidFile);

    89             launchTests(info.pid, clientTmpDir);

    90         } catch (Throwable t) {

    91             System.out.println("TempDirTest got unexpected exception: " + t);

    92             t.printStackTrace();

    93             throw t;

    94         } finally {

    95             // Make sure the Application process is stopped.

    96             RunnerUtil.stopApplication(info.shutdownPort, processThread);

    97         }

    98     }

    99 

   100     /**

   101      * Runs the actual tests in nested class TestMain.

   102      * The reason for running the tests in a separate process

   103      * is that we need to modify the class path and

   104      * the -Djava.io.tmpdir property.

   105      */

   106     private static void launchTests(int pid, Path clientTmpDir) throws Throwable {

   107         final String sep = File.separator;

   108 

   109         // Need to add jdk/lib/tools.jar to classpath.

   110         String classpath =

   111             System.getProperty("test.class.path", "") + File.pathSeparator +

   112             System.getProperty("test.jdk", ".") + sep + "lib" + sep + "tools.jar";

   113 

   114         String[] tmpDirArg = null;

   115         if (clientTmpDir != null) {

   116             tmpDirArg = new String [] {"-Djava.io.tmpdir=" + clientTmpDir};

   117         }

   118 

   119         // Arguments : [-Djava.io.tmpdir=] -classpath cp TempDirTest$TestMain pid

   120         String[] args = RunnerUtil.concat(

   121                 tmpDirArg,

   122                 new String[] {

   123                     "-classpath",

   124                     classpath,

   125                     "TempDirTest$TestMain",

   126                     Integer.toString(pid) });

   127         OutputAnalyzer output = ProcessTools.executeTestJvm(args);

   128         output.shouldHaveExitValue(0);

   129     }

   130 

   131     /**

   132      * This is the actual test. It will attach to the running Application

   133      * and perform a number of basic attach tests.

   134      */

   135     public static class TestMain {

   136         public static void main(String args[]) throws Exception {

   137             String pid = args[0];

   138 

   139             // Test 1 - list method should list the target VM

   140             System.out.println(" - Test: VirtualMachine.list");

   141             List<VirtualMachineDescriptor> l = VirtualMachine.list();

   142             boolean found = false;

   143             for (VirtualMachineDescriptor vmd: l) {

   144                 if (vmd.id().equals(pid)) {

   145                     found = true;

   146                     break;

   147                 }

   148             }

   149             if (found) {

   150                 System.out.println(" - " + pid + " found.");

   151             } else {

   152                 throw new RuntimeException(pid + " not found in VM list");

   153             }

   154 

   155             // Test 2 - try to attach and verify connection

   156 

   157             System.out.println(" - Attaching to application ...");

   158             VirtualMachine vm = VirtualMachine.attach(pid);

   159 

   160             System.out.println(" - Test: system properties in target VM");

   161             Properties props = vm.getSystemProperties();

   162             String value = props.getProperty("attach.test");

   163             if (value == null || !value.equals("true")) {

   164                 throw new RuntimeException("attach.test property not set");

   165             }

   166             System.out.println(" - attach.test property set as expected");

   167         }

   168     }

   169 }