File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/salt/__pycache__/payload.cpython-310.pyc
Back
o ;jK5 � @ 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
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings