xref: /aosp_15_r20/bionic/tests/dlext_test.cpp (revision 8d67ca893c1523eb926b9080dbe4e2ffd2a27ba1)
1*8d67ca89SAndroid Build Coastguard Worker /*
2*8d67ca89SAndroid Build Coastguard Worker  * Copyright (C) 2014 The Android Open Source Project
3*8d67ca89SAndroid Build Coastguard Worker  *
4*8d67ca89SAndroid Build Coastguard Worker  * Licensed under the Apache License, Version 2.0 (the "License");
5*8d67ca89SAndroid Build Coastguard Worker  * you may not use this file except in compliance with the License.
6*8d67ca89SAndroid Build Coastguard Worker  * You may obtain a copy of the License at
7*8d67ca89SAndroid Build Coastguard Worker  *
8*8d67ca89SAndroid Build Coastguard Worker  *      http://www.apache.org/licenses/LICENSE-2.0
9*8d67ca89SAndroid Build Coastguard Worker  *
10*8d67ca89SAndroid Build Coastguard Worker  * Unless required by applicable law or agreed to in writing, software
11*8d67ca89SAndroid Build Coastguard Worker  * distributed under the License is distributed on an "AS IS" BASIS,
12*8d67ca89SAndroid Build Coastguard Worker  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*8d67ca89SAndroid Build Coastguard Worker  * See the License for the specific language governing permissions and
14*8d67ca89SAndroid Build Coastguard Worker  * limitations under the License.
15*8d67ca89SAndroid Build Coastguard Worker  */
16*8d67ca89SAndroid Build Coastguard Worker 
17*8d67ca89SAndroid Build Coastguard Worker #include <gtest/gtest.h>
18*8d67ca89SAndroid Build Coastguard Worker 
19*8d67ca89SAndroid Build Coastguard Worker #include <dlfcn.h>
20*8d67ca89SAndroid Build Coastguard Worker #include <elf.h>
21*8d67ca89SAndroid Build Coastguard Worker #include <errno.h>
22*8d67ca89SAndroid Build Coastguard Worker #include <fcntl.h>
23*8d67ca89SAndroid Build Coastguard Worker #include <inttypes.h>
24*8d67ca89SAndroid Build Coastguard Worker #include <link.h>
25*8d67ca89SAndroid Build Coastguard Worker #include <stdio.h>
26*8d67ca89SAndroid Build Coastguard Worker #include <string.h>
27*8d67ca89SAndroid Build Coastguard Worker #include <unistd.h>
28*8d67ca89SAndroid Build Coastguard Worker 
29*8d67ca89SAndroid Build Coastguard Worker #include <android/dlext.h>
30*8d67ca89SAndroid Build Coastguard Worker #include <android-base/file.h>
31*8d67ca89SAndroid Build Coastguard Worker #include <android-base/strings.h>
32*8d67ca89SAndroid Build Coastguard Worker #include <android-base/test_utils.h>
33*8d67ca89SAndroid Build Coastguard Worker 
34*8d67ca89SAndroid Build Coastguard Worker #include <sys/mman.h>
35*8d67ca89SAndroid Build Coastguard Worker #include <sys/stat.h>
36*8d67ca89SAndroid Build Coastguard Worker #include <sys/types.h>
37*8d67ca89SAndroid Build Coastguard Worker #include <sys/vfs.h>
38*8d67ca89SAndroid Build Coastguard Worker #include <sys/wait.h>
39*8d67ca89SAndroid Build Coastguard Worker 
40*8d67ca89SAndroid Build Coastguard Worker #include <meminfo/procmeminfo.h>
41*8d67ca89SAndroid Build Coastguard Worker #include <procinfo/process_map.h>
42*8d67ca89SAndroid Build Coastguard Worker #include <ziparchive/zip_archive.h>
43*8d67ca89SAndroid Build Coastguard Worker 
44*8d67ca89SAndroid Build Coastguard Worker #include "bionic/mte.h"
45*8d67ca89SAndroid Build Coastguard Worker #include "bionic/page.h"
46*8d67ca89SAndroid Build Coastguard Worker #include "core_shared_libs.h"
47*8d67ca89SAndroid Build Coastguard Worker #include "dlext_private.h"
48*8d67ca89SAndroid Build Coastguard Worker #include "dlfcn_symlink_support.h"
49*8d67ca89SAndroid Build Coastguard Worker #include "gtest_globals.h"
50*8d67ca89SAndroid Build Coastguard Worker #include "utils.h"
51*8d67ca89SAndroid Build Coastguard Worker 
52*8d67ca89SAndroid Build Coastguard Worker #define ASSERT_DL_NOTNULL(ptr) \
53*8d67ca89SAndroid Build Coastguard Worker     ASSERT_TRUE((ptr) != nullptr) << "dlerror: " << dlerror()
54*8d67ca89SAndroid Build Coastguard Worker 
55*8d67ca89SAndroid Build Coastguard Worker #define ASSERT_DL_ZERO(i) \
56*8d67ca89SAndroid Build Coastguard Worker     ASSERT_EQ(0, i) << "dlerror: " << dlerror()
57*8d67ca89SAndroid Build Coastguard Worker 
58*8d67ca89SAndroid Build Coastguard Worker #define ASSERT_NOERROR(i) \
59*8d67ca89SAndroid Build Coastguard Worker     ASSERT_NE(-1, i) << "errno: " << strerror(errno)
60*8d67ca89SAndroid Build Coastguard Worker 
61*8d67ca89SAndroid Build Coastguard Worker #define ASSERT_SUBSTR(needle, haystack) \
62*8d67ca89SAndroid Build Coastguard Worker     ASSERT_PRED_FORMAT2(::testing::IsSubstring, needle, haystack)
63*8d67ca89SAndroid Build Coastguard Worker 
64*8d67ca89SAndroid Build Coastguard Worker 
65*8d67ca89SAndroid Build Coastguard Worker typedef int (*fn)(void);
66*8d67ca89SAndroid Build Coastguard Worker constexpr const char* kLibName = "libdlext_test.so";
67*8d67ca89SAndroid Build Coastguard Worker constexpr const char* kLibNameRecursive = "libdlext_test_recursive.so";
68*8d67ca89SAndroid Build Coastguard Worker constexpr const char* kLibNameNoRelro = "libdlext_test_norelro.so";
69*8d67ca89SAndroid Build Coastguard Worker constexpr const char* kLibZipSimpleZip = "libdir/libatest_simple_zip.so";
70*8d67ca89SAndroid Build Coastguard Worker constexpr auto kLibSize = 1024 * 1024; // how much address space to reserve for it
71*8d67ca89SAndroid Build Coastguard Worker 
72*8d67ca89SAndroid Build Coastguard Worker class DlExtTest : public ::testing::Test {
73*8d67ca89SAndroid Build Coastguard Worker protected:
SetUp()74*8d67ca89SAndroid Build Coastguard Worker   void SetUp() override {
75*8d67ca89SAndroid Build Coastguard Worker     handle_ = nullptr;
76*8d67ca89SAndroid Build Coastguard Worker     // verify that we don't have the library loaded already
77*8d67ca89SAndroid Build Coastguard Worker     void* h = dlopen(kLibName, RTLD_NOW | RTLD_NOLOAD);
78*8d67ca89SAndroid Build Coastguard Worker     ASSERT_TRUE(h == nullptr);
79*8d67ca89SAndroid Build Coastguard Worker     h = dlopen(kLibNameNoRelro, RTLD_NOW | RTLD_NOLOAD);
80*8d67ca89SAndroid Build Coastguard Worker     ASSERT_TRUE(h == nullptr);
81*8d67ca89SAndroid Build Coastguard Worker     // call dlerror() to swallow the error, and check it was the one we wanted
82*8d67ca89SAndroid Build Coastguard Worker     ASSERT_EQ(std::string("dlopen failed: library \"") + kLibNameNoRelro + "\" wasn't loaded and RTLD_NOLOAD prevented it", dlerror());
83*8d67ca89SAndroid Build Coastguard Worker   }
84*8d67ca89SAndroid Build Coastguard Worker 
TearDown()85*8d67ca89SAndroid Build Coastguard Worker   void TearDown() override {
86*8d67ca89SAndroid Build Coastguard Worker     if (handle_ != nullptr) {
87*8d67ca89SAndroid Build Coastguard Worker       ASSERT_DL_ZERO(dlclose(handle_));
88*8d67ca89SAndroid Build Coastguard Worker     }
89*8d67ca89SAndroid Build Coastguard Worker   }
90*8d67ca89SAndroid Build Coastguard Worker 
91*8d67ca89SAndroid Build Coastguard Worker   void* handle_;
92*8d67ca89SAndroid Build Coastguard Worker   const size_t kPageSize = getpagesize();
93*8d67ca89SAndroid Build Coastguard Worker };
94*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtTest,ExtInfoNull)95*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtTest, ExtInfoNull) {
96*8d67ca89SAndroid Build Coastguard Worker   handle_ = android_dlopen_ext(kLibName, RTLD_NOW, nullptr);
97*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(handle_);
98*8d67ca89SAndroid Build Coastguard Worker   fn f = reinterpret_cast<fn>(dlsym(handle_, "getRandomNumber"));
99*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(f);
100*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(4, f());
101*8d67ca89SAndroid Build Coastguard Worker }
102*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtTest,ExtInfoNoFlags)103*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtTest, ExtInfoNoFlags) {
104*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
105*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = 0;
106*8d67ca89SAndroid Build Coastguard Worker   handle_ = android_dlopen_ext(kLibName, RTLD_NOW, &extinfo);
107*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(handle_);
108*8d67ca89SAndroid Build Coastguard Worker   fn f = reinterpret_cast<fn>(dlsym(handle_, "getRandomNumber"));
109*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(f);
110*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(4, f());
111*8d67ca89SAndroid Build Coastguard Worker }
112*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtTest,ExtInfoUseFd)113*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtTest, ExtInfoUseFd) {
114*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_path = GetTestLibRoot() + "/libdlext_test_fd/libdlext_test_fd.so";
115*8d67ca89SAndroid Build Coastguard Worker 
116*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
117*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_LIBRARY_FD;
118*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_fd = TEMP_FAILURE_RETRY(open(lib_path.c_str(), O_RDONLY | O_CLOEXEC));
119*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(extinfo.library_fd != -1);
120*8d67ca89SAndroid Build Coastguard Worker   handle_ = android_dlopen_ext(lib_path.c_str(), RTLD_NOW, &extinfo);
121*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(handle_);
122*8d67ca89SAndroid Build Coastguard Worker   fn f = reinterpret_cast<fn>(dlsym(handle_, "getRandomNumber"));
123*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(f);
124*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(4, f());
125*8d67ca89SAndroid Build Coastguard Worker 
126*8d67ca89SAndroid Build Coastguard Worker   uint32_t* taxicab_number = reinterpret_cast<uint32_t*>(dlsym(handle_, "dlopen_testlib_taxicab_number"));
127*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(taxicab_number);
128*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(1729U, *taxicab_number);
129*8d67ca89SAndroid Build Coastguard Worker }
130*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtTest,ExtInfoUseFdWithOffset)131*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtTest, ExtInfoUseFdWithOffset) {
132*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_path = GetTestLibRoot() + "/libdlext_test_zip/libdlext_test_zip_zipaligned.zip";
133*8d67ca89SAndroid Build Coastguard Worker 
134*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
135*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_LIBRARY_FD | ANDROID_DLEXT_USE_LIBRARY_FD_OFFSET;
136*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_fd = TEMP_FAILURE_RETRY(open(lib_path.c_str(), O_RDONLY | O_CLOEXEC));
137*8d67ca89SAndroid Build Coastguard Worker 
138*8d67ca89SAndroid Build Coastguard Worker   // Find the offset of the shared library in the zip.
139*8d67ca89SAndroid Build Coastguard Worker   ZipArchiveHandle handle;
140*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(0, OpenArchive(lib_path.c_str(), &handle));
141*8d67ca89SAndroid Build Coastguard Worker   ZipEntry zip_entry;
142*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(0, FindEntry(handle, kLibZipSimpleZip, &zip_entry));
143*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_fd_offset = zip_entry.offset;
144*8d67ca89SAndroid Build Coastguard Worker   CloseArchive(handle);
145*8d67ca89SAndroid Build Coastguard Worker 
146*8d67ca89SAndroid Build Coastguard Worker   handle_ = android_dlopen_ext(lib_path.c_str(), RTLD_NOW, &extinfo);
147*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(handle_);
148*8d67ca89SAndroid Build Coastguard Worker 
149*8d67ca89SAndroid Build Coastguard Worker   uint32_t* taxicab_number = reinterpret_cast<uint32_t*>(dlsym(handle_, "dlopen_testlib_taxicab_number"));
150*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(taxicab_number);
151*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(1729U, *taxicab_number);
152*8d67ca89SAndroid Build Coastguard Worker }
153*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtTest,ExtInfoUseFdWithInvalidOffset)154*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtTest, ExtInfoUseFdWithInvalidOffset) {
155*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_path = GetTestLibRoot() + "/libdlext_test_zip/libdlext_test_zip_zipaligned.zip";
156*8d67ca89SAndroid Build Coastguard Worker 
157*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
158*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_LIBRARY_FD | ANDROID_DLEXT_USE_LIBRARY_FD_OFFSET;
159*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_fd = TEMP_FAILURE_RETRY(open(lib_path.c_str(), O_RDONLY | O_CLOEXEC));
160*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_fd_offset = 17;
161*8d67ca89SAndroid Build Coastguard Worker 
162*8d67ca89SAndroid Build Coastguard Worker   handle_ = android_dlopen_ext("libname_placeholder", RTLD_NOW, &extinfo);
163*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_ == nullptr);
164*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("dlopen failed: file offset for the library \"libname_placeholder\" is not page-aligned: 17", dlerror());
165*8d67ca89SAndroid Build Coastguard Worker 
166*8d67ca89SAndroid Build Coastguard Worker   // Test an address above 2^44, for http://b/18178121 .
167*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_fd_offset = (5LL << 48) + kPageSize;
168*8d67ca89SAndroid Build Coastguard Worker   handle_ = android_dlopen_ext("libname_placeholder", RTLD_NOW, &extinfo);
169*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_ == nullptr);
170*8d67ca89SAndroid Build Coastguard Worker   ASSERT_SUBSTR("dlopen failed: file offset for the library \"libname_placeholder\" >= file size", dlerror());
171*8d67ca89SAndroid Build Coastguard Worker 
172*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_fd_offset = 0LL - kPageSize;
173*8d67ca89SAndroid Build Coastguard Worker   handle_ = android_dlopen_ext("libname_placeholder", RTLD_NOW, &extinfo);
174*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_ == nullptr);
175*8d67ca89SAndroid Build Coastguard Worker   ASSERT_SUBSTR("dlopen failed: file offset for the library \"libname_placeholder\" is negative", dlerror());
176*8d67ca89SAndroid Build Coastguard Worker 
177*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_fd_offset = 0;
178*8d67ca89SAndroid Build Coastguard Worker   handle_ = android_dlopen_ext("libname_ignored", RTLD_NOW, &extinfo);
179*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_ == nullptr);
180*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ("dlopen failed: \"" + lib_path + "\" has bad ELF magic: 504b0304", dlerror());
181*8d67ca89SAndroid Build Coastguard Worker 
182*8d67ca89SAndroid Build Coastguard Worker   // Check if dlsym works after unsuccessful dlopen().
183*8d67ca89SAndroid Build Coastguard Worker   // Supply non-exiting one to make linker visit every soinfo.
184*8d67ca89SAndroid Build Coastguard Worker   void* sym = dlsym(RTLD_DEFAULT, "this_symbol_does_not_exist___");
185*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(sym == nullptr);
186*8d67ca89SAndroid Build Coastguard Worker 
187*8d67ca89SAndroid Build Coastguard Worker   close(extinfo.library_fd);
188*8d67ca89SAndroid Build Coastguard Worker }
189*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtTest,ExtInfoUseOffsetWithoutFd)190*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtTest, ExtInfoUseOffsetWithoutFd) {
191*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
192*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_LIBRARY_FD_OFFSET;
193*8d67ca89SAndroid Build Coastguard Worker   // This offset will not be used, so it doesn't matter.
194*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_fd_offset = 0;
195*8d67ca89SAndroid Build Coastguard Worker 
196*8d67ca89SAndroid Build Coastguard Worker   handle_ = android_dlopen_ext("/some/lib/that/does_not_exist", RTLD_NOW, &extinfo);
197*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_ == nullptr);
198*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("dlopen failed: invalid extended flag combination (ANDROID_DLEXT_USE_LIBRARY_FD_OFFSET without ANDROID_DLEXT_USE_LIBRARY_FD): 0x20", dlerror());
199*8d67ca89SAndroid Build Coastguard Worker }
200*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,android_dlopen_ext_force_load_smoke)201*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, android_dlopen_ext_force_load_smoke) {
202*8d67ca89SAndroid Build Coastguard Worker   DlfcnSymlink symlink("android_dlopen_ext_force_load_smoke");
203*8d67ca89SAndroid Build Coastguard Worker   const std::string symlink_name = basename(symlink.get_symlink_path().c_str());
204*8d67ca89SAndroid Build Coastguard Worker   // 1. Open actual file
205*8d67ca89SAndroid Build Coastguard Worker   void* handle = dlopen("libdlext_test.so", RTLD_NOW);
206*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(handle);
207*8d67ca89SAndroid Build Coastguard Worker   // 2. Open link with force_load flag set
208*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
209*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_FORCE_LOAD;
210*8d67ca89SAndroid Build Coastguard Worker   void* handle2 = android_dlopen_ext(symlink_name.c_str(), RTLD_NOW, &extinfo);
211*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(handle2);
212*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != handle2);
213*8d67ca89SAndroid Build Coastguard Worker 
214*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle2);
215*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
216*8d67ca89SAndroid Build Coastguard Worker }
217*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,android_dlopen_ext_force_load_soname_exception)218*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, android_dlopen_ext_force_load_soname_exception) {
219*8d67ca89SAndroid Build Coastguard Worker   DlfcnSymlink symlink("android_dlopen_ext_force_load_soname_exception");
220*8d67ca89SAndroid Build Coastguard Worker   const std::string symlink_name = basename(symlink.get_symlink_path().c_str());
221*8d67ca89SAndroid Build Coastguard Worker   // Check if soname lookup still returns already loaded library
222*8d67ca89SAndroid Build Coastguard Worker   // when ANDROID_DLEXT_FORCE_LOAD flag is specified.
223*8d67ca89SAndroid Build Coastguard Worker   void* handle = dlopen(symlink_name.c_str(), RTLD_NOW);
224*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(handle);
225*8d67ca89SAndroid Build Coastguard Worker 
226*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
227*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_FORCE_LOAD;
228*8d67ca89SAndroid Build Coastguard Worker 
229*8d67ca89SAndroid Build Coastguard Worker   // Note that 'libdlext_test.so' is dt_soname for the symlink_name
230*8d67ca89SAndroid Build Coastguard Worker   void* handle2 = android_dlopen_ext("libdlext_test.so", RTLD_NOW, &extinfo);
231*8d67ca89SAndroid Build Coastguard Worker 
232*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(handle2);
233*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == handle2);
234*8d67ca89SAndroid Build Coastguard Worker 
235*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle2);
236*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
237*8d67ca89SAndroid Build Coastguard Worker }
238*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlfcn,dlopen_from_nullptr_android_api_level_28)239*8d67ca89SAndroid Build Coastguard Worker TEST(dlfcn, dlopen_from_nullptr_android_api_level_28) {
240*8d67ca89SAndroid Build Coastguard Worker   // Regression test for http://b/123972211. Testing dlopen(nullptr) when target sdk is P
241*8d67ca89SAndroid Build Coastguard Worker   android_set_application_target_sdk_version(28);
242*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(dlopen(nullptr, RTLD_NOW) != nullptr);
243*8d67ca89SAndroid Build Coastguard Worker }
244*8d67ca89SAndroid Build Coastguard Worker 
245*8d67ca89SAndroid Build Coastguard Worker // Test system path translation for backward compatibility. http://b/130219528
TEST(dlfcn,dlopen_system_libicuuc_android_api_level_28)246*8d67ca89SAndroid Build Coastguard Worker TEST(dlfcn, dlopen_system_libicuuc_android_api_level_28) {
247*8d67ca89SAndroid Build Coastguard Worker   android_set_application_target_sdk_version(28);
248*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(dlopen(PATH_TO_SYSTEM_LIB "libicuuc.so", RTLD_NOW) != nullptr);
249*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(dlopen(PATH_TO_SYSTEM_LIB "libicui18n.so", RTLD_NOW) != nullptr);
250*8d67ca89SAndroid Build Coastguard Worker }
251*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlfcn,dlopen_system_libicuuc_android_api_level_29)252*8d67ca89SAndroid Build Coastguard Worker TEST(dlfcn, dlopen_system_libicuuc_android_api_level_29) {
253*8d67ca89SAndroid Build Coastguard Worker   android_set_application_target_sdk_version(29);
254*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(dlopen(PATH_TO_SYSTEM_LIB "libicuuc.so", RTLD_NOW) == nullptr);
255*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(dlopen(PATH_TO_SYSTEM_LIB "libicui18n.so", RTLD_NOW) == nullptr);
256*8d67ca89SAndroid Build Coastguard Worker }
257*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlfcn,dlopen_system_libicuuc_android_api_level_current)258*8d67ca89SAndroid Build Coastguard Worker TEST(dlfcn, dlopen_system_libicuuc_android_api_level_current) {
259*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(dlopen(PATH_TO_SYSTEM_LIB "libicuuc.so", RTLD_NOW) == nullptr);
260*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(dlopen(PATH_TO_SYSTEM_LIB "libicui18n.so", RTLD_NOW) == nullptr);
261*8d67ca89SAndroid Build Coastguard Worker }
262*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlfcn,dlopen_from_zip_absolute_path)263*8d67ca89SAndroid Build Coastguard Worker TEST(dlfcn, dlopen_from_zip_absolute_path) {
264*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_zip_path = "/libdlext_test_zip/libdlext_test_zip_zipaligned.zip";
265*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_path = GetTestLibRoot() + lib_zip_path;
266*8d67ca89SAndroid Build Coastguard Worker 
267*8d67ca89SAndroid Build Coastguard Worker   void* handle = dlopen((lib_path + "!/libdir/libatest_simple_zip.so").c_str(), RTLD_NOW);
268*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
269*8d67ca89SAndroid Build Coastguard Worker 
270*8d67ca89SAndroid Build Coastguard Worker   uint32_t* taxicab_number = reinterpret_cast<uint32_t*>(dlsym(handle, "dlopen_testlib_taxicab_number"));
271*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(taxicab_number);
272*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(1729U, *taxicab_number);
273*8d67ca89SAndroid Build Coastguard Worker 
274*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
275*8d67ca89SAndroid Build Coastguard Worker }
276*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlfcn,dlopen_from_zip_with_dt_runpath)277*8d67ca89SAndroid Build Coastguard Worker TEST(dlfcn, dlopen_from_zip_with_dt_runpath) {
278*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_zip_path = "/libdlext_test_runpath_zip/libdlext_test_runpath_zip_zipaligned.zip";
279*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_path = GetTestLibRoot() + lib_zip_path;
280*8d67ca89SAndroid Build Coastguard Worker 
281*8d67ca89SAndroid Build Coastguard Worker   void* handle = dlopen((lib_path + "!/libdir/libtest_dt_runpath_d_zip.so").c_str(), RTLD_NOW);
282*8d67ca89SAndroid Build Coastguard Worker 
283*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
284*8d67ca89SAndroid Build Coastguard Worker 
285*8d67ca89SAndroid Build Coastguard Worker   typedef void *(* dlopen_b_fn)();
286*8d67ca89SAndroid Build Coastguard Worker   dlopen_b_fn fn = (dlopen_b_fn)dlsym(handle, "dlopen_b");
287*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(fn != nullptr) << dlerror();
288*8d67ca89SAndroid Build Coastguard Worker 
289*8d67ca89SAndroid Build Coastguard Worker   void *p = fn();
290*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(p != nullptr) << dlerror();
291*8d67ca89SAndroid Build Coastguard Worker 
292*8d67ca89SAndroid Build Coastguard Worker   dlclose(p);
293*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
294*8d67ca89SAndroid Build Coastguard Worker }
295*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlfcn,dlopen_from_zip_ld_library_path)296*8d67ca89SAndroid Build Coastguard Worker TEST(dlfcn, dlopen_from_zip_ld_library_path) {
297*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_zip_path = "/libdlext_test_zip/libdlext_test_zip_zipaligned.zip";
298*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_path = GetTestLibRoot() + lib_zip_path + "!/libdir";
299*8d67ca89SAndroid Build Coastguard Worker 
300*8d67ca89SAndroid Build Coastguard Worker   typedef void (*fn_t)(const char*);
301*8d67ca89SAndroid Build Coastguard Worker   fn_t android_update_LD_LIBRARY_PATH =
302*8d67ca89SAndroid Build Coastguard Worker       reinterpret_cast<fn_t>(dlsym(RTLD_DEFAULT, "android_update_LD_LIBRARY_PATH"));
303*8d67ca89SAndroid Build Coastguard Worker 
304*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_update_LD_LIBRARY_PATH != nullptr) << dlerror();
305*8d67ca89SAndroid Build Coastguard Worker 
306*8d67ca89SAndroid Build Coastguard Worker   void* handle = dlopen("libdlext_test_zip.so", RTLD_NOW);
307*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == nullptr);
308*8d67ca89SAndroid Build Coastguard Worker 
309*8d67ca89SAndroid Build Coastguard Worker   android_update_LD_LIBRARY_PATH(lib_path.c_str());
310*8d67ca89SAndroid Build Coastguard Worker 
311*8d67ca89SAndroid Build Coastguard Worker   handle = dlopen("libdlext_test_zip.so", RTLD_NOW);
312*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
313*8d67ca89SAndroid Build Coastguard Worker 
314*8d67ca89SAndroid Build Coastguard Worker   int (*fn)(void);
315*8d67ca89SAndroid Build Coastguard Worker   fn = reinterpret_cast<int (*)(void)>(dlsym(handle, "getRandomNumber"));
316*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(fn != nullptr);
317*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(4, fn());
318*8d67ca89SAndroid Build Coastguard Worker 
319*8d67ca89SAndroid Build Coastguard Worker   uint32_t* taxicab_number =
320*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<uint32_t*>(dlsym(handle, "dlopen_testlib_taxicab_number"));
321*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(taxicab_number);
322*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(1729U, *taxicab_number);
323*8d67ca89SAndroid Build Coastguard Worker 
324*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
325*8d67ca89SAndroid Build Coastguard Worker }
326*8d67ca89SAndroid Build Coastguard Worker 
327*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtTest,Reserved)328*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtTest, Reserved) {
329*8d67ca89SAndroid Build Coastguard Worker   void* start = mmap(nullptr, kLibSize, PROT_NONE, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
330*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(start != MAP_FAILED);
331*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
332*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_RESERVED_ADDRESS;
333*8d67ca89SAndroid Build Coastguard Worker   extinfo.reserved_addr = start;
334*8d67ca89SAndroid Build Coastguard Worker   extinfo.reserved_size = kLibSize;
335*8d67ca89SAndroid Build Coastguard Worker   handle_ = android_dlopen_ext(kLibName, RTLD_NOW, &extinfo);
336*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(handle_);
337*8d67ca89SAndroid Build Coastguard Worker   fn f = reinterpret_cast<fn>(dlsym(handle_, "getRandomNumber"));
338*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(f);
339*8d67ca89SAndroid Build Coastguard Worker   EXPECT_GE(reinterpret_cast<void*>(f), start);
340*8d67ca89SAndroid Build Coastguard Worker   EXPECT_LT(reinterpret_cast<void*>(f),
341*8d67ca89SAndroid Build Coastguard Worker             reinterpret_cast<char*>(start) + kLibSize);
342*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(4, f());
343*8d67ca89SAndroid Build Coastguard Worker 
344*8d67ca89SAndroid Build Coastguard Worker   // Check that after dlclose reserved address space is unmapped (and can be reused)
345*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle_);
346*8d67ca89SAndroid Build Coastguard Worker   handle_ = nullptr;
347*8d67ca89SAndroid Build Coastguard Worker 
348*8d67ca89SAndroid Build Coastguard Worker   void* new_start = mmap(start, kPageSize, PROT_NONE, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
349*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NE(start, new_start) << "dlclose unmapped reserved space";
350*8d67ca89SAndroid Build Coastguard Worker }
351*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtTest,ReservedTooSmall)352*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtTest, ReservedTooSmall) {
353*8d67ca89SAndroid Build Coastguard Worker   void* start = mmap(nullptr, kPageSize, PROT_NONE, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
354*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(start != MAP_FAILED);
355*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
356*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_RESERVED_ADDRESS;
357*8d67ca89SAndroid Build Coastguard Worker   extinfo.reserved_addr = start;
358*8d67ca89SAndroid Build Coastguard Worker   extinfo.reserved_size = kPageSize;
359*8d67ca89SAndroid Build Coastguard Worker   handle_ = android_dlopen_ext(kLibName, RTLD_NOW, &extinfo);
360*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(nullptr, handle_);
361*8d67ca89SAndroid Build Coastguard Worker }
362*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtTest,ReservedRecursive)363*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtTest, ReservedRecursive) {
364*8d67ca89SAndroid Build Coastguard Worker   void* start = mmap(nullptr, kLibSize, PROT_NONE, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
365*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(start != MAP_FAILED);
366*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
367*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_RESERVED_ADDRESS | ANDROID_DLEXT_RESERVED_ADDRESS_RECURSIVE;
368*8d67ca89SAndroid Build Coastguard Worker   extinfo.reserved_addr = start;
369*8d67ca89SAndroid Build Coastguard Worker   extinfo.reserved_size = kLibSize;
370*8d67ca89SAndroid Build Coastguard Worker   handle_ = android_dlopen_ext(kLibNameRecursive, RTLD_NOW, &extinfo);
371*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(handle_);
372*8d67ca89SAndroid Build Coastguard Worker 
373*8d67ca89SAndroid Build Coastguard Worker   fn f = reinterpret_cast<fn>(dlsym(handle_, "getRandomNumber"));
374*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(f);
375*8d67ca89SAndroid Build Coastguard Worker   EXPECT_GE(reinterpret_cast<void*>(f), start);
376*8d67ca89SAndroid Build Coastguard Worker   EXPECT_LT(reinterpret_cast<void*>(f),
377*8d67ca89SAndroid Build Coastguard Worker             reinterpret_cast<char*>(start) + kLibSize);
378*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(4, f());
379*8d67ca89SAndroid Build Coastguard Worker 
380*8d67ca89SAndroid Build Coastguard Worker   f = reinterpret_cast<fn>(dlsym(handle_, "getBiggerRandomNumber"));
381*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(f);
382*8d67ca89SAndroid Build Coastguard Worker   EXPECT_GE(reinterpret_cast<void*>(f), start);
383*8d67ca89SAndroid Build Coastguard Worker   EXPECT_LT(reinterpret_cast<void*>(f),
384*8d67ca89SAndroid Build Coastguard Worker             reinterpret_cast<char*>(start) + kLibSize);
385*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(8, f());
386*8d67ca89SAndroid Build Coastguard Worker 
387*8d67ca89SAndroid Build Coastguard Worker   uint32_t* taxicab_number = reinterpret_cast<uint32_t*>(dlsym(handle_, "dlopen_testlib_taxicab_number"));
388*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(taxicab_number);
389*8d67ca89SAndroid Build Coastguard Worker   // Untag the pointer so that it can be compared with start, which will be untagged.
390*8d67ca89SAndroid Build Coastguard Worker   void* addr = reinterpret_cast<void*>(untag_address(taxicab_number));
391*8d67ca89SAndroid Build Coastguard Worker   EXPECT_GE(addr, start);
392*8d67ca89SAndroid Build Coastguard Worker   EXPECT_LT(addr, reinterpret_cast<char*>(start) + kLibSize);
393*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(1729U, *taxicab_number);
394*8d67ca89SAndroid Build Coastguard Worker }
395*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtTest,ReservedRecursiveTooSmall)396*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtTest, ReservedRecursiveTooSmall) {
397*8d67ca89SAndroid Build Coastguard Worker   void* start = mmap(nullptr, kPageSize, PROT_NONE, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
398*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(start != MAP_FAILED);
399*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
400*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_RESERVED_ADDRESS | ANDROID_DLEXT_RESERVED_ADDRESS_RECURSIVE;
401*8d67ca89SAndroid Build Coastguard Worker   extinfo.reserved_addr = start;
402*8d67ca89SAndroid Build Coastguard Worker   extinfo.reserved_size = kPageSize;
403*8d67ca89SAndroid Build Coastguard Worker   handle_ = android_dlopen_ext(kLibNameRecursive, RTLD_NOW, &extinfo);
404*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(nullptr, handle_);
405*8d67ca89SAndroid Build Coastguard Worker }
406*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtTest,ReservedHint)407*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtTest, ReservedHint) {
408*8d67ca89SAndroid Build Coastguard Worker   void* start = mmap(nullptr, kLibSize, PROT_NONE, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
409*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(start != MAP_FAILED);
410*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
411*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_RESERVED_ADDRESS_HINT;
412*8d67ca89SAndroid Build Coastguard Worker   extinfo.reserved_addr = start;
413*8d67ca89SAndroid Build Coastguard Worker   extinfo.reserved_size = kLibSize;
414*8d67ca89SAndroid Build Coastguard Worker   handle_ = android_dlopen_ext(kLibName, RTLD_NOW, &extinfo);
415*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(handle_);
416*8d67ca89SAndroid Build Coastguard Worker   fn f = reinterpret_cast<fn>(dlsym(handle_, "getRandomNumber"));
417*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(f);
418*8d67ca89SAndroid Build Coastguard Worker   EXPECT_GE(reinterpret_cast<void*>(f), start);
419*8d67ca89SAndroid Build Coastguard Worker   EXPECT_LT(reinterpret_cast<void*>(f),
420*8d67ca89SAndroid Build Coastguard Worker             reinterpret_cast<char*>(start) + kLibSize);
421*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(4, f());
422*8d67ca89SAndroid Build Coastguard Worker }
423*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtTest,ReservedHintTooSmall)424*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtTest, ReservedHintTooSmall) {
425*8d67ca89SAndroid Build Coastguard Worker   void* start = mmap(nullptr, kPageSize, PROT_NONE, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
426*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(start != MAP_FAILED);
427*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
428*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_RESERVED_ADDRESS_HINT;
429*8d67ca89SAndroid Build Coastguard Worker   extinfo.reserved_addr = start;
430*8d67ca89SAndroid Build Coastguard Worker   extinfo.reserved_size = kPageSize;
431*8d67ca89SAndroid Build Coastguard Worker   handle_ = android_dlopen_ext(kLibName, RTLD_NOW, &extinfo);
432*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(handle_);
433*8d67ca89SAndroid Build Coastguard Worker   fn f = reinterpret_cast<fn>(dlsym(handle_, "getRandomNumber"));
434*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(f);
435*8d67ca89SAndroid Build Coastguard Worker   EXPECT_TRUE(reinterpret_cast<void*>(f) < start ||
436*8d67ca89SAndroid Build Coastguard Worker               (reinterpret_cast<void*>(f) >= reinterpret_cast<char*>(start) + kPageSize));
437*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(4, f());
438*8d67ca89SAndroid Build Coastguard Worker }
439*8d67ca89SAndroid Build Coastguard Worker 
440*8d67ca89SAndroid Build Coastguard Worker class DlExtRelroSharingTest : public DlExtTest {
441*8d67ca89SAndroid Build Coastguard Worker protected:
SetUp()442*8d67ca89SAndroid Build Coastguard Worker   void SetUp() override {
443*8d67ca89SAndroid Build Coastguard Worker     DlExtTest::SetUp();
444*8d67ca89SAndroid Build Coastguard Worker     void* start = mmap(nullptr, kLibSize, PROT_NONE, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
445*8d67ca89SAndroid Build Coastguard Worker     ASSERT_TRUE(start != MAP_FAILED);
446*8d67ca89SAndroid Build Coastguard Worker     extinfo_.flags = ANDROID_DLEXT_RESERVED_ADDRESS;
447*8d67ca89SAndroid Build Coastguard Worker     extinfo_.reserved_addr = start;
448*8d67ca89SAndroid Build Coastguard Worker     extinfo_.reserved_size = kLibSize;
449*8d67ca89SAndroid Build Coastguard Worker     extinfo_.relro_fd = -1;
450*8d67ca89SAndroid Build Coastguard Worker   }
451*8d67ca89SAndroid Build Coastguard Worker 
TearDown()452*8d67ca89SAndroid Build Coastguard Worker   void TearDown() override {
453*8d67ca89SAndroid Build Coastguard Worker     DlExtTest::TearDown();
454*8d67ca89SAndroid Build Coastguard Worker   }
455*8d67ca89SAndroid Build Coastguard Worker 
CreateRelroFile(const char * lib,const char * relro_file,bool recursive)456*8d67ca89SAndroid Build Coastguard Worker   void CreateRelroFile(const char* lib, const char* relro_file, bool recursive) {
457*8d67ca89SAndroid Build Coastguard Worker     int relro_fd = open(relro_file, O_RDWR | O_TRUNC | O_CLOEXEC);
458*8d67ca89SAndroid Build Coastguard Worker     ASSERT_NOERROR(relro_fd);
459*8d67ca89SAndroid Build Coastguard Worker 
460*8d67ca89SAndroid Build Coastguard Worker     if (recursive) {
461*8d67ca89SAndroid Build Coastguard Worker       extinfo_.flags |= ANDROID_DLEXT_RESERVED_ADDRESS_RECURSIVE;
462*8d67ca89SAndroid Build Coastguard Worker     }
463*8d67ca89SAndroid Build Coastguard Worker 
464*8d67ca89SAndroid Build Coastguard Worker     pid_t pid = fork();
465*8d67ca89SAndroid Build Coastguard Worker     if (pid == 0) {
466*8d67ca89SAndroid Build Coastguard Worker       // child process
467*8d67ca89SAndroid Build Coastguard Worker       extinfo_.flags |= ANDROID_DLEXT_WRITE_RELRO;
468*8d67ca89SAndroid Build Coastguard Worker       extinfo_.relro_fd = relro_fd;
469*8d67ca89SAndroid Build Coastguard Worker       void* handle = android_dlopen_ext(lib, RTLD_NOW, &extinfo_);
470*8d67ca89SAndroid Build Coastguard Worker       if (handle == nullptr) {
471*8d67ca89SAndroid Build Coastguard Worker         fprintf(stderr, "in child: %s\n", dlerror());
472*8d67ca89SAndroid Build Coastguard Worker         exit(1);
473*8d67ca89SAndroid Build Coastguard Worker       }
474*8d67ca89SAndroid Build Coastguard Worker       fn f = reinterpret_cast<fn>(dlsym(handle, "getRandomNumber"));
475*8d67ca89SAndroid Build Coastguard Worker       ASSERT_DL_NOTNULL(f);
476*8d67ca89SAndroid Build Coastguard Worker       EXPECT_EQ(4, f());
477*8d67ca89SAndroid Build Coastguard Worker 
478*8d67ca89SAndroid Build Coastguard Worker       if (recursive) {
479*8d67ca89SAndroid Build Coastguard Worker         fn f = reinterpret_cast<fn>(dlsym(handle, "getBiggerRandomNumber"));
480*8d67ca89SAndroid Build Coastguard Worker         ASSERT_DL_NOTNULL(f);
481*8d67ca89SAndroid Build Coastguard Worker         EXPECT_EQ(8, f());
482*8d67ca89SAndroid Build Coastguard Worker       }
483*8d67ca89SAndroid Build Coastguard Worker 
484*8d67ca89SAndroid Build Coastguard Worker       uint32_t* taxicab_number =
485*8d67ca89SAndroid Build Coastguard Worker               reinterpret_cast<uint32_t*>(dlsym(handle, "dlopen_testlib_taxicab_number"));
486*8d67ca89SAndroid Build Coastguard Worker       ASSERT_DL_NOTNULL(taxicab_number);
487*8d67ca89SAndroid Build Coastguard Worker       EXPECT_EQ(1729U, *taxicab_number);
488*8d67ca89SAndroid Build Coastguard Worker       exit(testing::Test::HasFailure());
489*8d67ca89SAndroid Build Coastguard Worker     }
490*8d67ca89SAndroid Build Coastguard Worker 
491*8d67ca89SAndroid Build Coastguard Worker     // continuing in parent
492*8d67ca89SAndroid Build Coastguard Worker     ASSERT_NOERROR(close(relro_fd));
493*8d67ca89SAndroid Build Coastguard Worker     ASSERT_NOERROR(pid);
494*8d67ca89SAndroid Build Coastguard Worker     AssertChildExited(pid, 0);
495*8d67ca89SAndroid Build Coastguard Worker 
496*8d67ca89SAndroid Build Coastguard Worker     // reopen file for reading so it can be used
497*8d67ca89SAndroid Build Coastguard Worker     relro_fd = open(relro_file, O_RDONLY | O_CLOEXEC);
498*8d67ca89SAndroid Build Coastguard Worker     ASSERT_NOERROR(relro_fd);
499*8d67ca89SAndroid Build Coastguard Worker     extinfo_.flags |= ANDROID_DLEXT_USE_RELRO;
500*8d67ca89SAndroid Build Coastguard Worker     extinfo_.relro_fd = relro_fd;
501*8d67ca89SAndroid Build Coastguard Worker   }
502*8d67ca89SAndroid Build Coastguard Worker 
TryUsingRelro(const char * lib,bool recursive)503*8d67ca89SAndroid Build Coastguard Worker   void TryUsingRelro(const char* lib, bool recursive) {
504*8d67ca89SAndroid Build Coastguard Worker     handle_ = android_dlopen_ext(lib, RTLD_NOW, &extinfo_);
505*8d67ca89SAndroid Build Coastguard Worker     ASSERT_DL_NOTNULL(handle_);
506*8d67ca89SAndroid Build Coastguard Worker     fn f = reinterpret_cast<fn>(dlsym(handle_, "getRandomNumber"));
507*8d67ca89SAndroid Build Coastguard Worker     ASSERT_DL_NOTNULL(f);
508*8d67ca89SAndroid Build Coastguard Worker     EXPECT_EQ(4, f());
509*8d67ca89SAndroid Build Coastguard Worker 
510*8d67ca89SAndroid Build Coastguard Worker     if (recursive) {
511*8d67ca89SAndroid Build Coastguard Worker       fn f = reinterpret_cast<fn>(dlsym(handle_, "getBiggerRandomNumber"));
512*8d67ca89SAndroid Build Coastguard Worker       ASSERT_DL_NOTNULL(f);
513*8d67ca89SAndroid Build Coastguard Worker       EXPECT_EQ(8, f());
514*8d67ca89SAndroid Build Coastguard Worker     }
515*8d67ca89SAndroid Build Coastguard Worker 
516*8d67ca89SAndroid Build Coastguard Worker     uint32_t* taxicab_number =
517*8d67ca89SAndroid Build Coastguard Worker             reinterpret_cast<uint32_t*>(dlsym(handle_, "dlopen_testlib_taxicab_number"));
518*8d67ca89SAndroid Build Coastguard Worker     ASSERT_DL_NOTNULL(taxicab_number);
519*8d67ca89SAndroid Build Coastguard Worker     EXPECT_EQ(1729U, *taxicab_number);
520*8d67ca89SAndroid Build Coastguard Worker   }
521*8d67ca89SAndroid Build Coastguard Worker 
522*8d67ca89SAndroid Build Coastguard Worker   void SpawnChildrenAndMeasurePss(const char* lib, const char* relro_file, bool share_relro,
523*8d67ca89SAndroid Build Coastguard Worker                                   size_t* pss_out);
524*8d67ca89SAndroid Build Coastguard Worker 
525*8d67ca89SAndroid Build Coastguard Worker   std::string FindMappingName(void* ptr);
526*8d67ca89SAndroid Build Coastguard Worker 
527*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo_;
528*8d67ca89SAndroid Build Coastguard Worker };
529*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtRelroSharingTest,ChildWritesGoodData)530*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtRelroSharingTest, ChildWritesGoodData) {
531*8d67ca89SAndroid Build Coastguard Worker   TemporaryFile tf; // Use tf to get an unique filename.
532*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NOERROR(close(tf.fd));
533*8d67ca89SAndroid Build Coastguard Worker 
534*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NO_FATAL_FAILURE(CreateRelroFile(kLibName, tf.path, false));
535*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NO_FATAL_FAILURE(TryUsingRelro(kLibName, false));
536*8d67ca89SAndroid Build Coastguard Worker   void* relro_data = dlsym(handle_, "lots_of_relro");
537*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(relro_data);
538*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(tf.path, FindMappingName(relro_data));
539*8d67ca89SAndroid Build Coastguard Worker 
540*8d67ca89SAndroid Build Coastguard Worker   // Use destructor of tf to close and unlink the file.
541*8d67ca89SAndroid Build Coastguard Worker   tf.fd = extinfo_.relro_fd;
542*8d67ca89SAndroid Build Coastguard Worker }
543*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtRelroSharingTest,ChildWritesGoodDataRecursive)544*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtRelroSharingTest, ChildWritesGoodDataRecursive) {
545*8d67ca89SAndroid Build Coastguard Worker   TemporaryFile tf; // Use tf to get an unique filename.
546*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NOERROR(close(tf.fd));
547*8d67ca89SAndroid Build Coastguard Worker 
548*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NO_FATAL_FAILURE(CreateRelroFile(kLibNameRecursive, tf.path, true));
549*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NO_FATAL_FAILURE(TryUsingRelro(kLibNameRecursive, true));
550*8d67ca89SAndroid Build Coastguard Worker   void* relro_data = dlsym(handle_, "lots_of_relro");
551*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(relro_data);
552*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(tf.path, FindMappingName(relro_data));
553*8d67ca89SAndroid Build Coastguard Worker   void* recursive_relro_data = dlsym(handle_, "lots_more_relro");
554*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(recursive_relro_data);
555*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(tf.path, FindMappingName(recursive_relro_data));
556*8d67ca89SAndroid Build Coastguard Worker 
557*8d67ca89SAndroid Build Coastguard Worker 
558*8d67ca89SAndroid Build Coastguard Worker   // Use destructor of tf to close and unlink the file.
559*8d67ca89SAndroid Build Coastguard Worker   tf.fd = extinfo_.relro_fd;
560*8d67ca89SAndroid Build Coastguard Worker }
561*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtRelroSharingTest,CheckRelroSizes)562*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtRelroSharingTest, CheckRelroSizes) {
563*8d67ca89SAndroid Build Coastguard Worker   TemporaryFile tf1, tf2;
564*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NOERROR(close(tf1.fd));
565*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NOERROR(close(tf2.fd));
566*8d67ca89SAndroid Build Coastguard Worker 
567*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NO_FATAL_FAILURE(CreateRelroFile(kLibNameRecursive, tf1.path, false));
568*8d67ca89SAndroid Build Coastguard Worker   struct stat no_recursive;
569*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NOERROR(fstat(extinfo_.relro_fd, &no_recursive));
570*8d67ca89SAndroid Build Coastguard Worker   tf1.fd = extinfo_.relro_fd;
571*8d67ca89SAndroid Build Coastguard Worker 
572*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NO_FATAL_FAILURE(CreateRelroFile(kLibNameRecursive, tf2.path, true));
573*8d67ca89SAndroid Build Coastguard Worker   struct stat with_recursive;
574*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NOERROR(fstat(extinfo_.relro_fd, &with_recursive));
575*8d67ca89SAndroid Build Coastguard Worker   tf2.fd = extinfo_.relro_fd;
576*8d67ca89SAndroid Build Coastguard Worker 
577*8d67ca89SAndroid Build Coastguard Worker   // RELRO file should end up bigger when we use the recursive flag, since it
578*8d67ca89SAndroid Build Coastguard Worker   // includes data for more than one library.
579*8d67ca89SAndroid Build Coastguard Worker   ASSERT_GT(with_recursive.st_size, no_recursive.st_size);
580*8d67ca89SAndroid Build Coastguard Worker }
581*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtRelroSharingTest,ChildWritesNoRelro)582*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtRelroSharingTest, ChildWritesNoRelro) {
583*8d67ca89SAndroid Build Coastguard Worker   TemporaryFile tf; // // Use tf to get an unique filename.
584*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NOERROR(close(tf.fd));
585*8d67ca89SAndroid Build Coastguard Worker 
586*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NO_FATAL_FAILURE(CreateRelroFile(kLibNameNoRelro, tf.path, false));
587*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NO_FATAL_FAILURE(TryUsingRelro(kLibNameNoRelro, false));
588*8d67ca89SAndroid Build Coastguard Worker 
589*8d67ca89SAndroid Build Coastguard Worker   // Use destructor of tf to close and unlink the file.
590*8d67ca89SAndroid Build Coastguard Worker   tf.fd = extinfo_.relro_fd;
591*8d67ca89SAndroid Build Coastguard Worker }
592*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtRelroSharingTest,RelroFileEmpty)593*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtRelroSharingTest, RelroFileEmpty) {
594*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NO_FATAL_FAILURE(TryUsingRelro(kLibName, false));
595*8d67ca89SAndroid Build Coastguard Worker }
596*8d67ca89SAndroid Build Coastguard Worker 
TEST_F(DlExtRelroSharingTest,VerifyMemorySaving)597*8d67ca89SAndroid Build Coastguard Worker TEST_F(DlExtRelroSharingTest, VerifyMemorySaving) {
598*8d67ca89SAndroid Build Coastguard Worker   if (geteuid() != 0) GTEST_SKIP() << "This test must be run as root";
599*8d67ca89SAndroid Build Coastguard Worker 
600*8d67ca89SAndroid Build Coastguard Worker   TemporaryFile tf; // Use tf to get an unique filename.
601*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NOERROR(close(tf.fd));
602*8d67ca89SAndroid Build Coastguard Worker 
603*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NO_FATAL_FAILURE(CreateRelroFile(kLibName, tf.path, false));
604*8d67ca89SAndroid Build Coastguard Worker 
605*8d67ca89SAndroid Build Coastguard Worker   int pipefd[2];
606*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NOERROR(pipe(pipefd));
607*8d67ca89SAndroid Build Coastguard Worker 
608*8d67ca89SAndroid Build Coastguard Worker   size_t without_sharing, with_sharing;
609*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NO_FATAL_FAILURE(SpawnChildrenAndMeasurePss(kLibName, tf.path, false, &without_sharing));
610*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NO_FATAL_FAILURE(SpawnChildrenAndMeasurePss(kLibName, tf.path, true, &with_sharing));
611*8d67ca89SAndroid Build Coastguard Worker   ASSERT_LT(with_sharing, without_sharing);
612*8d67ca89SAndroid Build Coastguard Worker 
613*8d67ca89SAndroid Build Coastguard Worker   // We expect the sharing to save at least 50% of the library's total PSS.
614*8d67ca89SAndroid Build Coastguard Worker   // In practice it saves 80%+ for this library in the test.
615*8d67ca89SAndroid Build Coastguard Worker   size_t pss_saved = without_sharing - with_sharing;
616*8d67ca89SAndroid Build Coastguard Worker   size_t expected_min_saved = without_sharing / 2;
617*8d67ca89SAndroid Build Coastguard Worker 
618*8d67ca89SAndroid Build Coastguard Worker   EXPECT_LT(expected_min_saved, pss_saved);
619*8d67ca89SAndroid Build Coastguard Worker 
620*8d67ca89SAndroid Build Coastguard Worker   // Use destructor of tf to close and unlink the file.
621*8d67ca89SAndroid Build Coastguard Worker   tf.fd = extinfo_.relro_fd;
622*8d67ca89SAndroid Build Coastguard Worker }
623*8d67ca89SAndroid Build Coastguard Worker 
GetPss(bool shared_relro,const char * lib,const char * relro_file,pid_t pid,size_t * total_pss)624*8d67ca89SAndroid Build Coastguard Worker void GetPss(bool shared_relro, const char* lib, const char* relro_file, pid_t pid,
625*8d67ca89SAndroid Build Coastguard Worker             size_t* total_pss) {
626*8d67ca89SAndroid Build Coastguard Worker   android::meminfo::ProcMemInfo proc_mem(pid);
627*8d67ca89SAndroid Build Coastguard Worker   const std::vector<android::meminfo::Vma>& maps = proc_mem.MapsWithoutUsageStats();
628*8d67ca89SAndroid Build Coastguard Worker   ASSERT_GT(maps.size(), 0UL);
629*8d67ca89SAndroid Build Coastguard Worker 
630*8d67ca89SAndroid Build Coastguard Worker   // Calculate total PSS of the library.
631*8d67ca89SAndroid Build Coastguard Worker   *total_pss = 0;
632*8d67ca89SAndroid Build Coastguard Worker   bool saw_relro_file = false;
633*8d67ca89SAndroid Build Coastguard Worker   for (auto& vma : maps) {
634*8d67ca89SAndroid Build Coastguard Worker     if (android::base::EndsWith(vma.name, lib) || (vma.name == relro_file)) {
635*8d67ca89SAndroid Build Coastguard Worker       if (vma.name == relro_file) {
636*8d67ca89SAndroid Build Coastguard Worker           saw_relro_file = true;
637*8d67ca89SAndroid Build Coastguard Worker       }
638*8d67ca89SAndroid Build Coastguard Worker 
639*8d67ca89SAndroid Build Coastguard Worker       android::meminfo::Vma update_vma(vma);
640*8d67ca89SAndroid Build Coastguard Worker       ASSERT_TRUE(proc_mem.FillInVmaStats(update_vma));
641*8d67ca89SAndroid Build Coastguard Worker       *total_pss += update_vma.usage.pss;
642*8d67ca89SAndroid Build Coastguard Worker     }
643*8d67ca89SAndroid Build Coastguard Worker   }
644*8d67ca89SAndroid Build Coastguard Worker 
645*8d67ca89SAndroid Build Coastguard Worker   if (shared_relro) ASSERT_TRUE(saw_relro_file);
646*8d67ca89SAndroid Build Coastguard Worker }
647*8d67ca89SAndroid Build Coastguard Worker 
SpawnChildrenAndMeasurePss(const char * lib,const char * relro_file,bool share_relro,size_t * pss_out)648*8d67ca89SAndroid Build Coastguard Worker void DlExtRelroSharingTest::SpawnChildrenAndMeasurePss(const char* lib, const char* relro_file,
649*8d67ca89SAndroid Build Coastguard Worker                                                        bool share_relro, size_t* pss_out) {
650*8d67ca89SAndroid Build Coastguard Worker   const int CHILDREN = 20;
651*8d67ca89SAndroid Build Coastguard Worker 
652*8d67ca89SAndroid Build Coastguard Worker   // Create children
653*8d67ca89SAndroid Build Coastguard Worker   pid_t child_pids[CHILDREN];
654*8d67ca89SAndroid Build Coastguard Worker   int childpipe[CHILDREN];
655*8d67ca89SAndroid Build Coastguard Worker   for (int i=0; i<CHILDREN; ++i) {
656*8d67ca89SAndroid Build Coastguard Worker     char read_buf;
657*8d67ca89SAndroid Build Coastguard Worker     int child_done_pipe[2], parent_done_pipe[2];
658*8d67ca89SAndroid Build Coastguard Worker     ASSERT_NOERROR(pipe(child_done_pipe));
659*8d67ca89SAndroid Build Coastguard Worker     ASSERT_NOERROR(pipe(parent_done_pipe));
660*8d67ca89SAndroid Build Coastguard Worker 
661*8d67ca89SAndroid Build Coastguard Worker     pid_t child = fork();
662*8d67ca89SAndroid Build Coastguard Worker     if (child == 0) {
663*8d67ca89SAndroid Build Coastguard Worker       // close the 'wrong' ends of the pipes in the child
664*8d67ca89SAndroid Build Coastguard Worker       close(child_done_pipe[0]);
665*8d67ca89SAndroid Build Coastguard Worker       close(parent_done_pipe[1]);
666*8d67ca89SAndroid Build Coastguard Worker 
667*8d67ca89SAndroid Build Coastguard Worker       // open the library
668*8d67ca89SAndroid Build Coastguard Worker       void* handle;
669*8d67ca89SAndroid Build Coastguard Worker       if (share_relro) {
670*8d67ca89SAndroid Build Coastguard Worker         handle = android_dlopen_ext(lib, RTLD_NOW, &extinfo_);
671*8d67ca89SAndroid Build Coastguard Worker       } else {
672*8d67ca89SAndroid Build Coastguard Worker         handle = dlopen(lib, RTLD_NOW);
673*8d67ca89SAndroid Build Coastguard Worker       }
674*8d67ca89SAndroid Build Coastguard Worker       if (handle == nullptr) {
675*8d67ca89SAndroid Build Coastguard Worker         fprintf(stderr, "in child: %s\n", dlerror());
676*8d67ca89SAndroid Build Coastguard Worker         exit(1);
677*8d67ca89SAndroid Build Coastguard Worker       }
678*8d67ca89SAndroid Build Coastguard Worker 
679*8d67ca89SAndroid Build Coastguard Worker       // close write end of child_done_pipe to signal the parent that we're done.
680*8d67ca89SAndroid Build Coastguard Worker       close(child_done_pipe[1]);
681*8d67ca89SAndroid Build Coastguard Worker 
682*8d67ca89SAndroid Build Coastguard Worker       // wait for the parent to close parent_done_pipe, then exit
683*8d67ca89SAndroid Build Coastguard Worker       read(parent_done_pipe[0], &read_buf, 1);
684*8d67ca89SAndroid Build Coastguard Worker       exit(0);
685*8d67ca89SAndroid Build Coastguard Worker     }
686*8d67ca89SAndroid Build Coastguard Worker 
687*8d67ca89SAndroid Build Coastguard Worker     ASSERT_NOERROR(child);
688*8d67ca89SAndroid Build Coastguard Worker 
689*8d67ca89SAndroid Build Coastguard Worker     // close the 'wrong' ends of the pipes in the parent
690*8d67ca89SAndroid Build Coastguard Worker     close(child_done_pipe[1]);
691*8d67ca89SAndroid Build Coastguard Worker     close(parent_done_pipe[0]);
692*8d67ca89SAndroid Build Coastguard Worker 
693*8d67ca89SAndroid Build Coastguard Worker     // wait for the child to be done
694*8d67ca89SAndroid Build Coastguard Worker     read(child_done_pipe[0], &read_buf, 1);
695*8d67ca89SAndroid Build Coastguard Worker     close(child_done_pipe[0]);
696*8d67ca89SAndroid Build Coastguard Worker 
697*8d67ca89SAndroid Build Coastguard Worker     // save the child's pid and the parent_done_pipe
698*8d67ca89SAndroid Build Coastguard Worker     child_pids[i] = child;
699*8d67ca89SAndroid Build Coastguard Worker     childpipe[i] = parent_done_pipe[1];
700*8d67ca89SAndroid Build Coastguard Worker   }
701*8d67ca89SAndroid Build Coastguard Worker 
702*8d67ca89SAndroid Build Coastguard Worker   // Sum the PSS of tested library of all the children
703*8d67ca89SAndroid Build Coastguard Worker   size_t total_pss = 0;
704*8d67ca89SAndroid Build Coastguard Worker   for (int i=0; i<CHILDREN; ++i) {
705*8d67ca89SAndroid Build Coastguard Worker     size_t child_pss;
706*8d67ca89SAndroid Build Coastguard Worker     ASSERT_NO_FATAL_FAILURE(GetPss(share_relro, lib, relro_file, child_pids[i], &child_pss));
707*8d67ca89SAndroid Build Coastguard Worker     total_pss += child_pss;
708*8d67ca89SAndroid Build Coastguard Worker   }
709*8d67ca89SAndroid Build Coastguard Worker   *pss_out = total_pss;
710*8d67ca89SAndroid Build Coastguard Worker 
711*8d67ca89SAndroid Build Coastguard Worker   // Close pipes and wait for children to exit
712*8d67ca89SAndroid Build Coastguard Worker   for (int i=0; i<CHILDREN; ++i) {
713*8d67ca89SAndroid Build Coastguard Worker     ASSERT_NOERROR(close(childpipe[i]));
714*8d67ca89SAndroid Build Coastguard Worker   }
715*8d67ca89SAndroid Build Coastguard Worker   for (int i = 0; i < CHILDREN; ++i) {
716*8d67ca89SAndroid Build Coastguard Worker     AssertChildExited(child_pids[i], 0);
717*8d67ca89SAndroid Build Coastguard Worker   }
718*8d67ca89SAndroid Build Coastguard Worker }
719*8d67ca89SAndroid Build Coastguard Worker 
FindMappingName(void * ptr)720*8d67ca89SAndroid Build Coastguard Worker std::string DlExtRelroSharingTest::FindMappingName(void* ptr) {
721*8d67ca89SAndroid Build Coastguard Worker   uint64_t addr = reinterpret_cast<uint64_t>(untag_address(ptr));
722*8d67ca89SAndroid Build Coastguard Worker   std::string found_name = "<not found>";
723*8d67ca89SAndroid Build Coastguard Worker 
724*8d67ca89SAndroid Build Coastguard Worker   EXPECT_TRUE(android::procinfo::ReadMapFile("/proc/self/maps",
725*8d67ca89SAndroid Build Coastguard Worker                                              [&](const android::procinfo::MapInfo& mapinfo) {
726*8d67ca89SAndroid Build Coastguard Worker                                                if (addr >= mapinfo.start && addr < mapinfo.end) {
727*8d67ca89SAndroid Build Coastguard Worker                                                  found_name = mapinfo.name;
728*8d67ca89SAndroid Build Coastguard Worker                                                }
729*8d67ca89SAndroid Build Coastguard Worker                                              }));
730*8d67ca89SAndroid Build Coastguard Worker 
731*8d67ca89SAndroid Build Coastguard Worker   return found_name;
732*8d67ca89SAndroid Build Coastguard Worker }
733*8d67ca89SAndroid Build Coastguard Worker 
734*8d67ca89SAndroid Build Coastguard Worker // Testing namespaces
735*8d67ca89SAndroid Build Coastguard Worker static const char* g_public_lib = "libnstest_public.so";
736*8d67ca89SAndroid Build Coastguard Worker 
737*8d67ca89SAndroid Build Coastguard Worker // These are libs shared with default namespace
738*8d67ca89SAndroid Build Coastguard Worker static const std::string g_core_shared_libs = kCoreSharedLibs;
739*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_smoke)740*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_smoke) {
741*8d67ca89SAndroid Build Coastguard Worker   static const char* root_lib = "libnstest_root.so";
742*8d67ca89SAndroid Build Coastguard Worker   std::string shared_libs = g_core_shared_libs + ":" + g_public_lib;
743*8d67ca89SAndroid Build Coastguard Worker 
744*8d67ca89SAndroid Build Coastguard Worker   ASSERT_FALSE(android_init_anonymous_namespace("", nullptr));
745*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("android_init_anonymous_namespace failed: error linking namespaces"
746*8d67ca89SAndroid Build Coastguard Worker                " \"(anonymous)\"->\"(default)\": the list of shared libraries is empty.",
747*8d67ca89SAndroid Build Coastguard Worker                dlerror());
748*8d67ca89SAndroid Build Coastguard Worker 
749*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_public_path = GetTestLibRoot() + "/public_namespace_libs/" + g_public_lib;
750*8d67ca89SAndroid Build Coastguard Worker   void* handle_public = dlopen(lib_public_path.c_str(), RTLD_NOW);
751*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_public != nullptr) << dlerror();
752*8d67ca89SAndroid Build Coastguard Worker 
753*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(shared_libs.c_str(), nullptr)) << dlerror();
754*8d67ca89SAndroid Build Coastguard Worker 
755*8d67ca89SAndroid Build Coastguard Worker   // Check that libraries added to public namespace are not NODELETE
756*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle_public);
757*8d67ca89SAndroid Build Coastguard Worker   handle_public = dlopen(lib_public_path.c_str(), RTLD_NOW | RTLD_NOLOAD);
758*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_public == nullptr);
759*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(std::string("dlopen failed: library \"") + lib_public_path +
760*8d67ca89SAndroid Build Coastguard Worker                "\" wasn't loaded and RTLD_NOLOAD prevented it", dlerror());
761*8d67ca89SAndroid Build Coastguard Worker 
762*8d67ca89SAndroid Build Coastguard Worker   handle_public = dlopen(lib_public_path.c_str(), RTLD_NOW);
763*8d67ca89SAndroid Build Coastguard Worker 
764*8d67ca89SAndroid Build Coastguard Worker   // create "public namespace", share limited set of public libraries with
765*8d67ca89SAndroid Build Coastguard Worker 
766*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns1 =
767*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private",
768*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
769*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/private_namespace_libs").c_str(),
770*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_REGULAR,
771*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
772*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
773*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns1 != nullptr) << dlerror();
774*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns1, nullptr, shared_libs.c_str())) << dlerror();
775*8d67ca89SAndroid Build Coastguard Worker 
776*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns2 =
777*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private_isolated",
778*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
779*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/private_namespace_libs").c_str(),
780*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
781*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
782*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
783*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns2 != nullptr) << dlerror();
784*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns2, nullptr, shared_libs.c_str())) << dlerror();
785*8d67ca89SAndroid Build Coastguard Worker 
786*8d67ca89SAndroid Build Coastguard Worker   // This should not have affect search path for default namespace:
787*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(dlopen(root_lib, RTLD_NOW) == nullptr);
788*8d67ca89SAndroid Build Coastguard Worker   void* handle = dlopen(g_public_lib, RTLD_NOW);
789*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
790*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
791*8d67ca89SAndroid Build Coastguard Worker 
792*8d67ca89SAndroid Build Coastguard Worker   // dlopen for a public library using an absolute path should work
793*8d67ca89SAndroid Build Coastguard Worker   // 1. For isolated namespaces
794*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
795*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
796*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns2;
797*8d67ca89SAndroid Build Coastguard Worker   handle = android_dlopen_ext(lib_public_path.c_str(), RTLD_NOW, &extinfo);
798*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
799*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == handle_public);
800*8d67ca89SAndroid Build Coastguard Worker 
801*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
802*8d67ca89SAndroid Build Coastguard Worker 
803*8d67ca89SAndroid Build Coastguard Worker   // 1.1 even if it wasn't loaded before
804*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle_public);
805*8d67ca89SAndroid Build Coastguard Worker 
806*8d67ca89SAndroid Build Coastguard Worker   handle_public = dlopen(lib_public_path.c_str(), RTLD_NOW | RTLD_NOLOAD);
807*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_public == nullptr);
808*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(std::string("dlopen failed: library \"") + lib_public_path +
809*8d67ca89SAndroid Build Coastguard Worker                "\" wasn't loaded and RTLD_NOLOAD prevented it", dlerror());
810*8d67ca89SAndroid Build Coastguard Worker 
811*8d67ca89SAndroid Build Coastguard Worker   handle = android_dlopen_ext(lib_public_path.c_str(), RTLD_NOW, &extinfo);
812*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
813*8d67ca89SAndroid Build Coastguard Worker 
814*8d67ca89SAndroid Build Coastguard Worker   handle_public = dlopen(lib_public_path.c_str(), RTLD_NOW);
815*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == handle_public);
816*8d67ca89SAndroid Build Coastguard Worker 
817*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
818*8d67ca89SAndroid Build Coastguard Worker 
819*8d67ca89SAndroid Build Coastguard Worker   // 2. And for regular namespaces (make sure it does not load second copy of the library)
820*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns1;
821*8d67ca89SAndroid Build Coastguard Worker   handle = android_dlopen_ext(lib_public_path.c_str(), RTLD_NOW, &extinfo);
822*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
823*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == handle_public);
824*8d67ca89SAndroid Build Coastguard Worker 
825*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
826*8d67ca89SAndroid Build Coastguard Worker 
827*8d67ca89SAndroid Build Coastguard Worker   // 2.1 Unless it was not loaded before - in which case it will load a duplicate.
828*8d67ca89SAndroid Build Coastguard Worker   // TODO(dimitry): This is broken. Maybe we need to deprecate non-isolated namespaces?
829*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle_public);
830*8d67ca89SAndroid Build Coastguard Worker 
831*8d67ca89SAndroid Build Coastguard Worker   handle_public = dlopen(lib_public_path.c_str(), RTLD_NOW | RTLD_NOLOAD);
832*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_public == nullptr);
833*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(std::string("dlopen failed: library \"") + lib_public_path +
834*8d67ca89SAndroid Build Coastguard Worker                "\" wasn't loaded and RTLD_NOLOAD prevented it", dlerror());
835*8d67ca89SAndroid Build Coastguard Worker 
836*8d67ca89SAndroid Build Coastguard Worker   handle = android_dlopen_ext(lib_public_path.c_str(), RTLD_NOW, &extinfo);
837*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
838*8d67ca89SAndroid Build Coastguard Worker 
839*8d67ca89SAndroid Build Coastguard Worker   handle_public = dlopen(lib_public_path.c_str(), RTLD_NOW);
840*8d67ca89SAndroid Build Coastguard Worker 
841*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != handle_public);
842*8d67ca89SAndroid Build Coastguard Worker 
843*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
844*8d67ca89SAndroid Build Coastguard Worker 
845*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns1;
846*8d67ca89SAndroid Build Coastguard Worker 
847*8d67ca89SAndroid Build Coastguard Worker   void* handle1 = android_dlopen_ext(root_lib, RTLD_NOW, &extinfo);
848*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle1 != nullptr) << dlerror();
849*8d67ca89SAndroid Build Coastguard Worker 
850*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns2;
851*8d67ca89SAndroid Build Coastguard Worker   void* handle2 = android_dlopen_ext(root_lib, RTLD_NOW, &extinfo);
852*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle2 != nullptr) << dlerror();
853*8d67ca89SAndroid Build Coastguard Worker 
854*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle1 != handle2);
855*8d67ca89SAndroid Build Coastguard Worker 
856*8d67ca89SAndroid Build Coastguard Worker   typedef const char* (*fn_t)();
857*8d67ca89SAndroid Build Coastguard Worker 
858*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_local_string1 = reinterpret_cast<fn_t>(dlsym(handle1, "ns_get_local_string"));
859*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_local_string1 != nullptr) << dlerror();
860*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_local_string2 = reinterpret_cast<fn_t>(dlsym(handle2, "ns_get_local_string"));
861*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_local_string2 != nullptr) << dlerror();
862*8d67ca89SAndroid Build Coastguard Worker 
863*8d67ca89SAndroid Build Coastguard Worker   EXPECT_STREQ("This string is local to root library", ns_get_local_string1());
864*8d67ca89SAndroid Build Coastguard Worker   EXPECT_STREQ("This string is local to root library", ns_get_local_string2());
865*8d67ca89SAndroid Build Coastguard Worker 
866*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_local_string1() != ns_get_local_string2());
867*8d67ca89SAndroid Build Coastguard Worker 
868*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_private_extern_string1 =
869*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<fn_t>(dlsym(handle1, "ns_get_private_extern_string"));
870*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_private_extern_string1 != nullptr) << dlerror();
871*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_private_extern_string2 =
872*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<fn_t>(dlsym(handle2, "ns_get_private_extern_string"));
873*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_private_extern_string2 != nullptr) << dlerror();
874*8d67ca89SAndroid Build Coastguard Worker 
875*8d67ca89SAndroid Build Coastguard Worker   EXPECT_STREQ("This string is from private namespace", ns_get_private_extern_string1());
876*8d67ca89SAndroid Build Coastguard Worker   EXPECT_STREQ("This string is from private namespace", ns_get_private_extern_string2());
877*8d67ca89SAndroid Build Coastguard Worker 
878*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_private_extern_string1() != ns_get_private_extern_string2());
879*8d67ca89SAndroid Build Coastguard Worker 
880*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_public_extern_string1 =
881*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<fn_t>(dlsym(handle1, "ns_get_public_extern_string"));
882*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_public_extern_string1 != nullptr) << dlerror();
883*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_public_extern_string2 =
884*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<fn_t>(dlsym(handle2, "ns_get_public_extern_string"));
885*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_public_extern_string2 != nullptr) << dlerror();
886*8d67ca89SAndroid Build Coastguard Worker 
887*8d67ca89SAndroid Build Coastguard Worker   EXPECT_STREQ("This string is from public namespace", ns_get_public_extern_string1());
888*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_public_extern_string1() == ns_get_public_extern_string2());
889*8d67ca89SAndroid Build Coastguard Worker 
890*8d67ca89SAndroid Build Coastguard Worker   // and now check that dlopen() does the right thing in terms of preserving namespace
891*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_dlopened_string1 = reinterpret_cast<fn_t>(dlsym(handle1, "ns_get_dlopened_string"));
892*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_dlopened_string1 != nullptr) << dlerror();
893*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_dlopened_string2 = reinterpret_cast<fn_t>(dlsym(handle2, "ns_get_dlopened_string"));
894*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_dlopened_string2 != nullptr) << dlerror();
895*8d67ca89SAndroid Build Coastguard Worker 
896*8d67ca89SAndroid Build Coastguard Worker   EXPECT_STREQ("This string is from private namespace (dlopened library)", ns_get_dlopened_string1());
897*8d67ca89SAndroid Build Coastguard Worker   EXPECT_STREQ("This string is from private namespace (dlopened library)", ns_get_dlopened_string2());
898*8d67ca89SAndroid Build Coastguard Worker 
899*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_dlopened_string1() != ns_get_dlopened_string2());
900*8d67ca89SAndroid Build Coastguard Worker 
901*8d67ca89SAndroid Build Coastguard Worker   // Check that symbols from non-shared libraries a shared library depends on are not visible
902*8d67ca89SAndroid Build Coastguard Worker   // from original namespace.
903*8d67ca89SAndroid Build Coastguard Worker 
904*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_internal_extern_string =
905*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<fn_t>(dlsym(handle1, "ns_get_internal_extern_string"));
906*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_internal_extern_string != nullptr) << dlerror();
907*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_internal_extern_string() == nullptr) <<
908*8d67ca89SAndroid Build Coastguard Worker       "ns_get_internal_extern_string() expected to return null but returns \"" <<
909*8d67ca89SAndroid Build Coastguard Worker       ns_get_internal_extern_string() << "\"";
910*8d67ca89SAndroid Build Coastguard Worker 
911*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle1);
912*8d67ca89SAndroid Build Coastguard Worker 
913*8d67ca89SAndroid Build Coastguard Worker   // Check if handle2 is still alive (and well)
914*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is local to root library", ns_get_local_string2());
915*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from private namespace", ns_get_private_extern_string2());
916*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from public namespace", ns_get_public_extern_string2());
917*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from private namespace (dlopened library)", ns_get_dlopened_string2());
918*8d67ca89SAndroid Build Coastguard Worker 
919*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle2);
920*8d67ca89SAndroid Build Coastguard Worker }
921*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,dlopen_ext_use_o_tmpfile_fd)922*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, dlopen_ext_use_o_tmpfile_fd) {
923*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_path = GetTestLibRoot() + "/libtest_simple.so";
924*8d67ca89SAndroid Build Coastguard Worker 
925*8d67ca89SAndroid Build Coastguard Worker   int tmpfd = TEMP_FAILURE_RETRY(
926*8d67ca89SAndroid Build Coastguard Worker         open(GetTestLibRoot().c_str(), O_TMPFILE | O_CLOEXEC | O_RDWR | O_EXCL, 0));
927*8d67ca89SAndroid Build Coastguard Worker 
928*8d67ca89SAndroid Build Coastguard Worker   // Ignore kernels without O_TMPFILE flag support
929*8d67ca89SAndroid Build Coastguard Worker   if (tmpfd == -1 && (errno == EISDIR || errno == EINVAL || errno == EOPNOTSUPP)) {
930*8d67ca89SAndroid Build Coastguard Worker     return;
931*8d67ca89SAndroid Build Coastguard Worker   }
932*8d67ca89SAndroid Build Coastguard Worker 
933*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(tmpfd != -1) << strerror(errno);
934*8d67ca89SAndroid Build Coastguard Worker 
935*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns =
936*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("testing-o_tmpfile",
937*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
938*8d67ca89SAndroid Build Coastguard Worker                                    GetTestLibRoot().c_str(),
939*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
940*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
941*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
942*8d67ca89SAndroid Build Coastguard Worker 
943*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(ns);
944*8d67ca89SAndroid Build Coastguard Worker 
945*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns, nullptr, g_core_shared_libs.c_str())) << dlerror();
946*8d67ca89SAndroid Build Coastguard Worker 
947*8d67ca89SAndroid Build Coastguard Worker   std::string content;
948*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android::base::ReadFileToString(lib_path, &content)) << strerror(errno);
949*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android::base::WriteStringToFd(content, tmpfd)) << strerror(errno);
950*8d67ca89SAndroid Build Coastguard Worker 
951*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
952*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_LIBRARY_FD | ANDROID_DLEXT_USE_NAMESPACE;
953*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_fd = tmpfd;
954*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns;
955*8d67ca89SAndroid Build Coastguard Worker 
956*8d67ca89SAndroid Build Coastguard Worker   void* handle = android_dlopen_ext("foobar", RTLD_NOW, &extinfo);
957*8d67ca89SAndroid Build Coastguard Worker 
958*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(handle);
959*8d67ca89SAndroid Build Coastguard Worker 
960*8d67ca89SAndroid Build Coastguard Worker   uint32_t* taxicab_number = reinterpret_cast<uint32_t*>(dlsym(handle, "dlopen_testlib_taxicab_number"));
961*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(taxicab_number);
962*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(1729U, *taxicab_number);
963*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
964*8d67ca89SAndroid Build Coastguard Worker }
965*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,dlopen_ext_use_memfd)966*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, dlopen_ext_use_memfd) {
967*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_path = GetTestLibRoot() + "/libtest_simple.so";
968*8d67ca89SAndroid Build Coastguard Worker 
969*8d67ca89SAndroid Build Coastguard Worker   // create memfd
970*8d67ca89SAndroid Build Coastguard Worker   int memfd = memfd_create("foobar", MFD_CLOEXEC);
971*8d67ca89SAndroid Build Coastguard Worker   if (memfd == -1 && errno == ENOSYS) GTEST_SKIP() << "no memfd_create() in this kernel";
972*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(memfd != -1) << strerror(errno);
973*8d67ca89SAndroid Build Coastguard Worker 
974*8d67ca89SAndroid Build Coastguard Worker   // Check st.f_type is TMPFS_MAGIC for memfd
975*8d67ca89SAndroid Build Coastguard Worker   struct statfs st;
976*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(TEMP_FAILURE_RETRY(fstatfs(memfd, &st)) == 0) << strerror(errno);
977*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(static_cast<decltype(st.f_type)>(TMPFS_MAGIC), st.f_type);
978*8d67ca89SAndroid Build Coastguard Worker 
979*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns =
980*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("testing-memfd",
981*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
982*8d67ca89SAndroid Build Coastguard Worker                                    GetTestLibRoot().c_str(),
983*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
984*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
985*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
986*8d67ca89SAndroid Build Coastguard Worker 
987*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(ns);
988*8d67ca89SAndroid Build Coastguard Worker 
989*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns, nullptr, g_core_shared_libs.c_str())) << dlerror();
990*8d67ca89SAndroid Build Coastguard Worker 
991*8d67ca89SAndroid Build Coastguard Worker   // read file into memfd backed one.
992*8d67ca89SAndroid Build Coastguard Worker   std::string content;
993*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android::base::ReadFileToString(lib_path, &content)) << strerror(errno);
994*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android::base::WriteStringToFd(content, memfd)) << strerror(errno);
995*8d67ca89SAndroid Build Coastguard Worker 
996*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
997*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_LIBRARY_FD | ANDROID_DLEXT_USE_NAMESPACE;
998*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_fd = memfd;
999*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns;
1000*8d67ca89SAndroid Build Coastguard Worker 
1001*8d67ca89SAndroid Build Coastguard Worker   void* handle = android_dlopen_ext("foobar", RTLD_NOW, &extinfo);
1002*8d67ca89SAndroid Build Coastguard Worker 
1003*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(handle);
1004*8d67ca89SAndroid Build Coastguard Worker 
1005*8d67ca89SAndroid Build Coastguard Worker   uint32_t* taxicab_number = reinterpret_cast<uint32_t*>(dlsym(handle, "dlopen_testlib_taxicab_number"));
1006*8d67ca89SAndroid Build Coastguard Worker   ASSERT_DL_NOTNULL(taxicab_number);
1007*8d67ca89SAndroid Build Coastguard Worker   EXPECT_EQ(1729U, *taxicab_number);
1008*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
1009*8d67ca89SAndroid Build Coastguard Worker }
1010*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_symbol_visibilty_one_namespace)1011*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_symbol_visibilty_one_namespace) {
1012*8d67ca89SAndroid Build Coastguard Worker   static const char* root_lib = "libnstest_root.so";
1013*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(g_core_shared_libs.c_str(), nullptr));
1014*8d67ca89SAndroid Build Coastguard Worker 
1015*8d67ca89SAndroid Build Coastguard Worker   const std::string ns_search_path = GetTestLibRoot() + "/public_namespace_libs:" +
1016*8d67ca89SAndroid Build Coastguard Worker                                      GetTestLibRoot() + "/private_namespace_libs";
1017*8d67ca89SAndroid Build Coastguard Worker 
1018*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns =
1019*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("one",
1020*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1021*8d67ca89SAndroid Build Coastguard Worker                                    ns_search_path.c_str(),
1022*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1023*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1024*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1025*8d67ca89SAndroid Build Coastguard Worker 
1026*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns, nullptr, g_core_shared_libs.c_str())) << dlerror();
1027*8d67ca89SAndroid Build Coastguard Worker 
1028*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
1029*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1030*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns;
1031*8d67ca89SAndroid Build Coastguard Worker 
1032*8d67ca89SAndroid Build Coastguard Worker   void* handle = android_dlopen_ext(root_lib, RTLD_NOW, &extinfo);
1033*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
1034*8d67ca89SAndroid Build Coastguard Worker 
1035*8d67ca89SAndroid Build Coastguard Worker   typedef const char* (*fn_t)();
1036*8d67ca89SAndroid Build Coastguard Worker 
1037*8d67ca89SAndroid Build Coastguard Worker   // Check that relocation worked correctly
1038*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_internal_extern_string =
1039*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<fn_t>(dlsym(handle, "ns_get_internal_extern_string"));
1040*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_internal_extern_string != nullptr) << dlerror();
1041*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from a library a shared library depends on", ns_get_internal_extern_string());
1042*8d67ca89SAndroid Build Coastguard Worker 
1043*8d67ca89SAndroid Build Coastguard Worker   fn_t internal_extern_string_fn =
1044*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<fn_t>(dlsym(handle, "internal_extern_string"));
1045*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(internal_extern_string_fn != nullptr) << dlerror();
1046*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from a library a shared library depends on", internal_extern_string_fn());
1047*8d67ca89SAndroid Build Coastguard Worker }
1048*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_symbol_visibilty_between_namespaces)1049*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_symbol_visibilty_between_namespaces) {
1050*8d67ca89SAndroid Build Coastguard Worker   static const char* root_lib = "libnstest_root.so";
1051*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(g_core_shared_libs.c_str(), nullptr));
1052*8d67ca89SAndroid Build Coastguard Worker 
1053*8d67ca89SAndroid Build Coastguard Worker   const std::string public_ns_search_path =  GetTestLibRoot() + "/public_namespace_libs";
1054*8d67ca89SAndroid Build Coastguard Worker   const std::string private_ns_search_path = GetTestLibRoot() + "/private_namespace_libs";
1055*8d67ca89SAndroid Build Coastguard Worker 
1056*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_public =
1057*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("public",
1058*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1059*8d67ca89SAndroid Build Coastguard Worker                                    public_ns_search_path.c_str(),
1060*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1061*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1062*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1063*8d67ca89SAndroid Build Coastguard Worker 
1064*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_public, nullptr, g_core_shared_libs.c_str())) << dlerror();
1065*8d67ca89SAndroid Build Coastguard Worker 
1066*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_private =
1067*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private",
1068*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1069*8d67ca89SAndroid Build Coastguard Worker                                    private_ns_search_path.c_str(),
1070*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1071*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1072*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1073*8d67ca89SAndroid Build Coastguard Worker 
1074*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_private, ns_public, g_public_lib)) << dlerror();
1075*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_private, nullptr, g_core_shared_libs.c_str())) << dlerror();
1076*8d67ca89SAndroid Build Coastguard Worker 
1077*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
1078*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1079*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_private;
1080*8d67ca89SAndroid Build Coastguard Worker 
1081*8d67ca89SAndroid Build Coastguard Worker   void* handle = android_dlopen_ext(root_lib, RTLD_NOW, &extinfo);
1082*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
1083*8d67ca89SAndroid Build Coastguard Worker 
1084*8d67ca89SAndroid Build Coastguard Worker   typedef const char* (*fn_t)();
1085*8d67ca89SAndroid Build Coastguard Worker 
1086*8d67ca89SAndroid Build Coastguard Worker   // Check that relocation worked correctly
1087*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_internal_extern_string =
1088*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<fn_t>(dlsym(handle, "ns_get_internal_extern_string"));
1089*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_internal_extern_string != nullptr) << dlerror();
1090*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_internal_extern_string() == nullptr) <<
1091*8d67ca89SAndroid Build Coastguard Worker       "ns_get_internal_extern_string() expected to return null but returns \"" <<
1092*8d67ca89SAndroid Build Coastguard Worker       ns_get_internal_extern_string() << "\"";
1093*8d67ca89SAndroid Build Coastguard Worker 
1094*8d67ca89SAndroid Build Coastguard Worker   fn_t internal_extern_string_fn =
1095*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<fn_t>(dlsym(handle, "internal_extern_string"));
1096*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(internal_extern_string_fn == nullptr);
1097*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("undefined symbol: internal_extern_string", dlerror());
1098*8d67ca89SAndroid Build Coastguard Worker }
1099*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_unload_between_namespaces)1100*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_unload_between_namespaces) {
1101*8d67ca89SAndroid Build Coastguard Worker   static const char* root_lib = "libnstest_root.so";
1102*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(g_core_shared_libs.c_str(), nullptr));
1103*8d67ca89SAndroid Build Coastguard Worker 
1104*8d67ca89SAndroid Build Coastguard Worker   const std::string public_ns_search_path =  GetTestLibRoot() + "/public_namespace_libs";
1105*8d67ca89SAndroid Build Coastguard Worker   const std::string private_ns_search_path = GetTestLibRoot() + "/private_namespace_libs";
1106*8d67ca89SAndroid Build Coastguard Worker 
1107*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_public =
1108*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("public",
1109*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1110*8d67ca89SAndroid Build Coastguard Worker                                    public_ns_search_path.c_str(),
1111*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1112*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1113*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1114*8d67ca89SAndroid Build Coastguard Worker 
1115*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_public, nullptr, g_core_shared_libs.c_str())) << dlerror();
1116*8d67ca89SAndroid Build Coastguard Worker 
1117*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_private =
1118*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private",
1119*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1120*8d67ca89SAndroid Build Coastguard Worker                                    private_ns_search_path.c_str(),
1121*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1122*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1123*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1124*8d67ca89SAndroid Build Coastguard Worker 
1125*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_private, ns_public, g_public_lib)) << dlerror();
1126*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_private, nullptr, g_core_shared_libs.c_str())) << dlerror();
1127*8d67ca89SAndroid Build Coastguard Worker 
1128*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
1129*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1130*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_private;
1131*8d67ca89SAndroid Build Coastguard Worker 
1132*8d67ca89SAndroid Build Coastguard Worker   void* handle = android_dlopen_ext(root_lib, RTLD_NOW, &extinfo);
1133*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
1134*8d67ca89SAndroid Build Coastguard Worker 
1135*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
1136*8d67ca89SAndroid Build Coastguard Worker   // Check that root_lib was unloaded
1137*8d67ca89SAndroid Build Coastguard Worker   handle = android_dlopen_ext(root_lib, RTLD_NOW | RTLD_NOLOAD, &extinfo);
1138*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == nullptr);
1139*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(std::string("dlopen failed: library \"") + root_lib +
1140*8d67ca89SAndroid Build Coastguard Worker             "\" wasn't loaded and RTLD_NOLOAD prevented it", dlerror());
1141*8d67ca89SAndroid Build Coastguard Worker 
1142*8d67ca89SAndroid Build Coastguard Worker   // Check that shared library was unloaded in public ns
1143*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_public;
1144*8d67ca89SAndroid Build Coastguard Worker   handle = android_dlopen_ext(g_public_lib, RTLD_NOW | RTLD_NOLOAD, &extinfo);
1145*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == nullptr);
1146*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(std::string("dlopen failed: library \"") + g_public_lib +
1147*8d67ca89SAndroid Build Coastguard Worker             "\" wasn't loaded and RTLD_NOLOAD prevented it", dlerror());
1148*8d67ca89SAndroid Build Coastguard Worker }
1149*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_unload_between_namespaces_missing_symbol_direct)1150*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_unload_between_namespaces_missing_symbol_direct) {
1151*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(g_core_shared_libs.c_str(), nullptr));
1152*8d67ca89SAndroid Build Coastguard Worker 
1153*8d67ca89SAndroid Build Coastguard Worker   const std::string public_ns_search_path =  GetTestLibRoot() + "/public_namespace_libs";
1154*8d67ca89SAndroid Build Coastguard Worker   const std::string private_ns_search_path = GetTestLibRoot() + "/private_namespace_libs";
1155*8d67ca89SAndroid Build Coastguard Worker 
1156*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_public =
1157*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("public",
1158*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1159*8d67ca89SAndroid Build Coastguard Worker                                    public_ns_search_path.c_str(),
1160*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1161*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1162*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1163*8d67ca89SAndroid Build Coastguard Worker 
1164*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_public, nullptr, g_core_shared_libs.c_str())) << dlerror();
1165*8d67ca89SAndroid Build Coastguard Worker 
1166*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_private =
1167*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private",
1168*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1169*8d67ca89SAndroid Build Coastguard Worker                                    private_ns_search_path.c_str(),
1170*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1171*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1172*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1173*8d67ca89SAndroid Build Coastguard Worker 
1174*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_private, ns_public, "libtest_missing_symbol.so")) << dlerror();
1175*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_private, nullptr, g_core_shared_libs.c_str())) << dlerror();
1176*8d67ca89SAndroid Build Coastguard Worker 
1177*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
1178*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1179*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_private;
1180*8d67ca89SAndroid Build Coastguard Worker 
1181*8d67ca89SAndroid Build Coastguard Worker   void* handle = android_dlopen_ext((public_ns_search_path + "/libtest_missing_symbol.so").c_str(),
1182*8d67ca89SAndroid Build Coastguard Worker                                     RTLD_NOW,
1183*8d67ca89SAndroid Build Coastguard Worker                                     &extinfo);
1184*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == nullptr);
1185*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(std::string("dlopen failed: cannot locate symbol \"dlopen_testlib_missing_symbol\" referenced by \"") +
1186*8d67ca89SAndroid Build Coastguard Worker             public_ns_search_path + "/libtest_missing_symbol.so\"...",
1187*8d67ca89SAndroid Build Coastguard Worker             dlerror());
1188*8d67ca89SAndroid Build Coastguard Worker }
1189*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_unload_between_namespaces_missing_symbol_indirect)1190*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_unload_between_namespaces_missing_symbol_indirect) {
1191*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(g_core_shared_libs.c_str(), nullptr));
1192*8d67ca89SAndroid Build Coastguard Worker 
1193*8d67ca89SAndroid Build Coastguard Worker   const std::string public_ns_search_path =  GetTestLibRoot() + "/public_namespace_libs";
1194*8d67ca89SAndroid Build Coastguard Worker   const std::string private_ns_search_path = GetTestLibRoot() + "/private_namespace_libs";
1195*8d67ca89SAndroid Build Coastguard Worker 
1196*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_public =
1197*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("public",
1198*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1199*8d67ca89SAndroid Build Coastguard Worker                                    public_ns_search_path.c_str(),
1200*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1201*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1202*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1203*8d67ca89SAndroid Build Coastguard Worker 
1204*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_public, nullptr, g_core_shared_libs.c_str())) << dlerror();
1205*8d67ca89SAndroid Build Coastguard Worker 
1206*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_private =
1207*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private",
1208*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1209*8d67ca89SAndroid Build Coastguard Worker                                    private_ns_search_path.c_str(),
1210*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1211*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1212*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1213*8d67ca89SAndroid Build Coastguard Worker 
1214*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_private,
1215*8d67ca89SAndroid Build Coastguard Worker                                       ns_public,
1216*8d67ca89SAndroid Build Coastguard Worker                                       "libnstest_public.so:libtest_missing_symbol_child_public.so")
1217*8d67ca89SAndroid Build Coastguard Worker               ) << dlerror();
1218*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_private, nullptr, g_core_shared_libs.c_str())) << dlerror();
1219*8d67ca89SAndroid Build Coastguard Worker 
1220*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
1221*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1222*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_private;
1223*8d67ca89SAndroid Build Coastguard Worker 
1224*8d67ca89SAndroid Build Coastguard Worker   void* handle = android_dlopen_ext("libtest_missing_symbol_root.so", RTLD_NOW, &extinfo);
1225*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == nullptr);
1226*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(std::string("dlopen failed: cannot locate symbol \"dlopen_testlib_missing_symbol\" referenced by \"") +
1227*8d67ca89SAndroid Build Coastguard Worker             private_ns_search_path + "/libtest_missing_symbol_root.so\"...",
1228*8d67ca89SAndroid Build Coastguard Worker             dlerror());
1229*8d67ca89SAndroid Build Coastguard Worker }
1230*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_exempt_list_enabled)1231*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_exempt_list_enabled) {
1232*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(g_core_shared_libs.c_str(), nullptr));
1233*8d67ca89SAndroid Build Coastguard Worker 
1234*8d67ca89SAndroid Build Coastguard Worker   const std::string ns_search_path = GetTestLibRoot() + "/private_namespace_libs";
1235*8d67ca89SAndroid Build Coastguard Worker 
1236*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns =
1237*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("namespace",
1238*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1239*8d67ca89SAndroid Build Coastguard Worker                                    ns_search_path.c_str(),
1240*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED | ANDROID_NAMESPACE_TYPE_EXEMPT_LIST_ENABLED,
1241*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1242*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1243*8d67ca89SAndroid Build Coastguard Worker 
1244*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns, nullptr, g_core_shared_libs.c_str())) << dlerror();
1245*8d67ca89SAndroid Build Coastguard Worker 
1246*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
1247*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1248*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns;
1249*8d67ca89SAndroid Build Coastguard Worker 
1250*8d67ca89SAndroid Build Coastguard Worker   // An app targeting M can open libnativehelper.so because it's on the exempt-list.
1251*8d67ca89SAndroid Build Coastguard Worker   android_set_application_target_sdk_version(23);
1252*8d67ca89SAndroid Build Coastguard Worker   void* handle = android_dlopen_ext("libnativehelper.so", RTLD_NOW, &extinfo);
1253*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
1254*8d67ca89SAndroid Build Coastguard Worker 
1255*8d67ca89SAndroid Build Coastguard Worker   // Check that loader did not load another copy of libdl.so while loading exempted library.
1256*8d67ca89SAndroid Build Coastguard Worker   void* dlsym_ptr = dlsym(handle, "dlsym");
1257*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(dlsym_ptr != nullptr) << dlerror();
1258*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(&dlsym, dlsym_ptr);
1259*8d67ca89SAndroid Build Coastguard Worker 
1260*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
1261*8d67ca89SAndroid Build Coastguard Worker 
1262*8d67ca89SAndroid Build Coastguard Worker   // An app targeting N no longer has the exempt-list.
1263*8d67ca89SAndroid Build Coastguard Worker   android_set_application_target_sdk_version(24);
1264*8d67ca89SAndroid Build Coastguard Worker   handle = android_dlopen_ext("libnativehelper.so", RTLD_NOW, &extinfo);
1265*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == nullptr);
1266*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("dlopen failed: library \"libnativehelper.so\" not found", dlerror());
1267*8d67ca89SAndroid Build Coastguard Worker }
1268*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_exempt_list_disabled_by_default)1269*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_exempt_list_disabled_by_default) {
1270*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(g_core_shared_libs.c_str(), nullptr));
1271*8d67ca89SAndroid Build Coastguard Worker 
1272*8d67ca89SAndroid Build Coastguard Worker   const std::string ns_search_path = GetTestLibRoot() + "/private_namespace_libs";
1273*8d67ca89SAndroid Build Coastguard Worker 
1274*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns =
1275*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("namespace",
1276*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1277*8d67ca89SAndroid Build Coastguard Worker                                    ns_search_path.c_str(),
1278*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1279*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1280*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1281*8d67ca89SAndroid Build Coastguard Worker 
1282*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns, nullptr, g_core_shared_libs.c_str())) << dlerror();
1283*8d67ca89SAndroid Build Coastguard Worker 
1284*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
1285*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1286*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns;
1287*8d67ca89SAndroid Build Coastguard Worker 
1288*8d67ca89SAndroid Build Coastguard Worker   android_set_application_target_sdk_version(23);
1289*8d67ca89SAndroid Build Coastguard Worker   void* handle = android_dlopen_ext("libnativehelper.so", RTLD_NOW, &extinfo);
1290*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == nullptr);
1291*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("dlopen failed: library \"libnativehelper.so\" not found", dlerror());
1292*8d67ca89SAndroid Build Coastguard Worker }
1293*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_cyclic_namespaces)1294*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_cyclic_namespaces) {
1295*8d67ca89SAndroid Build Coastguard Worker   // Test that ns1->ns2->ns1 link does not break the loader
1296*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(g_core_shared_libs.c_str(), nullptr));
1297*8d67ca89SAndroid Build Coastguard Worker   std::string shared_libs = g_core_shared_libs + ":libthatdoesnotexist.so";
1298*8d67ca89SAndroid Build Coastguard Worker 
1299*8d67ca89SAndroid Build Coastguard Worker   const std::string ns_search_path =  GetTestLibRoot() + "/public_namespace_libs";
1300*8d67ca89SAndroid Build Coastguard Worker 
1301*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns1 =
1302*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("ns1",
1303*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1304*8d67ca89SAndroid Build Coastguard Worker                                    ns_search_path.c_str(),
1305*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1306*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1307*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1308*8d67ca89SAndroid Build Coastguard Worker 
1309*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns1, nullptr, g_core_shared_libs.c_str())) << dlerror();
1310*8d67ca89SAndroid Build Coastguard Worker 
1311*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns2 =
1312*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("ns1",
1313*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1314*8d67ca89SAndroid Build Coastguard Worker                                    ns_search_path.c_str(),
1315*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1316*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1317*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1318*8d67ca89SAndroid Build Coastguard Worker 
1319*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns2, nullptr, g_core_shared_libs.c_str())) << dlerror();
1320*8d67ca89SAndroid Build Coastguard Worker 
1321*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns2, ns1, shared_libs.c_str())) << dlerror();
1322*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns1, ns2, shared_libs.c_str())) << dlerror();
1323*8d67ca89SAndroid Build Coastguard Worker 
1324*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
1325*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1326*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns1;
1327*8d67ca89SAndroid Build Coastguard Worker 
1328*8d67ca89SAndroid Build Coastguard Worker   void* handle = android_dlopen_ext("libthatdoesnotexist.so", RTLD_NOW, &extinfo);
1329*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == nullptr);
1330*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("dlopen failed: library \"libthatdoesnotexist.so\" not found", dlerror());
1331*8d67ca89SAndroid Build Coastguard Worker }
1332*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_isolated)1333*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_isolated) {
1334*8d67ca89SAndroid Build Coastguard Worker   static const char* root_lib = "libnstest_root_not_isolated.so";
1335*8d67ca89SAndroid Build Coastguard Worker   std::string shared_libs = g_core_shared_libs + ":" + g_public_lib;
1336*8d67ca89SAndroid Build Coastguard Worker 
1337*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_public_path = GetTestLibRoot() + "/public_namespace_libs/" + g_public_lib;
1338*8d67ca89SAndroid Build Coastguard Worker   void* handle_public = dlopen(lib_public_path.c_str(), RTLD_NOW);
1339*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_public != nullptr) << dlerror();
1340*8d67ca89SAndroid Build Coastguard Worker 
1341*8d67ca89SAndroid Build Coastguard Worker   android_set_application_target_sdk_version(42U); // something > 23
1342*8d67ca89SAndroid Build Coastguard Worker 
1343*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(shared_libs.c_str(), nullptr)) << dlerror();
1344*8d67ca89SAndroid Build Coastguard Worker 
1345*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_not_isolated =
1346*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private",
1347*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1348*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/private_namespace_libs").c_str(),
1349*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_REGULAR,
1350*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1351*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1352*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_not_isolated != nullptr) << dlerror();
1353*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_not_isolated, nullptr, shared_libs.c_str())) << dlerror();
1354*8d67ca89SAndroid Build Coastguard Worker 
1355*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_isolated =
1356*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private_isolated1",
1357*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1358*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/private_namespace_libs").c_str(),
1359*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1360*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1361*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1362*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_isolated != nullptr) << dlerror();
1363*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_isolated, nullptr, shared_libs.c_str())) << dlerror();
1364*8d67ca89SAndroid Build Coastguard Worker 
1365*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_isolated2 =
1366*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private_isolated2",
1367*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/private_namespace_libs").c_str(),
1368*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1369*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1370*8d67ca89SAndroid Build Coastguard Worker                                    GetTestLibRoot().c_str(),
1371*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1372*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_isolated2 != nullptr) << dlerror();
1373*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_isolated2, nullptr, shared_libs.c_str())) << dlerror();
1374*8d67ca89SAndroid Build Coastguard Worker 
1375*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(dlopen(root_lib, RTLD_NOW) == nullptr);
1376*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("dlopen failed: library \"libnstest_root_not_isolated.so\" not found", dlerror());
1377*8d67ca89SAndroid Build Coastguard Worker 
1378*8d67ca89SAndroid Build Coastguard Worker   std::string lib_private_external_path =
1379*8d67ca89SAndroid Build Coastguard Worker       GetTestLibRoot() + "/private_namespace_libs_external/libnstest_private_external.so";
1380*8d67ca89SAndroid Build Coastguard Worker 
1381*8d67ca89SAndroid Build Coastguard Worker   // Load lib_private_external_path to default namespace
1382*8d67ca89SAndroid Build Coastguard Worker   // (it should remain invisible for the isolated namespaces after this)
1383*8d67ca89SAndroid Build Coastguard Worker   void* handle = dlopen(lib_private_external_path.c_str(), RTLD_NOW);
1384*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
1385*8d67ca89SAndroid Build Coastguard Worker 
1386*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
1387*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1388*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_not_isolated;
1389*8d67ca89SAndroid Build Coastguard Worker 
1390*8d67ca89SAndroid Build Coastguard Worker   void* handle1 = android_dlopen_ext(root_lib, RTLD_NOW, &extinfo);
1391*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle1 != nullptr) << dlerror();
1392*8d67ca89SAndroid Build Coastguard Worker 
1393*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_isolated;
1394*8d67ca89SAndroid Build Coastguard Worker 
1395*8d67ca89SAndroid Build Coastguard Worker   void* handle2 = android_dlopen_ext(root_lib, RTLD_NOW, &extinfo);
1396*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle2 == nullptr);
1397*8d67ca89SAndroid Build Coastguard Worker   const char* error = dlerror();
1398*8d67ca89SAndroid Build Coastguard Worker   ASSERT_MATCH(error,
1399*8d67ca89SAndroid Build Coastguard Worker                R"(dlopen failed: library "libnstest_private_external.so" not found: needed by )"
1400*8d67ca89SAndroid Build Coastguard Worker                R"(\S+libnstest_root_not_isolated.so in namespace private_isolated1)");
1401*8d67ca89SAndroid Build Coastguard Worker 
1402*8d67ca89SAndroid Build Coastguard Worker   // Check dlopen by absolute path
1403*8d67ca89SAndroid Build Coastguard Worker   handle2 = android_dlopen_ext(lib_private_external_path.c_str(), RTLD_NOW, &extinfo);
1404*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle2 == nullptr);
1405*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ("dlopen failed: library \"" + lib_private_external_path + "\" needed"
1406*8d67ca89SAndroid Build Coastguard Worker             " or dlopened by \"" + android::base::GetExecutablePath() +  "\" is not accessible"
1407*8d67ca89SAndroid Build Coastguard Worker             " for the namespace \"private_isolated1\"", dlerror());
1408*8d67ca89SAndroid Build Coastguard Worker 
1409*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_isolated2;
1410*8d67ca89SAndroid Build Coastguard Worker 
1411*8d67ca89SAndroid Build Coastguard Worker   // this should work because isolation_path for private_isolated2 includes GetTestLibRoot()
1412*8d67ca89SAndroid Build Coastguard Worker   handle2 = android_dlopen_ext(root_lib, RTLD_NOW, &extinfo);
1413*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle2 != nullptr) << dlerror();
1414*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle2);
1415*8d67ca89SAndroid Build Coastguard Worker 
1416*8d67ca89SAndroid Build Coastguard Worker   // Check dlopen by absolute path
1417*8d67ca89SAndroid Build Coastguard Worker   handle2 = android_dlopen_ext(lib_private_external_path.c_str(), RTLD_NOW, &extinfo);
1418*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle2 != nullptr) << dlerror();
1419*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle2);
1420*8d67ca89SAndroid Build Coastguard Worker 
1421*8d67ca89SAndroid Build Coastguard Worker   typedef const char* (*fn_t)();
1422*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_local_string = reinterpret_cast<fn_t>(dlsym(handle1, "ns_get_local_string"));
1423*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_local_string != nullptr) << dlerror();
1424*8d67ca89SAndroid Build Coastguard Worker 
1425*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is local to root library", ns_get_local_string());
1426*8d67ca89SAndroid Build Coastguard Worker 
1427*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_private_extern_string =
1428*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<fn_t>(dlsym(handle1, "ns_get_private_extern_string"));
1429*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_private_extern_string != nullptr) << dlerror();
1430*8d67ca89SAndroid Build Coastguard Worker 
1431*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from private namespace", ns_get_private_extern_string());
1432*8d67ca89SAndroid Build Coastguard Worker 
1433*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_public_extern_string =
1434*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<fn_t>(dlsym(handle1, "ns_get_public_extern_string"));
1435*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_public_extern_string != nullptr) << dlerror();
1436*8d67ca89SAndroid Build Coastguard Worker 
1437*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from public namespace", ns_get_public_extern_string());
1438*8d67ca89SAndroid Build Coastguard Worker 
1439*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_dlopened_string = reinterpret_cast<fn_t>(dlsym(handle1, "ns_get_dlopened_string"));
1440*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_dlopened_string != nullptr) << dlerror();
1441*8d67ca89SAndroid Build Coastguard Worker 
1442*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from private namespace (dlopened library)", ns_get_dlopened_string());
1443*8d67ca89SAndroid Build Coastguard Worker 
1444*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle1);
1445*8d67ca89SAndroid Build Coastguard Worker }
1446*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_shared)1447*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_shared) {
1448*8d67ca89SAndroid Build Coastguard Worker   static const char* root_lib = "libnstest_root_not_isolated.so";
1449*8d67ca89SAndroid Build Coastguard Worker   static const char* root_lib_isolated = "libnstest_root.so";
1450*8d67ca89SAndroid Build Coastguard Worker 
1451*8d67ca89SAndroid Build Coastguard Worker   std::string shared_libs = g_core_shared_libs + ":" + g_public_lib;
1452*8d67ca89SAndroid Build Coastguard Worker 
1453*8d67ca89SAndroid Build Coastguard Worker   // create a parent namespace to use instead of the default namespace. This is
1454*8d67ca89SAndroid Build Coastguard Worker   // to make this test be independent from the configuration of the default
1455*8d67ca89SAndroid Build Coastguard Worker   // namespace.
1456*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_parent =
1457*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("parent",
1458*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1459*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1460*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_REGULAR,
1461*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1462*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1463*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_parent != nullptr) << dlerror();
1464*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_parent, nullptr, g_core_shared_libs.c_str())) << dlerror();
1465*8d67ca89SAndroid Build Coastguard Worker 
1466*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
1467*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1468*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_parent;
1469*8d67ca89SAndroid Build Coastguard Worker 
1470*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_public_path = GetTestLibRoot() + "/public_namespace_libs/" + g_public_lib;
1471*8d67ca89SAndroid Build Coastguard Worker   void* handle_public = android_dlopen_ext(lib_public_path.c_str(), RTLD_NOW, &extinfo);
1472*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_public != nullptr) << dlerror();
1473*8d67ca89SAndroid Build Coastguard Worker 
1474*8d67ca89SAndroid Build Coastguard Worker   android_set_application_target_sdk_version(42U); // something > 23
1475*8d67ca89SAndroid Build Coastguard Worker 
1476*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(shared_libs.c_str(), nullptr)) << dlerror();
1477*8d67ca89SAndroid Build Coastguard Worker 
1478*8d67ca89SAndroid Build Coastguard Worker   // preload this library to the parent namespace to check if it
1479*8d67ca89SAndroid Build Coastguard Worker   // is shared later on.
1480*8d67ca89SAndroid Build Coastguard Worker   void* handle_dlopened =
1481*8d67ca89SAndroid Build Coastguard Worker           android_dlopen_ext((GetTestLibRoot() + "/private_namespace_libs/libnstest_dlopened.so").c_str(), RTLD_NOW, &extinfo);
1482*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_dlopened != nullptr) << dlerror();
1483*8d67ca89SAndroid Build Coastguard Worker 
1484*8d67ca89SAndroid Build Coastguard Worker   // create two child namespaces of 'ns_parent'. One with regular, the other
1485*8d67ca89SAndroid Build Coastguard Worker   // with isolated & shared.
1486*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_not_isolated =
1487*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private",
1488*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1489*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/private_namespace_libs").c_str(),
1490*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_REGULAR,
1491*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1492*8d67ca89SAndroid Build Coastguard Worker                                    ns_parent);
1493*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_not_isolated != nullptr) << dlerror();
1494*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_not_isolated, ns_parent, g_public_lib)) << dlerror();
1495*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_not_isolated, nullptr, g_core_shared_libs.c_str())) << dlerror();
1496*8d67ca89SAndroid Build Coastguard Worker 
1497*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_isolated_shared =
1498*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private_isolated_shared",
1499*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1500*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/private_namespace_libs").c_str(),
1501*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED | ANDROID_NAMESPACE_TYPE_SHARED,
1502*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1503*8d67ca89SAndroid Build Coastguard Worker                                    ns_parent);
1504*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_isolated_shared != nullptr) << dlerror();
1505*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_isolated_shared, ns_parent, g_public_lib)) << dlerror();
1506*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_isolated_shared, nullptr, g_core_shared_libs.c_str())) << dlerror();
1507*8d67ca89SAndroid Build Coastguard Worker 
1508*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_dlopen_ext(root_lib, RTLD_NOW, &extinfo) == nullptr);
1509*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("dlopen failed: library \"libnstest_root_not_isolated.so\" not found", dlerror());
1510*8d67ca89SAndroid Build Coastguard Worker 
1511*8d67ca89SAndroid Build Coastguard Worker   std::string lib_private_external_path =
1512*8d67ca89SAndroid Build Coastguard Worker       GetTestLibRoot() + "/private_namespace_libs_external/libnstest_private_external.so";
1513*8d67ca89SAndroid Build Coastguard Worker 
1514*8d67ca89SAndroid Build Coastguard Worker   // Load lib_private_external_path to the parent namespace
1515*8d67ca89SAndroid Build Coastguard Worker   // (it should remain invisible for the isolated namespaces after this)
1516*8d67ca89SAndroid Build Coastguard Worker   void* handle = android_dlopen_ext(lib_private_external_path.c_str(), RTLD_NOW, &extinfo);
1517*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
1518*8d67ca89SAndroid Build Coastguard Worker 
1519*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_not_isolated;
1520*8d67ca89SAndroid Build Coastguard Worker 
1521*8d67ca89SAndroid Build Coastguard Worker   void* handle1 = android_dlopen_ext(root_lib, RTLD_NOW, &extinfo);
1522*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle1 != nullptr) << dlerror();
1523*8d67ca89SAndroid Build Coastguard Worker 
1524*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_isolated_shared;
1525*8d67ca89SAndroid Build Coastguard Worker 
1526*8d67ca89SAndroid Build Coastguard Worker   void* handle2 = android_dlopen_ext(root_lib, RTLD_NOW, &extinfo);
1527*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle2 == nullptr);
1528*8d67ca89SAndroid Build Coastguard Worker   ASSERT_MATCH(dlerror(),
1529*8d67ca89SAndroid Build Coastguard Worker                R"(dlopen failed: library "libnstest_private_external.so" not found: needed by )"
1530*8d67ca89SAndroid Build Coastguard Worker                R"(\S+libnstest_root_not_isolated.so in namespace private_isolated_shared)");
1531*8d67ca89SAndroid Build Coastguard Worker 
1532*8d67ca89SAndroid Build Coastguard Worker   // Check dlopen by absolute path
1533*8d67ca89SAndroid Build Coastguard Worker   handle2 = android_dlopen_ext(lib_private_external_path.c_str(), RTLD_NOW, &extinfo);
1534*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle2 == nullptr);
1535*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ("dlopen failed: library \"" + lib_private_external_path + "\" needed"
1536*8d67ca89SAndroid Build Coastguard Worker             " or dlopened by \"" + android::base::GetExecutablePath() + "\" is not accessible"
1537*8d67ca89SAndroid Build Coastguard Worker             " for the namespace \"private_isolated_shared\"", dlerror());
1538*8d67ca89SAndroid Build Coastguard Worker 
1539*8d67ca89SAndroid Build Coastguard Worker   // load libnstest_root.so to shared namespace in order to check that everything is different
1540*8d67ca89SAndroid Build Coastguard Worker   // except shared libnstest_dlopened.so
1541*8d67ca89SAndroid Build Coastguard Worker 
1542*8d67ca89SAndroid Build Coastguard Worker   handle2 = android_dlopen_ext(root_lib_isolated, RTLD_NOW, &extinfo);
1543*8d67ca89SAndroid Build Coastguard Worker 
1544*8d67ca89SAndroid Build Coastguard Worker   typedef const char* (*fn_t)();
1545*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_local_string = reinterpret_cast<fn_t>(dlsym(handle1, "ns_get_local_string"));
1546*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_local_string != nullptr) << dlerror();
1547*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_local_string_shared = reinterpret_cast<fn_t>(dlsym(handle2, "ns_get_local_string"));
1548*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_local_string_shared != nullptr) << dlerror();
1549*8d67ca89SAndroid Build Coastguard Worker 
1550*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is local to root library", ns_get_local_string());
1551*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is local to root library", ns_get_local_string_shared());
1552*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_local_string() != ns_get_local_string_shared());
1553*8d67ca89SAndroid Build Coastguard Worker 
1554*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_private_extern_string =
1555*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<fn_t>(dlsym(handle1, "ns_get_private_extern_string"));
1556*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_private_extern_string != nullptr) << dlerror();
1557*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_private_extern_string_shared =
1558*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<fn_t>(dlsym(handle2, "ns_get_private_extern_string"));
1559*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_private_extern_string_shared() != nullptr) << dlerror();
1560*8d67ca89SAndroid Build Coastguard Worker 
1561*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from private namespace", ns_get_private_extern_string());
1562*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from private namespace", ns_get_private_extern_string_shared());
1563*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_private_extern_string() != ns_get_private_extern_string_shared());
1564*8d67ca89SAndroid Build Coastguard Worker 
1565*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_public_extern_string =
1566*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<fn_t>(dlsym(handle1, "ns_get_public_extern_string"));
1567*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_public_extern_string != nullptr) << dlerror();
1568*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_public_extern_string_shared =
1569*8d67ca89SAndroid Build Coastguard Worker           reinterpret_cast<fn_t>(dlsym(handle2, "ns_get_public_extern_string"));
1570*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_public_extern_string_shared != nullptr) << dlerror();
1571*8d67ca89SAndroid Build Coastguard Worker 
1572*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from public namespace", ns_get_public_extern_string());
1573*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from public namespace", ns_get_public_extern_string_shared());
1574*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_public_extern_string() == ns_get_public_extern_string_shared());
1575*8d67ca89SAndroid Build Coastguard Worker 
1576*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_dlopened_string = reinterpret_cast<fn_t>(dlsym(handle1, "ns_get_dlopened_string"));
1577*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_dlopened_string != nullptr) << dlerror();
1578*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_dlopened_string_shared = reinterpret_cast<fn_t>(dlsym(handle2, "ns_get_dlopened_string"));
1579*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_dlopened_string_shared != nullptr) << dlerror();
1580*8d67ca89SAndroid Build Coastguard Worker   const char** ns_dlopened_string = static_cast<const char**>(dlsym(handle_dlopened, "g_private_dlopened_string"));
1581*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_dlopened_string != nullptr) << dlerror();
1582*8d67ca89SAndroid Build Coastguard Worker 
1583*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from private namespace (dlopened library)", ns_get_dlopened_string());
1584*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from private namespace (dlopened library)", *ns_dlopened_string);
1585*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from private namespace (dlopened library)", ns_get_dlopened_string_shared());
1586*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_dlopened_string() != ns_get_dlopened_string_shared());
1587*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(*ns_dlopened_string == ns_get_dlopened_string_shared());
1588*8d67ca89SAndroid Build Coastguard Worker 
1589*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle1);
1590*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle2);
1591*8d67ca89SAndroid Build Coastguard Worker }
1592*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_shared_links_and_paths)1593*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_shared_links_and_paths) {
1594*8d67ca89SAndroid Build Coastguard Worker   // Create parent namespace (isolated, not shared)
1595*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_isolated =
1596*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private_isolated",
1597*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1598*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/private_namespace_libs").c_str(),
1599*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1600*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/public_namespace_libs").c_str(),
1601*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1602*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_isolated != nullptr) << dlerror();
1603*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_isolated, nullptr, g_core_shared_libs.c_str())) << dlerror();
1604*8d67ca89SAndroid Build Coastguard Worker 
1605*8d67ca89SAndroid Build Coastguard Worker   // Create shared namespace with ns_isolated parent
1606*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_shared =
1607*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private_shared",
1608*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1609*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1610*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_SHARED | ANDROID_NAMESPACE_TYPE_ISOLATED,
1611*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1612*8d67ca89SAndroid Build Coastguard Worker                                    ns_isolated);
1613*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_shared != nullptr) << dlerror();
1614*8d67ca89SAndroid Build Coastguard Worker 
1615*8d67ca89SAndroid Build Coastguard Worker   // 1. Load a library in ns_shared to check that it has inherited
1616*8d67ca89SAndroid Build Coastguard Worker   // search path and the link to the default namespace.
1617*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
1618*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1619*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_shared;
1620*8d67ca89SAndroid Build Coastguard Worker 
1621*8d67ca89SAndroid Build Coastguard Worker   {
1622*8d67ca89SAndroid Build Coastguard Worker     void* handle = android_dlopen_ext("libnstest_private.so", RTLD_NOW, &extinfo);
1623*8d67ca89SAndroid Build Coastguard Worker     ASSERT_TRUE(handle != nullptr) << dlerror();
1624*8d67ca89SAndroid Build Coastguard Worker     const char** ns_private_extern_string = static_cast<const char**>(dlsym(handle, "g_private_extern_string"));
1625*8d67ca89SAndroid Build Coastguard Worker     ASSERT_TRUE(ns_private_extern_string != nullptr) << dlerror();
1626*8d67ca89SAndroid Build Coastguard Worker     ASSERT_STREQ("This string is from private namespace", *ns_private_extern_string);
1627*8d67ca89SAndroid Build Coastguard Worker 
1628*8d67ca89SAndroid Build Coastguard Worker     dlclose(handle);
1629*8d67ca89SAndroid Build Coastguard Worker   }
1630*8d67ca89SAndroid Build Coastguard Worker   // 2. Load another test library by absolute path to check that
1631*8d67ca89SAndroid Build Coastguard Worker   // it has inherited permitted_when_isolated_path
1632*8d67ca89SAndroid Build Coastguard Worker   {
1633*8d67ca89SAndroid Build Coastguard Worker     void* handle = android_dlopen_ext(
1634*8d67ca89SAndroid Build Coastguard Worker             (GetTestLibRoot() + "/public_namespace_libs/libnstest_public.so").c_str(),
1635*8d67ca89SAndroid Build Coastguard Worker             RTLD_NOW,
1636*8d67ca89SAndroid Build Coastguard Worker             &extinfo);
1637*8d67ca89SAndroid Build Coastguard Worker 
1638*8d67ca89SAndroid Build Coastguard Worker     ASSERT_TRUE(handle != nullptr) << dlerror();
1639*8d67ca89SAndroid Build Coastguard Worker     const char** ns_public_extern_string = static_cast<const char**>(dlsym(handle, "g_public_extern_string"));
1640*8d67ca89SAndroid Build Coastguard Worker     ASSERT_TRUE(ns_public_extern_string != nullptr) << dlerror();
1641*8d67ca89SAndroid Build Coastguard Worker     ASSERT_STREQ("This string is from public namespace", *ns_public_extern_string);
1642*8d67ca89SAndroid Build Coastguard Worker 
1643*8d67ca89SAndroid Build Coastguard Worker     dlclose(handle);
1644*8d67ca89SAndroid Build Coastguard Worker   }
1645*8d67ca89SAndroid Build Coastguard Worker 
1646*8d67ca89SAndroid Build Coastguard Worker   // 3. Check that it is still isolated.
1647*8d67ca89SAndroid Build Coastguard Worker   {
1648*8d67ca89SAndroid Build Coastguard Worker     void* handle = android_dlopen_ext(
1649*8d67ca89SAndroid Build Coastguard Worker             (GetTestLibRoot() + "/libtest_empty.so").c_str(),
1650*8d67ca89SAndroid Build Coastguard Worker             RTLD_NOW,
1651*8d67ca89SAndroid Build Coastguard Worker             &extinfo);
1652*8d67ca89SAndroid Build Coastguard Worker 
1653*8d67ca89SAndroid Build Coastguard Worker     ASSERT_TRUE(handle == nullptr);
1654*8d67ca89SAndroid Build Coastguard Worker   }
1655*8d67ca89SAndroid Build Coastguard Worker }
1656*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_shared_dlclose)1657*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_shared_dlclose) {
1658*8d67ca89SAndroid Build Coastguard Worker   android_set_application_target_sdk_version(42U); // something > 23
1659*8d67ca89SAndroid Build Coastguard Worker 
1660*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(g_core_shared_libs.c_str(), nullptr)) << dlerror();
1661*8d67ca89SAndroid Build Coastguard Worker 
1662*8d67ca89SAndroid Build Coastguard Worker   // preload this library to the default namespace to check if it
1663*8d67ca89SAndroid Build Coastguard Worker   // is shared later on.
1664*8d67ca89SAndroid Build Coastguard Worker   void* handle_dlopened =
1665*8d67ca89SAndroid Build Coastguard Worker           dlopen((GetTestLibRoot() + "/private_namespace_libs/libnstest_dlopened.so").c_str(), RTLD_NOW);
1666*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_dlopened != nullptr) << dlerror();
1667*8d67ca89SAndroid Build Coastguard Worker 
1668*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_isolated_shared =
1669*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private_isolated_shared",
1670*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1671*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/private_namespace_libs").c_str(),
1672*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED | ANDROID_NAMESPACE_TYPE_SHARED,
1673*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1674*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1675*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_isolated_shared != nullptr) << dlerror();
1676*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_isolated_shared, nullptr, g_core_shared_libs.c_str())) << dlerror();
1677*8d67ca89SAndroid Build Coastguard Worker 
1678*8d67ca89SAndroid Build Coastguard Worker   // Check if "libnstest_dlopened.so" is loaded (and the same)
1679*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
1680*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1681*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_isolated_shared;
1682*8d67ca89SAndroid Build Coastguard Worker 
1683*8d67ca89SAndroid Build Coastguard Worker   void* handle = android_dlopen_ext("libnstest_dlopened.so", RTLD_NOW | RTLD_NOLOAD, &extinfo);
1684*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
1685*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == handle_dlopened);
1686*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
1687*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle_dlopened);
1688*8d67ca89SAndroid Build Coastguard Worker 
1689*8d67ca89SAndroid Build Coastguard Worker   // And now check that the library cannot be found by soname (and is no longer loaded)
1690*8d67ca89SAndroid Build Coastguard Worker   handle = android_dlopen_ext("libnstest_dlopened.so", RTLD_NOW | RTLD_NOLOAD, &extinfo);
1691*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == nullptr)
1692*8d67ca89SAndroid Build Coastguard Worker       << "Error: libnstest_dlopened.so is still accessible in shared namespace";
1693*8d67ca89SAndroid Build Coastguard Worker 
1694*8d67ca89SAndroid Build Coastguard Worker   handle = android_dlopen_ext((GetTestLibRoot() + "/private_namespace_libs/libnstest_dlopened.so").c_str(),
1695*8d67ca89SAndroid Build Coastguard Worker                               RTLD_NOW | RTLD_NOLOAD, &extinfo);
1696*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == nullptr)
1697*8d67ca89SAndroid Build Coastguard Worker       << "Error: libnstest_dlopened.so is still accessible in shared namespace";
1698*8d67ca89SAndroid Build Coastguard Worker 
1699*8d67ca89SAndroid Build Coastguard Worker   handle = dlopen("libnstest_dlopened.so", RTLD_NOW | RTLD_NOLOAD);
1700*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == nullptr)
1701*8d67ca89SAndroid Build Coastguard Worker       << "Error: libnstest_dlopened.so is still accessible in default namespace";
1702*8d67ca89SAndroid Build Coastguard Worker 
1703*8d67ca89SAndroid Build Coastguard Worker   handle = dlopen((GetTestLibRoot() + "/private_namespace_libs/libnstest_dlopened.so").c_str(),
1704*8d67ca89SAndroid Build Coastguard Worker                   RTLD_NOW | RTLD_NOLOAD);
1705*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == nullptr)
1706*8d67ca89SAndroid Build Coastguard Worker       << "Error: libnstest_dlopened.so is still accessible in default namespace";
1707*8d67ca89SAndroid Build Coastguard Worker 
1708*8d67ca89SAndroid Build Coastguard Worker   // Now lets see if the soinfo area gets reused in the wrong way:
1709*8d67ca89SAndroid Build Coastguard Worker   // load a library to default namespace.
1710*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_public_path = GetTestLibRoot() + "/public_namespace_libs/" + g_public_lib;
1711*8d67ca89SAndroid Build Coastguard Worker   void* handle_public = dlopen(lib_public_path.c_str(), RTLD_NOW);
1712*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_public != nullptr) << dlerror();
1713*8d67ca89SAndroid Build Coastguard Worker 
1714*8d67ca89SAndroid Build Coastguard Worker   // try to find it in shared namespace
1715*8d67ca89SAndroid Build Coastguard Worker   handle = android_dlopen_ext(g_public_lib, RTLD_NOW | RTLD_NOLOAD, &extinfo);
1716*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == nullptr)
1717*8d67ca89SAndroid Build Coastguard Worker       << "Error: " << g_public_lib << " is accessible in shared namespace";
1718*8d67ca89SAndroid Build Coastguard Worker }
1719*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_isolated_rtld_global)1720*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_isolated_rtld_global) {
1721*8d67ca89SAndroid Build Coastguard Worker   static const char* root_lib = "libnstest_root.so";
1722*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(g_core_shared_libs.c_str(), nullptr));
1723*8d67ca89SAndroid Build Coastguard Worker 
1724*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_public_path = GetTestLibRoot() + "/public_namespace_libs";
1725*8d67ca89SAndroid Build Coastguard Worker 
1726*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns1 =
1727*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("isolated1",
1728*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1729*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/private_namespace_libs").c_str(),
1730*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1731*8d67ca89SAndroid Build Coastguard Worker                                    lib_public_path.c_str(),
1732*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1733*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns1 != nullptr) << dlerror();
1734*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns1, nullptr, g_core_shared_libs.c_str())) << dlerror();
1735*8d67ca89SAndroid Build Coastguard Worker 
1736*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns2 =
1737*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("isolated2",
1738*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1739*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/private_namespace_libs").c_str(),
1740*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1741*8d67ca89SAndroid Build Coastguard Worker                                    lib_public_path.c_str(),
1742*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1743*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns2 != nullptr) << dlerror();
1744*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns2, nullptr, g_core_shared_libs.c_str())) << dlerror();
1745*8d67ca89SAndroid Build Coastguard Worker 
1746*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
1747*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1748*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns1;
1749*8d67ca89SAndroid Build Coastguard Worker 
1750*8d67ca89SAndroid Build Coastguard Worker   void* handle_global = android_dlopen_ext((lib_public_path + "/" + g_public_lib).c_str(),
1751*8d67ca89SAndroid Build Coastguard Worker                                            RTLD_GLOBAL,
1752*8d67ca89SAndroid Build Coastguard Worker                                            &extinfo);
1753*8d67ca89SAndroid Build Coastguard Worker 
1754*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_global != nullptr) << dlerror();
1755*8d67ca89SAndroid Build Coastguard Worker 
1756*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns1_child =
1757*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("isolated1_child",
1758*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1759*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/private_namespace_libs").c_str(),
1760*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1761*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1762*8d67ca89SAndroid Build Coastguard Worker                                    ns1);
1763*8d67ca89SAndroid Build Coastguard Worker 
1764*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns1_child != nullptr) << dlerror();
1765*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns1_child, nullptr, g_core_shared_libs.c_str())) << dlerror();
1766*8d67ca89SAndroid Build Coastguard Worker 
1767*8d67ca89SAndroid Build Coastguard Worker   // Now - only ns1 and ns1 child should be able to dlopen root_lib
1768*8d67ca89SAndroid Build Coastguard Worker   // attempt to use ns2 should result in dlerror()
1769*8d67ca89SAndroid Build Coastguard Worker 
1770*8d67ca89SAndroid Build Coastguard Worker   // Check ns1_child first.
1771*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1772*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns1_child;
1773*8d67ca89SAndroid Build Coastguard Worker 
1774*8d67ca89SAndroid Build Coastguard Worker   void* handle1 = android_dlopen_ext(root_lib, RTLD_NOW, &extinfo);
1775*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle1 != nullptr) << dlerror();
1776*8d67ca89SAndroid Build Coastguard Worker 
1777*8d67ca89SAndroid Build Coastguard Worker   // now ns1
1778*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1779*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns1;
1780*8d67ca89SAndroid Build Coastguard Worker 
1781*8d67ca89SAndroid Build Coastguard Worker   handle1 = android_dlopen_ext(root_lib, RTLD_NOW, &extinfo);
1782*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle1 != nullptr) << dlerror();
1783*8d67ca89SAndroid Build Coastguard Worker 
1784*8d67ca89SAndroid Build Coastguard Worker   // and ns2 should fail
1785*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1786*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns2;
1787*8d67ca89SAndroid Build Coastguard Worker 
1788*8d67ca89SAndroid Build Coastguard Worker   handle1 = android_dlopen_ext(root_lib, RTLD_NOW, &extinfo);
1789*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle1 == nullptr);
1790*8d67ca89SAndroid Build Coastguard Worker   ASSERT_MATCH(
1791*8d67ca89SAndroid Build Coastguard Worker       dlerror(),
1792*8d67ca89SAndroid Build Coastguard Worker       R"(dlopen failed: library "libnstest_public.so" not found: needed by \S+libnstest_root.so)"
1793*8d67ca89SAndroid Build Coastguard Worker       R"( in namespace isolated2)");
1794*8d67ca89SAndroid Build Coastguard Worker }
1795*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_inaccessible_error_message)1796*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_inaccessible_error_message) {
1797*8d67ca89SAndroid Build Coastguard Worker   // We set up 2 namespaces (a and b) and link a->b with a shared library
1798*8d67ca89SAndroid Build Coastguard Worker   // libtestshared.so. Then try to dlopen different library with the same
1799*8d67ca89SAndroid Build Coastguard Worker   // name from in namespace a. Note that library should not be accessible
1800*8d67ca89SAndroid Build Coastguard Worker   // in either namespace but since it's soname is in the list of shared libs
1801*8d67ca89SAndroid Build Coastguard Worker   // the linker will attempt to find it in linked namespace.
1802*8d67ca89SAndroid Build Coastguard Worker   //
1803*8d67ca89SAndroid Build Coastguard Worker   // Check the error message and make sure it mentions correct namespace name.
1804*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(g_core_shared_libs.c_str(), nullptr));
1805*8d67ca89SAndroid Build Coastguard Worker 
1806*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_a =
1807*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("ns_a",
1808*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1809*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/private_namespace_libs").c_str(),
1810*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1811*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1812*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1813*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_a != nullptr) << dlerror();
1814*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_a, nullptr, g_core_shared_libs.c_str())) << dlerror();
1815*8d67ca89SAndroid Build Coastguard Worker 
1816*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_b =
1817*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("ns_b",
1818*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1819*8d67ca89SAndroid Build Coastguard Worker                                    GetTestLibRoot().c_str(),
1820*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1821*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1822*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1823*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_b != nullptr) << dlerror();
1824*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_b, nullptr, g_core_shared_libs.c_str())) << dlerror();
1825*8d67ca89SAndroid Build Coastguard Worker 
1826*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_a, ns_b, "libtestshared.so")) << dlerror();
1827*8d67ca89SAndroid Build Coastguard Worker 
1828*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
1829*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1830*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_a;
1831*8d67ca89SAndroid Build Coastguard Worker 
1832*8d67ca89SAndroid Build Coastguard Worker   std::string library_path = GetTestLibRoot() + "/inaccessible_libs/libtestshared.so";
1833*8d67ca89SAndroid Build Coastguard Worker 
1834*8d67ca89SAndroid Build Coastguard Worker   void* handle = android_dlopen_ext(library_path.c_str(), RTLD_NOW, &extinfo);
1835*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle == nullptr);
1836*8d67ca89SAndroid Build Coastguard Worker   std::string expected_dlerror =
1837*8d67ca89SAndroid Build Coastguard Worker       android::base::StringPrintf("dlopen failed: library \"%s\" needed or dlopened by \"%s\""
1838*8d67ca89SAndroid Build Coastguard Worker                                   " is not accessible for the namespace \"ns_a\"",
1839*8d67ca89SAndroid Build Coastguard Worker                                   library_path.c_str(),
1840*8d67ca89SAndroid Build Coastguard Worker                                   android::base::GetExecutablePath().c_str());
1841*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(expected_dlerror, dlerror());
1842*8d67ca89SAndroid Build Coastguard Worker }
1843*8d67ca89SAndroid Build Coastguard Worker 
1844*8d67ca89SAndroid Build Coastguard Worker extern "C" bool __loader_android_link_namespaces_all_libs(android_namespace_t* namespace_from,
1845*8d67ca89SAndroid Build Coastguard Worker                                                           android_namespace_t* namespace_to);
1846*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_link_namespaces_invalid_arguments)1847*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_link_namespaces_invalid_arguments) {
1848*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(g_core_shared_libs.c_str(), nullptr));
1849*8d67ca89SAndroid Build Coastguard Worker 
1850*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns =
1851*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private",
1852*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1853*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/private_namespace_libs").c_str(),
1854*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_REGULAR,
1855*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1856*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1857*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns != nullptr) << dlerror();
1858*8d67ca89SAndroid Build Coastguard Worker 
1859*8d67ca89SAndroid Build Coastguard Worker   // Test android_link_namespaces()
1860*8d67ca89SAndroid Build Coastguard Worker   ASSERT_FALSE(android_link_namespaces(nullptr, nullptr, "libc.so"));
1861*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("android_link_namespaces failed: error linking namespaces: namespace_from is null.",
1862*8d67ca89SAndroid Build Coastguard Worker                dlerror());
1863*8d67ca89SAndroid Build Coastguard Worker 
1864*8d67ca89SAndroid Build Coastguard Worker   ASSERT_FALSE(android_link_namespaces(ns, nullptr, nullptr));
1865*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("android_link_namespaces failed: "
1866*8d67ca89SAndroid Build Coastguard Worker                "error linking namespaces \"private\"->\"(default)\": "
1867*8d67ca89SAndroid Build Coastguard Worker                "the list of shared libraries is empty.", dlerror());
1868*8d67ca89SAndroid Build Coastguard Worker 
1869*8d67ca89SAndroid Build Coastguard Worker   ASSERT_FALSE(android_link_namespaces(ns, nullptr, ""));
1870*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("android_link_namespaces failed: "
1871*8d67ca89SAndroid Build Coastguard Worker                "error linking namespaces \"private\"->\"(default)\": "
1872*8d67ca89SAndroid Build Coastguard Worker                "the list of shared libraries is empty.", dlerror());
1873*8d67ca89SAndroid Build Coastguard Worker 
1874*8d67ca89SAndroid Build Coastguard Worker   // Test __loader_android_link_namespaces_all_libs()
1875*8d67ca89SAndroid Build Coastguard Worker   ASSERT_FALSE(__loader_android_link_namespaces_all_libs(nullptr, nullptr));
1876*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("android_link_namespaces_all_libs failed: "
1877*8d67ca89SAndroid Build Coastguard Worker                "error linking namespaces: namespace_from is null.", dlerror());
1878*8d67ca89SAndroid Build Coastguard Worker 
1879*8d67ca89SAndroid Build Coastguard Worker   ASSERT_FALSE(__loader_android_link_namespaces_all_libs(nullptr, ns));
1880*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("android_link_namespaces_all_libs failed: "
1881*8d67ca89SAndroid Build Coastguard Worker                "error linking namespaces: namespace_from is null.", dlerror());
1882*8d67ca89SAndroid Build Coastguard Worker 
1883*8d67ca89SAndroid Build Coastguard Worker   ASSERT_FALSE(__loader_android_link_namespaces_all_libs(ns, nullptr));
1884*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("android_link_namespaces_all_libs failed: "
1885*8d67ca89SAndroid Build Coastguard Worker                "error linking namespaces: namespace_to is null.", dlerror());
1886*8d67ca89SAndroid Build Coastguard Worker }
1887*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_allow_all_shared_libs)1888*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_allow_all_shared_libs) {
1889*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_init_anonymous_namespace(g_core_shared_libs.c_str(), nullptr));
1890*8d67ca89SAndroid Build Coastguard Worker 
1891*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_a =
1892*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("ns_a",
1893*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1894*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/ns_a").c_str(),
1895*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1896*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1897*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1898*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_a != nullptr) << dlerror();
1899*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_a, nullptr, g_core_shared_libs.c_str())) << dlerror();
1900*8d67ca89SAndroid Build Coastguard Worker 
1901*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns_b =
1902*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("ns_b",
1903*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1904*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/ns_b").c_str(),
1905*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_ISOLATED,
1906*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1907*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1908*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_b != nullptr) << dlerror();
1909*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_b, nullptr, g_core_shared_libs.c_str())) << dlerror();
1910*8d67ca89SAndroid Build Coastguard Worker 
1911*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns_b, ns_a, "libnstest_ns_a_public1.so")) << dlerror();
1912*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(__loader_android_link_namespaces_all_libs(ns_a, ns_b)) << dlerror();
1913*8d67ca89SAndroid Build Coastguard Worker 
1914*8d67ca89SAndroid Build Coastguard Worker   // Load libs with android_dlopen_ext() from namespace b
1915*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
1916*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
1917*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_b;
1918*8d67ca89SAndroid Build Coastguard Worker 
1919*8d67ca89SAndroid Build Coastguard Worker   void* ns_b_handle1 = android_dlopen_ext("libnstest_ns_a_public1.so", RTLD_NOW, &extinfo);
1920*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_b_handle1 != nullptr) << dlerror();
1921*8d67ca89SAndroid Build Coastguard Worker 
1922*8d67ca89SAndroid Build Coastguard Worker   void* ns_b_handle1_internal =
1923*8d67ca89SAndroid Build Coastguard Worker       android_dlopen_ext("libnstest_ns_a_public1_internal.so", RTLD_NOW, &extinfo);
1924*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_b_handle1_internal == nullptr);
1925*8d67ca89SAndroid Build Coastguard Worker 
1926*8d67ca89SAndroid Build Coastguard Worker   void* ns_b_handle2 = android_dlopen_ext("libnstest_ns_b_public2.so", RTLD_NOW, &extinfo);
1927*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_b_handle2 != nullptr) << dlerror();
1928*8d67ca89SAndroid Build Coastguard Worker 
1929*8d67ca89SAndroid Build Coastguard Worker   void* ns_b_handle3 = android_dlopen_ext("libnstest_ns_b_public3.so", RTLD_NOW, &extinfo);
1930*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_b_handle3 != nullptr) << dlerror();
1931*8d67ca89SAndroid Build Coastguard Worker 
1932*8d67ca89SAndroid Build Coastguard Worker   // Load libs with android_dlopen_ext() from namespace a
1933*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns_a;
1934*8d67ca89SAndroid Build Coastguard Worker 
1935*8d67ca89SAndroid Build Coastguard Worker   void* ns_a_handle1 = android_dlopen_ext("libnstest_ns_a_public1.so", RTLD_NOW, &extinfo);
1936*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_a_handle1 != nullptr) << dlerror();
1937*8d67ca89SAndroid Build Coastguard Worker 
1938*8d67ca89SAndroid Build Coastguard Worker   void* ns_a_handle1_internal =
1939*8d67ca89SAndroid Build Coastguard Worker       android_dlopen_ext("libnstest_ns_a_public1_internal.so", RTLD_NOW, &extinfo);
1940*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_a_handle1_internal != nullptr) << dlerror();
1941*8d67ca89SAndroid Build Coastguard Worker 
1942*8d67ca89SAndroid Build Coastguard Worker   void* ns_a_handle2 = android_dlopen_ext("libnstest_ns_b_public2.so", RTLD_NOW, &extinfo);
1943*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_a_handle2 != nullptr) << dlerror();
1944*8d67ca89SAndroid Build Coastguard Worker 
1945*8d67ca89SAndroid Build Coastguard Worker   void* ns_a_handle3 = android_dlopen_ext("libnstest_ns_b_public3.so", RTLD_NOW, &extinfo);
1946*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_a_handle3 != nullptr) << dlerror();
1947*8d67ca89SAndroid Build Coastguard Worker 
1948*8d67ca89SAndroid Build Coastguard Worker   // Compare the dlopen handle
1949*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(ns_b_handle1, ns_a_handle1);
1950*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(ns_b_handle2, ns_a_handle2);
1951*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(ns_b_handle3, ns_a_handle3);
1952*8d67ca89SAndroid Build Coastguard Worker 
1953*8d67ca89SAndroid Build Coastguard Worker   // Close libs
1954*8d67ca89SAndroid Build Coastguard Worker   dlclose(ns_b_handle1);
1955*8d67ca89SAndroid Build Coastguard Worker   dlclose(ns_b_handle2);
1956*8d67ca89SAndroid Build Coastguard Worker   dlclose(ns_b_handle3);
1957*8d67ca89SAndroid Build Coastguard Worker 
1958*8d67ca89SAndroid Build Coastguard Worker   dlclose(ns_a_handle1);
1959*8d67ca89SAndroid Build Coastguard Worker   dlclose(ns_a_handle1_internal);
1960*8d67ca89SAndroid Build Coastguard Worker   dlclose(ns_a_handle2);
1961*8d67ca89SAndroid Build Coastguard Worker   dlclose(ns_a_handle3);
1962*8d67ca89SAndroid Build Coastguard Worker }
1963*8d67ca89SAndroid Build Coastguard Worker 
MapPflagsToProtFlags(uint32_t flags)1964*8d67ca89SAndroid Build Coastguard Worker static inline int MapPflagsToProtFlags(uint32_t flags) {
1965*8d67ca89SAndroid Build Coastguard Worker   int prot_flags = 0;
1966*8d67ca89SAndroid Build Coastguard Worker   if (PF_X & flags) prot_flags |= PROT_EXEC;
1967*8d67ca89SAndroid Build Coastguard Worker   if (PF_W & flags) prot_flags |= PROT_WRITE;
1968*8d67ca89SAndroid Build Coastguard Worker   if (PF_R & flags) prot_flags |= PROT_READ;
1969*8d67ca89SAndroid Build Coastguard Worker   return prot_flags;
1970*8d67ca89SAndroid Build Coastguard Worker }
1971*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_anonymous)1972*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_anonymous) {
1973*8d67ca89SAndroid Build Coastguard Worker   static const char* root_lib = "libnstest_root.so";
1974*8d67ca89SAndroid Build Coastguard Worker   std::string shared_libs = g_core_shared_libs + ":" + g_public_lib;
1975*8d67ca89SAndroid Build Coastguard Worker 
1976*8d67ca89SAndroid Build Coastguard Worker   const std::string lib_public_path = GetTestLibRoot() + "/public_namespace_libs/" + g_public_lib;
1977*8d67ca89SAndroid Build Coastguard Worker   void* handle_public = dlopen(lib_public_path.c_str(), RTLD_NOW);
1978*8d67ca89SAndroid Build Coastguard Worker 
1979*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle_public != nullptr) << dlerror();
1980*8d67ca89SAndroid Build Coastguard Worker 
1981*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(
1982*8d67ca89SAndroid Build Coastguard Worker           android_init_anonymous_namespace(shared_libs.c_str(),
1983*8d67ca89SAndroid Build Coastguard Worker                                            (GetTestLibRoot() + "/private_namespace_libs").c_str())
1984*8d67ca89SAndroid Build Coastguard Worker       ) << dlerror();
1985*8d67ca89SAndroid Build Coastguard Worker 
1986*8d67ca89SAndroid Build Coastguard Worker   android_namespace_t* ns =
1987*8d67ca89SAndroid Build Coastguard Worker           android_create_namespace("private",
1988*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1989*8d67ca89SAndroid Build Coastguard Worker                                    (GetTestLibRoot() + "/private_namespace_libs").c_str(),
1990*8d67ca89SAndroid Build Coastguard Worker                                    ANDROID_NAMESPACE_TYPE_REGULAR,
1991*8d67ca89SAndroid Build Coastguard Worker                                    nullptr,
1992*8d67ca89SAndroid Build Coastguard Worker                                    nullptr);
1993*8d67ca89SAndroid Build Coastguard Worker 
1994*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns != nullptr) << dlerror();
1995*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(android_link_namespaces(ns, nullptr, shared_libs.c_str())) << dlerror();
1996*8d67ca89SAndroid Build Coastguard Worker 
1997*8d67ca89SAndroid Build Coastguard Worker   std::string private_library_absolute_path = GetTestLibRoot() + "/private_namespace_libs/" + root_lib;
1998*8d67ca89SAndroid Build Coastguard Worker 
1999*8d67ca89SAndroid Build Coastguard Worker   android_dlextinfo extinfo;
2000*8d67ca89SAndroid Build Coastguard Worker   extinfo.flags = ANDROID_DLEXT_USE_NAMESPACE;
2001*8d67ca89SAndroid Build Coastguard Worker   extinfo.library_namespace = ns;
2002*8d67ca89SAndroid Build Coastguard Worker 
2003*8d67ca89SAndroid Build Coastguard Worker   // we are going to copy this library to anonymous mmap and call the copy of ns_get_dlopened_string
2004*8d67ca89SAndroid Build Coastguard Worker   void* handle = android_dlopen_ext(private_library_absolute_path.c_str(), RTLD_NOW, &extinfo);
2005*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(handle != nullptr) << dlerror();
2006*8d67ca89SAndroid Build Coastguard Worker 
2007*8d67ca89SAndroid Build Coastguard Worker   uintptr_t ns_get_dlopened_string_addr =
2008*8d67ca89SAndroid Build Coastguard Worker       reinterpret_cast<uintptr_t>(dlsym(handle, "ns_get_dlopened_string"));
2009*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_dlopened_string_addr != 0) << dlerror();
2010*8d67ca89SAndroid Build Coastguard Worker   typedef const char* (*fn_t)();
2011*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_dlopened_string_private = reinterpret_cast<fn_t>(ns_get_dlopened_string_addr);
2012*8d67ca89SAndroid Build Coastguard Worker 
2013*8d67ca89SAndroid Build Coastguard Worker   Dl_info private_library_info;
2014*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NE(dladdr(reinterpret_cast<void*>(ns_get_dlopened_string_addr), &private_library_info), 0)
2015*8d67ca89SAndroid Build Coastguard Worker       << dlerror();
2016*8d67ca89SAndroid Build Coastguard Worker   std::vector<map_record> maps_to_copy;
2017*8d67ca89SAndroid Build Coastguard Worker   bool has_executable_segment = false;
2018*8d67ca89SAndroid Build Coastguard Worker   uintptr_t addr_start = 0;
2019*8d67ca89SAndroid Build Coastguard Worker   uintptr_t addr_end = 0;
2020*8d67ca89SAndroid Build Coastguard Worker   std::tuple dl_iterate_arg = {&private_library_info, &maps_to_copy, &has_executable_segment,
2021*8d67ca89SAndroid Build Coastguard Worker                                &addr_start, &addr_end};
2022*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(
2023*8d67ca89SAndroid Build Coastguard Worker       1, dl_iterate_phdr(
2024*8d67ca89SAndroid Build Coastguard Worker              [](dl_phdr_info* info, size_t /*size*/, void* data) -> int {
2025*8d67ca89SAndroid Build Coastguard Worker                auto [private_library_info, maps_to_copy, has_executable_segment, addr_start,
2026*8d67ca89SAndroid Build Coastguard Worker                      addr_end] = *reinterpret_cast<decltype(dl_iterate_arg)*>(data);
2027*8d67ca89SAndroid Build Coastguard Worker                if (info->dlpi_addr != reinterpret_cast<ElfW(Addr)>(private_library_info->dli_fbase))
2028*8d67ca89SAndroid Build Coastguard Worker                  return 0;
2029*8d67ca89SAndroid Build Coastguard Worker 
2030*8d67ca89SAndroid Build Coastguard Worker                for (size_t i = 0; i < info->dlpi_phnum; ++i) {
2031*8d67ca89SAndroid Build Coastguard Worker                  const ElfW(Phdr)* phdr = info->dlpi_phdr + i;
2032*8d67ca89SAndroid Build Coastguard Worker                  if (phdr->p_type != PT_LOAD) continue;
2033*8d67ca89SAndroid Build Coastguard Worker                  *has_executable_segment |= phdr->p_flags & PF_X;
2034*8d67ca89SAndroid Build Coastguard Worker                  uintptr_t mapping_start = page_start(info->dlpi_addr + phdr->p_vaddr);
2035*8d67ca89SAndroid Build Coastguard Worker                  uintptr_t mapping_end = page_end(info->dlpi_addr + phdr->p_vaddr + phdr->p_memsz);
2036*8d67ca89SAndroid Build Coastguard Worker                  if (*addr_start == 0 || mapping_start < *addr_start) *addr_start = mapping_start;
2037*8d67ca89SAndroid Build Coastguard Worker                  if (*addr_end == 0 || mapping_end > *addr_end) *addr_end = mapping_end;
2038*8d67ca89SAndroid Build Coastguard Worker                  maps_to_copy->push_back({
2039*8d67ca89SAndroid Build Coastguard Worker                      .addr_start = mapping_start,
2040*8d67ca89SAndroid Build Coastguard Worker                      .addr_end = mapping_end,
2041*8d67ca89SAndroid Build Coastguard Worker                      .perms = MapPflagsToProtFlags(phdr->p_flags),
2042*8d67ca89SAndroid Build Coastguard Worker                  });
2043*8d67ca89SAndroid Build Coastguard Worker                }
2044*8d67ca89SAndroid Build Coastguard Worker                return 1;
2045*8d67ca89SAndroid Build Coastguard Worker              },
2046*8d67ca89SAndroid Build Coastguard Worker              &dl_iterate_arg));
2047*8d67ca89SAndroid Build Coastguard Worker 
2048*8d67ca89SAndroid Build Coastguard Worker   // Some validity checks.
2049*8d67ca89SAndroid Build Coastguard Worker   ASSERT_NE(maps_to_copy.size(), 0u);
2050*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(addr_start > 0);
2051*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(addr_end > 0);
2052*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_dlopened_string_addr > addr_start);
2053*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_dlopened_string_addr < addr_end);
2054*8d67ca89SAndroid Build Coastguard Worker 
2055*8d67ca89SAndroid Build Coastguard Worker   if (!has_executable_segment) {
2056*8d67ca89SAndroid Build Coastguard Worker     // For some natively bridged environments this code might be missing
2057*8d67ca89SAndroid Build Coastguard Worker     // the executable flag. This is because the guest code is not supposed
2058*8d67ca89SAndroid Build Coastguard Worker     // to be executed directly and making it non-executable is more secure.
2059*8d67ca89SAndroid Build Coastguard Worker     // In this case we assume the segment with the function is executable.
2060*8d67ca89SAndroid Build Coastguard Worker     for (auto& rec : maps_to_copy) {
2061*8d67ca89SAndroid Build Coastguard Worker       if (ns_get_dlopened_string_addr >= rec.addr_start &&
2062*8d67ca89SAndroid Build Coastguard Worker           ns_get_dlopened_string_addr < rec.addr_end) {
2063*8d67ca89SAndroid Build Coastguard Worker         ASSERT_TRUE((rec.perms & PROT_WRITE) == 0);
2064*8d67ca89SAndroid Build Coastguard Worker         rec.perms |= PROT_EXEC;
2065*8d67ca89SAndroid Build Coastguard Worker         break;
2066*8d67ca89SAndroid Build Coastguard Worker       }
2067*8d67ca89SAndroid Build Coastguard Worker     }
2068*8d67ca89SAndroid Build Coastguard Worker   }
2069*8d67ca89SAndroid Build Coastguard Worker 
2070*8d67ca89SAndroid Build Coastguard Worker   // copy
2071*8d67ca89SAndroid Build Coastguard Worker   uintptr_t reserved_addr = reinterpret_cast<uintptr_t>(mmap(nullptr, addr_end - addr_start,
2072*8d67ca89SAndroid Build Coastguard Worker                                                              PROT_NONE, MAP_ANON | MAP_PRIVATE,
2073*8d67ca89SAndroid Build Coastguard Worker                                                              -1, 0));
2074*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(reinterpret_cast<void*>(reserved_addr) != MAP_FAILED);
2075*8d67ca89SAndroid Build Coastguard Worker 
2076*8d67ca89SAndroid Build Coastguard Worker   struct stat file_stat;
2077*8d67ca89SAndroid Build Coastguard Worker   int ret = TEMP_FAILURE_RETRY(stat(private_library_absolute_path.c_str(), &file_stat));
2078*8d67ca89SAndroid Build Coastguard Worker   ASSERT_EQ(ret, 0) << "Failed to stat library";
2079*8d67ca89SAndroid Build Coastguard Worker   size_t file_size = file_stat.st_size;
2080*8d67ca89SAndroid Build Coastguard Worker 
2081*8d67ca89SAndroid Build Coastguard Worker   {
2082*8d67ca89SAndroid Build Coastguard Worker     // Disable MTE while copying the PROT_MTE-protected global variables from
2083*8d67ca89SAndroid Build Coastguard Worker     // the existing mappings. We don't really care about turning on PROT_MTE for
2084*8d67ca89SAndroid Build Coastguard Worker     // the new copy of the mappings, as this isn't the behaviour under test and
2085*8d67ca89SAndroid Build Coastguard Worker     // tags will be ignored. This only applies for MTE-enabled devices.
2086*8d67ca89SAndroid Build Coastguard Worker     ScopedDisableMTE disable_mte_for_copying_global_variables;
2087*8d67ca89SAndroid Build Coastguard Worker     for (const auto& rec : maps_to_copy) {
2088*8d67ca89SAndroid Build Coastguard Worker       uintptr_t offset = rec.addr_start - addr_start;
2089*8d67ca89SAndroid Build Coastguard Worker       size_t size = rec.addr_end - rec.addr_start;
2090*8d67ca89SAndroid Build Coastguard Worker       void* addr = reinterpret_cast<void*>(reserved_addr + offset);
2091*8d67ca89SAndroid Build Coastguard Worker       void* map =
2092*8d67ca89SAndroid Build Coastguard Worker           mmap(addr, size, PROT_READ | PROT_WRITE, MAP_ANON | MAP_PRIVATE | MAP_FIXED, -1, 0);
2093*8d67ca89SAndroid Build Coastguard Worker       ASSERT_TRUE(map != MAP_FAILED);
2094*8d67ca89SAndroid Build Coastguard Worker       // Attempting the below memcpy from a portion of the map that is off the end of
2095*8d67ca89SAndroid Build Coastguard Worker       // the backing file will cause the kernel to throw a SIGBUS
2096*8d67ca89SAndroid Build Coastguard Worker       size_t _size =
2097*8d67ca89SAndroid Build Coastguard Worker           ::android::procinfo::MappedFileSize(rec.addr_start, rec.addr_end, rec.offset, file_size);
2098*8d67ca89SAndroid Build Coastguard Worker       memcpy(map, reinterpret_cast<void*>(rec.addr_start), _size);
2099*8d67ca89SAndroid Build Coastguard Worker       mprotect(map, size, rec.perms);
2100*8d67ca89SAndroid Build Coastguard Worker     }
2101*8d67ca89SAndroid Build Coastguard Worker   }
2102*8d67ca89SAndroid Build Coastguard Worker 
2103*8d67ca89SAndroid Build Coastguard Worker   // call the function copy
2104*8d67ca89SAndroid Build Coastguard Worker   uintptr_t ns_get_dlopened_string_offset  = ns_get_dlopened_string_addr - addr_start;
2105*8d67ca89SAndroid Build Coastguard Worker   fn_t ns_get_dlopened_string_anon = reinterpret_cast<fn_t>(reserved_addr + ns_get_dlopened_string_offset);
2106*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from private namespace (dlopened library)",
2107*8d67ca89SAndroid Build Coastguard Worker                ns_get_dlopened_string_anon());
2108*8d67ca89SAndroid Build Coastguard Worker 
2109*8d67ca89SAndroid Build Coastguard Worker   // They should belong to different namespaces (private and anonymous)
2110*8d67ca89SAndroid Build Coastguard Worker   ASSERT_STREQ("This string is from private namespace (dlopened library)",
2111*8d67ca89SAndroid Build Coastguard Worker                ns_get_dlopened_string_private());
2112*8d67ca89SAndroid Build Coastguard Worker 
2113*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(ns_get_dlopened_string_anon() != ns_get_dlopened_string_private());
2114*8d67ca89SAndroid Build Coastguard Worker }
2115*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,ns_hidden_child)2116*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, ns_hidden_child) {
2117*8d67ca89SAndroid Build Coastguard Worker   ExecTestHelper eth;
2118*8d67ca89SAndroid Build Coastguard Worker 
2119*8d67ca89SAndroid Build Coastguard Worker   std::string helper = GetTestLibRoot() + "/ns_hidden_child_helper";
2120*8d67ca89SAndroid Build Coastguard Worker   std::string app_ns_dir = GetTestLibRoot() + "/ns_hidden_child_app";
2121*8d67ca89SAndroid Build Coastguard Worker   eth.SetArgs({ helper.c_str(), app_ns_dir.c_str(), nullptr });
2122*8d67ca89SAndroid Build Coastguard Worker 
2123*8d67ca89SAndroid Build Coastguard Worker   // Add the main libns_hidden_child_*.so libraries to the search path of the default namespace.
2124*8d67ca89SAndroid Build Coastguard Worker   std::string env = "LD_LIBRARY_PATH=" + GetTestLibRoot();
2125*8d67ca89SAndroid Build Coastguard Worker   eth.SetEnv({ env.c_str(), nullptr });
2126*8d67ca89SAndroid Build Coastguard Worker 
2127*8d67ca89SAndroid Build Coastguard Worker   eth.Run([&]() { execve(helper.c_str(), eth.GetArgs(), eth.GetEnv()); }, 0,
2128*8d67ca89SAndroid Build Coastguard Worker           "public_function is non-null\n"
2129*8d67ca89SAndroid Build Coastguard Worker           "internal_function is null\n");
2130*8d67ca89SAndroid Build Coastguard Worker }
2131*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,dlopen_handle_value_platform)2132*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, dlopen_handle_value_platform) {
2133*8d67ca89SAndroid Build Coastguard Worker   void* handle = dlopen("libtest_dlsym_from_this.so", RTLD_NOW | RTLD_LOCAL);
2134*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE((reinterpret_cast<uintptr_t>(handle) & 1) != 0)
2135*8d67ca89SAndroid Build Coastguard Worker           << "dlopen should return odd value for the handle";
2136*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
2137*8d67ca89SAndroid Build Coastguard Worker }
2138*8d67ca89SAndroid Build Coastguard Worker 
TEST(dlext,dlopen_handle_value_app_compat)2139*8d67ca89SAndroid Build Coastguard Worker TEST(dlext, dlopen_handle_value_app_compat) {
2140*8d67ca89SAndroid Build Coastguard Worker   android_set_application_target_sdk_version(23);
2141*8d67ca89SAndroid Build Coastguard Worker   void* handle = dlopen("libtest_dlsym_from_this.so", RTLD_NOW | RTLD_LOCAL);
2142*8d67ca89SAndroid Build Coastguard Worker   ASSERT_TRUE(reinterpret_cast<uintptr_t>(handle) % sizeof(uintptr_t) == 0)
2143*8d67ca89SAndroid Build Coastguard Worker           << "dlopen should return valid pointer";
2144*8d67ca89SAndroid Build Coastguard Worker   dlclose(handle);
2145*8d67ca89SAndroid Build Coastguard Worker }
2146