mirror of
https://github.com/theupdateframework/python-tuf
synced 2026-05-24 10:08:28 +00:00
87 lines
2.3 KiB
Python
Executable file
87 lines
2.3 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
# Copyright 2012 - 2017, New York University and the TUF contributors
|
|
# SPDX-License-Identifier: MIT OR Apache-2.0
|
|
|
|
"""
|
|
<Program>
|
|
simple_server.py
|
|
|
|
<Author>
|
|
Konstantin Andrianov.
|
|
|
|
<Started>
|
|
February 15, 2012.
|
|
|
|
<Copyright>
|
|
See LICENSE-MIT or LICENSE for licensing information.
|
|
|
|
<Purpose>
|
|
This is a basic server that was designed to be used in conjunction with
|
|
test_download.py to test download.py module.
|
|
|
|
<Reference>
|
|
SimpleHTTPServer:
|
|
http://docs.python.org/library/simplehttpserver.html#module-SimpleHTTPServer
|
|
"""
|
|
|
|
# Help with Python 3 compatibility, where the print statement is a function, an
|
|
# implicit relative import is invalid, and the '/' operator performs true
|
|
# division. Example: print 'hello world' raises a 'SyntaxError' exception.
|
|
from __future__ import print_function
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import unicode_literals
|
|
|
|
import sys
|
|
import random
|
|
|
|
import six
|
|
from six.moves.SimpleHTTPServer import SimpleHTTPRequestHandler
|
|
|
|
PORT = 0
|
|
|
|
def _port_gen():
|
|
return random.randint(30000, 45000)
|
|
|
|
if len(sys.argv) > 1:
|
|
try:
|
|
PORT = int(sys.argv[1])
|
|
if PORT < 30000 or PORT > 45000:
|
|
raise ValueError
|
|
|
|
except ValueError:
|
|
PORT = _port_gen()
|
|
|
|
else:
|
|
PORT = _port_gen()
|
|
|
|
|
|
class QuietHTTPRequestHandler(SimpleHTTPRequestHandler):
|
|
"""A SimpleHTTPRequestHandler that does not write incoming requests to
|
|
stderr. """
|
|
def log_request(self, code='-', size='-'):
|
|
pass
|
|
|
|
# NOTE: On Windows/Python2 tests that use this simple_server.py in a
|
|
# subprocesses hang after a certain amount of requests (~68), if a PIPE is
|
|
# passed as Popen's stderr argument. This problem doesn't emerge if
|
|
# we silence the HTTP messages.
|
|
# If you decide to receive the HTTP messages, then this bug
|
|
# could reappear.
|
|
use_quiet_http_request_handler = True
|
|
|
|
if len(sys.argv) > 2:
|
|
use_quiet_http_request_handler = sys.argv[2]
|
|
|
|
if use_quiet_http_request_handler:
|
|
handler = QuietHTTPRequestHandler
|
|
else:
|
|
handler = SimpleHTTPRequestHandler
|
|
|
|
# Allow re-use so you can re-run tests as often as you want even if the
|
|
# tests re-use ports. Otherwise TCP TIME-WAIT prevents reuse for ~1 minute
|
|
six.moves.socketserver.TCPServer.allow_reuse_address = True
|
|
httpd = six.moves.socketserver.TCPServer(('', PORT), handler)
|
|
|
|
httpd.serve_forever()
|