o
b& @ 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|v r|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)
sendCommandr r r items
sendHeader
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 s6 | dv r| jj||g d S | 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 sd| _ | j | j dS dS )z
Finish the original request, indicating that the response has been
completely written to it, and disconnect the outgoing transport.
TN) _finishedr finishr loseConnection)r r r r handleResponseEndK s
z ProxyClient.handleResponseEndN) __name__
__module__ __qualname____doc__r1 r r# r( r. r0 r4 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 r2 )r connectorreasonr r r clientConnectionFailedk s z)ProxyClientFactory.clientConnectionFailedN) r5 r6 r7 r8 r
r: r r< r? r r r r r9 V s r9 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 t | || || _d S r r
r r r channelqueuedr r r r r
zProxyRequest.__init__c
C s t | j}|d }|d d}| j| }d|v r$|d\}}t|}td|dd }|s4|d }| j| }| }d|vrJ|
d|d< | j dd | j
}|| j|| j||| } | j||| d S ) Nr ascii:) rK / host)r uridecodeportssplitr% r protocols
getAllHeaderscopyencodecontentseekreadmethod clientprotor
connectTCP)
r parsedr: hostportr class_r s
clientFactoryr r r process s$
zProxyRequest.processN) r5 r6 r7 r8 r9 rS rQ r r r rc r r r r r@ v s r@ c @ 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)r5 r6 r7 r8 r@ requestFactoryr r r r re s re 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 rB r rC rD r r r r rG 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.
rN rI N)requestHeadersr+ factoryr^ rV proxyClientFactoryClassrZ rO r[ rT rW rY r r\ r_ )r rb r r r rc s zReverseProxyRequest.processN)
r5 r6 r7 r8 r9 rj r r r rc r r r r rg s
rg c @ rd ) ReverseProxyzo
Implements a simple reverse proxy.
For details of usage, see the file examples/reverse-proxy.py.
N)r5 r6 r7 r8 rg rf r r r r rk s rk 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 r^ r_ pathr )r r^ r_ rm 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.
rM rK )safezutf-8)rl r^ r_ rm urlquoterV r )r rm requestr r r getChild s z ReverseProxyResource.getChildc C s | j dkr | j}nd| j| j f }|jd|dg |jdd t|jd }|r4| j d | }n| j }|
|j ||j |
|j |}| j| j| j | tS )zJ
Render a request by forwarding it to the proxied server.
rA z%s:%drN rI r ?)r_ r^ rh r+ rV rW rX r rO rm rj rZ r[ rT rY r r\ r )r rp r^ qsr rb r r r render s&
zReverseProxyResource.renderN)
r5 r6 r7 r8 r9 rj r r rq ru r r r r rl s
rl N)r8 urllib.parser ro r r twisted.internetr twisted.internet.protocolr twisted.web.httpr r r r
twisted.web.resourcer twisted.web.serverr r
r9 r@ re rg rk rl r r r r s 7 $$