#!/usr/bin/env python
#  Copyright 2018 The ANGLE Project Authors. All rights reserved.
#  Use of this source code is governed by a BSD-style license that can be
#  found in the LICENSE file.

# Generate library file with compressed symbols per Android build
# process.
#

import argparse
import os
import subprocess
import sys


def main():
  parser = argparse.ArgumentParser(description=__doc__)
  parser.add_argument(
      '--objcopy',
      required=True,
      help='The objcopy binary to run',
      metavar='PATH')
  parser.add_argument(
      '--nm', required=True, help='The nm binary to run', metavar='PATH')
  parser.add_argument(
      '--sofile',
      required=True,
      help='Shared object file produced by linking command',
      metavar='FILE')
  parser.add_argument(
      '--output',
      required=True,
      help='Final output shared object file',
      metavar='FILE')
  parser.add_argument(
      '--unstrippedsofile',
      required=True,
      help='Unstripped shared object file produced by linking command',
      metavar='FILE')
  args = parser.parse_args()

  copy_cmd = ["cp", args.sofile, args.output]
  result = subprocess.call(copy_cmd)

  objcopy_cmd = [args.objcopy]
  objcopy_cmd.append('--only-keep-debug')
  objcopy_cmd.append(args.unstrippedsofile)
  objcopy_cmd.append(args.output + '.debug')
  result = subprocess.call(objcopy_cmd)

  nm_cmd = subprocess.Popen(
      [args.nm, args.unstrippedsofile, '--format=posix', '--defined-only'],
      stdout=subprocess.PIPE)

  awk_cmd = subprocess.Popen(['awk', '{ print $1}'],
                             stdin=nm_cmd.stdout,
                             stdout=subprocess.PIPE)

  dynsym_out = open(args.output + '.dynsyms', 'w')
  sort_cmd = subprocess.Popen(['sort'], stdin=awk_cmd.stdout, stdout=dynsym_out)
  dynsym_out.close()

  nm_cmd = subprocess.Popen(
      [args.nm, args.unstrippedsofile, '--format=posix', '--defined-only'],
      stdout=subprocess.PIPE)

  awk_cmd = subprocess.Popen(
      ['awk', '{ if ($2 == "T" || $2 == "t" ||' + ' $2 == "D") print $1 }'],
      stdin=nm_cmd.stdout,
      stdout=subprocess.PIPE)

  funcsyms_out = open(args.output + '.funcsyms', 'w')
  sort_cmd = subprocess.Popen(['sort'],
                              stdin=awk_cmd.stdout,
                              stdout=funcsyms_out)
  funcsyms_out.close()

  keep_symbols = open(args.output + '.keep_symbols', 'w')
  sort_cmd = subprocess.Popen(
      ['comm', '-13', args.output + '.dynsyms', args.output + '.funcsyms'],
      stdin=awk_cmd.stdout,
      stdout=keep_symbols)

  # Ensure that the keep_symbols file is not empty.
  keep_symbols.write("\n")
  keep_symbols.close()

  objcopy_cmd = [
      args.objcopy, '--rename-section', '.debug_frame=saved_debug_frame',
      args.output + '.debug', args.output + ".mini_debuginfo"
  ]
  subprocess.check_call(objcopy_cmd)

  objcopy_cmd = [
      args.objcopy, '-S', '--remove-section', '.gdb_index', '--remove-section',
      '.comment', '--keep-symbols=' + args.output + '.keep_symbols',
      args.output + '.mini_debuginfo'
  ]
  subprocess.check_call(objcopy_cmd)

  objcopy_cmd = [
      args.objcopy, '--rename-section', '.saved_debug_frame=.debug_frame',
      args.output + ".mini_debuginfo"
  ]
  subprocess.check_call(objcopy_cmd)

  xz_cmd = ['xz', args.output + '.mini_debuginfo']
  subprocess.check_call(xz_cmd)

  objcopy_cmd = [
      args.objcopy, '--add-section',
      '.gnu_debugdata=' + args.output + '.mini_debuginfo.xz', args.output
  ]
  subprocess.check_call(objcopy_cmd)

  # Clean out scratch files
  rm_cmd = [
      'rm', '-f', args.output + '.dynsyms', args.output + '.funcsyms',
      args.output + '.keep_symbols', args.output + '.debug',
      args.output + '.mini_debuginfo', args.output + '.mini_debuginfo.xz'
  ]
  result = subprocess.call(rm_cmd)

  return result


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