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

handle A/AAAA queries

Browse Source
This commit is contained in:
Jordan Whited 2020-05-12 17:35:05 -07:00
parent 8109291569
commit 8f7de25397
1 changed files with 56 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

93
wgsd.go
View File

@ -4,6 +4,7 @@ import (
"context" "context"
"encoding/base64" "encoding/base64"
"fmt" "fmt"
"net"
"strings" "strings"
"github.com/coredns/coredns/plugin" "github.com/coredns/coredns/plugin"
@ -56,7 +57,7 @@ func (p *WGSD) ServeDNS(ctx context.Context, w dns.ResponseWriter,
return dns.RcodeServerFailure, nil return dns.RcodeServerFailure, nil
} }
if len(device.Peers) == 0 { if len(device.Peers) == 0 {
return nxdomain(p.zone, w, r) return nxDomain(p.zone, w, r)
} }
// setup our reply message // setup our reply message
@ -86,10 +87,11 @@ func (p *WGSD) ServeDNS(ctx context.Context, w dns.ResponseWriter,
for _, peer := range device.Peers { for _, peer := range device.Peers {
if base64.StdEncoding.EncodeToString(peer.PublicKey[:]) == pubKey { if base64.StdEncoding.EncodeToString(peer.PublicKey[:]) == pubKey {
endpoint := peer.Endpoint endpoint := peer.Endpoint
if endpoint.IP == nil { hostRR := getHostRR(pubKey, p.zone, endpoint)
return nxdomain(p.zone, w, r) if hostRR == nil {
return nxDomain(p.zone, w, r)
} }
srvTarget := fmt.Sprintf("%s.%s", pubKey, p.zone) m.Extra = append(m.Extra, hostRR)
m.Answer = append(m.Answer, &dns.SRV{ m.Answer = append(m.Answer, &dns.SRV{
Hdr: dns.RR_Header{ Hdr: dns.RR_Header{
Name: state.Name(), Name: state.Name(),
@ -100,49 +102,66 @@ func (p *WGSD) ServeDNS(ctx context.Context, w dns.ResponseWriter,
Priority: 0, Priority: 0,
Weight: 0, Weight: 0,
Port: uint16(endpoint.Port), Port: uint16(endpoint.Port),
Target: srvTarget, Target: fmt.Sprintf("%s.%s", pubKey, p.zone),
}) })
switch { w.WriteMsg(m) // nolint: errcheck
case endpoint.IP.To4() != nil: return dns.RcodeSuccess, nil
m.Extra = append(m.Extra, &dns.A{ }
Hdr: dns.RR_Header{ }
Name: srvTarget, return nxDomain(p.zone, w, r)
Rrtype: dns.TypeA, case len(name) == keyLen+len(".") && (qtype == dns.TypeA ||
Class: dns.ClassINET, qtype == dns.TypeAAAA):
Ttl: 0, pubKey := name[:44]
}, for _, peer := range device.Peers {
A: endpoint.IP, if base64.StdEncoding.EncodeToString(peer.PublicKey[:]) == pubKey {
}) endpoint := peer.Endpoint
case endpoint.IP.To16() != nil: hostRR := getHostRR(pubKey, p.zone, endpoint)
m.Extra = append(m.Extra, &dns.AAAA{ if hostRR == nil {
Hdr: dns.RR_Header{ return nxDomain(p.zone, w, r)
Name: srvTarget,
Rrtype: dns.TypeAAAA,
Class: dns.ClassINET,
Ttl: 0,
},
AAAA: endpoint.IP,
})
default:
// TODO: this shouldn't happen
} }
w.WriteMsg(m) // nolint: errcheck w.WriteMsg(m) // nolint: errcheck
return dns.RcodeSuccess, nil return dns.RcodeSuccess, nil
} }
} }
return nxdomain(p.zone, w, r) return nxDomain(p.zone, w, r)
case len(name) == keyLen+len(".") && (qtype == dns.TypeA ||
qtype == dns.TypeAAAA):
// TODO: return A/AAAA for of peer
default: default:
return nxdomain(p.zone, w, r) return nxDomain(p.zone, w, r)
} }
w.WriteMsg(m) // nolint: errcheck
return dns.RcodeSuccess, nil
} }
func nxdomain(name string, w dns.ResponseWriter, r *dns.Msg) (int, error) { func getHostRR(pubKey, zone string, endpoint *net.UDPAddr) dns.RR {
if endpoint.IP == nil {
return nil
}
name := fmt.Sprintf("%s.%s", pubKey, zone)
switch {
case endpoint.IP.To4() != nil:
return &dns.A{
Hdr: dns.RR_Header{
Name: name,
Rrtype: dns.TypeA,
Class: dns.ClassINET,
Ttl: 0,
},
A: endpoint.IP,
}
case endpoint.IP.To16() != nil:
return &dns.AAAA{
Hdr: dns.RR_Header{
Name: name,
Rrtype: dns.TypeAAAA,
Class: dns.ClassINET,
Ttl: 0,
},
AAAA: endpoint.IP,
}
default:
// TODO: this shouldn't happen
return nil
}
}
func nxDomain(name string, w dns.ResponseWriter, r *dns.Msg) (int, error) {
m := new(dns.Msg) m := new(dns.Msg)
m.SetReply(r) m.SetReply(r)
m.Authoritative = true m.Authoritative = true