U
W[k& @ s d Z ddlmZmZ ddlmZmZ ddlmZ ddl m
Z
ddl m Z ddl
mZ ddlmZmZmZmZ G d d
d
eZG d d d e
ZG d
d deZG dd deZG dd deZG dd deZG 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_importdivision) urllib_parseurlquote)reactor)
ClientFactory)Resource) NOT_DONE_YET)
HTTPClientRequest 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)fathercommandrestpopheadersdataselfr 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 transportwriter )r headervaluer r r connectionMade5 s
zProxyClient.connectionMadec C s | j t|| d S r )r setResponseCodeint)r r codemessager 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| jS r )protocolr r r r r r )r Zaddrr r r
buildProtocolm s
z ProxyClientFactory.buildProtocolc C s8 | j dd | j jdd | 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 Could not connect
N)r r" r( r* r r0 )r Z connectorreasonr 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 ZdZdeiZddiZeefddZ 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 httpP 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 urlparseuridecodeportssplitr# Z
urlunparse protocols
getAllHeaderscopyencodecontentseekreadmethod clientprotor
connectTCP)
r Zparsedr7 hostportr 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 ZdZeZeefddZdd 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| jjdg | | 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 ZdZeZefddZdd 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|ddd | jS )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 }|jd|dg |jdd 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 s <"'#