Nirmal Unnikrishnan
128 lines
5.5 KiB
Python
128 lines
5.5 KiB
Python
#!/usr/bin/env python3
|
|
################################################################################
|
|
##
|
|
## MIT License
|
|
##
|
|
## Copyright (c) 2017 - 2023 Advanced Micro Devices, Inc. All rights Reserved.
|
|
##
|
|
## Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
## of this software and associated documentation files (the "Software"), to deal
|
|
## in the Software without restriction, including without limitation the rights
|
|
## to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
## copies of the Software, and to permit persons to whom the Software is
|
|
## furnished to do so, subject to the following conditions:
|
|
##
|
|
## The above copyright notice and this permission notice shall be included in all
|
|
## copies or substantial portions of the Software.
|
|
##
|
|
## THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
## IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
## FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
## AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
## LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
## OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
## SOFTWARE.
|
|
##
|
|
################################################################################
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
import argparse
|
|
import pathlib
|
|
|
|
try:
|
|
from elftools.elf.elffile import ELFFile
|
|
from elftools.elf.dynamic import DynamicSection
|
|
from elftools.common.exceptions import ELFError
|
|
except ImportError:
|
|
print("Error : pyelftools failed to import.\n"
|
|
"Run \'pip3 install pyelftools\' to install the prerequisite\n")
|
|
|
|
# Function remove the runpath from the binary
|
|
def remove_runpath(filename) :
|
|
''' Remove the DT_RUNPATH tag from filename'''
|
|
REMOVE_RPATH_COMMAND = ['patchelf', '--remove-rpath', filename]
|
|
print("Executing the command : ", REMOVE_RPATH_COMMAND)
|
|
subprocess.run(REMOVE_RPATH_COMMAND, check=True)
|
|
|
|
def add_rpath(filename, runpath) :
|
|
''' Use runpath and add it as DT_PATH tag in filename'''
|
|
ADD_RPATH_COMMAND = ['patchelf', '--force-rpath', '--set-rpath', runpath, filename]
|
|
print("Executing the command : ", ADD_RPATH_COMMAND)
|
|
subprocess.run(ADD_RPATH_COMMAND, check=True)
|
|
|
|
def update_rpath(search_path, excludes) :
|
|
''' Function helps to change DT_RUNPATH in libraries and binaries in search_path to DT_RPATH.
|
|
Its done with the following steps :
|
|
1. Check all if the file is an ELF except in excludes folder
|
|
2. Read and store the DT_RUNPATH from the binaries/libraries.
|
|
3. Delete DT_RUNPATH and readd it as DT_RPATH '''
|
|
for path, dirs, files in os.walk(search_path, topdown=True, followlinks=True):
|
|
dirs[:] = [d for d in dirs if d not in excludes]
|
|
print( dirs )
|
|
for filename in files:
|
|
filename = os.path.join(path, filename)
|
|
print("Opening file ", filename)
|
|
# Open the file and check if its ELF file
|
|
try :
|
|
with open(filename, 'rb') as file:
|
|
elffile = ELFFile(file)
|
|
for section in elffile.iter_sections():
|
|
if not isinstance(section, DynamicSection):
|
|
continue
|
|
for tag in section.iter_tags():
|
|
if tag.entry.d_tag == 'DT_RUNPATH':
|
|
runpath = tag.runpath
|
|
print(runpath)
|
|
remove_runpath(filename)
|
|
add_rpath(filename, runpath)
|
|
break
|
|
break
|
|
except ELFError:
|
|
print("Discarding file as its not an ELF file", filename)
|
|
continue
|
|
except FileNotFoundError:
|
|
print("Discarding file with bad links", filename)
|
|
continue
|
|
except OSError:
|
|
print("Discarding file with OS error", filename)
|
|
continue
|
|
except Exception as ex:
|
|
print("Discarding file ", filename, ex)
|
|
continue
|
|
|
|
def main():
|
|
# The script expect a search folder as parameter. It finds all ELF files and updates RPATH
|
|
argparser = argparse.ArgumentParser(
|
|
usage='usage: %(prog)s <folder-to-search>',
|
|
description='Find the ELF files in the specified folder and convert the RUNPATH to RPATH. \n',
|
|
add_help=False,
|
|
prog='runpath_to_rpath.py')
|
|
|
|
argparser.add_argument('searchdir',
|
|
nargs='?', type=pathlib.Path, default=None,
|
|
help='Folder to search for ELF file. \nPlease note: Any folder with name llvm in that path will be discarded')
|
|
argparser.add_argument('-h', '--help',
|
|
action='store_true', dest='help',
|
|
help='Display this information')
|
|
|
|
args = argparser.parse_args()
|
|
if args.help or not args.searchdir:
|
|
argparser.print_help()
|
|
sys.exit(0)
|
|
|
|
# pyelftools is a mandatory requirement for this script. Exit if requirement is not met
|
|
if 'ELFFile' not in globals():
|
|
print('Please install pyelftools using \'pip3 install pyelftools\' ' +
|
|
'before using the script : runpath_to_rpath.py')
|
|
sys.exit(0)
|
|
|
|
# Find the elf files in the serach path and update RPATH
|
|
excludes = ['llvm', 'rocm-llvm', 'llvm-alt']
|
|
update_rpath(args.searchdir, excludes)
|
|
print("Done with rpath update")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|