109 lines
2.5 KiB
Go
109 lines
2.5 KiB
Go
package tendrils
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/gopatchy/sacn"
|
|
)
|
|
|
|
func (t *Tendrils) startSACNDiscoveryListener(ctx context.Context, iface net.Interface) {
|
|
receiver, err := sacn.NewDiscoveryReceiver(&iface)
|
|
if err != nil {
|
|
log.Printf("[ERROR] failed to create sacn discovery receiver on %s: %v", iface.Name, err)
|
|
return
|
|
}
|
|
defer receiver.Stop()
|
|
|
|
receiver.SetHandler(func(src *net.UDPAddr, pkt interface{}) {
|
|
if disc, ok := pkt.(*sacn.DiscoveryPacket); ok {
|
|
t.handleSACNDiscoveryPacket(src.IP, disc)
|
|
}
|
|
})
|
|
|
|
receiver.Start()
|
|
<-ctx.Done()
|
|
}
|
|
|
|
func (t *Tendrils) handleSACNDiscoveryPacket(srcIP net.IP, pkt *sacn.DiscoveryPacket) {
|
|
if t.DebugSACN {
|
|
log.Printf("[sacn] discovery from %q cid=%s ip=%s universes=%v", pkt.SourceName, sacn.FormatCID(pkt.CID), srcIP, pkt.Universes)
|
|
}
|
|
|
|
if srcIP != nil && pkt.SourceName != "" {
|
|
t.nodes.Update(nil, nil, []net.IP{srcIP}, "", pkt.SourceName, "sacn")
|
|
}
|
|
|
|
node := t.nodes.GetByIP(srcIP)
|
|
if node != nil {
|
|
intUniverses := make([]int, len(pkt.Universes))
|
|
for i, u := range pkt.Universes {
|
|
intUniverses[i] = int(u)
|
|
}
|
|
t.nodes.UpdateSACN(node, intUniverses)
|
|
}
|
|
t.NotifyUpdate()
|
|
}
|
|
|
|
func (n *Nodes) UpdateSACN(node *Node, outputs []int) {
|
|
n.mu.Lock()
|
|
defer n.mu.Unlock()
|
|
|
|
if node.SACNOutputs == nil {
|
|
node.SACNOutputs = SACNUniverseSet{}
|
|
}
|
|
|
|
for _, u := range outputs {
|
|
node.SACNOutputs.Add(SACNUniverse(u))
|
|
}
|
|
}
|
|
|
|
func (n *Nodes) UpdateSACNUnicastInputs(node *Node, inputs []int) {
|
|
n.mu.Lock()
|
|
defer n.mu.Unlock()
|
|
|
|
if node.SACNUnicastInputs == nil {
|
|
node.SACNUnicastInputs = SACNUniverseSet{}
|
|
}
|
|
|
|
for _, u := range inputs {
|
|
node.SACNUnicastInputs.Add(SACNUniverse(u))
|
|
}
|
|
}
|
|
|
|
func (n *Nodes) expireSACN() {
|
|
for _, node := range n.nodes {
|
|
if node.SACNUnicastInputs != nil {
|
|
node.SACNUnicastInputs.Expire(60 * time.Second)
|
|
}
|
|
if node.SACNOutputs != nil {
|
|
node.SACNOutputs.Expire(60 * time.Second)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (n *Nodes) mergeSACN(keep, merge *Node) {
|
|
if merge.SACNUnicastInputs != nil {
|
|
if keep.SACNUnicastInputs == nil {
|
|
keep.SACNUnicastInputs = SACNUniverseSet{}
|
|
}
|
|
for u, lastSeen := range merge.SACNUnicastInputs {
|
|
if existing, ok := keep.SACNUnicastInputs[u]; !ok || lastSeen.After(existing) {
|
|
keep.SACNUnicastInputs[u] = lastSeen
|
|
}
|
|
}
|
|
}
|
|
if merge.SACNOutputs != nil {
|
|
if keep.SACNOutputs == nil {
|
|
keep.SACNOutputs = SACNUniverseSet{}
|
|
}
|
|
for u, lastSeen := range merge.SACNOutputs {
|
|
if existing, ok := keep.SACNOutputs[u]; !ok || lastSeen.After(existing) {
|
|
keep.SACNOutputs[u] = lastSeen
|
|
}
|
|
}
|
|
}
|
|
}
|