Fix km_ebpfcore_restart_test to run inside VM

.github/workflows/cicd.yml

@@ -669,7 +669,7 @@ jobs:
     with:
       name: km_ebpfcore_restart_test
       pre_test: .\setup_ebpf_cicd_tests.ps1 -KmTracing $true -KmTraceType "memory"
-      test_command: .\ebpf_restart_test_controller.exe
+      test_command: .\execute_ebpf_cicd_tests.ps1 -TestMode "Stress" -Options @("RestartEbpfCore")
       post_test: .\cleanup_ebpf_cicd_tests.ps1 -KmTracing $true
       build_artifact: Build-x64
       environment: '["self-hosted", "1ES.Pool=ebpf-cicd-runner-pool-server-2019", "1ES.ImageOverride=server2022"]'

scripts/run_driver_tests.psm1

@@ -450,11 +450,24 @@ function Invoke-CICDStressTests
           [parameter(Mandatory = $false)][int] $TestHangTimeout = (120*60),
           [parameter(Mandatory = $false)][string] $UserModeDumpFolder = "C:\Dumps",
           [parameter(Mandatory = $false)][bool] $NeedKernelDump = $true,
-          [parameter(Mandatory = $false)][bool] $RestartExtension = $false)
+          [parameter(Mandatory = $false)][bool] $RestartExtension = $false,
+          [parameter(Mandatory = $false)][bool] $RestartEbpfCore = $false)
 
     Push-Location $WorkingDirectory
     $env:EBPF_ENABLE_WER_REPORT = "yes"
 
+    if ($RestartEbpfCore) {
+        Write-Log "Executing eBPF core restart stress test."
+
+        $LASTEXITCODE = 0
+
+        $TestCommand = ".\ebpf_restart_test_controller.exe"
+        Invoke-Test -TestName $TestCommand -VerboseLogs $VerboseLogs -TestHangTimeout $TestHangTimeout -TracingProfileName "EbpfForWindowsProvider"
+
+        Pop-Location
+        return
+    }
+
     Write-Log "Executing eBPF kernel mode multi-threaded stress tests (restart extension:$RestartExtension)."
 
     $LASTEXITCODE = 0

scripts/vm_run_tests.psm1

@@ -336,9 +336,12 @@ function Run-KernelTests {
             "stress" {
                 # Set RestartExtension to true if options contains that string.
                 $RestartExtension = $Options -contains "RestartExtension"
+                # Set RestartEbpfCore to true if options contains that string.
+                $RestartEbpfCore = $Options -contains "RestartEbpfCore"
                 Invoke-CICDStressTests `
                     -VerboseLogs $VerboseLogs `
                     -RestartExtension $RestartExtension `
+                    -RestartEbpfCore $RestartEbpfCore `
                     2>&1 | Write-Log
             }
             "performance" {