Function main

tools/prepare_lief.py:103–306

()

def main():
    ap = argparse.ArgumentParser(description='Prepare LIEF third-party and configured headers for checked-in sources')
    ap.add_argument('--source-dir', required=True, help='Path to the unzipped LIEF source directory (e.g. /tmp/LIEF-<ver>/)')
    ap.add_argument('--lief-dir', required=True, help='Target path to deps/LIEF (the vendored destination in this repo)')
    ap.add_argument('--version', help='LIEF version (X.Y.Z). If omitted, try to infer a fallback (0.17.0).')
    ap.add_argument('--commit', default='', help='Commit hash to embed into version.h (optional)')
    ap.add_argument('--tag', default='', help='Git tag to embed into version.h (optional)')
    args = ap.parse_args()

    source_dir = Path(args.source_dir).resolve()
    lief_dir = Path(args.lief_dir).resolve()
    if not source_dir.exists():
        print(f"source-dir does not exist: {source_dir}", file=sys.stderr)
        return 1
    if not lief_dir.exists():
        ensure_dir(lief_dir)

    # 1) Copy only the relevant files from source_dir to lief_dir
    included_dirs = ['src', 'include', 'config', 'README.md', 'LICENSE']
    for item in included_dirs:
        src = source_dir / item
        dst = lief_dir / item
        if not src.exists():
            print(f"Warning: expected item missing: {src}", file=sys.stderr)
            continue
        print(f"Copying {src} -> {dst}")
        if src.is_dir():
            if dst.exists():
                shutil.rmtree(dst)
            shutil.copytree(src, dst)
        else:
            shutil.copy2(src, dst)

    # 2) Extract selected third-party zips into third-party/<base>
    # Keep this generic via a small allowlist so we can extend later if needed.
    # For now, we only extract 'mbedtls'.
    src_third_party = source_dir / 'third-party'
    dst_third_party = lief_dir / 'third-party'
    ensure_dir(dst_third_party)

    # Use a temporary third party directory, then move needed files to lief_dir
    tmp_third_party = Path(tempfile.mkdtemp())
    # For example, src_third_party/mbedtls-<ver>.zip contains mbedtls-<ver>/,
    # we extract that to tmp_third_party/mbedtls, and in the extracted directory, we
    # only want to copy the 'library' folder, so tmp_third_party/mbedtls/library ->
    # dst_third_party/mbedtls/library. Any other files under the extracted mbedtls
    # directory will be ignored.
    libs_to_extract = {
        'mbedtls': ['library', 'include', 'README.md', 'LICENSE'],  # #include "mbedtls/private_access.h"
        'spdlog': ['include', 'src', 'LICENSE', 'README.md'],  # #include "spdlog/fmt/..."
        'frozen': ['include', 'LICENSE', 'README.rst'],      # #include <frozen/map.h>
        'expected': ['include', 'README.md', 'COPYING'],  # #include <LIEF/third-party/internal/expected.hpp>
        'utfcpp': ['source', 'README.md', 'LICENSE'],     # #include <internal/utfcpp/utf8/unchecked.h> | #include "utf8/unchecked.h"
        'tcb-span': ['span.hpp'],  # #include <LIEF/third-party/internal/span.hpp>
    }
    extracted_libs = []
    with tempfile.TemporaryDirectory() as _tmp_dir:
        tmp_third_party = Path(_tmp_dir)