mirror of
https://github.com/lief-project/LIEF
synced 2026-06-08 15:30:44 +00:00
Romain Thomas
147 lines
4.5 KiB
Python
147 lines
4.5 KiB
Python
#!/usr/bin/env python
|
|
import pytest
|
|
|
|
import lief
|
|
import subprocess
|
|
from subprocess import Popen
|
|
from pathlib import Path
|
|
from utils import (
|
|
get_sample, has_private_samples, check_layout,
|
|
is_linux, is_x86_64, check_layout
|
|
)
|
|
|
|
def test_issue_863(tmp_path: Path):
|
|
elf = lief.ELF.parse(get_sample('ELF/issue_863.elf'))
|
|
|
|
assert elf.sysv_hash.nchain == 7
|
|
|
|
elf.remove_dynamic_symbol("puts")
|
|
|
|
out = tmp_path / "issue_863.modified"
|
|
elf.write(out)
|
|
|
|
check_layout(elf)
|
|
|
|
new = lief.ELF.parse(out)
|
|
assert new.sysv_hash.nchain == 6
|
|
|
|
def test_pr_968():
|
|
elf = lief.ELF.parse(get_sample('ELF/echo.mips_r3000.bin'))
|
|
sym: lief.ELF.Symbol = elf.get_symbol("strstr")
|
|
assert sym.imported
|
|
|
|
def test_issue_1023():
|
|
"""
|
|
Make sure that get_content_from_virtual_address return an empty
|
|
buffer when trying to read bss segment
|
|
"""
|
|
elf = lief.ELF.parse(get_sample('ELF/nopie_bss_671.elf'))
|
|
|
|
bss_segment = elf.segments[3]
|
|
bss_start = bss_segment.virtual_address + bss_segment.physical_size
|
|
bss_content = elf.get_content_from_virtual_address(bss_start + 1, 1)
|
|
|
|
assert len(bss_content) == 0
|
|
|
|
def test_issue_1082():
|
|
"""
|
|
Make sure RISC-V imported symbols are correctly exported
|
|
"""
|
|
elf = lief.ELF.parse(get_sample("ELF/issue-1082-pie.elf"))
|
|
imp_symbols = [s.name for s in elf.imported_symbols]
|
|
assert len(imp_symbols) == 6
|
|
assert imp_symbols[0] == "__libc_start_main"
|
|
assert imp_symbols[1] == "printf"
|
|
assert imp_symbols[2] == "__cxa_finalize"
|
|
assert imp_symbols[3] == "__libc_start_main@GLIBC_2.34"
|
|
assert imp_symbols[4] == "printf@GLIBC_2.27"
|
|
assert imp_symbols[5] == "__cxa_finalize@GLIBC_2.27"
|
|
|
|
elf = lief.ELF.parse(get_sample("ELF/issue-1082-no_pie.elf"))
|
|
imp_symbols = [s.name for s in elf.imported_symbols]
|
|
assert len(imp_symbols) == 4
|
|
assert imp_symbols[0] == "printf"
|
|
assert imp_symbols[2] == "__libc_start_main@GLIBC_2.34"
|
|
assert imp_symbols[3] == "printf@GLIBC_2.27"
|
|
|
|
def test_issue_1089(tmp_path: Path):
|
|
elf = lief.ELF.parse(get_sample("ELF/libip4tc.so.2.0.0"))
|
|
|
|
original_nb_relocations = len(elf.dynamic_relocations)
|
|
|
|
elf.remove_dynamic_symbol("iptc_read_counter")
|
|
|
|
out = tmp_path / "libip4tc.so.2.0.0"
|
|
elf.write(out)
|
|
|
|
new = lief.ELF.parse(out)
|
|
check_layout(new)
|
|
|
|
assert new.get_symbol("iptc_read_counter") is None
|
|
assert len(new.dynamic_relocations) == original_nb_relocations - 2
|
|
|
|
@pytest.mark.skipif(not has_private_samples(), reason="needs private samples")
|
|
def test_issue_1097(tmp_path: Path):
|
|
elf = lief.ELF.parse(get_sample("private/ELF/libhwui.so"))
|
|
deps = [entry.name for entry in elf.dynamic_entries if isinstance(entry, lief.ELF.DynamicEntryLibrary)]
|
|
|
|
out = tmp_path / "libhwui.so"
|
|
elf.write(out)
|
|
|
|
new = lief.ELF.parse(out)
|
|
check_layout(new)
|
|
new_deps = [entry.name for entry in new.dynamic_entries if isinstance(entry, lief.ELF.DynamicEntryLibrary)]
|
|
|
|
assert new_deps == deps
|
|
|
|
def test_issue_1277():
|
|
elf = lief.ELF.parse(get_sample("ELF/issue_1277.elf"))
|
|
assert elf is not None
|
|
|
|
def test_issue_1309(tmp_path: Path):
|
|
elf = lief.ELF.parse(get_sample("ELF/issue_1309.so"))
|
|
elf[lief.ELF.DynamicEntry.TAG.SONAME].name = "lib" + "a" * 100 + ".so"
|
|
|
|
out = tmp_path / "new.so"
|
|
elf.write(out)
|
|
|
|
new = lief.ELF.parse(out)
|
|
check_layout(new)
|
|
assert list(new.dynamic_symbols)[129].name == "base64_decode_utf16le"
|
|
|
|
@pytest.mark.skipif(condition=not has_private_samples(), reason="needs private samples")
|
|
@pytest.mark.slow
|
|
def test_issue_1315(tmp_path: Path):
|
|
# NOTE(romain): there is an exhaustive test here: 'private/ELF/issue_1315'
|
|
# that includes on-device testing.
|
|
elf = lief.ELF.parse(get_sample("private/ELF/rocksdb-native@3.bare"))
|
|
assert elf is not None
|
|
out = tmp_path / "lib.so"
|
|
|
|
config = lief.ELF.Builder.config_t()
|
|
config.force_relocate = True
|
|
|
|
elf.write(out, config)
|
|
new = lief.ELF.parse(out)
|
|
check_layout(new)
|
|
|
|
|
|
def test_issue_743(tmp_path: Path):
|
|
elf = lief.ELF.parse(get_sample('ELF/ld-linux-x86-64.so.2'))
|
|
|
|
out = tmp_path / "ld-linux-x86-64.so.2"
|
|
config = lief.ELF.Builder.config_t()
|
|
config.force_relocate = True
|
|
|
|
elf.write(out, config)
|
|
new = lief.ELF.parse(out)
|
|
check_layout(new)
|
|
|
|
if is_linux() and is_x86_64():
|
|
out.chmod(0o755)
|
|
with Popen([out.as_posix(), "--version"], universal_newlines=True,
|
|
stdout=subprocess.PIPE, stderr=subprocess.STDOUT) as proc:
|
|
stdout = proc.stdout.read()
|
|
proc.poll()
|
|
assert "Debian GLIBC 2.36-9+deb12u8" in stdout
|