setup.py

#
# This file is part of libdebug Python library (https://github.com/libdebug/libdebug).
# Copyright (c) 2023-2024 Gabriele Digregorio, Roberto Alessandro Bertolini. All rights reserved.
# Licensed under the MIT license. See LICENSE file in the project root for details.
#

import os

from setuptools import find_packages, setup

try:
    from setuptools.command.build import build
except ImportError:
    from distutils.command.build import build

# Check if the user has the required C libraries installed
if not (
    os.path.isfile("/usr/include/sys/ptrace.h")
    or os.path.isfile("/usr/include/x86_64-linux-gnu/sys/ptrace.h")
    or os.path.isfile("/usr/include/aarch64-linux-gnu/sys/ptrace.h")
    or os.path.isfile("/usr/include/i386-linux-gnu/sys/ptrace.h")
):
    print("Required C libraries not found. Please install ptrace or kernel headers")
    exit(1)

if not (
    os.path.isfile("/usr/include/demangle.h")
    or os.path.isfile("/usr/include/libiberty/demangle.h")
):
    print(
        "Required C libraries not found. Please install libiberty-dev or binutils-devel"
    )
    exit(1)

if not os.path.isfile("/usr/include/libelf.h"):
    print("Required C libraries not found. Please install libelf-dev or elfutils")
    exit(1)

if os.path.isfile("/usr/include/libdwarf/dwarf.h") and os.path.isfile(
    "/usr/include/libdwarf/libdwarf.h"
):
    debug_sym_cffi = "debug_sym_cffi_build_legacy"
elif (
    os.path.isfile("/usr/include/libdwarf/libdwarf-0/dwarf.h")
    and os.path.isfile("/usr/include/libdwarf/libdwarf-0/libdwarf.h")
) or (
    os.path.isfile("/usr/include/libdwarf-0/dwarf.h")
    and os.path.isfile("/usr/include/libdwarf-0/libdwarf.h")
):
    debug_sym_cffi = "debug_sym_cffi_build"
else:
    print(
        "Required C libraries not found. Please install libdwarf-dev or libdwarf-devel"
    )
    exit(1)


class JumpstartBuildCommand(build):
    def run(self):
        os.system(
            "cc -o libdebug/ptrace/jumpstart/jumpstart libdebug/ptrace/jumpstart/jumpstart.c"
        )
        build.run(self)

    def get_outputs(self):
        outputs = build.get_outputs(self)
        outputs.append("libdebug/ptrace/jumpstart/jumpstart")
        return outputs


setup(
    name="libdebug",
    version="0.7.0",
    author="JinBlack, Io_no, MrIndeciso, Frank01001",
    description="A library to debug binary programs",
    packages=find_packages(include=["libdebug", "libdebug.*"]),
    install_requires=["capstone", "pyelftools", "cffi", "requests", "psutil", "prompt-toolkit"],
    setup_requires=["cffi"],
    cffi_modules=[
        "./libdebug/cffi/ptrace_cffi_build.py:ffibuilder",
        "./libdebug/cffi/personality_cffi_build.py:ffibuilder",
        f"./libdebug/cffi/{debug_sym_cffi}.py:ffibuilder",
    ],
    cmdclass={"build": JumpstartBuildCommand},
    package_data={
        "libdebug.ptrace.jumpstart": ["jumpstart", "jumpstart.c"],
        "libdebug.cffi": ["*.c"],
        "libdebug": ["py.typed"],
    },
    include_package_data=True,
)