From bdfdd7cd028518e0bbbc6a7e0ed8322190b80582 Mon Sep 17 00:00:00 2001 From: Wei-Ning Huang Date: Sat, 17 Nov 2018 13:18:34 +0800 Subject: core: more fix on light node synchronization (#32) Fix gas calculation in governance contract. Correctly register round height when processing pending blocks. We should register the mapping when we get the pending block instead of waiting for block confirmation. --- core/vm/governance.go | 92 +++++++++++++++++++++------------------------------ 1 file changed, 37 insertions(+), 55 deletions(-) (limited to 'core/vm') diff --git a/core/vm/governance.go b/core/vm/governance.go index cbc7f458f..b6a5038c5 100644 --- a/core/vm/governance.go +++ b/core/vm/governance.go @@ -1498,8 +1498,16 @@ func newGovernanceContract(evm *EVM, contract *Contract) *GovernanceContract { } } -func (g *GovernanceContract) penalize() { - g.contract.UseGas(g.contract.Gas) +func (g *GovernanceContract) UseGas(gas uint64) ([]byte, error) { + if !g.contract.UseGas(gas) { + return nil, ErrOutOfGas + } + return nil, nil +} + +func (g *GovernanceContract) penalize() ([]byte, error) { + g.UseGas(g.contract.Gas) + return nil, errExecutionReverted } func (g *GovernanceContract) inDKGSet(round *big.Int, nodeID coreTypes.NodeID) bool { @@ -1532,16 +1540,14 @@ func (g *GovernanceContract) inDKGSet(round *big.Int, nodeID coreTypes.NodeID) b func (g *GovernanceContract) addDKGComplaint(round *big.Int, comp []byte) ([]byte, error) { if round.Cmp(g.state.Round()) != 0 { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } caller := g.contract.Caller() // Finalized caller is not allowed to propose complaint. if g.state.DKGFinalized(round, caller) { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } // Calculate 2f @@ -1556,34 +1562,29 @@ func (g *GovernanceContract) addDKGComplaint(round *big.Int, comp []byte) ([]byt var dkgComplaint dkgTypes.Complaint if err := rlp.DecodeBytes(comp, &dkgComplaint); err != nil { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } // DKGComplaint must belongs to someone in DKG set. if !g.inDKGSet(round, dkgComplaint.ProposerID) { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } verified, _ := core.VerifyDKGComplaintSignature(&dkgComplaint) if !verified { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } g.state.PushDKGComplaint(round, comp) // Set this to relatively high to prevent spamming - g.contract.UseGas(10000000) - return nil, nil + return g.UseGas(5000000) } func (g *GovernanceContract) addDKGMasterPublicKey(round *big.Int, mpk []byte) ([]byte, error) { // Can only add DKG master public key of current and next round. if round.Cmp(new(big.Int).Add(g.state.Round(), big.NewInt(1))) > 0 { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } caller := g.contract.Caller() @@ -1596,53 +1597,44 @@ func (g *GovernanceContract) addDKGMasterPublicKey(round *big.Int, mpk []byte) ( var dkgMasterPK dkgTypes.MasterPublicKey if err := rlp.DecodeBytes(mpk, &dkgMasterPK); err != nil { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } // DKGMasterPublicKey must belongs to someone in DKG set. if !g.inDKGSet(round, dkgMasterPK.ProposerID) { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } verified, _ := core.VerifyDKGMasterPublicKeySignature(&dkgMasterPK) if !verified { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } g.state.PushDKGMasterPublicKey(round, mpk) - // DKG operation is expensive. - g.contract.UseGas(100000) - return nil, nil + return g.UseGas(100000) } func (g *GovernanceContract) addDKGFinalize(round *big.Int, finalize []byte) ([]byte, error) { if round.Cmp(g.state.Round()) != 0 { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } caller := g.contract.Caller() var dkgFinalize dkgTypes.Finalize if err := rlp.DecodeBytes(finalize, &dkgFinalize); err != nil { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } // DKGFInalize must belongs to someone in DKG set. if !g.inDKGSet(round, dkgFinalize.ProposerID) { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } verified, _ := core.VerifyDKGFinalizeSignature(&dkgFinalize) if !verified { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } if !g.state.DKGFinalized(round, caller) { @@ -1650,9 +1642,7 @@ func (g *GovernanceContract) addDKGFinalize(round *big.Int, finalize []byte) ([] g.state.IncDKGFinalizedsCount(round) } - // DKG operation is expensive. - g.contract.UseGas(100000) - return nil, nil + return g.UseGas(100000) } func (g *GovernanceContract) updateConfiguration(config *params.DexconConfig) ([]byte, error) { @@ -1671,8 +1661,7 @@ func (g *GovernanceContract) stake( // Reject invalid inputs. if len(name) >= 32 || len(email) >= 32 || len(location) >= 32 || len(url) >= 128 { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } caller := g.contract.Caller() @@ -1680,14 +1669,12 @@ func (g *GovernanceContract) stake( // Need to stake at least minStake. if g.contract.Value().Cmp(g.state.MinStake()) < 0 { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } // Can not stake if already staked. if offset.Cmp(big.NewInt(0)) >= 0 { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } offset = g.state.NodesLength() @@ -1702,8 +1689,7 @@ func (g *GovernanceContract) stake( }) g.state.PutOffset(caller, offset) - g.contract.UseGas(21000) - return nil, nil + return g.UseGas(0) } func (g *GovernanceContract) unstake() ([]byte, error) { @@ -1729,8 +1715,7 @@ func (g *GovernanceContract) unstake() ([]byte, error) { // TODO(w): add this to debug trace so it shows up as internal transaction. g.evm.Transfer(g.evm.StateDB, GovernanceContractAddress, caller, node.Staked) - g.contract.UseGas(21000) - return nil, nil + return g.UseGas(0) } func (g *GovernanceContract) proposeCRS(nextRound *big.Int, signedCRS []byte) ([]byte, error) { @@ -1781,8 +1766,7 @@ func (g *GovernanceContract) proposeCRS(nextRound *big.Int, signedCRS []byte) ([ Signature: signedCRS, } if !dkgGPK.VerifySignature(coreCommon.Hash(prevCRS), signature) { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } // Save new CRS into state and increase round. @@ -1794,8 +1778,7 @@ func (g *GovernanceContract) proposeCRS(nextRound *big.Int, signedCRS []byte) ([ // To encourage DKG set to propose the correct value, correctly submitting // this should cause nothing. - g.contract.UseGas(0) - return nil, nil + return g.UseGas(0) } func (g *GovernanceContract) transferOwnership(newOwner common.Address) ([]byte, error) { @@ -1812,20 +1795,19 @@ func (g *GovernanceContract) snapshotRound(round, height *big.Int) ([]byte, erro if g.evm.IsBlockProposer() { realHeight, ok := g.evm.Context.GetRoundHeight(round.Uint64()) if !ok { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } if height.Cmp(new(big.Int).SetUint64(realHeight)) != 0 { - g.penalize() - return nil, errExecutionReverted + return g.penalize() } } // Only allow updating the next round. nextRound := g.state.LenRoundHeight() if round.Cmp(nextRound) != 0 { - g.penalize() + // No need to penalize, since the only possibility at this point is the + // round height is already snapshoted. return nil, errExecutionReverted } -- cgit v1.2.3