ContentLengthAlgorithmpublic class ContentLengthAlgorithm extends StreamAlgorithmBase Predict if the NIO channel has been fully read or not. This lagorithm will
first search for the content-length header, and use that value to determine if
the bytes has been fully read or not. If the content-length isn't included,
it will search for the end of the HTTP stream, which is a '\r\n'
Note: the parsing algorithm is an adaptation of:
org.apache.coyote.http11.InternalInputBuffer
written by Remy Maucherat |
Fields Summary |
---|
protected static final byte[] | POST_METHOD | protected static final byte[] | PUT_METHOD | protected static final byte[] | CL_HEADER | public byte[] | ascbufPointer to the US-ASCII header buffer. | protected int | lastValidLast valid byte. | protected int | posPosition in the buffer. | protected boolean | isFoundIs the content-length fully read. | protected boolean | requestLineParsedIs the request line parsed? | public int | startReqThe request bytes position. | public int | lengthReqThe request bytes length |
Constructors Summary |
---|
public ContentLengthAlgorithm()
// ---------------------------------------------- Constructor ----------/
if (embeddedInGlassFish){
handler = new ContentLengthHandler(this);
} else {
handler = new DummyHandler();
}
|
Methods Summary |
---|
protected int | findBytes(byte[] buff, int start, int end, byte[] b)Compare two bytes array and return > 0 if true.
byte first = b[0];
// Look for first char
int srcEnd = b.length;
for (int i = start; i <= (end - srcEnd); i++) {
if (Ascii.toLower(buff[i]) != first) continue;
// found first char, now look for a match
int myPos = i+1;
for (int srcPos = 1; srcPos < srcEnd; ) {
if (Ascii.toLower(buff[myPos++]) != b[srcPos++])
break;
if (srcPos == srcEnd) return i - start; // found it
}
}
return -1;
| public com.sun.enterprise.web.connector.grizzly.Handler | getHandler()Return the Handler used by this algorithm.;
return handler;
| public java.lang.Class | getReadTask(com.sun.enterprise.web.connector.grizzly.SelectorThread selectorThread)Return the class responsible for handling OP_READ.
return com.sun.enterprise.web.connector.grizzly.XAReadTask.class;
| public boolean | parse(java.nio.ByteBuffer byteBuffer)Parse the ByteBuffer and try to determine if the bytes
stream has been fully read from the SocketChannel .
Drain the SocketChannel and determine if the request bytes
has been fully read. For POST method, parse the bytes and seek for the
content-type header to determine the length of the request bytes.
isFound = false;
curLimit = byteBuffer.limit();
curPosition = byteBuffer.position();
// Rule a - if we know the content length, verify all bytes are read
// Return true only if all bytes has been read.
if ( contentLength != -1 ){
isFound =
((contentLength + headerLength) <= byteBuffer.position());
if (isFound)
byteBuffer.flip();
return isFound;
}
try{
// Rule b - If nothing, return to the Selector.
if (byteBuffer.position() == 0)
return false;
byteBuffer.flip();
lastValid = byteBuffer.limit();
// Rule c - Parse the request line
if ( !requestLineParsed ) {
requestLineParsed = parseRequestLine(byteBuffer);
if ( !requestLineParsed ) {
return false;
}
}
// Rule d -- Parse the headers, looking for content-length
while (parseHeader(byteBuffer));
// Rule e - for POST/PUT, make sure all the body is loaded from
// the socket channel.
if ( headerLength != -1 && isFound ){
isFound = ((contentLength + headerLength)
<= byteBuffer.limit());
} else {
// The content-length was found, but the headers
// aren't yet bufferred entirely.
isFound = false;
}
return isFound;
} catch (BufferUnderflowException bue) {
SelectorThread.logger().log(Level.SEVERE,
"readTask.bufferunderflow", bue);
return false;
} finally {
byteBuffer.limit(curLimit);
byteBuffer.position(curPosition);
if (isFound){
byteBuffer.flip();
}
}
| protected boolean | parseHeader(java.nio.ByteBuffer byteBuffer)Parse the headers, looking for content-length header and value.
boolean headerFound = false;
byte chr = 0;
if ( state == 4 ){
while (true) {
// Read new bytes if needed
if (pos >= lastValid)
return false;
chr = byteBuffer.get(pos);
if ((chr == Constants.CR) || (chr == Constants.LF)) {
if (chr == Constants.LF) {
pos++;
headerLength = pos;
isFound = true;
state = 4;
return false;
}
} else {
break;
}
pos++;
}
state = 5;
}
// Mark the current buffer position
int start = pos;
if ( state == 5 ){
boolean colon = false;
while (!colon) {
// Read new bytes if needed
if (pos >= lastValid)
return false;
chr = byteBuffer.get(pos);
if (chr == Constants.COLON) {
colon = true;
if ( contentLength == -1
&& findBytes(ascbuf, start, start + (pos - start),
CL_HEADER ) != -1){
headerFound = true;
}
}
if ((chr >= Constants.A) && (chr <= Constants.Z)) {
chr = (byte) (chr - Constants.LC_OFFSET);
}
ascbuf[pos] = chr;
pos++;
}
state = 6;
}
// Mark the current buffer position
start = pos;
int realPos = pos;
boolean eol = false;
boolean validLine = true;
while (validLine) {
boolean space = true;
if ( state == 6 ){
// Skipping spaces
while (space) {
// Read new bytes if needed
if (pos >= lastValid)
return false;
chr = byteBuffer.get(pos);
if (( chr == Constants.SP) || (chr == Constants.HT)) {
pos++;
} else {
space = false;
}
}
state = 7;
}
int lastSignificantChar = realPos;
// Reading bytes until the end of the line
if ( state == 7){
while (!eol) {
// Read new bytes if needed
if (pos >= lastValid)
return false;
chr = byteBuffer.get(pos);
if (chr == Constants.CR) {
} else if (chr == Constants.LF) {
eol = true;
} else if (chr == Constants.SP) {
realPos++;
} else {
realPos++;
lastSignificantChar = realPos;
}
pos++;
}
state = 8;
realPos = lastSignificantChar;
}
if ( state == 8){
// Read new bytes if needed
if (pos >= lastValid)
return false;
chr = byteBuffer.get(pos);
if ((chr != Constants.SP) && (chr != Constants.HT)) {
validLine = false;
} else {
eol = false;
realPos++;
}
state = 4;
}
}
if ( headerFound ){
byteBuffer.position(start+1);
StringBuilder sb = new StringBuilder();
for(int i=0; i < realPos - start; i++) {
sb.append((char) byteBuffer.get());
}
contentLength = Integer.parseInt(sb.toString());
}
return true;
| protected boolean | parseRequestLine(java.nio.ByteBuffer byteBuffer)Parse the request line, looking for a POST method.
int start = 0;
byte chr = 0;
if ( state == 0 ){
do {
if (pos >= lastValid)
return false;
chr = byteBuffer.get(pos++);
} while ((chr == Constants.CR) || (chr == Constants.LF));
state = 1;
pos--;
byteBuffer.position(pos);
}
// Mark the current buffer position
start = pos;
boolean space = false;
if ( state == 1 ) {
while (!space) {
if (pos >= lastValid)
return false;
byte c = byteBuffer.get(pos);
ascbuf[pos] = c;
if (c == Constants.SP) {
space = true;
if ( !isFound &&
(findBytes(ascbuf,start, pos, POST_METHOD) == -1
|| findBytes(ascbuf,start, pos, PUT_METHOD) == -1)){
isFound = true;
}
}
pos++;
}
state = 2;
}
// Mark the current buffer position
start = pos;
int end = 0;
space = false;
boolean eol = false;
int questionPos = -1;
if ( state == 2 ){
while (!space) {
// Read new bytes if needed
if (pos >= lastValid)
return false;
byte b = byteBuffer.get(pos);
ascbuf[pos] = b;
if ( b == Constants.SP) {
space = true;
end = pos;
} else if ((b == Constants.CR)
|| (b == Constants.LF)) {
// HTTP/0.9 style request
eol = true;
space = true;
end = pos;
} else if ((b == Constants.QUESTION) && (questionPos == -1)){
questionPos = pos;
}
pos++;
}
state = 3;
startReq = start;
if (questionPos >= 0) {
lengthReq = questionPos - start;
} else {
lengthReq = end - start;
}
}
if ( state == 3 ) {
// Mark the current buffer position
start = pos;
end = 0;
while (!eol) {
// Read new bytes if needed
if (pos >= lastValid)
return false;
byte b = byteBuffer.get(pos);
if (b == Constants.CR) {
end = pos;
} else if (b == Constants.LF) {
if (end == 0)
end = pos;
eol = true;
}
pos++;
}
state = 4;
}
return eol;
| public void | recycle()Recylce this object.
contentLength = -1;
headerLength = -1;
curLimit = -1;
curPosition = -1;
pos = 0;
lastValid = 0;
requestLineParsed = false;
isFound = false;
state = 0;
lengthReq = -1;
startReq = -1;
socketChannel = null;
if ( handler != null){
handler.attachChannel(null);
}
|
|