Rev 9373 | Rev 9376 | Go to most recent revision | 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 | |||
9370 | Boppan | 14 | from lib.makeflop import Floppy |
15 | from lib.platform import is_win32, path |
||
16 | from lib.logging import log |
||
17 | from lib.network import download |
||
9367 | Boppan | 18 | |
19 | # TODO: Move into _tools/lib |
||
20 | def get_file_directory(path): |
||
21 | path = path.replace("\\", "/") |
||
22 | if "/" in path: |
||
23 | folder = "/".join(path.split("/")[:-1]) |
||
24 | if folder == "": |
||
25 | return "/" # It was a file in the root folder |
||
26 | return folder |
||
27 | else: |
||
28 | return "." # Just a filename, let's return current folder |
||
29 | |||
30 | # TODO: Move into _tools/lib |
||
31 | def run_qemu(start_dir = "workspace"): |
||
32 | qemu_command = f"qemu-system-i386" |
||
33 | flags = "" |
||
34 | flags += "-L . " # IDK why it does not work without this |
||
35 | flags += "-m 128 " |
||
36 | flags += f"-drive format=raw,file={start_dir}/kolibri.img,index=0,if=floppy -boot a " |
||
37 | flags += "-vga vmware " |
||
38 | flags += "-net nic,model=rtl8139 -net user " |
||
39 | flags += "-soundhw ac97 " |
||
40 | if is_win32(): |
||
41 | qemu_full_path = shutil.which(qemu_command) |
||
42 | qemu_directory = get_file_directory(qemu_full_path) |
||
43 | flags += f"-L {qemu_directory} " |
||
44 | s = f"{qemu_command} {flags}" |
||
45 | qemu_stdout = open(f"{start_dir}/qemu_stdout.log", "w") |
||
46 | qemu_stderr = open(f"{start_dir}/qemu_stderr.log", "w") |
||
47 | if is_win32(): |
||
48 | return subprocess.Popen(s, bufsize = 0, stdout = qemu_stdout, stderr = qemu_stderr, stdin = subprocess.DEVNULL, shell = True, start_new_session = True) |
||
49 | else: |
||
50 | a = shlex.split(s) |
||
51 | return subprocess.Popen(a, bufsize = 0, stdout = qemu_stdout, stderr = qemu_stderr, stdin = subprocess.DEVNULL, start_new_session = True) |
||
52 | |||
53 | if __name__ == "__main__": |
||
9374 | Boppan | 54 | program_files = build() |
9367 | Boppan | 55 | |
56 | os.makedirs("workspace", exist_ok = True) |
||
57 | |||
9373 | Boppan | 58 | if not os.path.exists("workspace/kolibri.unmodified.img"): |
9367 | Boppan | 59 | img_url = "http://builds.kolibrios.org/eng/data/data/kolibri.img" |
9373 | Boppan | 60 | download(img_url, "workspace/kolibri.unmodified.img") |
9367 | Boppan | 61 | |
9373 | Boppan | 62 | # Create a copy of IMG |
63 | shutil.copyfile("workspace/kolibri.unmodified.img", "workspace/kolibri.img") |
||
64 | |||
9367 | Boppan | 65 | # Open the IMG |
66 | with open("workspace/kolibri.img", "rb") as img: |
||
67 | img_data = img.read() |
||
68 | img = Floppy(img_data) |
||
69 | |||
70 | # Remove unuseful folders |
||
71 | img.delete_path("GAMES") |
||
72 | img.delete_path("DEMOS") |
||
73 | img.delete_path("3D") |
||
74 | |||
75 | log("Moving program files into kolibri image... ", end = "") |
||
76 | for file_name in program_files: |
||
77 | with open(file_name, "rb") as file: |
||
78 | file_data = file.read() |
||
9373 | Boppan | 79 | if not img.add_file_path(file_name.upper(), file_data): |
9367 | Boppan | 80 | print(f"Coudn't move {file_name} into IMG") |
81 | log("Done") |
||
82 | |||
9373 | Boppan | 83 | # TODO: Figure out which of compiled files is a program executable and only run it |
84 | log("Adding program to autorun.dat", end = "") |
||
85 | lines_to_add = b"" |
||
86 | for file_name in program_files: |
||
87 | lines_to_add += bytes(f"\r\n/SYS/{file_name.upper()}\t\t""\t0\t# Your program", "ascii") |
||
88 | autorun_dat = img.extract_file_path("SETTINGS\AUTORUN.DAT") |
||
89 | place_for_new_lines = autorun_dat.index(b"\r\n/SYS/@TASKBAR")# b"\r\n### Hello, ASM World! ###") |
||
90 | autorun_dat = autorun_dat[:place_for_new_lines] + lines_to_add + autorun_dat[place_for_new_lines:] |
||
91 | print(autorun_dat) |
||
92 | img.delete_path("SETTINGS\AUTORUN.DAT") |
||
93 | img.add_file_path("SETTINGS\AUTORUN.DAT", autorun_dat) |
||
94 | log("Done") |
||
95 | |||
96 | img.save("workspace/kolibri.img") |
||
97 | |||
9367 | Boppan | 98 | run_qemu() |