wesher/common/routes.go

33 lines
781 B
Go
Raw Normal View History

package common
import (
"net"
"github.com/vishvananda/netlink"
)
// Routes pushes list of local routes to a channel, after filtering using the provided network
// The full list is pushed after every routing change
func Routes(filter *net.IPNet) <-chan []net.IPNet {
routesc := make(chan []net.IPNet)
updatec := make(chan netlink.RouteUpdate)
netlink.RouteSubscribe(updatec, make(chan struct{}))
go func() {
for {
<-updatec
routes, err := netlink.RouteList(nil, netlink.FAMILY_ALL)
if err != nil {
continue
}
result := make([]net.IPNet, 0)
for _, route := range routes {
2020-05-21 18:21:43 +02:00
if route.Dst != nil && route.Gw == nil && filter.Contains(route.Dst.IP) {
result = append(result, *route.Dst)
}
}
routesc <- result
}
}()
return routesc
}