Edit File: connections.cpython-36.pyc
3 �c�% � @ s& d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlm Z y(d dl Z d dl m Z m Z m Z d dlZW n< ek r� d dljZ d dlm Z m Z m Z d dljZY nX 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 tj| j�| _| jj|||||� d S )N)� isinstance�fpr ZNonBlockingFile� set_callback)�self�rs�ws�ex� user_data�callback� r �!/usr/lib/python3.6/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� ntjj| ||� 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 )N)r r r r r )r r r r r r r r r r 5 s zHTTPConnection.set_callbackc C s || _ d S )N)r! )r Z user_agentr r r �set_user_agent= s z HTTPConnection.set_user_agentc C s� | j r| j j� rd| _ | jtks&| j r,t� �| jdkrH| j| j| j�}n | j| j�}| jrv|j | j | j | j | j | j� |j� |jtjks�t�t| _|jr�| j� 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_UNKNOWN�AssertionErrorr �close)r �responser r r � getresponseB s" zHTTPConnection.getresponsec C s t jj| � | jj| j� d S )N)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| j | _| _|| _ || _ d S )N)r ) 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| j� tj| � | j | j| j� d S )N)r3 r1 r2 r r- r4 r5 )r r r r r- � s zHTTPProxyConnection.connectr c C sN | j }| j| jkr$|d t| j� }d||f }tj| |||d� | j� d S )N�:z http://%s%s)� skip_host)r4 r5 � default_port�strr � putrequest�_add_proxy_headers)r �method�urlr< �hostnameZnewurlr r r r? � s z HTTPProxyConnection.putrequestc C sX | j s d S d| j | jf }tjtj|��jtjd�tjd��}| jddtj|� � d S )Nz%s:%s� � zProxy-AuthorizationzBasic %s) r6 r7 �base64Z encodestringr �bstr�replace� putheaderZsstr)r ZuserpassZ enc_userpassr r r r@ � s $z&HTTPProxyConnection._add_proxy_headersc C s | j ||�\| _| _d S )N)Z _get_hostportr" r# )r r" r# r r r r3 � s z!HTTPProxyConnection._set_hostport)r ) r r r r r/ r r- r? r@ r3 r r r r r0 x s r0 c @ s2 e Zd ZeZejjZddej fdd�Z dd� Z dS )�HTTPSConnectionNc C s$ t j| |||d� |pg }|| _d S )N)r )r r � trusted_certs)r r" r# rK r r r r r � s zHTTPSConnection.__init__c C s� t j| j| jt jt j�}x�|D ]�}|\}}}}}yt j |||�}W n t jk r^ d}w Y nX y"|j| j| jf� |j| j � W n$ t jk r� |j � d}w Y nX P q W |dkr�t jd��t j || j �| _| jj� 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 � SSLSocketrK r& �init_ssl) r �results�rZafZsocktype�protoZ canonnameZsar& r r r r- � s* zHTTPSConnection.connect) r r r r r% r rJ r= r r/ r r- r r r r rJ � s rJ c @ s e Zd Zdd� ZdS )�HTTPSProxyResponsec C s t j| � d| _d S )Nr )r r' r( )r r r r r' � s zHTTPSProxyResponse.beginN)r r r r' r r r r rS � s rS c @ sB e Zd ZejZddddejfdd�Zdd� Zd dd�Z d d � Z dS ) �HTTPSProxyConnectionNc C s* t j| ||||||d� |p g }|| _d S )N)r )r0 r rK )r r8 r" r# r9 r: rK r r r r r � s z HTTPSProxyConnection.__init__c C s� t j| � d| j| jf }tj| d|� | j� tj| � | j}t | _tj | �}|| _|j � |j dkr�| j � t j||j |j|j��tj| j| j�| _| jj� d S )Nz%s:%sZCONNECT�� )r0 r- r4 r5 r r? r@ Z endheadersr% rS r, r* Zstatus� xmlrpclibZ ProtocolError�reason�msgr rN r& rK rO )r r" r% r+ r r r r- � s z HTTPSProxyConnection.connectr c C s t j| |||d�S )N)r< )r r? )r rA rB r< r r r r? � s z HTTPSProxyConnection.putrequestc C s t j| � | jd| j� d S )Nz User-Agent)r0 r@ rI r! )r r r r r@ � s z'HTTPSProxyConnection._add_proxy_headers)r ) r r r rJ r= r r/ r r- r? r@ r r r r rT � s rT c C s$ | dkr dS t j| �} | jd�S dS )zJ Convert Internationalized domain name from Punycode (RFC3492) to Unicode N�idna)r rG �decode)rC r r r �idn_puny_to_unicode� s r[ c C s* | dkr dS t j| �} t j| jd��S dS )zZ Convert domain name to Punycode (RFC3492). Hostname can be instance of string or Unicode NrY )r Zustr�encode)rC r r r �idn_ascii_to_puny s r] )rF Zencodings.idnaZ encodingsrL �platformr Zrhnr r r r r r r rV � ImportErrorZ http.clientZclientZ xmlrpc.clientr r r0 rJ rS rT r[ r] r r r r �<module> s, S,&.