File: //opt/saltstack/salt/lib/python3.10/site-packages/relenv/__pycache__/relocate.cpython-310.pyc
o
;j@ � @ sb U d Z ddlmZ ddlZddlZddlZddlZddl Z
ddl Z e�
e�ZeZeZe
Z e Z g d�Zg d�ZdZdZdZdad ed
< dad ed < dEdd�ZdEdd�ZdFdd�ZdFdd�ZdGdd �Z dHd d �Z dId!d"�Z dJd#d$�Z dKd'd(�Z dLd+d,�Z!dFd-d.�Z" /dMdNd3d4�Z# dOdPd9d:�Z$ / ; <dQdRd@dA�Z%edBkr�ddl Z e&e dC�s�e'dD��e%e j(� dS dS )SzO
A script to ensure the proper rpaths are in place for the relenv environment.
� )� annotationsN)�is_macho�is_elf�
parse_otool_l�parse_readelf_d� parse_macho� parse_rpath� handle_macho� is_in_dir�
handle_elf� patch_rpath�main�os�shutil�
subprocess)zlinux-vdso.so.1z libc.so.6�
librt.so.1z libm.so.6z
libmd.so.0zlibpthread.so.0z
libdl.so.2zlibmemusage.soz libnsl.so.1zlibnss_compat.so.2zlibnss_db.so.2zlibnss_dns.so.2zlibnss_files.so.2zlibnss_hesiod.so.2zlibpcprofile.so.2zlibresolv.so.2r zlibthread_db.so.1z libutil.so.1z libutil.so.2z
libgcc_s.so.2z
libgcc_s.so.1� LC_ID_DYLIB�
LC_LOAD_DYLIB�LC_RPATHz
str | None�_READELF_BINARY�_PATCHELF_BINARY�return�strc C s� t durt S tjdkrFz.ddlm} m} | � }|r0|� }|d |� d� }|�� r5t|�a t W S W da t S W da t S tyE Y da t S w da t S )a9
Get the path to readelf binary, preferring toolchain version.
Returns the cached value if already computed. On Linux, prefers the
toolchain's readelf over the system version. Falls back to "readelf"
from PATH if toolchain is unavailable.
:return: Path to readelf binary
:rtype: str
N�linuxr )�
get_toolchain� get_triplet�binz-readelf�readelf) r �sys�platform�
relenv.commonr r �existsr � Exception)r r � toolchain�tripletZtoolchain_readelf� r% �C/opt/saltstack/salt/lib/python3.10/site-packages/relenv/relocate.py�_get_readelf_binaryQ s2
� � ��r' c C s� t durt S tjdkr>z&ddlm} | � }|r(|d d }|�� r-t|�a t W S W da t S W da t S ty= Y da t S w da t S )a=
Get the path to patchelf binary, preferring toolchain version.
Returns the cached value if already computed. On Linux, prefers the
toolchain's patchelf over the system version. Falls back to "patchelf"
from PATH if toolchain is unavailable.
:return: Path to patchelf binary
:rtype: str
Nr r )r r �patchelf)r r r r r r! r r" )r r# Ztoolchain_patchelfr% r% r& �_get_patchelf_binaryu s0
�
� ��r) �path�str | os.PathLike[str]�boolc C sB t | d��}|�d�}W d � |dv S 1 sw Y |dv S )z�
Determines whether the given file is a macho file.
:param path: The path to the file to check
:type path: str
:return: Whether the file is a macho file
:rtype: bool
�rb� N)s ���open�read�r* �fp�magicr% r% r& r � s
��r c C sB t | d��}|�d�}W d � |dkS 1 sw Y |dkS )z�
Determines whether the given file is an ELF file.
:param path: The path to the file to check
:type path: str
:return: Whether the file is an ELF file
:rtype: bool
r- r. Ns ELFr/ r2 r% r% r& r � s
��r �stdout�dict[str, list[str]]c C s d}d}d}i }dd� | � d�D �D ][}|sq|� � d dkrGd}|dur6|dur6|�|g ��|� d}d}|� � d }|ttd fv rG|}d
}|rm|� � }|d d kr]t|�d kr]|d }|d d
krmt|�d krm|d }q|r�|dur�|dur�|�|g ��|� |S )z�
Parse the output of ``otool -l <path>``.
:param stdout: The output of the ``otool -l <path>`` command
:type stdout: str
:return: The parsed relevant output with command keys and path values
:rtype: dict
FNc S � g | ]}|� � �qS r% ��strip)�.0�xr% r% r& �
<listcomp>� � z!parse_otool_l.<locals>.<listcomp>�
r �cmd���r T�name� r* )�split�
setdefault�appendr r �len)r5 Zin_cmdr? rA �data�lineZcommand�partsr% r% r& r � s6
�r � list[str]c C sJ | � � D ] }|�d�dkrq|�dd�d �� �d��d��d� S g S )z�
Parse the output of ``readelf -d <path>``.
:param stdout: The output of the ``readelf -d <path>`` command
:type stdout: str
:return: The RPATH values
:rtype: list
�PATHr@ �:rB �[�])�
splitlines�findrC r9 )r5 rH r% r% r& r � s
*r �dict[str, list[str]] | Nonec C s> t jdd| gt jt jd�}|j�� }|�d�dkrdS t|�S )z�
Run ``otool -l <path>`` and return its parsed output.
:param path: The path to the file
:type path: str
:return: The parsed relevant RPATH content, or None if it isn't an object file
:rtype: dict or None
�otoolz-l��stderrr5 zis not an object filer@ N)r �run�PIPEr5 �decoderP r )r* �procr5 r% r% r& r � s
r c C s. t � }tj|d| gtjtjd�}t|j�� �S )z�
Run ``readelf -d <path>`` and return its parsed output.
:param path: The path to the file
:type path: str
:return: The RPATH's found.
:rtype: list
z-d)r5 rT )r' r rU rV r r5 rW )r* r rX r% r% r& r s
r �root_dir�
rpath_onlyc C s* t �| �}t|�}t|�}t�d||� |sdS t|v r�|t D ]r}|�d�r.t�d|� q tj �
|�r�t �|�� � }|tj � |� }tj �
|�sf|rSt�
d||� q t�||� t�||� t�d||� t�d||� t �d�tj �||� � j� } d d
|t| �|g}
t�|
� t�d || |� q |S ) a�
Ensure the given macho file has the correct rpath and is in th correct location.
:param path: The path to a macho file
:type path: str
:param root_dir: The directory the file needs to reside under
:type root_dir: str
:param rpath_only: If true, only ensure the correct rpaths are present and don't copy the file
:type rpath_only: bool
:return: The information from ``parse_macho`` on the macho file.
zProcessing file %s %rN�@zSkipping dynamic load: %s�(In `rpath_only mode` but %s is not in %szCopied %s to %sz Use %s to %sz @loader_pathZinstall_name_toolz-changezChanged %s to %s in %s)�pathlib�Pathr r �log�infor �
startswithr r* r! �resolve�basename�warningr �copy�copymode�relpath�parentr rU ) r* rY rZ Zpath_obj�objZpath_strr; �
target_dir�y�zr? r% r% r& r s8
�r �filepath� directoryc C s t j�| ��t j�|�t j �S )a5
Determines whether a file is contained within a directory.
:param filepath: The path to the file to check
:type filepath: str
:param directory: The directory to check within
:type directory: str
:return: Whether the file is contained within the given directory
:rtype: bool
)r r* �realpathra �sep)rm rn r% r% r& r
B s r
c C s` t | �}|sdS t�d| |� t� }tj|d| gtjtjd�}|jr.t�d| |j �
� � dS dS )z�
Remove the rpath from a given ELF file.
:param path: The path to an ELF file
:type path: str
:return: True if successful, False otherwise
:rtype: bool
Tz Remove RPATH from %s (was: %s)z--remove-rpathrS z"Failed to remove RPATH from %s: %sF) r r_ r` r) r rU rV �
returncode�errorrT rW )r* � old_rpathr( rX r% r% r& � remove_rpathQ s
�rt T� new_rpath�
only_relative�
str | boolc C s| t | �}|r
dd� |D �}||vr9d�|g| �}t�d|| � t� }tj|dd|| gtjtjd�}|jr7dS |S d�|�S ) a�
Patch the rpath of a given ELF file.
:param path: The path to an ELF file
:type path: str
:param new_rpath: The new rpath to add
:type new_rpath: str
:param only_relative: Whether or not to remove non-relative rpaths, defaults to True
:type only_relative: bool, optional
:return: The new rpath, or False if patching failed
:rtype: str or bool
c S s g | ] }|� d �r|�qS )�$ORIGIN)ra )r: Zrpathr% r% r& r< � s z patch_rpath.<locals>.<listcomp>rL zSet RPATH=%s %sz
--force-rpathz --set-rpathrS F) r �joinr_ r` r) r rU rV rq )r* ru rv rs Z
patched_rpathr( rX r% r% r& r n s �
r �libs�root� str | os.PathLike[str] | None�Nonec C s� |du r|}t jd| gt jt jd�}d}|j�� �� D ]�}|�d�dkr+t�d|� qdd � |� dd
�D �\}}|d krDt�
d || � q|� d
d
�d � � } t
j�| �}
|tv r_t�d|� qt| |�rmd}t�d| � qt
j�||
�} t
j�| �r�t�d| � q|r�t�
d| |� qt�d| | � t�| | � t�| | � d}q|r�t
j�|t
j�| ��} | dkr�d}
n tt� d�| �}
t�d| |
� t | |
� dS t | � dS )a�
Handle the parsing and pathcing of an ELF file.
:param path: The path of the ELF file
:type path: str
:param libs: The libs directory
:type libs: str
:param rpath_only: If true, only ensure the correct rpaths are present and don't copy the file
:type rpath_only: bool
:param root: The directory to ensure the file is under, defaults to None
:type root: str, optional
NZlddrS Fz=>r@ zSkip ldd output line: %sc S r7 r% r8 )r: �_r% r% r& r< � r= z handle_elf.<locals>.<listcomp>rB z not foundz(Unable to find library %s linked from %s� r zSkipping glibc lib %sTz&File already within root directory: %sz Relocated library exists: %sr\ z
Copy %s to %s�.rx zAdjust rpath of %s to %s) r rU rV r5 rW rO rP r_ �debugrC rd �rsplitr9 r r* rc �LIBCLIBSr
r` ry r! r re rf rg �dirnamer r] r^ r rt )r* rz rZ r{ rX Z needs_rpathrH Zlib_nameZ
location_infoZ
linked_libZ lib_basenameZrelocated_pathZrelpartrg r% r% r& r � sL
r �DEBUG�<stdout>�libs_dir� log_level�
log_file_namec C s( t �|�� �}|dkrt j|d|dd� nt j|dd� tt�| ��� �}|du r/t�|d�}tt�|��� �}i }d}|r�d }t� |�D ]H\} } }
|
D ]@} tj
� | | �} | |v rZqLt �
d
| � t| �r{t �d | � t| ||�}
|
durz|
|| < d}qLt| �r�t �d | � t| |||� qLqE|s>dS dS )
a
The entrypoint into the relocate script.
:param root: The root directory to operate traverse for files to be patched
:type root: str
:param libs_dir: The directory to place the libraries in, defaults to None
:type libs_dir: str, optional
:param rpath_only: If true, only ensure the correct rpaths are present and don't copy the file, defaults to True
:type rpath_only: bool, optional
:param log_level: The level to log at, defaults to "INFO"
:type log_level: str, optional
r� z%(asctime)s %(message)s�w)�level�format�filename�filemode)r� r� N�libTFz Checking %szFound Mach-O %sz Found ELF %s)�logging� getLevelName�upper� basicConfigr r] r^ rb r �walkr* ry r_ r� r r` r r r )r{ r� rZ r� r� r� rY Z processed�found�dirs�files�filer* r~ r% r% r& r
� sN �� � �� �r
�__main__�RELENVzNot in a relenv environment)r r )r* r+ r r, )r5 r r r6 )r5 r r rJ )r* r+ r rQ )r* r+ r rJ )r* r+ rY r+ rZ r, r rQ )rm r+ rn r+ r r, )T)r* r+ ru r rv r, r rw )N)
r* r+ rz r+ rZ r, r{ r| r r} )NTr� r� ) r{ r+ r� r| rZ r, r� r r� r r r} ))�__doc__�
__future__r r� r �_osr] r �_shutilr � _subprocessr �_sys� getLogger�__name__r_ �__all__r� r r r r �__annotations__r r' r) r r r r r r r r
rt r r r
�hasattr� RuntimeErrorr� r% r% r% r& �<module> s\
$
#
*
/
�, �L �:
�