andreacavalli/netty5
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed NETTY-107 HttpMessageDecoder can not handle the content with no 'Content-Length' header.

Browse Source 
* Added HttpMessage.getContentLength(int defaultValue)
* Renamed state enum values
This commit is contained in:
Trustin Lee 2009-02-12 07:17:29 +00:00
parent 5faa7e9ea8
commit 2d682dc2a3
3 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/main/java/org/jboss/netty/handler/codec/http/DefaultHttpMessage.java
View File

@ -82,11 +82,15 @@ public class DefaultHttpMessage implements HttpMessage {
}
public int getContentLength() {
return getContentLength(0);
}
public int getContentLength(int defaultValue) {
List<String> contentLength = headers.get(HttpHeaders.Names.CONTENT_LENGTH);
if (contentLength != null && contentLength.size() > 0) {
return Integer.parseInt(contentLength.get(0));
}
return 0;
return defaultValue;
}
public boolean isChunked() {

2
src/main/java/org/jboss/netty/handler/codec/http/HttpMessage.java
View File

@ -57,6 +57,8 @@ public interface HttpMessage {
int getContentLength();
int getContentLength(int defaultValue);
void setContent(ChannelBuffer content);
boolean isChunked();

14
src/main/java/org/jboss/netty/handler/codec/http/HttpMessageDecoder.java
View File

@ -65,7 +65,7 @@ public abstract class HttpMessageDecoder extends ReplayingDecoder<HttpMessageDec
SKIP_CONTROL_CHARS,
READ_INITIAL,
READ_HEADER,
READ_CONTENT,
READ_VARIABLE_LENGTH_CONTENT,
READ_FIXED_LENGTH_CONTENT,
READ_CHUNK_SIZE,
READ_CHUNKED_CONTENT,
@ -103,18 +103,18 @@ public abstract class HttpMessageDecoder extends ReplayingDecoder<HttpMessageDec
if (!mergeChunks) {
return message;
}
} else if (message.getContentLength() == 0) {
} else if (message.getContentLength(-1) == 0) {
content = ChannelBuffers.EMPTY_BUFFER;
return reset();
}
//we return null here, this forces decode to be called again where we will decode the content
return null;
}
case READ_CONTENT: {
case READ_VARIABLE_LENGTH_CONTENT: {
if (content == null) {
content = ChannelBuffers.dynamicBuffer(channel.getConfig().getBufferFactory());
}
//this will cause a replay error until the channel is closed where this will read whats left in the buffer
//this will cause a replay error until the channel is closed where this will read what's left in the buffer
content.writeBytes(buffer.readBytes(buffer.readableBytes()));
return reset();
}
@ -207,7 +207,7 @@ public abstract class HttpMessageDecoder extends ReplayingDecoder<HttpMessageDec
}
private void readFixedLengthContent(ChannelBuffer buffer) {
int length = message.getContentLength();
int length = message.getContentLength(-1);
if (content == null) {
content = buffer.readBytes(length);
} else {
@ -238,10 +238,10 @@ public abstract class HttpMessageDecoder extends ReplayingDecoder<HttpMessageDec
State nextState;
if (message.isChunked()) {
nextState = State.READ_CHUNK_SIZE;
} else if (message.getContentLength() >= 0) {
} else if (message.getContentLength(-1) >= 0) {
nextState = State.READ_FIXED_LENGTH_CONTENT;
} else {
nextState = State.READ_CONTENT;
nextState = State.READ_VARIABLE_LENGTH_CONTENT;
}
checkpoint(nextState);
}