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 /*
  25  * @test
  26  * @key nmt jcmd
  27  * @library /testlibrary /testlibrary/whitebox
  28  * @build ThreadedVirtualAllocTestType
  29  * @run main/othervm/bootclasspath -XX:+UnlockDiagnosticVMOptions -XX:+WhiteBoxAPI -XX:NativeMemoryTracking=detail ThreadedVirtualAllocTestType
  30  */
  31 
  32 import com.oracle.java.testlibrary.*;
  33 import sun.hotspot.WhiteBox;
  34 
  35 public class ThreadedVirtualAllocTestType {
  36   public static long addr;
  37   public static final WhiteBox wb = WhiteBox.getWhiteBox();
  38   public static final long commitSize = 128 * 1024;
  39   public static final long reserveSize = 512 * 1024;
  40 
  41   public static void main(String args[]) throws Exception {
  42     OutputAnalyzer output;
  43 
  44     String pid = Integer.toString(ProcessTools.getProcessId());
  45     ProcessBuilder pb = new ProcessBuilder();
  46 
  47     boolean has_nmt_detail = wb.NMTIsDetailSupported();
  48     if (has_nmt_detail) {
  49       System.out.println("NMT detail support detected.");
  50     } else {
  51       System.out.println("NMT detail support not detected.");
  52     }
  53 
  54     Thread reserveThread = new Thread() {
  55       public void run() {
  56         addr = wb.NMTReserveMemory(reserveSize);
  57       }
  58     };
  59     reserveThread.start();
  60     reserveThread.join();
  61 
  62     mergeData();
  63 
  64     pb.command(new String[] { JDKToolFinder.getJDKTool("jcmd"), pid, "VM.native_memory", "detail"});
  65     output = new OutputAnalyzer(pb.start());
  66     output.shouldContain("Test (reserved=512KB, committed=0KB)");
  67     if (has_nmt_detail) {
  68       output.shouldMatch("\\[0x[0]*" + Long.toHexString(addr) + " - 0x[0]*" + Long.toHexString(addr + reserveSize) + "\\] reserved 512KB for Test");
  69     }
  70 
  71     Thread commitThread = new Thread() {
  72       public void run() {
  73         wb.NMTCommitMemory(addr, commitSize);
  74       }
  75     };
  76     commitThread.start();
  77     commitThread.join();
  78 
  79     mergeData();
  80 
  81     output = new OutputAnalyzer(pb.start());
  82     output.shouldContain("Test (reserved=512KB, committed=128KB)");
  83     if (has_nmt_detail) {
  84       output.shouldMatch("\\[0x[0]*" + Long.toHexString(addr) + " - 0x[0]*" + Long.toHexString(addr + commitSize) + "\\] committed 128KB");
  85     }
  86 
  87     Thread uncommitThread = new Thread() {
  88       public void run() {
  89         wb.NMTUncommitMemory(addr, commitSize);
  90       }
  91     };
  92     uncommitThread.start();
  93     uncommitThread.join();
  94 
  95     mergeData();
  96 
  97     output = new OutputAnalyzer(pb.start());
  98     output.shouldContain("Test (reserved=512KB, committed=0KB)");
  99     output.shouldNotMatch("\\[0x[0]*" + Long.toHexString(addr) + " - 0x[0]*" + Long.toHexString(addr + commitSize) + "\\] committed");
 100 
 101     Thread releaseThread = new Thread() {
 102       public void run() {
 103         wb.NMTReleaseMemory(addr, reserveSize);
 104       }
 105     };
 106     releaseThread.start();
 107     releaseThread.join();
 108 
 109     mergeData();
 110 
 111     output = new OutputAnalyzer(pb.start());
 112     output.shouldNotContain("Test (reserved=");
 113     output.shouldNotContain("\\[0x[0]*" + Long.toHexString(addr) + " - 0x[0]*" + Long.toHexString(addr + reserveSize) + "\\] reserved");
 114   }
 115 
 116   public static void mergeData() throws Exception {
 117     // Use WB API to ensure that all data has been merged before we continue
 118     if (!wb.NMTWaitForDataMerge()) {
 119       throw new Exception("Call to WB API NMTWaitForDataMerge() failed");
 120     }
 121   }
 122 }