0x1949 Team - FAZEMRX - MANAGER
Edit File: test_cgi.cpython-310.pyc
o �b; � @ sT d Z ddlZddlZddlZddlmZ ddlmZmZm Z m Z ddl m Z ddl mZmZmZ ddlmZ ddlmZmZmZmZmZmZ dd lmZm Z dd l m Z dd l m Z m!Z! d Z"d Z#dZ$dZ%dZ&dZ'dZ(dZ)dZ*G dd� dej+�Z,G dd� d�Z-G dd� de-ej.�Z/G dd � d e-ej.�Z0G d d � d ej.�Z1G d d � d ej.�Z2d!d"� Z3dS )#z! Tests for L{twisted.web.twcgi}. � N)�BytesIO)�address�error� interfaces�reactor)�ConnectionLost)�failure�log�util)�unittest)�client�http� http_headers�resource�server�twcgi)�INTERNAL_SERVER_ERROR� NOT_FOUND)�_render)� DummyChannel� DummyRequestz2print("Header: OK") print("") print("cgi output") zJprint("Header: spam") print("Header: eggs") print("") print("cgi output") z+print("XYZ") print("") print("cgi output") zPprint("Server: monkeys") print("Date: last year") print("") print("cgi output") a& # This is an example of a correctly-written CGI script which reads a body # from stdin, which only reads env['CONTENT_LENGTH'] bytes. import os, sys body_length = int(os.environ.get('CONTENT_LENGTH',0)) indata = sys.stdin.read(body_length) print("Header: OK") print("") print("readinput ok") a+ # This is an example of the typical (incorrect) CGI script which expects # the server to close stdin when the body of the request is complete. # A correct CGI should only read env['CONTENT_LENGTH'] bytes. import sys indata = sys.stdin.read() print("Header: OK") print("") print("readallinput ok") zMprint("content-type: text/cgi-duplicate-test") print("") print("cgi output") z�import json import os print("") print("") vals = {x:y for x,y in os.environ.items() if x.startswith("HTTP_")} print(json.dumps(vals)) zkimport cgi fs = cgi.FieldStorage() param = fs.getvalue("param") print("Header: OK") print("") print(param) c @ s e Zd ZejZdS )� PythonScriptN)�__name__� __module__� __qualname__�sys� executable�filter� r r �;/usr/lib/python3/dist-packages/twisted/web/test/test_cgi.pyr c s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� _StartServerAndTearDownMixinc C sH t �� }t�t|�}|�dt|�� t�|�}t � d|�| _ | j � � j S )Ns cgir )r �Resourcer �sibpath�__file__�putChildr r �Siter � listenTCP�p�getHost�port)�self�cgi�root�cgipath�siter r r � startServerh s z(_StartServerAndTearDownMixin.startServerc C s t | dd �r | j�� S d S )Nr' )�getattrr' � stopListening�r* r r r �tearDownp s �z%_StartServerAndTearDownMixin.tearDownc C sJ t j�| �� �}t|d��}|�|� W d � |S 1 s w Y |S )N�wt)�os�path�abspath�mktemp�open�write)r* �source� cgiFilename�cgiFiler r r �writeCGIt s ��z%_StartServerAndTearDownMixin.writeCGIN)r r r r/ r3 r> r r r r r g s r c @ s� e Zd ZdZej�e�sdZdd� Z dd� Z dd� Z d d � Z d d � Z d d� Zdd� Zdd� Zde_dd� Zdd� Zde_dd� Zdd� Zde_d d � Zd d � Zd S )!�CGITestsz, Tests for L{twcgi.FilteredScript}. z5CGI tests require a functional reactor.spawnProcess()c C sV | � t�}| �|�}d|f }|�d�}t�t��d|�}|�tj � |�| j � |S )N�http://localhost:%d/cgi�ascii� GET) r> � DUMMY_CGIr/ �encoder �Agentr �request� addCallback�readBody� _testCGI_1)r* r< �portnum�url�dr r r �test_CGI� s zCGITests.test_CGIc C s | � |dtj�d� � d S )Ns cgi outputrA )� assertEqualr5 �lineseprD )r* �resr r r rI � � zCGITests._testCGI_1c �b � � t�}� �|�}d|f }|�d�}t�t�}|�d|�}|�t � � fdd�}|�|� |S )zc If the CGI script emits a I{Server} or I{Date} header, these are ignored. r@ rA rB c s, � � d| j�d�� � � d| j�d�� d S )N�monkeysr z last year�date)� assertNotIn�headers� getRawHeaders��responser2 r r � checkResponse� s z;CGITests.test_protectedServerAndDate.<locals>.checkResponse) r> �SPECIAL_HEADER_CGIr/ rD r rE r rF rG � discardBody�r* r< rJ rK �agentrL rZ r r2 r �test_protectedServerAndDate� s z$CGITests.test_protectedServerAndDatec rR )z� If the CGI script emits a I{content-type} header, make sure that the server doesn't add an additional (duplicate) one, as per ticket 4786. r@ rA rB c s � � | j�d�dg� | S )Nz content-typeztext/cgi-duplicate-test�rN rV rW rX r2 r r rZ � s �zBCGITests.test_noDuplicateContentTypeHeaders.<locals>.checkResponse) r> �$NO_DUPLICATE_CONTENT_TYPE_HEADER_CGIr/ rD r rE r rF rG r\ r] r r2 r �"test_noDuplicateContentTypeHeaders� s z+CGITests.test_noDuplicateContentTypeHeadersc s| � � t�}� �|�}d|f }|�d�}t�t�}t�dgdgd��}|j d||d�}� fdd �}|� tj � |� |� |S ) zV The CGI script is never called with the Proxy header passed through. r@ rA s foos bar)s Proxys X-Innocent-HeaderrB )rV c s, t �| �d��}� �t|�� �h d�� d S )NrA > � HTTP_HOST�HTTP_CONNECTION�HTTP_X_INNOCENT_HEADER)�json�loads�decoderN �set�keys)rY rV r2 r r rZ � s �z7CGITests.test_noProxyPassthrough.<locals>.checkResponse) r> �HEADER_OUTPUT_CGIr/ rD r rE r r �HeadersrF rG rH )r* r< rJ rK r^ rV rL rZ r r2 r �test_noProxyPassthrough� s � z CGITests.test_noProxyPassthroughc rR )zp If a CGI script emits two instances of the same header, both are sent in the response. r@ rA rB c s � � | j�d�ddg� d S )N�header�spam�eggsr` rX r2 r r rZ � rQ z7CGITests.test_duplicateHeaderCGI.<locals>.checkResponse) r> �DUAL_HEADER_CGIr/ rD r rE r rF rG r\ r] r r2 r �test_duplicateHeaderCGI� s z CGITests.test_duplicateHeaderCGIc s� �� t�}��|�}d|f }|�d�}t�t�}|�d|�}|�t � g � � fdd�}t � |� �� t j |� � �fdd�}|�|� |S )zF Check for the error message in the duplicated header r@ rA rB c s � � t�| �� d S �N)�appendr �textFromEventDict)� eventDict)�loggedMessagesr r � addMessage� s z4CGITests.test_malformedHeaderCGI.<locals>.addMessagec s �� dtd� � � d S )Nz ignoring malformed CGI header: s XYZ)�assertIn�repr��ignored�rw r* r r rZ s �z7CGITests.test_malformedHeaderCGI.<locals>.checkResponse)r> �BROKEN_HEADER_CGIr/ rD r rE r rF rG r\ r � addObserver� addCleanup�removeObserver)r* r< rJ rK r^ rL rx rZ r r} r �test_malformedHeaderCGI� s z CGITests.test_malformedHeaderCGIc C s� t j�| �� �}t|d�� }|�t� W d � n1 s w Y | �|�}t� t �}d|f }|� d�}|� d|�}|� tj� |� | j� |S )Nr4 r@ rA rB )r5 r6 r7 r8 r9 r: � READINPUT_CGIr/ r rE r rD rF rG rH �_test_ReadEmptyInput_1�r* r< r= rJ r^ rK rL r r r �test_ReadEmptyInput s � z CGITests.test_ReadEmptyInput� c C �& dt j� �}|�d�}| �||� d S �Nz readinput okrA �r5 rO rD rN �r* rP �expectedr r r r� � z CGITests._test_ReadEmptyInput_1c C s� t j�| �� �}t|d�� }|�t� W d � n1 s w Y | �|�}t� t �}d|f }|� d�}|j |dt� td��d�}|�tj� |�| j� |S �Nr4 r@ rA s POSTs Here is your stdin)�uri�method� bodyProducer)r5 r6 r7 r8 r9 r: r� r/ r rE r rD rF �FileBodyProducerr rG rH �_test_ReadInput_1r� r r r �test_ReadInput s � � zCGITests.test_ReadInputc C r� r� r� r� r r r r� 3 r� zCGITests._test_ReadInput_1c C s� t j�| �� �}t|d�� }|�t� W d � n1 s w Y | �|�}d|f }|�d�}t � t �j |dt � td��d�}|�t j� |�| j� |S r� )r5 r6 r7 r8 r9 r: �READALLINPUT_CGIr/ rD r rE r rF r� r rG rH �_test_ReadAllInput_1)r* r< r= rJ rK rL r r r �test_ReadAllInput8 s � � zCGITests.test_ReadAllInputc C r� )Nzreadallinput okrA r� r� r r r r� K r� z CGITests._test_ReadAllInput_1c C sX G dd� d�}|� }t ddg�}t�ddd�|_tjd|d �}t||� | �|j� d S ) zw L{twcgi.FilteredScript.runProcess} uses the reactor passed as an argument to the constructor. c @ s e Zd ZdZdZdd� ZdS )z5CGITests.test_useReactorArgument.<locals>.FakeReactorzR A fake reactor recording whether spawnProcess is called. Fc _ s d| _ dS )z� Set the C{called} flag to C{True} if C{spawnProcess} is called. @param args: Positional arguments. @param kwargs: Keyword arguments. TN)�called)r* �args�kwargsr r r � spawnProcess] s zBCGITests.test_useReactorArgument.<locals>.FakeReactor.spawnProcessN)r r r �__doc__r� r� r r r r � FakeReactorV s r� �a�b�TCP� 127.0.0.1�90 z dummy-file�r N) r r � IPv4Addressr r �FilteredScriptr � assertTruer� )r* r� � fakeReactorrF r r r r �test_useReactorArgumentP s z CGITests.test_useReactorArgumentN)r r r r� r �IReactorProcess� providedByr �skiprM rI r_ rb rm rr r� r� �timeoutr� r� r� r� r� r� r r r r r? { s* r? c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �CGIScriptTestsz' Tests for L{twcgi.CGIScript}. c C sP | � t�}| �|�}d|f }t�t�}|�d|�}|�tj� |�| j � |S )zk If the CGI script is passed URL parameters, do not fall over, as per ticket 9887. s" http://localhost:%d/cgi?param=1234rB ) r> �URL_PARAMETER_CGIr/ r rE r rF rG rH �_test_urlParameters_1)r* r< rJ rK r^ rL r r r �test_urlParameterst s z!CGIScriptTests.test_urlParametersc C r� )N�1234rA r� r� r r r r� � r� z$CGIScriptTests._test_urlParameters_1c C sb G dd� d�}|� }t j| �� |d�}tddg�}t�ddd�|_t||� | �|j d d � d S ) zt L{twcgi.CGIScript.render} sets the process environment I{PATH_INFO} from the request path. c @ s e Zd ZdZdd� ZdS )z1CGIScriptTests.test_pathInfo.<locals>.FakeReactorzZ A fake reactor recording the environment passed to spawnProcess. c S s || _ dS )a9 Store the C{env} L{dict} to an instance attribute. @param process: Ignored @param filename: Ignored @param args: Ignored @param env: The environment L{dict} which will be stored @param wdir: Ignored N)� process_env)r* �process�filenamer� �env�wdirr r r r� � s z>CGIScriptTests.test_pathInfo.<locals>.FakeReactor.spawnProcessN)r r r r� r� r r r r r� � s r� r� r� r� r� r� r� � PATH_INFOz/a/bN) r � CGIScriptr8 r r r� r r rN r� )r* r� �_reactorr rF r r r � test_pathInfo� s z CGIScriptTests.test_pathInfoN)r r r r� r� r� r� r r r r r� o s r� c @ � e Zd ZdZdd� Zdd� ZdS )�CGIDirectoryTestsz* Tests for L{twcgi.CGIDirectory}. c s> t ���� �}tdg�� t|� �}� �fdd�}|�|� |S )zc L{twcgi.CGIDirectory.render} sets the HTTP response code to I{NOT FOUND}. � c � �� � jt� d S rs �rN � responseCoder r{ �rF r* r r � cbRendered� � z1CGIDirectoryTests.test_render.<locals>.cbRendered)r � CGIDirectoryr8 r r rG )r* r rL r� r r� r � test_render� s z CGIDirectoryTests.test_renderc sX �� � }t�|� t�|�}tdg�� |�d� �}t|� �}� �fdd�}|�|� |S )a L{twcgi.CGIDirectory.getChild} returns a resource which renders an response with the HTTP I{NOT FOUND} status code if the indicated child does not exist as an entry in the directory used to initialized the L{twcgi.CGIDirectory}. �fooc r� rs r� r{ r� r r r� � r� z8CGIDirectoryTests.test_notFoundChild.<locals>.cbRendered) r8 r5 �makedirsr r� r �getChildr rG )r* r6 r �childrL r� r r� r �test_notFoundChild� s z$CGIDirectoryTests.test_notFoundChildN)r r r r� r� r� r r r r r� � s r� c @ r� )�CGIProcessProtocolTestsz0 Tests for L{twcgi.CGIProcessProtocol}. c C s: t dg�}t�|�}|�t�t�� �� | �|j t � dS )z� If the process communicating with L{CGIProcessProtocol} ends before finishing writing out headers, the response has I{INTERNAL SERVER ERROR} as its status code. r� N) r r �CGIProcessProtocol� processEndedr �Failurer �ProcessTerminatedrN r� r )r* rF �protocolr r r �test_prematureEndOfHeaders� s z2CGIProcessProtocolTests.test_prematureEndOfHeadersc C sH t � }t�|d�}t�|�}|�t�td��� |� t�t � � �� dS )zg Ensure that the CGI process ends cleanly when the request connection is lost. TzConnection doneN) r r �Requestr r� �connectionLostr r� r r� r r� )r* rL rF r� r r r �test_connectionLost� s z+CGIProcessProtocolTests.test_connectionLostN)r r r r� r� r� r r r r r� � s r� c s t �� ��� fdd��S )zn Discard the body of a HTTP response. @param response: The response. @return: The response. c s � S rs r )�_rX r r �<lambda>� s z discardBody.<locals>.<lambda>)r rH rG rX r rX r r\ � s r\ )4r� rf r5 r �ior �twisted.internetr r r r �twisted.internet.errorr �twisted.pythonr r r � twisted.trialr � twisted.webr r r r r r �twisted.web.httpr r �twisted.web.test._utilr � twisted.web.test.requesthelperr r rC rq r~ r[ r� r� ra rk r� r� r r �TestCaser? r� r� r� r\ r r r r �<module> s< u8)