mirror of
https://github.com/theupdateframework/python-tuf
synced 2026-05-24 10:08:28 +00:00
34db5095aa
and disable unused sections for clarity as well. Signed-off-by: Sebastien Awwad <sebastien.awwad@gmail.com>
137 lines
3.3 KiB
Python
137 lines
3.3 KiB
Python
#!/usr/bin/env python
|
|
|
|
# Copyright 2018, New York University and the TUF contributors
|
|
# SPDX-License-Identifier: MIT OR Apache-2.0
|
|
|
|
"""
|
|
<Program>
|
|
simple_proxy.py
|
|
|
|
<Copyright>
|
|
See LICENSE-MIT OR LICENSE for licensing information.
|
|
|
|
<Purpose>
|
|
Provide a simple http proxy server that can be used by TUF tests.
|
|
See test_proxy_use.py to see how it's used in the tests.
|
|
|
|
Arguments are optional, but:
|
|
|
|
1. Proxy type:
|
|
If provided, the first argument is assumed to choose HTTP vs HTTPS proxy
|
|
vs HTTP CONNECT supporting HTTP proxy.
|
|
Currently, only basic HTTP is supported. The tests use proxy2.py for
|
|
the other two types of proxy.
|
|
The basic HTTP proxy is currently provided by dependency twisted.
|
|
|
|
2. Port:
|
|
If provided, the port on which the proxy should run. Userspace ports
|
|
only (1023<port<65536). If the port is not provided or not in that
|
|
range, a random port will be chosen in the smaller range
|
|
8090<port<30000.
|
|
|
|
Example use:
|
|
python simple_proxy.py http_dumb 8989
|
|
|
|
|
|
"""
|
|
# Help with Python 3 compatibility, where the print statement is a function, an
|
|
# implicit relative import is invalid, and the '/' operator performs true
|
|
# division.
|
|
from __future__ import print_function
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import unicode_literals
|
|
|
|
import sys
|
|
import random
|
|
|
|
import twisted.web.proxy
|
|
import twisted.web.http
|
|
import twisted.internet
|
|
import twisted.python.log
|
|
|
|
def _generate_random_port():
|
|
return random.randint(8090, 30000)
|
|
|
|
|
|
|
|
|
|
|
|
def main():
|
|
port = 0
|
|
proxy_type = 'http_dumb'
|
|
|
|
# if len(sys.argv) > 1:
|
|
# if sys.argv[1].lower() in ['https', 'http_smart', 'http_dumb']:
|
|
# proxy_type = sys.argv[1].lower()
|
|
# else:
|
|
# print('\n\nUNEXPECTED PROTOCOL FOR PROXY SERVER. Using ' + proxy_type +
|
|
# ' instead.\n\n')
|
|
|
|
if len(sys.argv) > 2:
|
|
try:
|
|
port = int(sys.argv[2])
|
|
except:
|
|
print('Ignoring argument. Second argument expected to be port, an int '
|
|
'i > 1023 and i < 65536. Generating randomly between 8090 and 30000.')
|
|
pass
|
|
|
|
if port < 1024 or port > 65536:
|
|
port = _generate_random_port()
|
|
|
|
if proxy_type == 'http_dumb':
|
|
run_http_dumb_proxy(port)
|
|
# elif proxy_type == 'https':
|
|
# run_https_proxy(port)
|
|
# elif proxy_type == 'http_smart':
|
|
# run_http_smart_proxy(port)
|
|
else:
|
|
raise Exception('Unexpected proxy type requested....')
|
|
|
|
|
|
|
|
|
|
|
|
def run_http_dumb_proxy(port):
|
|
"""
|
|
This proxy doesn't know what to do with HTTP CONNECT requests, so it can't
|
|
pass on HTTPS requests.
|
|
"""
|
|
|
|
twisted.python.log.startLogging(sys.stdout)
|
|
|
|
class ProxyFactory(twisted.web.http.HTTPFactory):
|
|
protocol = twisted.web.proxy.Proxy
|
|
|
|
twisted.internet.reactor.listenTCP(port, ProxyFactory())
|
|
twisted.internet.reactor.run()
|
|
|
|
|
|
|
|
|
|
|
|
# def run_http_smart_proxy(port):
|
|
# """
|
|
# This proxy needs to support HTTP CONNECT requests so that it can create a
|
|
# TCP tunnel for HTTPS requests to the target server to go through.
|
|
# """
|
|
# raise NotImplementedError()
|
|
|
|
|
|
|
|
|
|
|
|
# def run_https_proxy(port):
|
|
# """
|
|
# Run a proxy that connects using HTTPS (has its own validated certificate
|
|
# that the client accepts, etc.) before passing the request on to the target
|
|
# server.
|
|
# """
|
|
# raise NotImplementedError()
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|