File: //usr/local/lib/python3.10/test/__pycache__/test_urllib2_localnet.cpython-310.opt-2.pyc
o
�i�e � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z d dl m
Z
d dl mZ zd dlZW n eyG dZY nw ej�e�Zej�ed�Zej�ed�ZG dd� dejj�ZG d d
� d
ej�Z G d d � d �Z G d
d� dejj �Z G dd� dejj �Z G dd� de j!�Z"e j#ddd�G dd� de j!��Z$dd� Z%G dd� de j!�Z&d d � Z'e(d kr�e �)� dS dS ) � N)�hashlib_helper)�threading_helper)�warnings_helperz keycert.pemz keycert2.pemc @ s e Zd Z dd� Zdd� ZdS )�LoopbackHttpServerc C s" t jj�| ||� | j�d� d S )Ng�������?)�http�server�
HTTPServer�__init__�socket�
settimeout)�selfZserver_addressZRequestHandlerClass� r
�7/usr/local/lib/python3.10/test/test_urllib2_localnet.pyr " s
�zLoopbackHttpServer.__init__c C s" | j �� \}}|�d� ||fS )Ng $@)r
�acceptr )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 @ �& e Zd Z dd� 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_server�Event�readyZprotocol_versionr �httpdZ server_port�port�r �request_handlerr
r
r r : s
�z!LoopbackHttpServerThread.__init__c C s d| _ | �� | j�� d S )NT)r �joinr Z server_close�r r
r
r �stopC s z LoopbackHttpServerThread.stopc C s( | j �� | js| j�� | jrd S d S �N)r �setr r �handle_requestr! r
r
r �runK s
�z LoopbackHttpServerThread.runN)r r r r r" r& r
r
r
r r 7 s
r c @ sV e Zd Z dd� 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�_qopr! r
r
r r U s
zDigestAuthHandler.__init__c C �
|| _ d S r# )r- )r Zqopr
r
r �set_qop\ �
zDigestAuthHandler.set_qopc C r. r# )r+ )r Zusersr
r
r � set_users_ s
zDigestAuthHandler.set_usersc C r. r# )r, )r Zrealmr
r
r � set_realmc r0 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_noncef s z!DigestAuthHandler._generate_noncec C s� |� d�}||d d � }|�d�}i }|D ](}|�d�\}}|�� }|d dkr6|d dkr6|dd� }n|�� }|||< q|S )N� r3 �,�=r �"���)�find�split�strip)r Zauth_strZfirst_space_index�parts� auth_dict�part�name�valuer
r
r �_create_auth_dictl 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)sr4 z%(method)s:%(uri)s�HA1�HA2z3%(HA1)s:%(nonce)s:%(nc)s:%(cnonce)s:%(qop)s:%(HA2)s�response)�updater5 r6 r8 r9 ) r rF rK rL rM Z
final_dictZHA1_strrN ZHA2_strrO Z response_strrP 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 Required� Content-Type� text/htmlzProxy-Authenticatez(Digest realm="%s", qop="%s",nonce="%s", s Proxy Authentication Required.F)�
send_response� send_headerr, r- r< � end_headers�wfile�writer r
r
r �_return_auth_challenge� s �� z(DigestAuthHandler._return_auth_challengec C s� t | j�dkr
dS d|jvr| �|�S | �|jd �}|d | jv r+| j|d }n| �|�S |�d�| jvr=| �|�S | j�|d � d}|j|j fD ]
}| �
|||j |�rZd}qM|sb| �|�S dS )Nr TzProxy-Authorization�usernamer; F) �lenr+ �headersrZ rJ �getr* �remove�path�
short_pathrR Zcommand)r r rF rK Zauth_validatedr` r
r
r r% � s4
�
��
z DigestAuthHandler.handle_requestN) r r r r r/ r1 r2 r< rJ rR rZ r% r
r
r
r r' R s r' c @ sd e Zd Z dZdZdZdeef Ze�e� d���
d�Z dd� Z dd � Z
d
d � Zd d
� Zdd� ZdS )�BasicAuthHandler�testUser�testPass�Testz%s:%sr4 c O s t jjj| g|�R i |�� d S r# )r r �BaseHTTPRequestHandlerr )r �args�kwargsr
r
r r � s zBasicAuthHandler.__init__c G � d S r# r
�r �formatrg r
r
r � log_message� s z BasicAuthHandler.log_messagec C s" | � d� | �dd� | �� d S )N�� � Content-typerT )rU rV rW r! r
r
r �do_HEAD� s
zBasicAuthHandler.do_HEADc C s4 | � d� | �dd| j � | �dd� | �� d S )Ni� zWWW-AuthenticatezBasic realm="%s"rn rT )rU rV �REALMrW r! r
r
r � do_AUTHHEAD� s
z BasicAuthHandler.do_AUTHHEADc C sl | j �dd�s| �� | j�d� d S | j �dd�d| j kr0| �d� | �� | j�d� d S | �� d S )NZ
Authorization� s No Auth header receivedzBasic rm s It works)r] r^ rq rX rY � ENCODED_AUTHrU rW r! r
r
r �do_GET� s ��
zBasicAuthHandler.do_GETN)r r r �USER�PASSWDrp Z USER_PASSWD�base64Z b64encoder8 �decoders r rl ro rq rt r
r
r
r rb � s rb c @ r )�FakeProxyHandlerc O s& || _ tjjj| g|�R i |�� d S r# )�digest_auth_handlerr r rf r )r rz rg rh r
r
r r � s zFakeProxyHandler.__init__c G ri r# r
rj r
r
r rl s z FakeProxyHandler.log_messagec C sz t j�| jd�\}}}}}}|| _| j�| �r;| �dd� | �dd� | � � | j
� t d| j d�� | j
� d� d S d S ) Nr rm ZOKrS rT zYou've reached %s!<BR>r4 sF Our apologies, but our server is down due to a sudden zombie invasion.)
�urllib�parse�urlparser` ra rz r% rU rV rW rX rY �bytes)r Zscm�netlocr` �params�query�fragmentr
r
r rt s � ��zFakeProxyHandler.do_GETN)r r r r rl rt r
r
r
r ry � s
ry c sP e Zd ZdZdZdZdZ� fdd�Zdd� Z� fd d
�Z d d � Z
d
d� Z � Z S )�BasicAuthTestsrc rd Z Incorrectre c sT t t| ��� dd� }t|�| _| �| j� d| jj | _| j� � | jj
� � d S )Nc _ s t | i |��S r# )rb �rg rh r
r
r �#http_server_with_basic_auth_handler! s zABasicAuthTests.setUp.<locals>.http_server_with_basic_auth_handlerzhttp://127.0.0.1:%s) �superr� �setUpr r �
addCleanup� stop_serverr �
server_url�startr �wait)r r� �� __class__r
r r� s
zBasicAuthTests.setUpc C � | j �� d | _ d S r# �r r" r! r
r
r r� ) �
zBasicAuthTests.stop_serverc s t t| ��� d S r# )r� r� �tearDownr! r� r
r r� - s zBasicAuthTests.tearDownc C sz t j�� }|�| j| j| j| j� t j�t j� |�� z
| �
t j� | j�� W d S t j j
y< | �d| j � Y d S w )Nz!Basic auth failed for the url: %s)r{ r �HTTPBasicAuthHandler� add_passwordrp r� ru rv �install_opener� build_opener�
assertTrue�urlopen�error� HTTPError�fail�r Zahr
r
r �test_basic_auth_success0 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 r# )r{ r r� r� rp r� ru �INCORRECT_PASSWDr� r� � assertRaisesr� r� r� r� r
r
r �test_basic_auth_httperror9 s
z(BasicAuthTests.test_basic_auth_httperror)
r r r ru rv r� rp r� r� r� r� r� �
__classcell__r
r
r� r r� s r� r6 T)Zopensslc 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 � t j�� t j�| � d S r# ��os�environ�clearrQ �Z old_environr
r
r �restore_environK �
z-ProxyAuthTests.setUp.<locals>.restore_environrr �NO_PROXY�no_proxyc s t � jg| �R i |��S r# )ry rz r� r! r
r �create_fake_proxy_handlerV s z7ProxyAuthTests.setUp.<locals>.create_fake_proxy_handlerzhttp://127.0.0.1:%dr ) r� r� r� r� r� r� �copyr' rz r1 ru rv r2 rp r r r� r� r r� r r{ r Z ProxyHandlerZProxyDigestAuthHandler�proxy_digest_handlerr� �opener)r r� r� Z proxy_url�handlerr� r! r r� H s&
�zProxyAuthTests.setUpc C r� r# r� r! r
r
r r� c r� zProxyAuthTests.stop_serverc C sF | j �| j| j| j| jd � | j�d� | �t j
j | j j
| j� d S )NZbadr( )r� r� rp �URLru rv rz r/ r� r{ r� r� r� �openr! r
r
r �-test_proxy_with_bad_password_raises_httperrorg s �
�z<ProxyAuthTests.test_proxy_with_bad_password_raises_httperrorc C s( | j �d� | �tjj| jj| j� d S �Nr( ) rz r/ r� r{ r� r� r� r� r� r! r
r
r �,test_proxy_with_no_password_raises_httperroro s
�z;ProxyAuthTests.test_proxy_with_no_password_raises_httperrorc C s| | j �| j| j| j| j� | j�d� | j� | j��}|�
� r, |�
� s W d � d S W d � d S 1 s7w Y d S r� ) r� r� rp r� ru rv rz r/ r� r� �read�r �resultr
r
r �test_proxy_qop_auth_worksu s �
��"�z(ProxyAuthTests.test_proxy_qop_auth_worksc C s� | j �| j| j| j| j� | j�d� z | j� | j�}W n t
j j y( Y d S w |� |�
� r= |�
� s0W d � d S W d � d S 1 sHw Y d S )Nzauth-int)r� r� rp r� ru rv rz r/ r� r� r{ r� �URLErrorr� 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� ru rv rp r� r� r� r� r� r� r� r
r
r� 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 s0 | � � }|r| j�|�}||d � }|sd S d S r# )� send_headrX rY )r �body�doner
r
r rt � s
�z8GetRequestHandler.<locals>.FakeHTTPRequestHandler.do_GETc S s2 | j d }| j�t|��}| �� | j�|� d S )NzContent-Length)r] Zrfiler� �intrt �requestsr: )r Zcontent_lengthZ post_datar
r
r �do_POST� s
z9GetRequestHandler.<locals>.FakeHTTPRequestHandler.do_POSTc s| | j � _| j�| j� ��d�\}}}| �|� |D ]\}}| �||d| ji � q|r8| �dd� | � � |S | � � d S )Nr r rn z
text/plain)
r] �headers_receivedr� r: r` �poprU rV r rW )r Z
response_coder] r� �headerrI ��FakeHTTPRequestHandler� responsesr
r r� � s
z;GetRequestHandler.<locals>.FakeHTTPRequestHandler.send_headc W ri r# r
)r rg r
r
r rl � s z=GetRequestHandler.<locals>.FakeHTTPRequestHandler.log_messageN) r r r Zserver_versionr� r� r rt r� r� rl r
r� r
r r� � s r� )r r rf )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)� Zd*d+� Zd,d-� Z� ZS )/� TestUrlopenc sN t t| ��� | �tjj� dd� }| �|tj� � � dtjd< dtjd< d S )Nc S r� r# r� r� r
r
r r� � r� z*TestUrlopen.setUp.<locals>.restore_environ�*r� r� )
r� r� r� r� r{ r �
urlcleanupr� r� r� )r r� r� r
r r� � s
zTestUrlopen.setUpNc K sx g }t jj||fi |��}z$|�|�d�� |�|�� � |�|�d�� |�|�� � W |�� n|�� w d� |�S )Nrm i � )
r{ r r� �extend� readlinesr: �readliner� �closer )r �url�datarh �l�fr
r
r r� � s
zTestUrlopen.urlopenc C r� r# r� r! r
r
r r� � r� zTestUrlopen.stop_serverc C sZ |d u r
dg dfg}t |�}t|�| _| �| j� | j�� | jj�� | jj}||_|S )Nrm s
we don't care) r� r r r� r� r� r r� r )r r� r� r r
r
r � start_server� s
zTestUrlopen.start_serverc K s^ t tjd�s | �d� ddlm} |d u rdg dfg}t|�}|| fd|i|��}|j|_|S )NZ HTTPSHandlerzssl support requiredr )�make_https_serverrm �
we care a bitZ
handler_class)�hasattrr{ r �skipTestZtest.ssl_serversr� r� r )r r� rh 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_elserr rm �http://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
rm )zTransfer-EncodingZchunkedr� )r� r� r r� )r r� Z
chunked_startrP r� r� r
r
r � test_chunked s �
zTestUrlopen.test_chunkedc
C s� d}| � dg |fg�}z
| �d|j � W n tjjy1 } z|�� }|�� W Y d }~n
d }~ww | �d� | � ||� | � |j
dg� d S )Ns Bad bad bad...i� zhttp://localhost:%s/weeblez404 should raise URLErrorz/weeble) r� r� r r{ 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 )N�
pycon 2008...rm �http://localhost:%s/bizarre�/bizarrer� �r r� r� r� r
r
r �test_200# s
zTestUrlopen.test_200c C sJ d}| � dg |fg�}| �d|j d�}| �||� | �|jddg� d S )Nr� rm r� s get=with_feelingr� 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�Zcafile� https://localhost:%s/bizarre��contextr� )r� �ssl�create_default_context�CERT_localhostr� r r� )r r� r� r� r
r
r �
test_https2 s zTestUrlopen.test_httpsc C s | j td�}t�dtf��i | jd|j td�}| �|d� | �t j
j ��}| jd|j t d� W d � n1 s:w Y | j t d�}| �t j
j ��}| jd|j t d� W d � n1 saw Y W d � d S W d � d S 1 syw Y d S )N��certfilerr r� r� r� )
r� r� r �check_warnings�DeprecationWarningr� r r� r� r{ r� r� �CERT_fakehostname)r r� r� �cmr
r
r �test_https_with_cafile8 s( � � � � �� "�z"TestUrlopen.test_https_with_cafilec C s� | j td�}t�dtf��2 | �tjj��}| j d|j
dd� W d � n1 s*w Y W d � d S W d � d S 1 sBw Y d S )Nr� rr r� T)Z cadefault) r� r� r r� r� r� r{ r� r� r� r )r r� r� r
r
r �test_https_with_cadefaultI s � ��"�z%TestUrlopen.test_https_with_cadefaultc s� t d u r | �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 r# r
)Zssl_sockZ server_nameZinitial_context�Zsni_namer
r �cb_sniW s z*TestUrlopen.test_https_sni.<locals>.cb_sni)r� r� r� zhttps://localhost:%sr� � localhost) r� r� ZHAS_SNIZ
SSLContextZPROTOCOL_TLS_SERVERZset_servername_callbackr� r� r� r� r r� )r r� r� r� r
r� r �test_https_sniQ s
zTestUrlopen.test_https_snic C sd | � � }tjjd|j ddid�}tj�|�� W d � n1 s"w Y | �|jd d� d S )Nr� ZRangez bytes=20-39�r] )r� r{ r �Requestr r� r� r� �r r� Zreqr
r
r �test_sending_headersa s � �z TestUrlopen.test_sending_headersc C sv | � � }tjjd|j ddid�}tj�|�� W d � n1 s"w Y | �d|j�� � | � d|j�� � d S )Nr� z
X-SoMe-hEaderZfoobarr� z
X-Some-Header)
r� r{ r r r r� ZassertInr� �keysZ assertNotInr r
r
r �test_sending_headers_cameli s � �z&TestUrlopen.test_sending_headers_camelc C sp | � � }tj�d|j ��!}dD ]
}| �t||�d| � q| �|�� d� W d � d S 1 s1w Y d S )N�http://localhost:%s)r� r� �info�geturlz3object returned from urlopen lacks the %s attributezcalling 'read' failed)r� r{ r r� r r� r� r� )r r� �open_url�attrr
r
r �
test_basicr s �"�zTestUrlopen.test_basicc C sl | � � }tj�d|j �}|� |�� }W d � n1 s w Y | �|tjj d� | �
|� � d� d S )Nr zEobject returned by 'info' is not an instance of email.message.MessageZplain) r� r{ r r� r r ZassertIsInstance�email�messageZMessager� Zget_content_subtype)r r� r Zinfo_objr
r
r � test_infoz s �
� �zTestUrlopen.test_infoc C s\ | � � }tj�d|j �}|� |�� }W d � n1 s w Y | �|d|j � d S )Nr )r� r{ r r� r r r� )r r� r r� r
r
r � test_geturl� s
�zTestUrlopen.test_geturlc C sB d}| � dg |fg�}tj�d|j �}|D ]}| �||� qd S )Nr� rm r )r� r{ 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 �}t|�D ]\}}| �||| d
|t|| �t|�f � q%| �|d t|�� d S ) Ns We
s got
s here
s verylong i �
r� rm r zQFetched line number %s doesn't match expected:
Expected length was %s, got %sr3 ) r r� r{ r r� r � enumerater� r\ )r �linesr� r� r� �indexr r
r
r �test_line_iteration� s
��z TestUrlopen.test_line_iterationc C s� | � dg dfdg dfg�}tj�� }tj�d|j �}| �d |j� |�|d� d�� | �d|j� | �d|�
d�� |�|d� d�� | �d |j� | �d
|�
d�� d S ) Nrm s anyr �1zus-ascii� 1zContent-lengthZ
1234567890s
1234567890Z10) r� r{ r r� r r r� r� r� r8 Z
get_header)r r� r� r r
r
r �test_issue16464� s �
zTestUrlopen.test_issue16464r# )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� r
r
r� r r� � s0
r� c C s t �� } tjt jg| �R � d S r# )r Zthreading_setup�unittestZaddModuleCleanupZthreading_cleanup)� thread_infor
r
r � setUpModule� s r �__main__)*rw r� r � urllib.parser{ Zurllib.requestZ http.serverr r r r5 Z test.supportr r r r� � ImportErrorr` �dirname�__file__�herer r� r� r r r r r r' rf rb ry ZTestCaser� Zrequires_hashdigestr� r� r� r r �mainr
r
r
r �<module> sF � y,! (M+ { �