status-go/vendor/github.com/pion/ice/v2/mdns.go

66 lines
2.1 KiB
Go
Raw Normal View History

2024-01-03 21:57:33 +00:00
// SPDX-FileCopyrightText: 2023 The Pion community <https://pion.ly>
// SPDX-License-Identifier: MIT
2022-03-10 09:44:48 +00:00
package ice
import (
"github.com/google/uuid"
"github.com/pion/logging"
"github.com/pion/mdns"
2024-01-03 21:57:33 +00:00
"github.com/pion/transport/v2"
2022-03-10 09:44:48 +00:00
"golang.org/x/net/ipv4"
)
// MulticastDNSMode represents the different Multicast modes ICE can run in
type MulticastDNSMode byte
// MulticastDNSMode enum
const (
// MulticastDNSModeDisabled means remote mDNS candidates will be discarded, and local host candidates will use IPs
MulticastDNSModeDisabled MulticastDNSMode = iota + 1
// MulticastDNSModeQueryOnly means remote mDNS candidates will be accepted, and local host candidates will use IPs
MulticastDNSModeQueryOnly
// MulticastDNSModeQueryAndGather means remote mDNS candidates will be accepted, and local host candidates will use mDNS
MulticastDNSModeQueryAndGather
)
func generateMulticastDNSName() (string, error) {
// https://tools.ietf.org/id/draft-ietf-rtcweb-mdns-ice-candidates-02.html#gathering
// The unique name MUST consist of a version 4 UUID as defined in [RFC4122], followed by “.local”.
u, err := uuid.NewRandom()
return u.String() + ".local", err
}
2024-01-03 21:57:33 +00:00
func createMulticastDNS(n transport.Net, mDNSMode MulticastDNSMode, mDNSName string, log logging.LeveledLogger) (*mdns.Conn, MulticastDNSMode, error) {
2022-03-10 09:44:48 +00:00
if mDNSMode == MulticastDNSModeDisabled {
return nil, mDNSMode, nil
}
2024-01-03 21:57:33 +00:00
addr, mdnsErr := n.ResolveUDPAddr("udp4", mdns.DefaultAddress)
2022-03-10 09:44:48 +00:00
if mdnsErr != nil {
return nil, mDNSMode, mdnsErr
}
2024-01-03 21:57:33 +00:00
l, mdnsErr := n.ListenUDP("udp4", addr)
2022-03-10 09:44:48 +00:00
if mdnsErr != nil {
// If ICE fails to start MulticastDNS server just warn the user and continue
log.Errorf("Failed to enable mDNS, continuing in mDNS disabled mode: (%s)", mdnsErr)
return nil, MulticastDNSModeDisabled, nil
}
switch mDNSMode {
case MulticastDNSModeQueryOnly:
conn, err := mdns.Server(ipv4.NewPacketConn(l), &mdns.Config{})
return conn, mDNSMode, err
case MulticastDNSModeQueryAndGather:
conn, err := mdns.Server(ipv4.NewPacketConn(l), &mdns.Config{
LocalNames: []string{mDNSName},
})
return conn, mDNSMode, err
default:
return nil, mDNSMode, nil
}
}