paritytech · lexnv · Dec 3, 2024 · Dec 2, 2024 · Dec 3, 2024 · Dec 3, 2024
diff --git a/src/protocol/notification/mod.rs b/src/protocol/notification/mod.rs
@@ -435,18 +435,33 @@ impl NotificationProtocol {
                             },
                         );
                     }
-                    // user either initiated an outbound substream or an outbound substream was
-                    // opened/being opened as a result of an accepted inbound substream but was not
-                    // yet fully open
+                    // User initiated an outbound substream but the connection was closed before the
+                    // substream was fully open.
+                    // To have consistent state tracking in the user protocol, substream rejection
+                    // must be reported to the user.
+                    (OutboundState::OutboundInitiated { substream }, _) => {
+                        tracing::debug!(
+                            target: LOG_TARGET,
+                            ?peer,
+                            protocol = %self.protocol,
+                            "connection closed outbound substream initiated ",
+                        );
+                        // We need to remove this state to avoid a memory leak.
+                        self.pending_outbound.remove(&substream);
+
+                        self.event_handle
+                            .report_notification_stream_open_failure(
+                                peer,
+                                NotificationError::Rejected,
+                            )
+                            .await;
+                    }
+                    // An outbound substream was opened/being opened as a result of an accepted
+                    // inbound substream but was not yet fully open.
                     //
-                    // to have consistent state tracking in the user protocol, substream rejection
-                    // must be reported to the user
-                    (
-                        OutboundState::OutboundInitiated { .. }
-                        | OutboundState::Negotiating
-                        | OutboundState::Open { .. },
-                        _,
-                    ) => {
+                    // To have consistent state tracking in the user protocol, substream rejection
+                    // must be reported to the user.
+                    (OutboundState::Negotiating | OutboundState::Open { .. }, _) => {
                         tracing::debug!(
                             target: LOG_TARGET,
                             ?peer,