[ELFLOADER] Better handling of STB_GNU_UNIQUE binded symbol (prevent dlclose)

2 files changed, 16 insertions, 0 deletions

diff --git a/src/elfs/elfloader.c b/src/elfs/elfloader.c
index 9f01f75b..f68658d7 100644
--- a/src/elfs/elfloader.c
+++ b/src/elfs/elfloader.c
@@ -499,6 +499,21 @@ void GrabX64CopyMainElfReloc(elfheader_t* head)
         }
     }
 }
+void CheckGNUUniqueBindings(elfheader_t* head)
+{
+    if(head->rela) {
+        int cnt = head->relasz / head->relaent;
+        Elf64_Rela* rela = (Elf64_Rela *)(head->rela + head->delta);
+        printf_dump(LOG_DEBUG, "Checking for symbol with STB_GNU_UNIQUE bindingsfor %s\n", head->name);
+        for (int i=0; i<cnt; ++i) {
+            int bind = ELF64_ST_BIND(rela[i].r_info);
+            if(bind == STB_GNU_UNIQUE) {
+                head->gnuunique = 1;
+                return; // can stop searching
+            }
+        }
+    }
+}
 
 static elfheader_t* checkElfLib(elfheader_t* h, library_t* lib)
 {
diff --git a/src/elfs/elfloader_private.h b/src/elfs/elfloader_private.h
index 4487e972..91b3b510 100644
--- a/src/elfs/elfloader_private.h
+++ b/src/elfs/elfloader_private.h
@@ -102,6 +102,7 @@ typedef struct elfheader_s {
     int         fini_done;
     int         refcnt;     // ref count for the elf
     int         malloc_hook_2;  // this elf hook malloc, hacking it
+    int         gnuunique;  // set if contains some STB_GNU_UNIQUE binding, preventing dlclose to unload the lib
 
     char*       memory;     // char* and not void* to allow math on memory pointer
     multiblock_t*  multiblocks;