Linux下O_DIRECT直接存储IO的对齐、偏移与计数计算问题

原代码存在的问题 #

1. aligned_offset计算逻辑错误：offset >> 12得到的是4096字节块的索引（3），而非对齐后的字节偏移量，需要左移12位才能得到正确的起始地址（12288）。
2. aligned_count未考虑起始偏移的剩余部分，仅对数据长度做了简单的向上取整，导致最终读取范围无法覆盖全部请求数据。

修正后的计算代码 #

// Given: count=7000, offset=14000
const u64 ALIGNMENT = 4096;
const u64 ALIGN_MASK = ALIGNMENT - 1; // 等价于 0xFFF，用于位运算对齐

u64 aligned_offset = offset & ~ALIGN_MASK;       // 向下对齐到4096字节边界，结果：12288
u64 end_offset = offset + count;                 // 原始请求的结束位置，结果：21000
u64 aligned_end_offset = (end_offset + ALIGN_MASK) & ~ALIGN_MASK; // 向上对齐到4096字节边界，结果：24576
u64 aligned_count = aligned_end_offset - aligned_offset;         // 最终需要读取的对齐后字节数，结果：12288

关键逻辑说明 #

• 向下对齐起始偏移：通过& ~ALIGN_MASK位运算，直接将原始偏移的低12位清零，快速得到小于等于原始偏移的最大4096字节对齐地址。
• 向上对齐结束位置：给原始结束位置加上ALIGN_MASK后再做清零低12位操作，确保得到大于等于原始结束位置的最小4096字节对齐地址，保证覆盖所有请求数据。
• 计算对齐后长度：用对齐后的结束地址减去起始地址，得到的长度既满足Direct-IO的对齐要求，又能完整覆盖用户请求的全部数据范围。

内容的提问来源于stack exchange，提问作者gitman-2021