1 /*- 2 * Copyright (c) 2011 The NetBSD Foundation, Inc. 3 * All rights reserved. 4 * 5 * This code is derived from software contributed to The NetBSD Foundation 6 * by Joerg Sonnenberger. 7 * 8 * Redistribution and use in source and binary forms, with or without 9 * modification, are permitted provided that the following conditions 10 * are met: 11 * 1. Redistributions of source code must retain the above copyright 12 * notice, this list of conditions and the following disclaimer. 13 * 2. Redistributions in binary form must reproduce the above copyright 14 * notice, this list of conditions and the following disclaimer in the 15 * documentation and/or other materials provided with the distribution. 16 * 17 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 18 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 19 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 20 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 21 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 22 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 23 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 24 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 25 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 26 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 27 * POSSIBILITY OF SUCH DAMAGE. 28 */ 29 30 #include <dlfcn.h> 31 #include <link_elf.h> 32 #include <pthread.h> 33 #include <stdlib.h> 34 #include <stdio.h> 35 #include <string.h> 36 #include <unistd.h> 37 38 int sleep_init; 39 int sleep_fini; 40 int dlopen_cookie; 41 int dlclose_cookie; 42 43 void (*tls_callback_sym)(void); 44 45 static int 46 dl_iterate_phdr_cb(struct dl_phdr_info *info, size_t size, void *data) 47 { 48 (*tls_callback_sym)(); 49 return 0; 50 } 51 52 static void * 53 test_dl_iterate_phdr_helper(void *dummy) 54 { 55 sleep(10); 56 _exit(1); 57 } 58 59 static void 60 test_dl_iterate_phdr(void) 61 { 62 pthread_t t; 63 void *dso; 64 sleep_init = 0; 65 sleep_fini = 0; 66 if ((dso = dlopen("libh_helper_dso2.so", RTLD_LAZY)) == NULL) { 67 fprintf(stderr, "opening helper failed\n"); 68 _exit(1); 69 } 70 tls_callback_sym = dlsym(dso, "tls_callback"); 71 if (tls_callback_sym == NULL) { 72 fprintf(stderr, "bad helper\n"); 73 _exit(1); 74 } 75 pthread_create(&t, NULL, test_dl_iterate_phdr_helper, NULL); 76 if (dl_iterate_phdr(dl_iterate_phdr_cb, NULL)) 77 _exit(1); 78 _exit(0); 79 } 80 81 static void * 82 init_fini_helper(void *arg) 83 { 84 void *dso; 85 if ((dso = dlopen(arg, RTLD_LAZY)) == NULL) { 86 fprintf(stderr, "opening %s failed\n", (char *)arg); 87 exit(1); 88 } 89 dlclose(dso); 90 return NULL; 91 } 92 93 static void 94 test_dlopen(void) 95 { 96 pthread_t t1, t2; 97 sleep_init = 1; 98 sleep_fini = 0; 99 printf("%d\n", dlopen_cookie); 100 pthread_create(&t1, NULL, init_fini_helper, 101 __UNCONST("libh_helper_dso2.so")); 102 sleep(1); 103 printf("%d\n", dlopen_cookie); 104 if (dlopen_cookie != 1) 105 _exit(1); 106 sleep(1); 107 pthread_create(&t2, NULL, init_fini_helper, 108 __UNCONST("libutil.so")); 109 printf("%d\n", dlopen_cookie); 110 if (dlopen_cookie != 1) 111 _exit(1); 112 _exit(0); 113 } 114 115 static void 116 test_dlclose(void) 117 { 118 pthread_t t1, t2; 119 sleep_init = 0; 120 sleep_fini = 1; 121 printf("%d\n", dlclose_cookie); 122 pthread_create(&t1, NULL, init_fini_helper, 123 __UNCONST("libh_helper_dso2.so")); 124 sleep(1); 125 printf("%d\n", dlclose_cookie); 126 if (dlclose_cookie != 2) 127 _exit(1); 128 pthread_create(&t2, NULL, init_fini_helper, 129 __UNCONST("libutil.so")); 130 sleep(1); 131 printf("%d\n", dlclose_cookie); 132 if (dlclose_cookie != 2) 133 _exit(1); 134 _exit(0); 135 } 136 137 int 138 main(int argc, char **argv) 139 { 140 if (argc != 2) 141 return 1; 142 if (strcmp(argv[1], "dl_iterate_phdr") == 0) 143 test_dl_iterate_phdr(); 144 if (strcmp(argv[1], "dlopen") == 0) 145 test_dlopen(); 146 if (strcmp(argv[1], "dlclose") == 0) 147 test_dlclose(); 148 return 1; 149 } 150