difftreelog
fix AbiWrite implementations
in: master
4 files changed
crates/evm-coder/procedural/src/abi_derive.rsdiffbeforeafterboth137 let abi_write = if is_named_fields {137 let abi_write = if is_named_fields {138 quote!(138 quote!(139 #(139 #(140 self.#field_names.abi_write(writer);140 self.#field_names.abi_write(sub);141 )*141 )*142 )142 )143 } else {143 } else {146 .map(proc_macro2::Literal::usize_unsuffixed);146 .map(proc_macro2::Literal::usize_unsuffixed);147 quote!(147 quote!(148 #(148 #(149 self.#field_names.abi_write(writer);149 self.#field_names.abi_write(sub);150 )*150 )*151 )151 )152 };152 };153 quote!(153 quote!(154 impl ::evm_coder::abi::AbiWrite for #name {154 impl ::evm_coder::abi::AbiWrite for #name {155 fn abi_write(&self, writer: &mut ::evm_coder::abi::AbiWriter) {155 fn abi_write(&self, writer: &mut ::evm_coder::abi::AbiWriter) {156 if <Self as ::evm_coder::abi::AbiType>::is_dynamic() {157 let mut sub = ::evm_coder::abi::AbiWriter::new();158 {159 let sub = &mut sub;160 #abi_write161 }162 writer.write_subresult(sub);163 } else {164 let sub = writer;156 #abi_write165 #abi_write166 }157 }167 }158 }168 }159 )169 )crates/evm-coder/src/abi/impls.rsdiffbeforeafterboth310 #[allow(non_snake_case)]310 #[allow(non_snake_case)]311 impl<$($ident),+> AbiWrite for ($($ident,)+)311 impl<$($ident),+> AbiWrite for ($($ident,)+)312 where312 where313 $($ident: AbiWrite,)+313 $($ident: AbiWrite + AbiType,)+314 {314 {315 fn abi_write(&self, writer: &mut AbiWriter) {315 fn abi_write(&self, writer: &mut AbiWriter) {316 let ($($ident,)+) = self;316 let ($($ident,)+) = self;317 if writer.is_dynamic {317 if <Self as AbiType>::is_dynamic() {318 let mut sub = AbiWriter::new();318 let mut sub = AbiWriter::new();319 $($ident.abi_write(&mut sub);)+319 $($ident.abi_write(&mut sub);)+320 writer.write_subresult(sub);320 writer.write_subresult(sub);crates/evm-coder/src/lib.rsdiffbeforeafterboth137 #[cfg(feature = "std")]137 #[cfg(feature = "std")]138 pub type string = ::std::string::String;138 pub type string = ::std::string::String;139139140 #[derive(Default, Debug, PartialEq)]140 #[derive(Default, Debug, PartialEq, Clone)]141 pub struct bytes(pub Vec<u8>);141 pub struct bytes(pub Vec<u8>);142142143 /// Solidity doesn't have `void` type, however we have special implementation143 /// Solidity doesn't have `void` type, however we have special implementationcrates/evm-coder/tests/abi_derive_generation.rsdiffbeforeafterboth1use evm_coder_procedural::AbiCoder;1use evm_coder_procedural::AbiCoder;2use evm_coder::{2use evm_coder::{3 types::*,3 types::*,4 abi::{AbiType},4 abi::{AbiType, AbiRead, AbiWrite},5};5};667// TODO: move to build_failed tests7// TODO: move to build_failed tests8// #[derive(AbiCoder)]8// #[derive(AbiCoder, PartialEq, Debug)]9// struct TypeStructUnit {}9// struct TypeStructUnit {}101011#[derive(AbiCoder)]11#[derive(AbiCoder, PartialEq, Debug)]12struct TypeStruct1SimpleParam {12struct TypeStruct1SimpleParam {13 _a: u8,13 _a: u8,14}14}151516#[derive(AbiCoder)]16#[derive(AbiCoder, PartialEq, Debug)]17struct TypeStruct1DynamicParam {17struct TypeStruct1DynamicParam {18 _a: String,18 _a: String,19}19}202021#[derive(AbiCoder)]21#[derive(AbiCoder, PartialEq, Debug)]22struct TypeStruct2SimpleParam {22struct TypeStruct2SimpleParam {23 _a: u8,23 _a: u8,24 _b: u32,24 _b: u32,25}25}262627#[derive(AbiCoder)]27#[derive(AbiCoder, PartialEq, Debug)]28struct TypeStruct2DynamicParam {28struct TypeStruct2DynamicParam {29 _a: String,29 _a: String,30 _b: bytes,30 _b: bytes,31}31}323233#[derive(AbiCoder)]33#[derive(AbiCoder, PartialEq, Debug)]34struct TypeStruct2MixedParam {34struct TypeStruct2MixedParam {35 _a: u8,35 _a: u8,36 _b: bytes,36 _b: bytes,37}37}383839#[derive(AbiCoder)]39#[derive(AbiCoder, PartialEq, Debug)]40struct TypeStruct1DerivedSimpleParam {40struct TypeStruct1DerivedSimpleParam {41 _a: TypeStruct1SimpleParam,41 _a: TypeStruct1SimpleParam,42}42}434344#[derive(AbiCoder)]44#[derive(AbiCoder, PartialEq, Debug)]45struct TypeStruct2DerivedSimpleParam {45struct TypeStruct2DerivedSimpleParam {46 _a: TypeStruct1SimpleParam,46 _a: TypeStruct1SimpleParam,47 _b: TypeStruct2SimpleParam,47 _b: TypeStruct2SimpleParam,48}48}494950#[derive(AbiCoder)]50#[derive(AbiCoder, PartialEq, Debug)]51struct TypeStruct1DerivedDynamicParam {51struct TypeStruct1DerivedDynamicParam {52 _a: TypeStruct1DynamicParam,52 _a: TypeStruct1DynamicParam,53}53}545455#[derive(AbiCoder)]55#[derive(AbiCoder, PartialEq, Debug)]56struct TypeStruct2DerivedDynamicParam {56struct TypeStruct2DerivedDynamicParam {57 _a: TypeStruct1DynamicParam,57 _a: TypeStruct1DynamicParam,58 _b: TypeStruct2DynamicParam,58 _b: TypeStruct2DynamicParam,59}59}606061#[derive(AbiCoder)]61#[derive(AbiCoder, PartialEq, Debug)]62struct TypeStruct3DerivedMixedParam {62struct TypeStruct3DerivedMixedParam {63 _a: TypeStruct1SimpleParam,63 _a: TypeStruct1SimpleParam,64 _b: TypeStruct2DynamicParam,64 _b: TypeStruct2DynamicParam,200 );200 );201}201}202202203#[derive(AbiCoder)]203#[derive(AbiCoder, PartialEq, Debug)]204struct TupleStruct1SimpleParam(u8);204struct TupleStruct1SimpleParam(u8);205205206#[derive(AbiCoder)]206#[derive(AbiCoder, PartialEq, Debug)]207struct TupleStruct1DynamicParam(String);207struct TupleStruct1DynamicParam(String);208208209#[derive(AbiCoder)]209#[derive(AbiCoder, PartialEq, Debug)]210struct TupleStruct2SimpleParam(u8, u32);210struct TupleStruct2SimpleParam(u8, u32);211211212#[derive(AbiCoder)]212#[derive(AbiCoder, PartialEq, Debug)]213struct TupleStruct2DynamicParam(String, bytes);213struct TupleStruct2DynamicParam(String, bytes);214214215#[derive(AbiCoder)]215#[derive(AbiCoder, PartialEq, Debug)]216struct TupleStruct2MixedParam(u8, bytes);216struct TupleStruct2MixedParam(u8, bytes);217217218#[derive(AbiCoder)]218#[derive(AbiCoder, PartialEq, Debug)]219struct TupleStruct1DerivedSimpleParam(TupleStruct1SimpleParam);219struct TupleStruct1DerivedSimpleParam(TupleStruct1SimpleParam);220220221#[derive(AbiCoder)]221#[derive(AbiCoder, PartialEq, Debug)]222struct TupleStruct2DerivedSimpleParam(TupleStruct1SimpleParam, TupleStruct2SimpleParam);222struct TupleStruct2DerivedSimpleParam(TupleStruct1SimpleParam, TupleStruct2SimpleParam);223223224#[derive(AbiCoder)]224#[derive(AbiCoder, PartialEq, Debug)]225struct TupleStruct1DerivedDynamicParam(TupleStruct1DynamicParam);225struct TupleStruct1DerivedDynamicParam(TupleStruct1DynamicParam);226226227#[derive(AbiCoder)]227#[derive(AbiCoder, PartialEq, Debug)]228struct TupleStruct2DerivedDynamicParam(TupleStruct1DynamicParam, TupleStruct2DynamicParam);228struct TupleStruct2DerivedDynamicParam(TupleStruct1DynamicParam, TupleStruct2DynamicParam);229229230#[derive(AbiCoder)]230#[derive(AbiCoder, PartialEq, Debug)]231struct TupleStruct3DerivedMixedParam(231struct TupleStruct3DerivedMixedParam(232 TupleStruct1SimpleParam,232 TupleStruct1SimpleParam,233 TupleStruct2DynamicParam,233 TupleStruct2DynamicParam,406 );406 );407}407}408408409// #[test]409fn test_impl<TypeStruct, TupleStruct, Tuple>(410// fn impl_abi_read() {410 type_struct_data: TypeStruct,411// TypeStruct1SimpleParam::411 tuple_struct_data: TupleStruct,412// }412 tuple_data: Tuple,413) where414 TypeStruct: AbiWrite + AbiRead + std::cmp::PartialEq + std::fmt::Debug,415 Tuple: AbiWrite + AbiRead + std::cmp::PartialEq + std::fmt::Debug,416{417 use evm_coder::abi::{AbiReader, AbiWriter};418 const FUNCTION_IDENTIFIER: u32 = 0xdeadbeef;419420 let mut writer = AbiWriter::new_call(FUNCTION_IDENTIFIER);421 tuple_data.abi_write(&mut writer);422 let encoded_tuple = writer.finish();423424 let mut writer = AbiWriter::new_call(FUNCTION_IDENTIFIER);425 type_struct_data.abi_write(&mut writer);426 let encoded_struct = writer.finish();427428 similar_asserts::assert_eq!(encoded_tuple, encoded_struct);429430 // let (_, mut decoder) = AbiReader::new_call(&encoded_tuple).unwrap();431 // let restored_struct_data = <TypeStruct>::abi_read(&mut decoder).unwrap();432 // assert_eq!(restored_struct_data, type_struct_data);433434 // let (_, mut decoder) = AbiReader::new_call(&encoded_struct).unwrap();435 // let restored_tuple_data = <Tuple>::abi_read(&mut decoder).unwrap();436 // assert_eq!(restored_tuple_data, tuple_data);437}438439#[test]440fn codec_struct_1_simple() {441 let _a = 0xff;442 test_impl::<TypeStruct1SimpleParam, TupleStruct1SimpleParam, (uint8,)>(443 TypeStruct1SimpleParam { _a },444 TupleStruct1SimpleParam(_a),445 (_a,),446 );447}448449#[test]450fn codec_struct_1_dynamic() {451 let _a: String = "some string".into();452 test_impl::<TypeStruct1DynamicParam, TupleStruct1DynamicParam, (String,)>(453 TypeStruct1DynamicParam { _a: _a.clone() },454 TupleStruct1DynamicParam(_a.clone()),455 (_a,),456 );457}458459#[test]460fn codec_struct_2_dynamic() {461 let _a: String = "some string".into();462 let _b: bytes = bytes(vec![0x11, 0x22, 0x33]);463 test_impl::<TypeStruct2DynamicParam, TupleStruct2DynamicParam, (String, bytes)>(464 TypeStruct2DynamicParam {465 _a: _a.clone(),466 _b: _b.clone(),467 },468 TupleStruct2DynamicParam(_a.clone(), _b.clone()),469 (_a, _b),470 );471}413472