1 /*
   2  * Copyright (c) 2013, 2016, 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.BufferedReader;
  25 import java.io.File;
  26 import java.io.FileNotFoundException;
  27 import java.io.FileReader;
  28 import java.io.IOException;
  29 import java.io.Reader;
  30 import java.nio.CharBuffer;
  31 import java.util.Arrays;
  32 import java.util.Scanner;
  33 
  34 import jdk.test.lib.Asserts;
  35 import jdk.test.lib.JDKToolFinder;
  36 import jdk.test.lib.JDKToolLauncher;
  37 import jdk.test.lib.OutputAnalyzer;
  38 import jdk.test.lib.Platform;
  39 import jdk.test.lib.ProcessTools;
  40 
  41 /*
  42  * @test
  43  * @bug 6313383
  44  * @key regression
  45  * @summary Regression test for hprof export issue due to large heaps (>2G)
  46  * @library /testlibrary
  47  * @modules java.base/jdk.internal.misc
  48  *          java.compiler
  49  *          java.management/sun.management
  50  *          jdk.jvmstat/sun.jvmstat.monitor
  51  * @build jdk.test.lib.* JMapHProfLargeHeapProc
  52  * @run main JMapHProfLargeHeapTest
  53  */
  54 
  55 public class JMapHProfLargeHeapTest {
  56     private static final String HEAP_DUMP_FILE_NAME = "heap.bin";
  57     private static final String HPROF_HEADER_1_0_1 = "JAVA PROFILE 1.0.1";
  58     private static final String HPROF_HEADER_1_0_2 = "JAVA PROFILE 1.0.2";
  59     private static final long M = 1024L;
  60     private static final long G = 1024L * M;
  61 
  62     public static void main(String[] args) throws Exception {
  63         if (!Platform.shouldSAAttach()) {
  64             System.out.println("SA attach not expected to work - test skipped.");
  65             return;
  66         }
  67 
  68         // All heap dumps should create 1.0.2 file format
  69         // Hotspot internal heapdumper always use HPROF_HEADER_1_0_2 format,
  70         // but SA heapdumper still use HPROF_HEADER_1_0_1 for small heaps
  71         testHProfFileFormat("-Xmx1g", 22 * M, HPROF_HEADER_1_0_1);
  72 
  73         /**
  74          * This test was deliberately commented out since the test system lacks
  75          * support to handle the requirements for this kind of heap size in a
  76          * good way. If or when it becomes possible to run this kind of tests in
  77          * the test environment the test should be enabled again.
  78          * */
  79         // Large heap 2,2 gigabytes, should create 1.0.2 file format
  80         // testHProfFileFormat("-Xmx4g", 2 * G + 2 * M, HPROF_HEADER_1_0_2);
  81     }
  82 
  83     private static void testHProfFileFormat(String vmArgs, long heapSize,
  84             String expectedFormat) throws Exception, IOException,
  85             InterruptedException, FileNotFoundException {
  86         ProcessBuilder procBuilder = ProcessTools.createJavaProcessBuilder(
  87                 "-XaddExports:java.management/sun.management=ALL-UNNAMED", vmArgs, "JMapHProfLargeHeapProc", String.valueOf(heapSize));
  88         procBuilder.redirectError(ProcessBuilder.Redirect.INHERIT);
  89         Process largeHeapProc = procBuilder.start();
  90 
  91         try (Scanner largeHeapScanner = new Scanner(
  92                 largeHeapProc.getInputStream());) {
  93             String pidstring = null;
  94             while ((pidstring = largeHeapScanner.findInLine("PID\\[[0-9].*\\]")) == null) {
  95                 Thread.sleep(500);
  96             }
  97             int pid = Integer.parseInt(pidstring.substring(4,
  98                     pidstring.length() - 1));
  99             System.out.println("Extracted pid: " + pid);
 100 
 101             JDKToolLauncher jMapLauncher = JDKToolLauncher
 102                     .createUsingTestJDK("jhsdb");
 103             jMapLauncher.addToolArg("jmap");
 104             jMapLauncher.addToolArg("--binaryheap");
 105             jMapLauncher.addToolArg("--pid");
 106             jMapLauncher.addToolArg(String.valueOf(pid));
 107 
 108             ProcessBuilder jMapProcessBuilder = new ProcessBuilder(
 109                     jMapLauncher.getCommand());
 110             System.out.println("jmap command: "
 111                     + Arrays.toString(jMapLauncher.getCommand()));
 112 
 113             Process jMapProcess = jMapProcessBuilder.start();
 114             OutputAnalyzer analyzer = new OutputAnalyzer(jMapProcess);
 115             analyzer.shouldHaveExitValue(0);
 116             analyzer.shouldContain(HEAP_DUMP_FILE_NAME);
 117 
 118             largeHeapProc.getOutputStream().write('\n');
 119 
 120             File dumpFile = new File(HEAP_DUMP_FILE_NAME);
 121             Asserts.assertTrue(dumpFile.exists(), "Heap dump file not found.");
 122 
 123             try (Reader reader = new BufferedReader(new FileReader(dumpFile))) {
 124                 CharBuffer buf = CharBuffer.allocate(expectedFormat.length());
 125                 reader.read(buf);
 126                 buf.clear();
 127                 Asserts.assertEQ(buf.toString(), expectedFormat,
 128                         "Wrong file format. Expected '" + expectedFormat
 129                                 + "', but found '" + buf.toString() + "'");
 130             }
 131 
 132             System.out.println("Success!");
 133 
 134         } finally {
 135             largeHeapProc.destroyForcibly();
 136         }
 137     }
 138 }