github.com/chike0905/go-ethereum@v1.9.7/signer/fourbyte/validation_test.go (about) 1 // Copyright 2019 The go-ethereum Authors 2 // This file is part of the go-ethereum library. 3 // 4 // The go-ethereum library is free software: you can redistribute it and/or modify 5 // it under the terms of the GNU Lesser General Public License as published by 6 // the Free Software Foundation, either version 3 of the License, or 7 // (at your option) any later version. 8 // 9 // The go-ethereum library is distributed in the hope that it will be useful, 10 // but WITHOUT ANY WARRANTY; without even the implied warranty of 11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12 // GNU Lesser General Public License for more details. 13 // 14 // You should have received a copy of the GNU Lesser General Public License 15 // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>. 16 17 package fourbyte 18 19 import ( 20 "math/big" 21 "testing" 22 23 "github.com/ethereum/go-ethereum/common" 24 "github.com/ethereum/go-ethereum/common/hexutil" 25 "github.com/ethereum/go-ethereum/signer/core" 26 ) 27 28 func mixAddr(a string) (*common.MixedcaseAddress, error) { 29 return common.NewMixedcaseAddressFromString(a) 30 } 31 func toHexBig(h string) hexutil.Big { 32 b := big.NewInt(0).SetBytes(common.FromHex(h)) 33 return hexutil.Big(*b) 34 } 35 func toHexUint(h string) hexutil.Uint64 { 36 b := big.NewInt(0).SetBytes(common.FromHex(h)) 37 return hexutil.Uint64(b.Uint64()) 38 } 39 func dummyTxArgs(t txtestcase) *core.SendTxArgs { 40 to, _ := mixAddr(t.to) 41 from, _ := mixAddr(t.from) 42 n := toHexUint(t.n) 43 gas := toHexUint(t.g) 44 gasPrice := toHexBig(t.gp) 45 value := toHexBig(t.value) 46 var ( 47 data, input *hexutil.Bytes 48 ) 49 if t.d != "" { 50 a := hexutil.Bytes(common.FromHex(t.d)) 51 data = &a 52 } 53 if t.i != "" { 54 a := hexutil.Bytes(common.FromHex(t.i)) 55 input = &a 56 57 } 58 return &core.SendTxArgs{ 59 From: *from, 60 To: to, 61 Value: value, 62 Nonce: n, 63 GasPrice: gasPrice, 64 Gas: gas, 65 Data: data, 66 Input: input, 67 } 68 } 69 70 type txtestcase struct { 71 from, to, n, g, gp, value, d, i string 72 expectErr bool 73 numMessages int 74 } 75 76 func TestTransactionValidation(t *testing.T) { 77 var ( 78 // use empty db, there are other tests for the abi-specific stuff 79 db = newEmpty() 80 ) 81 testcases := []txtestcase{ 82 // Invalid to checksum 83 {from: "000000000000000000000000000000000000dead", to: "000000000000000000000000000000000000dead", 84 n: "0x01", g: "0x20", gp: "0x40", value: "0x01", numMessages: 1}, 85 // valid 0x000000000000000000000000000000000000dEaD 86 {from: "000000000000000000000000000000000000dead", to: "0x000000000000000000000000000000000000dEaD", 87 n: "0x01", g: "0x20", gp: "0x40", value: "0x01", numMessages: 0}, 88 // conflicting input and data 89 {from: "000000000000000000000000000000000000dead", to: "0x000000000000000000000000000000000000dEaD", 90 n: "0x01", g: "0x20", gp: "0x40", value: "0x01", d: "0x01", i: "0x02", expectErr: true}, 91 // Data can't be parsed 92 {from: "000000000000000000000000000000000000dead", to: "0x000000000000000000000000000000000000dEaD", 93 n: "0x01", g: "0x20", gp: "0x40", value: "0x01", d: "0x0102", numMessages: 1}, 94 // Data (on Input) can't be parsed 95 {from: "000000000000000000000000000000000000dead", to: "0x000000000000000000000000000000000000dEaD", 96 n: "0x01", g: "0x20", gp: "0x40", value: "0x01", i: "0x0102", numMessages: 1}, 97 // Send to 0 98 {from: "000000000000000000000000000000000000dead", to: "0x0000000000000000000000000000000000000000", 99 n: "0x01", g: "0x20", gp: "0x40", value: "0x01", numMessages: 1}, 100 // Create empty contract (no value) 101 {from: "000000000000000000000000000000000000dead", to: "", 102 n: "0x01", g: "0x20", gp: "0x40", value: "0x00", numMessages: 1}, 103 // Create empty contract (with value) 104 {from: "000000000000000000000000000000000000dead", to: "", 105 n: "0x01", g: "0x20", gp: "0x40", value: "0x01", expectErr: true}, 106 // Small payload for create 107 {from: "000000000000000000000000000000000000dead", to: "", 108 n: "0x01", g: "0x20", gp: "0x40", value: "0x01", d: "0x01", numMessages: 1}, 109 } 110 for i, test := range testcases { 111 msgs, err := db.ValidateTransaction(nil, dummyTxArgs(test)) 112 if err == nil && test.expectErr { 113 t.Errorf("Test %d, expected error", i) 114 for _, msg := range msgs.Messages { 115 t.Logf("* %s: %s", msg.Typ, msg.Message) 116 } 117 } 118 if err != nil && !test.expectErr { 119 t.Errorf("Test %d, unexpected error: %v", i, err) 120 } 121 if err == nil { 122 got := len(msgs.Messages) 123 if got != test.numMessages { 124 for _, msg := range msgs.Messages { 125 t.Logf("* %s: %s", msg.Typ, msg.Message) 126 } 127 t.Errorf("Test %d, expected %d messages, got %d", i, test.numMessages, got) 128 } else { 129 //Debug printout, remove later 130 for _, msg := range msgs.Messages { 131 t.Logf("* [%d] %s: %s", i, msg.Typ, msg.Message) 132 } 133 t.Log() 134 } 135 } 136 } 137 }