Your IP : 216.73.216.196
B
YI~da � @ sf d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z y d dl
Z
W n ek
rx dZ
Y nX ej�e�Zej�ed�Zej�ed�ZG dd� dejj�ZG dd� dej�ZG d d
� d
�ZG d d � d ejj �Z G d
d� dejj �Z G dd� de j �Z G dd� de j �Z!dd� Z"G dd� de j �Z#da$dd� Z%dd� Z&e'dk�rbe �(� dS ) � N)�supportz keycert.pemz keycert2.pemc @ s e Zd Zdd� Zdd� ZdS )�LoopbackHttpServerc C s" t jj�| ||� | j�d� d S )Ng�������?)�http�server�
HTTPServer�__init__�socket�
settimeout)�selfZserver_addressZRequestHandlerClass� r �?/opt/alt/python37/lib64/python3.7/test/test_urllib2_localnet.pyr s
zLoopbackHttpServer.__init__c C s | j �� \}}|�d� ||fS )Ng $@)r Zacceptr )r
�requestZclient_addressr r r � get_request) s
z LoopbackHttpServer.get_requestN)�__name__�
__module__� __qualname__r r r r r r r s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�LoopbackHttpServerThreadc C s< t j�| � d| _t �� | _d|_td|�| _| jj | _
d S )NFzHTTP/1.0)z 127.0.0.1r ) � threading�Threadr � _stop_serverZEvent�readyZprotocol_versionr �httpdZ server_port�port)r
�request_handlerr r r r 8 s
z!LoopbackHttpServerThread.__init__c C s d| _ | �� | j�� d S )NT)r �joinr Z server_close)r
r r r �stopA s z LoopbackHttpServerThread.stopc C s$ | j �� x| js | j�� q W d S )N)r �setr r �handle_request)r
r r r �runI s
z LoopbackHttpServerThread.runN)r r r r r r r r r r r 5 s r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�DigestAuthHandlerc C s" d| _ g | _i | _d| _d| _d S )Nr z
Test Realm�auth)� _request_num�_nonces�_users� _realm_name�_qop)r
r r r r S s
zDigestAuthHandler.__init__c C s
|| _ d S )N)r% )r
Zqopr r r �set_qopZ s zDigestAuthHandler.set_qopc C s
|| _ d S )N)r# )r
Zusersr r r � set_users] s zDigestAuthHandler.set_usersc C s
|| _ d S )N)r$ )r
Zrealmr r r � set_realma s zDigestAuthHandler.set_realmc C s8 | j d7 _ t�t| j ��d���� }| j�|� |S )N� �ascii)r! �hashlib�md5�str�encode� hexdigestr" �append)r
�noncer r r �_generate_nonced s z!DigestAuthHandler._generate_noncec C s� |� d�}||d d � }|�d�}i }xX|D ]P}|�d�\}}|�� }|d dkrn|d dkrn|dd� }n|�� }|||< q.W |S )N� r) �,�=r �"���)�find�split�strip)r
Zauth_strZfirst_space_index�parts� auth_dict�part�name�valuer r r �_create_auth_dictj s
z#DigestAuthHandler._create_auth_dictc C s� i }|� |� ||d< ||d< ||d< d| }t�|�d���� }d| }t�|�d���� } ||d< | |d< d | }
t�|
�d���� } | |d
kS ) N�password�method�uriz#%(username)s:%(realm)s:%(password)sr* z%(method)s:%(uri)s�HA1�HA2z3%(HA1)s:%(nonce)s:%(nc)s:%(cnonce)s:%(qop)s:%(HA2)s�response)�updater+ r, r. r/ ) r
r< rA rB rC Z
final_dictZHA1_strrD ZHA2_strrE Z response_strrF r r r �_validate_auth{ s
z DigestAuthHandler._validate_authc C sN |� dd� |�dd� |�dd| j| j| �� f � |�� |j�d� dS ) Ni� z Proxy Authentication Requiredz Content-Typez text/htmlzProxy-Authenticatez(Digest realm="%s", qop="%s",nonce="%s", s Proxy Authentication Required.F)�
send_response� send_headerr$ r% r2 � end_headers�wfile�write)r
r r r r �_return_auth_challenge� s z(DigestAuthHandler._return_auth_challengec C s� t | j�dkrdS d|jkr&| �|�S | �|jd �}|d | jkrT| j|d }n
| �|�S |�d�| jkrx| �|�S | j�|d � d}x*|j|j gD ]}| �
|||j |�r�d}q�W |s�| �|�S dS d S )Nr TzProxy-AuthorizationZusernamer1 F) �lenr# �headersrN r@ �getr" �remove�path�
short_pathrH Zcommand)r
r r< rA Zauth_validatedrS r r r r � s,
z DigestAuthHandler.handle_requestN) r r r r r& r' r( r2 r@ rH rN r r r r r r P s r c @ sb e Zd ZdZdZdZdeef Ze�e� d���
d�Z dd� Z dd � Z
d
d � Zd d
� Zdd� ZdS )�BasicAuthHandler�testUser�testPass�Testz%s:%sr* c O s t jjj| f|�|� d S )N)r r �BaseHTTPRequestHandlerr )r
�args�kwargsr r r r � s zBasicAuthHandler.__init__c G s d S )Nr )r
�formatrZ r r r � log_message� s z BasicAuthHandler.log_messagec C s" | � d� | �dd� | �� d S )N�� z Content-typez text/html)rI rJ rK )r
r r r �do_HEAD� s
zBasicAuthHandler.do_HEADc C s4 | � d� | �dd| j � | �dd� | �� d S )Ni� zWWW-AuthenticatezBasic realm="%s"z Content-typez text/html)rI rJ �REALMrK )r
r r r � do_AUTHHEAD� s
z BasicAuthHandler.do_AUTHHEADc C sh | j �dd�s$| �� | j�d� n@| j �dd�d| j kr\| �d� | �� | j�d� n| �� d S )NZ
Authorization� s No Auth header receivedzBasic r^ s It works)rP rQ ra rL rM � ENCODED_AUTHrI rK )r
r r r �do_GET� s
zBasicAuthHandler.do_GETN)r r r �USER�PASSWDr` Z USER_PASSWD�base64Z b64encoder. �decoderc r r] r_ ra rd r r r r rU � s rU c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�FakeProxyHandlerc O s || _ tjjj| f|�|� d S )N)�digest_auth_handlerr r rY r )r
rj rZ r[ r r r r � s zFakeProxyHandler.__init__c G s d S )Nr )r
r\ rZ r r r r] s z FakeProxyHandler.log_messagec C sv t j�| jd�\}}}}}}|| _| j�| �rr| �dd� | �dd� | � � | j
� t d| j d�� | j
� d� d S ) Nr r^ ZOKz Content-Typez text/htmlzYou've reached %s!<BR>r* sF Our apologies, but our server is down due to a sudden zombie invasion.)
�urllib�parseZurlparserS rT rj r rI rJ rK rL rM �bytes)r
ZscmZnetlocrS �paramsZqueryZfragmentr r r rd s zFakeProxyHandler.do_GETN)r r r r r] rd r r r r ri � s ri c sP e Zd ZdZdZdZdZ� fdd�Zdd� Z� fd d
�Z d d � Z
d
d� Z � Z S )�BasicAuthTestsrV rW Z IncorrectrX c sT t t| ��� dd� }t|�| _| �| j� d| jj | _| j� � | jj
� � d S )Nc _ s
t | |�S )N)rU )rZ r[ r r r �#http_server_with_basic_auth_handler s zABasicAuthTests.setUp.<locals>.http_server_with_basic_auth_handlerzhttp://127.0.0.1:%s) �superro �setUpr r �
addCleanup� stop_serverr �
server_url�startr �wait)r
rp )� __class__r r rr s
zBasicAuthTests.setUpc C s | j �� d | _ d S )N)r r )r
r r r rt ' s
zBasicAuthTests.stop_serverc s t t| ��� d S )N)rq ro �tearDown)r
)rx r r ry + s zBasicAuthTests.tearDownc C s| t j�� }|�| j| j| j| j� t j�t j� |�� y| �
t j� | j�� W n( t j j
k
rv | �d| j � Y nX d S )Nz!Basic auth failed for the url: %s)rk r
�HTTPBasicAuthHandler� add_passwordr` ru re rf �install_opener� build_opener�
assertTrue�urlopen�error� HTTPError�fail)r
�ahr r r �test_basic_auth_success. s
z&BasicAuthTests.test_basic_auth_successc C sR t j�� }|�| j| j| j| j� t j�t j� |�� | �
t j j t jj
| j� d S )N)rk r
rz r{ r` ru re �INCORRECT_PASSWDr| r} � assertRaisesr� r� r )r
r� r r r �test_basic_auth_httperror7 s
z(BasicAuthTests.test_basic_auth_httperror)
r r r re rf r� r` rr rt ry r� r� �
__classcell__r r )rx r ro s ro c sT e Zd ZdZdZdZdZ� fdd�Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z � Z
S )�ProxyAuthTestszhttp://localhostZtesterZtest123Z TestRealmc s� t t� ��� dd� }� �|tj�� � dtjd< dtjd< t� � _� j� � j
� j i� � j� � j
� � fdd�}t|�� _� �� j� � j�� � jj�� d� jj }tj�d |i�}tj�� � _tj�|� j�� _d S )
Nc S s t j�� t j�| � d S )N)�os�environ�clearrG )� old_environr r r �restore_environH s
z-ProxyAuthTests.setUp.<locals>.restore_environrb �NO_PROXY�no_proxyc s t � jf| �|�S )N)ri rj )rZ r[ )r
r r �create_fake_proxy_handlerS s z7ProxyAuthTests.setUp.<locals>.create_fake_proxy_handlerzhttp://127.0.0.1:%dr ) rq r� rr rs r� r� �copyr rj r' re rf r( r` r r rt rv r rw r rk r
Z ProxyHandlerZProxyDigestAuthHandler�proxy_digest_handlerr} �opener)r
r� r� Z proxy_url�handler)rx )r
r rr E s$
zProxyAuthTests.setUpc C s | j �� d | _ d S )N)r r )r
r r r rt ` s
zProxyAuthTests.stop_serverc C sF | j �| j| j| j| jd � | j�d� | �t j
j | j j
| j� d S )NZbadr )r� r{ r` �URLre rf rj r& r� rk r� r� r� �open)r
r r r �-test_proxy_with_bad_password_raises_httperrord s
z<ProxyAuthTests.test_proxy_with_bad_password_raises_httperrorc C s( | j �d� | �tjj| jj| j� d S )Nr ) rj r& r� rk r� r� r� r� r� )r
r r r �,test_proxy_with_no_password_raises_httperrorl s
z;ProxyAuthTests.test_proxy_with_no_password_raises_httperrorc C sN | j �| j| j| j| j� | j�d� | j� | j�}x |�
� r@q6W |� � d S )Nr ) r� r{ r` r� re rf rj r& r� r� �read�close)r
�resultr r r �test_proxy_qop_auth_worksr s
z(ProxyAuthTests.test_proxy_qop_auth_worksc C st | j �| j| j| j| j� | j�d� y| j� | j�}W n t
j j k
rT d }Y nX |rpx |�
� rfq\W |�� d S )Nzauth-int)r� r{ r` r� re rf rj r& r� r� rk r� �URLErrorr� r� )r
r� r r r �0test_proxy_qop_auth_int_works_or_throws_urlerror{ s
z?ProxyAuthTests.test_proxy_qop_auth_int_works_or_throws_urlerror)r r r r� re rf r` rr rt r� r� r� r� r� r r )rx r r� > s r� c s G � �fdd�dt jj�� � S )Nc sB e Zd ZdZg Zg ZdZdd� Zdd� Z� �fdd�Z d d
� Z
d S ) z1GetRequestHandler.<locals>.FakeHTTPRequestHandlerz TestHTTP/�P c S s. | � � }x |r(| j�|�}||d � }q
W d S )N)� send_headrL rM )r
�bodyZdoner r r rd � s z8GetRequestHandler.<locals>.FakeHTTPRequestHandler.do_GETc S s2 | j d }| j�t|��}| �� | j�|� d S )NzContent-Length)rP Zrfiler� �intrd �requestsr0 )r
Zcontent_lengthZ post_datar r r �do_POST� s
z9GetRequestHandler.<locals>.FakeHTTPRequestHandler.do_POSTc s� | j � _| j�| j� ��d�\}}}| �|� x&|D ] \}}| �||d| ji � q6W |rt| �dd� | � � |S | � � d S )Nr r z Content-typez
text/plain)
rP �headers_receivedr� r0 rS �poprI rJ r rK )r
Z
response_coderP r� �headerr? )�FakeHTTPRequestHandler� responsesr r r� � s
z;GetRequestHandler.<locals>.FakeHTTPRequestHandler.send_headc W s d S )Nr )r
rZ r r r r] � s z=GetRequestHandler.<locals>.FakeHTTPRequestHandler.log_messageN) r r r Zserver_versionr� r� r rd r� r� r] r )r� r� r r r� � s r� )r r rY )r� r )r� r� r �GetRequestHandler� s &r� c s� e Zd Z� fdd�Zd*dd�Zdd� Zd+dd �Zd,d
d �Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Z� ZS )-� TestUrlopenc s@ t t| ��� dd� }| �|tj�� � dtjd< dtjd< d S )Nc S s t j�� t j�| � d S )N)r� r� r� rG )r� r r r r� � s
z*TestUrlopen.setUp.<locals>.restore_environ�*r� r� )rq r� rr rs r� r� r� )r
r� )rx r r rr � s
zTestUrlopen.setUpNc K sl g }t jj||f|�}z@|�|�d�� |�|�� � |�|�d�� |�|�� � W d |�� X d� |�S )Nr^ i � )
rk r
r �extend� readlinesr0 �readliner� r� r )r
�url�datar[ �l�fr r r r � s
zTestUrlopen.urlopenc C s | j �� d | _ d S )N)r r )r
r r r rt � s
zTestUrlopen.stop_serverc C sZ |d krdg dfg}t |�}t|�| _| �| j� | j�� | jj�� | jj}||_|S )Nr^ s
we don't care) r� r r rs rt rv r rw r )r
r� r� r r r r � start_server� s
zTestUrlopen.start_serverc K s^ t tjd�s| �d� ddlm} |d kr6dg dfg}t|�}|| fd|i|��}|j|_|S )NZ HTTPSHandlerzssl support requiredr )�make_https_serverr^ s
we care a bitZ
handler_class)�hasattrrk r
�skipTestZtest.ssl_serversr� r� r )r
r� r[ r� r� r r r r �start_https_server� s
z TestUrlopen.start_https_serverc C sV d}ddgdfdg |fg}| � |�}| �d|j �}| �||� | �|jddg� d S ) Ns We got here...i. )ZLocationz(http://localhost:%(port)s/somewhere_elserb r^ zhttp://localhost:%s/�/z/somewhere_else)r� r r � assertEqualr� )r
�expected_responser� r� r� r r r �test_redirection� s
z TestUrlopen.test_redirectionc C s@ d}d}ddg|fg}| � |�}| �d|j �}| �||� d S )Ns hello worlds a
hello worl
1
d
0
r^ )zTransfer-EncodingZchunkedzhttp://localhost:%s/)r� r r r� )r
r� Z
chunked_startrF r� r� r r r � test_chunked s
zTestUrlopen.test_chunkedc
C s� d}| � dg |fg�}y| �d|j � W n6 tjjk
r` } z|�� }|�� W d d }~X Y n X | �d� | � ||� | � |j
dg� d S )Ns Bad bad bad...i� zhttp://localhost:%s/weeblez404 should raise URLErrorz/weeble) r� r r rk r� r� r� r� r� r� r� )r
r� r� r� r� r r r �test_404 s
zTestUrlopen.test_404c C sF d}| � dg |fg�}| �d|j �}| �||� | �|jdg� d S )Ns
pycon 2008...r^ zhttp://localhost:%s/bizarrez/bizarre)r� r r r� r� )r
r� r� r� r r r �test_200 s
zTestUrlopen.test_200c C sJ d}| � dg |fg�}| �d|j d�}| �||� | �|jddg� d S )Ns
pycon 2008...r^ zhttp://localhost:%s/bizarres get=with_feelingz/bizarre)r� r r r� r� )r
r� r� r� r r r �test_200_with_parameters% s z$TestUrlopen.test_200_with_parametersc C s8 | � � }tjtd�}| jd|j |d�}| �|d� d S )N)�cafilez https://localhost:%s/bizarre)�contexts
we care a bit)r� �ssl�create_default_context�CERT_localhostr r r� )r
r� r� r� r r r �
test_https- s zTestUrlopen.test_httpsc
C s� | j td�}t�dtf��� | jd|j td�}| �|d� | �t j
j ��}| jd|j t d� W d Q R X | j t d�}| �t j
j ��}| jd|j t d� W d Q R X W d Q R X d S )N)�certfilerb z https://localhost:%s/bizarre)r� s
we care a bit)
r� r� r �check_warnings�DeprecationWarningr r r� r� rk r� r� �CERT_fakehostname)r
r� r� �cmr r r �test_https_with_cafile3 s z"TestUrlopen.test_https_with_cafilec
C sX | j td�}t�dtf��4 | �tjj��}| j d|j
dd� W d Q R X W d Q R X d S )N)r� rb z https://localhost:%s/bizarreT)Z cadefault) r� r� r r� r� r� rk r� r� r r )r
r� r� r r r �test_https_with_cadefaultD s
z%TestUrlopen.test_https_with_cadefaultc s� t d kr| �d� t js"| �d� d � � fdd�}t �t j�}|�|� | j|td�}t jtd�}| j d|j
|d� | � � d � d S )
Nzssl module requiredz SNI support required in OpenSSLc s |� d S )Nr )Zssl_sockZ server_nameZinitial_context)�sni_namer r �cb_sniR s z*TestUrlopen.test_https_sni.<locals>.cb_sni)r� r� )r� zhttps://localhost:%s)r� Z localhost) r� r� ZHAS_SNIZ
SSLContextZPROTOCOL_TLS_SERVERZset_servername_callbackr� r� r� r r r� )r
r� r� r� r )r� r �test_https_sniL s
zTestUrlopen.test_https_snic C sP | � � }tjjd|j ddid�}tj�|�� W d Q R X | �|jd d� d S )Nzhttp://localhost:%s/ZRangez bytes=20-39)rP )r� rk r
ZRequestr r r� r� )r
r� Zreqr r r �test_sending_headers\ s
z TestUrlopen.test_sending_headersc C sb | � � }tj�d|j �}x"dD ]}| �t||�d| � q W z| �|�� d� W d |�� X d S )Nzhttp://localhost:%s)r� r� �info�geturlz3object returned from urlopen lacks the %s attributezcalling 'read' failed) r� rk r
r r r~ r� r� r� )r
r� �open_url�attrr r r �
test_basicd s
zTestUrlopen.test_basicc C sX | � � }tj�d|j �}|� |�� }W d Q R X | �|tjj d� | �
|� � d� d S )Nzhttp://localhost:%szEobject returned by 'info' is not an instance of email.message.MessageZplain) r� rk r
r r r� ZassertIsInstance�email�messageZMessager� Zget_content_subtype)r
r� r� Zinfo_objr r r � test_infoo s zTestUrlopen.test_infoc C sH | � � }tj�d|j �}|� |�� }W d Q R X | �|d|j � d S )Nzhttp://localhost:%s)r� rk r
r r r� r� )r
r� r� r� r r r � test_geturlz s
zTestUrlopen.test_geturlc C sF d}| � dg |fg�}tj�d|j �}x|D ]}| �||� q.W d S )Ns
pycon 2008...r^ zhttp://localhost:%s)r� rk r
r r r� )r
r� r� r� �liner r r �test_iteration� s
zTestUrlopen.test_iterationc
C s� ddddd d g}d� |�}| �dg |fg�}tj�d |j �}x<t|�D ]0\}}| �||| d
|t|| �t|�f � qLW | �|d t|�� d S ) Ns We
s got
s here
s verylong i �
r� r^ zhttp://localhost:%szQFetched line number %s doesn't match expected:
Expected length was %s, got %sr) ) r r� rk r
r r � enumerater� rO )r
�linesr� r� r� �indexr� r r r �test_line_iteration� s
z TestUrlopen.test_line_iteration)N)N)N)r r r rr r rt r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r )rx r r� � s(
r� c C s t �� ad S )N)r Zthreading_setup� threads_keyr r r r � setUpModule� s r� c C s t rtjt � d S )N)r� r Zthreading_cleanupr r r r �tearDownModule� s r� �__main__))rg r� r� Z urllib.parserk Zurllib.requestZ http.serverr r Zunittestr+ Ztestr r� � ImportErrorrS �dirname�__file__�herer r� r� r r r r r r rY rU ri ZTestCasero r� r� r� r� r� r� r �mainr r r r �<module> s>
y,!(N+ `