robustness

0xPolygonZero · Oct 17, 2024 · 146ed08 · 146ed08
1 parent 3cab56b
commit 146ed08
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/trace_decoder/src/type2.rs b/trace_decoder/src/type2.rs
@@ -173,8 +173,10 @@ fn visit(
             address,
             value,
         }) => {
+            ensure!(address.len() == Address::len_bytes());
             let address = Address::from_slice(&address);
             let collated = leaves.entry(address).or_default();
+            ensure!(value.len() <= 32);
             let value = U256::from_big_endian(&value);
             macro_rules! ensure {
                 ($expr:expr) => {
@@ -195,6 +197,7 @@ fn visit(
                     collated.code = Some(value)
                 }
                 SmtLeafType::Storage(slot) => {
+                    ensure!(slot.len() <= 32);
                     let clobbered = collated.storage.insert(U256::from_big_endian(&slot), value);
                     ensure!(clobbered.is_none())
                 }

diff --git a/trace_decoder/src/wire.rs b/trace_decoder/src/wire.rs
@@ -80,17 +80,13 @@ pub enum Instruction {
 }
 
 #[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
-// TODO(0xaatif): https://github.com/0xPolygonZero/zk_evm/issues/705
-//                `address` and `value` should be fixed length fields
 pub struct SmtLeaf {
     pub node_type: SmtLeafType,
     pub address: NonEmpty<Vec<u8>>,
     pub value: NonEmpty<Vec<u8>>,
 }
 
 #[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
-// TODO(0xaatif): https://github.com/0xPolygonZero/zk_evm/issues/705
-//                `Storage` should contain a fixed length field
 pub enum SmtLeafType {
     Balance,
     Nonce,