109 lines
4.5 KiB
C++
109 lines
4.5 KiB
C++
/*
|
|
* Copyright (C) 2021 The Android Open Source Project
|
|
* 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.
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <link.h>
|
|
|
|
#include "ldso_trampoline_phdr.h"
|
|
|
|
struct PhdrTrimSegmentTestParameters {
|
|
std::string name;
|
|
size_t in_vaddr, in_filesz, in_memsz;
|
|
size_t out_vaddr, out_filesz, out_memsz;
|
|
size_t trim_start, trim_end;
|
|
};
|
|
|
|
class PhdrTrimSegmentTests : public ::testing::TestWithParam<PhdrTrimSegmentTestParameters> {
|
|
protected:
|
|
};
|
|
|
|
TEST_P(PhdrTrimSegmentTests, Tests) {
|
|
auto params = GetParam();
|
|
ElfW(Phdr) phdr{};
|
|
phdr.p_vaddr = params.in_vaddr;
|
|
phdr.p_paddr = params.in_vaddr;
|
|
phdr.p_filesz = params.in_filesz;
|
|
phdr.p_memsz = params.in_memsz;
|
|
phdr_trim_segment(&phdr, params.trim_start, params.trim_end);
|
|
ASSERT_EQ(phdr.p_vaddr, params.out_vaddr);
|
|
ASSERT_EQ(phdr.p_paddr, params.out_vaddr);
|
|
ASSERT_EQ(phdr.p_filesz, params.out_filesz);
|
|
ASSERT_EQ(phdr.p_memsz, params.out_memsz);
|
|
}
|
|
|
|
INSTANTIATE_TEST_CASE_P(ldso_trampoline, PhdrTrimSegmentTests,
|
|
::testing::Values(
|
|
PhdrTrimSegmentTestParameters{
|
|
.name = "noop",
|
|
.in_vaddr = 1,
|
|
.in_filesz = 2,
|
|
.in_memsz = 2,
|
|
.trim_start = 1,
|
|
.trim_end = 3,
|
|
.out_vaddr = 1,
|
|
.out_filesz = 2,
|
|
.out_memsz = 2,
|
|
},
|
|
PhdrTrimSegmentTestParameters{
|
|
.name = "trim_beginning",
|
|
.in_vaddr = 1,
|
|
.in_filesz = 2,
|
|
.in_memsz = 2,
|
|
.trim_start = 2,
|
|
.trim_end = 3,
|
|
.out_vaddr = 2,
|
|
.out_filesz = 1,
|
|
.out_memsz = 1,
|
|
},
|
|
PhdrTrimSegmentTestParameters{
|
|
.name = "trim_end",
|
|
.in_vaddr = 1,
|
|
.in_filesz = 2,
|
|
.in_memsz = 2,
|
|
.trim_start = 1,
|
|
.trim_end = 2,
|
|
.out_vaddr = 1,
|
|
.out_filesz = 1,
|
|
.out_memsz = 1,
|
|
},
|
|
PhdrTrimSegmentTestParameters{
|
|
.name = "trim_data_bss",
|
|
.in_vaddr = 1,
|
|
.in_filesz = 2,
|
|
.in_memsz = 3,
|
|
.trim_start = 2,
|
|
.trim_end = 4,
|
|
.out_vaddr = 2,
|
|
.out_filesz = 1,
|
|
.out_memsz = 2,
|
|
}),
|
|
[](const testing::TestParamInfo<PhdrTrimSegmentTestParameters>& info) {
|
|
return info.param.name;
|
|
});
|