View Javadoc
1   /*
2    * Copyright 2017 The Netty Project
3    *
4    * The Netty Project licenses this file to you under the Apache License,
5    * version 2.0 (the "License"); you may not use this file except in compliance
6    * with the License. You may obtain a copy of the License at:
7    *
8    *   https://www.apache.org/licenses/LICENSE-2.0
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
12   * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
13   * License for the specific language governing permissions and limitations
14   * under the License.
15   */
16  
17  package io.netty.testsuite.http2;
18  
19  import static io.netty.util.internal.ObjectUtil.checkPositiveOrZero;
20  
21  import io.netty.channel.ChannelHandlerContext;
22  import io.netty.channel.ChannelInboundHandlerAdapter;
23  import io.netty.channel.ChannelInitializer;
24  import io.netty.channel.ChannelPipeline;
25  import io.netty.channel.SimpleChannelInboundHandler;
26  import io.netty.channel.socket.SocketChannel;
27  import io.netty.handler.codec.http.HttpMessage;
28  import io.netty.handler.codec.http.HttpObjectAggregator;
29  import io.netty.handler.codec.http.HttpServerCodec;
30  import io.netty.handler.codec.http.HttpServerUpgradeHandler;
31  import io.netty.handler.codec.http.HttpServerUpgradeHandler.UpgradeCodec;
32  import io.netty.handler.codec.http.HttpServerUpgradeHandler.UpgradeCodecFactory;
33  import io.netty.handler.codec.http2.CleartextHttp2ServerUpgradeHandler;
34  import io.netty.handler.codec.http2.Http2CodecUtil;
35  import io.netty.handler.codec.http2.Http2ServerUpgradeCodec;
36  import io.netty.util.AsciiString;
37  import io.netty.util.ReferenceCountUtil;
38  
39  /**
40   * Sets up the Netty pipeline for the example server. Depending on the endpoint config, sets up the
41   * pipeline for NPN or cleartext HTTP upgrade to HTTP/2.
42   */
43  public class Http2ServerInitializer extends ChannelInitializer<SocketChannel> {
44  
45      private static final UpgradeCodecFactory upgradeCodecFactory = new UpgradeCodecFactory() {
46          @Override
47          public UpgradeCodec newUpgradeCodec(CharSequence protocol) {
48              if (AsciiString.contentEquals(Http2CodecUtil.HTTP_UPGRADE_PROTOCOL_NAME, protocol)) {
49                  return new Http2ServerUpgradeCodec(new HelloWorldHttp2HandlerBuilder().build());
50              } else {
51                  return null;
52              }
53          }
54      };
55  
56      private final int maxHttpContentLength;
57  
58      Http2ServerInitializer() {
59          this(16 * 1024);
60      }
61  
62      Http2ServerInitializer(int maxHttpContentLength) {
63          checkPositiveOrZero(maxHttpContentLength, "maxHttpContentLength");
64          this.maxHttpContentLength = maxHttpContentLength;
65      }
66  
67      @Override
68      public void initChannel(SocketChannel ch) {
69          configureClearText(ch);
70      }
71  
72      /**
73       * Configure the pipeline for a cleartext upgrade from HTTP to HTTP/2.0
74       */
75      private void configureClearText(SocketChannel ch) {
76          final ChannelPipeline p = ch.pipeline();
77          final HttpServerCodec sourceCodec = new HttpServerCodec();
78          final HttpServerUpgradeHandler upgradeHandler = new HttpServerUpgradeHandler(sourceCodec, upgradeCodecFactory);
79          final CleartextHttp2ServerUpgradeHandler cleartextHttp2ServerUpgradeHandler =
80                  new CleartextHttp2ServerUpgradeHandler(sourceCodec, upgradeHandler,
81                                                         new HelloWorldHttp2HandlerBuilder().build());
82  
83          p.addLast(cleartextHttp2ServerUpgradeHandler);
84          p.addLast(new SimpleChannelInboundHandler<HttpMessage>() {
85              @Override
86              protected void channelRead0(ChannelHandlerContext ctx, HttpMessage msg) throws Exception {
87                  // If this handler is hit then no upgrade has been attempted and the client is just talking HTTP.
88                  System.err.println("Directly talking: " + msg.protocolVersion() + " (no upgrade was attempted)");
89                  ChannelPipeline pipeline = ctx.pipeline();
90                  ChannelHandlerContext thisCtx = pipeline.context(this);
91                  pipeline.addAfter(thisCtx.name(), null, new HelloWorldHttp1Handler("Direct. No Upgrade Attempted."));
92                  pipeline.replace(this, null, new HttpObjectAggregator(maxHttpContentLength));
93                  ctx.fireChannelRead(ReferenceCountUtil.retain(msg));
94              }
95          });
96  
97          p.addLast(new UserEventLogger());
98      }
99  
100     /**
101      * Class that logs any User Events triggered on this channel.
102      */
103     private static class UserEventLogger extends ChannelInboundHandlerAdapter {
104         @Override
105         public void userEventTriggered(ChannelHandlerContext ctx, Object evt) {
106             System.out.println("User Event Triggered: " + evt);
107             ctx.fireUserEventTriggered(evt);
108         }
109     }
110 }