Motivation: Http2MultiplexCodec doesn't currently have an API for using the response of a h2c upgrade request. Modifications: Add a new API to the Http2MultiplexCodecBuilder which allows for setting an upgrade handler and wire it into the Http2MultiplexCodec implementation. Result: When using the Http2MultiplexCodec with h2c upgrades the upgrade handler will get added to the Http2StreamChannel which represents the half-closed (local) response of stream 1. It is then up to the user to manage the transition from the IO channel pipeline configuration necessary for making the h2c upgrade request to a form where it can read the response from the new stream channel. Fixes #7947.
71 lines
2.7 KiB
Java
71 lines
2.7 KiB
Java
/*
|
|
* Copyright 2017 The Netty Project
|
|
*
|
|
* The Netty Project licenses this file to you under the Apache License, version 2.0 (the
|
|
* "License"); you may not use this file except in compliance with the License. You may obtain a
|
|
* copy of the License at:
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software distributed under the License
|
|
* is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
|
|
* or implied. See the License for the specific language governing permissions and limitations under
|
|
* the License.
|
|
*/
|
|
package io.netty.handler.codec.http2;
|
|
|
|
import io.netty.channel.Channel;
|
|
import io.netty.channel.ChannelHandler;
|
|
import io.netty.channel.ChannelHandlerContext;
|
|
import io.netty.channel.ChannelInboundHandlerAdapter;
|
|
import io.netty.channel.ChannelInitializer;
|
|
import io.netty.channel.embedded.EmbeddedChannel;
|
|
import io.netty.handler.codec.http.DefaultFullHttpRequest;
|
|
import io.netty.handler.codec.http.FullHttpRequest;
|
|
import io.netty.handler.codec.http.HttpMethod;
|
|
import io.netty.handler.codec.http.HttpVersion;
|
|
|
|
import static org.junit.Assert.assertNotNull;
|
|
import static org.junit.Assert.assertTrue;
|
|
|
|
|
|
import org.junit.Test;
|
|
|
|
public class Http2ClientUpgradeCodecTest {
|
|
|
|
@Test
|
|
public void testUpgradeToHttp2ConnectionHandler() throws Exception {
|
|
testUpgrade(new Http2ConnectionHandlerBuilder().server(false).frameListener(new Http2FrameAdapter()).build());
|
|
}
|
|
|
|
@Test
|
|
public void testUpgradeToHttp2FrameCodec() throws Exception {
|
|
testUpgrade(Http2FrameCodecBuilder.forClient().build());
|
|
}
|
|
|
|
@Test
|
|
public void testUpgradeToHttp2MultiplexCodec() throws Exception {
|
|
testUpgrade(Http2MultiplexCodecBuilder.forClient(new HttpInboundHandler())
|
|
.withUpgradeStreamHandler(new ChannelInboundHandlerAdapter()).build());
|
|
}
|
|
|
|
private static void testUpgrade(Http2ConnectionHandler handler) throws Exception {
|
|
FullHttpRequest request = new DefaultFullHttpRequest(HttpVersion.HTTP_1_1, HttpMethod.OPTIONS, "*");
|
|
|
|
EmbeddedChannel channel = new EmbeddedChannel(new ChannelInboundHandlerAdapter());
|
|
ChannelHandlerContext ctx = channel.pipeline().firstContext();
|
|
Http2ClientUpgradeCodec codec = new Http2ClientUpgradeCodec("connectionHandler", handler);
|
|
codec.setUpgradeHeaders(ctx, request);
|
|
// Flush the channel to ensure we write out all buffered data
|
|
channel.flush();
|
|
|
|
codec.upgradeTo(ctx, null);
|
|
assertNotNull(channel.pipeline().get("connectionHandler"));
|
|
|
|
assertTrue(channel.finishAndReleaseAll());
|
|
}
|
|
|
|
@ChannelHandler.Sharable
|
|
private static final class HttpInboundHandler extends ChannelInboundHandlerAdapter { }
|
|
}
|