mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-06-23 15:14:32 +00:00
8363641b2c
* check against master node list before sending out anything * remove duplicated signatures from QC * add break when checking allowed to send
91 lines
3 KiB
Go
91 lines
3 KiB
Go
package tests
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/XinFinOrg/XDPoSChain/accounts"
|
|
"github.com/XinFinOrg/XDPoSChain/consensus/XDPoS"
|
|
"github.com/XinFinOrg/XDPoSChain/consensus/XDPoS/utils"
|
|
"github.com/XinFinOrg/XDPoSChain/params"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestShouldVerifyBlock(t *testing.T) {
|
|
b, err := json.Marshal(params.TestXDPoSMockChainConfig)
|
|
assert.Nil(t, err)
|
|
configString := string(b)
|
|
|
|
var config params.ChainConfig
|
|
err = json.Unmarshal([]byte(configString), &config)
|
|
assert.Nil(t, err)
|
|
// Enable verify
|
|
config.XDPoS.V2.SkipV2Validation = false
|
|
// Skip the mining time validation by set mine time to 0
|
|
config.XDPoS.V2.MinePeriod = 0
|
|
// Block 901 is the first v2 block with round of 1
|
|
blockchain, _, _, _, _, _ := PrepareXDCTestBlockChainForV2Engine(t, 910, &config, 0)
|
|
adaptor := blockchain.Engine().(*XDPoS.XDPoS)
|
|
|
|
// Happy path
|
|
err = adaptor.VerifyHeader(blockchain, blockchain.GetBlockByNumber(901).Header(), true)
|
|
assert.Nil(t, err)
|
|
|
|
// Verify non-epoch switch block
|
|
err = adaptor.VerifyHeader(blockchain, blockchain.GetBlockByNumber(902).Header(), true)
|
|
assert.Nil(t, err)
|
|
|
|
nonEpochSwitchWithValidators := blockchain.GetBlockByNumber(902).Header()
|
|
nonEpochSwitchWithValidators.Validators = acc1Addr.Bytes()
|
|
err = adaptor.VerifyHeader(blockchain, nonEpochSwitchWithValidators, true)
|
|
assert.Equal(t, utils.ErrInvalidFieldInNonEpochSwitch, err)
|
|
}
|
|
|
|
func TestShouldFailIfNotEnoughQCSignatures(t *testing.T) {
|
|
b, err := json.Marshal(params.TestXDPoSMockChainConfig)
|
|
assert.Nil(t, err)
|
|
configString := string(b)
|
|
|
|
var config params.ChainConfig
|
|
err = json.Unmarshal([]byte(configString), &config)
|
|
assert.Nil(t, err)
|
|
// Enable verify
|
|
config.XDPoS.V2.SkipV2Validation = false
|
|
// Skip the mining time validation by set mine time to 0
|
|
config.XDPoS.V2.MinePeriod = 0
|
|
// Block 901 is the first v2 block with round of 1
|
|
blockchain, _, currentBlock, signer, signFn, _ := PrepareXDCTestBlockChainForV2Engine(t, 902, &config, 0)
|
|
adaptor := blockchain.Engine().(*XDPoS.XDPoS)
|
|
|
|
parentBlock := blockchain.GetBlockByNumber(901)
|
|
proposedBlockInfo := &utils.BlockInfo{
|
|
Hash: parentBlock.Hash(),
|
|
Round: utils.Round(1),
|
|
Number: parentBlock.Number(),
|
|
}
|
|
signedHash, err := signFn(accounts.Account{Address: signer}, utils.VoteSigHash(proposedBlockInfo).Bytes())
|
|
assert.Nil(t, err)
|
|
var signatures []utils.Signature
|
|
// Duplicate the signatures
|
|
signatures = append(signatures, signedHash, signedHash, signedHash, signedHash, signedHash, signedHash)
|
|
quorumCert := &utils.QuorumCert{
|
|
ProposedBlockInfo: proposedBlockInfo,
|
|
Signatures: signatures,
|
|
}
|
|
|
|
extra := utils.ExtraFields_v2{
|
|
Round: utils.Round(2),
|
|
QuorumCert: quorumCert,
|
|
}
|
|
extraInBytes, err := extra.EncodeToBytes()
|
|
if err != nil {
|
|
panic(fmt.Errorf("Error encode extra into bytes: %v", err))
|
|
}
|
|
headerWithDuplicatedSignatures := currentBlock.Header()
|
|
headerWithDuplicatedSignatures.Extra = extraInBytes
|
|
// Happy path
|
|
err = adaptor.VerifyHeader(blockchain, headerWithDuplicatedSignatures, true)
|
|
assert.Equal(t, utils.ErrInvalidQC, err)
|
|
|
|
}
|