ns.go

// sean at shanghai 2021

package main

import (
	"log"

	"github.com/miekg/dns"
)

type NSHandler struct{}

func (ns *NSHandler) FillRecords(req *dns.Msg, records []DNSRecord) *dns.Msg {
	m := new(dns.Msg)
	m.SetReply(req)
	m.Authoritative = true
	rr := make([]dns.NS, len(records))
	for idx, record := range records {
		rr[idx].Hdr.Name = req.Question[0].Name
		rr[idx].Hdr.Rrtype = dns.TypeNS
		rr[idx].Hdr.Class = dns.ClassINET
		rr[idx].Hdr.Ttl = record.Ttl
		rr[idx].Ns = record.Ns
		m.Answer = append(m.Answer, &rr[idx])
	}
	return m
}

func (ns *NSHandler) CheckRecord(record *DNSRecord) error {
	is := GoodName(record.Ns)
	if !is {
		log.Println("bad ns value:", record.Ns)
		return ErrBadValue
	}
	record.Ns = AppendDot(record.Ns)
	return nil
}

func (ns *NSHandler) RRToRecord(msg dns.RR) DNSRecord {
	var record DNSRecord
	v := msg.(*dns.NS)
	record.Ns = v.Ns
	record.Ttl = v.Hdr.Ttl
	return record
}