mirror of
https://github.com/KDE/kde-linux.git
synced 2025-12-24 00:18:23 -05:00
79 lines
2.5 KiB
Python
Executable File
79 lines
2.5 KiB
Python
Executable File
#!/bin/env python3
|
|
# SPDX-License-Identifier: GPL-2.0-only OR GPL-3.0-only OR LicenseRef-KDE-Accepted-GPL
|
|
# SPDX-FileCopyrightText: 2025 Harald Sitter <sitter@kde.org>
|
|
|
|
import atexit
|
|
import http.server
|
|
import sys
|
|
import subprocess
|
|
import os
|
|
import time
|
|
|
|
from pathlib import Path
|
|
|
|
class Handler(http.server.BaseHTTPRequestHandler):
|
|
def do_GET(self):
|
|
if self.path == '/good':
|
|
sys.exit(0)
|
|
if self.path == '/bad':
|
|
print("==Received /bad callback==")
|
|
content_len = int(self.headers.get('Content-Length'))
|
|
body = self.rfile.read(content_len)
|
|
print(body.decode('utf-8'))
|
|
sys.exit(1)
|
|
self.send_response(200)
|
|
self.end_headers()
|
|
|
|
server = http.server.HTTPServer(server_address=('', 0), RequestHandlerClass=Handler)
|
|
print("serving at port", server.server_port)
|
|
|
|
img = sys.argv[1]
|
|
if not img:
|
|
print("No image specified")
|
|
sys.exit(1)
|
|
test_img = img.replace('.raw', '.test.raw')
|
|
|
|
efi_base = sys.argv[2]
|
|
if not efi_base:
|
|
print("No EFI base image specified")
|
|
sys.exit(1)
|
|
|
|
subprocess.check_call(['cp', '--reflink=auto', img, test_img])
|
|
subprocess.check_call(['systemd-dissect', test_img, '--with', f'{os.path.dirname(os.path.realpath(__file__))}/basic-test-efi-addon.sh'],
|
|
env={'PORT': str(server.server_port),
|
|
'UKI': efi_base},
|
|
stdout=sys.stdout, stderr=sys.stderr)
|
|
|
|
# I ought to point out that this leaks the process in case of failure. It will however get reaped by the docker container shutdown.
|
|
qemu = subprocess.Popen([
|
|
"qemu-system-x86_64",
|
|
"-drive",
|
|
f"file={test_img},format=raw",
|
|
"-m",
|
|
"4G",
|
|
"-enable-kvm",
|
|
"-cpu",
|
|
"host",
|
|
"-bios",
|
|
"/usr/share/OVMF/x64/OVMF.4m.fd",
|
|
])
|
|
atexit.register(lambda: (qemu.kill()))
|
|
|
|
def on_timeout():
|
|
print("\n\n\n== Test timed out ==")
|
|
print("Download the image for inspection from http://images.kde-linux.haraldsitter.eu/")
|
|
print("(location may have changed to something like https://qoomon.github.io/aws-s3-bucket-browser/index.html?bucket=https://storage.kde.org/ci-artifacts/#).")
|
|
print("Once downloaded run:\n")
|
|
print(f"./basic-test.py {img} {efi_base}\n\n\n")
|
|
|
|
qemu.kill()
|
|
sys.exit(2)
|
|
|
|
server.timeout = 5 * 60 # 5 minutes
|
|
server.handle_timeout = on_timeout
|
|
while True: # kinda garbage but there seems to be no nice (non-private) poll-or-timeout api
|
|
server.handle_request()
|
|
time.sleep(8)
|
|
qemu.kill()
|
|
sys.exit(1) # if we get here we timed out = fail
|