diff options
| author | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2008-06-29 14:29:36 +0000 |
|---|---|---|
| committer | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2008-06-29 14:29:36 +0000 |
| commit | d791807addd686660fd75d721b3cb0cf14d52f67 (patch) | |
| tree | c882dc44d5456d37fb8738e53cc46618b1b87bab /src/GZipDecoder.cc | |
| parent | 6c4dd16c82fe8dadd7a598cca998b4273436f0bd (diff) | |
2008-06-29 Tatsuhiro Tsujikawa <tujikawa at rednoah dot com>
Added gzip, deflate decoding support in HTTP using libz. If
compiled
with this feature, aria2 sends "Accept-Encoding: deflate, gzip"
header
to a HTTP server. If a server returns "Content-Encoding: gzip"
or
"Content-Encoding: deflate" then, aria2 decodes the response
body on the
fly and writes decoded data to a local disk.
* README
* README.html
* configure.ac
* m4/aria2_arg.m4: Added ARIA2_ARG_WITH and ARIA2_ARG_ENABLE,
they are
wrapper function for AC_ARG_WITH and AC_ARG_ENABLE respectively.
* m4/libz.m4
* src/Decoder.h
* src/DownloadCommand.cc
* src/DownloadCommand.h
* src/Exception.h
* src/GZipDecoder.cc
* src/GZipDecoder.h
* src/HttpHeader.cc
* src/HttpHeader.h
* src/HttpRequest.cc
* src/HttpRequest.h
* src/HttpResponse.cc
* src/HttpResponse.h
* src/HttpResponseCommand.cc
* src/Makefile.am
* test/GZipDecoderTest.cc
* test/HttpRequestTest.cc
* test/HttpResponseTest.cc
* test/Makefile.am
* test/Makefile.in
* test/gzip_decode_test.gz
Diffstat (limited to 'src/GZipDecoder.cc')
| -rw-r--r-- | src/GZipDecoder.cc | 119 |
1 files changed, 119 insertions, 0 deletions
diff --git a/src/GZipDecoder.cc b/src/GZipDecoder.cc new file mode 100644 index 00000000..81f46d2c --- /dev/null +++ b/src/GZipDecoder.cc @@ -0,0 +1,119 @@ +/* <!-- copyright */ +/* + * aria2 - The high speed download utility + * + * 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + * + * In addition, as a special exception, the copyright holders give + * permission to link the code of portions of this program with the + * OpenSSL library under certain conditions as described in each + * individual source file, and distribute linked combinations + * including the two. + * You must obey the GNU General Public License in all respects + * for all of the code used other than OpenSSL. If you modify + * file(s) with this exception, you may extend this exception to your + * version of the file(s), but you are not obligated to do so. If you + * do not wish to do so, delete this exception statement from your + * version. If you delete this exception statement from all source + * files in the program, then also delete it here. + */ +/* copyright --> */ +#include "GZipDecoder.h" +#include "StringFormat.h" +#include "DlAbortEx.h" + +namespace aria2 { + +const std::string GZipDecoder::NAME("GZipDecoder"); + +GZipDecoder::GZipDecoder():_strm(0), _finished(false) {} + +GZipDecoder::~GZipDecoder() +{ + release(); +} + +void GZipDecoder::init() +{ + _strm = new z_stream(); + _strm->zalloc = Z_NULL; + _strm->zfree = Z_NULL; + _strm->opaque = Z_NULL; + _strm->avail_in = 0; + _strm->next_in = Z_NULL; + + // initalize z_stream with gzip/zlib format auto detection enabled. + if(Z_OK != inflateInit2(_strm, 47)) { + throw DlAbortEx("Initializing z_stream failed."); + } +} + +void GZipDecoder::release() +{ + if(_strm) { + inflateEnd(_strm); + _strm = 0; + } +} + +std::string GZipDecoder::decode(const unsigned char* in, size_t length) +{ + std::string out; + + if(length == 0) { + return out; + } + + _strm->avail_in = length; + _strm->next_in = const_cast<unsigned char*>(in); + + unsigned char outbuf[OUTBUF_LENGTH]; + while(1) { + _strm->avail_out = OUTBUF_LENGTH; + _strm->next_out = outbuf; + + int ret = ::inflate(_strm, Z_NO_FLUSH); + + if(ret == Z_STREAM_END) { + _finished = true; + } else if(ret != Z_OK) { + throw DlAbortEx(StringFormat("libz::inflate() failed. cause:%s", + _strm->msg).str()); + } + + size_t produced = OUTBUF_LENGTH-_strm->avail_out; + + out.insert(out.end(), &outbuf[0], &outbuf[produced]); + + if(_strm->avail_out > 0) { + break; + } + } + return out; +} + +bool GZipDecoder::finished() +{ + return _finished; +} + +const std::string& GZipDecoder::getName() const +{ + return NAME; +} + +} // namespace aria2 |
