diff --git a/poshc2/client/command_handlers/ImplantHandler.py b/poshc2/client/command_handlers/ImplantHandler.py index 3eec8456..e3e8f39a 100644 --- a/poshc2/client/command_handlers/ImplantHandler.py +++ b/poshc2/client/command_handlers/ImplantHandler.py @@ -427,26 +427,30 @@ def do_output_to_html(user, command): clear() +def generate_csv(tableName): + print_good(f"Generating {PoshProjectDirectory}/reports{tableName}.csv") + os.system(f"sqlite3 -header -csv {PoshProjectDirectory}PowershellC2.SQLite 'select * from {tableName};' > {PoshProjectDirectory}reports/{tableName}.csv") + + def do_generate_reports(user, command): generate_table("Tasks") generate_table("C2Server") generate_table("Creds") generate_table("Implants") graphviz() - do_generate_csvs(user, command) + generate_csv("Tasks") + generate_csv("C2Server") + generate_csv("Creds") + generate_csv("Implants") input("Press Enter to continue...") clear() -def generate_csvs(tableName): - os.system(f"sqlite3 -header -csv {PoshProjectDirectory}PowershellC2.SQLite 'select * from {tableName};' > {PoshProjectDirectory}reports/{tableName}.csv") - - def do_generate_csvs(user, command): - generate_csvs("Tasks") - generate_csvs("C2Server") - generate_csvs("Creds") - generate_csvs("Implants") + generate_csv("Tasks") + generate_csv("C2Server") + generate_csv("Creds") + generate_csv("Implants") input("Press Enter to continue...") clear() diff --git a/resources/scripts/posh-docker-debug b/resources/scripts/posh-docker-debug index 2e7e18dc..cbbf886e 100755 --- a/resources/scripts/posh-docker-debug +++ b/resources/scripts/posh-docker-debug @@ -18,5 +18,5 @@ if [[ ! -d "$POSH_PROJECT_DIR" ]]; then mkdir -p "$POSH_PROJECT_DIR" fi -sudo -E docker run -ti --rm -p "$port":443 -v "$POSH_PROJECT_DIR:$POSH_PROJECT_DIR" -v "$POSH_DIR:$POSH_DIR" -e POSHC2_DIR="${POSH_DIR}" nettitude/poshc2 /bin/bash +sudo -E docker run -ti --rm -v "$POSH_PROJECT_DIR:$POSH_PROJECT_DIR" -v "$POSH_DIR:$POSH_DIR" -e POSHC2_DIR="${POSH_DIR}" nettitude/poshc2 /bin/bash