4,13 → 4,24 |
|
import io |
import os |
import sys |
import subprocess |
import timeit |
import time |
import shlex |
import signal |
import shutil |
from . makeflop import Floppy |
|
def is_win32(): |
return True if sys.platform == "win32" else False |
|
def is_linux(): |
return True if sys.platform == "linux" or sys.platform == "linux2" else False |
|
def is_osx(): |
return True if sys.platform == "darwin" else False |
|
class TestTimeoutException(Exception): |
pass |
|
24,7 → 35,6 |
while not os.path.exists("debug.log"): |
self.wait() |
self.debug = open("debug.log", "rb") |
self.monitor_in = open("monitor.pipe.in", "wb") |
|
def wait_for_debug_log(self, needle, timeout = 1): |
needle = bytes(needle, "utf-8") |
51,6 → 61,10 |
self.timeout() |
|
def kill(self): |
if is_win32(): |
# FIXME: This is shit, isn't there anything better? |
subprocess.Popen(f"TASKKILL /F /PID {self.popen.pid} /T", stdout = subprocess.DEVNULL, stderr = subprocess.DEVNULL, stdin = subprocess.DEVNULL) |
else: |
os.killpg(os.getpgid(self.popen.pid), signal.SIGTERM) |
|
def failure(self): |
64,32 → 78,41 |
def wait(self, seconds = 0.25): |
time.sleep(seconds) |
|
def send_keys(self, keys): |
for key in keys.split(): |
self.send_monitor_command(f"sendkey {key}") |
def get_file_directory(path): |
path = path.replace("\\", "/") |
if "/" in path: |
folder = "/".join(path.split("/")[:-1]) |
if folder == "": |
return "/" # It was a file in the root folder |
return folder |
else: |
return "." # Just a filename, let's return current folder |
|
def take_screenshot(self, fname): |
self.send_monitor_command(f"screendump {fname}") |
def run_qemu(): |
qemu_command = f"qemu-system-i386" |
flags = "" |
flags += "-nographic " # Makes it faster |
flags += "-debugcon file:debug.log " # 0xe9 port output |
flags += "-L . " # IDK why it does not work without this |
flags += "-m 128 " |
flags += "-drive format=raw,file=../../kolibri_test.img,index=0,if=floppy -boot a " |
flags += "-vga vmware " |
flags += "-net nic,model=rtl8139 -net user " |
flags += "-soundhw ac97 " |
if is_win32(): |
qemu_full_path = shutil.which(qemu_command) |
qemu_directory = get_file_directory(qemu_full_path) |
flags += f"-L {qemu_directory} " |
s = f"{qemu_command} {flags}" |
if is_win32(): |
return subprocess.Popen(s, bufsize = 0, stdout = subprocess.DEVNULL, stderr = subprocess.DEVNULL, stdin = subprocess.DEVNULL, shell = True, start_new_session = True) |
else: |
a = shlex.split(s) |
return subprocess.Popen(a, bufsize = 0, stdout = subprocess.DEVNULL, stderr = subprocess.DEVNULL, stdin = subprocess.DEVNULL, start_new_session = True) |
|
def send_monitor_command(self, command): |
self.monitor_in.write(bytes(command + "\n", "utf-8")) |
self.monitor_in.flush() |
|
def images_diff(self, i0, i1, expect=True): |
diff = bool(os.system(f"perceptualdiff {i0} {i1} > /dev/null")) |
if diff != expect: |
self.failure() |
|
def run(): |
if os.path.exists("debug.log"): |
os.remove("debug.log") |
if os.path.exists("monitor.pipe.in"): |
os.remove("monitor.pipe.in") |
if os.path.exists("monitor.pipe.out"): |
os.remove("monitor.pipe.out") |
os.system("mkfifo monitor.pipe.in monitor.pipe.out") |
s = f"qemu-system-i386 -nographic -monitor pipe:monitor.pipe -debugcon file:debug.log -L . -m 128 -drive format=raw,file=../../kolibri_test.img,index=0,if=floppy -boot a -vga vmware -net nic,model=rtl8139 -net user -soundhw ac97" |
a = shlex.split(s) |
popen = subprocess.Popen(a, bufsize = 0, stdout = subprocess.DEVNULL, stderr = subprocess.DEVNULL, stdin = subprocess.DEVNULL, start_new_session = True) |
popen = run_qemu() |
return Qemu(popen) |
|