1 /*
   2  * Copyright (c) 2018, 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.util.Arrays;
  25 import jdk.internal.platform.Metrics;
  26 
  27 public class MetricsMemoryTester {
  28     public static void main(String[] args) {
  29         System.out.println(Arrays.toString(args));
  30         switch (args[0]) {
  31             case "memory":
  32                 testMemoryLimit(args[1]);
  33                 break;
  34             case "memoryswap":
  35                 testMemoryAndSwapLimit(args[1], args[2]);
  36                 break;
  37             case "kernelmem":
  38                 testKernelMemoryLimit(args[1]);
  39                 break;
  40             case "oomkill":
  41                 testOomKillFlag(Boolean.parseBoolean(args[2]));
  42                 break;
  43             case "failcount":
  44                 testMemoryFailCount();
  45                 break;
  46             case "softlimit":
  47                 testMemorySoftLimit(args[1]);
  48                 break;
  49         }
  50     }
  51 
  52     private static void testMemoryLimit(String value) {
  53         long limit = getMemoryValue(value);
  54 
  55         if (limit != Metrics.systemMetrics().getMemoryLimit()) {
  56             throw new RuntimeException("Memory limit not equal, expected : ["
  57                     + limit + "]" + ", got : ["
  58                     + Metrics.systemMetrics().getMemoryLimit() + "]");
  59         }
  60         System.out.println("TEST PASSED!!!");
  61     }
  62 
  63     private static void testMemoryFailCount() {
  64         long count = Metrics.systemMetrics().getMemoryFailCount();
  65 
  66         // Allocate 512M of data
  67         byte[][] bytes = new byte[64][];
  68         for (int i = 0; i < 64; i++) {
  69             try {
  70                 bytes[i] = new byte[8 * 1024 * 1024];
  71                 // Break out as soon as we see an increase in failcount
  72                 // to avoid getting killed by the OOM killer.
  73                 if (Metrics.systemMetrics().getMemoryFailCount() > count) {
  74                     break;
  75                 }
  76             } catch (Error e) { // OOM error
  77                 break;
  78             }
  79         }
  80         if (Metrics.systemMetrics().getMemoryFailCount() <= count) {
  81             throw new RuntimeException("Memory fail count : new : ["
  82                     + Metrics.systemMetrics().getMemoryFailCount() + "]"
  83                     + ", old : [" + count + "]");
  84         }
  85         System.out.println("TEST PASSED!!!");
  86     }
  87 
  88     private static void testMemorySoftLimit(String softLimit) {
  89 
  90         long memorySoftLimit = Metrics.systemMetrics().getMemorySoftLimit();
  91         long newmemorySoftLimit = getMemoryValue(softLimit);
  92 
  93         if (newmemorySoftLimit != memorySoftLimit) {
  94             throw new RuntimeException("Memory softlimit not equal, Actual : ["
  95                     + newmemorySoftLimit + "]" + ", Expected : ["
  96                     + memorySoftLimit + "]");
  97         }
  98         System.out.println("TEST PASSED!!!");
  99     }
 100 
 101     private static void testKernelMemoryLimit(String value) {
 102         long limit = getMemoryValue(value);
 103         long kmemlimit = Metrics.systemMetrics().getKernelMemoryLimit();
 104         if (kmemlimit != 0 && limit != kmemlimit) {
 105             throw new RuntimeException("Kernel Memory limit not equal, expected : ["
 106                     + limit + "]" + ", got : ["
 107                     + kmemlimit + "]");
 108         }
 109         System.out.println("TEST PASSED!!!");
 110     }
 111 
 112     private static void testMemoryAndSwapLimit(String memory, String memAndSwap) {
 113         long expectedMem = getMemoryValue(memory);
 114         long expectedMemAndSwap = getMemoryValue(memAndSwap);
 115 
 116         if (expectedMem != Metrics.systemMetrics().getMemoryLimit()
 117                 || expectedMemAndSwap != Metrics.systemMetrics().getMemoryAndSwapLimit()) {
 118             System.err.println("Memory and swap limit not equal, expected : ["
 119                     + expectedMem + ", " + expectedMemAndSwap + "]"
 120                     + ", got : [" + Metrics.systemMetrics().getMemoryLimit()
 121                     + ", " + Metrics.systemMetrics().getMemoryAndSwapLimit() + "]");
 122         }
 123         System.out.println("TEST PASSED!!!");
 124     }
 125 
 126     private static long getMemoryValue(String value) {
 127         long result;
 128         if (value.endsWith("m")) {
 129             result = Long.parseLong(value.substring(0, value.length() - 1))
 130                     * 1024 * 1024;
 131         } else if (value.endsWith("g")) {
 132             result = Long.parseLong(value.substring(0, value.length() - 1))
 133                     * 1024 * 1024 * 1024;
 134         } else {
 135             result = Long.parseLong(value);
 136         }
 137         return result;
 138     }
 139 
 140     private static void testOomKillFlag(boolean oomKillFlag) {
 141         if (!(oomKillFlag ^ Metrics.systemMetrics().isMemoryOOMKillEnabled())) {
 142             throw new RuntimeException("oomKillFlag error");
 143         }
 144         System.out.println("TEST PASSED!!!");
 145     }
 146 }