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:+PrintGCDetails",

    62                     "-XX:Flags=" + flagsFile.getAbsolutePath(),

    63                     "-cp", System.getProperty("test.class.path") + File.pathSeparator + getToolsJarPath(),

    64                     "CheckOrigin",

    65                     "-runtests");

    66 

    67             Map<String, String> env = pb.environment();

    68             env.put("_JAVA_OPTIONS", "-XX:+PrintOopAddress");

    69 

    70             pb.redirectOutput(ProcessBuilder.Redirect.INHERIT);

    71             pb.redirectError(ProcessBuilder.Redirect.INHERIT);

    72             Process p = pb.start();

    73             int exit = p.waitFor();

    74             System.out.println("sub process exit == " + exit);

    75             if (exit != 0) {

    76                 throw new Exception("Unexpected exit code from subprocess == " + exit);

    77             }

    78         } else {

    79             mbean =

    80                 ManagementFactory.getPlatformMXBean(HotSpotDiagnosticMXBean.class);

    81 

    82             // set a few more options

    83             mbean.setVMOption("HeapDumpOnOutOfMemoryError", "true");

    84             setOptionUsingAttach("HeapDumpPath", "/a/sample/path");

    85 

    86             // check the origin field for all the options we set

    87             checkOrigin("ManagementServer", Origin.DEFAULT);

    88             checkOrigin("PrintGCDetails", Origin.VM_CREATION);

    89             checkOrigin("PrintOopAddress", Origin.ENVIRON_VAR);

    90             checkOrigin("PrintSafepointStatistics", Origin.CONFIG_FILE);

    91             checkOrigin("HeapDumpOnOutOfMemoryError", Origin.MANAGEMENT);

    92             checkOrigin("NewSize", Origin.ERGONOMIC);

    93             checkOrigin("HeapDumpPath", Origin.ATTACH_ON_DEMAND);

    94         }

    95     }

    96 

    97     private static void checkOrigin(String option, Origin origin) throws Exception

    98     {

    99         Origin o = mbean.getVMOption(option).getOrigin();

   100         if (!o.equals(origin)) {

   101             throw new Exception("Option '" + option + "' should have origin '" + origin + "' but had '" + o + "'");

   102         }

   103         System.out.println("Option '" + option + "' verified origin = '" + origin + "'");

   104     }

   105 

   106     // use attach to set a manageable vm option

   107     private static void setOptionUsingAttach(String option, String value) throws Exception {

   108         HotSpotVirtualMachine vm = (HotSpotVirtualMachine) VirtualMachine.attach(ProcessTools.getProcessId()+"");

   109         InputStream in = vm.setFlag(option, value);

   110         System.out.println("Result from setting '" + option + "' to '" + value + "' using attach:");

   111         drain(vm, in);

   112         System.out.println("-- end -- ");

   113     }

   114 

   115     // Read the stream from the target VM until EOF, print to output, then detach

   116     private static void drain(VirtualMachine vm, InputStream in) throws Exception {

   117         byte b[] = new byte[256];

   118         int n;

   119         do {

   120             n = in.read(b);

   121             if (n > 0) {

   122                 String s = new String(b, 0, n, "UTF-8");

   123                 System.out.print(s);

   124             }

   125         } while (n > 0);

   126         in.close();

   127         vm.detach();

   128     }

   129 

   130     private static String getToolsJarPath() {

   131         return System.getProperty("java.home") +

   132             "/../lib/tools.jar".replace("/", File.separator);

   133     }

   134 }