File: //lib/python3/dist-packages/twisted/web/__pycache__/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 <"'#