RUSTRIAL OS

pub fn show_menu() {
    loop {
        println!("1. Normal Mode");
        println!("2. RustrialScript");
        println!("3. Graphics Demo");
        println!("4. Hardware Info");
        println!("5. Help");
        println!("6. Desktop");
        
        match read_key() {
            b'1' => normal_mode(),
            b'2' => script_menu(),
            b'6' => desktop::launch(),
            _ => continue,
        }
    }
}

pub async fn tcp_connect(
    dst_ip: Ipv4Addr,
    dst_port: u16,
) -> Result<TcpSocket> {
    let socket = TcpSocket::new();
    socket.send_syn(dst_ip, dst_port);
    socket.wait_for_syn_ack().await?;
    socket.send_ack();
    socket.state = Connected;
    
    Ok(socket)
}

match cmd.as_str() {
    "ls" => list_directory(args),
    "cat" => read_file(args),
    "mkdir" => create_dir(args),
    "ping" => ping_host(args).await,
    "ifconfig" => show_interfaces(),
    "run" => execute_script(args),
    "clear" => vga_clear(),
    _ => println!("Unknown: {}", cmd),
}