浏览代码

add check node private for namserver

abhishek9686 1 天之前
父节点
当前提交
eb0c1db6ca
共有 1 个文件被更改,包括 8 次插入2 次删除
  1. 8 2
      migrate/migrate.go

+ 8 - 2
migrate/migrate.go

@@ -5,6 +5,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"log"
+	"net"
 	"os"
 	"time"
 
@@ -78,8 +79,13 @@ func migrateNameservers() {
 				Status:    true,
 				CreatedBy: user.UserName,
 			}
-			for _, ip := range netI.NameServers {
-				ns.Servers = append(ns.Servers, ip)
+			for _, nsIP := range netI.NameServers {
+				_, cidr, err := net.ParseCIDR(netI.AddressRange)
+				if err == nil {
+					if !cidr.Contains(net.IP(nsIP)) {
+						ns.Servers = append(ns.Servers, nsIP)
+					}
+				}
 			}
 			ns.Create(db.WithContext(context.TODO()))
 			netI.NameServers = []string{}