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

Better white space handling in HTTP

Browse Source
This commit is contained in:
Trustin Lee 2009-02-12 05:02:22 +00:00
parent 0198da7b6c
commit 9cdc4a959e
3 changed files with 29 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -58,24 +58,33 @@ public abstract class HttpMessageDecoder extends ReplayingDecoder<HttpMessageDec
* @apiviz.exclude
*/
protected enum State {
SKIP_CONTROL_CHARS,
READ_INITIAL,
READ_HEADER,
READ_CONTENT,
READ_FIXED_LENGTH_CONTENT,
READ_CHUNK_SIZE,
READ_CHUNKED_CONTENT,
READ_CRLF,;
READ_CRLF;
}
private State nextState;
protected HttpMessageDecoder() {
super(State.READ_INITIAL);
super(State.SKIP_CONTROL_CHARS);
}
@Override
protected Object decode(ChannelHandlerContext ctx, Channel channel, ChannelBuffer buffer, State state) throws Exception {
switch (state) {
case SKIP_CONTROL_CHARS: {
try {
skipControlCharacters(buffer);
checkpoint(State.READ_INITIAL);
} finally {
checkpoint();
}
}
case READ_INITIAL: {
readInitial(buffer);
}
@ -144,10 +153,21 @@ public abstract class HttpMessageDecoder extends ReplayingDecoder<HttpMessageDec
message.setContent(content);
content = null;
nextState = null;
checkpoint(State.READ_INITIAL);
checkpoint(State.SKIP_CONTROL_CHARS);
return message;
}
private void skipControlCharacters(ChannelBuffer buffer) {
for (;;) {
char c = (char) buffer.readUnsignedByte();
if (!Character.isISOControl(c) &&
!Character.isWhitespace(c)) {
buffer.readerIndex(buffer.readerIndex() - 1);
break;
}
}
}
private void readChunkedContent(Channel channel, ChannelBuffer buffer) {
if (content == null) {
content = ChannelBuffers.dynamicBuffer(

5
src/main/java/org/jboss/netty/handler/codec/http/HttpMethod.java
View File

@ -125,8 +125,9 @@ public class HttpMethod implements Comparable<HttpMethod> {
}
for (int i = 0; i < name.length(); i ++) {
if (Character.isISOControl(name.charAt(i))) {
throw new IllegalArgumentException("control character in name");
if (Character.isISOControl(name.charAt(i)) ||
Character.isWhitespace(name.charAt(i))) {
throw new IllegalArgumentException("invalid character in name");
}
}

5
src/main/java/org/jboss/netty/handler/codec/http/HttpVersion.java
View File

@ -95,8 +95,9 @@ public class HttpVersion implements Comparable<HttpVersion> {
}
for (int i = 0; i < protocolName.length(); i ++) {
if (Character.isISOControl(protocolName.charAt(i))) {
throw new IllegalArgumentException("control character in protocolName");
if (Character.isISOControl(protocolName.charAt(i)) ||
Character.isWhitespace(protocolName.charAt(i))) {
throw new IllegalArgumentException("invalid character in protocolName");
}
}