File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/ssdp.cpython-310.pyc
o
�N�g�: � @ s@ d Z ddlZddlZddlZddlZddlZddlZddlZddl Zzddl
m Z W n e y9 ddl
m Z Y nw ejj�� Zeed�sGdZz ddlZde_W n e yn z ddlZde_W n e yk dZY nw Y nw G dd� de�ZG d d
� d
e�ZG d d � d �ZG d
d� de�ZG dd� de�ZG dd� de�Z dS )zq
Salt Service Discovery Protocol.
JSON-based service discovery protocol, used by minions to find running Master.
� N)� OrderedDict�dumpsFTc @ � e Zd ZdS )�TimeOutExceptionN��__name__�
__module__� __qualname__� r
r
�C/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/ssdp.pyr 7 � r c @ r )�TimeStampExceptionNr r
r
r
r r
; r r
c
@ s^ e Zd ZdZe�e �ZdZdZdZ dZ
dZ ede de
d e d
ei iZ e
d d � �Ze
d
d� �ZdS )�SSDPBasez*
Salt Service Discovery Protocol.
� signature�answer�port� listen_ip�timeoutZ__salt_master_servicei� z0.0.0.0� c C s t tot�S )zW
Return True if the USSDP dependencies are satisfied.
:return:
)�bool�asyncio�_jsonr
r
r
r �
_is_availableV s zSSDPBase._is_availablec C st t � t jt j�} z,z
| �d� | �� d }W n ty& t �t �� �}Y nw W | �� |S W | �� |S | �� w )zB
Find out localhost outside IP.
:return:
)z
1.255.255.255� r ) �socket�AF_INET�
SOCK_DGRAM�connect� getsockname� Exception�
gethostbyname� gethostname�close)ZsckZip_addrr
r
r � get_self_ip^ s
���
�zSSDPBase.get_self_ipN)r r r �__doc__�logging� getLogger�log� SIGNATURE�ANSWER�PORT� LISTEN_IP�TIMEOUT�DEFAULTS� staticmethodr r# r
r
r
r r ? s$
�
r c @ s: e Zd ZdZdd� Zdd� Zdd� Zdd
d �Zd d
� ZdS )� SSDPFactoryz"
Socket protocol factory.
c K sL | j | jfD ]}t| ||�|| j| �� qd| _d| _t�t� � �| _
dS )z4
Initialize
:param config:
FN) r( r) �setattr�getr- �disable_hidden� transportr r r! Zmy_ip��self�config�attrr
r
r �__init__u s
zSSDPFactory.__init__c O s | S )zo
Return instance on Factory call.
:param args:
:param kwargs:
:return:
r
)r5 �args�kwargsr
r
r �__call__� s zSSDPFactory.__call__c C s
|| _ dS )zL
On connection.
:param transport:
:return:
N)r3 )r5 r3 r
r
r �connection_made� s
zSSDPFactory.connection_madeN�
c
C s� d}dd� }|� }||k rLz| j j||d� | j�d� W dS tyE } z| j�d|� t�|� |d7 }||� 7 }W Y d}~nd}~ww ||k s
dS dS ) a�
On multi-master environments, running on the same machine,
transport sending to the destination can be allowed only at once.
Since every machine will immediately respond, high chance to
get sending fired at the same time, which will result to a PermissionError
at socket level. We are attempting to send it in a different time.
:param data:
:param addr:
:return:
r c S s dt �dd� S )Ng �?r= � )�random�randintr
r
r
r �slp_time� s z%SSDPFactory._sendto.<locals>.slp_time)�addrzSent successfullyNzPermission error: %sr )r3 �sendtor' �debug�AttributeError�time�sleep)r5 �datarB ZattemptsZtriesrA Zslp�exr
r
r �_sendto� s
���zSSDPFactory._sendtoc C s\ t jj�|�}|�| j�r�z
t|t| j�d� �}W n$ tt fy> | j
j dg|�R � | j r;| �
d�| jd�|� Y dS w tj�|�tj�� tjdd� k rj| j r^| �
d�| jd�|� | j
j dg|�R � dS | j
j d |g|�R � | �
t jj�d
�| jt jjj| jtd ���|� dS | j r�| �
t jj�d�| jd �|�� | j
j d
g|�R � dS )zb
On datagram receive.
:param data:
:param addr:
:return:
Nz0Received invalid timestamp in package from %s:%sz{}:E:{}zInvalid timestamp� )�secondszTimestamp is too oldz$Received outdated package from %s:%szReceived "%s" from %s:%sz{}:@:{}�Z _json_modulezInvalid packet signaturez!Received bad signature from %s:%s)�salt�utils� stringutils�
to_unicode�
startswithr �float�len� TypeError�
ValueErrorr' rD r2 rJ �format�datetime�
fromtimestamp�now� timedelta�to_bytes�jsonr r r )r5 rH rB �message� timestampr
r
r �datagram_received� sV ����
�����
��z SSDPFactory.datagram_received)Nr= ) r r r r$ r8 r; r< rJ r` r
r
r
r r/ p s
r/ c @ sD e Zd ZdZedd� �Zdd� Ze d dd ��Zd
d � ZdS )
�SSDPDiscoveryServerz'
Discovery service publisher.
c C � t �� S )zE
Return availability of the Server.
:return:
�r r r
r
r
r � is_available� � z SSDPDiscoveryServer.is_availablec K sB t �|�| _| j| jvri | j| j< | j| j �d| �� i� dS )z5
Initialize.
:param config:
ZmasterN)�copy�deepcopy�_configr) �updater# )r5 r6 r
r
r r8 � s z SSDPDiscoveryServer.__init__Nr c # sR � � s�s|s t d��||fdff}njt� }d� fd�ffD ]L\} }
|
durjt|
t�r1t|
�dks5J d��tj�| j|
|t j
||d���V } | sLt d ��| D ]\} }
}}
}| |f}||vrcddg||< ||| | < qNq � �fd
d �|� � D �}|s}t d ��g }|D ]s\\}}\}}d }}z7t j |t j
|d
�}t j
t jfD ]
}|�t j|d� q�|�d� � r�|�|� �r�tj�| �||��V |}W n. t y� } z|dur�|�� |�|� W Y d}~q�d}~w ty� |dur�|�� � w n|d �|� }tjj| d�}| �||||�}z tj�|�V W n
t�y! |�� � w tj�||��)z�
Create datagram connection.
Based on code from Python 3.5 version, this method is used
only in Python 2.7+ versions, since Trollius library did not
ported UDP packets broadcast.
zunexpected address family)NNr r N� z2-tuple is expected)�family�type�proto�flagsz!getaddrinfo() returned empty listc s8 g | ]\}}� r|d du s�r|d du s||f�qS )r Nr r
)�.0�keyZ addr_pair��
local_addr� remote_addrr
r �
<listcomp>* s � � �z@SSDPDiscoveryServer.create_datagram_endpoint.<locals>.<listcomp>z can not get address information)rk rl rm F)�loop)rV r �
isinstance�tuplerT r Z
coroutinesZFrom� getaddrinfor r �OSError�items� SO_REUSEADDR� SO_BROADCAST�
setsockopt�
SOL_SOCKET� setblocking�bindZ sock_connectr" �appendr ZfuturesZFutureZ_make_datagram_transportZReturn)ru Zprotocol_factoryrr rs rk rm rn Zaddr_pairs_infoZ
addr_infos�idxrB �infosZfam�_Zpro�addressrp �
exceptionsZ
local_addressZremote_address�sockZr_addr�opt�exc�protocol�waiterr3 r
rq r �create_datagram_endpoint� s� ����� � �
��� ��z,SSDPDiscoveryServer.create_datagram_endpointc C s� | j �| j| j| j �}| j �| j| j| j �}| j�d||� t�� }t | j | j
d�}tj r@|� t
j||||fd��\}}n|� |j|||fdd��\}}z|�� W | j�d� |�� |�� dS | j�d� |�� |�� w )z.
Run server.
:return:
z2Starting service discovery listener on udp://%s:%s)r )rr T)rr Zallow_broadcastz$Stopping service discovery listener.N)rh r1 r+ r- r* r' �infor Zget_event_loopr/ r) �portedZrun_until_completera r� Z run_foreverr" )r5 r r ru r� r3 r
r
r �runZ s4 �
�
�
��
�
zSSDPDiscoveryServer.run)NNr r r ) r r r r$ r. rd r8 r� r� r
r
r
r ra � s
� [ra c @ s< e Zd ZdZedd� �Zdd� Zdd� Zdd � Zd
d � Z d S )
�SSDPDiscoveryClientz:
Class to discover Salt Master via UDP broadcast.
c C rb )zD
Return availability of the Client
:return:
rc r
r
r
r rd re z SSDPDiscoveryClient.is_availablec K s� || _ t�tjtj�| _| j�tjtjd� | j�| j � | j
| j | j
�� | j | j
| j
fD ]}t| || j � || j | �� q.dS )z
Initialize
r N)rh r r r �_socketr} r~ r| �
settimeoutr1 r, r- r( r* r0 r4 r
r
r r8 � s � �z SSDPDiscoveryClient.__init__c C s4 t jj�| j� t�� � ��}| j�|d| jf� |S )zL
Query the broadcast for defined services.
:return:
z <broadcast>) rN rO rP r\ r rF r� rC r )r5 �queryr
r
r �_query� s zSSDPDiscoveryClient._queryc
C sx z | j �d�\}}|r ||vrg ||< || �|� nW dS W n ty: } z| j�d|� W Y d}~dS d}~ww q)zH
Collect masters map from the network.
:return:
Ti z'Discovery master collection failure: %sN)r� �recvfromr� r r' �error)r5 �responserH rB �errr
r
r �_collect_masters_map� s ����z(SSDPDiscoveryClient._collect_masters_mapc C s i }i }| j �d� | �� | �|� |s d}| j �|� |S |�� D ]e\}}|D ]^}tjj�|�}|� | j
�r�|� | j
�d }| j � d|d |d |� d|v rl|� d�d }| j �
d||� d |v rk| j �
d
|� q)||vrtg ||< || �tjjj|� d �d td �� q)q#|S )
zY
Gather the information of currently declared servers.
:return:
z Looking for a server discoveryz No master has been discovered.���z/Service announcement at '%s:%s'. Response: '%s'r r z:E:z3Error response from the service publisher at %s: %sr_ z(Publisher sent shifted timestamp from %sz:@:rM )r' r� r� r� rz rN rO rP rQ rR r �splitrD r� r� r] �loadsr )r5 r� Zmasters�msgrB Z descriptionsrH r� r
r
r �discover� sX
"�� �������� z SSDPDiscoveryClient.discoverN)
r r r r$ r. rd r8 r� r� r� r
r
r
r r� z s
r� ) r$ rf rX r% r? r rF Zsalt.utils.jsonrN Zsalt.utils.stringutilsZsalt.utils.odictr � ImportError� collectionsrO r] Z import_jsonr �hasattrr r� Ztrolliusr r r
r r/ ra r� r
r
r
r �<module> sH �
���
1u