1*a4527715Sguenther /* $OpenBSD: atexit_test.c,v 1.1 2014/11/23 08:46:49 guenther Exp $ */
2*a4527715Sguenther
3*a4527715Sguenther /*
4*a4527715Sguenther * Copyright (c) 2014 Philip Guenther <guenther@openbsd.org>
5*a4527715Sguenther *
6*a4527715Sguenther * Permission to use, copy, modify, and distribute this software for any
7*a4527715Sguenther * purpose with or without fee is hereby granted, provided that the above
8*a4527715Sguenther * copyright notice and this permission notice appear in all copies.
9*a4527715Sguenther *
10*a4527715Sguenther * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11*a4527715Sguenther * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12*a4527715Sguenther * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13*a4527715Sguenther * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14*a4527715Sguenther * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15*a4527715Sguenther * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16*a4527715Sguenther * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17*a4527715Sguenther */
18*a4527715Sguenther
19*a4527715Sguenther #include <dlfcn.h>
20*a4527715Sguenther #include <err.h>
21*a4527715Sguenther #include <stdio.h>
22*a4527715Sguenther #include <stdlib.h>
23*a4527715Sguenther
24*a4527715Sguenther void *libaa, *libab;
25*a4527715Sguenther
26*a4527715Sguenther #define CALLBACK(name) static void name(void) { printf("exe "#name"\n"); }
27*a4527715Sguenther
28*a4527715Sguenther CALLBACK(cleanup1)
CALLBACK(cleanup2)29*a4527715Sguenther CALLBACK(cleanup2)
30*a4527715Sguenther CALLBACK(cleanup3)
31*a4527715Sguenther
32*a4527715Sguenther static void
33*a4527715Sguenther cleanup_dlclose(void)
34*a4527715Sguenther {
35*a4527715Sguenther printf("exe cleanup_dlclose begin\n");
36*a4527715Sguenther dlclose(libaa);
37*a4527715Sguenther dlclose(libab);
38*a4527715Sguenther printf("exe cleanup_dlclose end\n");
39*a4527715Sguenther }
40*a4527715Sguenther
41*a4527715Sguenther static void
aa(void)42*a4527715Sguenther aa(void)
43*a4527715Sguenther {
44*a4527715Sguenther void (*func)(void) = dlsym(libaa, "aa");
45*a4527715Sguenther if (func == NULL)
46*a4527715Sguenther errx(1, "dlsym(libaa, aa): %s", dlerror());
47*a4527715Sguenther func();
48*a4527715Sguenther }
49*a4527715Sguenther
50*a4527715Sguenther static void
ab(void)51*a4527715Sguenther ab(void)
52*a4527715Sguenther {
53*a4527715Sguenther void (*func)(void) = dlsym(libab, "ab");
54*a4527715Sguenther if (func == NULL)
55*a4527715Sguenther errx(1, "dlsym(libab, ab): %s", dlerror());
56*a4527715Sguenther func();
57*a4527715Sguenther }
58*a4527715Sguenther
59*a4527715Sguenther int
main(int argc,char ** argv)60*a4527715Sguenther main(int argc, char **argv)
61*a4527715Sguenther {
62*a4527715Sguenther int test;
63*a4527715Sguenther
64*a4527715Sguenther libaa = dlopen(LIBAA, RTLD_LAZY);
65*a4527715Sguenther if (libaa == NULL)
66*a4527715Sguenther errx(1, "dlopen(%s, RTLD_LAZY): %s", LIBAA, dlerror());
67*a4527715Sguenther
68*a4527715Sguenther libab = dlopen(LIBAB, RTLD_LAZY);
69*a4527715Sguenther if (libab == NULL)
70*a4527715Sguenther errx(1, "dlopen(%s, RTLD_LAZY): %s", LIBAB, dlerror());
71*a4527715Sguenther
72*a4527715Sguenther if (argc != 2)
73*a4527715Sguenther test = 0;
74*a4527715Sguenther else
75*a4527715Sguenther test = atoi(argv[1]);
76*a4527715Sguenther
77*a4527715Sguenther switch (test) {
78*a4527715Sguenther case 0:
79*a4527715Sguenther /* 1, aa, 2, ab, 3, then exit */
80*a4527715Sguenther atexit(cleanup1);
81*a4527715Sguenther aa();
82*a4527715Sguenther atexit(cleanup2);
83*a4527715Sguenther ab();
84*a4527715Sguenther atexit(cleanup3);
85*a4527715Sguenther exit(0);
86*a4527715Sguenther
87*a4527715Sguenther case 1:
88*a4527715Sguenther /* 1, aa, 2, ab, 3, then dlclose aa, then bb */
89*a4527715Sguenther atexit(cleanup1);
90*a4527715Sguenther aa();
91*a4527715Sguenther atexit(cleanup2);
92*a4527715Sguenther ab();
93*a4527715Sguenther atexit(cleanup3);
94*a4527715Sguenther dlclose(libaa);
95*a4527715Sguenther dlclose(libab);
96*a4527715Sguenther exit(0);
97*a4527715Sguenther
98*a4527715Sguenther case 2:
99*a4527715Sguenther /* 1, aa, cleanup_dlclose, ab, 3, then exit */
100*a4527715Sguenther atexit(cleanup1);
101*a4527715Sguenther aa();
102*a4527715Sguenther atexit(cleanup_dlclose);
103*a4527715Sguenther ab();
104*a4527715Sguenther atexit(cleanup3);
105*a4527715Sguenther exit(0);
106*a4527715Sguenther
107*a4527715Sguenther case 3:
108*a4527715Sguenther /* 1, aa, 2, ab, cleanup_dlclose, then exit */
109*a4527715Sguenther atexit(cleanup1);
110*a4527715Sguenther aa();
111*a4527715Sguenther atexit(cleanup2);
112*a4527715Sguenther ab();
113*a4527715Sguenther atexit(cleanup_dlclose);
114*a4527715Sguenther exit(0);
115*a4527715Sguenther
116*a4527715Sguenther }
117*a4527715Sguenther
118*a4527715Sguenther return (0);
119*a4527715Sguenther }
120