jdk/src/java.rmi/share/classes/sun/rmi/transport/proxy/HttpAwareServerSocket.java
changeset 38414 3e22d8fd4912
parent 38413 d9e49b45fd7d
parent 37898 ac36c71a44f9
child 38415 acea5f7d354b
equal deleted inserted replaced
38413:d9e49b45fd7d 38414:3e22d8fd4912
     1 /*
       
     2  * Copyright (c) 1996, 2005, 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.  Oracle designates this
       
     8  * particular file as subject to the "Classpath" exception as provided
       
     9  * by Oracle in the LICENSE file that accompanied this code.
       
    10  *
       
    11  * This code is distributed in the hope that it will be useful, but WITHOUT
       
    12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
       
    13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
       
    14  * version 2 for more details (a copy is included in the LICENSE file that
       
    15  * accompanied this code).
       
    16  *
       
    17  * You should have received a copy of the GNU General Public License version
       
    18  * 2 along with this work; if not, write to the Free Software Foundation,
       
    19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
       
    20  *
       
    21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
       
    22  * or visit www.oracle.com if you need additional information or have any
       
    23  * questions.
       
    24  */
       
    25 package sun.rmi.transport.proxy;
       
    26 
       
    27 import java.io.BufferedInputStream;
       
    28 import java.io.IOException;
       
    29 import java.net.ServerSocket;
       
    30 import java.net.Socket;
       
    31 import sun.rmi.runtime.Log;
       
    32 
       
    33 /**
       
    34  * The HttpAwareServerSocket class extends the java.net.ServerSocket
       
    35  * class.  It behaves like a ServerSocket, except that if
       
    36  * the first four bytes of an accepted socket are the letters "POST",
       
    37  * then it returns an HttpReceiveSocket instead of a java.net.Socket.
       
    38  * This means that the accept method blocks until four bytes have been
       
    39  * read from the new socket's input stream.
       
    40  */
       
    41 class HttpAwareServerSocket extends ServerSocket {
       
    42 
       
    43     /**
       
    44      * Create a server socket on a specified port.
       
    45      * @param port the port
       
    46      * @exception IOException IO error when opening the socket.
       
    47      */
       
    48     public HttpAwareServerSocket(int port) throws IOException
       
    49     {
       
    50         super(port);
       
    51     }
       
    52 
       
    53     /**
       
    54      * Create a server socket, bind it to the specified local port
       
    55      * and listen to it.  You can connect to an annonymous port by
       
    56      * specifying the port number to be 0.  <i>backlog</i> specifies
       
    57      * how many connection requests the system will queue up while waiting
       
    58      * for the ServerSocket to execute accept().
       
    59      * @param port the specified port
       
    60      * @param backlog the number of queued connect requests pending accept
       
    61      */
       
    62     public HttpAwareServerSocket(int port, int backlog) throws IOException
       
    63     {
       
    64         super(port, backlog);
       
    65     }
       
    66 
       
    67     /**
       
    68      * Accept a connection. This method will block until the connection
       
    69      * is made and four bytes can be read from the input stream.
       
    70      * If the first four bytes are "POST", then an HttpReceiveSocket is
       
    71      * returned, which will handle the HTTP protocol wrapping.
       
    72      * Otherwise, a WrappedSocket is returned.  The input stream will be
       
    73      * reset to the beginning of the transmission.
       
    74      * In either case, a BufferedInputStream will already be on top of
       
    75      * the underlying socket's input stream.
       
    76      * @exception IOException IO error when waiting for the connection.
       
    77      */
       
    78     public Socket accept() throws IOException
       
    79     {
       
    80         Socket socket = super.accept();
       
    81         BufferedInputStream in =
       
    82             new BufferedInputStream(socket.getInputStream());
       
    83 
       
    84         RMIMasterSocketFactory.proxyLog.log(Log.BRIEF,
       
    85             "socket accepted (checking for POST)");
       
    86 
       
    87         in.mark(4);
       
    88         boolean isHttp = (in.read() == 'P') &&
       
    89                          (in.read() == 'O') &&
       
    90                          (in.read() == 'S') &&
       
    91                          (in.read() == 'T');
       
    92         in.reset();
       
    93 
       
    94         if (RMIMasterSocketFactory.proxyLog.isLoggable(Log.BRIEF)) {
       
    95             RMIMasterSocketFactory.proxyLog.log(Log.BRIEF,
       
    96                 (isHttp ? "POST found, HTTP socket returned" :
       
    97                           "POST not found, direct socket returned"));
       
    98         }
       
    99 
       
   100         if (isHttp)
       
   101             return new HttpReceiveSocket(socket, in, null);
       
   102         else
       
   103             return new WrappedSocket(socket, in, null);
       
   104     }
       
   105 
       
   106     /**
       
   107      * Return the implementation address and implementation port of
       
   108      * the HttpAwareServerSocket as a String.
       
   109      */
       
   110     public String toString()
       
   111     {
       
   112         return "HttpAware" + super.toString();
       
   113     }
       
   114 }