forked from mozilla/gzipServer
-
Notifications
You must be signed in to change notification settings - Fork 0
/
gzipServer.py
executable file
·91 lines (80 loc) · 2.96 KB
/
gzipServer.py
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
#!/c/mozilla-build/python/python.exe
#
# Usage: gzipServer.py [port number to listen on]
#
# Default port number is 80
#
import time
import sys
import http.server
import gzip
import json
import argparse
HOST_NAME = '127.0.0.1'
gPort = None
gReportCount = 0
class HTTPHandler(http.server.BaseHTTPRequestHandler):
#
# Boilerplate web server, for testing with browser
#
def do_HEAD(s):
s.send_response(200)
s.send_header("Content-type", "text/html")
s.end_headers()
def do_GET(s):
global gArgs
s.send_response(200)
s.send_header("Content-type", "text/html")
s.end_headers()
s.wfile.write("<html>")
s.wfile.write(" <head><title>Success</title></head>")
s.wfile.write(" <body>")
s.wfile.write(" <p>The server is working correctly. Firefox should send a POST request on port %d</p>" % gArgs.port)
s.wfile.write(" </body>")
s.wfile.write("</html>")
def do_POST(s):
global gPort
global gReportCount
global gArgs
gReportCount += 1
length = int(s.headers["Content-Length"])
postData = s.rfile.read(length)
plainData = ""
if "Content-Encoding" in s.headers and s.headers["Content-Encoding"] == "gzip":
gzData = postData
gzFile = open('temp.gz', 'wb')
gzFile.write(gzData)
gzFile.close()
unzippedFile = gzip.open('temp.gz', 'rb')
plainData = unzippedFile.read()
unzippedFile.close()
else:
plainData = postData
filename = f"port_{gPort}_report{gReportCount:03}.json"
plainFile = open(filename, "w")
if gArgs.saveRawPings:
plainFile.write(plainData)
else:
jsonData = json.loads(plainData)
json.dump(jsonData, plainFile, sort_keys = False, indent = 2)
plainFile.close()
s.send_response(200)
s.send_header("Content-type", "text/html")
s.end_headers()
print(time.asctime() + "\tReceived submission #" + str(gReportCount) + ":", str(length), "bytes. Saved in", filename)
if gArgs.printHeaders:
for h in s.headers:
print(h + ": " + s.headers[h])
parser = argparse.ArgumentParser(description="Telemetry test server.")
parser.add_argument("-p", "--port", dest="port", default=80, type=int,
help="The port to listen on.")
parser.add_argument("-r", "--raw-pings", dest="saveRawPings", action="store_true",
help="Store the raw pings instead of pretty formatting.")
parser.add_argument("-H", "--headers", dest="printHeaders", action="store_true",
help="Print the headers of received requests")
gArgs = parser.parse_args()
gPort = gArgs.port
server_class = http.server.HTTPServer
httpd = server_class((HOST_NAME, gArgs.port), HTTPHandler)
print("%s\tServer started - %s:%s" % (time.asctime(), HOST_NAME, gArgs.port))
httpd.serve_forever()