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

ap: allow batch operation against multiple package files

Browse Source
This commit is contained in:
Vladislav Yarmak 2019-09-04 18:20:51 +03:00
parent ed2642e83f
commit cd93eb85e2
1 changed files with 40 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -22,7 +22,8 @@ def parse_args():
description="Generates .1337 patch for Nvidia drivers for Windows", description="Generates .1337 patch for Nvidia drivers for Windows",
formatter_class=argparse.ArgumentDefaultsHelpFormatter) formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument("installer_file", parser.add_argument("installer_file",
help="location of installer executable") nargs="+",
help="location of installer executable(s)")
parser.add_argument("-7", "--7zip", parser.add_argument("-7", "--7zip",
default="7z", default="7z",
dest="sevenzip", dest="sevenzip",
@ -154,44 +155,45 @@ def main():
replacement = unhexlify(args.replacement) replacement = unhexlify(args.replacement)
assert len(search) == len(replacement), "len() of search and replacement"\ assert len(search) == len(replacement), "len() of search and replacement"\
" is not equal" " is not equal"
patch = make_patch(args.installer_file, for installer_file in args.installer_file:
arch_tgt=args.target, patch = make_patch(installer_file,
search=search, arch_tgt=args.target,
replacement=replacement, search=search,
sevenzip=args.sevenzip) replacement=replacement,
patch_content = format_patch(patch, args.target_name) sevenzip=args.sevenzip)
if args.stdout: patch_content = format_patch(patch, args.target_name)
with open(sys.stdout.fileno(), mode='wb', closefd=False) as out: if args.stdout:
out.write(patch_content) with open(sys.stdout.fileno(), mode='wb', closefd=False) as out:
else: out.write(patch_content)
version, product_type = identify_driver(args.installer_file,
sevenzip=args.sevenzip)
drv_prefix = {
"100": "quadro_",
"300": "",
"301": "nsd_",
}
installer_name = os.path.basename(args.installer_file).lower()
if 'winserv2008' in installer_name:
os_prefix = 'ws2012_x64'
elif 'winserv-2016' in installer_name:
os_prefix = 'ws2016_x64'
elif 'win10' in installer_name:
os_prefix = 'win10_x64'
elif 'win7' in installer_name:
os_prefix = 'win7_x64'
else: else:
raise UnknownPlatformException("Can't infer platform from filename %s" version, product_type = identify_driver(installer_file,
% (repr(installer_name),)) sevenzip=args.sevenzip)
driver_name = drv_prefix[product_type] + version drv_prefix = {
out_dir = os.path.join( "100": "quadro_",
os.path.dirname( "300": "",
os.path.abspath(__file__)), '..', '..', os_prefix, driver_name) "301": "nsd_",
os.makedirs(out_dir, 0o755, True) }
out_filename = os.path.join(out_dir, installer_name = os.path.basename(installer_file).lower()
os.path.splitext(args.target_name)[0] + PATCH_EXT) if 'winserv2008' in installer_name:
with open(out_filename, 'xb') as out: os_prefix = 'ws2012_x64'
out.write(patch_content) elif 'winserv-2016' in installer_name:
os_prefix = 'ws2016_x64'
elif 'win10' in installer_name:
os_prefix = 'win10_x64'
elif 'win7' in installer_name:
os_prefix = 'win7_x64'
else:
raise UnknownPlatformException("Can't infer platform from filename %s"
% (repr(installer_name),))
driver_name = drv_prefix[product_type] + version
out_dir = os.path.join(
os.path.dirname(
os.path.abspath(__file__)), '..', '..', os_prefix, driver_name)
os.makedirs(out_dir, 0o755, True)
out_filename = os.path.join(out_dir,
os.path.splitext(args.target_name)[0] + PATCH_EXT)
with open(out_filename, 'xb') as out:
out.write(patch_content)
if __name__ == '__main__': if __name__ == '__main__':