Sindbad~EG File Manager
a
��g�e � @ sr d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z d dl
mZ z d dlZW n ey� dZY n0 ej�e�Zej�ed�Zej�ed�ZG dd� dejj�ZG dd � d ej�Z G d
d � d �Z G d d
� d
ejj �Z G dd� dejj �Z G dd� de j!�Z"ej#ddd�G dd� de j!��Z$dd� Z%G dd� de j!�Z&dd � Z'e(d k�rne �)� dS ) � N)�support)�hashlib_helperz keycert.pemz keycert2.pemc @ s e Zd ZdZdd� Zdd� ZdS )�LoopbackHttpServerz^HTTP server w/ a few modifications that make it useful for
loopback testing purposes.
c C s" t jj�| ||� | j�d� d S )Ng�������?)�http�server�
HTTPServer�__init__�socket�
settimeout)�selfZserver_addressZRequestHandlerClass� r �6/usr/local/lib/python3.9/test/test_urllib2_localnet.pyr ! s
�zLoopbackHttpServer.__init__c C s | j �� \}}|�d� ||fS )z HTTPServer method, overridden.g $@)r �acceptr
)r �requestZclient_addressr r r
� get_request* s
z LoopbackHttpServer.get_requestN)�__name__�
__module__� __qualname__�__doc__r r r r r r
r s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �LoopbackHttpServerThreadz2Stoppable thread that runs a loopback http server.c 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 9 s
�z!LoopbackHttpServerThread.__init__c C s d| _ | �� | j�� dS )z.Stops the webserver if it's currently running.TN)r �joinr Z server_close�r r r r
�stopB s z LoopbackHttpServerThread.stopc C s | j �� | js | j�� q
d S �N)r �setr r �handle_requestr r r r
�runJ s
z LoopbackHttpServerThread.runN)r r r r r r! r% r r r r
r 6 s r c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )�DigestAuthHandlerz-Handler for performing digest authentication.c 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 T s
zDigestAuthHandler.__init__c C s
|| _ d S r" )r, )r Zqopr r r
�set_qop[ s zDigestAuthHandler.set_qopc C s t |t�sJ �|| _d S r" )�
isinstance�dictr* )r Zusersr r r
� set_users^ s zDigestAuthHandler.set_usersc C s
|| _ d S r" )r+ )r Zrealmr r r
� set_realmb 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_noncee s z!DigestAuthHandler._generate_noncec C s� |� d�}||d d � }|�d�}i }|D ]P}|�d�\}}|�� }|d dkrl|d dkrl|dd� }n|�� }|||< q,|S )N� r2 �,�=r �"���)�find�split�strip)r Zauth_strZfirst_space_index�parts� auth_dict�part�name�valuer r r
�_create_auth_dictk 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)sr3 z%(method)s:%(uri)s�HA1�HA2z3%(HA1)s:%(nonce)s:%(nc)s:%(cnonce)s:%(qop)s:%(HA2)s�response)�updater4 r5 r7 r8 ) r rE rJ rK rL Z
final_dictZHA1_strrM ZHA2_strrN Z response_strrO 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�dkrdS d|jvr&| �|�S | �|jd �}|d | jv rT| j|d }n
| �|�S |�d�| jvrx| �|�S | j�|d � d}|j|j fD ]}| �
|||j |�r�d}q�|s�| �|�S dS dS )a Performs digest authentication on the given HTTP request
handler. Returns True if authentication was successful, False
otherwise.
If no users have been set, then digest auth is effectively
disabled and this method will always return True.
r TzProxy-Authorization�usernamer: FN) �lenr* �headersrY rI �getr) �remove�path�
short_pathrQ Zcommand)r r rE rJ Zauth_validatedr_ r r r
r$ � s0
�
�
z DigestAuthHandler.handle_requestN)
r r r r r r- r0 r1 r; rI rQ rY r$ r r r r
r& Q s r& c @ sf e Zd ZdZdZdZdZdeef Ze� e�
d��� d�Z dd� Z
d d
� Zd d � Zd
d� Zdd� ZdS )�BasicAuthHandlerz,Handler for performing basic authentication.�testUser�testPass�Testz%s:%sr3 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 s d S r" r �r �formatrf r r r
� log_message� s z BasicAuthHandler.log_messagec C s" | � d� | �dd� | �� d S )N�� � Content-typerS )rT rU rV r r r r
�do_HEAD� s
zBasicAuthHandler.do_HEADc C s4 | � d� | �dd| j � | �dd� | �� d S )Ni� zWWW-AuthenticatezBasic realm="%s"rl rS )rT rU �REALMrV 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 rk s It works)r\ r] ro rW rX � ENCODED_AUTHrT rV r r r r
�do_GET� s ��
zBasicAuthHandler.do_GETN)r r r r �USER�PASSWDrn Z USER_PASSWD�base64Z b64encoder7 �decoderq r rj rm ro rr r r r r
ra � s ra c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �FakeProxyHandlerz�This is a 'fake proxy' that makes it look like the entire
internet has gone down due to a sudden zombie invasion. It main
utility is in providing us with authentication support for
testing.
c O s& || _ tjjj| g|�R i |�� d S r" )�digest_auth_handlerr r re r )r rx rf rg r r r
r � s zFakeProxyHandler.__init__c G s d S r" r rh r r r
rj s z FakeProxyHandler.log_messagec C sv t j�| jd�\}}}}}}|| _| j�| �rr| �dd� | �dd� | � � | j
� t d| j d�� | j
� d� d S ) Nr rk ZOKrR rS zYou've reached %s!<BR>r3 sF Our apologies, but our server is down due to a sudden zombie invasion.)
�urllib�parse�urlparser_ r` rx r$ rT rU rV rW rX �bytes)r Zscm�netlocr_ �params�query�fragmentr r r
rr s � �zFakeProxyHandler.do_GETN)r r r r r rj rr r r r r
rw � s rw c sP e Zd ZdZdZdZdZ� fdd�Zdd� Z� fd d
�Z d d � Z
d
d� Z � Z S )�BasicAuthTestsrb rc Z Incorrectrd c sT t t| ��� dd� }t|�| _| �| j� d| jj | _| j� � | jj
� � d S )Nc _ s t | i |��S r" )ra �rf rg 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 s | j �� d | _ d S r" �r r! r r r r
r� ( s
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 n& t j j
yt | �d| j � Y n0 d S )Nz!Basic auth failed for the url: %s)ry r �HTTPBasicAuthHandler� add_passwordrn r� rs rt �install_opener� build_opener�
assertTrue�urlopen�error� HTTPError�fail�r Zahr 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 r" )ry r r� r� rn r� rs �INCORRECT_PASSWDr� r� � assertRaisesr� r� r� r� r r r
�test_basic_auth_httperror8 s
z(BasicAuthTests.test_basic_auth_httperror)
r r r rs rt r� rn r� r� r� r� r� �
__classcell__r r r� r
r� s r� r5 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 s t j�� t j�| � d S r" ��os�environ�clearrP �Z old_environr r r
�restore_environJ s
z-ProxyAuthTests.setUp.<locals>.restore_environrp �NO_PROXY�no_proxyc s t � jg| �R i |��S r" )rw rx r� r r r
�create_fake_proxy_handlerU s z7ProxyAuthTests.setUp.<locals>.create_fake_proxy_handlerzhttp://127.0.0.1:%dr ) r� r� r� r� r� r� �copyr& rx r0 rs rt r1 rn r r r� r� r r� r ry r Z ProxyHandlerZProxyDigestAuthHandler�proxy_digest_handlerr� �opener)r r� r� Z proxy_url�handlerr� r r
r� G s&
�zProxyAuthTests.setUpc C s | j �� d | _ d S r" r� r r r r
r� b s
zProxyAuthTests.stop_serverc C sF | j �| j| j| j| jd � | j�d� | �t j
j | j j
| j� d S )NZbadr' )r� r� rn �URLrs rt rx r- r� ry r� r� r� �openr r r r
�-test_proxy_with_bad_password_raises_httperrorf s �
�z<ProxyAuthTests.test_proxy_with_bad_password_raises_httperrorc C s( | j �d� | �tjj| jj| j� d S �Nr' ) rx r- r� ry r� r� r� r� r� r r r r
�,test_proxy_with_no_password_raises_httperrorn s
�z;ProxyAuthTests.test_proxy_with_no_password_raises_httperrorc C sb | j �| j| j| j| j� | j�d� | j� | j��}|�
� r@q6W d � n1 sT0 Y d S r� ) r� r� rn r� rs rt rx r- r� r� �read�r �resultr r r
�test_proxy_qop_auth_workst 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 yN Y n00 |� |�
� r`qVW d � n1 st0 Y d S )Nzauth-int)r� r� rn r� rs rt rx r- r� r� ry 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� rs rt rn 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 s* | � � }|r&| j�|�}||d � }qd S r" )� send_headrW rX )r �body�doner r r
rr � s z8GetRequestHandler.<locals>.FakeHTTPRequestHandler.do_GETc S s2 | j d }| j�t|��}| �� | j�|� d S )NzContent-Length)r\ Zrfiler� �intrr �requestsr9 )r Zcontent_lengthZ post_datar r r
�do_POST� s
z9GetRequestHandler.<locals>.FakeHTTPRequestHandler.do_POSTc s| | j � _| j�| j� ��d�\}}}| �|� |D ] \}}| �||d| ji � q4|rp| �dd� | � � |S | � � d S )Nr r rl z
text/plain)
r\ �headers_receivedr� r9 r_ �poprT rU r rV )r Z
response_coder\ r� �headerrH ��FakeHTTPRequestHandler� responsesr r
r� � s
z;GetRequestHandler.<locals>.FakeHTTPRequestHandler.send_headc W s d S r" r )r rf r r r
rj � s z=GetRequestHandler.<locals>.FakeHTTPRequestHandler.log_messageN) r r r Zserver_versionr� r� r rr r� r� rj r r� r r
r� � s r� )r r re )r� r r� r
�GetRequestHandler� s &r� c s� e Zd ZdZ� fdd�Zd/dd�Zdd� Zd0d d
�Zd1d d �Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z� ZS )2� TestUrlopenaN Tests urllib.request.urlopen using the network.
These tests are not exhaustive. Assuming that testing using files does a
good job overall of some of the basic interface features. There are no
tests exercising the optional 'data' and 'proxies' arguments. No tests
for transparent redirection have been written.
c sN t t| ��� | �tjj� dd� }| �|tj� � � dtjd< dtjd< d S )Nc S s t j�� t j�| � d S r" r� r� r r r
r� � s
z*TestUrlopen.setUp.<locals>.restore_environ�*r� r� )
r� r� r� r� ry r �
urlcleanupr� r� r� )r r� r� r r
r� � s
zTestUrlopen.setUpNc K sx g }t jj||fi |��}zH|�|�d�� |�|�� � |�|�d�� |�|�� � W |�� n
|�� 0 d� |�S )Nrk i � )
ry r r� �extend� readlinesr9 �readliner� �closer )r �url�datarg �l�fr r r
r� � s zTestUrlopen.urlopenc C s | j �� d | _ d S r" r� r r r r
r� � s
zTestUrlopen.stop_serverc C sZ |d u rdg dfg}t |�}t|�| _| �| j� | j�� | jj�� | jj}||_|S )Nrk 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 r6dg dfg}t|�}|| fd|i|��}|j|_|S )NZ HTTPSHandlerzssl support requiredr )�make_https_serverrk �
we care a bitZ
handler_class)�hasattrry r �skipTestZtest.ssl_serversr� r� r )r r� rg 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_elserp rk �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
rk )zTransfer-EncodingZchunkedr� )r� r� r r� )r r� Z
chunked_startrO r� r� r r r
� test_chunked s �
zTestUrlopen.test_chunkedc
C s� d}| � dg |fg�}z| �d|j � W n8 tjjyb } z |�� }|�� W Y d }~nd }~0 0 | �d� | � ||� | � |j
dg� d S )Ns Bad bad bad...i� zhttp://localhost:%s/weeblez404 should raise URLErrorz/weeble) r� r� r ry 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...rk �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� rk 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_https1 s zTestUrlopen.test_httpsc C s� | j td�}t�dtf��� | jd|j td�}| �|d� | �t j
j ��$}| jd|j t d� W d � n1 st0 Y | j t d�}| �t j
j ��$}| jd|j t d� W d � n1 s�0 Y W d � n1 s�0 Y d S )N��certfilerp r� r� r� )
r� r� r �check_warnings�DeprecationWarningr� r r� r� ry r� r� �CERT_fakehostname)r r� r� �cmr r r
�test_https_with_cafile7 s � �$ �z"TestUrlopen.test_https_with_cafilec C s� | j td�}t�dtf��R | �tjj��$}| j d|j
dd� W d � n1 sT0 Y W d � n1 sr0 Y d S )Nr� rp r� T)Z cadefault) r� r� r r� r� r� ry r� r� r� r )r r� r� r r r
�test_https_with_cadefaultH 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_sniV 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_sniP s
zTestUrlopen.test_https_snic C sd | � � }tjjd|j ddid�}tj�|�� W d � n1 sD0 Y | �|jd d� d S )Nr� ZRangez bytes=20-39�r\ )r� ry r �Requestr r� r� r� �r r� Zreqr r r
�test_sending_headers` s � z TestUrlopen.test_sending_headersc C sv | � � }tjjd|j ddid�}tj�|�� W d � n1 sD0 Y | �d|j�� � | � d|j�� � d S )Nr� z
X-SoMe-hEaderZfoobarr� z
X-Some-Header)
r� ry r r� r r� ZassertInr� �keysZ assertNotInr� r r r
�test_sending_headers_camelh s � z&TestUrlopen.test_sending_headers_camelc C sn | � � }tj�d|j ��@}dD ]}| �t||�d| � q | �|�� d� W d � n1 s`0 Y d S )N�http://localhost:%s)r� r� �info�geturlz3object returned from urlopen lacks the %s attributezcalling 'read' failed)r� ry r r� r r� r� r� )r r� �open_url�attrr r r
�
test_basicq s �zTestUrlopen.test_basicc C sl | � � }tj�d|j �}|� |�� }W d � n1 s<0 Y | �|tjj d� | �
|� � d� d S )Nr� zEobject returned by 'info' is not an instance of email.message.MessageZplain) r� ry r r� r r ZassertIsInstance�email�messageZMessager� Zget_content_subtype)r r� r Zinfo_objr r r
� test_infoy s �& �zTestUrlopen.test_infoc C s\ | � � }tj�d|j �}|� |�� }W d � n1 s<0 Y | �|d|j � d S )Nr� )r� ry 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 ]}| �||� q,d S )Nr� rk r� )r� ry 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 ]0\}}| �||| d
|t|| �t|�f � qJ| �|d t|�� d S ) Ns We
s got
s here
s verylong i �
r� rk r� zQFetched line number %s doesn't match expected:
Expected length was %s, got %sr2 ) r r� ry 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 ) Nrk s anyr� �1zus-ascii� 1zContent-lengthZ
1234567890s
1234567890Z10) r� ry r r� r� r r� r� r� r7 Z
get_header)r r� r� r r r r
�test_issue16464� s �
zTestUrlopen.test_issue16464)N)N)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 r� r r r� r
r� � s.
r� c C s t �� } tjt jg| �R � d S r" )r Zthreading_setup�unittestZaddModuleCleanupZthreading_cleanup)� thread_infor r r
� setUpModule� s r �__main__)*ru r� r � urllib.parsery Zurllib.requestZ http.serverr r r r4 �testr Z test.supportr r� � ImportErrorr_ �dirname�__file__�herer r� r� r r r r r r& re ra rw ZTestCaser� Zrequires_hashdigestr� r� r� r r �mainr r r r
�<module> s>
y,!( M+ {
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists