fixup! Merge branch 'grarco/nullifier-uniqueness' (#2240)

anoma · Dec 12, 2023 · bd4374b · bd4374b
1 parent 42b0595
commit bd4374b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/core/src/types/token.rs b/core/src/types/token.rs
@@ -1203,13 +1203,17 @@ pub fn is_denom_key(token_addr: &Address, key: &Key) -> bool {
 
 /// Check if the given storage key is a masp key
 pub fn is_masp_key(key: &Key) -> bool {
-    matches!(&key.segments[..],
+    if key.segments.len() >= 2 {
+        matches!(&key.segments[..2],
         [DbKeySeg::AddressSeg(addr), DbKeySeg::StringSeg(key)]
             if *addr == MASP
                 && (key == HEAD_TX_KEY
                     || key.starts_with(TX_KEY_PREFIX)
                     || key.starts_with(PIN_KEY_PREFIX)
                     || key.starts_with(MASP_NULLIFIERS_KEY_PREFIX)))
+    } else {
+        false
+    }
 }
 
 /// Check if the given storage key is a masp nullifier key