如何使用boost iostream进行流式解压缩

时间:2017-06-21 15:38:46

标签: c++ boost zlib boost-iostreams

我正在使用boost iostreams(1.64.0)来解压缩zlib数据。 我想做流式解压缩。这意味着压缩数据的大小不可预测。 我编写了以下代码示例。

#include <sstream>
#include <string>
#include <iostream>

#include <boost/iostreams/filtering_streambuf.hpp>
#include <boost/iostreams/copy.hpp>
#include <boost/iostreams/filter/zlib.hpp>

int main() {
    // Compress
    std::stringstream sender;
    boost::iostreams::filtering_streambuf<boost::iostreams::input> out;
    out.push(boost::iostreams::zlib_compressor());
    out.push(sender);
    sender << "Hello World";
    std::stringstream compressed;
    boost::iostreams::copy(out, compressed);

    // Decompress
    boost::iostreams::filtering_streambuf<boost::iostreams::input> in;
    in.push(boost::iostreams::zlib_decompressor());
    in.push(compressed);
    std::istream is(&in);
    std::size_t const buf_size = 256;
    char buf[buf_size] = { '\0' };
#if 0
    is.getline(buf, buf_size); // works fine
#else
    std::size_t read_size = is.readsome(buf, buf_size);
    std::cout << "read_size:" << read_size << std::endl;
#endif
    // http://www.cplusplus.com/reference/ios/ios/rdstate/
    std::cout << "rdstate:" << is.rdstate() << std::endl;
    std::cout << buf << std::endl;

}

我使用readsome(),因为数据的大小是不可预测的。 我得到了以下输出:

read_size:0
rdstate:0

这对我来说意外。

如果我使用getline()代替readsome(),我会得到以下输出:

rdstate:2
Hello World

预计产量。

我认为当我使用readsome()时,输出应该是相同的。 我不能在实际代码中使用getline()因为原始数据是二进制格式。

有没有办法使用readsome() filtering_streambuf或任何好的方法来解压缩不可预测的长度二进制数据?

1 个答案:

答案 0 :(得分:2)

感谢sehe's comment,问题就解决了。

我为评论写了the response,但很难阅读,因为代码格式不正确。所以我自己回答。我希望它能帮助其他有类似问题的人。

我替换了

std::size_t read_size = is.readsome(buf, buf_size);

is.read(buf, buf_size);
std::size_t read_size = is.gcount();

,然后问题就解决了。

我误解了std::istream::read阻塞,直到读取buf_size长度数据。这不是真的。即使实际读取大小小于buf_size,函数也会返回。见http://www.cplusplus.com/reference/istream/istream/read/。为了获得read_size,我致电std::istream::gcount()。请参阅http://www.cplusplus.com/reference/istream/istream/gcount/

注意:我对boost::asio::readboost::asio::ip::tcp::socket::read_some感到困惑。但他们的行为与std::istream的行为不同。

以下是修复版的完整代码:

#include <sstream>
#include <string>
#include <iostream>

#include <boost/iostreams/filtering_streambuf.hpp>
#include <boost/iostreams/copy.hpp>
#include <boost/iostreams/filter/zlib.hpp>

int main() {
    // Compress
    std::stringstream sender;
    boost::iostreams::filtering_streambuf<boost::iostreams::input> out;
    out.push(boost::iostreams::zlib_compressor());
    out.push(sender);
    sender << "Hello World";
    std::stringstream compressed;
    boost::iostreams::copy(out, compressed);

    // Decompress
    boost::iostreams::filtering_streambuf<boost::iostreams::input> in;
    in.push(boost::iostreams::zlib_decompressor());
    in.push(compressed);
    std::istream is(&in);
    std::size_t const buf_size = 256;
    char buf[buf_size] = { '\0' };
    is.read(buf, buf_size);
    std::size_t read_size = is.gcount();
    std::cout << "read_size:" << read_size << std::endl;
    // http://www.cplusplus.com/reference/ios/ios/rdstate/
    std::cout << "rdstate:" << is.rdstate() << std::endl;
    std::cout << buf << std::endl;

}