-
Notifications
You must be signed in to change notification settings - Fork 2
/
native.go
166 lines (140 loc) · 5 KB
/
native.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
// Copyright 2017 Monax Industries Limited
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package bvm
import (
"crypto/sha256"
"math/big"
"github.com/bcbchain/bclib/tendermint/tmlibs/common"
"github.com/bcbchain/bcbchain/hyperledger/burrow/execution/bvm/ecrypto"
"github.com/bcbchain/bcbchain/hyperledger/burrow/crypto"
"github.com/bcbchain/bclib/tendermint/tmlibs/log"
. "github.com/bcbchain/bcbchain/hyperledger/burrow/binary"
"github.com/bcbchain/bcbchain/hyperledger/burrow/execution/errors"
"golang.org/x/crypto/ripemd160"
)
const SignatureLength = 65
var registeredNativeContracts = make(map[crypto.BVMAddress]NativeContract)
func IsRegisteredNativeContract(address crypto.BVMAddress) bool {
_, ok := registeredNativeContracts[address]
return ok
}
func init() {
registerNativeContracts()
}
func registerNativeContracts() {
registeredNativeContracts[crypto.BVMAddress(Int64ToWord256(1).Word160())] = ecRecoverFunc
registeredNativeContracts[crypto.BVMAddress(Int64ToWord256(2).Word160())] = sha256Func
registeredNativeContracts[crypto.BVMAddress(Int64ToWord256(3).Word160())] = ripemd160Func
registeredNativeContracts[crypto.BVMAddress(Int64ToWord256(4).Word160())] = identityFunc
}
//-----------------------------------------------------------------------------
func ExecuteNativeContract(address crypto.BVMAddress, st Interface, caller crypto.BVMAddress, input []byte, gas *uint64,
logger log.Logger) ([]byte, errors.CodedError) {
contract, ok := registeredNativeContracts[address]
if !ok {
return nil, errors.ErrorCodef(errors.ErrorCodeNativeFunction,
"no native contract registered at address: %v", address)
}
output, err := contract(st, caller, input, gas, logger)
if err != nil {
return nil, errors.NewException(errors.ErrorCodeNativeFunction, err.Error())
}
return output, nil
}
type NativeContract func(state Interface, caller crypto.BVMAddress, input []byte, gas *uint64,
logger log.Logger) (output []byte, err error)
func ecRecoverFunc(state Interface, caller crypto.BVMAddress, input []byte, gas *uint64, logger log.Logger) (output []byte, err error) {
// Deduct gas
gasRequired := GasEcRecover
if *gas < gasRequired {
return nil, errors.ErrorCodeInsufficientGas
} else {
*gas -= gasRequired
}
// Recover
const ecRecoverInputLength = 128
input = common.RightPadBytes(input, ecRecoverInputLength)
// "input" is (hash, v, r, s), each 32 bytes
// but for ecrecover we want (r, s, v)
r := new(big.Int).SetBytes(input[64:96])
s := new(big.Int).SetBytes(input[96:128])
v := input[63] - 27
// tighter sig s values input homestead only apply to tx sigs
if !allZero(input[32:63]) || !ecrypto.ValidateSignatureValues(v, r, s, false) {
return nil, nil
}
hash := input[:32]
// We must make sure not to modify the 'input', so placing the 'v' along with
// the signature needs to be done on a new allocation
sig := make([]byte, 65)
copy(sig, input[64:128])
sig[64] = v
pub, err := ecrypto.EcRecover(hash, sig)
// make sure the public key is a valid one
if err != nil {
return nil, nil
}
// the first byte of pubKey is bitcoin heritage
return common.LeftPadBytes(ecrypto.Keccak256(pub[1:])[12:], 32), nil
}
func sha256Func(state Interface, caller crypto.BVMAddress, input []byte, gas *uint64,
logger log.Logger) (output []byte, err error) {
// Deduct gas
gasRequired := uint64((len(input)+31)/32)*GasSha256Word + GasSha256Base
if *gas < gasRequired {
return nil, errors.ErrorCodeInsufficientGas
} else {
*gas -= gasRequired
}
// Hash
hasher := sha256.New()
// CONTRACT: this does not err
hasher.Write(input)
return hasher.Sum(nil), nil
}
func ripemd160Func(state Interface, caller crypto.BVMAddress, input []byte, gas *uint64,
logger log.Logger) (output []byte, err error) {
// Deduct gas
gasRequired := uint64((len(input)+31)/32)*GasRipemd160Word + GasRipemd160Base
if *gas < gasRequired {
return nil, errors.ErrorCodeInsufficientGas
} else {
*gas -= gasRequired
}
// Hash
hasher := ripemd160.New()
// CONTRACT: this does not err
hasher.Write(input)
return LeftPadBytes(hasher.Sum(nil), 32), nil
}
func identityFunc(state Interface, caller crypto.BVMAddress, input []byte, gas *uint64,
logger log.Logger) (output []byte, err error) {
// Deduct gas
gasRequired := uint64((len(input)+31)/32)*GasIdentityWord + GasIdentityBase
if *gas < gasRequired {
return nil, errors.ErrorCodeInsufficientGas
} else {
*gas -= gasRequired
}
// Return identity
return input, nil
}
func allZero(b []byte) bool {
for _, byte := range b {
if byte != 0 {
return false
}
}
return true
}