Switch to more performant API

Signed-off-by: Alan Jowett <alan.jowett@microsoft.com>
microsoft · Oct 22, 2024 · 5641ea7 · 5641ea7
1 parent 14df17e
commit 5641ea7
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 10 deletions.
diff --git a/libs/runtime/ebpf_platform.c b/libs/runtime/ebpf_platform.c
@@ -59,16 +59,6 @@ ebpf_is_preemptible()
     return irql < DISPATCH_LEVEL;
 }
 
-uint32_t
-ebpf_get_current_cpu()
-{
-    PROCESSOR_NUMBER processor_number;
-
-    KeGetCurrentProcessorNumberEx(&processor_number);
-
-    return KeGetProcessorIndexFromNumber(&processor_number);
-}
-
 uint64_t
 ebpf_get_current_thread_id()
 {

diff --git a/libs/runtime/ebpf_platform.h b/libs/runtime/ebpf_platform.h
@@ -264,6 +264,7 @@ extern "C"
      *   running on. Only valid if ebpf_is_preemptible() == true.
      * @retval Zero based index of CPUs.
      */
+    EBPF_INLINE_HINT
     uint32_t
     ebpf_get_current_cpu();
 

diff --git a/libs/runtime/kernel/ebpf_platform_kernel.c b/libs/runtime/kernel/ebpf_platform_kernel.c
@@ -262,3 +262,9 @@ _IRQL_requires_max_(PASSIVE_LEVEL) _Must_inspect_result_ ebpf_result_t
 
     return EBPF_SUCCESS;
 }
+
+uint32_t
+ebpf_get_current_cpu()
+{
+    return KeGetCurrentProcessorIndex();
+}
diff --git a/libs/runtime/user/ebpf_platform_user.cpp b/libs/runtime/user/ebpf_platform_user.cpp
@@ -281,3 +281,13 @@ _IRQL_requires_max_(PASSIVE_LEVEL) _Must_inspect_result_ ebpf_result_t
     NTSTATUS status = usersim_platform_get_authentication_id(authentication_id);
     return ntstatus_to_ebpf_result(status);
 }
+
+uint32_t
+ebpf_get_current_cpu()
+{
+    PROCESSOR_NUMBER processor_number;
+
+    KeGetCurrentProcessorNumberEx(&processor_number);
+
+    return KeGetProcessorIndexFromNumber(&processor_number);
+}