From fe6524016fba71eee580a56bb6fa418d2d79bd3d Mon Sep 17 00:00:00 2001
From: pyro57000 <github@rocketweasle.site>
Date: Thu, 13 Mar 2025 15:27:15 -0500
Subject: [PATCH] added a check to make sure cobalt strike exists before
 spawning a thread.

---
 pentest_tool/src/box_controls.rs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/pentest_tool/src/box_controls.rs b/pentest_tool/src/box_controls.rs
index 7e21fab..0639aa1 100644
--- a/pentest_tool/src/box_controls.rs
+++ b/pentest_tool/src/box_controls.rs
@@ -181,6 +181,13 @@ pub fn clean_unused_boxes(projects: &Vec<Project>, boxtemplate: &String) -> Opti
 }
 
 pub fn launch_cobalt_strike(project: Project) -> Option<JoinHandle<()>>{
+    let mut check_path = PathBuf::new();
+    check_path.push(&project.files_folder);
+    check_path.push("cobaltstrike/client");
+    if check_path.as_path().exists() == false{
+        println!("looks like you don't have cobalt strike set up on this beacon... try copying your cobalt strike folder here!");
+        return None;
+    }
     let handle = thread::spawn(move ||{
         let mut cs_dir = PathBuf::new();
         cs_dir.push(project.files_folder);