diff options
Diffstat (limited to 'trunk/src/ChunkedEncoding.cc')
| -rw-r--r-- | trunk/src/ChunkedEncoding.cc | 173 |
1 files changed, 0 insertions, 173 deletions
diff --git a/trunk/src/ChunkedEncoding.cc b/trunk/src/ChunkedEncoding.cc deleted file mode 100644 index 80670601..00000000 --- a/trunk/src/ChunkedEncoding.cc +++ /dev/null @@ -1,173 +0,0 @@ -/* <!-- copyright */ -/* - * aria2 - a simple utility for downloading files faster - * - * Copyright (C) 2006 Tatsuhiro Tsujikawa - * - * This program is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation; either version 2 of the License, or - * (at your option) any later version. - * - * This program 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 General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with this program; if not, write to the Free Software - * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA - */ -/* copyright --> */ -#include "ChunkedEncoding.h" -#include "DlAbortEx.h" -#include "message.h" -#include <string.h> -#include <errno.h> - -#define MAX_BUFSIZE (1024*1024) - -ChunkedEncoding::ChunkedEncoding() { - strbufSize = 4096; - strbuf = new char[strbufSize]; - strbuf[0] = '\0'; - state = READ_SIZE; - chunkSize = 0; -} - -ChunkedEncoding::~ChunkedEncoding() { - if(strbuf != NULL) { - delete [] strbuf; - } -} - -void ChunkedEncoding::init() { -} - -bool ChunkedEncoding::finished() { - return state == FINISH ? true : false; -} - -void ChunkedEncoding::end() {} - -void ChunkedEncoding::inflate(char* outbuf, int& outlen, const char* inbuf, int inlen) { - addBuffer(inbuf, inlen); - char* p = strbuf; - int clen = 0; - while(1) { - if(state == READ_SIZE) { - if(readChunkSize(&p) == 0) { - if(chunkSize == 0) { - state = FINISH; - } else { - state = READ_DATA; - } - } else { - // chunk size is not fully received. - break; - } - } else if(state == READ_DATA) { - if(readData(&p, outbuf, clen, outlen) == 0) { - state = READ_SIZE; - } else { - break; - } - } else { - break; - } - // Was all bytes in strbuf examined? - if(strbuf+strlen(strbuf) <= p) { - break; - } - } - if(strbuf+strlen(strbuf) <= p) { - // make strbuf NULL-string - strbuf[0] = '\0'; - } else { - // copy string between [p, strbuf+strlen(strbuf)+1], +1 is for NULL - // character. - char* temp = new char[strbufSize]; - memcpy(temp, p, strbuf+strlen(strbuf)-p+1); - delete [] strbuf; - strbuf = temp; - } - outlen = clen; -} - -int ChunkedEncoding::readData(char** pp, char* buf, int& len, int maxlen) { - if(buf+len == buf+maxlen) { - return -1; - } - if(chunkSize == 0) { - return readDataEOL(pp); - } - int wsize; - if(strlen(*pp) < (unsigned long int)chunkSize) { - wsize = strlen(*pp) <= (unsigned int)maxlen-len ? strlen(*pp) : maxlen-len; - } else { - wsize = chunkSize <= maxlen-len ? chunkSize : maxlen-len; - } - memcpy(buf+len, *pp, wsize); - chunkSize -= wsize; - len += wsize; - *pp += wsize; - - if(chunkSize == 0) { - return readDataEOL(pp); - } else { - return -1; - } -} - -int ChunkedEncoding::readDataEOL(char** pp) { - if(strstr(*pp, "\r\n") == *pp) { - *pp += 2; - return 0; - } else if(strlen(*pp) < 2) { - return -1; - } else { - throw new DlAbortEx(EX_INVALID_CHUNK_SIZE); - } -} - -// strbuf is NULL terminated string, and inlen is strlen(strbuf). -// therefore, strbuf[inlen] = '\0' -int ChunkedEncoding::readChunkSize(char** pp) { - // we read chunk-size from *pp - char* p = strstr(*pp, "\r\n"); - // \r\n is not found. Return -1 - if(p == NULL) { - return -1; - } - // We ignore chunk-extension - char* exsp = index(*pp, ';'); - if(exsp == NULL) { - exsp = p; - } - // TODO check invalid characters in buffer - chunkSize = strtol(*pp, NULL, 16); - if(chunkSize < 0) { - throw new DlAbortEx(EX_INVALID_CHUNK_SIZE); - } else if(errno == ERANGE && (chunkSize == LONG_MAX || chunkSize == LONG_MIN)) { - throw new DlAbortEx(strerror(errno)); - } - *pp = p+2; - - return 0; -} - -void ChunkedEncoding::addBuffer(const char* inbuf, int inlen) { - if(strlen(strbuf)+inlen >= (unsigned int)strbufSize) { - if(strlen(strbuf)+inlen+1 > MAX_BUFSIZE) { - throw new DlAbortEx(EX_TOO_LARGE_CHUNK, strlen(strbuf)+inlen+1); - } - strbufSize = strlen(strbuf)+inlen+1; - char* temp = new char[strbufSize]; - memcpy(temp, strbuf, strlen(strbuf)+1); - delete [] strbuf; - strbuf = temp; - } - int origlen = strlen(strbuf); - memcpy(strbuf+origlen, inbuf, inlen); - strbuf[origlen+inlen] = '\0'; -} |
