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

udated the exit operations to only prompt for

Browse Source 
work vs personal if work and personal configs
exist.
This commit is contained in:
Pyro57000
2025-06-04 09:50:13 -05:00
parent 6d92032d06
commit e211433772
2 changed files with 61 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
pentest_tool/src/cli.rs
View File

@@ -11,6 +11,7 @@ use chrono::Local;
use colored::Colorize;
use crate::print_error;
use crate::print_informational;
use crate::print_success;
use crate::Project;
use crate::project_controls;
use crate::box_controls;
@@ -265,6 +266,7 @@ fn print_banner(banner: &str){
print!("{}", banner.custom_color((255,165,0)));
}
#[allow(unused)]
pub fn cli(interactive: bool,
mut projects: Vec<Project>,
config_path: PathBuf,
@@ -376,31 +378,39 @@ for help enter help or ?. for information about a specific command enter help (c
let mut workspace_config_path = config_path.clone();
workspace_config_path.pop();
let mut project_conf_path = config_path.clone();
let mut success_message = String::new();
project_conf_path.pop();
project_conf_path.push("projects.conf");
if get_user_input("are your work projects currently loaded? (not yoru personal projects...)").to_lowercase().contains("y"){
workspace_config_path.push("projects.work");
}
else{
workspace_config_path.push("projects.personal");
}
let open_res = OpenOptions::new().create(true).write(true).open(workspace_config_path);
if open_res.is_err(){
print_error("error opeing workspace config file!", open_res.err().unwrap().to_string());
}
else{
let mut workspace_config_file = open_res.unwrap();
let projects_read_res = read_to_string(project_conf_path);
if projects_read_res.is_ok(){
let project_string = projects_read_res.unwrap();
let write_res = write!(workspace_config_file, "{}", project_string);
if write_res.is_err(){
print_error("error writing workspace config file!", write_res.err().unwrap().to_string());
}
project_conf_path.push("projects.work");
if project_conf_path.exists(){
if get_user_input("are your work projects currently loaded? (not yoru personal projects...)").to_lowercase().contains("y"){
workspace_config_path.push("projects.work");
success_message = String::from("projects.conf saved to projects.work");
}
else{
print_error("error reading projects config file!", projects_read_res.err().unwrap().to_string());
workspace_config_path.push("projects.personal");
success_message = String::from("projects.conf saved to projects.personal");
}
let open_res = OpenOptions::new().create(true).write(true).open(workspace_config_path);
if open_res.is_err(){
print_error("error opeing workspace config file!", open_res.err().unwrap().to_string());
}
else{
project_conf_path.pop();
project_conf_path.push("projects.conf");
let mut workspace_config_file = open_res.unwrap();
let projects_read_res = read_to_string(project_conf_path);
if projects_read_res.is_ok(){
let project_string = projects_read_res.unwrap();
let write_res = write!(workspace_config_file, "{}", project_string);
if write_res.is_err(){
print_error("error writing workspace config file!", write_res.err().unwrap().to_string());
}
}
else{
print_error("error reading projects config file!", projects_read_res.err().unwrap().to_string());
}
}
print_success(success_message);
}
if threads.len() > 0{
println!("closing threads...");

32
pentest_tool/src/enumeration.rs
View File

@@ -193,10 +193,38 @@ pub fn bruteforce_subs(project: &Project, given_domains: Option<&Vec<String>>, g
print_informational("sleeping for 10 seconds to allow for sudo password input.");
sleep(Duration::from_secs(10));
let gobuser_output = gobuster_cmd_res.unwrap().stdout;
print_success("Gobuster enumeration Done!");
let gobuster_string = String::from_utf8_lossy(&gobuser_output);
let mut final_string = String::new();
if gobuster_string.contains("specify the '--wildcard' switch"){
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())
.arg("--wildcard")
.output();
if gobuster_cmd_res.is_err(){
let error = gobuster_cmd_res.err().unwrap();
println!("{}","From gobuster thread: Error running gobuster command!".red());
println!("{}", error.to_string().red());
return;
}
let new_gobuser_output = gobuster_cmd_res.unwrap().stdout;
let new_gobuser_string = String::from_utf8_lossy(&new_gobuser_output);
final_string = new_gobuser_string.to_string();
}
else{
final_string = gobuster_string.to_string();
}
print_success("Gobuster enumeration Done!");
let mut domain_names = Vec::new();
let lines: Vec<&str> = gobuster_string.split("\n").collect();
let lines: Vec<&str> = final_string.split("\n").collect();
for line in lines{
if line.contains("Found:"){
let domain = line.split_whitespace().collect::<Vec<&str>>()[1];