foosinn/webhook-rfc2136
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity

feat(v4-only): allow ipv4 only support

Browse source
This commit is contained in:
foosinn 2023-03-20 01:13:12 +01:00
parent a82fd88827
commit f0844e50b5
1 changed files with 17 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

24
main.go
View file

@ -38,7 +38,7 @@ func main() {
} }
} }
func (a *App) handler (w http.ResponseWriter, r *http.Request) { func (a *App) handler(w http.ResponseWriter, r *http.Request) {
query := r.URL.Query() query := r.URL.Query()
token := query.Get("token") token := query.Get("token")
@ -50,7 +50,7 @@ func (a *App) handler (w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "no :(") fmt.Fprintf(w, "no :(")
return return
} }
if v4 == nil || v6 == nil { if v4 == nil {
w.WriteHeader(http.StatusBadRequest) w.WriteHeader(http.StatusBadRequest)
fmt.Fprintf(w, "no :(") fmt.Fprintf(w, "no :(")
return return
@ -73,21 +73,31 @@ func (a *App) dnsUpdate(v4, v6 net.IP) error {
// dns key name as fqdn // dns key name as fqdn
key := dns.Fqdn(a.DNSKeyName) key := dns.Fqdn(a.DNSKeyName)
// remove records (always includes v6)
delRra, _ := dns.NewRR(fmt.Sprintf("%s IN A 127.0.0.1", a.DNSRecord))
delRraaaa, _ := dns.NewRR(fmt.Sprintf("%s IN AAAA ::1", a.DNSRecord))
delRrs := []dns.RR{delRra, delRraaaa}
// create records // create records
rra := &dns.A{ rra := &dns.A{
Hdr: dns.RR_Header{Name: a.DNSRecord, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 300}, Hdr: dns.RR_Header{Name: a.DNSRecord, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 300},
A: v4, A: v4,
} }
rraaaa := &dns.AAAA{ rrs := []dns.RR{rra}
Hdr: dns.RR_Header{Name: a.DNSRecord, Rrtype: dns.TypeAAAA, Class: dns.ClassINET, Ttl: 300},
AAAA: v6, // only v6 if set
if v6 != nil {
rraaaa := &dns.AAAA{
Hdr: dns.RR_Header{Name: a.DNSRecord, Rrtype: dns.TypeAAAA, Class: dns.ClassINET, Ttl: 300},
AAAA: v6,
}
rrs = []dns.RR{rra, rraaaa}
} }
rrs := []dns.RR{rra, rraaaa}
// create message // create message
m := new(dns.Msg) m := new(dns.Msg)
m.SetUpdate(a.DNSZone) m.SetUpdate(a.DNSZone)
m.RemoveRRset(rrs) m.RemoveRRset(delRrs)
m.Insert(rrs) m.Insert(rrs)
m.SetTsig(key, a.DNSKeyAlgo, 300, time.Now().Unix()) m.SetTsig(key, a.DNSKeyAlgo, 300, time.Now().Unix())