akern-gkgoat-fork/ableos/src/kmain.rs

#![allow(clippy::empty_loop)]

use vga::{colors::Color16, writers::GraphicsWriter};

use crate::{
    relib::network::socket::SocketReturns,
    vga_e::{self, VGAE},
};

use {
    crate::{
        arch::{drivers::graphics::GraphicsBuffer, init, sloop},
        driver_traits::graphics::Graphics,
        experiments::{
            info::master,
            systeminfo::{KERNEL_VERSION, RELEASE_TYPE},
        },
        // log::LOG_STATE,
        relib::math::rand::RAND_HANDLE,
        relib::network::socket::{Socket, SOCKETS},
        scheduler::{test_fn, Thread, ThreadList},
    },
    alloc::{
        string::{String, ToString},
        vec,
    },
    lazy_static::lazy_static,
};

#[no_mangle]
#[allow(unconditional_recursion)]
pub extern "C" fn stack_overflow() -> u8 {
    stack_overflow();
    // meme number
    69 // NOTE: Any specific reason for this number aside from memes?
}

lazy_static! {
    pub static ref KEY_BUFFER: spin::Mutex<String> = spin::Mutex::new("".to_string());
    pub static ref THREAD_LIST: spin::Mutex<ThreadList> = spin::Mutex::new(vec![]);
    pub static ref TICK: spin::Mutex<u64> = spin::Mutex::new(0);
}

#[no_mangle]
pub fn kernel_main() -> ! {
    init::init();

    {
        let mut a_thread = Thread::new();

        a_thread.new_task(test_fn);
        a_thread.new_task(test_fn);
        THREAD_LIST.lock().push(a_thread);
        GraphicsBuffer::draw();
        GraphicsBuffer::hide_cursor();
    }

    crate::wasm::evaluate();
    {
        info!("{} v{}", RELEASE_TYPE, KERNEL_VERSION);
        info!(
            "Brand String: {:?}",
            master().unwrap().brand_string().unwrap()
        );
        if false {
            println!("$PINK$Hi$RED$ from$GREEN$ able!$RESET$");
            println!("$RED$hi$RESET$");
        }
    }

    {
        use crate::relib::network::socket::SimpleSock;

        let mut xyz = SimpleSock::new();

        xyz.peek();

        xyz.write(vec![0, 1, 2, 3]);

        let x = "simple 🧦".to_string().into_bytes();
        xyz.write(x);

        info!("{:?}", &xyz.read(4).unwrap());

        println!("{:?}", &xyz.peek().unwrap());

        match &xyz.peek() {
            SocketReturns::ReadOk(strr) => {
                let out = String::from_utf8_lossy(strr);

                info!("{}", out);
            }
            SocketReturns::ReadIndexOutOfBounds => todo!(),
            SocketReturns::WriteOk => todo!(),
        }
    }

    println!("$GREEN$able$RESET$@$LIGHTBLUE$stAble$ $RED$->$RESET$");

    VGAE.lock().clear_screen(Color16::Black);

    for x in "μkernel".chars().enumerate() {
        vga_e::draw_char(x.1, x.0);
    }
    // vga_e::test_it_fucko();
    // stack_overflow();

    sloop()
}

/// called by arch specific timers to tick up all kernel related functions
pub fn tick() {
    let mut data = TICK.lock();
    *data += 1;
    RAND_HANDLE.lock().seed_entropy_timer(*data);

    // println!("{}", *data);
    crate::kernel_state::KERNEL_STATE.lock().update_state();
}

pub fn key_entropy(key: u8) {
    RAND_HANDLE.lock().seed_entropy_keyboard(key);
}