基于固定大小的简单内存池实现

基于固定大小的简单内存池实现

#include <stddef.h>
#include "memory_poll.h"


#define BLOCK_SIZE 64  // 每个块的大小
#define POOL_SIZE 16   // 总块数
#define HEAP_SIZE (BLOCK_SIZE * POOL_SIZE)

static char memory_pool[HEAP_SIZE];  // 内存池
static char* free_list[POOL_SIZE];   // 空闲链表

// 初始化内存池
void init_pool() {
    for (int i = 0; i < POOL_SIZE; i++) {
        free_list[i] = &memory_pool[i * BLOCK_SIZE];
    }
}

// 分配内存块
void* pool_malloc() {
    for (int i = 0; i < POOL_SIZE; i++) {
        if (free_list[i] != NULL) {
            char* block = free_list[i];
            free_list[i] = NULL;  // 将块从空闲列表中移除
            return block;
        }
    }
    return NULL;  // 没有空闲块
}

// 释放内存块
void pool_free(void* ptr) {
    for (int i = 0; i < POOL_SIZE; i++) {
        if (free_list[i] == NULL) {
            free_list[i] = ptr;  // 将块放回空闲列表
            break;
        }
    }
}