Subversion Repositories Kolibri OS

#!/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) == tuple)

    for name_package_pair in tools:

        assert(type(name_package_pair) == tuple)

        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:

            with open("lang.inc", "w") as lang_inc:

                lang_inc.write("lang fix en\n")

            execute("fasm bootbios.asm bootbios.bin.pretest -dpretest_build=1")

            command = "fasm "

            command += "-dpretest_build=1 -ddebug_com_base=0xe9 "

            command += "-m 65536 "

            command += "kernel.asm kernel.mnt.pretest"

            execute(command)

        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):

    command = "clang "

    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 sys.platform == "linux" or sys.platform == "linux2":

        linker_script = "-T umka/umka.ld"

    else:

        linker_script = "-Wl,/ALIGN:65536 -Wl,/MAP:umka.map "

    command = "clang "

    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")

    exit()

if __name__ == "__main__":

    root_dir = os.getcwd()

    # Check available tools

    tools = (("qemu-system-i386", "qemu-system-x86"),

             ("fasm", "fasm"))

    check_tools(tools)

    prepare_test_img()

    build_umka()

    tests = collect_tests()

    serial_executor_thread = run_tests_serially(tests, root_dir)

    serial_executor_thread.join()