e354a5
commit 72d36ffd7db55ae599f4c77feb0eae25a0f3714e
e354a5
Author: Florian Weimer <fweimer@redhat.com>
e354a5
Date:   Thu Oct 8 10:57:09 2020 +0200
e354a5
e354a5
    elf: Implement __rtld_malloc_is_complete
e354a5
    
e354a5
    In some cases, it is difficult to determine the kind of malloc
e354a5
    based on the execution context, so a function to determine that
e354a5
    is helpful.
e354a5
    
e354a5
    Reviewed-by: Adhemerval Zanella  <adhemerval.zanella@linaro.org>
e354a5
e354a5
diff --git a/elf/dl-minimal.c b/elf/dl-minimal.c
e354a5
index 4335f1bd24289b01..cf7df8a8d5eabe9d 100644
e354a5
--- a/elf/dl-minimal.c
e354a5
+++ b/elf/dl-minimal.c
e354a5
@@ -59,6 +59,14 @@ __rtld_malloc_init_stubs (void)
e354a5
   __rtld_realloc = &rtld_realloc;
e354a5
 }
e354a5
 
e354a5
+bool
e354a5
+__rtld_malloc_is_complete (void)
e354a5
+{
e354a5
+  /* The caller assumes that there is an active malloc.  */
e354a5
+  assert (__rtld_malloc != NULL);
e354a5
+  return __rtld_malloc != &rtld_malloc;
e354a5
+}
e354a5
+
e354a5
 /* Lookup NAME at VERSION in the scope of MATCH.  */
e354a5
 static void *
e354a5
 lookup_malloc_symbol (struct link_map *main_map, const char *name,
e354a5
diff --git a/include/rtld-malloc.h b/include/rtld-malloc.h
e354a5
index b026a3270cd24819..9266ec532f3f3376 100644
e354a5
--- a/include/rtld-malloc.h
e354a5
+++ b/include/rtld-malloc.h
e354a5
@@ -66,6 +66,10 @@ realloc (void *ptr, size_t size)
e354a5
    implementation.  */
e354a5
 void __rtld_malloc_init_stubs (void) attribute_hidden;
e354a5
 
e354a5
+/* Return false if the active malloc is the ld.so minimal malloc, true
e354a5
+   if it is the full implementation from libc.so.  */
e354a5
+_Bool __rtld_malloc_is_complete (void) attribute_hidden;
e354a5
+
e354a5
 /* Called shortly before the final self-relocation (when RELRO
e354a5
    variables are still writable) to activate the real malloc
e354a5
    implementation.  MAIN_MAP is the link map of the executable.  */