Fix CI: kernel/official build guards, Rust bindings

- Guard QUIC_PARAM_GLOBAL_WORKER_STATISTICS case with #ifndef _KERNEL_MODE
  (worker pool API not available in kernel builds)
- Guard test code with #ifdef QUIC_API_ENABLE_PREVIEW_FEATURES for official
  builds that don't define preview features
- Exclude worker stats tests from kernel test driver registration
- Add QUIC_WORKER_STATISTICS and QUIC_WORKER_STATISTICS_LIST to Rust FFI
  bindings (win_bindings.rs and linux_bindings.rs)

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: guhetier

src/core/library.c

@@ -1876,6 +1876,7 @@ QuicLibraryGetGlobalParam(
         break;
     }
 
+#ifndef _KERNEL_MODE
     case QUIC_PARAM_GLOBAL_WORKER_STATISTICS: {
 
         if (MsQuicLib.WorkerPool == NULL) {
@@ -1911,6 +1912,7 @@ QuicLibraryGetGlobalParam(
         Status = QUIC_STATUS_SUCCESS;
         break;
     }
+#endif
 
     default:
         Status = QUIC_STATUS_INVALID_PARAMETER;

src/rs/ffi/linux_bindings.rs

@@ -176,6 +176,7 @@ pub const QUIC_PARAM_GLOBAL_STATELESS_RESET_KEY: u32 = 16777227;
 pub const QUIC_PARAM_GLOBAL_STATISTICS_V2_SIZES: u32 = 16777228;
 pub const QUIC_PARAM_GLOBAL_STATELESS_RETRY_CONFIG: u32 = 16777229;
 pub const QUIC_PARAM_GLOBAL_XDP_MAP_CONFIG: u32 = 16777230;
+pub const QUIC_PARAM_GLOBAL_WORKER_STATISTICS: u32 = 16777231;
 pub const QUIC_PARAM_CONFIGURATION_SETTINGS: u32 = 50331648;
 pub const QUIC_PARAM_CONFIGURATION_TICKET_KEYS: u32 = 50331649;
 pub const QUIC_PARAM_CONFIGURATION_VERSION_SETTINGS: u32 = 50331650;
@@ -1644,6 +1645,45 @@ const _: () = {
     ["Offset of field: QUIC_LISTENER_STATISTICS::BindingRecvDroppedPackets"]
         [::std::mem::offset_of!(QUIC_LISTENER_STATISTICS, BindingRecvDroppedPackets) - 16usize];
 };
+#[repr(C)]
+#[derive(Debug, Copy, Clone)]
+pub struct QUIC_WORKER_STATISTICS {
+    pub IdealProcessor: u16,
+    pub Reserved: [u8; 6usize],
+    pub CumulativeActiveTimeUs: u64,
+    pub CumulativeWallTimeUs: u64,
+}
+#[allow(clippy::unnecessary_operation, clippy::identity_op)]
+const _: () = {
+    ["Size of QUIC_WORKER_STATISTICS"][::std::mem::size_of::<QUIC_WORKER_STATISTICS>() - 24usize];
+    ["Alignment of QUIC_WORKER_STATISTICS"]
+        [::std::mem::align_of::<QUIC_WORKER_STATISTICS>() - 8usize];
+    ["Offset of field: QUIC_WORKER_STATISTICS::IdealProcessor"]
+        [::std::mem::offset_of!(QUIC_WORKER_STATISTICS, IdealProcessor) - 0usize];
+    ["Offset of field: QUIC_WORKER_STATISTICS::Reserved"]
+        [::std::mem::offset_of!(QUIC_WORKER_STATISTICS, Reserved) - 2usize];
+    ["Offset of field: QUIC_WORKER_STATISTICS::CumulativeActiveTimeUs"]
+        [::std::mem::offset_of!(QUIC_WORKER_STATISTICS, CumulativeActiveTimeUs) - 8usize];
+    ["Offset of field: QUIC_WORKER_STATISTICS::CumulativeWallTimeUs"]
+        [::std::mem::offset_of!(QUIC_WORKER_STATISTICS, CumulativeWallTimeUs) - 16usize];
+};
+#[repr(C)]
+#[derive(Debug, Copy, Clone)]
+pub struct QUIC_WORKER_STATISTICS_LIST {
+    pub WorkerCount: u32,
+    pub WorkerStatsSize: u32,
+}
+#[allow(clippy::unnecessary_operation, clippy::identity_op)]
+const _: () = {
+    ["Size of QUIC_WORKER_STATISTICS_LIST"]
+        [::std::mem::size_of::<QUIC_WORKER_STATISTICS_LIST>() - 8usize];
+    ["Alignment of QUIC_WORKER_STATISTICS_LIST"]
+        [::std::mem::align_of::<QUIC_WORKER_STATISTICS_LIST>() - 4usize];
+    ["Offset of field: QUIC_WORKER_STATISTICS_LIST::WorkerCount"]
+        [::std::mem::offset_of!(QUIC_WORKER_STATISTICS_LIST, WorkerCount) - 0usize];
+    ["Offset of field: QUIC_WORKER_STATISTICS_LIST::WorkerStatsSize"]
+        [::std::mem::offset_of!(QUIC_WORKER_STATISTICS_LIST, WorkerStatsSize) - 4usize];
+};
 pub const QUIC_PERFORMANCE_COUNTERS_QUIC_PERF_COUNTER_CONN_CREATED: QUIC_PERFORMANCE_COUNTERS = 0;
 pub const QUIC_PERFORMANCE_COUNTERS_QUIC_PERF_COUNTER_CONN_HANDSHAKE_FAIL:
     QUIC_PERFORMANCE_COUNTERS = 1;

src/rs/ffi/win_bindings.rs

@@ -170,6 +170,7 @@ pub const QUIC_PARAM_GLOBAL_STATELESS_RESET_KEY: u32 = 16777227;
 pub const QUIC_PARAM_GLOBAL_STATISTICS_V2_SIZES: u32 = 16777228;
 pub const QUIC_PARAM_GLOBAL_STATELESS_RETRY_CONFIG: u32 = 16777229;
 pub const QUIC_PARAM_GLOBAL_XDP_MAP_CONFIG: u32 = 16777230;
+pub const QUIC_PARAM_GLOBAL_WORKER_STATISTICS: u32 = 16777231;
 pub const QUIC_PARAM_CONFIGURATION_SETTINGS: u32 = 50331648;
 pub const QUIC_PARAM_CONFIGURATION_TICKET_KEYS: u32 = 50331649;
 pub const QUIC_PARAM_CONFIGURATION_VERSION_SETTINGS: u32 = 50331650;
@@ -1637,6 +1638,45 @@ const _: () = {
     ["Offset of field: QUIC_LISTENER_STATISTICS::BindingRecvDroppedPackets"]
         [::std::mem::offset_of!(QUIC_LISTENER_STATISTICS, BindingRecvDroppedPackets) - 16usize];
 };
+#[repr(C)]
+#[derive(Debug, Copy, Clone)]
+pub struct QUIC_WORKER_STATISTICS {
+    pub IdealProcessor: u16,
+    pub Reserved: [u8; 6usize],
+    pub CumulativeActiveTimeUs: u64,
+    pub CumulativeWallTimeUs: u64,
+}
+#[allow(clippy::unnecessary_operation, clippy::identity_op)]
+const _: () = {
+    ["Size of QUIC_WORKER_STATISTICS"][::std::mem::size_of::<QUIC_WORKER_STATISTICS>() - 24usize];
+    ["Alignment of QUIC_WORKER_STATISTICS"]
+        [::std::mem::align_of::<QUIC_WORKER_STATISTICS>() - 8usize];
+    ["Offset of field: QUIC_WORKER_STATISTICS::IdealProcessor"]
+        [::std::mem::offset_of!(QUIC_WORKER_STATISTICS, IdealProcessor) - 0usize];
+    ["Offset of field: QUIC_WORKER_STATISTICS::Reserved"]
+        [::std::mem::offset_of!(QUIC_WORKER_STATISTICS, Reserved) - 2usize];
+    ["Offset of field: QUIC_WORKER_STATISTICS::CumulativeActiveTimeUs"]
+        [::std::mem::offset_of!(QUIC_WORKER_STATISTICS, CumulativeActiveTimeUs) - 8usize];
+    ["Offset of field: QUIC_WORKER_STATISTICS::CumulativeWallTimeUs"]
+        [::std::mem::offset_of!(QUIC_WORKER_STATISTICS, CumulativeWallTimeUs) - 16usize];
+};
+#[repr(C)]
+#[derive(Debug, Copy, Clone)]
+pub struct QUIC_WORKER_STATISTICS_LIST {
+    pub WorkerCount: u32,
+    pub WorkerStatsSize: u32,
+}
+#[allow(clippy::unnecessary_operation, clippy::identity_op)]
+const _: () = {
+    ["Size of QUIC_WORKER_STATISTICS_LIST"]
+        [::std::mem::size_of::<QUIC_WORKER_STATISTICS_LIST>() - 8usize];
+    ["Alignment of QUIC_WORKER_STATISTICS_LIST"]
+        [::std::mem::align_of::<QUIC_WORKER_STATISTICS_LIST>() - 4usize];
+    ["Offset of field: QUIC_WORKER_STATISTICS_LIST::WorkerCount"]
+        [::std::mem::offset_of!(QUIC_WORKER_STATISTICS_LIST, WorkerCount) - 0usize];
+    ["Offset of field: QUIC_WORKER_STATISTICS_LIST::WorkerStatsSize"]
+        [::std::mem::offset_of!(QUIC_WORKER_STATISTICS_LIST, WorkerStatsSize) - 4usize];
+};
 pub const QUIC_PERFORMANCE_COUNTERS_QUIC_PERF_COUNTER_CONN_CREATED: QUIC_PERFORMANCE_COUNTERS = 0;
 pub const QUIC_PERFORMANCE_COUNTERS_QUIC_PERF_COUNTER_CONN_HANDSHAKE_FAIL:
     QUIC_PERFORMANCE_COUNTERS = 1;

src/test/MsQuicTests.h

@@ -89,8 +89,10 @@ void QuicTestTlsParam();
 void QuicTestTlsHandshakeInfo(const bool& EnableResumption);
 void QuicTestStreamParam();
 void QuicTestGetPerfCounters();
+#ifdef QUIC_API_ENABLE_PREVIEW_FEATURES
 void QuicTestGetWorkerStatistics();
 void QuicTestValidateWorkerStatistics();
+#endif
 void QuicTestVersionSettings();
 void QuicTestValidateParamApi();
 void QuicTestCredentialLoad(const QUIC_CREDENTIAL_BLOB& Config);

src/test/bin/quic_gtest.cpp

@@ -390,6 +390,7 @@ TEST(ParameterValidation, ValidateGetPerfCounters) {
     }
 }
 
+#ifdef QUIC_API_ENABLE_PREVIEW_FEATURES
 TEST(ParameterValidation, ValidateGetWorkerStatistics) {
     TestLogger Logger("QuicTestGetWorkerStatistics");
     if (TestingKernelMode) {
@@ -407,6 +408,7 @@ TEST(ParameterValidation, ValidateWorkerStatistics) {
         QuicTestValidateWorkerStatistics();
     }
 }
+#endif // QUIC_API_ENABLE_PREVIEW_FEATURES
 
 TEST(ParameterValidation, ValidateConfiguration) {
 #ifdef QUIC_TEST_SCHANNEL_FLAGS

src/test/bin/winkernel/control.cpp

@@ -463,8 +463,10 @@ ExecuteTestRequest(
     RegisterTestFunction(QuicTestTlsHandshakeInfo);
     RegisterTestFunction(QuicTestStreamParam);
     RegisterTestFunction(QuicTestGetPerfCounters);
+#ifndef _KERNEL_MODE
     RegisterTestFunction(QuicTestGetWorkerStatistics);
     RegisterTestFunction(QuicTestValidateWorkerStatistics);
+#endif
     RegisterTestFunction(QuicTestValidateConfiguration);
     RegisterTestFunction(QuicTestValidateListener);
     RegisterTestFunction(QuicTestValidateConnection);

src/test/lib/ApiTest.cpp

@@ -6045,6 +6045,7 @@ QuicTestGetPerfCounters()
     TEST_EQUAL(BufferLength, (sizeof(uint64_t) * (QUIC_PERF_COUNTER_MAX - 4)));
 }
 
+#ifdef QUIC_API_ENABLE_PREVIEW_FEATURES
 void
 QuicTestGetWorkerStatistics()
 {
@@ -6225,6 +6226,7 @@ QuicTestValidateWorkerStatistics()
     delete[] BeforeBuffer;
     delete[] AfterBuffer;
 }
+#endif // QUIC_API_ENABLE_PREVIEW_FEATURES
 
 #ifdef QUIC_API_ENABLE_PREVIEW_FEATURES
 void