File: //proc/self/root/usr/lib/python3.9/site-packages/cockpit/channels/__pycache__/stream.cpython-39.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 � � � t |t
j��s:J �|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�asyncio� Transport) r r r �pathZlabel� transport�_r �host�errorZproblemr r r �create_transport" s2
"�
(z$SocketStreamChannel.create_transportN) �__name__�
__module__� __qualname__�payloadr- �AbstractEventLoopr r. r4 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< t | jt�sJ �d| j�� i}| j�� }|d ur8||d< |S )Nz exit-statusr )r) �
_transportr Zget_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 rA 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)�defaultrA �environrD rE 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 )rG rA �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�osrI �update�
ValueErrorr r Zget_pidZ _ready_infor% r& �FileNotFoundError�PermissionErrorr' r( r+ )r r r r> rC rQ rG rA rI r? rP r0 rR r3 r r r r4 Y s:
z(SubprocessStreamChannel.create_transport)
r5 r6 r7 r8 Z restrictionsr< r r@ rB r- r9 r r4 r r r r r: E s r: )r- ZloggingrV rS �typingr Zchannelr r Zjsonutilr r r r r
r r r
Z
transportsr r r Z getLoggerr5 r% r r: r r r r �<module> s (
&