jdk/src/share/classes/com/sun/tools/hat/Main.java
changeset 2 90ce3da70b43
child 468 642c8c0be52e
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/src/share/classes/com/sun/tools/hat/Main.java	Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,220 @@
+/*
+ * Copyright 2005-2006 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.  Sun designates this
+ * particular file as subject to the "Classpath" exception as provided
+ * by Sun in the LICENSE file that accompanied this code.
+ *
+ * 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.
+ */
+
+
+/*
+ * The contents of this file are subject to the Sun Public License
+ * Version 1.0 (the "License"); you may not use this file except in
+ * compliance with the License. A copy of the License is available at
+ * http://www.sun.com/, and in the file LICENSE.html in the
+ * doc directory.
+ *
+ * The Original Code is HAT. The Initial Developer of the
+ * Original Code is Bill Foote, with contributions from others
+ * at JavaSoft/Sun. Portions created by Bill Foote and others
+ * at Javasoft/Sun are Copyright (C) 1997-2004. All Rights Reserved.
+ */
+
+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;
+
+/**
+ *
+ * @author      Bill Foote
+ */
+
+
+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);
+    }
+
+    //
+    // Convert s to a boolean.  If it's invalid, abort the program.
+    //
+    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;       // Never happens
+        }
+    }
+
+    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)) {
+                // Undocumented option. To be used for testing purpose only
+                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;    // Guard against conservative GC
+        }
+        if ( debugLevel == 2 ) {
+            System.out.println("No server, -debug 2 was used.");
+            System.exit(0);
+        }
+
+        if (parseonly) {
+            // do not start web server.
+            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.");
+    }
+}