Mirrored_Repos/nvidia-patch
1
0
Fork 0
mirror of https://github.com/keylase/nvidia-patch.git synced 2024-11-23 05:57:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

ap: direct mode

Browse Source
This commit is contained in:
Vladislav Yarmak 2019-09-14 16:10:28 +03:00
parent 8490006b82
commit 0f12e20b91
1 changed files with 19 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
win/tools/autopatch/autopatch.py
View File

@ -44,6 +44,10 @@ def parse_args():
parser.add_argument("-o", "--stdout", parser.add_argument("-o", "--stdout",
action="store_true", action="store_true",
help="output into stdout") help="output into stdout")
parser.add_argument("-D", "--direct",
action="store_true",
help="supply patched library directly instead of "
"installer file. Implies --stdout option.")
args = parser.parse_args() args = parser.parse_args()
return args return args
@ -113,13 +117,18 @@ def make_patch(archive, *,
arch_tgt, arch_tgt,
search, search,
replacement, replacement,
sevenzip="7z"): sevenzip="7z",
with tempfile.TemporaryDirectory() as tmpdir: direct=False):
with ExtractedTarget(archive, if direct:
tmpdir, with open(archive, 'rb') as fo:
arch_tgt, f = fo.read()
sevenzip=sevenzip) as tgt: else:
f = expand(tgt, sevenzip=sevenzip) with tempfile.TemporaryDirectory() as tmpdir:
with ExtractedTarget(archive,
tmpdir,
arch_tgt,
sevenzip=sevenzip) as tgt:
f = expand(tgt, sevenzip=sevenzip)
offset = f.find(search) offset = f.find(search)
del f del f
if offset == -1: if offset == -1:
@ -160,9 +169,10 @@ def main():
arch_tgt=args.target, arch_tgt=args.target,
search=search, search=search,
replacement=replacement, replacement=replacement,
sevenzip=args.sevenzip) sevenzip=args.sevenzip,
direct=args.direct)
patch_content = format_patch(patch, args.target_name) patch_content = format_patch(patch, args.target_name)
if args.stdout: if args.stdout or args.direct:
with open(sys.stdout.fileno(), mode='wb', closefd=False) as out: with open(sys.stdout.fileno(), mode='wb', closefd=False) as out:
out.write(patch_content) out.write(patch_content)
else: else: