B
u9aa ã @ 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'dkrbe (¡ 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 ú+/usr/lib/python3.7/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!
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..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..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..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 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..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..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..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..FakeHTTPRequestHandler.send_headc W s d S )Nr )r
rZ r r r r] ° s z=GetRequestHandler..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..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..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 s>
y,!(N+ `