Java tutorial
/* * Copyright 2012 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 com.hzmsc.scada.Jmtis.server; import com.hzmsc.scada.JmtisLE.codec.JmtisBodyEncoder; import com.hzmsc.scada.JmtisLE.codec.JmtisDecoder; import com.hzmsc.scada.JmtisLE.codec.JmtisHeaderEncoder; import com.hzmsc.scada.JmtisLE.server.JmtisServerHandler; import io.netty.buffer.ByteBuf; import io.netty.channel.ChannelHandlerContext; import io.netty.channel.ChannelPipeline; import io.netty.handler.codec.ByteToMessageDecoder; import io.netty.handler.codec.LengthFieldBasedFrameDecoder; import io.netty.handler.codec.compression.ZlibCodecFactory; import io.netty.handler.codec.compression.ZlibWrapper; import io.netty.handler.codec.http.HttpContentCompressor; import io.netty.handler.codec.http.HttpRequestDecoder; import io.netty.handler.codec.http.HttpResponseEncoder; import io.netty.handler.ssl.SslContext; import io.netty.handler.ssl.SslHandler; import java.util.List; /** * Manipulates the current pipeline dynamically to switch protocols or enable * SSL or GZIP. */ public class PortUnificationServerHandler extends ByteToMessageDecoder { private final SslContext sslCtx; private final boolean detectSsl; private final boolean detectGzip; public PortUnificationServerHandler(SslContext sslCtx) { this(sslCtx, true, true); } private PortUnificationServerHandler(SslContext sslCtx, boolean detectSsl, boolean detectGzip) { this.sslCtx = sslCtx; this.detectSsl = detectSsl; this.detectGzip = detectGzip; } @Override protected void decode(ChannelHandlerContext ctx, ByteBuf in, List<Object> out) throws Exception { System.out.println(in.getUnsignedByte(in.readerIndex()) + ".........................."); // Will use the first five bytes to detect a protocol. if (in.readableBytes() < 5) { return; } if (isSsl(in)) { enableSsl(ctx); } else { final int magic1 = in.getUnsignedByte(in.readerIndex()); final int magic2 = in.getUnsignedByte(in.readerIndex() + 1); if (isGzip(magic1, magic2)) { enableGzip(ctx); } else if (isHttp(magic1, magic2)) { switchToHttp(ctx); } else if (isJMAC(magic1, magic2)) { switchToJMAC(ctx); } else { // Unknown protocol; discard everything and close the connection. in.clear(); ctx.close(); } } } private boolean isSsl(ByteBuf buf) { if (detectSsl) { return SslHandler.isEncrypted(buf); } return false; } private boolean isGzip(int magic1, int magic2) { if (detectGzip) { return magic1 == 31 && magic2 == 139; } return false; } private static boolean isHttp(int magic1, int magic2) { return magic1 == 'G' && magic2 == 'E' || // GET magic1 == 'P' && magic2 == 'O' || // POST magic1 == 'P' && magic2 == 'U' || // PUT magic1 == 'H' && magic2 == 'E' || // HEAD magic1 == 'O' && magic2 == 'P' || // OPTIONS magic1 == 'P' && magic2 == 'A' || // PATCH magic1 == 'D' && magic2 == 'E' || // DELETE magic1 == 'T' && magic2 == 'R' || // TRACE magic1 == 'C' && magic2 == 'O'; // CONNECT } private static boolean isJMAC(int magic1, int magic2) { return magic1 == 'C' && magic2 == 'A'; } private void enableSsl(ChannelHandlerContext ctx) { ChannelPipeline p = ctx.pipeline(); p.addLast("ssl", sslCtx.newHandler(ctx.alloc())); p.addLast("unificationA", new PortUnificationServerHandler(sslCtx, false, detectGzip)); p.remove(this); } private void enableGzip(ChannelHandlerContext ctx) { ChannelPipeline p = ctx.pipeline(); p.addLast("gzipdeflater", ZlibCodecFactory.newZlibEncoder(ZlibWrapper.GZIP)); p.addLast("gzipinflater", ZlibCodecFactory.newZlibDecoder(ZlibWrapper.GZIP)); p.addLast("unificationB", new PortUnificationServerHandler(sslCtx, detectSsl, false)); p.remove(this); } private void switchToHttp(ChannelHandlerContext ctx) { ChannelPipeline p = ctx.pipeline(); p.addLast("decoder", new HttpRequestDecoder()); p.addLast("encoder", new HttpResponseEncoder()); p.addLast("deflater", new HttpContentCompressor()); p.addLast("handler", new HttpHelloWorldServerHandler()); p.remove(this); } private void switchToJMAC(ChannelHandlerContext ctx) { ChannelPipeline pipeline = ctx.pipeline(); pipeline.addLast(new LengthFieldBasedFrameDecoder(1024, 4, 4)); pipeline.addLast(new JmtisDecoder()); pipeline.addLast(new JmtisHeaderEncoder()); pipeline.addLast(new JmtisBodyEncoder()); pipeline.addLast(new JmtisServerHandler()); pipeline.remove(this); } }