Add Kernel test mode integration.

microsoft · Jan 5, 2024 · 11b4cbb · 11b4cbb
1 parent 42c2089
commit 11b4cbb
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 2 deletions.
diff --git a/src/test/MsQuicTests.h b/src/test/MsQuicTests.h
@@ -1251,4 +1251,17 @@ typedef struct {
 #define IOCTL_QUIC_RUN_CONN_CLOSE_BEFORE_STREAM_CLOSE \
     QUIC_CTL_CODE(117, METHOD_BUFFERED, FILE_WRITE_DATA)
 
-#define QUIC_MAX_IOCTL_FUNC_CODE 117
+#pragma pack(push)
+#pragma pack(1)
+
+typedef struct {
+    bool DropPackets;
+} QUIC_RUN_CANCEL_ON_LOSS_PARAMS;
+
+#pragma pack(pop)
+
+#define IOCTL_QUIC_RUN_CANCEL_ON_LOSS \
+    QUIC_CTL_CODE(118, METHOD_BUFFERED, FILE_WRITE_DATA)
+// QUIC_RUN_CANCEL_ON_LOSS_PARAMS
+
+#define QUIC_MAX_IOCTL_FUNC_CODE 118
diff --git a/src/test/bin/quic_gtest.cpp b/src/test/bin/quic_gtest.cpp
@@ -1970,7 +1970,10 @@ TEST_P(WithAbortiveArgs, AbortiveShutdown) {
 TEST_P(WithCancelOnLossArgs, CancelOnLossSend) {
     TestLoggerT<ParamType> Logger("QuicCancelOnLossSend", GetParam());
     if (TestingKernelMode) {
-        // TODO
+        QUIC_RUN_CANCEL_ON_LOSS_PARAMS Params = {
+            GetParam().DropPackets
+        };
+        ASSERT_TRUE(DriverClient.Run(IOCTL_QUIC_RUN_CANCEL_ON_LOSS, Params));
     } else {
         QuicCancelOnLossSend(GetParam().DropPackets);
     }

diff --git a/src/test/bin/winkernel/control.cpp b/src/test/bin/winkernel/control.cpp
@@ -1532,6 +1532,13 @@ QuicTestCtlEvtIoDeviceControl(
         QuicTestCtlRun(QuicTestConnectionCloseBeforeStreamClose());
         break;
 
+    case IOCTL_QUIC_RUN_CANCEL_ON_LOSS:
+        CXPLAT_FRE_ASSERT(Params != nullptr);
+        QuicTestCtlRun(
+            QuicCancelOnLossSend(
+                Params->DropPackets));
+        break;
+
     default:
         Status = STATUS_NOT_IMPLEMENTED;
         break;