1   /*
2    * $HeadURL: https://svn.apache.org/repos/asf/httpcomponents/oac.hc3x/trunk/src/test/org/apache/commons/httpclient/server/TransparentProxyRequestHandler.java $
3    * $Revision$
4    * $Date$
5    *
6    * ====================================================================
7    *
8    *  Licensed to the Apache Software Foundation (ASF) under one or more
9    *  contributor license agreements.  See the NOTICE file distributed with
10   *  this work for additional information regarding copyright ownership.
11   *  The ASF licenses this file to You under the Apache License, Version 2.0
12   *  (the "License"); you may not use this file except in compliance with
13   *  the License.  You may obtain a copy of the License at
14   *
15   *      http://www.apache.org/licenses/LICENSE-2.0
16   *
17   *  Unless required by applicable law or agreed to in writing, software
18   *  distributed under the License is distributed on an "AS IS" BASIS,
19   *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
20   *  See the License for the specific language governing permissions and
21   *  limitations under the License.
22   * ====================================================================
23   *
24   * This software consists of voluntary contributions made by many
25   * individuals on behalf of the Apache Software Foundation.  For more
26   * information on the Apache Software Foundation, please see
27   * <http://www.apache.org/>.
28   *
29   */
30  
31  package org.apache.commons.httpclient.server;
32  
33  import java.io.IOException;
34  import java.io.InputStream;
35  import java.io.InterruptedIOException;
36  import java.io.OutputStream;
37  import java.net.Socket;
38  
39  import org.apache.commons.httpclient.Header;
40  import org.apache.commons.httpclient.HttpStatus;
41  import org.apache.commons.httpclient.HttpVersion;
42  
43  /***
44   * This request handler can handle the CONNECT method. It does nothing for any
45   * other HTTP methods.
46   * 
47   * @author Ortwin Glueck
48   */
49  public class TransparentProxyRequestHandler implements HttpRequestHandler {
50  
51      /*
52       * (non-Javadoc)
53       * 
54       * @see org.apache.commons.httpclient.server.HttpRequestHandler#processRequest(org.apache.commons.httpclient.server.SimpleHttpServerConnection)
55       */
56      public boolean processRequest(
57          final SimpleHttpServerConnection conn,
58          final SimpleRequest request) throws IOException
59      {
60  
61          RequestLine line = request.getRequestLine();
62          HttpVersion ver = line.getHttpVersion();
63          String method = line.getMethod();
64          if (!"CONNECT".equalsIgnoreCase(method)) {
65              return false;
66          }
67          Socket targetSocket = null;
68          try {
69              targetSocket = connect(line.getUri());
70          } catch (IOException e) {
71              SimpleResponse response = new SimpleResponse();
72              response.setStatusLine(ver, HttpStatus.SC_NOT_FOUND);
73              response.setHeader(new Header("Server", "test proxy"));
74              response.setBodyString("Cannot connect to " + line.getUri());
75              conn.writeResponse(response);
76              return true;
77          }
78          SimpleResponse response = new SimpleResponse();
79          response.setHeader(new Header("Server", "test proxy"));
80          response.setStatusLine(ver, HttpStatus.SC_OK, "Connection established");
81          conn.writeResponse(response);
82          
83          SimpleHttpServerConnection target = new SimpleHttpServerConnection(targetSocket); 
84          pump(conn, target);
85          return true;
86      }
87  
88      private void pump(final SimpleHttpServerConnection source, final SimpleHttpServerConnection target)
89          throws IOException {
90  
91          source.setSocketTimeout(100);
92          target.setSocketTimeout(100);
93  
94          InputStream sourceIn = source.getInputStream();
95          OutputStream sourceOut = source.getOutputStream();
96          InputStream targetIn = target.getInputStream();
97          OutputStream targetOut = target.getOutputStream();
98          
99          byte[] tmp = new byte[1024];
100         int l;
101         for (;;) {
102             if (!source.isOpen() || !target.isOpen()) { 
103                 break;
104             }
105             try {
106                 l = sourceIn.read(tmp);
107                 if (l == -1) {
108                     break;
109                 }
110                 targetOut.write(tmp, 0, l);
111             } catch (InterruptedIOException ignore) {
112                 if (Thread.interrupted()) {
113                     break;
114                 }
115             }
116             try {
117                 l = targetIn.read(tmp);
118                 if (l == -1) {
119                     break;
120                 }
121                 sourceOut.write(tmp, 0, l);
122             } catch (InterruptedIOException ignore) {
123                 if (Thread.interrupted()) {
124                     break;
125                 }
126             }
127         }
128     }
129     
130     private static Socket connect(final String host) throws IOException {
131         String hostname = null; 
132         int port; 
133         int i = host.indexOf(':');
134         if (i != -1) {
135             hostname = host.substring(0, i);
136             try {
137                 port = Integer.parseInt(host.substring(i + 1));
138             } catch (NumberFormatException ex) {
139                 throw new IOException("Invalid host address: " + host);
140             }
141         } else {
142             hostname = host;
143             port = 80;
144         }
145         return new Socket(hostname, port);        
146     }
147     
148 }