1 /*
2 * Copyright (c) 2022 Jeppe Odgaard
3 *
4 * SPDX-License-Identifier: Apache-2.0
5 */
6
7 #include <zephyr/kernel.h>
8 #include <zephyr/sys/sys_heap.h>
9
10 #define HEAP_SIZE 256
11
12 K_HEAP_DEFINE(my_kernel_heap, HEAP_SIZE);
13
14 static char heap_mem[HEAP_SIZE];
15 static struct sys_heap heap;
16
17 static void print_sys_memory_stats(struct sys_heap *);
18 static void print_static_heaps(void);
19 static void print_all_heaps(void);
20
main(void)21 int main(void)
22 {
23 void *p;
24
25 printk("System heap sample\n\n");
26
27 sys_heap_init(&heap, heap_mem, HEAP_SIZE);
28 print_sys_memory_stats(&heap);
29
30 p = sys_heap_alloc(&heap, 150);
31 print_sys_memory_stats(&heap);
32
33 p = sys_heap_realloc(&heap, p, 100);
34 print_sys_memory_stats(&heap);
35
36 sys_heap_free(&heap, p);
37 print_sys_memory_stats(&heap);
38
39 /* Allocate from our custom k_heap to make the output
40 * more interesting and prevent compiler optimisations from
41 * discarding it.
42 */
43 p = k_heap_alloc(&my_kernel_heap, 10, K_FOREVER);
44
45 print_static_heaps();
46 print_all_heaps();
47
48 /* Cleanup memory */
49 k_heap_free(&my_kernel_heap, p);
50 return 0;
51 }
52
print_sys_memory_stats(struct sys_heap * hp)53 static void print_sys_memory_stats(struct sys_heap *hp)
54 {
55 struct sys_memory_stats stats;
56
57 sys_heap_runtime_stats_get(hp, &stats);
58
59 printk("allocated %zu, free %zu, max allocated %zu, heap size %u\n",
60 stats.allocated_bytes, stats.free_bytes,
61 stats.max_allocated_bytes, HEAP_SIZE);
62 }
63
print_static_heaps(void)64 static void print_static_heaps(void)
65 {
66 struct k_heap *ha;
67 int n;
68
69 n = k_heap_array_get(&ha);
70 printk("%d static heap(s) allocated:\n", n);
71
72 for (int i = 0; i < n; i++) {
73 printk("\t%d - address %p ", i, &ha[i]);
74 print_sys_memory_stats(&ha[i].heap);
75 }
76 }
77
print_all_heaps(void)78 static void print_all_heaps(void)
79 {
80 struct sys_heap **ha;
81 int n;
82
83 n = sys_heap_array_get(&ha);
84 printk("%d heap(s) allocated (including static):\n", n);
85
86 for (int i = 0; i < n; i++) {
87 printk("\t%d - address %p ", i, ha[i]);
88 print_sys_memory_stats(ha[i]);
89 }
90 }
91