dlfcn_test.cpp revision f4cb6313645ef65cc0eea0a439e51b6788cd3439
1/* 2 * Copyright (C) 2012 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17#include <gtest/gtest.h> 18 19#include <dlfcn.h> 20#include <libgen.h> 21#include <limits.h> 22#include <stdio.h> 23#include <stdint.h> 24 25#include "private/ScopeGuard.h" 26 27#include <string> 28 29#define ASSERT_SUBSTR(needle, haystack) \ 30 ASSERT_PRED_FORMAT2(::testing::IsSubstring, needle, haystack) 31 32static bool g_called = false; 33extern "C" void DlSymTestFunction() { 34 g_called = true; 35} 36 37static int g_ctor_function_called = 0; 38 39extern "C" void ctor_function() __attribute__ ((constructor)); 40 41extern "C" void ctor_function() { 42 g_ctor_function_called = 17; 43} 44 45TEST(dlfcn, ctor_function_call) { 46 ASSERT_EQ(17, g_ctor_function_called); 47} 48 49TEST(dlfcn, dlsym_in_self) { 50 dlerror(); // Clear any pending errors. 51 void* self = dlopen(NULL, RTLD_NOW); 52 ASSERT_TRUE(self != NULL); 53 ASSERT_TRUE(dlerror() == NULL); 54 55 void* sym = dlsym(self, "DlSymTestFunction"); 56 ASSERT_TRUE(sym != NULL); 57 58 void (*function)() = reinterpret_cast<void(*)()>(sym); 59 60 g_called = false; 61 function(); 62 ASSERT_TRUE(g_called); 63 64 ASSERT_EQ(0, dlclose(self)); 65} 66 67TEST(dlfcn, dlsym_with_dependencies) { 68 void* handle = dlopen("libtest_with_dependency.so", RTLD_NOW); 69 ASSERT_TRUE(handle != NULL); 70 dlerror(); 71 // This symbol is in DT_NEEDED library. 72 void* sym = dlsym(handle, "getRandomNumber"); 73 ASSERT_TRUE(sym != NULL); 74 int (*fn)(void); 75 fn = reinterpret_cast<int (*)(void)>(sym); 76 EXPECT_EQ(4, fn()); 77 dlclose(handle); 78} 79 80TEST(dlfcn, dlopen_noload) { 81 void* handle = dlopen("libtest_simple.so", RTLD_NOW | RTLD_NOLOAD); 82 ASSERT_TRUE(handle == NULL); 83 handle = dlopen("libtest_simple.so", RTLD_NOW); 84 void* handle2 = dlopen("libtest_simple.so", RTLD_NOW | RTLD_NOLOAD); 85 ASSERT_TRUE(handle != NULL); 86 ASSERT_TRUE(handle2 != NULL); 87 ASSERT_TRUE(handle == handle2); 88 ASSERT_EQ(0, dlclose(handle)); 89 ASSERT_EQ(0, dlclose(handle2)); 90} 91 92// ifuncs are only supported on intel and arm64 for now 93#if defined (__aarch64__) || defined(__i386__) || defined(__x86_64__) 94TEST(dlfcn, ifunc) { 95 typedef const char* (*fn_ptr)(); 96 97 // ifunc's choice depends on whether IFUNC_CHOICE has a value 98 // first check the set case 99 setenv("IFUNC_CHOICE", "set", 1); 100 void* handle = dlopen("libtest_ifunc.so", RTLD_NOW); 101 ASSERT_TRUE(handle != NULL); 102 fn_ptr foo_ptr = reinterpret_cast<fn_ptr>(dlsym(handle, "foo")); 103 fn_ptr foo_library_ptr = reinterpret_cast<fn_ptr>(dlsym(handle, "foo_library")); 104 ASSERT_TRUE(foo_ptr != NULL); 105 ASSERT_TRUE(foo_library_ptr != NULL); 106 ASSERT_EQ(strncmp("set", foo_ptr(), 3), 0); 107 ASSERT_EQ(strncmp("set", foo_library_ptr(), 3), 0); 108 dlclose(handle); 109 110 // then check the unset case 111 unsetenv("IFUNC_CHOICE"); 112 handle = dlopen("libtest_ifunc.so", RTLD_NOW); 113 ASSERT_TRUE(handle != NULL); 114 foo_ptr = reinterpret_cast<fn_ptr>(dlsym(handle, "foo")); 115 foo_library_ptr = reinterpret_cast<fn_ptr>(dlsym(handle, "foo_library")); 116 ASSERT_TRUE(foo_ptr != NULL); 117 ASSERT_TRUE(foo_library_ptr != NULL); 118 ASSERT_EQ(strncmp("unset", foo_ptr(), 5), 0); 119 ASSERT_EQ(strncmp("unset", foo_library_ptr(), 3), 0); 120 dlclose(handle); 121} 122 123TEST(dlfcn, ifunc_ctor_call) { 124 typedef const char* (*fn_ptr)(); 125 126 void* handle = dlopen("libtest_ifunc.so", RTLD_NOW); 127 ASSERT_TRUE(handle != nullptr) << dlerror(); 128 fn_ptr is_ctor_called = reinterpret_cast<fn_ptr>(dlsym(handle, "is_ctor_called_irelative")); 129 ASSERT_TRUE(is_ctor_called != nullptr) << dlerror(); 130 ASSERT_STREQ("false", is_ctor_called()); 131 132 is_ctor_called = reinterpret_cast<fn_ptr>(dlsym(handle, "is_ctor_called_jump_slot")); 133 ASSERT_TRUE(is_ctor_called != nullptr) << dlerror(); 134 ASSERT_STREQ("true", is_ctor_called()); 135 dlclose(handle); 136} 137#endif 138 139TEST(dlfcn, dlopen_check_relocation_dt_needed_order) { 140 // This is the structure of the test library and 141 // its dt_needed libraries 142 // libtest_relo_check_dt_needed_order.so 143 // | 144 // +-> libtest_relo_check_dt_needed_order_1.so 145 // | 146 // +-> libtest_relo_check_dt_needed_order_2.so 147 // 148 // The root library references relo_test_get_answer_lib - which is defined 149 // in both dt_needed libraries, the correct relocation should 150 // use the function defined in libtest_relo_check_dt_needed_order_1.so 151 void* handle = nullptr; 152 auto guard = make_scope_guard([&]() { 153 dlclose(handle); 154 }); 155 156 handle = dlopen("libtest_relo_check_dt_needed_order.so", RTLD_NOW); 157 ASSERT_TRUE(handle != nullptr) << dlerror(); 158 159 typedef int (*fn_t) (void); 160 fn_t fn = reinterpret_cast<fn_t>(dlsym(handle, "relo_test_get_answer")); 161 ASSERT_TRUE(fn != nullptr) << dlerror(); 162 ASSERT_EQ(1, fn()); 163} 164 165TEST(dlfcn, dlopen_check_order) { 166 // Here is how the test library and its dt_needed 167 // libraries are arranged 168 // 169 // libtest_check_order.so 170 // | 171 // +-> libtest_check_order_1_left.so 172 // | | 173 // | +-> libtest_check_order_a.so 174 // | | 175 // | +-> libtest_check_order_b.so 176 // | 177 // +-> libtest_check_order_2_right.so 178 // | | 179 // | +-> libtest_check_order_d.so 180 // | | 181 // | +-> libtest_check_order_b.so 182 // | 183 // +-> libtest_check_order_3_c.so 184 // 185 // load order should be (1, 2, 3, a, b, d) 186 // 187 // get_answer() is defined in (2, 3, a, b, c) 188 // get_answer2() is defined in (b, d) 189 void* sym = dlsym(RTLD_DEFAULT, "dlopen_test_get_answer"); 190 ASSERT_TRUE(sym == nullptr); 191 void* handle = dlopen("libtest_check_order.so", RTLD_NOW); 192 ASSERT_TRUE(handle != nullptr); 193 typedef int (*fn_t) (void); 194 fn_t fn, fn2; 195 fn = reinterpret_cast<fn_t>(dlsym(RTLD_DEFAULT, "dlopen_test_get_answer")); 196 ASSERT_TRUE(fn != NULL); 197 fn2 = reinterpret_cast<fn_t>(dlsym(RTLD_DEFAULT, "dlopen_test_get_answer2")); 198 ASSERT_TRUE(fn2 != NULL); 199 200 ASSERT_EQ(42, fn()); 201 ASSERT_EQ(43, fn2()); 202 dlclose(handle); 203} 204 205// libtest_with_dependency_loop.so -> libtest_with_dependency_loop_a.so -> 206// libtest_with_dependency_loop_b.so -> libtest_with_dependency_loop_c.so -> 207// libtest_with_dependency_loop_a.so 208TEST(dlfcn, dlopen_check_loop) { 209 void* handle = dlopen("libtest_with_dependency_loop.so", RTLD_NOW); 210 ASSERT_TRUE(handle == NULL); 211 ASSERT_STREQ("dlopen failed: recursive link to \"libtest_with_dependency_loop_a.so\"", dlerror()); 212} 213 214TEST(dlfcn, dlopen_failure) { 215 void* self = dlopen("/does/not/exist", RTLD_NOW); 216 ASSERT_TRUE(self == NULL); 217#if defined(__BIONIC__) 218 ASSERT_STREQ("dlopen failed: library \"/does/not/exist\" not found", dlerror()); 219#else 220 ASSERT_STREQ("/does/not/exist: cannot open shared object file: No such file or directory", dlerror()); 221#endif 222} 223 224static void* ConcurrentDlErrorFn(void*) { 225 dlopen("/child/thread", RTLD_NOW); 226 return reinterpret_cast<void*>(strdup(dlerror())); 227} 228 229TEST(dlfcn, dlerror_concurrent) { 230 dlopen("/main/thread", RTLD_NOW); 231 const char* main_thread_error = dlerror(); 232 ASSERT_SUBSTR("/main/thread", main_thread_error); 233 234 pthread_t t; 235 ASSERT_EQ(0, pthread_create(&t, NULL, ConcurrentDlErrorFn, NULL)); 236 void* result; 237 ASSERT_EQ(0, pthread_join(t, &result)); 238 char* child_thread_error = static_cast<char*>(result); 239 ASSERT_SUBSTR("/child/thread", child_thread_error); 240 free(child_thread_error); 241 242 ASSERT_SUBSTR("/main/thread", main_thread_error); 243} 244 245TEST(dlfcn, dlsym_failures) { 246 dlerror(); // Clear any pending errors. 247 void* self = dlopen(NULL, RTLD_NOW); 248 ASSERT_TRUE(self != NULL); 249 ASSERT_TRUE(dlerror() == NULL); 250 251 void* sym; 252 253#if defined(__BIONIC__) && !defined(__LP64__) 254 // RTLD_DEFAULT in lp32 bionic is not (void*)0 255 // so it can be distinguished from the NULL handle. 256 sym = dlsym(NULL, "test"); 257 ASSERT_TRUE(sym == NULL); 258 ASSERT_SUBSTR("dlsym library handle is null", dlerror()); 259#endif 260 261 // NULL symbol name. 262#if defined(__BIONIC__) 263 // glibc marks this parameter non-null and SEGVs if you cheat. 264 sym = dlsym(self, NULL); 265 ASSERT_TRUE(sym == NULL); 266 ASSERT_SUBSTR("", dlerror()); 267#endif 268 269 // Symbol that doesn't exist. 270 sym = dlsym(self, "ThisSymbolDoesNotExist"); 271 ASSERT_TRUE(sym == NULL); 272 ASSERT_SUBSTR("undefined symbol: ThisSymbolDoesNotExist", dlerror()); 273 274 ASSERT_EQ(0, dlclose(self)); 275} 276 277TEST(dlfcn, dladdr) { 278 dlerror(); // Clear any pending errors. 279 void* self = dlopen(NULL, RTLD_NOW); 280 ASSERT_TRUE(self != NULL); 281 ASSERT_TRUE(dlerror() == NULL); 282 283 void* sym = dlsym(self, "DlSymTestFunction"); 284 ASSERT_TRUE(sym != NULL); 285 286 // Deliberately ask dladdr for an address inside a symbol, rather than the symbol base address. 287 void* addr = reinterpret_cast<void*>(reinterpret_cast<uintptr_t>(sym) + 2); 288 289 Dl_info info; 290 int rc = dladdr(addr, &info); 291 ASSERT_NE(rc, 0); // Zero on error, non-zero on success. 292 293 // Get the name of this executable. 294 char executable_path[PATH_MAX]; 295 rc = readlink("/proc/self/exe", executable_path, sizeof(executable_path)); 296 ASSERT_NE(rc, -1); 297 executable_path[rc] = '\0'; 298 std::string executable_name(basename(executable_path)); 299 300 // The filename should be that of this executable. 301 // Note that we don't know whether or not we have the full path, so we want an "ends_with" test. 302 std::string dli_fname(info.dli_fname); 303 dli_fname = basename(&dli_fname[0]); 304 ASSERT_EQ(dli_fname, executable_name); 305 306 // The symbol name should be the symbol we looked up. 307 ASSERT_STREQ(info.dli_sname, "DlSymTestFunction"); 308 309 // The address should be the exact address of the symbol. 310 ASSERT_EQ(info.dli_saddr, sym); 311 312 // Look in /proc/pid/maps to find out what address we were loaded at. 313 // TODO: factor /proc/pid/maps parsing out into a class and reuse all over bionic. 314 void* base_address = NULL; 315 char line[BUFSIZ]; 316 FILE* fp = fopen("/proc/self/maps", "r"); 317 ASSERT_TRUE(fp != NULL); 318 while (fgets(line, sizeof(line), fp) != NULL) { 319 uintptr_t start = strtoul(line, 0, 16); 320 line[strlen(line) - 1] = '\0'; // Chomp the '\n'. 321 char* path = strchr(line, '/'); 322 if (path != NULL && strcmp(executable_path, path) == 0) { 323 base_address = reinterpret_cast<void*>(start); 324 break; 325 } 326 } 327 fclose(fp); 328 329 // The base address should be the address we were loaded at. 330 ASSERT_EQ(info.dli_fbase, base_address); 331 332 ASSERT_EQ(0, dlclose(self)); 333} 334 335TEST(dlfcn, dladdr_invalid) { 336 Dl_info info; 337 338 dlerror(); // Clear any pending errors. 339 340 // No symbol corresponding to NULL. 341 ASSERT_EQ(dladdr(NULL, &info), 0); // Zero on error, non-zero on success. 342 ASSERT_TRUE(dlerror() == NULL); // dladdr(3) doesn't set dlerror(3). 343 344 // No symbol corresponding to a stack address. 345 ASSERT_EQ(dladdr(&info, &info), 0); // Zero on error, non-zero on success. 346 ASSERT_TRUE(dlerror() == NULL); // dladdr(3) doesn't set dlerror(3). 347} 348 349// Our dynamic linker doesn't support GNU hash tables. 350#if defined(__BIONIC__) 351// GNU-style ELF hash tables are incompatible with the MIPS ABI. 352// MIPS requires .dynsym to be sorted to match the GOT but GNU-style requires sorting by hash code. 353#if !defined(__mips__) 354TEST(dlfcn, dlopen_library_with_only_gnu_hash) { 355 dlerror(); // Clear any pending errors. 356 void* handle = dlopen("no-elf-hash-table-library.so", RTLD_NOW); 357 ASSERT_TRUE(handle == NULL); 358 ASSERT_STREQ("dlopen failed: empty/missing DT_HASH in \"no-elf-hash-table-library.so\" (built with --hash-style=gnu?)", dlerror()); 359} 360#endif 361#endif 362 363TEST(dlfcn, dlopen_bad_flags) { 364 dlerror(); // Clear any pending errors. 365 void* handle; 366 367#if defined(__GLIBC__) 368 // glibc was smart enough not to define RTLD_NOW as 0, so it can detect missing flags. 369 handle = dlopen(NULL, 0); 370 ASSERT_TRUE(handle == NULL); 371 ASSERT_SUBSTR("invalid", dlerror()); 372#endif 373 374 handle = dlopen(NULL, 0xffffffff); 375 ASSERT_TRUE(handle == NULL); 376 ASSERT_SUBSTR("invalid", dlerror()); 377 378 // glibc actually allows you to choose both RTLD_NOW and RTLD_LAZY at the same time, and so do we. 379 handle = dlopen(NULL, RTLD_NOW|RTLD_LAZY); 380 ASSERT_TRUE(handle != NULL); 381 ASSERT_SUBSTR(NULL, dlerror()); 382} 383 384TEST(dlfcn, rtld_default_unknown_symbol) { 385 void* addr = dlsym(RTLD_DEFAULT, "ANY_UNKNOWN_SYMBOL_NAME"); 386 ASSERT_TRUE(addr == NULL); 387} 388 389TEST(dlfcn, rtld_default_known_symbol) { 390 void* addr = dlsym(RTLD_DEFAULT, "fopen"); 391 ASSERT_TRUE(addr != NULL); 392} 393 394TEST(dlfcn, rtld_next_unknown_symbol) { 395 void* addr = dlsym(RTLD_NEXT, "ANY_UNKNOWN_SYMBOL_NAME"); 396 ASSERT_TRUE(addr == NULL); 397} 398 399TEST(dlfcn, rtld_next_known_symbol) { 400 void* addr = dlsym(RTLD_NEXT, "fopen"); 401 ASSERT_TRUE(addr != NULL); 402} 403 404TEST(dlfcn, dlsym_weak_func) { 405 dlerror(); 406 void* handle = dlopen("libtest_dlsym_weak_func.so",RTLD_NOW); 407 ASSERT_TRUE(handle != NULL); 408 409 int (*weak_func)(); 410 weak_func = reinterpret_cast<int (*)()>(dlsym(handle, "weak_func")); 411 ASSERT_TRUE(weak_func != NULL) << "dlerror: " << dlerror(); 412 EXPECT_EQ(42, weak_func()); 413 dlclose(handle); 414} 415 416TEST(dlfcn, dlopen_symlink) { 417 void* handle1 = dlopen("libdlext_test.so", RTLD_NOW); 418 void* handle2 = dlopen("libdlext_test_v2.so", RTLD_NOW); 419 ASSERT_TRUE(handle1 != NULL); 420 ASSERT_TRUE(handle2 != NULL); 421 ASSERT_EQ(handle1, handle2); 422} 423