Merge pull request #2232 from iserrano76/AMSI

Fix execution without parameters and servername
microsoft · Nov 27, 2024 · 1661fa5 · 1661fa5
2 parents 293e235 + 6615883
commit 1661fa5
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/Admin/Test-AMSI.ps1 b/Admin/Test-AMSI.ps1
@@ -116,11 +116,11 @@ param(
     [Parameter(ParameterSetName = 'RestartIISAll', Mandatory = $true)]
     [switch]$RestartIIS,
     [Alias("ExchangeServerFQDN")]
-    [Parameter(ParameterSetName = 'TestAMSI', Mandatory = $false, ValueFromPipeline = $true)]
-    [Parameter(ParameterSetName = 'EnableAMSI', Mandatory = $false, ValueFromPipeline = $true)]
-    [Parameter(ParameterSetName = 'DisableAMSI', Mandatory = $false, ValueFromPipeline = $true)]
-    [Parameter(ParameterSetName = 'CheckAMSIConfig', Mandatory = $false, ValueFromPipeline = $true)]
-    [Parameter(ParameterSetName = 'RestartIIS', Mandatory = $false, ValueFromPipeline = $true)]
+    [Parameter(Position = 0, ParameterSetName = 'TestAMSI', Mandatory = $false, ValueFromPipeline = $true)]
+    [Parameter(Position = 0, ParameterSetName = 'EnableAMSI', Mandatory = $false, ValueFromPipeline = $true)]
+    [Parameter(Position = 0, ParameterSetName = 'DisableAMSI', Mandatory = $false, ValueFromPipeline = $true)]
+    [Parameter(Position = 0, ParameterSetName = 'CheckAMSIConfig', Mandatory = $false, ValueFromPipeline = $true)]
+    [Parameter(Position = 0, ParameterSetName = 'RestartIIS', Mandatory = $false, ValueFromPipeline = $true)]
     [string[]]$ServerList = $null,
     [Parameter(ParameterSetName = 'TestAMSIAll', Mandatory = $true)]
     [Parameter(ParameterSetName = 'CheckAMSIConfigAll', Mandatory = $true)]
@@ -154,6 +154,10 @@ param(
 
 begin {
 
+    if (-not $ServerList -and $args.Count -gt 0) {
+        $ServerList = @($args[0])
+    }
+
     . $PSScriptRoot\..\Shared\Confirm-Administrator.ps1
     . $PSScriptRoot\..\Shared\Confirm-ExchangeShell.ps1
     . $PSScriptRoot\..\Shared\Invoke-ScriptBlockHandler.ps1