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 /*
  25  * @test
  26  * @key nmt jcmd
  27  * @library /test/lib
  28  * @modules java.base/jdk.internal.misc
  29  *          java.management
  30  * @build sun.hotspot.WhiteBox
  31  * @run main ClassFileInstaller sun.hotspot.WhiteBox
  32  *                              sun.hotspot.WhiteBox$WhiteBoxPermission
  33  * @run main/othervm -Xbootclasspath/a:. -XX:+UnlockDiagnosticVMOptions -XX:+WhiteBoxAPI -XX:NativeMemoryTracking=detail ThreadedMallocTestType
  34  */
  35 
  36 import jdk.test.lib.process.ProcessTools;
  37 import jdk.test.lib.process.OutputAnalyzer;
  38 import jdk.test.lib.JDKToolFinder;
  39 import sun.hotspot.WhiteBox;
  40 
  41 public class ThreadedMallocTestType {
  42   public static long memAlloc1;
  43   public static long memAlloc2;
  44   public static long memAlloc3;
  45 
  46   public static void main(String args[]) throws Exception {
  47     OutputAnalyzer output;
  48     final WhiteBox wb = WhiteBox.getWhiteBox();
  49 
  50     // Grab my own PID
  51     String pid = Long.toString(ProcessTools.getProcessId());
  52     ProcessBuilder pb = new ProcessBuilder();
  53 
  54     Thread allocThread = new Thread() {
  55       public void run() {
  56         // Alloc memory using the WB api
  57         memAlloc1 = wb.NMTMalloc(128 * 1024);
  58         memAlloc2 = wb.NMTMalloc(256 * 1024);
  59         memAlloc3 = wb.NMTMalloc(512 * 1024);
  60       }
  61     };
  62 
  63     allocThread.start();
  64     allocThread.join();
  65 
  66     // Run 'jcmd <pid> VM.native_memory summary'
  67     pb.command(new String[] { JDKToolFinder.getJDKTool("jcmd"), pid, "VM.native_memory", "summary"});
  68     output = new OutputAnalyzer(pb.start());
  69     output.shouldContain("Test (reserved=896KB, committed=896KB)");
  70 
  71     Thread freeThread = new Thread() {
  72       public void run() {
  73         // Free the memory allocated by NMTMalloc
  74         wb.NMTFree(memAlloc1);
  75         wb.NMTFree(memAlloc2);
  76         wb.NMTFree(memAlloc3);
  77       }
  78     };
  79 
  80     freeThread.start();
  81     freeThread.join();
  82 
  83     output = new OutputAnalyzer(pb.start());
  84     output.shouldNotContain("Test (reserved=");
  85   }
  86 }