File: //usr/lib/python3.9/site-packages/rhn/__pycache__/transports.cpython-39.pyc
a
쑐c�| � @ sZ d dl Z d dlZd dlZd dlmZ d dlmZmZ d dlm Z d dl
m Z z d dl Z d dl
mZmZmZ W n* ey� d dlmZ eZeZeZY n0 dZdZG d d
� d
e�ZG d d � d e j�ZG d
d� de�Z G dd� de�Z G dd� de �Z G dd� d�Z d%dd�Z d&dd�Z!G dd � d �Z"G d d � d �Z#d'd d �Z$d(d!d"�Z%e#Z&G d#d$� d$�Z'dS ))� N)� connections)�sstr�bstr)�SmartIO)� UserDictCase)�IntType�
StringType�ListTypez2.8.62.8.6-8.el98.el9� c @ s e Zd ZdS )� NotProcessedN)�__name__�
__module__� __qualname__� r r �2/usr/lib/python3.9/site-packages/rhn/transports.pyr $ s r c @ s� e Zd Zde Zd'dd�Zd(dd�Zd d
� Zd d � Zd
d� Z d)dd�Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd*d d �Zd d � Zd!d"� Zd#d$� Zd%d&� ZdS )+� Transportzrhn.rpclib.py/%sr Nc C sz d| _ ddd�| _| j||d� t� | _d| _d | _d| _d | _|| _ || _
d| _ d | _ d | _
d | _d | _|| _|| _d S )NFr ��transfer�encoding�POST� @ )Z_use_builtin_types�_transport_flags�set_transport_flagsr �_headers�verbose�
connection�method�_lang�refreshCallback�progressCallback�
bufferSize�
headers_in�response_status�response_reason� _redirectedZ
_use_datetime�timeout)�selfr r r r Z use_datetimer% r r r �__init__* s" zTransport.__init__r c C s || _ || _d S �N�r r )r& r r r r r �set_progress_callback? s z Transport.set_progress_callbackc C s
|| _ d S r( )r )r& r r r r �set_refresh_callbackD s z Transport.set_refresh_callbackc C s |d u r d}|| _ d S )Nr �r )r& r r r r �set_buffer_sizeJ s zTransport.set_buffer_sizec C s |dvrt d| ��|| _d S )N)ZGETr zUnknown request method %s)�IOErrorr )r& r r r r �
set_methodR s zTransport.set_methodc K s< | j �|� |d ur || j d< |d ur0|| j d< | �� d S )Nr r )r �update�validate_transport_flags�r& r r �kwargsr r r r X s
z Transport.set_transport_flagsc C s
| j �� S r( )r �copy�r& r r r �get_transport_flagsc s z Transport.get_transport_flagsc C sH | j �d�}t|dd�}|| j d< | j �d�}t|dd�}|| j d< d S )Nr � )�strictr )r �get�lookupTransfer�lookupEncoding)r& r r r r r r1 f s
z"Transport.validate_transport_flagsc C s@ t |�t g �t d�fv r.dd� |D �| j|< nt|�| j|< d S )Nr c S s g | ] }t |��qS r )�str)�.0�ar r r �
<listcomp>t � z(Transport.set_header.<locals>.<listcomp>)�typer r< �r& �name�argr r r �
set_headerq s zTransport.set_headerc C sF || j v r&| j | }t|t�s4|g}ng }| j |< |�t|�� d S r( )r �
isinstancer �appendr< )r& rC rD Zvlistr r r �
add_headerx s
zTransport.add_headerc C s | j �� d S r( )r �clearr5 r r r �
clear_headers� s zTransport.clear_headersc C s8 | j rtd|f � | jr*tj|| jd�S t�|�S d S )NzConnecting via http to %s)r% )r �printr% r �HTTPConnection�r& �hostr r r �get_connection� s
zTransport.get_connectionc C sP || _ | �|�\}}}|s g }| �|�}|�| j� | j rJ|�| j d � t|| jd�}|jf i | j �� |�
d| j� t | j �
� �| D ]\} }
|�
| |
� q�|�
dd� |�|� dD ]} |�| � q�|�||�\} }
| j �r td� | �
� D ]\} }
td| |
f � q�|
jd v �r.| d
| _|
j| _d S | | _|
j| _|
j| _| �|
|�S ) Nr7 )r r z
User-Agent� Content-Typeztext/xml)�Content-LengthZHostzIncoming headers:z %s : %s�i- i. ZLocation)r �
get_host_inforO Zset_user_agent�
user_agentZset_debuglevel�Outputr r r rE �listr �items�process� clear_header� send_httprK �statusr$ r" r! �reasonr# �_process_response)r& rN �handlerZ request_bodyr �
extra_headers�x509r Zreq�header�value�h�headers�fdr r r �request� s<
zTransport.requestc C sb t | j| j| jd�}|�|�}t|t�rPt|j|j |j
| j| jd�}|j |_ |S |� � | � |�S )Nr) �r r )
�Inputr! r r �decoderF � InputStream�Filere �lengthrC �close�parse_response)r& re r Zresp�fr r r r] � s
�
�zTransport._process_responsec C s | j S r( )r$ r5 r r r �
redirected� s zTransport.redirectedc C sb | � � \}}|�d�}|s qJ| jr*| �� | jr>tdt|�� |�|� q |�� |�� |�� S )N� zbody:)Z getparser�readr r rK �reprZfeedrm )r& ro �p�u�responser r r rn � s
zTransport.parse_responsec C s
|| _ d S r( )r )r& �langr r r �setlang� s zTransport.setlang)r r NNNN)r )NN)r )r r
r � __version__rT r' r* r+ r- r/ r r6 r1 rE rH rJ rO rf r] rp rn rx r r r r r ' s( �
8 r c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�
SafeTransportr Nc C s8 t j| |||||d� g | _|p"g D ]}| �|� q$d S �N)r r r% )r r' �
trusted_certs�add_trusted_cert)r& r r r r r| r% �certfiler r r r' � s
� zSafeTransport.__init__c C s* t �|t j�std| ��| j�|� d S �Nz%Certificate file %s is not accessible��os�access�R_OK�
ValueErrorr| rG �r& r~ r r r r} s z SafeTransport.add_trusted_certc C sR | � |�\}}}| jr$td|f � | jr>tj|| j| jd�S tj|| jd�S d S )NzConnecting via https to %s)r| r% )r| )rS r rK r% r ZHTTPSConnectionr| )r& rN r_ r` r r r rO s ��z SafeTransport.get_connection)r r NNNN�r r
r r' r} rO r r r r rz � s
�
rz c @ s e Zd Zddd�Zdd� ZdS )�ProxyTransportNr c C s, t j| |||||d� || _|| _|| _d S r{ )r r' �_proxy�_proxy_username�_proxy_password) r& �proxy�
proxyUsername�
proxyPasswordr r r r r% r r r r' s
�zProxyTransport.__init__c C s^ | j r td|| j| j| jf � | jrBtj| j|| j| j| jd�S tj| j|| j| jd�S d S )Nz8Connecting via http to %s proxy %s, username %s, pass %s)�username�passwordr% )r� r� )r rK r� r� r� r% r ZHTTPProxyConnectionrM r r r rO # s �
�
�z ProxyTransport.get_connection)NNr r NNN)r r
r r' rO r r r r r� s
�
r� c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�SafeProxyTransportNr c
C s> t j| |||||||| d� g | _|p(g D ]}
| �|
� q*d S )N)r� r� r r r r r% )r� r' r| r} ) r& r� r� r� r r r r r| r% r~ r r r r' 0 s � zSafeProxyTransport.__init__c C s* t �|t j�std| ��| j�|� d S r r� r� r r r r} = s z#SafeProxyTransport.add_trusted_certc C sf | j r td|| j| j| jf � | jrFtj| j|| j| j| j| jd�S tj| j|| j| j| jd�S d S )Nz9Connecting via https to %s proxy %s, username %s, pass %s)r� r� r| r% )r� r� r| ) r rK r� r� r� r% r ZHTTPSProxyConnectionr| rM r r r rO B s �
�
�z!SafeProxyTransport.get_connection)NNr r NNNNr� r r r r r� / s �
r� c @ s: e Zd Zd dd�Zejfdd�Zejfdd �Zd
d � ZdS )
rh Nrq r c C s� d | _ d | _d | _d| _d| _d| _|| _|| _|| _|s�dt j
v rTt j
d � � | _ dt j
v rnt j
d � � | _dt j
v r�t j
d � � | _dt j
v r�t t j
d �| _dt j
v r�t j
d | _d t j
v r�t j
d | _n�|�
� D ]�}|| }|� � }|d
k�r"zt |�| _W n t�y d| _Y n0 q�|d k�r8|� � | _ q�|d k�rN|� � | _q�|d
k�rd|� � | _q�|dk�rz|� � | _q�|dkr�|| _q�d | _d S )Nr �C� Z HTTP_CONTENT_TRANSFER_ENCODINGZHTTP_CONTENT_ENCODINGz CONTENT-TYPEZCONTENT_LENGTHZHTTP_ACCEPT_LANGUAGEZHTTP_X_PACKAGE_FILENAMEzcontent-lengthzcontent-transfer-encodingzcontent-encodingz content-typezaccept-languagezx-package-filename)r r rA rl rw rC r r � max_mem_sizer� �environ�lower�int�keysr� �io)r& rd r r r� ra rb rc r r r r' h sR
zInput.__init__c C s� | j dkrd S | jr2t|| j| j| j| jd�| _n
t|�| _| jrL| jdkrPd S | jdkr�dd l }| j}|�
dd� t | jd�| _|� || j� n
t
| j��d S )N�application/octet-stream)r r r� �binary�base64r �r� )rA rl � _smart_readr r r� r� �_smart_total_readr r� �seekr ri �NotImplementedError)r& re r� Zold_ior r r rr � s$
�
z
Input.readc C s\ | j dkr t|| j| j|jd�S | js0| �|� t|d�rB|�� | jrR| jdkrTn�| jdv r�dd l }|�
� }| j� dd� |� | j�� �|�
� }~t|�| _t| jd�| _| j�|� n�| jdv �r dd l}| j� dd� |jd t| jd
�}|�� }t|�| _t| jd�| _| j�|� n*| jd k�r:t| j| j��nt| j| j��| j� dd� | jS ) Nr� �rm rm �__plain)�x-zlib�deflater r� )�x-gzip�gzip�rb��mode�
compresslevelZfileobj�x-gpg)rA rj rl rC rm r� rr �hasattrr �zlib�
decompressobjr� �
decompress�flush�lenr r� �writer� �GzipFile�COMPRESS_LEVELr� r )r& re r� �obj�datar� �gzr r r ri � s@
�
z Input.decodec C s | j S r( )rw r5 r r r �getlang� s z
Input.getlang)NNrq r ) r r
r r' �sys�stdinrr ri r� r r r r rh g s �
80rh rq r c C s* t |d�}| �|�}|sq&|�|� q
|S )z�
Tries to read data from the supplied stream, and puts the results into a
StmartIO object. The data will be in memory or in a temporary file,
depending on how much it's been read
Returns a SmartIO object
r� )r rr r� )re r r� r� �chunkr r r r� � s
r� c C s� t � � }|}t|d�}|}|dkr�t � � } t||�}
| �|
�} t| �}
|
sLq�||
}|�| � |d u rhq| | dks||dkr| }t|| �} | | | d }
|dkr�d}n||
}|| ||
|� q|�dd� |S )Nr� r r7 g���ư>)�timer �minrr r� r� �floatr� )re �amtr r r� Z startTimeZlastTime�bufZorigsizeZcurTime�lr� Z bytesReadZspeedZsecsr r r r� s2
r� c @ s e Zd Zddd�Zdd� ZdS )rj � <unknown>Nc C s || _ t|�| _|| _|| _d S r( )re r� rl rC rm )r& re rl rC rm r r r r' = s
zInputStream.__init__c C s d| j | jf S )Nz0Input data is a stream of %d bytes for file %s.
)rl rC r5 r r r �__repr__C s zInputStream.__repr__)r� N)r r
r r' r� r r r r rj < s
rj c @ s� e Zd ZdZdZdZdZdZdZdZ ddgddgd d
gd ggZ
g d �Z d dd�Z dd� Z
dd� Zdd� Zd dd�Zd!dd�Zdd � Zd d � ZdS )"�
BaseOutputr r7 � � Nr� r� r� r� r� r� )Nr� r� r c C s� |rt |tj�std��|| _|| _d | _t� | _d| _ d| _
i | _ d | _ d | _
d | _d | _d | _d | _| j||d� d| _d S )Nz&Expected an HTTPConnection type objectr r )rF r rL � Exceptionr � _connectionr� r rd r r �transport_flagsr� r� �_hostZ_handlerZ
_http_typeZ _protocolr �_BaseOutput__processed)r& r r r r r r r r' c s$ zBaseOutput.__init__c C sB t |�t g �t d�fv r0d�tt|��| j|< nt|�| j|< d S )Nr �,)rA �join�mapr< rd rB r r r rE � s zBaseOutput.set_headerc C s || j v r| j |= d S r( )rd )r& rC r r r rY � s
zBaseOutput.clear_headerc C s� || _ | j| jkr�dd l}| j| j d }| �d|� tdd�}|jdt|d�}t j
d dkrn|� t |�� n|� t
|�� |�� |�� | _ |�� n�| j| jkr�dd l}| j| j d }| �d|� |�t�}|�|�|�� | _ n8| j| jk�r t| j| j��| j| j d }| �d|� | j| jk�rT| j| j }| �d|� | �d d
� nH| j| jk�r�dd l} | j| j }| �d|� | �d d � | � | j �| _ | �d t | j �� t }
t t � � �dk�r�t � � d }
| �d
d|
� d| _ d S )Nr zContent-Encodingr7 )Z force_mem�wbr� r� zContent-Transfer-EncodingrP zapplication/binaryz text/base64rQ zX-Transport-Infoz=Extended Capabilities Transport (C) Red Hat, Inc (version %s))!r� r � ENCODE_GZIPr� � encodingsrE r r� r� r� � version_infor� r r rm �getvalue� ENCODE_ZLIBr� � compressobj�compressr� �
ENCODE_GPGr� r �TRANSFER_BINARY� transfers�TRANSFER_BASE64r� Z encodestringr� ry �splitr� ) r& r� r� Z
encoding_namero r� r� r� Z
transfer_namer� Z rpc_versionr r r rX � sZ
�
��zBaseOutput.processc K s || _ || _| j�|� d S r( )r r r� r0 r2 r r r r � s z BaseOutput.set_transport_flags�/RPC2c C s� | j s
t�|| _| jd u r"td��| j�� | jj| j|t| j �| j
d� | j� � }| � |�s~t
�d| j|f |j|j|j��|j|fS )NzNo connection object found)�bodyrd z%s %s)r� r r� r� r� Zconnectrf r r r� rd Z getresponse�response_acceptable� xmlrpclibZ
ProtocolErrorr[ r\ �msg)r& rN r^ rv r r r rZ � s
� �zBaseOutput.send_httpc C sP |j dkrdS |j dv r dS |j dkr*dS | j�d�s:dS |jd dkrLdS dS ) z*Returns true if the response is acceptable�� r7 rR �� r Zallow_partial_contentrP r� )r[ r� r9 r� )r& rv r r r r� � s
z BaseOutput.response_acceptablec C s | j r| j �� d | _ d S r( )r� rm r5 r r r rm � s
zBaseOutput.close)r r Nr )r r )r� )r r
r Z ENCODE_NONEr� r� r� Z
TRANSFER_NONEr� r� r� r� r' rE rY rX r rZ r� rm r r r r r� I s* �
8
r� c C s� | du r dS t | t�r8d| kr0ttj�k r8n n| S t | t�rptttj��D ] }tj| | �� krP| S qP|r�td| ��dS )zTGiven a string or numeric representation of a transfer, return the
transfer codeNr zUnsupported transfer %s) rF r r� rU r� r �ranger� r� )r r8 �ir r r r: s (
r: c C s� | du r dS t | t�r8d| kr0ttj�k r8n n| S t | t�rptttj��D ] }| �� tj| v rP| S qP|r�td| ��dS )zUGiven a string or numeric representation of an encoding, return the
encoding codeNr zUnsupported encoding %s) rF r r� rU r� r r� r� r� )r r8 r� r r r r; s (
r; c @ s@ e Zd Zddd�Zdd� Zddd �Zd
d � Zd d
� Zdd� ZdS )rk r Nr c C sF || _ || _|j| _|| _d| _|r<||�d�d d � | _|| _d S )Nr� �/r7 )rl �file_objrm r rC �rfindr )r& r� rl rC r r r r r r' & s z
File.__init__c C s | j S r( )rl r5 r r r �__len__1 s z File.__len__c C s$ |d u r| � � }|�� S | j�|�S r( )� _get_filerr r� )r& r� re r r r rr 4 s z File.readc C sL | � � }|�| j�}|sqHtjd dkr8|�t|�� q|�t|�� q|S )zHCopies the contents of this File object into another file
objectr r� )r� rr r r� r� r� r r )r& �filere r� r r r � read_to_file< s zFile.read_to_filec C sL | j r,t| j| j | j| jd�}|�dd� nt| j| jd�}|�dd� |S )z�Read everything into a temporary file and call the progress
callbacks if the file length is defined, or just reads till EOFrg r r, )rl r� r� r r r� r� )r& r� r r r r� J s
� zFile._get_filec C s | j r| � � d | _ d S r( r� r5 r r r �__del__X s z File.__del__)r NNr )N) r r
r r' r� rr r� r� r� r r r r rk % s �
rk )rq r )rq Nr )r )r )(r� r� r� Zrhnr Zrhn.i18nr r Z rhn.SmartIOr Zrhn.UserDictCaser r� �typesr r r � ImportErrorZ
xmlrpc.clientZclientr� �bytesrV ry r� r� r r rz r� r� rh r� r� rj r� r: r; rU rk r r r r �<module>
sF
V 8
�
<
8