B
    Ka              	   @   sx  d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl	m
Z
mZmZmZmZ e
e eeedf dddZe
e ee dd	d
ZG dd deZeee dddZe eee dddZeee dddZedkrtddlZe Zedstdede edeej  eddd x4ee!dde"d d!d" D ]Z#ee#d#d q^W dS )$zPEP 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returnc             C   s   t || t |S )N)structunpackreadcalcsize)r   r    r   3lib/python3.7/site-packages/packaging/_musllinux.py_read_unpacked   s    r   )r   r	   c          	   C   sd  |  d yt| d}W n tjk
r.   dS X |dd tdkrHdS |  tdd ydd	d
|d  \}}}W n tk
r   dS X tj| }yt| |\}}}}}}}	W n tjk
r   dS X xt	|	d D ]}
|  |||
   y|t| |\}}}W n tjk
r   dS X |dkr,q|  | t
| |d}d|krZdS |S W dS )zDetect 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   Z16BN   s   ELFZHHI   )ZIIIIHHHZIIIIIIII)r   r   r   )ZQQQIHHHZIIQQQQQQ)r         )r   r       musl)seekr   r
   errortupler   KeyErroroperator
itemgetterrangeosfsdecoder   strip)r   ZidentZe_fmtZp_fmtZp_idxZp_get_Ze_phoffZe_phentsizeZe_phnumiZp_typeZp_offsetZp_fileszZinterpreterr   r   r   _parse_ld_musl_from_elf   sB    




r$   c               @   s   e Zd ZU eed< eed< dS )_MuslVersionmajorminorN)__name__
__module____qualname__int__annotations__r   r   r   r   r%   G   s   
r%   )outputr	   c             C   sx   dd dd |   D D }t|dk s<|d d d dkr@d S td	|d
 }|sXd S tt|d
t|ddS )Nc             S   s   g | ]}|r|qS r   r   ).0nr   r   r   
<listcomp>M   s    z'_parse_musl_version.<locals>.<listcomp>c             s   s   | ]}|  V  qd S )N)r!   )r.   r/   r   r   r   	<genexpr>M   s    z&_parse_musl_version.<locals>.<genexpr>r   r   r   r   zVersion (\d+)\.(\d+)r   )r&   r'   )
splitlineslenrematchr%   r+   group)r-   linesmr   r   r   _parse_musl_versionL   s     r9   )
executabler	   c          	   C   sl   t  8}y|t| d}W n tk
r2   dS X t|}W dQ R X |sNdS tj|gtjdd}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)stderrZuniversal_newlines)
contextlib	ExitStackenter_contextopenOSErrorr$   
subprocessrunPIPEr9   r<   )r:   stackr   Zldprocr   r   r   _get_musl_versionV   s    
rG   )archr	   c             c   sL   t tj}|dkrdS x0t|jddD ]}d|j d| d|  V  q&W dS )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.
    NZ
musllinux_r"   )rG   sysr:   r   r'   r&   )rH   Zsys_muslr'   r   r   r   platform_tagsn   s
    	
rK   __main__zlinux-z	not linuxzplat:zmusl:ztags: )endz[.-]r"   -r   rI   z
      )$__doc__r=   	functoolsr   r   r4   r
   rB   rJ   typingr   r   r   r   r   bytesstrr+   r   r$   r%   r9   	lru_cacherG   rK   r(   Z	sysconfigZget_platformZplat
startswithAssertionErrorprintr:   subsplittr   r   r   r   <module>   s2   1


$