2019-06-09 07:24:20 +00:00
|
|
|
package ssdp
|
|
|
|
|
2020-01-10 18:59:01 +00:00
|
|
|
import (
|
|
|
|
"net"
|
|
|
|
"sync"
|
|
|
|
)
|
2019-06-09 07:24:20 +00:00
|
|
|
|
|
|
|
// Interfaces specify target interfaces to multicast. If no interfaces are
|
|
|
|
// specified, all interfaces will be used.
|
|
|
|
var Interfaces []net.Interface
|
|
|
|
|
2020-01-10 18:59:01 +00:00
|
|
|
var ifLock sync.Mutex
|
|
|
|
var ifList []net.Interface
|
|
|
|
|
|
|
|
// interfaces gets list of net.Interface to multicast UDP packet.
|
|
|
|
func interfaces() ([]net.Interface, error) {
|
|
|
|
ifLock.Lock()
|
|
|
|
defer ifLock.Unlock()
|
|
|
|
if len(Interfaces) > 0 {
|
|
|
|
return Interfaces, nil
|
|
|
|
}
|
|
|
|
if len(ifList) > 0 {
|
|
|
|
return ifList, nil
|
|
|
|
}
|
|
|
|
l, err := interfacesIPv4()
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
2019-06-09 07:24:20 +00:00
|
|
|
}
|
2020-01-10 18:59:01 +00:00
|
|
|
ifList = l
|
|
|
|
return ifList, nil
|
2019-06-09 07:24:20 +00:00
|
|
|
}
|
|
|
|
|
2020-01-10 18:59:01 +00:00
|
|
|
// interfacesIPv4 lists net.Interface on IPv4.
|
|
|
|
func interfacesIPv4() ([]net.Interface, error) {
|
2019-06-09 07:24:20 +00:00
|
|
|
iflist, err := net.Interfaces()
|
|
|
|
if err != nil {
|
2020-01-10 18:59:01 +00:00
|
|
|
return nil, err
|
2019-06-09 07:24:20 +00:00
|
|
|
}
|
|
|
|
list := make([]net.Interface, 0, len(iflist))
|
|
|
|
for _, ifi := range iflist {
|
|
|
|
if !hasLinkUp(&ifi) || !hasIPv4Address(&ifi) {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
list = append(list, ifi)
|
|
|
|
}
|
2020-01-10 18:59:01 +00:00
|
|
|
return list, nil
|
2019-06-09 07:24:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// hasLinkUp checks an I/F have link-up or not.
|
|
|
|
func hasLinkUp(ifi *net.Interface) bool {
|
|
|
|
return ifi.Flags&net.FlagUp != 0
|
|
|
|
}
|
|
|
|
|
|
|
|
// hasIPv4Address checks an I/F have IPv4 address.
|
|
|
|
func hasIPv4Address(ifi *net.Interface) bool {
|
|
|
|
addrs, err := ifi.Addrs()
|
|
|
|
if err != nil {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
for _, a := range addrs {
|
|
|
|
ip, _, err := net.ParseCIDR(a.String())
|
|
|
|
if err != nil {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
if len(ip.To4()) == net.IPv4len && !ip.IsUnspecified() {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
}
|