Support rsp in linker intercept

This commit is contained in:
Andrea Fioraldi 2023-07-24 17:37:39 +02:00
parent e5cd808701
commit dcbf46b8ef

View File

@ -1,6 +1,6 @@
#!/usr/bin/env python3
import subprocess, json, sys, os
import subprocess, shutil, json, sys, os
FILTER = ['-shared']
@ -12,28 +12,44 @@ args = sys.argv[1:]
out_args = []
shareds = []
search = []
prev_o = False
for i in range(len(args)):
if prev_o:
prev_o = False
continue
elif args[i] in FILTER:
continue
elif args[i].endswith('.so') and not args[i].startswith('-'):
name = os.path.basename(args[i])[3:-3] # remove prefix and suffix
shareds.append(name)
continue
elif args[i] == '-o':
prev_o = True
continue
elif args[i].startswith('-l'):
shareds.append(args[i][2:])
continue
elif args[i].startswith('-L'):
search.append(args[i][2:])
out_args.append(args[i])
def process_args(args):
global out_args, shareds, search
prev_o = False
for i in range(len(args)):
if prev_o:
prev_o = False
continue
elif args[i] in FILTER:
continue
elif args[i].endswith('.so') and not args[i].startswith('-'):
name = os.path.basename(args[i])[3:-3] # remove prefix and suffix
shareds.append(name)
continue
elif args[i] == '-o':
prev_o = True
continue
elif args[i].startswith('-l'):
shareds.append(args[i][2:])
continue
elif args[i].endswith('.rsp'):
fname = args[i][1:] # remove initial @
with open(fname) as f:
rsp = f.read().split()
process_args(rsp)
continue
elif args[i].startswith('-L'):
search.append(args[i][2:])
out_args.append(args[i])
process_args(args)
with open(OUT, 'w') as f:
json.dump({'cmd': out_args, 'libs': shareds, 'search': search}, f, indent=2)
json.dump({
'cmd': out_args,
'libs': shareds,
'search': search,
}, f, indent=2)
subprocess.run([CXX] + args)