1 files changed, 23 insertions, 0 deletions
diff --git a/att script/4(v6 DDoS)/code/辅助权威服务器/plugin/atk/atkutil.go b/att script/4(v6 DDoS)/code/辅助权威服务器/plugin/atk/atkutil.go
new file mode 100644
index 0000000..b397fb2
--- /dev/null
+++ b/att script/4(v6 DDoS)/code/辅助权威服务器/plugin/atk/atkutil.go
@@ -0,0 +1,23 @@
+package atk
+
+import (
+	"strings"
+)
+
+func (a Atk) validRequest(qname string) int {
+	//判断是否为第一阶段目标域名（放大）
+	if strings.Contains(qname, a.zoneip4) {
+		if len(strings.Split(qname, ".")) == 5 {
+			//需要放大
+			return 0
+		}
+		// 请求被修改，返回权威信息
+		return -1
+	}
+	if strings.Contains(qname, a.zoneip6) {
+		//需要放大
+		return 1
+	}
+	// 均不满足，返回权威信息
+	return 2
+}