blob: 86b09cb3d9447e30079af84abd96f837e12be2f7 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
use crate::tls::TlsStream;
use zagan_bet::server::core::{ConnectionState, ServerCore};
pub fn handle_client_tls(mut stream: TlsStream) {
let client_addr = match stream.peer_addr() {
Ok(addr) => addr.to_string(),
Err(error) => {
eprintln!("Could not get peer address: {error}");
return;
}
};
let db_path = std::env::var("SERVER_DB").unwrap_or_else(|_| "server.db".to_string());
let mut core = match ServerCore::open(&db_path) {
Ok(core) => core,
Err(error) => {
eprintln!("Failed to open database {db_path}: {error}");
return;
}
};
let mut state = ConnectionState::default();
println!("[TLS Client connected] {client_addr}");
loop {
let mut line = String::new();
match stream.read_line(&mut line) {
Ok(0) => break,
Ok(_) => {}
Err(error) => {
eprintln!("Read error: {error}");
break;
}
}
let line = line.trim();
if line.is_empty() {
continue;
}
let user = state.user().unwrap_or("guest").to_string();
println!("[{user} @ {client_addr}] Received: {line}");
let response = core.handle_message(&mut state, line);
if let Err(error) = stream.write_line(&response) {
eprintln!("Write error: {error}");
break;
}
let user = state.user().unwrap_or("guest");
println!("[{user} @ {client_addr}] Sent: {response}");
}
core.disconnect(&state);
println!("[TLS Client disconnected] {client_addr}");
}
|
