Route watch working

network/routewatch_windows.go

@@ -0,0 +1,42 @@
+package network
+
+import (
+	"context"
+
+	"golang.zx2c4.com/wireguard/windows/tunnel/winipcfg"
+)
+
+// watchRouteEvents calls notify for every route, interface and unicast
+// address change notification.
+func watchRouteEvents(ctx context.Context, notify func()) error {
+	routeCb, err := winipcfg.RegisterRouteChangeCallback(func(winipcfg.MibNotificationType, *winipcfg.MibIPforwardRow2) {
+		notify()
+	})
+	if err != nil {
+		return err
+	}
+	ifaceCb, err := winipcfg.RegisterInterfaceChangeCallback(func(winipcfg.MibNotificationType, *winipcfg.MibIPInterfaceRow) {
+		notify()
+	})
+	if err != nil {
+		_ = routeCb.Unregister()
+		return err
+	}
+	addrCb, err := winipcfg.RegisterUnicastAddressChangeCallback(func(winipcfg.MibNotificationType, *winipcfg.MibUnicastIPAddressRow) {
+		notify()
+	})
+	if err != nil {
+		_ = routeCb.Unregister()
+		_ = ifaceCb.Unregister()
+		return err
+	}
+
+	go func() {
+		<-ctx.Done()
+		_ = routeCb.Unregister()
+		_ = ifaceCb.Unregister()
+		_ = addrCb.Unregister()
+	}()
+
+	return nil
+}