mirror of
https://github.com/lief-project/LIEF
synced 2026-06-08 15:30:44 +00:00
lief-agent
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
#!/usr/bin/env python
|
|
import os
|
|
import stat
|
|
import ctypes
|
|
import lief
|
|
import pytest
|
|
|
|
from utils import (
|
|
get_sample, has_recent_glibc, is_linux, is_x86_64, check_layout
|
|
)
|
|
|
|
SYMBOLS = {
|
|
"myinstance": 0x1159,
|
|
"myinit": 0x1175,
|
|
"mycalc": 0x1199,
|
|
"mydelete": 0x1214,
|
|
}
|
|
|
|
@pytest.mark.skipif(not is_linux() or not is_x86_64(), reason="requires Linux x86-64")
|
|
@pytest.mark.skipif(not has_recent_glibc(), reason="needs a recent GLIBC version")
|
|
def test_gnu_hash(tmpdir):
|
|
target_path = get_sample('ELF/ELF64_x86-64_binary_empty-gnu-hash.bin')
|
|
output = os.path.join(tmpdir, "libnoempty.so")
|
|
|
|
binary = lief.ELF.parse(target_path)
|
|
entry_flag: lief.ELF.DynamicEntryFlags = binary[lief.ELF.DynamicEntry.TAG.FLAGS_1]
|
|
entry_flag.remove(lief.ELF.DynamicEntryFlags.FLAG.PIE)
|
|
|
|
for name, addr in SYMBOLS.items():
|
|
binary.add_exported_function(addr, name)
|
|
binary.write(output)
|
|
|
|
check_layout(output)
|
|
|
|
st = os.stat(output)
|
|
os.chmod(output, st.st_mode | stat.S_IEXEC)
|
|
lief.logging.info(output)
|
|
|
|
lib = ctypes.cdll.LoadLibrary(output)
|
|
|
|
# Raise 'AttributeError' if not exported
|
|
lief.logging.info(lib.myinstance)
|
|
assert lib.myinstance is not None
|