mirror of
https://github.com/keylase/nvidia-patch
synced 2024-11-01 09:20:09 +00:00
75 lines
2.7 KiB
Python
Executable File
75 lines
2.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
import argparse
|
|
import json
|
|
import os.path
|
|
import posixpath
|
|
from string import Template
|
|
from itertools import groupby
|
|
from functools import partial
|
|
import urllib.request
|
|
import urllib.parse
|
|
|
|
from constants import OSKind, Product, WinSeries, DATAFILE_PATH, \
|
|
DRIVER_URL_TEMPLATE, DRIVER_DIR_PREFIX, BASE_PATH, REPO_BASE
|
|
from utils import find_driver, linux_driver_key, windows_driver_key
|
|
|
|
def posixpath_components(path):
|
|
result = []
|
|
while True:
|
|
head, tail = posixpath.split(path)
|
|
if head == path:
|
|
break
|
|
result.append(tail)
|
|
path = head
|
|
result.reverse()
|
|
if result and not result[-1]:
|
|
result.pop()
|
|
return result
|
|
|
|
def validate_patch(patch64, patch32):
|
|
wc_base = os.path.abspath(os.path.join(BASE_PATH, "..", ".."))
|
|
base_parse = urllib.parse.urlsplit(REPO_BASE, scheme='http')
|
|
p64_parse = urllib.parse.urlsplit(patch64, scheme='http')
|
|
p32_parse = urllib.parse.urlsplit(patch32, scheme='http')
|
|
if not (p64_parse[0] == p32_parse[0] == base_parse[0]):
|
|
raise Exception("URL scheme doesn't match repo base URL scheme")
|
|
if not (p64_parse[1] == p32_parse[1] == base_parse[1]):
|
|
raise Exception("URL network location doesn't match repo base URL network location")
|
|
if posixpath.commonpath((base_parse[2], p64_parse[2], p32_parse[2])) != \
|
|
posixpath.commonpath((base_parse[2],)):
|
|
raise Exception("URL is not subpath of repo base path")
|
|
p64_posix_relpath = posixpath.relpath(p64_parse[2], base_parse[2])
|
|
p32_posix_relpath = posixpath.relpath(p32_parse[2], base_parse[2])
|
|
p64_comp = posixpath_components(p64_posix_relpath)
|
|
p32_comp = posixpath_components(p32_posix_relpath)
|
|
p64_filepath = os.path.join(wc_base, *p64_comp)
|
|
p32_filepath = os.path.join(wc_base, *p32_comp)
|
|
if not os.path.exists(p64_filepath):
|
|
raise Exception("File %s not found!" % p64_filepath)
|
|
if not os.path.exists(p32_filepath):
|
|
raise Exception("File %s not found!" % p32_filepath)
|
|
if os.path.getsize(p64_filepath) == 0:
|
|
raise Exception("File %s empty!" % p64_filepath)
|
|
if os.path.exists(p32_filepath) == 0:
|
|
raise Exception("File %s empty!" % p32_filepath)
|
|
|
|
def main():
|
|
with open(DATAFILE_PATH) as data_file:
|
|
data = json.load(data_file)
|
|
|
|
drivers = data[OSKind.Linux.value]['x86_64']['drivers']
|
|
for d in drivers:
|
|
d['nvenc_patch'] = True
|
|
d['nvfbc_patch'] = False
|
|
if 'driver_url' in d:
|
|
driver_url = d['driver_url']
|
|
del d['driver_url']
|
|
d['driver_url'] = driver_url
|
|
with open(DATAFILE_PATH, 'w') as data_file:
|
|
json.dump(data, data_file, indent=4)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|