[29] | 1 | // (C) Copyright Jonathan Turkanis 2003. |
---|
| 2 | // Distributed under the Boost Software License, Version 1.0. (See accompanying |
---|
| 3 | // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt.) |
---|
| 4 | |
---|
| 5 | // See http://www.boost.org/libs/iostreams for documentation. |
---|
| 6 | |
---|
| 7 | // To configure Boost to work with libbz2, see the |
---|
| 8 | // installation instructions here: |
---|
| 9 | // http://boost.org/libs/iostreams/doc/index.html?path=7 |
---|
| 10 | |
---|
| 11 | // Define BOOST_IOSTREAMS_SOURCE so that <boost/iostreams/detail/config.hpp> |
---|
| 12 | // knows that we are building the library (possibly exporting code), rather |
---|
| 13 | // than using it (possibly importing code). |
---|
| 14 | #define BOOST_IOSTREAMS_SOURCE |
---|
| 15 | |
---|
| 16 | #include <boost/iostreams/detail/config/dyn_link.hpp> |
---|
| 17 | #include <boost/iostreams/filter/bzip2.hpp> |
---|
| 18 | #include "bzlib.h" // Julian Seward's "bzip.h" header. |
---|
| 19 | // To configure Boost to work with libbz2, see the |
---|
| 20 | // installation instructions here: |
---|
| 21 | // http://boost.org/libs/iostreams/doc/index.html?path=7 |
---|
| 22 | |
---|
| 23 | namespace boost { namespace iostreams { |
---|
| 24 | |
---|
| 25 | namespace bzip2 { |
---|
| 26 | |
---|
| 27 | // Status codes |
---|
| 28 | |
---|
| 29 | const int ok = BZ_OK; |
---|
| 30 | const int run_ok = BZ_RUN_OK; |
---|
| 31 | const int flush_ok = BZ_FLUSH_OK; |
---|
| 32 | const int finish_ok = BZ_FINISH_OK; |
---|
| 33 | const int stream_end = BZ_STREAM_END; |
---|
| 34 | const int sequence_error = BZ_SEQUENCE_ERROR; |
---|
| 35 | const int param_error = BZ_PARAM_ERROR; |
---|
| 36 | const int mem_error = BZ_MEM_ERROR; |
---|
| 37 | const int data_error = BZ_DATA_ERROR; |
---|
| 38 | const int data_error_magic = BZ_DATA_ERROR_MAGIC; |
---|
| 39 | const int io_error = BZ_IO_ERROR; |
---|
| 40 | const int unexpected_eof = BZ_UNEXPECTED_EOF; |
---|
| 41 | const int outbuff_full = BZ_OUTBUFF_FULL; |
---|
| 42 | const int config_error = BZ_CONFIG_ERROR; |
---|
| 43 | |
---|
| 44 | // Action codes |
---|
| 45 | |
---|
| 46 | const int finish = BZ_FINISH; |
---|
| 47 | const int run = BZ_RUN; |
---|
| 48 | |
---|
| 49 | } // End namespace bzip2. |
---|
| 50 | |
---|
| 51 | //------------------Implementation of bzip2_error-----------------------------// |
---|
| 52 | |
---|
| 53 | bzip2_error::bzip2_error(int error) |
---|
| 54 | : BOOST_IOSTREAMS_FAILURE("bzip2 error"), error_(error) |
---|
| 55 | { } |
---|
| 56 | |
---|
| 57 | void bzip2_error::check(int error) |
---|
| 58 | { |
---|
| 59 | switch (error) { |
---|
| 60 | case BZ_OK: |
---|
| 61 | case BZ_RUN_OK: |
---|
| 62 | case BZ_FLUSH_OK: |
---|
| 63 | case BZ_FINISH_OK: |
---|
| 64 | case BZ_STREAM_END: |
---|
| 65 | return; |
---|
| 66 | case BZ_MEM_ERROR: |
---|
| 67 | throw std::bad_alloc(); |
---|
| 68 | default: |
---|
| 69 | throw bzip2_error(error); |
---|
| 70 | } |
---|
| 71 | } |
---|
| 72 | |
---|
| 73 | //------------------Implementation of bzip2_base------------------------------// |
---|
| 74 | |
---|
| 75 | namespace detail { |
---|
| 76 | |
---|
| 77 | bzip2_base::bzip2_base(const bzip2_params& params) |
---|
| 78 | : params_(params), stream_(new bz_stream), ready_(false) |
---|
| 79 | { } |
---|
| 80 | |
---|
| 81 | bzip2_base::~bzip2_base() { delete static_cast<bz_stream*>(stream_); } |
---|
| 82 | |
---|
| 83 | void bzip2_base::before( const char*& src_begin, const char* src_end, |
---|
| 84 | char*& dest_begin, char* dest_end ) |
---|
| 85 | { |
---|
| 86 | bz_stream* s = static_cast<bz_stream*>(stream_); |
---|
| 87 | s->next_in = const_cast<char*>(src_begin); |
---|
| 88 | s->avail_in = static_cast<unsigned>(src_end - src_begin); |
---|
| 89 | s->next_out = reinterpret_cast<char*>(dest_begin); |
---|
| 90 | s->avail_out= static_cast<unsigned>(dest_end - dest_begin); |
---|
| 91 | } |
---|
| 92 | |
---|
| 93 | void bzip2_base::after(const char*& src_begin, char*& dest_begin) |
---|
| 94 | { |
---|
| 95 | bz_stream* s = static_cast<bz_stream*>(stream_); |
---|
| 96 | src_begin = const_cast<char*>(s->next_in); |
---|
| 97 | dest_begin = s->next_out; |
---|
| 98 | } |
---|
| 99 | |
---|
| 100 | void bzip2_base::end(bool compress) |
---|
| 101 | { |
---|
| 102 | ready_ = false; |
---|
| 103 | bz_stream* s = static_cast<bz_stream*>(stream_); |
---|
| 104 | bzip2_error::check( |
---|
| 105 | compress ? |
---|
| 106 | BZ2_bzCompressEnd(s) : |
---|
| 107 | BZ2_bzDecompressEnd(s) |
---|
| 108 | ); |
---|
| 109 | } |
---|
| 110 | |
---|
| 111 | int bzip2_base::compress(int action) |
---|
| 112 | { |
---|
| 113 | return BZ2_bzCompress(static_cast<bz_stream*>(stream_), action); |
---|
| 114 | } |
---|
| 115 | |
---|
| 116 | int bzip2_base::decompress() |
---|
| 117 | { |
---|
| 118 | return BZ2_bzDecompress(static_cast<bz_stream*>(stream_)); |
---|
| 119 | } |
---|
| 120 | |
---|
| 121 | void bzip2_base::do_init |
---|
| 122 | ( bool compress, |
---|
| 123 | #if !BOOST_WORKAROUND(BOOST_MSVC, < 1300) |
---|
| 124 | bzip2::alloc_func alloc, |
---|
| 125 | bzip2::free_func free, |
---|
| 126 | #endif |
---|
| 127 | void* derived ) |
---|
| 128 | { |
---|
| 129 | bz_stream* s = static_cast<bz_stream*>(stream_); |
---|
| 130 | |
---|
| 131 | // Current interface for customizing memory management |
---|
| 132 | // is non-conforming and has been disabled: |
---|
| 133 | //#if !BOOST_WORKAROUND(BOOST_MSVC, < 1300) |
---|
| 134 | // s->bzalloc = alloc; |
---|
| 135 | // s->bzfree = free; |
---|
| 136 | //#else |
---|
| 137 | s->bzalloc = 0; |
---|
| 138 | s->bzfree = 0; |
---|
| 139 | //#endif |
---|
| 140 | s->opaque = derived; |
---|
| 141 | bzip2_error::check( |
---|
| 142 | compress ? |
---|
| 143 | BZ2_bzCompressInit( s, |
---|
| 144 | params_.block_size, |
---|
| 145 | 0, |
---|
| 146 | params_.work_factor ) : |
---|
| 147 | BZ2_bzDecompressInit( s, |
---|
| 148 | 0, |
---|
| 149 | params_.small ) |
---|
| 150 | ); |
---|
| 151 | ready_ = true; |
---|
| 152 | } |
---|
| 153 | |
---|
| 154 | } // End namespace detail. |
---|
| 155 | |
---|
| 156 | //----------------------------------------------------------------------------// |
---|
| 157 | |
---|
| 158 | } } // End namespaces iostreams, boost. |
---|