mirror of
https://github.com/flatpak/flatpak.git
synced 2026-05-10 09:00:20 -04:00
fad37dfa8c
Add an integration test that verifies the Flatpak-Upgrade-From HTTP header is correctly sent during update operations but absent during fresh installs. This header is used by Flathub to distinguish updates from new installs in download statistics. To support the test, extend web-server.py to optionally log Flatpak-* request headers to a separate file.
83 lines
2.6 KiB
Python
Executable File
83 lines
2.6 KiB
Python
Executable File
#!/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):
|
|
headers_log_file = None
|
|
|
|
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 log_flatpak_headers(self):
|
|
if self.headers_log_file is None:
|
|
return
|
|
|
|
for name in ("Flatpak-Ref", "Flatpak-Upgrade-From", "Flatpak-Is-Update"):
|
|
value = self.headers.get(name)
|
|
if value is not None:
|
|
with open(self.headers_log_file, 'a+') as f:
|
|
f.write("%s: %s\n" % (name, value))
|
|
|
|
def do_GET(self):
|
|
self.log_flatpak_headers()
|
|
if self.handle_tokens():
|
|
return None
|
|
self.headers.__delitem__("If-Modified-Since")
|
|
return super().do_GET()
|
|
|
|
def run(dir, headers_log=None):
|
|
RequestHandler.protocol_version = "HTTP/1.0"
|
|
RequestHandler.headers_log_file = headers_log
|
|
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
|
|
headers_log = None
|
|
if len(sys.argv) >= 2 and len(sys.argv[1]) > 0:
|
|
dir = sys.argv[1]
|
|
if len(sys.argv) >= 3 and len(sys.argv[2]) > 0:
|
|
headers_log = sys.argv[2]
|
|
|
|
run(dir, headers_log)
|