fix(hermes): fix ip whitelisting logic

pyth-network · Oct 11, 2023 · eb2fd83 · eb2fd83
1 parent bb922c3
commit eb2fd83
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/hermes/src/api/ws.rs b/hermes/src/api/ws.rs
@@ -294,7 +294,8 @@ impl Subscriber {
             if !self
                 .ws_state
                 .bytes_limit_whitelist
-                .contains(&self.ip_addr.into())
+                .iter()
+                .any(|ip_net| ip_net.contains(&self.ip_addr))
                 && self.ws_state.rate_limiter.check_key_n(
                     &self.ip_addr,
                     NonZeroU32::new(message.len().try_into()?).ok_or(anyhow!("Empty message"))?,