From 6dd0690fdd54682e40af270b159af8d8ab6723c7 Mon Sep 17 00:00:00 2001
From: Arshdeep54 <balarsh535@gmail.com>
Date: Fri, 20 Dec 2024 20:28:21 +0530
Subject: [PATCH] fix log implementation

---
 src/auth.rs | 15 +++------------
 src/ssh.rs  | 20 +++++---------------
 src/su.rs   | 14 +++-----------
 src/sudo.rs | 14 +++-----------
 4 files changed, 14 insertions(+), 49 deletions(-)

diff --git a/src/auth.rs b/src/auth.rs
index a37e335..249f555 100644
--- a/src/auth.rs
+++ b/src/auth.rs
@@ -30,21 +30,12 @@ pub fn handle_auth(ssh_host_username: &str, ssh_key: &str) -> Result<()> {
 
         Ok(false) => {
             let name = get_name(&config, ssh_key)?;
-
+            if let Err(e) = logger::log(AUTH_LOG_PATH, "SUCCESS", &format!("User: {}", name)) {
+                println!("Failed to log: {}", e);
+            }
             match fork() {
                 Ok(ForkResult::Parent { .. }) => {}
                 Ok(ForkResult::Child) => {
-                    match fork() {
-                        Ok(ForkResult::Parent { .. }) => {}
-                        Ok(ForkResult::Child) => {
-                            if let Err(e) = logger::log(AUTH_LOG_PATH, "SUCCESS", &format!("User: {}", name)) {
-                                println!("Failed to log: {}", e);
-                            }
-                            std::process::exit(0); 
-                        }
-                        Err(_) => println!("Fork failed"),
-                    }
-
                     notifier::post_ssh_summary(
                         &config,
                         false,
diff --git a/src/ssh.rs b/src/ssh.rs
index ecf9c2e..ff92d00 100644
--- a/src/ssh.rs
+++ b/src/ssh.rs
@@ -9,7 +9,7 @@ use lib::errors::*;
 use lib::init::init;
 use lib::keyhouse::get_name;
 use lib::notifier;
-use lib::utils::clear_file;
+//use lib::utils::clear_file;
 use lib::logger;
 use lib::utils::SSH_LOG_PATH;
 
@@ -23,22 +23,12 @@ pub fn handle_ssh() -> Result<()> {
 
         let env = read_temp_env("/opt/watchdog/ssh_env")?;
         let name = get_name(&config, &env.ssh_key)?;
-
+        if let Err(e) = logger::log(SSH_LOG_PATH, "SUCCESS", &format!("User: {}", name)) {
+            println!("Failed to log: {}", e);
+        }
         match fork() {
-            Ok(ForkResult::Parent { .. }) => {
-                clear_file("/opt/watchdog/ssh_env")?;
-            }
+            Ok(ForkResult::Parent { .. }) => {}
             Ok(ForkResult::Child) => {
-                match fork() {
-                    Ok(ForkResult::Parent { .. }) => {}
-                    Ok(ForkResult::Child) => {
-                        if let Err(e) = logger::log(SSH_LOG_PATH, "SUCCESS", &format!("User: {}", name)) {
-                            println!("Failed to log: {}", e);
-                        }
-                    }
-                    Err(_) => println!("Fork failed"),
-                }
-
                 notifier::post_ssh_summary(&config, true, name, env.ssh_host_username)?;
             }
             Err(_) => println!("Fork failed"),
diff --git a/src/su.rs b/src/su.rs
index 65d7f09..9d1cb12 100644
--- a/src/su.rs
+++ b/src/su.rs
@@ -23,20 +23,12 @@ pub fn handle_su() -> Result<()> {
     if pam_type == "open_session" {
         let config = read_config()?;
         init(&config)?;
-
+        if let Err(e) = logger::log(SU_LOG_PATH, "SUCCESS", &format!("User: {}", pam_user)) {
+            println!("Failed to log: {}", e);
+        }
         match fork() {
             Ok(ForkResult::Parent { .. }) => {}
             Ok(ForkResult::Child) => {
-                match fork() {
-                    Ok(ForkResult::Parent { .. }) => {}
-                    Ok(ForkResult::Child) => {
-                        if let Err(e) = logger::log(SU_LOG_PATH, "SUCCESS", &format!("User: {}", pam_user)) {
-                            println!("Failed to log: {}", e);
-                        }
-                    }
-                    Err(_) => println!("Fork failed"),
-                }
-
                 notifier::post_su_summary(&config, pam_ruser, pam_user)?;
             }
             Err(_) => println!("Fork failed"),
diff --git a/src/sudo.rs b/src/sudo.rs
index 51445a5..4a36694 100644
--- a/src/sudo.rs
+++ b/src/sudo.rs
@@ -20,20 +20,12 @@ pub fn handle_sudo() -> Result<()> {
     if pam_type == "open_session" {
         let config = read_config()?;
         init(&config)?;
-
+        if let Err(e) = logger::log(SUDO_LOG_PATH, "SUCCESS", &format!("User: {}", pam_ruser)) {
+            println!("Failed to log: {}", e);
+        }
         match fork() {
             Ok(ForkResult::Parent { .. }) => {}
             Ok(ForkResult::Child) => {
-                match fork() {
-                    Ok(ForkResult::Parent { .. }) => {}
-                    Ok(ForkResult::Child) => {
-                        if let Err(e) = logger::log(SUDO_LOG_PATH, "SUCCESS", &format!("User: {}", pam_ruser)) {
-                            println!("Failed to log: {}", e);
-                        }
-                    }
-                    Err(_) => println!("Fork failed"),
-                }
-
                 notifier::post_sudo_summary(&config, pam_ruser)?;
             }
             Err(_) => println!("Fork failed"),