Current File : //lib/python3.9/site-packages/ansible/module_utils/__pycache__/connection.cpython-39.pyc
a
�)g� � @ s� d dl mZmZmZ eZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
mZ d dlmZmZ d dlmZ d dlmZ d dlmZ dd � Zd
d � Zd d
� Zdd� Zdd� Z G dd� de �Z G dd� de �Z dS )� )�absolute_import�division�print_functionN)�partial)�to_bytes�to_text)�AnsibleJSONEncoder)� iteritems)�cPicklec C s` t j|dd�}|�dd�}tt�|��� �}t�| dt |� � t�| |� t�| d| � dS )z�Handles making sure all data is properly written to file descriptor fd.
In particular, that data is encoded in a character stream-friendly way and
that all data gets written before returning.
r )Zprotocol�
s \rs %d
s %s
N)
r
�dumps�replacer �hashlibZsha1Z hexdigest�os�write�len)�fd�obj�srcZ data_hash� r �C/usr/lib/python3.9/site-packages/ansible/module_utils/connection.py�write_to_file_descriptor/ s r c C s t �dt|��}| �|| �S )N�!Q)�struct�packr Zsendall)�s�dataZ
packed_lenr r r � send_dataF s r c C s� d}t d�}t|�|k r<| �|t|� �}|s2d S ||7 }q t�d|d |� �d }||d � }t|�|k r�| �|t|� �}|s�d S ||7 }q`|S )N� � r r )r r Zrecvr �unpack)r Z
header_lenr �dZdata_lenr r r � recv_dataK s
r" c
C sr t | j�}z|�|�}W nN tyf } z6t|dd�}t|d|�}|dt|dd�fW Y d }~S d }~0 0 d|dfS )N�code� �errr �surrogate_then_replace��errorsr )�
ConnectionZ _socket_path� exec_command�ConnectionError�getattrr )�moduleZcommandZ
connection�out�excr# �messager r r r* ] s
(r* c O s( t t�� �}d| |d�}||f|d< |S )Nz2.0)Zjsonrpc�method�id�params)�str�uuidZuuid4)Zmethod_�args�kwargs�reqid�reqr r r �request_builderh s r: c s e Zd Z� fdd�Z� ZS )r+ c s2 t t| ��|� t|�D ]\}}t| ||� qd S )N)�superr+ �__init__r �setattr)�selfr0 r6 r7 �k�v�� __class__r r r< r s zConnectionError.__init__)�__name__�
__module__� __qualname__r< �
__classcell__r r rA r r+ p s r+ c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) r) c C s |d u rt d��|| _d S )Nzsocket_path must be a value)�AssertionError� socket_path)r> rH r r r r< z s zConnection.__init__c C sN z | j | W S tyH |�d�r8td| jj|f ��t| j|� Y S 0 d S )N�_z!'%s' object has no attribute '%s')�__dict__�KeyError�
startswith�AttributeErrorrB rC r �__rpc__)r> �namer r r � __getattr__ s
zConnection.__getattr__c
O s� t |g|�R i |��}|d }tj�| j�s:td| j ��ztj|tdd�}W n4 t y� } z tdt
|� ��W Y d }~n
d }~0 0 z| � |�}W nF t j
y� } z,td| j t
| dd�t�� d ��W Y d } ~ n
d } ~ 0 0 zt�|�}
W nl t�yT |�d
��rtd �||���d d
� |D �dd
� t|�D � } d�| �} td�|| |���Y n0 |
d |k�rltd��d|
v �r�t�t|
d ��|
d< |
S )Nr2 z�socket path %s does not exist or cannot be found. See Troubleshooting socket path issues in the Network Debug and Troubleshooting GuideT)�clsZ
vault_to_textzqFailed to encode some variables as JSON for communication with ansible-connection. The original exception was: %szuunable to connect to socket %s. See Troubleshooting socket path issues in the Network Debug and Troubleshooting Guider& r' �r% � exceptionZ
set_optionz;Unable to decode JSON from response to {0}. Received '{1}'.c S s g | ] }t |��qS r )�repr)�.0�argr r r �
<listcomp>� � z,Connection._exec_jsonrpc.<locals>.<listcomp>c S s g | ]\}}d � ||��qS )z {0}={1!r})�format)rU r? r@ r r r rW � rX z, z@Unable to decode JSON from response to {0}({1}). Received '{2}'.z invalid json-rpc id receivedZ result_type�result)r: r �path�existsrH r+ �jsonr r � TypeErrorr �send�socket�error� traceback�
format_exc�loads�
ValueErrorrL rY r �joinr
r ) r> rO r6 r7 r9 r8 r r/ r. �e�responser3 r r r �
_exec_jsonrpc� sT ������
�
�
zConnection._exec_jsonrpcc O s` | j |g|�R i |��}d|v rX|�d�}|�d�p:|d }|d }tt|dd�|d��|d S ) a� Executes the json-rpc and returns the output received
from remote device.
:name: rpc method to be executed over connection plugin that implements jsonrpc 2.0
:args: Ordered list of params passed as arguments to rpc method
:kwargs: Dict of valid key, value pairs passed as arguments to rpc method
For usage refer the respective connection plugin docs.
ra r r0 r# r& r' )r# rZ )ri �getr+ r )r> rO r6 r7 rh r% �msgr# r r r rN � s
zConnection.__rpc__c
C s� z6t � t jt j�}|�| j� t|t|�� t|�}W nN t jy� } z4|� � t
d| j t |dd�t �
� d��W Y d }~n
d }~0 0 |� � t |dd�S )Nzmunable to connect to socket %s. See the socket path issue category in Network Debug and Troubleshooting Guider& r' rR Zsurrogate_or_strict)r` ZAF_UNIXZ SOCK_STREAMZconnectrH r r r" ra �closer+ r rb rc )r> r Zsfrh rg r r r r_ � s �� zConnection.sendN)rC rD rE r< rP ri rN r_ r r r r r) x s
1r) )!Z
__future__r r r �typeZ
__metaclass__r r r] r` r rb r5 � functoolsr Zansible.module_utils._textr r Z ansible.module_utils.common.jsonr Zansible.module_utils.sixr Z ansible.module_utils.six.movesr
r r r" r* r: � Exceptionr+ �objectr) r r r r �<module> s(
Mini Shell