mirror of
https://github.com/sigcn/pg.git
synced 2025-10-07 02:02:45 +08:00
55 lines
1.6 KiB
Go
55 lines
1.6 KiB
Go
//go:build windows
|
|
|
|
package netlink
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"os/exec"
|
|
|
|
"golang.zx2c4.com/wireguard/windows/tunnel/winipcfg"
|
|
)
|
|
|
|
func SetupLink(ifName, cidr string) error {
|
|
ip, ipnet, err := net.ParseCIDR(cidr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if ip.To4() == nil { // ipv6
|
|
info.IPv6 = ip.String()
|
|
return exec.Command("netsh", "interface", "ipv6", "add", "address", ifName, cidr).Run()
|
|
}
|
|
// ipv4
|
|
info.IPv4 = ip.String()
|
|
addrMask := fmt.Sprintf("%d.%d.%d.%d", ipnet.Mask[0], ipnet.Mask[1], ipnet.Mask[2], ipnet.Mask[3])
|
|
return exec.Command("netsh", "interface", "ipv4", "set", "address", ifName, "static", ip.String(), addrMask).Run()
|
|
}
|
|
|
|
func AddRoute(ifName string, to *net.IPNet, via net.IP) error {
|
|
if via.To4() == nil { // ipv6
|
|
return exec.Command("netsh", "interface", "ipv6", "add", "route", to.String(), ifName, via.String()).Run()
|
|
}
|
|
// ipv4
|
|
addrMask := fmt.Sprintf("%d.%d.%d.%d", to.Mask[0], to.Mask[1], to.Mask[2], to.Mask[3])
|
|
return exec.Command("route", "add", to.IP.String(), "mask", addrMask, via.String()).Run()
|
|
}
|
|
|
|
func DelRoute(ifName string, to *net.IPNet, via net.IP) error {
|
|
if via.To4() == nil { // ipv6
|
|
return exec.Command("netsh", "interface", "ipv6", "delete", "route", to.String(), ifName, via.String()).Run()
|
|
}
|
|
return exec.Command("route", "delete", to.IP.String()).Run()
|
|
}
|
|
|
|
func LinkByIndex(index int) (*Link, error) {
|
|
luid, err := winipcfg.LUIDFromIndex(uint32(index))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ifInfo, err := luid.Interface()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &Link{Name: ifInfo.Alias(), Type: uint32(ifInfo.Type), Index: index}, nil
|
|
}
|