LIEF/tests/elf/test_empty_gnu_hash.py

#!/usr/bin/env python
import unittest
import logging
import os
import sys
import stat
import re
import subprocess
import tempfile
import shutil
from subprocess import Popen
import ctypes
import lief

from unittest import TestCase
from utils import get_sample, has_recent_glibc

class TestEmptyGNUHash(TestCase):
    SYMBOLS = {
        "myinstance": 0x1159,
        "myinit":     0x1175,
        "mycalc":     0x1199,
        "mydelete":   0x1214,
    }

    def setUp(self):
        self.logger = logging.getLogger(__name__)
        self.tmp_dir = tempfile.mkdtemp(suffix='_lief_test_empty_gnu_hash')
        self.logger.debug("temp dir: {}".format(self.tmp_dir))


    @unittest.skipUnless(sys.platform.startswith("linux"), "requires Linux")
    @unittest.skipUnless(has_recent_glibc(), "Need a recent GLIBC version")
    def test_export(self):
        target_path = get_sample('ELF/ELF64_x86-64_binary_empty-gnu-hash.bin')
        output      = os.path.join(self.tmp_dir, "libnoempty.so")

        binary = lief.parse(target_path)

        binary[lief.ELF.DYNAMIC_TAGS.FLAGS_1].remove(lief.ELF.DYNAMIC_FLAGS_1.PIE)

        for name, addr in TestEmptyGNUHash.SYMBOLS.items():
            binary.add_exported_function(addr, name)
        binary.write(output)

        st = os.stat(output)
        os.chmod(output, st.st_mode | stat.S_IEXEC)

        lib = ctypes.cdll.LoadLibrary(output)
        # Raise 'AttributeError' if not exported
        print(lib.myinstance)
        self.assertIsNotNone(lib.myinstance)

    def tearDown(self):
        # Delete it
        if os.path.isdir(self.tmp_dir):
            shutil.rmtree(self.tmp_dir)

if __name__ == '__main__':
    root_logger = logging.getLogger()
    root_logger.setLevel(logging.DEBUG)

    ch = logging.StreamHandler()
    ch.setLevel(logging.DEBUG)
    root_logger.addHandler(ch)

    unittest.main(verbosity=2)