#!/usr/bin/python3
# Copyright 2021 Magomed Kostoev
# Published under MIT License
import os
import sys
import urllib
from importlib.machinery import SourceFileLoader
from shutil import which
import timeit
import urllib.request
import subprocess
from threading import Thread
import filecmp
sys.path.append('test')
import common
use_umka = False
def log(s, end="\n"):
print(s, end=end, flush=True)
def execute(s, mute=False):
mute = ">/dev/null" if mute else ""
code = os.system(f"{s}{mute}")
if code:
print(f"Command returned {code}: \"{s}\"")
exit(-1)
def stage(name, command, mute=False):
print(f"{name}... ", end="")
execute(command, mute=mute)
print("Done.")
def download(link, path):
log(f"Downloading {path}... ", end="")
urllib.request.urlretrieve(link, path)
log("Done.")
def tool_exists(name):
assert(type(name) == str)
return which(name) is not None
def check_tools(tools):
assert(type(tools) == list)
for name_package_pair in tools:
assert(type(name_package_pair) == list)
assert(len(name_package_pair) == 2)
assert(type(name_package_pair[0]) == str)
assert(type(name_package_pair[1]) == str)
not_exists = []
for name, package in tools:
if not tool_exists(name):
not_exists.append((name, package))
if len(not_exists) != 0:
log("Sorry, I can't find some tools:")
header_name = "Name"
header_package = "Package (probably)"
max_name_len = len(header_name)
max_package_name_len = len(header_package)
for name, package in not_exists:
if len(package) > max_package_name_len:
max_package_name_len = len(package)
if len(name) > max_name_len:
max_name_len = len(name)
def draw_row(name, package):
log((f" | {name.ljust(max_name_len)}" +
f" | {package.ljust(max_package_name_len)} |"))
def draw_line():
log(f" +-{'-' * max_name_len}-+-{'-' * max_package_name_len}-+")
draw_line()
draw_row(header_name, header_package)
draw_line()
for name, package in not_exists:
draw_row(name, package)
draw_line()
exit()
def prepare_test_img():
# TODO: Always recompile the kernel (after build system is done?)
# Get IMG
if not os.path.exists("kolibri_test.img"):
if len(sys.argv) == 1:
download("http://builds.kolibrios.org/eng/data/data/kolibri.img",
"kolibri_test.img")
else:
builds_eng = sys.argv[1]
execute(f"cp {builds_eng}/data/data/kolibri.img kolibri_test.img")
# Open the IMG
with open("kolibri_test.img", "rb") as img:
img_data = img.read()
img = common.Floppy(img_data)
# Remove unuseful folders
img.delete_path("GAMES")
img.delete_path("DEMOS")
img.delete_path("3D")
# Get test kernel
if not os.path.exists("kernel.mnt.pretest"):
if len(sys.argv) == 1:
execute("tup kernel.mnt.pretest")
else:
builds_eng = sys.argv[1]
kernel_mnt_pretest_subpath = "data/kernel/trunk/kernel.mnt.pretest"
kernel_mnt_pretest = f"{builds_eng}/{kernel_mnt_pretest_subpath}"
execute(f"cp {kernel_mnt_pretest} kernel.mnt.pretest", mute=True)
# Put the kernel into IMG
with open("kernel.mnt.pretest", "rb") as kernel_mnt_pretest:
kernel_mnt_pretest_data = kernel_mnt_pretest.read()
img.add_file_path("KERNEL.MNT", kernel_mnt_pretest_data)
img.save("kolibri_test.img")
def collect_tests():
tests = []
# Collect tests from test folder (not recursively yet)
for test_folder in os.listdir("test"):
test_folder_path = f"test/{test_folder}"
test_file = f"{test_folder_path}/test.py"
if not os.path.isdir(test_folder_path):
continue
if os.path.exists(test_file):
tests.append(test_folder_path)
return tests
def run_tests_serially_thread(test, root_dir):
test_number = 1
for test in tests:
test_dir = f"{root_dir}/{test}"
print(f"[{test_number}/{len(tests)}] {test}... ", end="", flush=True)
start = timeit.default_timer()
try:
loader = SourceFileLoader("test", f"{test_dir}/test.py")
loader.load_module().run(root_dir, test_dir)
except common.TestTimeoutException:
result = "TIMEOUT"
except common.TestFailureException:
result = "FAILURE"
else:
result = "SUCCESS"
finish = timeit.default_timer()
print(f"{result} ({finish - start:.2f} seconds)")
test_number += 1
def run_tests_serially(tests, root_dir):
thread = Thread(target=run_tests_serially_thread, args=(tests, root_dir))
thread.start()
return thread
def build_umka_asm(object_output_dir):
umka_o = f"{object_output_dir}/umka.o"
kolibri_kernel_trunk_runtests_py = os.path.abspath(__file__)
kolibri_kernel_trunk = os.path.dirname(kolibri_kernel_trunk_runtests_py)
kolibri_kernel = os.path.dirname(kolibri_kernel_trunk)
kolibrios_folder = os.path.dirname(kolibri_kernel)
env = os.environ
libcrash = "programs/develop/libraries/libcrash/hash"
env["INCLUDE"] = ""
env["INCLUDE"] += f"{kolibrios_folder}/kernel/trunk;"
env["INCLUDE"] += f"{kolibrios_folder}/{libcrash}"
command = "fasm "
command += "-dWIN32=1 " if sys.platform == "win32" else ""
command += "-dUEFI=1 -dextended_primary_loader=1 -dUMKA=1 "
command += "umka/umka.asm umka/build/umka.o -s umka/build/umka.fas "
command += "-m 2000000 "
print(command)
stdout = subprocess.check_output(command, shell=True, env=env)
print(stdout.decode("ascii"))
return umka_o
def cc(src, obj, include_path):
if tool_exists("i686-w64-mingw32-gcc"):
compiler = "i686-w64-mingw32-gcc"
elif tool_exists("clang"):
compiler = "clang"
else:
print("No compiler found to compile UMKa (tried i686-w64-mingw32-gcc and clang)")
print(" Please make sure you have installed llvm-mingw or clang")
print(" If they're installled consider updating PATH variable")
command = f"{compiler} "
command += "-Wno-everything -std=c11 -g -O0 -fno-pie -m32 -masm=intel -c "
command += "-D_FILE_OFFSET_BITS=64 -DNDEBUG -D_POSIX_C_SOURCE=200809L "
command += f"-I {include_path} {src} -o {obj}"
print(command)
if os.system(command) != 0:
exit()
def link(objects):
if tool_exists("i686-w64-mingw32-gcc"):
compiler = "i686-w64-mingw32-gcc"
elif tool_exists("clang"):
compiler = "clang"
else:
print("No compiler found to compile UMKa (tried i686-w64-mingw32-gcc and clang)")
print(" Please make sure you have installed llvm-mingw or clang")
print(" If they're installled consider updating PATH variable")
if sys.platform == "linux" or sys.platform == "linux2":
linker_script = "-T umka/umka.ld"
else:
linker_script = "-Wl,/ALIGN:65536 -Wl,/MAP:umka.map "
command = f"{compiler} "
command += "-Wno-everything "
command += f"-no-pie -m32 -o umka_shell.exe -static {linker_script} "
command += " ".join(objects)
print(command)
if os.system(command) != 0:
exit()
def build_umka():
if not use_umka:
return
os.makedirs("umka/build", exist_ok=True)
platform = "win32" if sys.platform == "win32" else "linux"
os.makedirs(f"umka/build/{platform}", exist_ok=True)
c_sources = [
"umka_shell.c",
"shell.c",
"trace.c",
"trace_lbr.c",
"vdisk.c",
"vnet.c",
"getopt.c",
"isatty.c",
"lodepng.c",
f"{platform}/pci.c",
f"{platform}/thread.c",
"util.c",
]
src_obj_pairs = [
(f"umka/{source}", f"umka/build/{source}.o") for source in c_sources
]
for src, obj in src_obj_pairs:
cc(src, obj, "umka/linux")
umka_o = build_umka_asm("umka/build")
objects = [obj for src, obj in src_obj_pairs] + [umka_o]
link(objects)
os.chdir("umka/test")
for test in [t for t in os.listdir(".") if t.endswith(".t")]:
out_log = f"{test[:-2]}.out.log"
ref_log = f"{test[:-2]}.ref.log"
cmd_umka = f"..{os.sep}..{os.sep}umka_shell.exe < {test} > {out_log}"
print(cmd_umka)
os.system(cmd_umka)
with open(out_log, "rb") as f:
crlf = bytes([0x0D, 0x0A])
lf = bytes([0x0A])
out_log_contents = f.read().replace(crlf, lf)
with open(out_log, "wb") as f:
f.write(out_log_contents)
with open(ref_log, "rb") as f:
ref_log_contents = f.read()
if out_log_contents != ref_log_contents:
print("FAILURE")
exit()
os.chdir("../../")
print("SUCCESS")
def download_umka():
if not use_umka:
return
if not os.path.exists("umka"):
if os.system("git clone https://github.com/KolibriOS/umka") != 0:
print("Couldn't clone UMKa repo")
exit()
os.chdir("umka")
if os.system("git checkout trunk") != 0:
print("Couldn't checkout trunk branch of UMKa")
exit()
os.system("git pull")
os.chdir("../")
def download_umka_imgs():
if not use_umka:
return
imgs = [
"fat32_test0.img",
"jfs.img",
"kolibri.img",
"xfs_borg_bit.img",
"xfs_v4_btrees_l2.img",
"xfs_v4_files_s05k_b4k_n8k.img",
"xfs_v4_ftype0_s05k_b2k_n8k_xattr.img",
"xfs_v4_ftype0_s05k_b2k_n8k.img",
"xfs_v4_ftype0_s4k_b4k_n8k.img",
"xfs_v4_ftype1_s05k_b2k_n8k.img",
"xfs_v4_unicode.img",
"xfs_v4_xattr.img",
"xfs_v5_files_s05k_b4k_n8k.img",
"xfs_v5_ftype1_s05k_b2k_n8k.img",
]
for img in imgs:
if not os.path.exists(f"umka/img/{img}"):
download(f"http://ftp.kolibrios.org/users/Boppan/img/{img}",
f"umka/img/{img}")
if __name__ == "__main__":
root_dir = os.getcwd()
# Check available tools
tools = [["qemu-system-i386", "qemu-system-x86"],
["fasm", "fasm"]]
if use_umka:
tools.append(["git", "git"]);
check_tools(tools)
prepare_test_img()
download_umka()
download_umka_imgs()
build_umka()
tests = collect_tests()
serial_executor_thread = run_tests_serially(tests, root_dir)
serial_executor_thread.join()