refactor: 🔨 Return Result directly and remove all exit code

Because Result implements Terminaison trait, the cli will return the right exit code

src/actions.rs

@@ -33,11 +33,11 @@ pub enum Commands {
 }
 
 pub trait Runnable {
-    fn run(&self) -> Result<i32, Box<dyn Error>>;
+    fn run(&self) -> Result<(), Box<dyn Error>>;
 }
 
 impl Runnable for Actions {
-    fn run(&self) -> Result<i32, Box<dyn Error>> {
+    fn run(&self) -> Result<(), Box<dyn Error>> {
         match &self.command {
             Commands::AlternateDataStreams(alternate_data_streams) => {
                 alternate_data_streams as &dyn Runnable

src/actions/ads.rs

@@ -20,7 +20,7 @@ pub enum Commands {
 }
 
 impl Runnable for AlternateDataStreams {
-    fn run(&self) -> Result<i32, Box<dyn Error>> {
+    fn run(&self) -> Result<(), Box<dyn Error>> {
         match &self.command {
             Commands::Create(create) => create as &dyn Runnable,
         }

src/actions/ads/create.rs

@@ -64,7 +64,7 @@ fn create_ads(fullpath: String, adsname: String, hex_data: Vec<u8>) -> bool {
 
 impl Runnable for Create {
     /* Version 20230908 */
-    fn run(&self) -> Result<i32, Box<dyn Error>> {
+    fn run(&self) -> Result<(), Box<dyn Error>> {
         println!("Alternate Data Stream");
 
         if !self.filename.is_empty() {
@@ -77,9 +77,9 @@ impl Runnable for Create {
             let payload: Vec<u8> = general_purpose::STANDARD.decode(self.data.as_str())?;
             let ret_ads: bool = create_ads(fullname, barrow_ads, payload);
 
-            return Ok(!ret_ads as i32);
+            return Ok(());
         }
 
-        Ok(1)
+        Ok(())
     }
 }

src/actions/drivers.rs

@@ -20,7 +20,7 @@ pub enum Commands {
 }
 
 impl Runnable for Drivers {
-    fn run(&self) -> Result<i32, Box<dyn Error>> {
+    fn run(&self) -> Result<(), Box<dyn Error>> {
         match &self.command {
             Commands::Create(create) => create as &dyn Runnable,
         }

src/actions/drivers/create.rs

@@ -114,18 +114,18 @@ fn create_driver_service(name: &str, details: &str, path: &str) -> bool {
 
 impl Runnable for Create {
     /* Version 20230908 */
-    fn run(&self) -> Result<i32, Box<dyn Error>> {
+    fn run(&self) -> Result<(), Box<dyn Error>> {
         println!("Bring Your Own Vulnerable Driver");
 
         if !is_administrator()? {
             println!("Need to have Administrator right to create the service");
-            return Ok(1);
+            return Ok(());
         }
 
         // Todo check path is valid or not :)
 
         let result: bool = create_driver_service(&self.internal, &self.display, &self.path);
 
-        Ok(!result as i32)
+        Ok(())
     }
 }

src/actions/files.rs

@@ -20,7 +20,7 @@ pub enum Commands {
 }
 
 impl Runnable for Files {
-    fn run(&self) -> Result<i32, Box<dyn Error>> {
+    fn run(&self) -> Result<(), Box<dyn Error>> {
         match &self.command {
             Commands::Create(create) => create as &dyn Runnable,
         }

src/actions/files/create.rs

@@ -90,10 +90,10 @@ fn create_file(fullpath: String, hex_data: Vec<u8>) -> bool {
 }
 
 impl Runnable for Create {
-    fn run(&self) -> Result<i32, Box<dyn Error>> {
+    fn run(&self) -> Result<(), Box<dyn Error>> {
         if self.admin && !is_administrator()? {
             println!("Need to have Administrator right to create the file");
-            return Ok(1);
+            return Ok(());
         }
 
         let mut generator: Generator<rand::rngs::ThreadRng> =
@@ -107,6 +107,6 @@ impl Runnable for Create {
         let payload: Vec<u8> = general_purpose::STANDARD.decode(self.magicbyte.as_str())?;
         let ret: bool = create_file(fullname, payload);
 
-        Ok(!ret as i32)
+        Ok(())
     }
 }

src/actions/mutexes.rs

@@ -20,7 +20,7 @@ pub enum Commands {
 }
 
 impl Runnable for Mutexes {
-    fn run(&self) -> Result<i32, Box<dyn Error>> {
+    fn run(&self) -> Result<(), Box<dyn Error>> {
         match &self.command {
             Commands::Create(create) => create as &dyn Runnable,
         }

src/actions/mutexes/create.rs

@@ -39,7 +39,7 @@ fn create_mutex(name: &String, wait: u64) {
 }
 
 impl Runnable for Create {
-    fn run(&self) -> Result<i32, Box<dyn Error>> {
+    fn run(&self) -> Result<(), Box<dyn Error>> {
         println!("Create Mutex");
 
         let mut generator: Generator<rand::rngs::ThreadRng> =
@@ -52,6 +52,6 @@ impl Runnable for Create {
 
         create_mutex(&payload, 2000);
 
-        Ok(0)
+        Ok(())
     }
 }

src/actions/pipes.rs

@@ -20,7 +20,7 @@ pub enum Commands {
 }
 
 impl Runnable for Pipes {
-    fn run(&self) -> Result<i32, Box<dyn Error>> {
+    fn run(&self) -> Result<(), Box<dyn Error>> {
         match &self.command {
             Commands::Create(create) => create as &dyn Runnable,
         }

src/actions/pipes/create.rs

@@ -51,7 +51,7 @@ fn create_name_pipe(name: &String, wait: u64) {
 }
 
 impl Runnable for Create {
-    fn run(&self) -> Result<i32, Box<dyn Error>> {
+    fn run(&self) -> Result<(), Box<dyn Error>> {
         println!("Create NamePipe");
 
         let mut generator: Generator<rand::rngs::ThreadRng> =
@@ -64,6 +64,6 @@ impl Runnable for Create {
 
         create_name_pipe(&payload, 2000);
 
-        Ok(0)
+        Ok(())
     }
 }

src/actions/processes.rs

@@ -20,7 +20,7 @@ pub enum Commands {
 }
 
 impl Runnable for Processes {
-    fn run(&self) -> Result<i32, Box<dyn Error>> {
+    fn run(&self) -> Result<(), Box<dyn Error>> {
         match &self.command {
             Commands::Spoofing(spoofing) => spoofing as &dyn Runnable,
         }

src/actions/processes/spoofing.rs

@@ -97,9 +97,7 @@ fn spoof(executable: &str, parent_pid: u32) -> Result<(), Box<dyn Error>> {
 }
 
 impl Runnable for Spoofing {
-    fn run(&self) -> Result<i32, Box<dyn Error>> {
-        spoof(&self.executable, get_pid(&self.parent_executable)?)?;
-
-        Ok(0)
+    fn run(&self) -> Result<(), Box<dyn Error>> {
+        Ok(spoof(&self.executable, get_pid(&self.parent_executable)?)?)
     }
 }

src/main.rs

@@ -9,6 +9,7 @@ mod windows;
 use actions::Runnable;
 use clap::Parser;
 use cli::{Arguments, Commands};
+use std::error::Error;
 
 fn banner() {
     let banner: &str = "
@@ -23,17 +24,10 @@ fn banner() {
     println!("{}", banner);
 }
 
-fn main() {
+fn main() -> Result<(), Box<dyn Error>> {
     banner();
 
-    match Arguments::parse().command {
-        Commands::Actions(action) => match action.run() {
-            Ok(code) => std::process::exit(code),
-            Err(error) => {
-                println!("Error: {}", error);
-
-                std::process::exit(1);
-            }
-        },
-    };
+    Ok(match Arguments::parse().command {
+        Commands::Actions(action) => action.run()?,
+    })
 }