概述

? ? ? ? 在實際項目中，有些單片機資源緊缺，需要mallloc內(nèi)存，庫又沒有自帶malloc函數(shù)時，此時，就需要手動編寫，在此做個筆錄。（已在項目上使用），還可進入對齊管理機制。

直接上源碼 ^_^

一、示例1：

1、mem_malloc.h文件

#ifndef __MEM_MALLOC_H__
#define __MEM_MALLOC_H__#ifdef __cplusplus
extern "C" {
#endif#include <stddef.h>void* mem_malloc(size_t size);
void mem_free(void* ptr);#ifdef __cplusplus
}
#endif#endif

2、mem_malloc.c文件

#include "mem_malloc.h"
#include "stdint.h"// 定義內(nèi)存塊結(jié)構(gòu)
typedef struct MemoryBlock {size_t size;struct MemoryBlock* next;
} MemoryBlock;// 定義內(nèi)存池起始地址
#define MEMORY_POOL_SIZE 1024
static uint8_t memoryPool[MEMORY_POOL_SIZE];// 全局指針，指向內(nèi)存池起始位置
static MemoryBlock* memoryPoolPtr = NULL;// 初始化內(nèi)存池
void memoryPoolInit(void)
{memoryPoolPtr = (MemoryBlock*)memoryPool;memoryPoolPtr->size = MEMORY_POOL_SIZE - sizeof(MemoryBlock);memoryPoolPtr->next = NULL;
}// 分配內(nèi)存
void* mem_malloc(size_t size)
{//memoryPoolInit();if (size == 0)return NULL;MemoryBlock* currentBlock = memoryPoolPtr;//MemoryBlock* prevBlock = NULL;// 遍歷內(nèi)存池中的內(nèi)存塊，找到合適大小的內(nèi)存塊while (currentBlock != NULL) {if (currentBlock->size >= size + sizeof(MemoryBlock)) {// 如果當(dāng)前內(nèi)存塊大于所需內(nèi)存，分配內(nèi)存if (currentBlock->size >= size + sizeof(MemoryBlock) + sizeof(size_t)) {MemoryBlock* newBlock = (MemoryBlock*)((uint8_t*)currentBlock + sizeof(MemoryBlock) + size);newBlock->size = currentBlock->size - sizeof(MemoryBlock) - size;newBlock->next = currentBlock->next;currentBlock->next = newBlock;currentBlock->size = size;}// 返回分配的內(nèi)存塊的地址return (uint8_t*)currentBlock + sizeof(MemoryBlock);}//prevBlock = currentBlock;currentBlock = currentBlock->next;}// 沒有足夠的內(nèi)存塊可供分配return NULL;
}// 釋放內(nèi)存
void mem_free(void* ptr)
{if (ptr == NULL)return;MemoryBlock* block = (MemoryBlock*)((uint8_t*)ptr - sizeof(MemoryBlock));block->next = memoryPoolPtr;memoryPoolPtr = block;
}// 示例代碼
int main(void) 
{// 初始化內(nèi)存池memoryPoolInit();// 分配內(nèi)存int* intPtr = (int*)mem_malloc(sizeof(int));if (intPtr != NULL) {*intPtr = 42;free(intPtr);}return 0;
}

二、示例2：

#include <stdio.h>#define MEMORY_POOL_SIZE (1024 * 2)unsigned char memoryBuff[MEMORY_POOL_SIZE];typedef struct {unsigned char* start;size_t size;
} MemoryBlock;static MemoryBlock memoryBlocks[MEMORY_POOL_SIZE] = { 0 };
static int numBlocks = 0;// 從內(nèi)存池中分配內(nèi)存
void* mem_malloc(size_t size)
{// 尋找空閑塊for (int i = 0; i < numBlocks; i++) {if (memoryBlocks[i].size == 0 && size <= MEMORY_POOL_SIZE) {// 找到合適大小的內(nèi)存塊，返回內(nèi)存塊地址memoryBlocks[i].start = memoryBuff;memoryBlocks[i].size = size;return memoryBlocks[i].start;}}// 分配新的塊if (numBlocks < MEMORY_POOL_SIZE) {memoryBlocks[numBlocks].start = memoryBuff + numBlocks;memoryBlocks[numBlocks].size = size;numBlocks++;return memoryBlocks[numBlocks - 1].start;}// 分配失敗return NULL;
}void mem_free(void* ptr)
{// 查找要釋放的塊for (int i = 0; i < numBlocks; i++) {if (memoryBlocks[i].start == ptr) {memoryBlocks[i].size = 0;break;}}
}int main(void)
{// 使用mem_malloc和mem_free進行內(nèi)存管理unsigned char* ptr1 = (unsigned char*)mem_malloc(50);unsigned char* ptr2 = (unsigned char*)mem_malloc(100);if (ptr1 != NULL && ptr2 != NULL) {// 使用分配的內(nèi)存for (int i = 0; i < 50; i++) {ptr1[i] = i;}for (int i = 0; i < 100; i++) {ptr2[i] = i + 50;}// 打印分配的內(nèi)存printf("ptr1: ");for (int i = 0; i < 50; i++) {printf("%d ", ptr1[i]);}printf("\n");printf("ptr2: ");for (int i = 0; i < 100; i++) {printf("%d ", ptr2[i]);}printf("\n");}mem_free(ptr1);mem_free(ptr2);return 0;
}

三、運行結(jié)果

四、總結(jié)

? ? ? ? 希望能幫助到你。