46 lines
1.3 KiB
Python
Executable File
46 lines
1.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
|
|
EXCLUDE_LIST = ('AFLplusplus', 'target')
|
|
|
|
old_ver = sys.argv[1]
|
|
new_ver = sys.argv[2]
|
|
|
|
result = subprocess.run("git config --file .gitmodules --get-regexp path | awk '{ print $2 }'", shell=True, stdout=subprocess.PIPE)
|
|
submodules = filter(lambda x: len(x) > 0, result.stdout.decode('utf-8').split('\n'))
|
|
|
|
for subdir, dirs, files in os.walk(os.getcwd()):
|
|
exclude = False
|
|
for word in EXCLUDE_LIST:
|
|
if word in subdir.split(os.sep):
|
|
exclude = True
|
|
break
|
|
for sub in submodules:
|
|
if subdir.startswith(sub):
|
|
exclude = True
|
|
break
|
|
if exclude:
|
|
continue
|
|
|
|
for file in files:
|
|
if file != 'Cargo.toml':
|
|
continue
|
|
fname = os.path.join(subdir, file)
|
|
print(fname)
|
|
|
|
with open(fname, 'r') as f:
|
|
toml = f.read()
|
|
lines = toml.split('\n')
|
|
|
|
for i in range(len(lines)):
|
|
if lines[i].startswith('version = "%s"' % old_ver):
|
|
lines[i] = 'version = "%s"' % new_ver
|
|
if lines[i].startswith('libafl') and 'version = "%s"' % old_ver in lines[i]:
|
|
lines[i] = lines[i].replace('version = "%s"' % old_ver, 'version = "%s"' % new_ver)
|
|
|
|
with open(fname, 'w') as f:
|
|
f.write('\n'.join(lines))
|