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  * @bug 8035668
  27  * @library /lib/testlibrary
  28  * @summary Test checks case when target application finishes execution and jstat didn't complete work.
  29             jstat is started with interval = 100 (jstat -compiler 100) and monitored application finishes
  30             after 500ms. This shouldn't cause crash or hang in target application or in jstat.
  31  * @build jdk.testlibrary.ProcessTools jdk.testlibrary.JDKToolLauncher
  32  * @build JStatInterval
  33  * @run main JStatInterval
  34  */
  35 
  36 import jdk.testlibrary.ProcessTools;
  37 import jdk.testlibrary.JDKToolLauncher;
  38 
  39 import java.io.BufferedReader;
  40 import java.io.InputStreamReader;
  41 import java.util.concurrent.TimeUnit;
  42 import java.util.concurrent.atomic.AtomicBoolean;
  43 import java.util.concurrent.atomic.AtomicInteger;
  44 
  45 public class JStatInterval {
  46     private static final String PID = "PID";
  47     private static final String READY = "READY";
  48     private static final String ERROR = "!ERROR";
  49 
  50     public static class Application {
  51         public static void main(String[] args) {
  52             try {
  53                 System.out.println(PID + ":" + ProcessTools.getProcessId());
  54                 System.out.println(READY);
  55                 System.out.flush();
  56                 int exitCode = System.in.read();
  57                 Thread.sleep(500);
  58                 System.exit(exitCode);
  59             } catch (Exception e) {
  60                 System.out.println(ERROR);
  61                 System.out.flush();
  62                 throw new Error(e);
  63             }
  64         }
  65     }
  66     public static void main(String[] args) throws Exception {
  67         ProcessBuilder pb = ProcessTools.createJavaProcessBuilder(
  68             "-cp",
  69             System.getProperty("test.class.path"),
  70             "-XX:+UsePerfData",
  71             Application.class.getName()
  72         );
  73         AtomicInteger pid = new AtomicInteger(-1);
  74         AtomicBoolean error = new AtomicBoolean(false);
  75         Process app = ProcessTools.startProcess(
  76             "application",
  77             pb,
  78             line -> {
  79                 if (line.startsWith(PID)) {
  80                     pid.set(Integer.parseInt(line.split("\\:")[1]));
  81                 } else if (line.equals(READY)) {
  82                     return true;
  83                 } else if (line.equals(ERROR)) {
  84                     error.set(true);
  85                     return true;
  86                 }
  87                 return false;
  88             },
  89             10,
  90             TimeUnit.SECONDS
  91         );
  92         if (error.get()) {
  93             throw new Error("Unable to start the monitored application.");
  94         }
  95 
  96         String pidStr = String.valueOf(pid.get());
  97         JDKToolLauncher l = JDKToolLauncher.createUsingTestJDK("jstat");
  98         l.addToolArg("-compiler");
  99         l.addToolArg(pidStr);
 100         l.addToolArg("100");
 101 
 102         ProcessBuilder jstatDef = new ProcessBuilder(l.getCommand());
 103         Process jstat = ProcessTools.startProcess(
 104             "jstat",
 105             jstatDef,
 106             line -> {
 107                 if (line.trim().toLowerCase().startsWith("compiled")) {
 108                     return true;
 109                 }
 110                 return false;
 111             },
 112             10,
 113             TimeUnit.SECONDS
 114         );
 115 
 116         app.getOutputStream().write(0);
 117         app.getOutputStream().flush();
 118 
 119         if (app.waitFor() != 0) {
 120             throw new Error("Error detected upon exiting the monitored application with active jstat");
 121         }
 122         if (jstat.waitFor() != 0) {
 123             throw new Error("Error detected in jstat when monitored application has exited prematurely");
 124         }
 125     }
 126 }