diff options
Diffstat (limited to 'src/bin/server/handler.rs')
| -rw-r--r-- | src/bin/server/handler.rs | 64 |
1 files changed, 0 insertions, 64 deletions
diff --git a/src/bin/server/handler.rs b/src/bin/server/handler.rs deleted file mode 100644 index 2132d09..0000000 --- a/src/bin/server/handler.rs +++ /dev/null | |||
| @@ -1,64 +0,0 @@ | |||
| 1 | use std::io::{BufRead, BufReader, Write}; | ||
| 2 | use std::net::TcpStream; | ||
| 3 | use std::sync::Mutex; | ||
| 4 | use once_cell::sync::Lazy; | ||
| 5 | use rusqlite::Connection; | ||
| 6 | use crate::database; | ||
| 7 | |||
| 8 | static DB_CONN: Lazy<Mutex<Connection>> = Lazy::new(|| Mutex::new(database::init_db())); | ||
| 9 | |||
| 10 | pub fn handle_client(stream: TcpStream) { | ||
| 11 | let client_addr = stream.peer_addr().unwrap().to_string(); | ||
| 12 | let mut user = "guest".to_string(); | ||
| 13 | println!("[{} @ {}] Client connected", user, client_addr); | ||
| 14 | let reader_stream = stream.try_clone().unwrap(); | ||
| 15 | let reader = BufReader::new(reader_stream); | ||
| 16 | let mut writer = stream; | ||
| 17 | for line in reader.lines() { | ||
| 18 | let line = line.unwrap(); | ||
| 19 | println!("[{} @ {}] Received: {}", user, client_addr, line); | ||
| 20 | let response = process_message(&line, &mut user); | ||
| 21 | writeln!(writer, "{}", response).unwrap(); | ||
| 22 | } | ||
| 23 | println!("[{} @ {}] Client disconnected", user, client_addr); | ||
| 24 | } | ||
| 25 | |||
| 26 | fn process_message(msg: &str, user: &mut String) -> String { | ||
| 27 | let parts: Vec<&str> = msg.splitn(3, ' ').collect(); | ||
| 28 | match parts[0] { | ||
| 29 | "REGISTER" => { | ||
| 30 | if parts.len() != 3 { | ||
| 31 | "FAIL Invalid REGISTER format".to_string() | ||
| 32 | } else { | ||
| 33 | let username = parts[1]; | ||
| 34 | let hashed_pass = parts[2]; | ||
| 35 | let db = DB_CONN.lock().unwrap(); | ||
| 36 | match database::add_user(&db, username, hashed_pass) { | ||
| 37 | Ok(_) => format!("OK Registered {}", username), | ||
| 38 | Err(_) => "FAIL User already exists".to_string(), | ||
| 39 | } | ||
| 40 | } | ||
| 41 | } | ||
| 42 | "AUTH" => { | ||
| 43 | if parts.len() != 3 { | ||
| 44 | "FAIL Invalid AUTH format".to_string() | ||
| 45 | } else { | ||
| 46 | let username = parts[1]; | ||
| 47 | let provided_hash = parts[2]; | ||
| 48 | let db = DB_CONN.lock().unwrap(); | ||
| 49 | if let Some(stored_hash) = database::get_hashed_password(&db, username) { | ||
| 50 | if provided_hash == stored_hash { | ||
| 51 | *user = username.to_string(); | ||
| 52 | format!("OK {}", username) | ||
| 53 | } else { | ||
| 54 | "FAIL Invalid credentials".to_string() | ||
| 55 | } | ||
| 56 | } else { | ||
| 57 | "FAIL User not found".to_string() | ||
| 58 | } | ||
| 59 | } | ||
| 60 | } | ||
| 61 | "PING" => "PING".to_string(), | ||
| 62 | _ => msg.to_string(), // Echo back for compatibility | ||
| 63 | } | ||
| 64 | } | ||
