remove duplicate powershellhistory file
This commit is contained in:
parent
1029a7407d
commit
fd8eb6665a
|
@ -1,60 +0,0 @@
|
|||
import os
|
||||
import ntpath
|
||||
from typing import Any
|
||||
from dploot.lib.target import Target
|
||||
from dploot.lib.smb import DPLootSMBConnection
|
||||
from donpapi.core import DonPAPICore
|
||||
from donpapi.lib.logger import DonPAPIAdapter
|
||||
|
||||
|
||||
TAG = "PowerShellHistory"
|
||||
|
||||
class PowerShellHistoryDump:
|
||||
false_positive = [".", "..", "desktop.ini", "Public", "Default", "Default User", "All Users", ".NET v4.5", ".NET v4.5 Classic"]
|
||||
user_directories = ["\\Users\\{username}\\AppData\\Roaming\\Microsoft\\Windows\\PowerShell\\PSReadLine\\"]
|
||||
max_filesize = 5000000
|
||||
|
||||
def __init__(self, target: Target, conn: DPLootSMBConnection, masterkeys: list, options: Any, logger: DonPAPIAdapter, context: DonPAPICore) -> None:
|
||||
self.target = target
|
||||
self.conn = conn
|
||||
self.masterkeys = masterkeys
|
||||
self.options = options
|
||||
self.logger = logger
|
||||
self.context = context
|
||||
self.found = 0
|
||||
|
||||
def run(self):
|
||||
|
||||
self.logger.display("Gathering powershell history files")
|
||||
for user in self.context.users:
|
||||
for directory in self.user_directories:
|
||||
directory_path = directory.format(username=user)
|
||||
self.dig_files(directory_path=directory_path, recurse_level=0, recurse_max=10)
|
||||
self.logger.secret(f"Found {self.found} powershell history files", TAG)
|
||||
|
||||
def dig_files(self, directory_path, recurse_level=0, recurse_max=10):
|
||||
directory_list = self.conn.remote_list_dir(self.context.share, directory_path)
|
||||
if directory_list is not None:
|
||||
for item in directory_list:
|
||||
if item.get_longname() not in self.false_positive:
|
||||
self.found += 1
|
||||
new_path = ntpath.join(directory_path, item.get_longname())
|
||||
file_content = self.conn.readFile(self.context.share, new_path)
|
||||
local_filepath = os.path.join(self.context.output_dir, *(new_path.split('\\')))
|
||||
|
||||
os.makedirs(os.path.dirname(local_filepath), exist_ok=True)
|
||||
with open(local_filepath, "wb") as f:
|
||||
if file_content is None:
|
||||
file_content = b""
|
||||
f.write(file_content)
|
||||
|
||||
os.makedirs(f"{self.context.output_dir}/../PowerShellHistory", exist_ok=True)
|
||||
local_filepath = os.path.join(
|
||||
f"{self.context.output_dir}/../PowerShellHistory",
|
||||
f"{item.get_longname()}-{self.found}"
|
||||
)
|
||||
with open(local_filepath, "wb") as f:
|
||||
if file_content is None:
|
||||
file_content = b""
|
||||
f.write(file_content)
|
||||
|
Loading…
Reference in New Issue