CasperSecurity
o
{ۓa � @ s~ d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
m Z m Z m
Z
mZ de
e dedeedf fdd �Zde
e de
e fd
d �ZG d d
� d
e �Zdede
e fdd�Ze�� dede
e fdd��Zdede e fdd�Zedkr�ddlZe�� Z e � d�s�J d��e de � e deej �� e d d d � ee� d d e �!d!d"�d# ��D ]
Z"e e"d$d � q�dS dS )%z�PEP 656 support.
This module implements logic to detect if the currently running Python is
linked against musl, and what musl version is used.
� N)�IO�Iterator�
NamedTuple�Optional�Tuple�f�fmt�return.c C s t �|| �t �|���S �N)�struct�unpack�read�calcsize)r r � r �6/usr/lib/python3/dist-packages/packaging/_musllinux.py�_read_unpacked s r c C sb | � d� zt| d�}W n tjy Y dS w |dd� td�kr$dS | � t�d�d� zdd d
�|d \}}}W n
tyE Y dS w tj|� }zt| |�\}}}}}}} W n tjyd Y dS w t | d �D ]C}
| � |||
� z |t| |��\} } }
W n tjy� Y dS w | d kr�qk| � | � t
� | � |
���
d �}d
|vr� dS | S dS )z�Detect musl libc location by parsing the Python executable.
Based on: https://gist.github.com/lyssdod/f51579ae8d93c8657a5564aefc2ffbca
ELF header: https://refspecs.linuxfoundation.org/elf/gabi4+/ch4.eheader.html
r �16BN� s ELF�HHI� )�IIIIHHH�IIIIIIII)r r r )�QQQIHHH�IIQQQQQQ)r � � )r r � � �musl)�seekr r �error�tupler �KeyError�operator�
itemgetter�range�os�fsdecoder
�strip)r �ident�e_fmt�p_fmt�p_idx�p_get�_�e_phoff� e_phentsize�e_phnum�i�p_type�p_offset�p_filesz� interpreterr r r �_parse_ld_musl_from_elf sN
��� �
��
r7 c @ s e Zd ZU eed< eed< dS )� _MuslVersion�major�minorN)�__name__�
__module__� __qualname__�int�__annotations__r r r r r8 G s
r8 �outputc C sx dd� dd� | � � D �D �}t|�dk s |d d d� dkr d S t�d |d
�}|s,d S tt|�d
��t|�d��d �S ) Nc S s g | ]}|r|�qS r r ��.0�nr r r �
<listcomp>M s z'_parse_musl_version.<locals>.<listcomp>c s s � | ]}|� � V qd S r
)r( rA r r r � <genexpr>M s � z&_parse_musl_version.<locals>.<genexpr>r r r r zVersion (\d+)\.(\d+)r )r9 r: )�
splitlines�len�re�matchr8 r> �group)r@ �lines�mr r r �_parse_musl_versionL s rM �
executablec C s� t �� �'}z
|�t| d��}W n ty Y W d � dS w t|�}W d � n1 s.w Y |s7dS tj|gtjdd�}t |j
�S )a` Detect currently-running musl runtime version.
This is done by checking the specified executable's dynamic linking
information, and invoking the loader to parse its output for a version
string. If the loader is musl, the output would be something like::
musl libc (x86_64)
Version 1.2.2
Dynamic Program Loader
�rbNT)�stderr�universal_newlines) �
contextlib� ExitStack�
enter_context�open�OSErrorr7 �
subprocess�run�PIPErM rP )rN �stackr �ld�procr r r �_get_musl_versionV s
�
�
r] �archc c sJ � t tj�}|du r dS t|jdd�D ]}d|j� d|� d| � �V qdS )aT Generate musllinux tags compatible to the current platform.
:param arch: Should be the part of platform tag after the ``linux_``
prefix, e.g. ``x86_64``. The ``linux_`` prefix is assumed as a
prerequisite for the current platform to be musllinux-compatible.
:returns: An iterator of compatible musllinux tags.
N����
musllinux_r. )r] �sysrN r% r: r9 )r^ �sys_muslr: r r r �
platform_tagsn s �
�rc �__main__zlinux-z not linuxzplat:zmusl:ztags:� )�endz[.-]r. �-r r_ z
)#�__doc__rR � functoolsr# r& rH r rW ra �typingr r r r r �bytes�strr> r r7 r8 rM � lru_cacher] rc r; � sysconfig� get_platform�plat�
startswith�printrN �sub�split�tr r r r �<module> s8 "1
"�