1 /*
2 * Copyright (C) 2024 The Android Open Source Project
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * * Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in
12 * the documentation and/or other materials provided with the
13 * distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
18 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
19 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
21 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
22 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
23 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
24 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
25 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29 #pragma once
30
31 #include "elf_max_page_size.h"
32 #include "gtest_globals.h"
33
34 #include <android-base/stringprintf.h>
35
36 #include <string>
37
38 #include <dlfcn.h>
39 #include <gtest/gtest.h>
40 #include <unistd.h>
41
OpenTestLibrary(std::string lib,bool expect_fail,void ** handle)42 static inline void OpenTestLibrary(std::string lib, bool expect_fail, void** handle) {
43 void* _handle = dlopen(lib.c_str(), RTLD_NODELETE);
44 const char* dlopen_error = dlerror();
45
46 if (expect_fail) {
47 ASSERT_EQ(_handle, nullptr);
48
49 const std::string expected_error = android::base::StringPrintf(
50 "dlopen failed: \"%s\" program alignment (%d) cannot be smaller than system page size (%d)",
51 lib.c_str(), 4096, getpagesize());
52
53 ASSERT_EQ(expected_error, dlopen_error);
54 } else {
55 ASSERT_NE(_handle, nullptr) << "Failed to dlopen shared library \"" << lib
56 << "\": " << dlopen_error;
57 }
58
59 *handle = _handle;
60 }
61
CallTestFunction(void * handle)62 static inline void CallTestFunction(void* handle) {
63 loader_test_func_t loader_test_func = (loader_test_func_t)dlsym(handle, "loader_test_func");
64 const char* dlsym_error = dlerror();
65
66 ASSERT_EQ(dlsym_error, nullptr) << "Failed to locate symbol \"loader_test_func\": "
67 << dlsym_error;
68
69 int res = loader_test_func();
70 ASSERT_EQ(res, TEST_RESULT_BASE + TEST_RESULT_INCREMENT);
71
72 // Call loader_test_func() twice to ensure we can modify writeable data and bss data
73 res = loader_test_func();
74 ASSERT_EQ(res, TEST_RESULT_BASE + (2 * TEST_RESULT_INCREMENT));
75 }
76