pyro/pentest_tool
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update main.rs

Browse Source 
added the functionality to run my initial recon script to import nessus findings from external scans into obsidian
This commit is contained in:
Pyro57000
2024-06-11 20:12:12 +00:00
committed by GitHub
parent ec351f733b
commit 7ba70d03b7
1 changed files with 21 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
pentest_tool/src/main.rs
View File

@@ -473,6 +473,19 @@ fn gnerate_userpass(project: &Project){
} }
} }
fn run_initial_enum(project: &Project){
let mut csv = String::new();
println!("path to the csv?");
std::io::stdin().read_line(&mut csv).unwrap();
let status = process::Command::new("initial_recon").arg(&csv).arg(&project.customer).arg(&project.project_name).status().expect("error running initial_recon program");
if status.success(){
println!("execllent! hosts should be imported to {}/host_notes.md", &project.notes_folder.display());
}
else{
println!("ooof something went wrong, host notes may not have saved correctly!");
}
}
fn main_menu(mut projects: Vec<Project>, config_path: &PathBuf, base_files: &PathBuf, base_notes: &PathBuf, tools_dir: &PathBuf, boxtemplate: String){ fn main_menu(mut projects: Vec<Project>, config_path: &PathBuf, base_files: &PathBuf, base_notes: &PathBuf, tools_dir: &PathBuf, boxtemplate: String){
let mut loopize = true; let mut loopize = true;
@@ -545,9 +558,10 @@ Current Project: {} {}
10.) Open Project Files Folder In Dolphin 10.) Open Project Files Folder In Dolphin
11.) Open Project Notes Folder In Dolphin 11.) Open Project Notes Folder In Dolphin
12.) generate userpass file from your obsidian notes 12.) generate userpass file from your obsidian notes
13.) Print Project Info For Report 13.) run pyro's initail enum script on a nessus csv for the current project
14.) Stop All Distroboxes 14.) Print Project Info For Report
15.) Quit Application 15.) Stop All Distroboxes
16.) Quit Application
\n", active_project.customer, active_project.project_name); \n", active_project.customer, active_project.project_name);
std::io::stdin().read_line(&mut response).expect("error getting menu input"); std::io::stdin().read_line(&mut response).expect("error getting menu input");
clear().expect("error clearing screen"); clear().expect("error clearing screen");
@@ -567,9 +581,10 @@ Current Project: {} {}
"10" => open_in_dolphin("files", active_project.clone()), "10" => open_in_dolphin("files", active_project.clone()),
"11" => open_in_dolphin("notes", active_project.clone()), "11" => open_in_dolphin("notes", active_project.clone()),
"12" => gnerate_userpass(&active_project), "12" => gnerate_userpass(&active_project),
"13" => print_report_information(active_project.clone()), "13" => run_initial_enum(&active_project),
"14" => stop_all_boxes(&projects), "14" => print_report_information(active_project.clone()),
"15" => {save_projects(&projects, &config_path); "15" => stop_all_boxes(&projects),
"16" => {save_projects(&projects, &config_path);
let mut stop = String::new(); let mut stop = String::new();
println!("stop all boxes?\ny/n"); println!("stop all boxes?\ny/n");
std::io::stdin().read_line(&mut stop).unwrap(); std::io::stdin().read_line(&mut stop).unwrap();