jdk/test/sun/tools/jhat/HatRun.java
changeset 25144 e2bf17cee34b
parent 25143 aade36e70da0
parent 25086 f5f41ed620d5
child 25145 82fa3e8aabee
child 25200 360a141ded8f
equal deleted inserted replaced
25143:aade36e70da0 25144:e2bf17cee34b
     1 /*
       
     2  * Copyright (c) 2005, 2013, 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 /*
       
    26  *
       
    27  * Support classes for running jhat tests
       
    28  *
       
    29  */
       
    30 
       
    31 import java.io.InputStream;
       
    32 import java.io.IOException;
       
    33 import java.io.File;
       
    34 import java.io.BufferedInputStream;
       
    35 import java.io.PrintStream;
       
    36 
       
    37 /*
       
    38  * Helper class to direct process output to a StringBuffer
       
    39  */
       
    40 class MyInputStream implements Runnable {
       
    41     private String              name;
       
    42     private BufferedInputStream in;
       
    43     private StringBuffer        buffer;
       
    44 
       
    45     /* Create MyInputStream that saves all output to a StringBuffer */
       
    46     MyInputStream(String name, InputStream in) {
       
    47         this.name = name;
       
    48         this.in = new BufferedInputStream(in);
       
    49         buffer = new StringBuffer(4096);
       
    50         Thread thr = new Thread(this);
       
    51         thr.setDaemon(true);
       
    52         thr.start();
       
    53     }
       
    54 
       
    55     /* Dump the buffer */
       
    56     void dump(PrintStream x) {
       
    57         String str = buffer.toString();
       
    58         x.println("<beginning of " + name + " buffer>");
       
    59         x.println(str);
       
    60         x.println("<end of buffer>");
       
    61     }
       
    62 
       
    63     /* Check to see if a pattern is inside the output. */
       
    64     boolean contains(String pattern) {
       
    65         String str = buffer.toString();
       
    66         return str.contains(pattern);
       
    67     }
       
    68 
       
    69     /* Runs as a separate thread capturing all output in a StringBuffer */
       
    70     public void run() {
       
    71         try {
       
    72             byte b[] = new byte[100];
       
    73             for (;;) {
       
    74                 int n = in.read(b);
       
    75                 String str;
       
    76                 if (n < 0) {
       
    77                     break;
       
    78                 }
       
    79                 str = new String(b, 0, n);
       
    80                 buffer.append(str);
       
    81                 System.out.print(str);
       
    82             }
       
    83         } catch (IOException ioe) { /* skip */ }
       
    84     }
       
    85 }
       
    86 
       
    87 /*
       
    88  * Main jhat run
       
    89  */
       
    90 public class HatRun {
       
    91 
       
    92     private String        all_hprof_options;
       
    93     private String        all_hat_options;
       
    94     private String        dumpfile;
       
    95     private MyInputStream output;
       
    96     private MyInputStream error;
       
    97 
       
    98     /* Create a Hat run process */
       
    99     public HatRun(String hprof_options, String hat_options)
       
   100     {
       
   101         all_hprof_options = hprof_options;
       
   102         all_hat_options   = hat_options;
       
   103     }
       
   104 
       
   105     /*
       
   106      * Execute a process with an -agentpath or -agentlib command option
       
   107      */
       
   108     public void runit(String class_name)
       
   109     {
       
   110         runit(class_name, null);
       
   111     }
       
   112 
       
   113     /*
       
   114      * Execute a command.
       
   115      */
       
   116     private void execute(String cmd[])
       
   117     {
       
   118         /* Begin process */
       
   119         Process p;
       
   120         String cmdLine = "";
       
   121         int i;
       
   122 
       
   123         for ( i = 0 ; i < cmd.length; i++ ) {
       
   124           cmdLine += cmd[i];
       
   125           cmdLine += " ";
       
   126         }
       
   127         System.out.println("Starting: " + cmdLine);
       
   128 
       
   129         try {
       
   130             p = Runtime.getRuntime().exec(cmd);
       
   131         } catch ( IOException e ) {
       
   132             throw new RuntimeException("Test failed - exec got IO exception");
       
   133         }
       
   134 
       
   135         /* Save process output in StringBuffers */
       
   136         output = new MyInputStream("Input Stream", p.getInputStream());
       
   137         error  = new MyInputStream("Error Stream", p.getErrorStream());
       
   138 
       
   139         /* Wait for process to complete, and if exit code is non-zero we fail */
       
   140         try {
       
   141             int exitStatus;
       
   142             exitStatus = p.waitFor();
       
   143             if ( exitStatus != 0) {
       
   144                 System.out.println("Exit code is " + exitStatus);
       
   145                 error.dump(System.out);
       
   146                 output.dump(System.out);
       
   147                 throw new RuntimeException("Test failed - " +
       
   148                                     "exit return code non-zero " +
       
   149                                     "(exitStatus==" + exitStatus + ")");
       
   150             }
       
   151         } catch ( InterruptedException e ) {
       
   152             throw new RuntimeException("Test failed - process interrupted");
       
   153         }
       
   154         System.out.println("Completed: " + cmdLine);
       
   155     }
       
   156 
       
   157     /*
       
   158      * Execute a process with an -agentpath or -agentlib command option
       
   159      *    plus any set of other java options.
       
   160      */
       
   161     public void runit(String class_name, String vm_options[])
       
   162     {
       
   163         String jre_home  = System.getProperty("java.home");
       
   164         String sdk_home  = (jre_home.endsWith("jre") ?
       
   165                             (jre_home + File.separator + "..") :
       
   166                             jre_home );
       
   167         String cdir      = System.getProperty("test.classes", ".");
       
   168         String os_arch   = System.getProperty("os.arch");
       
   169         String os_name   = System.getProperty("os.name");
       
   170         String java      = jre_home
       
   171                              + File.separator + "bin"
       
   172                              + File.separator + "java";
       
   173         String jhat      = sdk_home + File.separator + "bin"
       
   174                            + File.separator + "jhat";
       
   175         /* Array of strings to be passed in for exec:
       
   176          *   1. java
       
   177          *   2. -Dtest.classes=.
       
   178          *   3. -Xcheck:jni          (Just because it finds bugs)
       
   179          *   4. -Xverify:all         (Make sure verification is on full blast)
       
   180          *   5. -agent
       
   181          *       vm_options
       
   182          *   6+i. classname
       
   183          */
       
   184         int nvm_options = 0;
       
   185         if ( vm_options != null ) nvm_options = vm_options.length;
       
   186         String cmd[]     = new String[1 + 7 + nvm_options];
       
   187         int i,j;
       
   188 
       
   189         i = 0;
       
   190         cmd[i++] = java;
       
   191         cmd[i++] = "-cp";
       
   192         cmd[i++] = cdir;
       
   193         cmd[i++] = "-Dtest.classes=" + cdir;
       
   194         cmd[i++] = "-Xcheck:jni";
       
   195         cmd[i++] = "-Xverify:all";
       
   196         dumpfile= cdir + File.separator + class_name + ".hdump";
       
   197         cmd[i++] = "-agentlib:hprof=" + all_hprof_options
       
   198                     + ",format=b,file=" + dumpfile;
       
   199         /* Add any special VM options */
       
   200         for ( j = 0; j < nvm_options; j++ ) {
       
   201             cmd[i++] = vm_options[j];
       
   202         }
       
   203         /* Add classname */
       
   204         cmd[i++] = class_name;
       
   205 
       
   206         /* Execute process */
       
   207         execute(cmd);
       
   208 
       
   209         /* Run jhat */
       
   210         String jhat_cmd[] = new String[4];
       
   211         jhat_cmd[0] = jhat;
       
   212         jhat_cmd[1] = "-debug";
       
   213         jhat_cmd[2] = "2";
       
   214         jhat_cmd[3] = dumpfile;
       
   215 
       
   216         /* Execute process */
       
   217         execute(jhat_cmd);
       
   218 
       
   219     }
       
   220 
       
   221     /* Does the pattern appear in the output of this process */
       
   222     public boolean output_contains(String pattern)
       
   223     {
       
   224         return output.contains(pattern) || error.contains(pattern);
       
   225     }
       
   226 }