File: //opt/saltstack/salt/lib/python3.10/site-packages/__pycache__/portend.cpython-310.pyc
o
�N�gN � @ s d Z ddlmZmZ ddlZddlZddlZddlZddlZddl Z ddl
Z
zddl m Z W n
e
y; ddl Z Y nw ddlmZ dd� ZG dd � d e�ZG d
d � d e�ZG d d
� d
e�Zed�fdd�ZeZed�fdd�ZeZdd� ZG dd� de �Z dd� Z e dkr�e � dS dS )z:
A simple library for managing the availability of ports.
� )�print_function�divisionN)�abc)�timingc C s | dkrdS | dv r dS | S )zDReturn the host on which a client can connect to the given listener.z0.0.0.0z 127.0.0.1)z::z::0z ::0.0.0.0z::1� )Z server_hostr r �;/opt/saltstack/salt/lib/python3.10/site-packages/portend.py� client_host s
r c @ s( e Zd Zd dd�Zd
dd�Zdd� ZdS ) �Checker� �?c C s
|| _ d S �N��timeout)�selfr
r r r �__init__' �
zChecker.__init__Nc C sb |du rt |tj�r|dd� \}}t�� dkr t|�}t�||tjtj �}t
t � | j
|�� dS )aJ
Assert that the given addr is free
in that all attempts to connect fail within the timeout
or raise a PortNotFree exception.
>>> free_port = find_available_local_port()
>>> Checker().assert_free('localhost', free_port)
>>> Checker().assert_free('127.0.0.1', free_port)
>>> Checker().assert_free('::1', free_port)
Also accepts an addr tuple
>>> addr = '::1', free_port, 0, 0
>>> Checker().assert_free(addr)
Host might refer to a server bind address like '::', which
should use localhost to perform the check.
>>> Checker().assert_free('::', free_port)
N� �Windows)�
isinstancer �Sequence�platform�systemr �socket� getaddrinfo� AF_UNSPEC� SOCK_STREAM�list� itertools�starmap�_connect)r �host�port�infor r r � assert_free* s �zChecker.assert_freec
C s� t � |||�}|�| j� t�|��! z|�|� W n t jy+ Y W d � d S w W d � n1 s6w Y |d d� \}}d} t| jdi t � ����)Nr z Port {port} is in use on {host}.r )
r �
settimeoutr
�
contextlib�closing�connect�error� PortNotFree�format�locals)
r �af�socktype�proto� canonname�sa�sr r Ztmplr r r r I s �� �zChecker._connect)r
r )�__name__�
__module__� __qualname__r r" r r r r r r & s
r c @ � e Zd ZdS )�TimeoutN�r1 r2 r3 r r r r r5 Z � r5 c @ r4 )r( Nr6 r r r r r( ^ r7 r( ZInfc C sr | st d��t�|�}|�� s.z tdd��| |� W dS ty) t�d� Y nw |�� rt dj
di t � ����)uG
Wait for the specified port to become free (dropping or rejecting
requests). Return when the port is free or raise a Timeout if timeout has
elapsed.
Timeout may be specified in seconds or as a timedelta.
If timeout is None or ∞, the routine will run indefinitely.
>>> free('localhost', find_available_local_port())
�*Host values of '' or None are not allowed.皙�����?r Nz Port {port} not free on {host}.r ) �
ValueErrorr �Timer�expiredr r" r( �time�sleepr5 r) r* �r r r
Ztimerr r r �freeb s
�� r@ c C sr | st d��t�|�}|�� s.ztdd��| |� t�d� W n
ty) Y dS w |�� rt dj
di t � ����)u�
Wait for the specified port to become occupied (accepting requests).
Return when the port is occupied or raise a Timeout if timeout has
elapsed.
Timeout may be specified in seconds or as a timedelta.
If timeout is None or ∞, the routine will run indefinitely.
>>> occupied('localhost', find_available_local_port(), .1)
Traceback (most recent call last):
...
Timeout: Port ... not bound on localhost.
r8 g �?r r9 Nz Port {port} not bound on {host}.r ) r: r r; r< r r" r= r> r( r5 r) r* r? r r r �occupied� s
�� rA c C s` t �ddt jt j�} tt| ��\}}}}}t � ||�}|�|� |�� dd� \}}|�� |S )zV
Find a free port on localhost.
>>> 0 < find_available_local_port() < 65536
True
Nr r ) r r r r �next�iter�bind� getsockname�close)�infos�familyr- �_�addr�sockr r r r �find_available_local_port� s
rL c @ s( e Zd ZdZedd� �Zedd� �ZdS )�HostPortz�
A simple representation of a host/port pair as a string
>>> hp = HostPort('localhost:32768')
>>> hp.host
'localhost'
>>> hp.port
32768
>>> len(hp)
15
c C s | � d�\}}}|S �N�:)� partition�r r �sepr r r r r � s z
HostPort.hostc C s | � d�\}}}t|�S rN )rP �intrQ r r r r � s z
HostPort.portN)r1 r2 r3 �__doc__�propertyr r r r r r rM � s
rM c
C s� t �� } dd� }| jddtd� | jdd|d� | jdd d td
� | �� }z|j|jj|jj |j
d � W d S t yL } z t |t
jd � td
��d }~ww )Nc S s
t � | S r )�globals)�keyr r r �
global_lookup� r z _main.<locals>.global_lookup�targetz host:port)�metavar�type�func�statez-tz --timeout)�defaultr[ r )�file� )�argparse�ArgumentParser� add_argumentrM �float�
parse_argsr\ rY r r r
r5 �print�sys�stderr�
SystemExit)�parserrX �argsr
r r r �_main� s ��rl �__main__) rT Z
__future__r r r= r ra rg r r$ r � collectionsr � ImportErrorZtemporar r �objectr �IOErrorr5 r( rd r@ Zwait_for_free_portrA Zwait_for_occupied_portrL �strrM rl r1 r r r r �<module> s: �
4
�