自定义malloc内存对齐问题：为何替代方案未通过特定测试？

我在实现自定义malloc时遇到测试失败，已经改用alignof(max_align_t)（或alignof(long double)）解决了问题，但想搞明白原方案无效的原因，验证自己对内存对齐的理解。

原思路是：如果当前地址不能被申请类型的实际大小整除，就填充到满足条件，让编译器高效访问内存。但原代码用申请大小sz作为对齐基准，通不过测试，尤其是调用malloc(1)时，相关对齐断言直接失败。

原malloc实现代码 #

void *malloc(size_t sz, const char *file, int line) {
  (void)file, (void)line; // avoid uninitialized variable warnings
  if (default_buffer.pos + sz > default_buffer.size) {
    // Not enough space left in default buffer for allocation
    gstats.nfail++;
    gstats.fail_size += sz;
    return nullptr;
  }
  // Here i'm checking for correct allocation alignement
  if (default_buffer.pos % sz != 0) {
    default_buffer.pos += default_buffer.pos % sz;
  }
  // Otherwise there is enough space; claim the next `sz` bytes
  void *ptr = &default_buffer.buffer[default_buffer.pos];

  default_buffer.pos += sz;
  gstats.ntotal++;
  gstats.total_size += sz;
  return ptr;
}

测试代码（test.c） #

int main() {
double* ptr = (double*) malloc(sizeof(double));
assert((uintptr_t) ptr % alignof(double) == 0);
assert((uintptr_t) ptr % alignof(unsigned long long) == 0);
assert((uintptr_t) ptr % alignof(std::max_align_t) == 0);

    char* ptr2 = (char*) malloc(1);
    assert((uintptr_t) ptr2 % alignof(double) == 0);
    assert((uintptr_t) ptr2 % alignof(unsigned long long) == 0);
    assert((uintptr_t) ptr2 % alignof(std::max_align_t) == 0);
    
    free(ptr);
    free(ptr2);

}

原方案无效的核心原因 #

原代码的对齐逻辑完全搞错了内存对齐的核心要求：

1. 对齐基准错误：内存对齐的要求不是按「申请的内存块大小」来定，而是要满足「系统中所有基础类型的最大对齐要求」。C/C++标准明确规定，malloc返回的指针必须能安全转换为任何标量类型的指针，因此必须对齐到std::max_align_t的对齐值——这是当前平台所有基础类型中最严格的对齐要求。
2. 极端场景失效：当调用malloc(1)时，sz=1，任何地址对1取模结果都是0，原代码不会做任何对齐填充，返回的地址仅满足1字节对齐，但测试要求它必须满足double、unsigned long long这类大类型的对齐要求，必然断言失败。
3. 无法覆盖全局对齐要求：即使是malloc(sizeof(double))，如果平台上std::max_align_t的对齐要求高于double（比如部分平台long double的对齐是16字节，而double是8字节），原代码的对齐逻辑也无法满足max_align_t的断言。

正确的对齐逻辑 #

不管申请的内存大小是多少，都要将地址对齐到alignof(std::max_align_t)的倍数。修正后的代码如下：

void *malloc(size_t sz, const char *file, int line) {
  (void)file, (void)line; // 避免未初始化变量警告
  const size_t align = alignof(std::max_align_t);
  // 计算需要填充的字节数，确保pos对齐到align的倍数
  size_t padding = (align - (default_buffer.pos % align)) % align;
  
  // 先检查对齐后是否有足够空间
  if (default_buffer.pos + padding + sz > default_buffer.size) {
    gstats.nfail++;
    gstats.fail_size += sz;
    return nullptr;
  }
  
  default_buffer.pos += padding;
  void *ptr = &default_buffer.buffer[default_buffer.pos];

  default_buffer.pos += sz;
  gstats.ntotal++;
  gstats.total_size += sz;
  return ptr;
}

总结 #

原方案的错误在于混淆了「内存块大小」和「对齐要求」两个概念。内存对齐的本质是为了让CPU高效访问数据，而标准malloc必须满足全局最严格的对齐要求，才能保证任何类型的对象都能被正确放置在分配的内存中。

内容的提问来源于stack exchange，提问作者Grainme