commit
This commit is contained in:
parent
fb58747539
commit
9b44b03ce0
18
src/main.rs
18
src/main.rs
|
@ -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");
|
||||
|
|
Loading…
Reference in New Issue