/*

 * Copyright (c) 2014, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

/*

 * @test CodeCacheTest

 * @bug 8054889

 * @build DcmdUtil CodeCacheTest

 * @run main/othervm -XX:+SegmentedCodeCache CodeCacheTest

 * @run main/othervm -XX:-SegmentedCodeCache CodeCacheTest

 * @summary Test of diagnostic command Compiler.codecache

 */

import java.io.BufferedReader;

import java.io.StringReader;

import java.lang.reflect.Method;

import java.util.regex.Matcher;

import java.util.regex.Pattern;

public class CodeCacheTest {

    /**

     * This test calls Jcmd (diagnostic command tool) Compiler.codecache and then parses the output,

     * making sure that all numbers look ok

     *

     *

     * Expected output without code cache segmentation:

     *

     * CodeCache: size=245760Kb used=4680Kb max_used=4680Kb free=241079Kb

     * bounds [0x00007f5bd9000000, 0x00007f5bd94a0000, 0x00007f5be8000000]

     * total_blobs=575 nmethods=69 adapters=423

     * compilation: enabled

     *

     * Expected output with code cache segmentation (number of segments may change):

     *

     * CodeHeap 'non-nmethods': size=5696Kb used=2236Kb max_used=2238Kb free=3459Kb

     *  bounds [0x00007fa0f0ffe000, 0x00007fa0f126e000, 0x00007fa0f158e000]

     * CodeHeap 'profiled nmethods': size=120036Kb used=8Kb max_used=8Kb free=120027Kb

     *  bounds [0x00007fa0f158e000, 0x00007fa0f17fe000, 0x00007fa0f8ac7000]

     * CodeHeap 'non-profiled nmethods': size=120036Kb used=2Kb max_used=2Kb free=120034Kb

     *  bounds [0x00007fa0f8ac7000, 0x00007fa0f8d37000, 0x00007fa100000000]

     * total_blobs=486 nmethods=8 adapters=399

     * compilation: enabled

     */

    static Pattern line1 = Pattern.compile("(CodeCache|CodeHeap.*): size=(\\p{Digit}*)Kb used=(\\p{Digit}*)Kb max_used=(\\p{Digit}*)Kb free=(\\p{Digit}*)Kb");

    static Pattern line2 = Pattern.compile(" bounds \\[0x(\\p{XDigit}*), 0x(\\p{XDigit}*), 0x(\\p{XDigit}*)\\]");

    static Pattern line3 = Pattern.compile(" total_blobs=(\\p{Digit}*) nmethods=(\\p{Digit}*) adapters=(\\p{Digit}*)");

    static Pattern line4 = Pattern.compile(" compilation: (.*)");

    private static boolean getFlagBool(String flag, String where) {

      Matcher m = Pattern.compile(flag + "\\s+:?= (true|false)").matcher(where);

      if (!m.find()) {

        throw new RuntimeException("Could not find value for flag " + flag + " in output string");

      }

      return m.group(1).equals("true");

    }

    private static int getFlagInt(String flag, String where) {

      Matcher m = Pattern.compile(flag + "\\s+:?=\\s+\\d+").matcher(where);

      if (!m.find()) {

        throw new RuntimeException("Could not find value for flag " + flag + " in output string");

      }

      String match = m.group();

      return Integer.parseInt(match.substring(match.lastIndexOf(" ") + 1, match.length()));

    }

    public static void main(String arg[]) throws Exception {

        // Get number of code cache segments

        int segmentsCount = 0;

        String flags = DcmdUtil.executeDcmd("VM.flags", "-all");

        if (!getFlagBool("SegmentedCodeCache", flags) || !getFlagBool("UseCompiler", flags)) {

          // No segmentation

          segmentsCount = 1;

        } else if (getFlagBool("TieredCompilation", flags) && getFlagInt("TieredStopAtLevel", flags) > 1) {

          // Tiered compilation: use all segments

          segmentsCount = 3;

        } else {

          // No TieredCompilation: only non-nmethod and non-profiled segment

          segmentsCount = 2;

        }

        // Get output from dcmd (diagnostic command)

        String result = DcmdUtil.executeDcmd("Compiler.codecache");

        BufferedReader r = new BufferedReader(new StringReader(result));

        // Validate code cache segments

        String line;

        Matcher m;

        for (int s = 0; s < segmentsCount; ++s) {

          // Validate first line

          line = r.readLine();

          m = line1.matcher(line);

          if (m.matches()) {

              for (int i = 2; i <= 5; i++) {

                  int val = Integer.parseInt(m.group(i));

                  if (val < 0) {

                      throw new Exception("Failed parsing dcmd codecache output");

                  }

              }

          } else {

              throw new Exception("Regexp 1 failed");

          }

          // Validate second line

          line = r.readLine();

          m = line2.matcher(line);

          if (m.matches()) {

              String start = m.group(1);

              String mark  = m.group(2);

              String top   = m.group(3);

              // Lexical compare of hex numbers to check that they look sane.

              if (start.compareTo(mark) > 1) {

                  throw new Exception("Failed parsing dcmd codecache output");

              }

              if (mark.compareTo(top) > 1) {

                  throw new Exception("Failed parsing dcmd codecache output");

              }

          } else {

              throw new Exception("Regexp 2 failed line: " + line);

          }

        }

        // Validate third line

        line = r.readLine();

        m = line3.matcher(line);

        if (m.matches()) {

            int blobs = Integer.parseInt(m.group(1));

            if (blobs <= 0) {

                throw new Exception("Failed parsing dcmd codecache output");

            }

            int nmethods = Integer.parseInt(m.group(2));

            if (nmethods < 0) {

                throw new Exception("Failed parsing dcmd codecache output");

            }

            int adapters = Integer.parseInt(m.group(3));

            if (adapters <= 0) {

                throw new Exception("Failed parsing dcmd codecache output");

            }

            if (blobs < (nmethods + adapters)) {

                throw new Exception("Failed parsing dcmd codecache output");

            }

        } else {

            throw new Exception("Regexp 3 failed");

        }

        // Validate fourth line

        line = r.readLine();

        m = line4.matcher(line);

        if (!m.matches()) {

            throw new Exception("Regexp 4 failed");

        }

    }

}