Guard against sid != MULTIPLEXER_SID

nats-io · Aug 14, 2023 · a0b7a58 · a0b7a58
1 parent 618cf14
commit a0b7a58
Showing 1 changed file with 20 additions and 18 deletions.
diff --git a/async-nats/src/lib.rs b/async-nats/src/lib.rs
@@ -501,24 +501,26 @@ impl ConnectionHandler {
                             self.handle_flush().await?;
                         }
                     }
-                } else if let Some(multiplexer) = self.multiplexer.as_mut() {
-                    let maybe_token = subject.strip_prefix(&multiplexer.prefix).to_owned();
-
-                    if let Some(token) = maybe_token {
-                        if let Some(sender) = multiplexer.senders.remove(token) {
-                            let message = Message {
-                                subject,
-                                reply,
-                                payload,
-                                headers,
-                                status,
-                                description,
-                                length,
-                            };
-
-                            sender.send(message).map_err(|_| {
-                                io::Error::new(io::ErrorKind::Other, "request receiver closed")
-                            })?;
+                } else if sid == MULTIPLEXER_SID {
+                    if let Some(multiplexer) = self.multiplexer.as_mut() {
+                        let maybe_token = subject.strip_prefix(&multiplexer.prefix).to_owned();
+
+                        if let Some(token) = maybe_token {
+                            if let Some(sender) = multiplexer.senders.remove(token) {
+                                let message = Message {
+                                    subject,
+                                    reply,
+                                    payload,
+                                    headers,
+                                    status,
+                                    description,
+                                    length,
+                                };
+
+                                sender.send(message).map_err(|_| {
+                                    io::Error::new(io::ErrorKind::Other, "request receiver closed")
+                                })?;
+                            }
                         }
                     }
                 }