B
    2‹d8  ã               @   sX  U d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm	Z	 ddl
mZ g Ze	ed< g Ze	ed	< d
aee	dœdd„Zdd„ Zdd„ Zdd„ ZG dd„ deƒZdd„ Ze dej¡Zdd„ Zdd„ Ze dej¡Zdd„ Zd d!„ Zd"d#„ Zd$d%„ Z ed3d'd(„ƒZ!ed)d*„ ƒZ"ed+d,„ ƒZ#ed-d.„ ƒZ$ed/d0„ ƒZ%ed1d2„ ƒZ&dS )4z-Utilities for identifying local IP addresses.é    N)ÚPIPE)ÚPopen)ÚIterable)ÚList)ÚwarnÚ	LOCAL_IPSÚ
PUBLIC_IPSÚ )ÚelemsÚreturnc             C   s8   t ƒ }g }x(| D ] }||kr| |¡ | |¡ qW |S )z£uniq_stable(elems) -> list

    Return from an iterable, a list of all the unique elements in the input,
    maintaining the order in which they first appear.
    )ÚsetÚappendÚadd)r
   ÚseenÚvalueÚx© r   úk/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/jupyter_client/localinterfaces.pyÚ_uniq_stable   s    

r   c             C   sl   d}t jdkr&t ¡ }| jtjO  _t| tt|d}| ¡ \}}|j	r`t
d| | dd¡f ƒ‚| dd¡S )z4Get output of a command, raising IOError if it failsNÚnt)ÚstdoutÚstderrÚstartupinfozFailed to run %s: %sÚutf8Úreplace)ÚosÚnameÚ
subprocessZSTARTUPINFOZdwFlagsZSTARTF_USESHOWWINDOWr   r   ÚcommunicateÚ
returncodeÚIOErrorÚdecode)Úcmdr   Úpr   r   r   r   r   Ú_get_output#   s    
r$   c                s   dˆ _ ‡ fdd„}|S )z%decorator to only run a function onceFc                 s   ˆ j r
d S ˆ f | Ž}dˆ _ |S )NT)Úcalled)ÚkwargsÚret)Úfr   r   Úwrapped4   s
    
z_only_once.<locals>.wrapped)r%   )r(   r)   r   )r(   r   Ú
_only_once0   s    r*   c                s   ‡ fdd„}|S )z2decorator to ensure load_ips has been run before fc                 s   t ƒ  ˆ | |ŽS )N)Ú	_load_ips)Úargsr&   )r(   r   r   Ú
ips_loadedA   s    z!_requires_ips.<locals>.ips_loadedr   )r(   r-   r   )r(   r   Ú_requires_ips>   s    r.   c               @   s   e Zd ZdS )ÚNoIPAddressesN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r/   I   s   r/   c             C   s”   | st ‚g }g }x4| D ],}| |¡ | d¡s:| |¡ qts|aqW trRtdkrbda| dt¡ | ddg¡ t|ƒtdd…< t|ƒtdd…< dS )z7populate local and public IPs from flat list of all IPsz127.z	127.0.0.1r   z0.0.0.0r	   N)	r/   r   Ú
startswithÚ	LOCALHOSTÚinsertÚextendr   r   r   )ÚaddrsÚ
public_ipsÚ	local_ipsÚipr   r   r   Ú_populate_from_listM   s     


r;   zinet\b.*?(\d+\.\d+\.\d+\.\d+)c              C   sr   yt dƒ} W n tk
r(   t dƒ} Y nX |  ¡ }g }x.|D ]&}t | ¡ ¡}|r<| | d¡¡ q<W t|ƒ dS )z0load ip addresses from `ifconfig` output (posix)Úifconfigz/sbin/ifconfigé   N)	r$   ÚOSErrorÚ
splitlinesÚ_ifconfig_ipv4_patÚmatchÚstripr   Úgroupr;   )ÚoutÚlinesr7   ÚlineÚmr   r   r   Ú_load_ips_ifconfigj   s    
rH   c              C   sr   t ddddgƒ} |  ¡ }g }xH|D ]@}| ¡  ¡ }t|ƒdkr"|d dkr"| |d  d¡d ¡ q"W t|ƒ d	S )
z/load ip addresses from `ip addr` output (Linux)r:   z-fZinetÚaddré   r   r=   ú/N)r$   r?   ÚlowerÚsplitÚlenr   r;   )rD   rE   r7   rF   Úblocksr   r   r   Ú_load_ips_ip|   s    
rP   zipv4.*?(\d+\.\d+\.\d+\.\d+)$c              C   sP   t dƒ} |  ¡ }g }x.|D ]&}t | ¡ ¡}|r| | d¡¡ qW t|ƒ dS )z2load ip addresses from `ipconfig` output (Windows)Úipconfigr=   N)r$   r?   Ú_ipconfig_ipv4_patrA   rB   r   rC   r;   )rD   rE   r7   rF   rG   r   r   r   Ú_load_ips_ipconfigŒ   s    
rS   c              C   sÌ   ddl } g }g }xt|  ¡ D ]h}|  |¡ | jg ¡}xN|D ]F}| d¡}|sLq8| d¡sl| d¡sl| |¡ ntst|a| |¡ q8W qW tsšda| dt¡ | 	ddg¡ t
|ƒtdd…< t
|ƒtdd…< dS )	z load ip addresses with netifacesr   NrI   Úloz127.z	127.0.0.1z0.0.0.0r	   )Ú	netifacesZ
interfacesZifaddressesÚgetÚAF_INETr3   r   r4   r5   r6   r   r   r   )rU   r9   r8   ZifaceZipv4sÚentryrI   r   r   r   Ú_load_ips_netifaces™   s(    

rY   c           
   C   s   yt  d¡d tdd…< W n$ t jk
r>   dgtdd…< Y nX zxy\t  ¡ } t  | ¡d tdd…< |  d¡sœtdd„ tD ƒƒrœt  t  ¡ d ¡d tdd…< W n t jk
r´   Y nX W dttƒtdd…< t 	t¡ X t 	dd	g¡ ttƒtdd…< td
 a
dS )zJload ip addresses with socket.gethostbyname_ex

    This can be slow.
    Ú	localhostrJ   Nz	127.0.0.1z.localc             s   s   | ]}|  d ¡V  qdS )Z127N)r3   )Ú.0r:   r   r   r   ú	<genexpr>Ç   s    z*_load_ips_gethostbyname.<locals>.<genexpr>z0.0.0.0r	   r   )ÚsocketÚgethostbyname_exr   ÚerrorÚgethostnamer   ÚendswithÚallr   r6   r4   )Úhostnamer   r   r   Ú_load_ips_gethostbyname·   s     "
rd   c               C   s&   da t ddgtdd…< g tdd…< dS )z&Fallback in case of unexpected failurez	127.0.0.1z0.0.0.0r	   N)r4   r   r   r   r   r   r   Ú_load_ips_dumb×   s    re   Tc          
   C   sÎ   yŽyt ƒ S  tk
r   Y nX tjdkrJytƒ S  ttfk
rF   Y qŠX n@ytƒ S  ttfk
rh   Y nX ytƒ S  ttfk
rˆ   Y nX t	ƒ S  t
k
rÂ } z| s¦‚ td| ƒ W dd}~X Y nX tƒ  dS )aG  load the IPs that point to this machine

    This function will only ever be called once.

    It will use netifaces to do it quickly if available.
    Then it will fallback on parsing the output of ifconfig / ip addr / ipconfig, as appropriate.
    Finally, it will fallback on socket.gethostbyname_ex, which can be slow.
    r   z9Unexpected error discovering local network interfaces: %sN)rY   ÚImportErrorr   r   rS   r>   r/   rP   rH   rd   Ú	Exceptionr   re   )Zsuppress_exceptionsÚer   r   r   r+   ß   s0    
r+   c               C   s   t S )z2return the IP addresses that point to this machine)r   r   r   r   r   r9     s    r9   c               C   s   t S )zKreturn the IP addresses for this machine that are visible to other machines)r   r   r   r   r   r8     s    r8   c               C   s   t S )z1return ip for localhost (almost always 127.0.0.1))r4   r   r   r   r   rZ     s    rZ   c             C   s   | t kS )z does `ip` point to this machine?)r   )r:   r   r   r   Úis_local_ip  s    ri   c             C   s   | t kS )z#is `ip` a publicly visible address?)r   )r:   r   r   r   Úis_public_ip%  s    rj   )T)'Ú__doc__r   Úrer]   r   r   r   Útypingr   r   Úwarningsr   r   Ú__annotations__r   r4   r   r$   r*   r.   rg   r/   r;   ÚcompileÚ
IGNORECASEr@   rH   rP   rR   rS   rY   rd   re   r+   r9   r8   rZ   ri   rj   r   r   r   r   Ú<module>   sB    -