mirror of
https://github.com/QuasarApp/LIEF.git
synced 2025-04-26 12:24:32 +00:00
3550e45f18
* Temporarily disable Melkor in test suite The Makefile uses the -executable flag which doesn't work on RHEL5. Disabling until I find a workaround for this. * Patch Makefile in Melkor fuzzer This replaces the `-executable` flag which is not supported by `find` on CentOS 5. * Respect CC environment variable Several tests hard coded /usr/bin/cc which might not exist in some environments. We first check the CC environment variable and fall back to the hard coded path if CC is unset. * Skip tests on GLIBC < 2.17 Some test binaries were linked against GLIBC 2.17. Skip tests which use this binary if the platform does not have the required GLIBC version. * Enable ccache in Docker in Travis builds * Run `auditwheel repair` on the produced wheels This will vendor the needed external shared libraries into the wheel and tag it as manylinux1. * Install ccache in Docker image * Avoid using bind mount volume in Docker build The bind mount volume wrote files as 'root' which causes issues with the deploy script in Travis. Copying the source code into the image and retrieving the built wheels instead of mounting the source tree fixes this issue. * Fix missing build folder when building with Docker After finishing the build inside Docker we need the build directory from the container to be able to deploy the built artifacts with deploy.sh. * Use the right Python interpreter for Linux builds The Dockerized .travis.yml builds attempt to invoke the interpreter in the PYTHON_BINARY environment variable, which is only valid inside the Docker image. To fix this, override the variable on Linux for tasks which require the host's Python interpreter. * Fix missing pip installation in Travis The Ubuntu image in Travis does not come with `pip` preinstalled for Python 3. * Remove .git directory from .dockerignore As `setup.py` uses `git` to determine the version number we need to copy the contents of `.git` into the image.
293 lines
8.5 KiB
Python
293 lines
8.5 KiB
Python
import unittest
|
|
import logging
|
|
import shlex
|
|
import subprocess
|
|
import tempfile
|
|
import lief
|
|
import os
|
|
import stat
|
|
import shutil
|
|
import sys
|
|
from collections import namedtuple
|
|
|
|
from utils import get_compiler
|
|
|
|
class CommandResult(object):
|
|
def __init__(self, output, error, retcode, process=None):
|
|
self.output = output
|
|
self.error = error
|
|
self.retcode = retcode
|
|
self.process = process
|
|
|
|
def __bool__(self):
|
|
return not self.retcode
|
|
|
|
def __str__(self):
|
|
if bool(self):
|
|
return self.output
|
|
return self.error
|
|
|
|
|
|
|
|
|
|
lief.Logger.set_level(lief.LOGGING_LEVEL.INFO)
|
|
|
|
LOGGING_LEVEL = logging.DEBUG
|
|
|
|
LIBADD = """\
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#define LOCAL __attribute__ ((visibility ("hidden")))
|
|
|
|
LOCAL int add_hidden(int a, int b) {
|
|
printf("[LOCAL] %d + %d = %d\\n", a, b, a+b);
|
|
return a + b;
|
|
}
|
|
|
|
|
|
int main(int argc, char** argv) {
|
|
|
|
if (argc != 3) {
|
|
printf("Usage: %s <a> <b>\\n", argv[0]);
|
|
exit(-1);
|
|
}
|
|
|
|
printf("Hello\\n");
|
|
int res = add_hidden(atoi(argv[1]), atoi(argv[2]));
|
|
printf("From add_hidden@libadd.so a + b = %d\\n", res);
|
|
return 0;
|
|
}
|
|
"""
|
|
|
|
|
|
BINADD = """\
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
extern int add_hidden(int a, int b);
|
|
|
|
int main(int argc, char **argv) {
|
|
if (argc != 3) {
|
|
printf("Usage: %s <a> <b>\\n", argv[0]);
|
|
exit(-1);
|
|
}
|
|
|
|
printf("Hello\\n");
|
|
int res = add_hidden(atoi(argv[1]), atoi(argv[2]));
|
|
printf("From add_hidden@libadd.so a + b = %d\\n", res);
|
|
return 0;
|
|
}
|
|
"""
|
|
|
|
|
|
class TestBin2Lib(unittest.TestCase):
|
|
LOGGER = logging.getLogger(__name__)
|
|
|
|
def setUp(self):
|
|
self._logger = logging.getLogger(__name__)
|
|
|
|
@staticmethod
|
|
def run_cmd(cmd):
|
|
TestBin2Lib.LOGGER.debug("Running: '{}'".format(cmd))
|
|
cmd = shlex.split(cmd)
|
|
p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True)
|
|
stdout, stderr = p.communicate()
|
|
|
|
if stdout:
|
|
TestBin2Lib.LOGGER.debug(stdout)
|
|
|
|
if stderr:
|
|
TestBin2Lib.LOGGER.error(stderr)
|
|
|
|
return CommandResult(stdout, stderr, p.returncode)
|
|
|
|
@unittest.skipUnless(sys.platform.startswith("linux"), "requires Linux")
|
|
def test_libadd(self):
|
|
|
|
|
|
_, binaddc = tempfile.mkstemp(prefix="binadd_", suffix=".c")
|
|
_, libaddc = tempfile.mkstemp(prefix="libadd_", suffix=".c")
|
|
|
|
self._logger.debug(binaddc)
|
|
self._logger.debug(libaddc)
|
|
|
|
fd, binadd = tempfile.mkstemp(prefix="binadd_", suffix=".bin")
|
|
_, libadd = tempfile.mkstemp(prefix="libadd_", suffix=".so")
|
|
_, libadd2 = tempfile.mkstemp(prefix="libadd2_", suffix=".so")
|
|
|
|
self._logger.debug(binadd)
|
|
self._logger.debug(libadd)
|
|
self._logger.debug(libadd2)
|
|
|
|
|
|
with open(binaddc, 'w') as f:
|
|
f.write(BINADD)
|
|
|
|
with open(libaddc, 'w') as f:
|
|
f.write(LIBADD)
|
|
|
|
compiler = get_compiler()
|
|
|
|
# Compile libadd
|
|
r = self.run_cmd("{compiler} -Wl,--export-dynamic -mcmodel=large -fPIE -pie -o {output} {input}".format(
|
|
compiler=compiler,
|
|
output=libadd,
|
|
input=libaddc))
|
|
self.assertTrue(r, msg="Unable to compile libadd")
|
|
|
|
libadd = lief.parse(libadd)
|
|
|
|
libadd_hidden = libadd.get_symbol("add_hidden")
|
|
libadd_hidden.binding = lief.ELF.SYMBOL_BINDINGS.GLOBAL
|
|
libadd_hidden.visibility = lief.ELF.SYMBOL_VISIBILITY.DEFAULT
|
|
libadd_hidden = libadd.add_dynamic_symbol(libadd_hidden, lief.ELF.SymbolVersion.global_)
|
|
self._logger.debug(libadd_hidden)
|
|
|
|
libadd.add(lief.ELF.DynamicSharedObject(os.path.basename(libadd2)))
|
|
|
|
libadd.write(libadd2)
|
|
|
|
lib_directory = os.path.dirname(libadd2)
|
|
libname = os.path.basename(libadd2)[3:-3] # libadd.so ---> add
|
|
|
|
r = self.run_cmd("{compiler} -Wl,--export-dynamic -mcmodel=large -fPIE -pie -Wl,-rpath={libdir} -L{libdir} -o {output} {input} -l{libadd2}".format(
|
|
compiler=compiler,
|
|
libdir=lib_directory,
|
|
libadd2=libname,
|
|
output=binadd,
|
|
input=binaddc))
|
|
self.assertTrue(r, msg="Unable to compile binadd")
|
|
|
|
os.close(fd)
|
|
st = os.stat(binadd)
|
|
os.chmod(binadd, st.st_mode | stat.S_IEXEC)
|
|
|
|
r = self.run_cmd(binadd + " 1 2")
|
|
self.assertTrue(r)
|
|
self.assertIn("From add_hidden@libadd.so a + b = 3", r.output)
|
|
|
|
|
|
@unittest.skipUnless(sys.platform.startswith("linux"), "requires Linux")
|
|
def test_libadd_api(self):
|
|
_, binaddc = tempfile.mkstemp(prefix="binadd_", suffix=".c")
|
|
_, libaddc = tempfile.mkstemp(prefix="libadd_", suffix=".c")
|
|
|
|
self._logger.debug(binaddc)
|
|
self._logger.debug(libaddc)
|
|
|
|
fd, binadd = tempfile.mkstemp(prefix="binadd_", suffix=".bin")
|
|
_, libadd = tempfile.mkstemp(prefix="libadd_", suffix=".so")
|
|
_, libadd2 = tempfile.mkstemp(prefix="libadd2_", suffix=".so")
|
|
|
|
self._logger.debug(binadd)
|
|
self._logger.debug(libadd)
|
|
self._logger.debug(libadd2)
|
|
|
|
|
|
with open(binaddc, 'w') as f:
|
|
f.write(BINADD)
|
|
|
|
with open(libaddc, 'w') as f:
|
|
f.write(LIBADD)
|
|
|
|
compiler = get_compiler()
|
|
|
|
# Compile libadd
|
|
r = self.run_cmd("{compiler} -Wl,--export-dynamic -mcmodel=large -fPIE -pie -o {output} {input}".format(
|
|
compiler=compiler,
|
|
output=libadd,
|
|
input=libaddc))
|
|
self.assertTrue(r, msg="Unable to compile libadd")
|
|
|
|
libadd = lief.parse(libadd)
|
|
libadd.export_symbol("add_hidden")
|
|
libadd.write(libadd2)
|
|
|
|
lib_directory = os.path.dirname(libadd2)
|
|
libname = os.path.basename(libadd2)[3:-3] # libadd.so ---> add
|
|
|
|
r = self.run_cmd("{compiler} -Wl,--export-dynamic -mcmodel=large -fPIE -pie -Wl,-rpath={libdir} -L{libdir} -o {output} {input} -l{libadd2}".format(
|
|
compiler=compiler,
|
|
libdir=lib_directory,
|
|
libadd2=libname,
|
|
output=binadd,
|
|
input=binaddc))
|
|
self.assertTrue(r, msg="Unable to compile binadd")
|
|
|
|
os.close(fd)
|
|
st = os.stat(binadd)
|
|
os.chmod(binadd, st.st_mode | stat.S_IEXEC)
|
|
|
|
r = self.run_cmd(binadd + " 1 2")
|
|
self.assertTrue(r)
|
|
self.assertIn("From add_hidden@libadd.so a + b = 3", r.output)
|
|
|
|
|
|
@unittest.skipUnless(sys.platform.startswith("linux"), "requires Linux")
|
|
def test_libadd_api2(self):
|
|
_, binaddc = tempfile.mkstemp(prefix="binadd_", suffix=".c")
|
|
_, libaddc = tempfile.mkstemp(prefix="libadd_", suffix=".c")
|
|
|
|
self._logger.debug(binaddc)
|
|
self._logger.debug(libaddc)
|
|
|
|
fd, binadd = tempfile.mkstemp(prefix="binadd_", suffix=".bin")
|
|
_, libadd = tempfile.mkstemp(prefix="libadd_", suffix=".so")
|
|
_, libadd2 = tempfile.mkstemp(prefix="libadd2_", suffix=".so")
|
|
|
|
self._logger.debug(binadd)
|
|
self._logger.debug(libadd)
|
|
self._logger.debug(libadd2)
|
|
|
|
|
|
with open(binaddc, 'w') as f:
|
|
f.write(BINADD)
|
|
|
|
with open(libaddc, 'w') as f:
|
|
f.write(LIBADD)
|
|
|
|
compiler = get_compiler()
|
|
|
|
# Compile libadd
|
|
r = self.run_cmd("{compiler} -Wl,--export-dynamic -mcmodel=large -fPIE -pie -o {output} {input}".format(
|
|
compiler=compiler,
|
|
output=libadd,
|
|
input=libaddc))
|
|
self.assertTrue(r, msg="Unable to compile libadd")
|
|
|
|
libadd = lief.parse(libadd)
|
|
add_hidden_static = libadd.get_static_symbol("add_hidden")
|
|
libadd.add_exported_function(add_hidden_static.value, add_hidden_static.name)
|
|
libadd.write(libadd2)
|
|
|
|
lib_directory = os.path.dirname(libadd2)
|
|
libname = os.path.basename(libadd2)[3:-3] # libadd.so ---> add
|
|
|
|
r = self.run_cmd("{compiler} -Wl,--export-dynamic -mcmodel=large -fPIE -pie -Wl,-rpath={libdir} -L{libdir} -o {output} {input} -l{libadd2}".format(
|
|
compiler=compiler,
|
|
libdir=lib_directory,
|
|
libadd2=libname,
|
|
output=binadd,
|
|
input=binaddc))
|
|
self.assertTrue(r, msg="Unable to compile binadd")
|
|
|
|
os.close(fd)
|
|
st = os.stat(binadd)
|
|
os.chmod(binadd, st.st_mode | stat.S_IEXEC)
|
|
|
|
r = self.run_cmd(binadd + " 1 2")
|
|
self.assertTrue(r)
|
|
self.assertIn("From add_hidden@libadd.so a + b = 3", r.output)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
root_logger = logging.getLogger()
|
|
root_logger.addHandler(logging.StreamHandler())
|
|
root_logger.setLevel(LOGGING_LEVEL)
|
|
for h in root_logger.handlers:
|
|
h.setLevel(LOGGING_LEVEL)
|
|
|
|
unittest.main(verbosity=2)
|
|
|
|
|