jdk-sandbox: comparison jdk/test/java/net/httpclient/ProxyTest.java

equal deleted inserted replaced

-:14df107500cc
+:5982afcbf9dc
+/*
+* Copyright (c) 2017, 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.
+*/
+import com.sun.net.httpserver.HttpContext;
+import com.sun.net.httpserver.HttpExchange;
+import com.sun.net.httpserver.HttpHandler;
+import com.sun.net.httpserver.HttpServer;
+import com.sun.net.httpserver.HttpsConfigurator;
+import com.sun.net.httpserver.HttpsParameters;
+import com.sun.net.httpserver.HttpsServer;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.io.OutputStreamWriter;
+import java.io.PrintWriter;
+import java.io.Writer;
+import java.net.HttpURLConnection;
+import java.net.InetAddress;
+import java.net.InetSocketAddress;
+import java.net.Proxy;
+import java.net.ProxySelector;
+import java.net.ServerSocket;
+import java.net.Socket;
+import java.net.URI;
+import java.net.URISyntaxException;
+import java.nio.charset.StandardCharsets;
+import java.security.NoSuchAlgorithmException;
+import javax.net.ssl.HostnameVerifier;
+import javax.net.ssl.HttpsURLConnection;
+import javax.net.ssl.SSLContext;
+import javax.net.ssl.SSLSession;
+import jdk.incubator.http.HttpClient;
+import jdk.incubator.http.HttpRequest;
+import jdk.incubator.http.HttpResponse;
+import jdk.testlibrary.SimpleSSLContext;
+/**
+* @test
+* @bug 8185852
+* @summary verifies that passing a proxy with an unresolved address does
+*          not cause  java.nio.channels.UnresolvedAddressException
+* @modules jdk.incubator.httpclient
+* @library /lib/testlibrary/
+* @build jdk.testlibrary.SimpleSSLContext ProxyTest
+* @run main/othervm ProxyTest
+* @author danielfuchs
+*/
+public class ProxyTest {
+static {
+try {
+HttpsURLConnection.setDefaultHostnameVerifier(new HostnameVerifier() {
+public boolean verify(String hostname, SSLSession session) {
+return true;
+}
+});
+SSLContext.setDefault(new SimpleSSLContext().get());
+} catch (IOException ex) {
+throw new ExceptionInInitializerError(ex);
+}
+}
+static final String RESPONSE = "<html><body><p>Hello World!</body></html>";
+static final String PATH = "/foo/";
+static HttpServer createHttpsServer() throws IOException, NoSuchAlgorithmException {
+HttpsServer server = com.sun.net.httpserver.HttpsServer.create();
+HttpContext context = server.createContext(PATH);
+context.setHandler(new HttpHandler() {
+@Override
+public void handle(HttpExchange he) throws IOException {
+he.getResponseHeaders().add("encoding", "UTF-8");
+he.sendResponseHeaders(200, RESPONSE.length());
+he.getResponseBody().write(RESPONSE.getBytes(StandardCharsets.UTF_8));
+he.close();
+}
+});
+server.setHttpsConfigurator(new Configurator(SSLContext.getDefault()));
+server.bind(new InetSocketAddress(InetAddress.getLoopbackAddress(), 0), 0);
+return server;
+}
+public static void main(String[] args)
+throws IOException,
+URISyntaxException,
+NoSuchAlgorithmException,
+InterruptedException
+{
+HttpServer server = createHttpsServer();
+server.start();
+try {
+test(server, HttpClient.Version.HTTP_1_1);
+// test(server, HttpClient.Version.HTTP_2);
+} finally {
+server.stop(0);
+System.out.println("Server stopped");
+}
+}
+public static void test(HttpServer server, HttpClient.Version version)
+throws IOException,
+URISyntaxException,
+NoSuchAlgorithmException,
+InterruptedException
+{
+System.out.println("Server is: " + server.getAddress().toString());
+System.out.println("Verifying communication with server");
+URI uri = new URI("https:/" + server.getAddress().toString() + PATH + "x");
+try (InputStream is = uri.toURL().openConnection().getInputStream()) {
+String resp = new String(is.readAllBytes(), StandardCharsets.UTF_8);
+System.out.println(resp);
+if (!RESPONSE.equals(resp)) {
+throw new AssertionError("Unexpected response from server");
+}
+}
+System.out.println("Communication with server OK");
+TunnelingProxy proxy = new TunnelingProxy(server);
+proxy.start();
+try {
+System.out.println("Proxy started");
+Proxy p = new Proxy(Proxy.Type.HTTP,
+InetSocketAddress.createUnresolved("localhost", proxy.getAddress().getPort()));
+System.out.println("Verifying communication with proxy");
+HttpURLConnection conn = (HttpURLConnection)uri.toURL().openConnection(p);
+try (InputStream is = conn.getInputStream()) {
+String resp = new String(is.readAllBytes(), StandardCharsets.UTF_8);
+System.out.println(resp);
+if (!RESPONSE.equals(resp)) {
+throw new AssertionError("Unexpected response from proxy");
+}
+}
+System.out.println("Communication with proxy OK");
+System.out.println("\nReal test begins here.");
+System.out.println("Setting up request with HttpClient for version: "
++ version.name());
+ProxySelector ps = ProxySelector.of(
+InetSocketAddress.createUnresolved("localhost", proxy.getAddress().getPort()));
+HttpClient client = HttpClient.newBuilder()
+.version(version)
+.proxy(ps)
+.build();
+HttpRequest request = HttpRequest.newBuilder()
+.uri(uri)
+.GET()
+.build();
+System.out.println("Sending request with HttpClient");
+HttpResponse<String> response
+= client.send(request, HttpResponse.BodyHandler.asString());
+System.out.println("Got response");
+String resp = response.body();
+System.out.println("Received: " + resp);
+if (!RESPONSE.equals(resp)) {
+throw new AssertionError("Unexpected response");
+}
+} finally {
+System.out.println("Stopping proxy");
+proxy.stop();
+System.out.println("Proxy stopped");
+}
+}
+static class TunnelingProxy {
+final Thread accept;
+final ServerSocket ss;
+final boolean DEBUG = false;
+final HttpServer serverImpl;
+TunnelingProxy(HttpServer serverImpl) throws IOException {
+this.serverImpl = serverImpl;
+ss = new ServerSocket();
+accept = new Thread(this::accept);
+}
+void start() throws IOException {
+ss.bind(new InetSocketAddress(InetAddress.getLoopbackAddress(), 0));
+accept.start();
+}
+// Pipe the input stream to the output stream.
+private synchronized Thread pipe(InputStream is, OutputStream os, char tag) {
+return new Thread("TunnelPipe("+tag+")") {
+@Override
+public void run() {
+try {
+try {
+int c;
+while ((c = is.read()) != -1) {
+os.write(c);
+os.flush();
+// if DEBUG prints a + or a - for each transferred
+// character.
+if (DEBUG) System.out.print(tag);
+}
+is.close();
+} finally {
+os.close();
+}
+} catch (IOException ex) {
+if (DEBUG) ex.printStackTrace(System.out);
+}
+}
+};
+}
+public InetSocketAddress getAddress() {
+return new InetSocketAddress(ss.getInetAddress(), ss.getLocalPort());
+}
+// This is a bit shaky. It doesn't handle continuation
+// lines, but our client shouldn't send any.
+// Read a line from the input stream, swallowing the final
+// \r\n sequence. Stops at the first \n, doesn't complain
+// if it wasn't preceded by '\r'.
+//
+String readLine(InputStream r) throws IOException {
+StringBuilder b = new StringBuilder();
+int c;
+while ((c = r.read()) != -1) {
+if (c == '\n') break;
+b.appendCodePoint(c);
+}
+if (b.codePointAt(b.length() -1) == '\r') {
+b.delete(b.length() -1, b.length());
+}
+return b.toString();
+}
+public void accept() {
+Socket clientConnection = null;
+try {
+while (true) {
+System.out.println("Tunnel: Waiting for client");
+Socket previous = clientConnection;
+try {
+clientConnection = ss.accept();
+} catch (IOException io) {
+if (DEBUG) io.printStackTrace(System.out);
+break;
+} finally {
+// we have only 1 client at a time, so it is safe
+// to close the previous connection here
+if (previous != null) previous.close();
+}
+System.out.println("Tunnel: Client accepted");
+Socket targetConnection = null;
+InputStream  ccis = clientConnection.getInputStream();
+OutputStream ccos = clientConnection.getOutputStream();
+Writer w = new OutputStreamWriter(ccos, "UTF-8");
+PrintWriter pw = new PrintWriter(w);
+System.out.println("Tunnel: Reading request line");
+String requestLine = readLine(ccis);
+System.out.println("Tunnel: Request status line: " + requestLine);
+if (requestLine.startsWith("CONNECT ")) {
+// We should probably check that the next word following
+// CONNECT is the host:port of our HTTPS serverImpl.
+// Some improvement for a followup!
+// Read all headers until we find the empty line that
+// signals the end of all headers.
+while(!requestLine.equals("")) {
+System.out.println("Tunnel: Reading header: "
++ (requestLine = readLine(ccis)));
+}
+// Open target connection
+targetConnection = new Socket(
+serverImpl.getAddress().getAddress(),
+serverImpl.getAddress().getPort());
+// Then send the 200 OK response to the client
+System.out.println("Tunnel: Sending "
++ "HTTP/1.1 200 OK\r\n\r\n");
+pw.print("HTTP/1.1 200 OK\r\nContent-Length: 0\r\n\r\n");
+pw.flush();
+} else {
+// This should not happen.
+throw new IOException("Tunnel: Unexpected status line: "
++ requestLine);
+}
+// Pipe the input stream of the client connection to the
+// output stream of the target connection and conversely.
+// Now the client and target will just talk to each other.
+System.out.println("Tunnel: Starting tunnel pipes");
+Thread t1 = pipe(ccis, targetConnection.getOutputStream(), '+');
+Thread t2 = pipe(targetConnection.getInputStream(), ccos, '-');
+t1.start();
+t2.start();
+// We have only 1 client... wait until it has finished before
+// accepting a new connection request.
+// System.out.println("Tunnel: Waiting for pipes to close");
+// t1.join();
+// t2.join();
+System.out.println("Tunnel: Done - waiting for next client");
+}
+} catch (Throwable ex) {
+try {
+ss.close();
+} catch (IOException ex1) {
+ex.addSuppressed(ex1);
+}
+ex.printStackTrace(System.err);
+}
+}
+void stop() throws IOException {
+ss.close();
+}
+}
+static class Configurator extends HttpsConfigurator {
+public Configurator(SSLContext ctx) {
+super(ctx);
+}
+@Override
+public void configure (HttpsParameters params) {
+params.setSSLParameters (getSSLContext().getSupportedSSLParameters());
+}
+}
+}

changeset 46138	5982afcbf9dc
child 46157	f3c2dcb8d8fe