slab.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670
  1. #include "slab.h"
  2. /**
  3. * @brief 创建一个内存池
  4. *
  5. * @param size 内存池容量大小
  6. * @param constructor 构造函数
  7. * @param destructor 析构函数
  8. * @param arg 参数
  9. * @return struct slab* 构建好的内存池对象
  10. */
  11. struct slab *slab_create(ul size, void *(*constructor)(void *vaddr, ul arg), void *(*destructor)(void *vaddr, ul arg), ul arg)
  12. {
  13. struct slab *slab_pool = (struct slab *)kmalloc(sizeof(struct slab), 0);
  14. // BUG
  15. if (slab_pool == NULL)
  16. {
  17. kBUG("slab_create()->kmalloc()->slab == NULL");
  18. return NULL;
  19. }
  20. memset(slab_pool, 0, sizeof(struct slab));
  21. slab_pool->size = SIZEOF_LONG_ALIGN(size);
  22. slab_pool->count_total_using = 0;
  23. slab_pool->count_total_free = 0;
  24. // 直接分配cache_pool_entry结构体,避免每次访问都要检测是否为NULL,提升效率
  25. slab_pool->cache_pool_entry = (struct slab_obj *)kmalloc(sizeof(struct slab_obj), 0);
  26. // BUG
  27. if (slab_pool->cache_pool_entry == NULL)
  28. {
  29. kBUG("slab_create()->kmalloc()->slab->cache_pool_entry == NULL");
  30. kfree(slab_pool);
  31. return NULL;
  32. }
  33. memset(slab_pool->cache_pool_entry, 0, sizeof(struct slab_obj));
  34. // dma内存池设置为空
  35. slab_pool->cache_dma_pool_entry = NULL;
  36. // 设置构造及析构函数
  37. slab_pool->constructor = constructor;
  38. slab_pool->destructor = destructor;
  39. list_init(&slab_pool->cache_pool_entry->list);
  40. // 分配属于内存池的内存页
  41. slab_pool->cache_pool_entry->page = alloc_pages(ZONE_NORMAL, 1, PAGE_KERNEL);
  42. // BUG
  43. if (slab_pool->cache_pool_entry->page == NULL)
  44. {
  45. kBUG("slab_create()->kmalloc()->slab->cache_pool_entry == NULL");
  46. kfree(slab_pool->cache_pool_entry);
  47. kfree(slab_pool);
  48. return NULL;
  49. }
  50. // page_init(slab_pool->cache_pool_entry->page, PAGE_KERNEL);
  51. slab_pool->cache_pool_entry->count_using = 0;
  52. slab_pool->cache_pool_entry->count_free = PAGE_2M_SIZE / slab_pool->size;
  53. slab_pool->count_total_free = slab_pool->cache_pool_entry->count_free;
  54. slab_pool->cache_pool_entry->vaddr = phys_2_virt(slab_pool->cache_pool_entry->page->addr_phys);
  55. // bitmap有多少有效位
  56. slab_pool->cache_pool_entry->bmp_count = slab_pool->cache_pool_entry->count_free;
  57. // 计算位图所占的空间 占用多少byte(按unsigned long大小的上边缘对齐)
  58. slab_pool->cache_pool_entry->bmp_len = ((slab_pool->cache_pool_entry->bmp_count + sizeof(ul) * 8 - 1) >> 6) << 3;
  59. // 初始化位图
  60. slab_pool->cache_pool_entry->bmp = (ul *)kmalloc(slab_pool->cache_pool_entry->bmp_len, 0);
  61. // BUG
  62. if (slab_pool->cache_pool_entry->bmp == NULL)
  63. {
  64. kBUG("slab_create()->kmalloc()->slab->cache_pool_entry == NULL");
  65. free_pages(slab_pool->cache_pool_entry->page, 1);
  66. kfree(slab_pool->cache_pool_entry);
  67. kfree(slab_pool);
  68. return NULL;
  69. }
  70. // 将位图清空
  71. memset(slab_pool->cache_pool_entry->bmp, 0, slab_pool->cache_pool_entry->bmp_len);
  72. return slab_pool;
  73. }
  74. /**
  75. * @brief 销毁内存池
  76. * 只有当slab是空的时候才能销毁
  77. * @param slab_pool 要销毁的内存池
  78. * @return ul
  79. *
  80. */
  81. ul slab_destroy(struct slab *slab_pool)
  82. {
  83. struct slab_obj *slab_obj_ptr = slab_pool->cache_pool_entry;
  84. if (slab_pool->count_total_using)
  85. {
  86. kBUG("slab_cache->count_total_using != 0");
  87. return ESLAB_NOTNULL;
  88. }
  89. struct slab_obj *tmp_slab_obj = NULL;
  90. while (!list_empty(&slab_obj_ptr->list))
  91. {
  92. tmp_slab_obj = slab_obj_ptr;
  93. // 获取下一个slab_obj的起始地址
  94. slab_obj_ptr = container_of(list_next(&slab_obj_ptr->list), struct slab_obj, list);
  95. list_del(&tmp_slab_obj->list);
  96. kfree(tmp_slab_obj->bmp);
  97. page_clean(tmp_slab_obj->page);
  98. free_pages(tmp_slab_obj->page, 1);
  99. kfree(tmp_slab_obj);
  100. }
  101. kfree(slab_obj_ptr->bmp);
  102. page_clean(slab_obj_ptr->page);
  103. free_pages(slab_obj_ptr->page, 1);
  104. kfree(slab_obj_ptr);
  105. kfree(slab_pool);
  106. return 0;
  107. }
  108. /**
  109. * @brief 分配SLAB内存池中的内存对象
  110. *
  111. * @param slab_pool slab内存池
  112. * @param arg 传递给内存对象构造函数的参数
  113. * @return void* 内存空间的虚拟地址
  114. */
  115. void *slab_malloc(struct slab *slab_pool, ul arg)
  116. {
  117. struct slab_obj *slab_obj_ptr = slab_pool->cache_pool_entry;
  118. struct slab_obj *tmp_slab_obj = NULL;
  119. // slab内存池中已经没有空闲的内存对象,进行扩容
  120. if (slab_pool->count_total_free == 0)
  121. {
  122. tmp_slab_obj = (struct slab_obj *)kmalloc(sizeof(struct slab_obj), 0);
  123. // BUG
  124. if (tmp_slab_obj == NULL)
  125. {
  126. kBUG("slab_malloc()->kmalloc()->slab->tmp_slab_obj == NULL");
  127. return NULL;
  128. }
  129. memset(tmp_slab_obj, 0, sizeof(struct slab_obj));
  130. list_init(&tmp_slab_obj->list);
  131. tmp_slab_obj->page = alloc_pages(ZONE_NORMAL, 1, PAGE_KERNEL);
  132. // BUG
  133. if (tmp_slab_obj->page == NULL)
  134. {
  135. kBUG("slab_malloc()->kmalloc()=>tmp_slab_obj->page == NULL");
  136. kfree(tmp_slab_obj);
  137. return NULL;
  138. }
  139. tmp_slab_obj->count_using = 0;
  140. tmp_slab_obj->count_free = PAGE_2M_SIZE / slab_pool->size;
  141. tmp_slab_obj->vaddr = phys_2_virt(tmp_slab_obj->page->addr_phys);
  142. tmp_slab_obj->bmp_count = tmp_slab_obj->count_free;
  143. // 计算位图所占的空间 占用多少byte(按unsigned long大小的上边缘对齐)
  144. tmp_slab_obj->bmp_len = ((tmp_slab_obj->bmp_count + sizeof(ul) * 8 - 1) >> 6) << 3;
  145. tmp_slab_obj->bmp = (ul *)kmalloc(tmp_slab_obj->bmp_len, 0);
  146. // BUG
  147. if (tmp_slab_obj->bmp == NULL)
  148. {
  149. kBUG("slab_malloc()->kmalloc()=>tmp_slab_obj->bmp == NULL");
  150. free_pages(tmp_slab_obj->page, 1);
  151. kfree(tmp_slab_obj);
  152. return NULL;
  153. }
  154. memset(tmp_slab_obj->bmp, 0, tmp_slab_obj->bmp_len);
  155. list_add(&slab_pool->cache_pool_entry->list, &tmp_slab_obj->list);
  156. slab_pool->count_total_free += tmp_slab_obj->count_free;
  157. slab_obj_ptr = tmp_slab_obj;
  158. }
  159. // 扩容完毕或无需扩容,开始分配内存对象
  160. int tmp_md;
  161. do
  162. {
  163. if (slab_obj_ptr->count_free == 0)
  164. {
  165. slab_obj_ptr = container_of(list_next(&slab_obj_ptr->list), struct slab_obj, list);
  166. continue;
  167. }
  168. for (int i = 0; i < slab_obj_ptr->bmp_count; ++i)
  169. {
  170. // 当前bmp对应的内存对象都已经被分配
  171. if (*(slab_obj_ptr->bmp + (i >> 6)) == 0xffffffffffffffffUL)
  172. {
  173. i += 63;
  174. continue;
  175. }
  176. // 第i个内存对象是空闲的
  177. tmp_md = i % 64;
  178. if ((*(slab_obj_ptr->bmp + (i >> 6)) & (1UL << tmp_md)) == 0)
  179. {
  180. // 置位bmp
  181. *(slab_obj_ptr->bmp + (i >> 6)) |= (1UL << tmp_md);
  182. // 更新当前slab对象的计数器
  183. ++(slab_obj_ptr->count_using);
  184. --(slab_obj_ptr->count_free);
  185. // 更新slab内存池的计数器
  186. ++(slab_pool->count_total_using);
  187. --(slab_pool->count_total_free);
  188. if (slab_pool->constructor != NULL)
  189. {
  190. // 返回内存对象指针(要求构造函数返回内存对象指针)
  191. return slab_pool->constructor((char *)slab_obj_ptr->vaddr + slab_pool->size * i, arg);
  192. }
  193. // 返回内存对象指针
  194. else
  195. return (void *)((char *)slab_obj_ptr->vaddr + slab_pool->size * i);
  196. }
  197. }
  198. } while (slab_obj_ptr != slab_pool->cache_pool_entry);
  199. // should not be here
  200. kBUG("slab_malloc() ERROR: can't malloc");
  201. // 释放内存
  202. if (tmp_slab_obj != NULL)
  203. {
  204. list_del(&tmp_slab_obj->list);
  205. kfree(tmp_slab_obj->bmp);
  206. page_clean(tmp_slab_obj->page);
  207. free_pages(tmp_slab_obj->page, 1);
  208. kfree(tmp_slab_obj);
  209. }
  210. return NULL;
  211. }
  212. /**
  213. * @brief 回收slab内存池中的对象
  214. *
  215. * @param slab_pool 对应的内存池
  216. * @param addr 内存对象的虚拟地址
  217. * @param arg 传递给虚构函数的参数
  218. * @return ul
  219. */
  220. ul slab_free(struct slab *slab_pool, void *addr, ul arg)
  221. {
  222. struct slab_obj *slab_obj_ptr = slab_pool->cache_pool_entry;
  223. do
  224. {
  225. // 虚拟地址不在当前内存池对象的管理范围内
  226. if (!(slab_obj_ptr->vaddr <= addr && addr <= (slab_obj_ptr->vaddr + PAGE_2M_SIZE)))
  227. {
  228. slab_obj_ptr = container_of(list_next(&slab_obj_ptr->list), struct slab_obj, list);
  229. }
  230. else
  231. {
  232. // 计算出给定内存对象是第几个
  233. int index = (addr - slab_obj_ptr->vaddr) / slab_pool->size;
  234. // 复位位图中对应的位
  235. *(slab_obj_ptr->bmp + (index >> 6)) ^= (1UL << index % 64);
  236. ++(slab_obj_ptr->count_free);
  237. --(slab_obj_ptr->count_using);
  238. ++(slab_pool->count_total_free);
  239. --(slab_pool->count_total_using);
  240. // 有对应的析构函数,调用析构函数
  241. if (slab_pool->destructor != NULL)
  242. slab_pool->destructor((char *)slab_obj_ptr->vaddr + slab_pool->size * index, arg);
  243. // 当前内存对象池的正在使用的内存对象为0,且内存池的空闲对象大于当前对象池的2倍,则销毁当前对象池,以减轻系统内存压力
  244. if ((slab_obj_ptr->count_using == 0) && ((slab_pool->count_total_free >> 1) >= slab_obj_ptr->count_free) && (slab_obj_ptr != slab_pool->cache_pool_entry))
  245. {
  246. list_del(&slab_obj_ptr->list);
  247. slab_pool->count_total_free -= slab_obj_ptr->count_free;
  248. kfree(slab_obj_ptr->bmp);
  249. page_clean(slab_obj_ptr->page);
  250. free_pages(slab_obj_ptr->page, 1);
  251. kfree(slab_obj_ptr);
  252. }
  253. }
  254. return 0;
  255. } while (slab_obj_ptr != slab_pool->cache_pool_entry);
  256. kwarn("slab_free(): address not in current slab");
  257. return ENOT_IN_SLAB;
  258. }
  259. /**
  260. * @brief 初始化内存池组
  261. * 在初始化通用内存管理单元期间,尚无内存空间分配函数,需要我们手动为SLAB内存池指定存储空间
  262. * @return ul
  263. */
  264. ul slab_init()
  265. {
  266. kinfo("Initializing SLAB...");
  267. // 将slab的内存池空间放置在mms的后方
  268. ul tmp_addr = memory_management_struct.end_of_struct;
  269. for (int i = 0; i < 16; ++i)
  270. {
  271. // 将slab内存池对象的空间放置在mms的后面,并且预留4个unsigned long 的空间以防止内存越界
  272. kmalloc_cache_group[i].cache_pool_entry = (struct slab_obj *)memory_management_struct.end_of_struct;
  273. memory_management_struct.end_of_struct += sizeof(struct slab_obj) + (sizeof(ul) << 2);
  274. list_init(&kmalloc_cache_group[i].cache_pool_entry->list);
  275. // 初始化内存池对象
  276. kmalloc_cache_group[i].cache_pool_entry->count_using = 0;
  277. kmalloc_cache_group[i].cache_pool_entry->count_free = PAGE_2M_SIZE / kmalloc_cache_group[i].size;
  278. kmalloc_cache_group[i].cache_pool_entry->bmp_len = (((kmalloc_cache_group[i].cache_pool_entry->count_free + sizeof(ul) * 8 - 1) >> 6) << 3);
  279. kmalloc_cache_group[i].cache_pool_entry->bmp_count = kmalloc_cache_group[i].cache_pool_entry->count_free;
  280. // 在slab对象后方放置bmp
  281. kmalloc_cache_group[i].cache_pool_entry->bmp = (ul *)memory_management_struct.end_of_struct;
  282. // bmp后方预留4个unsigned long的空间防止内存越界,且按照8byte进行对齐
  283. memory_management_struct.end_of_struct = (ul)(memory_management_struct.end_of_struct + kmalloc_cache_group[i].cache_pool_entry->bmp_len + (sizeof(ul) << 2)) & (~(sizeof(ul) - 1));
  284. // @todo:此处可优化,直接把所有位设置为0,然后再对部分不存在对应的内存对象的位设置为1
  285. memset(kmalloc_cache_group[i].cache_pool_entry->bmp, 0xff, kmalloc_cache_group[i].cache_pool_entry->bmp_len);
  286. for (int j = 0; j < kmalloc_cache_group[i].cache_pool_entry->bmp_count; ++j)
  287. *(kmalloc_cache_group[i].cache_pool_entry->bmp + (j >> 6)) ^= 1UL << (j % 64);
  288. kmalloc_cache_group[i].count_total_using = 0;
  289. kmalloc_cache_group[i].count_total_free = kmalloc_cache_group[i].cache_pool_entry->count_free;
  290. }
  291. struct Page *page = NULL;
  292. // 将上面初始化内存池组时,所占用的内存页进行初始化
  293. ul tmp_page_mms_end = virt_2_phys(memory_management_struct.end_of_struct) >> PAGE_2M_SHIFT;
  294. ul page_num = 0;
  295. for (int i = PAGE_2M_ALIGN(virt_2_phys(tmp_addr)) >> PAGE_2M_SHIFT; i <= tmp_page_mms_end; ++i)
  296. {
  297. page = memory_management_struct.pages_struct + i;
  298. page_num = page->addr_phys >> PAGE_2M_SHIFT;
  299. *(memory_management_struct.bmp + (page_num >> 6)) |= (1UL << (page_num % 64));
  300. ++page->zone->count_pages_using;
  301. --page->zone->count_pages_free;
  302. page_init(page, PAGE_KERNEL_INIT | PAGE_KERNEL | PAGE_PGT_MAPPED);
  303. }
  304. kdebug("2.memory_management_struct.bmp:%#018lx\tzone_struct->count_pages_using:%d\tzone_struct->count_pages_free:%d", *memory_management_struct.bmp, memory_management_struct.zones_struct->count_pages_using, memory_management_struct.zones_struct->count_pages_free);
  305. // 为slab内存池对象分配内存空间
  306. ul *virt = NULL;
  307. for (int i = 0; i < 16; ++i)
  308. {
  309. // 获取一个新的空页并添加到空页表,然后返回其虚拟地址
  310. virt = (ul *)((memory_management_struct.end_of_struct + PAGE_2M_SIZE * i + PAGE_2M_SIZE - 1) & PAGE_2M_MASK);
  311. page = Virt_To_2M_Page(virt);
  312. page_num = page->addr_phys >> PAGE_2M_SHIFT;
  313. *(memory_management_struct.bmp + (page_num >> 6)) |= (1UL << (page_num % 64));
  314. ++page->zone->count_pages_using;
  315. --page->zone->count_pages_free;
  316. page_init(page, PAGE_PGT_MAPPED | PAGE_KERNEL | PAGE_KERNEL_INIT);
  317. kmalloc_cache_group[i].cache_pool_entry->page = page;
  318. kmalloc_cache_group[i].cache_pool_entry->vaddr = virt;
  319. }
  320. kdebug("3.memory_management_struct.bmp:%#018lx\tzone_struct->count_pages_using:%d\tzone_struct->count_pages_free:%d", *memory_management_struct.bmp, memory_management_struct.zones_struct->count_pages_using, memory_management_struct.zones_struct->count_pages_free);
  321. kinfo("SLAB initialized successfully!");
  322. return 0;
  323. }
  324. /**
  325. * @brief 在kmalloc中创建slab_obj的函数(与slab_malloc()中的类似)
  326. *
  327. * @param size
  328. * @return struct slab_obj* 创建好的slab_obj
  329. */
  330. struct slab_obj *kmalloc_create_slab_obj(ul size)
  331. {
  332. struct Page *page = alloc_pages(ZONE_NORMAL, 1, 0);
  333. // BUG
  334. if (page == NULL)
  335. {
  336. kBUG("kmalloc_create()->alloc_pages()=>page == NULL");
  337. return NULL;
  338. }
  339. page_init(page, PAGE_KERNEL);
  340. ul *vaddr = NULL;
  341. ul struct_size = 0;
  342. struct slab_obj *slab_obj_ptr;
  343. // 根据size大小,选择不同的分支来处理
  344. // 之所以选择512byte为分界点,是因为,此时bmp大小刚好为512byte。显而易见,选择过小的话会导致kmalloc函数与当前函数反复互相调用,最终导致栈溢出
  345. switch (size)
  346. {
  347. // ============ 对于size<=512byte的内存池对象,将slab_obj结构体和bmp放置在物理页的内部 ========
  348. // 由于这些对象的特征是,bmp占的空间大,而内存块的空间小,这样做的目的是避免再去申请一块内存来存储bmp,减少浪费。
  349. case 32:
  350. case 64:
  351. case 128:
  352. case 256:
  353. case 512:
  354. vaddr = phys_2_virt(page->addr_phys);
  355. // slab_obj结构体的大小 (本身的大小+bmp的大小)
  356. struct_size = sizeof(struct slab_obj) + PAGE_2M_SIZE / size / 8;
  357. // 将slab_obj放置到物理页的末尾
  358. slab_obj_ptr = (struct slab_obj *)((unsigned char *)vaddr + PAGE_2M_SIZE - struct_size);
  359. slab_obj_ptr->bmp = (ul *)slab_obj_ptr + sizeof(struct slab_obj);
  360. slab_obj_ptr->count_free = (PAGE_2M_SIZE - struct_size) / size;
  361. slab_obj_ptr->count_using = 0;
  362. slab_obj_ptr->bmp_count = slab_obj_ptr->count_free;
  363. slab_obj_ptr->vaddr = vaddr;
  364. slab_obj_ptr->page = page;
  365. list_init(&slab_obj_ptr->list);
  366. slab_obj_ptr->bmp_len = ((slab_obj_ptr->bmp_count + sizeof(ul) * 8 - 1) >> 6) << 3;
  367. // @todo:此处可优化,直接把所有位设置为0,然后再对部分不存在对应的内存对象的位设置为1
  368. memset(slab_obj_ptr->bmp, 0xff, slab_obj_ptr->bmp_len);
  369. for (int i = 0; i < slab_obj_ptr->bmp_count; ++i)
  370. *(slab_obj_ptr->bmp + (i >> 6)) ^= 1UL << (i % 64);
  371. break;
  372. // ================= 较大的size时,slab_obj和bmp不再放置于当前物理页内部 ============
  373. // 因为在这种情况下,bmp很短,继续放置在当前物理页内部则会造成可分配的对象少,加剧了内存空间的浪费
  374. case 1024: // 1KB
  375. case 2048:
  376. case 4096: // 4KB
  377. case 8192:
  378. case 16384:
  379. case 32768:
  380. case 65536:
  381. case 131072: // 128KB
  382. case 262144:
  383. case 524288:
  384. case 1048576: // 1MB
  385. slab_obj_ptr = (struct slab_obj *)kmalloc(sizeof(struct slab_obj), 0);
  386. slab_obj_ptr->count_free = PAGE_2M_SIZE / size;
  387. slab_obj_ptr->count_using = 0;
  388. slab_obj_ptr->bmp_count = slab_obj_ptr->count_free;
  389. slab_obj_ptr->bmp_len = ((slab_obj_ptr->bmp_count + sizeof(ul) * 8 - 1) >> 6) << 3;
  390. slab_obj_ptr->bmp = (ul *)kmalloc(slab_obj_ptr->bmp_len, 0);
  391. // @todo:此处可优化,直接把所有位设置为0,然后再对部分不存在对应的内存对象的位设置为1
  392. memset(slab_obj_ptr->bmp, 0xff, slab_obj_ptr->bmp_len);
  393. for (int i = 0; i < slab_obj_ptr->bmp_count; ++i)
  394. *(slab_obj_ptr->bmp + (i >> 6)) ^= 1UL << (i % 64);
  395. slab_obj_ptr->vaddr = phys_2_virt(page->addr_phys);
  396. slab_obj_ptr->page = page;
  397. list_init(&slab_obj_ptr->list);
  398. break;
  399. // size 错误
  400. default:
  401. kerror("kamlloc_create(): Wrong size%d", size);
  402. free_pages(page, 1);
  403. return NULL;
  404. break;
  405. }
  406. return slab_obj_ptr;
  407. }
  408. /**
  409. * @brief 通用内存分配函数
  410. *
  411. * @param size 要分配的内存大小
  412. * @param flags 内存的flag
  413. * @return void* 内核内存虚拟地址
  414. */
  415. void *kmalloc(unsigned long size, unsigned long flags)
  416. {
  417. if (size > 1048576)
  418. {
  419. kwarn("kmalloc(): Can't alloc such memory: %ld bytes, because it is too large.", size);
  420. return NULL;
  421. }
  422. int index;
  423. for (int i = 0; i < 16; ++i)
  424. if (kmalloc_cache_group[i].size >= size)
  425. {
  426. index = i;
  427. break;
  428. }
  429. struct slab_obj *slab_obj_ptr = kmalloc_cache_group[index].cache_pool_entry;
  430. //kdebug("count_total_free=%d", kmalloc_cache_group[index].count_total_free);
  431. // 内存池没有可用的内存对象,需要进行扩容
  432. if (kmalloc_cache_group[index].count_total_free == 0)
  433. {
  434. // 创建slab_obj
  435. slab_obj_ptr = kmalloc_create_slab_obj(kmalloc_cache_group[index].size);
  436. // BUG
  437. if (slab_obj_ptr == NULL)
  438. {
  439. kBUG("kmalloc()->kmalloc_create_slab_obj()=>slab == NULL");
  440. return NULL;
  441. }
  442. kmalloc_cache_group[index].count_total_free += slab_obj_ptr->count_free;
  443. list_add(&kmalloc_cache_group[index].cache_pool_entry->list, &slab_obj_ptr->list);
  444. }
  445. else // 内存对象充足
  446. {
  447. do
  448. {
  449. // 跳转到下一个内存池对象
  450. if (slab_obj_ptr->count_free == 0)
  451. slab_obj_ptr = container_of(list_next(&slab_obj_ptr->list), struct slab_obj, list);
  452. else
  453. break;
  454. } while (slab_obj_ptr != kmalloc_cache_group[index].cache_pool_entry);
  455. }
  456. // 寻找一块可用的内存对象
  457. int md;
  458. for (int i = 0; i < slab_obj_ptr->bmp_count; ++i)
  459. {
  460. // 当前bmp全部被使用
  461. if (*(slab_obj_ptr->bmp + (i >> 6)) == 0xffffffffffffffffUL)
  462. {
  463. i += 63;
  464. continue;
  465. }
  466. md = i % 64;
  467. // 找到相应的内存对象
  468. if ((*(slab_obj_ptr->bmp + (i >> 6)) & (1UL << md)) == 0)
  469. {
  470. *(slab_obj_ptr->bmp + (i >> 6)) |= (1UL << md);
  471. ++(slab_obj_ptr->count_using);
  472. --(slab_obj_ptr->count_free);
  473. --kmalloc_cache_group[index].count_total_free;
  474. ++kmalloc_cache_group[index].count_total_using;
  475. return (void *)((char *)slab_obj_ptr->vaddr + kmalloc_cache_group[index].size * i);
  476. }
  477. }
  478. kerror("kmalloc(): Cannot alloc more memory: %d bytes", size);
  479. return NULL;
  480. }
  481. /**
  482. * @brief 通用内存释放函数
  483. *
  484. * @param address 要释放的内存线性地址
  485. * @return unsigned long
  486. */
  487. unsigned long kfree(void *address)
  488. {
  489. struct slab_obj *slab_obj_ptr = NULL;
  490. // 将线性地址按照2M物理页对齐, 获得所在物理页的起始线性地址
  491. void *page_base_addr = (void *)((ul)address & PAGE_2M_MASK);
  492. int index;
  493. for (int i = 0; i < 16; ++i)
  494. {
  495. slab_obj_ptr = kmalloc_cache_group[i].cache_pool_entry;
  496. do
  497. {
  498. // 不属于当前slab_obj的管理范围
  499. if (slab_obj_ptr->vaddr != page_base_addr)
  500. {
  501. slab_obj_ptr = container_of(list_next(&slab_obj_ptr->list), struct slab_obj, list);
  502. }
  503. else
  504. {
  505. // 计算地址属于哪一个内存对象
  506. index = (address - slab_obj_ptr->vaddr) / kmalloc_cache_group[i].size;
  507. // 复位bmp
  508. *(slab_obj_ptr->bmp + (index >> 6)) ^= 1UL << (index % 64);
  509. ++(slab_obj_ptr->count_free);
  510. --(slab_obj_ptr->count_using);
  511. ++kmalloc_cache_group[i].count_total_free;
  512. --kmalloc_cache_group[i].count_total_using;
  513. // 回收空闲的slab_obj
  514. // 条件:当前slab_obj_ptr的使用为0、总空闲内存对象>=当前slab_obj的总对象的2倍 且当前slab_pool不为起始slab_obj
  515. if ((slab_obj_ptr->count_using == 0) && (kmalloc_cache_group[i].count_total_free >= ((slab_obj_ptr->bmp_count) << 1)) && (kmalloc_cache_group[i].cache_pool_entry != slab_obj_ptr))
  516. {
  517. switch (kmalloc_cache_group[i].size)
  518. {
  519. case 32:
  520. case 64:
  521. case 128:
  522. case 256:
  523. case 512:
  524. // 在这种情况下,slab_obj是被安放在page内部的
  525. list_del(&slab_obj_ptr->list);
  526. kmalloc_cache_group[i].count_total_free -= slab_obj_ptr->bmp_count;
  527. page_clean(slab_obj_ptr->page);
  528. free_pages(slab_obj_ptr->page, 1);
  529. break;
  530. default:
  531. // 在这种情况下,slab_obj是被安放在额外获取的内存对象中的
  532. list_del(&slab_obj_ptr->list);
  533. kmalloc_cache_group[i].count_total_free -= slab_obj_ptr->bmp_count;
  534. kfree(slab_obj_ptr->bmp);
  535. page_clean(slab_obj_ptr->page);
  536. free_pages(slab_obj_ptr->page, 1);
  537. kfree(slab_obj_ptr);
  538. break;
  539. }
  540. }
  541. return 0;
  542. }
  543. } while (slab_obj_ptr != kmalloc_cache_group[i].cache_pool_entry);
  544. }
  545. kBUG("kfree(): Can't free memory.");
  546. return ECANNOT_FREE_MEM;
  547. }