| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459 |
- use parity_scale_codec::{Decode, Encode};
- use rand::Rng;
- use crate::build_solidity;
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Val32(u32);
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Val8(u8);
- #[test]
- fn const_array_array() {
- let mut runtime = build_solidity(
- r##"
- contract foo {
- int8[8] constant bar = [ int8(1), 2, 3, 4, 5, 6, 7, 8 ];
- function f(uint32 i1) public returns (int8) {
- return bar[i1];
- }
- }"##,
- );
- runtime.function("f", Val32(1).encode());
- assert_eq!(runtime.vm.output, Val8(2).encode());
- }
- #[test]
- fn votes() {
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Votes([bool; 11]);
- let mut runtime = build_solidity(
- r##"
- contract foo {
- /// In a jury, do the ayes have it?
- function f(bool[11] votes) public pure returns (bool) {
- uint32 i;
- uint32 ayes = 0;
- for (i=0; i<votes.length; i++) {
- if (votes[i]) {
- ayes += 1;
- }
- }
- return ayes > votes.length / 2;
- }
- }"##,
- );
- runtime.function(
- "f",
- Votes([
- true, true, true, true, true, true, false, false, false, false, false,
- ])
- .encode(),
- );
- assert_eq!(runtime.vm.output, true.encode());
- runtime.function(
- "f",
- Votes([
- true, true, true, true, true, false, false, false, false, false, false,
- ])
- .encode(),
- );
- assert_eq!(runtime.vm.output, false.encode());
- }
- #[test]
- fn return_array() {
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Res([u64; 4]);
- let mut runtime = build_solidity(
- r##"
- contract foo {
- function array() pure public returns (int64[4]) {
- return [ int64(4), 84564, 31213, 1312 ];
- }
- }"##,
- );
- runtime.function("array", Vec::new());
- assert_eq!(runtime.vm.output, Res([4, 84564, 31213, 1312]).encode());
- }
- #[test]
- fn storage_arrays() {
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Val(i32);
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct SetArg(u64, i32);
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct GetArg(u64);
- let mut runtime = build_solidity(
- r##"
- contract foo {
- int32[8589934592] bigarray;
- function set(uint64 index, int32 val) public {
- bigarray[index] = val;
- }
- function get(uint64 index) public returns (int32) {
- return bigarray[index];
- }
- }"##,
- );
- let mut rng = rand::thread_rng();
- let mut vals = Vec::new();
- for _ in 0..100 {
- let index = rng.gen::<u64>() % 0x200_0000;
- let val = rng.gen::<i32>();
- runtime.function("set", SetArg(index, val).encode());
- vals.push((index, val));
- }
- for val in vals {
- runtime.function("get", GetArg(val.0).encode());
- assert_eq!(runtime.vm.output, Val(val.1).encode());
- }
- }
- #[test]
- fn enum_arrays() {
- #[derive(Encode, Decode)]
- struct Arg([u8; 100]);
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Ret(u32);
- let mut runtime = build_solidity(
- r##"
- contract enum_array {
- enum Foo { Bar1, Bar2, Bar3, Bar4 }
- function count_bar2(Foo[100] calldata foos) public returns (uint32) {
- uint32 count = 0;
- uint32 i;
- for (i = 0; i < foos.length; i++) {
- if (foos[i] == Foo.Bar2) {
- count++;
- }
- }
- return count;
- }
- }"##,
- );
- let mut rng = rand::thread_rng();
- let mut a = [0u8; 100];
- let mut count = 0;
- #[allow(clippy::needless_range_loop)]
- for i in 0..a.len() {
- a[i] = rng.gen::<u8>() % 4;
- if a[i] == 1 {
- count += 1;
- }
- }
- runtime.function("count_bar2", Arg(a).encode());
- assert_eq!(runtime.vm.output, Ret(count).encode());
- }
- #[test]
- fn storage_ref_arg() {
- let mut runtime = build_solidity(
- r##"
- contract storage_refs {
- int32[10] a;
- int32[10] b;
- function set(int32[10] storage array, uint8 index, int32 val) private {
- array[index] = val;
- }
- function test() public {
- set(a, 2, 5);
- set(b, 2, 102);
- assert(a[2] == 5);
- assert(b[2] == 102);
- }
- }"##,
- );
- runtime.function("test", Vec::new());
- }
- #[test]
- fn storage_ref_var() {
- let mut runtime = build_solidity(
- r##"
- contract storage_refs {
- int32[10] a;
- int32[10] b;
- function set(int32[10] storage array, uint8 index, int32 val) private {
- array[index] = val;
- }
- function test() public {
- int32[10] storage ref = a;
- set(ref, 2, 5);
- ref = b;
- set(ref, 2, 102);
- assert(a[2] == 5);
- assert(b[2] == 102);
- }
- }"##,
- );
- runtime.function("test", Vec::new());
- }
- #[test]
- fn storage_ref_returns() {
- let mut runtime = build_solidity(
- r##"
- contract storage_refs {
- int32[10] a;
- int32[10] b;
- function a_or_b(bool want_a) private returns (int32[10] storage) {
- if (want_a) {
- return a;
- }
- return b;
- }
- function set(int32[10] storage array, uint8 index, int32 val) private {
- array[index] = val;
- }
- function test() public {
- int32[10] storage ref = a_or_b(true);
- set(ref, 2, 5);
- ref = a_or_b(false);
- set(ref, 2, 102);
- assert(a[2] == 5);
- assert(b[2] == 102);
- }
- }"##,
- );
- runtime.function("test", Vec::new());
- }
- #[test]
- fn storage_to_memory() {
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Ret([u32; 10]);
- let mut runtime = build_solidity(
- r##"
- contract storage_refs {
- uint32[10] a;
- function test() public returns (uint32[10]) {
- for (uint32 i = 0; i < 10; ) {
- uint32 index = i;
- a[index] = 7 * ++i;
- }
- return a;
- }
- }"##,
- );
- runtime.function("test", Vec::new());
- let val = Ret([7, 14, 21, 28, 35, 42, 49, 56, 63, 70]);
- assert_eq!(runtime.vm.output, val.encode());
- }
- #[test]
- fn memory_to_storage() {
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Ret([u32; 10]);
- let mut runtime = build_solidity(
- r##"
- contract storage_refs {
- int32[10] a;
- function test() public returns (int32[10]) {
- int32[10] b = [ int32(7), 14, 21, 28, 35, 42, 49, 56, 63, 70 ];
- a = b;
- return a;
- }
- }"##,
- );
- runtime.function("test", Vec::new());
- let val = Ret([7, 14, 21, 28, 35, 42, 49, 56, 63, 70]);
- assert_eq!(runtime.vm.output, val.encode());
- }
- #[test]
- fn array_dimensions() {
- let mut runtime = build_solidity(
- r##"
- contract storage_refs {
- int32[2**16] a;
- function test() public {
- assert(a.length == 65536);
- }
- }"##,
- );
- runtime.function("test", Vec::new());
- }
- #[test]
- fn array_in_struct() {
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Ret([u32; 10]);
- let mut runtime = build_solidity(
- r##"
- contract storage_refs {
- struct foo {
- int32[10] f1;
- }
- function test() public returns (int32[10]) {
- foo a = foo({f1: [ int32(7), 14, 21, 28, 35, 42, 49, 56, 63, 0 ]});
- assert(a.f1[1] == 14);
- a.f1[9] = 70;
- return a.f1;
- }
- }"##,
- );
- runtime.function("test", Vec::new());
- let val = Ret([7, 14, 21, 28, 35, 42, 49, 56, 63, 70]);
- assert_eq!(runtime.vm.output, val.encode());
- }
- #[test]
- fn struct_in_array() {
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct S(u64, bool);
- let mut runtime = build_solidity(
- r##"
- struct S {
- uint64 f1;
- bool f2;
- }
- contract foo {
- S[] store;
- function set(S[] memory n) public {
- store = n;
- }
- function copy() public returns (S[] memory) {
- return store;
- }
- }"##,
- );
- let val = vec![S(102, true), S(u64::MAX, false)];
- runtime.function("set", val.encode());
- runtime.function("copy", vec![]);
- assert_eq!(runtime.vm.output, val.encode());
- }
- #[test]
- fn struct_in_fixed_array() {
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct S(u64, bool);
- let mut runtime = build_solidity(
- r##"
- struct S {
- uint64 f1;
- bool f2;
- }
- contract foo {
- S[2] store;
- function set(S[2] memory n) public {
- store = n;
- }
- function copy() public returns (S[2] memory) {
- return store;
- }
- }"##,
- );
- let val = [S(102, true), S(u64::MAX, false)];
- runtime.function("set", val.encode());
- runtime.function("copy", vec![]);
- assert_eq!(runtime.vm.output, val.encode());
- }
- #[test]
- fn struct_array_struct() {
- let mut runtime = build_solidity(
- r##"
- contract flipper {
- struct bool_struct {
- bool foo_bool;
- }
- struct struct_bool_struct_array {
- bool_struct[1] foo_struct_array;
- }
- function get_memory() public pure returns (bool) {
- bool_struct memory foo = bool_struct({foo_bool: true});
- bool_struct[1] memory foo_array = [foo];
- struct_bool_struct_array memory foo_struct = struct_bool_struct_array({foo_struct_array: foo_array});
- /* return foo_array[0].foo_bool; */
- return foo_struct.foo_struct_array[0].foo_bool;
- }
- }"##,
- );
- runtime.function("get_memory", Vec::new());
- assert_eq!(runtime.vm.output, true.encode());
- }
- #[test]
- fn struct_array_struct_abi() {
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Foo {
- f1: u32,
- f2: bool,
- }
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Bar {
- bars: [Foo; 10],
- }
- let mut runtime = build_solidity(
- r##"
- contract flipper {
- struct foo {
- int32 f1;
- bool f2;
- }
- struct bar {
- foo[10] bars;
- }
- bar s;
- function get_bar() public returns (bar) {
- bar memory a = bar({ bars: [
- foo({ f1: 1, f2: true}),
- foo({ f1: 2, f2: true}),
- foo({ f1: 3, f2: true}),
- foo({ f1: 4, f2: true}),
- foo({ f1: 5, f2: true}),
- foo({ f1: 6, f2: true}),
- foo({ f1: 7, f2: false}),
- foo({ f1: 8, f2: true}),
- foo({ f1: 9, f2: true}),
- foo({ f1: 10, f2: true})
- ]});
- s = a;
- return a;
- }
- function set_bar(bar memory a) public {
- for (uint32 i = 0; i < 10; i++) {
- assert(a.bars[i].f1 == i + 1);
- assert(a.bars[i].f2 == (i != 6));
- }
- a = s;
- }
- }"##,
- );
- let b = Bar {
- bars: [
- Foo { f1: 1, f2: true },
- Foo { f1: 2, f2: true },
- Foo { f1: 3, f2: true },
- Foo { f1: 4, f2: true },
- Foo { f1: 5, f2: true },
- Foo { f1: 6, f2: true },
- Foo { f1: 7, f2: false },
- Foo { f1: 8, f2: true },
- Foo { f1: 9, f2: true },
- Foo { f1: 10, f2: true },
- ],
- };
- runtime.function("get_bar", Vec::new());
- assert_eq!(runtime.vm.output, b.encode());
- runtime.function("set_bar", b.encode());
- }
- #[test]
- fn memory_dynamic_array_new() {
- let mut runtime = build_solidity(
- r#"
- contract foo {
- function test() public {
- int32[] memory a = new int32[](5);
- assert(a.length == 5);
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- // The Ethereum Foundation solc allows you to create arrays of length 0
- let mut runtime = build_solidity(
- r#"
- contract foo {
- function test() public {
- bool[] memory a = new bool[](0);
- assert(a.length == 0);
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- }
- #[test]
- fn memory_dynamic_array_deref() {
- // The Ethereum Foundation solc allows you to create arrays of length 0
- let mut runtime = build_solidity(
- r#"
- contract foo {
- function test() public {
- int32[] memory a = new int32[](5);
- assert(a.length == 5);
- a[0] = 102;
- a[1] = -5;
- a[4] = 0x7cafeeed;
- assert(a[0] == 102);
- assert(a[1] == -5);
- assert(a[4] == 0x7cafeeed);
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- }
- #[test]
- #[should_panic]
- fn array_bounds_dynamic_array() {
- let mut runtime = build_solidity(
- r#"
- contract foo {
- function test() public returns (int32) {
- int32[] memory a = new int32[](5);
- a[5] = 102;
- return a[3];
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- }
- #[test]
- #[should_panic]
- fn empty_array_bounds_dynamic_array() {
- let mut runtime = build_solidity(
- r#"
- contract foo {
- function test() public returns (bytes32) {
- bytes32[] memory a = new bytes32[](0);
- a[0] = "yo";
- return a[0];
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- }
- #[test]
- fn memory_dynamic_array_types_call_return() {
- let mut runtime = build_solidity(
- r#"
- contract foo {
- function a(bool cond) public returns (bytes27[]) {
- bytes27[] foo;
- foo = new bytes27[](5);
- foo[1] = "cond was true";
- return foo;
- }
- function b(bytes27[] x) private {
- x[1] = "b was called";
- x = new bytes27[](3);
- x[1] = "should not be";
- }
- function test() public {
- bytes27[] x = a(true);
- assert(x.length == 5);
- assert(x[1] == "cond was true");
- b(x);
- assert(x.length == 5);
- assert(x[1] == "b was called");
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- }
- #[test]
- fn dynamic_arrays_need_phi_node() {
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- enum bar { bar1, bar2, bar3, bar4 }
- function a(bool cond) public returns (bar[] memory) {
- bar[] memory foo;
- if (cond) {
- foo = new bar[](5);
- foo[1] = bar.bar2;
- } else {
- foo = new bar[](3);
- foo[1] = bar.bar3;
- }
- return foo;
- }
- function test() public {
- bar[] memory x = a(true);
- assert(x.length == 5);
- assert(x[1] == bar.bar2);
- x = a(false);
- assert(x.length == 3);
- assert(x[1] == bar.bar3);
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- }
- // test:
- // alignment of array elements
- // arrays of other structs/arrays/darrays
- // nil pointer should fail
- // copy to/from storage <=> memory
- // abi encode/decode
- // string/bytes
- #[test]
- fn storage_dynamic_array_length() {
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- int32[] bar;
- function test() public {
- assert(bar.length == 0);
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- }
- #[test]
- fn dynamic_array_push() {
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- function test() public {
- int[] bar = (new int[])(1);
- bar[0] = 128;
- bar.push(64);
- assert(bar.length == 2);
- assert(bar[1] == 64);
- }
- }
- "#,
- );
- runtime.function("test", Vec::new());
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- function test() public {
- bytes bar = (new bytes)(1);
- bar[0] = 128;
- bar.push(64);
- assert(bar.length == 2);
- assert(bar[1] == 64);
- }
- }
- "#,
- );
- runtime.function("test", Vec::new());
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- struct s {
- int32 f1;
- bool f2;
- }
- function test() public {
- s[] bar = new s[](1);
- bar[0] = s({f1: 0, f2: false});
- bar.push(s({f1: 1, f2: true}));
- assert(bar.length == 2);
- assert(bar[1].f1 == 1);
- assert(bar[1].f2 == true);
- }
- }
- "#,
- );
- runtime.function("test", Vec::new());
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- enum enum1 { val1, val2, val3 }
- function test() public {
- enum1[] bar = new enum1[](1);
- bar[0] = enum1.val1;
- bar.push(enum1.val2);
- assert(bar.length == 2);
- assert(bar[1] == enum1.val2);
- }
- }
- "#,
- );
- runtime.function("test", Vec::new());
- // push() returns a reference to the thing
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- struct s {
- int32 f1;
- bool f2;
- }
- function test() public {
- s[] bar = new s[](0);
- s memory n = bar.push();
- n.f1 = 102;
- n.f2 = true;
- assert(bar[0].f1 == 102);
- assert(bar[0].f2 == true);
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- }
- #[test]
- fn dynamic_array_pop() {
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- function test() public {
- int[] bar = new int[](1);
- bar[0] = 128;
- assert(bar.length == 1);
- assert(128 == bar.pop());
- assert(bar.length == 0);
- }
- }
- "#,
- );
- runtime.function("test", Vec::new());
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- function test() public {
- bytes bar = new bytes(1);
- bar[0] = 128;
- assert(bar.length == 1);
- assert(128 == bar.pop());
- assert(bar.length == 0);
- }
- }
- "#,
- );
- runtime.function("test", Vec::new());
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- struct s {
- int32 f1;
- bool f2;
- }
- function test() public {
- s[] bar = new s[](1);
- bar[0] = s(128, true);
- assert(bar.length == 1);
- s baz = bar.pop();
- assert(baz.f1 == 128);
- assert(baz.f2 == true);
- assert(bar.length == 0);
- }
- }
- "#,
- );
- runtime.function("test", Vec::new());
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- enum enum1 { val1, val2, val3 }
- function test() public {
- enum1[] bar = new enum1[](1);
- bar[0] = enum1.val2;
- assert(bar.length == 1);
- assert(enum1.val2 == bar.pop());
- assert(bar.length == 0);
- }
- }
- "#,
- );
- runtime.function("test", Vec::new());
- }
- #[test]
- #[should_panic]
- fn dynamic_array_pop_empty_array() {
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- function test() public {
- int[] bar = new int[](0);
- bar.pop();
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- }
- #[test]
- #[should_panic]
- fn dynamic_array_pop_bounds() {
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- function test() public {
- int[] bar = new int[](1);
- bar[0] = 12;
- bar.pop();
- assert(bar[0] == 12);
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- }
- #[test]
- fn storage_dynamic_array_push() {
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- int32[] bar;
- function test() public {
- assert(bar.length == 0);
- bar.push(102);
- assert(bar[0] == 102);
- assert(bar.length == 1);
- bar.push();
- assert(bar[1] == 0);
- assert(bar.length == 2);
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- // push() returns a reference to the thing
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- struct s {
- int32 f1;
- bool f2;
- }
- s[] bar;
- function test() public {
- s storage n = bar.push();
- n.f1 = 102;
- n.f2 = true;
- assert(bar[0].f1 == 102);
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- }
- #[test]
- fn storage_dynamic_array_pop() {
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- int32[] bar;
- function test() public {
- assert(bar.length == 0);
- bar.push(102);
- assert(bar[0] == 102);
- assert(bar.length == 1);
- int32 v = bar.pop();
- assert(bar.length == 0);
- assert(v == 102);
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- // We should have one entry for the length; pop should have removed the 102 entry
- assert_eq!(runtime.store.len(), 1);
- // ensure that structs and fixed arrays are wiped by pop
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- enum enum1 { val1, val2, val3 }
- struct s {
- bool f1;
- bytes3 f2;
- enum1 f3;
- uint64 f4;
- int64[2] f5;
- }
- s[] bar;
- function test() public {
- s storage first = bar.push();
- first.f1 = true;
- first.f2 = "abc";
- first.f3 = enum1.val2;
- first.f4 = 65536;
- first.f5[0] = -1;
- first.f5[1] = 5;
- assert(bar[0].f5[1] == 5);
- // now erase it again
- bar.pop();
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- // We should have one entry for the length; pop should have removed the 102 entry
- assert_eq!(runtime.store.len(), 1);
- }
- #[test]
- fn storage_delete() {
- // ensure that structs and fixed arrays are wiped by pop
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- uint64 bar;
- function test() public {
- bar = 0xdeaddeaddeaddead;
- delete bar;
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- // We should have one entry for the length; pop should have removed the 102 entry
- assert!(runtime.store.is_empty());
- // ensure that structs and fixed arrays are wiped by delete
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- enum enum1 { val1, val2, val3 }
- struct s {
- bool f1;
- bytes3 f2;
- enum1 f3;
- uint64 f4;
- int64[2] f5;
- }
- s[] bar;
- function test() public {
- s storage first = bar.push();
- first.f1 = true;
- first.f2 = "abc";
- first.f3 = enum1.val2;
- first.f4 = 65536;
- first.f5[0] = -1;
- first.f5[1] = 5;
- assert(bar[0].f5[1] == 5);
- // now erase it again
- delete bar[0];
- }
- }"#,
- );
- runtime.function("test", Vec::new());
- // We should have one entry for the length; delete should have removed the entry
- assert_eq!(runtime.store.len(), 1);
- // ensure that structs and fixed arrays are wiped by delete
- let mut runtime = build_solidity(
- r#"
- pragma solidity 0;
- contract foo {
- int[] bar;
- function setup() public {
- bar.push(102);
- bar.push(305);
- }
- function clear() public {
- delete bar;
- }
- }"#,
- );
- runtime.function("setup", Vec::new());
- assert_eq!(runtime.store.len(), 3);
- runtime.function("clear", Vec::new());
- assert_eq!(runtime.store.len(), 0);
- // our delete operator has to iterate over the dynamic array. Ensure it works if the array is empty
- runtime.function("clear", Vec::new());
- assert_eq!(runtime.store.len(), 0);
- }
- #[test]
- fn storage_dynamic_copy() {
- let mut runtime = build_solidity(
- r#"
- contract c {
- int32[] foo;
- constructor() public {
- for (int32 i = 0; i <11; i++) {
- foo.push(i * 3);
- }
- }
- function storage_to_memory() view public {
- int32[] memory bar = foo;
- assert(bar.length == 11);
- for (int32 i = 0; i <11; i++) {
- assert(bar[uint32(i)] == i * 3);
- }
- }
- function memory_to_storage() public {
- int32[] memory bar = new int32[](5);
- for (int32 i = 0; i < 5; i++) {
- bar[uint32(i)] = 5 * (i + 7);
- }
- foo = bar;
- assert(foo.length == 5);
- for (int32 i = 0; i < 5; i++) {
- assert(foo[uint32(i)] == 5 * (i + 7));
- }
- }
- }
- "#,
- );
- runtime.constructor(0, Vec::new());
- runtime.function("storage_to_memory", Vec::new());
- runtime.function("memory_to_storage", Vec::new());
- assert_eq!(runtime.store.len(), 6);
- }
- #[test]
- fn abi_encode_dynamic_array() {
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Int32Array(Vec<i32>);
- let mut runtime = build_solidity(
- r#"
- contract c {
- function encode() pure public returns (int32[]) {
- int32[] memory bar = new int32[](11);
- for (int32 i = 0; i <11; i++) {
- bar[uint32(i)] = i * 3;
- }
- return bar;
- }
- }
- "#,
- );
- runtime.constructor(0, Vec::new());
- runtime.function("encode", Vec::new());
- assert_eq!(
- runtime.vm.output,
- Int32Array(vec!(0, 3, 6, 9, 12, 15, 18, 21, 24, 27, 30)).encode()
- );
- }
- #[test]
- fn abi_decode_dynamic_array() {
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Int32Array(Vec<i32>);
- let mut runtime = build_solidity(
- r#"
- contract c {
- function decode(int32[] bar) pure public {
- assert(bar.length == 11);
- for (int32 i = 0; i <11; i++) {
- assert(bar[uint32(i)] == i * 3);
- }
- }
- function decode_empty(int32[] bar) pure public {
- assert(bar.length == 0);
- }
- }
- "#,
- );
- runtime.constructor(0, Vec::new());
- runtime.function(
- "decode",
- Int32Array(vec![0, 3, 6, 9, 12, 15, 18, 21, 24, 27, 30]).encode(),
- );
- runtime.function("decode_empty", Int32Array(vec![]).encode());
- }
- #[test]
- fn abi_encode_dynamic_array2() {
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Array(Vec<(bool, u32)>);
- let mut runtime = build_solidity(
- r#"
- contract structs {
- struct foo {
- bool x;
- uint32 y;
- }
- function test() public returns (foo[]) {
- foo[] x = new foo[](3);
- x[0] = foo({x: true, y: 64});
- x[1] = foo({x: false, y: 102});
- x[2] = foo({x: true, y: 0x800});
- return x;
- }
- }
- "#,
- );
- runtime.constructor(0, Vec::new());
- runtime.function("test", Vec::new());
- assert_eq!(
- runtime.vm.output,
- Array(vec!((true, 64), (false, 102), (true, 0x800))).encode()
- );
- }
- #[test]
- fn abi_encode_dynamic_array3() {
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Array(Vec<(bool, u32, String)>);
- let mut runtime = build_solidity(
- r#"
- contract structs {
- struct foo {
- bool x;
- uint32 y;
- string z;
- }
- function test() public returns (foo[]) {
- foo[] x = new foo[](3);
- x[0] = foo({x: true, y: 64, z: "abc"});
- x[1] = foo({x: false, y: 102, z: "a"});
- x[2] = foo({x: true, y: 0x800, z: "abcdef"});
- return x;
- }
- }
- "#,
- );
- runtime.constructor(0, Vec::new());
- runtime.function("test", Vec::new());
- assert_eq!(
- runtime.vm.output,
- Array(vec!(
- (true, 64, "abc".to_owned()),
- (false, 102, "a".to_owned()),
- (true, 0x800, "abcdef".to_owned())
- ))
- .encode()
- );
- }
- #[test]
- fn abi_encode_dynamic_array4() {
- #[derive(Debug, PartialEq, Encode, Decode)]
- struct Array([(bool, u32, String); 3]);
- let mut runtime = build_solidity(
- r#"
- contract structs {
- struct foo {
- bool x;
- uint32 y;
- string z;
- }
- function test() public returns (foo[3]) {
- foo[3] x;
- x[0] = foo({x: true, y: 64, z: "abc"});
- x[1] = foo({x: false, y: 102, z: "a"});
- x[2] = foo({x: true, y: 0x800, z: "abcdef"});
- return x;
- }
- }
- "#,
- );
- runtime.constructor(0, Vec::new());
- runtime.function("test", Vec::new());
- runtime.heap_verify();
- assert_eq!(
- runtime.vm.output,
- Array([
- (true, 64, "abc".to_owned()),
- (false, 102, "a".to_owned()),
- (true, 0x800, "abcdef".to_owned())
- ])
- .encode()
- );
- }
- #[test]
- fn large_index_ty_in_bounds() {
- let mut runtime = build_solidity(
- r#"
- contract foo {
- uint128 storage_index;
- function test(uint128 index) public returns (uint16) {
- uint16[] memory a = new uint16[](16);
- storage_index = index;
- return a[storage_index];
- }
- }"#,
- );
- runtime.constructor(0, Vec::new());
- runtime.function("test", 15u128.encode());
- runtime.function_expect_failure("test", 17u128.encode());
- runtime.function_expect_failure("test", 0xfffffffffffffu128.encode());
- }
- #[test]
- fn alloc_size_from_storage() {
- let mut runtime = build_solidity(
- r#"
- contract Test {
- uint32 length = 1;
- function contfunc() public view returns (uint64[] memory) {
- uint64[] memory values = new uint64[](length);
- return values;
- }
- }"#,
- );
- runtime.constructor(0, Vec::new());
- runtime.function("contfunc", Vec::new());
- assert_eq!(runtime.vm.output, vec![0u64].encode());
- }
|