File: //lib/python3.9/site-packages/rhn/__pycache__/connections.cpython-39.pyc
a
ꑐc�% � @ s( d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlm Z z(d dl
Z
d dl
m Z m Z m
Z
d dlZW n> ey� d dlmZ
d dlm Z m Z m
Z
d dlmZ Y n0 G dd� de
j�ZG d d
� d
e
j�ZG d d � d e�ZG d
d� de�ZG dd� de�ZG dd� de�Zdd� Zdd� ZdS )� N)�python_version)�SSL)� nonblocking)�i18n)� _CS_REQ_SENT�_CS_IDLE�ResponseNotReadyc @ s e Zd Zdd� ZdS )� HTTPResponsec C s4 t | jtj�s t�| j�| _| j�|||||� d S �N)�
isinstance�fpr ZNonBlockingFile� set_callback��selfZrs�ws�exZ user_data�callback� r �3/usr/lib/python3.9/site-packages/rhn/connections.pyr
s zHTTPResponse.set_callbackN)�__name__�
__module__� __qualname__r
r r r r r s r c @ s@ e Zd ZeZdejfdd�Zdd� Zdd� Z dd � Z
d
d � Z dS ) �HTTPConnectionNc C s^ t � dkr tjj| |||d� ntj�| ||� g | _g | _g | _d | _d | _d| _ || _
d S )Nz2.6.1��timeoutz#rhn.connections $Revision$ (python)) r �httplibr �__init__�_cb_rs�_cb_ws�_cb_ex�
_cb_user_data� _cb_callback� _user_agentr )r �host�portr r r r r ( s
zHTTPConnection.__init__c C s" || _ || _|| _|| _|| _d S r
)r r r r r! r r r r r
5 s
zHTTPConnection.set_callbackc C s
|| _ d S r
)r" )r Z
user_agentr r r �set_user_agent= s z HTTPConnection.set_user_agentc C s� | j r| j �� rd| _ | jtks&| j r,t� �| jdkrH| �| j| j�}n | �| j�}| jrv|� | j
| j | j | j
| j� |�� |jtjks�J �t| _|jr�| �� n|| _ |S )z!Get the response from the server.Nr )Z_HTTPConnection__responseZisclosedZ_HTTPConnection__stater r Z
debuglevel�response_class�sockr! r
r r r r �begin�
will_closer Z_UNKNOWNr �close)r �responser r r � getresponseB s$
�
zHTTPConnection.getresponsec C s t j�| � | j�| j� d S r
)r r �connectr' �
settimeoutr �r r r r r- s s zHTTPConnection.connect) r r r r r& r �DEFAULT_TIMEOUTr r
r% r, r- r r r r r % s
1r c @ sB e Zd Zdddejfdd�Zdd� Zd
dd�Zd d
� Zd d � Z dS )�HTTPProxyConnectionNc C sP t j| ||d� | j| j | _| _| �||� | j| j | _| _|| _ || _
d S �Nr ) r r r# r$ �_HTTPProxyConnection__proxy� _HTTPProxyConnection__proxy_port�
_set_hostport�_host�_port� _HTTPProxyConnection__username� _HTTPProxyConnection__password)r �proxyr# r$ �username�passwordr r r r r y s z HTTPProxyConnection.__init__c C s. | � | j| j� t�| � | � | j| j� d S r
)r5 r3 r4 r r- r6 r7 r/ r r r r- � s
zHTTPProxyConnection.connectr c C sN | j }| j| jkr$|d t| j� }d||f }tj| |||d� | �� d S )N�:z http://%s%s�� skip_host)r6 r7 � default_port�strr �
putrequest�_add_proxy_headers)r �method�urlr? �hostnameZnewurlr r r rB � s z HTTPProxyConnection.putrequestc C sX | j s
d S d| j | jf }t�t�|���t�d�t�d��}| �ddt�|� � d S )N�%s:%s�
� zProxy-AuthorizationzBasic %s) r8 r9 �base64Z encodestringr �bstr�replace� putheaderZsstr)r ZuserpassZ enc_userpassr r r rC � s
$z&HTTPProxyConnection._add_proxy_headersc C s | � ||�\| _| _d S r
)Z
_get_hostportr# r$ )r r# r$ r r r r5 � s z!HTTPProxyConnection._set_hostport)r )
r r r r r0 r r- rB rC r5 r r r r r1 x s �
r1 c @ s2 e Zd ZeZejjZddej fdd�Z
dd� Z dS )�HTTPSConnectionNc C s$ t j| |||d� |pg }|| _d S r2 )r r �
trusted_certs)r r# r$ rO r r r r r � s zHTTPSConnection.__init__c C s� t �| j| jt jt j�}|D ]�}|\}}}}}zt � |||�}W n t jy\ d}Y q Y n0 z"|�| j| jf� |�| j � W n$ t jy� |�
� d}Y q Y n0 q�q |du r�t �d��t � || j
�| _| j�� dS )z'Connect to a host on a given (SSL) portNz0Unable to connect to the host and port specified)�socketZ getaddrinfor# r$ Z AF_UNSPECZ SOCK_STREAM�errorr- r. r r* r � SSLSocketrO r' �init_ssl) r �results�r�afZsocktype�protoZ canonnameZsar' r r r r- � s, �
zHTTPSConnection.connect) r r r r r& r rN r@ r r0 r r- r r r r rN � s �
rN c @ s e Zd Zdd� ZdS )�HTTPSProxyResponsec C s t �| � d| _d S )Nr )r r( r) r/ r r r r( � s
zHTTPSProxyResponse.beginN)r r r r( r r r r rX � s rX c @ sB e Zd ZejZddddejfdd�Zdd� Zd dd�Z d d
� Z
dS ) �HTTPSProxyConnectionNc C s* t j| ||||||d� |p g }|| _d S r2 )r1 r rO )r r: r# r$ r; r< rO r r r r r � s
�z HTTPSProxyConnection.__init__c C s� t �| � d| j| jf }t�| d|� | �� t�| � | j}t | _t�
| �}|| _|� � |j dkr�| � � t
�||j |j|j��t�| j| j�| _| j�� d S )NrG ZCONNECT�� )r1 r- r6 r7 r rB rC Z
endheadersr& rX r, r* �status� xmlrpclibZ
ProtocolError�reason�msgr rR r' rO rS )r r# r& r+ r r r r- � s"
�z HTTPSProxyConnection.connectr c C s t j| |||d�S )Nr> )r rB )r rD rE r? r r r rB � s z HTTPSProxyConnection.putrequestc C s t �| � | �d| j� d S )Nz
User-Agent)r1 rC rM r" r/ r r r rC � s
z'HTTPSProxyConnection._add_proxy_headers)r ) r r r rN r@ r r0 r r- rB rC r r r r rY � s �
rY c C s$ | du r dS t �| �} | �d�S dS )zJ Convert Internationalized domain name from Punycode (RFC3492) to Unicode N�idna)r rK �decode�rF r r r �idn_puny_to_unicode� s
rb c C s* | du r dS t �| �} t �| �d��S dS )zZ Convert domain name to Punycode (RFC3492). Hostname can be instance of string or Unicode Nr_ )r Zustr�encodera r r r �idn_ascii_to_puny s
rd )rJ Zencodings.idna� encodingsrP �platformr Zrhnr r r r r r r r\ � ImportErrorZ http.clientZclientZ
xmlrpc.clientr r r1 rN rX rY rb rd r r r r �<module>
s, S,&.