178 lines
6.3 KiB
Python
178 lines
6.3 KiB
Python
import os
|
||
import shutil
|
||
import subprocess
|
||
import time
|
||
from multiprocessing import Pool
|
||
|
||
|
||
ARCH_FLAGS = "-m64 -march=x86-64 -mabi=sysv -mno-red-zone -mcmodel=kernel -MMD -MP"
|
||
WARN_FLAGS = "-Wall -Wextra -nostdlib "
|
||
STANDART_FLAGS = "-std=gnu11"
|
||
PROTECT_FLAGS = "-O0 -pipe -ffreestanding -fno-stack-protector -fno-lto -fno-stack-check -fno-PIC -fno-PIE"
|
||
CHARSET_FLAGS = "-finput-charset=UTF-8 -fexec-charset=cp1251"
|
||
LIBS_FLAGS = "-Ilimine -Iinclude"
|
||
|
||
def version_build():
|
||
with open("include/version.h", "r") as file:
|
||
lines = file.readlines()
|
||
|
||
major = 0
|
||
minor = 0
|
||
build = 0
|
||
|
||
with open("include/version.h", "w") as file:
|
||
for line in lines:
|
||
if line.startswith("#define VERSION_BUILD"):
|
||
parts = line.split()
|
||
build = int(parts[2]) + 1
|
||
if build > 999:
|
||
build = 0
|
||
minor += 1
|
||
file.write(f"#define VERSION_MINOR {minor}\n")
|
||
file.write(f"#define VERSION_BUILD {build}\n")
|
||
elif line.startswith("#define VERSION_MAJOR"):
|
||
parts = line.split()
|
||
major = int(parts[2])
|
||
file.write(line)
|
||
elif line.startswith("#define VERSION_MINOR"):
|
||
parts = line.split()
|
||
minor = int(parts[2])
|
||
file.write(line)
|
||
else:
|
||
file.write(line)
|
||
|
||
return [major, minor, build]
|
||
|
||
def sort_strings(strings):
|
||
return sorted(strings, key=lambda x: not x.endswith('.s.o'))
|
||
|
||
def find_files(directory, extensions):
|
||
file_list = []
|
||
for root, dirs, files in os.walk(directory):
|
||
for file in files:
|
||
if any(file.endswith(extension) for extension in extensions):
|
||
file_list.append(os.path.join(root, file))
|
||
return file_list
|
||
|
||
|
||
def compile(file: str):
|
||
CC = "gcc"
|
||
output_file = file.replace('/', '_')
|
||
obj_file = f"bin/{output_file}.o"
|
||
cmd = f"{CC} {WARN_FLAGS} {PROTECT_FLAGS} {ARCH_FLAGS} {CHARSET_FLAGS} {LIBS_FLAGS} -c {file} -o {obj_file}"
|
||
print(cmd)
|
||
os.system(cmd)
|
||
return obj_file
|
||
|
||
|
||
def compile_all():
|
||
file_list = find_files("kernel/", [".s", ".cpp", ".c"])
|
||
file_list += find_files("kernel/*/*", [".s", ".cpp", ".c"])
|
||
|
||
with Pool() as pool:
|
||
results = pool.map(compile, file_list)
|
||
|
||
while not all(results):
|
||
print(results)
|
||
time.sleep(1)
|
||
print(results)
|
||
cmd = f"ld -nostdlib -static -m elf_x86_64 -z max-page-size=0x1000 -T configs/linker.ld -o kernel.elf {' '.join(sort_strings(results))}"
|
||
print(cmd)
|
||
os.system(cmd)
|
||
|
||
|
||
def check_limine():
|
||
if not os.path.isdir("limine"):
|
||
subprocess.run(["git", "clone", "https://github.com/limine-bootloader/limine.git", "--branch=v5.x-branch-binary", "--depth=1"])
|
||
else:
|
||
os.chdir("limine")
|
||
subprocess.run(["git", "pull"])
|
||
os.chdir("..")
|
||
os.chdir("limine")
|
||
subprocess.run(["make"])
|
||
os.chdir("..")
|
||
|
||
|
||
|
||
def check_os():
|
||
import platform
|
||
using_distro = False
|
||
try:
|
||
import distro
|
||
using_distro = True
|
||
except ImportError:
|
||
pass
|
||
if using_distro:
|
||
linux_distro = distro.like()
|
||
else:
|
||
linux_distro = platform.linux_distribution()[0]
|
||
if linux_distro.lower() in ['debian', 'ubuntu']:
|
||
return 1
|
||
return 0
|
||
|
||
def check_tools():
|
||
required_tools = ["gcc", "g++", "xorriso", "make", "mtools", "curl"]
|
||
missing_tools = []
|
||
|
||
for tool in required_tools:
|
||
if shutil.which(tool) is None:
|
||
missing_tools.append(tool)
|
||
|
||
if len(missing_tools) > 0:
|
||
if check_os():
|
||
subprocess.run(["sudo", "apt", "install"] + missing_tools)
|
||
return
|
||
subprocess.run(["sudo", "pacman", "-S"] + missing_tools)
|
||
|
||
|
||
def create_hdd(IMAGE_NAME):
|
||
os.system(f"rm -f {IMAGE_NAME}.hdd".format())
|
||
os.system(f"dd if=/dev/zero bs=1M count=0 seek=4 of={IMAGE_NAME}.hdd")
|
||
os.system(f"sgdisk {IMAGE_NAME}.hdd -n 1:2048 -t 1:ef00")
|
||
os.system(f"./limine/limine bios-install {IMAGE_NAME}.hdd")
|
||
os.system(f"mformat -i {IMAGE_NAME}.hdd@@1M")
|
||
os.system(f"mmd -i {IMAGE_NAME}.hdd@@1M ::/mod ::/EFI ::/EFI/BOOT ::/user")
|
||
os.system(f"mcopy -i {IMAGE_NAME}.hdd@@1M kernel.elf configs/limine.cfg limine/limine-bios.sys ::/")
|
||
os.system(f"mcopy -i {IMAGE_NAME}.hdd@@1M modules/pci/pci.ko modules/simd/simd.ko modules/cpubench/cpubench.ko ::/mod")
|
||
os.system(f"mcopy -i {IMAGE_NAME}.hdd@@1M limine/BOOTX64.EFI limine/BOOTIA32.EFI ::/EFI/BOOT")
|
||
os.system(f"mcopy -i {IMAGE_NAME}.hdd@@1M boot.jpg boot.tga ::/")
|
||
os.system(f"./limine/limine bios-install {IMAGE_NAME}.hdd")
|
||
|
||
|
||
def create_iso(IMAGE_NAME):
|
||
os.system(f"rm -f {IMAGE_NAME}.iso")
|
||
os.system(f"rm -rf iso_root")
|
||
os.system(f"mkdir -p iso_root")
|
||
os.system(f"cp -v iso_root/")
|
||
os.system(f"cp -v kernel.elf boot.jpg boot.tga configs/limine.cfg limine/limine-bios.sys limine/limine-bios-cd.bin limine/limine-uefi-cd.bin iso_root/")
|
||
os.system(f"mkdir -p iso_root/EFI/BOOT")
|
||
os.system(f"mkdir -p iso_root/mod")
|
||
os.system(f"cp -v modules/pci/pci.ko iso_root/mod/")
|
||
os.system(f"cp -v modules/simd/simd.ko iso_root/mod/")
|
||
os.system(f"cp -v modules/cpubench/cpubench.ko iso_root/mod/")
|
||
os.system(f"cp -v limine/BOOTX64.EFI iso_root/EFI/BOOT/")
|
||
os.system(f"cp -v limine/BOOTIA32.EFI iso_root/EFI/BOOT/")
|
||
os.system(f"xorriso -as mkisofs -b limine-bios-cd.bin -no-emul-boot -boot-load-size 4 -boot-info-table --efi-boot limine-uefi-cd.bin -efi-boot-part --efi-boot-image --protective-msdos-label iso_root -o {IMAGE_NAME}.iso")
|
||
os.system(f"./limine/limine bios-install {IMAGE_NAME}.iso")
|
||
|
||
if __name__ == "__main__":
|
||
os.system("""find . \( -name "*.c" -o -name "*.h" -o -name "*.cpp" -o -name "*.hpp" \) -print0 | xargs -0 clang-format -i -style=file""")
|
||
subprocess.run(["rm", "-rf", "bin"])
|
||
subprocess.run(["mkdir", "-p", "bin"])
|
||
|
||
if not os.path.isdir("ovmf"):
|
||
subprocess.run(["mkdir", "-p", "ovmf"])
|
||
os.chdir("ovmf")
|
||
subprocess.run(["curl", "-Lo", "OVMF.fd", "https://retrage.github.io/edk2-nightly/bin/RELEASEX64_OVMF.fd"])
|
||
os.chdir("..")
|
||
|
||
if not os.path.isdir("limine"):
|
||
check_limine()
|
||
check_tools()
|
||
major, minor, build = version_build()
|
||
compile_all()
|
||
create_iso("bmosp")
|
||
create_hdd("bmosp")
|
||
|
||
print(f"Не забудьте сохранить изменения! Номер сборки: {major}.{minor}.{build}")
|