在二进制文件中写入对齐的数据

Nick 发表于 Dev

缺口

我正在创建一个包含一些数据对象的文件。数据对象有不同的大小，是这样的（非常简化）：

struct Data{
   uint64_t size;
   char     blob[MAX_SIZE];
// ... methods here:
}

在稍后的某个步骤中，该文件将是内存中的 mmap()，因此我希望每个数据对象的开头都从uint64_t size将要存储的 8 个字节对齐的内存地址开始（让我们忽略字节序）。

代码看起来或多或少是这样的（目前硬编码 8 个字节）：

size_t calcAlign(size_t const value, size_t const align_size){
    return align_size - value % align_size;
}

template<class ITERATOR>
void process(std::ofstream &file_data, ITERATOR begin, ITERATOR end){
    for(auto it = begin; it != end; ++it){
        const auto &data = *it;

        size_t bytesWriten = data.writeToFile(file_data);

        size_t const alignToBeAdded = calcAlign(bytesWriten, 8);

        if (alignToBeAdded != 8){
            uint64_t const placeholder = 0;

            file_data.write( (const char *) & placeholder, (std::streamsize) alignToBeAdded);
        }
    }
}

这是在文件内实现对齐的最佳方法吗？

贝尔杜斯

你不需要依赖writeToFile返回大小，你可以使用ofstream::tellp

const auto beginPos = file_data.tellp();
// write stuff to file
const auto alignSize = (file_data.tellp()-beginPos)%8;
if(alignSize)
    file_data.write("\0\0\0\0\0\0\0\0",8-alignSize);

编辑发表 OP 评论：在一个最小的例子上进行了测试并且它有效。

#include <iostream>
#include <fstream>
int main(){
    using namespace std;
    ofstream file_data;
    file_data.open("tempfile.dat", ios::out | ios::binary);
    const auto beginPos = file_data.tellp();
    file_data.write("something", 9);
    const auto alignSize = (file_data.tellp() - beginPos) % 8;
    if (alignSize)
        file_data.write("\0\0\0\0\0\0\0\0", 8 - alignSize);
    file_data.close();
    return 0;
}

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。