Merge pull request #1167 from Gustav-Simonsson/check_ec_recover_err

Add missing err checks on From()
This commit is contained in:
Jeffrey Wilcke 2015-06-03 10:29:47 -07:00
commit 8610314918
4 changed files with 25 additions and 18 deletions

View File

@ -160,7 +160,6 @@ func (js *jsre) pendingTransactions(call otto.FunctionCall) otto.Value {
//ltxs := make([]*tx, len(txs)) //ltxs := make([]*tx, len(txs))
var ltxs []*tx var ltxs []*tx
for _, tx := range txs { for _, tx := range txs {
// no need to check err
if from, _ := tx.From(); accountSet.Has(from) { if from, _ := tx.From(); accountSet.Has(from) {
ltxs = append(ltxs, newTx(tx)) ltxs = append(ltxs, newTx(tx))
} }

View File

@ -62,7 +62,6 @@ type Message interface {
func AddressFromMessage(msg Message) common.Address { func AddressFromMessage(msg Message) common.Address {
from, _ := msg.From() from, _ := msg.From()
return crypto.CreateAddress(from, msg.Nonce()) return crypto.CreateAddress(from, msg.Nonce())
} }
@ -109,9 +108,12 @@ func NewStateTransition(env vm.Environment, msg Message, coinbase *state.StateOb
func (self *StateTransition) Coinbase() *state.StateObject { func (self *StateTransition) Coinbase() *state.StateObject {
return self.state.GetOrNewStateObject(self.coinbase) return self.state.GetOrNewStateObject(self.coinbase)
} }
func (self *StateTransition) From() *state.StateObject { func (self *StateTransition) From() (*state.StateObject, error) {
f, _ := self.msg.From() f, err := self.msg.From()
return self.state.GetOrNewStateObject(f) if err != nil {
return nil, err
}
return self.state.GetOrNewStateObject(f), nil
} }
func (self *StateTransition) To() *state.StateObject { func (self *StateTransition) To() *state.StateObject {
if self.msg == nil { if self.msg == nil {
@ -140,7 +142,10 @@ func (self *StateTransition) AddGas(amount *big.Int) {
func (self *StateTransition) BuyGas() error { func (self *StateTransition) BuyGas() error {
var err error var err error
sender := self.From() sender, err := self.From()
if err != nil {
return err
}
if sender.Balance().Cmp(MessageGasValue(self.msg)) < 0 { if sender.Balance().Cmp(MessageGasValue(self.msg)) < 0 {
return fmt.Errorf("insufficient ETH for gas (%x). Req %v, has %v", sender.Address().Bytes()[:4], MessageGasValue(self.msg), sender.Balance()) return fmt.Errorf("insufficient ETH for gas (%x). Req %v, has %v", sender.Address().Bytes()[:4], MessageGasValue(self.msg), sender.Balance())
} }
@ -159,10 +164,11 @@ func (self *StateTransition) BuyGas() error {
} }
func (self *StateTransition) preCheck() (err error) { func (self *StateTransition) preCheck() (err error) {
var ( msg := self.msg
msg = self.msg sender, err := self.From()
sender = self.From() if err != nil {
) return err
}
// Make sure this transaction's nonce is correct // Make sure this transaction's nonce is correct
if sender.Nonce() != msg.Nonce() { if sender.Nonce() != msg.Nonce() {
@ -185,10 +191,8 @@ func (self *StateTransition) transitionState() (ret []byte, usedGas *big.Int, er
return return
} }
var ( msg := self.msg
msg = self.msg sender, _ := self.From() // err checked in preCheck
sender = self.From()
)
// Pay intrinsic gas // Pay intrinsic gas
if err = self.UseGas(IntrinsicGas(self.msg)); err != nil { if err = self.UseGas(IntrinsicGas(self.msg)); err != nil {
@ -212,7 +216,7 @@ func (self *StateTransition) transitionState() (ret []byte, usedGas *big.Int, er
} else { } else {
// Increment the nonce for the next transaction // Increment the nonce for the next transaction
self.state.SetNonce(sender.Address(), sender.Nonce()+1) self.state.SetNonce(sender.Address(), sender.Nonce()+1)
ret, err = vmenv.Call(self.From(), self.To().Address(), self.msg.Data(), self.gas, self.gasPrice, self.value) ret, err = vmenv.Call(sender, self.To().Address(), self.msg.Data(), self.gas, self.gasPrice, self.value)
} }
if err != nil && IsValueTransferErr(err) { if err != nil && IsValueTransferErr(err) {
@ -226,7 +230,8 @@ func (self *StateTransition) transitionState() (ret []byte, usedGas *big.Int, er
} }
func (self *StateTransition) refundGas() { func (self *StateTransition) refundGas() {
coinbase, sender := self.Coinbase(), self.From() coinbase := self.Coinbase()
sender, _ := self.From() // err already checked
// Return remaining gas // Return remaining gas
remaining := new(big.Int).Mul(self.gas, self.msg.GasPrice()) remaining := new(big.Int).Mul(self.gas, self.msg.GasPrice())
sender.AddBalance(remaining) sender.AddBalance(remaining)

View File

@ -271,7 +271,7 @@ func (pool *TxPool) Stop() {
} }
func (self *TxPool) queueTx(tx *types.Transaction) { func (self *TxPool) queueTx(tx *types.Transaction) {
from, _ := tx.From() from, _ := tx.From() // already validated
self.queue[from] = append(self.queue[from], tx) self.queue[from] = append(self.queue[from], tx)
} }

View File

@ -139,6 +139,10 @@ type Transaction struct {
} }
func NewTx(tx *types.Transaction) *Transaction { func NewTx(tx *types.Transaction) *Transaction {
sender, err := tx.From()
if err != nil {
return nil
}
hash := tx.Hash().Hex() hash := tx.Hash().Hex()
var receiver string var receiver string
@ -147,7 +151,6 @@ func NewTx(tx *types.Transaction) *Transaction {
} else { } else {
receiver = core.AddressFromMessage(tx).Hex() receiver = core.AddressFromMessage(tx).Hex()
} }
sender, _ := tx.From()
createsContract := core.MessageCreatesContract(tx) createsContract := core.MessageCreatesContract(tx)
var data string var data string