224 lines · 8.6 KB · Java
RawDownload
/* * Copyright (c) 1997, 2008, Oracle and/or its affiliates. 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.  Oracle designates this * particular file as subject to the "Classpath" exception as provided * by Oracle 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA * or visit www.oracle.com if you need additional information or have any * questions. */  /* * The Original Code is HAT. The Initial Developer of the * Original Code is Bill Foote, with contributions from others * at JavaSoft/Sun. */ package com.sun.tools.hat.internal.server; /** * Reads a single HTTP query from a socket, and starts up a QueryHandler * to server it. * * @author      Bill Foote */  import java.net.Socket;import java.net.ServerSocket;import java.net.InetAddress; import java.io.InputStream;import java.io.BufferedInputStream;import java.io.IOException;import java.io.Writer;import java.io.BufferedWriter;import java.io.PrintWriter;import java.io.OutputStream;import java.io.OutputStreamWriter;import java.io.BufferedOutputStream; import com.sun.tools.hat.internal.model.Snapshot;import com.sun.tools.hat.internal.oql.OQLEngine; public class HttpReader implements Runnable {      private Socket socket;    private PrintWriter out;    private Snapshot snapshot;    private OQLEngine engine;     public HttpReader (Socket s, Snapshot snapshot, OQLEngine engine) {        this.socket = s;        this.snapshot = snapshot;        this.engine = engine;    }     public void run() {        InputStream in = null;        try {            in = new BufferedInputStream(socket.getInputStream());            out = new PrintWriter(new BufferedWriter(                            new OutputStreamWriter(                                socket.getOutputStream())));            out.println("HTTP/1.0 200 OK");            out.println("Cache-Control: no-cache");            out.println("Pragma: no-cache");            out.println();            if (in.read() != 'G' || in.read() != 'E'                    || in.read() != 'T' || in.read() != ' ') {                outputError("Protocol error");            }            int data;            StringBuffer queryBuf = new StringBuffer();            while ((data = in.read()) != -1 && data != ' ') {                char ch = (char) data;                queryBuf.append(ch);            }            String query = queryBuf.toString();            query = java.net.URLDecoder.decode(query, "UTF-8");            QueryHandler handler = null;            if (snapshot == null) {                outputError("The heap snapshot is still being read.");                return;            } else if (query.equals("/")) {                handler = new AllClassesQuery(true, engine != null);                handler.setUrlStart("");                handler.setQuery("");            } else if (query.startsWith("/oql/")) {                if (engine != null) {                    handler = new OQLQuery(engine);                    handler.setUrlStart("");                    handler.setQuery(query.substring(5));                }            } else if (query.startsWith("/oqlhelp/")) {                if (engine != null) {                    handler = new OQLHelp();                    handler.setUrlStart("");                    handler.setQuery("");                }            } else if (query.equals("/allClassesWithPlatform/")) {                handler = new AllClassesQuery(false, engine != null);                handler.setUrlStart("../");                handler.setQuery("");            } else if (query.equals("/showRoots/")) {                handler = new AllRootsQuery();                handler.setUrlStart("../");                handler.setQuery("");            } else if (query.equals("/showInstanceCounts/includePlatform/")) {                handler = new InstancesCountQuery(false);                handler.setUrlStart("../../");                handler.setQuery("");            } else if (query.equals("/showInstanceCounts/")) {                handler = new InstancesCountQuery(true);                handler.setUrlStart("../");                handler.setQuery("");            } else if (query.startsWith("/instances/")) {                handler = new InstancesQuery(false);                handler.setUrlStart("../");                handler.setQuery(query.substring(11));            }  else if (query.startsWith("/newInstances/")) {                handler = new InstancesQuery(false, true);                handler.setUrlStart("../");                handler.setQuery(query.substring(14));            }  else if (query.startsWith("/allInstances/")) {                handler = new InstancesQuery(true);                handler.setUrlStart("../");                handler.setQuery(query.substring(14));            }  else if (query.startsWith("/allNewInstances/")) {                handler = new InstancesQuery(true, true);                handler.setUrlStart("../");                handler.setQuery(query.substring(17));            } else if (query.startsWith("/object/")) {                handler = new ObjectQuery();                handler.setUrlStart("../");                handler.setQuery(query.substring(8));            } else if (query.startsWith("/class/")) {                handler = new ClassQuery();                handler.setUrlStart("../");                handler.setQuery(query.substring(7));            } else if (query.startsWith("/roots/")) {                handler = new RootsQuery(false);                handler.setUrlStart("../");                handler.setQuery(query.substring(7));            } else if (query.startsWith("/allRoots/")) {                handler = new RootsQuery(true);                handler.setUrlStart("../");                handler.setQuery(query.substring(10));            } else if (query.startsWith("/reachableFrom/")) {                handler = new ReachableQuery();                handler.setUrlStart("../");                handler.setQuery(query.substring(15));            } else if (query.startsWith("/rootStack/")) {                handler = new RootStackQuery();                handler.setUrlStart("../");                handler.setQuery(query.substring(11));            } else if (query.startsWith("/histo/")) {                handler = new HistogramQuery();                handler.setUrlStart("../");                handler.setQuery(query.substring(7));            } else if (query.startsWith("/refsByType/")) {                handler = new RefsByTypeQuery();                handler.setUrlStart("../");                handler.setQuery(query.substring(12));            } else if (query.startsWith("/finalizerSummary/")) {                handler = new FinalizerSummaryQuery();                handler.setUrlStart("../");                handler.setQuery("");            } else if (query.startsWith("/finalizerObjects/")) {                handler = new FinalizerObjectsQuery();                handler.setUrlStart("../");                handler.setQuery("");            }             if (handler != null) {                handler.setOutput(out);                handler.setSnapshot(snapshot);                handler.run();            } else {                outputError("Query '" + query + "' not implemented");            }        } catch (IOException ex) {            ex.printStackTrace();        } finally {            if (out != null) {                out.close();            }            try {                if (in != null) {                    in.close();                }            } catch (IOException ignored) {            }            try {                socket.close();            } catch (IOException ignored) {            }        }    }     private void outputError(String msg) {        out.println();        out.println("<html><body bgcolor=\"#ffffff\">");        out.println(msg);        out.println("</body></html>");    } }