File: //usr/lib/python3.9/site-packages/cockpit/channels/__pycache__/http_channel.cpython-39.pyc
a
�/�h� � @ sn d dl Zd dlZd dlZd dlZddlmZmZ ddlm Z m
Z
m Z m Z m
Z
mZ e�e�ZG dd� de�ZdS )� N� )� AsyncChannel� ChannelError)�
JsonObject�get_dict�get_int�
get_object�get_str� typecheckedc @ s� e Zd ZdZeejjee d�dd��Z
ee ejj d�dd��Z eejj dd d
�d d ��Z
eejj eed
eejjd�dd��Ze d d�dd�Zd S )� HttpChannelz http-stream2)�response�binary�returnc s0 ddh� |s� � ddh� � fdd�| �� D �S )NZ
ConnectionzTransfer-EncodingzContent-LengthZRangec s i | ]\}}|� vr||�qS � r )�.0�key�value��remover �A/usr/lib/python3.9/site-packages/cockpit/channels/http_channel.py�
<dictcomp>) � z+HttpChannel.get_headers.<locals>.<dictcomp>)�updateZ
getheaders)r r
r r r � get_headers s zHttpChannel.get_headers)�optionsr c C s* t | dd�}t| dd �}t | dd �}t| dd �}|d urL|d urLtddd��|d u rh|d u rhtdd d��|d ur�|d ur�tdd
d��|d u�rt|d d �}|d ur�t |d d �}|d ur�tj|d
�}q�tjt |d�d�}nt�� }d|v �r|d �sd|_tjj|_ t
j j |||d�S t
j j
||d�S d S )N�addressZ localhostZtls�unix�port�protocol-errorz#TLS on Unix socket is not supported��messagez&no "port" or "unix" option for channelz-cannot specify both "port" and "unix" options� authority�data)Zcadata�file)ZcafileZvalidateF)r �context)r )r r r r �sslZcreate_default_contextZcheck_hostnameZ
VerifyModeZ CERT_NONEZ verify_mode�http�clientZHTTPSConnection�HTTPConnection)r Z opt_addressZopt_tls�opt_unixZopt_portr! r" r$ r r r �
create_client+ s.
zHttpChannel.create_clientz
str | NoneN)�
connectionr) r c C s, |r t � t j�| _| j�|� n| �� d S )N)�socketZAF_UNIXZsock�connect)r+ r) r r r r- N s zHttpChannel.connectzdict[str, str])r+ �method�path�headers�bodyr c C s | j |||pi |d� | �� S )N)r0 r1 )�requestZ getresponse)r+ r. r/ r0 r1 r r r r2 Y s zHttpChannel.requestc
� sV t �d|� t|d�}t|d�}t|ddd� d �}d|v rFtdd d
��| �|�}| �� d }| �� I d H }|d u rtq~||7 }q\z"| �| j |t|d d ��I d H W nr t
j y� } z td
t |�d
�|�W Y d }~nDd }~0 t
tf�y } z tdt |�d
�|�W Y d }~n
d }~0 0 z&| �| j||||�p,i |�I d H } W n@ tjjt
f�yz } z tdt |�d
�|�W Y d }~n
d }~0 0 | jd| j| j| j| | jd�d� zj| �| j| j�I d H }
|
�s��q�| �|
�I d H �q�t �d� | �� }
|
d k�s�J �| �|j�I d H W n@ tjjt
f�yH } z tdt |�d
�|�W Y d }~n
d }~0 0 | � � d S )Nzopen %sr. r/ r0 c S s dd� | � � D �S )Nc S s i | ]\}}|t |t��qS r )r
�str)r �k�vr r r r f r z5HttpChannel.run.<locals>.<lambda>.<locals>.<dictcomp>)�items)�dr r r �<lambda>f r z!HttpChannel.run.<locals>.<lambda>r+ r z4connection sharing is not implemented on this bridger r r zunknown-hostkeyz not-foundZ
terminatedr )r
)Zcommand�status�reasonr0 zreading response done) �logger�debugr r r r* Zready�readZ in_threadr- r% ZSSLCertVerificationErrorr3 �OSError�IOErrorr2 r& r' Z
HTTPExceptionZ send_controlr9 r: r Z is_binary�read1Z
BLOCK_SIZE�write�closeZdone) �selfr r. r/ r0 r+ r1 r" �excr �blockr r r �runa sP
"&(&(�
(zHttpChannel.run)�__name__�
__module__� __qualname__Zpayload� staticmethodr& r' Z HTTPResponse�boolr r r( r* r- r3 �bytesr2 rF r r r r r s
"
�r )Z http.clientr& Zloggingr, r% Zchannelr r Zjsonutilr r r r r r
Z getLoggerrG r; r r r r r �<module> s