ncnn fastMalloc函数内存对齐式分配代码浅析

工作需要，简单熟悉下ncnn（腾讯的一个神经网络前向计算框架）相关的源码。看到有关内存对齐的一段代码，忍不住分析一番。

涉及到C语言的指针、内存，总会让人有些头大，但我们又不得不承认，它也确是非常有趣。

nccc git地址
https://github.com/Tencent/ncnn
简单Demo
将 allocator.h 中的 alignPtr 和 fastMalloc 两个函数提取出来，做一个简单的调用Demo，具体如下。

Demo & 问题

如果对上述问题感兴趣，你可以继续往下看了：）

Step n 和第2部分代码中的是一一对应的哦！

unsigned char* udata = (unsigned char*)malloc(size + sizeof(void*) + MALLOC_ALIGN);

内存分配

问题1：为什么要多分配 sizeof(void *) 大小的内存?（图中8B部分）
因为设计者期望保存对其前的内存指针。
问题2：为什么要多分配 MALLOC_ALIGN 大小的内存？（图中16B部分）
因为对齐单位是16(MALLOC_ALIGN)，而对其后，数据指针不一定指在分配的内存的起始位置，为了保证对其后仍然有100B目标内存可用，要+16。

unsigned char** adata = alignPtr((unsigned char**)udata + 1, MALLOC_ALIGN);

关于udata+1

问题3：为什么要将udata转换成(unsigned char **)？
1）现在udata是char *类型，假设udata指向的内存是0x0000000102803030，那么udata+1 = 0x0000000102803031, 而(unsigned char**)udata + 1 = 0x0000000102803038，我们期望在内存地址的开始初留出一个存放指针的空间(8B)，所以选择后者。
2）我们要使用留出的8B的空间存储对齐前的内存地址，而对齐前的指针的类型（在这个函数体中）是unsigned char *，那么将内存看做一个unsigned char * 的数组的指针，再以此存储我们的对齐前内存地址（unsigned char *类型），多么的优雅，多么的符合概念啊！

template<typename _Tp> static inline _Tp* alignPtr(_Tp* ptr, int n=(int)sizeof(_Tp)) {
    return (_Tp*)(((size_t)ptr + n-1) & -n);
}

实现地址偏移

问题4：_Tp是什么类型？
unsigned char *
问题5：& -n 如何理解？
1）目的：n = 16 (B)，这边的目的是让内存指向的地址为16的整数倍
2）目的达成分析: 16的二进制值是 0001 0000，-16的二进制值是16二进制值的各位取反(...0001 1111)，再加1(...1111 0000)。如果一个地址是64位的，这么操作后就是0xffff ffff ffff fff0，与目标数字按位与后，自然就是一个能被16整除的数字了。
3）举一反三: 不仅16，2^n的数字都可以这么操作哦
问题6：什么是&，什么是按位与
……这个……不解释……

adata[-1] = udata;

设置对齐前内存地址

static inline void fastFree(void* ptr) {
    if (ptr) {
        unsigned char* udata = ((unsigned char**)ptr)[-1];
        free(udata);
    }
}

return adata;

image.png

用户拿到的内存指针指向的是数据的开始位置哦！

完~

ncnn fastMalloc函数 内存对齐式分配 代码浅析