1 /*
   2  * Copyright (c) 2013, 2015, 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 com.oracle.java.testlibrary.Asserts;
  35 import com.oracle.java.testlibrary.JDKToolFinder;
  36 import com.oracle.java.testlibrary.JDKToolLauncher;
  37 import com.oracle.java.testlibrary.OutputAnalyzer;
  38 import com.oracle.java.testlibrary.Platform;
  39 import com.oracle.java.testlibrary.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/sun.misc
  48  *          java.compiler
  49  *          java.management/sun.management
  50  *          jdk.jvmstat/sun.jvmstat.monitor
  51  * @build com.oracle.java.testlibrary.* JMapHProfLargeHeapProc
  52  * @run main JMapHProfLargeHeapTest
  53  */
  54 
  55 public class JMapHProfLargeHeapTest {
  56     private static final String HEAP_DUMP_FILE_NAME = "heap.hprof";
  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 we are on MacOSX, test if JMap tool is signed, otherwise return
  64         // since test will fail with privilege error.
  65         if (Platform.isOSX()) {
  66             String jmapToolPath = JDKToolFinder.getTestJDKTool("jmap");
  67             ProcessBuilder codesignProcessBuilder = new ProcessBuilder(
  68                     "codesign", "-v", jmapToolPath);
  69             Process codesignProcess = codesignProcessBuilder.start();
  70             OutputAnalyzer analyser = new OutputAnalyzer(codesignProcess);
  71             try {
  72                 analyser.shouldNotContain("code object is not signed at all");
  73                 System.out.println("Signed jmap found at: " + jmapToolPath);
  74             } catch (Exception e) {
  75                 // Abort since we can't know if the test will work
  76                 System.out
  77                         .println("Test aborted since we are on MacOSX and the jmap tool is not signed.");
  78                 return;
  79             }
  80         }
  81 
  82         // Small heap 22 megabytes, should create 1.0.1 file format
  83         testHProfFileFormat("-Xmx1g", 22 * M, HPROF_HEADER_1_0_1);
  84 
  85         /**
  86          * This test was deliberately commented out since the test system lacks
  87          * support to handle the requirements for this kind of heap size in a
  88          * good way. If or when it becomes possible to run this kind of tests in
  89          * the test environment the test should be enabled again.
  90          * */
  91         // Large heap 2,2 gigabytes, should create 1.0.2 file format
  92         // testHProfFileFormat("-Xmx4g", 2 * G + 2 * M, HPROF_HEADER_1_0_2);
  93     }
  94 
  95     private static void testHProfFileFormat(String vmArgs, long heapSize,
  96             String expectedFormat) throws Exception, IOException,
  97             InterruptedException, FileNotFoundException {
  98         ProcessBuilder procBuilder = ProcessTools.createJavaProcessBuilder(
  99                 vmArgs, "JMapHProfLargeHeapProc", String.valueOf(heapSize));
 100         procBuilder.redirectError(ProcessBuilder.Redirect.INHERIT);
 101         Process largeHeapProc = procBuilder.start();
 102 
 103         try (Scanner largeHeapScanner = new Scanner(
 104                 largeHeapProc.getInputStream());) {
 105             String pidstring = null;
 106             while ((pidstring = largeHeapScanner.findInLine("PID\\[[0-9].*\\]")) == null) {
 107                 Thread.sleep(500);
 108             }
 109             int pid = Integer.parseInt(pidstring.substring(4,
 110                     pidstring.length() - 1));
 111             System.out.println("Extracted pid: " + pid);
 112 
 113             JDKToolLauncher jMapLauncher = JDKToolLauncher
 114                     .createUsingTestJDK("jmap");
 115             jMapLauncher.addToolArg("-dump:format=b,file=" + pid + "-"
 116                     + HEAP_DUMP_FILE_NAME);
 117             jMapLauncher.addToolArg(String.valueOf(pid));
 118 
 119             ProcessBuilder jMapProcessBuilder = new ProcessBuilder(
 120                     jMapLauncher.getCommand());
 121             System.out.println("jmap command: "
 122                     + Arrays.toString(jMapLauncher.getCommand()));
 123 
 124             Process jMapProcess = jMapProcessBuilder.start();
 125             OutputAnalyzer analyzer = new OutputAnalyzer(jMapProcess);
 126             analyzer.shouldHaveExitValue(0);
 127             analyzer.shouldContain(pid + "-" + HEAP_DUMP_FILE_NAME);
 128             analyzer.shouldContain("Heap dump file created");
 129 
 130             largeHeapProc.getOutputStream().write('\n');
 131 
 132             File dumpFile = new File(pid + "-" + HEAP_DUMP_FILE_NAME);
 133             Asserts.assertTrue(dumpFile.exists(), "Heap dump file not found.");
 134 
 135             try (Reader reader = new BufferedReader(new FileReader(dumpFile))) {
 136                 CharBuffer buf = CharBuffer.allocate(expectedFormat.length());
 137                 reader.read(buf);
 138                 buf.clear();
 139                 Asserts.assertEQ(buf.toString(), expectedFormat,
 140                         "Wrong file format. Expected '" + expectedFormat
 141                                 + "', but found '" + buf.toString() + "'");
 142             }
 143 
 144             System.out.println("Success!");
 145 
 146         } finally {
 147             largeHeapProc.destroyForcibly();
 148         }
 149     }
 150 }