You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
3.6 KiB
114 lines
3.6 KiB
package host
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"net"
|
|
"runtime"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/log"
|
|
"github.com/harmony-one/harmony/p2p"
|
|
)
|
|
|
|
// SendMessage is to connect a socket given a port and send the given message.
|
|
// TODO(minhdoan, rj): need to check if a peer is reachable or not.
|
|
func SendMessage(host p2p.Host, p p2p.Peer, message []byte, lostPeer chan p2p.Peer) {
|
|
// Construct normal p2p message
|
|
content := ConstructP2pMessage(byte(0), message)
|
|
go send(host, p, content, lostPeer)
|
|
}
|
|
|
|
// BroadcastMessage sends the message to a list of peers
|
|
func BroadcastMessage(h p2p.Host, peers []p2p.Peer, msg []byte, lostPeer chan p2p.Peer) {
|
|
if len(peers) == 0 {
|
|
return
|
|
}
|
|
// Construct broadcast p2p message
|
|
content := ConstructP2pMessage(byte(17), msg)
|
|
length := len(content)
|
|
|
|
log.Info("Start Broadcasting", "gomaxprocs", runtime.GOMAXPROCS(0), "Size", length)
|
|
start := time.Now()
|
|
for _, peer := range peers {
|
|
peerCopy := peer
|
|
go send(h, peerCopy, content, lostPeer)
|
|
}
|
|
log.Info("Broadcasting Done", "time spent(s)", time.Since(start).Seconds())
|
|
|
|
// Keep track of block propagation time
|
|
// Assume 1M message is for block propagation
|
|
if length > 1000000 {
|
|
log.Debug("NET: START BLOCK PROPAGATION", "Size", length)
|
|
}
|
|
}
|
|
|
|
// BroadcastMessageFromLeader sends the message to a list of peers from a leader.
|
|
func BroadcastMessageFromLeader(h p2p.Host, peers []p2p.Peer, msg []byte, lostPeer chan p2p.Peer) {
|
|
// TODO(minhdoan): Enable back for multicast.
|
|
peers = SelectMyPeers(peers, 1, MaxBroadCast)
|
|
BroadcastMessage(h, peers, msg, lostPeer)
|
|
}
|
|
|
|
// ConstructP2pMessage constructs the p2p message as [messageType, contentSize, content]
|
|
func ConstructP2pMessage(msgType byte, content []byte) []byte {
|
|
message := make([]byte, 5+len(content))
|
|
message[0] = 17 // messageType 0x11
|
|
binary.BigEndian.PutUint32(message[1:5], uint32(len(content)))
|
|
copy(message[5:], content)
|
|
return message
|
|
}
|
|
|
|
// BroadcastMessageFromValidator sends the message to a list of peers from a validator.
|
|
func BroadcastMessageFromValidator(h p2p.Host, selfPeer p2p.Peer, peers []p2p.Peer, msg []byte) {
|
|
peers = SelectMyPeers(peers, selfPeer.ValidatorID*MaxBroadCast+1, (selfPeer.ValidatorID+1)*MaxBroadCast)
|
|
BroadcastMessage(h, peers, msg, nil)
|
|
}
|
|
|
|
// MaxBroadCast is the maximum number of neighbors to broadcast
|
|
const MaxBroadCast = 20
|
|
|
|
// SelectMyPeers chooses a list of peers based on the range of ValidatorID
|
|
// This is a quick hack of the current p2p networking model
|
|
func SelectMyPeers(peers []p2p.Peer, min int, max int) []p2p.Peer {
|
|
res := []p2p.Peer{}
|
|
for _, peer := range peers {
|
|
if peer.ValidatorID >= min && peer.ValidatorID <= max {
|
|
res = append(res, peer)
|
|
}
|
|
}
|
|
return res
|
|
}
|
|
|
|
// Send a message to another node with given port.
|
|
func send(h p2p.Host, peer p2p.Peer, message []byte, lostPeer chan p2p.Peer) {
|
|
// Add attack code here.
|
|
//attack.GetInstance().Run()
|
|
backoff := p2p.NewExpBackoff(250*time.Millisecond, 5*time.Second, 2)
|
|
|
|
for trial := 0; trial < 3; trial++ {
|
|
err := h.SendMessage(peer, message)
|
|
// No need to retry if new stream error or no error
|
|
if err == nil || err == p2p.ErrNewStream {
|
|
if trial > 0 {
|
|
log.Warn("retry send", "rety", trial)
|
|
}
|
|
return
|
|
}
|
|
log.Info("sleeping before trying to send again",
|
|
"duration", backoff.Cur, "addr", net.JoinHostPort(peer.IP, peer.Port))
|
|
backoff.Sleep()
|
|
}
|
|
log.Error("gave up sending a message", "addr", net.JoinHostPort(peer.IP, peer.Port))
|
|
|
|
if lostPeer != nil {
|
|
// Notify lostPeer channel
|
|
lostPeer <- peer
|
|
}
|
|
}
|
|
|
|
// DialWithSocketClient joins host port and establishes connection
|
|
func DialWithSocketClient(ip, port string) (conn net.Conn, err error) {
|
|
addr := net.JoinHostPort(ip, port)
|
|
conn, err = net.Dial("tcp", addr)
|
|
return
|
|
}
|
|
|