2017-08-28 14:43:44 -07:00
|
|
|
package raft
|
|
|
|
|
|
|
|
import (
|
|
|
|
"io"
|
|
|
|
"net"
|
|
|
|
|
|
|
|
"fmt"
|
2019-01-29 03:08:50 -08:00
|
|
|
"github.com/ethereum/go-ethereum/p2p/enode"
|
|
|
|
"github.com/ethereum/go-ethereum/p2p/enr"
|
2017-08-28 14:43:44 -07:00
|
|
|
"github.com/ethereum/go-ethereum/rlp"
|
2019-02-08 01:29:01 -08:00
|
|
|
"log"
|
2017-08-28 14:43:44 -07:00
|
|
|
)
|
|
|
|
|
|
|
|
// Serializable information about a Peer. Sufficient to build `etcdRaft.Peer`
|
2019-02-26 00:33:29 -08:00
|
|
|
// or `enode.Node`.
|
|
|
|
// As NodeId is mainly used to derive the `ecdsa.pubkey` to build `enode.Node` it is kept as [64]byte instead of ID [32]byte used by `enode.Node`.
|
2017-08-28 14:43:44 -07:00
|
|
|
type Address struct {
|
2019-02-08 01:29:01 -08:00
|
|
|
RaftId uint16 `json:"raftId"`
|
|
|
|
NodeId enode.EnodeID `json:"nodeId"`
|
|
|
|
Ip net.IP `json:"ip"`
|
|
|
|
P2pPort enr.TCP `json:"p2pPort"`
|
2019-02-24 23:01:53 -08:00
|
|
|
RaftPort enr.RaftPort `json:"raftPort"`
|
2017-08-28 14:43:44 -07:00
|
|
|
}
|
|
|
|
|
2019-01-29 03:08:50 -08:00
|
|
|
func newAddress(raftId uint16, raftPort int, node *enode.Node) *Address {
|
2019-02-25 00:30:54 -08:00
|
|
|
// derive 64 byte nodeID from 128 byte enodeID
|
2019-02-08 01:29:01 -08:00
|
|
|
id, err := enode.RaftHexID(node.EnodeID())
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
2017-08-28 14:43:44 -07:00
|
|
|
return &Address{
|
2018-09-11 12:05:42 -07:00
|
|
|
RaftId: raftId,
|
2019-02-08 01:29:01 -08:00
|
|
|
NodeId: id,
|
2019-01-29 03:08:50 -08:00
|
|
|
Ip: node.IP(),
|
|
|
|
P2pPort: enr.TCP(node.TCP()),
|
2019-02-24 23:01:53 -08:00
|
|
|
RaftPort: enr.RaftPort(raftPort),
|
2017-08-28 14:43:44 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// A peer that we're connected to via both raft's http transport, and ethereum p2p
|
|
|
|
type Peer struct {
|
2019-01-29 03:08:50 -08:00
|
|
|
address *Address // For raft transport
|
|
|
|
p2pNode *enode.Node // For ethereum transport
|
2017-08-28 14:43:44 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
func (addr *Address) EncodeRLP(w io.Writer) error {
|
2018-09-11 12:05:42 -07:00
|
|
|
return rlp.Encode(w, []interface{}{addr.RaftId, addr.NodeId, addr.Ip, addr.P2pPort, addr.RaftPort})
|
2017-08-28 14:43:44 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
func (addr *Address) DecodeRLP(s *rlp.Stream) error {
|
|
|
|
// These fields need to be public:
|
|
|
|
var temp struct {
|
|
|
|
RaftId uint16
|
2019-02-08 01:29:01 -08:00
|
|
|
NodeId enode.EnodeID
|
2017-08-28 14:43:44 -07:00
|
|
|
Ip net.IP
|
2019-01-29 03:08:50 -08:00
|
|
|
P2pPort enr.TCP
|
2019-02-24 23:01:53 -08:00
|
|
|
RaftPort enr.RaftPort
|
2017-08-28 14:43:44 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
if err := s.Decode(&temp); err != nil {
|
|
|
|
return err
|
|
|
|
} else {
|
2018-09-11 12:05:42 -07:00
|
|
|
addr.RaftId, addr.NodeId, addr.Ip, addr.P2pPort, addr.RaftPort = temp.RaftId, temp.NodeId, temp.Ip, temp.P2pPort, temp.RaftPort
|
2017-08-28 14:43:44 -07:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// RLP Address encoding, for transport over raft and storage in LevelDB.
|
|
|
|
|
|
|
|
func (addr *Address) toBytes() []byte {
|
|
|
|
size, r, err := rlp.EncodeToReader(addr)
|
|
|
|
if err != nil {
|
|
|
|
panic(fmt.Sprintf("error: failed to RLP-encode Address: %s", err.Error()))
|
|
|
|
}
|
|
|
|
var buffer = make([]byte, uint32(size))
|
|
|
|
r.Read(buffer)
|
|
|
|
|
|
|
|
return buffer
|
|
|
|
}
|
|
|
|
|
|
|
|
func bytesToAddress(bytes []byte) *Address {
|
|
|
|
var addr Address
|
|
|
|
if err := rlp.DecodeBytes(bytes, &addr); err != nil {
|
|
|
|
log.Fatalf("failed to RLP-decode Address: %v", err)
|
|
|
|
}
|
|
|
|
return &addr
|
|
|
|
}
|