create Http2FrameAdapter test and InboundHttp2ToHttpAdapter test
This commit is contained in:
parent
1e1b8469b2
commit
4ca2ff395d
3 changed files with 344 additions and 76 deletions
|
@ -65,7 +65,8 @@ jar {
|
||||||
|
|
||||||
test {
|
test {
|
||||||
jvmArgs "-javaagent:" + configurations.alpnagent.asPath
|
jvmArgs "-javaagent:" + configurations.alpnagent.asPath
|
||||||
//include 'org/xbib/netty/http/client/test/Http2Test*'
|
include 'org/xbib/netty/http/client/test/Http2FrameAdapterTest*'
|
||||||
|
include 'org/xbib/netty/http/client/test/InboundHttp2ToHttpAdapterTest*'
|
||||||
testLogging {
|
testLogging {
|
||||||
showStandardStreams = false
|
showStandardStreams = false
|
||||||
exceptionFormat = 'full'
|
exceptionFormat = 'full'
|
||||||
|
|
|
@ -4,16 +4,13 @@ import io.netty.bootstrap.Bootstrap;
|
||||||
import io.netty.buffer.ByteBuf;
|
import io.netty.buffer.ByteBuf;
|
||||||
import io.netty.channel.Channel;
|
import io.netty.channel.Channel;
|
||||||
import io.netty.channel.ChannelHandlerContext;
|
import io.netty.channel.ChannelHandlerContext;
|
||||||
import io.netty.channel.ChannelInboundHandlerAdapter;
|
|
||||||
import io.netty.channel.ChannelInitializer;
|
import io.netty.channel.ChannelInitializer;
|
||||||
|
import io.netty.channel.ChannelPromise;
|
||||||
import io.netty.channel.EventLoopGroup;
|
import io.netty.channel.EventLoopGroup;
|
||||||
import io.netty.channel.nio.NioEventLoopGroup;
|
import io.netty.channel.nio.NioEventLoopGroup;
|
||||||
import io.netty.channel.socket.nio.NioSocketChannel;
|
import io.netty.channel.socket.nio.NioSocketChannel;
|
||||||
import io.netty.handler.codec.http.HttpMethod;
|
import io.netty.handler.codec.http.HttpMethod;
|
||||||
import io.netty.handler.codec.http2.DefaultHttp2Connection;
|
|
||||||
import io.netty.handler.codec.http2.DefaultHttp2Headers;
|
import io.netty.handler.codec.http2.DefaultHttp2Headers;
|
||||||
import io.netty.handler.codec.http2.DelegatingDecompressorFrameListener;
|
|
||||||
import io.netty.handler.codec.http2.Http2Connection;
|
|
||||||
import io.netty.handler.codec.http2.Http2ConnectionHandler;
|
import io.netty.handler.codec.http2.Http2ConnectionHandler;
|
||||||
import io.netty.handler.codec.http2.Http2ConnectionHandlerBuilder;
|
import io.netty.handler.codec.http2.Http2ConnectionHandlerBuilder;
|
||||||
import io.netty.handler.codec.http2.Http2Exception;
|
import io.netty.handler.codec.http2.Http2Exception;
|
||||||
|
@ -21,8 +18,8 @@ import io.netty.handler.codec.http2.Http2FrameAdapter;
|
||||||
import io.netty.handler.codec.http2.Http2FrameLogger;
|
import io.netty.handler.codec.http2.Http2FrameLogger;
|
||||||
import io.netty.handler.codec.http2.Http2SecurityUtil;
|
import io.netty.handler.codec.http2.Http2SecurityUtil;
|
||||||
import io.netty.handler.codec.http2.Http2Settings;
|
import io.netty.handler.codec.http2.Http2Settings;
|
||||||
import io.netty.handler.codec.http2.InboundHttp2ToHttpAdapterBuilder;
|
|
||||||
import io.netty.handler.logging.LogLevel;
|
import io.netty.handler.logging.LogLevel;
|
||||||
|
import io.netty.handler.logging.LoggingHandler;
|
||||||
import io.netty.handler.ssl.ApplicationProtocolConfig;
|
import io.netty.handler.ssl.ApplicationProtocolConfig;
|
||||||
import io.netty.handler.ssl.ApplicationProtocolNames;
|
import io.netty.handler.ssl.ApplicationProtocolNames;
|
||||||
import io.netty.handler.ssl.SslContext;
|
import io.netty.handler.ssl.SslContext;
|
||||||
|
@ -32,14 +29,11 @@ import io.netty.handler.ssl.SslProvider;
|
||||||
import io.netty.handler.ssl.SupportedCipherSuiteFilter;
|
import io.netty.handler.ssl.SupportedCipherSuiteFilter;
|
||||||
import io.netty.handler.ssl.util.InsecureTrustManagerFactory;
|
import io.netty.handler.ssl.util.InsecureTrustManagerFactory;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
import org.xbib.netty.http.client.Http2Handler;
|
|
||||||
import org.xbib.netty.http.client.TrafficLoggingHandler;
|
|
||||||
|
|
||||||
import javax.net.ssl.SNIHostName;
|
import javax.net.ssl.SNIHostName;
|
||||||
import javax.net.ssl.SNIServerName;
|
import javax.net.ssl.SNIServerName;
|
||||||
import javax.net.ssl.SSLEngine;
|
import javax.net.ssl.SSLEngine;
|
||||||
import javax.net.ssl.SSLParameters;
|
import javax.net.ssl.SSLParameters;
|
||||||
import java.io.InputStream;
|
|
||||||
import java.net.InetSocketAddress;
|
import java.net.InetSocketAddress;
|
||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
import java.util.concurrent.CountDownLatch;
|
import java.util.concurrent.CountDownLatch;
|
||||||
|
@ -52,7 +46,7 @@ import java.util.logging.SimpleFormatter;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
*/
|
*/
|
||||||
public class Http2Test {
|
public class Http2FrameAdapterTest {
|
||||||
|
|
||||||
static {
|
static {
|
||||||
System.setProperty("java.util.logging.SimpleFormatter.format",
|
System.setProperty("java.util.logging.SimpleFormatter.format",
|
||||||
|
@ -69,13 +63,11 @@ public class Http2Test {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
private static final Logger logger = Logger.getLogger("");
|
private static final Logger logger = Logger.getLogger("");
|
||||||
|
|
||||||
private final int serverExpectedDataFrames = 1;
|
@Test
|
||||||
|
public void testHttp2FrameAdapter() throws Exception {
|
||||||
|
final int serverExpectedDataFrames = 1;
|
||||||
public void testGeneric() throws Exception {
|
|
||||||
final InetSocketAddress inetSocketAddress = new InetSocketAddress("http2-push.io", 443);
|
final InetSocketAddress inetSocketAddress = new InetSocketAddress("http2-push.io", 443);
|
||||||
final CountDownLatch dataLatch = new CountDownLatch(serverExpectedDataFrames);
|
final CountDownLatch dataLatch = new CountDownLatch(serverExpectedDataFrames);
|
||||||
EventLoopGroup group = new NioEventLoopGroup();
|
EventLoopGroup group = new NioEventLoopGroup();
|
||||||
|
@ -91,14 +83,12 @@ public class Http2Test {
|
||||||
SslContext sslContext = SslContextBuilder.forClient()
|
SslContext sslContext = SslContextBuilder.forClient()
|
||||||
.sslProvider(SslProvider.OPENSSL)
|
.sslProvider(SslProvider.OPENSSL)
|
||||||
.trustManager(InsecureTrustManagerFactory.INSTANCE)
|
.trustManager(InsecureTrustManagerFactory.INSTANCE)
|
||||||
.keyManager((InputStream) null, null, null)
|
|
||||||
.ciphers(Http2SecurityUtil.CIPHERS, SupportedCipherSuiteFilter.INSTANCE)
|
.ciphers(Http2SecurityUtil.CIPHERS, SupportedCipherSuiteFilter.INSTANCE)
|
||||||
.applicationProtocolConfig(new ApplicationProtocolConfig(
|
.applicationProtocolConfig(new ApplicationProtocolConfig(
|
||||||
ApplicationProtocolConfig.Protocol.ALPN,
|
ApplicationProtocolConfig.Protocol.ALPN,
|
||||||
ApplicationProtocolConfig.SelectorFailureBehavior.NO_ADVERTISE,
|
ApplicationProtocolConfig.SelectorFailureBehavior.NO_ADVERTISE,
|
||||||
ApplicationProtocolConfig.SelectedListenerFailureBehavior.ACCEPT,
|
ApplicationProtocolConfig.SelectedListenerFailureBehavior.ACCEPT,
|
||||||
ApplicationProtocolNames.HTTP_2,
|
ApplicationProtocolNames.HTTP_2))
|
||||||
ApplicationProtocolNames.HTTP_1_1))
|
|
||||||
.build();
|
.build();
|
||||||
SslHandler sslHandler = sslContext.newHandler(ch.alloc());
|
SslHandler sslHandler = sslContext.newHandler(ch.alloc());
|
||||||
SSLEngine engine = sslHandler.engine();
|
SSLEngine engine = sslHandler.engine();
|
||||||
|
@ -113,6 +103,7 @@ public class Http2Test {
|
||||||
@Override
|
@Override
|
||||||
public void onSettingsRead(ChannelHandlerContext ctx, Http2Settings settings)
|
public void onSettingsRead(ChannelHandlerContext ctx, Http2Settings settings)
|
||||||
throws Http2Exception {
|
throws Http2Exception {
|
||||||
|
logger.log(Level.FINE, "settings received, now writing headers");
|
||||||
Http2ConnectionHandler handler = ctx.pipeline().get(Http2ConnectionHandler.class);
|
Http2ConnectionHandler handler = ctx.pipeline().get(Http2ConnectionHandler.class);
|
||||||
handler.encoder().writeHeaders(ctx, 3,
|
handler.encoder().writeHeaders(ctx, 3,
|
||||||
new DefaultHttp2Headers().method(HttpMethod.GET.asciiName())
|
new DefaultHttp2Headers().method(HttpMethod.GET.asciiName())
|
||||||
|
@ -135,7 +126,9 @@ public class Http2Test {
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
clientChannel = bs.connect(inetSocketAddress).syncUninterruptibly().channel();
|
clientChannel = bs.connect(inetSocketAddress).syncUninterruptibly().channel();
|
||||||
|
logger.log(Level.INFO, () -> "waiting for HTTP/2 data");
|
||||||
dataLatch.await();
|
dataLatch.await();
|
||||||
|
logger.log(Level.INFO, () -> "done, data arrived");
|
||||||
} finally {
|
} finally {
|
||||||
if (clientChannel != null) {
|
if (clientChannel != null) {
|
||||||
clientChannel.close();
|
clientChannel.close();
|
||||||
|
@ -144,66 +137,34 @@ public class Http2Test {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
class TrafficLoggingHandler extends LoggingHandler {
|
||||||
|
|
||||||
public void testHttpAdapter() throws Exception {
|
TrafficLoggingHandler() {
|
||||||
final InetSocketAddress inetSocketAddress = new InetSocketAddress("http2-push.io", 443);
|
super("client", LogLevel.TRACE);
|
||||||
final CountDownLatch dataLatch = new CountDownLatch(serverExpectedDataFrames);
|
}
|
||||||
EventLoopGroup group = new NioEventLoopGroup();
|
|
||||||
Channel clientChannel = null;
|
@Override
|
||||||
try {
|
public void channelRegistered(ChannelHandlerContext ctx) throws Exception {
|
||||||
Bootstrap bs = new Bootstrap();
|
ctx.fireChannelRegistered();
|
||||||
bs.group(group);
|
}
|
||||||
bs.channel(NioSocketChannel.class);
|
|
||||||
bs.handler(new ChannelInitializer<Channel>() {
|
@Override
|
||||||
@Override
|
public void channelUnregistered(ChannelHandlerContext ctx) throws Exception {
|
||||||
protected void initChannel(Channel ch) throws Exception {
|
ctx.fireChannelUnregistered();
|
||||||
ch.pipeline().addLast(new TrafficLoggingHandler());
|
}
|
||||||
SslContext sslContext = SslContextBuilder.forClient()
|
|
||||||
.sslProvider(SslProvider.OPENSSL)
|
@Override
|
||||||
.trustManager(InsecureTrustManagerFactory.INSTANCE)
|
public void flush(ChannelHandlerContext ctx) throws Exception {
|
||||||
.keyManager((InputStream) null, null, null)
|
ctx.flush();
|
||||||
.ciphers(Http2SecurityUtil.CIPHERS, SupportedCipherSuiteFilter.INSTANCE)
|
}
|
||||||
.applicationProtocolConfig(new ApplicationProtocolConfig(
|
|
||||||
ApplicationProtocolConfig.Protocol.ALPN,
|
@Override
|
||||||
ApplicationProtocolConfig.SelectorFailureBehavior.NO_ADVERTISE,
|
public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception {
|
||||||
ApplicationProtocolConfig.SelectedListenerFailureBehavior.ACCEPT,
|
if (msg instanceof ByteBuf && !((ByteBuf) msg).isReadable()) {
|
||||||
ApplicationProtocolNames.HTTP_2,
|
ctx.write(msg, promise);
|
||||||
ApplicationProtocolNames.HTTP_1_1))
|
} else {
|
||||||
.build();
|
super.write(ctx, msg, promise);
|
||||||
SslHandler sslHandler = sslContext.newHandler(ch.alloc());
|
|
||||||
SSLEngine engine = sslHandler.engine();
|
|
||||||
String fullQualifiedHostname = inetSocketAddress.getHostName();
|
|
||||||
SSLParameters params = engine.getSSLParameters();
|
|
||||||
params.setServerNames(Arrays.asList(new SNIServerName[]{new SNIHostName(fullQualifiedHostname)}));
|
|
||||||
engine.setSSLParameters(params);
|
|
||||||
ch.pipeline().addLast(sslHandler);
|
|
||||||
// settings handler
|
|
||||||
final Http2Connection http2Connection = new DefaultHttp2Connection(false);
|
|
||||||
Http2ConnectionHandler http2ConnectionHandler = new Http2ConnectionHandlerBuilder()
|
|
||||||
.frameLogger(new Http2FrameLogger(LogLevel.INFO, "client"))
|
|
||||||
.frameListener(new DelegatingDecompressorFrameListener(http2Connection,
|
|
||||||
new InboundHttp2ToHttpAdapterBuilder(http2Connection)
|
|
||||||
.maxContentLength(1024 * 1024)
|
|
||||||
.propagateSettings(true)
|
|
||||||
.validateHttpHeaders(false)
|
|
||||||
.build()))
|
|
||||||
.build();
|
|
||||||
ch.pipeline().addLast(http2ConnectionHandler);
|
|
||||||
ch.pipeline().addLast(new ChannelInboundHandlerAdapter() {
|
|
||||||
@Override
|
|
||||||
public void channelInactive(ChannelHandlerContext ctx) throws Exception {
|
|
||||||
dataLatch.countDown();
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
});
|
|
||||||
clientChannel = bs.connect(inetSocketAddress).syncUninterruptibly().channel();
|
|
||||||
dataLatch.await();
|
|
||||||
} finally {
|
|
||||||
if (clientChannel != null) {
|
|
||||||
clientChannel.close();
|
|
||||||
}
|
}
|
||||||
group.shutdownGracefully();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
|
@ -0,0 +1,306 @@
|
||||||
|
package org.xbib.netty.http.client.test;
|
||||||
|
|
||||||
|
import io.netty.bootstrap.Bootstrap;
|
||||||
|
import io.netty.buffer.ByteBuf;
|
||||||
|
import io.netty.channel.Channel;
|
||||||
|
import io.netty.channel.ChannelFuture;
|
||||||
|
import io.netty.channel.ChannelHandlerContext;
|
||||||
|
import io.netty.channel.ChannelInboundHandlerAdapter;
|
||||||
|
import io.netty.channel.ChannelInitializer;
|
||||||
|
import io.netty.channel.ChannelPromise;
|
||||||
|
import io.netty.channel.EventLoopGroup;
|
||||||
|
import io.netty.channel.SimpleChannelInboundHandler;
|
||||||
|
import io.netty.channel.nio.NioEventLoopGroup;
|
||||||
|
import io.netty.channel.socket.ChannelInputShutdownReadComplete;
|
||||||
|
import io.netty.channel.socket.nio.NioSocketChannel;
|
||||||
|
import io.netty.handler.codec.http.DefaultFullHttpRequest;
|
||||||
|
import io.netty.handler.codec.http.FullHttpResponse;
|
||||||
|
import io.netty.handler.codec.http.HttpMethod;
|
||||||
|
import io.netty.handler.codec.http.HttpRequest;
|
||||||
|
import io.netty.handler.codec.http.HttpVersion;
|
||||||
|
import io.netty.handler.codec.http2.DefaultHttp2Connection;
|
||||||
|
import io.netty.handler.codec.http2.DelegatingDecompressorFrameListener;
|
||||||
|
import io.netty.handler.codec.http2.Http2Connection;
|
||||||
|
import io.netty.handler.codec.http2.Http2ConnectionPrefaceWrittenEvent;
|
||||||
|
import io.netty.handler.codec.http2.Http2FrameLogger;
|
||||||
|
import io.netty.handler.codec.http2.Http2SecurityUtil;
|
||||||
|
import io.netty.handler.codec.http2.Http2Settings;
|
||||||
|
import io.netty.handler.codec.http2.HttpConversionUtil;
|
||||||
|
import io.netty.handler.codec.http2.HttpToHttp2ConnectionHandler;
|
||||||
|
import io.netty.handler.codec.http2.HttpToHttp2ConnectionHandlerBuilder;
|
||||||
|
import io.netty.handler.codec.http2.InboundHttp2ToHttpAdapterBuilder;
|
||||||
|
import io.netty.handler.logging.LogLevel;
|
||||||
|
import io.netty.handler.logging.LoggingHandler;
|
||||||
|
import io.netty.handler.ssl.ApplicationProtocolConfig;
|
||||||
|
import io.netty.handler.ssl.ApplicationProtocolNames;
|
||||||
|
import io.netty.handler.ssl.ApplicationProtocolNegotiationHandler;
|
||||||
|
import io.netty.handler.ssl.SslCloseCompletionEvent;
|
||||||
|
import io.netty.handler.ssl.SslContext;
|
||||||
|
import io.netty.handler.ssl.SslContextBuilder;
|
||||||
|
import io.netty.handler.ssl.SslHandler;
|
||||||
|
import io.netty.handler.ssl.SslProvider;
|
||||||
|
import io.netty.handler.ssl.SupportedCipherSuiteFilter;
|
||||||
|
import io.netty.handler.ssl.util.InsecureTrustManagerFactory;
|
||||||
|
import io.netty.util.internal.PlatformDependent;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import javax.net.ssl.SNIHostName;
|
||||||
|
import javax.net.ssl.SNIServerName;
|
||||||
|
import javax.net.ssl.SSLEngine;
|
||||||
|
import javax.net.ssl.SSLParameters;
|
||||||
|
import java.net.InetSocketAddress;
|
||||||
|
import java.net.URL;
|
||||||
|
import java.util.AbstractMap;
|
||||||
|
import java.util.Arrays;
|
||||||
|
import java.util.Iterator;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.concurrent.TimeUnit;
|
||||||
|
import java.util.logging.ConsoleHandler;
|
||||||
|
import java.util.logging.Handler;
|
||||||
|
import java.util.logging.Level;
|
||||||
|
import java.util.logging.LogManager;
|
||||||
|
import java.util.logging.Logger;
|
||||||
|
import java.util.logging.SimpleFormatter;
|
||||||
|
|
||||||
|
/**
|
||||||
|
*/
|
||||||
|
public class InboundHttp2ToHttpAdapterTest {
|
||||||
|
|
||||||
|
static {
|
||||||
|
System.setProperty("java.util.logging.SimpleFormatter.format",
|
||||||
|
"%1$tY-%1$tm-%1$td %1$tH:%1$tM:%1$tS.%1$tL %4$-7s [%3$s] %2$s %5$s %6$s%n");
|
||||||
|
LogManager.getLogManager().reset();
|
||||||
|
Logger rootLogger = LogManager.getLogManager().getLogger("");
|
||||||
|
Handler handler = new ConsoleHandler();
|
||||||
|
handler.setFormatter(new SimpleFormatter());
|
||||||
|
rootLogger.addHandler(handler);
|
||||||
|
rootLogger.setLevel(Level.ALL);
|
||||||
|
for (Handler h : rootLogger.getHandlers()) {
|
||||||
|
handler.setFormatter(new SimpleFormatter());
|
||||||
|
h.setLevel(Level.ALL);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static final Logger logger = Logger.getLogger("");
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testInboundHttp2ToHttpAdapter() throws Exception {
|
||||||
|
URL url = new URL("https://http2-push.io");
|
||||||
|
final InetSocketAddress inetSocketAddress = new InetSocketAddress(url.getHost(), 443);
|
||||||
|
EventLoopGroup group = new NioEventLoopGroup();
|
||||||
|
Channel clientChannel = null;
|
||||||
|
SettingsHandler settingsHandler = new SettingsHandler();
|
||||||
|
ResponseHandler responseHandler = new ResponseHandler();
|
||||||
|
try {
|
||||||
|
Bootstrap bs = new Bootstrap();
|
||||||
|
bs.group(group);
|
||||||
|
bs.channel(NioSocketChannel.class);
|
||||||
|
bs.handler(new ChannelInitializer<Channel>() {
|
||||||
|
@Override
|
||||||
|
protected void initChannel(Channel ch) throws Exception {
|
||||||
|
ch.pipeline().addLast(new TrafficLoggingHandler());
|
||||||
|
SslContext sslContext = SslContextBuilder.forClient()
|
||||||
|
.sslProvider(SslProvider.OPENSSL)
|
||||||
|
.trustManager(InsecureTrustManagerFactory.INSTANCE)
|
||||||
|
.ciphers(Http2SecurityUtil.CIPHERS, SupportedCipherSuiteFilter.INSTANCE)
|
||||||
|
.applicationProtocolConfig(new ApplicationProtocolConfig(
|
||||||
|
ApplicationProtocolConfig.Protocol.ALPN,
|
||||||
|
ApplicationProtocolConfig.SelectorFailureBehavior.NO_ADVERTISE,
|
||||||
|
ApplicationProtocolConfig.SelectedListenerFailureBehavior.ACCEPT,
|
||||||
|
ApplicationProtocolNames.HTTP_2))
|
||||||
|
.build();
|
||||||
|
SslHandler sslHandler = sslContext.newHandler(ch.alloc());
|
||||||
|
SSLEngine engine = sslHandler.engine();
|
||||||
|
String fullQualifiedHostname = inetSocketAddress.getHostName();
|
||||||
|
SSLParameters params = engine.getSSLParameters();
|
||||||
|
params.setServerNames(Arrays.asList(new SNIServerName[]{new SNIHostName(fullQualifiedHostname)}));
|
||||||
|
engine.setSSLParameters(params);
|
||||||
|
ch.pipeline().addLast(sslHandler);
|
||||||
|
ch.pipeline().addLast(new Http2NegotiationHandler(settingsHandler, responseHandler));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
clientChannel = bs.connect(inetSocketAddress).syncUninterruptibly().channel();
|
||||||
|
settingsHandler.awaitSettings(clientChannel.newPromise());
|
||||||
|
HttpRequest httpRequest = new DefaultFullHttpRequest(HttpVersion.valueOf("HTTP/2.0"),
|
||||||
|
HttpMethod.GET, url.toExternalForm());
|
||||||
|
logger.log(Level.FINE, "HTTP2: sending request");
|
||||||
|
responseHandler.put(3, clientChannel.write(httpRequest), clientChannel.newPromise());
|
||||||
|
clientChannel.flush();
|
||||||
|
logger.log(Level.FINE, "HTTP2: waiting for responses");
|
||||||
|
responseHandler.awaitResponses();
|
||||||
|
logger.log(Level.FINE, "HTTP2: done");
|
||||||
|
} finally {
|
||||||
|
if (clientChannel != null) {
|
||||||
|
clientChannel.close();
|
||||||
|
}
|
||||||
|
group.shutdownGracefully();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private HttpToHttp2ConnectionHandler createHttp2ConnectionHandler() {
|
||||||
|
final Http2Connection http2Connection = new DefaultHttp2Connection(false);
|
||||||
|
return new HttpToHttp2ConnectionHandlerBuilder()
|
||||||
|
.connection(http2Connection)
|
||||||
|
.frameLogger(new Http2FrameLogger(LogLevel.INFO, "client"))
|
||||||
|
.frameListener(new DelegatingDecompressorFrameListener(http2Connection,
|
||||||
|
new InboundHttp2ToHttpAdapterBuilder(http2Connection)
|
||||||
|
.maxContentLength(10 * 1024 * 1024)
|
||||||
|
.propagateSettings(true)
|
||||||
|
.validateHttpHeaders(false)
|
||||||
|
.build()))
|
||||||
|
.build();
|
||||||
|
}
|
||||||
|
|
||||||
|
class Http2NegotiationHandler extends ApplicationProtocolNegotiationHandler {
|
||||||
|
|
||||||
|
private final SettingsHandler settingsHandler;
|
||||||
|
|
||||||
|
private final ResponseHandler responseHandler;
|
||||||
|
|
||||||
|
Http2NegotiationHandler(SettingsHandler settingsHandler, ResponseHandler responseHandler) {
|
||||||
|
super("");
|
||||||
|
this.settingsHandler = settingsHandler;
|
||||||
|
this.responseHandler = responseHandler;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void configurePipeline(ChannelHandlerContext ctx, String protocol) throws Exception {
|
||||||
|
if (ApplicationProtocolNames.HTTP_2.equals(protocol)) {
|
||||||
|
ctx.pipeline().addLast(createHttp2ConnectionHandler());
|
||||||
|
ctx.pipeline().addLast(settingsHandler);
|
||||||
|
ctx.pipeline().addLast(new UserEventLogger());
|
||||||
|
ctx.pipeline().addLast(responseHandler);
|
||||||
|
logger.log(Level.FINE, "negotiated HTTP/2: pipeline = " + ctx.pipeline().names());
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
ctx.close();
|
||||||
|
throw new IllegalStateException("unexpected protocol: " + protocol);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
class SettingsHandler extends SimpleChannelInboundHandler<Http2Settings> {
|
||||||
|
|
||||||
|
private ChannelPromise promise;
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void channelRead0(ChannelHandlerContext ctx, Http2Settings msg) throws Exception {
|
||||||
|
promise.setSuccess();
|
||||||
|
ctx.pipeline().remove(this);
|
||||||
|
}
|
||||||
|
|
||||||
|
void awaitSettings(ChannelPromise promise) throws Exception {
|
||||||
|
this.promise = promise;
|
||||||
|
int timeout = 5000;
|
||||||
|
if (!promise.awaitUninterruptibly(timeout, TimeUnit.MILLISECONDS)) {
|
||||||
|
throw new IllegalStateException("time out while waiting for HTTP/2 settings");
|
||||||
|
}
|
||||||
|
if (!promise.isSuccess()) {
|
||||||
|
throw new RuntimeException(promise.cause());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
class ResponseHandler extends SimpleChannelInboundHandler<FullHttpResponse> {
|
||||||
|
|
||||||
|
private final Map<Integer, Map.Entry<ChannelFuture, ChannelPromise>> streamidPromiseMap;
|
||||||
|
|
||||||
|
ResponseHandler() {
|
||||||
|
this.streamidPromiseMap = PlatformDependent.newConcurrentHashMap();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void channelRead0(ChannelHandlerContext ctx, FullHttpResponse httpResponse) throws Exception {
|
||||||
|
Integer streamId = httpResponse.headers().getInt(HttpConversionUtil.ExtensionHeaderNames.STREAM_ID.text());
|
||||||
|
if (streamId == null) {
|
||||||
|
logger.log(Level.WARNING, () -> "stream ID missing");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
Map.Entry<ChannelFuture, ChannelPromise> entry = streamidPromiseMap.get(streamId);
|
||||||
|
if (entry != null) {
|
||||||
|
entry.getValue().setSuccess();
|
||||||
|
} else {
|
||||||
|
logger.log(Level.WARNING, () -> "stream id not found in promise map: " + streamId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
|
||||||
|
logger.log(Level.FINE, () -> "exception caught " + cause.getMessage());
|
||||||
|
}
|
||||||
|
|
||||||
|
void put(int streamId, ChannelFuture channelFuture, ChannelPromise promise) {
|
||||||
|
logger.log(Level.FINE, () -> "put stream ID " + streamId);
|
||||||
|
streamidPromiseMap.put(streamId, new AbstractMap.SimpleEntry<>(channelFuture, promise));
|
||||||
|
}
|
||||||
|
|
||||||
|
void awaitResponses() {
|
||||||
|
int timeout = 5000;
|
||||||
|
Iterator<Map.Entry<Integer, Map.Entry<ChannelFuture, ChannelPromise>>> iterator = streamidPromiseMap.entrySet().iterator();
|
||||||
|
while (iterator.hasNext()) {
|
||||||
|
Map.Entry<Integer, Map.Entry<ChannelFuture, ChannelPromise>> entry = iterator.next();
|
||||||
|
ChannelFuture channelFuture = entry.getValue().getKey();
|
||||||
|
if (!channelFuture.awaitUninterruptibly(timeout, TimeUnit.MILLISECONDS)) {
|
||||||
|
throw new IllegalStateException("time out while waiting to write for stream id " + entry.getKey());
|
||||||
|
}
|
||||||
|
if (!channelFuture.isSuccess()) {
|
||||||
|
throw new RuntimeException(channelFuture.cause());
|
||||||
|
}
|
||||||
|
ChannelPromise promise = entry.getValue().getValue();
|
||||||
|
if (!promise.awaitUninterruptibly(timeout, TimeUnit.MILLISECONDS)) {
|
||||||
|
throw new IllegalStateException("time out while waiting for response on stream id " + entry.getKey());
|
||||||
|
}
|
||||||
|
if (!promise.isSuccess()) {
|
||||||
|
throw new RuntimeException(promise.cause());
|
||||||
|
}
|
||||||
|
iterator.remove();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
class UserEventLogger extends ChannelInboundHandlerAdapter {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
|
||||||
|
logger.log(Level.FINE, () -> "got user event " + evt);
|
||||||
|
if (evt instanceof Http2ConnectionPrefaceWrittenEvent ||
|
||||||
|
evt instanceof SslCloseCompletionEvent ||
|
||||||
|
evt instanceof ChannelInputShutdownReadComplete) {
|
||||||
|
// Expected events
|
||||||
|
logger.log(Level.FINE, () -> "user event is expected: " + evt);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
super.userEventTriggered(ctx, evt);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
class TrafficLoggingHandler extends LoggingHandler {
|
||||||
|
|
||||||
|
TrafficLoggingHandler() {
|
||||||
|
super("client", LogLevel.TRACE);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void channelRegistered(ChannelHandlerContext ctx) throws Exception {
|
||||||
|
ctx.fireChannelRegistered();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void channelUnregistered(ChannelHandlerContext ctx) throws Exception {
|
||||||
|
ctx.fireChannelUnregistered();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void flush(ChannelHandlerContext ctx) throws Exception {
|
||||||
|
ctx.flush();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception {
|
||||||
|
if (msg instanceof ByteBuf && !((ByteBuf) msg).isReadable()) {
|
||||||
|
ctx.write(msg, promise);
|
||||||
|
} else {
|
||||||
|
super.write(ctx, msg, promise);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue