内存池的C实现

最新推荐文章于 2024-02-03 15:47:57 发布

DingUXiu

最新推荐文章于 2024-02-03 15:47:57 发布

阅读量321

点赞数

文章标签： c语言开发语言

本文链接：https://blog.csdn.net/u010650845/article/details/126345606

版权

C/C++进阶指南专栏收录该内容

20 篇文章 3 订阅

订阅专栏

mempool.h

#ifndef _MEMPOOL_H
#define _MEMPOOL_H

#include "sys.h"
#include "FreeRTOS.h"
#include "task.h"
#include "string.h"
#include "delay.h"

/* 将内存设置为0 */
#define memset0(mem, size)  memset((mem), 0, (size))

/* 一定要保证申请内存成功! */
#define my_alloc_msg(_type) _alloc_msg(sizeof(_type))
#define my_free_msg(_msg) do {if(_msg==NULL) break;_free_msg(_msg); _msg = 0;}while(0);

void msg_pool_init(void);

#endif

mempool.c

#include "mempool.h"

//>全部消息池：msg_pool[MAX_MSG][MAX_MSG_LEN]
#define     MAX_MSG_2       32
#define     MAX_MSG_4       64
#define     MAX_MSG_8       128
#define     MAX_MSG_32      128
#define     MAX_MSG_64      64
#define     MAX_MSG_128     8
#define     MAX_MSG_256     4

uint8_t     msg_pool_2[MAX_MSG_2][2];
uint8_t     msg_pool_4[MAX_MSG_4][4];
uint8_t     msg_pool_8[MAX_MSG_8][8];
uint8_t     msg_pool_32[MAX_MSG_32][32];
uint8_t     msg_pool_64[MAX_MSG_64][64];
uint8_t     msg_pool_128[MAX_MSG_128][128];
uint8_t     msg_pool_256[MAX_MSG_256][256];

void msg_pool_init(void){
    memset0(msg_pool_2,   MAX_MSG_2*2);
    memset0(msg_pool_4,   MAX_MSG_4*4);
    memset0(msg_pool_8,   MAX_MSG_8*8);
    memset0(msg_pool_32,  MAX_MSG_32*32);
    memset0(msg_pool_64,  MAX_MSG_64*64);
    memset0(msg_pool_128, MAX_MSG_128*128);
    memset0(msg_pool_256, MAX_MSG_256*256);
}

//移除消息
#define LOOP_POOL_REMOVE(MAX_MSG_x,msg_pool_x,tick_1s) \
for( i=0; i<MAX_MSG_x; ++i) {\
    if(msg_pool_x[i][0] > 0 && tick_1s-msg_pool_x[i][1] > 15) {\
        msg_pool_x[i][0] = 0;\
    }\
}

//遍历消息池查询是否存在可用空间
#define LOOP_POOL(MAX_MSG_x,msg_pool_x) \
for( i=0; i<MAX_MSG_x; ++i) {\
    if(msg_pool_x[i][0] == 0) {\
        ret = (uint8_t*)(&msg_pool_x[i]);\
        goto ok_end;\
    }\
}
//进入临界区
#define OS_ENTER_CRITICAL() \
if(0==__get_CONTROL()){ \
	state_value = taskENTER_CRITICAL_FROM_ISR(); \
}else{ \
		taskENTER_CRITICAL();\
}
//退出临界区
#define OS_EXIT_CRITICAL() \
if(0==__get_CONTROL()){\
	taskEXIT_CRITICAL_FROM_ISR(state_value);\
}else{\
	taskEXIT_CRITICAL();\
}

//申请消息，并清0
void* _alloc_msg(uint16_t size){
    uint32_t i = 0;
	uint32_t state_value = 0;
    uint8_t* ret = NULL;

//_again:
	
    OS_ENTER_CRITICAL();

    if(size <=2 ){
        LOOP_POOL(MAX_MSG_2, msg_pool_2);
        LOOP_POOL(MAX_MSG_4, msg_pool_4);
        LOOP_POOL(MAX_MSG_8, msg_pool_8);
        LOOP_POOL(MAX_MSG_32, msg_pool_32);
        LOOP_POOL(MAX_MSG_64, msg_pool_64);
        LOOP_POOL(MAX_MSG_128, msg_pool_128);
        LOOP_POOL(MAX_MSG_256, msg_pool_256);
    }
    else if(size <=4 ){
        LOOP_POOL(MAX_MSG_4, msg_pool_4);
        LOOP_POOL(MAX_MSG_8, msg_pool_8);
        LOOP_POOL(MAX_MSG_32, msg_pool_32);
        LOOP_POOL(MAX_MSG_64, msg_pool_64);
        LOOP_POOL(MAX_MSG_128, msg_pool_128);
        LOOP_POOL(MAX_MSG_256, msg_pool_256);
    }
    else if(size <= 8){
        LOOP_POOL(MAX_MSG_8, msg_pool_8);
        LOOP_POOL(MAX_MSG_32, msg_pool_32);
        LOOP_POOL(MAX_MSG_64, msg_pool_64);
        LOOP_POOL(MAX_MSG_128, msg_pool_128);
        LOOP_POOL(MAX_MSG_256, msg_pool_256);
    }
    else if(size <= 32){
        LOOP_POOL(MAX_MSG_32, msg_pool_32);
        LOOP_POOL(MAX_MSG_64, msg_pool_64);
        LOOP_POOL(MAX_MSG_128, msg_pool_128);
        LOOP_POOL(MAX_MSG_256, msg_pool_256);
    }
    else if(size <= 64){
        LOOP_POOL(MAX_MSG_64, msg_pool_64);
        LOOP_POOL(MAX_MSG_128, msg_pool_128);
        LOOP_POOL(MAX_MSG_256, msg_pool_256);
    }
    else if(size <= 128){
        LOOP_POOL(MAX_MSG_128, msg_pool_128);
        LOOP_POOL(MAX_MSG_256, msg_pool_256);
    }
    else if(size <= 256){
        LOOP_POOL(MAX_MSG_256, msg_pool_256);
    }
    else{
        // 不应该出现大于256的消息
        ret = NULL;
    }
ok_end:
    if(ret != NULL) {
        ret[0] = 0xff;
        ret[1] = systemMs();
        OS_EXIT_CRITICAL();
        memset0(ret+2, size-2);
    } else {
        OS_EXIT_CRITICAL();
        //goto _again;  
    }
    return ret;
}

void _free_msg(void* msg){

    if(msg != NULL){
        uint8_t* p = msg;
        uint32_t state_value = 0;
        OS_ENTER_CRITICAL();
        p[0] = 0;
        p[1] = 0;
        OS_EXIT_CRITICAL();
    }
}