U
j& @ s d Z ddlmZmZmZ ddlmZ ddlm Z ddl
m Z m Z m
Z
mZ ddlmZ ddlmZ G dd d e
ZG d
d d e ZG d d
d
eZG dd de ZG dd deZG dd de ZG dd deZdS )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.
)quoteurlparse
urlunparse)reactor)
ClientFactory)_QUEUED_SENTINEL HTTPChannel
HTTPClientRequest)Resource) NOT_DONE_YETc @ 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 5/tmp/pip-unpacked-wheel-j5h_c8_q/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 connectionMade4 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 handleHeader> s zProxyClient.handleHeaderc C s | j | d S r )r r )r bufferr r r handleResponsePartH 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 handleResponseEndK s
z ProxyClient.handleResponseEndN) __name__
__module__ __qualname____doc__r. r r r% r+ r- r0 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 ^ s zProxyClientFactory.__init__c C s | | j| j| j| j| j| jS r )protocolr r r r r r )r addrr r r
buildProtocolf 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 r/ )r Z connectorreasonr r r clientConnectionFailedk s z)ProxyClientFactory.clientConnectionFailedN) r1 r2 r3 r4 r
r6 r r8 r: r r r r r5 V s
r5 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|krH|d\}}t|}td|dd }|sh|d }| j| }| }d|kr|
d|d< | j dd | j
}|| j|| j||| } | j||| d S ) Nr ascii:) rB / host)r uridecodeportssplitr" r protocols
getAllHeaderscopyencodecontentseekreadmethod clientprotor
connectTCP)
r parsedr6 hostportr class_r s
clientFactoryr r r process s$
zProxyRequest.processN) r1 r2 r3 r4 r5 rJ rH r r r rZ r r r r r; v 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)r1 r2 r3 r4 r; requestFactoryr r r r r[ s r[ 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 rM proxyClientFactoryClassrQ rF rR rK rN rP r rS rV )r rY r r r rZ s zReverseProxyRequest.processN)
r1 r2 r3 r4 r5 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)r1 r2 r3 r4 r] r\ r r r r ra s ra 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 )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|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 )safezutf-8)rb rU rV rc urlquoterM r )r rc requestr r r getChild s z ReverseProxyResource.getChildc C s | j dkr| j}nd| j| j f }|jd|dg |jdd t|jd }|rh| 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:%drE r@ r ?)rV rU r^ r( rM rN rO r rF rc r` rQ rR rK rP r rS r )r rf rU qsr rY r r r render s&
zReverseProxyResource.renderN)
r1 r2 r3 r4 r5 r` r r rg rk r r r r rb s
rb N)r4 urllib.parser re r r Ztwisted.internetr Ztwisted.internet.protocolr Ztwisted.web.httpr r r r
Ztwisted.web.resourcer Ztwisted.web.serverr r
r5 r; r[ r] ra rb r r r r s 7 $$