routes_windows.go 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. package local
  2. import (
  3. "fmt"
  4. "net"
  5. "regexp"
  6. "strings"
  7. "time"
  8. "github.com/gravitl/netmaker/netclient/ncutils"
  9. )
  10. // GetDefaultRoute - Gets the default route (ip and interface) on a windows machine
  11. func GetDefaultRoute() (string, string, error) {
  12. var ipaddr string
  13. var iface string
  14. var err error
  15. var outLine string
  16. output, err := ncutils.RunCmd("netsh interface ipv4 show route", false)
  17. if err != nil {
  18. return ipaddr, iface, err
  19. }
  20. for _, line := range strings.Split(strings.TrimSuffix(output, "\n"), "\n") {
  21. if strings.Contains(line, "0.0.0.0/0") {
  22. outLine = line
  23. break
  24. }
  25. }
  26. if outLine == "" {
  27. return ipaddr, iface, fmt.Errorf("could not find default gateway")
  28. }
  29. space := regexp.MustCompile(`\s+`)
  30. outputSlice := strings.Split(strings.TrimSpace(space.ReplaceAllString(outLine, " ")), " ")
  31. ipaddr = outputSlice[len(outputSlice)-1]
  32. if err = ncutils.CheckIPAddress(ipaddr); err != nil {
  33. return ipaddr, iface, fmt.Errorf("invalid output for ip address check: " + err.Error())
  34. }
  35. iface = "irrelevant"
  36. return ipaddr, iface, err
  37. }
  38. func setRoute(iface string, addr *net.IPNet, address string) error {
  39. var err error
  40. _, err = ncutils.RunCmd("route ADD "+addr.String()+" "+address, false)
  41. time.Sleep(time.Second >> 2)
  42. ncutils.RunCmd("route CHANGE "+addr.IP.String()+" MASK "+addr.Mask.String()+" "+address, false)
  43. return err
  44. }
  45. func deleteRoute(iface string, addr *net.IPNet, address string) error {
  46. var err error
  47. _, err = ncutils.RunCmd("route DELETE "+addr.IP.String()+" MASK "+addr.Mask.String()+" "+address, false)
  48. return err
  49. }
  50. func setCidr(iface, address string, addr *net.IPNet) {
  51. ncutils.RunCmd("route ADD "+addr.String()+" "+address, false)
  52. time.Sleep(time.Second >> 2)
  53. ncutils.RunCmd("route CHANGE "+addr.IP.String()+" MASK "+addr.Mask.String()+" "+address, false)
  54. }
  55. func removeCidr(iface string, addr *net.IPNet, address string) {
  56. ncutils.RunCmd("route DELETE "+addr.String(), false)
  57. }