regress/lib/libc/atexit/atexit.c

   1 /*      $NetBSD: atexit.c,v 1.1 2003/03/01 04:20:54 thorpej Exp $       */
   2
   3 /*
   4  * Program to test atexit(3) and __cxa_atexit()/__cxa_finalize().
   5  *
   6  * Written by Jason R. Thorpe, February 2003.
   7  * Public domain.
   8  */
   9
  10 #include <stdio.h>
  11 #include <stdlib.h>
  12 #include <signal.h>
  13 #include <string.h>
  14 #include <unistd.h>
  15
  16 extern int __cxa_atexit(void (*func)(void *), void *, void *);
  17 extern void __cxa_finalize(void *);
  18
  19 static int dso_handle_1;
  20 static int dso_handle_2;
  21 static int dso_handle_3;
  22
  23 static int arg_1;
  24 static int arg_2;
  25 static int arg_3;
  26
  27 static int exiting_state;
  28
  29 #define ASSERT(expr)                                                    \
  30 do {                                                                    \
  31         if ((expr) == 0) {                                              \
  32                 write(STDERR_FILENO, __func__, strlen(__func__));       \
  33                 write(STDERR_FILENO, ": ", 2);                          \
  34                 write(STDERR_FILENO, __STRING(expr),                    \
  35                       strlen(__STRING(expr)));                          \
  36                 write(STDERR_FILENO, "\n", 1);                          \
  37                 my_abort();                                             \
  38         }                                                               \
  39 } while (/*CONSTCOND*/0)
  40
  41 #define SUCCESS()                                                       \
  42 do {                                                                    \
  43         write(STDOUT_FILENO, __func__, strlen(__func__));               \
  44         write(STDOUT_FILENO, "\n", 1);                                  \
  45 } while (/*CONSTCOND*/0)
  46
  47 static void
  48 my_abort(void)
  49 {
  50
  51         (void)raise(SIGABRT);
  52         /* NOTREACHED */
  53 }
  54
  55 static void
  56 cxa_handler_5(void *arg)
  57 {
  58         static int cxa_handler_5_called;
  59
  60         ASSERT(arg == (void *)&arg_1);
  61         ASSERT(cxa_handler_5_called == 0);
  62         ASSERT(exiting_state == 5);
  63
  64         exiting_state--;
  65         cxa_handler_5_called = 1;
  66         SUCCESS();
  67 }
  68
  69 static void
  70 cxa_handler_4(void *arg)
  71 {
  72         static int cxa_handler_4_called;
  73
  74         ASSERT(arg == (void *)&arg_1);
  75         ASSERT(cxa_handler_4_called == 0);
  76         ASSERT(exiting_state == 4);
  77
  78         exiting_state--;
  79         cxa_handler_4_called = 1;
  80         SUCCESS();
  81 }
  82
  83 static void
  84 cxa_handler_3(void *arg)
  85 {
  86         static int cxa_handler_3_called;
  87
  88         ASSERT(arg == (void *)&arg_2);
  89         ASSERT(cxa_handler_3_called == 0);
  90         ASSERT(exiting_state == 3);
  91
  92         exiting_state--;
  93         cxa_handler_3_called = 1;
  94         SUCCESS();
  95 }
  96
  97 static void
  98 cxa_handler_2(void *arg)
  99 {
 100         static int cxa_handler_2_called;
 101
 102         ASSERT(arg == (void *)&arg_3);
 103         ASSERT(cxa_handler_2_called == 0);
 104         ASSERT(exiting_state == 2);
 105
 106         exiting_state--;
 107         cxa_handler_2_called = 1;
 108         SUCCESS();
 109 }
 110
 111 static void
 112 normal_handler_1(void)
 113 {
 114         static int normal_handler_1_called;
 115
 116         ASSERT(normal_handler_1_called == 0);
 117         ASSERT(exiting_state == 1);
 118
 119         exiting_state--;
 120         normal_handler_1_called = 1;
 121         SUCCESS();
 122 }
 123
 124 static void
 125 normal_handler_0(void)
 126 {
 127         static int normal_handler_0_called;
 128
 129         ASSERT(normal_handler_0_called == 0);
 130         ASSERT(exiting_state == 0);
 131
 132         normal_handler_0_called = 1;
 133         SUCCESS();
 134 }
 135
 136 int
 137 main(int argc, char *argv[])
 138 {
 139
 140         exiting_state = 5;
 141
 142         ASSERT(0 == atexit(normal_handler_0));
 143         ASSERT(0 == atexit(normal_handler_1));
 144         ASSERT(0 == __cxa_atexit(cxa_handler_4, &arg_1, &dso_handle_1));
 145         ASSERT(0 == __cxa_atexit(cxa_handler_5, &arg_1, &dso_handle_1));
 146         ASSERT(0 == __cxa_atexit(cxa_handler_3, &arg_2, &dso_handle_2));
 147         ASSERT(0 == __cxa_atexit(cxa_handler_2, &arg_3, &dso_handle_3));
 148
 149         __cxa_finalize(&dso_handle_1);
 150         __cxa_finalize(&dso_handle_2);
 151         exit(0);
 152 }