Sanitize address before inbox ID lookup

xmtp · Jun 24, 2024 · 1c40c6c · 1c40c6c
1 parent 1e09690
commit 1c40c6c
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/xmtp_mls/src/client.rs b/xmtp_mls/src/client.rs
@@ -270,8 +270,15 @@ where
         &self,
         address: String,
     ) -> Result<Option<String>, ClientError> {
-        let mut results = self.api_client.get_inbox_ids(vec![address.clone()]).await?;
-        Ok(results.remove(&address))
+        if let Some(sanitized_address) = sanitize_evm_addresses(vec![address])?.pop() {
+            let mut results = self
+                .api_client
+                .get_inbox_ids(vec![sanitized_address.clone()])
+                .await?;
+            Ok(results.remove(&sanitized_address))
+        } else {
+            Ok(None)
+        }
     }
 
     /// Get sequence id, may not be consistent with the backend