| /* |
| * Copyright (c) 2003, 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. |
| * |
| * 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. |
| */ |
| |
| /* @test |
| * @bug 4920526 |
| * @summary Needs per connection proxy support for URLs |
| * @library ../../../sun/net/www/httptest/ |
| * @build ClosedChannelList HttpServer HttpTransaction HttpCallback |
| * @compile PerConnectionProxy.java |
| * @run main/othervm -Dhttp.proxyHost=inexistant -Dhttp.proxyPort=8080 PerConnectionProxy |
| */ |
| |
| import java.net.*; |
| import java.io.*; |
| import sun.net.www.*; |
| |
| public class PerConnectionProxy implements HttpCallback { |
| static HttpServer server; |
| |
| public void request (HttpTransaction req) { |
| req.setResponseEntityBody ("Hello ."); |
| try { |
| req.sendResponse (200, "Ok"); |
| req.orderlyClose(); |
| } catch (IOException e) { |
| } |
| } |
| |
| public static void main(String[] args) { |
| try { |
| server = new HttpServer (new PerConnectionProxy(), 1, 10, 0); |
| ProxyServer pserver = new ProxyServer(InetAddress.getByName("localhost"), server.getLocalPort()); |
| // start proxy server |
| new Thread(pserver).start(); |
| |
| URL url = new URL("http://localhost:"+server.getLocalPort()); |
| |
| // for non existing proxy expect an IOException |
| try { |
| InetSocketAddress isa = InetSocketAddress.createUnresolved("inexistent", 8080); |
| Proxy proxy = new Proxy(Proxy.Type.HTTP, isa); |
| HttpURLConnection urlc = (HttpURLConnection)url.openConnection (proxy); |
| InputStream is = urlc.getInputStream (); |
| is.close(); |
| throw new RuntimeException("non existing per connection proxy should lead to IOException"); |
| } catch (IOException ioex) { |
| // expected |
| } |
| |
| // for NO_PROXY, expect direct connection |
| try { |
| HttpURLConnection urlc = (HttpURLConnection)url.openConnection (Proxy.NO_PROXY); |
| int respCode = urlc.getResponseCode(); |
| urlc.disconnect(); |
| } catch (IOException ioex) { |
| throw new RuntimeException("direct connection should succeed :"+ioex.getMessage()); |
| } |
| |
| // for a normal proxy setting expect to see connection |
| // goes through that proxy |
| try { |
| InetSocketAddress isa = InetSocketAddress.createUnresolved("localhost", pserver.getPort()); |
| Proxy p = new Proxy(Proxy.Type.HTTP, isa); |
| HttpURLConnection urlc = (HttpURLConnection)url.openConnection (p); |
| int respCode = urlc.getResponseCode(); |
| urlc.disconnect(); |
| } catch (IOException ioex) { |
| throw new RuntimeException("connection through a local proxy should succeed :"+ioex.getMessage()); |
| } |
| |
| } catch (Exception e) { |
| throw new RuntimeException(e); |
| } finally { |
| if (server != null) { |
| server.terminate(); |
| } |
| } |
| |
| } |
| |
| static class ProxyServer extends Thread { |
| private static ServerSocket ss = null; |
| |
| // client requesting for a tunnel |
| private Socket clientSocket = null; |
| |
| /* |
| * Origin server's address and port that the client |
| * wants to establish the tunnel for communication. |
| */ |
| private InetAddress serverInetAddr; |
| private int serverPort; |
| |
| public ProxyServer(InetAddress server, int port) throws IOException { |
| serverInetAddr = server; |
| serverPort = port; |
| ss = new ServerSocket(0); |
| } |
| |
| public void run() { |
| try { |
| clientSocket = ss.accept(); |
| processRequests(); |
| } catch (Exception e) { |
| System.out.println("Proxy Failed: " + e); |
| e.printStackTrace(); |
| try { |
| ss.close(); |
| } |
| catch (IOException excep) { |
| System.out.println("ProxyServer close error: " + excep); |
| excep.printStackTrace(); |
| } |
| } |
| } |
| |
| private void processRequests() throws Exception { |
| // connection set to the tunneling mode |
| |
| Socket serverSocket = new Socket(serverInetAddr, serverPort); |
| ProxyTunnel clientToServer = new ProxyTunnel( |
| clientSocket, serverSocket); |
| ProxyTunnel serverToClient = new ProxyTunnel( |
| serverSocket, clientSocket); |
| clientToServer.start(); |
| serverToClient.start(); |
| System.out.println("Proxy: Started tunneling......."); |
| |
| clientToServer.join(); |
| serverToClient.join(); |
| System.out.println("Proxy: Finished tunneling........"); |
| |
| clientToServer.close(); |
| serverToClient.close(); |
| |
| } |
| |
| /** |
| *************************************************************** |
| * helper methods follow |
| *************************************************************** |
| */ |
| public int getPort() { |
| return ss.getLocalPort(); |
| } |
| /* |
| * This inner class provides unidirectional data flow through the sockets |
| * by continuously copying bytes from the input socket onto the output |
| * socket, until both sockets are open and EOF has not been received. |
| */ |
| static class ProxyTunnel extends Thread { |
| Socket sockIn; |
| Socket sockOut; |
| InputStream input; |
| OutputStream output; |
| |
| public ProxyTunnel(Socket sockIn, Socket sockOut) |
| throws Exception { |
| this.sockIn = sockIn; |
| this.sockOut = sockOut; |
| input = sockIn.getInputStream(); |
| output = sockOut.getOutputStream(); |
| } |
| |
| public void run() { |
| int BUFFER_SIZE = 400; |
| byte[] buf = new byte[BUFFER_SIZE]; |
| int bytesRead = 0; |
| int count = 0; // keep track of the amount of data transfer |
| |
| try { |
| while ((bytesRead = input.read(buf)) >= 0) { |
| output.write(buf, 0, bytesRead); |
| output.flush(); |
| count += bytesRead; |
| } |
| } catch (IOException e) { |
| /* |
| * The peer end has closed the connection |
| * we will close the tunnel |
| */ |
| close(); |
| } |
| } |
| |
| public void close() { |
| try { |
| if (!sockIn.isClosed()) |
| sockIn.close(); |
| if (!sockOut.isClosed()) |
| sockOut.close(); |
| } catch (IOException ignored) { } |
| } |
| } |
| |
| } |
| } |