root/gost_software
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tun: fix windows route

Browse Source
This commit is contained in:
ginuerzh 2020-01-01 11:44:33 +08:00
parent 21bee37da1
commit 2e428aaba9
2 changed files with 17 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tun.go
View File

@ -26,6 +26,7 @@ type tunHandler struct {
raddr string raddr string
options *HandlerOptions options *HandlerOptions
ipNet *net.IPNet ipNet *net.IPNet
routes sync.Map
} }
// TunHandler creates a handler for tun tunnel. // TunHandler creates a handler for tun tunnel.
@ -97,7 +98,6 @@ func (h *tunHandler) createTun() (conn net.Conn, err error) {
} }
func (h *tunHandler) transportTun(tun net.Conn, conn net.PacketConn, raddr net.Addr) error { func (h *tunHandler) transportTun(tun net.Conn, conn net.PacketConn, raddr net.Addr) error {
var routes sync.Map
errc := make(chan error, 1) errc := make(chan error, 1)
go func() { go func() {
@ -130,7 +130,7 @@ func (h *tunHandler) transportTun(tun net.Conn, conn net.PacketConn, raddr net.A
} }
addr := raddr addr := raddr
if v, ok := routes.Load(header.Dst.String()); ok { if v, ok := h.routes.Load(header.Dst.String()); ok {
addr = v.(net.Addr) addr = v.(net.Addr)
} }
if addr == nil { if addr == nil {
@ -195,11 +195,11 @@ func (h *tunHandler) transportTun(tun net.Conn, conn net.PacketConn, raddr net.A
} }
if h.ipNet != nil && h.ipNet.IP.Equal(header.Src.Mask(h.ipNet.Mask)) { if h.ipNet != nil && h.ipNet.IP.Equal(header.Src.Mask(h.ipNet.Mask)) {
if actual, loaded := routes.LoadOrStore(header.Src.String(), addr); loaded { if actual, loaded := h.routes.LoadOrStore(header.Src.String(), addr); loaded {
if actual.(net.Addr).String() != addr.String() { if actual.(net.Addr).String() != addr.String() {
log.Logf("[tun] %s <- %s: update route: %s -> %s (old %s)", log.Logf("[tun] %s <- %s: update route: %s -> %s (old %s)",
tun.LocalAddr(), addr, header.Src, addr, actual.(net.Addr)) tun.LocalAddr(), addr, header.Src, addr, actual.(net.Addr))
routes.Store(header.Src.String(), addr) h.routes.Store(header.Src.String(), addr)
} }
} else { } else {
log.Logf("[tun] %s: new route: %s -> %s", tun.LocalAddr(), header.Src, addr) log.Logf("[tun] %s: new route: %s -> %s", tun.LocalAddr(), header.Src, addr)

21
tun_windows.go
View File

@ -38,7 +38,7 @@ func createTun(cfg TunConfig) (conn net.Conn, ipNet *net.IPNet, err error) {
return return
} }
if err = addRoutes(ip.String(), cfg.Routes...); err != nil { if err = addRoutes(ifce.Name(), cfg.Routes...); err != nil {
return return
} }
@ -49,18 +49,16 @@ func createTun(cfg TunConfig) (conn net.Conn, ipNet *net.IPNet, err error) {
return return
} }
func addRoutes(ifIP string, routes ...string) error { func addRoutes(ifName string, routes ...string) error {
for _, route := range routes { for _, route := range routes {
if route == "" { if route == "" {
continue continue
} }
_, inet, err := net.ParseCIDR(route)
if err != nil {
return err
}
cmd := fmt.Sprintf("route ADD %s MASK %s %s", deleteRoute(ifName, route)
inet.IP, ipMask(inet.Mask), ifIP)
cmd := fmt.Sprintf("netsh interface ip add route prefix=%s interface=%s store=active",
route, ifName)
log.Log("[tun]", cmd) log.Log("[tun]", cmd)
args := strings.Split(cmd, " ") args := strings.Split(cmd, " ")
if er := exec.Command(args[0], args[1:]...).Run(); er != nil { if er := exec.Command(args[0], args[1:]...).Run(); er != nil {
@ -70,6 +68,13 @@ func addRoutes(ifIP string, routes ...string) error {
return nil return nil
} }
func deleteRoute(ifName string, route string) error {
cmd := fmt.Sprintf("netsh interface ip delete route prefix=%s interface=%s store=active",
route, ifName)
args := strings.Split(cmd, " ")
return exec.Command(args[0], args[1:]...).Run()
}
func ipMask(mask net.IPMask) string { func ipMask(mask net.IPMask) string {
return fmt.Sprintf("%d.%d.%d.%d", mask[0], mask[1], mask[2], mask[3]) return fmt.Sprintf("%d.%d.%d.%d", mask[0], mask[1], mask[2], mask[3])
} }