a
äze 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#dddG dd„ de j!ƒƒZ$dd„ Z%G dd„ de j!ƒZ&dd „ Z'e(d krne )¡ dS ) é N)Úsupport)Úhashlib_helperz 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 ú0/usr/lib/python3.9/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 @ 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_serverÚEventÚreadyZprotocol_versionr ÚhttpdZ server_portÚport©r Úrequest_handlerr r r
r 9 s
ÿz!LoopbackHttpServerThread.__init__c C s d| _ | ¡ | j ¡ d S )NT)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 6 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Ú_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
|| _ d S r! )r) )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ú r/ ú,ú=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)sr0 z%(method)s:%(uri)sÚHA1ÚHA2z3%(HA1)s:%(nonce)s:%(nc)s:%(cnonce)s:%(qop)s:%(HA2)sÚresponse)Úupdater1 r2 r4 r5 ) r rB rG rH rI Z
final_dictZHA1_strrJ ZHA2_strrK Z response_strrL 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+ r8 Ú 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 d S )Nr TzProxy-AuthorizationÚusernamer7 F) Úlenr) ÚheadersrV rF Úgetr( ÚremoveÚpathÚ
short_pathrN Zcommand)r r rB rG Zauth_validatedr\ r r r
r# s0
ÿ
ý
z DigestAuthHandler.handle_requestN) r r r r r, r- r. r8 rF rN rV r# r r r r
r% Q 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:%sr0 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 Úformatrc r r r
Ú log_messageÖ s z BasicAuthHandler.log_messagec C s" | d¡ | dd¡ | ¡ d S )NéÈ ú Content-typerP )rQ rR rS r r r r
Údo_HEADÚ s
zBasicAuthHandler.do_HEADc C s4 | d¡ | dd| j ¡ | dd¡ | ¡ d S )Ni‘ zWWW-AuthenticatezBasic realm="%s"ri rP )rQ rR ÚREALMrS 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 rh s It works)rY rZ rl rT rU Ú ENCODED_AUTHrQ rS r r r r
Údo_GETå s ÿÿ
zBasicAuthHandler.do_GETN)r r r ÚUSERÚPASSWDrk Z USER_PASSWDÚbase64Z b64encoder4 Údecodern r rg rj rl ro r r r r
r^ Ê s r^ c @ s$ e Zd Zdd„ Zdd„ Zdd„ ZdS )ÚFakeProxyHandlerc O s& || _ tjjj| g|¢R i |¤Ž d S r! )Údigest_auth_handlerr r rb r )r ru rc rd r r r
r ý s zFakeProxyHandler.__init__c G s d S r! r re r r r
rg s z FakeProxyHandler.log_messagec C sv t j | jd¡\}}}}}}|| _| j | ¡rr| dd¡ | dd¡ | ¡ | j
t d| j dƒ¡ | j
d¡ d S ) Nr rh ZOKrO rP zYou've reached %s!
r0 sF Our apologies, but our server is down due to a sudden zombie invasion.)
ÚurllibÚparseÚurlparser\ r] ru r# rQ rR rS rT rU Úbytes)r ZscmÚnetlocr\ ÚparamsÚqueryÚfragmentr r r
ro s ÿ ÿzFakeProxyHandler.do_GETN)r r r r rg ro r r r r
rt ö s rt c sP e Zd ZdZdZdZdZ‡ fdd„Zdd„ Z‡ fd d
„Z d d „ Z
d
d„ Z ‡ Z S )ÚBasicAuthTestsr_ r` Z Incorrectra c sT t t| ƒ ¡ dd„ }t|ƒ| _| | j¡ d| jj | _| j ¡ | jj
¡ d S )Nc _ s t | i |¤ŽS r! )r^ ©rc rd r r r
Ú#http_server_with_basic_auth_handler s zABasicAuthTests.setUp..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)rv r ÚHTTPBasicAuthHandlerÚ add_passwordrk r… rp rq Ú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! )rv r rŒ r rk r… rp ÚINCORRECT_PASSWDrŽ r Ú assertRaisesr’ r“ r‘ r• r r r
Útest_basic_auth_httperror8 s
z(BasicAuthTests.test_basic_auth_httperror)
r r r rp rq r— rk r‚ r„ r‹ r– r™ Ú
__classcell__r r rˆ r
r~ s r~ r2 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ÚclearrM ©Z old_environr r r
Úrestore_environJ s
z-ProxyAuthTests.setUp..restore_environrm ÚNO_PROXYÚno_proxyc s t ˆ jg| ¢R i |¤ŽS r! )rt ru r r r r
Úcreate_fake_proxy_handlerU s z7ProxyAuthTests.setUp..create_fake_proxy_handlerzhttp://127.0.0.1:%dr ) r r› r‚ rƒ r rž Úcopyr% ru r- rp rq r. rk r r r„ r† r r‡ r rv 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 rk ÚURLrp rq ru r, r˜ rv r’ r“ r§ Úopenr r r r
Ú-test_proxy_with_bad_password_raises_httperrorf s ÿ
þz.FakeHTTPRequestHandlerz TestHTTP/éP c S s* | ¡ }|r&| j |¡}||d … }qd S r! )Ú send_headrT rU )r ÚbodyÚdoner r r
ro – s z8GetRequestHandler..FakeHTTPRequestHandler.do_GETc S s2 | j d }| j t|ƒ¡}| ¡ | j |¡ d S )NzContent-Length)rY Zrfiler® Úintro Úrequestsr6 )r Zcontent_lengthZ post_datar r r
Údo_POSTœ s
z9GetRequestHandler..FakeHTTPRequestHandler.do_POSTc s| | j ˆ _| j | j¡ ˆ d¡\}}}| |¡ |D ] \}}| ||d| ji ¡ q4|rp| dd¡ | ¡ |S | ¡ d S )Nr r ri z
text/plain)
rY Úheaders_receivedr¹ r6 r\ ÚpoprQ rR r rS )r Z
response_coderY r¶ ÚheaderrE ©ÚFakeHTTPRequestHandlerÚ responsesr r
rµ ¢ s
z;GetRequestHandler..FakeHTTPRequestHandler.send_headc W s d S r! r )r rc r r r
rg ± s z=GetRequestHandler..FakeHTTPRequestHandler.log_messageN) r r r Zserver_versionr¹ r» r ro rº rµ rg r r¾ r r
r¿ s r¿ )r r rb )rÀ r r¾ r
ÚGetRequestHandler s &rÁ c sÊ e Zd Z‡ fdd„Zd.dd„Zdd„ Zd/dd „Zd0d
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 )1Ú TestUrlopenc 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..restore_environÚ*r¢ r£ )
r r r‚ rƒ rv 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 )Nrh i ó )
rv r r‘ ÚextendÚ readlinesr6 Úreadliner® Úcloser )r ÚurlÚdatard Ú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 )Nrh 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_serverrh ó
we care a bitZ
handler_class)Úhasattrrv r ÚskipTestZtest.ssl_serversrÏ rÁ r )r rÀ rd 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_elserm rh ú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
rh )zTransfer-EncodingZchunkedrÔ )rÎ r‘ r r× )r rØ Z
chunked_startrL 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 rv 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...rh ú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Ü rh 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©Úcertfilerm rã râ rÐ )
rÓ rè r Úcheck_warningsÚDeprecationWarningr‘ r r× r˜ rv 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ê rm rã T)Z cadefault) rÓ rè r rì rí r˜ rv 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..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©rY )rÎ rv 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Î rv 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Î rv 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Î rv 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Î rv 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Ü rh rü )rÎ rv 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Å rh rü zQFetched line number %s doesn't match expected:
Expected length was %s, got %sr/ ) r rÎ rv r r‘ r Ú enumerater× rX )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 ) Nrh s anyrü Ú1zus-asciió 1zContent-lengthZ
1234567890s
1234567890Z10) rÎ rv r r r÷ r r× rË rª r4 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 ¸ 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__)*rr r r Ú urllib.parserv Zurllib.requestZ http.serverr r r r1 Útestr Z test.supportr ræ Ú ImportErrorr\ ÚdirnameÚ__file__Úherer rè rî r r r r r r% rb r^ rt ZTestCaser~ Zrequires_hashdigestr› rÁ r r r Úmainr r r r
Ú s>
y,!( M+ {