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.

71 lines
2.2 KiB

// Copyright 2018 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/debug/elf_reader_linux.h"
#include <dlfcn.h>
#include "base/files/memory_mapped_file.h"
#include "base/strings/string_util.h"
#include "build/build_config.h"
#include "testing/gtest/include/gtest/gtest.h"
extern char __executable_start;
namespace base {
namespace debug {
// The linker flag --build-id is passed only on official builds. Clang does not
// enable it by default and we do not have build id section in non-official
// builds.
#if defined(OFFICIAL_BUILD)
TEST(ElfReaderTest, ReadElfBuildId) {
Optional<std::string> build_id = ReadElfBuildId(&__executable_start);
ASSERT_TRUE(build_id);
const size_t kGuidBytes = 20;
EXPECT_EQ(2 * kGuidBytes, build_id.value().size());
for (char c : *build_id) {
EXPECT_TRUE(IsHexDigit(c));
EXPECT_FALSE(IsAsciiLower(c));
}
}
#endif
TEST(ElfReaderTest, ReadElfLibraryName) {
#if defined(OS_ANDROID)
// On Android the library loader memory maps the full so file.
const char kLibraryName[] = "lib_base_unittests__library";
const void* addr = &__executable_start;
#else
// On Linux the executable does not contain soname and is not mapped till
// dynamic segment. So, use malloc wrapper so file on which the test already
// depends on.
const char kLibraryName[] = MALLOC_WRAPPER_LIB;
// Find any symbol in the loaded file.
void* handle = dlopen(kLibraryName, RTLD_NOW | RTLD_LOCAL);
const void* init_addr = dlsym(handle, "_init");
// Use this symbol to get full path to the loaded library.
Dl_info info;
int res = dladdr(init_addr, &info);
ASSERT_NE(0, res);
std::string filename(info.dli_fname);
EXPECT_FALSE(filename.empty());
EXPECT_NE(std::string::npos, filename.find(kLibraryName));
// Memory map the so file and use it to test reading so name.
MemoryMappedFile file;
file.Initialize(FilePath(filename));
const void* addr = file.data();
#endif
auto name = ReadElfLibraryName(addr);
ASSERT_TRUE(name);
EXPECT_NE(std::string::npos, name->find(kLibraryName))
<< "Library name " << *name << " doesn't contain expected "
<< kLibraryName;
}
} // namespace debug
} // namespace base