feat: add Bytecode::original_bytecode_slice to match BytecodeLocked

bluealloy · Apr 9, 2024 · d896e0e · d896e0e
1 parent 1edfeb6
commit d896e0e
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/crates/primitives/src/bytecode.rs b/crates/primitives/src/bytecode.rs
@@ -92,7 +92,7 @@ impl Bytecode {
         if self.is_empty() {
             KECCAK_EMPTY
         } else {
-            keccak256(&self.original_bytes())
+            keccak256(self.original_bytecode_slice())
         }
     }
 
@@ -130,7 +130,18 @@ impl Bytecode {
         match self.state {
             BytecodeState::Raw => self.bytecode.clone(),
             BytecodeState::Checked { len } | BytecodeState::Analysed { len, .. } => {
-                self.bytecode.slice(0..len)
+                self.bytecode.slice(..len)
+            }
+        }
+    }
+
+    /// Returns the original bytecode as a byte slice.
+    #[inline]
+    pub fn original_bytecode_slice(&self) -> &[u8] {
+        match self.state {
+            BytecodeState::Raw => &self.bytecode,
+            BytecodeState::Checked { len } | BytecodeState::Analysed { len, .. } => {
+                &self.bytecode[..len]
             }
         }
     }