View Javadoc
1   /*
2    * Copyright 2016 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    *   http://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.example.http2.helloworld.multiplex.server;
18  
19  import io.netty.channel.ChannelHandlerContext;
20  import io.netty.channel.ChannelInboundHandlerAdapter;
21  import io.netty.channel.ChannelInitializer;
22  import io.netty.channel.ChannelPipeline;
23  import io.netty.channel.SimpleChannelInboundHandler;
24  import io.netty.channel.socket.SocketChannel;
25  import io.netty.example.http2.helloworld.server.HelloWorldHttp1Handler;
26  import io.netty.handler.codec.http.HttpMessage;
27  import io.netty.handler.codec.http.HttpObjectAggregator;
28  import io.netty.handler.codec.http.HttpServerCodec;
29  import io.netty.handler.codec.http.HttpServerUpgradeHandler;
30  import io.netty.handler.codec.http.HttpServerUpgradeHandler.UpgradeCodec;
31  import io.netty.handler.codec.http.HttpServerUpgradeHandler.UpgradeCodecFactory;
32  import io.netty.handler.codec.http2.Http2FrameCodecBuilder;
33  import io.netty.handler.codec.http2.Http2CodecUtil;
34  import io.netty.handler.codec.http2.Http2MultiplexHandler;
35  import io.netty.handler.codec.http2.Http2ServerUpgradeCodec;
36  import io.netty.handler.ssl.SslContext;
37  import io.netty.util.AsciiString;
38  import io.netty.util.ReferenceCountUtil;
39  
40  /**
41   * Sets up the Netty pipeline for the example server. Depending on the endpoint config, sets up the
42   * pipeline for NPN or cleartext HTTP upgrade to HTTP/2.
43   */
44  public class Http2ServerInitializer extends ChannelInitializer<SocketChannel> {
45  
46      private static final UpgradeCodecFactory upgradeCodecFactory = new UpgradeCodecFactory() {
47          @Override
48          public UpgradeCodec newUpgradeCodec(CharSequence protocol) {
49              if (AsciiString.contentEquals(Http2CodecUtil.HTTP_UPGRADE_PROTOCOL_NAME, protocol)) {
50                  return new Http2ServerUpgradeCodec(
51                          Http2FrameCodecBuilder.forServer().build(),
52                          new Http2MultiplexHandler(new HelloWorldHttp2Handler()));
53              } else {
54                  return null;
55              }
56          }
57      };
58  
59      private final SslContext sslCtx;
60      private final int maxHttpContentLength;
61  
62      public Http2ServerInitializer(SslContext sslCtx) {
63          this(sslCtx, 16 * 1024);
64      }
65  
66      public Http2ServerInitializer(SslContext sslCtx, int maxHttpContentLength) {
67          if (maxHttpContentLength < 0) {
68              throw new IllegalArgumentException("maxHttpContentLength (expected >= 0): " + maxHttpContentLength);
69          }
70          this.sslCtx = sslCtx;
71          this.maxHttpContentLength = maxHttpContentLength;
72      }
73  
74      @Override
75      public void initChannel(SocketChannel ch) {
76          if (sslCtx != null) {
77              configureSsl(ch);
78          } else {
79              configureClearText(ch);
80          }
81      }
82  
83      /**
84       * Configure the pipeline for TLS NPN negotiation to HTTP/2.
85       */
86      private void configureSsl(SocketChannel ch) {
87          ch.pipeline().addLast(sslCtx.newHandler(ch.alloc()), new Http2OrHttpHandler());
88      }
89  
90      /**
91       * Configure the pipeline for a cleartext upgrade from HTTP to HTTP/2.0
92       */
93      private void configureClearText(SocketChannel ch) {
94          final ChannelPipeline p = ch.pipeline();
95          final HttpServerCodec sourceCodec = new HttpServerCodec();
96  
97          p.addLast(sourceCodec);
98          p.addLast(new HttpServerUpgradeHandler(sourceCodec, upgradeCodecFactory));
99          p.addLast(new SimpleChannelInboundHandler<HttpMessage>() {
100             @Override
101             protected void channelRead0(ChannelHandlerContext ctx, HttpMessage msg) throws Exception {
102                 // If this handler is hit then no upgrade has been attempted and the client is just talking HTTP.
103                 System.err.println("Directly talking: " + msg.protocolVersion() + " (no upgrade was attempted)");
104                 ChannelPipeline pipeline = ctx.pipeline();
105                 pipeline.addAfter(ctx.name(), null, new HelloWorldHttp1Handler("Direct. No Upgrade Attempted."));
106                 pipeline.replace(this, null, new HttpObjectAggregator(maxHttpContentLength));
107                 ctx.fireChannelRead(ReferenceCountUtil.retain(msg));
108             }
109         });
110 
111         p.addLast(new UserEventLogger());
112     }
113 
114     /**
115      * Class that logs any User Events triggered on this channel.
116      */
117     private static class UserEventLogger extends ChannelInboundHandlerAdapter {
118         @Override
119         public void userEventTriggered(ChannelHandlerContext ctx, Object evt) {
120             System.out.println("User Event Triggered: " + evt);
121             ctx.fireUserEventTriggered(evt);
122         }
123     }
124 }