KarxOS/src/main.rs

#![allow(non_snake_case)]
#![no_std]
#![no_main]
#![feature(custom_test_frameworks)]
#![test_runner(crate::test_runner)]
#![reexport_test_harness_main = "test_main"]
#![feature(abi_x86_interrupt)]
#![feature(alloc_error_handler)]

mod allocator;
mod clock;
mod gdt;
mod interrupts;
mod memory;
mod shell;
mod vga_buffer;

use bootloader::entry_point;
use bootloader::BootInfo;
use core::panic::PanicInfo;

extern crate alloc;

#[panic_handler]
fn panic(info: &PanicInfo) -> ! {
    println!("{}", info);
    loop {}
}

#[alloc_error_handler]
fn alloc_error_handler(layout: alloc::alloc::Layout) -> ! {
    panic!("allocation error: {:?}", layout)
}

fn init() {
    gdt::init_gdt();
    interrupts::init();
    unsafe {
        interrupts::PICS.lock().initialize();
    }
    x86_64::instructions::interrupts::enable();
}

macro_rules! status {
    ($n:expr) => {
        print!("[ ");
        change_color(Color::Green, Color::Black);
        print!("OK");
        change_color(Color::White, Color::Black);
        println!(" ] {}", $n);
    };
}

entry_point!(main);

fn main(boot_info: &'static BootInfo) -> ! {
    use crate::vga_buffer::{change_color, Color};
    use memory::BootInfoFrameAllocator;
    use x86_64::VirtAddr; // For status! macro

    init();
    status!("Initialized GDT and Interrupts");

    clock::init();
    status!("Initialized system clock");

    let phys_mem_offset = VirtAddr::new(boot_info.physical_memory_offset);
    let mut mapper = unsafe { memory::init(phys_mem_offset) };
    let mut frame_allocator = unsafe { BootInfoFrameAllocator::init(&boot_info.memory_map) };
    status!("Initialized Mapper and Frame allocator");

    allocator::init_heap(&mut mapper, &mut frame_allocator).expect("Heap initialization failed");
    status!("Initialized heap");

    println!();
    print!("Welcome to ");
    change_color(Color::Blue, Color::Black);
    println!("KarxOS!");
    change_color(Color::White, Color::Black);

    // First prompt, future prompts will be handled by shell::evaluate
    print!(">>> ");

    #[cfg(test)]
    test_main();

    loop {
        // Halt CPU so that usage isn't 100% all the time
        x86_64::instructions::hlt();
    }
}

#[cfg(test)]
fn test_runner(tests: &[&dyn Fn()]) {
    println!("Running {} tests", tests.len());
    for test in tests {
        test();
    }
}