123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116 |
- // Copyright 2017 The go-ethereum Authors
- // This file is part of the go-ethereum library.
- //
- // The go-ethereum library is free software: you can redistribute it and/or modify
- // it under the terms of the GNU Lesser General Public License as published by
- // the Free Software Foundation, either version 3 of the License, or
- // (at your option) any later version.
- //
- // The go-ethereum library is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- // GNU Lesser General Public License for more details.
- //
- // You should have received a copy of the GNU Lesser General Public License
- // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
- package core
- import (
- "fmt"
- "math/big"
- "strings"
- "sync"
- "github.com/ethereum/go-ethereum/common"
- "github.com/ethereum/go-ethereum/consensus/istanbul"
- ibfttypes "github.com/ethereum/go-ethereum/consensus/istanbul/ibft/types"
- )
- // Construct a new message set to accumulate messages for given sequence/view number.
- func newMessageSet(valSet istanbul.ValidatorSet) *messageSet {
- return &messageSet{
- view: &istanbul.View{
- Round: new(big.Int),
- Sequence: new(big.Int),
- },
- messagesMu: new(sync.Mutex),
- messages: make(map[common.Address]*ibfttypes.Message),
- valSet: valSet,
- }
- }
- // ----------------------------------------------------------------------------
- type messageSet struct {
- view *istanbul.View
- valSet istanbul.ValidatorSet
- messagesMu *sync.Mutex
- messages map[common.Address]*ibfttypes.Message
- }
- func (ms *messageSet) View() *istanbul.View {
- return ms.view
- }
- func (ms *messageSet) Add(msg *ibfttypes.Message) error {
- ms.messagesMu.Lock()
- defer ms.messagesMu.Unlock()
- if err := ms.verify(msg); err != nil {
- return err
- }
- return ms.addVerifiedMessage(msg)
- }
- func (ms *messageSet) Values() (result []*ibfttypes.Message) {
- ms.messagesMu.Lock()
- defer ms.messagesMu.Unlock()
- for _, v := range ms.messages {
- result = append(result, v)
- }
- return result
- }
- func (ms *messageSet) Size() int {
- ms.messagesMu.Lock()
- defer ms.messagesMu.Unlock()
- return len(ms.messages)
- }
- func (ms *messageSet) Get(addr common.Address) *ibfttypes.Message {
- ms.messagesMu.Lock()
- defer ms.messagesMu.Unlock()
- return ms.messages[addr]
- }
- // ----------------------------------------------------------------------------
- func (ms *messageSet) verify(msg *ibfttypes.Message) error {
- // verify if the message comes from one of the validators
- if _, v := ms.valSet.GetByAddress(msg.Address); v == nil {
- return istanbul.ErrUnauthorizedAddress
- }
- // TODO: check view number and sequence number
- return nil
- }
- func (ms *messageSet) addVerifiedMessage(msg *ibfttypes.Message) error {
- ms.messages[msg.Address] = msg
- return nil
- }
- func (ms *messageSet) String() string {
- ms.messagesMu.Lock()
- defer ms.messagesMu.Unlock()
- addresses := make([]string, 0, len(ms.messages))
- for _, v := range ms.messages {
- addresses = append(addresses, v.Address.String())
- }
- return fmt.Sprintf("[%v]", strings.Join(addresses, ", "))
- }
|