File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/__pycache__/payload.cpython-310.pyc
o
�N�gK5 � @ s� d Z ddlZddlZddlZddlZddlZddlm m
Z
ddl Zddl Zddl
mZ ddlmZmZ ddlmZ zddlZW n eyJ Y nw e�e�Zdd� Zdd � Zd
d � Zdd
d�Z ddd�Z dd� Z dd� Z G dd� d�Z G dd� d�Z!dS )z�
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
� N)� _Constant)�SaltDeserializationError�SaltReqTimeoutError)�CaseInsensitiveDictc C s t jj�| �S )z�
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.
)�salt�utils�msgpack�dumps)�payload� r �@/opt/saltstack/salt/lib/python3.10/site-packages/salt/payload.py�package s r
c C s t jjj| dd�S )z
Unpackages a payload
T)�use_list)r r r �loads)Zpackage_r r r � unpackage% s r c K s2 d| i}i }|D ]}|| ||< q||d< t |�S )z�
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.
�enc�load)r
)r �kwargsr
r �keyr r r �format_payload, s r Fc
C s� zxzQdd� }t �� d|d�}|du rd|d< nd|d< z
tjjj| fi |��}W n tyC |�dd� tjjj| fi |��}Y nw |du rQ|sQtj j
� |�}W n t yq } zt
�d|� t
�d | � d
}t|�|�d}~ww W t �� |S t �� w ) aL
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 sP | dkrt jj�|�}tj�|d�S | dkr&t jjj|dd�\}}t||�S |S )N�N �%Y%m%dT%H:%M:%S.%f�O F)�raw) r r � stringutilsZ
to_unicode�datetime�strptimer r r )�code�data�name�valuer r r �ext_type_decoderL s
z loads.<locals>.ext_type_decoderT)r Zext_hookNr Fz�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: %sz.Msgpack deserialization failure on message: %sz=Could not deserialize msgpack message. See log for more info.)�gc�disabler r r Zunpackb�UnicodeDecodeError�popr Z transport�frameZdecode_embedded_strs� Exception�log�critical�debugr �enable)�msg�encodingr r! Z loads_kwargs�ret�excZexc_msgr r r r 9 s<
� ��
���
�r c sj dd� }z t jjj| ||d�W S tt jjjjfy4 � fdd�� � | t� �} t jjj| ||d� Y S w )a�
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�r t| �S t | tjtjf�r"tjj�dtjj �
| � d���S t | t �r:tjj�dtjjj
| j| jfdd��S t | tj�rDt| �S t | tj�rNt| �S t | ttjf�rZt| �S t | t�rct| �S t | tjj�rnt| �S | S )Nr r r T�� use_bin_type) �
isinstance�int�strr �dater r r ZExtTyper �to_bytes�strftimer r r r �immutabletypesZ
ImmutableDict�dictZ
ImmutableList�list�setZ ImmutableSet�tupler � collections�abc�MutableMapping)�objr r r �ext_type_encoder~ s.
�
�
z dumps.<locals>.ext_type_encoder)�defaultr1 c s� t | �}||v rt| tttf�rd�t| �jt | ��S |�|� t| t�r;| � � �
� D ] \}}� ||�| |< q+t| �S t| ttf�rXt| �} t | �D ] \}}� ||�| |< qJ| S t| t �rh| t
dd�krht| �S | S )Nz <Recursion on {} with id={}>� �@ )�idr2 r9 r: r< �format�type�__name__�add�copy�items� enumerater3 �powr4 )r@ �contextZobjidr r �idx�entry��verylong_encoderr r rR � s$ �
z dumps.<locals>.verylong_encoder)r r r Zpackb�
OverflowError�
exceptionsZPackValueErrorr; )r, r1 rA r rQ r r s s "�
��r c C s$ | � � }| �� |rt|dd�S dS )z6
Run the correct serialization to load a file
zutf-8)r- N)�read�closer )�fn_r r r r r � s
�r c C s |� t| dd�� |�� dS )z?
Serialize the correct data into the named file object
Tr0 N)�writer rV )r, rW r r r �dump� s rY c @ s8 e Zd ZdZdd� Zee�Zee�Zee�Zee �Z dS )�Serialzb
Create a serialization object, this object manages all message
serialization in Salt
c O s t jj�dd� d S )NZChlorinez�The `salt.payload.Serial` class has been deprecated, and is set to be removed in {version}. Please use `salt.payload.loads` and `salt.payload.dumps`.)r r ZversionsZ
warn_until)�self�argsr r r r �__init__� s �zSerial.__init__N)
rH �
__module__� __qualname__�__doc__r] � staticmethodr r rY r r r r r rZ � s rZ c @ sZ e Zd ZdZddd�Zedd � �Zd
d � Zd d
� Zddd�Z ddd�Z
dd� Z dd� Z dS )�SREQzC
Create a generic interface to wrap salt zeromq req calls.
� r r Nc C s0 || _ || _|| _t�� | _t�� | _|| _d S �N) �master�id_�linger�zmqZContextrN ZPoller�poller�opts)r[ re rf �serialrg rj r r r r] � s
z
SREQ.__init__c C s� t | d�sW| j�tj�| _t td�r| j�tjd� | �� | j �
d�r?t td�r2| j�tj d� n
t td�r?| j�tj d� | j
| j_
| jrP| j�tj| j� | j�| j � | jS ) z+
Lazily create the socket.
�_socket�RECONNECT_IVL_MAXi� ztcp://[�IPV6� �IPV4ONLYr )�hasattrrN �socketrh ZREQrl �
setsockoptrm �_set_tcp_keepalivere �
startswithrn rp rg rf ZIDENTITY�connect�r[ r r r rr s
z SREQ.socketc C s� t td�rJ| jrLd| jv r| j�tj| jd � d| jv r(| j�tj| jd � d| jv r8| j�tj| jd � d| jv rN| j�tj| jd � d S d S d S d S )N�
TCP_KEEPALIVEZ
tcp_keepaliveZtcp_keepalive_idleZtcp_keepalive_cntZtcp_keepalive_intvl) rq rh rj rl rs rx ZTCP_KEEPALIVE_IDLEZTCP_KEEPALIVE_CNTZTCP_KEEPALIVE_INTVLrw r r r rt s"
�
�
�� zSREQ._set_tcp_keepalivec C s� t | d�r?t| jjt�r&t| jj�� �}|D ]}t�d|� | j� |� qn| jjD ]}t�d|� | j� |d � q*| `
dS dS )z.
delete socket if you have it
rl zUnregistering socket: %sr N) rq r2 ri �socketsr9 r: �keysr( Ztrace�
unregisterrl �r[ ry rr r r r � clear_socket* s
� �zSREQ.clear_socketro �< c C s� d|i}||d< t |�}| j�|� | j�| jtj� d} | j�|d �}|d7 }|r-n |dkr9t� d|||� ||krK| �
� t d� || |���q t
| j�� �S ) zO
Takes two arguments, the encryption type and the base payload
r r r Ti� ro z5SaltReqTimeoutError: after %s seconds. (Try %s of %s)z3SaltReqTimeoutError: after {} seconds, ran {} tries)r rr �sendri �registerrh �POLLIN�pollr( �infor} r rF r �recv) r[ r r �tries�timeoutr
ZpkgZtriedZpolledr r r r : s6 ����z SREQ.sendc C s( |� dd�}|� di �}| �||||�S )zA
Detect the encryption type based on the payload
r �clearr )�getr )r[ r
r� r� r r r r r � send_autoY s zSREQ.send_autoc C s� t | jjt�r+t| jj�� �}|D ]}|jdu r#|�tj d� |�
� | j� |� qn%| jjD ] }|d jdu rG|d �tj d� |d �
� | j� |d � q/| j jdu rc| j �tj d� | j �
� | j
jdu rp| j
�� d S d S )NFro r )r2 ri ry r9 r: rz �closedrs rh ZLINGERrV r{ rr rN Ztermr| r r r �destroya s&
�
�z SREQ.destroyc C s | � � d S rd )r� rw r r r �__del__v s z SREQ.__del__)rc r r N)ro r~ )
rH r^ r_ r` r] �propertyrr rt r} r r� r� r� r r r r rb � s
rb )NF)F)"r` �collections.abcr= r r" �loggingZsalt.transport.framer Zsalt.utils.immutabletypesr r8 Zsalt.utils.msgpackZsalt.utils.stringutilsZ
salt.defaultsr Zsalt.exceptionsr r Zsalt.utils.datar rh � ImportError� getLoggerrH r( r
r r r r r rY rZ rb r r r r �<module> s6 �
:X