Your IP : 216.73.216.213
B
;��]m& � @ 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 �D/opt/alt/python37/lib64/python3.7/site-packages/twisted/web/proxy.py�__init__) s zProxyClient.__init__c C sN | � | j| j� x"| j�� D ]\}}| �||� q W | �� | 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 )N)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 )N)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, r/ 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 )N)r r r r r 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 )N)�protocolr r r r r r )r �addrr 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 r. )r Z connector�reasonr r r �clientConnectionFailedr s z)ProxyClientFactory.clientConnectionFailedN) r0 r1 r2 r3 r r5 r r7 r9 r r r r r4 \ s
r4 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 )N)r r r )r �channel�queuedr 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�:)� rA � � /s host)r �urlparse�uri�decode�ports�splitr! �
urlunparse� protocols�
getAllHeaders�copy�encode�content�seek�read�method� clientprotor �
connectTCP)
r �parsedr5 �host�portr �class_r �s�
clientFactoryr r r �process� s&
zProxyRequest.processN) r0 r1 r2 r3 r4 rJ rG r
r r rZ 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)r0 r1 r2 r3 r: �requestFactoryr r r r r[ � s
r[ 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 )N)r r r )r 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.
s hostr? N)�requestHeadersr' �factoryrU rM �proxyClientFactoryClassrQ rE rR rK rN rP r rS rV )r rY r r r rZ � s zReverseProxyRequest.processN)
r0 r1 r2 r3 r4 r` r
r r rZ 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)r0 r1 r2 r3 r] r\ r r r r ra � s ra 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 )aU
@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{bytes}
N)r r rU rV �pathr )r rU rV rc 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.
rC rA )�safezutf-8)rb rU rV rc r rM r )r rc �requestr r r �getChild s "z ReverseProxyResource.getChildc C s� | j dkr| j}nd| j| j f }|j�d|�d�g� |j�dd� t�|j �d }|rj| j
d | }n| j
}| � |j ||j
|�� |j�� |�}| j�| j| j |� tS )zJ
Render a request by forwarding it to the proxied server.
r; z%s:%ds hostr? r � � ?)rV rU r^ r' rM rN rO r rD rE rc r` rQ rR rK rP r rS r )r re rU �qsr rY r r r �render s
zReverseProxyResource.renderN)
r0 r1 r2 r3 r4 r` r r rf rj r r r r rb � s
rb N)r3 �
__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 r4 r: r[ r] ra rb r r r r �<module> s <"'#