zishuo/netmaker
1
0
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2025-10-15 21:32:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

added time to prevent UDP reset

Browse Source
This commit is contained in:
0xdcarns
2022-02-03 09:51:16 -05:00
parent a693217feb
commit 06e2603c20
6 changed files with 27 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
netclient/local/routes.go
View File

@@ -7,8 +7,8 @@ import (
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
)
// SetPeerRoutes - sets/removes ip routes for each peer on a network
func SetPeerRoutes(iface string, oldPeers map[string][]net.IPNet, newPeers []wgtypes.PeerConfig) {
// traverse through all recieved peers
for _, peer := range newPeers {
// if pubkey found in existing peers, check against existing peer
@@ -29,9 +29,9 @@ func SetPeerRoutes(iface string, oldPeers map[string][]net.IPNet, newPeers []wgt
}
}
}
delete(oldPeers, peer.PublicKey.String())
delete(oldPeers, peer.PublicKey.String()) // remove peer as it was found and processed
} else {
for _, allowedIP := range peer.AllowedIPs {
for _, allowedIP := range peer.AllowedIPs { // add all routes as peer doesn't exist
if err := setRoute(iface, &allowedIP); err != nil {
ncutils.PrintLog(err.Error(), 1)
}
@@ -39,7 +39,7 @@ func SetPeerRoutes(iface string, oldPeers map[string][]net.IPNet, newPeers []wgt
}
}
// traverse through all existing peers
// traverse through all remaining existing peers
for _, allowedIPs := range oldPeers {
for _, allowedIP := range allowedIPs {
deleteRoute(iface, &allowedIP)