package com.sun.tools.hat;
import java.io.IOException;
import java.io.File;
import com.sun.tools.hat.internal.model.Snapshot;
import com.sun.tools.hat.internal.model.ReachableExcludesImpl;
import com.sun.tools.hat.internal.server.QueryListener;
public class Main {
private static String VERSION_STRING = "jhat version 2.0";
private static void usage(String message) {
if ( message != null ) {
System.err.println("ERROR: " + message);
}
System.err.println("Usage: jhat [-stack <bool>] [-refs <bool>] [-port <port>] [-baseline <file>] [-debug <int>] [-version] [-h|-help] <file>");
System.err.println();
System.err.println("\t-J<flag> Pass <flag> directly to the runtime system. For");
System.err.println("\t\t\t example, -J-mx512m to use a maximum heap size of 512MB");
System.err.println("\t-stack false: Turn off tracking object allocation call stack.");
System.err.println("\t-refs false: Turn off tracking of references to objects");
System.err.println("\t-port <port>: Set the port for the HTTP server. Defaults to 7000");
System.err.println("\t-exclude <file>: Specify a file that lists data members that should");
System.err.println("\t\t\t be excluded from the reachableFrom query.");
System.err.println("\t-baseline <file>: Specify a baseline object dump. Objects in");
System.err.println("\t\t\t both heap dumps with the same ID and same class will");
System.err.println("\t\t\t be marked as not being \"new\".");
System.err.println("\t-debug <int>: Set debug level.");
System.err.println("\t\t\t 0: No debug output");
System.err.println("\t\t\t 1: Debug hprof file parsing");
System.err.println("\t\t\t 2: Debug hprof file parsing, no server");
System.err.println("\t-version Report version number");
System.err.println("\t-h|-help Print this help and exit");
System.err.println("\t<file> The file to read");
System.err.println();
System.err.println("For a dump file that contains multiple heap dumps,");
System.err.println("you may specify which dump in the file");
System.err.println("by appending \"#<number>\" to the file name, i.e. \"foo.hprof#3\".");
System.err.println();
System.err.println("All boolean options default to \"true\"");
System.exit(1);
}
private static boolean booleanValue(String s) {
if ("true".equalsIgnoreCase(s)) {
return true;
} else if ("false".equalsIgnoreCase(s)) {
return false;
} else {
usage("Boolean value must be true or false");
return false;
}
}
public static void main(String[] args) {
if (args.length < 1) {
usage("No arguments supplied");
}
boolean parseonly = false;
int portNumber = 7000;
boolean callStack = true;
boolean calculateRefs = true;
String baselineDump = null;
String excludeFileName = null;
int debugLevel = 0;
for (int i = 0; ; i += 2) {
if (i > (args.length - 1)) {
usage("Option parsing error");
}
if ("-version".equals(args[i])) {
System.out.print(VERSION_STRING);
System.out.println(" (java version " + System.getProperty("java.version") + ")");
System.exit(0);
}
if ("-h".equals(args[i]) || "-help".equals(args[i])) {
usage(null);
}
if (i == (args.length - 1)) {
break;
}
String key = args[i];
String value = args[i+1];
if ("-stack".equals(key)) {
callStack = booleanValue(value);
} else if ("-refs".equals(key)) {
calculateRefs = booleanValue(value);
} else if ("-port".equals(key)) {
portNumber = Integer.parseInt(value, 10);
} else if ("-exclude".equals(key)) {
excludeFileName = value;
} else if ("-baseline".equals(key)) {
baselineDump = value;
} else if ("-debug".equals(key)) {
debugLevel = Integer.parseInt(value, 10);
} else if ("-parseonly".equals(key)) {
parseonly = booleanValue(value);
}
}
String fileName = args[args.length - 1];
Snapshot model = null;
File excludeFile = null;
if (excludeFileName != null) {
excludeFile = new File(excludeFileName);
if (!excludeFile.exists()) {
System.out.println("Exclude file " + excludeFile
+ " does not exist. Aborting.");
System.exit(1);
}
}
System.out.println("Reading from " + fileName + "...");
try {
model = com.sun.tools.hat.internal.parser.Reader.readFile(fileName, callStack, debugLevel);
} catch (IOException ex) {
ex.printStackTrace();
System.exit(1);
} catch (RuntimeException ex) {
ex.printStackTrace();
System.exit(1);
}
System.out.println("Snapshot read, resolving...");
model.resolve(calculateRefs);
System.out.println("Snapshot resolved.");
if (excludeFile != null) {
model.setReachableExcludes(new ReachableExcludesImpl(excludeFile));
}
if (baselineDump != null) {
System.out.println("Reading baseline snapshot...");
Snapshot baseline = null;
try {
baseline = com.sun.tools.hat.internal.parser.Reader.readFile(baselineDump, false,
debugLevel);
} catch (IOException ex) {
ex.printStackTrace();
System.exit(1);
} catch (RuntimeException ex) {
ex.printStackTrace();
System.exit(1);
}
baseline.resolve(false);
System.out.println("Discovering new objects...");
model.markNewRelativeTo(baseline);
baseline = null;
}
if ( debugLevel == 2 ) {
System.out.println("No server, -debug 2 was used.");
System.exit(0);
}
if (parseonly) {
System.out.println("-parseonly is true, exiting..");
System.exit(0);
}
QueryListener listener = new QueryListener(portNumber);
listener.setModel(model);
Thread t = new Thread(listener, "Query Listener");
t.setPriority(Thread.NORM_PRIORITY+1);
t.start();
System.out.println("Started HTTP server on port " + portNumber);
System.out.println("Server is ready.");
}
}
