Mirrored_Repos/nvidia-patch
1
0
Fork 0
mirror of https://github.com/keylase/nvidia-patch.git synced 2024-11-26 15:37:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

ap: ensure only one occurence

Browse Source
This commit is contained in:
Vladislav Yarmak 2019-10-22 19:00:25 +03:00
parent 0ea7227035
commit 8eb2b50a9a
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -59,6 +59,9 @@ class ExtractException(Exception):
class PatternNotFoundException(Exception): class PatternNotFoundException(Exception):
pass pass
class MultipleOccurencesException(Exception):
pass
class UnknownPlatformException(Exception): class UnknownPlatformException(Exception):
pass pass
@ -130,9 +133,11 @@ def make_patch(archive, *,
sevenzip=sevenzip) as tgt: sevenzip=sevenzip) as tgt:
f = expand(tgt, sevenzip=sevenzip) f = expand(tgt, sevenzip=sevenzip)
offset = f.find(search) offset = f.find(search)
del f
if offset == -1: if offset == -1:
raise PatternNotFoundException("Pattern not found.") raise PatternNotFoundException("Pattern not found.")
if f[offset+len(search):].find(search) != -1:
raise MultipleOccurencesException("Multiple occurences of pattern found!")
del f
print("Pattern found @ %016X" % (offset,), file=sys.stderr) print("Pattern found @ %016X" % (offset,), file=sys.stderr)
res = [] res = []