Add __attribute__((__malloc__)) declaration to internal malloc funcs.

2 files changed, 3 insertions, 2 deletions

diff --git a/src/pmm.c b/src/pmm.c
index 1b90f215..58c0e8b0 100644
--- a/src/pmm.c
+++ b/src/pmm.c
@@ -117,7 +117,7 @@ zonelow_expand(u32 size, u32 align)
  ****************************************************************/
 
 // Obtain memory from a given zone.
-static void *
+static void * __malloc
 zone_malloc(struct zone_s *zone, u32 size, u32 align)
 {
     u32 oldpos = GET_PMMVAR(zone->cur);
@@ -186,7 +186,7 @@ struct pmmalloc_s {
 struct pmmalloc_s *PMMAllocs VAR32FLATVISIBLE;
 
 // Allocate memory from the given zone and track it as a PMM allocation
-void *
+void * __malloc
 pmm_malloc(struct zone_s *zone, u32 handle, u32 size, u32 align)
 {
     u32 oldallocdata = GET_PMMVAR(ZoneTmpHigh.cur);
diff --git a/src/types.h b/src/types.h
index e0133589..50839418 100644
--- a/src/types.h
+++ b/src/types.h
@@ -129,6 +129,7 @@ extern void __force_link_error__only_in_16bit(void) __noreturn;
 
 #define noinline __attribute__((noinline))
 #define __always_inline inline __attribute__((always_inline))
+#define __malloc __attribute__((__malloc__))
 #define __attribute_const __attribute__((__const__))
 
 #define __stringify_1(x)        #x