299 lines
11 KiB
Java
299 lines
11 KiB
Java
/*
|
|
* Copyright (c) 2015, 2018, 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 java.io.IOException;
|
|
import java.net.SocketException;
|
|
import java.net.URI;
|
|
import java.util.ArrayList;
|
|
import java.util.EnumSet;
|
|
import java.util.HashSet;
|
|
import java.util.LinkedHashSet;
|
|
import java.util.List;
|
|
import java.util.concurrent.CompletableFuture;
|
|
import javax.net.ssl.SSLContext;
|
|
import javax.net.ServerSocketFactory;
|
|
import javax.net.ssl.SSLException;
|
|
import javax.net.ssl.SSLServerSocketFactory;
|
|
import java.net.http.HttpClient;
|
|
import java.net.http.HttpClient.Version;
|
|
import java.net.http.HttpRequest;
|
|
import java.net.http.HttpResponse;
|
|
import java.util.stream.Stream;
|
|
|
|
import jdk.test.lib.net.SimpleSSLContext;
|
|
import static java.lang.System.out;
|
|
import static java.lang.String.format;
|
|
import static java.nio.charset.StandardCharsets.ISO_8859_1;
|
|
import static java.net.http.HttpResponse.BodyHandlers.ofString;
|
|
|
|
/**
|
|
* @test
|
|
* @bug 8087112
|
|
* @library /test/lib
|
|
* @build jdk.test.lib.net.SimpleSSLContext
|
|
* @build MockServer
|
|
* @run main/othervm
|
|
* -Djdk.internal.httpclient.debug=true
|
|
* -Djdk.httpclient.HttpClient.log=all
|
|
* SplitResponse HTTP connection:CLOSE mode:SYNC
|
|
*/
|
|
|
|
/**
|
|
* Similar test to QuickResponses except that each byte of the response
|
|
* is sent in a separate packet, which tests the stability of the implementation
|
|
* for receiving unusual packet sizes. Additionally, tests scenarios there
|
|
* connections that are retrieved from the connection pool may reach EOF before
|
|
* being reused.
|
|
*/
|
|
public class SplitResponse {
|
|
|
|
static String response(String body, boolean serverKeepalive) {
|
|
StringBuilder sb = new StringBuilder();
|
|
sb.append("HTTP/1.1 200 OK\r\n");
|
|
if (!serverKeepalive)
|
|
sb.append("Connection: Close\r\n");
|
|
|
|
sb.append("Content-length: ")
|
|
.append(body.getBytes(ISO_8859_1).length)
|
|
.append("\r\n");
|
|
sb.append("\r\n");
|
|
sb.append(body);
|
|
return sb.toString();
|
|
}
|
|
|
|
static final String responses[] = {
|
|
"Lorem ipsum",
|
|
"dolor sit amet",
|
|
"consectetur adipiscing elit, sed do eiusmod tempor",
|
|
"quis nostrud exercitation ullamco",
|
|
"laboris nisi",
|
|
"ut",
|
|
"aliquip ex ea commodo consequat." +
|
|
"Duis aute irure dolor in reprehenderit in voluptate velit esse" +
|
|
"cillum dolore eu fugiat nulla pariatur.",
|
|
"Excepteur sint occaecat cupidatat non proident."
|
|
};
|
|
|
|
final ServerSocketFactory factory;
|
|
final SSLContext context;
|
|
final boolean useSSL;
|
|
SplitResponse(boolean useSSL) throws IOException {
|
|
this.useSSL = useSSL;
|
|
context = new SimpleSSLContext().get();
|
|
SSLContext.setDefault(context);
|
|
factory = useSSL ? SSLServerSocketFactory.getDefault()
|
|
: ServerSocketFactory.getDefault();
|
|
}
|
|
|
|
public HttpClient newHttpClient() {
|
|
HttpClient client;
|
|
if (useSSL) {
|
|
client = HttpClient.newBuilder()
|
|
.sslContext(context)
|
|
.build();
|
|
} else {
|
|
client = HttpClient.newHttpClient();
|
|
}
|
|
return client;
|
|
}
|
|
|
|
enum Protocol {
|
|
HTTP, HTTPS
|
|
}
|
|
enum Connection {
|
|
KEEP_ALIVE,
|
|
CLOSE
|
|
}
|
|
enum Mode {
|
|
SYNC, ASYNC
|
|
}
|
|
|
|
|
|
public static void main(String[] args) throws Exception {
|
|
boolean useSSL = false;
|
|
if (args != null && args.length >= 1) {
|
|
useSSL = Protocol.valueOf(args[0]).equals(Protocol.HTTPS);
|
|
} else {
|
|
args = new String[] {"HTTP", "connection:KEEP_ALIVE:CLOSE", "mode:SYNC:ASYNC"};
|
|
}
|
|
|
|
LinkedHashSet<Mode> modes = new LinkedHashSet<>();
|
|
LinkedHashSet<Connection> keepAlive = new LinkedHashSet<>();
|
|
Stream.of(args).skip(1).forEach(s -> {
|
|
if (s.startsWith("connection:")) {
|
|
Stream.of(s.split(":")).skip(1).forEach(c -> {
|
|
keepAlive.add(Connection.valueOf(c));
|
|
});
|
|
} else if (s.startsWith("mode:")) {
|
|
Stream.of(s.split(":")).skip(1).forEach(m -> {
|
|
modes.add(Mode.valueOf(m));
|
|
});
|
|
} else {
|
|
System.err.println("Illegal argument: " + s);
|
|
System.err.println("Allowed syntax is: HTTP|HTTPS [connection:KEEP_ALIVE[:CLOSE]] [mode:SYNC[:ASYNC]");
|
|
throw new IllegalArgumentException(s);
|
|
}
|
|
});
|
|
|
|
if (keepAlive.isEmpty()) keepAlive.addAll(EnumSet.allOf(Connection.class));
|
|
if (modes.isEmpty()) modes.addAll(EnumSet.allOf(Mode.class));
|
|
|
|
SplitResponse sp = new SplitResponse(useSSL);
|
|
|
|
for (Version version : Version.values()) {
|
|
for (Connection serverKeepalive : keepAlive) {
|
|
// Note: the mock server doesn't support Keep-Alive, but
|
|
// pretending that it might exercises code paths in and out of
|
|
// the connection pool, and retry logic
|
|
for (Mode mode : modes) {
|
|
sp.test(version,serverKeepalive == Connection.KEEP_ALIVE,mode == Mode.ASYNC);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// @Test
|
|
void test(Version version, boolean serverKeepalive, boolean async)
|
|
throws Exception
|
|
{
|
|
out.println(format("*** version %s, serverKeepAlive: %s, async: %s ***",
|
|
version, serverKeepalive, async));
|
|
MockServer server = new MockServer(0, factory);
|
|
URI uri = new URI(server.getURL());
|
|
out.println("server is: " + uri);
|
|
server.start();
|
|
|
|
|
|
// The following code can be uncommented to verify that the
|
|
// MockServer will reject rogue requests whose URI does not
|
|
// contain "/foo/".
|
|
//
|
|
// Thread rogue = new Thread() {
|
|
// public void run() {
|
|
// try {
|
|
// HttpClient client = newHttpClient();
|
|
// URI uri2 = URI.create(uri.toString().replace("/foo/","/"));
|
|
// HttpRequest request = HttpRequest
|
|
// .newBuilder(uri2).version(version).build();
|
|
// while (true) {
|
|
// try {
|
|
// client.send(request, HttpResponse.BodyHandlers.ofString());
|
|
// } catch (IOException ex) {
|
|
// System.out.println("Client rejected " + request);
|
|
// }
|
|
// sleep(250);
|
|
// }
|
|
// } catch ( Throwable x) {
|
|
// }
|
|
// }
|
|
// };
|
|
// rogue.setDaemon(true);
|
|
// rogue.start();
|
|
|
|
|
|
HttpClient client = newHttpClient();
|
|
HttpRequest request = HttpRequest.newBuilder(uri).version(version).build();
|
|
HttpResponse<String> r;
|
|
CompletableFuture<HttpResponse<String>> cf1;
|
|
|
|
try {
|
|
for (int i=0; i<responses.length; i++) {
|
|
out.println("----- iteration " + i + " -----");
|
|
String body = responses[i];
|
|
Thread t = sendSplitResponse(response(body, serverKeepalive), server);
|
|
|
|
if (async) {
|
|
out.println("send async: " + request);
|
|
cf1 = client.sendAsync(request, ofString());
|
|
r = cf1.get();
|
|
} else { // sync
|
|
out.println("send sync: " + request);
|
|
r = client.send(request, ofString());
|
|
}
|
|
|
|
out.println("response " + r);
|
|
String rxbody = r.body();
|
|
out.println("response body:[" + rxbody + "]");
|
|
|
|
if (r.statusCode() != 200)
|
|
throw new RuntimeException("Expected 200, got:" + r.statusCode());
|
|
|
|
if (!rxbody.equals(body))
|
|
throw new RuntimeException(format("Expected:%s, got:%s", body, rxbody));
|
|
|
|
t.join();
|
|
conn.close();
|
|
}
|
|
} finally {
|
|
server.close();
|
|
}
|
|
System.out.println("OK");
|
|
}
|
|
|
|
// required for cleanup
|
|
volatile MockServer.Connection conn;
|
|
|
|
// Sends the response, mostly, one byte at a time with a small delay
|
|
// between bytes, to encourage that each byte is read in a separate read
|
|
Thread sendSplitResponse(String s, MockServer server) {
|
|
System.out.println("Server: creating new thread to send ... ");
|
|
Thread t = new Thread(() -> {
|
|
System.out.println("Server: waiting for server to receive headers");
|
|
conn = server.activity();
|
|
System.out.println("Server: Start sending response");
|
|
|
|
try {
|
|
int len = s.length();
|
|
out.println("Server: going to send [" + s + "]");
|
|
for (int i = 0; i < len; i++) {
|
|
String onechar = s.substring(i, i + 1);
|
|
try {
|
|
conn.send(onechar);
|
|
} catch(SocketException | SSLException x) {
|
|
if (!useSSL || i != len - 1) throw x;
|
|
if (x.getMessage().contains("closed by remote host")) {
|
|
String osname = System.getProperty("os.name", "unknown");
|
|
// On Solaris we can receive an exception when
|
|
// the client closes the connection after receiving
|
|
// the last expected char.
|
|
if (osname.contains("SunO")) {
|
|
System.out.println(osname + " detected");
|
|
System.out.println("WARNING: ignoring " + x);
|
|
System.err.println(osname + " detected");
|
|
System.err.println("WARNING: ignoring " + x);
|
|
}
|
|
}
|
|
}
|
|
Thread.sleep(10);
|
|
}
|
|
out.println("Server: sent [" + s + "]");
|
|
} catch (IOException | InterruptedException e) {
|
|
throw new RuntimeException(e);
|
|
}
|
|
});
|
|
t.setDaemon(true);
|
|
t.start();
|
|
return t;
|
|
}
|
|
}
|