1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
from http.server import HTTPServer, BaseHTTPRequestHandler
import argparse
import time
import subprocess
import sys
import os
import threading
args = None
outerthread = None
barrier = threading.Barrier(2)
class SimpleHTTPRequestHandler(BaseHTTPRequestHandler):
def do_GET(self):
barrier.wait()
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
def runServer(fileName):
httpd = HTTPServer(('localhost', 0), SimpleHTTPRequestHandler)
with open(fileName,"w") as f:
f.write('http://localhost:{}/test'.format(httpd.socket.getsockname()[1]))
httpd.handle_request()
os.remove(fileName)
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:
serverThread = threading.Thread(target=runServer,args=(args.file,))
serverThread.daemon = True
serverThread.start()
barrier.wait(60)
serverThread.join(20)
|
