浏览代码

Merge pull request #230 from gravitl/v0.7.1

V0.7.1
dcarns 4 年之前
父节点
当前提交
47bc50b437
共有 1 个文件被更改,包括 10 次插入0 次删除
  1. 10 0
      netclient/server/grpc.go

+ 10 - 0
netclient/server/grpc.go

@@ -208,6 +208,16 @@ func GetPeers(macaddress string, network string, server string, dualstack bool,
 		}
 		var allowedips []net.IPNet
 		allowedips = append(allowedips, peeraddr)
+		// handle manually set peers
+		for _, allowedIp := range node.AllowedIPs {
+			if _, ipnet, err := net.ParseCIDR(allowedIp); err == nil {
+				nodeEndpointArr := strings.Split(node.Endpoint, ":")
+				if !ipnet.Contains(net.IP(nodeEndpointArr[0])) { // don't need to add an allowed ip that already exists..
+					allowedips = append(allowedips, *ipnet)
+				}
+			}
+		}
+		// handle egress gateway peers
 		if node.IsEgressGateway == "yes" {
 			hasGateway = true
 			ranges := node.EgressGatewayRanges