有這么一道題目,要求按任意字節對齊分配內存,接口:
char * aligned_malloc(int size, int alignment)//size 為分配的內存大小,alignment對齊基數(可以為任意數)
這個在gcc庫函數里能找到源碼,在ffdshow等開源代碼里面也能找到源碼,實現如下:
void * aligned_malloc (int size, int alignment)
{
? void* ptr = malloc(size + alignment);
? if (ptr)
??? {
????? void* aligned = (void*)(((long)ptr + alignment) & ~(alignment - 1));
????? ((void**)aligned)[-1] = ptr;
????? return aligned;
??? }
? else
??? return NULL;
}
void *aligned_free(void? *paligned)
{
delete [ ]paligned;
}
解這道題目關鍵是和釋放聯系起來,利用分配的內存首地址的前一地址記錄內存真正分配的起始位置,想不到這點就玩完了。
面試的時候經常會考這類題目,大家可以多看看linux下面庫函數的源碼,對自己本身就是一種提高。