222 lines
7.2 KiB
Java
222 lines
7.2 KiB
Java
|
/*
|
||
|
* JBoss, Home of Professional Open Source
|
||
|
* Copyright 2005-2008, Red Hat Middleware LLC, and individual contributors
|
||
|
* by the @authors tag. See the copyright.txt in the distribution for a
|
||
|
* full listing of individual contributors.
|
||
|
*
|
||
|
* This is free software; you can redistribute it and/or modify it
|
||
|
* under the terms of the GNU Lesser General Public License as
|
||
|
* published by the Free Software Foundation; either version 2.1 of
|
||
|
* the License, or (at your option) any later version.
|
||
|
*
|
||
|
* This software is distributed in the hope that it will be useful,
|
||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||
|
* Lesser General Public License for more details.
|
||
|
*
|
||
|
* You should have received a copy of the GNU Lesser General Public
|
||
|
* License along with this software; if not, write to the Free
|
||
|
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
|
||
|
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
|
||
|
*/
|
||
|
package org.jboss.netty.handler.codec.http;
|
||
|
|
||
|
import java.util.List;
|
||
|
|
||
|
import org.jboss.netty.buffer.ChannelBuffer;
|
||
|
import org.jboss.netty.buffer.ChannelBuffers;
|
||
|
import org.jboss.netty.channel.Channel;
|
||
|
import org.jboss.netty.channel.ChannelHandlerContext;
|
||
|
import org.jboss.netty.handler.codec.replay.ReplayingDecoder;
|
||
|
|
||
|
/**
|
||
|
* Decodes an Http type message.
|
||
|
*
|
||
|
* @author The Netty Project (netty-dev@lists.jboss.org)
|
||
|
* @author Andy Taylor (andy.taylor@jboss.org)
|
||
|
* @author Trustin Lee (tlee@redhat.com)
|
||
|
*/
|
||
|
public abstract class HttpMessageDecoder extends ReplayingDecoder<HttpMessageDecoder.ResponseState> {
|
||
|
protected HttpMessage message;
|
||
|
|
||
|
private ChannelBuffer content;
|
||
|
|
||
|
private int chunkSize;
|
||
|
|
||
|
public enum ResponseState {
|
||
|
READ_INITIAL,
|
||
|
READ_HEADER,
|
||
|
READ_CONTENT,
|
||
|
READ_FIXED_LENGTH_CONTENT,
|
||
|
READ_CHUNK_SIZE,
|
||
|
READ_CHUNKED_CONTENT,
|
||
|
READ_CRLF,;
|
||
|
}
|
||
|
|
||
|
private ResponseState nextState;
|
||
|
|
||
|
protected HttpMessageDecoder() {
|
||
|
super(ResponseState.READ_INITIAL);
|
||
|
}
|
||
|
|
||
|
@Override
|
||
|
protected Object decode(ChannelHandlerContext ctx, Channel channel, ChannelBuffer buffer, ResponseState state) throws Exception {
|
||
|
switch (state) {
|
||
|
case READ_INITIAL: {
|
||
|
readInitial(buffer);
|
||
|
}
|
||
|
case READ_HEADER: {
|
||
|
readHeaders(buffer);
|
||
|
//we return null here, this forces decode to be called again where we will decode the content
|
||
|
return null;
|
||
|
}
|
||
|
case READ_CONTENT: {
|
||
|
if (content == null) {
|
||
|
content = ChannelBuffers.dynamicBuffer();
|
||
|
}
|
||
|
//this will cause a replay error until the channel is closed where this will read whats left in the buffer
|
||
|
content.writeBytes(buffer.readBytes(buffer.readableBytes()));
|
||
|
return reset();
|
||
|
}
|
||
|
case READ_FIXED_LENGTH_CONTENT: {
|
||
|
//we have a content-length so we just read the correct number of bytes
|
||
|
readFixedLengthContent(buffer);
|
||
|
return reset();
|
||
|
}
|
||
|
/**
|
||
|
* everything else after this point takes care of reading chunked content. basically, read chunk size,
|
||
|
* read chunk, read and ignore the CRLF and repeat until 0
|
||
|
*/
|
||
|
case READ_CHUNK_SIZE: {
|
||
|
String line = readIntoCurrentLine(buffer);
|
||
|
|
||
|
chunkSize = getChunkSize(line);
|
||
|
if (chunkSize == 0) {
|
||
|
byte next = buffer.readByte();
|
||
|
if (next == HttpCodecUtil.CR) {
|
||
|
buffer.readByte();
|
||
|
}
|
||
|
return reset();
|
||
|
}
|
||
|
else {
|
||
|
checkpoint(ResponseState.READ_CHUNKED_CONTENT);
|
||
|
}
|
||
|
}
|
||
|
case READ_CHUNKED_CONTENT: {
|
||
|
readChunkedContent(buffer);
|
||
|
}
|
||
|
case READ_CRLF: {
|
||
|
byte next = buffer.readByte();
|
||
|
if (next == HttpCodecUtil.CR) {
|
||
|
buffer.readByte();
|
||
|
}
|
||
|
checkpoint(nextState);
|
||
|
return null;
|
||
|
}
|
||
|
default: {
|
||
|
throw new Error("Shouldn't reach here.");
|
||
|
}
|
||
|
|
||
|
}
|
||
|
}
|
||
|
|
||
|
private Object reset() {
|
||
|
message.setContent(content);
|
||
|
content = null;
|
||
|
nextState = null;
|
||
|
checkpoint(ResponseState.READ_INITIAL);
|
||
|
return message;
|
||
|
}
|
||
|
|
||
|
private void readChunkedContent(ChannelBuffer buffer) {
|
||
|
if (content == null) {
|
||
|
content = ChannelBuffers.dynamicBuffer(chunkSize);
|
||
|
}
|
||
|
content.writeBytes(buffer, chunkSize);
|
||
|
nextState = ResponseState.READ_CHUNK_SIZE;
|
||
|
checkpoint(ResponseState.READ_CRLF);
|
||
|
}
|
||
|
|
||
|
private void readFixedLengthContent(ChannelBuffer buffer) {
|
||
|
int length = message.getContentLength();
|
||
|
if (content == null) {
|
||
|
content = ChannelBuffers.buffer(length);
|
||
|
}
|
||
|
content.writeBytes(buffer.readBytes(length));
|
||
|
}
|
||
|
|
||
|
private void readHeaders(ChannelBuffer buffer) {
|
||
|
message.clearHeaders();
|
||
|
String line = readIntoCurrentLine(buffer);
|
||
|
String lastHeader = null;
|
||
|
while (!line.equals("")) {
|
||
|
if (line.startsWith(" ") || line.startsWith("\t")) {
|
||
|
List<String> current = message.getHeaders(lastHeader);
|
||
|
int lastPos = current.size() - 1;
|
||
|
String newString = current.get(lastPos) + line.trim();
|
||
|
current.remove(lastPos);
|
||
|
current.add(newString);
|
||
|
}
|
||
|
else {
|
||
|
String[] header = splitHeader(line);
|
||
|
message.addHeader(header[0], header[1]);
|
||
|
lastHeader = header[0];
|
||
|
}
|
||
|
line = readIntoCurrentLine(buffer);
|
||
|
}
|
||
|
ResponseState nextState;
|
||
|
if (message.getContentLength() > 0) {
|
||
|
nextState = ResponseState.READ_FIXED_LENGTH_CONTENT;
|
||
|
}
|
||
|
else if (message.isChunked()) {
|
||
|
nextState = ResponseState.READ_CHUNK_SIZE;
|
||
|
}
|
||
|
else {
|
||
|
nextState = ResponseState.READ_CONTENT;
|
||
|
}
|
||
|
checkpoint(nextState);
|
||
|
}
|
||
|
|
||
|
protected abstract void readInitial(ChannelBuffer buffer) throws Exception;
|
||
|
|
||
|
private int getChunkSize(String hex) {
|
||
|
return Integer.valueOf(hex, 16);
|
||
|
}
|
||
|
|
||
|
protected String readIntoCurrentLine(ChannelBuffer channel) {
|
||
|
StringBuffer sb = new StringBuffer();
|
||
|
while (true) {
|
||
|
byte nextByte = channel.readByte();
|
||
|
if (nextByte == HttpCodecUtil.CR) {
|
||
|
channel.readByte();
|
||
|
return sb.toString();
|
||
|
}
|
||
|
else if (nextByte == HttpCodecUtil.LF) {
|
||
|
return sb.toString();
|
||
|
}
|
||
|
else {
|
||
|
sb.append((char) nextByte);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
protected String[] splitInitial(String sb) {
|
||
|
String[] split = sb.split(" ");
|
||
|
if (split.length != 3) {
|
||
|
throw new IllegalArgumentException(sb + "does not contain all 3 parts");
|
||
|
}
|
||
|
return split;
|
||
|
}
|
||
|
|
||
|
private String[] splitHeader(String sb) {
|
||
|
String[] split = sb.split(":", 2);
|
||
|
if (split.length != 2) {
|
||
|
throw new IllegalArgumentException(sb + "does not contain all 2 parts");
|
||
|
}
|
||
|
for (int i = 0; i < split.length; i++) {
|
||
|
split[i] = split[i].trim();
|
||
|
}
|
||
|
return split;
|
||
|
}
|
||
|
}
|