File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/ssdp.cpython-310.pyc
Back
o ;j�: � @ s@ d Z ddlZddlZddlZddlZddlZddlZddlZddl Zzddl m Z W n e y9 ddl m Z Y nw ej j�� 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�ZdS )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 �attemptsZtriesrA 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 rK �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< rK ra 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)rl rm rn F)�loop)rW r � isinstance�tuplerU r Z coroutinesZFrom� getaddrinfor r �OSError�items� SO_REUSEADDR� SO_BROADCAST� setsockopt� SOL_SOCKET� setblocking�bindZ sock_connectr" �appendr ZfuturesZFutureZ_make_datagram_transportZReturn)rv Zprotocol_factoryrs rt rl rn ro Zaddr_pairs_infoZ addr_infos�idxrB �infosZfam�_Zpro�addressrq � exceptionsZ local_addressZremote_address�sockZr_addr�opt�exc�protocol�waiterr3 r rr 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 )rs T)rs Zallow_broadcastz$Stopping service discovery listener.N)ri r1 r+ r- r* r' �infor Zget_event_loopr/ r) �portedZrun_until_completerb r� Z run_foreverr" )r5 r r rv r� r3 r r r �runZ s4 � � � �� � zSSDPDiscoveryServer.run)NNr r r ) r r r r$ r. re r8 r� r� r r r r rb � s � [rb c @ s< e Zd ZdZedd� �Zdd� Zdd� Zdd � Zd d � Z d S ) �SSDPDiscoveryClientz: Class to discover Salt Master via UDP broadcast. c C rc )zD Return availability of the Client :return: rd r r r r re rf 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)ri 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>) rO rP rQ 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:@:rN )r' r� r� r� r{ rO rP rQ rR rS 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. re r8 r� r� r� r r r r r� z s r� ) r$ rg rY r% r? r rF Zsalt.utils.jsonrO Zsalt.utils.stringutils� collectionsr � ImportErrorrP r^ Z import_jsonr �hasattrr r� Ztrolliusr r r r r/ rb r� r r r r �<module> sH � ��� 1u
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings