collectkit

Param(
	[parameter(Position = 0)] [String] $Command,
	[parameter(Position = 1)] [String] $Arg1
	)

function Get-Version {
    Write-Output "  ____      _ _           _   _  ___ _     "
    Write-Output " / ___|___ | | | ___  ___| |_| |/ (_) |_   "
    Write-Output "| |   / _ \| | |/ _ \/ __| __| ' /| | __| "
    Write-Output "| |__| (_) | | |  __/ (__| |_| . \| | |_  "
    Write-Output " \____\___/|_|_|\___|\___|\__|_|\_\_|\__| "
    Write-Output ""
	Write-Output "Author: Nicholas Dhaeyer - nicholas.dhaeyer@nviso.eu | @DhaeyerWolf"
    Write-Output "V2024-10-10"
}

function help {
    <#
    .NOTES
    Author: nicholas.dhaeyer@nviso.eu | @DhaeyerWolf
    Updated: 2022-05-25
	Summary: Gives you the instructions
    #>
	Get-Version	
	
	Write-Output ""
	Write-Output ""
	Write-Output "Usage: 'run collectkit [Command] [argument]'"
	
	Write-Output ""
	Write-Output "Possible commands:"
	Write-Output "------------------"
	Write-Output ""
	Write-Output "===== COLLECTION ====="
	Write-Output "- Redline Captures -"
	Write-Output "RL-Win-Basic:	Runs a Basic Fast redline script for windows: Memory, Network, Services, Tasks, Persitence, Users. - collection can take longer depending on how much calculations are needed."
	Write-Output "RL-Win-Mem-Fast:Runs a redline Windows script that gets Memory, Network, Services & Tasks. - collection can take longer depending on how much calculations are needed."
	
	Write-Output ""
	Write-Output "- Memory Captures -"
	Write-Output "Get-Mem:	Makes a Memdump. Can take longer depending on how much memory the system has. 8GB RAM can take up to 15+ min."
	Write-Output "Get-Proc:	Makes a dump of a process using procdump. Usage: <run collectkit 'Get-Proc <procID|procName>'>"
	
	Write-Output ""
	Write-Output "- Browser Captures -"
	Write-Output "Get-Extentions:	Gets Lists & gets the user's browser extentions. - Only Chrome supported (for now) "
	Write-Output "Get-BrowserFiles:	Get all files stored by a browser currently supported: IE, MS Edge, Chrome, Firefox "
	
	
	Write-Output ""
	Write-Output "- List Files -"
	Write-Output "List-DefaultFolder:Lists the files in the folder C:\ProgramData\Microsoft\Windows Defender Advanced Threat Protection\Downloads\"
	Write-Output "Get-Folder:	Zips the specified folder and makes it available for download (default MDE download size is max 3GB)"
	
	Write-Output ""
	Write-Output "- Other collections -"
	Write-Output "Get-Basic-Info:
			- runs a modified version of mamun-sec dfirt tool
			- runs nov3mb3r's trident tool
			- Gets PowerShell history of all users "
	
	
	Write-Output ""
	Write-Output "===== REMEDIATION ====="
	Write-Output "Remediate-Item:	removes a given file/folder by using 'Remove-Item -Recurse -Force [dir/file]'"
	
	Write-Output ""
	Write-Output "===== CLEANUP ====="
	Write-Output "Rm-Collection:	Removes the collection.zip file used in all collections."
	## TODO:
	
	List-DefaultFolder
}

# Default folder listing display
function List-DefaultFolder{
	<#
    .NOTES
    Author: nicholas.dhaeyer@nviso.eu | @DhaeyerWolf
    Updated: 2022-10-05
	Summary: Lists the files in the folder C:\ProgramData\Microsoft\Windows Defender Advanced Threat Protection\Downloads\
	Usage: 'run collectkit List-DefaultFolder'
    #>
	Write-Output ""
	Write-Output ""
	Write-Output "
	+++++++++++++++++++++++++++++++++++++++++++++
	+   Current Files in the default folder:    +
	+ C:\ProgramData\Microsoft\Windows Defender +
	+   Advanced Threat Protection\Downloads\   +
	+++++++++++++++++++++++++++++++++++++++++++++"
	Get-ChildItem ./
}

# Redline Captures
function RL-Win-Basic {
	<#
    .NOTES
    Author: nicholas.dhaeyer@nviso.eu | @DhaeyerWolf
    Updated: 2022-05-25
	Summary: Runs a Basic Fast redline script for windows: Memory, Network, Services, Tasks, Persitence, Users.
	Usage: 'run collectkit RL-Win-Basic'
    #>
	
	Redline -URL "https://github.com/D4rk5t0rM/RemoteAcquisition/raw/main/RedlineCaptures/Redline-Win-Basic.zip"

}

function RL-Win-Mem-Fast {
	<#
    .NOTES
    Author: nicholas.dhaeyer@nviso.eu | @DhaeyerWolf
    Updated: 2022-05-25
	Summary: Runs a redline Windows script that gets Memory, Network, Services & Tasks.
	Usage: 'run collectkit RL-Win-Mem-Fast'
    #>
	
	Redline -URL "https://github.com/D4rk5t0rM/RemoteAcquisition/raw/main/RedlineCaptures/Redline-Win-Memory-Fast.zip"
}

function RL-Full-Disk {
	<#
    .NOTES
    Author: nicholas.dhaeyer@nviso.eu | @DhaeyerWolf
    Updated: 2024-10-10
	Summary: Runs a redline Windows script that gets Memory, Network, Services & Tasks.
	Usage: 'run collectkit RL-Full-Disk
    #>
	
	Redline -URL "https://github.com/D4rk5t0rM/RemoteAcquisition/raw/main/RedlineCaptures/Redline-Basic+fullDiskEnum.zip"
}

# Collections

function Redline([string]$URL) {
    <#
    .NOTES
    Author: nicholas.dhaeyer@nviso.eu | @DhaeyerWolf
    Updated: 2022-05-25
	Summary: Uses redeline A redline script defined in a RL-* command. - Do not run this on it's own
	Usage: 'run collectkit RL-[Redline Script]'
    #>
	
	# Download & Unpack:
	Write-Output "Downloading $URL"
	Start-BitsTransfer -Source $URL -Destination Redline.zip

	Expand-Archive Redline.zip -Force
		
	# Execution:
	start .\Redline\RunRedlineAudit.bat -Wait
	
	# Needed to collect the created sessions
	#Start-Sleep -Seconds $timeout
	
	Compress-Archive -path .\Redline\Sessions\ .\collection.zip -CompressionLevel optimal -Force
	
	Write-Output "Please collect the file collection.zip. Run: getfile '$(pwd)\collection.zip'"
	Write-Output "To clean up the collection file run: <run collectkit Rm-Collection>"	
	
	List-DefaultFolder
}

function Get-Mem {
	<#
    .NOTES
    Author: nicholas.dhaeyer@nviso.eu | @DhaeyerWolf
    Updated: 2022-05-23
	Summary: Gets a Memdump of the host.
	Usage: 'run collectkit Get-Mem'
    #>	
	$hostname = hostname
	# Download:
	Start-BitsTransfer -Source https://github.com/D4rk5t0rM/RemoteAcquisition/raw/main/Setup/7z.exe  -Destination 7z.exe
	Start-BitsTransfer -Source https://github.com/D4rk5t0rM/RemoteAcquisition/raw/main/Setup/winpmem_mini_x64_rc2.exe -Destination winpmem.exe
	Start-BitsTransfer -Source https://github.com/D4rk5t0rM/RemoteAcquisition/raw/main/Setup/7z.dll  -Destination 7z.dll
			
	# Create Memdump:
	.\winpmem.exe .\$hostname-Memdump.raw
	
	#Zip if bigger than 3GB - limit of MDE getfile command
	if((Get-Item .\$hostname-Memdump.raw).length -gt 3GB) {
        .\7z.exe a -tzip .\$hostname-Memdump.zip .\$hostname-Memdump.raw
		Write-Output "run command: <getfile '$(pwd)\$hostname-Memdump.zip'>"
    }
	else{
		Write-Output "run command: <getfile '$(pwd)\$hostname-Memdump.raw'>"
	}
	
	List-DefaultFolder
}

function Get-Proc {
	<#
    .NOTES
    Author: nicholas.dhaeyer@nviso.eu | @DhaeyerWolf
    Updated: 2022-05-25
	Summary: Dumps a process according to procdump arguments
	Usage: 'run collectkit "Get-Proc [process name/id]"'
    #>	
	# Download:
	Start-BitsTransfer -Source https://github.com/D4rk5t0rM/RemoteAcquisition/raw/main/Setup/procdump64.exe -Destination procdump64.exe
	if ([System.IO.File]::Exists(".\7z.exe")){
		Start-BitsTransfer -Source https://github.com/D4rk5t0rM/RemoteAcquisition/raw/main/Setup/7z.exe  -Destination 7z.exe}
	if ([System.IO.File]::Exists(".\7z.dll")){
		Start-BitsTransfer -Source https://github.com/D4rk5t0rM/RemoteAcquisition/raw/main/Setup/7z.dll  -Destination 7z.dll}
	
	# Create ProcDump:
	.\procdump64.exe $Arg1 -ma -o proc-$Arg1.dmp /accepteula
	
	#Zip if bigger than 3GB - limit of MDE getfile command
	if((Get-Item .\proc-$Arg1.dmp).length -gt 200MB) {
        .\7z.exe a -tzip .\proc-$Arg1.zip .\proc-$Arg1.dmp
		Write-Output "run command: <getfile '$(pwd)\proc-$Arg1.zip'>"
    }
	else{
		Write-Output "run command: <getfile '$(pwd)\proc-$Arg1.dmp'>"
	}
	
	List-DefaultFolder
}

function Get-Extentions {
	<#
    .NOTES
    Author: nicholas.dhaeyer@nviso.eu | @DhaeyerWolf
    Updated: 2022-05-23
	Summary: Gets Lists & gets the user's browser extentions.
	Usage: 'run collectkit Get-Extentions'
    #>	
		
	# Chrome
	$chrome = @()
	$chromeTmp = New-Object System.Collections.Generic.List[System.Object]
	foreach($User in Get-ChildItem c:\Users\){
		$chromeTmp.Add("C:\Users\$User\AppData\Local\Google\Chrome\User Data\Default\Extensions\")
	}
	
	$chromeTmp.ToArray()
	
	foreach($path in $chromeTmp){
		if(Test-Path -Path $path){
			$chrome += $path
		}
	}
	foreach($path in $chrome){
		Get-ChildItem $path
	}
	
	Compress-Archive -path $chrome .\Chrome-Extentions.zip -CompressionLevel optimal
	Write-Output ""
	Write-Output "Please collect the file .\Chrome-Extentions.zip Run: getfile '$(pwd)\Chrome-Extentions.zip'"
	
	List-DefaultFolder	
}

function Get-BrowserFiles {
	<#
    .NOTES
    Author: nicholas.dhaeyer@nviso.eu | @DhaeyerWolf
    Updated: 2022-10-05
	Summary: Gets a ZIP file of all files of a browser on a system
	Usage: 'run collectkit Get-BrowserFiles'
    #>	
	
	# Setup
	Start-BitsTransfer -Source https://raw.githubusercontent.com/D4rk5t0rM/RemoteAcquisition/main/Setup/CollectBrowserfiles.cmd -Destination CollectBrowserfiles.cmd
	
	# Execution
	cmd.exe /c CollectBrowserfiles.cmd
	
	Compress-Archive -path .\Browser_Files .\Browser_Files.zip -CompressionLevel optimal
	
	# Cleanup
	Remove-Item -Recurse -Force .\Browser_Files 
	Remove-Item -Recurse -Force .\CollectBrowserfiles.cmd
	
	
	Write-Output "Please collect the file .\Browser_Files.zip Run: getfile '$(pwd)\Browser_Files.zip'"
	
	List-DefaultFolder
}


function Get-Folder {
	<#
    .NOTES
    Author: nicholas.dhaeyer@nviso.eu | @DhaeyerWolf
    Updated: 2022-10-05
	Summary: Gets Lists & gets the user's browser extentions.
	Usage: run collectkit "Get-Folder '<folderpath>'"
    #>	
	
	$OutputFile = $Arg1.Split("\")[-1]
	$OutputFile
	
	Compress-Archive -path $Arg1 .\"$OutputFile".zip -CompressionLevel optimal
	
	Write-Output "Please collect the file .\$OutputFile.zip Run: getfile '$(pwd)\$OutputFile.zip'"
	
	List-DefaultFolder
}

# Run external scripts
function Get-Basic-Info {
	<#
    .NOTES
    Author: nicholas.dhaeyer@nviso.eu | @DhaeyerWolf
    Updated: 2022-10-05
	Summary:- runs a modified version of mamun-sec dfirt tool
			- runs nov3mb3r's trident tool
			- Gets PowerShell history of all users
	Usage: 'run collectkit Get-Basic-Info"
    #>
	
	# Default setup
	mkdir .\Basic-Info
	
	
	# dfirt
	## Setup
	Start-BitsTransfer -Source https://raw.githubusercontent.com/D4rk5t0rM/Forensics-tools/main/.scripts/dfirt.ps1 -Destination .\dfirt.ps1
	
	## Execution
	PowerShell -ep bypass .\dfirt.ps1
	
	## prepare export
	Move-Item .\report.txt .\Basic-Info\dfirt_$(hostname).txt
	Get-Content .\Basic-Info\dfirt_$(hostname).txt
	
	
	# Trident
	## Setup
	Start-BitsTransfer -Source https://raw.githubusercontent.com/nov3mb3r/trident/main/trident.ps1 -Destination .\trident.ps1
	
	## Execution
	PowerShell -ep bypass .\trident.ps1
	
	# prepare export
	Move-Item .\trident_$(hostname).txt .\Basic-Info\trident_$(hostname).txt
	Get-Content .\Basic-Info\trident_$(hostname).txt
	
	
	# Powershell History
	foreach ($u in $(get-localuser | % {$_.Name})){
			Copy-Item c:\Users\$u\AppData\Roaming\Microsoft\Windows\PowerShell\PSReadLine\ConsoleHost_history.txt .\Basic-Info\"$u"_PowershellHistory.txt 2> .\error.txt
			
			Write-Output "
			+++++++++++++++++++++++++++++
			+  Powerhsell History for:  +
			+            $u     
			+++++++++++++++++++++++++++++
			"
			Get-Content .\Basic-Info\"$u"_PowershellHistory.txt
		}
	
	
	# --------------------
	
	# Collection
	Compress-Archive -path .\Basic-Info .\Basic-Info.zip -CompressionLevel optimal
	
	Write-Output "Please collect the file .\Basic-Info.zip Run: getfile '$(pwd)\Basic-Info.zip'"
	
	
	
	# Cleanup
	Remove-Item -Recurse -Force .\dfirt.ps1
	Remove-Item -Recurse -Force .\trident.ps1
	Remove-Item -Recurse -Force .\Basic-Info
	Remove-Item -Recurse -Force .\error.txt
	
	List-DefaultFolder
	
}

# Remediation actions
function Remediate-Item {
    <#
    .NOTES
    Author: nicholas.dhaeyer@nviso.eu | @DhaeyerWolf
    Updated: 2022-09-09
	Summary: emoves a given file/folder by using 'Remove-Item -Recurse -Force [dir/file]' 
	Usage: 'run collectkit "rm [dir/file]"' full path is required. full path with spaces, use "rm '[path]'"
    #>
	Remove-Item -Recurse -Force $Arg1
	Write-Output "Executed the command"
	Write-Output "Files left on disk: ...."
	Get-ChildItem $(Split-Path -Path $Arg1)
}

# Cleanup:
function Rm-Collection {
    <#
    .NOTES
    Author: nicholas.dhaeyer@nviso.eu | @DhaeyerWolf
    Updated: 2022-05-22
	Summary: Removes the collection file. should be run after the Get-* commands
	Usage: 'run collectkit Rm-Collection'
    #>	
	
	Remove-Item -Recurse -Force *
	Write-Output "Done! following files are still on disk:"
	
	List-DefaultFolder
	
}


Function Invoke-CollectKit {
	$available = (Get-ChildItem function: | Where-Object CommandType -EQ "Function" | Where-Object Helpfile -EQ $null | Where-Object Source -EQ "" | Where-Object Name -NE "Invoke-CollectKit").Name
	if ($available.Contains($Command)) {
		&$Command
	} else {
		Write-Output "Valid remediation actions are:"
		ForEach ($c in $available) {
			Write-Output $c
		}
	}
}

Invoke-CollectKit