build: rework mallinfo test & find malloc_size

[mirror_frr.git] / lib / memory_vty.c

diff --git a/lib/memory_vty.c b/lib/memory_vty.c
index 972914bf243b840727bc096b8f823d2fd187a59f..388273128e251ea247a8f388cc62ed746ddc679e 100644 (file)
--- a/lib/memory_vty.c
+++ b/lib/memory_vty.c
@@ -21,9 +21,12 @@
 
 #include <zebra.h>
 /* malloc.h is generally obsolete, however GNU Libc mallinfo wants it. */
-#if (defined(GNU_LINUX) && defined(HAVE_MALLINFO))
+#ifdef HAVE_MALLOC_H
 #include <malloc.h>
-#endif /* HAVE_MALLINFO */
+#endif
+#ifdef HAVE_MALLOC_MALLOC_H
+#include <malloc/malloc.h>
+#endif
 #include <dlfcn.h>
 #include <link.h>