apps/kubevpn
1
0
Fork 0
mirror of https://github.com/kubenetworks/kubevpn.git synced 2025-12-24 11:51:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix windows dns server primary issue

Browse Source
This commit is contained in:
naison
2021-09-12 19:16:28 +08:00
parent 7483bb287e
commit 999f6c02ca
2 changed files with 52 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
dns/dns_windows.go
View File

@@ -1,3 +1,4 @@
//go:build windows
// +build windows
package dns
@@ -11,8 +12,9 @@ import (
func DNS(ip string, namespace string) error {
tunName := os.Getenv("tunName")
fmt.Println("tun name: " + tunName)
args := []string{
log.Info("tun name: " + tunName)
_ = cleanDnsServer(tunName)
cmd := exec.Command("netsh", []string{
"interface",
"ipv4",
"add",
@@ -20,29 +22,58 @@ func DNS(ip string, namespace string) error {
fmt.Sprintf("name=\"%s\"", tunName),
fmt.Sprintf("address=%s", ip),
"index=1",
}
output, err := exec.Command("netsh", args...).CombinedOutput()
fmt.Println(exec.Command("netsh", args...).Args)
log.Info(string(output))
_ = addNicSuffix(namespace)
}...)
output, err := cmd.CombinedOutput()
if err != nil {
log.Error(err)
return nil
log.Warnf("error while set dns server, error: %v, output: %s, command: %v", err, string(output), cmd.Args)
}
_ = addNicSuffixSearchList(namespace)
_ = updateNicMetric(tunName)
return nil
}
// @see https://docs.microsoft.com/en-us/powershell/module/dnsclient/set-dnsclientglobalsetting?view=windowsserver2019-ps#example-1--set-the-dns-suffix-search-list
func addNicSuffix(namespace string) error {
func addNicSuffixSearchList(namespace string) error {
cmd := exec.Command("PowerShell", []string{
"Set-DnsClientGlobalSetting",
"-SuffixSearchList",
fmt.Sprintf("@(\"%s.svc.cluster.local\", \"svc.cluster.local\")", namespace),
}...)
output, err := cmd.CombinedOutput()
log.Info(cmd.Args)
if err != nil {
log.Warn(err)
log.Warnf("error while set dns suffix search list, err: %v, output: %s, command: %v", err, string(output), cmd.Args)
}
return err
}
func updateNicMetric(name string) error {
cmd := exec.Command("PowerShell", []string{
"Set-NetIPInterface",
"-InterfaceAlias",
fmt.Sprintf("\"%s\"", name),
"-InterfaceMetric",
"1",
}...)
out, err := cmd.CombinedOutput()
if err != nil {
log.Warnf("error while update nic metrics, error: %v, output: %s, command: %v", err, string(out), cmd.Args)
}
return err
}
func cleanDnsServer(name string) error {
cmd := exec.Command("netsh", []string{
"interface",
"ipv4",
"delete",
"dnsservers",
fmt.Sprintf("\"%s\"", name),
"all",
}...)
out, err := cmd.CombinedOutput()
if err != nil {
log.Warnf("clean dnsservers failed, error: %v, output: %s, command: %v", err, string(out), cmd.Args)
}
log.Info(string(output))
return err
}

9
pkg/main_test.go
View File

@@ -5,12 +5,14 @@ import (
"crypto/md5"
"fmt"
log "github.com/sirupsen/logrus"
"io"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
net2 "k8s.io/apimachinery/pkg/util/net"
"k8s.io/client-go/kubernetes"
"k8s.io/client-go/tools/clientcmd"
"kubevpn/remote"
"net"
"net/http"
"os/exec"
"testing"
"time"
@@ -100,3 +102,10 @@ func checkSum(msg []byte) uint16 {
sum += sum >> 16
return uint16(^sum)
}
func TestHttpServer(t *testing.T) {
http.HandleFunc("/", func(writer http.ResponseWriter, request *http.Request) {
_, _ = io.WriteString(writer, "hello")
})
_ = http.ListenAndServe(":9080", nil)
}