--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/test/sun/tools/jhat/HatRun.java Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,230 @@
+/*
+ * Copyright 2005 Sun Microsystems, Inc. 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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
+ * CA 95054 USA or visit www.sun.com if you need additional information or
+ * have any questions.
+ */
+
+
+/*
+ *
+ * Support classes for running jhat tests
+ *
+ */
+
+import java.io.InputStream;
+import java.io.IOException;
+import java.io.File;
+import java.io.BufferedInputStream;
+import java.io.PrintStream;
+
+/*
+ * Helper class to direct process output to a StringBuffer
+ */
+class MyInputStream implements Runnable {
+ private String name;
+ private BufferedInputStream in;
+ private StringBuffer buffer;
+
+ /* Create MyInputStream that saves all output to a StringBuffer */
+ MyInputStream(String name, InputStream in) {
+ this.name = name;
+ this.in = new BufferedInputStream(in);
+ buffer = new StringBuffer(4096);
+ Thread thr = new Thread(this);
+ thr.setDaemon(true);
+ thr.start();
+ }
+
+ /* Dump the buffer */
+ void dump(PrintStream x) {
+ String str = buffer.toString();
+ x.println("<beginning of " + name + " buffer>");
+ x.println(str);
+ x.println("<end of buffer>");
+ }
+
+ /* Check to see if a pattern is inside the output. */
+ boolean contains(String pattern) {
+ String str = buffer.toString();
+ return str.contains(pattern);
+ }
+
+ /* Runs as a separate thread capturing all output in a StringBuffer */
+ public void run() {
+ try {
+ byte b[] = new byte[100];
+ for (;;) {
+ int n = in.read(b);
+ String str;
+ if (n < 0) {
+ break;
+ }
+ str = new String(b, 0, n);
+ buffer.append(str);
+ System.out.print(str);
+ }
+ } catch (IOException ioe) { /* skip */ }
+ }
+}
+
+/*
+ * Main jhat run
+ */
+public class HatRun {
+
+ private String all_hprof_options;
+ private String all_hat_options;
+ private String dumpfile;
+ private MyInputStream output;
+ private MyInputStream error;
+
+ /* Create a Hat run process */
+ public HatRun(String hprof_options, String hat_options)
+ {
+ all_hprof_options = hprof_options;
+ all_hat_options = hat_options;
+ }
+
+ /*
+ * Execute a process with an -agentpath or -agentlib command option
+ */
+ public void runit(String class_name)
+ {
+ runit(class_name, null);
+ }
+
+ /*
+ * Execute a command.
+ */
+ private void execute(String cmd[])
+ {
+ /* Begin process */
+ Process p;
+ String cmdLine = "";
+ int i;
+
+ for ( i = 0 ; i < cmd.length; i++ ) {
+ cmdLine += cmd[i];
+ cmdLine += " ";
+ }
+ System.out.println("Starting: " + cmdLine);
+
+ try {
+ p = Runtime.getRuntime().exec(cmd);
+ } catch ( IOException e ) {
+ throw new RuntimeException("Test failed - exec got IO exception");
+ }
+
+ /* Save process output in StringBuffers */
+ output = new MyInputStream("Input Stream", p.getInputStream());
+ error = new MyInputStream("Error Stream", p.getErrorStream());
+
+ /* Wait for process to complete, and if exit code is non-zero we fail */
+ try {
+ int exitStatus;
+ exitStatus = p.waitFor();
+ if ( exitStatus != 0) {
+ System.out.println("Exit code is " + exitStatus);
+ error.dump(System.out);
+ output.dump(System.out);
+ throw new RuntimeException("Test failed - " +
+ "exit return code non-zero " +
+ "(exitStatus==" + exitStatus + ")");
+ }
+ } catch ( InterruptedException e ) {
+ throw new RuntimeException("Test failed - process interrupted");
+ }
+ System.out.println("Completed: " + cmdLine);
+ }
+
+ /*
+ * Execute a process with an -agentpath or -agentlib command option
+ * plus any set of other java options.
+ */
+ public void runit(String class_name, String vm_options[])
+ {
+ String jre_home = System.getProperty("java.home");
+ String sdk_home = (jre_home.endsWith("jre") ?
+ (jre_home + File.separator + "..") :
+ jre_home );
+ String cdir = System.getProperty("test.classes", ".");
+ String os_arch = System.getProperty("os.arch");
+ boolean d64 = os_arch.equals("sparcv9") ||
+ os_arch.equals("amd64");
+ String isa_dir = d64?(File.separator+os_arch):"";
+ String java = jre_home
+ + File.separator + "bin" + isa_dir
+ + File.separator + "java";
+ String jhat = sdk_home + File.separator + "bin"
+ + File.separator + "jhat";
+ /* Array of strings to be passed in for exec:
+ * 1. java
+ * 2. -Dtest.classes=.
+ * 3. -d64 (optional)
+ * 4. -Xcheck:jni (Just because it finds bugs)
+ * 5. -Xverify:all (Make sure verification is on full blast)
+ * 6. -agent
+ * vm_options
+ * 7+i. classname
+ */
+ int nvm_options = 0;
+ if ( vm_options != null ) nvm_options = vm_options.length;
+ String cmd[] = new String[1 + (d64?1:0) + 5 + nvm_options];
+ int i,j;
+
+ i = 0;
+ cmd[i++] = java;
+ cmd[i++] = "-Dtest.classes=" + cdir;
+ if ( d64 ) {
+ cmd[i++] = "-d64";
+ }
+ cmd[i++] = "-Xcheck:jni";
+ cmd[i++] = "-Xverify:all";
+ dumpfile= cdir + File.separator + class_name + ".hdump";
+ cmd[i++] = "-agentlib:hprof=" + all_hprof_options
+ + ",format=b,file=" + dumpfile;
+ /* Add any special VM options */
+ for ( j = 0; j < nvm_options; j++ ) {
+ cmd[i++] = vm_options[j];
+ }
+ /* Add classname */
+ cmd[i++] = class_name;
+
+ /* Execute process */
+ execute(cmd);
+
+ /* Run jhat */
+ String jhat_cmd[] = new String[4];
+ jhat_cmd[0] = jhat;
+ jhat_cmd[1] = "-debug";
+ jhat_cmd[2] = "2";
+ jhat_cmd[3] = dumpfile;
+
+ /* Execute process */
+ execute(jhat_cmd);
+
+ }
+
+ /* Does the pattern appear in the output of this process */
+ public boolean output_contains(String pattern)
+ {
+ return output.contains(pattern) || error.contains(pattern);
+ }
+}