From f9a774cc58d0bc802907fb4b37a3cc9d23f560c9 Mon Sep 17 00:00:00 2001
From: Kevin O'Connor <kevin@koconnor.net>
Date: Sat, 17 Apr 2010 16:58:32 -0400
Subject: [PATCH] Add __attribute__((__malloc__)) declaration to internal
 malloc funcs.

---
 src/pmm.c   | 4 ++--
 src/types.h | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/pmm.c b/src/pmm.c
index 1b90f21..58c0e8b 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 e013358..5083941 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
-- 
2.39.5