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
|
#!/usr/bin/python3
from wsgiref.handlers import format_date_time
from email.utils import parsedate
from calendar import timegm
import gzip
import sys
import time
import zlib
import os
from http import HTTPStatus
from urllib.parse import parse_qs
import http.server as http_server
from io import BytesIO
import sys
class RequestHandler(http_server.SimpleHTTPRequestHandler):
def handle_tokens(self):
need_token_path = self.translate_path(self.path) + ".need_token"
if os.path.isfile(need_token_path):
with open(need_token_path, 'r') as content_file:
token_content = content_file.read()
token = None
auth = self.headers.get("Authorization")
if auth and auth.startswith("Bearer "):
token = auth[7:]
if token == None:
self.send_response(HTTPStatus.UNAUTHORIZED, "No token")
self.end_headers()
return True
if token != token_content:
self.send_response(HTTPStatus.UNAUTHORIZED, "Wrong token")
self.end_headers()
return True
return False
def do_GET(self):
if self.handle_tokens():
return None
self.headers.__delitem__("If-Modified-Since")
return super().do_GET()
def run(dir):
RequestHandler.protocol_version = "HTTP/1.0"
httpd = http_server.HTTPServer( ("127.0.0.1", 0), RequestHandler)
host, port = httpd.socket.getsockname()[:2]
with open("httpd-port", 'w') as file:
file.write("%d" % port)
with open("httpd-pid", 'w') as file:
file.write("%d" % os.getpid())
try:
os.write(3, bytes("Started\n", 'utf-8'));
except:
pass
print("Serving HTTP on port %d" % port);
if dir:
os.chdir(dir)
httpd.serve_forever()
if __name__ == '__main__':
dir = None
if len(sys.argv) >= 2 and len(sys.argv[1]) > 0:
dir = sys.argv[1]
run(dir)
|
