Run kernel mode perf tests with a kernel mode client (microsoft#1501)

* Run kernel mode perf tests with a kernel mode client * fix user mode tests
boscorona · Apr 23, 2021 · f1dc5cc · f1dc5cc
1 parent c2e5bfe
commit f1dc5cc
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/scripts/performance-helper.psm1 b/scripts/performance-helper.psm1
@@ -282,6 +282,21 @@ function Get-ExeName {
 
 function Remove-PerfServices {
     if ($IsWindows) {
+        if ($null -ne (Get-Service -Name "secnetperfdrvpriv" -ErrorAction Ignore)) {
+            try {
+                net.exe stop secnetperfdrvpriv /y | Out-Null
+            }
+            catch {}
+            sc.exe delete secnetperfdrvpriv /y | Out-Null
+        }
+        if ($null -ne (Get-Service -Name "msquicpriv" -ErrorAction Ignore)) {
+            try {
+                net.exe stop msquicpriv /y | Out-Null
+            }
+            catch {}
+            sc.exe delete msquicpriv /y | Out-Null
+        }
+
         Invoke-TestCommand -Session $Session -ScriptBlock {
             if ($null -ne (Get-Service -Name "secnetperfdrvpriv" -ErrorAction Ignore)) {
                 try {

diff --git a/scripts/performance.ps1 b/scripts/performance.ps1
@@ -394,6 +394,17 @@ function Invoke-Test {
         $RemoteArguments += " -stats:1"
     }
 
+    if ($Kernel) {
+        $Arch = Split-Path (Split-Path $LocalExe -Parent) -Leaf
+        $RootBinPath = Split-Path (Split-Path (Split-Path $LocalExe -Parent) -Parent) -Parent
+        $KernelDir = Join-Path $RootBinPath "winkernel" $Arch
+
+        Copy-Item (Join-Path $KernelDir "secnetperfdrvpriv.sys") (Split-Path $LocalExe -Parent)
+        Copy-Item (Join-Path $KernelDir "msquicpriv.sys") (Split-Path $LocalExe -Parent)
+
+        $LocalArguments = "-driverNamePriv:secnetperfdrvpriv $LocalArguments"
+    }
+
     Write-LogAndDebug "Running Remote: $RemoteExe Args: $RemoteArguments"
 
     # Starting the server
@@ -438,6 +449,12 @@ function Invoke-Test {
             $LocalResults | Write-LogAndDebug
         }
     } finally {
+        if ($Kernel) {
+            net.exe stop msquicpriv /y | Out-Null
+            sc.exe delete msquictestpriv | Out-Null
+            sc.exe delete msquicpriv | Out-Null
+        }
+
         $RemoteResults = Wait-ForRemote -Job $RemoteJob
         Write-LogAndDebug $RemoteResults.ToString()