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 }