File: //usr/lib/python3.9/site-packages/cockpit/channels/__pycache__/stream.cpython-39.opt-1.pyc
a
�/�hJ � @ s� d dl Z d dlZd dlZd dlZd dlmZ ddlmZmZ ddl m
Z
m Z m Z m
Z
mZmZmZmZ ddlmZmZmZ e�e�ZG dd� de�ZG d d
� d
ee�ZdS ) � N)�Dict� )� ChannelError�ProtocolChannel)�JsonDict�
JsonObject�get_bool�get_enum�get_int�
get_object�get_str�get_strv)�SubprocessProtocol�SubprocessTransport�
WindowSizec @ s& e Zd ZdZejeejd�dd�ZdS )�SocketStreamChannel�stream��loop�options�returnc
� s, d|v r d|v r t ddd��z�d|v rZt|d�}d|� �}|�� fdd�|�I d H \}}n\d|v r�t|d�}t|d d
�}d |� d |� �}|�� fd
d�||�I d H \}}n t ddd��t�d|� W nZ t�y } z@t�d|| � t | t
�r�d}
nd}
t |
t | �d�| �W Y d } ~ n
d } ~ 0 0 � � � |S )NZunix�port�protocol-errorz-cannot specify both "port" and "unix" options��messagez Unix socket c s � S �N� r ��selfr �;/usr/lib/python3.9/site-packages/cockpit/channels/stream.py�<lambda>+ � z6SocketStreamChannel.create_transport.<locals>.<lambda>�addressZ localhostz TCP socket �:c s � S r r r r r r r 3 r! z7no "port" or "unix" or other address option for channelz$SocketStreamChannel: connected to %sz0SocketStreamChannel: connecting to %s failed: %s� not-foundZ
terminated)
r r Zcreate_unix_connectionr
Zcreate_connection�logger�debug�OSError�info�
isinstance�ConnectionRefusedError�str� close_on_eof) r r r �pathZlabel� transport�_r �host�errorZproblemr r r �create_transport" s0
"�
(z$SocketStreamChannel.create_transportN) �__name__�
__module__� __qualname__�payload�asyncio�AbstractEventLoopr Z Transportr2 r r r r r s r c @ sL e Zd ZdZdZdd�dd�Zed�dd�Zd d
� Ze j
ee d �d d
�Z dS )�SubprocessStreamChannelr ))�spawnNN)r c C s | � � d S r )r, r r r r �process_exitedI s z&SubprocessStreamChannel.process_exitedc C s, d| j �� i}| j �� }|d ur(||d< |S )Nz exit-statusr )�
_transportZget_returncodeZ
get_stderr)r �args�stderrr r r �_get_close_argsL s
z'SubprocessStreamChannel._get_close_argsc C s& t |dtd �}|d ur"| j�|� d S )N�window)r r r< Zset_window_size)r r r@ r r r �
do_optionsT s z"SubprocessStreamChannel.do_optionsr c
� s� t |d�}t|dg d�d�}t|dd�}t|ddd �}t|d
td �}t |d g �}|d kr`tj} n|d
krptj} ntj } t
t j �}
z |
�
t
dd� |D ��� W n ty� tddd�d �Y n0 z>t|| ||||
|| d�} | �� } d| i| _t�d|| � | W S t�y, }
ztd�|
�W Y d }
~
nxd }
~
0 t�yZ }
ztd�|
�W Y d }
~
nJd }
~
0 t�y� }
z(t�d|t|
�� td�|
�W Y d }
~
n
d }
~
0 0 d S )Nr: �err)�out�ignorer r Z directory�.�ptyF)�defaultr@ �environrC rD c s s | ]}|� d d�V qdS )�=� N)�split)�.0�er r r � <genexpr>j r! z;SubprocessStreamChannel.create_transport.<locals>.<genexpr>r z+invalid "environ" option for stream channelr )rF r@ �env�cwdr> �pidz Spawned process args=%s pid=%ir$ z
access-deniedzFailed to spawn %s: %szinternal-error)r
r r r r r �
subprocessZSTDOUTZDEVNULL�PIPE�dict�osrH �update�
ValueErrorr r Zget_pidZ _ready_infor% r& �FileNotFoundError�PermissionErrorr' r( r+ )r r r r= rB rP rF r@ rH r> rO r. rQ r1 r r r r2 Y s:
z(SubprocessStreamChannel.create_transport)
r3 r4 r5 r6 Z restrictionsr; r r? rA r7 r8 r r2 r r r r r9 E s r9 )r7 ZloggingrU rR �typingr Zchannelr r Zjsonutilr r r r r
r r r
Z
transportsr r r Z getLoggerr3 r% r r9 r r r r �<module> s (
&