自建DNS增加解析测试

2026-01-03 17:46:37 +08:00 · 2021-08-09 18:42:00 +08:00
parent 69c1d35406
commit 4d092f329b
13 changed files with 404 additions and 1 deletions
--- a/internal/web/actions/default/ns/test/index.go
+++ b/internal/web/actions/default/ns/test/index.go
@@ -0,0 +1,156 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+
+package test
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/dns/domains/domainutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/dnsconfigs"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/actions"
+	"github.com/iwind/TeaGo/maps"
+	"github.com/iwind/TeaGo/types"
+	"github.com/miekg/dns"
+	"net"
+	"regexp"
+)
+
+type IndexAction struct {
+	actionutils.ParentAction
+}
+
+func (this *IndexAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *IndexAction) RunGet(params struct{}) {
+	// 集群列表
+	clustersResp, err := this.RPC().NSClusterRPC().FindAllEnabledNSClusters(this.AdminContext(), &pb.FindAllEnabledNSClustersRequest{})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var clusterMaps = []maps.Map{}
+	for _, cluster := range clustersResp.NsClusters {
+		if !cluster.IsOn {
+			continue
+		}
+
+		countNodesResp, err := this.RPC().NSNodeRPC().CountAllEnabledNSNodesMatch(this.AdminContext(), &pb.CountAllEnabledNSNodesMatchRequest{
+			NsClusterId:  cluster.Id,
+			InstallState: 0,
+			ActiveState:  0,
+			Keyword:      "",
+		})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		var countNodes = countNodesResp.Count
+		if countNodes <= 0 {
+			continue
+		}
+
+		clusterMaps = append(clusterMaps, maps.Map{
+			"id":         cluster.Id,
+			"name":       cluster.Name,
+			"countNodes": countNodes,
+		})
+	}
+	this.Data["clusters"] = clusterMaps
+
+	// 记录类型
+	this.Data["recordTypes"] = dnsconfigs.FindAllRecordTypeDefinitions()
+
+	this.Show()
+}
+
+func (this *IndexAction) RunPost(params struct {
+	NodeId   int64
+	Domain   string
+	Type     string
+	Ip       string
+	ClientIP string
+
+	Must *actions.Must
+}) {
+	nodeResp, err := this.RPC().NSNodeRPC().FindEnabledNSNode(this.AdminContext(), &pb.FindEnabledNSNodeRequest{NsNodeId: params.NodeId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var node = nodeResp.NsNode
+	if node == nil {
+		this.Fail("找不到要测试的节点")
+	}
+
+	var isOk = false
+	var errMsg string
+	var isNetError = false
+	var result string
+
+	defer func() {
+		this.Data["isOk"] = isOk
+		this.Data["err"] = errMsg
+		this.Data["isNetErr"] = isNetError
+		this.Data["result"] = result
+		this.Success()
+	}()
+
+	if !domainutils.ValidateDomainFormat(params.Domain) {
+		errMsg = "域名格式错误"
+		return
+	}
+
+	recordType, ok := dns.StringToType[params.Type]
+	if !ok {
+		errMsg = "不支持此记录类型"
+		return
+	}
+
+	if len(params.ClientIP) > 0 && net.ParseIP(params.ClientIP) == nil {
+		errMsg = "客户端IP格式不正确"
+		return
+	}
+
+	var optionId int64
+	if len(params.ClientIP) > 0 {
+		optionResp, err := this.RPC().NSQuestionOptionRPC().CreateNSQuestionOption(this.AdminContext(), &pb.CreateNSQuestionOptionRequest{
+			Name:       "setRemoteAddr",
+			ValuesJSON: maps.Map{"ip": params.ClientIP}.AsJSON(),
+		})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		optionId = optionResp.NsQuestionOptionId
+		defer func() {
+			_, err = this.RPC().NSQuestionOptionRPC().DeleteNSQuestionOption(this.AdminContext(), &pb.DeleteNSQuestionOptionRequest{NsQuestionOptionId: optionId})
+			if err != nil {
+				this.ErrorPage(err)
+			}
+		}()
+	}
+
+	c := new(dns.Client)
+	m := new(dns.Msg)
+	var domain = params.Domain + "."
+	if optionId > 0 {
+		domain = "$" + types.String(optionId) + "-" + domain
+	}
+	m.SetQuestion(domain, recordType)
+	r, _, err := c.Exchange(m, params.Ip+":53")
+	if err != nil {
+		errMsg = "解析过程中出错：" + err.Error()
+
+		// 是否为网络错误
+		if regexp.MustCompile(`timeout|connect`).MatchString(err.Error()) {
+			isNetError = true
+		}
+
+		return
+	}
+	result = r.String()
+	result = regexp.MustCompile(`\$\d+-`).ReplaceAllString(result, "")
+	isOk = true
+}