71 lines
2.2 KiB
C++
71 lines
2.2 KiB
C++
// 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
|