Java tutorial
// ======================================================================== // $Id: AJP13Connection.java,v 1.38 2006/11/22 20:21:30 gregwilkins Exp $ // Copyright 200-2004 Mort Bay Consulting Pty. Ltd. // ------------------------------------------------------------------------ // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // http://www.apache.org/licenses/LICENSE-2.0 // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. // ======================================================================== package org.browsermob.proxy.jetty.http.ajp; import org.apache.commons.logging.Log; import org.browsermob.proxy.jetty.http.*; import org.browsermob.proxy.jetty.log.LogFactory; import org.browsermob.proxy.jetty.util.LineInput; import org.browsermob.proxy.jetty.util.LogSupport; import org.browsermob.proxy.jetty.util.URI; import java.io.ByteArrayInputStream; import java.io.IOException; import java.io.InputStream; import java.io.OutputStream; import java.net.InetAddress; import java.net.Socket; import java.net.SocketException; import java.security.cert.CertificateFactory; import java.security.cert.X509Certificate; /* ------------------------------------------------------------ */ /** * @version $Id: AJP13Connection.java,v 1.38 2006/11/22 20:21:30 gregwilkins Exp $ * @author Greg Wilkins (gregw) */ public class AJP13Connection extends HttpConnection { private static Log log = LogFactory.getLog(AJP13Connection.class); private AJP13Listener _listener; private AJP13InputStream _ajpIn; private AJP13OutputStream _ajpOut; private String _remoteHost; private String _remoteAddr; private String _serverName; private int _serverPort; private boolean _isSSL; /* ------------------------------------------------------------ */ public AJP13Connection(AJP13Listener listener, InputStream in, OutputStream out, Socket socket, int bufferSize) throws IOException { super(listener, null, new AJP13InputStream(in, out, bufferSize), out, socket); LineInput lin = (LineInput) getInputStream().getInputStream(); _ajpIn = (AJP13InputStream) lin.getInputStream(); _ajpOut = new AJP13OutputStream(getOutputStream().getOutputStream(), bufferSize); _ajpOut.setCommitObserver(this); getOutputStream().setBufferedOutputStream(_ajpOut); _listener = listener; } /* ------------------------------------------------------------ */ /** * Get the Remote address. * * @return the remote address */ public InetAddress getRemoteInetAddress() { return null; } /* ------------------------------------------------------------ */ public void destroy() { if (_ajpIn != null) _ajpIn.destroy(); _ajpIn = null; if (_ajpOut != null) _ajpOut.destroy(); _ajpOut = null; _remoteHost = null; _remoteAddr = null; _serverName = null; } /* ------------------------------------------------------------ */ /** * Get the Remote address. * * @return the remote host name */ public String getRemoteAddr() { return _remoteAddr; } /* ------------------------------------------------------------ */ /** * Get the Remote address. * * @return the remote host name */ public String getRemoteHost() { return _remoteHost; } /* ------------------------------------------------------------ */ /** * Get the listeners HttpServer . Conveniance method equivalent to * getListener().getHost(). * * @return HttpServer. */ public String getServerName() { return _serverName; } /* ------------------------------------------------------------ */ /** * Get the listeners Port . Conveniance method equivalent to * getListener().getPort(). * * @return HttpServer. */ public int getServerPort() { return _serverPort; } /* ------------------------------------------------------------ */ /** * Get the listeners Default scheme. Conveniance method equivalent to * getListener().getDefaultProtocol(). * * @return HttpServer. */ public String getDefaultScheme() { return _isSSL ? HttpMessage.__SSL_SCHEME : super.getDefaultScheme(); } /* ------------------------------------------------------------ */ public boolean isSSL() { return _isSSL; } /* ------------------------------------------------------------ */ public boolean handleNext() { AJP13RequestPacket packet = null; HttpRequest request = getRequest(); HttpResponse response = getResponse(); HttpContext context = null; boolean gotRequest = false; _persistent = true; _keepAlive = true; try { try { packet = null; packet = _ajpIn.nextPacket(); if (packet == null) return false; if (packet.getDataSize() == 0) return true; } catch (IOException e) { LogSupport.ignore(log, e); return false; } int type = packet.getByte(); if (log.isDebugEnabled()) log.debug("AJP13 type=" + type + " size=" + packet.unconsumedData()); switch (type) { case AJP13Packet.__FORWARD_REQUEST: request.setTimeStamp(System.currentTimeMillis()); request.setState(HttpMessage.__MSG_EDITABLE); request.setMethod(packet.getMethod()); request.setVersion(packet.getString()); String version = packet.getString(); try { request.setVersion(version); } catch (Exception e) { log.warn("Bad version" + version, e); log.warn(packet.toString()); } String path = packet.getString(); int sc = path.lastIndexOf(";"); if (sc < 0) request.setPath(URI.encodePath(path)); else request.setPath(URI.encodePath(path.substring(0, sc)) + path.substring(sc)); _remoteAddr = packet.getString(); _remoteHost = packet.getString(); _serverName = packet.getString(); _serverPort = packet.getInt(); _isSSL = packet.getBoolean(); // Check keep alive _keepAlive = request.getDotVersion() >= 1; // Headers int h = packet.getInt(); for (int i = 0; i < h; i++) { String hdr = packet.getHeader(); String val = packet.getString(); request.addField(hdr, val); if (!_keepAlive && hdr.equalsIgnoreCase(HttpFields.__Connection) && val.equalsIgnoreCase(HttpFields.__KeepAlive)) _keepAlive = true; } // Handler other attributes byte attr = packet.getByte(); while ((0xFF & attr) != 0xFF) { String value = (attr == 11) ? null : packet.getString(); switch (attr) { case 11: // key size request.setAttribute("javax.servlet.request.key_size", new Integer(packet.getInt())); break; case 10: // request attribute request.setAttribute(value, packet.getString()); break; case 9: // SSL session request.setAttribute("javax.servlet.request.ssl_session", value); break; case 8: // SSL cipher request.setAttribute("javax.servlet.request.cipher_suite", value); break; case 7: // SSL cert // request.setAttribute("javax.servlet.request.X509Certificate",value); CertificateFactory cf = CertificateFactory.getInstance("X.509"); InputStream certstream = new ByteArrayInputStream(value.getBytes()); X509Certificate cert = (X509Certificate) cf.generateCertificate(certstream); X509Certificate certs[] = { cert }; request.setAttribute("javax.servlet.request.X509Certificate", certs); break; case 6: // JVM Route request.setAttribute("org.browsermob.proxy.jetty.http.ajp.JVMRoute", value); break; case 5: // Query String request.setQuery(value); break; case 4: // AuthType request.setAuthType(value); break; case 3: // Remote User request.setAuthUser(value); break; case 2: // servlet path not implemented case 1: // _context not implemented default: log.warn("Unknown attr: " + attr + "=" + value); } attr = packet.getByte(); } _listener.customizeRequest(this, request); gotRequest = true; statsRequestStart(); request.setState(HttpMessage.__MSG_RECEIVED); // Complete response if (request.getContentLength() == 0 && request.getField(HttpFields.__TransferEncoding) == null) _ajpIn.close(); // Prepare response response.setState(HttpMessage.__MSG_EDITABLE); response.setVersion(HttpMessage.__HTTP_1_1); response.setDateField(HttpFields.__Date, _request.getTimeStamp()); if (!Version.isParanoid()) response.setField(HttpFields.__Server, Version.getDetail()); // Service request if (log.isDebugEnabled()) log.debug("REQUEST:\n" + request); context = service(request, response); if (log.isDebugEnabled()) log.debug("RESPONSE:\n" + response); break; default: if (log.isDebugEnabled()) log.debug("Ignored: " + packet); _persistent = false; } } catch (SocketException e) { LogSupport.ignore(log, e); _persistent = false; } catch (Exception e) { log.warn(LogSupport.EXCEPTION, e); _persistent = false; try { if (gotRequest) _ajpOut.close(); } catch (IOException e2) { LogSupport.ignore(log, e2); } } finally { // abort if nothing received. if (packet == null || !gotRequest) return false; // flush and end the output try { // Consume unread input. // while(_ajpIn.skip(4096)>0 || _ajpIn.read()>=0); // end response getOutputStream().close(); if (!_persistent) _ajpOut.end(); // Close the outout _ajpOut.close(); // reset streams getOutputStream().resetStream(); getOutputStream().addObserver(this); getInputStream().resetStream(); _ajpIn.resetStream(); _ajpOut.resetStream(); } catch (Exception e) { log.debug(LogSupport.EXCEPTION, e); _persistent = false; } finally { statsRequestEnd(); if (context != null) context.log(request, response, -1); } } return _persistent; } /* ------------------------------------------------------------ */ protected void firstWrite() throws IOException { log.debug("ajp13 firstWrite()"); } /* ------------------------------------------------------------ */ protected void commit() throws IOException { log.debug("ajp13 commit()"); if (_response.isCommitted()) return; _request.setHandled(true); getOutputStream().writeHeader(_response); } /* ------------------------------------------------------------ */ protected void setupOutputStream() throws IOException { // Nobble the OutputStream for HEAD requests if (HttpRequest.__HEAD.equals(getRequest().getMethod())) getOutputStream().nullOutput(); } }