malloc.c 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233
  1. #include "malloc.h"
  2. //内存池(32字节对齐)
  3. __align(32) uint8_t mem1base[MEM1_MAX_SIZE]; //内部SRAM内存池
  4. __align(32) uint8_t mem2base[MEM2_MAX_SIZE] __attribute__((at(0X68000000))); //外部SRAM内存池
  5. //内存管理表
  6. uint16_t mem1mapbase[MEM1_ALLOC_TABLE_SIZE]; //内部SRAM内存池MAP
  7. uint16_t mem2mapbase[MEM2_ALLOC_TABLE_SIZE] __attribute__((at(0X68000000+MEM2_MAX_SIZE))); //外部SRAM内存池MAP
  8. //内存管理参数
  9. const uint32_t memtblsize[SRAMBANK]={MEM1_ALLOC_TABLE_SIZE,MEM2_ALLOC_TABLE_SIZE}; //内存表大小
  10. const uint32_t memblksize[SRAMBANK]={MEM1_BLOCK_SIZE,MEM2_BLOCK_SIZE}; //内存分块大小
  11. const uint32_t memsize[SRAMBANK]={MEM1_MAX_SIZE,MEM2_MAX_SIZE}; //内存总大小
  12. //内存管理控制器
  13. struct _m_mallco_dev mallco_dev=
  14. {
  15. my_mem_init, //内存初始化
  16. my_mem_perused, //内存使用率
  17. mem1base,mem2base, //内存池
  18. mem1mapbase,mem2mapbase, //内存管理状态表
  19. 0,0, //内存管理未就绪
  20. };
  21. //复制内存
  22. //*des:目的地址
  23. //*src:源地址
  24. //n:需要复制的内存长度(字节为单位)
  25. void mymemcpy(void *des,void *src,uint32_t n)
  26. {
  27. uint8_t *xdes=des;
  28. uint8_t *xsrc=src;
  29. while(n--)*xdes++=*xsrc++;
  30. }
  31. //设置内存
  32. //*s:内存首地址
  33. //c :要设置的值
  34. //count:需要设置的内存大小(字节为单位)
  35. void mymemset(void *s,uint8_t c,uint32_t count)
  36. {
  37. uint8_t *xs = s;
  38. while(count--)*xs++=c;
  39. }
  40. //内存管理初始化
  41. //memx:所属内存块
  42. void my_mem_init(uint8_t memx)
  43. {
  44. mymemset(mallco_dev.memmap[memx], 0,memtblsize[memx]*2);//内存状态表数据清零
  45. mymemset(mallco_dev.membase[memx], 0,memsize[memx]); //内存池所有数据清零
  46. mallco_dev.memrdy[memx]=1; //内存管理初始化OK
  47. }
  48. //获取内存使用率
  49. //memx:所属内存块
  50. //返回值:使用率(0~100)
  51. uint8_t my_mem_perused(uint8_t memx)
  52. {
  53. uint32_t used=0;
  54. uint32_t i;
  55. for(i=0;i<memtblsize[memx];i++)
  56. {
  57. if(mallco_dev.memmap[memx][i])used++;
  58. }
  59. return (used*100)/(memtblsize[memx]);
  60. }
  61. //内存分配(内部调用)
  62. //memx:所属内存块
  63. //size:要分配的内存大小(字节)
  64. //返回值:0XFFFFFFFF,代表错误;其他,内存偏移地址
  65. uint32_t my_mem_malloc(uint8_t memx,uint32_t size)
  66. {
  67. signed long offset=0;
  68. uint32_t nmemb; //需要的内存块数
  69. uint32_t cmemb=0;//连续空内存块数
  70. uint32_t i;
  71. if(!mallco_dev.memrdy[memx])mallco_dev.init(memx);//未初始化,先执行初始化
  72. if(size==0)
  73. return 0XFFFFFFFF;//不需要分配
  74. nmemb=size/memblksize[memx]; //获取需要分配的连续内存块数
  75. if(size%memblksize[memx])nmemb++;
  76. for(offset=memtblsize[memx]-1;offset>=0;offset--)//搜索整个内存控制区
  77. {
  78. if(!mallco_dev.memmap[memx][offset])cmemb++;//连续空内存块数增加
  79. else cmemb=0; //连续内存块清零
  80. if(cmemb==nmemb) //找到了连续nmemb个空内存块
  81. {
  82. for(i=0;i<nmemb;i++) //标注内存块非空
  83. {
  84. mallco_dev.memmap[memx][offset+i]=nmemb;
  85. }
  86. return (offset*memblksize[memx]);//返回偏移地址
  87. }
  88. }
  89. return 0XFFFFFFFF;//未找到符合分配条件的内存块
  90. }
  91. //释放内存(内部调用)
  92. //memx:所属内存块
  93. //offset:内存地址偏移
  94. //返回值:0,释放成功;1,释放失败;
  95. uint8_t my_mem_free(uint8_t memx,uint32_t offset)
  96. {
  97. int i;
  98. if(!mallco_dev.memrdy[memx])//未初始化,先执行初始化
  99. {
  100. mallco_dev.init(memx);
  101. return 1;//未初始化
  102. }
  103. if(offset<memsize[memx])//偏移在内存池内.
  104. {
  105. int index=offset/memblksize[memx]; //偏移所在内存块号码
  106. int nmemb=mallco_dev.memmap[memx][index]; //内存块数量
  107. for(i=0;i<nmemb;i++) //内存块清零
  108. {
  109. mallco_dev.memmap[memx][index+i]=0;
  110. }
  111. return 0;
  112. }
  113. else
  114. return 2;//偏移超区了.
  115. }
  116. //释放内存(外部调用)
  117. //memx:所属内存块
  118. //ptr:内存首地址
  119. void myfree(uint8_t memx,void *ptr)
  120. {
  121. uint32_t offset;
  122. if(ptr==NULL)return;//地址为0.
  123. offset=(uint32_t)ptr-(uint32_t)mallco_dev.membase[memx];
  124. my_mem_free(memx,offset); //释放内存
  125. }
  126. //分配内存(外部调用)
  127. //memx:所属内存块
  128. //size:内存大小(字节)
  129. //返回值:分配到的内存首地址.
  130. void *mymalloc(uint8_t memx,uint32_t size)
  131. {
  132. uint32_t offset;
  133. offset=my_mem_malloc(memx,size);
  134. if(offset==0XFFFFFFFF)return NULL;
  135. else return (void*)((uint32_t)mallco_dev.membase[memx]+offset);
  136. }
  137. //重新分配内存(外部调用)
  138. //memx:所属内存块
  139. //*ptr:旧内存首地址
  140. //size:要分配的内存大小(字节)
  141. //返回值:新分配到的内存首地址.
  142. void *myrealloc(uint8_t memx,void *ptr,uint32_t size)
  143. {
  144. uint32_t offset;
  145. offset=my_mem_malloc(memx,size);
  146. if(offset==0XFFFFFFFF)return NULL;
  147. else
  148. {
  149. mymemcpy((void*)((uint32_t)mallco_dev.membase[memx]+offset),ptr,size); //拷贝旧内存内容到新内存
  150. myfree(memx,ptr); //释放旧内存
  151. return (void*)((uint32_t)mallco_dev.membase[memx]+offset); //返回新内存首地址
  152. }
  153. }
  154. #include "string.h"
  155. void *mycalloc(size_t num, size_t size) {
  156. void* ptr = mymalloc(SRAMEX,num * size); // 分配内存
  157. if (ptr == NULL) {
  158. return NULL; // 检查内存分配是否成功
  159. }
  160. memset(ptr, 0, num * size); // 初始化内存为零
  161. return ptr;
  162. }