Rev 9411 | Details | Compare with Previous | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
9367 | Boppan | 1 | import os |
2 | import sys |
||
9368 | Boppan | 3 | import shlex |
9367 | Boppan | 4 | import shutil |
5 | import urllib.request |
||
6 | import subprocess |
||
7 | |||
9374 | Boppan | 8 | path_to_tools_workspace = os.path.dirname(os.path.abspath(__file__)) |
9 | path_to_tools = os.path.dirname(path_to_tools_workspace) |
||
9370 | Boppan | 10 | sys.path.append(path_to_tools) |
9367 | Boppan | 11 | |
9374 | Boppan | 12 | from workspace.build import build |
13 | |||
9390 | Boppan | 14 | from lib.builds import builds_get, builds_get_contents |
9370 | Boppan | 15 | from lib.makeflop import Floppy |
16 | from lib.platform import is_win32, path |
||
9411 | Boppan | 17 | from lib.logging import log, require_tools |
9383 | Boppan | 18 | from lib.constants import tools_cache_kolibri_img |
9367 | Boppan | 19 | |
20 | # TODO: Move into _tools/lib |
||
21 | def run_qemu(start_dir = "workspace"): |
||
9411 | Boppan | 22 | require_tools(("qemu-system-i386",)) |
23 | |||
9367 | Boppan | 24 | qemu_command = f"qemu-system-i386" |
25 | flags = "" |
||
26 | flags += "-L . " # IDK why it does not work without this |
||
27 | flags += "-m 128 " |
||
28 | flags += f"-drive format=raw,file={start_dir}/kolibri.img,index=0,if=floppy -boot a " |
||
29 | flags += "-vga vmware " |
||
30 | flags += "-net nic,model=rtl8139 -net user " |
||
31 | if is_win32(): |
||
32 | qemu_full_path = shutil.which(qemu_command) |
||
9381 | Boppan | 33 | qemu_directory = os.path.dirname(qemu_full_path) |
9367 | Boppan | 34 | flags += f"-L {qemu_directory} " |
35 | s = f"{qemu_command} {flags}" |
||
36 | qemu_stdout = open(f"{start_dir}/qemu_stdout.log", "w") |
||
37 | qemu_stderr = open(f"{start_dir}/qemu_stderr.log", "w") |
||
38 | if is_win32(): |
||
39 | return subprocess.Popen(s, bufsize = 0, stdout = qemu_stdout, stderr = qemu_stderr, stdin = subprocess.DEVNULL, shell = True, start_new_session = True) |
||
40 | else: |
||
41 | a = shlex.split(s) |
||
42 | return subprocess.Popen(a, bufsize = 0, stdout = qemu_stdout, stderr = qemu_stderr, stdin = subprocess.DEVNULL, start_new_session = True) |
||
43 | |||
44 | if __name__ == "__main__": |
||
9377 | Boppan | 45 | program_name = build() |
9367 | Boppan | 46 | |
47 | os.makedirs("workspace", exist_ok = True) |
||
48 | |||
9373 | Boppan | 49 | # Create a copy of IMG |
9390 | Boppan | 50 | kolibri_img = builds_get("eng/data/data/kolibri.img", "workspace/kolibri.img") |
9373 | Boppan | 51 | |
9367 | Boppan | 52 | # Open the IMG |
9390 | Boppan | 53 | with open(kolibri_img, "rb") as img: |
9367 | Boppan | 54 | img_data = img.read() |
55 | img = Floppy(img_data) |
||
56 | |||
57 | # Remove unuseful folders |
||
58 | img.delete_path("GAMES") |
||
59 | img.delete_path("DEMOS") |
||
60 | img.delete_path("3D") |
||
61 | |||
9391 | Boppan | 62 | # Insert faster kernel if no --compressed-kernel flag passed |
63 | if "--compressed-kernel" not in sys.argv: |
||
9389 | Boppan | 64 | new_kernel = builds_get_contents("eng/data/kernel/trunk/kernel.mnt.pretest") |
65 | img.add_file_path("KERNEL.MNT", new_kernel) |
||
66 | |||
9377 | Boppan | 67 | log("Moving program into kolibri image... ", end = "") |
68 | with open(program_name, "rb") as file: |
||
69 | file_data = file.read() |
||
70 | if not img.add_file_path(program_name.upper(), file_data): |
||
71 | print(f"Coudn't move {program_name} into IMG") |
||
9367 | Boppan | 72 | log("Done") |
73 | |||
9377 | Boppan | 74 | log("Adding program to autorun.dat... ", end = "") |
75 | lines_to_add = bytes(f"\r\n/SYS/{program_name.upper()}\t\t""\t0\t# Your program", "ascii") |
||
9373 | Boppan | 76 | autorun_dat = img.extract_file_path("SETTINGS\AUTORUN.DAT") |
77 | place_for_new_lines = autorun_dat.index(b"\r\n/SYS/@TASKBAR")# b"\r\n### Hello, ASM World! ###") |
||
78 | autorun_dat = autorun_dat[:place_for_new_lines] + lines_to_add + autorun_dat[place_for_new_lines:] |
||
79 | img.delete_path("SETTINGS\AUTORUN.DAT") |
||
80 | img.add_file_path("SETTINGS\AUTORUN.DAT", autorun_dat) |
||
81 | log("Done") |
||
82 | |||
9390 | Boppan | 83 | img.save(kolibri_img) |
9373 | Boppan | 84 | |
9367 | Boppan | 85 | run_qemu() |