1 files changed, 119 insertions, 0 deletions
diff --git a/network/netadaptercx/netvadapterlibrary/code/enlthreads.h b/network/netadaptercx/netvadapterlibrary/code/enlthreads.h
new file mode 100644
index 00000000..062d728e
--- /dev/null
+++ b/network/netadaptercx/netvadapterlibrary/code/enlthreads.h
@@ -0,0 +1,119 @@
+// Copyright (C) Microsoft Corporation. All rights reserved.
+#pragma once
+
+#include <wil/resource.h>
+// #include <KMacros.h>
+
+#ifdef _KERNEL_MODE
+    using ENL_START_ROUTINE = KSTART_ROUTINE;
+    using ENL_THREAD_ROUTINE_RETURN = VOID;
+    using ENL_THREAD = PKTHREAD;
+    // static const EC_THREAD EC_THREAD_INVALID = nullptr;
+    using ENL_THREAD_PRIORITY = LONG;
+    // #define EC_INVALID_THREAD_PRIORITY MAXLONG
+
+    using unique_zw_handle = wil::unique_any<HANDLE, decltype(&::ZwClose), &::ZwClose>;
+    using unique_pkthread = wil::unique_any<PKTHREAD, decltype(&ObfDereferenceObject), &ObfDereferenceObject>;
+    using unique_pkevent = wil::unique_any<PKEVENT, decltype(&ObfDereferenceObject), &ObfDereferenceObject>;
+    using unique_completionport = wil::unique_any<void *, decltype(&ObfDereferenceObject), &ObfDereferenceObject>;
+
+    struct unique_thread
+    {
+        unique_zw_handle
+            NtHandle;
+
+        unique_pkthread
+            ObHandle;
+
+        operator bool(
+            void
+        ) const;
+
+        void reset(
+        );
+    };
+#else
+
+#ifndef NOTHING
+#define NOTHING
+#endif
+
+#ifndef AFFINITY_MASK
+#define AFFINITY_MASK(n) ((KAFFINITY)1 << (n))
+#endif
+
+typedef struct _NDTBUS_POWER_INTERFACE_STANDARD
+{
+} NDTBUS_POWER_INTERFACE_STANDARD;
+
+_IRQL_requires_same_
+typedef DWORD (WINAPI ENL_START_ROUTINE)(
+    LPVOID lpThreadParameter
+);
+using ENL_THREAD_ROUTINE_RETURN = DWORD;
+using ENL_THREAD = DWORD;
+// static const EC_THREAD EC_THREAD_INVALID = 0;
+using ENL_THREAD_PRIORITY = int;
+// #define EC_INVALID_THREAD_PRIORITY THREAD_PRIORITY_ERROR_RETURN
+using unique_thread = wil::unique_handle;
+
+#define KeGetProcessorIndexFromNumber(_processor) (_processor)->Number
+#define KeMemoryBarrier() MemoryBarrier()
+
+inline
+NTSTATUS
+KeGetProcessorNumberFromIndex (
+    _In_ ULONG ProcIndex,
+    _Out_ PPROCESSOR_NUMBER ProcNumber
+    )
+{
+    ProcNumber->Number = static_cast<UCHAR>(ProcIndex);
+    ProcNumber->Group = 0;
+    ProcNumber->Reserved = 0; 
+
+    return STATUS_SUCCESS;
+}
+
+inline
+VOID
+KeQueryNodeActiveAffinity (
+    __in USHORT NodeNumber,
+    __out_opt PGROUP_AFFINITY Affinity,
+    __out_opt PUSHORT Count
+)
+{
+    UNREFERENCED_PARAMETER(NodeNumber);
+    UNREFERENCED_PARAMETER(Count);
+    
+    GetThreadGroupAffinity(GetCurrentThread(), Affinity);
+}
+#endif
+
+ENL_THREAD
+EnlGetCurrentThread(
+    void
+);
+
+NTSTATUS
+EnlThreadCreate(
+    _In_ ENL_START_ROUTINE StartRoutine,
+    _In_opt_ void * Context,
+    _Out_ unique_thread & Thread
+);
+
+void
+EnlThreadSetPriority(
+    _In_ unique_thread & Thread,
+    _In_ ENL_THREAD_PRIORITY Priority
+);
+
+void
+EnlThreadSetAffinity(
+    _In_ PGROUP_AFFINITY GroupAffinity,
+    _Out_opt_ PGROUP_AFFINITY PreviousAffinity
+);
+
+void
+EnlThreadWaitForTermination(
+    _In_ unique_thread & Thread
+);