implement neighbor list

net/lux_router.go

@@ -295,6 +295,11 @@ func (r *LuxRouter) GetRouteKey(route *LuxRoute) (crypto.LuxKey, bool) {
 	return crypto.LuxKey{}, false
 }
 
+func (r *LuxRouter) HasKeyFor(id proto.LuxID) bool {
+	_, ok := r.keyStore.Get(id)
+	return ok
+}
+
 func (r *LuxRouter) Recv() (LuxPacket, error) {
 	dgram := r.RecvDgram()
 

node/lux_node.go

@@ -6,12 +6,16 @@ import (
 	"lux/net"
 	"lux/proto"
 	"sync"
+
+	ipnet "net"
 )
 
 type LuxNode struct {
 	router  net.LuxRouter
 	running bool
 
+	neighbors map[proto.LuxID]*ipnet.UDPAddr
+
 	state     LuxNodeState
 	stateLock sync.RWMutex
 }
@@ -20,6 +24,7 @@ func NewLuxNode(nodeKey crypto.LuxKey, ks crypto.LuxKeyStore) LuxNode {
 	return LuxNode{
 		router:    net.NewLuxRouter(nodeKey, ks),
 		running:   false,
+		neighbors: make(map[proto.LuxID]*ipnet.UDPAddr),
 		state:     NewNodeState(),
 	}
 }
@@ -40,6 +45,27 @@ func (node *LuxNode) AddExterior(udpAddr string) error {
 	return node.router.CreateInboundChannel(net.LuxChannelExterior, udpAddr)
 }
 
+func (node *LuxNode) AddNeighbor(id proto.LuxID, udpAddr string) error {
+	udpIp, err := ipnet.ResolveUDPAddr("udp", udpAddr)
+	if err != nil {
+		return err
+	}
+
+	if node.router.HasKeyFor(id) {
+		// we have key for this node, so we can route
+		err = node.router.CreateOutboundRoute(id, net.LuxChannelInterior, udpAddr)
+		if err != nil {
+			return err
+		}
+	}
+
+	// just add to neighbor list to let other nodes know
+	node.neighbors[id] = udpIp
+
+	log.Infof("added neighbor %s at %s", id.String(), udpIp.String())
+	return nil
+}
+
 func (node *LuxNode) GetState() *LuxNodeState {
 	return &node.state
 }