kuschelyagi/lil-bits
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

blasphemy: allow passing in args after --run for convenience

Browse Source
This commit is contained in:
Adrian Kuschelyagi Malacoda 2020-10-24 05:04:06 -05:00
parent 00bbf3300b
commit cb8f881ce2
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
blasphemy
View File

@ -84,7 +84,7 @@ def print_paths(paths):
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--wrap", action="store_true", help="output wrapper script to stdout") parser.add_argument("--wrap", action="store_true", help="output wrapper script to stdout")
parser.add_argument("--run", action="store_true", help="run command with specified args") parser.add_argument("--run", nargs="*", help="run command with specified args")
parser.add_argument("--command", help="command to wrap or run (default to the first input binary)") parser.add_argument("--command", help="command to wrap or run (default to the first input binary)")
parser.add_argument("--arguments", nargs="+", default=[], help="Additional arguments to pass to command") parser.add_argument("--arguments", nargs="+", default=[], help="Additional arguments to pass to command")
parser.add_argument("--inputs", nargs="+", default=[], help="input binaries to scan for dependencies") parser.add_argument("--inputs", nargs="+", default=[], help="input binaries to scan for dependencies")
@ -107,10 +107,10 @@ if (not command) and args.inputs:
#command[0] = os.path.abspath(command[0]) #command[0] = os.path.abspath(command[0])
command = pre_command + [command] + args.arguments command = pre_command + [command] + args.arguments
if args.run: if args.run is not None:
environment = {"LD_LIBRARY_PATH": make_ld_library_path(paths)} environment = {"LD_LIBRARY_PATH": make_ld_library_path(paths)}
environment.update(os.environ) environment.update(os.environ)
completed_process = run(command, env=environment) completed_process = run(command + args.run, env=environment)
sys.exit(completed_process.returncode) sys.exit(completed_process.returncode)
elif args.wrap: elif args.wrap:
print(generate_wrapper(paths, command)) print(generate_wrapper(paths, command))