Current File : //lib/python2.7/site-packages/salt/payload.pyo
�
���^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 j j
Z
d d l Z d d l m Z m Z d d l m Z d d l m Z y d d l Z Wn e k
r� n Xe j e � Z e Z y� d d l Z e j d k rHe j e j d d d
g d e �d e! �d k r~e � q~n6 e j e j d d d
g � d e! �d k r~e � n e! Z Wna e k
r�y d d l# Z e! Z Wq�e k
r�d
Z$ e j j% d e$ � e j& d � q�Xn Xe r:e' e d � r:d e( f d � � YZ) d e* f d � � YZ+ e+ � e _+ n d � Z, d � Z- d � Z. d e* f d � � YZ/ d e* f d � � YZ0 d S( u�
Many aspects of the salt payload need to be managed, from the return of
encrypted keys to general payload dynamics and packaging, these happen
in here
i ( t absolute_importt print_functiont unicode_literalsN( t SaltReqTimeoutErrort SaltDeserializationError( t CaseInsensitiveDict( t sixi i i i t use_bin_typet use_listu [%(levelname)-8s] %(message)st
log_formatu7 Unable to import msgpack or msgpack_pure python modulesu
exceptionst PackValueErrorc B@ s e Z d Z RS( uF
older versions of msgpack do not have PackValueError
( t __name__t
__module__t __doc__( ( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyR
? s t
exceptionsc B@ s e Z d Z e � Z RS( uL
older versions of msgpack do not have an exceptions module
( R R R
R
( ( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyR D s c C@ s
t j | � S( u�
This method for now just wraps msgpack.dumps, but it is here so that
we can make the serialization a custom option in the future with ease.
( t msgpackt dumps( t payload( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyt packageM s c C@ s t j | d t �S( u
Unpackages a payload
R ( R t loadst True( t package_( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyt unpackageU s c K@ sF i | d 6} i } x | D] } | | | | <q W| | d <t | � S( u�
Pass in the required arguments for a payload, the enc type and the cmd,
then a list of keyword args to generate the body of the load dict.
u encu load( R ( t enct kwargsR t loadt key( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyt format_payload\ s
t Serialc B@ sD e Z d Z d � Z d e d � Z d � Z e d � Z d � Z RS( ub
Create a serialization object, this object manages all message
serialization in Salt
c C@ sR t | t � r' | j d d � | _ n' t | t j � rE | | _ n d | _ d S( Nu serialu msgpack( t
isinstancet dictt gett serialR t string_types( t selft opts( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyt __init__n s
c C@ s� zzy d � } t j � i t d 6| d 6} t j d k r� t j d k rq | d
k rd t | d <q{ t | d <n
| | d <y t j | | � } Wq� t k
r� | j d d
� | j d d
� t j | | � } q� Xn t j | | � } t
j r | d
k r | r t j
j j | � } n WnU t k
rx} t j d
| � t j d | � t j � t
j t d � | � � n XWd
t j � X| S( u�
Run the correct loads serialization format
:param encoding: Useful for Python 3 support. If the msgpack data
was encoded using "use_bin_type=True", this will
differentiate between the 'bytes' type and the
'str' type by decoding contents with 'str' type
to what the encoding was set as. Recommended
encoding is 'utf-8' when using Python 3.
If the msgpack data was not encoded using
"use_bin_type=True", it will try to decode
all 'bytes' and 'str' data (the distinction has
been lost in this case) to what the encoding is
set as. In this case, it will fail if any of
the contents cannot be converted.
c S@ s8 | d k r4 t j j j | � } t j j | d � S| S( NiN u %Y%m%dT%H:%M:%S.%f( t saltt utilst stringutilst
to_unicodet datetimet strptime( t codet data( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyt ext_type_decoder� s u use_listu ext_hooki i i i u rawu encodingu� Could not deserialize msgpack message. This often happens when trying to read a file not in binary mode. To see message payload, enable debug logging and retry. Exception: %su. Msgpack deserialization failure on message: %su= Could not deserialize msgpack message. See log for more info.N( i i i ( i i i ( t gct disableR R t versiont Nonet FalseR t UnicodeDecodeErrort popR t PY3R% t transportt framet decode_embedded_strst Exceptiont logt criticalt debugt collectt
raise_fromR t enable( R" t msgt encodingt rawR- t loads_kwargst rett exc( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyR v sB
c C@ sL | j � } | j � | rH t j r8 | j | d d �S| j | � Sn d S( u>
Run the correct serialization to load a file
RA u utf-8N( t readt closeR R5 R ( R" t fn_R, ( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyR � s
c @ s� d � } y? t j d k r4 t j | d | d | �St j | d | �SWns t t j j f k
r� � f d � � � | � } t j d k r� t j | d | d | �St j | d | �Sn Xd S(
u�
Run the correct dumps serialization format
:param use_bin_type: Useful for Python 3 support. Tells msgpack to
differentiate between 'str' and 'bytes' types
by encoding them differently.
Since this changes the wire protocol, this
option should not be used outside of IPC.
c S@ s� t | t j � r t j | � St | t j t j f � rb t j d t j j
j | j d � � � St | t
j � r~ t | � St | t
j � r� t | � St | t t
j f � r� t | � St | t � r� t | � S| S( NiN u %Y%m%dT%H:%M:%S.%f( R R t
integer_typest text_typeR) t dateR t ExtTypeR% R& R' t to_bytest strftimet immutabletypest
ImmutableDictR t
ImmutableListt listt sett ImmutableSett tupleR ( t obj( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyt ext_type_encoder� s
i i t defaultR c @ s� t | t � rO x3 t j | j � � D] \ } } � | � | | <q% Wt | � St | t t f � r� t | � } x* t | � D] \ } } � | � | | <q} W| St | t j � r� | t d d � k r� t j
| � S| Sd S( Ni i@ ( R R R t iteritemst copyRR RU t enumerateRI t powRJ ( RV R t valuet idxt entry( t verylong_encoder( s0 /usr/lib/python2.7/site-packages/salt/payload.pyR` � s "
'
N( i i i ( i i i ( R R0 R t
OverflowErrorR R
( R" R@ R RW ( ( R` s0 /usr/lib/python2.7/site-packages/salt/payload.pyR � s
c C@ sL t j r" | j | j | � � n | j | j | d t �� | j � d S( uG
Serialize the correct data into the named file object
R N( R t PY2t writeR R RG ( R" R@ RH ( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyt dump
s N(
R R R
R$ R1 R2 R R R Rd ( ( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyR i s F Et SREQc B@ st e Z d Z d d d d d � Z e d � � Z d � Z d � Z d d d
� Z d d d � Z
d � Z d
� Z RS( uC
Create a generic interface to wrap salt zeromq req calls.
u u msgpacki c C@ sU | | _ | | _ t | � | _ | | _ t j � | _ t j � | _ | | _
d S( N( t mastert id_R R t lingert zmqt Contextt contextt Pollert pollerR# ( R" Rf Rg R Rh R# ( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyR$ s c C@ s t | d � s| j j t j � | _ t t d � rO | j j t j d � n | j � | j j
d � r� t t d � r� | j j t j d � q� t t d � r� | j j t j d � q� n | j
| j _
| j r� | j j t j | j � n | j j | j � n | j S( u+
Lazily create the socket.
u _socketu RECONNECT_IVL_MAXi� u tcp://[u IPV6i u IPV4ONLYi ( t hasattrRk t socketRi t REQt _sockett
setsockoptt RECONNECT_IVL_MAXt _set_tcp_keepaliveRf t
startswitht IPV6t IPV4ONLYRh Rg t IDENTITYt connect( R" ( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyRo ( s
c C@ s� t t d � r� | j r� d | j k rG | j j t j | j d � n d | j k rv | j j t j | j d � n d | j k r� | j j t j | j d � n d | j k r� | j j t j | j d � q� n d S( Nu
TCP_KEEPALIVEu
tcp_keepaliveu tcp_keepalive_idleu tcp_keepalive_cntu tcp_keepalive_intvl( Rn Ri R# Rq Rr t
TCP_KEEPALIVEt TCP_KEEPALIVE_IDLEt TCP_KEEPALIVE_CNTt TCP_KEEPALIVE_INTVL( R" ( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyRt B s c C@ s� t | d � r� t | j j t � rp t | j j j � � } xl | D]& } t j d | � | j j | � qC Wn; x8 | j j D]* } t j d | � | j j | d � q} W| `
n d S( u.
delete socket if you have it
u _socketu Unregistering socket: %si N( Rn R Rm t socketsR RR t keysR: t tracet
unregisterRq ( R" R~ Ro ( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyt clear_socketU s
i i< c C@ s i | d 6} | | d <| j j | � } | j j | � | j j | j t j � d } x� t r� | j j | d � } | d 7} | r� Pn | d k r� t
j d | | | � n | | k r[ | j � t
d j | | | � � � q[ q[ W| j j | j j � � S( uO
Takes two arguments, the encryption type and the base payload
u encu loadi i� i u5 SaltReqTimeoutError: after %s seconds. (Try %s of %s)u5 SaltReqTimeoutError: after {0} seconds, ran {1} tries( R R Ro t sendRm t registerRi t POLLINR t pollR: t infoR� R t formatR t recv( R" R R t triest timeoutR t pkgt triedt polled( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyR� e s*
c C@ s: | j d d � } | j d i � } | j | | | | � S( uA
Detect the encryption type based on the payload
u encu clearu load( R R� ( R" R R� R� R R ( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyt send_auto� s c C@ sD t | j j t � r� t | j j j � � } x� | D]E } | j t k ri | j t j
d � | j � n | j j | � q4 Wnf xc | j j D]U } | d j t k r� | d j t j
d � | d j � n | j j | d � q� W| j
j t k r | j
j t j
d � | j
j � n | j j t k r@| j j � n d S( Ni i ( R Rm R~ R RR R t closedR2 Rr Ri t LINGERRG R� Ro Rk t term( R" R~ Ro ( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyt destroy� s"
c C@ s | j � d S( N( R� ( R" ( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyt __del__� s N(
R R R
R1 R$ t propertyRo Rt R� R� R� R� R� ( ( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyRe s ( i i i (1 R
t
__future__R R R t loggingR. R) t salt.logR% t salt.transport.framet salt.utils.immutabletypesR& RO t salt.utils.stringutilst salt.exceptionsR R t salt.utils.dataR t salt.extR Ri t ImportErrort getLoggerR R: R2 t HAS_MSGPACKR R0 R R R R1 t msgpack_puret
LOG_FORMATt setup_console_loggert fatalRn R9 R
t objectR R R R R Re ( ( ( s0 /usr/lib/python2.7/site-packages/salt/payload.pyt <module> sT
3 -
�
Mini Shell