Merge pull request #5793 from ton31337/fix/formatting_show_bgp_summary_failed

[mirror_frr.git] / lib / memory.h

diff --git a/lib/memory.h b/lib/memory.h
index d5facad583363030caf0f641bd0f4fec48e2ed5c..44ea19b5579b17abedcc061f52ff269ed7eaa83a 100644 (file)
--- a/lib/memory.h
+++ b/lib/memory.h
@@ -18,16 +18,30 @@
 #define _QUAGGA_MEMORY_H
 
 #include <stdlib.h>
+#include <stdio.h>
 #include <frratomic.h>
+#include "compiler.h"
 
-#define array_size(ar) (sizeof(ar) / sizeof(ar[0]))
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+#if defined(HAVE_MALLOC_SIZE) && !defined(HAVE_MALLOC_USABLE_SIZE)
+#define malloc_usable_size(x) malloc_size(x)
+#define HAVE_MALLOC_USABLE_SIZE
+#endif
 
 #define SIZE_VAR ~0UL
 struct memtype {
        struct memtype *next, **ref;
        const char *name;
-       _Atomic size_t n_alloc;
-       _Atomic size_t size;
+       atomic_size_t n_alloc;
+       atomic_size_t n_max;
+       atomic_size_t size;
+#ifdef HAVE_MALLOC_USABLE_SIZE
+       atomic_size_t total;
+       atomic_size_t max_size;
+#endif
 };
 
 struct memgroup {
@@ -36,41 +50,6 @@ struct memgroup {
        const char *name;
 };
 
-#if defined(__clang__)
-#if __clang_major__ > 3 || (__clang_major__ == 3 && __clang_minor__ >= 5)
-#  define _RET_NONNULL  , returns_nonnull
-#endif
-# define _CONSTRUCTOR(x) constructor(x)
-#elif defined(__GNUC__)
-#if __GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 9)
-#  define _RET_NONNULL  , returns_nonnull
-#endif
-#if __GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 3)
-#  define _CONSTRUCTOR(x) constructor(x)
-#  define _DESTRUCTOR(x)  destructor(x)
-#  define _ALLOC_SIZE(x)  alloc_size(x)
-#endif
-#endif
-
-#ifdef __sun
-/* Solaris doesn't do constructor priorities due to linker restrictions */
-#undef _CONSTRUCTOR
-#undef _DESTRUCTOR
-#endif
-
-#ifndef _RET_NONNULL
-# define _RET_NONNULL
-#endif
-#ifndef _CONSTRUCTOR
-# define _CONSTRUCTOR(x) constructor
-#endif
-#ifndef _DESTRUCTOR
-# define _DESTRUCTOR(x) destructor
-#endif
-#ifndef _ALLOC_SIZE
-# define _ALLOC_SIZE(x)
-#endif
-
 /* macro usage:
  *
  *  mydaemon.h
@@ -122,41 +101,44 @@ struct memgroup {
                *_mg_##mname.ref = _mg_##mname.next;                           \
        }
 
-
 #define DECLARE_MTYPE(name)                                                    \
-       extern struct memtype _mt_##name;                                      \
-       static struct memtype *const MTYPE_##name = &_mt_##name;
+       extern struct memtype MTYPE_##name[1];                                 \
+       /* end */
 
 #define DEFINE_MTYPE_ATTR(group, mname, attr, desc)                            \
-       attr struct memtype _mt_##mname                                        \
-               __attribute__((section(".data.mtypes"))) = {                   \
+       attr struct memtype MTYPE_##mname[1]                                   \
+               __attribute__((section(".data.mtypes"))) = { {                 \
                        .name = desc,                                          \
                        .next = NULL,                                          \
                        .n_alloc = 0,                                          \
                        .size = 0,                                             \
                        .ref = NULL,                                           \
-       };                                                                     \
+       } };                                                                   \
        static void _mtinit_##mname(void) __attribute__((_CONSTRUCTOR(1001))); \
        static void _mtinit_##mname(void)                                      \
        {                                                                      \
                if (_mg_##group.insert == NULL)                                \
                        _mg_##group.insert = &_mg_##group.types;               \
-               _mt_##mname.ref = _mg_##group.insert;                          \
-               *_mg_##group.insert = &_mt_##mname;                            \
-               _mg_##group.insert = &_mt_##mname.next;                        \
+               MTYPE_##mname->ref = _mg_##group.insert;                       \
+               *_mg_##group.insert = MTYPE_##mname;                           \
+               _mg_##group.insert = &MTYPE_##mname->next;                      \
        }                                                                      \
        static void _mtfini_##mname(void) __attribute__((_DESTRUCTOR(1001)));  \
        static void _mtfini_##mname(void)                                      \
        {                                                                      \
-               if (_mt_##mname.next)                                          \
-                       _mt_##mname.next->ref = _mt_##mname.ref;               \
-               *_mt_##mname.ref = _mt_##mname.next;                           \
-       }
+               if (MTYPE_##mname->next)                                       \
+                       MTYPE_##mname->next->ref = MTYPE_##mname->ref;         \
+               *MTYPE_##mname->ref = MTYPE_##mname->next;                     \
+       }                                                                      \
+       /* end */
+
+#define DEFINE_MTYPE(group, name, desc)                                        \
+       DEFINE_MTYPE_ATTR(group, name, , desc)                                 \
+       /* end */
 
-#define DEFINE_MTYPE(group, name, desc) DEFINE_MTYPE_ATTR(group, name, , desc)
 #define DEFINE_MTYPE_STATIC(group, name, desc)                                 \
        DEFINE_MTYPE_ATTR(group, name, static, desc)                           \
-       static struct memtype *const MTYPE_##name = &_mt_##name;
+       /* end */
 
 DECLARE_MGROUP(LIB)
 DECLARE_MTYPE(TMP)
@@ -194,8 +176,13 @@ static inline size_t mtype_stats_alloc(struct memtype *mt)
  * last value from qmem_walk_fn. */
 typedef int qmem_walk_fn(void *arg, struct memgroup *mg, struct memtype *mt);
 extern int qmem_walk(qmem_walk_fn *func, void *arg);
-extern void log_memstats_stderr(const char *);
+extern int log_memstats(FILE *fp, const char *);
+#define log_memstats_stderr(prefix) log_memstats(stderr, prefix)
 
 extern void memory_oom(size_t size, const char *name);
 
+#ifdef __cplusplus
+}
+#endif
+
 #endif /* _QUAGGA_MEMORY_H */