basic_stringbuf::in_avail() 和 basic_stringbuf::str() 的可靠性如何？

Question

我需要从流中获取所有内容，而不实际提取它们（就像 stringstream::str() 一样）。我尝试过 basic_stringbuf::str() ，但当流为空时它的行为不正确。为了避免这种情况，我尝试了 basic_stringbuf::in_avail()，但效果也不是很好。

在以下测试用例中，in_avail() 不返回流上可用元素的数量，而 str() 返回更多 个元素比当前存在的内容：

#include <iostream>
#include <iterator>
#include <vector>
#include <sstream>

// extracts everything from the stream
std::vector<unsigned char> stream2vector(std::basic_istream<unsigned char>& stream)
{
    std::vector<unsigned char> retreivedData;
    std::istreambuf_iterator<unsigned char> it(stream);
    const std::istreambuf_iterator<unsigned char> endOfStream;
    retreivedData.insert(retreivedData.begin(), it, endOfStream);
    return retreivedData;
}

int main() {
    std::basic_stringbuf<unsigned char> buf;
    std::basic_iostream<unsigned char> stream(&buf);
    unsigned char array[5] = { 1, 2, 3, 4, 5 };
    stream.write(array, 5);


    std::cout << "rdbuf()->in_avail(): " << buf.in_avail() << "\n";
    std::vector<unsigned char> d1 = stream2vector(stream);
    std::cout << "d1.size(): " << d1.size() << "\n";

    std::cout << "\n";
    // d2 should be empty
    std::vector<unsigned char> d2 = stream2vector(stream);
    std::cout << "d2.size(): " << d2.size() << "\n";
    std::basic_string<unsigned char> s = buf.str();
    std::cout << "buf.str().size(): " << buf.str().size() << "\n";
}

在 g++ 4.4 上编译，输出是：

rdbuf()->in_avail(): 1 // expected: 5
d1.size(): 5 // as expected

d2.size(): 0 // as expected
buf.str().size(): 5 // expected: 0

我做错了什么？做我正在尝试的事情的最佳方法是什么？

多谢。

Answer 1

in_avail 是准备从缓冲区读取的字符数，而不是缓冲区本身的大小。这里确实允许返回任何非零值。

但是，我无法回答您正在做的事情的最佳方法是什么，因为我不知道您在做什么。如果您已经拥有 unsigned char 数组的内容，那么您将想要执行以下操作：

std::vector<unsigned char> data(array, array + sizeof(array)/sizeof(unsigned char));

如果您只是尝试将整个流读入向量，那么我会按照您的方式执行操作正在做；我只是用这个等效的、更简单的函数替换您的stream2vector函数：

// extracts everything from the stream
std::vector<unsigned char> stream2vector(std::basic_istream<unsigned char>& stream)
{
    std::istreambuf_iterator<unsigned char> it(stream);
    const std::istreambuf_iterator<unsigned char> endOfStream;
    return std::vector<unsigned char>(it, endOfStream);
}

我不完全确定为什么您将此处的每个操作专门用于unsigned char - 我只会使用默认值char 版本，因为 unsigned char 允许与 short 具有相同的大小，这可能不是您想要的（但我不是了解执行此操作的任何实现）。