Add minimum matchmaking lobby server

rc_lobby/Cargo.toml

@@ -0,0 +1,17 @@
+[package]
+name = "oj_rc_lobby"
+version.workspace = true
+edition.workspace = true
+readme.workspace = true
+license.workspace = true
+repository.workspace = true
+authors.workspace = true
+
+[dependencies]
+log.workspace = true
+env_logger.workspace = true
+tokio = { version = "1.43", features = [ "net", "macros", "rt-multi-thread", "io-util" ] }
+clap.workspace = true
+polariton.workspace = true
+oj_polariton_auth = { version = "*", path = "../polariton_auth" }
+polariton_server.workspace = true

rc_lobby/run_debug.sh

@@ -0,0 +1,3 @@
+#!/bin/bash
+
+RUST_BACKTRACE=1 RUST_LOG=debug cargo run -- -1

rc_lobby/src/cli.rs

@@ -0,0 +1,31 @@
+use clap::Parser;
+
+#[derive(Parser, Debug)]
+#[command(version, about, long_about = None)]
+pub struct CliArgs {
+    /// TCP port on which to accept connections
+    #[arg(short, long, default_value_t = 4540)]
+    pub port: u16,
+
+    /// IP Address on which to accept connections
+    #[arg(long, default_value_t = {"127.0.0.1".to_string()})]
+    pub ip: String,
+
+    /// Domain and port of the game server to send new connections
+    #[arg(long, default_value_t = {"127.0.0.1:4541".to_string()})]
+    pub redirect: String,
+
+    /// Name of game server to send new connections
+    #[arg(long, default_value_t = {"ngram_is_ngnius".to_string()})]
+    pub room_name: String,
+
+    /// Handle one connection and then exit
+    #[arg(short = '1', long)]
+    pub once: bool,
+}
+
+impl CliArgs {
+    pub fn get() -> Self {
+        Self::parse()
+    }
+}

rc_lobby/src/main.rs

@@ -0,0 +1,263 @@
+#![forbid(unsafe_code)]
+mod cli;
+
+use oj_polariton_auth::Handshake;
+use tokio::net;
+
+use polariton::packet::{Data, Message, Packet, StandardMessage};
+use polariton::operation::{OperationResponse, Typed};
+
+#[tokio::main]
+async fn main() -> std::io::Result<()> {
+    env_logger::init();
+    let args = cli::CliArgs::get();
+    log::debug!("Got cli args {:?}", args);
+
+    let ip_addr: std::net::IpAddr = args.ip.parse().expect("Invalid IP address");
+
+    // memory leak, but only once (so not a big deal)
+    let redirect_static = Box::leak(Box::new(args.redirect.clone()));
+    let room_name_static = Box::leak(Box::new(args.room_name.clone()));
+
+    let listener = net::TcpListener::bind(std::net::SocketAddr::new(ip_addr, args.port)).await?;
+
+    if args.once {
+        log::warn!("Handling first connection and then exiting");
+        let (socket, address) = listener.accept().await?;
+        process_socket(socket, address, redirect_static, room_name_static).await;
+        Ok(())
+    } else {
+        loop {
+            let (socket, address) = listener.accept().await?;
+            tokio::spawn(process_socket(socket, address, redirect_static, room_name_static));
+        }
+    }
+}
+
+async fn process_socket(mut socket: net::TcpStream, address: std::net::SocketAddr, redirect_url: &str, lobby_name: &str) {
+    log::debug!("Accepting connection from address {}", address);
+
+    let enc = match do_connect_handshake(&mut socket, lobby_name, redirect_url).await {
+        Some(x) => x,
+        None => {
+            log::error!("Failed to do connect handshake with {}", address);
+            return;
+        }
+    };
+    let ctx = polariton::packet::SerdesContext::from_boxed(Default::default(), enc);
+    while let Ok(packet) = polariton_server::utils::receive_packet_async(&mut socket, &ctx).await {
+        match packet {
+            Packet::Ping(ping) => {
+                polariton_server::utils::handle_ping_async(ping, &mut socket, &ctx).await.unwrap_or_default();
+            },
+            Packet::Packet(packet) => log::warn!("Not handling packet {:?}", packet),
+        }
+    }
+    log::debug!("Goodbye connection from address {}", address);
+}
+
+const APP_ID: &str = "LobbyServer";
+
+struct AuthImpl;
+
+const TOKEN_KEY: u8 = 216; // token;refresh_token
+//const UNKNOWN_BYTE_KEY: u8 = 217;
+const SERVICE_KEY: u8 = 224;
+const USERNAME_KEY: u8 = 225;
+
+//const CCU_KEY: u8 = 245;
+
+#[derive(Debug)]
+enum AuthError {
+    WrongService { expected: String, actual: String },
+    MissingService,
+    MissingToken,
+    MissingUsername,
+}
+
+impl AuthError {
+    fn log_err(&self) {
+        match self {
+            Self::WrongService { expected, actual } => log::error!("(auth fail) Got unexpected service {}, expected {}", actual, expected),
+            Self::MissingService => log::error!("(auth fail) No service name param ({}) received", SERVICE_KEY),
+            Self::MissingToken => log::error!("(auth fail) No token param ({}) received", TOKEN_KEY),
+            Self::MissingUsername => log::error!("(auth fail) No username param ({}) received", USERNAME_KEY),
+        }
+    }
+}
+
+impl oj_polariton_auth::AuthProvider<AuthError> for AuthImpl {
+    fn validate(&mut self, params: &std::collections::HashMap<u8, Typed>) -> Result<std::collections::HashMap<u8, Typed>, AuthError> {
+        if let Some(Typed::Str(token)) = params.get(&TOKEN_KEY) {
+            if let Some(Typed::Str(service)) = params.get(&SERVICE_KEY) {
+                if let Some(Typed::Str(user)) = params.get(&USERNAME_KEY) {
+                    if service.string == APP_ID {
+                        let params_resp = std::collections::HashMap::<u8, Typed>::new();
+                        //params_resp.insert(CCU_KEY, Typed::Byte(0));
+                        log::debug!("Auth success for {} (token: {})", user.string, token.string);
+                        Ok(params_resp)
+                    } else { Err(AuthError::WrongService { expected: APP_ID.to_owned(), actual: service.string.to_owned() }) }
+                } else { Err(AuthError::MissingUsername) }
+            } else { Err(AuthError::MissingService) }
+        } else { Err(AuthError::MissingToken) }
+    }
+}
+
+async fn do_connect_handshake(
+    socket: &mut net::TcpStream,
+    game_server_name: &str,
+    game_server_url: &str,
+) -> Option<Box<dyn polariton::packet::Cryptographer>> {
+    let handshake = Handshake::new(APP_ID);
+    // connect
+    log::debug!("(connect) Handling first packet");
+    let packet1 = match polariton_server::utils::receive_packet_async(socket, &Default::default()).await {
+        Ok(x) => x,
+        Err(e) => {
+            log::error!("Failed to read connect packet: {}", e);
+            return None;
+        }
+    };
+    let (handshake, to_send) = match handshake.connect(&packet1) {
+        Ok(x) => (x.handshake, x.extra),
+        Err(e) => {
+            log::error!("Failed to handle connect handshake: {:?}", e.extra);
+            return None;
+        }
+    };
+    match polariton_server::utils::send_packet_async(&to_send, socket, &Default::default()).await {
+        Ok(_) => {},
+        Err(e) => {
+            log::error!("Failed to send connect ack packet: {}", e);
+            return None;
+        }
+    }
+    // encrypt
+    log::debug!("(connect) Handling second packet");
+    let mut packet2 = match polariton_server::utils::receive_packet_async(socket, &Default::default()).await {
+        Ok(x) => x,
+        Err(e) => {
+            log::error!("Failed to read (maybe) public key packet: {}", e);
+            return None;
+        }
+    };
+    while let Packet::Ping(ping) = packet2 {
+        polariton_server::utils::handle_ping_async(ping, socket, &Default::default()).await.unwrap_or_default();
+        packet2 = match polariton_server::utils::receive_packet_async(socket, &Default::default()).await {
+            Ok(x) => x,
+            Err(e) => {
+                log::error!("Failed to read (maybe) public key packet: {}", e);
+                return None;
+            }
+        };
+    }
+    let (handshake, to_send, crypto) = match handshake.encrypt(&packet2) {
+        Ok(x) => (x.handshake, x.extra.0, x.extra.1),
+        Err(e) => {
+            log::error!("Failed to handle encryption handshake: {:?}", e.extra);
+            return None;
+        }
+    };
+    match polariton_server::utils::send_packet_async(&to_send, socket, &Default::default()).await {
+        Ok(_) => {},
+        Err(e) => {
+            log::error!("Failed to send encryption ack packet: {}", e);
+            return None;
+        }
+    }
+    // pre-auth
+    let handshake = handshake.with_auth(AuthImpl);
+    let op_ctx = polariton::serdes::SerdesContext::default();
+    let ctx = polariton::packet::SerdesContext::new(op_ctx, crypto);
+    // authenticate
+    log::debug!("(connect) Handling third packet");
+    let mut packet3 = match polariton_server::utils::receive_packet_async(socket, &ctx).await {
+        Ok(x) => x,
+        Err(e) => {
+            log::error!("Failed to read (maybe) auth packet: {}", e);
+            return None;
+        }
+    };
+    while let Packet::Ping(ping) = packet3 {
+        polariton_server::utils::handle_ping_async(ping, socket, &Default::default()).await.unwrap_or_default();
+        packet3 = match polariton_server::utils::receive_packet_async(socket, &ctx).await {
+            Ok(x) => x,
+            Err(e) => {
+                log::error!("Failed to read (maybe) auth packet: {}", e);
+                return None;
+            }
+        };
+    }
+    let to_send = match handshake.authenticate(&packet3, &ctx) {
+        Ok(x) => x,
+        Err(h) => match h.extra {
+            oj_polariton_auth::AuthError::Validation(e) => {
+                e.log_err();
+                return None;
+            },
+            e => {
+                log::error!("Failed to handle auth handshake: {:?}", e);
+                return None;
+            },
+        },
+    };
+    match polariton_server::utils::send_packet_async(&to_send, socket, &ctx).await {
+        Ok(_) => {},
+        Err(e) => {
+            log::error!("Failed to send auth ack packet: {}", e);
+            return None;
+        }
+    }
+
+    // redirect to lobby
+    log::debug!("(connect) Handling fourth packet");
+    let mut packet_j = match polariton_server::utils::receive_packet_async(socket, &ctx).await {
+        Ok(x) => x,
+        Err(e) => {
+            log::error!("Failed to read (maybe) join packet: {}", e);
+            return None;
+        }
+    };
+    while let Packet::Ping(ping) = packet_j {
+        polariton_server::utils::handle_ping_async(ping, socket, &ctx).await.unwrap_or_default();
+        packet_j = match polariton_server::utils::receive_packet_async(socket, &ctx).await {
+            Ok(x) => x,
+            Err(e) => {
+                log::error!("Failed to read (maybe) join packet: {}", e);
+                return None;
+            }
+        };
+    }
+    log::debug!("(connect) Got fourth packet {:?}", packet_j);
+    if let Packet::Packet(msg) = &packet_j {
+        if let Message::Standard(st) = &msg.message {
+            if let Data::OpReq(req) = &st.data {
+                if req.code == 226 { // join lobby
+                    log::debug!("Max players from lobby join request: {:?}", req.params.to_owned().to_dict().get(&255));
+                    let mut params = std::collections::HashMap::<u8, Typed>::new();
+                    params.insert(230 /* game server address */, Typed::Str(game_server_url.into()));
+                    params.insert(255 /* room name */, Typed::Str(game_server_name.into()));
+                    let resp = Packet::from_message(
+                        Message::Standard(
+                            StandardMessage { flags: 0,
+                                data: Data::OpResp(OperationResponse {
+                                    code: req.code,
+                                    return_code: 0,
+                                    message: Typed::Null,
+                                    params: params.into(),
+                                }),
+                            }.encrypt(true)), 0, true, &ctx).unwrap();
+                    match polariton_server::utils::send_packet_async(&resp, socket, &ctx).await {
+                        Ok(_) => {},
+                        Err(e) => {
+                            log::error!("Failed to send lobby ack packet: {}", e);
+                            return None;
+                        }
+                    }
+                }
+            }
+        }
+    }
+
+    Some(ctx.into_crypto())
+}