| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139 |
- #!/usr/bin/env python
- #
- # Simple asynchronous HTTP proxy with tunnelling (CONNECT).
- #
- # GET/POST proxying based on
- # http://groups.google.com/group/python-tornado/msg/7bea08e7a049cf26
- #
- # Copyright (C) 2012 Senko Rasic <senko.rasic@dobarkod.hr>
- #
- # Permission is hereby granted, free of charge, to any person obtaining a copy
- # of this software and associated documentation files (the "Software"), to deal
- # in the Software without restriction, including without limitation the rights
- # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
- # copies of the Software, and to permit persons to whom the Software is
- # furnished to do so, subject to the following conditions:
- #
- # The above copyright notice and this permission notice shall be included in
- # all copies or substantial portions of the Software.
- #
- # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
- # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
- # THE SOFTWARE.
- import sys
- import socket
- import tornado.httpserver
- import tornado.ioloop
- import tornado.iostream
- import tornado.web
- import tornado.httpclient
- __all__ = ['ProxyHandler', 'run_proxy']
- class ProxyHandler(tornado.web.RequestHandler):
- SUPPORTED_METHODS = ['GET', 'POST', 'CONNECT']
- @tornado.web.asynchronous
- def get(self):
- def handle_response(response):
- if response.error and not isinstance(response.error,
- tornado.httpclient.HTTPError):
- self.set_status(500)
- self.write('Internal server error:\n' + str(response.error))
- self.finish()
- else:
- self.set_status(response.code)
- for header in ('Date', 'Cache-Control', 'Server',
- 'Content-Type', 'Location'):
- v = response.headers.get(header)
- if v:
- self.set_header(header, v)
- if response.body:
- self.write(response.body)
- self.finish()
- req = tornado.httpclient.HTTPRequest(
- url=self.request.uri,
- method=self.request.method, body=self.request.body,
- headers=self.request.headers, follow_redirects=False,
- allow_nonstandard_methods=True)
- client = tornado.httpclient.AsyncHTTPClient()
- try:
- client.fetch(req, handle_response)
- except tornado.httpclient.HTTPError as e:
- if hasattr(e, 'response') and e.response:
- self.handle_response(e.response)
- else:
- self.set_status(500)
- self.write('Internal server error:\n' + str(e))
- self.finish()
- @tornado.web.asynchronous
- def post(self):
- return self.get()
- @tornado.web.asynchronous
- def connect(self):
- host, port = self.request.uri.split(':')
- client = self.request.connection.stream
- def read_from_client(data):
- upstream.write(data)
- def read_from_upstream(data):
- client.write(data)
- def client_close(data=None):
- if upstream.closed():
- return
- if data:
- upstream.write(data)
- upstream.close()
- def upstream_close(data=None):
- if client.closed():
- return
- if data:
- client.write(data)
- client.close()
- def start_tunnel():
- client.read_until_close(client_close, read_from_client)
- upstream.read_until_close(upstream_close, read_from_upstream)
- client.write(b'HTTP/1.0 200 Connection established\r\n\r\n')
- s = socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0)
- upstream = tornado.iostream.IOStream(s)
- upstream.connect((host, int(port)), start_tunnel)
- def run_proxy(port, start_ioloop=True):
- """
- Run proxy on the specified port. If start_ioloop is True (default),
- the tornado IOLoop will be started immediately.
- """
- app = tornado.web.Application([
- (r'.*', ProxyHandler),
- ])
- app.listen(port)
- ioloop = tornado.ioloop.IOLoop.instance()
- if start_ioloop:
- ioloop.start()
- if __name__ == '__main__':
- port = 8888
- if len(sys.argv) > 1:
- port = int(sys.argv[1])
- print("Starting HTTP proxy on port %d" % port)
- run_proxy(port)
|