관리-도구
편집 파일: 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(