129 lines
2.9 KiB
Go
129 lines
2.9 KiB
Go
package models
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
type Graph struct {
|
|
Version int `json:"version"`
|
|
Batadv struct {
|
|
Links []*GraphLink `json:"links"`
|
|
} `json:"batadv"`
|
|
}
|
|
|
|
type GraphLink struct {
|
|
Source string `json:"source"`
|
|
Target string `json:"target"`
|
|
VPN bool `json:"vpn"`
|
|
TQ float32 `json:"tq"`
|
|
Bidirect bool `json:"bidirect"`
|
|
}
|
|
|
|
type GraphBuilder struct {
|
|
macToID map[string]string // mapping from MAC address to node id
|
|
links map[string]*GraphLink // mapping from $idA-$idB to existing link
|
|
vpn map[string]interface{} // IDs/addresses of VPN servers
|
|
}
|
|
|
|
func (nodes *Nodes) BuildGraph(vpnAddresses []string) *Graph {
|
|
builder := &GraphBuilder{
|
|
macToID: make(map[string]string),
|
|
links: make(map[string]*GraphLink),
|
|
vpn: make(map[string]interface{}),
|
|
}
|
|
|
|
// read VPN addresses into map
|
|
for _, address := range vpnAddresses {
|
|
builder.vpn[address] = nil
|
|
}
|
|
|
|
builder.readNodes(nodes.List)
|
|
|
|
graph := &Graph{Version: 2}
|
|
graph.Batadv.Links = builder.Links()
|
|
return graph
|
|
}
|
|
|
|
func (builder *GraphBuilder) readNodes(nodes map[string]*Node) {
|
|
// Fill mac->id map
|
|
for sourceId, node := range nodes {
|
|
if nodeinfo := node.Nodeinfo; nodeinfo != nil {
|
|
interfaces := nodeinfo.Network.Mesh.Bat0.Interfaces
|
|
addresses := append(append(interfaces.Other, interfaces.Tunnel...), interfaces.Wireless...)
|
|
|
|
for _, sourceAddress := range addresses {
|
|
builder.macToID[sourceAddress] = sourceId
|
|
|
|
// is VPN address?
|
|
if _, found := builder.vpn[sourceAddress]; found {
|
|
builder.vpn[sourceId] = nil
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Add links
|
|
for sourceId, node := range nodes {
|
|
if neighbours := node.Neighbours; neighbours != nil {
|
|
for _, batadvNeighbours := range neighbours.Batadv {
|
|
for targetAddress, link := range batadvNeighbours.Neighbours {
|
|
if targetId, found := builder.macToID[targetAddress]; found {
|
|
builder.addLink(targetId, sourceId, link.Tq)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (builder *GraphBuilder) Links() []*GraphLink {
|
|
i := 0
|
|
links := make([]*GraphLink, len(builder.links))
|
|
|
|
for _, link := range builder.links {
|
|
links[i] = link
|
|
i += 1
|
|
}
|
|
return links
|
|
}
|
|
|
|
func (builder *GraphBuilder) isVPN(ids ...string) bool {
|
|
for _, id := range ids {
|
|
if _, found := builder.vpn[id]; found {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (builder *GraphBuilder) addLink(targetId string, sourceId string, linkTq int) {
|
|
// Sort IDs to generate the key
|
|
var key string
|
|
if strings.Compare(sourceId, targetId) > 0 {
|
|
key = fmt.Sprintf("%s-%s", sourceId, targetId)
|
|
} else {
|
|
key = fmt.Sprintf("%s-%s", targetId, sourceId)
|
|
}
|
|
|
|
var tq float32
|
|
if linkTq > 0 {
|
|
tq = float32(1.0 / (float32(linkTq) / 255.0))
|
|
}
|
|
|
|
if link, ok := builder.links[key]; !ok {
|
|
builder.links[key] = &GraphLink{
|
|
Source: sourceId,
|
|
Target: targetId,
|
|
VPN: builder.isVPN(sourceId, targetId),
|
|
TQ: tq,
|
|
}
|
|
} else {
|
|
// Use lowest of both link qualities
|
|
if tq < link.TQ {
|
|
link.TQ = tq
|
|
}
|
|
link.Bidirect = true
|
|
}
|
|
}
|