001: /*
002: * Copyright (c) 1998-2008 Caucho Technology -- all rights reserved
003: *
004: * This file is part of Resin(R) Open Source
005: *
006: * Each copy or derived work must preserve the copyright notice and this
007: * notice unmodified.
008: *
009: * Resin Open Source is free software; you can redistribute it and/or modify
010: * it under the terms of the GNU General Public License as published by
011: * the Free Software Foundation; either version 2 of the License, or
012: * (at your option) any later version.
013: *
014: * Resin Open Source is distributed in the hope that it will be useful,
015: * but WITHOUT ANY WARRANTY; without even the implied warranty of
016: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE, or any warranty
017: * of NON-INFRINGEMENT. See the GNU General Public License for more
018: * details.
019: *
020: * You should have received a copy of the GNU General Public License
021: * along with Resin Open Source; if not, write to the
022: * Free SoftwareFoundation, Inc.
023: * 59 Temple Place, Suite 330
024: * Boston, MA 02111-1307 USA
025: *
026: * @author Scott Ferguson
027: */
028:
029: package com.caucho.server.http;
030:
031: import com.caucho.vfs.ReadStream;
032: import com.caucho.vfs.StreamImpl;
033:
034: import java.io.IOException;
035:
036: /**
037: * StreamImpl so servlets can read POST data as a normal stream.
038: */
039: class ChunkedInputStream extends StreamImpl {
040: private ReadStream _next;
041: private int _available;
042:
043: void init(ReadStream next) {
044: _next = next;
045: _available = 0;
046: }
047:
048: public boolean canRead() {
049: return true;
050: }
051:
052: public int getAvailable() {
053: return _available;
054: }
055:
056: /**
057: * Reads more data from the input stream.
058: */
059: public int read(byte[] buf, int offset, int len) throws IOException {
060: // The chunk still has more data left
061: if (_available > 0) {
062: if (_available < len)
063: len = _available;
064:
065: len = _next.read(buf, offset, len);
066:
067: if (len > 0)
068: _available -= len;
069: }
070: // The chunk is done, so read the next chunk
071: else if (_available == 0) {
072: _available = readChunkLength();
073:
074: // the new chunk has data
075: if (_available > 0) {
076: if (_available < len)
077: len = _available;
078:
079: len = _next.read(buf, offset, len);
080:
081: if (len > 0)
082: _available -= len;
083: }
084: // the new chunk is the last
085: else {
086: _available = -1;
087: len = -1;
088: }
089: } else
090: len = -1;
091:
092: return len;
093: }
094:
095: /**
096: * Reads the next chunk length from the input stream.
097: */
098: private int readChunkLength() throws IOException {
099: int length = 0;
100: int ch;
101:
102: // skip whitespace
103: for (ch = _next.read(); ch == '\r' || ch == ' ' || ch == '\n'; ch = _next
104: .read()) {
105: }
106:
107: // XXX: This doesn't properly handle the case when when the browser
108: // sends headers at the end of the data. See the HTTP/1.1 spec.
109: for (; ch > 0 && ch != '\r' && ch != '\n'; ch = _next.read()) {
110: if (ch >= '0' && ch <= '9')
111: length = 16 * length + ch - '0';
112: else if (ch >= 'a' && ch <= 'f')
113: length = 16 * length + ch - 'a' + 10;
114: else if (ch >= 'A' && ch <= 'F')
115: length = 16 * length + ch - 'A' + 10;
116: else if (ch == ' ' || ch == '\t') {
117: //if (dbg.canWrite())
118: // dbg.println("unexpected chunk whitespace.");
119: } else
120: throw new IOException(
121: "HTTP/1.1 protocol error: bad chunk");
122: }
123:
124: if (ch == '\r')
125: ch = _next.read();
126:
127: return length;
128: }
129: }
|