관리-도구
편집 파일: stream.cpython-39.pyc
a ��gJ � @ 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 ( &