vpn: implement link.RouteSubscribe for windows

This commit is contained in:
rkonfj 2024-07-07 15:34:14 +08:00
parent 90be7dccdd
commit 8d1a5433b2
No known key found for this signature in database
GPG Key ID: 9595E3A991503741
4 changed files with 32 additions and 1 deletions

1
go.mod
View File

@ -16,6 +16,7 @@ require (
golang.org/x/sys v0.19.0
golang.org/x/time v0.5.0
golang.zx2c4.com/wireguard v0.0.0-20231211153847-12269c276173
golang.zx2c4.com/wireguard/windows v0.5.3
gopkg.in/yaml.v2 v2.4.0
storj.io/common v0.0.0-20240425113201-9815a85cbc32
tailscale.com v1.56.1

2
go.sum
View File

@ -72,6 +72,8 @@ golang.zx2c4.com/wintun v0.0.0-20230126152724-0fa3db229ce2 h1:B82qJJgjvYKsXS9jeu
golang.zx2c4.com/wintun v0.0.0-20230126152724-0fa3db229ce2/go.mod h1:deeaetjYA+DHMHg+sMSMI58GrEteJUUzzw7en6TJQcI=
golang.zx2c4.com/wireguard v0.0.0-20231211153847-12269c276173 h1:/jFs0duh4rdb8uIfPMv78iAJGcPKDeqAFnaLBropIC4=
golang.zx2c4.com/wireguard v0.0.0-20231211153847-12269c276173/go.mod h1:tkCQ4FQXmpAgYVh++1cq16/dH4QJtmvpRv19DWGAHSA=
golang.zx2c4.com/wireguard/windows v0.5.3 h1:On6j2Rpn3OEMXqBq00QEDC7bWSZrPIHKIus8eIuExIE=
golang.zx2c4.com/wireguard/windows v0.5.3/go.mod h1:9TEe8TJmtwyQebdFwAkEWOPr3prrtqm+REGFifP60hI=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=

View File

@ -1,4 +1,4 @@
//go:build !linux
//go:build !linux && !windows
package link

28
vpn/link/route_windows.go Normal file
View File

@ -0,0 +1,28 @@
package link
import (
"context"
"net"
"golang.zx2c4.com/wireguard/windows/tunnel/winipcfg"
)
func RouteSubscribe(ctx context.Context, ch chan<- RouteUpdate) error {
cb, err := winipcfg.RegisterRouteChangeCallback(func(notificationType winipcfg.MibNotificationType, route *winipcfg.MibIPforwardRow2) {
dst := route.DestinationPrefix.Prefix()
ch <- RouteUpdate{
Type: uint16(notificationType),
Dst: &net.IPNet{IP: net.IP(dst.Addr().AsSlice()), Mask: net.CIDRMask(dst.Bits(), dst.Addr().BitLen())},
Via: net.IP(route.NextHop.Addr().AsSlice()),
}
})
if err != nil {
return err
}
go func() {
<-ctx.Done()
cb.Unregister()
close(ch)
}()
return nil
}