github/jwhited-wgsd
1
0
Fork 0
mirror of https://github.com/jwhited/wgsd.git synced 2025-04-04 11:09:31 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

cleaner comments

Browse Source
This commit is contained in:
George Harvey 2021-04-06 15:15:53 +01:00 committed by GitHub
parent b3919d152b
commit b213b48876
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
wgsd.go
View File

@ -178,16 +178,17 @@ func getSelfPeer(zone *Zone, device *wgtypes.Device, state request.Request) (wgt
return self, nil return self, nil
} }
func networkSizes(ips []*net.IPNet) (int, float64) { // takes a list of IPNets e.g "192.168.0.2/32" and
var len int // and the total number of addresses exposed float64
// assumed that the networks do not overlap
func networkSizes(ips []*net.IPNet) float64 {
var tot float64 var tot float64
for _, ip := range ips { for _, ip := range ips {
ones, bits := ip.Mask.Size() ones, bits := ip.Mask.Size()
bitsize := (bits - ones) bitsize := (bits - ones)
len ++
tot += math.Pow(2,float64(bitsize)) tot += math.Pow(2,float64(bitsize))
} }
return len, tot return tot
} }
func getPeers(client wgctrlClient, zone *Zone, state request.Request) ( func getPeers(client wgctrlClient, zone *Zone, state request.Request) (
@ -201,7 +202,7 @@ func getPeers(client wgctrlClient, zone *Zone, state request.Request) (
peers = append(peers, device.Peers...) peers = append(peers, device.Peers...)
} else { } else {
for _, peer in range device.Peers{ for _, peer in range device.Peers{
if l , t := networkSizes(networks); l > 1 || t > 1 { if t := networkSizes(networks); t > 1 { // could be more complex, e.g allow user to specify a threshold
peers = append(peers, peer) peers = append(peers, peer)
} }
} }
@ -223,7 +224,7 @@ func (p *WGSD) ServeDNS(ctx context.Context, w dns.ResponseWriter,
state := request.Request{W: w, Req: r} state := request.Request{W: w, Req: r}
// Check if the request is // Check if the request is
a zone we are serving. If it doesn't match we //a zone we are serving. If it doesn't match we
// pass the request on to the next plugin. // pass the request on to the next plugin.
zoneName := plugin.Zones(p.Names).Matches(state.Name()) zoneName := plugin.Zones(p.Names).Matches(state.Name())
if zoneName == "" { if zoneName == "" {
@ -340,7 +341,7 @@ func soa(zone string) dns.RR {
Refresh: 86400, Refresh: 86400,
Retry: 7200, Retry: 7200,
Expire: 3600000, Expire: 3600000,
Minttl: 60,pe Minttl: 60,
} }
} }