mirror of
https://github.com/Kitware/CMake.git
synced 2026-01-12 09:20:41 -06:00
116b06870d
In order to abort transfers on slow connections the ExternalProject command support passing the INACTIVITY_TIMEOUT argument. Fixes: #20992
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
from http.server import HTTPServer, BaseHTTPRequestHandler
|
|
import argparse
|
|
import time
|
|
import subprocess
|
|
import sys
|
|
import os
|
|
args = None
|
|
outerthread = None
|
|
|
|
class SimpleHTTPRequestHandler(BaseHTTPRequestHandler):
|
|
def do_GET(self):
|
|
self.send_response(200)
|
|
self.end_headers()
|
|
data = b'D'
|
|
|
|
if args.speed_limit:
|
|
slow_deadline = time.time()+args.limit_duration
|
|
|
|
while time.time() < slow_deadline:
|
|
self.wfile.write(data)
|
|
if args.speed_limit:
|
|
time.sleep(1.1)
|
|
|
|
data = data * 100
|
|
self.wfile.write(data)
|
|
self.close_connection = True
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--speed_limit', help='transfer rate limitation', action='store_true',default=False)
|
|
parser.add_argument('--limit_duration', help='duration of the transfer rate limitation',default=1, type=float)
|
|
parser.add_argument('--file', help='file to write the url to connect to')
|
|
parser.add_argument('--subprocess', action='store_true')
|
|
args = parser.parse_args()
|
|
if not args.subprocess:
|
|
subprocess.Popen([sys.executable]+sys.argv+['--subprocess'],stdin=subprocess.DEVNULL, stderr=subprocess.DEVNULL,stdout=subprocess.DEVNULL)
|
|
else:
|
|
httpd = HTTPServer(('localhost', 0), SimpleHTTPRequestHandler)
|
|
with open(args.file,"w") as f:
|
|
f.write('http://localhost:{}/test'.format(httpd.socket.getsockname()[1]))
|
|
httpd.handle_request()
|
|
os.remove(args.file)
|