From 8f66cfa156f04cc6f1f6a78bff261122de6801ca Mon Sep 17 00:00:00 2001
From: Nikias Bassen
Date: Thu, 6 Jan 2022 09:22:30 +0100
Subject: Check availability of constructor attribute and use it on Windows in
 favor of DllMain

---
 src/libirecovery.c | 28 ++++++++++++++++++----------
 1 file changed, 18 insertions(+), 10 deletions(-)

(limited to 'src')

diff --git a/src/libirecovery.c b/src/libirecovery.c
index 0e971a0..b0f6409 100644
--- a/src/libirecovery.c
+++ b/src/libirecovery.c
@@ -442,7 +442,23 @@ static void _irecv_deinit(void)
 static thread_once_t init_once = THREAD_ONCE_INIT;
 static thread_once_t deinit_once = THREAD_ONCE_INIT;
 
-#ifdef WIN32
+#ifndef HAVE_ATTRIBUTE_CONSTRUCTOR
+  #if defined(__llvm__) || defined(__GNUC__)
+    #define HAVE_ATTRIBUTE_CONSTRUCTOR
+  #endif
+#endif
+
+#ifdef HAVE_ATTRIBUTE_CONSTRUCTOR
+static void __attribute__((constructor)) libirecovery_initialize(void)
+{
+	thread_once(&init_once, _irecv_init);
+}
+
+static void __attribute__((destructor)) libirecovery_deinitialize(void)
+{
+	thread_once(&deinit_once, _irecv_deinit);
+}
+#elif defined(WIN32)
 BOOL WINAPI DllMain(HINSTANCE hModule, DWORD dwReason, LPVOID lpReserved)
 {
 	switch (dwReason) {
@@ -458,15 +474,7 @@ BOOL WINAPI DllMain(HINSTANCE hModule, DWORD dwReason, LPVOID lpReserved)
 	return 1;
 }
 #else
-static void __attribute__((constructor)) libirecovery_initialize(void)
-{
-	thread_once(&init_once, _irecv_init);
-}
-
-static void __attribute__((destructor)) libirecovery_deinitialize(void)
-{
-	thread_once(&deinit_once, _irecv_deinit);
-}
+#warning No compiler support for constructor/destructor attributes, some features might not be available.
 #endif
 
 #ifdef HAVE_IOKIT
-- 
cgit v1.1-32-gdbae