check that the buffer has been read before returning contract

input-output-hk · Oct 26, 2021 · db10b94 · db10b94
1 parent c6d06b8
commit db10b94
Showing 1 changed file with 13 additions and 9 deletions.
diff --git a/chain-impl-mockchain/src/smartcontract/mod.rs b/chain-impl-mockchain/src/smartcontract/mod.rs
@@ -83,15 +83,19 @@ impl Readable for Contract {
                             return Err(ReadError::StructureInvalid("Invalid byte sequence".into()))
                         }
                     };
-                    let contract = Contract::EVM {
-                        from,
-                        to,
-                        gas,
-                        gas_price,
-                        value,
-                        data,
-                    };
-                    Ok(contract)
+
+                    if let Err(e) = buf.expect_end() {
+                        Err(e)
+                    } else {
+                        Ok(Contract::EVM {
+                            from,
+                            to,
+                            gas,
+                            gas_price,
+                            value,
+                            data,
+                        })
+                    }
                 }
             }
             n => {