Another fix

microsoft · Nov 28, 2023 · 208d9b4 · 208d9b4
1 parent 2ab4711
commit 208d9b4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/core/connection.c b/src/core/connection.c
@@ -5836,9 +5836,9 @@ QuicConnRecvDatagrams(
         }
     }
 
-    if (!Connection->State.UpdateWorker &&
-        Connection->State.Connected &&
-        RecvState.UpdatePartitionId) {
+    if (!Connection->State.UpdateWorker && Connection->State.Connected &&
+        !Connection->State.ShutdownComplete && RecvState.UpdatePartitionId) {
+        CXPLAT_DBG_ASSERT(Connection->Registration);
         CXPLAT_DBG_ASSERT(!Connection->Registration->NoPartitioning);
         CXPLAT_DBG_ASSERT(RecvState.PartitionIndex != QuicPartitionIdGetIndex(Connection->PartitionID));
         Connection->PartitionID = QuicPartitionIdCreate(RecvState.PartitionIndex);