static
1 year ago
commit
3d7a9318d1
@ -1,8 +1,127 @@ |
|||||||
package core |
package core |
||||||
|
|
||||||
type leaderRotationMeta struct { |
import ( |
||||||
pub []byte |
"bytes" |
||||||
epoch uint64 |
"hash/crc32" |
||||||
count uint64 |
"strconv" |
||||||
shifts uint64 |
"strings" |
||||||
|
|
||||||
|
"github.com/harmony-one/harmony/block" |
||||||
|
"github.com/harmony-one/harmony/crypto/bls" |
||||||
|
"github.com/harmony-one/harmony/internal/utils" |
||||||
|
"github.com/pkg/errors" |
||||||
|
) |
||||||
|
|
||||||
|
// LeaderRotationMeta contains information about leader rotation
|
||||||
|
type LeaderRotationMeta struct { |
||||||
|
Pub []byte // bls public key of previous block miner
|
||||||
|
Epoch uint64 // epoch number of previously inserted block
|
||||||
|
Count uint64 // quantity of continuous blocks inserted by the same leader
|
||||||
|
Shifts uint64 // number of leader shifts, shift happens when leader changes
|
||||||
|
} |
||||||
|
|
||||||
|
// ShortString returns string representation of the struct
|
||||||
|
func (a LeaderRotationMeta) ShortString() string { |
||||||
|
s := strings.Builder{} |
||||||
|
s.Write(a.Pub[3:]) |
||||||
|
s.WriteString(" ") |
||||||
|
s.WriteString(strconv.FormatUint(a.Epoch, 10)) |
||||||
|
s.WriteString(" ") |
||||||
|
s.WriteString(strconv.FormatUint(a.Count, 10)) |
||||||
|
s.WriteString(" ") |
||||||
|
s.WriteString(strconv.FormatUint(a.Shifts, 10)) |
||||||
|
return s.String() |
||||||
|
} |
||||||
|
|
||||||
|
// Hash returns hash of the struct
|
||||||
|
func (a LeaderRotationMeta) Hash() []byte { |
||||||
|
c := crc32.NewIEEE() |
||||||
|
c.Write(a.Pub) |
||||||
|
c.Write([]byte(strconv.FormatUint(a.Epoch, 10))) |
||||||
|
c.Write([]byte(strconv.FormatUint(a.Count, 10))) |
||||||
|
c.Write([]byte(strconv.FormatUint(a.Shifts, 10))) |
||||||
|
return c.Sum(nil) |
||||||
|
} |
||||||
|
|
||||||
|
// Clone returns a copy of the struct
|
||||||
|
func (a LeaderRotationMeta) Clone() LeaderRotationMeta { |
||||||
|
return LeaderRotationMeta{ |
||||||
|
Pub: append([]byte{}, a.Pub...), |
||||||
|
Epoch: a.Epoch, |
||||||
|
Count: a.Count, |
||||||
|
Shifts: a.Shifts, |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// buildLeaderRotationMeta builds leader rotation meta if feature is activated.
|
||||||
|
func (bc *BlockChainImpl) buildLeaderRotationMeta(curHeader *block.Header) error { |
||||||
|
if !bc.chainConfig.IsLeaderRotationInternalValidators(curHeader.Epoch()) { |
||||||
|
return nil |
||||||
|
} |
||||||
|
if curHeader.NumberU64() == 0 { |
||||||
|
return errors.New("current header is genesis") |
||||||
|
} |
||||||
|
curPubKey, err := bc.getLeaderPubKeyFromCoinbase(curHeader) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
for i := curHeader.NumberU64() - 1; i >= 0; i-- { |
||||||
|
header := bc.GetHeaderByNumber(i) |
||||||
|
if header == nil { |
||||||
|
return errors.New("header is nil") |
||||||
|
} |
||||||
|
blockPubKey, err := bc.getLeaderPubKeyFromCoinbase(header) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
if curPubKey.Bytes != blockPubKey.Bytes || curHeader.Epoch().Uint64() != header.Epoch().Uint64() { |
||||||
|
for j := i; j <= curHeader.NumberU64(); j++ { |
||||||
|
header := bc.GetHeaderByNumber(j) |
||||||
|
if header == nil { |
||||||
|
return errors.New("header is nil") |
||||||
|
} |
||||||
|
err := bc.saveLeaderRotationMeta(header) |
||||||
|
if err != nil { |
||||||
|
utils.Logger().Error().Err(err).Msg("save leader continuous blocks count error") |
||||||
|
return err |
||||||
|
} |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
|
} |
||||||
|
return errors.New("no leader rotation meta to save") |
||||||
|
} |
||||||
|
|
||||||
|
// saveLeaderRotationMeta saves leader rotation meta if feature is activated.
|
||||||
|
func (bc *BlockChainImpl) saveLeaderRotationMeta(h *block.Header) error { |
||||||
|
blockPubKey, err := bc.getLeaderPubKeyFromCoinbase(h) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
bc.leaderRotationMeta = processRotationMeta(h.Epoch().Uint64(), blockPubKey.Bytes, bc.leaderRotationMeta) |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func processRotationMeta(epoch uint64, blockPubKey bls.SerializedPublicKey, s LeaderRotationMeta) LeaderRotationMeta { |
||||||
|
// increase counter only if the same leader and epoch
|
||||||
|
if bytes.Equal(s.Pub, blockPubKey[:]) && s.Epoch == epoch { |
||||||
|
s.Count++ |
||||||
|
} else { |
||||||
|
s.Count = 1 |
||||||
|
} |
||||||
|
// we should increase shifts if the leader has changed.
|
||||||
|
if !bytes.Equal(s.Pub, blockPubKey[:]) { |
||||||
|
s.Shifts++ |
||||||
|
} |
||||||
|
// but set to zero if new
|
||||||
|
if s.Epoch != epoch { |
||||||
|
s.Shifts = 0 |
||||||
|
} |
||||||
|
s.Epoch = epoch |
||||||
|
return LeaderRotationMeta{ |
||||||
|
Pub: blockPubKey[:], |
||||||
|
Epoch: s.Epoch, |
||||||
|
Count: s.Count, |
||||||
|
Shifts: s.Shifts, |
||||||
|
} |
||||||
} |
} |
||||||
|
|||||||
@ -0,0 +1,57 @@ |
|||||||
|
package core |
||||||
|
|
||||||
|
import ( |
||||||
|
"testing" |
||||||
|
|
||||||
|
"github.com/harmony-one/harmony/crypto/bls" |
||||||
|
"github.com/stretchr/testify/require" |
||||||
|
) |
||||||
|
|
||||||
|
var k1 = bls.SerializedPublicKey{1, 2, 3} |
||||||
|
|
||||||
|
func TestRotationMetaProcess(t *testing.T) { |
||||||
|
t.Run("same_leader_increase_count", func(t *testing.T) { |
||||||
|
rs := processRotationMeta(1, bls.SerializedPublicKey{}, LeaderRotationMeta{ |
||||||
|
Pub: bls.SerializedPublicKey{}.Bytes(), |
||||||
|
Epoch: 1, |
||||||
|
Count: 1, |
||||||
|
Shifts: 1, |
||||||
|
}) |
||||||
|
require.Equal(t, LeaderRotationMeta{ |
||||||
|
Pub: bls.SerializedPublicKey{}.Bytes(), |
||||||
|
Epoch: 1, |
||||||
|
Count: 2, |
||||||
|
Shifts: 1, |
||||||
|
}, rs) |
||||||
|
}) |
||||||
|
|
||||||
|
t.Run("new_leader_increase_shifts", func(t *testing.T) { |
||||||
|
rs := processRotationMeta(1, k1, LeaderRotationMeta{ |
||||||
|
Pub: bls.SerializedPublicKey{}.Bytes(), |
||||||
|
Epoch: 1, |
||||||
|
Count: 1, |
||||||
|
Shifts: 1, |
||||||
|
}) |
||||||
|
require.Equal(t, LeaderRotationMeta{ |
||||||
|
Pub: k1.Bytes(), |
||||||
|
Epoch: 1, |
||||||
|
Count: 1, |
||||||
|
Shifts: 2, |
||||||
|
}, rs) |
||||||
|
}) |
||||||
|
|
||||||
|
t.Run("new_epoch_reset_count", func(t *testing.T) { |
||||||
|
rs := processRotationMeta(2, k1, LeaderRotationMeta{ |
||||||
|
Pub: bls.SerializedPublicKey{}.Bytes(), |
||||||
|
Epoch: 1, |
||||||
|
Count: 1, |
||||||
|
Shifts: 1, |
||||||
|
}) |
||||||
|
require.Equal(t, LeaderRotationMeta{ |
||||||
|
Pub: k1.Bytes(), |
||||||
|
Epoch: 2, |
||||||
|
Count: 1, |
||||||
|
Shifts: 0, |
||||||
|
}, rs) |
||||||
|
}) |
||||||
|
} |
||||||
Loading…
Reference in new issue