Fix AV in netebpfext (#3769)

* fix * fix * fix
microsoft · Aug 14, 2024 · 03e8c10 · 03e8c10
1 parent 94a66d6
commit 03e8c10
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/netebpfext/net_ebpf_ext_sock_addr.c b/netebpfext/net_ebpf_ext_sock_addr.c
@@ -991,15 +991,17 @@ _Requires_exclusive_lock_held_(_net_ebpf_ext_sock_addr_blocked_contexts
 
 #pragma warning(suppress : 6001) /* entry and list entry are non-null */
         RemoveEntryList(&entry->list_entry);
+        uint64_t transport_endpoint_handle = entry->transport_endpoint_handle;
         BOOLEAN result =
             RtlDeleteElementGenericTableAvl(&_net_ebpf_ext_sock_addr_blocked_contexts.blocked_context_table, entry);
         ebpf_assert(result);
+        entry = NULL;
         _net_ebpf_ext_sock_addr_blocked_contexts.blocked_context_count--;
         NET_EBPF_EXT_LOG_MESSAGE_UINT64(
             NET_EBPF_EXT_TRACELOG_LEVEL_VERBOSE,
             NET_EBPF_EXT_TRACELOG_KEYWORD_SOCK_ADDR,
             "_net_ebpf_ext_purge_block_connect_contexts: Delete",
-            entry->transport_endpoint_handle);
+            transport_endpoint_handle);
     }
 
     // Free entries from low-memory list.