You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

79 lines
2.5 KiB

/*
* Copyright (C) 2016 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "dlfcn_symlink_support.h"
#include <gtest/gtest.h>
#include <dlfcn.h>
#include <libgen.h>
#include <link.h>
#include <unistd.h>
#include <android-base/strings.h>
#include <algorithm>
#include <string>
#include <vector>
static const constexpr char* source_file_name = "libdlext_test.so";
static const constexpr char* symlink_name_prefix = "libdlext_test_";
static int dl_callback(struct dl_phdr_info *info, size_t /* size */, void *data) {
// The case when path is not absolute and is equal to source_file_name
// is disregarded intentionally since in bionic dlpi_name should always
// be realpath to a shared object.
const std::string suffix = std::string("/") + source_file_name;
// TODO (dimitry): remove this check once fake libdl.so is gone
if (info->dlpi_name == nullptr) {
// This is linker imposing as libdl.so - skip it
return 0;
}
if (android::base::EndsWith(info->dlpi_name, suffix)) {
std::string* path = reinterpret_cast<std::string*>(data);
*path = info->dlpi_name;
return 1; // found
}
return 0;
}
void create_dlfcn_test_symlink(const char* suffix, std::string* result) {
void* handle = dlopen(source_file_name, RTLD_NOW);
std::string source_file_path;
ASSERT_TRUE(handle != nullptr) << dlerror();
ASSERT_TRUE(dl_iterate_phdr(dl_callback, &source_file_path) == 1)
<< "dl_phdr_info for \"" << source_file_name << "\" was not found.";
dlclose(handle);
std::vector<char> buf;
std::copy(source_file_path.begin(), source_file_path.end(), std::back_inserter(buf));
buf.push_back('\0');
std::string path_dir = dirname(&buf[0]);
std::string link_path = path_dir + "/" + symlink_name_prefix + suffix + ".so";
ASSERT_TRUE(symlink(source_file_path.c_str(), link_path.c_str()) == 0) << strerror(errno);
*result = link_path;
}
void remove_dlfcn_test_symlink(const std::string& path) {
ASSERT_TRUE(unlink(path.c_str()) == 0) << strerror(errno);
}