Current File : //lib/python2.7/site-packages/salt/utils/ssdp.pyc
�
���^c @@ s� d Z d d l m Z m Z m 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 d d l
Z d d l Z e j j j � Z e e d � s� d Z n y d d l Z e e _ WnD e k
ry d d l Z e e _ Wq e k
rd Z q Xn Xd e f d � � YZ d e f d � � YZ d
e f d � � YZ d e f d
� � YZ! d e f d � � YZ" d e f d � � YZ# d S( uq
Salt Service Discovery Protocol.
JSON-based service discovery protocol, used by minions to find running Master.
i ( t absolute_importt print_functiont unicode_literalsN( t OrderedDictu dumpst TimeOutExceptionc B@ s e Z RS( ( t __name__t
__module__( ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyR 2 s t TimeStampExceptionc B@ s e Z RS( ( R R ( ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyR 6 s t SSDPBasec B@ s� e Z d Z e j e � Z d Z d Z d Z d Z d Z
i d e 6d e 6d e 6d e
6i e 6Z e d
� � Z
e d � � Z RS( u*
Salt Service Discovery Protocol.
u signatureu answeru portu listen_ipu timeoutu __salt_master_servicei� u 0.0.0.0i c C@ s t t o t � S( uW
Return True if the USSDP dependencies are satisfied.
:return:
( t boolt asynciot _json( ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyt
_is_availableP s c C@ sx t j t j t j � } zN y! | j d � | j � d } Wn& t k
rd t j t j � � } n XWd | j � X| S( uB
Find out localhost outside IP.
:return:
u
1.255.255.255i i N( u
1.255.255.255i ( t sockett AF_INETt
SOCK_DGRAMt connectt getsocknamet Exceptiont
gethostbynamet gethostnamet close( t sckt ip_addr( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyt get_self_ipX s
( R R t __doc__t loggingt getLoggert logt SIGNATUREt ANSWERt PORTt LISTEN_IPt TIMEOUTt DEFAULTSt staticmethodR R ( ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyR : s
t SSDPFactoryc B@ sA e Z d Z d � Z d � Z d � Z d d d � Z d � Z RS( u"
Socket protocol factory.
c K@ sn x= | j | j f D]) } t | | | j | | j | � � q Wt | _ d | _ t j
t j � � | _ d S( u4
Initialize
:param config:
N(
R R t setattrt getR" t Falset disable_hiddent Nonet transportR
R R t my_ip( t selft configt attr( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyt __init__o s
' c O@ s | S( uo
Return instance on Factory call.
:param args:
:param kwargs:
:return:
( ( R, t argst kwargs( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyt __call__{ s c C@ s
| | _ d S( uL
On connection.
:param transport:
:return:
N( R* ( R, R* ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyt connection_made� s i
c C@ s� d } d � } | � } x� | | k r� y. | j j | d | �| j j d � d SWq t k
r� } | j j d | � t j | � | d 7} | | � 7} q Xq Wd S( u�
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:
i c S@ s d t j d d � S( Ng �?i
i ( t randomt randint( ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyt <lambda>� s t addru Sent successfullyNu Permission error: %si ( R* t sendtoR t debugt AttributeErrort timet sleep( R, t dataR7 t attemptst triest slp_timet slpt ex( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyt _sendto� s
c C@ s� t j j j | � } | j | j � ruy t | t | j � � } WnV t t f k
r� | j
j d | � | j r� | j
d j | j d � | � n d SXt j j | � t j j � t j d d � k r| j r� | j
d j | j d � | � n | j
j d | � d S| j
j d | | � | j
t j j j t d
� j | j t j j j | j d t �� � | � nM | j r�| j
t j j j d j | j d � | � � n | j
j d
| � d S( ub
On datagram receive.
:param data:
:param addr:
:return:
u0 Received invalid timestamp in package from %s:%su {0}:E:{1}u Invalid timestampNt secondsi u Timestamp is too oldu$ Received outdated package from %s:%su Received "%s" from %s:%su {0}:@:{1}t _json_moduleu Invalid packet signatureu! Received bad signature from %s:%s( t saltt utilst stringutilst
to_unicodet
startswitht signaturet floatt lent TypeErrort
ValueErrorR R9 R( RC t formatt datetimet
fromtimestampt nowt timedeltat to_bytest strt jsont dumpst answerR ( R, R= R7 t messaget timestamp( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyt datagram_received� s8 "1 "!
N( R R R R/ R2 R3 R) RC R\ ( ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyR$ j s t SSDPDiscoveryServerc B@ sM e Z d Z e d � � Z d � Z e d d d d d d � � Z d � Z RS( u'
Discovery service publisher.
c C@ s
t j � S( uE
Return availability of the Server.
:return:
( R R ( ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyt is_available� s c K@ s_ t j | � | _ | j | j k r7 i | j | j <n | j | j j i | j � d 6� d S( u5
Initialize.
:param config:
u masterN( t copyt deepcopyt _configR t updateR ( R, R- ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyR/ � s i c c@ s� | p | s9 | s! t d � � n | | f d f f } n�t � } x
d | f d | f f D]� \ } }
|
d
k r[ t |
t � r� t |
� d k s� t d � � t j j | j
d | d t j d | d | |
� � V} | s� t j
d
� � n xY | D]N \ } }
} }
} | | f } | | k r6d
d
g | | <n | | | | <q� Wq[ q[ Wg | j � D]D \ } } | r~| d d
k p�| o�| d d
k s\| | f ^ q\} | s�t d � � n g } xG| D]5\ \ } } \ } } d
} } y� t j d | d t j d | � } x0 t j t j g D] } | j t j | d � q!W| j t � | rd| j | � n | r�t j j | j | | � � V| } n Wnh t j
k
r�} | d
k r�| j � n | j | � q�t k
r�| d
k r�| j � n � q�XPq�W| d � | � } t j j d | � } | j | | | | � } y t j j | � VWn t k
rv| j � � n Xt j j | | � � d
S( u�
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.
u unexpected address familyi i i u 2-tuple is expectedt familyt typet protot flagsu! getaddrinfo() returned empty listu can not get address informationt loopN( NN( RO R) R t
isinstancet tupleRM t AssertionErrorR
t
coroutinest Fromt getaddrinfoR
R t errort itemst SO_REUSEADDRt SO_BROADCASTt
setsockoptt
SOL_SOCKETt setblockingR' t bindt sock_connectR t appendR t futurest Futuret _make_datagram_transportt Return( Rg t protocol_factoryt
local_addrt remote_addrRc Re Rf t addr_pairs_infot
addr_infost idxR7 t infost famt _t prot addresst keyt addr_pairt
exceptionst
local_addresst remote_addresst sockt r_addrt optt exct protocolt waiterR* ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyt create_datagram_endpoint� sn % -% (
!
c C@ s" | j j | j | j | j � } | j j | j | j | j � } | j j d | | � t j � } t d | j | j
� } t j r� | j t
j | | d | | f �� \ } } n0 | j | j | d | | f d t �� \ } } z | j � Wd | j j d � | j � | j � Xd S( u.
Run server.
:return:
u2 Starting service discovery listener on udp://%s:%sRY R} t allow_broadcastNu$ Stopping service discovery listener.( Ra R& R R" R R t infoR
t get_event_loopR$ R t portedt run_until_completeR] R� t Truet run_foreverR ( R, t listen_ipt portRg R� R* ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyt run3 s "" **
N( R R R R# R^ R/ R) R� R� ( ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyR] � s Et SSDPDiscoveryClientc B@ sA e Z d Z e d � � Z d � Z d � Z d � Z d � Z RS( u:
Class to discover Salt Master via UDP broadcast.
c C@ s
t j � S( uD
Return availability of the Client
:return:
( R R ( ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyR^ O s c K@ s� | | _ t j t j t j � | _ | j j t j t j d � | j j | j j | j
| j | j
� � xF | j | j
| j
g D], } t | | | j j | | j | � � q� Wd S( u
Initialize
i N( Ra R
R R t _socketRr Rs Rq t
settimeoutR& R! R" R R R% ( R, R- R. ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyR/ W s , c C@ sJ t j j j d j | j t j � � � } | j j | d | j f � | S( uL
Query the broadcast for defined services.
:return:
u {}{}u <broadcast>(
RF RG RH RU RP RK R; R� R8 R� ( R, t query( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyt _queryc s c C@ s� x� t r� yP | j j d � \ } } | rW | | k rC g | | <n | | j | � n PWq t k
r� } | j j d | � Pq Xq Wd S( uH
Collect masters map from the network.
:return:
i u' Discovery master collection failure: %sN( R� R� t recvfromRw R R Rn ( R, t responseR= R7 t err( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyt _collect_masters_mapn s
c C@ s� i } i } | j j d � | j � | j | � | sR d } | j j | � n5x2| j � D]$\ } } x| D]
} t j j j | � } | j | j
� rr | j | j
� d } | j j d | d | d | � d | k r0| j d � d } | j j
d | | � d | k r|| j j
d
| � q|q| | k rIg | | <n | | j t j j j | j d � d d t �� qr qr Wq_ W| S(
uY
Gather the information of currently declared servers.
:return:
u Looking for a server discoveryu No master has been discovered.i����u/ Service announcement at '%s:%s'. Response: '%s'i i u :E:u3 Error response from the service publisher at %s: %su timestampu( Publisher sent shifted timestamp from %su :@:RE ( R R� R� R� Ro RF RG RH RI RJ RK t splitR9 Rn Rw RW t loadsR ( R, R� t masterst msgR7 t descriptionsR= R� ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyt discover� s8
4( R R R R# R^ R/ R� R� R� ( ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyR� K s ($ R t
__future__R R R RQ R; R R4 R
R_ t collectionsR t salt.utils.jsonRF t salt.utils.stringutilsRG RW t import_jsonR t hasattrR) R
R' R� t ImportErrort trolliusR� R R R t objectR R$ R] R� ( ( ( s3 /usr/lib/python2.7/site-packages/salt/utils/ssdp.pyt <module> s8
0kv
Mini Shell