Your IP : 216.73.216.196
B
� f?� � @ s� d Z ddlmZmZmZmZ ddlmZmZ ddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlmZ ddl m Z m Z ddlZddl m Z G dd � d �Z!G d
d � d ej"�Z#G d d
� d
ej$�Z%G dd� de%�Z&G dd� de%�Z'G dd� de%�Z(dZ)dZ*dZ+e�,e-e d��o\e �.� dkd�G dd� de%��Z/G dd � d e�Z0G d d � d e0�Z1G d d � d �Z2G d!d"� d"ej$�Z3G d#d$� d$ej$�Z4G d%d&� d&ej$�Z5d*d'd(�Z6e7d)k�r�e6� dS )+z�Unittests for the various HTTPServer modules.
Written by Cody A.W. Somerville <cody-somerville@ubuntu.com>,
Josip Dzolonga, and Michael Otteneder for the 2007/08 GHOP contest.
� )�BaseHTTPRequestHandler�
HTTPServer�SimpleHTTPRequestHandler�CGIHTTPRequestHandler)�server�
HTTPStatusN)�mock)�BytesIO�StringIO)�supportc @ s e Zd Zdd� Zddd�ZdS )�NoLogRequestHandlerc G s d S )N� )�self�argsr
r
�:/opt/alt/python37/lib64/python3.7/test/test_httpservers.py� log_message" s z NoLogRequestHandler.log_messageNc C s dS )N� r
)r �nr
r
r �read& s zNoLogRequestHandler.read)N)�__name__�
__module__� __qualname__r r r
r
r
r r ! s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestServerThreadc C s t j�| � || _|| _d S )N)� threading�Thread�__init__�request_handler� test_object)r r r r
r
r r + s zTestServerThread.__init__c C sZ t d| j�| _| jj�� \| j_| j_| jj� � d | _z| j�
d� W d | j� � X d S )N)Z localhostr g�������?) r r r ZsocketZ getsocknamer �HOST�PORT�server_started�setZ
serve_foreverZ server_close)r r
r
r �run0 s zTestServerThread.runc C s | j �� | �� d S )N)r Zshutdown�join)r r
r
r �stop: s
zTestServerThread.stopN)r r r r r" r$ r
r
r
r r * s
r c @ s, e Zd Zdd� Zdd� Zddi fdd�ZdS ) � BaseTestCasec C sD t �� | _t �� t_t�� | _t | | j
�| _ | j � � | j�
� d S )N)r Zthreading_setup�_threadsZEnvironmentVarGuard�os�environr ZEventr r r �thread�start�wait)r r
r
r �setUp@ s
zBaseTestCase.setUpc C s* | j �� d | _ tj�� tj| j� d S )N)r) r$ r' r( �__exit__r Zthreading_cleanupr& )r r
r
r �tearDownH s
zBaseTestCase.tearDown�GETNc C s0 t j�| j| j�| _| j�||||� | j�� S )N)�http�client�HTTPConnectionr r Z
connection�request� getresponse)r Zuri�method�body�headersr
r
r r3 N s zBaseTestCase.request)r r r r, r. r3 r
r
r
r r% ? s r% c @ s� e Zd ZG dd� dee�Zdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+S ),�BaseHTTPServerTestCasec @ s\ e Zd ZdZdZdd� Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dS )z&BaseHTTPServerTestCase.request_handlerzHTTP/1.1c C s0 | � tj� | �dd� | �dd� | �� d S )Nz Content-Typez text/html�
Connection�close)�
send_responser �
NO_CONTENT� send_header� end_headers)r r
r
r �do_TESTY s z.BaseHTTPServerTestCase.request_handler.do_TESTc C s0 | � tj� | �dd� | �dd� | �� d S )Nz Content-Typez text/htmlr9 z
keep-alive)r; r r< r= r> )r r
r
r �do_KEEP_ s z.BaseHTTPServerTestCase.request_handler.do_KEEPc C s | � d� d S )Ni� )�
send_error)r r
r
r � do_KEYERRORe s z2BaseHTTPServerTestCase.request_handler.do_KEYERRORc C s | � tj� d S )N)rA r � NOT_FOUND)r r
r
r � do_NOTFOUNDh s z2BaseHTTPServerTestCase.request_handler.do_NOTFOUNDc C s | � ddd� d S )Ni� z
Short Messagez This is a long
explanation)rA )r r
r
r �do_EXPLAINERRORk s z6BaseHTTPServerTestCase.request_handler.do_EXPLAINERRORc C s. | � d� | �dd� | �dd� | �� d S )Ni� z Content-Typez text/htmlr9 r: )r; r= r> )r r
r
r � do_CUSTOMo s
z0BaseHTTPServerTestCase.request_handler.do_CUSTOMc C sJ | � d� | �dd� | �dd� | �� | jd �d�}| j�|� d S )Ni� z X-Specialu Dängerous Mindr9 r: zx-special-incomingzutf-8)r; r= r> r7 �encode�wfile�write)r r6 r
r
r �do_LATINONEHEADERu s
z8BaseHTTPServerTestCase.request_handler.do_LATINONEHEADERc C s | � t| jdd � �� d S )N� )rA �int�path)r r
r
r �
do_SEND_ERROR} s z4BaseHTTPServerTestCase.request_handler.do_SEND_ERRORc C s | � t| jdd � �� d S )NrK )rA rL rM )r r
r
r �do_HEAD� s z.BaseHTTPServerTestCase.request_handler.do_HEADN)r r r �protocol_version�default_request_versionr? r@ rB rD rE rF rJ rN rO r
r
r
r r U s r c C s, t �| � tj�| j| j�| _| j�� d S )N) r% r, r0 r1 r2 r r �con�connect)r r
r
r r, � s
z BaseHTTPServerTestCase.setUpc C s, | j �dd� | j �� }| �|jtj� d S )Nr/ �/)rR r3 r4 � assertEqual�statusr �NOT_IMPLEMENTED)r �resr
r
r � test_command� s
z#BaseHTTPServerTestCase.test_commandc C s> d| j _| j �dd� | j �� | j �� }| �|jtj� d S )Nz HTTP/1.1
ZXYZBOGUSrT ) rR �
_http_vsn_str�
putrequest�
endheadersr4 rU rV r rW )r rX r
r
r �test_request_line_trimming� s
z1BaseHTTPServerTestCase.test_request_line_trimmingc C s> d| j _| j �dd� | j �� | j �� }| �|jtj� d S )NZFUBARr/ rT ) rR rZ r[ r\ r4 rU rV r � BAD_REQUEST)r rX r
r
r �test_version_bogus� s
z)BaseHTTPServerTestCase.test_version_bogusc C s> d| j _| j �dd� | j �� | j �� }| �|jtj� d S )Nz
HTTP/9.9.9r/ rT ) rR rZ r[ r\ r4 rU rV r r^ )r rX r
r
r �test_version_digits� s
z*BaseHTTPServerTestCase.test_version_digitsc C s> d| j _| j �dd� | j �� | j �� }| �|jtj� d S )Nr r/ rT ) rR rZ r[ r\ r4 rU rV r rW )r rX r
r
r �test_version_none_get� s
z,BaseHTTPServerTestCase.test_version_none_getc C s> d| j _| j �dd� | j �� | j �� }| �|jtj� d S )Nr �CUSTOMrT ) rR rZ r[ r\ r4 rU rV r r^ )r rX r
r
r �test_version_none� s
z(BaseHTTPServerTestCase.test_version_nonec C sF d| j _d| j _| j �dd� | j �� | j �� }| �|jtj � d S )N�c zHTTP/9.9r/ rT )
rR Z _http_vsnrZ r[ r\ r4 rU rV r ZHTTP_VERSION_NOT_SUPPORTED)r rX r
r
r �test_version_invalid� s
z+BaseHTTPServerTestCase.test_version_invalidc C s> d| j _| j �dd� | j �� | j �� }| �|jtj� d S )Nr ) rR rZ r[ r\ r4 rU rV r r^ )r rX r
r
r �test_send_blank� s
z&BaseHTTPServerTestCase.test_send_blankc C sD | j �dd� | j �dd� | j �� | j �� }| �|jtj� d S )Nr/ rT r9 r: ) rR r[ � putheaderr\ r4 rU rV r rW )r rX r
r
r �test_header_close� s
z(BaseHTTPServerTestCase.test_header_closec C sL d| j _| j �dd� | j �dd� | j �� | j �� }| �|jtj � d S )NzHTTP/1.1r/ rT r9 z
keep-alive)
rR rZ r[ rg r\ r4 rU rV r rW )r rX r
r
r �test_header_keep_alive� s
z-BaseHTTPServerTestCase.test_header_keep_alivec C s, | j �dd� | j �� }| �|jtj� d S )N�TESTrT )rR r3 r4 rU rV r r< )r rX r
r
r � test_handler� s
z#BaseHTTPServerTestCase.test_handlerc C sJ | j �dd� | j �� }| �|�d�d� | j �dd� | �| j j� d S )NZKEEPrT r9 z
keep-aliverj )rR r3 r4 rU � getheaderZ
addCleanupr: )r rX r
r
r � test_return_header_keep_alive� s
z4BaseHTTPServerTestCase.test_return_header_keep_alivec C s* | j �dd� | j �� }| �|jd� d S )NZKEYERRORrT i� )rR r3 r4 rU rV )r rX r
r
r �test_internal_key_error� s
z.BaseHTTPServerTestCase.test_internal_key_errorc C s* | j �dd� | j �� }| �|jd� d S )Nrb rT i� )rR r3 r4 rU rV )r rX r
r
r �test_return_custom_status� s
z0BaseHTTPServerTestCase.test_return_custom_statusc C s> | j �dd� | j �� }| �|jd� | �t|�d��� d S )NZ EXPLAINERRORrT i� zContent-Length)rR r3 r4 rU rV �
assertTruerL rl )r rX r
r
r �test_return_explain_error� s
z0BaseHTTPServerTestCase.test_return_explain_errorc C sL | j jddddid� | j �� }| �|�d�d� | �|�� d�d�� d S ) NZLATINONEHEADERrT zX-Special-Incomingu Ärger mit Unicode)r7 z X-Specialu Dängerous Mindzutf-8)rR r3 r4 rU rl r rG )r rX r
r
r �test_latin1_header� s
z)BaseHTTPServerTestCase.test_latin1_headerc C sN | j �dd� | j �� }| �|jtj� |�� }| �t|� d��t
|�� d S )NZNOTFOUNDrT zContent-Length) rR r3 r4 rU rV r rC r rL rl �len)r rX �datar
r
r �test_error_content_length� s
z0BaseHTTPServerTestCase.test_error_content_lengthc C s� t jt jf}x�t jt jt jt jt jfD ]�}| j�dd�|�� | j� � }| �
||j � | �
d |� d�� | �
d |� d�� ||kr�| �
d |� d�� |�
� }| �
d|� q&W d S )NZ
SEND_ERRORz/{}zContent-Lengthz Content-TypezTransfer-Encoding� )r � NOT_MODIFIED�
RESET_CONTENTr< Z
PROCESSING�SWITCHING_PROTOCOLSrR r3 �formatr4 rU rV rl r )r � allow_transfer_encoding_codes�coderX rt r
r
r �test_send_error� s
z&BaseHTTPServerTestCase.test_send_errorc C s� t jt jf}x�t jt jt jt jt jfD ]�}| j�dd�|�� | j� � }| �
||j � |t jkr�| � t
|�d��dk� | �d|�d�� n$| �
d |�d�� | �
d |�d�� ||kr�| �
d |�d�� |�� }| �
d|� q&W d S ) N�HEADz/{}zContent-Lengthr z text/htmlz Content-TypezTransfer-Encodingrv )r rw rx �OKr< ry rR r3 rz r4 rU rV rp rL rl �assertInr )r r{ r| rX rt r
r
r �test_head_via_send_error
s"
z/BaseHTTPServerTestCase.test_head_via_send_errorN)r r r r r r r, rY r] r_ r` ra rc re rf rh ri rk rm rn ro rq rr ru r} r� r
r
r
r r8 T s* . r8 c @ s, e Zd ZG dd� de�Zdd� Zdd� ZdS )� RequestHandlerLoggingTestCasec @ s$ e Zd ZdZdZdd� Zdd� ZdS )z-RequestHandlerLoggingTestCase.request_handlerzHTTP/1.1c C s | � tj� | �� d S )N)r; r r r> )r r
r
r �do_GET( s z4RequestHandlerLoggingTestCase.request_handler.do_GETc C s | � tjd� d S )NzFile not found)rA r rC )r r
r
r �do_ERROR, s z6RequestHandlerLoggingTestCase.request_handler.do_ERRORN)r r r rP rQ r� r� r
r
r
r r $ s r c C sb t j�| j| j�| _| j�� t�� � }| j� dd� | j�
� W d Q R X | � |� � �
d�� d S )Nr/ rT z"GET / HTTP/1.1" 200 -
)r0 r1 r2 r r rR rS r �captured_stderrr3 r4 rp �getvalue�endswith)r �errr
r
r �test_get/ s
z&RequestHandlerLoggingTestCase.test_getc C s� t j�| j| j�| _| j�� t�� � }| j� dd� | j�
� W d Q R X |� � � d�}| �
|d �d�� | �
|d �d�� d S )NZERRORrT �
r z code 404, message File not foundrK z"ERROR / HTTP/1.1" 404 -)r0 r1 r2 r r rR rS r r� r3 r4 r� �splitrp r� )r r� �linesr
r
r �test_err: s
z&RequestHandlerLoggingTestCase.test_errN)r r r r r r� r� r
r
r
r r� # s r� c s� e Zd ZG dd� dee�Z� fdd�Z� fdd�Zd'dd �Ze �
e j d
kd �e �
e j d kd
�e �
ejd�dd� ���Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Z� Z S )(�SimpleHTTPServerTestCasec @ s e Zd ZdS )z(SimpleHTTPServerTestCase.request_handlerN)r r r r
r
r
r r H s r c s� t � �� t�� | _t�� }t�|� d| _tj |d�| _
tj � | j
�| _
d| j
| _tj �| j
d�}t|d��}|�| j� |�� W d Q R X t�|�j}tj�|tjj�}|jdd�| _tj j |� � dd �| _ d S )
Ns We are the knights who say Ni!)�dirrT �test�wbr )Z microsecondT)�usegmt) �superr, r' �getcwd�cwd�tempfileZ
gettempdir�chdirrt �mkdtemp�tempdirrM �basename� tempdir_name�base_urlr# �openrI �flush�stat�st_mtime�datetimeZ
fromtimestampZtimezoneZutc�replace�last_modif_datetime�email�utilsZ
formatdate� timestamp�last_modif_header)r Z basetempdirZtempnameZtemp�mtimeZ
last_modif)� __class__r
r r, K s$
z SimpleHTTPServerTestCase.setUpc s@ z.t �| j� yt�| j� W n Y nX W d t� �� X d S )N)r' r� r� �shutilZrmtreer� r� r. )r )r� r
r r. ` s
z!SimpleHTTPServerTestCase.tearDownNc s� � �fdd�}d � |�_ ��� }| ��� | ��j|� | ��j� |rT| �||� | ��jd� | ��j� dd�d� | �� �d�dd� � �
� |S ) Nc s �j � d�_ dS )zWDon't close reader yet so we can check if there was leftover
buffered inputN)�fpr
)�reader�responser
r �
close_connk s zDSimpleHTTPServerTestCase.check_status_and_reason.<locals>.close_conn�
r9 r: � rv zConnection should be closed) Z _close_connr rp rU rV �assertIsNotNone�reason�version�msg�getr: )r r� rV rt r� r6 r
)r� r� r �check_status_and_reasonj s
z0SimpleHTTPServerTestCase.check_status_and_reason�darwinz2undecodable name cannot always be decoded on macOSZwin32z+undecodable name cannot be decoded on win32z need support.TESTFN_UNDECODABLEc C s t �� }t�tj�d }ttj�| j |�d��}|�
tj� W d Q R X | � | j d �}t j
dkr�x"t�| j �D ]}|dkrn|}P qnW | �|tj�}tjj|dd�}| �d| �|d �|� | �d
tj|d d � �|d �|� | � | j d | �}| j|tjtjd
� d S )Nz.txtr� rT r� r� �
surrogatepass)�errorsz href="%s"�surrogateescapez>%s<F)�quote)rt )�sys�getfilesystemencodingr' �fsdecoder �TESTFN_UNDECODABLEr� rM r# r� rI r3 r� �platform�listdirr� r r �urllib�parser� r� rG �html�escape)r �enc�filename�fr� �namer6 Z
quotednamer
r
r �test_undecodable_filename� s&
z2SimpleHTTPServerTestCase.test_undecodable_filenamec C sd | � | jd ��� }| �|d| j� d��d�� | � | jd ��� }| �|d| j� d��d�� d S )Nz/?x=123z listing for z/\?x=123�latin1z/?x=%bbu /\?x=�)r3 r� r Z assertRegexrG )r r� r
r
r �test_undecodable_parameter� s z3SimpleHTTPServerTestCase.test_undecodable_parameterc C s2 t �t j�| jd�� d| j� d�}|� d�}| �|�}| �|tj � |�
d�}| j ||dd� d|� �}| �|�}| �|tj � |�
d�}| j |�
d�|d� | j ||d d� d|� �}| �|�}| �|tj � | � |�
d�|� d
|� �}|� d�}| �|�}| �|tj � |�
d�}| j|�
d
�|d� d S ) aI Ensure //evil.co/..%2f../../X does not put //evil.co/ in Location.
//netloc/ in a Location header is a redirect to a new host.
https://github.com/python/cpython/issues/87389
This checks that a path resolving to a directory on our server cannot
resolve into a redirect to another server.
Zexisting_directoryz2/python.org/..%2f..%2f..%2f..%2f..%2f../%0a%0d/../z/existing_directoryrT �Locationznon-attack failed!)r� z//zcExpected Location header to start with a single / and end with a / as this is a directory redirect.zhttps://pypi.org/N)r' �mkdirrM r# r� r� r3 r� r �MOVED_PERMANENTLYrl rU � assertFalse�
startswithrp ) r ZurlZexpected_locationr� �locationZ
attack_urlZ attack3_urlZ attack_scheme_netloc_2slash_urlZ expected_scheme_netloc_locationr
r
r �3test_get_dir_redirect_location_domain_injection_bug� s0
zLSimpleHTTPServerTestCase.test_get_dir_redirect_location_domain_injection_bugc
C s� | � | jd �}| j|tj| jd� | � | jd �}| �|tj� | � | jd �}| �|tj� | � | j�}| �|tj� | � | jd �}| �|tj� | � | jd �}| �|tj� | �|� d�| jd � | � d �}| �|tj� | � d
�}| �|tj� d }t
t j �
| jd �d
��}|�|� W d Q R X | � | jd �}| �|tj|� t jdk�r�t �� dk�r�t �| jd� z"| � | jd �}| �|tj� W d t �| jd� X d S )Nz/test)rt z/test/rT z/?hi=2z?hi=1r� z/?hi=1z/ThisDoesNotExistz/ThisDoesNotExist/s Dummy index file
z
index.htmlr� �posixr i� )r3 r� r� r r rt rC r� rU rl r� r' rM r# r� rI r� �geteuid�chmodr� )r r� rt r� r
r
r r� � s:
z!SimpleHTTPServerTestCase.test_getc C sT | j | jd dd�}| �|tj� | �|�d�tt| j ��� | �|�d�d� d S )Nz/testr~ )r5 zcontent-lengthz content-typezapplication/octet-stream)
r3 r� r� r r rU rl �strrs rt )r r� r
r
r � test_head� s z"SimpleHTTPServerTestCase.test_headc C s� t j�� }| j|d< | j| jd |d�}| �|tj� | j t
j dd� }t j�� }t j j
|dd�|d< | j| jd |d�}| �|tj� dS ) z�Check that when a request to /test is sent with the request header
If-Modified-Since set to date of last modification, the server returns
status code 304, not 200
zIf-Modified-Sincez/test)r7 rK )ZhoursT)r� N)r� �message�Messager� r3 r� r� r rw r� r� � timedeltar� �format_datetime)r r7 r� Znew_dtr
r
r �test_browser_cache s
z+SimpleHTTPServerTestCase.test_browser_cachec C sZ | j }|tjdd� }tj�� }tjj|dd�|d< | j| j d |d�}| �
|t j � d S )Nim )ZdaysT)r� zIf-Modified-Sincez/test)r7 )
r� r� r� r� r� r� r� r� r3 r� r� r r )r ZdtZold_dtr7 r� r
r
r � test_browser_cache_file_changed s
z8SimpleHTTPServerTestCase.test_browser_cache_file_changedc C sB t j�� }| j|d< d|d< | j| jd |d�}| �|tj� d S )NzIf-Modified-Since�*z
If-None-Matchz/test)r7 ) r� r� r� r� r3 r� r� r r )r r7 r� r
r
r �,test_browser_cache_with_If_None_Match_header% s
zESimpleHTTPServerTestCase.test_browser_cache_with_If_None_Match_headerc C sX | j ddd�}| �|tj� | j ddd�}| �|tj� | j ddd�}| �|tj� d S )NrT ZFOO)r5 ZcustomZGETs)r3 r� r rW )r r� r
r
r �test_invalid_requests. s z.SimpleHTTPServerTestCase.test_invalid_requestsc C s@ | � | jd �}| j|tj| jd� |jd }| �|| j� dS )z�Checks that the datetime returned in Last-Modified response header
is the actual datetime of last modification, rounded to the second
z/test)rt z
Last-modifiedN) r3 r� r� r r rt r7 rU r� )r r� r� r
r
r �test_last_modified7 s
z+SimpleHTTPServerTestCase.test_last_modifiedc C s� | � | jd �}| j|tj| jd� | � | jd �}| �|tj� | � | jd �}| �|tj� | � | j�}| �|tj� | � | jd �}| �|tj� | � | jd �}| �|tj� | �|� d�| jd � d S ) Nz/test)rt z/test/rT z/?hi=2z?hi=1r� z/?hi=1)
r3 r� r� r r rt rC r� rU rl )r r� r
r
r � test_path_without_leading_slash@ s z8SimpleHTTPServerTestCase.test_path_without_leading_slashc C s� d}t j�| j|�}yt|d��� W n" tk
rH t�d| ��Y nX z,| � | j
d �}| � |t j
�}|j�� }W d t �|� X | �|� dtj|dd� }| �|�|�|� d S )Nz <test&>.txt�wz-Can not create file %s on current file systemrT z>%s<F)r� )r' rM r# r� r� r: �OSError�unittestZSkipTestr3 r� r� r r r7 Zget_content_charset�unlinkr� r� r� r� rG )r r� Zfullpathr� r6 r� Z html_textr
r
r �test_html_escape_filenameP s
z2SimpleHTTPServerTestCase.test_html_escape_filename)N) r r r r r r r, r. r� r� �skipIfr� r� Z
skipUnlessr r� r� r� r� r� r� r� r� r� r� r� r� r� �
__classcell__r
r
)r� r r� G s*
1& r� zD#!%s
print("Content-type: text/html")
print()
print("Hello World")
z�#!%s
import cgi
print("Content-type: text/html")
print()
form = cgi.FieldStorage()
print("%%s, %%s, %%s" %% (form.getfirst("spam"), form.getfirst("eggs"),
form.getfirst("bacon")))
zR#!%s
import os
print("Content-type: text/html")
print()
print(os.environ["%s"])
r� z7This test can't be run reliably as root (issue #13308).c @ s� e Zd ZG dd� dee�Zej�d�Zdd� Z dd� Z
dd � Z d
d � Z d d
� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )!�CGIHTTPServerTestCasec @ s e Zd ZdS )z%CGIHTTPServerTestCase.request_handlerN)r r r r
r
r
r r � s r �asciic C sp t �| � t�� | _t�� | _tj� | jd�| _
tj� | j
d�| _ t� | j
� t� | j � d | _
d | _d | _d | _d | _t�� r�tj� | jd�| _t�tj| j� ntj| _y| j�d� W n& tk
r� | �� | �d� Y nX tj� | jd�| _
t | j
d��}|� t | j � W d Q R X t� | j
d� tj� | j
d �| _t | jddd
��}|� t | j � W d Q R X t� | jd� tj� | j
d �| _t | jddd
��}|� t | j � W d Q R X t� | jd� tj� | j d �| _t | jddd
��}|� t | j � W d Q R X t� | jd� tj� | j
d
�| _t | jddd
��}|� t!| jdf � W d Q R X t� | jd� t�"| j� d S )Nzcgi-binz child-dir�pythonzutf-8z0Python executable path is not encodable to utf-8znocgi.pyr� i� zfile1.py)�encodingzfile2.pyzfile3.pyzfile4.pyZ QUERY_STRING)#r% r, r' r� r� r� r� �
parent_dirrM r# �cgi_dir�
cgi_child_dirr� �
nocgi_path�
file1_path�
file2_path�
file3_path�
file4_pathr Z can_symlink� pythonexe�symlinkr� �
executablerG �UnicodeEncodeErrorr. ZskipTestr� rI � cgi_file1r� � cgi_file2� cgi_file4r� )r r� Zfile1Zfile2Zfile3Zfile4r
r
r r, � sT
zCGIHTTPServerTestCase.setUpc C s� z�t �| j� | jtjkr&t �| j� | jr8t �| j� | jrJt �| j� | j r\t �| j � | j
rnt �| j
� | j r�t �| j � t � | j
� t � | j� t � | j� W d t�| � X d S )N)r' r� r� r� r� r� �remover� r� r� r� r� �rmdirr� r� r� r% r. )r r
r
r r. � s$ z CGIHTTPServerTestCase.tearDownc C s� dt t ddddddddddddddddd d
d
t d d t ddd
� }x^|�� D ]R\}}t|t�rvt|t�rv| �|tj|� qHt�|�}| j ||d|||f d� qHW d S )Nz//z//\z/cgi-bin/file1.pyz//az/C:/z/a/bz/a/b/z/a/b/dz/a/b/d/fz/a/b/fz/a/fz//f) r z..z/.//..rT z//z/\z/.//zcgi-bin/file1.pyz/cgi-bin/file1.py�az/az//az./az./C:/z/a/bz/a/b/z/a/b/.z /a/b/c/..z /a/b/c/../dz/a/b/c/../d/e/../fz/a/b/c/../d/e/../../fz /a/b/c/../d/e/.././././..//fz ../a/b/c/../d/e/.././././..//fz/a/b/c/../d/e/../../../fz/a/b/c/../d/e/../../../../fz /a/b/c/../d/e/../../../../../fz /a/b/c/../d/e/../../../../f/..z /a/b/c/../d/e/../../../../f/../.z path = %r
Got: %r
Wanted: %r)r� )
�
IndexError�items�
isinstance�type�
issubclass� Exception� assertRaisesr Z_url_collapse_pathrU )r Z test_vectorsrM �expectedZactualr
r
r �test_url_collapse_path� sH
z,CGIHTTPServerTestCase.test_url_collapse_pathc C s: | � d�}| �|�� |�d�|jfd| j dtjf� d S )Nz/cgi-bin/file1.pyz Content-types Hello Worldz text/html)r3 rU r rl rV �linesepr r )r rX r
r
r �test_headers_and_content s
z.CGIHTTPServerTestCase.test_headers_and_contentc C s | � d�}| �|jtj� d S )Nz)///////////nocgi.py/../cgi-bin/nothere.sh)r3 rU rV r rC )r rX r
r
r �test_issue19435 s
z%CGIHTTPServerTestCase.test_issue19435c C sF t j�dddd��}ddi}| �dd||�}| �|�� d | j � d S )
NrK r� i@� )ZspamZeggsZbaconz Content-typez!application/x-www-form-urlencodedz/cgi-bin/file2.pyZPOSTs 1, python, 123456)r� r� Z urlencoder3 rU r r
)r Zparamsr7 rX r
r
r � test_post s
z CGIHTTPServerTestCase.test_postc C s& | � d�}|�� | �|jtj� d S )Nz/cgi-bin/invalid)r3 r rU rV r rC )r rX r
r
r �test_invaliduri s
z%CGIHTTPServerTestCase.test_invaliduric C sR ddt �d� i}| jdd|d�}| �d| j dtjf|�� |�d �|j f� d S )
Ns
Authorizations Basic s
username:passz/cgi-bin/file1.pyr/ )r7 s Hello Worldz text/htmlz Content-type)
�base64Z b64encoder3 rU r
r r r rl rV )r r7 rX r
r
r �test_authorization s z(CGIHTTPServerTestCase.test_authorizationc C s: | � d�}| �d| j dtjf|�� |�d�|jf� d S )Nzcgi-bin/file1.pys Hello Worldz text/htmlz Content-type)r3 rU r
r r r rl rV )r rX r
r
r �test_no_leading_slash% s
z+CGIHTTPServerTestCase.test_no_leading_slashc C sZ d}|t jd< | �d�}| �d| j dtjf|�� |�d�|j f� | �t jd |� d S )NzTest CGI ServerZSERVER_SOFTWAREz/cgi-bin/file1.pys Hello Worldz text/htmlz Content-type)
r' r( r3 rU r
r r r rl rV )r Z signaturerX r
r
r � test_os_environ_is_not_altered, s
z4CGIHTTPServerTestCase.test_os_environ_is_not_alteredc C s: | � d�}| �d| j dtjf|�� |�d�|jf� d S )Nz/cgi-bin%2ffile1.pys Hello Worldz text/htmlz Content-type)r3 rU r
r r r rl rV )r rX r
r
r �#test_urlquote_decoding_in_cgi_check5 s
z9CGIHTTPServerTestCase.test_urlquote_decoding_in_cgi_checkc C s: | � d�}| �d| j dtjf|�� |�d�|jf� d S )Nz/cgi-bin/child-dir/file3.pys Hello Worldz text/htmlz Content-type)r3 rU r
r r r rl rV )r rX r
r
r � test_nested_cgi_path_issue21323; s
z5CGIHTTPServerTestCase.test_nested_cgi_path_issue21323c C s: | � d�}| �d| j dtjf|�� |�d�|jf� d S )Nz/cgi-bin/file4.py?a=b?c=ds a=b?c=dz text/htmlz Content-type)r3 rU r
r r r rl rV )r rX r
r
r �&test_query_with_multiple_question_markA s
z<CGIHTTPServerTestCase.test_query_with_multiple_question_markc C s: | � d�}| �d| j dtjf|�� |�d�|jf� d S )Nz0/cgi-bin/file4.py?k=aa%2F%2Fbb&//q//p//=//a//b//s k=aa%2F%2Fbb&//q//p//=//a//b//z text/htmlz Content-type)r3 rU r
r r r rl rV )r rX r
r
r �"test_query_with_continuous_slashesG s
z8CGIHTTPServerTestCase.test_query_with_continuous_slashesN)r r r r r r r' r
rG r, r. r r r r
r r r r r r r r r
r
r
r r� � s :* r� c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�SocketlessRequestHandlerc s2 t �� }t� |j_t� �|d d � d| _d| _d S )NFzHTTP/1.1) r ZMockr ZmakefileZ return_valuer� r �
get_calledrP )r r �kwargsr3 )r� r
r r P s
z!SocketlessRequestHandler.__init__c C s6 d| _ | �tj� | �dd� | �� | j�d� d S )NTz Content-Typez text/htmls <html><body>Data</body></html>
)r r; r r r= r> rH rI )r r
r
r r� X s
z SocketlessRequestHandler.do_GETc G s d S )Nr
)r rz r r
r
r r _ s z$SocketlessRequestHandler.log_message)r r r r r� r r� r
r
)r� r r O s r c @ s e Zd Zdd� ZdS )�!RejectingSocketlessRequestHandlerc C s | � tj� dS )NF)rA r ZEXPECTATION_FAILED)r r
r
r �handle_expect_100c s z3RejectingSocketlessRequestHandler.handle_expect_100N)r r r r r
r
r
r r b s r c @ s0 e Zd Zdd� Zdd� Zdd� Zedd� �Zd S )
�AuditableBytesIOc C s
g | _ d S )N)�datas)r r
r
r r j s zAuditableBytesIO.__init__c C s | j �|� d S )N)r �append)r rt r
r
r rI m s zAuditableBytesIO.writec C s d� | j�S )Nrv )r# r )r r
r
r �getDatap s zAuditableBytesIO.getDatac C s
t | j�S )N)rs r )r r
r
r � numWritess s zAuditableBytesIO.numWritesN)r r r r rI r �propertyr r
r
r
r r h s r c @ s� e Zd ZdZe�d�Zdd� Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1S )2� BaseHTTPRequestHandlerTestCasezrTest the functionality of the BaseHTTPServer.
Test the support for the Expect 100-continue header.
s HTTP/1.[0-9]+ 200 OKc C s t � | _d S )N)r �handler)r r
r
r r, � s z$BaseHTTPRequestHandlerTestCase.setUpc C s: t |�}t � }|| j_|| j_| j�� |�d� |�� S )Nr )r r# �rfilerH �handle_one_request�seek� readlines)r r� �input�outputr
r
r �send_typical_request� s
z3BaseHTTPRequestHandlerTestCase.send_typical_requestc C s | � | jj� d S )N)rp r# r )r r
r
r �verify_get_called� s z0BaseHTTPRequestHandlerTestCase.verify_get_calledc s0 x*dD ]"� | � t� fdd�|D ��d� qW d S )N)s Server: s Date: s Content-Type: c 3 s | ]}|� � �V qd S )N)r� )�.0�h)� fieldNamer
r � <genexpr>� s zIBaseHTTPRequestHandlerTestCase.verify_expected_headers.<locals>.<genexpr>rK )rU �sum)r r7 r
)r. r �verify_expected_headers� s
z6BaseHTTPRequestHandlerTestCase.verify_expected_headersc C s | j �|�}| �|� d S )N)�HTTPResponseMatch�searchr� )r r� �matchr
r
r �verify_http_server_response� s z:BaseHTTPRequestHandlerTestCase.verify_http_server_responsec C s� d| j _tj}tj�tdt� ��:}|| j d� || j d� || j dd� || j dd� W d Q R X |� � }| �
d|� | �
d |� |� � }| � d|d
� | � d |d � | � d
|d � | � d|d � d S )N)z 127.0.0.1i9 �stderrz/fooz/bar z /spam %s.r u beans�� r z/\x1bbar\x00\x1brK z/spam a.� u /spam \x1b\x7f\x9f beans.� )
r# Zclient_addressr r r Zpatch�objectr� r
r� Z assertNotIn�
splitlinesr� )r r Z fake_stderrr6 r� r
r
r �test_unprintable_not_logged� s z:BaseHTTPRequestHandlerTestCase.test_unprintable_not_loggedc C s� | � d�}| �|d � | �|dd� � | �� | �|d d� | �| jjd� | �| jjd� | �| jjd� | �| jj d � | �
| jj � � d
� d S ) Ns GET / HTTP/1.1
r rK ���s <html><body>Data</body></html>
zGET / HTTP/1.1r/ rT zHTTP/1.1r
)
r* r5 r1 r+ rU r# � requestline�commandrM �request_version�assertSequenceEqualr7 r )r �resultr
r
r �
test_http_1_1� s
z,BaseHTTPRequestHandlerTestCase.test_http_1_1c C s� | � d�}| �|d � | �|dd� � | �� | �|d d� | �| jjd� | �| jjd� | �| jjd� | �| jj d � | �
| jj � � d
� d S ) Ns GET / HTTP/1.0
r rK r> s <html><body>Data</body></html>
zGET / HTTP/1.0r/ rT zHTTP/1.0r
)
r* r5 r1 r+ rU r# r? r@ rM rA rB r7 r )r rC r
r
r �
test_http_1_0� s
z,BaseHTTPRequestHandlerTestCase.test_http_1_0c C s6 | � d�}| �t|�d� | �|d d� | �� d S )Ns GET / HTTP/0.9
rK r s <html><body>Data</body></html>
)r* rU rs r+ )r rC r
r
r �
test_http_0_9� s
z,BaseHTTPRequestHandlerTestCase.test_http_0_9c C sH | � d�}| �|d �d�� | �|d|�d�� � | �| jj� d S )Ns) GET /spaced out HTTP/1.1
Host: dummy
r s
HTTP/1.1 400 rK s
)r* rp r� r1 �indexr� r# r )r rC r
r
r �test_extra_space� s
z/BaseHTTPRequestHandlerTestCase.test_extra_spacec C s� | � d�}| �|d � | �|dd� � | �� | �|d d� | �| jjd� | �| jjd� | �| jjd� | �| jj d � d
}| �
| jj � � |� d S ) Ns( GET / HTTP/1.0
Expect: 100-continue
r rK r> s <html><body>Data</body></html>
zGET / HTTP/1.0r/ rT zHTTP/1.0))�Expectz 100-continue)
r* r5 r1 r+ rU r# r? r@ rM rA rB r7 r )r rC r7 r
r
r �test_with_continue_1_0� s
z5BaseHTTPRequestHandlerTestCase.test_with_continue_1_0c C s� | � d�}| �|d d� | �|d d� | �|d d� | �|dd� � | �� | �|d d � | �| jjd
� | �| jjd � | �| jjd � | �| jjd
� d}| � | jj
� � |� d S )Ns( GET / HTTP/1.1
Expect: 100-continue
r s HTTP/1.1 100 Continue
rK s
r9 s HTTP/1.1 200 OK
r> s <html><body>Data</body></html>
zGET / HTTP/1.1r/ rT zHTTP/1.1))rI z 100-continue) r* rU r1 r+ r# r? r@ rM rA rB r7 r )r rC r7 r
r
r �test_with_continue_1_1� s
z5BaseHTTPRequestHandlerTestCase.test_with_continue_1_1c C sN t d�}t� }t� }||_||_d|_d|_d |_|�d� | � |j
d� d S )Ns GET / HTTP/1.1
zHTTP/1.1r i� r9 ) r r r r$ rH rA r? r@ rA rU r )r r( r) r# r
r
r �#test_header_buffering_of_send_error� s
zBBaseHTTPRequestHandlerTestCase.test_header_buffering_of_send_errorc C sX t d�}t� }t� }||_||_d|_|�d� | �|jd� |� � | �|jd� d S )Ns GET / HTTP/1.1
zHTTP/1.1i� r rK )
r r r r$ rH rA Zsend_response_onlyrU r r> )r r( r) r# r
r
r �+test_header_buffering_of_send_response_only� s
zJBaseHTTPRequestHandlerTestCase.test_header_buffering_of_send_response_onlyc C sv t d�}t� }t� }||_||_d|_|�dd� |�dd� | �|jd� |� � | �|�
� d� | �|jd� d S ) Ns GET / HTTP/1.1
zHTTP/1.1ZFooZfooZbarr s Foo: foo
bar: bar
rK ) r r r r$ rH rA r= rU r r> r )r r( r) r# r
r
r �$test_header_buffering_of_send_header
s zCBaseHTTPRequestHandlerTestCase.test_header_buffering_of_send_headerc C s� dd� }t d�}t � }|| j_|| j_d| j_| j�� | �||�d� ||��d�}| �|d d� | �|d d� | �|d
d � d S ) Nc S s( | � � }| �d� | �� }| �|� |S )Nr )�tellr& r )r� �posrt r
r
r �_readAndReseek s
z[BaseHTTPRequestHandlerTestCase.test_header_unbuffered_when_continue.<locals>._readAndReseeks( GET / HTTP/1.1
Expect: 100-continue
zHTTP/1.1rv s
r s HTTP/1.1 100 ContinuerK r9 s HTTP/1.1 200 OK) r r# r$ rH rA r% ZassertNotEqualr� rU )r rQ r( r) rC r
r
r �$test_header_unbuffered_when_continue s
zCBaseHTTPRequestHandlerTestCase.test_header_unbuffered_when_continuec C st | j }t� | _ | �d�}| �|d d� | �|dd� � | �| j j� | �tdd� |dd� D ��d� || _ d S )Ns( GET / HTTP/1.1
Expect: 100-continue
r s! HTTP/1.1 417 Expectation Failed
rK r> c s s | ]}|d kV qdS )s Connection: close
Nr
)r, �rr
r
r r/ 9 s zMBaseHTTPRequestHandlerTestCase.test_with_continue_rejected.<locals>.<genexpr>)r# r r* rU r1 r� r r0 )r Z
usual_handlerrC r
r
r �test_with_continue_rejected0 s
"z:BaseHTTPRequestHandlerTestCase.test_with_continue_rejectedc C sD | � ddd �}| �|d d� | �| jj� | �| jjt� d S )Ns GET � xi r s# HTTP/1.1 414 Request-URI Too Long
)r* rU r� r# r ZassertIsInstancer? r� )r rC r
r
r �test_request_length< s z2BaseHTTPRequestHandlerTestCase.test_request_lengthc C sH | � ddd d �}| �|d d� | �| jj� | �| jjd� d S )Ns GET / HTTP/1.1
X-Foo: bar� ri s
r s HTTP/1.1 431 Line too long
zGET / HTTP/1.1)r* rU r� r# r r? )r rC r
r
r �test_header_lengthD s
z1BaseHTTPRequestHandlerTestCase.test_header_lengthc C s< | � d�}| �|d d� | �| jj� | �| jjd� d S )Ns� GET / HTTP/1.1
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
r s HTTP/1.1 431 Too many headers
zGET / HTTP/1.1)r* rU r� r# r r? )r rC r
r
r �test_too_many_headersL s
z4BaseHTTPRequestHandlerTestCase.test_too_many_headersc C s8 | � d�}d�|�}d}| �tj|dd��d�|� d S )Ns* <script>alert("hello")</script> / HTTP/1.1rv z <script>alert("hello")</script>F)r� r� )r* r# r� r� r� rG )r rC �textr
r
r �test_html_escape_on_errorS s
z8BaseHTTPRequestHandlerTestCase.test_html_escape_on_errorc sZ � �fdd�}|�j _td�� �j �� ��tt� � td�� �j �� ��tt� � d S )Nc s t � ��j_d S )N)�nextr# Zclose_connectionr
)� close_valuesr r
r r% ] s zPBaseHTTPRequestHandlerTestCase.test_close_connection.<locals>.handle_one_request)T)FFT)r# r% �iterZhandler �
StopIterationr\ )r r% r
)r] r r �test_close_connectionZ s
z4BaseHTTPRequestHandlerTestCase.test_close_connectionc C sd t � � }t �|�\ }}}}}}}} }
d| jj| || jj| ||||f } | �| jj|d�| � d S )Nz#%s, %02d %3s %4d %02d:%02d:%02d GMT)r� )�timeZgmtimer# Z weekdaynameZ monthnamerU Zdate_time_string) r ZnowZyearZmonthZdayZhhZmmZssZwd�y�zr r
r
r �test_date_time_stringi s
z4BaseHTTPRequestHandlerTestCase.test_date_time_stringN) r r r �__doc__�re�compiler2 r, r* r+ r1 r5 r= rD rE rF rH rJ rK rL rM rN rR rT rV rX rY r[ r` rd r
r
r
r r" x s2
r" c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) � SimpleHTTPRequestHandlerTestCasez Test url parsing c C s( t �� | _t j�| jd�| _t� | _d S )Nr� )r' r� �
translatedrM r# r r# )r r
r
r r, x s
z&SimpleHTTPRequestHandlerTestCase.setUpc C sR | j �d�}| �|| j� | j �d�}| �|| j� | j �d�}| �|| j� d S )Nz /filenamez/filename?foo=barz/filename?a=b&spam=eggs#zot)r# �translate_pathrU ri )r rM r
r
r �test_query_arguments} s z5SimpleHTTPRequestHandlerTestCase.test_query_argumentsc C s8 | j �d�}| �|| j� | j �d�}| �|| j� d S )Nz
//filenamez//filename?foo=bar)r# rj rU ri )r rM r
r
r � test_start_with_double_slash� s z=SimpleHTTPRequestHandlerTestCase.test_start_with_double_slashc C s� t �tjdt��� | j�d�}|�tjtj�}| � || j
� | j�d�}|�tjtj�}| � || j
� | j�d�}|�tjtj�}| � || j
� | j�d�}|�tjtj�}| � || j
� W d Q R X d S )NrM zc:c:c:foo/filenamez\c:../filenamezc:\c:..\foo/filenamezc:c:foo\c:c:bar/filename) r Z swap_attrr r' �ntpathr# rj r� �seprU ri )r rM r
r
r �test_windows_colon� s z3SimpleHTTPRequestHandlerTestCase.test_windows_colonN)r r r re r, rk rl ro r
r
r
r rh v s
rh c @ s e Zd Zdd� ZdS )� MiscTestCasec C sj g }dddh}xHt t�D ]<}|�d�s||kr0qtt|�}t|dd �dkr|�|� qW | �tj|� d S )Nr� Z
nobody_uidr� �_r z http.server)r� r r� �getattrr ZassertCountEqual�__all__)r r Z blacklistr� Z
module_objectr
r
r �test_all� s
zMiscTestCase.test_allN)r r r rt r
r
r
r rp � s rp c C s4 t �� }zt�tttttt t
� W d t � |� X d S )N) r' r� r Z run_unittestr� r" r8 r� r� rh rp r� )�verboser� r
r
r � test_main� s
rv �__main__)N)8re Z http.serverr r r r r0 r r r' r� rf r rm r� Z
email.messager� Z email.utilsr� Z http.clientZ urllib.parser� r� ra r� r r� r �ior r
r� r r r r ZTestCaser% r8 r� r� r� r� r� r� �hasattrr� r� r r r r" rh rp rv r r
r
r
r �<module> s` P$ '
I (