自定义 std::fstream,std::filebuf 的溢出和下溢函数未为每个字符调用

custom std::fstream, std::filebuf's overflow and underflow function not invoked for every character

本文关键字:std 字符 调用 函数 fstream filebuf 自定义 溢出 下溢      更新时间:2023-10-16

我正在尝试制作一个自定义std::fstream,它将在读取时对数据进行编码/解码。

template <class T>
class _filebuf : public std::filebuf {
public:
using transform_type = T;
int_type underflow() override {
auto c = std::filebuf::underflow();
return c < 0 ? c : transform.decode(c);
}
int_type overflow(int_type c) override {
return c < 0 ? c : std::filebuf::overflow(transform.encode(c));
}
private:
transform_type transform;
};
template <class T>
class _fstream : public std::iostream {
public:
using buffer_type = _filebuf<T>;

explicit _fstream(const std::string& path, std::ios::openmode openmode)
: std::iostream(0)
{
this->init(&buffer);
buffer.open(path, openmode);
}
private:
buffer_type buffer;
};

下面是一个使用示例:

class _transform {
public:
template <class T>
T encode(T value) const {
return value - 1;
}

template <class T>
T decode(T value) const {
return value + 1;
}    
};
int main() {
_fstream<_transform> ofs("test.txt", std::ios::out | std::ios::trunc);
ofs << "ABC"; // outputs "@BC" to the file (@ is 64 in ASCII, so only first character encoded properly)
_fstream<_transform> ifs("test.txt", std::ios::in);
std::string s;
ifs >> s; // inputs  "ABC" when "@BC" was in the file so again only first character is decoded
// ...
};

经过我自己的研究,我发现">溢出"函数在这个过程中被调用了两次(65和-1,其中-1可能是EOF(,"下溢">也被调用了两次(64和-1(。由于其他字符不会丢失,因此可能会在不调用这些函数的情况下以某种方式处理它们。

为什么会发生这种情况以及如何改变它?

std::streambuf<CharT,Traits>::underflow

确保get区域中至少有一个字符可用,std::filebuf的有效实现将始终尝试将完整的缓冲区字符读取到get区域中。除非您寻求流,否则underflow不会再次调用,直到对sgetn/xsgetnsbumpc的调用清空获取区域

我认为您可能会更成功地包装而不是扩展文件缓冲区。

Boost iostreams 使写入流过滤器变得更加简单。