summaryrefslogtreecommitdiff
path: root/dns/cilent.go
blob: 63e3734ff2031be89aee7f03aa52add8994bf2ae (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package dns

import (
	"code.crute.me/mcrute/go_ddns_manager/bind"
	"github.com/miekg/dns"
)

type DNSClient struct {
	Server string
}

func (c *DNSClient) AXFR(zone *bind.Zone) (chan *dns.Envelope, error) {
	k := zone.Keys()[0]
	t := &dns.Transfer{TsigSecret: k.AsMap()}

	m := &dns.Msg{}
	m.SetAxfr(zone.Name)
	k.Sign(m)

	return t.In(m, c.Server)
}

func (c *DNSClient) Insert(zone *bind.Zone, rrs ...RR) error {
	k := zone.Keys()[0]
	dc := &dns.Client{TsigSecret: k.AsMap()}

	m := &dns.Msg{}
	m.SetUpdate(zone.Name)
	m.Insert(toRRSet(zone, rrs...))
	k.Sign(m)

	if _, _, err := dc.Exchange(m, c.Server); err != nil {
		return err
	}

	return nil
}

func (c *DNSClient) Remove(zone *bind.Zone, rrs ...RR) error {
	k := zone.Keys()[0]
	dc := &dns.Client{TsigSecret: k.AsMap()}

	m := &dns.Msg{}
	m.SetUpdate(zone.Name)
	m.Remove(toRRSet(zone, rrs...))
	k.Sign(m)

	if _, _, err := dc.Exchange(m, c.Server); err != nil {
		return err
	}

	return nil
}

func (c *DNSClient) RemoveAll(zone *bind.Zone, rrs ...RR) error {
	k := zone.Keys()[0]
	dc := &dns.Client{TsigSecret: k.AsMap()}

	m := &dns.Msg{}
	m.SetUpdate(zone.Name)
	m.RemoveRRset(toRRSet(zone, rrs...))
	k.Sign(m)

	if _, _, err := dc.Exchange(m, c.Server); err != nil {
		return err
	}

	return nil
}