#!/usr/bin/env python3
#
# Copyright 2021 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import argparse
import subprocess
import sys


def main() -> int:
  parser = argparse.ArgumentParser(description="Strip binary using LLVM tools.")
  parser.add_argument("--llvm-strip-binary-path",
                      help="Path to llvm-strip executable.")
  parser.add_argument("--llvm-objcopy-binary-path",
                      required=True,
                      help="Path to llvm-objcopy executable.")
  parser.add_argument("--binary-input", help="Input ELF binary.")
  parser.add_argument("--symbol-output",
                      help="File to write extracted debug info (.debug).")
  parser.add_argument("--stripped-binary-output",
                      help="File to write stripped binary.")
  args = parser.parse_args()

  # Replicate the behavior of:
  # eu-strip <binary_input> -o <stripped_binary_output> -f <symbol_output>
  subprocess.check_output([
      args.llvm_objcopy_binary_path,
      "--only-keep-debug",
      args.binary_input,
      args.symbol_output,
  ])
  subprocess.check_output([
      args.llvm_strip_binary_path,
      "--strip-debug",
      "--strip-unneeded",
      "-o",
      args.stripped_binary_output,
      args.binary_input,
  ])
  subprocess.check_output([
      args.llvm_objcopy_binary_path,
      f"--add-gnu-debuglink={args.symbol_output}",
      args.stripped_binary_output,
  ])

  return 0


if __name__ == "__main__":
  sys.exit(main())