slab.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903
  1. /*
  2. * File : slab.c
  3. * This file is part of RT-Thread RTOS
  4. * COPYRIGHT (C) 2008 - 2011, RT-Thread Development Team
  5. *
  6. * The license and distribution terms for this file may be
  7. * found in the file LICENSE in this distribution or at
  8. * http://www.rt-thread.org/license/LICENSE
  9. *
  10. * Change Logs:
  11. * Date Author Notes
  12. * 2008-07-12 Bernard the first version
  13. * 2010-07-13 Bernard fix RT_ALIGN issue found by kuronca
  14. * 2010-10-23 yi.qiu add module memory allocator
  15. * 2010-12-18 yi.qiu fix zone release bug
  16. */
  17. /*
  18. * KERN_SLABALLOC.C - Kernel SLAB memory allocator
  19. *
  20. * Copyright (c) 2003,2004 The DragonFly Project. All rights reserved.
  21. *
  22. * This code is derived from software contributed to The DragonFly Project
  23. * by Matthew Dillon <dillon@backplane.com>
  24. *
  25. * Redistribution and use in source and binary forms, with or without
  26. * modification, are permitted provided that the following conditions
  27. * are met:
  28. *
  29. * 1. Redistributions of source code must retain the above copyright
  30. * notice, this list of conditions and the following disclaimer.
  31. * 2. Redistributions in binary form must reproduce the above copyright
  32. * notice, this list of conditions and the following disclaimer in
  33. * the documentation and/or other materials provided with the
  34. * distribution.
  35. * 3. Neither the name of The DragonFly Project nor the names of its
  36. * contributors may be used to endorse or promote products derived
  37. * from this software without specific, prior written permission.
  38. *
  39. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  40. * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  41. * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  42. * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  43. * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  44. * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
  45. * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  46. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  47. * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  48. * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  49. * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  50. * SUCH DAMAGE.
  51. *
  52. */
  53. #include <rthw.h>
  54. #include <rtthread.h>
  55. #include "kservice.h"
  56. #define RT_MEM_STATS
  57. #if defined (RT_USING_HEAP) && defined (RT_USING_SLAB)
  58. /* some statistical variable */
  59. #ifdef RT_MEM_STATS
  60. static rt_size_t used_mem, max_mem;
  61. #endif
  62. #ifdef RT_USING_HOOK
  63. static void (*rt_malloc_hook)(void *ptr, rt_size_t size);
  64. static void (*rt_free_hook)(void *ptr);
  65. /**
  66. * @addtogroup Hook
  67. */
  68. /*@{*/
  69. /**
  70. * This function will set a hook function, which will be invoked when a memory
  71. * block is allocated from heap memory.
  72. *
  73. * @param hook the hook function
  74. */
  75. void rt_malloc_sethook(void (*hook)(void *ptr, rt_size_t size))
  76. {
  77. rt_malloc_hook = hook;
  78. }
  79. /**
  80. * This function will set a hook function, which will be invoked when a memory
  81. * block is released to heap memory.
  82. *
  83. * @param hook the hook function
  84. */
  85. void rt_free_sethook(void (*hook)(void *ptr))
  86. {
  87. rt_free_hook = hook;
  88. }
  89. /*@}*/
  90. #endif
  91. /*
  92. * slab allocator implementation
  93. *
  94. * A slab allocator reserves a ZONE for each chunk size, then lays the
  95. * chunks out in an array within the zone. Allocation and deallocation
  96. * is nearly instantanious, and fragmentation/overhead losses are limited
  97. * to a fixed worst-case amount.
  98. *
  99. * The downside of this slab implementation is in the chunk size
  100. * multiplied by the number of zones. ~80 zones * 128K = 10MB of VM per cpu.
  101. * In a kernel implementation all this memory will be physical so
  102. * the zone size is adjusted downward on machines with less physical
  103. * memory. The upside is that overhead is bounded... this is the *worst*
  104. * case overhead.
  105. *
  106. * Slab management is done on a per-cpu basis and no locking or mutexes
  107. * are required, only a critical section. When one cpu frees memory
  108. * belonging to another cpu's slab manager an asynchronous IPI message
  109. * will be queued to execute the operation. In addition, both the
  110. * high level slab allocator and the low level zone allocator optimize
  111. * M_ZERO requests, and the slab allocator does not have to pre initialize
  112. * the linked list of chunks.
  113. *
  114. * XXX Balancing is needed between cpus. Balance will be handled through
  115. * asynchronous IPIs primarily by reassigning the z_Cpu ownership of chunks.
  116. *
  117. * XXX If we have to allocate a new zone and M_USE_RESERVE is set, use of
  118. * the new zone should be restricted to M_USE_RESERVE requests only.
  119. *
  120. * Alloc Size Chunking Number of zones
  121. * 0-127 8 16
  122. * 128-255 16 8
  123. * 256-511 32 8
  124. * 512-1023 64 8
  125. * 1024-2047 128 8
  126. * 2048-4095 256 8
  127. * 4096-8191 512 8
  128. * 8192-16383 1024 8
  129. * 16384-32767 2048 8
  130. * (if RT_MM_PAGE_SIZE is 4K the maximum zone allocation is 16383)
  131. *
  132. * Allocations >= zone_limit go directly to kmem.
  133. *
  134. * API REQUIREMENTS AND SIDE EFFECTS
  135. *
  136. * To operate as a drop-in replacement to the FreeBSD-4.x malloc() we
  137. * have remained compatible with the following API requirements:
  138. *
  139. * + small power-of-2 sized allocations are power-of-2 aligned (kern_tty)
  140. * + all power-of-2 sized allocations are power-of-2 aligned (twe)
  141. * + malloc(0) is allowed and returns non-RT_NULL (ahc driver)
  142. * + ability to allocate arbitrarily large chunks of memory
  143. */
  144. /*
  145. * Chunk structure for free elements
  146. */
  147. typedef struct slab_chunk
  148. {
  149. struct slab_chunk *c_next;
  150. } slab_chunk;
  151. /*
  152. * The IN-BAND zone header is placed at the beginning of each zone.
  153. */
  154. typedef struct slab_zone
  155. {
  156. rt_int32_t z_magic; /* magic number for sanity check */
  157. rt_int32_t z_nfree; /* total free chunks / ualloc space in zone */
  158. rt_int32_t z_nmax; /* maximum free chunks */
  159. struct slab_zone *z_next; /* zoneary[] link if z_nfree non-zero */
  160. rt_uint8_t *z_baseptr; /* pointer to start of chunk array */
  161. rt_int32_t z_uindex; /* current initial allocation index */
  162. rt_int32_t z_chunksize; /* chunk size for validation */
  163. rt_int32_t z_zoneindex; /* zone index */
  164. slab_chunk *z_freechunk; /* free chunk list */
  165. } slab_zone;
  166. #define ZALLOC_SLAB_MAGIC 0x51ab51ab
  167. #define ZALLOC_ZONE_LIMIT (16 * 1024) /* max slab-managed alloc */
  168. #define ZALLOC_MIN_ZONE_SIZE (32 * 1024) /* minimum zone size */
  169. #define ZALLOC_MAX_ZONE_SIZE (128 * 1024) /* maximum zone size */
  170. #define NZONES 72 /* number of zones */
  171. #define ZONE_RELEASE_THRESH 2 /* threshold number of zones */
  172. static slab_zone *zone_array[NZONES]; /* linked list of zones NFree > 0 */
  173. static slab_zone *zone_free; /* whole zones that have become free */
  174. static int zone_free_cnt;
  175. static int zone_size;
  176. static int zone_limit;
  177. static int zone_page_cnt;
  178. /*
  179. * Misc constants. Note that allocations that are exact multiples of
  180. * RT_MM_PAGE_SIZE, or exceed the zone limit, fall through to the kmem module.
  181. */
  182. #define MIN_CHUNK_SIZE 8 /* in bytes */
  183. #define MIN_CHUNK_MASK (MIN_CHUNK_SIZE - 1)
  184. /*
  185. * Array of descriptors that describe the contents of each page
  186. */
  187. #define PAGE_TYPE_FREE 0x00
  188. #define PAGE_TYPE_SMALL 0x01
  189. #define PAGE_TYPE_LARGE 0x02
  190. struct memusage
  191. {
  192. rt_uint32_t type:2 ; /* page type */
  193. rt_uint32_t size:30; /* pages allocated or offset from zone */
  194. };
  195. static struct memusage *memusage = RT_NULL;
  196. #define btokup(addr) (&memusage[((rt_uint32_t)(addr) - heap_start) >> RT_MM_PAGE_BITS])
  197. static rt_uint32_t heap_start, heap_end;
  198. /* page allocator */
  199. struct rt_page_head
  200. {
  201. struct rt_page_head *next; /* next valid page */
  202. rt_size_t page; /* number of page */
  203. /* dummy */
  204. char dummy[RT_MM_PAGE_SIZE - (sizeof(struct rt_page_head*) + sizeof (rt_size_t))];
  205. };
  206. static struct rt_page_head *rt_page_list;
  207. static struct rt_semaphore heap_sem;
  208. void *rt_page_alloc(rt_size_t npages)
  209. {
  210. struct rt_page_head *b, *n;
  211. struct rt_page_head **prev;
  212. if(npages == 0) return RT_NULL;
  213. /* lock heap */
  214. rt_sem_take(&heap_sem, RT_WAITING_FOREVER);
  215. for (prev = &rt_page_list; (b = *prev) != RT_NULL; prev = &(b->next))
  216. {
  217. if (b->page > npages)
  218. {
  219. /* splite pages */
  220. n = b + npages;
  221. n->next = b->next;
  222. n->page = b->page - npages;
  223. *prev = n;
  224. break;
  225. }
  226. if (b->page == npages)
  227. {
  228. /* this node fit, remove this node */
  229. *prev = b->next;
  230. break;
  231. }
  232. }
  233. /* unlock heap */
  234. rt_sem_release(&heap_sem);
  235. return b;
  236. }
  237. void rt_page_free(void *addr, rt_size_t npages)
  238. {
  239. struct rt_page_head *b, *n;
  240. struct rt_page_head **prev;
  241. RT_ASSERT(addr != RT_NULL);
  242. RT_ASSERT((rt_uint32_t)addr % RT_MM_PAGE_SIZE == 0);
  243. RT_ASSERT(npages != 0);
  244. n = (struct rt_page_head *)addr;
  245. /* lock heap */
  246. rt_sem_take(&heap_sem, RT_WAITING_FOREVER);
  247. for (prev = &rt_page_list; (b = *prev) != RT_NULL; prev = &(b->next))
  248. {
  249. RT_ASSERT(b->page > 0);
  250. RT_ASSERT(b > n || b + b->page <= n);
  251. if (b + b->page == n)
  252. {
  253. if (b + (b->page += npages) == b->next)
  254. {
  255. b->page += b->next->page;
  256. b->next = b->next->next;
  257. }
  258. goto _return;
  259. }
  260. if (b == n + npages)
  261. {
  262. n->page = b->page + npages;
  263. n->next = b->next;
  264. *prev = n;
  265. goto _return;
  266. }
  267. if (b > n + npages) break;
  268. }
  269. n->page = npages;
  270. n->next = b;
  271. *prev = n;
  272. _return:
  273. /* unlock heap */
  274. rt_sem_release(&heap_sem);
  275. }
  276. /*
  277. * Initialize the page allocator
  278. */
  279. static void rt_page_init(void *addr, rt_size_t npages)
  280. {
  281. RT_ASSERT(addr != RT_NULL);
  282. RT_ASSERT(npages != 0);
  283. rt_page_list = RT_NULL;
  284. rt_page_free(addr, npages);
  285. }
  286. /**
  287. * @ingroup SystemInit
  288. *
  289. * This function will init system heap
  290. *
  291. * @param begin_addr the beginning address of system page
  292. * @param end_addr the end address of system page
  293. *
  294. */
  295. void rt_system_heap_init(void *begin_addr, void *end_addr)
  296. {
  297. rt_uint32_t limsize, npages;
  298. RT_DEBUG_NOT_IN_INTERRUPT;
  299. /* align begin and end addr to page */
  300. heap_start = RT_ALIGN((rt_uint32_t)begin_addr, RT_MM_PAGE_SIZE);
  301. heap_end = RT_ALIGN_DOWN((rt_uint32_t)end_addr, RT_MM_PAGE_SIZE);
  302. if (heap_start >= heap_end)
  303. {
  304. rt_kprintf("rt_system_heap_init, wrong address[0x%x - 0x%x]\n",
  305. (rt_uint32_t)begin_addr, (rt_uint32_t)end_addr);
  306. return;
  307. }
  308. limsize = heap_end - heap_start;
  309. npages = limsize / RT_MM_PAGE_SIZE;
  310. /* initialize heap semaphore */
  311. rt_sem_init(&heap_sem, "heap", 1, RT_IPC_FLAG_FIFO);
  312. RT_DEBUG_LOG(RT_DEBUG_SLAB,
  313. ("heap[0x%x - 0x%x], size 0x%x, 0x%x pages\n", heap_start, heap_end, limsize, npages));
  314. /* init pages */
  315. rt_page_init((void *)heap_start, npages);
  316. /* calculate zone size */
  317. zone_size = ZALLOC_MIN_ZONE_SIZE;
  318. while (zone_size < ZALLOC_MAX_ZONE_SIZE && (zone_size << 1) < (limsize/1024))
  319. zone_size <<= 1;
  320. zone_limit = zone_size / 4;
  321. if (zone_limit > ZALLOC_ZONE_LIMIT) zone_limit = ZALLOC_ZONE_LIMIT;
  322. zone_page_cnt = zone_size / RT_MM_PAGE_SIZE;
  323. RT_DEBUG_LOG(RT_DEBUG_SLAB,
  324. ("zone size 0x%x, zone page count 0x%x\n", zone_size, zone_page_cnt));
  325. /* allocate memusage array */
  326. limsize = npages * sizeof(struct memusage);
  327. limsize = RT_ALIGN(limsize, RT_MM_PAGE_SIZE);
  328. memusage = rt_page_alloc(limsize/RT_MM_PAGE_SIZE);
  329. RT_DEBUG_LOG(RT_DEBUG_SLAB,
  330. ("memusage 0x%x, size 0x%x\n", (rt_uint32_t)memusage, limsize));
  331. }
  332. /*
  333. * Calculate the zone index for the allocation request size and set the
  334. * allocation request size to that particular zone's chunk size.
  335. */
  336. rt_inline int zoneindex(rt_uint32_t *bytes)
  337. {
  338. rt_uint32_t n = (rt_uint32_t)*bytes; /* unsigned for shift opt */
  339. if (n < 128)
  340. {
  341. *bytes = n = (n + 7) & ~7;
  342. return(n / 8 - 1); /* 8 byte chunks, 16 zones */
  343. }
  344. if (n < 256)
  345. {
  346. *bytes = n = (n + 15) & ~15;
  347. return(n / 16 + 7);
  348. }
  349. if (n < 8192)
  350. {
  351. if (n < 512)
  352. {
  353. *bytes = n = (n + 31) & ~31;
  354. return(n / 32 + 15);
  355. }
  356. if (n < 1024)
  357. {
  358. *bytes = n = (n + 63) & ~63;
  359. return(n / 64 + 23);
  360. }
  361. if (n < 2048)
  362. {
  363. *bytes = n = (n + 127) & ~127;
  364. return(n / 128 + 31);
  365. }
  366. if (n < 4096)
  367. {
  368. *bytes = n = (n + 255) & ~255;
  369. return(n / 256 + 39);
  370. }
  371. *bytes = n = (n + 511) & ~511;
  372. return(n / 512 + 47);
  373. }
  374. if (n < 16384)
  375. {
  376. *bytes = n = (n + 1023) & ~1023;
  377. return(n / 1024 + 55);
  378. }
  379. rt_kprintf("Unexpected byte count %d", n);
  380. return 0;
  381. }
  382. /**
  383. * @addtogroup MM
  384. */
  385. /*@{*/
  386. /**
  387. * This function will allocate a block from system heap memory.
  388. * - If the nbytes is less than zero,
  389. * or
  390. * - If there is no nbytes sized memory valid in system,
  391. * the RT_NULL is returned.
  392. *
  393. * @param size the size of memory to be allocated
  394. *
  395. * @return the allocated memory
  396. *
  397. */
  398. void *rt_malloc(rt_size_t size)
  399. {
  400. slab_zone *z;
  401. rt_int32_t zi;
  402. slab_chunk *chunk;
  403. struct memusage *kup;
  404. /* zero size, return RT_NULL */
  405. if (size == 0) return RT_NULL;
  406. #ifdef RT_USING_MODULE
  407. if (rt_module_self() != RT_NULL) return rt_module_malloc(size);
  408. #endif
  409. /*
  410. * Handle large allocations directly. There should not be very many of
  411. * these so performance is not a big issue.
  412. */
  413. if (size >= zone_limit)
  414. {
  415. size = RT_ALIGN(size, RT_MM_PAGE_SIZE);
  416. chunk = rt_page_alloc(size >> RT_MM_PAGE_BITS);
  417. if (chunk == RT_NULL) return RT_NULL;
  418. /* set kup */
  419. kup = btokup(chunk);
  420. kup->type = PAGE_TYPE_LARGE;
  421. kup->size = size >> RT_MM_PAGE_BITS;
  422. RT_DEBUG_LOG(RT_DEBUG_SLAB, ("malloc a large memory 0x%x, page cnt %d, kup %d\n",
  423. size,
  424. size >> RT_MM_PAGE_BITS,
  425. ((rt_uint32_t)chunk - heap_start) >> RT_MM_PAGE_BITS));
  426. /* lock heap */
  427. rt_sem_take(&heap_sem, RT_WAITING_FOREVER);
  428. #ifdef RT_MEM_STATS
  429. used_mem += size;
  430. if (used_mem > max_mem) max_mem = used_mem;
  431. #endif
  432. goto done;
  433. }
  434. /* lock heap */
  435. rt_sem_take(&heap_sem, RT_WAITING_FOREVER);
  436. /*
  437. * Attempt to allocate out of an existing zone. First try the free list,
  438. * then allocate out of unallocated space. If we find a good zone move
  439. * it to the head of the list so later allocations find it quickly
  440. * (we might have thousands of zones in the list).
  441. *
  442. * Note: zoneindex() will panic of size is too large.
  443. */
  444. zi = zoneindex(&size);
  445. RT_ASSERT(zi < NZONES);
  446. RT_DEBUG_LOG(RT_DEBUG_SLAB, ("try to malloc 0x%x on zone: %d\n", size, zi));
  447. if ((z = zone_array[zi]) != RT_NULL)
  448. {
  449. RT_ASSERT(z->z_nfree > 0);
  450. /* Remove us from the zone_array[] when we become empty */
  451. if (--z->z_nfree == 0)
  452. {
  453. zone_array[zi] = z->z_next;
  454. z->z_next = RT_NULL;
  455. }
  456. /*
  457. * No chunks are available but nfree said we had some memory, so
  458. * it must be available in the never-before-used-memory area
  459. * governed by uindex. The consequences are very serious if our zone
  460. * got corrupted so we use an explicit rt_kprintf rather then a KASSERT.
  461. */
  462. if (z->z_uindex + 1 != z->z_nmax)
  463. {
  464. z->z_uindex = z->z_uindex + 1;
  465. chunk = (slab_chunk *)(z->z_baseptr + z->z_uindex * size);
  466. }
  467. else
  468. {
  469. /* find on free chunk list */
  470. chunk = z->z_freechunk;
  471. /* remove this chunk from list */
  472. z->z_freechunk = z->z_freechunk->c_next;
  473. }
  474. #ifdef RT_MEM_STATS
  475. used_mem += z->z_chunksize;
  476. if (used_mem > max_mem) max_mem = used_mem;
  477. #endif
  478. goto done;
  479. }
  480. /*
  481. * If all zones are exhausted we need to allocate a new zone for this
  482. * index.
  483. *
  484. * At least one subsystem, the tty code (see CROUND) expects power-of-2
  485. * allocations to be power-of-2 aligned. We maintain compatibility by
  486. * adjusting the base offset below.
  487. */
  488. {
  489. rt_int32_t off;
  490. if ((z = zone_free) != RT_NULL)
  491. {
  492. /* remove zone from free zone list */
  493. zone_free = z->z_next;
  494. -- zone_free_cnt;
  495. }
  496. else
  497. {
  498. /* unlock heap, since page allocator will think about lock */
  499. rt_sem_release(&heap_sem);
  500. /* allocate a zone from page */
  501. z = rt_page_alloc(zone_size / RT_MM_PAGE_SIZE);
  502. if (z == RT_NULL) goto fail;
  503. /* lock heap */
  504. rt_sem_take(&heap_sem, RT_WAITING_FOREVER);
  505. RT_DEBUG_LOG(RT_DEBUG_SLAB, ("alloc a new zone: 0x%x\n", (rt_uint32_t)z));
  506. /* set message usage */
  507. for (off = 0, kup = btokup(z); off < zone_page_cnt; off ++)
  508. {
  509. kup->type = PAGE_TYPE_SMALL;
  510. kup->size = off;
  511. kup ++;
  512. }
  513. }
  514. /* clear to zero */
  515. rt_memset(z, 0, sizeof(slab_zone));
  516. /* offset of slab zone struct in zone */
  517. off = sizeof(slab_zone);
  518. /*
  519. * Guarentee power-of-2 alignment for power-of-2-sized chunks.
  520. * Otherwise just 8-byte align the data.
  521. */
  522. if ((size | (size - 1)) + 1 == (size << 1))
  523. off = (off + size - 1) & ~(size - 1);
  524. else
  525. off = (off + MIN_CHUNK_MASK) & ~MIN_CHUNK_MASK;
  526. z->z_magic = ZALLOC_SLAB_MAGIC;
  527. z->z_zoneindex = zi;
  528. z->z_nmax = (zone_size - off) / size;
  529. z->z_nfree = z->z_nmax - 1;
  530. z->z_baseptr = (rt_uint8_t *)z + off;
  531. z->z_uindex = 0;
  532. z->z_chunksize = size;
  533. chunk = (slab_chunk *)(z->z_baseptr + z->z_uindex * size);
  534. /* link to zone array */
  535. z->z_next = zone_array[zi];
  536. zone_array[zi] = z;
  537. #ifdef RT_MEM_STATS
  538. used_mem += z->z_chunksize;
  539. if (used_mem > max_mem) max_mem = used_mem;
  540. #endif
  541. }
  542. done:
  543. rt_sem_release(&heap_sem);
  544. RT_OBJECT_HOOK_CALL(rt_malloc_hook, ((char *)chunk, size));
  545. return chunk;
  546. fail:
  547. rt_sem_release(&heap_sem);
  548. return RT_NULL;
  549. }
  550. /**
  551. * This function will change the size of previously allocated memory block.
  552. *
  553. * @param ptr the previously allocated memory block
  554. * @param size the new size of memory block
  555. *
  556. * @return the allocated memory
  557. */
  558. void *rt_realloc(void *ptr, rt_size_t size)
  559. {
  560. void *nptr;
  561. slab_zone *z;
  562. struct memusage *kup;
  563. if (ptr == RT_NULL) return rt_malloc(size);
  564. if (size == 0)
  565. {
  566. rt_free(ptr);
  567. return RT_NULL;
  568. }
  569. #ifdef RT_USING_MODULE
  570. if (rt_module_self() != RT_NULL) return rt_module_realloc(ptr, size);
  571. #endif
  572. /*
  573. * Get the original allocation's zone. If the new request winds up
  574. * using the same chunk size we do not have to do anything.
  575. */
  576. kup = btokup((rt_uint32_t)ptr & ~RT_MM_PAGE_MASK);
  577. if (kup->type == PAGE_TYPE_LARGE)
  578. {
  579. rt_size_t osize;
  580. osize = kup->size << RT_MM_PAGE_BITS;
  581. if ((nptr = rt_malloc(size)) == RT_NULL) return RT_NULL;
  582. rt_memcpy(nptr, ptr, size > osize ? osize : size);
  583. rt_free(ptr);
  584. return nptr;
  585. }
  586. else if (kup->type == PAGE_TYPE_SMALL)
  587. {
  588. z = (slab_zone *)(((rt_uint32_t)ptr & ~RT_MM_PAGE_MASK) - kup->size * RT_MM_PAGE_SIZE);
  589. RT_ASSERT(z->z_magic == ZALLOC_SLAB_MAGIC);
  590. zoneindex(&size);
  591. if (z->z_chunksize == size) return(ptr); /* same chunk */
  592. /*
  593. * Allocate memory for the new request size. Note that zoneindex has
  594. * already adjusted the request size to the appropriate chunk size, which
  595. * should optimize our bcopy(). Then copy and return the new pointer.
  596. */
  597. if ((nptr = rt_malloc(size)) == RT_NULL) return RT_NULL;
  598. rt_memcpy(nptr, ptr, size > z->z_chunksize ? z->z_chunksize : size);
  599. rt_free(ptr);
  600. return nptr;
  601. }
  602. return RT_NULL;
  603. }
  604. /**
  605. * This function will contiguously allocate enough space for count objects
  606. * that are size bytes of memory each and returns a pointer to the allocated
  607. * memory.
  608. *
  609. * The allocated memory is filled with bytes of value zero.
  610. *
  611. * @param count number of objects to allocate
  612. * @param size size of the objects to allocate
  613. *
  614. * @return pointer to allocated memory / NULL pointer if there is an error
  615. */
  616. void *rt_calloc(rt_size_t count, rt_size_t size)
  617. {
  618. void *p;
  619. /* allocate 'count' objects of size 'size' */
  620. p = rt_malloc(count * size);
  621. /* zero the memory */
  622. if (p) rt_memset(p, 0, count * size);
  623. return p;
  624. }
  625. /**
  626. * This function will release the previously allocated memory block by rt_malloc.
  627. * The released memory block is taken back to system heap.
  628. *
  629. * @param ptr the address of memory which will be released
  630. */
  631. void rt_free(void *ptr)
  632. {
  633. slab_zone *z;
  634. slab_chunk *chunk;
  635. struct memusage *kup;
  636. /* free a RT_NULL pointer */
  637. if (ptr == RT_NULL) return ;
  638. RT_OBJECT_HOOK_CALL(rt_free_hook, (ptr));
  639. #ifdef RT_USING_MODULE
  640. if(rt_module_self() != RT_NULL)
  641. {
  642. rt_module_free(rt_module_self(), ptr);
  643. return;
  644. }
  645. #endif
  646. /* get memory usage */
  647. #if RT_DEBUG_SLAB
  648. {
  649. rt_uint32_t addr = ((rt_uint32_t)ptr & ~RT_MM_PAGE_MASK);
  650. RT_DEBUG_LOG(RT_DEBUG_SLAB,
  651. ("free a memory 0x%x and align to 0x%x, kup index %d\n",
  652. (rt_uint32_t)ptr,
  653. (rt_uint32_t)addr,
  654. ((rt_uint32_t)(addr) - heap_start) >> RT_MM_PAGE_BITS));
  655. }
  656. #endif
  657. kup = btokup((rt_uint32_t)ptr & ~RT_MM_PAGE_MASK);
  658. /* release large allocation */
  659. if (kup->type == PAGE_TYPE_LARGE)
  660. {
  661. rt_uint32_t size;
  662. /* lock heap */
  663. rt_sem_take(&heap_sem, RT_WAITING_FOREVER);
  664. /* clear page counter */
  665. size = kup->size;
  666. kup->size = 0;
  667. #ifdef RT_MEM_STATS
  668. used_mem -= size * RT_MM_PAGE_SIZE;
  669. #endif
  670. rt_sem_release(&heap_sem);
  671. RT_DEBUG_LOG(RT_DEBUG_SLAB,
  672. ("free large memory block 0x%x, page count %d\n", (rt_uint32_t)ptr, size));
  673. /* free this page */
  674. rt_page_free(ptr, size);
  675. return;
  676. }
  677. /* lock heap */
  678. rt_sem_take(&heap_sem, RT_WAITING_FOREVER);
  679. /* zone case. get out zone. */
  680. z = (slab_zone *)(((rt_uint32_t)ptr & ~RT_MM_PAGE_MASK) - kup->size * RT_MM_PAGE_SIZE);
  681. RT_ASSERT(z->z_magic == ZALLOC_SLAB_MAGIC);
  682. chunk = (slab_chunk *)ptr;
  683. chunk->c_next = z->z_freechunk;
  684. z->z_freechunk = chunk;
  685. #ifdef RT_MEM_STATS
  686. used_mem -= z->z_chunksize;
  687. #endif
  688. /*
  689. * Bump the number of free chunks. If it becomes non-zero the zone
  690. * must be added back onto the appropriate list.
  691. */
  692. if (z->z_nfree++ == 0)
  693. {
  694. z->z_next = zone_array[z->z_zoneindex];
  695. zone_array[z->z_zoneindex] = z;
  696. }
  697. /*
  698. * If the zone becomes totally free, and there are other zones we
  699. * can allocate from, move this zone to the FreeZones list. Since
  700. * this code can be called from an IPI callback, do *NOT* try to mess
  701. * with kernel_map here. Hysteresis will be performed at malloc() time.
  702. */
  703. if (z->z_nfree == z->z_nmax && (z->z_next || zone_array[z->z_zoneindex] != z))
  704. {
  705. slab_zone **pz;
  706. RT_DEBUG_LOG(RT_DEBUG_SLAB, ("free zone 0x%x\n", (rt_uint32_t)z, z->z_zoneindex));
  707. /* remove zone from zone array list */
  708. for (pz = &zone_array[z->z_zoneindex]; z != *pz; pz = &(*pz)->z_next);
  709. *pz = z->z_next;
  710. /* reset zone */
  711. z->z_magic = -1;
  712. /* insert to free zone list */
  713. z->z_next = zone_free;
  714. zone_free = z;
  715. ++ zone_free_cnt;
  716. /* release zone to page allocator */
  717. if (zone_free_cnt > ZONE_RELEASE_THRESH)
  718. {
  719. register rt_base_t i;
  720. z = zone_free;
  721. zone_free = z->z_next;
  722. -- zone_free_cnt;
  723. /* set message usage */
  724. for (i = 0, kup = btokup(z); i < zone_page_cnt; i ++)
  725. {
  726. kup->type = PAGE_TYPE_FREE;
  727. kup->size = 0;
  728. kup ++;
  729. }
  730. /* unlock heap */
  731. rt_sem_release(&heap_sem);
  732. /* release pages */
  733. rt_page_free(z, zone_size / RT_MM_PAGE_SIZE);
  734. return;
  735. }
  736. }
  737. /* unlock heap */
  738. rt_sem_release(&heap_sem);
  739. }
  740. #ifdef RT_MEM_STATS
  741. void rt_memory_info(rt_uint32_t *total, rt_uint32_t *used, rt_uint32_t *max_used)
  742. {
  743. if (total != RT_NULL) *total = heap_end - heap_start;
  744. if (used != RT_NULL) *used = used_mem;
  745. if (max_used != RT_NULL) *max_used = max_mem;
  746. }
  747. #ifdef RT_USING_FINSH
  748. #include <finsh.h>
  749. void list_mem(void)
  750. {
  751. rt_kprintf("total memory: %d\n", heap_end - heap_start);
  752. rt_kprintf("used memory : %d\n", used_mem);
  753. rt_kprintf("maximum allocated memory: %d\n", max_mem);
  754. }
  755. FINSH_FUNCTION_EXPORT(list_mem, list memory usage information)
  756. #endif
  757. #endif
  758. /*@}*/
  759. #endif