Your IP : 216.73.216.147
���]s` �
@ s+ d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
m Z e j d � Z
y d d l Z Wn e k
r� d Z Yn Xe j j e � Z e j j e d � Z e j j e d � Z Gd d � d e j j � Z Gd d � d e
j � Z Gd
d � d � Z Gd d
� d
e j j � Z Gd d � d e j j � Z e j e
d � Gd d � d e j! � � Z" e j e
d � Gd d � d e j! � � Z# d d � Z$ e j e
d � Gd d � d e j! � � Z% d a& d d � Z' d d � Z( e) d k r'e j* � d S) � N)�support� threadingz keycert.pemz keycert2.pemc @ s( e Z d Z d d � Z d d � Z d S)�LoopbackHttpServerc C s- t j j j | | | � | j j d � d S)Ng�������?)�http�server�
HTTPServer�__init__�socket�
settimeout)�selfZserver_addressZRequestHandlerClass� r �:/opt/alt/python35/lib64/python3.5/test_urllib2_localnet.pyr ! s zLoopbackHttpServer.__init__c C s, | j j � \ } } | j d � | | f S)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 @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�LoopbackHttpServerThreadc C sV t j j | � d | _ t j � | _ d | _ t d | � | _ | j j | _
d S)NFzHTTP/1.0� 127.0.0.1r )r r ) r �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 � | j j � d S)NT)r �joinr Z server_close)r r r r
�stopB s
z LoopbackHttpServerThread.stopc C s. | j j � x | j s) | j j � q Wd S)N)r �setr r �handle_request)r r r r
�runJ s
z LoopbackHttpServerThread.runN)r r r r r r! r r r r
r 6 s r c @ s| e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d S)�DigestAuthHandlerc C s1 d | _ g | _ i | _ d | _ d | _ d S)Nr z
Test Realm�auth)� _request_num�_nonces�_users� _realm_name�_qop)r r r r
r T s
zDigestAuthHandler.__init__c C s
| | _ d S)N)r( )r Zqopr r r
�set_qop[ 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_realmb s zDigestAuthHandler.set_realmc C sJ | j d 7_ t j t | j � j d � � j � } | j j | � | S)N� �ascii)r$ �hashlib�md5�str�encode� hexdigestr% �append)r �noncer r r
�_generate_noncee s 'z!DigestAuthHandler._generate_noncec C s� | j d � } | | d d � } | j d � } i } xx | D]p } | j d � \ } } | j � } | d d k r� | d d k r� | d d � } n | j � } | | | <q? W| S) N� r, �,�=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 } | j | � | | d <| | d <| | d <d | } t j | j d � � j � } d | } t j | j d � � j � } | | d <| | d <d | }
t j |
j d � � j � } | | d
k S) 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/ r1 r2 ) r r? rD rE rF Z
final_dictZHA1_strrG ZHA2_strrH Z response_strrI r r r
�_validate_auth| s
z DigestAuthHandler._validate_authc C sg | j d d � | j d d � | j d d | j | j | j � f � | j � | j j d � d S) 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( r5 � end_headers�wfile�write)r r r r r
�_return_auth_challenge� s
z(DigestAuthHandler._return_auth_challengec C s t | j � d k r d Sd | j k r5 | j | � S| j | j d � } | d | j k rr | j | d } n
| j | � S| j d � | j k r� | j | � S| j j | d � d } x; | j | j g D]' } | j
| | | j | � r� d } q� W| s| j | � Sd Sd S)Nr TzProxy-AuthorizationZusernamer4 F) �lenr& �headersrQ rC �getr% �remove�path�
short_pathrK Zcommand)r r r? rD Zauth_validatedrV r r r
r � s,
z DigestAuthHandler.handle_requestN) r r r r r) r* r+ r5 rC rK rQ r r r r r
r" Q s r" c @ s� e Z d Z d Z d Z d Z d e e f Z e j e j d � � j
d � Z d d � Z d d � Z
d
d � Z d d
� Z d d � Z d S)�BasicAuthHandler�testUser�testPass�Testz%s:%sr- c O s t j j j | | | � d S)N)r r �BaseHTTPRequestHandlerr )r �args�kwargsr r r
r � s zBasicAuthHandler.__init__c G s d S)Nr )r �formatr] r r r
� log_message� s z BasicAuthHandler.log_messagec C s+ | j d � | j d d � | j � d S)N�� z Content-typez text/html)rL rM rN )r r r r
�do_HEAD� s
zBasicAuthHandler.do_HEADc C sB | j d � | j d d | j � | j d d � | j � d S)Ni� zWWW-AuthenticatezBasic realm="%s"z Content-typez text/html)rL rM �REALMrN )r r r r
� do_AUTHHEAD� s
z BasicAuthHandler.do_AUTHHEADc C s� | j j d d � s2 | j � | j j d � nV | j j d d � d | j k r~ | j d � | j � | j j d � n
| j � d S)NZ
Authorization� s No Auth header receivedzBasic ra s It works)rS rT rd rO rP � ENCODED_AUTHrL rN )r r r r
�do_GET� s
zBasicAuthHandler.do_GETN)r r r �USER�PASSWDrc Z USER_PASSWD�base64Z b64encoder1 �decoderf r r` rb rd rg r r r r
rX � s ! rX c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�FakeProxyHandlerc O s& | | _ t j j j | | | � d S)N)�digest_auth_handlerr r r\ r )r rm r] r^ r r r
r � s zFakeProxyHandler.__init__c G s d S)Nr )r r_ r] r r r
r` s z FakeProxyHandler.log_messagec C s� t j j | j d � \ } } } } } } | | _ | j j | � r� | j d d � | j d d � | j � | j
j t d | j d � � | j
j d � d S) Nr ra ZOKz Content-Typez text/htmlzYou've reached %s!<BR>r- sF Our apologies, but our server is down due to a sudden zombie invasion.)
�urllib�parse�urlparserV rW rm r rL rM rN rO rP �bytes)r ZscmZnetlocrV �paramsZqueryZfragmentr r r
rg s !
zFakeProxyHandler.do_GETN)r r r r r` rg r r r r
rl � s rl z!Threading required for this test.c sd e Z d Z d Z d Z d Z d Z � f d d � Z � f d d � Z d d
� Z d d � Z
� S)
�BasicAuthTestsrY rZ Z Incorrectr[ c su t t | � j � d d � } t | � | _ | j | j j � d | j j | _ | j j � | j j
j � d S)Nc _ s
t | | � S)N)rX )r] r^ r r r
�#http_server_with_basic_auth_handler! s zABasicAuthTests.setUp.<locals>.http_server_with_basic_auth_handlerzhttp://127.0.0.1:%s) �superrs �setUpr r �
addCleanupr r �
server_url�startr �wait)r rt )� __class__r r
rv s
zBasicAuthTests.setUpc s t t | � j � d S)N)ru rs �tearDown)r )r{ r r
r| ) s zBasicAuthTests.tearDownc C s� t j j � } | j | j | j | j | j � t j j t j j | � � y | j
t j j | j � � Wn+ t j j
k
r� | j d | j � Yn Xd S)Nz!Basic auth failed for the url: %s)rn r �HTTPBasicAuthHandler� add_passwordrc rx rh ri �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 ss t j j � } | j | j | j | j | j � t j j t j j | � � | j
t j j t j j
| j � d S)N)rn r r} r~ rc rx rh �INCORRECT_PASSWDr r� � assertRaisesr� r� r� )r r� r r r
�test_basic_auth_httperror5 s " z(BasicAuthTests.test_basic_auth_httperror) r r r rh ri r� rc rv r| r� r� r r )r{ r
rs s rs c s| e Z d Z d Z d Z d Z d Z � f d d � Z � f d d � Z d d
� Z d d � Z
d
d � Z d d � Z � S)�ProxyAuthTestszhttp://localhostZtesterZtest123Z TestRealmc s$ t t � � j � d d � } � j | t j j � � d t j d <d t j d <t � � _ � j j � j
� j i � � j j � j
� � f d d � } t | � � _ � j j � � j j j � d � j j } t j j d | i � } t j j � � _ t j j | � j � � _ d S)
Nc S s! t j j � t j j | � d S)N)�os�environ�clearrJ )� old_environr r r
�restore_environG s
z-ProxyAuthTests.setUp.<locals>.restore_environre �NO_PROXY�no_proxyc s t � j | | � S)N)rl rm )r] r^ )r r r
�create_fake_proxy_handlerR s z7ProxyAuthTests.setUp.<locals>.create_fake_proxy_handlerzhttp://127.0.0.1:%dr )ru r� rv rw r� r� �copyr" rm r* rh ri r+ rc r r ry r rz r rn r Z ProxyHandlerZProxyDigestAuthHandler�proxy_digest_handlerr� �opener)r r� r� Z proxy_url�handler)r{ )r r
rv D s"
zProxyAuthTests.setUpc s$ | j j � t t | � j � d S)N)r r ru r� r| )r )r{ r r
r| ^ s
zProxyAuthTests.tearDownc C s_ | j j | j | j | j | j d � | j j d � | j t j
j | j j
| j � d S)NZbadr# )r� r~ rc �URLrh ri rm r) r� rn r� r� r� �open)r r r r
�-test_proxy_with_bad_password_raises_httperrorb s z<ProxyAuthTests.test_proxy_with_bad_password_raises_httperrorc C s6 | j j d � | j t j j | j j | j � d S)Nr# ) rm r) r� rn r� r� r� r� r� )r r r r
�,test_proxy_with_no_password_raises_httperrorj s z;ProxyAuthTests.test_proxy_with_no_password_raises_httperrorc C sk | j j | j | j | j | j � | j j d � | j j | j � } x | j
� r\ qM W| j � d S)Nr# ) r� r~ rc r� rh ri rm r) r� r� �read�close)r �resultr r r
�test_proxy_qop_auth_worksp s z(ProxyAuthTests.test_proxy_qop_auth_worksc C s� | j j | j | j | j | j � | j j d � y | j j | j � } Wn t
j j k
rn d } Yn X| r� x | j
� r� qx W| j � d S)Nzauth-int)r� r~ rc r� rh ri rm r) r� r� rn r� �URLErrorr� r� )r r� r r r
�0test_proxy_qop_auth_int_works_or_throws_urlerrory s z?ProxyAuthTests.test_proxy_qop_auth_int_works_or_throws_urlerror)
r r r r� rh ri rc rv r| r� r� r� r� r r )r{ r
r� < s r� c s) G� � f d d � d t j j � � � S)Nc sa e Z d Z d Z g Z g Z d Z d d � Z d d � Z � � f d d � Z d d
� Z
d S) z1GetRequestHandler.<locals>.FakeHTTPRequestHandlerz TestHTTP/�P c S s? | j � } x, | r: | j j | � } | | d � } q Wd S)N)� send_headrO rP )r �body�doner r r
rg � s z8GetRequestHandler.<locals>.FakeHTTPRequestHandler.do_GETc S sC | j d } | j j t | � � } | j � | j j | � d S)NzContent-Length)rS Zrfiler� �intrg �requestsr3 )r Zcontent_lengthZ post_datar r r
�do_POST� s
z9GetRequestHandler.<locals>.FakeHTTPRequestHandler.do_POSTc s� | j � _ | j j | j � � j d � \ } } } | j | � x1 | D]) \ } } | j | | d | j i � qK W| r� | j d d � | j � | S| j � d S)Nr r z Content-typez
text/plain)
rS �headers_receivedr� r3 rV �poprL rM r rN )r Z
response_coderS r� �headerrB )�FakeHTTPRequestHandler� responsesr r
r� � s
!
z;GetRequestHandler.<locals>.FakeHTTPRequestHandler.send_headc W s d S)Nr )r r] r r r
r` � s z=GetRequestHandler.<locals>.FakeHTTPRequestHandler.log_messageN) r r r Zserver_versionr� r� r rg r� r� r` r )r� r� r r
r� � s r� )r r r\ )r� r )r� r� r
�GetRequestHandler� s %&r� c s e Z d Z � f d d � Z d d d � Z d d d � Z d d d � Z d
d � Z d d
� Z d d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d! � Z d" d# � Z d$ d% � Z d& d' � Z � S)(� TestUrlopenc sV t t | � j � d d � } | j | t j j � � d t j d <d t j d <d S)Nc S s! t j j � t j j | � d S)N)r� r� r� rJ )r� r r r
r� � s
z*TestUrlopen.setUp.<locals>.restore_environ�*r� r� )ru r� rv rw r� r� r� )r r� )r{ r r
rv � s
zTestUrlopen.setUpNc
K s� g } t j j | | | � } zV | j | j d � � | j | j � � | j | j d � � | j | j � � Wd | j � Xd j | � S)Nra i � )
rn r r� �extend� readlinesr3 �readliner� r� r )r �url�datar^ �l�fr r r
r� � s zTestUrlopen.urlopenc C s� | d k r d g d f g } t | � } t | � | _ | j | j j � | j j � | j j j � | j j } | | _ | S)Nra s
we don't care) r� r r rw r ry r rz r )r r� r� r r r r
� start_server� s
zTestUrlopen.start_serverc K s~ t t j d � s | j d � d d l m } | d k rM d g d f g } t | � } | | d | | �} | j | _ | S)NZ HTTPSHandlerzssl support requiredr )�make_https_serverra s
we care a bitZ
handler_class)�hasattrrn 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 sy d } d d
g d f d g | f g } | j | � } | j d | j � } | j | | � | j | j d d g � d S) Ns We got here...i. �Location�(http://localhost:%(port)s/somewhere_elsere ra zhttp://localhost:%s/�/z/somewhere_else)r� r� )r� r� r � assertEqualr� )r �expected_responser� r� r� r r r
�test_redirection� s z TestUrlopen.test_redirectionc C sZ d } d } d d g | f g } | j | � } | j d | j � } | j | | � d S)Ns hello worlds a
hello worl
1
d
0
ra �Transfer-Encoding�chunkedzhttp://localhost:%s/)r� r� )r� r� r r� )r r� Z
chunked_startrI r� r� r r r
� test_chunked� s zTestUrlopen.test_chunkedc C s� d } | j d g | f g � } y | j d | j � WnA t j j k
r| } z | j � } | j � WYd d } ~ Xn X| j d � | j | | � | j | j
d g � d S)Ns Bad bad bad...i� zhttp://localhost:%s/weeblez404 should raise URLErrorz/weeble) r� r� r rn r� r� r� r� r� r� r� )r r� r� r� r� r r r
�test_404
s
zTestUrlopen.test_404c C sa d } | j d g | f g � } | j d | j � } | j | | � | j | j d g � d S)Ns
pycon 2008...ra zhttp://localhost:%s/bizarrez/bizarre)r� r� r r� r� )r r� r� r� r r r
�test_200 s
zTestUrlopen.test_200c C sg d } | j d g | f g � } | j d | j d � } | j | | � | j | j d d g � d S)Ns
pycon 2008...ra 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 sN | j � } t j d t � } | j d | j d | �} | j | 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 d t � } | j d | j d t �} | j | d � | j t j j � �! } | j d | j d t �Wd QRX| j d t � } | j t
j � �! } | j d | j d t �Wd QRXd S)N�certfilez https://localhost:%s/bizarrer� s
we care a bit) r� r� r� r r� r� rn r� r� �CERT_fakehostnamer� ZCertificateError)r r� r� �cmr r r
�test_https_with_cafile. s z"TestUrlopen.test_https_with_cafilec C sO | j d t � } | j t j j � �! } | j d | j d d �Wd QRXd S)Nr� z https://localhost:%s/bizarreZ cadefaultT)r� r� r� rn r� r� r� r )r r� r� r r r
�test_https_with_cadefault> s z%TestUrlopen.test_https_with_cadefaultc s� t d k r | j d � t j s/ | j d � d � � f d d � } t j t j � } | j | � | j d | d t � } t j d t � } | j d | j
d | �| j � d � d S)
Nzssl module requiredz SNI support required in OpenSSLc s
| � d S)Nr )Zssl_sockZ server_nameZinitial_context)�sni_namer r
�cb_sniK s z*TestUrlopen.test_https_sni.<locals>.cb_snir� r� r� zhttps://localhost:%s� localhost) r� r� ZHAS_SNIZ
SSLContextZPROTOCOL_TLSv1Zset_servername_callbackr� r� r� r� r r� )r r� r� r� r )r� r
�test_https_sniE s
zTestUrlopen.test_https_snic
C sf | j � } t j j d | j d d d i �} t j j | � � Wd QRX| j | j d d � d S)Nzhttp://localhost:%s/rS ZRangez bytes=20-39)r� rn r ZRequestr r� r� r� )r r� Zreqr r r
�test_sending_headersU s z TestUrlopen.test_sending_headersc
C s | j � } t j j d | j � } x+ d D]# } | j t | | � d | � q, Wz | j | j � d � Wd | j � Xd S) Nzhttp://localhost:%sr� r� �info�geturlz3object returned from urlopen lacks the %s attributezcalling 'read' failed)r� r� r� r� ) r� rn r r� r r� r� r� r� )r r� �open_url�attrr r r
�
test_basic] s
zTestUrlopen.test_basicc C sr | j � } t j j d | j � } | � | j � } Wd QRX| j | t j j d � | j
| j � d � d S)Nzhttp://localhost:%szEobject returned by 'info' is not an instance of email.message.MessageZplain) r� rn r r� r r� ZassertIsInstance�email�messageZMessager� Zget_content_subtype)r r� r� Zinfo_objr r r
� test_infoh s zTestUrlopen.test_infoc C sZ | j � } t j j d | j � } | � | j � } Wd QRX| j | d | j � d S)Nzhttp://localhost:%s)r� rn r r� r r� r� )r r� r� r� r r r
� test_geturls s
zTestUrlopen.test_geturlc C s_ d } | j d g | f g � } t j j d | j � } x | D] } | j | | � qA Wd S)Ns
pycon 2008...ra zhttp://localhost:%s)r� rn r r� r r� )r r� r� r� �liner r r
�test_iteration{ s
zTestUrlopen.test_iterationc C s� d d d d d d g } d j | � } | j d g | f g � } t j j d | j � } xN t | � D]@ \ } } | j | | | d
| t | | � t | � f � qj W| j | d t | � � d S) Ns We
s got
s here
s verylong i s
r� ra zhttp://localhost:%szQFetched line number %s doesn't match expected:
Expected length was %s, got %sr, ) r r� rn r r� r � enumerater� rR )r �linesr� r� r� �indexr� r r r
�test_line_iteration� s %z TestUrlopen.test_line_iteration)r r r rv 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 j � a d S)N)r Zthreading_setup� threads_keyr r r r
� setUpModule� s r� c C s t r t j t � d S)N)r� r Zthreading_cleanupr r r r
�tearDownModule� s r� �__main__)+rj r� r� Z urllib.parsern Zurllib.requestZ http.serverr Zunittestr. Ztestr �
import_moduler r� � ImportErrorrV �dirname�__file__�herer r� r� r r r r r r" r\ rX rl Z
skipUnlessZTestCasers r� r� r� r� r� r� r �mainr r r r
�<module> sB
y , ! $ M + �