← Back
Editing: proxy.cpython-38.pyc
U W[k& � @ s� d Z ddlmZmZ ddlmZmZ ddlmZ ddl m Z ddl m Z ddl mZ ddlmZmZmZmZ G d d � d e�ZG d d � d e �ZG d d� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de �ZdS )aT Simplistic HTTP proxy support. This comes in two main variants - the Proxy and the ReverseProxy. When a Proxy is in use, a browser trying to connect to a server (say, www.yahoo.com) will be intercepted by the Proxy, and the proxy will covertly connect to the server, and return the result. When a ReverseProxy is in use, the client connects directly to the ReverseProxy (say, www.yahoo.com) which farms off the request to one of a pool of servers, and returns the result. Normally, a Proxy is used on the client end of an Internet connection, while a ReverseProxy is used on the server end. � )�absolute_import�division)� urllib_parse�urlquote)�reactor)� ClientFactory)�Resource)� NOT_DONE_YET)� HTTPClient�Request� HTTPChannel�_QUEUED_SENTINELc @ sD e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dS )� ProxyClientz� Used by ProxyClientFactory to implement a simple web proxy. @ivar _finished: A flag which indicates whether or not the original request has been finished yet. Fc C sD || _ || _|| _d|kr |d= d|d< |�dd � || _|| _d S )Ns proxy-connections closes connections keep-alive)�father�command�rest�pop�headers�data��selfr r �versionr r r � r �3/usr/lib/python3/dist-packages/twisted/web/proxy.py�__init__) s zProxyClient.__init__c C sJ | � | j| j� | j�� D ]\}}| �||� q| �� | j�| j � d S �N) Z sendCommandr r r �itemsZ sendHeaderZ endHeaders� transport�writer )r �header�valuer r r �connectionMade5 s zProxyClient.connectionMadec C s | j �t|�|� d S r )r �setResponseCode�int)r r �code�messager r r � handleStatus= s zProxyClient.handleStatusc C s4 |� � dkr | jj�||g� n| jj�||� d S )N)s servers dates content-type)�lowerr �responseHeaders� setRawHeaders� addRawHeader)r �keyr r r r � handleHeaderA s zProxyClient.handleHeaderc C s | j �|� d S r )r r )r �bufferr r r �handleResponsePartL s z ProxyClient.handleResponsePartc C s$ | j s d| _ | j�� | j�� dS )z� Finish the original request, indicating that the response has been completely written to it, and disconnect the outgoing transport. TN)� _finishedr �finishr ZloseConnection)r r r r �handleResponseEndP s z ProxyClient.handleResponseEndN) �__name__� __module__� __qualname__�__doc__r/ r r! r& r, r. r1 r r r r r s r c @ s, e Zd ZdZeZdd� Zdd� Zdd� ZdS ) �ProxyClientFactoryz? Used by ProxyRequest to implement a simple web proxy. c C s( || _ || _|| _|| _|| _|| _d S r )r r r r r r r r r r r d s zProxyClientFactory.__init__c C s | � | j| j| j| j| j| j�S r )�protocolr r r r r r )r Zaddrr r r � buildProtocolm s �z ProxyClientFactory.buildProtocolc C s8 | j �dd� | j j�dd� | j �d� | j �� dS )zh Report a connection failure in a response to the incoming request as an error. i� s Gateway errors Content-Types text/htmls <H1>Could not connect</H1>N)r r" r( r* r r0 )r Z connector�reasonr r r �clientConnectionFailedr s z)ProxyClientFactory.clientConnectionFailedN) r2 r3 r4 r5 r r7 r r8 r: r r r r r6 \ s r6 c @ s6 e Zd ZdZdeiZddiZeefdd�Z dd� Z dS ) � ProxyRequestz� Used by Proxy to implement a simple web proxy. @ivar reactor: the reactor used to create connections. @type reactor: object providing L{twisted.internet.interfaces.IReactorTCP} s http�P c C s t �| ||� || _d S r �r r r �r ZchannelZqueuedr r r r r � s zProxyRequest.__init__c C s� t �| j�}|d }|d �d�}| j| }d|krJ|�d�\}}t|�}t �d|dd � �}|sl|d }| j| }| � � � � }d|kr�|� d�|d< | j � dd� | j �� }|| j|| j||| �} | j�||| � d S ) Nr � �ascii�:)� rB � � /� host)r �urlparse�uri�decode�ports�splitr# Z urlunparse� protocols� getAllHeaders�copy�encode�content�seek�read�method� clientprotor � connectTCP) r Zparsedr7 �host�portr Zclass_r �s� clientFactoryr r r �process� s* �zProxyRequest.processN) r2 r3 r4 r5 r6 rK rI r r r rY r r r r r; ~ s r; c @ s e Zd ZdZeZdS )�Proxyao This class implements a simple web proxy. Since it inherits from L{twisted.web.http.HTTPChannel}, to use it you should do something like this:: from twisted.web import http f = http.HTTPFactory() f.protocol = Proxy Make the HTTPFactory a listener on a port as per usual, and you have a fully-functioning web proxy! N)r2 r3 r4 r5 r; �requestFactoryr r r r rZ � s rZ c @ s* e Zd ZdZeZeefdd�Zdd� Z dS )�ReverseProxyRequestal Used by ReverseProxy to implement a simple reverse proxy. @ivar proxyClientFactoryClass: a proxy client factory class, used to create new connections. @type proxyClientFactoryClass: L{ClientFactory} @ivar reactor: the reactor used to create connections. @type reactor: object providing L{twisted.internet.interfaces.IReactorTCP} c C s t �| ||� || _d S r r= r> r r r r � s z ReverseProxyRequest.__init__c C sZ | j �d| jj�d�g� | �| j| j| j| � � | j � � | �}| j � | jj| jj|� dS )z� Handle this request by connecting to the proxied server and forwarding it there, then forwarding the response back as the response to this request. rE r@ N)�requestHeadersr) �factoryrU rN �proxyClientFactoryClassrR rG rS rL rO rQ r rT rV )r rX r r r rY � s � ��zReverseProxyRequest.processN) r2 r3 r4 r5 r6 r_ r r r rY r r r r r\ � s r\ c @ s e Zd ZdZeZdS )� ReverseProxyzo Implements a simple reverse proxy. For details of usage, see the file examples/reverse-proxy.py. N)r2 r3 r4 r5 r\ r[ r r r r r` � s r` c @ s0 e Zd ZdZeZefdd�Zdd� Zdd� Z dS ) �ReverseProxyResourcea� Resource that renders the results gotten from another server Put this resource in the tree to cause everything below it to be relayed to a different server. @ivar proxyClientFactoryClass: a proxy client factory class, used to create new connections. @type proxyClientFactoryClass: L{ClientFactory} @ivar reactor: the reactor used to create connections. @type reactor: object providing L{twisted.internet.interfaces.IReactorTCP} c C s& t �| � || _|| _|| _|| _dS )aS @param host: the host of the web server to proxy. @type host: C{str} @param port: the port of the web server to proxy. @type port: C{port} @param path: the base path to fetch data from. Note that you shouldn't put any trailing slashes in it, it will be added automatically in request. For example, if you put B{/foo}, a request on B{/bar} will be proxied to B{/foo/bar}. Any required encoding of special characters (such as " " or "/") should have been done already. @type path: C{str} N)r r rU rV �pathr )r rU rV rb r r r r r � s z ReverseProxyResource.__init__c C s, t | j| j| jd t|dd��d� | j�S )z� Create and return a proxy resource with the same proxy configuration as this one, except that its path also contains the segment given by C{path} at the end. rD rB )Zsafezutf-8)ra rU rV rb r rN r )r rb �requestr r r �getChild s �z ReverseProxyResource.getChildc C s� | j dkr| j}n| jd t| j � }|j�d|�d�g� |j�dd� t� |j �d }|rn| j d | }n| j }| � |j ||j|�� |j�� |�}| j�| j| j |� tS )zJ Render a request by forwarding it to the proxied server. r< rA rE r@ r � � ?)rV rU �strr] r) rN rO rP r rF rG rb r_ rR rS rL rQ r rT r )r rc rU Zqsr rX r r r �render s& �zReverseProxyResource.renderN) r2 r3 r4 r5 r6 r_ r r rd rh r r r r ra � s ra N)r5 Z __future__r r Ztwisted.python.compatr r Ztwisted.internetr Ztwisted.internet.protocolr Ztwisted.web.resourcer Ztwisted.web.serverr Ztwisted.web.httpr r r r r r6 r; rZ r\ r` ra r r r r �<module> s <"'#
Save File
Cancel