numam-dpdk/devtools/update-patches.py

54 lines
1.6 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
# SPDX-License-Identifier: BSD-3-Clause
# Copyright(c) 2021 Intel Corporation
import os
import sys
import shutil
from os.path import abspath, dirname, join
def fixup_library_renames(contents):
"""since library directory names have dropped the 'librte_' prefix,
add those prefixes back in for patches than need it"""
modified = False
# first get all the DPDK libs to build up replacement list
# stored in function attribute between calls
try:
libdirs = fixup_library_renames.libdirs
except AttributeError:
dpdk_libdir = abspath(join(dirname(sys.argv[0]), '..', 'lib'))
for root, dirs, files in os.walk(dpdk_libdir):
fixup_library_renames.libdirs = dirs
libdirs = dirs
break
for i in range(len(contents)):
# skip over any lines which don't have lib in it
if not "lib/" in contents[i]:
continue
for d in libdirs:
if f'lib/{d}' in contents[i]:
modified = True
contents[i] = contents[i].replace(f'lib/{d}', f'lib/librte_{d}')
return modified
def main():
"takes list of patches off argv and processes each"
for patch in sys.argv[1:]:
modified = False
with open(patch) as f:
contents = f.readlines()
modified |= fixup_library_renames(contents)
# other functions to change the patch go here
if not modified:
continue
shutil.copyfile(f'{patch}', f'{patch}.bak')
with open(patch, 'w') as f:
f.writelines(contents)
if __name__ == "__main__":
main()