mirror of
https://github.com/tomahawk-player/tomahawk.git
synced 2025-08-06 14:16:32 +02:00
285 lines
10 KiB
C++
285 lines
10 KiB
C++
// Copyright (c) 2010, Google Inc.
|
|
// All rights reserved.
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without
|
|
// modification, are permitted provided that the following conditions are
|
|
// met:
|
|
//
|
|
// * Redistributions of source code must retain the above copyright
|
|
// notice, this list of conditions and the following disclaimer.
|
|
// * Redistributions in binary form must reproduce the above
|
|
// copyright notice, this list of conditions and the following disclaimer
|
|
// in the documentation and/or other materials provided with the
|
|
// distribution.
|
|
// * Neither the name of Google Inc. nor the names of its
|
|
// contributors may be used to endorse or promote products derived from
|
|
// this software without specific prior written permission.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
// Unit tests for FileID
|
|
|
|
#include <elf.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "common/linux/file_id.h"
|
|
#include "common/linux/safe_readlink.h"
|
|
#include "common/linux/synth_elf.h"
|
|
#include "common/test_assembler.h"
|
|
#include "common/tests/auto_tempdir.h"
|
|
#include "breakpad_googletest_includes.h"
|
|
|
|
using namespace google_breakpad;
|
|
using google_breakpad::SafeReadLink;
|
|
using google_breakpad::synth_elf::BuildIDNote;
|
|
using google_breakpad::synth_elf::ELF;
|
|
using google_breakpad::test_assembler::kLittleEndian;
|
|
using google_breakpad::test_assembler::Section;
|
|
|
|
namespace {
|
|
|
|
// Simply calling Section::Append(size, byte) produces a uninteresting pattern
|
|
// that tends to get hashed to 0000...0000. This populates the section with
|
|
// data to produce better hashes.
|
|
void PopulateSection(Section* section, int size, int prime_number) {
|
|
for (int i = 0; i < size; i++)
|
|
section->Append(1, (i % prime_number) % 256);
|
|
}
|
|
|
|
} // namespace
|
|
|
|
TEST(FileIDStripTest, StripSelf) {
|
|
// Calculate the File ID of this binary using
|
|
// FileID::ElfFileIdentifier, then make a copy of this binary,
|
|
// strip it, and ensure that the result is the same.
|
|
char exe_name[PATH_MAX];
|
|
ASSERT_TRUE(SafeReadLink("/proc/self/exe", exe_name));
|
|
|
|
// copy our binary to a temp file, and strip it
|
|
AutoTempDir temp_dir;
|
|
std::string templ = temp_dir.path() + "/file-id-unittest";
|
|
char cmdline[4096];
|
|
sprintf(cmdline, "cp \"%s\" \"%s\"", exe_name, templ.c_str());
|
|
ASSERT_EQ(system(cmdline), 0);
|
|
sprintf(cmdline, "strip \"%s\"", templ.c_str());
|
|
ASSERT_EQ(system(cmdline), 0);
|
|
|
|
uint8_t identifier1[sizeof(MDGUID)];
|
|
uint8_t identifier2[sizeof(MDGUID)];
|
|
FileID fileid1(exe_name);
|
|
EXPECT_TRUE(fileid1.ElfFileIdentifier(identifier1));
|
|
FileID fileid2(templ.c_str());
|
|
EXPECT_TRUE(fileid2.ElfFileIdentifier(identifier2));
|
|
char identifier_string1[37];
|
|
char identifier_string2[37];
|
|
FileID::ConvertIdentifierToString(identifier1, identifier_string1,
|
|
37);
|
|
FileID::ConvertIdentifierToString(identifier2, identifier_string2,
|
|
37);
|
|
EXPECT_STREQ(identifier_string1, identifier_string2);
|
|
}
|
|
|
|
class FileIDTest : public testing::Test {
|
|
public:
|
|
void GetElfContents(ELF& elf) {
|
|
string contents;
|
|
ASSERT_TRUE(elf.GetContents(&contents));
|
|
ASSERT_LT(0, contents.size());
|
|
|
|
elfdata_v.clear();
|
|
elfdata_v.insert(elfdata_v.begin(), contents.begin(), contents.end());
|
|
elfdata = &elfdata_v[0];
|
|
}
|
|
|
|
vector<uint8_t> elfdata_v;
|
|
uint8_t* elfdata;
|
|
};
|
|
|
|
TEST_F(FileIDTest, ElfClass) {
|
|
uint8_t identifier[sizeof(MDGUID)];
|
|
const char expected_identifier_string[] =
|
|
"80808080-8080-0000-0000-008080808080";
|
|
char identifier_string[sizeof(expected_identifier_string)];
|
|
const size_t kTextSectionSize = 128;
|
|
|
|
ELF elf32(EM_386, ELFCLASS32, kLittleEndian);
|
|
Section text32(kLittleEndian);
|
|
for (size_t i = 0; i < kTextSectionSize; ++i) {
|
|
text32.D8(i * 3);
|
|
}
|
|
elf32.AddSection(".text", text32, SHT_PROGBITS);
|
|
elf32.Finish();
|
|
GetElfContents(elf32);
|
|
|
|
EXPECT_TRUE(FileID::ElfFileIdentifierFromMappedFile(elfdata, identifier));
|
|
|
|
FileID::ConvertIdentifierToString(identifier, identifier_string,
|
|
sizeof(identifier_string));
|
|
EXPECT_STREQ(expected_identifier_string, identifier_string);
|
|
|
|
memset(identifier, 0, sizeof(identifier));
|
|
memset(identifier_string, 0, sizeof(identifier_string));
|
|
|
|
ELF elf64(EM_X86_64, ELFCLASS64, kLittleEndian);
|
|
Section text64(kLittleEndian);
|
|
for (size_t i = 0; i < kTextSectionSize; ++i) {
|
|
text64.D8(i * 3);
|
|
}
|
|
elf64.AddSection(".text", text64, SHT_PROGBITS);
|
|
elf64.Finish();
|
|
GetElfContents(elf64);
|
|
|
|
EXPECT_TRUE(FileID::ElfFileIdentifierFromMappedFile(elfdata, identifier));
|
|
|
|
FileID::ConvertIdentifierToString(identifier, identifier_string,
|
|
sizeof(identifier_string));
|
|
EXPECT_STREQ(expected_identifier_string, identifier_string);
|
|
}
|
|
|
|
TEST_F(FileIDTest, BuildID) {
|
|
const uint8_t kExpectedIdentifier[sizeof(MDGUID)] =
|
|
{0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
|
|
0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F};
|
|
char expected_identifier_string[] =
|
|
"00000000-0000-0000-0000-000000000000";
|
|
FileID::ConvertIdentifierToString(kExpectedIdentifier,
|
|
expected_identifier_string,
|
|
sizeof(expected_identifier_string));
|
|
|
|
uint8_t identifier[sizeof(MDGUID)];
|
|
char identifier_string[sizeof(expected_identifier_string)];
|
|
|
|
ELF elf32(EM_386, ELFCLASS32, kLittleEndian);
|
|
Section text(kLittleEndian);
|
|
text.Append(4096, 0);
|
|
elf32.AddSection(".text", text, SHT_PROGBITS);
|
|
BuildIDNote::AppendSection(elf32,
|
|
kExpectedIdentifier,
|
|
sizeof(kExpectedIdentifier));
|
|
elf32.Finish();
|
|
GetElfContents(elf32);
|
|
|
|
EXPECT_TRUE(FileID::ElfFileIdentifierFromMappedFile(elfdata, identifier));
|
|
|
|
FileID::ConvertIdentifierToString(identifier, identifier_string,
|
|
sizeof(identifier_string));
|
|
EXPECT_STREQ(expected_identifier_string, identifier_string);
|
|
|
|
memset(identifier, 0, sizeof(identifier));
|
|
memset(identifier_string, 0, sizeof(identifier_string));
|
|
|
|
ELF elf64(EM_X86_64, ELFCLASS64, kLittleEndian);
|
|
// Re-use empty text section from previous test
|
|
elf64.AddSection(".text", text, SHT_PROGBITS);
|
|
BuildIDNote::AppendSection(elf64,
|
|
kExpectedIdentifier,
|
|
sizeof(kExpectedIdentifier));
|
|
elf64.Finish();
|
|
GetElfContents(elf64);
|
|
|
|
EXPECT_TRUE(FileID::ElfFileIdentifierFromMappedFile(elfdata, identifier));
|
|
|
|
FileID::ConvertIdentifierToString(identifier, identifier_string,
|
|
sizeof(identifier_string));
|
|
EXPECT_STREQ(expected_identifier_string, identifier_string);
|
|
}
|
|
|
|
// Test to make sure two files with different text sections produce
|
|
// different hashes when not using a build id.
|
|
TEST_F(FileIDTest, UniqueHashes32) {
|
|
char identifier_string_1[] =
|
|
"00000000-0000-0000-0000-000000000000";
|
|
char identifier_string_2[] =
|
|
"00000000-0000-0000-0000-000000000000";
|
|
uint8_t identifier_1[sizeof(MDGUID)];
|
|
uint8_t identifier_2[sizeof(MDGUID)];
|
|
|
|
{
|
|
ELF elf1(EM_386, ELFCLASS32, kLittleEndian);
|
|
Section foo_1(kLittleEndian);
|
|
PopulateSection(&foo_1, 32, 5);
|
|
elf1.AddSection(".foo", foo_1, SHT_PROGBITS);
|
|
Section text_1(kLittleEndian);
|
|
PopulateSection(&text_1, 4096, 17);
|
|
elf1.AddSection(".text", text_1, SHT_PROGBITS);
|
|
elf1.Finish();
|
|
GetElfContents(elf1);
|
|
}
|
|
|
|
EXPECT_TRUE(FileID::ElfFileIdentifierFromMappedFile(elfdata, identifier_1));
|
|
FileID::ConvertIdentifierToString(identifier_1, identifier_string_1,
|
|
sizeof(identifier_string_1));
|
|
|
|
{
|
|
ELF elf2(EM_386, ELFCLASS32, kLittleEndian);
|
|
Section text_2(kLittleEndian);
|
|
Section foo_2(kLittleEndian);
|
|
PopulateSection(&foo_2, 32, 5);
|
|
elf2.AddSection(".foo", foo_2, SHT_PROGBITS);
|
|
PopulateSection(&text_2, 4096, 31);
|
|
elf2.AddSection(".text", text_2, SHT_PROGBITS);
|
|
elf2.Finish();
|
|
GetElfContents(elf2);
|
|
}
|
|
|
|
EXPECT_TRUE(FileID::ElfFileIdentifierFromMappedFile(elfdata, identifier_2));
|
|
FileID::ConvertIdentifierToString(identifier_2, identifier_string_2,
|
|
sizeof(identifier_string_2));
|
|
|
|
EXPECT_STRNE(identifier_string_1, identifier_string_2);
|
|
}
|
|
|
|
// Same as UniqueHashes32, for x86-64.
|
|
TEST_F(FileIDTest, UniqueHashes64) {
|
|
char identifier_string_1[] =
|
|
"00000000-0000-0000-0000-000000000000";
|
|
char identifier_string_2[] =
|
|
"00000000-0000-0000-0000-000000000000";
|
|
uint8_t identifier_1[sizeof(MDGUID)];
|
|
uint8_t identifier_2[sizeof(MDGUID)];
|
|
|
|
{
|
|
ELF elf1(EM_X86_64, ELFCLASS64, kLittleEndian);
|
|
Section foo_1(kLittleEndian);
|
|
PopulateSection(&foo_1, 32, 5);
|
|
elf1.AddSection(".foo", foo_1, SHT_PROGBITS);
|
|
Section text_1(kLittleEndian);
|
|
PopulateSection(&text_1, 4096, 17);
|
|
elf1.AddSection(".text", text_1, SHT_PROGBITS);
|
|
elf1.Finish();
|
|
GetElfContents(elf1);
|
|
}
|
|
|
|
EXPECT_TRUE(FileID::ElfFileIdentifierFromMappedFile(elfdata, identifier_1));
|
|
FileID::ConvertIdentifierToString(identifier_1, identifier_string_1,
|
|
sizeof(identifier_string_1));
|
|
|
|
{
|
|
ELF elf2(EM_X86_64, ELFCLASS64, kLittleEndian);
|
|
Section text_2(kLittleEndian);
|
|
Section foo_2(kLittleEndian);
|
|
PopulateSection(&foo_2, 32, 5);
|
|
elf2.AddSection(".foo", foo_2, SHT_PROGBITS);
|
|
PopulateSection(&text_2, 4096, 31);
|
|
elf2.AddSection(".text", text_2, SHT_PROGBITS);
|
|
elf2.Finish();
|
|
GetElfContents(elf2);
|
|
}
|
|
|
|
EXPECT_TRUE(FileID::ElfFileIdentifierFromMappedFile(elfdata, identifier_2));
|
|
FileID::ConvertIdentifierToString(identifier_2, identifier_string_2,
|
|
sizeof(identifier_string_2));
|
|
|
|
EXPECT_STRNE(identifier_string_1, identifier_string_2);
|
|
}
|