116 lines
2.6 KiB
Go
116 lines
2.6 KiB
Go
// Package node maintains information about peers on the network and associated
|
|
// connection sessions.
|
|
package node
|
|
|
|
import (
|
|
"fmt"
|
|
"net/netip"
|
|
|
|
"github.com/Indra-Labs/indra"
|
|
"github.com/Indra-Labs/indra/pkg/ifc"
|
|
"github.com/Indra-Labs/indra/pkg/key/prv"
|
|
"github.com/Indra-Labs/indra/pkg/key/pub"
|
|
"github.com/Indra-Labs/indra/pkg/nonce"
|
|
log2 "github.com/cybriq/proc/pkg/log"
|
|
)
|
|
|
|
var (
|
|
log = log2.GetLogger(indra.PathBase)
|
|
check = log.E.Chk
|
|
)
|
|
|
|
// Node is a representation of a messaging counterparty. The net.IP can be nil
|
|
// for the case of a client node that is not in a direct open connection. For
|
|
// this reason all nodes are assigned an ID and will normally be handled by this
|
|
// except when the netip.AddrPort is known via the packet sender address.
|
|
type Node struct {
|
|
nonce.ID
|
|
Addr string
|
|
AddrPort *netip.AddrPort
|
|
HeaderPub, PayloadPub *pub.Key
|
|
HeaderPrv, PayloadPrv *prv.Key
|
|
ifc.Transport
|
|
}
|
|
|
|
// New creates a new Node. netip.AddrPort is optional if the counterparty is not
|
|
// in direct connection.
|
|
func New(addr *netip.AddrPort, hdr, pld *pub.Key, hdrPrv, pldPrv *prv.Key,
|
|
tpt ifc.Transport) (n *Node, id nonce.ID) {
|
|
|
|
id = nonce.NewID()
|
|
n = &Node{
|
|
ID: id,
|
|
Addr: addr.String(),
|
|
AddrPort: addr,
|
|
Transport: tpt,
|
|
HeaderPub: hdr,
|
|
PayloadPub: pld,
|
|
HeaderPrv: hdrPrv,
|
|
PayloadPrv: pldPrv,
|
|
}
|
|
return
|
|
}
|
|
|
|
type Nodes []*Node
|
|
|
|
// NewNodes creates an empty Nodes
|
|
func NewNodes() (n Nodes) { return Nodes{} }
|
|
|
|
func (n Nodes) Len() int {
|
|
return len(n)
|
|
}
|
|
|
|
// Add a Node to a Nodes.
|
|
func (n Nodes) Add(nn *Node) Nodes {
|
|
return append(n, nn)
|
|
}
|
|
|
|
// FindByID searches for a Node by ID.
|
|
func (n Nodes) FindByID(i nonce.ID) (no *Node) {
|
|
for _, nn := range n {
|
|
if nn.ID == i {
|
|
no = nn
|
|
break
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
// FindByAddrPort searches for a Node by netip.AddrPort.
|
|
func (n Nodes) FindByAddrPort(id *netip.AddrPort) (no *Node) {
|
|
for _, nn := range n {
|
|
if nn.AddrPort.String() == id.String() {
|
|
no = nn
|
|
break
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
// DeleteByID deletes a node identified by an ID.
|
|
func (n Nodes) DeleteByID(ii nonce.ID) (nn Nodes, e error) {
|
|
e = fmt.Errorf("id %x not found", ii)
|
|
for i := range n {
|
|
if n[i].ID == ii {
|
|
n = append(n[:i], n[i+1:]...)
|
|
e = nil
|
|
break
|
|
}
|
|
}
|
|
return n, e
|
|
}
|
|
|
|
// DeleteByAddrPort deletes a node identified by a netip.AddrPort.
|
|
func (n Nodes) DeleteByAddrPort(ip *netip.AddrPort) (nn Nodes, e error) {
|
|
e = fmt.Errorf("node with ip %v not found", ip)
|
|
nn = n
|
|
for i := range n {
|
|
if n[i].AddrPort.String() == ip.String() {
|
|
nn = append(n[:i], n[i+1:]...)
|
|
e = nil
|
|
break
|
|
}
|
|
}
|
|
return
|
|
}
|