tendrils/link.go

package tendrils

import (
	"encoding/json"
	"fmt"
)

type Link struct {
	ID         string `json:"id"`
	NodeA      *Node  `json:"node_a"`
	InterfaceA string `json:"interface_a,omitempty"`
	NodeB      *Node  `json:"node_b"`
	InterfaceB string `json:"interface_b,omitempty"`
}

func (l *Link) MarshalJSON() ([]byte, error) {
	type linkJSON struct {
		ID         string `json:"id"`
		NodeAID    string `json:"node_a_id"`
		InterfaceA string `json:"interface_a,omitempty"`
		NodeBID    string `json:"node_b_id"`
		InterfaceB string `json:"interface_b,omitempty"`
	}

	return json.Marshal(linkJSON{
		ID:         l.ID,
		NodeAID:    l.NodeA.ID,
		InterfaceA: l.InterfaceA,
		NodeBID:    l.NodeB.ID,
		InterfaceB: l.InterfaceB,
	})
}

func (l *Link) String() string {
	nameA := l.NodeA.DisplayName()
	if nameA == "" {
		nameA = l.NodeA.FirstMAC()
	}
	nameB := l.NodeB.DisplayName()
	if nameB == "" {
		nameB = l.NodeB.FirstMAC()
	}
	sideA := nameA
	if l.InterfaceA != "" {
		sideA = nameA + ":" + l.InterfaceA
	}
	sideB := nameB
	if l.InterfaceB != "" {
		sideB = nameB + ":" + l.InterfaceB
	}
	return fmt.Sprintf("%s <-> %s", sideA, sideB)
}

func (n *Nodes) getDirectLinks() []*Link {
	macToNode := map[string]*Node{}
	for _, node := range n.nodes {
		for _, iface := range node.Interfaces {
			if iface.MAC != "" {
				macToNode[string(iface.MAC)] = node
			}
		}
	}

	seen := map[string]bool{}
	var links []*Link

	for _, target := range n.nodes {
		seenMACs := map[string]bool{}
		for _, iface := range target.Interfaces {
			if iface.MAC == "" {
				continue
			}
			mac := string(iface.MAC)
			if seenMACs[mac] {
				continue
			}
			seenMACs[mac] = true

			lastHops := n.findAllLastHops(target, mac, macToNode)
			for _, lh := range lastHops {
				targetIface := n.findTargetInterface(target, lh.node, macToNode)
				key := makeLinkKey(lh.node, lh.port, target, targetIface)
				if !seen[key] {
					seen[key] = true
					if target.Type == "" {
						if lh.node.Type == NodeTypeSwitch {
							target.Type = NodeTypeWiredClient
						} else if lh.node.Type == NodeTypeAP {
							target.Type = NodeTypeWirelessClient
						}
					}
					links = append(links, &Link{
						ID:         newID("link"),
						NodeA:      lh.node,
						InterfaceA: lh.port,
						NodeB:      target,
						InterfaceB: targetIface,
					})
				}
			}
		}
	}

	return links
}

func (n *Nodes) findAllLastHops(target *Node, mac string, macToNode map[string]*Node) []struct {
	node *Node
	port string
} {
	var results []struct {
		node *Node
		port string
	}

	for _, node := range n.nodes {
		port, sees := node.MACTable[mac]
		if !sees || node == target {
			continue
		}

		if !n.hasCloserNode(node, target, mac, port, macToNode) {
			results = append(results, struct {
				node *Node
				port string
			}{node, port})
		}
	}
	return results
}

func (n *Nodes) hasCloserNode(node, target *Node, mac, port string, macToNode map[string]*Node) bool {
	nodeMACs := map[string]bool{}
	for _, iface := range node.Interfaces {
		if iface.MAC != "" {
			nodeMACs[string(iface.MAC)] = true
		}
	}

	for otherMAC, otherPort := range node.MACTable {
		if otherPort != port {
			continue
		}
		otherNode := macToNode[otherMAC]
		if otherNode == nil || otherNode == node || otherNode == target {
			continue
		}

		targetPort, alsoSees := otherNode.MACTable[mac]
		if !alsoSees {
			continue
		}

		seesNodeOnSamePort := false
		for nodeMAC := range nodeMACs {
			if nodePort, seesNode := otherNode.MACTable[nodeMAC]; seesNode && nodePort == targetPort {
				seesNodeOnSamePort = true
				break
			}
		}

		if !seesNodeOnSamePort {
			return true
		}
	}
	return false
}

func (n *Nodes) findTargetInterface(target, lastHop *Node, macToNode map[string]*Node) string {
	for lastHopMAC, targetPort := range target.MACTable {
		if macToNode[lastHopMAC] == lastHop {
			return targetPort
		}
	}
	return ""
}

func makeLinkKey(nodeA *Node, ifaceA string, nodeB *Node, ifaceB string) string {
	ptrA := fmt.Sprintf("%p", nodeA)
	ptrB := fmt.Sprintf("%p", nodeB)
	if ptrA < ptrB {
		return ptrA + ":" + ifaceA + "-" + ptrB + ":" + ifaceB
	}
	return ptrB + ":" + ifaceB + "-" + ptrA + ":" + ifaceA
}