Implement usage

[apps/pfixtools.git] / common.c

diff --git a/common.c b/common.c
index eba691b..7553779 100644 (file)
--- a/common.c
+++ b/common.c
@@ -40,7 +40,6 @@
 
 #include "common.h"
 
-sig_atomic_t cleanexit = false;
 sig_atomic_t sigint    = false;
 sig_atomic_t sighup    = false;
 
@@ -205,22 +204,27 @@ int drop_privileges(const char *user, const char *group)
     return 0;
 }
 
-void common_initialize(void)
-{
-    extern initcall_t __madinit_start, __madinit_end;
+extern initcall_t __madinit[], __madexit[];
 
-    initcall_t *call_p = &__madinit_start;
-    while (call_p < &__madinit_end) {
-        (*call_p++)();
+static void common_shutdown(void)
+{
+    for (int i = -1; __madexit[i]; i--) {
+        (*__madexit[i])();
     }
 }
 
-void common_shutdown(void)
+static void __attribute__((__constructor__,__used__))
+common_initialize(void)
 {
-    extern exitcall_t __madexit_start, __madexit_end;
+    if (atexit(common_shutdown)) {
+        fputs("Cannot hook my atexit function, quitting !\n", stderr);
+        abort();
+    }
 
-    exitcall_t *call_p = &__madexit_end;
-    while (call_p > &__madexit_start) {
-        (*--call_p)();
+    for (int i = 0; __madinit[i]; i++) {
+        if ((*__madinit[i])()) {
+            exit(EXIT_FAILURE);
+        }
     }
 }
+