mirror of
https://github.com/tailscale/tailscale.git
synced 2025-04-05 07:45:52 +00:00
c6b92ddda8
goupnp is an existing upnp client for go, which provides all the functionality we need, licensed under BSD-2-Clause, so we can copy it over and modify parts of it for our case. Specifically, we add contexts to all the methods so we can better handle timeouts, remove the dependency on large charsets, and (eventually) trim out extra components we don't need. Signed-off-by: julianknodt <julianknodt@gmail.com>
76 lines
1.8 KiB
Go
76 lines
1.8 KiB
Go
package goupnp
|
|
|
|
import (
|
|
"io"
|
|
"net"
|
|
|
|
"tailscale.com/net/upnp/httpu"
|
|
)
|
|
|
|
// httpuClient creates a HTTPU client that multiplexes to all multicast-capable
|
|
// IPv4 addresses on the host. Returns a function to clean up once the client is
|
|
// no longer required.
|
|
func httpuClient() (httpu.ClientInterface, func(), error) {
|
|
addrs, err := localIPv4MCastAddrs()
|
|
if err != nil {
|
|
return nil, nil, ctxError(err, "requesting host IPv4 addresses")
|
|
}
|
|
|
|
closers := make([]io.Closer, 0, len(addrs))
|
|
delegates := make([]httpu.ClientInterface, 0, len(addrs))
|
|
for _, addr := range addrs {
|
|
c, err := httpu.NewHTTPUClientAddr(addr)
|
|
if err != nil {
|
|
return nil, nil, ctxErrorf(err,
|
|
"creating HTTPU client for address %s", addr)
|
|
}
|
|
closers = append(closers, c)
|
|
delegates = append(delegates, c)
|
|
}
|
|
|
|
closer := func() {
|
|
for _, c := range closers {
|
|
c.Close()
|
|
}
|
|
}
|
|
|
|
return httpu.NewMultiClient(delegates), closer, nil
|
|
}
|
|
|
|
// localIPv4MCastAddrs returns the set of IPv4 addresses on multicast-able
|
|
// network interfaces.
|
|
func localIPv4MCastAddrs() ([]string, error) {
|
|
ifaces, err := net.Interfaces()
|
|
if err != nil {
|
|
return nil, ctxError(err, "requesting host interfaces")
|
|
}
|
|
|
|
// Find the set of addresses to listen on.
|
|
var addrs []string
|
|
for _, iface := range ifaces {
|
|
if iface.Flags&net.FlagMulticast == 0 || iface.Flags&net.FlagLoopback != 0 || iface.Flags&net.FlagUp == 0 {
|
|
// Does not support multicast or is a loopback address.
|
|
continue
|
|
}
|
|
ifaceAddrs, err := iface.Addrs()
|
|
if err != nil {
|
|
return nil, ctxErrorf(err,
|
|
"finding addresses on interface %s", iface.Name)
|
|
}
|
|
for _, netAddr := range ifaceAddrs {
|
|
addr, ok := netAddr.(*net.IPNet)
|
|
if !ok {
|
|
// Not an IPNet address.
|
|
continue
|
|
}
|
|
if addr.IP.To4() == nil {
|
|
// Not IPv4.
|
|
continue
|
|
}
|
|
addrs = append(addrs, addr.IP.String())
|
|
}
|
|
}
|
|
|
|
return addrs, nil
|
|
}
|