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, boxtemplate: &String) -> Opti } pub fn launch_cobalt_strike(project: Project) -> Option>{ + 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);