From 3d839e7bd63f67bf78707e2a87cbd44b94733eba Mon Sep 17 00:00:00 2001 From: Carrie Roberts Date: Mon, 21 Oct 2019 15:02:04 -0600 Subject: [PATCH] handle spaces in input param defaults (#590) --- .../Invoke-AtomicRedTeam/Public/Invoke-AtomicTest.ps1 | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/execution-frameworks/Invoke-AtomicRedTeam/Invoke-AtomicRedTeam/Public/Invoke-AtomicTest.ps1 b/execution-frameworks/Invoke-AtomicRedTeam/Invoke-AtomicRedTeam/Public/Invoke-AtomicTest.ps1 index a51dbe98..ba4a9ac2 100644 --- a/execution-frameworks/Invoke-AtomicRedTeam/Invoke-AtomicRedTeam/Public/Invoke-AtomicTest.ps1 +++ b/execution-frameworks/Invoke-AtomicRedTeam/Invoke-AtomicRedTeam/Public/Invoke-AtomicTest.ps1 @@ -103,12 +103,9 @@ function Invoke-AtomicTest { $isElevated = $false } function Get-InputArgs([hashtable]$ip) { - $inputArgsDefault = [Array]($ip.Keys).Split(" ") - $inputDefaults = [Array]($ip.Values | ForEach-Object { $_.default.toString() }).Split(" ") $defaultArgs = @{ } - for ($i = 0; $i -lt $inputArgsDefault.Length; $i++) { - $defaultArgs[$inputArgsDefault[$i]] = $inputDefaults[$i] - + foreach ($key in $ip.Keys) { + $defaultArgs[$key] = $ip[$key].default } # overwrite defaults with any user supplied values foreach ($key in $InputArgs.Keys) { @@ -290,4 +287,4 @@ function Invoke-AtomicTest { } # End of PROCESS block END { } # Intentionally left blank and can be removed -} +} \ No newline at end of file