This commit is contained in:
firebadnofire 2024-04-16 18:29:08 -04:00
parent fb58747539
commit 9b44b03ce0
Signed by: firebadnofire
SSH Key Fingerprint: SHA256:bnN1TGRauJN84CxL1IZ/2uHNvJualwYkFjOKaaOilJE
1 changed files with 7 additions and 11 deletions

View File

@ -25,7 +25,7 @@ fn main() {
});
// Ensure the log directory exists
std::fs::create_dir_all("./log").expect("Failed to create log directory");
std::fs::create_dir_all("/var/log/RMS-server").expect("Failed to create log directory");
log_verbose(format!("Server listening on port {}", port));
@ -48,22 +48,19 @@ fn main() {
// Function to handle a single client
fn handle_client(mut stream: std::net::TcpStream) {
let mut buffer = [0; 1024];
let peer_addr = stream.peer_addr().unwrap().to_string(); // Log client IP
loop {
match stream.read(&mut buffer) {
Ok(0) => {
log_verbose(format!("Client disconnected: {}", peer_addr));
break;
break; // Client disconnected
},
Ok(size) => {
let msg = String::from_utf8_lossy(&buffer[..size]);
println!("{}", msg); // Print what client sends
log_message(format!("Received from {}: {}", peer_addr, msg)); // Log to messages file
log_message(msg.into_owned()); // Log to messages file
},
Err(e) => {
log_verbose(format!("Failed to receive data from {}: {}", peer_addr, e));
break;
Err(_) => {
break; // Connection error
}
}
}
@ -75,7 +72,7 @@ fn log_verbose(message: String) {
.create(true)
.write(true)
.append(true)
.open(Path::new("./log/RMS-server.log"))
.open(Path::new("/var/log/RMS-server/verbose.log"))
.expect("Failed to open verbose log file");
writeln!(file, "{}", message).expect("Failed to write to verbose log file");
@ -83,12 +80,11 @@ fn log_verbose(message: String) {
// Log raw messages to a separate file
fn log_message(message: String) {
println!("{}", message); // Print to stdout
let mut file = OpenOptions::new()
.create(true)
.write(true)
.append(true)
.open(Path::new("./log/rawmessages.log"))
.open(Path::new("/var/log/RMS-server/rawmessages.log"))
.expect("Failed to open message log file");
writeln!(file, "{}", message).expect("Failed to write to message log file");