File: //proc/1233/root/usr/lib/python3/dist-packages/docker/transport/__pycache__/sshconn.cpython-310.pyc
o
�2.a� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z ddl
mZ d dlmZ z d dlm mZ W n
eyN d dlZY nw ejjZG dd� de j �ZG dd� dej�ZG d d
� d
ejj �Z G d d � d e �Z dS )
� N)�BaseHTTPAdapter� )� constantsc sT e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
� Z S )� SSHSocketc sj t � �tjtj� || _d | _d | _d| jv r!| j�d�\| _| _d| jv r0| j�d�\| _| _d | _ d S )N�:�@)
�super�__init__�socket�AF_INET� SOCK_STREAM�host�port�user�split�proc)�selfr
�� __class__� �:/usr/lib/python3/dist-packages/docker/transport/sshconn.pyr s �
zSSHSocket.__init__c K s� dg}| j r
|d| j g }| jr|d| jg }|d| jdg }d }tjs*dd� }|}ttj�}|�dd � |�d d � t j
d
� |�|d t j t j tjrLd n|d �| _
d S )
N�sshz-lz-pz--�docker system dial-stdioc S s t � t jt j� d S �N)�signal�SIGINT�SIG_IGNr r r r �f2 s z SSHSocket.connect.<locals>.f�LD_LIBRARY_PATH�
SSL_CERT_FILE� T)�env�shell�stdout�stdin�
preexec_fn)r r r
r �IS_WINDOWS_PLATFORM�dict�os�environ�pop�
subprocess�Popen�join�PIPEr )r �kwargs�args� preexec_funcr r! r r r �connect& s*
�zSSHSocket.connectc C s6 | j r| j jjr td��| j j�|�}| j j�� |S �Nz<SSH subprocess not initiated.connect() must be called first.)r r$ �closed� Exception�write�flush)r �data�writtenr r r �_writeD s
zSSHSocket._writec C s | � |� d S r �r: �r r8 r r r �sendallL s zSSHSocket.sendallc C s
| � |�S r r; r< r r r �sendO s
zSSHSocket.sendc C s | j std��| j j�|�S r3 )r r5 r# �read)r �nr r r �recvR s zSSHSocket.recvc C s | j s| �� | | j j_| j jS r )r r2 r# �channel)r �moder r r �makefileX s
zSSHSocket.makefilec C s< | j r| j jjr
d S | j j�d� | j j�� | j �� d S )Ns
)r r$ r4 r6 r7 � terminate�r r r r �close_ s
zSSHSocket.close) �__name__�
__module__� __qualname__r r2 r: r= r> rA rD rG �
__classcell__r r r r r s
r c s& e Zd Zd� fdd� Zdd� Z� ZS )�
SSHConnectionN�< c s& t � jd|d� || _|| _|| _d S )N� localhost)�timeout)r r �
ssh_transportrO �ssh_host)r rP rO r
r r r r h s �
zSSHConnection.__init__c C sP | j r| j �� }|�| j� |�d� nt| j�}|�| j� |�� || _d S )Nr ) rP � open_session�
settimeoutrO � exec_commandr rQ r2 �sock)r rU r r r r2 p s
zSSHConnection.connect)NrM N)rH rI rJ r r2 rK r r r r rL g s rL c s2 e Zd ZdZd � fdd� Zdd� Zd d
� Z� ZS ) �SSHConnectionPoolr NrM �
c s6 t � jd||d� d | _|| _|r|�� | _|| _d S )NrN )rO �maxsize)r r rP rO �
get_transportrQ )r �
ssh_clientrO rX r
r r r r � s �
zSSHConnectionPool.__init__c C s t | j| j| j�S r )rL rP rO rQ rF r r r � _new_conn� s zSSHConnectionPool._new_connc C sj d }z | j j| j|d�}W n! ty tj�| d�� tjy. | jr,tj� | d��Y nw |p4| �
� S )N)�blockrO zPool is closed.z>Pool reached maximum size and no more connections are allowed.) �pool�getr\ �AttributeError�urllib3�
exceptions�ClosedPoolError�queue�Empty�EmptyPoolErrorr[ )r rO �connr r r � _get_conn� s �� zSSHConnectionPool._get_conn)NrM rW N)rH rI rJ �schemer r[ rg rK r r r r rV } s
rV c sb e Zd Zejjjg d� Zdejej df� fdd� Z
dd� Z dd � Z dd d �Z
� fd
d�Z� ZS )�SSHHTTPAdapter)�poolsrO rZ �
ssh_params�
max_pool_sizerM Fc sj d | _ |s| �|� | �� || _|�d�r |td�d � | _|| _|| _t|dd� d�| _ t
� � � d S )Nzssh://c S s | � � S r )rG )�pr r r �<lambda>� s z)SSHHTTPAdapter.__init__.<locals>.<lambda>)� dispose_func) rZ �_create_paramiko_client�_connectrQ �
startswith�lenrO rl �RecentlyUsedContainerrj r r )r �base_urlrO �pool_connectionsrl � shell_outr r r r � s
�zSSHHTTPAdapter.__init__c C sB t �d��t j� t�� | _tj� |�}|j
|j |j d�| _
tj�d�}tj�|�r�t�� }t|��
}|�|� W d � n1 sBw Y |�|j
�}d|v r\t�| jd �| j
d< d|v rg|d | j
d< |j d u rwd|v rw|d | j
d< |j d u r�d|v r�|d | j
d < d
|v r�|d
| j
d < | j�� | j�t�� � d S ) N�paramiko)�hostnamer �usernamez
~/.ssh/config� proxycommandrU ry r r rz � identityfile� key_filename)�logging� getLogger�setLevel�WARNINGrx � SSHClientrZ �urllib�parse�urlparsery r rz rk r( �path�
expanduser�exists� SSHConfig�open�lookup� ProxyCommand�ssh_conf�load_system_host_keys�set_missing_host_key_policy�
WarningPolicy)r ru �ssh_config_file�confr � host_configr r r rp � s8
�
�
�
z&SSHHTTPAdapter._create_paramiko_clientc C s" | j r| j jdi | j�� d S d S )Nr )rZ r2 rk rF r r r rq � s �zSSHHTTPAdapter._connectNc C s� | j st| j | j| j| jd�S | jj�7 | j�|�}|r%|W d � S | j r1| j �� s1| � � t| j | j| j| jd�}|| j|< W d � |S 1 sMw Y |S )N)rZ rO rX r
)
rZ rV rO rl rQ rj �lockr^ rY rq )r �url�proxiesr] r r r �get_connection� s2 �
��
��z SSHHTTPAdapter.get_connectionc s" t � �� | jr| j�� d S d S r )r rG rZ rF r r r rG � s
�zSSHHTTPAdapter.closer )rH rI rJ �requests�adapters� HTTPAdapter� __attrs__r �DEFAULT_NUM_POOLS�DEFAULT_MAX_POOL_SIZEr rp rq r� rG rK r r r r ri � s �
ri ) rx rc � urllib.parser� �requests.adaptersr� r~ r( r r
r+ � docker.transport.basehttpadapterr � r � http.client�client�httplib�requests.packages.urllib3�packagesr` � ImportError� _collectionsrt r �HTTPConnectionrL �connectionpool�HTTPConnectionPoolrV ri r r r r �<module> s, �O'