Handle epoch_enter at APC_LEVEL

Signed-off-by: Alan Jowett <alanjo@microsoft.com>
microsoft · Mar 16, 2023 · 5f0501a · 5f0501a
1 parent 2cbb2de
commit 5f0501a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/libs/platform/ebpf_state.c b/libs/platform/ebpf_state.c
@@ -92,7 +92,7 @@ _ebpf_state_get_entry(
     // High frequency call, don't log entry/exit.
     ebpf_state_entry_t* local_entry = NULL;
 
-    if (execution_context_state->current_irql == PASSIVE_LEVEL) {
+    if (execution_context_state->current_irql <= DISPATCH_LEVEL >) {
         ebpf_result_t return_value;
         uint64_t current_thread_id = execution_context_state->id.thread;
 

diff --git a/libs/platform/kernel/ebpf_platform_kernel.c b/libs/platform/kernel/ebpf_platform_kernel.c
@@ -390,7 +390,7 @@ _Requires_lock_held_(*lock) _Releases_lock_(*lock) _IRQL_requires_(DISPATCH_LEVE
 _Must_inspect_result_ ebpf_result_t
 ebpf_set_current_thread_affinity(uintptr_t new_thread_affinity_mask, _Out_ uintptr_t* old_thread_affinity_mask)
 {
-    if (KeGetCurrentIrql() > PASSIVE_LEVEL) {
+    if (KeGetCurrentIrql() >= DISPATCH_LEVEL) {
         return EBPF_OPERATION_NOT_SUPPORTED;
     }
 
@@ -808,7 +808,7 @@ bool
 ebpf_should_yield_processor()
 {
     // Don't yield if we are at passive level as the scheduler can preempt us.
-    if (KeGetCurrentIrql() == PASSIVE_LEVEL)
+    if (KeGetCurrentIrql() <= DISPATCH_LEVEL)
         return false;
 
     // KeShouldYieldProcessor returns TRUE if the current thread should yield the processor.