o
IXîhü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#dddG 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
úD/opt/python-3.10.19/usr/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!
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..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..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..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.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..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..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..FakeHTTPRequestHandler.send_headc W ri r# r
)r rg r
r
r rl ² s z=GetRequestHandler..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..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..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 Ú sF ÿ y,! (M+ { ÿ