Restart mDNS every time the connection information changes (#155)

This commit is contained in:
Andrew 2025-02-13 12:10:47 -05:00 committed by GitHub
parent 2a99c2db9d
commit 951173ba19
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 18 additions and 3 deletions

View File

@ -13,6 +13,8 @@ import (
"github.com/vishvananda/netlink/nl" "github.com/vishvananda/netlink/nl"
) )
var mDNSConn *mdns.Conn
var networkState struct { var networkState struct {
Up bool Up bool
IPv4 string IPv4 string
@ -91,13 +93,26 @@ func checkNetworkState() {
} }
if newState != networkState { if newState != networkState {
networkState = newState
fmt.Println("network state changed") fmt.Println("network state changed")
//restart MDNS
startMDNS()
networkState = newState
requestDisplayUpdate() requestDisplayUpdate()
} }
} }
func startMDNS() error { func startMDNS() error {
//If server was previously running, stop it
if mDNSConn != nil {
fmt.Printf("Stopping mDNS server\n")
err := mDNSConn.Close()
if err != nil {
fmt.Printf("failed to stop mDNS server: %v\n", err)
}
}
//Start a new server
fmt.Printf("Starting mDNS server on jetkvm.local\n")
addr4, err := net.ResolveUDPAddr("udp4", mdns.DefaultAddressIPv4) addr4, err := net.ResolveUDPAddr("udp4", mdns.DefaultAddressIPv4)
if err != nil { if err != nil {
return err return err
@ -118,10 +133,11 @@ func startMDNS() error {
return err return err
} }
_, err = mdns.Server(ipv4.NewPacketConn(l4), ipv6.NewPacketConn(l6), &mdns.Config{ mDNSConn, err = mdns.Server(ipv4.NewPacketConn(l4), ipv6.NewPacketConn(l6), &mdns.Config{
LocalNames: []string{"jetkvm.local"}, //TODO: make it configurable LocalNames: []string{"jetkvm.local"}, //TODO: make it configurable
}) })
if err != nil { if err != nil {
mDNSConn = nil
return err return err
} }
//defer server.Close() //defer server.Close()
@ -157,7 +173,6 @@ func init() {
} }
} }
}() }()
fmt.Println("Starting mDNS server")
err := startMDNS() err := startMDNS()
if err != nil { if err != nil {
fmt.Println("failed to run mDNS: %v", err) fmt.Println("failed to run mDNS: %v", err)