xsteadfastx/dsnet
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

check to see if IP is already allocated

Browse Source
This commit is contained in:
Callan Bryant 2020-03-02 18:53:10 +00:00
parent 45ec0ff5b8
commit d0c33994f8
No known key found for this signature in database
GPG Key ID: C31FA9DF3ACBFFAA
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
types.go
View File

@ -7,6 +7,7 @@ import (
"strings" "strings"
"time" "time"
"fmt" "fmt"
"errors"
"golang.zx2c4.com/wireguard/wgctrl/wgtypes" "golang.zx2c4.com/wireguard/wgctrl/wgtypes"
) )
@ -82,8 +83,20 @@ func (conf *DsnetConfig) MustAddPeer(peer PeerConfig) {
conf.Peers = append(conf.Peers, peer) conf.Peers = append(conf.Peers, peer)
} }
func (conf DsnetConfig) IPAllocated(IP net.IP) bool {
for _, peer := range conf.Peers {
for _, peerIPNet := range peer.AllowedIPs {
if IP.Equal(peerIPNet.IP) {
return false
}
}
}
return true
}
// choose a free IP for a new Peer // choose a free IP for a new Peer
func (conf DsnetConfig) ChooseIP() net.IP { func (conf DsnetConfig) ChooseIP() (net.IP, error) {
network := conf.Network.IPNet network := conf.Network.IPNet
ones, bits := network.Mask.Size() ones, bits := network.Mask.Size()
zeros := bits - ones zeros := bits - ones
@ -98,12 +111,14 @@ func (conf DsnetConfig) ChooseIP() net.IP {
for j := 0; j < len(IP); j++ { for j := 0; j < len(IP); j++ {
shift := (len(IP) - j - 1) * 8 shift := (len(IP) - j - 1) * 8
IP[j] = IP[j] | byte(i>>shift) IP[j] = IP[j] | byte(i>>shift)
}
fmt.Println(IP) if conf.IPAllocated(IP) {
return IP, nil
}
}
} }
return net.IP{} return net.IP{}, errors.New("IP range exhausted")
} }
type Dsnet struct { type Dsnet struct {