github/nvidia-patch
1
0
Fork 0
mirror of https://github.com/keylase/nvidia-patch.git synced 2025-08-06 14:53:53 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

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

@ -59,6 +59,9 @@ class ExtractException(Exception):
class PatternNotFoundException(Exception):
pass
class MultipleOccurencesException(Exception):
pass
class UnknownPlatformException(Exception):
pass
@ -130,9 +133,11 @@ def make_patch(archive, *,
sevenzip=sevenzip) as tgt:
f = expand(tgt, sevenzip=sevenzip)
offset = f.find(search)
del f
if offset == -1:
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)
res = []