test/jdk/java/net/httpclient/ProxyTest.java
author chegar
Sun, 05 Nov 2017 21:19:55 +0000
branchhttp-client-branch
changeset 55764 34d7cc00f87a
parent 47216 71c04702a3d5
child 49765 ee6f7a61f3a5
child 56076 9a2855e0a796
permissions -rw-r--r--
http-client-branch: WebSocket permission checks, test updates, and more
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
46138
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
     1
/*
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
     2
 * Copyright (c) 2017, Oracle and/or its affiliates. All rights reserved.
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
     3
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
     4
 *
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
     5
 * This code is free software; you can redistribute it and/or modify it
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
     6
 * under the terms of the GNU General Public License version 2 only, as
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
     7
 * published by the Free Software Foundation.
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
     8
 *
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
     9
 * This code is distributed in the hope that it will be useful, but WITHOUT
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    10
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    11
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    12
 * version 2 for more details (a copy is included in the LICENSE file that
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    13
 * accompanied this code).
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    14
 *
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    15
 * You should have received a copy of the GNU General Public License version
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    16
 * 2 along with this work; if not, write to the Free Software Foundation,
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    17
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    18
 *
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    19
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    20
 * or visit www.oracle.com if you need additional information or have any
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    21
 * questions.
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    22
 */
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    23
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    24
import com.sun.net.httpserver.HttpContext;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    25
import com.sun.net.httpserver.HttpExchange;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    26
import com.sun.net.httpserver.HttpHandler;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    27
import com.sun.net.httpserver.HttpServer;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    28
import com.sun.net.httpserver.HttpsConfigurator;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    29
import com.sun.net.httpserver.HttpsParameters;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    30
import com.sun.net.httpserver.HttpsServer;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    31
import java.io.IOException;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    32
import java.io.InputStream;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    33
import java.io.OutputStream;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    34
import java.io.OutputStreamWriter;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    35
import java.io.PrintWriter;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    36
import java.io.Writer;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    37
import java.net.HttpURLConnection;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    38
import java.net.InetAddress;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    39
import java.net.InetSocketAddress;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    40
import java.net.Proxy;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    41
import java.net.ProxySelector;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    42
import java.net.ServerSocket;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    43
import java.net.Socket;
55764
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
    44
import java.net.SocketAddress;
46138
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    45
import java.net.URI;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    46
import java.net.URISyntaxException;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    47
import java.nio.charset.StandardCharsets;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    48
import java.security.NoSuchAlgorithmException;
55764
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
    49
import java.util.List;
46138
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    50
import javax.net.ssl.HostnameVerifier;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    51
import javax.net.ssl.HttpsURLConnection;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    52
import javax.net.ssl.SSLContext;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    53
import javax.net.ssl.SSLSession;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    54
import jdk.incubator.http.HttpClient;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    55
import jdk.incubator.http.HttpRequest;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    56
import jdk.incubator.http.HttpResponse;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    57
import jdk.testlibrary.SimpleSSLContext;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    58
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    59
/**
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    60
 * @test
46157
f3c2dcb8d8fe 8181422: ClassCastException in HTTP Client
dfuchs
parents: 46138
diff changeset
    61
 * @bug 8185852 8181422
f3c2dcb8d8fe 8181422: ClassCastException in HTTP Client
dfuchs
parents: 46138
diff changeset
    62
 * @summary Verifies that passing a proxy with an unresolved address does
f3c2dcb8d8fe 8181422: ClassCastException in HTTP Client
dfuchs
parents: 46138
diff changeset
    63
 *          not cause java.nio.channels.UnresolvedAddressException.
f3c2dcb8d8fe 8181422: ClassCastException in HTTP Client
dfuchs
parents: 46138
diff changeset
    64
 *          Verifies that downgrading from HTTP/2 to HTTP/1.1 works through
f3c2dcb8d8fe 8181422: ClassCastException in HTTP Client
dfuchs
parents: 46138
diff changeset
    65
 *          an SSL Tunnel connection when the client is HTTP/2 and the server
f3c2dcb8d8fe 8181422: ClassCastException in HTTP Client
dfuchs
parents: 46138
diff changeset
    66
 *          and proxy are HTTP/1.1
46138
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    67
 * @modules jdk.incubator.httpclient
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    68
 * @library /lib/testlibrary/
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    69
 * @build jdk.testlibrary.SimpleSSLContext ProxyTest
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    70
 * @run main/othervm ProxyTest
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    71
 * @author danielfuchs
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    72
 */
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    73
public class ProxyTest {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    74
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    75
    static {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    76
        try {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    77
            HttpsURLConnection.setDefaultHostnameVerifier(new HostnameVerifier() {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    78
                    public boolean verify(String hostname, SSLSession session) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    79
                        return true;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    80
                    }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    81
                });
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    82
            SSLContext.setDefault(new SimpleSSLContext().get());
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    83
        } catch (IOException ex) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    84
            throw new ExceptionInInitializerError(ex);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    85
        }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    86
    }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    87
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    88
    static final String RESPONSE = "<html><body><p>Hello World!</body></html>";
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    89
    static final String PATH = "/foo/";
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    90
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    91
    static HttpServer createHttpsServer() throws IOException, NoSuchAlgorithmException {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    92
        HttpsServer server = com.sun.net.httpserver.HttpsServer.create();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    93
        HttpContext context = server.createContext(PATH);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    94
        context.setHandler(new HttpHandler() {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    95
            @Override
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    96
            public void handle(HttpExchange he) throws IOException {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    97
                he.getResponseHeaders().add("encoding", "UTF-8");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    98
                he.sendResponseHeaders(200, RESPONSE.length());
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
    99
                he.getResponseBody().write(RESPONSE.getBytes(StandardCharsets.UTF_8));
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   100
                he.close();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   101
            }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   102
        });
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   103
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   104
        server.setHttpsConfigurator(new Configurator(SSLContext.getDefault()));
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   105
        server.bind(new InetSocketAddress(InetAddress.getLoopbackAddress(), 0), 0);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   106
        return server;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   107
    }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   108
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   109
    public static void main(String[] args)
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   110
            throws IOException,
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   111
            URISyntaxException,
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   112
            NoSuchAlgorithmException,
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   113
            InterruptedException
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   114
    {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   115
        HttpServer server = createHttpsServer();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   116
        server.start();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   117
        try {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   118
            test(server, HttpClient.Version.HTTP_1_1);
46157
f3c2dcb8d8fe 8181422: ClassCastException in HTTP Client
dfuchs
parents: 46138
diff changeset
   119
            test(server, HttpClient.Version.HTTP_2);
46138
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   120
        } finally {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   121
            server.stop(0);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   122
            System.out.println("Server stopped");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   123
        }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   124
    }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   125
55764
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   126
    /**
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   127
     * A Proxy Selector that wraps a ProxySelector.of(), and counts the number
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   128
     * of times its select method has been invoked. This can be used to ensure
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   129
     * that the Proxy Selector is invoked only once per HttpClient.sendXXX
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   130
     * invocation.
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   131
     */
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   132
    static class CountingProxySelector extends ProxySelector {
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   133
        private final ProxySelector proxySelector;
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   134
        private volatile int count; // 0
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   135
        private CountingProxySelector(InetSocketAddress proxyAddress) {
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   136
            proxySelector = ProxySelector.of(proxyAddress);
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   137
        }
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   138
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   139
        public static CountingProxySelector of(InetSocketAddress proxyAddress) {
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   140
            return new CountingProxySelector(proxyAddress);
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   141
        }
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   142
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   143
        int count() { return count; }
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   144
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   145
        @Override
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   146
        public List<Proxy> select(URI uri) {
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   147
            count++;
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   148
            return proxySelector.select(uri);
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   149
        }
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   150
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   151
        @Override
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   152
        public void connectFailed(URI uri, SocketAddress sa, IOException ioe) {
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   153
            proxySelector.connectFailed(uri, sa, ioe);
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   154
        }
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   155
    }
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   156
46138
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   157
    public static void test(HttpServer server, HttpClient.Version version)
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   158
            throws IOException,
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   159
            URISyntaxException,
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   160
            NoSuchAlgorithmException,
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   161
            InterruptedException
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   162
    {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   163
        System.out.println("Server is: " + server.getAddress().toString());
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   164
        System.out.println("Verifying communication with server");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   165
        URI uri = new URI("https:/" + server.getAddress().toString() + PATH + "x");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   166
        try (InputStream is = uri.toURL().openConnection().getInputStream()) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   167
            String resp = new String(is.readAllBytes(), StandardCharsets.UTF_8);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   168
            System.out.println(resp);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   169
            if (!RESPONSE.equals(resp)) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   170
                throw new AssertionError("Unexpected response from server");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   171
            }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   172
        }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   173
        System.out.println("Communication with server OK");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   174
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   175
        TunnelingProxy proxy = new TunnelingProxy(server);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   176
        proxy.start();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   177
        try {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   178
            System.out.println("Proxy started");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   179
            Proxy p = new Proxy(Proxy.Type.HTTP,
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   180
                    InetSocketAddress.createUnresolved("localhost", proxy.getAddress().getPort()));
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   181
            System.out.println("Verifying communication with proxy");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   182
            HttpURLConnection conn = (HttpURLConnection)uri.toURL().openConnection(p);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   183
            try (InputStream is = conn.getInputStream()) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   184
                String resp = new String(is.readAllBytes(), StandardCharsets.UTF_8);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   185
                System.out.println(resp);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   186
                if (!RESPONSE.equals(resp)) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   187
                    throw new AssertionError("Unexpected response from proxy");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   188
                }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   189
            }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   190
            System.out.println("Communication with proxy OK");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   191
            System.out.println("\nReal test begins here.");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   192
            System.out.println("Setting up request with HttpClient for version: "
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   193
                    + version.name());
55764
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   194
            CountingProxySelector ps = CountingProxySelector.of(
46138
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   195
                    InetSocketAddress.createUnresolved("localhost", proxy.getAddress().getPort()));
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   196
            HttpClient client = HttpClient.newBuilder()
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   197
                .version(version)
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   198
                .proxy(ps)
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   199
                .build();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   200
            HttpRequest request = HttpRequest.newBuilder()
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   201
                .uri(uri)
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   202
                .GET()
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   203
                .build();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   204
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   205
            System.out.println("Sending request with HttpClient");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   206
            HttpResponse<String> response
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   207
                = client.send(request, HttpResponse.BodyHandler.asString());
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   208
            System.out.println("Got response");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   209
            String resp = response.body();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   210
            System.out.println("Received: " + resp);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   211
            if (!RESPONSE.equals(resp)) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   212
                throw new AssertionError("Unexpected response");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   213
            }
55764
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   214
            if (ps.count() > 1) {
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   215
                throw new AssertionError("CountingProxySelector. Expected 1, got " + ps.count());
34d7cc00f87a http-client-branch: WebSocket permission checks, test updates, and more
chegar
parents: 47216
diff changeset
   216
            }
46138
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   217
        } finally {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   218
            System.out.println("Stopping proxy");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   219
            proxy.stop();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   220
            System.out.println("Proxy stopped");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   221
        }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   222
    }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   223
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   224
    static class TunnelingProxy {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   225
        final Thread accept;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   226
        final ServerSocket ss;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   227
        final boolean DEBUG = false;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   228
        final HttpServer serverImpl;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   229
        TunnelingProxy(HttpServer serverImpl) throws IOException {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   230
            this.serverImpl = serverImpl;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   231
            ss = new ServerSocket();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   232
            accept = new Thread(this::accept);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   233
        }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   234
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   235
        void start() throws IOException {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   236
            ss.bind(new InetSocketAddress(InetAddress.getLoopbackAddress(), 0));
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   237
            accept.start();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   238
        }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   239
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   240
        // Pipe the input stream to the output stream.
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   241
        private synchronized Thread pipe(InputStream is, OutputStream os, char tag) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   242
            return new Thread("TunnelPipe("+tag+")") {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   243
                @Override
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   244
                public void run() {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   245
                    try {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   246
                        try {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   247
                            int c;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   248
                            while ((c = is.read()) != -1) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   249
                                os.write(c);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   250
                                os.flush();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   251
                                // if DEBUG prints a + or a - for each transferred
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   252
                                // character.
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   253
                                if (DEBUG) System.out.print(tag);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   254
                            }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   255
                            is.close();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   256
                        } finally {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   257
                            os.close();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   258
                        }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   259
                    } catch (IOException ex) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   260
                        if (DEBUG) ex.printStackTrace(System.out);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   261
                    }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   262
                }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   263
            };
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   264
        }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   265
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   266
        public InetSocketAddress getAddress() {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   267
            return new InetSocketAddress(ss.getInetAddress(), ss.getLocalPort());
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   268
        }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   269
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   270
        // This is a bit shaky. It doesn't handle continuation
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   271
        // lines, but our client shouldn't send any.
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   272
        // Read a line from the input stream, swallowing the final
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   273
        // \r\n sequence. Stops at the first \n, doesn't complain
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   274
        // if it wasn't preceded by '\r'.
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   275
        //
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   276
        String readLine(InputStream r) throws IOException {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   277
            StringBuilder b = new StringBuilder();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   278
            int c;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   279
            while ((c = r.read()) != -1) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   280
                if (c == '\n') break;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   281
                b.appendCodePoint(c);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   282
            }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   283
            if (b.codePointAt(b.length() -1) == '\r') {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   284
                b.delete(b.length() -1, b.length());
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   285
            }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   286
            return b.toString();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   287
        }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   288
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   289
        public void accept() {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   290
            Socket clientConnection = null;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   291
            try {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   292
                while (true) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   293
                    System.out.println("Tunnel: Waiting for client");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   294
                    Socket previous = clientConnection;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   295
                    try {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   296
                        clientConnection = ss.accept();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   297
                    } catch (IOException io) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   298
                        if (DEBUG) io.printStackTrace(System.out);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   299
                        break;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   300
                    } finally {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   301
                        // we have only 1 client at a time, so it is safe
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   302
                        // to close the previous connection here
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   303
                        if (previous != null) previous.close();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   304
                    }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   305
                    System.out.println("Tunnel: Client accepted");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   306
                    Socket targetConnection = null;
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   307
                    InputStream  ccis = clientConnection.getInputStream();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   308
                    OutputStream ccos = clientConnection.getOutputStream();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   309
                    Writer w = new OutputStreamWriter(ccos, "UTF-8");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   310
                    PrintWriter pw = new PrintWriter(w);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   311
                    System.out.println("Tunnel: Reading request line");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   312
                    String requestLine = readLine(ccis);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   313
                    System.out.println("Tunnel: Request status line: " + requestLine);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   314
                    if (requestLine.startsWith("CONNECT ")) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   315
                        // We should probably check that the next word following
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   316
                        // CONNECT is the host:port of our HTTPS serverImpl.
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   317
                        // Some improvement for a followup!
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   318
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   319
                        // Read all headers until we find the empty line that
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   320
                        // signals the end of all headers.
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   321
                        while(!requestLine.equals("")) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   322
                            System.out.println("Tunnel: Reading header: "
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   323
                                               + (requestLine = readLine(ccis)));
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   324
                        }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   325
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   326
                        // Open target connection
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   327
                        targetConnection = new Socket(
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   328
                                serverImpl.getAddress().getAddress(),
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   329
                                serverImpl.getAddress().getPort());
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   330
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   331
                        // Then send the 200 OK response to the client
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   332
                        System.out.println("Tunnel: Sending "
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   333
                                           + "HTTP/1.1 200 OK\r\n\r\n");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   334
                        pw.print("HTTP/1.1 200 OK\r\nContent-Length: 0\r\n\r\n");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   335
                        pw.flush();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   336
                    } else {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   337
                        // This should not happen.
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   338
                        throw new IOException("Tunnel: Unexpected status line: "
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   339
                                           + requestLine);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   340
                    }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   341
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   342
                    // Pipe the input stream of the client connection to the
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   343
                    // output stream of the target connection and conversely.
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   344
                    // Now the client and target will just talk to each other.
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   345
                    System.out.println("Tunnel: Starting tunnel pipes");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   346
                    Thread t1 = pipe(ccis, targetConnection.getOutputStream(), '+');
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   347
                    Thread t2 = pipe(targetConnection.getInputStream(), ccos, '-');
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   348
                    t1.start();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   349
                    t2.start();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   350
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   351
                    // We have only 1 client... wait until it has finished before
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   352
                    // accepting a new connection request.
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   353
                    // System.out.println("Tunnel: Waiting for pipes to close");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   354
                    // t1.join();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   355
                    // t2.join();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   356
                    System.out.println("Tunnel: Done - waiting for next client");
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   357
                }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   358
            } catch (Throwable ex) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   359
                try {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   360
                    ss.close();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   361
                } catch (IOException ex1) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   362
                    ex.addSuppressed(ex1);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   363
                }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   364
                ex.printStackTrace(System.err);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   365
            }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   366
        }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   367
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   368
        void stop() throws IOException {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   369
            ss.close();
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   370
        }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   371
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   372
    }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   373
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   374
    static class Configurator extends HttpsConfigurator {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   375
        public Configurator(SSLContext ctx) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   376
            super(ctx);
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   377
        }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   378
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   379
        @Override
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   380
        public void configure (HttpsParameters params) {
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   381
            params.setSSLParameters (getSSLContext().getSupportedSSLParameters());
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   382
        }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   383
    }
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   384
5982afcbf9dc 8185852: HttpConnection should resolve addresses before SocketChannel.connect() is called
dfuchs
parents:
diff changeset
   385
}