File: //opt/saltstack/salt/lib/python3.10/site-packages/relenv/__pycache__/relocate.cpython-310.pyc
o
�N�g. � @ s� d Z ddlZddlZddlZddlZddlZe�e�Zg d�Z dZ
dZ dZ dd� Z
d d
� Zd d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zd#dd�Zd$dd�Z d%d d �Zed krmddlZeed!�sfed"��eej� dS dS )&zO
A script to ensure the proper rpaths are in place for the relenv environment.
� N)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_RPATHc 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��path�fp�magic� r �C/opt/saltstack/salt/lib/python3.10/site-packages/relenv/relocate.py�is_macho2 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 r r r r �is_elfB s
��r c C s� d}d}d}i }dd� | � d�D �D ]U}|sq|� � d dkrId}|r6||vr+g ||< || �|� d}d}|� � d ttd fv rI|� � d }d
}|rg|� � d d krY|� � d }|� � d d
krg|� � d }q|ry||vrrg ||< || �|� |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�appendr r )�stdoutZin_cmdr r �data�liner r r �
parse_otool_lQ s:
�r$ 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 �:r �[�])�
splitlines�findr r )r! r# r r r �parse_readelf_d~ s
*r+ c 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
Zotoolz-l��stderrr! zis not an object filer N)�
subprocess�run�PIPEr! �decoder* r$ )r �procr! r r r � parse_macho� s
�
r3 c C s( t jdd| gt jt jd�}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
Zreadelfz-d)r! r- )r. r/ r0 r+ r! r1 )r r2 r r r � parse_rpath� s
�r4 c C s t | �}t�d| |� t|v r�|t D ]q}| �d�r!t�d| � qtj�|�r�t� |��
� tj� |� }tj�|�sW|rDt� d||� qt
�||� t
�||� t�d||� t�d|| � t� d�tj�|t� | ��
� j� }dd ||| 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 %r�@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)r3 �log�infor �
startswith�osr �exists�pathlib�Path�resolve�basename�warning�shutil�copy�copymode�relpath�parentr. r/ )r �root_dir�
rpath_only�objr �y�zr r r r � handle_macho� s2
�
�rK c 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 �realpathr9 �sep)�filepath� directoryr r r � is_in_dir� s rP Tc C sv t | �}|r
dd� |D �}||vr6d�|g| �}t�d|| � tjddd|| gtjtjd�}|jr4d S |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)r9 )r Zrpathr r r r � s z patch_rpath.<locals>.<listcomp>r& zSet RPATH=%s %sZpatchelfz
--force-rpathz --set-rpathr, F)r4 �joinr7 r8 r. r/ r0 �
returncode)r Z new_rpathZ
only_relativeZ old_rpathZ
patched_rpathr2 r r r � patch_rpath� s �
rT c 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�d| � 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
NZlddr, Fz=>r zSkip ldd output line: %sc S r r r )r �_r r r r # r z handle_elf.<locals>.<listcomp>r 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: %sr6 z
Copy %s to %s�.rQ zAdjust rpath of %s to %szDo not adjust rpath of %s) r. r/ r0 r! r1 r) r* r7 �debugr r@ �rsplitr r: r r? �LIBCLIBSrP r8 rR r; rA rB rC rD �dirname�strr< r= rT )r ZlibsrG �rootr2 Z needs_rpathr# Zlib_nameZ
location_infoZ
linked_libZ lib_basenameZrelocated_pathZrelpartrD r r r �
handle_elf
sL
r^ �DEBUG�<stdout>c C s, |dkr
|dd�}ni }t jd
t �|�� �dd�|�� tt�| ��� �}|du r/t�|d�}tt�|��� �}|}i }d}|r�d }t� |�D ]H\} } }
|
D ]@} tj
� | | �} | |v r\qNt �
d
| � t| �r}t �d | � t| ||�}
|
dur||
|| < d}qNt| �r�t �d | � t| |||� qNqG|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` �w)�filename�filemodez%(asctime)s %(message)s)�level�formatN�libTFz Checking %szFound Mach-O %sz Found ELF %sr )�logging� basicConfig� getLevelName�upperr\ r< r= r> r: �walkr rR r7 rX r r8 rK r r^ )r] Zlibs_dirrG Z log_levelZ
log_file_name�kwargsrF Z processed�found�dirs�files�filer rU r r r �mainO sN � �� � �� �rq �__main__�RELENVzNot in a relenv environment)T)N)NTr_ r` ) �__doc__rg r: r< rA r. � getLogger�__name__r7 rZ r r r r r r$ r+ r3 r4 rK rP rT r^ rq �sys�hasattr� RuntimeErrorrs r r r r �<module> s:
-(
#C
�5
�