原文:http://www.openedv.com/posts/list/17718.htm
#ifndef __MALLOC_H
#define __MALLOC_H
#include "sys.h"
//
//本程序只供学习使用,未经作者许可,不得用于其它任何用途
//ALIENTEK 开发板
//内存管理 代码
//正点原子@ALIENTEK
//技术论坛:www.openedv.com
//创建日期:2011/7/5
//版本:V2.0
//版权所有,盗版必究。
//Copyright(C) 正点原子 2009-2019
//All rights reserved
//********************************************************************************
//V2.0 20111008
//1,大幅修改接口函数,增加多块内存的管理.
//
//typedef unsigned long u32;
//typedef unsigned short u16;
//typedef unsigned char u8;
#ifndef NULL
#define NULL 0
#endif
//mem内存参数设定.mem完全处于内部SRAM里面
#define MEM_BLOCK_SIZE 32 //内存块大小为32字节
#define MEM_MAX_SIZE 8*1024 //最大管理内存 2K;;STM32RBT6自带20KSRAM;
#define MEM_ALLOC_TABLE_SIZE MEM_MAX_SIZE/MEM_BLOCK_SIZE //内存表大小;384
//内存管理控制器
struct _m_mallco_dev
{
void (*init)(void); //初始化
u8 (*perused)(void); //内存使用率
u8 *membasex; //内存池,管理2个区域的内存
u16 *memmapx; //内存管理状态表
u8 memrdyx; //内存管理是否就绪
};
extern struct _m_mallco_dev mallco_dev; //在mallco.c里面定义 ;;这里进行外部声明;
void mymemset(void *s,u8 c,u32 count); //设置内存
void mymemcpy(void *des,void *src,u32 n);//复制内存
void mem_init(void); //内存管理初始化函数(外/内部调用)
u32 mem_malloc(u32 size); //内存分配(内部调用)
u8 mem_free(u32 offset); //内存释放(内部调用)
u8 mem_perused(void); //获得内存使用率(外/内部调用)
//用户调用函数
void myfree(void *ptr); //内存释放(外部调用)
void *mymalloc(u32 size); //内存分配(外部调用)
void *myrealloc(void *ptr,u32 size); //重新分配内存(外部调用)
#endif
================================================================
、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、、
//
=====================================================================
#include "malloc.h"
//
//本程序只供学习使用,未经作者许可,不得用于其它任何用途
//ALIENTEK 开发板
//内存管理 代码
//正点原子@ALIENTEK
//技术论坛:www.openedv.com
//创建日期:2011/7/5
//版本:V2.0
//版权所有,盗版必究。
//Copyright(C) 正点原子 2009-2019
//All rights reserved
//********************************************************************************
//V2.0 20111008
//1,大幅修改接口函数,增加多块内存的管理.
//
//内存池4字节对齐;__align(4)定义内存池为 4 字节对齐,这个非常重要!如果不加这个
//限制,在某些情况下[比如分配内存给结构体指针],可能出现错误,所以一定要加上这个;
__align(4) u8 membase[MEM_MAX_SIZE]; //内部SRAM内存池
//内存管理表
u16 memmapbase[MEM_ALLOC_TABLE_SIZE]; //内部SRAM内存池MAP
//内存管理参数
const u32 memtblsize=MEM_ALLOC_TABLE_SIZE; //内存表大小
const u32 memblksize=MEM_BLOCK_SIZE; //内存分块大小
const u32 memsize=MEM_MAX_SIZE; //内存池总大小
//内存管理控制器
struct _m_mallco_dev mallco_dev=
{
mem_init, //内存初始化
mem_perused, //内存使用率
membase, //内存池
memmapbase, //内存管理状态表
0, //内存管理未就绪
};
//复制内存
//*des:目的地址
//*src:源地址
//n:需要复制的内存长度(字节为单位)
void mymemcpy(void *des,void *src,u32 n)
{
u8 *xdes=des;
u8 *xsrc=src;
while(n--)*xdes++=*xsrc++;
}
//设置内存
//*s:内存首地址
//c :要设置的值
//count:需要设置的内存大小(字节为单位)
void mymemset(void *s,u8 c,u32 count)
{
u8 *xs = s;
while(count--)*xs++=c;
}
//内存管理初始化
//memx:所属内存块
void mem_init()
{
mymemset(mallco_dev.memmapx , 0,memtblsize*2); //内存状态表数据清零
mymemset(mallco_dev.membasex, 0,memsize); //内存池所有数据清零
mallco_dev.memrdyx=1; //内存管理初始化OK
}
//获取内存使用率
//返回值:使用率(0~100)
u8 mem_perused(void)
{
u32 used=0;
u32 i;
for(i=0;i<memtblsize;i++)
{
if(mallco_dev.memmapx[i])used++;
}
return (used*100)/(memtblsize);
}
//内存分配(内部调用)
//size:要分配的内存大小(字节)
//返回值:0XFFFFFFFF,代表错误;其他,内存偏移地址
u32 mem_malloc(u32 size)
{
signed long offset=0;
u16 nmemb; //需要的内存块数
u16 cmemb=0;//连续空内存块数
u32 i;
if(!mallco_dev.memrdyx)mallco_dev.init();//未初始化,先执行初始化
if(size==0)return 0XFFFFFFFF; //不需要分配
nmemb=size/memblksize; //获取需要分配的连续内存块数
if(size%memblksize)nmemb++;
for(offset=memtblsize-1;offset>=0;offset--) //搜索整个内存控制区
{
if(!mallco_dev.memmapx[offset])cmemb++; //连续空内存块数增加
else cmemb=0; //连续内存块清零
if(cmemb==nmemb) //找到了连续nmemb个空内存块
{
for(i=0;i<nmemb;i++) //标注内存块非空
{
mallco_dev.memmapx[offset+i]=nmemb;
}
return (offset*memblksize); //内存块的大小*偏移量=返回偏移地址
}
}
return 0XFFFFFFFF;//未找到符合分配条件的内存块
}
//释放内存(内部调用)
//offset:内存地址偏移
//返回值:0,释放成功;1,释放失败;
u8 mem_free(u32 offset)
{
int i;
if(!mallco_dev.memrdyx)//未初始化,先执行初始化
{
mallco_dev.init();
return 1;//未初始化
}
if(offset<memsize) //偏移必须发生在内存池内.
{
int index=offset/memblksize; //偏移所在内存块号码
int nmemb=mallco_dev.memmapx[index]; //内存块数量
for(i=0;i<nmemb;i++) //内存块清零
{
mallco_dev.memmapx[index+i]=0;
}
return 0;
}else return 2;//偏移超区了.
}
//释放内存(外部调用)
//ptr:内存首地址
void myfree(void *ptr)
{
u32 offset;
if(ptr==NULL)return; //地址为0.
offset=(u32)ptr-(u32)mallco_dev.membasex;
mem_free(offset); //释放内存
}
//分配内存(外部调用)
//size:内存大小(字节)
//返回值:分配到的内存首地址.
void *mymalloc(u32 size)
{
u32 offset;
offset=mem_malloc(size);
if(offset==0XFFFFFFFF)return NULL;
else return (void*)((u32)mallco_dev.membasex+offset);
}
//重新分配内存(外部调用)
//*ptr:旧内存首地址
//size:要分配的内存大小(字节)
//返回值:新分配到的内存首地址.
void *myrealloc(void *ptr,u32 size)
{
u32 offset;
offset=mem_malloc(size);
if(offset==0XFFFFFFFF)return NULL;
else
{
mymemcpy((void*)((u32)mallco_dev.membasex+offset),ptr,size); //拷贝旧内存内容到新内存
myfree(ptr); //释放旧内存
return (void*)((u32)mallco_dev.membasex+offset); //返回新内存首地址
}
}