Resolved issue: NETTY-261 Write a port unification example
This commit is contained in:
Trustin Lee
parent
0b16006fd7
commit
deb1c1b038
@ -0,0 +1,58 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2010 Red Hat, Inc.
|
||||||
|
*
|
||||||
|
* Red Hat 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 org.jboss.netty.example.portunification;
|
||||||
|
|
||||||
|
import java.net.InetSocketAddress;
|
||||||
|
import java.util.concurrent.Executors;
|
||||||
|
|
||||||
|
import org.jboss.netty.bootstrap.ServerBootstrap;
|
||||||
|
import org.jboss.netty.channel.ChannelPipeline;
|
||||||
|
import org.jboss.netty.channel.ChannelPipelineFactory;
|
||||||
|
import org.jboss.netty.channel.Channels;
|
||||||
|
import org.jboss.netty.channel.socket.nio.NioServerSocketChannelFactory;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Serves two protocols (HTTP and Factorial) using only one port, enabling
|
||||||
|
* either SSL or GZIP dynamically on demand.
|
||||||
|
* <p>
|
||||||
|
* Because SSL and GZIP are enabled on demand, 5 combinations per protocol
|
||||||
|
* are possible: none, SSL only, GZIP only, SSL + GZIP, and GZIP + SSL.
|
||||||
|
*
|
||||||
|
* @author <a href="http://www.jboss.org/netty/">The Netty Project</a>
|
||||||
|
* @author <a href="http://gleamynode.net/">Trustin Lee</a>
|
||||||
|
*
|
||||||
|
* @version $Rev$, $Date$
|
||||||
|
*/
|
||||||
|
public class PortUnificationServer {
|
||||||
|
|
||||||
|
public static void main(String[] args) throws Exception {
|
||||||
|
// Configure the server.
|
||||||
|
ServerBootstrap bootstrap = new ServerBootstrap(
|
||||||
|
new NioServerSocketChannelFactory(
|
||||||
|
Executors.newCachedThreadPool(),
|
||||||
|
Executors.newCachedThreadPool()));
|
||||||
|
|
||||||
|
// Set up the event pipeline factory.
|
||||||
|
bootstrap.setPipelineFactory(new ChannelPipelineFactory() {
|
||||||
|
public ChannelPipeline getPipeline() throws Exception {
|
||||||
|
return Channels.pipeline(new PortUnificationServerHandler());
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// Bind and start to accept incoming connections.
|
||||||
|
bootstrap.bind(new InetSocketAddress(8080));
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,161 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2010 Red Hat, Inc.
|
||||||
|
*
|
||||||
|
* Red Hat 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 org.jboss.netty.example.portunification;
|
||||||
|
|
||||||
|
import javax.net.ssl.SSLEngine;
|
||||||
|
|
||||||
|
import org.jboss.netty.buffer.ChannelBuffer;
|
||||||
|
import org.jboss.netty.channel.Channel;
|
||||||
|
import org.jboss.netty.channel.ChannelHandlerContext;
|
||||||
|
import org.jboss.netty.channel.ChannelPipeline;
|
||||||
|
import org.jboss.netty.example.factorial.BigIntegerDecoder;
|
||||||
|
import org.jboss.netty.example.factorial.FactorialServerHandler;
|
||||||
|
import org.jboss.netty.example.factorial.NumberEncoder;
|
||||||
|
import org.jboss.netty.example.http.snoop.HttpRequestHandler;
|
||||||
|
import org.jboss.netty.example.securechat.SecureChatSslContextFactory;
|
||||||
|
import org.jboss.netty.handler.codec.compression.ZlibDecoder;
|
||||||
|
import org.jboss.netty.handler.codec.compression.ZlibEncoder;
|
||||||
|
import org.jboss.netty.handler.codec.compression.ZlibWrapper;
|
||||||
|
import org.jboss.netty.handler.codec.frame.FrameDecoder;
|
||||||
|
import org.jboss.netty.handler.codec.http.HttpContentCompressor;
|
||||||
|
import org.jboss.netty.handler.codec.http.HttpRequestDecoder;
|
||||||
|
import org.jboss.netty.handler.codec.http.HttpResponseEncoder;
|
||||||
|
import org.jboss.netty.handler.ssl.SslHandler;
|
||||||
|
|
||||||
|
/**
|
||||||
|
*
|
||||||
|
*
|
||||||
|
* @author <a href="http://www.jboss.org/netty/">The Netty Project</a>
|
||||||
|
* @author <a href="http://gleamynode.net/">Trustin Lee</a>
|
||||||
|
*
|
||||||
|
* @version $Rev$, $Date$
|
||||||
|
*/
|
||||||
|
public class PortUnificationServerHandler extends FrameDecoder {
|
||||||
|
|
||||||
|
private final boolean detectSsl;
|
||||||
|
private final boolean detectGzip;
|
||||||
|
|
||||||
|
public PortUnificationServerHandler() {
|
||||||
|
this(true, true);
|
||||||
|
}
|
||||||
|
|
||||||
|
private PortUnificationServerHandler(boolean detectSsl, boolean detectGzip) {
|
||||||
|
this.detectSsl = detectSsl;
|
||||||
|
this.detectGzip = detectGzip;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected Object decode(ChannelHandlerContext ctx, Channel channel, ChannelBuffer buffer) throws Exception {
|
||||||
|
|
||||||
|
// Will use the first two bytes to detect a protocol.
|
||||||
|
if (buffer.readableBytes() < 2) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
final int magic1 = buffer.getUnsignedByte(buffer.readerIndex());
|
||||||
|
final int magic2 = buffer.getUnsignedByte(buffer.readerIndex() + 1);
|
||||||
|
|
||||||
|
if (isSsl(magic1)) {
|
||||||
|
enableSsl(ctx);
|
||||||
|
} else if (isGzip(magic1, magic2)) {
|
||||||
|
enableGzip(ctx);
|
||||||
|
} else if (isHttp(magic1, magic2)) {
|
||||||
|
switchToHttp(ctx);
|
||||||
|
} else if (isFactorial(magic1)) {
|
||||||
|
switchToFactorial(ctx);
|
||||||
|
} else {
|
||||||
|
// Unknown protocol
|
||||||
|
ctx.getChannel().close();
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Forward the current read buffer as is to the new handlers.
|
||||||
|
return buffer.readBytes(buffer.readableBytes());
|
||||||
|
}
|
||||||
|
|
||||||
|
private boolean isSsl(int magic1) {
|
||||||
|
if (detectSsl) {
|
||||||
|
switch (magic1) {
|
||||||
|
case 20: case 21: case 22: case 23: case 255:
|
||||||
|
return true;
|
||||||
|
default:
|
||||||
|
return magic1 >= 128;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
private boolean isGzip(int magic1, int magic2) {
|
||||||
|
if (detectGzip) {
|
||||||
|
return magic1 == 31 && magic2 == 139;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
private 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 boolean isFactorial(int magic1) {
|
||||||
|
return magic1 == 'F';
|
||||||
|
}
|
||||||
|
|
||||||
|
private void enableSsl(ChannelHandlerContext ctx) {
|
||||||
|
ChannelPipeline p = ctx.getPipeline();
|
||||||
|
|
||||||
|
SSLEngine engine =
|
||||||
|
SecureChatSslContextFactory.getServerContext().createSSLEngine();
|
||||||
|
engine.setUseClientMode(false);
|
||||||
|
|
||||||
|
p.addLast("ssl", new SslHandler(engine));
|
||||||
|
p.addLast("unificationA", new PortUnificationServerHandler(false, detectGzip));
|
||||||
|
p.remove(this);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void enableGzip(ChannelHandlerContext ctx) {
|
||||||
|
ChannelPipeline p = ctx.getPipeline();
|
||||||
|
p.addLast("gzipdeflater", new ZlibEncoder(ZlibWrapper.GZIP));
|
||||||
|
p.addLast("gzipinflater", new ZlibDecoder(ZlibWrapper.GZIP));
|
||||||
|
p.addLast("unificationB", new PortUnificationServerHandler(detectSsl, false));
|
||||||
|
p.remove(this);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void switchToHttp(ChannelHandlerContext ctx) {
|
||||||
|
ChannelPipeline p = ctx.getPipeline();
|
||||||
|
p.addLast("decoder", new HttpRequestDecoder());
|
||||||
|
p.addLast("encoder", new HttpResponseEncoder());
|
||||||
|
p.addLast("deflater", new HttpContentCompressor());
|
||||||
|
p.addLast("handler", new HttpRequestHandler());
|
||||||
|
p.remove(this);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void switchToFactorial(ChannelHandlerContext ctx) {
|
||||||
|
ChannelPipeline p = ctx.getPipeline();
|
||||||
|
p.addLast("decoder", new BigIntegerDecoder());
|
||||||
|
p.addLast("encoder", new NumberEncoder());
|
||||||
|
p.addLast("handler", new FactorialServerHandler());
|
||||||
|
p.remove(this);
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user