forked from KolibriOS/kolibrios
Magomed Kostoev (mkostoevr)
282c7690c8
git-svn-id: svn://kolibrios.org@9374 a494cfbc-eb01-0410-851d-a64ba20cac60
99 lines
3.5 KiB
Python
99 lines
3.5 KiB
Python
import os
|
|
import sys
|
|
import shlex
|
|
import shutil
|
|
import urllib.request
|
|
import subprocess
|
|
|
|
path_to_tools_workspace = os.path.dirname(os.path.abspath(__file__))
|
|
path_to_tools = os.path.dirname(path_to_tools_workspace)
|
|
sys.path.append(path_to_tools)
|
|
|
|
from workspace.build import build
|
|
|
|
from lib.makeflop import Floppy
|
|
from lib.platform import is_win32, path
|
|
from lib.logging import log
|
|
from lib.network import download
|
|
|
|
# TODO: Move into _tools/lib
|
|
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
|
|
|
|
# TODO: Move into _tools/lib
|
|
def run_qemu(start_dir = "workspace"):
|
|
qemu_command = f"qemu-system-i386"
|
|
flags = ""
|
|
flags += "-L . " # IDK why it does not work without this
|
|
flags += "-m 128 "
|
|
flags += f"-drive format=raw,file={start_dir}/kolibri.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}"
|
|
qemu_stdout = open(f"{start_dir}/qemu_stdout.log", "w")
|
|
qemu_stderr = open(f"{start_dir}/qemu_stderr.log", "w")
|
|
if is_win32():
|
|
return subprocess.Popen(s, bufsize = 0, stdout = qemu_stdout, stderr = qemu_stderr, stdin = subprocess.DEVNULL, shell = True, start_new_session = True)
|
|
else:
|
|
a = shlex.split(s)
|
|
return subprocess.Popen(a, bufsize = 0, stdout = qemu_stdout, stderr = qemu_stderr, stdin = subprocess.DEVNULL, start_new_session = True)
|
|
|
|
if __name__ == "__main__":
|
|
program_files = build()
|
|
|
|
os.makedirs("workspace", exist_ok = True)
|
|
|
|
if not os.path.exists("workspace/kolibri.unmodified.img"):
|
|
img_url = "http://builds.kolibrios.org/eng/data/data/kolibri.img"
|
|
download(img_url, "workspace/kolibri.unmodified.img")
|
|
|
|
# Create a copy of IMG
|
|
shutil.copyfile("workspace/kolibri.unmodified.img", "workspace/kolibri.img")
|
|
|
|
# Open the IMG
|
|
with open("workspace/kolibri.img", "rb") as img:
|
|
img_data = img.read()
|
|
img = Floppy(img_data)
|
|
|
|
# Remove unuseful folders
|
|
img.delete_path("GAMES")
|
|
img.delete_path("DEMOS")
|
|
img.delete_path("3D")
|
|
|
|
log("Moving program files into kolibri image... ", end = "")
|
|
for file_name in program_files:
|
|
with open(file_name, "rb") as file:
|
|
file_data = file.read()
|
|
if not img.add_file_path(file_name.upper(), file_data):
|
|
print(f"Coudn't move {file_name} into IMG")
|
|
log("Done")
|
|
|
|
# TODO: Figure out which of compiled files is a program executable and only run it
|
|
log("Adding program to autorun.dat", end = "")
|
|
lines_to_add = b""
|
|
for file_name in program_files:
|
|
lines_to_add += bytes(f"\r\n/SYS/{file_name.upper()}\t\t""\t0\t# Your program", "ascii")
|
|
autorun_dat = img.extract_file_path("SETTINGS\AUTORUN.DAT")
|
|
place_for_new_lines = autorun_dat.index(b"\r\n/SYS/@TASKBAR")# b"\r\n### Hello, ASM World! ###")
|
|
autorun_dat = autorun_dat[:place_for_new_lines] + lines_to_add + autorun_dat[place_for_new_lines:]
|
|
print(autorun_dat)
|
|
img.delete_path("SETTINGS\AUTORUN.DAT")
|
|
img.add_file_path("SETTINGS\AUTORUN.DAT", autorun_dat)
|
|
log("Done")
|
|
|
|
img.save("workspace/kolibri.img")
|
|
|
|
run_qemu()
|