如何在C++项目中实现一个aligned_malloc方法

本文主要介绍"怎么在C++项目中实现一个aligned_malloc方法"，希望能够解决您遇到有关问题，下面我们一起来看这篇 "怎么在C++项目中实现一个aligned_malloc方法" 文章。

malloc的默认行为

int main()
{
  void *p = malloc(1024);
  printf("0x%p\n", p);
  free(p);
}

请求了一个大小为1024的内存块并打印出来，一切都很完美。
我们看看这块内存的地址。

如何在C++项目中实现一个aligned_malloc方法

可以看到，在64bit机器上，malloc默认行为会将分配的地址以16-byte对齐，如果我们想改变这种默认行为，提供32-byte或者64-byte对齐，应该怎么做呢？

实现aligned_malloc

源代码

从C++17开始，可以使用aligned_alloc函数达到这个目的，但是如果使用较老的C++版本，如C++14，C++11，我们需要手动写一个实现。
话不多说，先贴代码如下，aligned_malloc和aligned_free，需要配合使用，否则会有内存泄漏问题。

#include <memory>

void* aligned_malloc(size_t size, size_t alignment)
{
	size_t offset = alignment - 1 + sizeof(void*);
	void * originalP = malloc(size + offset);
	size_t originalLocation = reinterpret_cast<size_t>(originalP);
	size_t realLocation = (originalLocation + offset) & ~(alignment - 1);
	void * realP = reinterpret_cast<void*>(realLocation);
	size_t originalPStorage = realLocation - sizeof(void*);
	*reinterpret_cast<void**>(originalPStorage) = originalP;
	return realP;
}

void aligned_free(void* p)
{
	size_t originalPStorage = reinterpret_cast<size_t>(p) - sizeof(void*);
	free(*reinterpret_cast<void**>(originalPStorage));
}

int main()
{	
	void * p = aligned_malloc(1024, 64);
	printf("0x%p\n", p);
	aligned_free(p);
	return 0;
}

添加一个测试程序，

#include <assert.h>

void TestAlignedMalloc()
{
  const int size = 100;
  const int alignment = 64;
  void* testArray[size];
  for (int i = 0; i < size; ++i)
  {
    void * p = aligned_malloc(1024, alignment);
    assert((reinterpret_cast<size_t>(p) & (alignment - 1)) == 0);
    printf("0x%p\n", p);
    testArray[i] = p;
  }
  for (int i = 0; i < size; ++i)
  {
    aligned_free(testArray[i]);
  }
}

int main()
{
  TestAlignedMalloc();
  return 0;
}

看看结果，

如何在C++项目中实现一个aligned_malloc方法

分配的内存地址都是以64-byte为边界，并且分配的内存最后也被成功释放了，函数是正确的。

源代码说明

本小段主要向不大了解解决思路的小伙伴做一些简单解释，程序大佬可以一笑而过哈。

首先我们要明确我们的解决方案，既然malloc分配的指针地址不能达到我们想要的字节对齐效果，我们就自己来调整这个指针。所以我们的做法是

  • 比用户实际需要的多分配一些内存，多分配的部分等于对齐大小减一再加上指针大小。加上对齐大小减一很好理解，是为了之后的对齐做准备，而加上指针大小是为了之后有空间保存原始指针，对应分配函数中的前2行

  • 在malloc返回的原始指针的基础上，加上指针大小，再对齐（采用的方法就是加上对齐大小减1再做位运算），这个运算结果就是我们想要的对齐后的指针，也是我们返回给用户的指针，对应分配函数中的3~5行

  • 我们还需要保存malloc返回的原始指针，否则free的时候会出问题。这时我们之前多分配的一个指针大小就有用武之地了，保存原始指针在那个地址，分配函数的最后几行就在做这个事

  • 当free的时候，我们知道原始指针存放在我们使用的指针的前一个指针大小偏移的内存里面，通过一些运算取得这个内存地址，再根据里面存放的原始指针调用free完成内存释放

关于 "怎么在C++项目中实现一个aligned_malloc方法" 就介绍到这。希望大家多多支持编程宝库。

如何解决C语言中for循环问题：本文主要介绍"怎么解决C语言中for循环问题"，希望能够解决您遇到有关问题，下面我们一起来看这篇 "怎么解决C语言中for循环问题" 文章。今天分享一下C语言中的for循环中我们常常忽略的小问题。举一个小例子 ...