瀏覽代碼

remove peers blocked by acl

Matthew R Kasun 2 年之前
父節點
當前提交
ea3b01f315
共有 2 個文件被更改,包括 3 次插入0 次删除
  1. 2 0
      logic/peers.go
  2. 1 0
      mq/relay.go

+ 2 - 0
logic/peers.go

@@ -791,6 +791,8 @@ func GetPeerUpdate(host *models.Host) []wgtypes.PeerConfig {
 				update.AllowedIPs = append(update.AllowedIPs, AddAllowedIPs(&peer)...)
 				peerUpdate = append(peerUpdate, update)
 			} else {
+				update.Remove = true
+				peerUpdate = append(peerUpdate, update)
 				log.Println("node not allowed", client.Host.Name, peer.Host.Name)
 			}
 		}

+ 1 - 0
mq/relay.go

@@ -47,6 +47,7 @@ func PubPeerUpdate(client, relay *models.Client, peers []models.Client) {
 			log.Println("node allowed", client.Host.Name, peer.Host.Name)
 			update.AllowedIPs = append(update.AllowedIPs, logic.AddAllowedIPs(&peer)...)
 		} else {
+			update.Remove = true
 			log.Println("node not allowed", client.Host.Name, client.Node.Address, peer.Host.Name, peer.Node.Address)
 		}
 		if relay != nil {