ableos_time/ableos/src/kmain.rs

#![allow(clippy::empty_loop)]

use {
    crate::{
        arch::{init, sloop},
        boot_conf,
        boot_conf::BootConfig,
        capabilities::FileAccess,
        experiments::{
            info::master,
            systeminfo::{KERNEL_VERSION, RELEASE_TYPE},
        },
        file::PathRep,
        relib::network::socket::{SimpleSock, Socket},
        scheduler::SCHEDULER,
        VgaBuffer, SCREEN_BUFFER,
    },
    alloc::{
        format,
        string::{String, ToString},
        vec,
    },
    core::sync::atomic::{AtomicU64, Ordering::*},
    facepalm::start_facepalm,
    lazy_static::lazy_static,
    log::*,
    shadeable::pixel_format::from_vga_16,
    vga::colors::Color16,
};

lazy_static! {
    pub static ref TICK: AtomicU64 = AtomicU64::new(0);
    pub static ref BOOT_CONF: BootConfig = boot_conf::BootConfig::new();
}

/// The main entry point of the kernel
#[no_mangle]
pub fn kernel_main() -> ! {
    init::init();
    log::set_max_level(BOOT_CONF.log_level());

    {
        {
            let mut scheduler = SCHEDULER.lock();

            use crate::scheduler::Priority::*;
            let mut process_1 = scheduler.new_process(High);
            process_1.capabilities.files = FileAccess::Some(vec![PathRep {
                location: FileLocations::Home,
                file_name: "test".to_string(),
            }]);
            scheduler.add_process(process_1);
            for ref_process in &scheduler.list {
                trace!("{:?}", ref_process);
            }
            drop(scheduler);
        }

        use crate::proto_filetable::file::FileLocations;

        {
            let mut sock_print_id = SimpleSock::new();
            sock_print_id.register_protocol("Screen Printer".to_string());
            sock_print_id.write(format!("🐑").into());

            let mut mode = SCREEN_BUFFER.lock();

            mode.force_redraw();
            for current in (*String::from_utf8_lossy(&sock_print_id.peek().unwrap())).chars() {
                mode.draw_char(0, 0, current, from_vga_16(Color16::Red));
            }
            mode.copy_to_buffer();
        }
    }

    start_facepalm();

    sloop()
}

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

    crate::kernel_state::KERNEL_STATE.lock().update_state();
    // let mut scheduler = SCHEDULER.lock();
    // scheduler.bump_exec();

    TICK.store(data, Relaxed)
}

pub fn cpu_socket_startup() {
    let mut cpu_info_socket = SimpleSock::new();
    cpu_info_socket.register_protocol("CPU_INFO".to_string());

    let x = master().unwrap();
    let _xyz = x.brand_string().unwrap();
}

pub fn log_version_data() {
    info!("{} v{}", RELEASE_TYPE, KERNEL_VERSION);
    info!(
        "Brand String: {}",
        master().unwrap().brand_string().unwrap()
    );
}

pub fn generate_process_pass() -> u128 {
    use rdrand::RdRand;
    let gen = RdRand::new().unwrap();
    let ret = (gen.try_next_u64().unwrap() as u128) << 64 | (gen.try_next_u64().unwrap() as u128);
    ret
}