From 36fc4e67d4b8c23d726234f34113ae518d4a12e7 Mon Sep 17 00:00:00 2001
From: Santo Cariotti <santo@dcariotti.me>
Date: Sun, 15 Oct 2023 18:27:15 +0200
Subject: Add multiprocessing

---
 src/main.rs | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index 893b9f5..0694aa5 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -2,7 +2,8 @@ mod trace;
 
 use crate::trace::{exec, trace};
 use clap::Parser;
-use std::process::Command;
+use fork::{fork, Fork};
+use nix::unistd::Pid;
 
 #[derive(Parser)]
 #[command(author, version, about, long_about = None)]
@@ -12,15 +13,13 @@ struct Args {
 
 fn main() -> anyhow::Result<()> {
     let args = Args::parse();
-    let params = args.command.split(' ').collect::<Vec<&str>>();
 
-    let mut command = Command::new(params[0]);
-    if params.len() > 1 {
-        for arg in &params[1..] {
-            command.arg(arg);
-        }
-    }
-    let pid = exec(&mut command)?;
+    let pid = match fork() {
+        Ok(Fork::Child) => return exec(&args.command),
+        Ok(Fork::Parent(child)) => Pid::from_raw(child as i32),
+        Err(err) => panic!("fork() failed: {err}"),
+    };
+
     trace(pid)?;
 
     Ok(())
-- 
cgit v1.2.3-18-g5258