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_all_heaps(void);
19 
main(void)20 int main(void)
21 {
22 	void *p;
23 
24 	printk("System heap sample\n\n");
25 
26 	sys_heap_init(&heap, heap_mem, HEAP_SIZE);
27 	print_sys_memory_stats(&heap);
28 
29 	p = sys_heap_alloc(&heap, 150);
30 	print_sys_memory_stats(&heap);
31 
32 	p = sys_heap_realloc(&heap, p, 100);
33 	print_sys_memory_stats(&heap);
34 
35 	sys_heap_free(&heap, p);
36 	print_sys_memory_stats(&heap);
37 
38 	print_all_heaps();
39 	return 0;
40 }
41 
print_sys_memory_stats(struct sys_heap * hp)42 static void print_sys_memory_stats(struct sys_heap *hp)
43 {
44 	struct sys_memory_stats stats;
45 
46 	sys_heap_runtime_stats_get(hp, &stats);
47 
48 	printk("allocated %zu, free %zu, max allocated %zu, heap size %u\n",
49 		stats.allocated_bytes, stats.free_bytes,
50 		stats.max_allocated_bytes, HEAP_SIZE);
51 }
52 
print_all_heaps(void)53 static void print_all_heaps(void)
54 {
55 	struct sys_heap **ha;
56 	size_t i, n;
57 
58 	n = sys_heap_array_get(&ha);
59 	printk("There are %zu heaps allocated:\n", n);
60 
61 	for (i = 0; i < n; i++) {
62 		printk("\t%zu - address %p ", i, ha[i]);
63 		print_sys_memory_stats(ha[i]);
64 	}
65 }
66