cleaned up all the warnings so it should compile without any!

dns enumeration, and added the ability to modify
the tool's configuration.

.gitignore

@@ -0,0 +1 @@
+/pentest_tool/target

README.md

@@ -53,11 +53,11 @@ Current engagements: client1 internal pentest, client 2 internal pentest
 
 This tool automatically creates the file structure, and if you use the start_pentest option populates the markdown note files with the templates I use.  
 
-For example if I get a new internal pentest engagement for client 3 but it hasn't started yet I'll start by creating the folder and note structure using option 4 in the menu system. This would create the folder structure and notes templates in the /home/pyro/pentests/upcomming/client3/internal_pentest and /home/pyro/notes/upcomming/client3/internal_pentest folders respectively. It does not create the distrobox or add it to the tracked project list yet.
+For example if I get a new internal pentest engagement for client 3 but it hasn't started yet I'll start by creating the folder and note structure using option 4 in the menu system. This would create the folder structure and notes templates in the /home/pyro/pentests/upcomming/client3/internal_pentest and /home/pyro/notes/upcomming/client3/internal_pentest folders respectively. It does not create the distrobox, but it does add it to the list of tracked projects, with a status of upcomming.
 
-Once the engagement starts I'd run option 6 "import new project - and set up new distrobox" from the menu system. It will ask if you have an existing folder structure to copy over, since I created that structure before with option 4 I would answer y, yes, or anything with the letter y in it. Then it prompts you to past the path to the folders to copy over. Here I browse to /home/pyro/pentests/upcomming in dolphin. I right click on the client3 folder and select "copy path" and then paste that into the terminal. I repeat this process for the notes folder.
+Once the engagement starts I'd run option 9 to promote a project from upcomming status to current status. This will automatically copy the needed files over, and will clean up the empty directory in the upcomming files directory, but the empty folder in the upcomming notes folder is still not cleaning correctly, I'm working on that.
 
-It then attempts to guess the client name and project name based on the folder structure, and asks you to confirm. If its correct the project object will be created, the folders will be copied, and a new distrobox will be set up. If it is not correct it will prompt you to correct it, then copy the files, create a new project object, and setup a new distrobox.  At the moment it is not very fault tollerant and will fail if there is an existing folder with the same name. I plan to fix that later, but haven't gotten around to it yet. Once the distro box is set up (this takes a while) it will return you to the main menu.
+It then will create a new distrobox and ask you if you need to set up cobalt strike for the project. if so it'll ask you for the path to your cobalt strike, it'll copy that into the project files folder so that you can run it if needed.
 
 During the test I use the menu system to spawn new terminals in the distrobox created for the engagement, use the various file generation options to get data in easily usable format for attack operations, etc.
 
@@ -75,7 +75,8 @@ Once the project is done and I'm ready to clean up the distrobox I use option 7
 1. clone this repository `git clone https://github.com/Pyro57000/pentest_tool.git`
 2. cd into the nested "pentest_tool" folder `cd pentest_tool/pentest_tool`
 3. use cargo to build the release binary `cargo build --release`
-4. follow the same installation instructions, skipping the step where you download the release binary.
+4. copy the compiled binary to a folder on your path `sudo cp ./target/release/pentest_tool /usr/bin/`
+5. follow the same installation instructions, skipping the step where you download the release binary.
 
 
 

ToDo.md

@@ -0,0 +1,29 @@
+# planned features
+1.) finish hash cracking with a dedictated cracking rig code.
+
+2.) cracte hash cracking with current computer code.
+
+
+4.) create code that moves projects to a "writing" state and folder.
+
+5.) create code that tracks "current, upcomming, and writing" states, maybe automatic zipping of folders after writing is done?
+    upcoming and current are done, but need to add writing yet...
+
+# Unplanned, but would be cool
+1.) create a "server" and "Client" infrastructure that can help manage the distrobox clients and the main server
+
+2.) maybe expand this server client model to interact with cracking rigs and what not.
+
+3.) implment a function to searchsploit and copy wanted exploits to the project folder.
+
+4.) implement a function to execute those copied exploits.
+
+5.) add a "client" model for compromised internal hosts, this would likely need to be a server from a networking perspective that listens on 127.0.0.1 and then have the main interface connect to that over a proxy to run jobs. a good starting point would be to run some of our enumeration commands and scripts and save the output to the notes directly.
+
+6.) add the ability to interact with the cobalt strike beacons... would need to learn agressor script... what a pain...
+
+
+# NOTE
+if you wish to contribute, please do! just fix a bug or implement any of the above features and make a pull request!! 
+
+I'll keep plugging away as I have time throughout my days as well.

pentest_tool/Cargo.toml

@@ -1,9 +1,15 @@
 [package]
 name = "pentest_tool"
-version = "0.2.0"
+version = "3.1.1"
 edition = "2021"
 
 [dependencies]
+chrono = "0.4.39"
 clearscreen = "3.0.0"
 directories = "5.0.1"
+dns-lookup = "2.0.4"
 fs_extra = "1.3.0"
+futures-io = { version = "0.2.0-beta" }
+reqwest = {version = "0.12.12", features = ["blocking", "json"]}
+rodio = "0.20.1"
+walkdir = "2.5.0"

pentest_tool/src/box_controls.rs

@@ -1,5 +1,11 @@
-use std::process;
+use std::process::Command;
-use crate::Project;
+use std::{path::PathBuf, process};
+use std::env;
+use std::fs;
+use std::io::Write;
+use std::thread::{self, JoinHandle};
+use std::time::Duration;
+use crate::{get_user_input, Project};
 
 pub fn stop_all_boxes(projects: &Vec<Project>){
     let mut problem_childs: Vec<Project> = Vec::new();
@@ -37,6 +43,10 @@ pub fn project_standalone_terminal(project: Project, mut terminal: String){
     if terminal.contains("profile"){
         profile = true;
     }
+    if terminal.contains("!!!"){
+        let terminal_launch_cmd = format!("distrobox enter --root {}", &project.boxname);
+        terminal = terminal.replace("!!!", &terminal_launch_cmd);
+    }
     let terminal_vec:Vec<&str> = terminal.split(" ").collect();
     let mut terminal_start = process::Command::new(terminal_vec[0]);
     let mut first = true;
@@ -63,3 +73,216 @@ pub fn project_standalone_terminal(project: Project, mut terminal: String){
 pub fn project_inline_terminal(project: Project){
     process::Command::new("distrobox").arg("enter").arg("--root").arg(project.boxname).arg("--").arg("script").arg("-a").arg("-B").arg("/pentest/working/terminal.log").status().expect("error opeing konsole");
 }
+
+#[allow(unused)]
+pub fn make_box(project: &Project, tools_dir: &PathBuf, boxtemplate: &String, new: bool, fingerprint: bool){
+    println!("stopping template box to ensure we can clone it!");
+    let stop_result = Command::new("distrobox").arg("stop").arg("--root").arg(boxtemplate).status();
+    if stop_result.is_err(){
+        println!("error stopping template!");
+    }
+    if !new{
+        let _distrobox_stop_status = process::Command::new("distrobox").arg("stop").arg("--root").arg(&project.boxname).status().expect("error stopping distrobox");
+        let distrobox_rm_status = process::Command::new("distrobox-rm")
+                                                                .arg("--root")
+                                                                .arg("-f")
+                                                                .arg(&project.boxname)
+                                                                .status().expect("error calling distrobox");
+        if distrobox_rm_status.success(){
+            println!("Distrobox Removal Successful!!!");
+        }
+        else{
+            println!("Distrobox Removal Failed, manual removal required!");
+        }
+    }
+    let mut box_name_path = project.files_folder.clone();
+    let box_name = format!("{}_{}", &boxtemplate, &project.customer);
+    box_name_path.push("boxname");
+    let mut box_name_file = fs::File::create(box_name_path).expect("Error creating box name file");
+    box_name_file.write_all(&box_name.as_bytes()).expect("error writing boxname to box file");
+    let pentest_volume = format!("{}:/pentest:rw", &project.files_folder.display());
+    let toold_volume = format!("{}:/tools:rw", tools_dir.display());
+    let mut distrobox_cmd = String::new();
+    if fingerprint{
+        println!("creating box with shared volume for fingerprints... note you will still need to set up fingerprint authentication in your distrobox");
+        println!("\nfor example, you may need to install fprintd and imagemegick on your template box, and set up the pam files to utilize finger print auth");
+        println!("\nsee https://wiki.archlinux.org/title/Fprint for more information and instructions");
+        distrobox_cmd = format!("distrobox create --root --init --clone {} --volume {} --volume {} --volume /var/lib/fprint:/var/lib/fprint:rw --name {}", boxtemplate, toold_volume,  pentest_volume, box_name);
+    }
+    else {
+        distrobox_cmd = format!("distrobox create --root --init --clone {} --volume {} --volume {} --name {}", boxtemplate, toold_volume,  pentest_volume, box_name); 
+    }
+    println!("{}", distrobox_cmd);
+    let distrobox_cmd_vec: Vec<&str> = distrobox_cmd.split_whitespace().collect();
+    let mut distrobox_cmd_build = process::Command::new("distrobox");
+    let mut first = true;
+    for word in &distrobox_cmd_vec{
+        if first == false{
+            println!("adding {} as an argument for the creation command...", word);
+            distrobox_cmd_build.arg(word);
+        }
+        else{
+            first = false;
+        }
+    }
+    let distrobox_result = distrobox_cmd_build.status();
+    if distrobox_result.is_err(){
+        println!("oooof we ran into trouble creating your distrobox!!");
+        println!("try creating it manually!");
+        println!("{}", distrobox_cmd);
+    }
+    else{                                                     
+        if distrobox_result.unwrap().success(){
+            println!("we made a distrobox oh boy!");
+            let distrobox_start_result = process::Command::new("distrobox")
+                                                                            .arg("enter")
+                                                                            .arg("--root")
+                                                                            .arg(&box_name)
+                                                                            .arg("--")
+                                                                            .arg("sudo")
+                                                                            .arg("-s")
+                                                                            .arg("ln")
+                                                                            .arg("-sf")
+                                                                            .arg("/pentest/boxname")
+                                                                            .arg("/etc/boxname")
+                                                                            .status();
+            if distrobox_start_result.is_err(){
+                println!("ooof did not start successfully try entering it yoruself");
+                println!("distrobox enter --root {} -- sudo -s ln -sf /pentest/boxname /etc/boxname", &box_name);
+            }
+            else if distrobox_start_result.unwrap().success(){
+                println!("distrobox was started as well!!!! good job me!");
+            }
+            else {
+                println!("ooof did not start successfully try entering it yoruself");
+                println!("distrobox enter --root {} -- sudo -s ln -sf /pentest/boxname /etc/boxname", &box_name);
+            }
+        }
+    }
+}
+
+
+pub fn clean_unused_boxes(projects: &Vec<Project>, boxtemplate: &String) -> Option<JoinHandle<()>>{
+    println!("starting template box: {}", boxtemplate);
+    let template_status = process::Command::new("distrobox").arg("enter").arg("--root").arg(boxtemplate).arg("--").arg("exit").status();
+    if template_status.is_err(){
+        let start_error = template_status.err().unwrap();
+        println!("OOOF issue starting template box, cancelling...");
+        println!("ERROR: {}", start_error);
+        return None;
+    }
+    println!("starting project boxes...");
+    for project in projects{
+        if project.stage.contains("current"){
+            let start_status = process::Command::new("distrobox").arg("enter").arg("--root").arg(&project.boxname).arg("--").arg("exit").status();
+            if start_status.is_err(){
+                let start_error = start_status.err().unwrap();
+                println!("OOOF issue starting {}, cancelling...", project.boxname);
+                println!("ERROR: {}", start_error);
+                return None;
+            }
+        }
+    }
+    println!("pruning unused containers...");
+    let handle = thread::spawn(move ||{
+        let spawn_result = process::Command::new("sudo").arg("podman").arg("system").arg("prune").arg("-a").arg("-f").output();
+    if spawn_result.is_err(){
+        let spawn_error = spawn_result.err().unwrap();
+        println!("oof trouble spawing prune command!, try manually");
+        println!("ERROR: {}", spawn_error);
+    }
+    println!("PRUNING COMPLETE!");
+    });
+    thread::sleep(Duration::from_secs(3));
+    println!("this will take some time, but its running on a different thread so you can continue working!");
+    return Some(handle);
+}
+
+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 to the following path!");
+        println!("{}", &check_path.display());
+        return None;
+    }
+    println!("you will need to enter your sudo password shortly, when you've done that successfully please type END and hit enter");
+    let handle = thread::spawn(move ||{
+        let mut cs_dir = PathBuf::new();
+        cs_dir.push(project.files_folder);
+        cs_dir.push("cobaltstrike/client");
+        let cd_res = env::set_current_dir(&cs_dir);
+        if cd_res.is_ok(){
+            let cobalt_strike_launch_result = Command::new("distrobox")
+                                                                                        .arg("enter")
+                                                                                        .arg("--root").arg(project.boxname)
+                                                                                        .arg("--")
+                                                                                        .arg("./cobaltstrike")
+                                                                                        .output();
+            if cobalt_strike_launch_result.is_err(){
+                let error = cobalt_strike_launch_result.err().unwrap();
+                println!("error launching cobalt strike!");
+                println!("{}", error);
+            }
+        }
+    });
+    thread::sleep(Duration::from_secs(5));
+    loop{
+        let end = get_user_input("please enter END if the sudo password was enter correctly!");
+        if end == "END".to_string(){
+            break;
+        }
+    }
+    return Some(handle);
+}
+
+#[allow(unused)]
+pub fn launch_bloodhound_gui(project: Project) -> Option<JoinHandle<()>>{
+    let mut bloodhound_command = String::new();
+    let version_response = get_user_input("do you want to use a specific bloodhound version?");
+    if version_response.to_lowercase().contains("y"){
+        bloodhound_command = get_user_input("path to the bloodhound binary you want to use?");
+    }
+    else{
+        bloodhound_command = "bloodhound".to_owned();
+    }
+    let mut neo4j_started = false;
+    let neo4jstart_res = Command::new("distrobox")
+                                                        .arg("enter")
+                                                        .arg("--root")
+                                                        .arg(&project.boxname)
+                                                        .arg("--")
+                                                        .arg("sudo")
+                                                        .arg("neo4j")
+                                                        .arg("start")
+                                                        .status();
+    if neo4jstart_res.is_err(){
+        let error = neo4jstart_res.err().unwrap();
+        println!("errror starting neo4j...");
+        println!("{}", error);
+    }
+    else{
+        neo4j_started = true;
+        thread::sleep(Duration::from_secs(3));
+    }
+    if neo4j_started == false{
+        println!("review previous errors, not spawning bloodhound thread...");
+        return None;
+    }
+    let handle = thread::spawn(move ||{
+        let bloodhound_res = Command::new("distrobox")
+                                                        .arg("enter")
+                                                        .arg("--root")
+                                                        .arg(&project.boxname)
+                                                        .arg("--")
+                                                        .arg(bloodhound_command)
+                                                        .output();
+        if bloodhound_res.is_err(){
+            let error = bloodhound_res.err().unwrap();
+            println!("error starting bloodhound!");
+            println!("{}", error);
+        }
+    });
+    return Some(handle);
+}

pentest_tool/src/cli.rs

@@ -0,0 +1,340 @@
+use std::path::PathBuf;
+use std::process::exit;
+use std::thread::JoinHandle;
+use chrono::Datelike;
+use clearscreen::clear;
+use clearscreen;
+use chrono::Local;
+use crate::Project;
+use crate::project_controls;
+use crate::box_controls;
+use crate::info_controls;
+use crate::start_pentest;
+use crate::get_user_input;
+use crate::menu;
+use crate::portscan_controls;
+use crate::victim_commands;
+use crate::enumeration;
+use crate:: tool_controls;
+use crate::configuration;
+
+
+fn help(command: Option<String>){
+    if command.is_some(){
+        let help_cmd = command.unwrap();
+        match help_cmd.as_str(){
+            "list projects" | "lp" | "listp" | "list p" => {println!("Command: list projects\nAliases: lp, listp, list p\n\nThis command lists all projects currently tracked by the pentest_tool"); return;},
+            "switch project" | "swp" | "switch p" | "switchp" => {println!("Command: switch project\nAliases: swp, switch p, switchp\n\nThis command will switch the active project from the current one to a new on of your choosing. It will prompt you to make a selection."); return;},
+            "show active project" | "show active" | "sa" | "show a" => {println!("Command: show active project\nAliases: sa, show a\n\nThis command shows information about the currently active project. NOTE the most useful information is already displayed above the CLI prompt."); return;},
+            "create new project" | "cnp" | "new project" | "np" => {println!("Command: create new project\nAliases: cnp, new project, np\n\nThis command creates a new project and default note structure based on pyro's perferred note structure. It will prompt you for any needed information."); return;},
+            "save projects" | "sp" | "save" | "s" => {println!("Command: save projects\nAliases: sp save\n\nThis command saves all project information to the ~/.config/pyro_pentest_tool/projects.conf file"); return;},
+            "import project" | "ip" | "import" => {println!("Command: import project\nAliases: ip, import\n\nThis command will impot an existing project and set up a new distrobox for the project if it is a current project."); return;},
+            "remove project" | "rp" | "remove" | "rmp" => {println!("Command:remove project\nAliases:rp, remove, rmp\n\nThis command removes a project from the list of projects tracked by the pentest_tool, and will destroy its distrobox. It does not remove any directories used or created by the distrobox. Run this after the report is written."); return;},
+            "show upcoming projects" | "sup" | "show upcoming" => {println!("Command:show upcoming projects\nAliases:sup, show upcoming\n\nThis command shows a list of upcoming projects. Use this to verify which project you want to promote when the project enters the active phase."); return;},
+            "promote project" | "pp" | "promote" => {println!("Command:promote project\nAliases:pp (ha), promote\n\nThis command promotes an upcoming project to a current project. It will copy the folders that were created over to the current project space and set up a new distrobox for the project."); return;},
+            "new terminal" | "enter" | "enter terminal" | "nt" | "et" => {println!("Command:new terminal\nAliases:enter, enter terminal, nt, et\n\nThis command spawns a new terminal window in the active project's distrobox. Use this to interact with your project's distrobox."); return;},
+            "inline terminal" | "it" | "enter inline" | "ei" => {println!("Command:inline terminal\nAliases:it enter, inline, ei\n\nThis command spawns a terminal in this window using the current active project's distrobox"); return;},
+            "cobalt strike" | "cs"  => {println!("Command:cobalt strike\nAliases:cs\n\nThis command opens cobalt strike in the active project's distrobox, and spins it off as a new thread to ensure it doesn't block the rest of this tools operation. NOTE the cobalt strike window will need to be closed before this tool exits sucessfully."); return;},
+            "recreate distrobox" | "rdb" | "ndb" | "new distrobox" => {println!("Command:recreate distrobox\nAliases:rdb, ndb, new distrobox\n\nThis command destroyes the existing distrobox for the currently active project, and clones a new one based on the current state of the template distrobox."); return;},
+            "generate userpass" | "userpass" | "gup" | "up" => {println!("Command:generate userpass\nAliases:userpass, gup, up\n\nThis command generates a userpass file based on the active project's notes. The file will be in the username:password format."); return;},
+            "inital enum" | "ie" | "enum" => {println!("Command:initial enum\nAliases:ie, enum\n\nThis command runs the initial enum script on a nessus csv and saves the output to the active project's notes in the host_notes.md file."); return;},
+            "build attack notes" | "ban" | "attack notes" | "hn" => {println!("Command:build attack notes\nAliases:ban, attack notes\n\nThis command builds the active project's attack note based on the active project's host notes (for external tests). It is expected that you'd run the initial enum command, then manually fill out the enumeration talbes with correct service names and ports."); return;},
+            "host discovery" | "build host discovery" | "hd" | "bhd"  => {println!("Command:host discovery\nAliases:build host discovery, hd, bhd\n\nThis command prints the host discovery ping command for the active project, based on the scope table in the general.md notes file."); return;},
+            "cobaltstrike port scan" | "cs port scan" | "csps" => {println!("Command:port scan\nAliases:cs port scan, cobaltstrike port scan, csps, ps (tell your cat I said that)\n\nThis command prints the cobalt strike portscan command based on the active project's scope table in the general.md notes file"); return;},
+            "parse port scan" | "pps" | "parse scan" => {println!("Command:parse port scan\nAliases:pps, parse scan\n\nThis commmand parses a cobalt strike portscan TSV and saves interesting hoests to files to the active project's files folder. The host files are designated with the service that was detected that might be interesting. Use this to generate target lists for specific protocols."); return;},
+            "stop boxes" | "stop distroboxes" | "sdb" => {println!("Command:stop boxes\nAliases:stop distroboxes, sdb\n\nThis command stops all distroboxes for the tracked projects. Note if the distrobox isn't running you will see errors in the console, you can safely ignore these."); return;},
+            "password spray" | "pass spray" | "pas" => {println!("Command:password spray\nAliases:pass spray, pas\n\nThis command iterates through the password spray note file and print the command to perform the spray, waiting the proper observation window beteen commands. It prompts you to save if needed. NOTE this will block execution for the rest of the program until it is either finished, or you save and exit the password spray function. I'm working on making this better."); return;},
+            "bloodhound" | "bh" => {println!("Command:bloodhound\nAliases:bh\n\nThis command launches bloodhound in the active project's distrobox. It will automatically start neo4j before staring bloodhound."); return;},
+            "parse gather contacts" | "pgc" | "parse contacts" | "pc" => {println!("Command:parse gather contacts\nAliases:pgc, parse contacts, pc\n\nThis command parses output saved from the gather contacts burpsuite extension."); return;},
+            "prune distroboxes" | "pdb" | "prune" => {println!("Command:prune distroboxes\nAliases:pdb, prune\n\nthis command prunes distroboxes for all projects that are not being tracked by this tool (frees up system storage). This will start all the currently acvtive distorboxes to ensure they don't get pruned, and then will delete all the not-started distrobox volumes and resources."); return;},
+            "clear" | "clear screen" | "cls" => {println!("Command:clear\nAliases:clear screen, cls\n\nThis command clears the screen of command output."); return},
+            "exit" => {println!("Command:exit\nAliases:(none)\nThis command exits the pentest_tool, it will save all project infomation, and wait for all threads to re-join the main thread before exiting."); return;},
+            "settings" => {println!("\n\nThe settings file is located at ~/.config/pyro_pentest_tool/conf\n\nThe format is setting_name:setting_value.\n\nNeeded settings are\n  project_files - the place to store current project files folders\n\n  projtect_notes - the place to store current project notes\n\n  tools_folder - the place to store custom tools like those downloaded from github\n\n  upcoming_files - the place to store upcoming project files folders\n\n  upcoming_notes -  The place to store upcoming project note files\n\n  box_template - the name of the distrobox you want to clone for project boxes\n\n  terminal - the command you use to launch a terminal, while running a specific command: Ex: konsole -e \n\n  cracking_rig - the user and host you use for a personal cracking rig in the openssh formating: Ex: pyro@cracking_rig or pyro@192.168.1.101 if you do not have a cracking rig the default is @n\n\n rockyou_location - the location on the cracking rig for the rockyou.txt file default is n\n\n  rule_location - the location on the cracking rig for the one rule to rule them all file. Default is n\n\n  pass_file - this is the location where you store your standard password spray file. If you do not have a custom one this tool provides one. The default is ~/.config/pyro_pentest_tool/passwordspary.md\n\n  fingerprint - this is whether you want fingerprint authentication within your distroboxes, takes y/n\n\n  vault_name - the name of your obsidian vault, default is notes\n\n"); return;},
+            "parse normal nmap file" | "pnnf" | "parse nmap" | "pn" => {println!("Command:parse normal nmap file\nAliases: pnnf, parse nmap, pn\n\nThis command parses the normal output of an nmap scan (like if you just tee'd or >'d it to a file) and outputs in host:port or int the coablt strike tsv format. It will attempt to find the file within the active project's files folder, and if it can't find the file it will prompt you for input.")},
+            "sharpersist command" | "spc" | "sharp scheduled task" | "sst" => {println!("Command: sharpersist command\nAliases: spc, sharp scheduled task, sst\n\nThis comand prints the commands to run to use sharpersist.exe to create a scheduled task that runs hourly called FRPersist.")},
+            "port scan" | "ps" | "nmap" | "nmap scan" | "ns" | "nm" => {println!("Command: port scan\nAliases: ps, nmap, nmap scan, ns, nm\n\nThis command runs an nmap scan against the scope in the active projects notes, and saves the output.")},
+            "show scope" | "ss" | "show s" | "s s" | "scope" => {println!("Command:show scope\nAliases:ss, show s, s s, scope\n\nThis command displays the current project's scope as just the hosts in the scope table in your notes.")},
+            "port scan command" | "psc" | "nmap command" | "nmc" => {println!("command:port scan command\nAliases:psc,nmap command, nmc\n\nThis command will print the nmap command to manually run a scan to the terminal so you can copy paste it.")},
+            "update git tools" | "ugt" | "update git" | "ug" => {println!("Command: update git tools\nAliases: update git, ugt, ug\n\nThis command attempts to update the git tools in your tools directory, it will attempt to update every directory as a git project. If the directory is not a git project it should just error out and continue to the next one.")},
+            "dns records" | "dr" => {println!("Command:dnsrecords\nAliases:dr\n\nThis command will run dns recon inside of your distrobox and save the results to your enumeration notes.")},
+            "brute force subdomains"| "bsd" | "gobuster dns" | "gd" => {println!("Command:brute force subdomains\nAliases:bsd,gobuster dns, gd\n\nthis command will run gobuster in the project's distrobox and save the results to your notes.")},
+            "dns enumeration" | "de" | "all dns stuff" | "ads" | "dns stuff" | "ds" => {println!("Command:dns enumeration\nAliases:de, all dns stuff, ads, dns stuff, de\n\nThis command will perform both dns record enumeration with dnsrecon, and subdomain enumeration using gobster inside of your distrobox and save the output to your notes.")},
+            "modify tool config" | "mtc"  => {println!("Command: modify tool config\nAliases: mtc\n\nThis command lets you modify the tool's configuration.");}
+            _ => ()
+        }
+    }
+    println!("Welcom to Pyro's pentest command line!");
+    println!("the pentest_tool uses a configuration file to store your settings. This configuration file is located at ~/.config/pyro_pentest_tool/conf.\nYou can modify this file, but do so cautiously, incorrect formatting will cause this program to have a bruh moment.\nThe correct format is setting_name:setting_value\nExample:\nprojecT_files:/var/home/pyro/pentests/current");
+    println!("for help configuring the settings file run help settings");
+    println!("available commands: name | aliases | ...");
+    print!("
+menu | main menu | mm
+list projects |  lp  |  listp  |  list p  
+switch project  |  sp  |  switch p  |  switchp  
+show active project  |  show active  |  sa  |  show a 
+create new project  |  cnp  |  new project  |  np  
+save projects  |  sp  |  save  |  s  
+import project  |  ip  |  import  
+remove project  |  rp  |  remove  |  rmp  
+show upcoming project  |  sup  |  show upcoming  
+promote project  |  pp  |  promote  
+new terminal  |  enter  |  enter terminal  |  nt  |  et  
+inline terminal  |  it  |  enter inline  |  ei  
+cobalt strike  |  cs   
+recreate distrobox  |  rdb  |  ndb  |  new distrobox  
+generate userpass  |  userpass  |  gup  |  up  
+inital enum  |  ie  |  enum  
+host discovery  |  build host discovery  |  hd  |  bhd   
+port scan  |  cs port scan  |  cobaltstrike port scan  |  csps  |  ps  
+parse port scan  |  pps  |  parse scan  
+stop boxes  |  stop distroboxes  |  sdb  
+password spray  |  pass spray  |  pas  
+bloodhound  |  bh 
+parse gather contacts  |  pgc  |  parse contacts  |  pc  
+prune distroboxes  |  pdb  |  prune  
+clear | clear screen | cls
+parse nomral nmap file | pnnf | parse nmap | pn
+show scope | ss | show s | s s | scope
+sharpersist command | spc | sharp scheduled task
+port scan | ps | nmap | nmap scan | ns | nm
+port scan command | psc | nmap command | nmc
+update git tools | ugt | update git | ug
+dns records | dr 
+brute force subdomains| bsd | gobuster dns | gd 
+dns enumeration | de | all dns stuff | ads | dns stuff | ds  
+modify tool config | mtc
+help  |  ?  |  -h 
+")
+}
+
+pub fn get_active_project(projects: &Vec<Project>) -> &Project{
+    let mut active_project = &projects[0];
+    for project in projects{
+        if project.active == true{
+            active_project = project
+        }
+    }
+    return active_project
+}
+
+pub fn next_project_id(config_path: &PathBuf) -> i32{
+    let projects = project_controls::get_projects(config_path, false);
+    let mut new_id = 0;
+    for project in projects.clone(){
+        if project.id > new_id{
+            new_id = project.id + 1;
+        }
+    }
+    return new_id;
+}
+#[allow(unused)]
+pub fn run_command(cmd: String,
+                   mut projects: &mut Vec<Project>, 
+                   config_path: PathBuf, 
+                   base_files: &PathBuf, 
+                   base_notes: &PathBuf, 
+                   tools_dir: &PathBuf, 
+                   boxtemplate: String, 
+                   terminal: String, 
+                   cracking_rig: String, 
+                   rockyou: String, 
+                   rule: String, 
+                   upcoming_files: &PathBuf, 
+                   upcoming_notes: &PathBuf, 
+                   password_spray_file: &PathBuf, 
+                   fingerprint: bool, 
+                   vault_name: String) -> Option<JoinHandle<()>> {
+    let mut new_id = next_project_id(&config_path);
+    let active_project = get_active_project(&projects);
+    let mut notes_folder_string = format!("{}", &active_project.notes_folder.display());
+    let mut obsidian_folder_vec = PathBuf::new();
+    let mut reached_vault_folder = false;
+    for folder in notes_folder_string.split("/").collect::<Vec<&str>>(){
+        if !folder.contains(&vault_name){
+            reached_vault_folder = true;
+            obsidian_folder_vec.push(folder);
+        }
+        else{
+            if reached_vault_folder{
+                obsidian_folder_vec.push(folder);
+            }
+        }
+    }
+    let obsidian_uri = format!("obsidian://open?vault={}&file={}", vault_name, obsidian_folder_vec.display().to_string().replace("/", "%2F"));
+    let mut response = String::new();
+    let now = Local::now();
+    let month = now.month();
+    let year = now.year();
+    let mut season = String::new();
+    let mut lseason = String::new();
+    match month{
+        12 | 01 | 02 => {season = "Winter".to_owned(); lseason = "Fall".to_owned()},
+        03 | 04 | 05 => {season = "Spring".to_owned(); lseason = "Winter".to_owned()},
+        06 | 07 | 08 => {season = "Summer".to_owned(); lseason = "Spring".to_owned()},
+        09 | 10 | 11 => {season = "Fall".to_owned(); lseason = "Summer".to_owned()},
+        _ => {println!("error getting season! Check code..."); exit(1)}
+    }
+    if cmd.contains("help"){
+        if cmd.contains(" "){
+            let help_with = &cmd.split(" ").collect::<Vec<&str>>()[1].to_owned();
+            help(Some(help_with.to_owned()));
+        }
+        else{
+            help(None);
+        }
+        return None;
+    }
+    match cmd.as_str(){
+                        "list projects" | "lp" | "listp" | "list p" => {project_controls::list_projects(&projects); return None},
+                        "switch project" | "swp" | "switch p" | "switchp" => {project_controls::switch_project(&mut projects); return None},
+                        "show active project" | "show active" | "sa" | "show a" => {println!("\nclient: {}\n\nproject: {}\n\nbox: {}\n\nproject files: {}\n\nproject notes: {}\n", active_project.customer ,active_project.project_name, active_project.boxname, active_project.files_folder.display(), active_project.notes_folder.display()); return None},
+                        "create new project" | "cnp" | "new project" | "np" => {new_id = new_id + 1; start_pentest::start_pentest(&config_path, &mut projects, new_id, upcoming_files, upcoming_notes, &boxtemplate, password_spray_file); return None},
+                        "save projects" | "sp" | "save" | "s" => {project_controls::save_projects(&projects, &config_path); return None},
+                        "import project" | "ip" | "import" => {new_id = new_id + 1; project_controls::new_project(&mut projects, &base_files, &base_notes, &tools_dir, &boxtemplate, &config_path, new_id, &upcoming_files, &upcoming_notes, fingerprint); return None},
+                        "remove project" | "rp" | "remove" | "rmp" => {project_controls::remove_project(&mut projects, &config_path); return None},
+                        "show upcoming projects" | "sup" | "show upcoming" => {project_controls::print_upcoming_projects(&projects); return None},
+                        "promote project" | "pp" | "promote" => {project_controls::promote_project(&mut projects, &config_path, base_files, base_notes, tools_dir, &boxtemplate, fingerprint); return None},
+                        "new terminal" | "enter" | "enter terminal" | "nt" | "et" => {box_controls::project_standalone_terminal(active_project.clone(), terminal.clone()); return None},
+                        "inline terminal" | "it" | "enter inline" | "ei" => {box_controls::project_inline_terminal(active_project.clone()); return None},
+                        "cobalt strike" | "cs"  => {let cs_thread = box_controls::launch_cobalt_strike(active_project.clone()); return cs_thread},
+                        "recreate distrobox" | "rdb" | "ndb" | "new distrobox" => {box_controls::make_box(&active_project, &tools_dir, &boxtemplate, false, fingerprint); return None},
+                        "generate userpass" | "userpass" | "gup" | "up" => {info_controls::generate_userpass(&active_project); return None},
+                        "inital enum" | "ie" | "enum" => {info_controls::run_initial_enum(&active_project); return None},
+                        "build attack notes" | "ban" | "attack notes" | "hn" => {portscan_controls::build_cmd_for_host_discovery(&active_project); return None;}
+                        "host discovery" | "build host discovery" | "hd" | "bhd"  => {portscan_controls::build_cmd_for_host_discovery(&active_project); return None},
+                        "cobaltstrike port scan" | "cs port scan" | "csps" => {portscan_controls::build_cs_portscan_cmd(&active_project); return None},
+                        "parse port scan" | "pps" | "parse scan" => {portscan_controls::parse_csportscan(&active_project); return None},
+                        "stop boxes" | "stop distroboxes" | "sdb" => {box_controls::stop_all_boxes(&projects); return None},
+                        "password spray" | "pass spray" | "pas" => {info_controls::password_spray_help(&active_project, season, lseason, year, &tools_dir, &config_path); return None},
+                        "bloodhound" | "bh" => {let bloodhound_handle = box_controls::launch_bloodhound_gui(active_project.clone()).unwrap(); return Some(bloodhound_handle);},
+                        "parse gather contacts" | "pgc" | "parse contacts" | "pc" => {info_controls::partse_gathercontacts(&active_project); return None},
+                        "prune distroboxes" | "pdb" | "prune" => {let prune_thread = box_controls::clean_unused_boxes(&projects, &boxtemplate); return prune_thread},
+                        "parse normal nmap file" | "pnnf" | "parse nmap" | "pn" => {portscan_controls::parse_normal_nmap_output(&active_project); return None;},
+                        "show scope" | "ss" | "show s" | "s s" | "scope" => {let scope_res = info_controls::get_scope_entries(&active_project); if scope_res.is_some(){for host in scope_res.unwrap(){println!("{}", host)}}return None},
+                        "update git tools" | "ugt" | "update git" | "ug" => {tool_controls::update_git_tools(tools_dir); return None},
+                        "port scan" | "ps" | "nmap" | "nmap scan" | "ns" | "nm" => {portscan_controls::run_nmap_portscan(&active_project); return None;},
+                        "port scan command" | "psc" | "nmap command" | "nmc" => {portscan_controls::build_nmap_command(&active_project); return None;}
+                        "sharpersist command" | "spc" | "sharp scheduled task" | "sst" => {victim_commands::sharp_persist_command(&tools_dir); return None;},
+                        "dns records" | "dr" => {let dns_handle = enumeration::run_dns_enumeration(&active_project, None, true); return dns_handle;},
+                        "brute force subdomains"| "bsd" | "gobuster dns" | "gd" => {let gobuster_handle = enumeration::bruteforce_subs(&active_project, None,None, true); return gobuster_handle},
+                        "dns enumeration" | "de" | "all dns stuff" | "ads" | "dns stuff" | "ds" => {let all_dns_handle = enumeration::do_all_dns_enumeration(&active_project); return all_dns_handle},
+                        "dns squatting scan" | "dnstwist" | "dss" => {let twist_handle = enumeration::dns_squatting(&active_project, None, true); return twist_handle},
+                        "print report information" | "pri" => {info_controls::print_report_information(&active_project); return None;},
+                        "modify tool config" | "mtc"  => {configuration::generate_tool_config(&config_path); return None;},
+                        _ => {help(None); println!("\n\n unknown command."); return None;}
+    }
+}
+
+fn print_banner(banner: &str){
+    print!("{}", banner);
+}
+
+pub fn cli(interactive: bool,
+           mut projects: Vec<Project>, 
+           config_path: PathBuf, 
+           base_files: &PathBuf, 
+           base_notes: &PathBuf, 
+           tools_dir: &PathBuf, 
+           boxtemplate: String, 
+           terminal: String, 
+           cracking_rig: String, 
+           rockyou: String, 
+           rule: String, 
+           upcoming_files: &PathBuf, 
+           upcoming_notes: &PathBuf, 
+           password_spray_file: &PathBuf, 
+           fingerprint: bool, 
+           vault_name: String) {
+    let mut threads = Vec::new();
+    if interactive{
+        let mut loopize = true;
+        let banner = "
+,,,;;::ccccc::;;;::c::;,;::cccccllc::::::;:::;;;;,,;,'',,;,,;;;;;;;:;;;;;,,,,,,,,,,,'''''',,,,,,''''
+,;;;::ccccc::::::ccc:;;;:ccccccclc::ccccccc::;;;;;;;;;;,,;;;;;;;;;;;;;;;,,,,,,,,,,,'''''''''',,,,,''
+,;;:::ccc:cc:::ccc:::::::ccccclcccllccccllc::::::;;;;;;;;;;;;;;;;;;;;;;,,,,,,,,,,,''''''''...'',,,,'
+,;;:::c::ccc::cc::::::::cclollllllolllllccccc::cc:::::;;;;;;;;;;;;;;;;;;,,,,,,,,,,'''''''''''..'',,,
+,;::::::ccc::cc::::::ccloodollooooollllcccccc:llc::::::;;;;;;:;;;;;;;;;;;,,,,,,,,,,''''''''''''''',,
+,;:::::c:::c::::ccccloddxxddxxddddodollllccclclcccccc:::::::::::::::;;;;;;;;,,,,,,,,,'''''''''''''',
+;;:::::::c::c::clllodxxO0OKX0kkOkkxxxxxdooooolcccccccc:::::::cllc::::::::;;;;;,,,,,,,,,,'''''''''',,
+;:::::c:cc::cclolclokO0KXNNX00KKK0O0KOxdxxdooccccclllccccccccdkdlcccccllcc::;;;;;;;;,,,,,,,,,,,',,,,
+::::::cc::::coxdlllok00KNWNXXX0KXKOKNOkO0kddocccllllllccccclx0Kkodddoodddoollc::;;;;;;;;;,,,,,,,,,,,
+:::::::c:::clkkodooxxkO0KX0xookKKkkKNKO0KkdoodolcllllollolldKNNXKKXKKKKKK0Okxdocc:cc:::;;;;;,,,,,,,,
+::cc::cc::cldxllolodxxdoddc'.,okkxOXXOdxkxolkOdlllllllldkdokXNNNNNNNNX0kxollcc:::::cclc::;;;;;;,,,,,
+:::::::cccldko:,.';cc:;:;....;clllOXOxxOkocoK0xooddollx0Odd0XNNNNNX0Oxdolcccc::;;;;;;:cllc:;;:;,,,,,
+;;::c:::ccldkl;'...,''''....',;,';dxdkkdc;cONKxxOOOxddOXOdx0XXNNWNOdddxkOOOOkdllc:;,,,;cool:;;;;;,,;
+,;;::;;::llco:,'..''..,.......''.';:ldl;,,:xXNOOXXX0xdkOOddkXNNWWWX00KXNNX0kxddddol:,''';lol:;;:;;,;
+,,,;;;;;:coc;;'..;;. .,,;'.......':dxdc;ldc,l00xkXNKxodkkkkk0XNWWMWWWNXKOxdooolooool:;'..,lol::::;;;
+',,,;;;;:cllc;,..',.  ','.  .....;odoo:;co:.'ldldOOx::x0KXX0kk0XNWWWXOxdoooollllllllcc:'..':lc:::;;;
+',,,;;;;;:cccc:,. .          ..;cccccc:,,''.',,:l:;;:oOXXKOOOkxOXNNNXOxddooooollllllllc,....:c:::;;;
+''',,,;;;;;;;cll,..   ..    .':lc:c:;,,......,,;:;;:cokkxxO00O0KXNNN0kxkkkxddoollllllll:'...':::::::
+.''',,,,,,,,,;:c:,.. ..'.  ..','',;;'........',,;:;:::cdxxxddkKXXXKKKKXXXXXX0kdoloolllol;....,;:::::
+..'''',,'',,,;;:::;..... ............... .'.....,;,',:ldc;:ldOKKK00KNWWWNNXK0xoooodooooo:'...';;;:;;
+....'''''',,;;::cll:,''......  .      ..........'...,;;l:,,;oddkOOKNWWWWNX0kdodxxxxdddooc,...',;;;;,
+......''''',;::cloodddol;.               ...........',.;;,,',:cxdd0KXXKKKXKOkxxkkkxdddooc,...';;,,,,
+........''',;:clloddxxdo:'.              .. ...........''.'',;c:;cccodddk0KX0OOkxddddddo:...';;;;,,'
+..........',;:cclodxxkxdl:,..          ...  ......'....'..':c,..'.,c,,,.,cxkO00Okxdddddc'..';:;;;,,'
+..........',;;:cloodxkkkdol:'.    .  ...    ...... ......';c'.'...:;',;,'..,lxO00Oxxxo:'...,::;;,,,,
+...........',;;:clodxkOOkxdol;.   ..  ..        ... ....',::'.''.',.........'oxdxxxdl;...';::;;;,,''
+............',;:clodxkkOOOxddo;.   ......       ........',,',................';:clc;,...';::;;,,,'''
+............',;:cldxkkOkkkxdddo;.    .....      .........,'...........'''','.',,'''....,cc:;;,,'''..
+.............';:cldxxkkkkxddddxl,.   ....       .;c;'...................',;;cc;'...';clolc:;,,'''...
+............'';clodxkkkkkxddddddl'    ...       .:lc;'................. ....',,''';lxkxdlc:;,'''....
+........',,;:;coddxkOOOOOkxxddddd:.   ...     ..,''..................      . ..;cdkkkkxoc:;,'''.....
+......',;::cllodxkkOOOOOOkxxxddddc.  ...      ..,;,'................... ..   .':odO0Okdl:;,'''......
+.....',;:cloddxxkOOOOOOOkkxxdoooo;.  ..       .........................      .';cokOOxlc:;,''.......
+....,;:clodxxkkOOOkO0OOOOxdlcc;;,......      .';,.................         ...',:ldxxxdlc;,''.......
+...,:clodooxkkkO0OxO00OOxo:;;,. ........   .''.......... ..  ..           ..,,,;:codxxdlc:;,'.......
+'',;clodolokOkxkOkkO00Oko:;;;.  ..... ..   .,,........'. ..  ..  ..   ..........;:codocclc:,,'......
+              ___       __   __         ___               __        ___  __  
+        |  | |__  |    /  ` /  \\  |\\/| |__     |__|  /\\  /  ` |__/ |__  |__) 
+        |/\\| |___ |___ \\__, \\__/  |  | |___    |  | /~~\\ \\__, |  \\ |___ |  \\ 
+                                                                             
+         __   ___ ___     __                    __                           
+        / _` |__   |     |__) |  | |\\ | | |\\ | / _`                          
+        \\__> |___  |     |    |/\\| | \\| | | \\| \\__>     
+    
+
+
+    ";
+        print_banner(banner);
+        while loopize{
+            project_controls::save_projects(&projects, &config_path);
+            let active_project = get_active_project(&projects);
+            let current_information = format!("
+Active Project: {}, {}
+Project Status: {}
+Files Folder: {}
+Notes Folder: {}
+Boxname: {}
+Obsidian URI: {}
+
+for help enter help or ?. for information about a specific command enter help (command)
+
+
+", active_project.customer, active_project.project_name, active_project.stage, active_project.files_folder.display(), active_project.notes_folder.display(), active_project.boxname, "coming soon");
+            let prompt = format!("\n{}:{}\nCommand?", active_project.customer, active_project.project_name);
+            let command = get_user_input(&prompt);
+            match command.as_str(){
+                "exit" => loopize = false,
+                "menu" | "main menu" | "mm" => {let menu_thread_option = menu::main_menu(&mut projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.clone(), cracking_rig.clone(), rockyou.clone(), rule.clone(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.clone()); if menu_thread_option.is_some(){for thread in menu_thread_option.unwrap(){threads.push(thread);}}},
+                "print banner" | "banner" => print_banner(banner),
+                "clear" | "clear screen" | "cls" => {clear().unwrap(); print_banner(banner);},
+                "list threads" | "threads" | "lst" => println!("There are {} threads still running.", threads.len()),
+                "info" => println!("{}", current_information), 
+                _ =>  {let thread_option = run_command(command, &mut projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.clone(), cracking_rig.clone(), rockyou.clone(), rule.clone(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.clone()); if thread_option.is_some(){threads.push(thread_option.unwrap())}},
+            }
+        }
+        project_controls::save_projects(&projects, &config_path);
+        if get_user_input("do you want to stop all the boxes?").contains("y"){
+            box_controls::stop_all_boxes(&projects);
+        }
+    }
+    if threads.len() > 0{
+        println!("closing threads...");
+        println!("note this will hang until all threads have completed");
+        println!("please make sure to close all spawned programs such as cobalt strike and bloodhound.");
+        for thread in threads{
+            let _ = thread.join();
+        }
+    }
+}

pentest_tool/src/configuration.rs

@@ -0,0 +1,148 @@
+use std::path::PathBuf;
+use std::fs::read_to_string;
+use std::io::Write;
+use crate::{get_user_input, open_overwrite};
+
+
+
+#[allow(unused)]
+pub fn generate_tool_config(config_dir: &PathBuf){
+    let mut config_file_path = config_dir.clone();
+    println!("{}", config_file_path.display());
+    let mut current_config = String::new();
+    let current_config_read_res = read_to_string(&config_file_path);
+    if current_config_read_res.is_ok(){
+        current_config = current_config_read_res.unwrap();
+        println!("current configuration loaded!");
+    }
+    print!("{}", current_config);
+    let mut project_base_folder = PathBuf::new();
+    let mut project_base_notes = PathBuf::new();
+    let mut tools_folder = PathBuf::new();
+    let mut terminal_command = String::new();
+    let mut box_template = String::new();
+    let mut cracking_rig = String::new();
+    let mut rockyou = String::new();
+    let mut rule = String::new();
+    let mut upcoming_files = PathBuf::new();
+    let mut upcoming_notes = PathBuf::new();
+    let mut pass_spray_file = PathBuf::new();
+    let mut fingerprint = false;
+    let mut vault_name = String::new();
+    let settings: Vec<&str> = current_config.split("\n").collect();
+    for line in settings{
+        if line.len() > 1{
+            let setting_vec: Vec<&str> = line.split(":").collect();
+            match setting_vec[0]{
+                "Project_files" => project_base_folder.push(setting_vec[1].trim_end()),
+                "Project_notes" => project_base_notes.push(setting_vec[1].trim_end()),
+                "tools_folder" => tools_folder.push(setting_vec[1].trim_end()),
+                "upcoming_files" => upcoming_files.push(setting_vec[1].trim_end()),
+                "upcoming_notes" => upcoming_notes.push(setting_vec[1].trim_end()),
+                "box_template" => box_template = setting_vec[1].trim_end().to_owned(),
+                "terminal" => terminal_command = setting_vec[1].trim_end().to_owned(),
+                "cracking_rig" => cracking_rig = setting_vec[1].trim_end().to_owned(),
+                "rockyou_location" => rockyou = setting_vec[1].trim_end().to_owned(),
+                "rule_location" => rule = setting_vec[1].trim_end().to_owned(),
+                "pass_file"=> pass_spray_file.push(setting_vec[1]),
+                "fingerprint" => {if setting_vec[1].contains("y"){fingerprint = true}},
+                "vault_name" => vault_name = setting_vec[1].trim_end().to_owned(),
+                _ => println!("error unknown setting: {}", setting_vec[0])
+            }
+        }
+    }
+    println!("1.) recreate entire configuration");
+    println!("2.) modify one setting");
+    if get_user_input("Selection?").contains("2"){
+        print!("
+1 .) project_files
+2 .) project_notes
+3 .) tools_folder
+4 .) upcoming_files
+5 .) upcoming_notes
+6 .) box_template
+7 .) terminal
+8 .) cracking_rig
+9 .) rockyou_location
+10.) rule_location
+11.) pass_file
+12.) fingerprint
+13.) vault_name 
+");
+        match get_user_input("which setting would you like to modify?").as_str(){
+            "1" => {project_base_folder.clear(); project_base_folder.push(get_user_input("full path to the base project files folder?"));},
+            "2" => {project_base_notes.clear(); project_base_notes.push(get_user_input("full path to the base project notes folder"));},
+            "3" => {tools_folder.clear(); tools_folder.push(get_user_input("full path to your custom tools folder?"));},
+            "4" => {upcoming_files.clear(); upcoming_files.push(get_user_input("full path to your upcoming projects folder"));},
+            "5" => {upcoming_notes.clear(); upcoming_notes.push(get_user_input("full path to your upcoming project nots folder"));},
+            "6" => {box_template = get_user_input("name of your distrobox template?")},
+            "7" => {terminal_command = get_user_input("comand to run your terminal while executing a specific command, ex: konsole -e ")},
+            "8" => {cracking_rig = get_user_input("username and address of your personal cracking rig, example pyro@crackingrig or pyro@192.168.1.10?")},
+            "9" => {rockyou = get_user_input("location of rockyou.txt on your cracking rig?")},
+            "10" => {rule = get_user_input("location of the one rule on your crakcing rig?")},
+            "11" => {pass_spray_file.clear(); pass_spray_file.push(get_user_input("location of your password spray list file?"));},
+            "12" => {fingerprint = get_user_input("will you be using fingerprint authentication in your distroboxes?").to_lowercase().contains("y")},
+            "13" => {vault_name = get_user_input("obsidian vault name?")},
+            _ => {println!("unknown selection, please try again...");}
+        };
+    }
+    else{
+    project_base_folder = PathBuf::from(get_user_input("path to store your active projects?"));
+    project_base_notes = PathBuf::from(get_user_input("path to store your active project notes?"));
+    upcoming_files = PathBuf::from(get_user_input("path to store your upcomming projects?"));
+    upcoming_notes = PathBuf::from(get_user_input("path to store your upcomming project notes?"));
+    tools_folder = PathBuf::from(get_user_input("path where you store your custom tools (like from github and places)?"));
+    box_template = get_user_input("Name of the distrobox you want to use as a template?");
+    cracking_rig = String::from("nobody@nothing");
+    rockyou = String::from("n/a");
+    rule = String::from("n/a");
+    let cracking_rig_response = get_user_input("do you have a separate machine to crack passwords on? (not the ambush cracking rig)");
+    if cracking_rig_response.to_lowercase().contains("y"){
+        let rig_ip = get_user_input("ip address or hostname of your cracking rig?");
+        let rig_user = get_user_input("username to log into your cracking rig with?");
+        rockyou = get_user_input("location of rockyou.txt on the cracking rig?");
+        rule = get_user_input("location of one rule to rule them all on the cracking rig?");
+        cracking_rig = format!("{}@{}", rig_user, rig_ip);
+    }
+    else{
+        println!("ok free loader");
+    }
+    fingerprint = get_user_input("will you be using fingerprint authentication for your distroboxes?").to_lowercase().contains("y");
+    terminal_command = get_user_input("command to launch your terminal, while executing a command, for example konsel -e ");
+    vault_name = get_user_input("the name of the vault you're going to use?");
+    }
+let new_config = format!("
+Project_files:{}
+Project_notes:{}
+tools_folder:{}
+upcoming_files:{}
+upcoming_notes:{}
+box_template:{}
+terminal:{}
+cracking_rig:{}
+rockyou_location:{}
+rule_location:{}
+pass_file:{}
+fingerprint:{}
+vault_name:{}    
+",project_base_folder.display(), project_base_notes.display(), tools_folder.display(), upcoming_files.display(), upcoming_notes.display(), box_template, terminal_command, cracking_rig, rockyou, rule, pass_spray_file.display(), fingerprint, vault_name); 
+    println!("this will be the new config that will be saved:\n");
+    println!("{}", new_config);
+    if get_user_input("is this ok?").to_lowercase().contains("y"){
+        let config_file_res = open_overwrite(&config_file_path);
+        if config_file_res.is_none(){
+            println!("failed to open config file in overwrite mode... quitting... nothing was saved.");
+            return;
+        }
+        let mut config_file = config_file_res.unwrap();
+        let write_res= write!(config_file, "{}", new_config);
+        if write_res.is_err(){
+            let error = write_res.err().unwrap();
+            println!("error writing config file!");
+            println!("{}", error);
+            println!("nothing was saved...");
+            return;
+        }
+        write_res.unwrap();
+    }
+}

pentest_tool/src/enumeration.rs

@@ -0,0 +1,362 @@
+use std::fs::{read_to_string, remove_file, OpenOptions};
+use std::process::Command;
+use std::thread::JoinHandle;
+use std::thread::{spawn, sleep};
+use std::io::Write;
+use std::time::Duration;
+use dns_lookup::lookup_host;
+use crate::get_user_input;
+use crate::Project;
+use crate::open_append;
+
+#[allow(unused)]
+pub fn run_dns_enumeration(project: &Project, given_domains: Option<&Vec<String>>, standalone: bool) -> Option<JoinHandle<()>>{
+    let notes_folder = project.notes_folder.clone();
+    let mut enumeration = notes_folder.clone();
+    enumeration.push("enumeration.md");
+    let enumeration_file_res = open_append(&enumeration);
+    if enumeration_file_res.is_none(){
+        println!("error opening enumeration_file!");
+        println!("try creating it manually.");
+        return None;
+    }
+    let mut enumeration_file = enumeration_file_res.unwrap();
+    let mut domaind = Vec::new();
+    if given_domains.is_none(){
+        loop{
+            let domain = get_user_input("domain to add? enter DONE in all caps when you're finsihed");
+            match domain.as_str(){
+                "DONE" => break,
+                _ => domaind.push(domain),
+            }
+        }
+    }
+    else{
+        for domain in given_domains.unwrap(){
+            domaind.push(domain.to_owned());
+        }
+    }
+    let working_project = project.clone();
+    let dns_handle = spawn(move || {
+        for domain in &domaind{
+            let output_res = Command::new("distrobox")
+                                                   .arg("enter")
+                                                   .arg("--root")
+                                                   .arg(working_project.boxname.to_owned())
+                                                   .arg("--")
+                                                   .arg("dnsrecon")
+                                                   .arg("-d")
+                                                   .arg(domain)
+                                                   .arg("-c")
+                                                   .arg("dns_temp.csv")
+                                                   .output();
+            if output_res.is_err(){
+                let error = output_res.err().unwrap();
+                println!("From DNS Enumeration Thread: error running dnsrecon in the project's distrobox!");
+                println!("{}", error);
+                return;
+            }
+            println!("sleping for 10 seconds to allow for sudo password input.");
+            sleep(Duration::from_secs(10));
+            let output_string_res = read_to_string("dns_temp.csv");
+            if output_string_res.is_err(){
+                let error = output_string_res.err().unwrap();
+                println!("From DNS Enumeration Thread: error reading output data!");
+                println!("{}", error);
+                return;
+            }
+            let output_string = output_string_res.unwrap();
+            let lines: Vec<&str> = output_string.split("\n").collect();
+            let mut out_data = String::new();
+            if standalone{
+                out_data.push_str("# DNS Enumeration\n");
+                out_data.push_str("## DNS Records\n");
+            }
+            let mut data_vec = Vec::new();
+            let mut first_line = true;
+            for line in lines{
+                if first_line == true{
+                    first_line = false;
+                }
+                else{
+                    if line.len() > 1{
+                        let words: Vec<&str> = line.split(",").collect();
+                        let domain_name = words[2].to_owned();
+                        let domain_type = words[1].to_owned();
+                        let mut data = String::new();
+                        if words[3].len() > 2{
+                            data = words[3].to_owned(); 
+                        }
+                        else{
+                            data = words[6].to_owned();
+                        }
+                        let data_line = format!("| {} | {} | {} |", domain_name, domain_type, data);
+                        data_vec.push(data_line);
+                    }
+                }
+            }
+            let domain_header = format!("#### {}\n", domain);
+            out_data.push_str(&domain_header);
+            if standalone{
+                out_data.push_str("#### DNS Records\n");
+            }
+            out_data.push_str("| Domain name | Record type | data |\n");
+            out_data.push_str("| ----------- | ----------- | ---- |\n");
+            for thang in data_vec{
+                let out_line = format!("{}\n", thang);
+                out_data.push_str(&out_line);
+            }
+            if standalone{
+                out_data.push_str("\n---\n");
+            }
+            println!("From DNS Enumeration Thread: Finished gathering data for {} writing to notes...", domain);
+            write!(enumeration_file, "{}", &out_data).unwrap();
+            let remove_res = remove_file("dns_temp.csv");
+            if remove_res.is_err(){
+                println!("From DNS Enumeration Thread: error removing temporay data file!");
+                println!("From DNS Enumeration Thread: please manually delete dns_temp.csv");
+            }
+        }
+    });
+    return Some(dns_handle);
+}
+
+#[allow(unused)]
+pub fn bruteforce_subs(project: &Project, given_domains: Option<&Vec<String>>, given_wordlist: Option<String>, standalone: bool) -> Option<JoinHandle<()>>{
+    let mut enumeration_path = project.notes_folder.clone();
+    enumeration_path.push("enumeration.md");
+    let enumeration_file_res = OpenOptions::new().append(true).create(true).open(enumeration_path);
+    if enumeration_file_res.is_err(){
+        let error = enumeration_file_res.err().unwrap();
+        println!("error opening enumeration notes file!");
+        println!("{}", error);
+        return None;
+    }
+    let mut enumeration_file = enumeration_file_res.unwrap();
+    let mut domains = Vec::new();
+    if given_domains.is_none(){
+        loop{
+            let domain = get_user_input("Domain to add? Enter DONE in all caps when done.");
+            if domain == "DONE".to_owned(){
+                break;
+            }
+            else{
+                domains.push(domain);
+            }
+        }
+    }
+    else{
+        for domain in given_domains.unwrap(){
+            domains.push(domain.to_owned());
+        };
+    }
+    let mut wordlist = String::new();
+    if given_wordlist.is_none(){
+        wordlist = get_user_input("path to wordlist?");
+    }
+    else{
+        wordlist = given_wordlist.unwrap();
+    }
+    let working_project = project.clone();
+    let mut out_data = String::new();
+    if standalone{
+        out_data.push_str("# DNS Enumeration\n");
+        out_data.push_str("## Subdomain Enumeration\n");
+    }
+    let gobuster_thread = spawn( move ||{
+        for domain in domains{
+            if standalone{
+                out_data.push_str(format!("#### {}\n", &domain).as_str());
+            }
+            let gobuster_cmd_res = Command::new("distrobox")
+                                                            .arg("enter")
+                                                            .arg("--root")
+                                                            .arg(working_project.boxname.to_owned())
+                                                            .arg("--")
+                                                            .arg("gobuster")
+                                                            .arg("dns")
+                                                            .arg("-d")
+                                                            .arg(&domain)
+                                                            .arg("-w")
+                                                            .arg(wordlist.to_owned())
+                                                            .output();
+            if gobuster_cmd_res.is_err(){
+                let error = gobuster_cmd_res.err().unwrap();
+                println!("From gobuster thread: Error running gobuster command!");
+                println!("{}", error);
+                return;
+            }
+            println!("sleeping for 10 seconds to allow for sudo password input.");
+            sleep(Duration::from_secs(10));
+            let gobuser_output = gobuster_cmd_res.unwrap().stdout;
+            println!("From Gobuster Thread: Sudomain enumeration Done!");
+            let gobuster_string = String::from_utf8_lossy(&gobuser_output);
+            let mut domain_names = Vec::new();
+            let lines: Vec<&str> = gobuster_string.split("\n").collect();
+            for line in lines{
+                if line.contains("Found:"){
+                    let domain = line.split_whitespace().collect::<Vec<&str>>()[1];
+                    domain_names.push(domain.to_owned());
+                }
+            }
+            out_data.push_str("\n| domain name | ips |\n");
+            out_data.push_str("| ----------- | --- |\n");
+            for name in domain_names{
+                let ips = lookup_host(&name);
+                if ips.is_ok(){
+                    let mut ip_string = String::new();
+                    for ip in ips.unwrap(){
+                        ip_string = format!("{},{}", ip, ip_string);
+                    }
+                    out_data.push_str(format!("| {} | {} |\n", name, ip_string).as_str());
+                }
+            }
+        }
+        if standalone{
+            out_data.push_str("\n---\n");
+        }
+        let write_res = write!(enumeration_file, "{}", out_data);
+        if write_res.is_err(){
+            let error = write_res.err().unwrap();
+            println!("FROM Gobuster Thread: error writing notes!");
+            println!("{}", error);
+            return;
+        }
+        write_res.unwrap();
+    });
+    return Some(gobuster_thread);
+}
+
+pub fn dns_squatting(project: &Project, given_domains: Option<&Vec<String>>, standalone: bool) -> Option<JoinHandle<()>>{
+    let mut enumeration_notes = project.notes_folder.clone();
+    enumeration_notes.push("enumeration.md");
+    let open_enumeration_notes_res = OpenOptions::new().append(true).create(true).open(enumeration_notes);
+    if open_enumeration_notes_res.is_err(){
+        let error = open_enumeration_notes_res.err().unwrap();
+        println!("Error opening enumeration notes");
+        println!("{}", error);
+        return None;
+    }
+    let mut enumeration_file = open_enumeration_notes_res.unwrap();
+    let mut domains = Vec::new();
+    if given_domains.is_none(){
+        loop{
+            let domain = get_user_input("Domain to add? enter DONE in all caps when you're finished");
+            if domain == "DONE"{
+                break;
+            }
+            else{
+                domains.push(domain);
+            }
+        }
+    }
+    else{
+        domains = given_domains.unwrap().to_owned();
+    }
+    let working_project = project.clone();
+    let squatting_thread = spawn(move || {
+        let write_res = write!(enumeration_file, "### Domain Squatting\n");
+        if write_res.is_err(){
+            let error = write_res.err().unwrap();
+            println!("error writing to enumeration notes file!");
+            println!("{}", error);
+            return;
+        }
+        write_res.unwrap();
+        for domain in domains{
+            if standalone{
+                write!(enumeration_file, "#### {}\n", domain).unwrap();
+            }
+            write!(enumeration_file, "\n| type | domain name | ns servers |\n").unwrap();
+            write!(enumeration_file, "| ---- | ----------- | ---------- |\n").unwrap();
+            let twist_output = Command::new("distrobox")
+                                                    .arg("enter")
+                                                    .arg("--root")
+                                                    .arg(working_project.boxname.to_owned())
+                                                    .arg("--")
+                                                    .arg("dnstwist")
+                                                    .arg("-r")
+                                                    .arg(domain)
+                                                    .output();
+            if twist_output.is_err(){
+                let error = twist_output.err().unwrap();
+                println!("From DNSTwist thread: Error running dnstwist command!");
+                println!("{}", error);
+                return;
+            }
+            println!("sleeping for 10 seconds to allow for sudo password input.");
+            sleep(Duration::from_secs(10));
+            let twist_output_vec = twist_output.unwrap().stdout;
+            let output_string = String::from_utf8_lossy(&twist_output_vec);
+            let output_lines = output_string.split("\n");
+            for line in output_lines{
+                if line.len() > 0{
+                    let words: Vec<&str> = line.split_whitespace().collect();
+                    let twist_type = words[0];
+                    let name = words[1];
+                    let ns_servers = words[2..].join(" ");
+                    write!(enumeration_file, "| {} | {} | {} |\n", twist_type, name, ns_servers).unwrap();
+                }
+            }
+        }
+    });
+    return Some(squatting_thread);
+}
+
+pub fn do_all_dns_enumeration(project: &Project) -> Option<JoinHandle<()>>{
+    let mut enumeration_path = project.notes_folder.clone();
+    enumeration_path.push("enumeration.md");
+    let enumeration_file_res = OpenOptions::new().append(true).create(true).open(enumeration_path);
+    if enumeration_file_res.is_err(){
+        let error = enumeration_file_res.err().unwrap();
+        println!("error opening enumeration notes file!");
+        println!("{}", error);
+        return None;
+    }
+    let mut enumeration_file = enumeration_file_res.unwrap();
+    let mut domains = Vec::new();
+    loop{
+        let domain = get_user_input("Domain to add? enter DONE in all caps when you're finished");
+        if domain == "DONE"{
+            break;
+        }
+        else{
+            domains.push(domain);
+        }
+    }
+    let wordlist = get_user_input("path to wordlist for sub domain bruteforcing?");
+    let working_project = project.clone();
+    let all_dns_handle = spawn(move ||{
+        let mut write_success = true;
+        let write_res = write!(enumeration_file, "# DNS Enumeration\n");
+        if write_res.is_err(){
+            let error = write_res.err().unwrap();
+            println!("From All DNS thread: Error writing notes file!");
+            println!("{}", error);
+            write_success = false;
+        }
+        if write_success{
+            for domain in &domains{
+                let thread_domain = vec![domain.to_owned()];
+                write!(enumeration_file, "## {}\n", &domain).unwrap();
+                write!(enumeration_file, "### DNS Records\n").unwrap();
+                let dns_enum_thread = run_dns_enumeration(&working_project, Some(&thread_domain), false);
+                if dns_enum_thread.is_some(){
+                    let _ = dns_enum_thread.unwrap().join();
+                }
+                write!(enumeration_file, "### Subdomain Enumeration\n").unwrap();
+                let gobuster_thread = bruteforce_subs(&working_project, Some(&thread_domain), Some(wordlist.to_owned()), false);
+                if gobuster_thread.is_some(){
+                    let _ = gobuster_thread.unwrap().join();
+                }
+                write!(enumeration_file, "### Domain Squatting\n").unwrap();
+                let twist_thread = dns_squatting(&working_project, Some(&thread_domain), false);
+                if twist_thread.is_some(){
+                    let _ = twist_thread.unwrap().join();
+                }
+                write!(enumeration_file, "\n---\n").unwrap();
+            }
+        }
+    });
+    return Some(all_dns_handle);
+}

pentest_tool/src/install.rs

@@ -1,130 +1,274 @@
 use std::collections::HashMap;
-use std::fs;
+use std::fs::{File, create_dir_all, remove_dir_all};
+use std::io::Read;
 use std::io::Write;
-use std::path::PathBuf;
+use std::io::copy;
-use std::process;
+use std::process::Command;
-
+use std::time::Duration;
+use reqwest::blocking::get;
+use std::{path::Path, path::PathBuf};
+use std::thread;
+use std::process::exit;
 
+use crate::get_user_input;
 
+#[allow(unused)]
 pub fn install(config_path: &PathBuf){
     let mut _terminal_commands = HashMap::from([
-        ("konsole", "konsole -e !!!"),
+        ("kde", "konsole -e !!!"),
+        ("plasma", "konsole -e !!!"),
         ("gnome", "gnome-terminal -- bash !!!"),
         ("xfce", "xfce4-terminal --execute '!!!'"),
         ("alacritty", "alacritty -e !!!"),
     ]);
+    let mut folder_creation = true;
+    let current_projects = PathBuf::from(get_user_input("path to store your active projects?"));
+    let current_notes = PathBuf::from(get_user_input("path to store your active project notes?"));
+    let upcoming_projects = PathBuf::from(get_user_input("path to store your upcomming projects?"));
+    let upcoming_notes = PathBuf::from(get_user_input("path to store your upcomming project notes?"));
+    let tools = PathBuf::from(get_user_input("path where you store your custom tools (like from github and places)?"));
+    let mut vault_name = String::new();
+    let folders = vec![&current_projects, &current_notes, &upcoming_projects, &upcoming_notes, &tools];
     let mut config_folder_path: PathBuf = config_path.clone();
     config_folder_path.pop();
+    let mut password_path = config_folder_path.clone();
+    password_path.push("password_spray.md");
     let mut projects_conf_path = config_folder_path.clone();
+    let mut bell_file_path = config_folder_path.clone();
     let del_on_fail = config_folder_path.clone();
     projects_conf_path.push("projects.conf");
-    fs::create_dir_all(&config_folder_path).expect("error creating config dir");
+    create_dir_all(&config_folder_path).expect("error creating config dir");
-    let mut config_file = fs::File::create(config_path).expect("error creating file");
+    bell_file_path.push("bell.mp3");
-    let mut projects_conf_file = fs::File::create(projects_conf_path).expect("error creating projects config file");
+    let bell_sound_url = "https://github.com/Pyro57000/pentest_tool/raw/refs/heads/main/resources/bell.mp3";
-    projects_conf_file.write_all(b"customer:name:notes:files:active:box_name\n").expect("error writing default project info");
+    let response = get(bell_sound_url).unwrap();
-    let mut terminal_response = String::new();
+    let response_length = response.content_length().unwrap_or(0);
-    let mut notes_response = String::new();
+    let mut bell_file = File::create(bell_file_path).unwrap();
-    let mut files_response = String::new();
+    copy(&mut response.take(response_length), &mut bell_file).unwrap();
-    let mut tools_response = String::new();
+    println!("bell notification tone sucessfully downloaded!");
-    let mut template_name = String::new();
+    println!("creating new folders if needed...");
-    let mut have_template = String::new();
+    for folder in &folders{
-    println!("terminal you use? (example: konsole, xfce, gnome, etc)");
+        if !Path::exists(folder){
-    std::io::stdin().read_line(&mut terminal_response).unwrap();
+            let create_res = create_dir_all(folder);
-    let mut _terminal_command = String::new();
+            if create_res.is_err(){
-    if terminal_response.contains("konsole"){
+                let error = create_res.err().unwrap();
-        let mut response_buffer = String::new();
+                    println!("error creating folder {}", folder.display());
-        println!("do you already have a custom profile setup?");
+                    println!("{}", error);
-        println!("this is pretty specific to pyro's setup, I do some special sauce for my konsole stuff");
+                    println!("you'll need to create this manually after the install is done!");
-        std::io::stdin().read_line(&mut response_buffer).unwrap();
+                    folder_creation = false;
-        if response_buffer.to_lowercase().contains("y"){
+                }
-            let mut profile = String::new();
+            }
-            println!("konsole profile name?");
+        }
-            std::io::stdin().read_line(&mut profile).unwrap();
+    let tool_volume = format!("{}:/tools:rw", &tools.display()); 
-            _terminal_command = format!("konsole --profile {}", profile);
+    println!("sweet, now let's get some distrobox stuff out of the way!");
+    let distrobox_run_res = Command::new("distrobox").arg("list").arg("--root").output();
+    if distrobox_run_res.is_err(){
+        let error = distrobox_run_res.err().unwrap();
+        println!("Distrobox file was not found!");
+        println!("This usually means that distrobox is not installed.");
+        println!("please install distrobox, if you're on kali run `sudo apt install distrobox`");
+        println!("{}", error);
+        println!("this project heavily relies on distrobox, as its primarily a distrobox mangement tool.");
+        println!("cleaning up up configuration folders, please install distrobox and re-run this program.");
+        let cleanup = remove_dir_all(&del_on_fail);
+        if cleanup.is_err(){
+            println!("error cleaning up configuration folder!");
+            println!("please manually delete {} before re-running this install.", &del_on_fail.display());
+            exit(1);
+        }
+        cleanup.unwrap();
+        exit(1);
+    }
+    let distrobox_output = distrobox_run_res.unwrap().stdout;
+    let distrobox_string = String::from_utf8_lossy(&distrobox_output);
+    let mut distrobox_names = Vec::new();
+    let distrobox_output_lines: Vec<&str> = distrobox_string.split("\n").collect();
+    for line in distrobox_output_lines{
+        if line.len() > 1{
+            let name = line.split(" | ").collect::<Vec<&str>>()[1];
+            distrobox_names.push(name.to_owned());
+        }
+    }
+    let mut have_box = false;
+    for name in distrobox_names{
+        if !name.contains("NAME"){
+            have_box = true;
+            println!("{}", name);
+        }
+    }
+    let mut template_box_name = String::new();
+    if have_box{
+        template_box_name = get_user_input("Name of the distrobox you want to use as a template?");
+    }
+    else{
+        println!("no exiting distroboxes created!");
+        println!("let's make one!");
+        template_box_name = get_user_input("name of the distrobox you want to create?");
+        println!("review images listed here: https://distrobox.it/compatibility/#containers-distros");
+        let box_image = get_user_input("image you want to base the template box on?");
+        let template_box_create_res = Command::new("distrobox")
+                                                            .arg("create")
+                                                            .arg("--root")
+                                                            .arg("--image")
+                                                            .arg(&box_image)
+                                                            .arg("--name")
+                                                            .arg(&template_box_name)
+                                                            .arg("--init")
+                                                            .arg("--volume")
+                                                            .arg(&tool_volume)
+                                                            .status();
+        if template_box_create_res.is_err(){
+            let error = template_box_create_res.err().unwrap();
+            println!("error creating template box!");
+            println!("{}", error);
+            println!("\n\n\nplese create it yourself with the following command.");
+            println!("\n\ndistrobox create --root --image {}, --name {}, --init --volume {}", &box_image, &template_box_name, &tool_volume);
+        }
+        else{
+            template_box_create_res.unwrap();
+        }
+    }
+    let mut cracking_rig = String::from("nobody@nothing");
+    let mut rockyou = String::from("n/a");
+    let mut rule = String::from("n/a");
+    let cracking_rig_response = get_user_input("do you have a separate machine to crack passwords on? (not the ambush cracking rig)");
+    if cracking_rig_response.to_lowercase().contains("y"){
+        let rig_ip = get_user_input("ip address or hostname of your cracking rig?");
+        let rig_user = get_user_input("username to log into your cracking rig with?");
+        rockyou = get_user_input("location of rockyou.txt on the cracking rig?");
+        rule = get_user_input("location of one rule to rule them all on the cracking rig?");
+        cracking_rig = format!("{}@{}", rig_user, rig_ip);
+    }
+    else{
+        println!("ok free loader");
+    }
+    let fingerprint = get_user_input("will you be using fingerprint authentication for your distroboxes?").to_lowercase();
+    let mut terminal = String::new();
+    for desktop in _terminal_commands.keys(){
+        println!("{}", desktop);
+    }
+    let desktop_response = get_user_input("do you use any of these desktops?").to_lowercase();
+    if desktop_response.contains("y"){
+        let default_response = get_user_input("do you use the default terminal for your desktop?").to_lowercase();
+        if default_response.contains("y"){
+            let de = get_user_input("which desktop do you use?");
+            terminal = _terminal_commands[&de.as_str()].to_owned();
         }
     }
     else{
-        _terminal_command = _terminal_commands[terminal_response.trim_end()].to_owned();
+        println!("OK, then please enter the command you'd use to launch a new terminal and run a command inside of it, replacing the command with three !s");
+        println!("for example for konsole you'd enter");
+        println!("konsole -e !!!");
+        terminal = get_user_input("");
     }
-    println!("path to save project notes?");
+    if terminal.contains("konsole"){
-    std::io::stdin().read_line(&mut notes_response).unwrap();
+        println!("do you use a specific profile for your attack boxes?");
-    println!("path to save project files?");
+        println!("this is pretty specific to Pyro's setup, so you probably don't");
-    std::io::stdin().read_line(&mut files_response).unwrap();
+        if get_user_input("").to_lowercase().contains("y"){
-    println!("path to folder with your custom tools?");
+            let profile_name = get_user_input("what is the name of your profile?");
-    std::io::stdin().read_line(&mut tools_response).unwrap();
+            terminal = format!("konsole --profile {}", profile_name);
-    print!("
-This tool is mainly to handle distrobox creation and usage.
-It's expecting you to have a distrobox that you will use as a template.
-Do you have a distrobox set up to function as your template for all new projects?
-");
-    std::io::stdin().read_line(&mut have_template).unwrap();
-    if have_template.contains("n"){
-        println!("please set up a distrobox with root as a template and re-run this tool");
-        println!("example distrobox setup command:");
-        println!("distrobox create --root --image archlinux --name template");
-        println!("then enter that distrobox and install all the tools you want and do what ever setup you need");
-        println!("and re-run this tool.");
-        process::Command::new("rm").arg(del_on_fail).spawn().expect("ERROR deleting config folder, please manually clean up");
-        std::process::exit(1);
         }
-    let _list = process::Command::new("distrobox").arg("list").arg("--root").status();
+    }
-    println!("distrobox template name?");
+    println!("many of the fuctions of this tool assume you're using obsidian or some other markdown editor for note taking");
-    std::io::stdin().read_line(&mut template_name).unwrap();
+    let obsidian_used = get_user_input("do you use obsidian for your notes?");
-    let config_string = format!("Project_files:{}\nProject_notes:{}\ntools_folder:{}\nterminal:{}", files_response.trim_end(), notes_response.trim_end(), tools_response.trim_end(), _terminal_command.trim_end());
+    if obsidian_used.to_lowercase().contains("y"){
-    config_file.write_all(config_string.as_bytes()).expect("error writing to config file");
+        vault_name = get_user_input("the name of the vault you're going to use?");
-    let default_projectline = format!("default:default:{}:{}:yes:{}", &notes_response.trim_end(), &files_response.trim_end(), &template_name.trim_end());
+    }
-    projects_conf_file.write_all(default_projectline.as_bytes()).expect("error writing default project line");
+    let configuration_string = format!("
-    println!("config file generated and saved to {}\n", config_path.display());
+Project_files:{}
-    println!("please rerun the program");
+Project_notes:{}
-    
+tools_folder:{}
-
+upcoming_files:{}
-    let config_path = &config_folder_path.clone();
+upcoming_notes:{}
-    let mut install_path = config_path.clone();
+box_template:{}
-    let mut password_spray_template_path = install_path.clone();
+terminal:{}
-    install_path.push("new_projects.conf");
+cracking_rig:{}
-    password_spray_template_path.push("passwordspray.md");
+rockyou_location:{}
-    let password_spray_template_path = install_path.clone();
+rule_location:{}
-    let mut conf_file = fs::File::create(install_path).expect("error creating config file");
+pass_file:{}
-    let mut project_folder_path = String::new();
+fingerprint:{}
-    let mut porject_note_path = String::new();
+vault_name:{}"
-    println!("path to the project folders directory?");
+, &current_projects.display(), &current_notes.display(), &tools.display(), &upcoming_projects.display(), &upcoming_notes.display(), &template_box_name, &terminal, cracking_rig, rockyou, rule, &password_path.display(), fingerprint, &vault_name);
-    std::io::stdin().read_line(&mut project_folder_path).expect("error reading project folder from stdin");
+    println!("cool everything, all folders and settings have been entered, now let's save this to a config file...");
-    println!("path to the project notes directory");
+    thread::sleep(Duration::from_secs(3));
-    std::io::stdin().read_line(&mut porject_note_path).expect("error reading project note folder form stdin");
+    let mut config_file_res = File::create_new(config_path);
-    write!(conf_file, "project_folder_path:{}
+    if config_file_res.is_err(){
-project_notes_path:{}
+        println!("ooof error creating configuration file...");
-", project_folder_path.trim_end(), porject_note_path.trim_end()).expect("error writing config file");
+        println!("try creating it manually.");
-    let mut passpray_file =  fs::File::create(password_spray_template_path).expect("error creating passwordspray file");
+        println!("copy the following configuration and save it to {}", config_path.display());
-    write!(passpray_file, "
+        print!("{}\n", configuration_string);
-- [ ] useraspass
+        exit(1);
-- [ ] Seasonyear!
+    }
-- [ ] Service123!
+    let mut config_file = config_file_res.unwrap();
-- [ ] admin
+    let config_write_res =write!(config_file, "{}", &configuration_string);
-- [ ] Admin
+    if config_write_res.is_err(){
-- [ ] Admin123!
+        println!("error writing configuration string to the configuration file!");
-- [ ] admin123
+        println!(" try creating it manually.");
-- [ ] admin1
+        println!("copy the following configuration and save it to {}", &configuration_string);
-- [ ] 1234567
+        exit(1);
-- [ ] Seasonyear
+    }
-- [ ] seasonyear!
+    config_write_res.unwrap();
-- [ ] seasonyear
+    println!("nice the configuration file was written correctly!");
-- [ ] COMPANYYEAR!
+    println!("creating project configuration file, and poplulating it with the default project...");
-- [ ] COMPANYYEAR
+    let project_conf_res = File::create_new(&projects_conf_path);
-- [ ] November2024!
+    if project_conf_res.is_err(){
-- [ ] September2024!
+        let error = project_conf_res.err().unwrap();
-- [ ] October2024!
+        println!("ooof error creating the projects configuration file.");
-- [ ] COMPANYfoundingyear!
+        println!("try creating it manually!");
-- [ ] COMPANYfoundingyear
+        println!("copy the following configuration and save it to {}", &projects_conf_path.display());
-- [ ] COMPANYstreetnumber!
+        println!("customer:name:notes:files:active:time:box_name:stage");
-- [ ] COMPANYstreetnumber
+        println!("default:default:{}:{}:yes:{}:current", &current_notes.display(), &current_projects.display(), &template_box_name);
-- [ ] Password
+        exit(1);
-- [ ] P@ssw0rd
+    }
-- [ ] Password1!
+    let mut project_conf_file = project_conf_res.unwrap();
-- [ ] Password123!
+    let project_write_res = write!(project_conf_file, "customer:name:notes:files:active:time:box_name:stage\ndefault:default:{}:{}:yes:{}:current", &current_notes.display(), &current_projects.display(), &template_box_name);
-- [ ] Passwordyear!
+    if project_write_res.is_err(){
-- [ ] P@55w0rd
+        println!("error writing project config file.");
-- [ ] Service
+        exit(1);
-- [ ] Service!
+    }
-- [ ] Serviceyear!").expect("error writing password spray template");
+    project_write_res.unwrap();
-    std::process::exit(0);
+    let password_file_res = File::create_new(password_path);
+    if password_file_res.is_err(){
+        println!("error creating password spray file");
+        exit(1);
+    }
+    let mut password_file = password_file_res.unwrap();
+    let password_write_res = write!(password_file, "
+    - [ ] useraspass
+    - [ ] Seasonyear!
+    - [ ] Service123!
+    - [ ] admin
+    - [ ] Admin
+    - [ ] Admin123!
+    - [ ] admin123
+    - [ ] admin1
+    - [ ] 1234567
+    - [ ] Seasonyear
+    - [ ] seasonyear!
+    - [ ] seasonyear
+    - [ ] COMPANYYEAR!
+    - [ ] COMPANYYEAR
+    - [ ] November2024!
+    - [ ] September2024!
+    - [ ] October2024!
+    - [ ] COMPANYfoundingyear!
+    - [ ] COMPANYfoundingyear
+    - [ ] COMPANYstreetnumber!
+    - [ ] COMPANYstreetnumber
+    - [ ] Password
+    - [ ] P@ssw0rd
+    - [ ] Password1!
+    - [ ] Password123!
+    - [ ] Passwordyear!
+    - [ ] P@55w0rd
+    - [ ] Service
+    - [ ] Service!
+    - [ ] Serviceyear! 
+    ");
+    if password_write_res.is_err(){
+        println!("error writing password file");
+        exit(1);
+    }
+    password_write_res.unwrap();
+    println!("install completed successfully!");
+    println!("re-run this to launch!");
 }

pentest_tool/src/main.rs

@@ -1,6 +1,7 @@
-use std::path::PathBuf;
+use std::{io::stdin, path::PathBuf, process::Command};
 use directories::UserDirs;
-use std::fs;
+use std::process::exit;
+use std::fs::{self, File};
 
 #[derive(Clone)]
 pub struct Project{
@@ -10,15 +11,66 @@ pub struct Project{
     pub files_folder: PathBuf,
     pub active: bool,
     pub boxname: String,
+    pub stage: String,
     pub id: i32,
 }
 
 mod install;
-mod menu;
 mod project_controls;
 mod box_controls;
 mod info_controls;
 mod start_pentest;
+mod cli;
+mod menu;
+mod portscan_controls;
+mod victim_commands;
+mod enumeration;
+mod tool_controls;
+mod configuration;
+
+pub fn open_overwrite(path: &PathBuf) -> Option<File>{
+    let file_create_res = fs::OpenOptions::new().create(true).write(true).open(path);
+    if file_create_res.is_err(){
+        let error = file_create_res.err().unwrap();
+        println!("error opening {} file!", path.display());
+        println!("{}", error);
+        return None;
+    }
+    else {
+        let file = file_create_res.unwrap();
+        return Some(file);
+    }
+}
+
+pub fn open_append(path: &PathBuf) -> Option<File>{
+    let file_create_res = fs::OpenOptions::new().create(true).append(true).open(path);
+    if file_create_res.is_err(){
+        let error = file_create_res.err().unwrap();
+        println!("error opening {} file!", path.display());
+        println!("{}", error);
+        return None;
+    }
+    else {
+        let file = file_create_res.unwrap();
+        return Some(file);
+    }
+}
+
+pub fn get_user_input(prompt: &str) -> String{
+    let mut response = String::new();
+    loop{
+        println!("{}", prompt);
+        let result = stdin().read_line(&mut response);
+        if result.is_err(){
+            println!("we need input here dummy! try again...");
+        }
+        else{
+            result.unwrap();
+            break;
+        }
+    }
+    return response.trim_end().to_owned();
+}
 
 
 fn main() {
@@ -42,8 +94,23 @@ fn main() {
     ⠀⠀⠀⠀⢿⡉⠳⡟⣸⠃⠀⠀⠀⠘⢷⣌⠉⠀⣿⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀
     ⠀⠀⠀⠀⠀⠙⢦⣴⠏⠀⠀⠀⠀⠀⠀⠉⠳⠶⠏⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀
     ");
-    let user_dirs = UserDirs::new().expect("error getting user directories");
+    let mut config_path = PathBuf::new();
-    let mut config_path = user_dirs.home_dir().to_path_buf();
+    let user_dirs_res = UserDirs::new();
+    if user_dirs_res.is_none(){
+        println!("oof couldn't get the config directory the right way, falling back to the stpuid way...");
+        let get_home_res = Command::new("echo").arg("$HOME").output();
+        if get_home_res.is_err(){
+            println!("ooof the stupid way didn't work... this is a bruh moment, gotta fix your environment variabls!!!");
+            exit(1);
+        }
+        let get_home_output = get_home_res.unwrap().stdout;
+        let home_string = String::from_utf8_lossy(&get_home_output);
+        config_path.push(home_string.trim());
+    }
+    else{
+        let user_dirs = user_dirs_res.unwrap();
+        config_path.push(user_dirs.home_dir());
+    }
     config_path.push(".config/pyro_pentest_tool/conf");
     if config_path.as_path().exists() == false{
         install::install(&config_path);
@@ -53,6 +120,14 @@ fn main() {
     let mut tools_folder = PathBuf::new();
     let mut terminal_command = String::new();
     let mut box_template = String::new();
+    let mut cracking_rig = String::new();
+    let mut rockyou = String::new();
+    let mut rule = String::new();
+    let mut upcoming_files = PathBuf::new();
+    let mut upcoming_notes = PathBuf::new();
+    let mut pass_spray_file = PathBuf::new();
+    let mut fingerprint = false;
+    let mut vault_name = String::new();
     println!("\nconfig already generated\nloading config file...\n");
     let settings_string = fs::read_to_string(&config_path).expect("error reading config file");
     let settings: Vec<&str> = settings_string.split("\n").collect();
@@ -63,8 +138,16 @@ fn main() {
                 "Project_files" => project_base_folder.push(setting_vec[1].trim_end()),
                 "Project_notes" => project_base_notes.push(setting_vec[1].trim_end()),
                 "tools_folder" => tools_folder.push(setting_vec[1].trim_end()),
+                "upcoming_files" => upcoming_files.push(setting_vec[1].trim_end()),
+                "upcoming_notes" => upcoming_notes.push(setting_vec[1].trim_end()),
                 "box_template" => box_template = setting_vec[1].trim_end().to_owned(),
                 "terminal" => terminal_command = setting_vec[1].trim_end().to_owned(),
+                "cracking_rig" => cracking_rig = setting_vec[1].trim_end().to_owned(),
+                "rockyou_location" => rockyou = setting_vec[1].trim_end().to_owned(),
+                "rule_location" => rule = setting_vec[1].trim_end().to_owned(),
+                "pass_file"=> pass_spray_file.push(setting_vec[1]),
+                "fingerprint" => {if setting_vec[1].contains("y"){fingerprint = true}},
+                "vault_name" => vault_name = setting_vec[1].trim_end().to_owned(),
                 _ => println!("error unknown setting: {}", setting_vec[0])
             }
         }
@@ -74,12 +157,13 @@ fn main() {
     Note Folders: {}
     Tools Folder: {}
     distrobox template: {}
-    terminal_command: {}\n
+    terminal_command: {}
-", project_base_folder.display(), project_base_notes.display(), tools_folder.display(), box_template, terminal_command);
+    cracking_rig: {}\n
+    upcoming project folders: {}
+    upcoming project notes: {}
+", &project_base_folder.display(), &project_base_notes.display(), &tools_folder.display(), box_template, terminal_command, cracking_rig, &upcoming_files.display(), &upcoming_notes.display());
     println!("loading project configs...");
-    let projects = project_controls::get_projects(&config_path);
+    let projects = project_controls::get_projects(&config_path, true);
-    println!("Enter to start main menu");
+    let _continue = get_user_input("press enter to load command line interface.");
-    let mut enter = String::new();
+    cli::cli(true, projects, config_path, &project_base_folder, &project_base_notes, &tools_folder, box_template, terminal_command, cracking_rig, rockyou, rule, &upcoming_files, &upcoming_notes, &pass_spray_file, fingerprint, vault_name);
-    std::io::stdin().read_line(&mut enter).unwrap();
-    menu::main_menu(projects, config_path, &project_base_folder, &project_base_notes, &tools_folder, box_template, terminal_command);
 }

pentest_tool/src/menu.rs

@@ -1,41 +1,29 @@
 use std::path::PathBuf;
+use std::thread::JoinHandle;
 use clearscreen::clear;
 use clearscreen;
+use crate::get_user_input;
 use crate::Project;
-use crate::project_controls;
+use crate::cli;
-use crate::box_controls;
-use crate::info_controls;
-use crate::start_pentest;
 
-fn next_project_id(config_path: &PathBuf) -> i32{
+pub fn main_menu(projects: &mut Vec<Project>,
-    let projects = project_controls::get_projects(config_path);
+    config_path: PathBuf, 
-    let mut new_id = 0;
+    base_files: &PathBuf, 
-    for project in projects.clone(){
+    base_notes: &PathBuf, 
-        if project.id > new_id{
+    tools_dir: &PathBuf, 
-            new_id = project.id + 1;
+    boxtemplate: String, 
-        }
+    terminal: String, 
-    }
+    cracking_rig: String, 
-    return new_id;
+    rockyou: String, 
-}
+    rule: String, 
-
+    upcoming_files: &PathBuf, 
-fn get_active_project(projects: &Vec<Project>) -> &Project{
+    upcoming_notes: &PathBuf, 
-    let mut active_project = &projects[0];
+    password_spray_file: &PathBuf, 
-    for project in projects{
+    fingerprint: bool, 
-        if project.active == true{
+    vault_name: String) -> Option<Vec<JoinHandle<()>>>{
-            active_project = project
-        }
-    }
-    return active_project
-}
-
-pub fn main_menu(mut projects: Vec<Project>, config_path: PathBuf, base_files: &PathBuf, base_notes: &PathBuf, tools_dir: &PathBuf, boxtemplate: String, terminal: String){
     let mut loopize = true;
-    let mut new_id = next_project_id(&config_path);
+    let mut threads = Vec::new(); 
-    loop {
+    let banner = "
-        let active_project = get_active_project(&projects);
-        let mut response = String::new();
-        clear().expect("error clearing screen");
-        print!("
 ,,,;;::ccccc::;;;::c::;,;::cccccllc::::::;:::;;;;,,;,'',,;,,;;;;;;;:;;;;;,,,,,,,,,,,'''''',,,,,,''''
 ,;;;::ccccc::::::ccc:;;;:ccccccclc::ccccccc::;;;;;;;;;;,,;;;;;;;;;;;;;;;,,,,,,,,,,,'''''''''',,,,,''
 ,;;:::ccc:cc:::ccc:::::::ccccclcccllccccllc::::::;;;;;;;;;;;;;;;;;;;;;;,,,,,,,,,,,''''''''...'',,,,'
@@ -72,78 +60,262 @@ pub fn main_menu(mut projects: Vec<Project>, config_path: PathBuf, base_files: &
 ....,;:clodxxkkOOOkO0OOOOxdlcc;;,......      .';,.................         ...',:ldxxxdlc;,''.......
 ...,:clodooxkkkO0OxO00OOxo:;;,. ........   .''.......... ..  ..           ..,,,;:codxxdlc:;,'.......
 '',;clodolokOkxkOkkO00Oko:;;;.  ..... ..   .,,........'. ..  ..  ..   ..........;:codocclc:,,'......
-              ___       __   __         ___               __        ___  __  
+";
-        |  | |__  |    /  ` /  \\  |\\/| |__     |__|  /\\  /  ` |__/ |__  |__) 
+    while loopize {
-        |/\\| |___ |___ \\__, \\__/  |  | |___    |  | /~~\\ \\__, |  \\ |___ |  \\ 
+        let _enter_to_clear = get_user_input("press enter to display main menu.");
-                                                                             
+        clear().expect("error clearing screen");
-         __   ___ ___     __                    __                           
+        print!("
-        / _` |__   |     |__) |  | |\\ | | |\\ | / _`                          
+{}
-        \\__> |___  |     |    |/\\| | \\| | | \\| \\__>     
+___  ___      _        ___  ___                   
-    
+|  \\/  |     (_)       |  \\/  |                 _ 
-
+| .  . | __ _ _ _ __   | .  . | ___ _ __  _   _(_)
-
+| |\\/| |/ _` | | '_ \\  | |\\/| |/ _ \\ '_ \\| | | |  
-NOTE OPTION 18 WILL SAVE YOUR PROJECTS BEFORE QUITTING
+| |  | | (_| | | | | | | |  | |  __/ | | | |_| |_ 
-
+\\_|  |_/\\__,_|_|_| |_| \\_|  |_/\\___|_| |_|\\__,_(_)
-Current Project: {} {}
-
-        Main Menu:
             1 .) Show Active Project
             2 .) List Projects
             3 .) Switch Active Project
-            4 .) create new project with Pyro's default tool
+            4 .) create new project with Pyro's default layout
             5 .) Save Project Information
-            6 .) Import New Project - and setup new Distrobox
+            6 .) Import New Project to Current projects list - and setup new Distrobox
             7 .) Remove Project
-            8 .) Open A New Terminal in Current Active Project
+            8 .) Print upcoming projects
-            9 .) Open A Terminal In this windows for the current active project
+            9. ) promote project from upcoming to current
-            10.) Open Project Files Folder In Dolphin
+            10.) Open A New Terminal in Current Active Project
-            11.) Open Project Notes Folder In Dolphin
+            11.) Open A Terminal In this windows for the current active project
-            12.) generate userpass file from your obsidian notes
+            12.) open current project's cobalt strike
-            13.) run pyro's initail enum script on a nessus csv for the current project
+            13.) re-create the distrobox for the current active project
-            14.) Print Project Info For Report
+            14.) generate userpass file from your obsidian notes
-            15.) Build host discovery cmd command from scope in notes
+            15.) run pyro's initail enum script on a nessus csv for the current project
-            16.) build portscan command from scope in notes
+            16.) build external attack notes from host notes
-            17.) Stop All Distroboxes
+            17.) Build host discovery cmd command from scope in notes
-            18.) Quit Application
+            18.) build portscan command from scope in notes
-\n", active_project.customer, active_project.project_name);
+            19.) parse a cs portscan services.tsv file
-        std::io::stdin().read_line(&mut response).expect("error getting menu input");
+            20.) Stop All Distroboxes
+            21.) Password Spray (will print password to spray, and wait the obervation window time)
+            22.) Launch bloodhound with the current project's distrobox
+            23.) Parse GatherContacts output file
+            24.) prune unused distroboxes (free up system storage)
+            25.) Gather DNS Records
+            26.) Bruteforce Subdomains
+            27.) Do all DNS enumeration 
+            28.) enter external only menu
+            29.) enter internal only menu
+            30.) exit menu
+\n", banner);
+        match get_user_input("selection?").as_str(){
+            "1" => {let thread_option = cli::run_command("show active project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "2" => {let thread_option = cli::run_command("list projects".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "3" => {let thread_option = cli::run_command("switch project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "4" => {let thread_option = cli::run_command("create_new_project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "5" => {let thread_option = cli::run_command("save projects".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "6" => {let thread_option = cli::run_command("import project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "7" => {let thread_option = cli::run_command("remove project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "8" => {let thread_option = cli::run_command("show upcoming projects".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "9" => {let thread_option = cli::run_command("promote project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "10" => {let thread_option = cli::run_command("new terminal".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "11" => {let thread_option = cli::run_command("inline terminal".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "12" => {let thread_option = cli::run_command("cobalt strike".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "13" => {let thread_option = cli::run_command("recreate distrobox".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "14" => {let thread_option = cli::run_command("generate userpass".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "15" => {let thread_option = cli::run_command("initial enum".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "16" => {let thread_option = cli::run_command("build attack notes".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "17" => {let thread_option = cli::run_command("host discovery".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "18" => {let thread_option = cli::run_command("port scan".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "19" => {let thread_option = cli::run_command("parse port scan".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "20" => {let thread_option = cli::run_command("stop boxes".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "21" => {let thread_option = cli::run_command("password spray".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "22" => {let thread_option = cli::run_command("bloodhound".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "23" => {let thread_option = cli::run_command("parse gather contacts".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "24" => {let thread_option = cli::run_command("prune distroboxes".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "25" => {let thread_option = cli::run_command("dns records".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "26" => {let thread_option = cli::run_command("brute force subdomains".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "27" => {let thread_option = cli::run_command("dns enumeration".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+            "28" => {let threads_option = external_menu(banner, projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.clone(), cracking_rig.clone(), rockyou.clone(), rule.clone(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.clone()); if threads_option.is_some(){for thread in threads_option.unwrap(){threads.push(thread)}}},
+            "29" => {let threads_option = internal_menu(banner, projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.clone(), cracking_rig.clone(), rockyou.clone(), rule.clone(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.clone()); if threads_option.is_some(){for thread in threads_option.unwrap(){threads.push(thread)}}},
+            "30" => loopize = false,
+            _ => println!("unknown selection, try again!"),
+        }
+    }
+    if threads.len() > 0{
+        return Some(threads);
+    }
+    else{
+        return None;
+    }
+    pub fn external_menu(
+        banner: &str,
+        projects: &mut Vec<Project>,
+        config_path: PathBuf, 
+        base_files: &PathBuf, 
+        base_notes: &PathBuf, 
+        tools_dir: &PathBuf, 
+        boxtemplate: String, 
+        terminal: String, 
+        cracking_rig: String, 
+        rockyou: String, 
+        rule: String, 
+        upcoming_files: &PathBuf, 
+        upcoming_notes: &PathBuf, 
+        password_spray_file: &PathBuf, 
+        fingerprint: bool, 
+        vault_name: String) -> Option<Vec<JoinHandle<()>>>{
+            let mut loopize = true;
+            let mut threads = Vec::new(); 
+            while loopize {
+                let _enter_to_clear = get_user_input("press enter to display external project menu.");
                 clear().expect("error clearing screen");
-        match response.as_str().trim_end(){
+                print!("
-            "1" => println!("\nclient: {}\n\nproject: {}\n\nbox: {}\n\nproject files: {}\n\nproject notes: {}\n", active_project.customer ,active_project.project_name, active_project.boxname, active_project.files_folder.display(), active_project.notes_folder.display()),
+        {}
-            "2" => {println!("+++++++++++++++++++++"); 
+ _____     _                        _  ___  ___                   
-                    for project in &projects{
+|  ___|   | |                      | | |  \\/  |                 _ 
-                        println!("++{}|{}++",project.customer ,project.project_name)}
+| |____  _| |_ ___ _ __ _ __   __ _| | | .  . | ___ _ __  _   _(_)
-                    println!("++++++++++++++++++++")},
+|  __\\ \\/ / __/ _ \\ '__| '_ \\ / _` | | | |\\/| |/ _ \\ '_ \\| | | |  
-            "3" => project_controls::switch_project(&mut projects),
+| |___>  <| ||  __/ |  | | | | (_| | | | |  | |  __/ | | | |_| |_ 
-            "4" => start_pentest::start_pentest(&config_path),
+\\____/_/\\_\\\\__\\___|_|  |_| |_|\\__,_|_| \\_|  |_/\\___|_| |_|\\__,_(_)
-            "5" => project_controls::save_projects(&projects, &config_path),
+        1 .) Show Active Project
-            "6" => {new_id = new_id + 1; project_controls::new_project(&mut projects, &base_files, &base_notes, &tools_dir, &boxtemplate, &config_path, new_id)},
+        2 .) List Projects
-            "7" => project_controls::remove_project(&mut projects, &config_path),
+        3 .) Switch Active Project
-            "8" => box_controls::project_standalone_terminal(active_project.clone(), terminal.clone()),
+        4 .) create new project with Pyro's default layout
-            "9" => box_controls::project_inline_terminal(active_project.clone()),
+        5 .) Save Project Information
-            "10" => info_controls::open_in_dolphin("files", active_project.clone()),
+        6 .) Import New Project to Current projects list - and setup new Distrobox
-            "11" => info_controls::open_in_dolphin("notes", active_project.clone()),
+        7 .) Remove Project
-            "12" => info_controls::generate_userpass(&active_project),
+        8 .) Print upcoming projects
-            "13" => info_controls::run_initial_enum(&active_project),
+        9. ) promote project from upcoming to current
-            "14" =>info_controls::print_report_information(active_project.clone()),
+        10.) Open A New Terminal in Current Active Project
-            "15" => info_controls::build_cmd_for_host_discovery(&active_project),
+        11.) Open A Terminal In this windows for the current active project
-            "16" => info_controls::build_cs_portscan_cmd(&active_project),
+        12.) re-create the distrobox for the current active project
-            "17" => box_controls::stop_all_boxes(&projects),
+        13.) generate userpass file from your obsidian notes
-            "18" => {project_controls::save_projects(&projects, &config_path);
+        14.) run pyro's initail enum script on a nessus csv for the current project
-                    let mut stop = String::new(); 
+        15.) build external attack notes from host notes
-                    println!("stop all boxes?\ny/n");
+        16.) Stop All Distroboxes
-                    std::io::stdin().read_line(&mut stop).unwrap();
+        17.) Password Spray (will print password to spray, and wait the obervation window time)
-                    if stop.contains("y"){
+        18.) Parse GatherContacts output file
-                        box_controls::stop_all_boxes(&projects);
+        19.) prune unused distroboxes (free up system storage)
+        20.) Gather DNS Records
+        21.) Brute force Subdomains
+        22.) Do all DNS Enumeration
+        23.) exit menu
+        \n", banner);
+                match get_user_input("selection?").as_str(){
+                    "1" => {let thread_option = cli::run_command("show active project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "2" => {let thread_option = cli::run_command("list projects".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "3" => {let thread_option = cli::run_command("switch project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "4" => {let thread_option = cli::run_command("create_new_project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "5" => {let thread_option = cli::run_command("save projects".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "6" => {let thread_option = cli::run_command("import project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "7" => {let thread_option = cli::run_command("remove project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "8" => {let thread_option = cli::run_command("show upcoming projects".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "9" => {let thread_option = cli::run_command("promote project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "10" => {let thread_option = cli::run_command("new terminal".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "11" => {let thread_option = cli::run_command("inline terminal".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "12" => {let thread_option = cli::run_command("recreate distrobox".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "13" => {let thread_option = cli::run_command("generate userpass".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "14" => {let thread_option = cli::run_command("initial enum".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "15" => {let thread_option = cli::run_command("build attack notes".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "26" => {let thread_option = cli::run_command("stop boxes".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "17" => {let thread_option = cli::run_command("password spray".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "18" => {let thread_option = cli::run_command("parse gather contacts".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "19" => {let thread_option = cli::run_command("prune distroboxes".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "20" => {let thread_option = cli::run_command("dns records".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "21" => {let thread_option = cli::run_command("brute force subdomains".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "22" => {let thread_option = cli::run_command("dns enumeration".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                    "23" => loopize = false,
+                    _ => println!("unknown selection, try again!"),
                 }
-                    loopize = false},
-            _ => println!("uknonwn selection")
             }
-        if loopize == false{
+            if threads.len() > 0{
-            break
+                return Some(threads);
+            }
+            else{
+                return None;
+            }
+        }
+
+        pub fn internal_menu(
+            banner: &str,
+            projects: &mut Vec<Project>,
+            config_path: PathBuf, 
+            base_files: &PathBuf, 
+            base_notes: &PathBuf, 
+            tools_dir: &PathBuf, 
+            boxtemplate: String, 
+            terminal: String, 
+            cracking_rig: String, 
+            rockyou: String, 
+            rule: String, 
+            upcoming_files: &PathBuf, 
+            upcoming_notes: &PathBuf, 
+            password_spray_file: &PathBuf, 
+            fingerprint: bool, 
+            vault_name: String) -> Option<Vec<JoinHandle<()>>>{
+                let mut loopize = true;
+                let mut threads = Vec::new(); 
+                while loopize {
+                    let _enter_to_clear = get_user_input("press enter to display internal project menu.");
+                    clear().expect("error clearing screen");
+                    print!("
+            {}
+ _____      _                        _  ___  ___                   
+|_   _|    | |                      | | |  \\/  |                 _ 
+  | | _ __ | |_ ___ _ __ _ __   __ _| | | .  . | ___ _ __  _   _(_)
+  | || '_ \\| __/ _ \\ '__| '_ \\ / _` | | | |\\/| |/ _ \\ '_ \\| | | |  
+ _| || | | | ||  __/ |  | | | | (_| | | | |  | |  __/ | | | |_| |_ 
+ \\___/_| |_|\\__\\___|_|  |_| |_|\\__,_|_| \\_|  |_/\\___|_| |_|\\__,_(_)
+        1 .) Show Active Project
+        2 .) List Projects
+        3 .) Switch Active Project
+        4 .) create new project with Pyro's default layout
+        5 .) Save Project Information
+        6 .) Import New Project to Current projects list - and setup new Distrobox
+        7 .) Remove Project
+        8 .) Print upcoming projects
+        9. ) promote project from upcoming to current
+        10.) Open A New Terminal in Current Active Project
+        11.) Open A Terminal In this windows for the current active project
+        12.) open cobalt strike in the current project's distrobox
+        13.) re-create the distrobox for the current active project
+        14.) generate userpass file from your obsidian notes
+        15.) print host discvoery command for internal pingsweep
+        16.) print cobalt strike portscan command to include all in scope ranges
+        17) parse a cobalt strike portscan CSV
+        18.) Stop All Distroboxes
+        19.) Password Spray (will print password to spray, and wait the obervation window time)
+        20.) open bloodhound in the current project's distrobox
+        21.) prune unused distroboxes (free up system storage)
+        22.) exit menu
+            \n", banner);
+                    match get_user_input("selection?").as_str(){
+                        "1" => {let thread_option = cli::run_command("show active project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "2" => {let thread_option = cli::run_command("list projects".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "3" => {let thread_option = cli::run_command("switch project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "4" => {let thread_option = cli::run_command("create_new_project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "5" => {let thread_option = cli::run_command("save projects".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "6" => {let thread_option = cli::run_command("import project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "7" => {let thread_option = cli::run_command("remove project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "8" => {let thread_option = cli::run_command("show upcoming projects".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "9" => {let thread_option = cli::run_command("promote project".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "10" => {let thread_option = cli::run_command("new terminal".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "11" => {let thread_option = cli::run_command("inline terminal".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "12" => {let thread_option = cli::run_command("cobalt strike".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "13" => {let thread_option = cli::run_command("recreate distrobox".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "14" => {let thread_option = cli::run_command("generate userpass".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "15" => {let thread_option = cli::run_command("host discovery".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "16" => {let thread_option = cli::run_command("port scan".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "17" => {let thread_option = cli::run_command("parse port scan".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "18" => {let thread_option = cli::run_command("stop boxes".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "19" => {let thread_option = cli::run_command("password spray".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "20" => {let thread_option = cli::run_command("bloodhound".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "21" => {let thread_option = cli::run_command("prune distroboxes".to_owned(), projects, config_path.clone(), base_files, base_notes, tools_dir, boxtemplate.clone(), terminal.to_owned(), cracking_rig.to_owned(), rockyou.to_owned(), rule.to_owned(), upcoming_files, upcoming_notes, password_spray_file, fingerprint, vault_name.to_owned()); if thread_option.is_some(){threads.push(thread_option.unwrap());}},
+                        "22" => loopize = false,
+                        _ => println!("unknown selection, try again!"),
+                    }
+                }
+                if threads.len() > 0{
+                    return Some(threads);
+                }
+                else{
+                    return None;
                 }
-        println!("\n\n\npress enter to return to the menu");
-        let mut enter = String::new();
-        std::io::stdin().read_line(&mut enter).unwrap();
             }
 }

pentest_tool/src/portscan_controls.rs

@@ -0,0 +1,830 @@
+use std::collections::HashMap;
+use std::fs;
+use std::fs::{read_to_string, create_dir_all};
+use std::io::Write;
+use std::path::PathBuf;
+ use std::process::Command;
+use std::thread::{sleep, spawn, JoinHandle};
+use std::time::Duration;
+use walkdir::WalkDir;
+use crate::get_user_input;
+use crate::Project;
+use crate::open_overwrite;
+use crate::open_append;
+use crate::info_controls;
+
+pub fn parse_normal_nmap_output(project: &Project){
+    let mut found_file = false;
+    let mut found_working = false;
+    let mut file_to_parse = PathBuf::new();
+    let mut save_path = PathBuf::new();
+    let files_folder = project.files_folder.clone();
+    println!("scanning files folder for an nmap file...");
+    let walkdir_result = WalkDir::new(files_folder);
+    for entry_res in walkdir_result{
+        if entry_res.is_ok(){
+            let entry = entry_res.unwrap();
+            let entry_name = entry.path().display().to_string();
+            if entry_name.contains("nmap") || entry_name.contains("port scan") || entry_name.contains("proxychains_output"){
+                file_to_parse = entry.into_path();
+            }
+            else if entry.file_type().is_dir(){
+                if entry.file_name().to_ascii_lowercase().to_string_lossy().contains("working"){
+                    save_path.push(entry.path());
+                    found_working = true;
+                }
+            }
+        }
+    }
+    if !found_working{
+        save_path.push(&project.files_folder);
+    }
+    if file_to_parse.display().to_string().len() > 1{
+        println!("we found {}", file_to_parse.display());
+        let res = get_user_input("is the the file you want to parse?");
+        if res.to_lowercase().contains("y"){
+            found_file = true;
+        }
+    }
+    if !found_file{
+        println!("ooof the right file was not found.");
+        let file_prompt = get_user_input("please enter the full path to the file you want to parse.");
+        file_to_parse.clear();
+        file_to_parse.push(file_prompt);
+    }
+    let file_read_res = fs::read_to_string(file_to_parse);
+    if file_read_res.is_err(){
+        let error = file_read_res.err().unwrap();
+        println!("ooof error reading the file!");
+        println!("{}", error);
+        return;
+    }
+    let nmap_string = file_read_res.unwrap();
+    let sections: Vec<&str> = nmap_string.split("Nmap scan report for").collect();
+    let mut host_ports = Vec::new();
+    for section in sections{
+        let mut host = String::new();
+        for line in section.split("\n").collect::<Vec<&str>>(){
+            if line.contains("."){
+                if !line.contains("latency"){
+                    host = line.trim().to_owned();
+                }
+            }
+            else{
+                if line.contains("open"){
+                    let port = line.split_whitespace().collect::<Vec<&str>>()[0];
+                    host_ports.push(format!("{}:{}", host, port));
+                }
+            }
+        }
+    }
+    println!("done parsing.");
+    if save_path.display().to_string().len() > 1{
+        save_path.push("parsed_namp.txt");
+        println!("default save location is {}", save_path.display());
+        if get_user_input("is this ok?").contains("n"){
+            let user_path = get_user_input("ok where would you like to save it instead?");
+            save_path.clear();
+            save_path.push(user_path);
+        }
+
+    }
+    else{
+        let user_path = get_user_input("where would you like to save the file?");
+        save_path.clear();
+        save_path.push(user_path);
+    }
+    let save_open_res = fs::OpenOptions::new().create(true).append(true).open(&save_path);
+    if save_open_res.is_err(){
+        let error = save_open_res.err().unwrap();
+        println!("oof error opening the save file!");
+        println!("{}", error);
+        if get_user_input("do you want to print the results to the console instead?").to_lowercase().contains("y"){
+            for host in &host_ports{
+                println!("{}", host);
+            }
+        }
+        return;
+    }
+    let mut host_port_file = save_open_res.unwrap();
+    save_path.pop();
+    save_path.push("parsed_nmap.tsv");
+    let tsv_open_res = fs::OpenOptions::new().create(true).append(true).open(save_path);
+    if tsv_open_res.is_err(){
+        let error = tsv_open_res.err().unwrap();
+        println!("error opening tsv file!");
+        println!("{}", error);
+        if get_user_input("do you want to print the results to the console instead?").to_lowercase().contains("y"){
+            for host in &host_ports{
+                println!("{}", host);
+            }
+        }
+        return;
+    }
+    let mut tsv_file = tsv_open_res.unwrap();
+    for host in &host_ports{
+        let outline = format!("{}\n", host);
+        let split_data: Vec<&str> = host.split(":").collect();
+        let ip = split_data[0];
+        let port_split: Vec<&str> = split_data[1].split("/").collect();
+        let port = port_split[0];
+        let proto = port_split[1];
+        let tsv_outline = format!("{}\t{}\t\t{}\n", ip, port, proto);
+        let host_port_write_res = write!(host_port_file, "{}", outline);
+        let tsv_write_res = write!(tsv_file, "{}", tsv_outline);
+        if host_port_write_res.is_err(){
+            println!("error writing data to file...");
+        }
+        if tsv_write_res.is_err(){
+            println!("error writing data to the tsv file...");
+        }
+    }
+}
+
+pub fn build_cs_portscan_cmd(project: &Project){
+    let mut general_note_path = project.notes_folder.clone();
+    general_note_path.push("general.md");
+    println!("Reading from: {}", general_note_path.display());
+    let mut ranges = Vec::new();
+    let general_note_string = read_to_string(general_note_path);
+    let mut _note_string = String::new();
+   match general_note_string{
+        Err(error) => {println!("error reading file to string!! {}", error); return;},
+        _=> _note_string = general_note_string.unwrap()
+    }
+    let lines: Vec<&str> = _note_string.split("\n").collect();
+    for line in lines{
+       if line.contains("|"){
+            if line.contains("."){
+                let ip = line.split("|").collect::<Vec<&str>>()[1];
+                if ip.contains(","){
+                    let ips: Vec<&str> = ip.split(",").collect();
+                    for ip in ips{
+                        ranges.push(ip.trim_end().trim_start());
+                    }
+                } 
+                else{
+                    ranges.push(ip.trim_end().trim_start());
+                }
+            }
+        }
+    }
+    let portscan_cmd = "portscan !!! 1-1024,1433,2222,3306,3389,5000-6000,8000,8080,8443 icmp 512";
+    let mut _range_command = String::new();
+    let combined_ranges = ranges.join(",");
+    let final_command = portscan_cmd.replace("!!!", &combined_ranges);
+    println!("{}", final_command);
+}
+
+pub fn build_cmd_for_host_discovery(project: &Project){
+    let mut cobalt_strike_response = String::new();
+    let mut need_shell = false;
+    println!("will you be running this via cobalt strike? (do you need it to start with \"shell\"?");
+    let input_result = std::io::stdin().read_line(&mut cobalt_strike_response);
+    if input_result.is_err(){
+        println!("error getting user input... not really sure how you did this... just pressing enter would work..... wow");
+        return;
+    }
+    if cobalt_strike_response.to_lowercase().contains("y"){
+        need_shell = true;
+    }
+    let mut general_note_path = project.notes_folder.clone();
+    general_note_path.push("general.md");
+    println!("Reading from: {}", general_note_path.display());
+    let mut ranges = Vec::new();
+    let general_note_string = read_to_string(general_note_path);
+    let mut _note_string = String::new();
+   match general_note_string{
+        Err(error) => {println!("error reading file to string!! {}", error); return;},
+        _=> _note_string = general_note_string.unwrap()
+    }
+    let lines: Vec<&str> = _note_string.split("\n").collect();
+    for line in lines{
+       if line.contains("|"){
+            if line.contains("."){
+                let ip = line.split("|").collect::<Vec<&str>>()[1];
+                if ip .contains(","){
+                    let ips: Vec<&str> = ip.split(",").collect();
+                    for ip in ips{
+                        ranges.push(ip.trim_end().trim_start());
+                    }
+                } 
+                else{
+                    ranges.push(ip.trim_end().trim_start());
+                }
+            }
+        }
+    }
+    let mut _discovery_command = "";
+    _discovery_command = "(for /L %a IN (1,1,254) DO ping /n 1 /w 3 !!!.%a) | find \"Reply\"";
+    let mut final_command = String::new();
+    for range in ranges{
+        let mut network: Vec<&str> = range.split(".").collect();
+        network.pop();
+        let network_string = network.join(".");
+        let mut _range_command = String::new();
+        if need_shell{
+            _range_command = format!("{}{}","shell ", _discovery_command.replace("!!!", &network_string));
+            need_shell = false;
+        }
+        else{
+            _range_command = _discovery_command.replace("!!!", &network_string);
+        }
+        final_command.push_str(_range_command.as_str());
+        final_command.push_str(">> ping_only_replies.txt &");
+    }
+    final_command.pop();
+    println!("{}", final_command);
+}
+
+pub fn parse_csportscan(project: &Project){
+    let mut tsv_path = project.files_folder.clone();
+    tsv_path.push("working/tsvs/services.tsv");
+    let mut outfile = tsv_path.clone();
+    outfile.pop();
+    outfile.pop();
+    let mut windows_hosts = Vec::new();
+    let mut ssh_hosts = Vec::new();
+    let mut ftp_hosts = Vec::new();
+    let mut rdp_hosts = Vec::new();
+    let mut dns_hosts = Vec::new();
+    let mut snmp_hosts = Vec::new();
+    let mut web_hosts = Vec::new();
+    let mut telnet_hosts = Vec::new();
+    let mut unknown_ports = Vec::new();
+    
+    if !get_user_input("do you have the tsv saved in the project folder under working/tsvs/services.tsv?").to_lowercase().contains("y"){
+        tsv_path.clear();
+        tsv_path.push(get_user_input("ooof ok, please enter the full path to your tsv file."));
+    }
+    let tsv_read_res = read_to_string(tsv_path);
+    if tsv_read_res.is_err(){
+        let error = tsv_read_res.err().unwrap();
+        println!("ooof error reading tsv file!");
+        println!("{}", error);
+        return;
+    }
+    println!("tsv read, parsing lines...");
+    let tsv_string = tsv_read_res.unwrap();
+    let lines: Vec<&str> = tsv_string.split("\n").collect();
+    for line in lines{
+        let words: Vec<&str> = line.split("\t").collect();
+        if words.len() > 1{
+            let host = words[0].to_lowercase().to_owned();
+            let port = words[1].to_lowercase().to_owned();
+            let host_entry = format!("{}:{}", &host, &port);
+            match words[1]{
+                "135" => {if !windows_hosts.contains(&host){windows_hosts.push(host)}},
+                "445" => {if !windows_hosts.contains(&host){windows_hosts.push(host)}},
+                "22" => {if !ssh_hosts.contains(&host){ssh_hosts.push(host);}},
+                "21" => {if !ftp_hosts.contains(&host){ftp_hosts.push(host);}},
+                "23" => {if !telnet_hosts.contains(&host){telnet_hosts.push(host)}},
+                "3389" => {if !rdp_hosts.contains(&host){rdp_hosts.push(host);}},
+                "80" | "443" | "8080" | "8443" | "4433" | "8000" => {if !web_hosts.contains(&host_entry){web_hosts.push(host_entry);}},
+                "53" => {if !dns_hosts.contains(&host){dns_hosts.push(host);}},
+                "161" => {if !snmp_hosts.contains(&host){snmp_hosts.push(host);}},
+                _ => {
+                    if words.len() == 3{
+                        let banner = words[2].to_lowercase().to_owned();
+                        if words[2].to_lowercase().contains("ssh"){
+                            if !ssh_hosts.contains(&host_entry){
+                                ssh_hosts.push(host_entry);
+                            }
+                        }
+                        else if banner.contains("ftp"){
+                            if !ftp_hosts.contains(&host_entry){
+                                ftp_hosts.push(host_entry);
+                            }
+                        }
+                        else if banner.contains("nginx") || banner.contains("apache"){
+                            if !web_hosts.contains(&host_entry){
+                                web_hosts.push(host_entry);
+                            }
+                        }
+                        else{
+                            continue;
+                        }
+                    }
+                    else if words.len() == 2{
+                        unknown_ports.push(host_entry);
+                    }
+                }
+            }
+        }
+    }
+    println!("is {} where you want to save your files?", outfile.display());
+    if get_user_input("").to_lowercase().contains("n"){
+        outfile.clear();
+        outfile.push(get_user_input("ok, please enter the full path to the folder you want to save them to."));
+    }
+    print!("
+{} Windows hosts found!
+{} SSH hosts found!
+{} FTP hosts found!
+{} Telnet hosts found!
+{} SNMP hosts found!
+{} DNS hosts found!
+{} RDP hosts found!
+{} untagged hosts found!
+", windows_hosts.len(), ssh_hosts.len(), ftp_hosts.len(), telnet_hosts.len(), snmp_hosts.len(), dns_hosts.len(), rdp_hosts.len(), unknown_ports.len());
+    println!("lines parsed! creating output files...");
+    outfile.push("windows_hosts.txt");
+    let file_option = open_overwrite(&outfile);
+    if file_option.is_some(){
+        let mut windows_file = file_option.unwrap();
+        for host in windows_hosts{
+            let write_res = write!(windows_file, "{}\n", host);
+            if write_res.is_err(){
+                let error = write_res.err().unwrap();
+                println!("oooof error writing windows_hosts.txt!!");
+                println!("{}", error);
+            }
+            else{
+                write_res.unwrap();
+            }
+        }
+    }
+    outfile.pop();
+    outfile.push("ssh_hosts.txt");
+    let file_option = open_overwrite(&outfile);
+    if file_option.is_some(){
+       let mut ssh_file = file_option.unwrap();
+        for host in ssh_hosts{
+            let write_res = write!(ssh_file, "{}\n", host);
+            if write_res.is_err(){
+                let error = write_res.err().unwrap();
+                println!("oooof error writing ssh_hosts.txt!!");
+                println!("{}", error);
+            }
+            else{
+                write_res.unwrap();
+            }
+        }
+    }
+    outfile.pop();
+    outfile.push("telnet_hosts.txt");
+    let file_option = open_overwrite(&outfile);
+    if file_option.is_some(){
+       let mut telnet_file = file_option.unwrap();
+        for host in telnet_hosts{
+            let write_res = write!(telnet_file, "{}\n", host);
+            if write_res.is_err(){
+                let error = write_res.err().unwrap();
+                println!("oooof error writing _hosts.txt!!");
+                println!("{}", error);
+            }
+            else{
+                write_res.unwrap();
+            }
+        }
+    }
+    outfile.pop();
+    outfile.push("ftp_hosts.txt");
+    let file_option = open_overwrite(&outfile);
+    if file_option.is_some(){
+        let mut ftp_file = file_option.unwrap();
+        for host in ftp_hosts{
+            let write_res = write!(ftp_file, "{}\n", host);
+            if write_res.is_err(){
+                let error = write_res.err().unwrap();
+                println!("oooof error writing _hosts.txt!!");
+                println!("{}", error);
+            }
+            else{
+                write_res.unwrap();
+            }
+        }
+    }
+    outfile.pop();
+    outfile.push("snmp_hosts.txt");
+    let file_option = open_overwrite(&outfile);
+    if file_option.is_some(){
+        let mut snmp_file = file_option.unwrap();
+        for host in snmp_hosts{
+            let write_res = write!(snmp_file, "{}\n", host);
+            if write_res.is_err(){
+                let error = write_res.err().unwrap();
+                println!("oooof error writing _hosts.txt!!");
+                println!("{}", error);
+            }
+            else{
+                write_res.unwrap();
+            }
+        }
+    }
+    outfile.pop();
+    outfile.push("dns_hosts.txt");
+    let file_option = open_overwrite(&outfile);
+    if file_option.is_some(){
+       let mut dns_file = file_option.unwrap();
+        for host in dns_hosts{
+            let write_res = write!(dns_file, "{}\n", host);
+            if write_res.is_err(){
+                let error = write_res.err().unwrap();
+                println!("oooof error writing _hosts.txt!!");
+                println!("{}", error);
+            }
+            else{
+                write_res.unwrap();
+            }
+        }
+    }
+    outfile.pop();
+    outfile.push("rdp_hosts.txt");
+    let file_option = open_overwrite(&outfile);
+    if file_option.is_some(){
+       let mut rdp_file = file_option.unwrap();
+        for host in rdp_hosts{
+            let write_res = write!(rdp_file, "{}\n", host);
+            if write_res.is_err(){
+                let error = write_res.err().unwrap();
+                println!("oooof error writing _hosts.txt!!");
+                println!("{}", error);
+            }
+            else{
+                write_res.unwrap();
+            }
+        }
+    }
+    outfile.pop();
+    outfile.push("web_hosts.txt");
+    let file_option = open_overwrite(&outfile);
+    if file_option.is_some(){
+       let mut web_file = file_option.unwrap();
+        for host in web_hosts{
+            let write_res = write!(web_file, "{}\n", host);
+            if write_res.is_err(){
+                let error = write_res.err().unwrap();
+                println!("oooof error writing _hosts.txt!!");
+                println!("{}", error);
+            }
+            else{
+                write_res.unwrap();
+            }
+        }
+    }
+    println!("interesting ports have been written to... writing untagged port files...");
+    outfile.pop();
+    outfile.push("untagged ports");
+    if !outfile.exists(){
+        let untagged_res = create_dir_all(&outfile);
+        if untagged_res.is_err(){
+            let error = untagged_res.err().unwrap();
+            println!("ooof error creating untagged folder!");
+            println!("{}", error);
+        }
+        else{
+            untagged_res.unwrap();
+        }
+    }
+    for line in unknown_ports{
+        let line_vec:Vec<&str> = line.split(":").collect();
+        let host = line_vec[0].to_owned();
+        let port = line_vec[1].to_owned();
+        let file_name = format!("{}_hosts.txt", port);
+        outfile.push(file_name);
+        let write_file_opt = open_append(&outfile);
+        if write_file_opt.is_some(){
+            let mut write_file = write_file_opt.unwrap();
+            let write_res = write!(write_file, "{}\n", host);
+            if write_res.is_err(){
+                let error = write_res.err().unwrap();
+                println!("ooof error writing to file...");
+                println!("{}", error);
+            }
+        }
+        outfile.pop();
+    }
+    println!("DONE all files saved to {}", outfile.display());
+    println!("note if no hosts were found for a protocol their files will be empty.");
+}
+
+#[allow(unused)]
+pub fn run_nmap_portscan(project: &Project) -> Option<JoinHandle<()>>{
+    let notes_folder = project.notes_folder.clone();
+    let mut ports_to_scan = vec![String::from("80"), String::from("443"),
+                                 String::from("161"),
+                                 String::from("22"),
+                                 String::from("21"),
+                                 String::from("23"),
+                                 String::from("8080"),
+                                 String::from("8443"),
+                                 String::from("4433"),
+                                 String::from("135"),
+                                 String::from("445"),
+                                 String::from("3389"),
+                                 String::from("5985"),
+                                 String::from("1433"),
+                                 String::from("3306"),
+                                 String::from("2222"),];
+    let mut general_notes = notes_folder.clone();
+    general_notes.push("general.md");
+    println!("building targets from scope in general notes...");
+    let targets_res = info_controls::get_scope_entries(project);
+    if targets_res.is_none(){
+        return None;
+    }
+    let mut targets = targets_res.unwrap();
+    println!("Got targets from scope!");
+    for target in &targets{
+        println!("{}", target);
+    }
+    if get_user_input("does this list look ok?").to_lowercase().contains("n"){
+        let mut modified_targets = targets.clone();
+        let mut targets_to_remove = Vec::new();
+        let mut targets_ready = false;
+        while !targets_ready{
+            println!("current target list:");
+            for target in &modified_targets{
+                if &targets_to_remove.contains(target) == &false{
+                    println!("{}", target);
+                }
+            }
+            let response = get_user_input("\nc to continue as is\na to add a target to the list\nr to remove a target from the list\nf to start fresh with a blank list of targets");
+            match response.as_str(){
+                "c" => targets_ready = true,
+                "a" => {let new_target = get_user_input("target to add?"); let _ = &mut modified_targets.push(new_target);},
+                "r" => {let _ = targets_to_remove.push(get_user_input("target to remove?"));},
+                "f" => {let _ = &mut modified_targets.clear();}
+                _ => println!("unknown selection"),
+            }
+        }
+        for target in &targets{
+            println!("{}", target);
+        }
+        println!("\nwill be replace with\n");
+        for target in &modified_targets{
+            println!("{}", target);
+        }
+        if get_user_input("continue?").to_lowercase().contains("n"){
+            println!("ok exiting this function, feel free to try again...");
+            return None;
+        }
+        targets = modified_targets;
+    }
+    println!("the standard ports I scan are:");
+    for port in &ports_to_scan{
+        println!("{}", port);
+    }
+    if get_user_input("is this ok?").to_lowercase().contains("n"){
+        let mut ports_ready = false;
+        ports_to_scan.clear();
+        while !ports_ready{
+            let response = get_user_input("enter a port to scan in the nmap format. For example 80-100 or 80,443 Enter END in all caps when you're done.");
+            match response.as_str(){
+                "END" => ports_ready = true,
+                _ => ports_to_scan.push(response),
+            }
+        }
+    }
+    let proxy = get_user_input("will you be using proxychains for this scan?").to_lowercase().contains("y");
+    println!("sweet we have what we need!");
+    println!("building portscan command...");
+    let working_project = project.clone();
+    let mut save_path = project.files_folder.clone();
+    save_path.push("working");
+    if !save_path.exists(){
+        save_path.clear();
+        save_path = project.files_folder.clone();
+    }
+    save_path.push("services.tsv");
+    let mut enumeration_notes_path = project.notes_folder.clone();
+    enumeration_notes_path.push("enumeration.md");
+    println!("{}", save_path.display());
+    let mut nmap_output = Vec::new();
+    let nmap_thread = spawn(move || {
+        if proxy{
+        let port_scancmd_res = Command::new("distrobox")
+                                                .arg("enter")
+                                                .arg("--root")
+                                                .arg(working_project.boxname)
+                                                .arg("--")
+                                                .arg("proxychains")
+                                                .arg("nmap")
+                                                .arg("-sT")
+                                                .arg("-p")
+                                                .arg(ports_to_scan.join(","))
+                                                .arg(targets.join(" "))
+                                                .output();
+            if port_scancmd_res.is_err(){
+                let error = port_scancmd_res.err().unwrap();
+                println!("FROM NMAP THREAD: error running portscan!");
+                println!("{}", error);
+                return;
+            }
+            nmap_output = port_scancmd_res.unwrap().stdout;
+        }
+        else{
+            let port_scancmd_res = Command::new("distrobox")
+                                                .arg("enter")
+                                                .arg("--root")
+                                                .arg(working_project.boxname)
+                                                .arg("--")
+                                                .arg("nmap")
+                                                .arg("-sT")
+                                                .arg("-p")
+                                                .arg(ports_to_scan.join(","))
+                                                .arg(targets.join(" "))
+                                                .output();
+            if port_scancmd_res.is_err(){
+                let error = port_scancmd_res.err().unwrap();
+                println!("FROM NMAP THREAD: error running portscan!");
+                println!("{}", error);
+                return;
+            }
+            nmap_output = port_scancmd_res.unwrap().stdout;
+        } 
+        println!("FROM NMAP THREAD: nmap scan finished, parsing the results...");
+        let nmap_output_string = String::from_utf8_lossy(&nmap_output);
+        let mut nmap_log_path = save_path.clone(); 
+        nmap_log_path.pop();
+        nmap_log_path.push("nmap.log");
+        let nmap_log_file = open_append(&nmap_log_path);
+        if nmap_log_file.is_some(){
+            let mut nmap_log_file = nmap_log_file.unwrap();
+            write!(nmap_log_file, "{}", nmap_output_string).unwrap();
+        }
+        let mut host_ports = Vec::new();
+        let host_sections: Vec<&str> = nmap_output_string.split("Nmap scan report ").collect();
+        for section in host_sections{
+            let mut host = String::new();
+            let mut ports = Vec::new();
+            let lines: Vec<&str> = section.split("\n").collect();
+            let mut port_section = false;
+            for line in lines{
+                if line.contains("for "){
+                    if !line.contains("(not scanned)"){
+                        host = line.split(" ").collect::<Vec<&str>>()[1..].join(" ").to_owned();
+                    }
+                }
+                else if line.contains("Host is up"){
+                    port_section = true;
+                }
+                else if port_section{
+                    if line.contains("open"){
+                        let words: Vec<&str> = line.split_whitespace().collect();
+                        if words.len() >= 3{
+                            let port = words [0];
+                            let service = words [2];
+                            let port_entry = format!("{}:{}", port, service);
+                            ports.push(port_entry);
+                        }
+                    }
+                }
+            }
+            for port in ports{
+                let entry_words: Vec<&str> = port.split(":").collect();
+                let port = entry_words[0];
+                let service = entry_words[1];
+                let host_entry = format!("{}\t{}\t{}", host, port, service);
+                host_ports.push(host_entry);
+            }
+        }
+        let services_file_open_res = open_overwrite(&save_path);
+        if services_file_open_res.is_none(){
+            println!("FROM NMAP THREAD: error opening the services.tsv file to write!");
+            println!("FROM NMAP THREAD: exiting");
+            return;
+        }
+        let mut services_file = services_file_open_res.unwrap();
+        let enumeration_open_res = open_append(&enumeration_notes_path);
+        if enumeration_open_res.is_none(){
+            println!("error opening enumeration notes file!");
+            println!("scan data will not be saved to enumeration notes!");
+        }
+        let services_write_res = write!(services_file, "host\tport\tbanner\tnotes\n");
+        if services_write_res.is_err(){
+            let error = services_write_res.err().unwrap();
+            println!("FROM NMAP THREAD: error writing to the services.tsv file!");
+            println!("{}", error);
+            return;
+        }
+        let mut enumeration_write = false;
+        if enumeration_open_res.is_some(){
+            enumeration_write = true;
+        }
+        let mut host_all_ports: HashMap<String, Vec<String>> = HashMap::new();
+        for host in &host_ports{
+            write!(services_file, "{}\n", host).unwrap();
+            let host_data: Vec<&str> = host.split_whitespace().collect();
+            let address = host_data[0].to_owned();
+            let port = host_data[1].to_owned();
+            let service = host_data[2..].join(" ");
+            if host_all_ports.contains_key(&address){
+                host_all_ports.get_mut(&address).unwrap().push(format!("{}:{}", port, service));
+            }
+            else {
+                host_all_ports.insert(address, vec![format!("{}:{}", port, service)]);
+            }
+        }
+        if enumeration_write{
+            let mut enumeration_file = enumeration_open_res.unwrap();
+            let write_res = write!(enumeration_file, "\n# Port Scan\n");
+            if write_res.is_ok(){
+                write_res.unwrap();
+                for host in host_all_ports.keys(){
+                    write!(enumeration_file, "## {}\n| HOST | PORT | SERVICE |\n| ---- | ---- | ------- |\n", host).unwrap();
+                    for port_entry in &host_all_ports[host]{
+                        println!("{}", port_entry);
+                        let parts: Vec<&str> = port_entry.split(":").collect();
+                        let port = parts[0];
+                        let services = parts[1];
+                        write!(enumeration_file, "| {} | {} |\n", port, services).unwrap();
+                    }
+                }
+                write!(enumeration_file, "---\n").unwrap();
+            }
+        }
+        println!("FROM NMAP THREAD: Parsing done! You're scan results are saved in cobalt strike services.tsv format at {}", save_path.display());
+    });
+    sleep(Duration::from_secs(10));
+    return Some(nmap_thread);
+}
+
+pub fn build_nmap_command(project: &Project){
+    fn get_targets() -> Vec<String>{
+        let mut targets = Vec::new();
+        println!("please enter the ranges/ips to scan one per line, and enter END all caps when done.");
+        loop {
+            let response = get_user_input("ip or range to add?");
+            if response == "END".to_owned(){
+                break;
+            }
+            else{
+                targets.push(response);
+            }
+        }
+        return targets;
+    }
+    let targets_res = info_controls::get_scope_entries(project);
+    let mut targets = Vec::new();
+    let mut ports_to_scan = vec![String::from("80"), String::from("443"),
+                                 String::from("161"),
+                                 String::from("22"),
+                                 String::from("21"),
+                                 String::from("23"),
+                                 String::from("8080"),
+                                 String::from("8443"),
+                                 String::from("4433"),
+                                 String::from("135"),
+                                 String::from("445"),
+                                 String::from("3389"),
+                                 String::from("5985"),
+                                 String::from("1433"),
+                                 String::from("3306"),
+                                 String::from("2222"),];
+    let mut save_path = project.files_folder.clone();
+    if targets_res.is_none(){
+        println!("couldn't get target list from your notes!");
+        targets = get_targets();
+    }
+    else{
+        println!("got targets from the cope in notes!");
+        for target in targets_res.unwrap(){
+            targets.push(target);
+        }
+    }
+    for target in &targets{
+        println!("{}", target);
+    }
+    if get_user_input("is this ok?").to_lowercase().contains("n"){
+        println!("oooof ok, we'll have you recreate it manually.");
+        targets = get_targets();
+    }
+    println!("These are the ports we're going to scan.");
+    for port in &ports_to_scan{
+        println!("{}", port);
+    }
+    if get_user_input("is this ok?").to_lowercase().contains("n"){
+        println!("oof ok, rebuild it manually.");
+        println!("please enter the ports you want to scan, one per line, enter END in all caps when you're finished.");
+        ports_to_scan.clear();
+        loop{
+            let port = get_user_input("port to add?");
+            if port.contains("END"){
+                break;
+            }
+            else{
+                ports_to_scan.push(port);
+            }
+        }
+    }
+    println!("we are going to save the output to {}/working/nmap_output.txt", save_path.display());
+    if get_user_input("is this ok").to_lowercase().contains("n"){
+        println!("oof ok.");
+        save_path.clear();
+        save_path.push(get_user_input("full path to where you want to save it then?"));
+    }
+    else{
+        save_path.push("working/nmap_output.txt");
+    }
+    let ports_string = ports_to_scan.join(",");
+    let targets_string = targets.join(" ");
+    println!("\nYour portscan command is:");
+    if get_user_input("will you be using proxychains for this scan?").to_lowercase().contains("y"){
+        println!("\n\nproxychains nmap -sT -p {} {} -Pn | tee {}", ports_string, targets_string, save_path.display());
+    }
+    else{
+        println!("nmap -p {} {} -Pn | tee {}", ports_string, targets_string, save_path.display());
+    }
+}

pentest_tool/src/project_controls.rs

@@ -4,12 +4,16 @@ use std::io::stdin;
 use std::io::Write;
 use std::path::PathBuf;
 use std::process;
+use std::process::Command;
 use std::thread;
 use std::time::Duration;
 use std::str::FromStr;
+use crate::get_user_input;
 use crate::Project;
+use crate::box_controls::make_box;
 
-pub fn switch_project(projects: &mut Vec<Project>){
+pub fn 
+switch_project(projects: &mut Vec<Project>){
     for project in projects.clone(){
         if project.active == false{
             println!("{} {}|{}", project.id, project.customer, project.project_name);
@@ -30,7 +34,7 @@ pub fn switch_project(projects: &mut Vec<Project>){
                 project.active = false;
             }
             else{
-                println!("error unknown project id")
+                println!("error unknown project id");
             }
         }
     }
@@ -46,30 +50,63 @@ pub fn save_projects(projects: &Vec<Project>, config_path: &PathBuf){
     save_file_path.pop();
     save_file_path.push("projects.conf");
     let mut save_file = fs::File::create(save_file_path).expect("error creating save_file");
-    save_file.write_all(b"customer:name:notes:files:active:time:box_name\n").expect("error writing first line to file");
+    save_file.write_all(b"customer:name:notes:files:active:time:box_name:stage\n").expect("error writing first line to file");
     for project in projects{
         let default = format!{"{}:{}:{}:{}:", project.customer, project.project_name, project.notes_folder.display(), project.files_folder.display()};
         let mut _outline = String::new();
         if project.active{
             if active_set == false{
-                _outline = format!("{}yes:{}\n", default, project.boxname);
+                _outline = format!("{}yes:{}:{}\n", default, project.boxname, project.stage);
                 active_set = true;
             }
 
         }
         else{
-            _outline = format!("{}no:{}\n", default, project.boxname);
+            _outline = format!("{}no:{}:{}\n", default, project.boxname, project.stage);
         }
         save_file.write_all(_outline.as_bytes()).expect("error writing outline");
     }
 }
 
-pub fn new_project(projects: &mut Vec<Project>, project_dir: &PathBuf, notes_dir: &PathBuf, tools_dir: &PathBuf, boxtemplate: &String, config_path: &PathBuf, new_id: i32){
+#[allow(unused)]
-    let mut new_project_dir = project_dir.clone();
+pub fn new_project(projects: &mut Vec<Project>, project_dir: &PathBuf, notes_dir: &PathBuf, tools_dir: &PathBuf, boxtemplate: &String, config_path: &PathBuf, new_id: i32, upcoming_files: &PathBuf, upcoming_notes: &PathBuf, fingerprint: bool){
-    let mut new_note_dir = notes_dir.clone();
+    let mut new_project_dir = PathBuf::new();
+    let mut new_note_dir = PathBuf::new();
     let mut existing_folders = String::new();
     let mut customer_name = String::new();
     let mut project_name = String::new();
+    let mut project_stage = String::new();
+    loop{
+        let mut stage_response = String::new();
+        println!("what stage is this project in?");
+        print!("
+1.) current
+2.) upcoming
+");
+        let stage_result = stdin().read_line(&mut stage_response);
+        if stage_result.is_err(){
+            println!("we need input here dummy, try again...");
+        }
+        else{
+            match &stage_response.trim_end(){
+                &"1" => {project_stage = "current".to_owned(); break;},
+                &"2" => {project_stage = "upcoming".to_owned(); break;},
+                _ => println!("unknown option, try again...")
+            }
+        }
+    }
+    if project_stage.contains("current"){
+        new_project_dir = project_dir.clone();
+        new_note_dir = notes_dir.clone();
+    }
+    else if project_stage.contains("upcoming"){
+        new_project_dir = upcoming_files.clone();
+        new_note_dir = upcoming_notes.clone();
+    }
+    else{
+        println!("unknown stage!!")
+    }
+    println!("{}", new_project_dir.display());
     println!("do you have an existing notes and folder structure to copy over?\ny/n");
     std::io::stdin().read_line(&mut existing_folders).unwrap();
     if existing_folders.contains("y") || existing_folders.contains("Y"){
@@ -183,54 +220,6 @@ pub fn new_project(projects: &mut Vec<Project>, project_dir: &PathBuf, notes_dir
     }
     thread::sleep(Duration::from_secs(2));
     let box_name = format!("atarchbox_{}", customer_name);
-    let mut box_name_path = new_project_dir.clone();
-    box_name_path.push("boxname");
-    let mut box_name_file = fs::File::create(box_name_path).expect("Error creating box name file");
-    box_name_file.write_all(&box_name.as_bytes()).expect("error writing boxname to box file");
-    let pentest_volume = format!("{}:/pentest:rw", new_project_dir.display());
-    let toold_volume = format!("{}:/tools:rw", tools_dir.display());
-    println!("distrobox create --root --clone {} --volume {} --volume {} --name {}", boxtemplate, toold_volume,  pentest_volume, box_name);
-    let distrobox_result = process::Command::new("distrobox")
-                                                                .arg("create")
-                                                                .arg("--root")
-                                                                .arg("--clone")
-                                                                .arg(boxtemplate)
-                                                                .arg("--volume")
-                                                                .arg(&toold_volume)
-                                                                .arg("--volume")
-                                                                .arg(&pentest_volume)
-                                                                .arg("--name")
-                                                                .arg(&box_name)
-                                                                .status()
-                                                                .expect("error getting distrobox status");
-    if distrobox_result.success(){
-        println!("we made a distrobox oh boy!");
-        let distrobox_start_result = process::Command::new("distrobox")
-                                                                        .arg("enter")
-                                                                        .arg("--root")
-                                                                        .arg(&box_name)
-                                                                        .arg("--")
-                                                                        .arg("sudo")
-                                                                        .arg("-s")
-                                                                        .arg("ln")
-                                                                        .arg("-sf")
-                                                                        .arg("/pentest/boxname")
-                                                                        .arg("/etc/boxname")
-                                                                        .status()
-                                                                        .expect("error getting response from distrobox start");
-        if distrobox_start_result.success(){
-            println!("distrobox was started as well!!!! good job me!");
-        }
-        else{
-            println!("ooof did not start successfully try entering it yoruself");
-            println!("distrobox enter --rrot {} -- sudo -s ln -sf /pentest/boxname /etc/boxname", &box_name);
-        }
-    }
-    else{
-        println!("ooof distrobox did not work.... try creating it yourself");
-        println!("distrobox create --root --clone {} --volume {} --volume {} --name {}", boxtemplate, &toold_volume, &pentest_volume, &box_name);
-        println!("distrobox enter --rrot {} -- sudo -s ln -sf /pentest/boxname /etc/boxname", &box_name);
-    }
     let new_project = Project{customer: customer_name.trim_end().to_owned(), 
                                     project_name: project_name.trim_end().to_owned(), 
                                     notes_folder: new_note_dir, 
@@ -238,7 +227,11 @@ pub fn new_project(projects: &mut Vec<Project>, project_dir: &PathBuf, notes_dir
                                     active: false,
                                     id: new_id,
                                     boxname: box_name,
+                                    stage: project_stage.to_owned()
                                       };
+    if project_stage.contains("current"){
+        make_box(&new_project, &tools_dir, &boxtemplate, true, fingerprint);
+    }
     projects.push(new_project);
     save_projects(projects, config_path);
 
@@ -290,18 +283,27 @@ pub fn remove_project(projects: &mut Vec<Project>, config_path: &PathBuf){
             
         }
         else{
-            println!("we need user in put here dummy!!");
+            println!("no project selected! returning...");
         }
     }
     else{
-        println!("we need input here dummy!");
+        println!("no project selected! returning...");
     }
 }
 
-pub fn get_projects(config_path: &PathBuf) -> Vec<Project>{
+pub fn get_projects(config_path: &PathBuf, show: bool) -> Vec<Project>{
     let mut mut_config_path = config_path.clone();
     mut_config_path.pop();
     mut_config_path.push("projects.conf");
+    let mut bkup_config_path = mut_config_path.clone();
+    bkup_config_path.pop();
+    bkup_config_path.push("projects.conf.bkup");
+    let bkup_result = fs::copy(&mut_config_path, &bkup_config_path);
+    if bkup_result.is_err(){
+        let error = bkup_result.err().unwrap();
+        println!("error backing up the projects.conf file!");
+        println!("error: {}", error);
+    }
     let mut projects = Vec::new();
     let projects_string = fs::read_to_string(mut_config_path).expect("error reading projects file");
     let project_lines:Vec<&str> = projects_string.split("\n").collect();
@@ -312,6 +314,12 @@ pub fn get_projects(config_path: &PathBuf) -> Vec<Project>{
         if first != 1{
             if line.len() > 1{
                 let settings: Vec<&str> = line.split(":").collect();
+                //debug config file...
+                /*let mut count = 0;
+                for settin in &settings{
+                    println!("{}: {}", count, settin);
+                    count = count + 1;
+                }*/
                 let customer = settings[0].to_owned();
                 let project = settings[1].to_owned();
                 let notes_string = settings[2].to_owned();
@@ -327,11 +335,132 @@ pub fn get_projects(config_path: &PathBuf) -> Vec<Project>{
                         active = true;
                     }
                 }
-                let new_project = Project{customer: customer, project_name: project, files_folder: project_folder, notes_folder: notes_folder, active: active, id: first, boxname: boxname};
+                let project_stage = settings[6].to_owned();
+                let new_project = Project{customer: customer, project_name: project, files_folder: project_folder, notes_folder: notes_folder, active: active, id: first, boxname: boxname, stage: project_stage};
+                if show{
                     println!("{} {} LOADED!", &new_project.customer, &new_project.project_name);
+                }
                 projects.push(new_project);
             }
         }
     }
     return projects
 }
+
+pub fn print_upcoming_projects(projects: &Vec<Project>){
+    for project in projects{
+        if project.stage.contains("upcoming"){
+            println!("{}:{}", project.customer, project.project_name);
+        }
+    }
+}
+
+pub fn promote_project(projects: &mut Vec<Project>, config_path: &PathBuf, project_dir: &PathBuf, notes_dir: &PathBuf, tools_dir: &PathBuf, boxtemplate: &String, fingerprint: bool){
+    let working_projects = projects.clone();
+    for project in &working_projects{
+        if project.stage.contains("upcoming"){
+            println!("{}.) {}:{}", project.id, project.customer, project.project_name);
+        }
+    }
+    println!("which project to promote?");
+    let mut selection = String::new();
+    let result = stdin().read_line(&mut selection);
+    if result.is_err(){
+        println!("we need input here dummy try again....");
+    } 
+    result.unwrap();
+    println!("{}", project_dir.display());
+    let promote_id: i32 = selection.trim_end().parse().unwrap(); 
+    let mut projects_to_save = Vec::new();
+    for project in &working_projects{
+        if project.id == promote_id{
+            let mut promoted_project = project.clone();
+            let mut new_files_dir = project_dir.clone();
+            let mut new_notes_dir = notes_dir.clone();
+            new_files_dir.push(&promoted_project.customer);
+            new_notes_dir.push(&promoted_project.customer);
+            fs::create_dir_all(&new_files_dir).unwrap();
+            fs::create_dir_all(&new_notes_dir).unwrap();
+            let folder_move_success = process::Command::new("mv")
+                                                .arg("-i")
+                                                .arg(&project.files_folder)
+                                                .arg(&new_files_dir.display().to_string())
+                                                .status().expect("unable to call the system mv command");
+            let note_move_success = process::Command::new("mv")
+                                                .arg("-i")
+                                                .arg(&project.notes_folder)
+                                                .arg(&new_notes_dir.display().to_string())
+                                                .status().expect("unable to call the system mv command");
+            if folder_move_success.success(){
+                println!("we copied the project folder correctly!!");
+                let mut remove_folder = PathBuf::new();
+                remove_folder.push(&project.files_folder);
+                remove_folder.pop();
+                let remove_files_res = fs::remove_dir_all(remove_folder);
+                if remove_files_res.is_err(){
+                    println!("error removing the original files folder form the upcomming folder, manual cleanup required");
+                }
+                else{
+                    remove_files_res.unwrap();
+                    println!("upcoming files folder cleanup successful!");
+                }
+            }
+            else{
+                println!("failed to copy the project folder, try to move it manually!");
+            }
+            if note_move_success.success(){
+                println!("we copied the notes folder correctly!!");
+                let mut remove_folder = PathBuf::new();
+                remove_folder.push(&project.files_folder);
+                remove_folder.pop();
+                let remove_notes_res = fs::remove_dir_all(remove_folder);
+                if remove_notes_res.is_err(){
+                    println!("error removing the original notes folder form the upcomming folder, manual cleanup required");
+                }
+                else{
+                    remove_notes_res.unwrap();
+                    println!("upcoming notes folder cleanup successful!");
+                }
+            }
+            else{
+                println!("failed to copy the notes folder, try to move it manually!");
+            }
+            new_files_dir.push(&promoted_project.project_name);
+            new_notes_dir.push(&promoted_project.project_name);
+            promoted_project.files_folder = new_files_dir;
+            promoted_project.notes_folder = new_notes_dir;
+            promoted_project.stage = "current".to_owned();
+            let cs_response = get_user_input("will you need to be using cobalt strike for this project?");
+            if cs_response.to_lowercase().contains("y"){
+                let cs_path = get_user_input("path to your current cobalt strike directory?");
+                let copy_result = Command::new("cp").arg("-R").arg(&cs_path).arg(&promoted_project.files_folder).status();
+                if copy_result.is_err(){
+                    println!("oof we had an error copying... you'll have to copy this manually");
+                }
+                else{
+                    let copy_exit = copy_result.unwrap();
+                    if copy_exit.success() == false{
+                        println!("oof we had an error copying... you'll have to copy this manually");
+                        println!("run cp -R {} {}", &cs_path, &promoted_project.files_folder.display());
+                    }
+                }
+            }
+            thread::sleep(Duration::from_secs(3));
+            make_box(&promoted_project, tools_dir, boxtemplate, true, fingerprint);
+            projects_to_save.push(promoted_project);
+        }
+        else{
+            projects_to_save.push(project.clone());
+        }
+    }
+    projects.clear();
+    projects.append(&mut projects_to_save);
+    save_projects(&projects_to_save, config_path);
+}
+
+pub fn list_projects(projects: &Vec<Project>){
+    println!("+++++++++++++++++++++"); 
+    for project in projects{
+        println!("++Customer: {}|Project name: {}|Stage: {}++",project.customer ,project.project_name, project.stage)}
+    println!("++++++++++++++++++++")
+}

pentest_tool/src/start_pentest.rs

@@ -1,25 +1,59 @@
 use std::fs;
-use std::io::stdin;
+use std::fs::File;
 use std::io::Write;
 use std::path::PathBuf;
+use crate::Project;
+use crate::project_controls;
+use crate::get_user_input;
 
-fn external(project_folder_path: String,comapny_name: &String, project_name: &String, passtemp: PathBuf){
+
-    let mut general_notes = fs::File::create(format!("{}/general.md", &project_folder_path)).expect("error creating general notes file");
+fn create_project_folder(path: &mut PathBuf, folder: &str){
-    let mut attack_notes = fs::File::create(format!("{}/attacks.md", &project_folder_path)).expect("error creating attack notes file");
+    path.push(folder);
-    let mut host_notes = fs::File::create(format!("{}/host_notes.md", &project_folder_path)).expect("error creating host notes file");
+    let result = fs::create_dir_all(&path);
-    let mut finding_notes = fs::File::create(format!("{}/findings.md", &project_folder_path)).expect("error creating findings notes file");
+    if result.is_err(){
+        println!("error creating {} directory!", folder);
+    }
+    else{
+        result.unwrap();
+    }
+    path.pop();
+}
+
+fn create_note_file(path: &PathBuf) -> Option<File>{
+    let result = fs::File::create(path);
+    if result.is_err(){
+        println!("error creating {} try manually!", path.display());
+        return None;
+    }
+    else{
+        let file = result.unwrap();
+        return Some(file);
+    } 
+}
+
+fn external(passtemp: &PathBuf, project: &Project){
+    // using a pathbuf to create files.
+    let mut notes_path = project.notes_folder.clone();
+    let file_creation_res = fs::create_dir_all(&notes_path);
+    if file_creation_res.is_err(){
+        let error = file_creation_res.err().unwrap();
+        println!("error creating notes folder! {}", error);
+    }
+    else{
+        file_creation_res.unwrap();
+    }
+    notes_path.push("general.md");
+    let general_notes_result = create_note_file(&notes_path);
+    if general_notes_result.is_some(){
+        let mut general_notes = general_notes_result.unwrap();
         // for tagging
-    let year = project_name.split("_").collect::<Vec<&str>>()[1];
         let project_type = "External";
-    writeln!(&mut general_notes, "#{} #{} #{} #general", comapny_name, project_type, year).expect("error writing to general notes file");
+        writeln!(&mut general_notes, "#{} #{} #general", project.customer, project_type).expect("error writing to general notes file");
-    writeln!(&mut attack_notes, "#{} #{} #{} #attack", comapny_name, project_type, year).expect("error writing tags on attack notes");
-    writeln!(&mut host_notes, "##{} #{} #{} #host_notes", comapny_name, project_type, year).expect("error writing tag lin in host notes");
-    writeln!(&mut finding_notes, "#{} #{} #{} #findings", comapny_name, project_type, year).expect("error writing tag line on findings");
         writeln!(&mut general_notes, "# Scope").expect("error writing to general notes file");
         writeln!(&mut general_notes, "\n| IP | Third Party | Approval |").expect("error writing to general notes file");
         writeln!(&mut general_notes, "| -- | ----------- | -------- |").expect("error writing to general notes file");
-writeln!(&mut general_notes, "# PPC").expect("failed to write general notes");
+        writeln!(&mut general_notes, "# PPC").expect("failed to write general notes");
-write!(&mut general_notes, "
+        write!(&mut general_notes, "
 Planning call notes:
 * methodolgy
 * whole month testing window
@@ -41,21 +75,16 @@ Planning call notes:
 * password attacks
     * password sprays 3-12 hours
     * lock out policy - 
+|\n").expect("faile to write pentest notes");
 
-## Enumeration in progress
+    }
-
+    notes_path.pop();
-### DNS Bruteforcing
+    notes_path.push("attacks.md");
-
+    let attack_notes_result = create_note_file(&notes_path);
-| domain name | IP | inscope? |
+    if attack_notes_result.is_some(){
-| ----------- | -- | -------- |
+        let mut attack_notes = attack_notes_result.unwrap();
-
+        writeln!(&mut attack_notes, "#{} #{} #attack", project.customer, "external").expect("error writing tags on attack notes");
-# Attack Surface Enumeration
+        write!(&mut attack_notes,"
-
-| IP | host notes | needs? |
-| -- | ---------- | ------ |\n").expect("faile to write pentest notes");
-
-// set up the basics for our attack notes
-write!(&mut attack_notes,"
 # Directory Bruteforcing
 - [ ] example.com
 
@@ -74,39 +103,61 @@ write!(&mut attack_notes,"
 passwords tried:
 * password\n
 " ).expect("failed to write attack notes template");
-let new_spray_path = format!("{}/passwordspray.md", &project_folder_path);
+    }
-fs::copy(passtemp,new_spray_path).unwrap();
+    notes_path.pop();
+    notes_path.push("password_spray.md");
+    let pass_result = fs::copy(&passtemp, &notes_path);
+    if pass_result.is_err(){
+        println!("error copying password spray file! try manually");
+    }
+    else{
+        pass_result.unwrap();
+    }
+    notes_path.pop();
+    notes_path.push("host_notes.md");
+    let host_notes_result = create_note_file(&notes_path);
+    if host_notes_result.is_some(){
+        let mut host_notes = host_notes_result.unwrap();
+        writeln!(&mut host_notes, "##{} #{} #host_notes", project.customer, "external").expect("error writing tag lin in host notes");
+    }
+    notes_path.pop();
+    notes_path.push("findings.md");
+    let findings_notes_result = create_note_file(&notes_path);
+    if findings_notes_result.is_some(){
+        let mut finding_notes = findings_notes_result.unwrap();
+        writeln!(&mut finding_notes, "#{} #{} #findings", project.customer, "external").expect("error writing tag line on findings");
+    }
+
 }
 
-fn internal(project_folder_path: String, comapny_name: &String, project_name: &String, passtemp: PathBuf){
+fn internal(passtemp: &PathBuf, project: &Project){
-    let loot_folder = format!("{}/l00t", project_folder_path);
+    let mut notes_path = project.notes_folder.clone();
-    fs::create_dir_all(&loot_folder).expect("error creating loot directory");
+    let file_creation_res = fs::create_dir_all(&notes_path);
-    let mut general_notes = fs::File::create(format!("{}/general.md", &project_folder_path)).expect("error creating general notes file");
+    if file_creation_res.is_err(){
-    let mut attack_notes = fs::File::create(format!("{}/attacks.md", &project_folder_path)).expect("error creating attack notes file");
+        let error = file_creation_res.err().unwrap();
-    let mut finding_notes = fs::File::create(format!("{}/findings.md", &project_folder_path)).expect("error creating findings notes file");
+        println!("error creating notes folder! {}", error);
-    let mut systeminfo = fs::File::create(format!("{}/systeminfo", &project_folder_path)).expect("error creating systeminfo note file");
+    }
-    let mut netsta = fs::File::create(format!("{}/netstat", &project_folder_path)).expect("error creating netstat file");
+    else{
-    let mut creds_notes = fs::File::create(format!("{}/creds.md", &loot_folder)).expect("error creating creds note");
+        file_creation_res.unwrap();
-    let mut todo_notes = fs::File::create(format!("{}/todo.md", &project_folder_path)).expect("error creating todo notes");
+    }
-    let mut cleanup_notes = fs::File::create(format!("{}/cleanup.md", &project_folder_path)).expect("error creating cleanup notes");
+    notes_path.push("password_spray.md");
-    let mut dump_notes = fs::File::create(format!("{}/dumps.md", &loot_folder)).expect("error creating password spray notes");
+    println!("copying from {} to {}", passtemp.display(), &notes_path.display());
-    let mut enum_notes = fs::File::create(format!("{}/initial_enum.md", &project_folder_path)).expect("error creating password spray notes");
+    let pass_result = fs::copy(&passtemp, &notes_path);
-    let enum_obsidian_path = format!("Hack_Notes/pentest_notes/upcomming/{customer}//initial_enum.md", customer = comapny_name);
+    if pass_result.is_err(){
-    // for tagging notes
+        let error = pass_result.err().unwrap();
-    let oyear = project_name.split("_").collect::<Vec<&str>>()[0];
+        println!("error copying password spray file, try again manually! {}", error);
-    let year = format!("year-{}", oyear);
+    }
-    let project_type = "Internal";
+    else{
-    writeln!(&mut creds_notes, "#{} #{} #{} #l00t #creds", comapny_name, project_type, year).expect("error writing creds notes"); 
+        pass_result.unwrap();
-    writeln!(&mut dump_notes, "#{} #{} #{} #l00t #dumps", comapny_name, project_type, year).expect("error writing creds notes");
+    }
-    writeln!(&mut cleanup_notes, "#{} #{} #{} #cleanup", comapny_name, project_type, year).expect("error writing to cleanup notes");
+    notes_path.pop();
-    writeln!(&mut general_notes, "#{} #{} #{} #general", comapny_name, project_type, year).expect("error writing to general notes file");
+    notes_path.push("l00t");
-    writeln!(&mut attack_notes, "#{} #{} #{} #attack", comapny_name, project_type, year).expect("error writing attack note tags");
+    fs::create_dir_all(&notes_path).expect("error creating loot directory");
-    writeln!(&mut todo_notes,  "#{} #{} #{} #todo", comapny_name, project_type, year).expect("error writing tag line on todo");
+    notes_path.push("creds.md");
-    writeln!(&mut finding_notes, "#{} #{} #{} #findings", comapny_name, project_type, year).expect("error writing tags line on findings");
+    let creds_notes_result = create_note_file(&notes_path);
-    writeln!(&mut systeminfo, "#{} #{} #{} #general", comapny_name, project_type, year).expect("error writing tag line for system info");
+    if creds_notes_result.is_some(){
-    writeln!(&mut systeminfo, "#{} #{} #{} #Password_sprays", comapny_name, project_type, year).expect("error writing tag line for password spraying");
+        let mut creds_notes = creds_notes_result.unwrap();
-    writeln!(&mut netsta, "#{} #{} #{} #general", comapny_name, project_type, year).expect("error writing tagline in the netstat file");
+        writeln!(&mut creds_notes, "#{} #{} #l00t #creds", project.customer, "internal").expect("error writing creds notes"); 
-    writeln!(&mut enum_notes, "#{} #{} #{} #enum", comapny_name, project_type, year).expect("error writing tagline in the netstat file");
         writeln!(&mut creds_notes, "# CREDS").expect("error writing creds notes");
         writeln!(&mut creds_notes, "\n| System | username | password |").expect("error writing creds notes");
         writeln!(&mut creds_notes, "| ------ | -------- | -------- |").expect("error writing creds notes");
@@ -114,18 +165,16 @@ fn internal(project_folder_path: String, comapny_name: &String, project_name: &S
         writeln!(&mut creds_notes, "# HASHES\n\n").expect("error writing creds notes");
         writeln!(&mut creds_notes, "| TYPE | USER | HASH |").expect("error writing creds notes");
         writeln!(&mut creds_notes, "| ---- | ---- | ---- |").expect("error writing creds notes");
-    writeln!(&mut cleanup_notes, "- [ ] Breach machine C-temp-fr").expect("error writing to cleanup notes");
+    } 
-    writeln!(&mut cleanup_notes, "- [ ] (continue to add as needed").expect("error writing ot cleanup notes");
+    notes_path.pop();
+    notes_path.pop();
+    notes_path.push("general.md");
+    let general_result = create_note_file(&notes_path);
+    if general_result.is_some(){
+        let mut general_notes = general_result.unwrap();
+        writeln!(&mut general_notes, "#{} #{} #general", project.customer, "internal").expect("error writing to general notes file");
         writeln!(&mut general_notes, "# Scope\n").expect("error writing to general notes file");
         writeln!(&mut general_notes, "PASTE SCOPE FROM EXCELL HERE (THE EXCEL TO MARKDOWN TABLE PLUGIN WILL FORMAT FOR YOU").expect("shouldn't ever fail");
-    writeln!(&mut dump_notes, "# SAM\n\n").expect("shouldn't ever fail");
-    writeln!(&mut dump_notes, "## system name\n").expect("shouldn't ever fail");
-    writeln!(&mut dump_notes, "```").expect("shouldn't ever fail");
-    writeln!(&mut dump_notes, "```\n\n").expect("shouldn't ever fail");
-    writeln!(&mut dump_notes, "# LSASS\n\n").expect("shouldn't ever fail");
-    writeln!(&mut dump_notes, "## system name\n").expect("shouldn't ever fail");
-    writeln!(&mut dump_notes, "```").expect("shouldn't ever fail");
-    writeln!(&mut dump_notes, "```").expect("shouldn't ever fail");
         write!(&mut general_notes, "
 On the call: 
 
@@ -151,6 +200,14 @@ Do they have a specific contact
 
 Email any follow-up items from the call to the PM 
 ").expect("error writing PPC text");
+    }
+    notes_path.pop();
+    let enum_path = format!("{}/initial_enum.md", notes_path.display());
+    notes_path.push("attacks.md");
+    let attack_result = create_note_file(&notes_path);
+    if attack_result.is_some(){
+        let mut attack_notes = attack_result.unwrap();
+        writeln!(&mut attack_notes, "#{} #{} #attack", project.customer, "internal").expect("error writing attack note tags");
         write!(&mut attack_notes,"
 # current dat (ex: 7/5)
 Got Persistence via (schtasks, bat schtasks, startup folder)
@@ -206,14 +263,28 @@ powerup.ps1/sharpup.exe notes.
 
 
 
-    ", enum = enum_obsidian_path).expect("error writing to attack notes for internal tests");
+    ", enum = enum_path).expect("error writing to attack notes for internal tests");
+    }
+    notes_path.pop();
+    notes_path.push("findings.md");
+    let findings_result = create_note_file(&notes_path);
+    if findings_result.is_some(){
+       let mut finding_notes = findings_result.unwrap();
+       writeln!(&mut finding_notes, "#{} #{} #findings", project.customer, "internal").expect("error writing tags line on findings");
        write!(&mut finding_notes, "
-# normal findings
+       # normal findings
        
-# data exfil
+       # data exfil
-## [Sarting Username]
+       ## [Sarting Username]
        
            ").expect("error writing to findings notes on internal");
+    }
+    notes_path.pop();
+    notes_path.push("todo.md");
+    let todo_result = create_note_file(&notes_path);
+    if todo_result.is_some(){
+        let mut todo_notes = todo_result.unwrap();
+        writeln!(&mut todo_notes,  "#{} #{} #todo", project.customer, "internal").expect("error writing tag line on todo");
         write!(&mut todo_notes, "
 
 - [ ] local checks
@@ -227,93 +298,135 @@ powerup.ps1/sharpup.exe notes.
 - [ ] passwords in AD Descriptions?
 - [ ] password spray
 ").expect("error writing todo list");
-write!(&mut netsta,"
+    }
-```
+    notes_path.pop();
-
+    notes_path.push("cleanup.md");
-```").expect("error writing code block to system info");
+    let cleanup_result = create_note_file(&notes_path);
-write!(&mut systeminfo,"
+    if cleanup_result.is_some(){
-```
+        let mut cleanup_notes = cleanup_result.unwrap();
-
+        writeln!(&mut cleanup_notes, "#{} #{} #cleanup", project.customer, "internal").expect("error writing to cleanup notes");
-```").expect("error writing code block to system info");
+        writeln!(&mut cleanup_notes, "- [ ] Breach machine C-temp-fr").expect("error writing to cleanup notes");
-let new_pass_path =  format!("{}/passwordspray.md", &project_folder_path);
+        writeln!(&mut cleanup_notes, "- [ ] (continue to add as needed").expect("error writing ot cleanup notes");
-println!("{}  |  {}", passtemp.display(), new_pass_path);
+    }
-fs::copy(passtemp, new_pass_path).unwrap();
 }
 
 
-pub fn start_pentest(config_path: &PathBuf) {
+fn vishing(project: &Project){
-    let mut pentest_notes = String::new();
+    let mut notes_path = project.notes_folder.clone();
-    let mut project_files = String::new();
+    let mknote_folder_res = fs::create_dir_all(&notes_path);
-    let mut company_name = String::new();
+    if mknote_folder_res.is_err(){
-    let mut project_name = String::new();
+        let error = mknote_folder_res.err().unwrap();
-    let mut config_file_path_buf = config_path.clone();
+        println!("Error creating notes folder!");
-    let mut passpray_path = config_file_path_buf.clone();
+        println!("{}", error);
-    passpray_path.push("passwordspray.md");
+        return;
-    config_file_path_buf.push("new_projects.conf");
-    let config_string = fs::read_to_string(config_file_path_buf).expect("error reading ");
-    if config_string.contains("folder_path") && config_string.contains("notes_path"){
-        let config_string_vec: Vec<&str> = config_string.split("\n").collect();
-        for line in config_string_vec{
-            if line.contains("project_folder_path"){
-                let line_vec: Vec<&str> = line.split(":").collect();
-                project_files = line_vec[1].to_owned();
     }
-            else if line.contains("project_notes_path"){
+    notes_path.push("general.md");
-                let line_vec: Vec<&str> = line.split(":").collect();
+    let general_notes_res = create_note_file(&notes_path);
-                pentest_notes = line_vec[1].to_owned();
+    if general_notes_res.is_some(){
-            }
+        let mut general_notes = general_notes_res.unwrap();
-        }
+        write!(general_notes, "# Scope\n\n").unwrap();
-    }
+        write!(general_notes, "
-    println!("Project files path: {}\nProject Notes path: {}", project_files, pentest_notes);
+Introductions
-    println!("Comapny Name?");
-    match std::io::stdin().read_line(&mut company_name){
-        Ok(_result) => (),
-        Err(_e) => {println!("we need input here dummy!"); return;}
-    }
-    println!("project Name?");
-    match stdin().read_line(&mut project_name){
-        Ok(_result) => (),
-        Err(_e) => {println!("we need input here dummy!"); return;}
-    }
-    //remove new lines from input
-    company_name = company_name.trim_end().to_owned();
-    project_name = project_name.trim_end().to_owned();
-    let project_folder_path = format!("{}/{}/{}", pentest_notes, company_name,project_name);
-    println!("setting folder creation paths...");
-    let project_files_folder_path = format!("{}/{}/{}", project_files, company_name, project_name);
-    let working_folder = format!("{}/working", &project_files_folder_path);
-    let writing_folder = format!("{}/writing", &project_files_folder_path);
-    let screeenshot_folder = format!("{}/screenshots",&writing_folder);
-    let delivery_folder = format!("{}/delivery", &project_files_folder_path);
-    // make the folders for this project's notes and files
-    println!("creating directory structure for notes and file folders...");
-    fs::create_dir_all(&project_folder_path).expect("Error creating project folder");
-    fs::create_dir_all(&project_files_folder_path).expect("Error creating project file folder");
-    fs::create_dir_all(&working_folder).expect("error creating working directory");
-    fs::create_dir_all(&writing_folder).expect("Error creating writing direcotry");
-    fs::create_dir(&screeenshot_folder).expect("error creating screenshots folder");
-    fs::create_dir_all(&delivery_folder).expect("Error creating delivery direcotry");
-    if project_name.contains("internal"){
-        println!("internal pentest type detected, auto populating notes with internal layout...");
-        internal(project_folder_path,&company_name, &project_name, passpray_path);
-    }
-    else if project_name.contains("external") {
-        println!("external pentest type detected, auto populating notes with external layout...");
-        external(project_folder_path, &company_name, &project_name, passpray_path);
-    }
-    else if project_name.contains("webapp") {
-        println!("not implemented yet sorry");
-        println!("default file folder structure used, and empty notes folder created...")
-    }
-    else{
-        println!("unknown project type detected, default folder strucutre and empty note folder created...");
-        print!("
-Known project types:
-Internal Penetration Test
-External Penetartion Test
 
-If this test is actually one of these types please include the type in the project name parameter, example: 2023_internal_pentest.
+have they been vished before?
-If this test is a common test and you would like a default note structure implemented in this script let Kevin \"Kage\" Gunter know and supply an example markdown note sheet.");
+if yes ask what the purpose of that vishing was, gain a foothold, or other?
-        }
 
+ask the purpose of this test, just see if we can get creds or testing a vetting process, or any specific goals for the engagement.
+
+four main aspects
+1. verbal confirmation and verification of information
+2. run commands on the system they're on
+3. go to a specific website
+4. join a screen sharing session with us
+
+pretexts:
+default is third party it.
+they have 2 dudes for helpdesk so this may not be the best pretext.  Try to impersonate a specific helpdesk user. impersonate William sounds like our plan. 
+
+Vector - 
+
+ask for any questions, comments, or concerns. 
+").unwrap();
+    }
+    notes_path.pop();
+    notes_path.push("findings.md");
+    let findings_res = create_note_file(&notes_path);
+    if findings_res.is_some(){
+        let mut findings_file = findings_res.unwrap();
+        write!(findings_file, "# Findings to write up\n\n\n# Delivery Notes\n").unwrap();
+    }
+    notes_path.pop();
+    notes_path.push("pretext.md");
+    let pretext_res = create_note_file(&notes_path);
+    if pretext_res.is_some(){
+        let mut pretext_file = pretext_res.unwrap();
+        write!(pretext_file, "\n\n\n#Script").unwrap();
+        write!(pretext_file, "
+Hello I'm [name] from [Find Local IT Shop]. We were brought in to help your normal IT guys with some of the menial tasks so they can focus on more import improvement projects. As part of this we're making sure our inventory management system is checking in correctly and up to date, this should only take a minute or two. Is now bad time to talk?
+
+Great I just need to confirm that my inventory report here is accurate.
+
+Are you currently running Windows 11?
+
+can you confirm your user name is [metadata username]?
+
+Your primary browser is firfox?
+
+Oh thats strange it seems our report is wrong then... I don't think our program on your computer is checking in correctly... uhhh I want to make sure you're getting all the windows updates we need to be compliant.
+
+Hold the windows key on your keyboard and press the r button. in the box that opens up type cmd.exe and press enter.
+
+This will open a scary black box, but don't worry I'll walk you through what we need here, it'll be pretty easy.
+
+In that box type systemifo all one word and press enter.
+
+Scroll up through that output and find the section that talks about hotfixes, how many are installed?
+
+That doesn't seem like the right number to me, can you read me the last 3 that are listed there?
+
+yeah you're definitely not getting all of the windows updates. This is going to take a bit of troubleshooting to figure out. Would you mind hopping in a Zoom call with me and sharing your screen so I can check a few things? This should only take a couple of minutes.
+
+(open up the services manager and scroll through it, check some program files folders, and run a few commands in cmd to act like I'm troubleshooting.)
+
+Hmmm everything looks ok on this end. I'm going to do some troubleshooting on the server side and see if we can get to the bottom of this. I don't think we'll need anything else from you to fix this, but if that changes I'll let you know.  Thank you for your time.        
+").unwrap();
+    }
+    notes_path.pop();
+    notes_path.push("calls.md");
+    create_note_file(&notes_path);
+}
+
+pub fn start_pentest(config_path: &PathBuf, projects: &mut Vec<Project>, id: i32, upcoming_files: &PathBuf, upcoming_notes: &PathBuf, boxtemplate: &String, password_spray_file: &PathBuf) {
+    let mut project_files = upcoming_files.clone();
+    let mut project_notes = upcoming_notes.clone();
+    let customer_name = get_user_input("Customer name?");
+    let project_name = get_user_input("Project Name?");
+    project_files.push(&customer_name);
+    project_files.push(&project_name);
+    project_notes.push(&customer_name);
+    project_notes.push(&project_name);
+    println!("Files: {}\nNotes: {}\n\n", project_files.display(), project_notes.display());
+    let confirm_response = get_user_input("does this look ok?");
+    if confirm_response.to_lowercase().contains("n"){
+        println!("oof sorry");
+        return;
+    }
+    let mut working = project_files.clone();
+    create_project_folder(&mut working, "working");
+    create_project_folder(&mut working, "writing");
+    create_project_folder(&mut working, "delivery");
+    let project_boxname = format!("{}_{}", boxtemplate, customer_name);
+    let new_prject = Project{customer:customer_name.clone(), project_name:project_name.clone(), notes_folder:project_notes.clone(), files_folder:project_files.clone(),active:false, boxname:project_boxname.clone(),stage:"upcoming".to_owned(), id};
+    if project_name.to_lowercase().contains("external"){
+        external(password_spray_file, &new_prject);
+    }
+    else if project_name.to_lowercase().contains("internal"){
+        internal(password_spray_file, &new_prject);
+    }
+    else if project_name.to_lowercase().contains("vishing"){
+        vishing(&new_prject);
+    }
+    projects.push(new_prject);
+    project_controls::save_projects(projects, config_path);
+    println!("project created and saved to the projects config file!");
 }

pentest_tool/src/tool_controls.rs

@@ -0,0 +1,41 @@
+use std::{env::set_current_dir, path::PathBuf};
+use std::process::Command;
+use walkdir::WalkDir;
+
+
+pub fn update_git_tools(tools_dir: &PathBuf){
+    let mut folders = Vec::new();
+    for entry in WalkDir::new(tools_dir).max_depth(2){
+        let entry = entry.unwrap();
+        let path = entry.path();
+        if path.is_dir(){
+            folders.push(path.to_owned());
+        }
+    }
+    for folder in folders{
+        let cd_res = set_current_dir(&folder);
+        if cd_res.is_err(){
+            let error = cd_res.err().unwrap();
+            println!("error changing directory!");
+            println!("{}", error);
+        }
+        else{
+            let _cd = cd_res.unwrap();
+            let git_command_res = Command::new("git")
+                                                        .arg("pull")
+                                                        .arg("--autostash")
+                                                        .status();
+            if git_command_res.is_err(){
+                let error = git_command_res.err().unwrap();
+                println!("error running git pull command!");
+                println!("{}", error);
+            }
+            else{
+                let git_command = git_command_res.unwrap();
+                if git_command.success(){
+                    println!("successfully updated {}", folder.display());
+                }
+            }
+        }
+    }
+}

pentest_tool/src/victim_commands.rs

@@ -0,0 +1,32 @@
+use std::path::PathBuf;
+use walkdir::WalkDir;
+use crate::get_user_input;
+
+pub fn sharp_persist_command(tools_dir: &PathBuf){
+    let filename = "SharPersist.exe";
+    let mut binary_path = String::new();
+    for entry in WalkDir::new(tools_dir){
+        let entry = entry.unwrap();
+        let path = entry.path();
+        if path.is_file() && path.file_name().unwrap() == filename{
+            binary_path = path.to_str().unwrap().to_string();
+        }
+    }
+    if get_user_input("will you be running this is cobalt strike?").to_lowercase().contains("y"){
+        let beacon_path = get_user_input("excellent! Please enter the path to your beacon file on the victim machine.");
+        println!("how will you run it?");
+        println!("1.) execute-assembly\n2.) inline-execute-assembly\n3.) from disk");
+        match get_user_input("selection?").as_str(){
+            "1" => println!("execute-assembly {} -t schtask -c \"{}\" -n FRPersist -m add -o hourly", binary_path, beacon_path),
+            "2" => println!("inlineExecute-Assembly --dotnetassembly {} --assemblyargs -t schtask -c \"{}\" -n FRPersist -m add -o hourly --etw --amsi", binary_path, beacon_path),
+            "3" => println!("upload {}\n\nrun .\\SharPersist.exe -t schtask -c \"{}\" -n FRPersist -m add -o hourly", binary_path, beacon_path),
+            _ => println!("unknown selection")
+        }
+        return;
+    }
+    let file_to_run = get_user_input("path to the file you want to run for persistence?");
+    println!("ok then, upload the following file to the machine");
+    println!("{}", binary_path);
+    println!("C:\\path\\to\\your\\sharpersist.exe -t schtask -c \"{}\" -n FRPersist -m add -o hourly", file_to_run);
+    return;
+}

resources/credits.txt

@@ -0,0 +1 @@
+https://www.soundjay.com/misc/small-bell-ring-01a.mp3