167 lines
3.5 KiB
Go
167 lines
3.5 KiB
Go
package dns_test
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
const TypeISBN uint16 = 0xFF00
|
|
|
|
// A crazy new RR type :)
|
|
type ISBN struct {
|
|
x string // rdata with 10 or 13 numbers, dashes or spaces allowed
|
|
}
|
|
|
|
func NewISBN() dns.PrivateRdata { return &ISBN{""} }
|
|
|
|
func (rd *ISBN) Len() int { return len([]byte(rd.x)) }
|
|
func (rd *ISBN) String() string { return rd.x }
|
|
|
|
func (rd *ISBN) Parse(txt []string) error {
|
|
rd.x = strings.TrimSpace(strings.Join(txt, " "))
|
|
return nil
|
|
}
|
|
|
|
func (rd *ISBN) Pack(buf []byte) (int, error) {
|
|
b := []byte(rd.x)
|
|
n := copy(buf, b)
|
|
if n != len(b) {
|
|
return n, dns.ErrBuf
|
|
}
|
|
return n, nil
|
|
}
|
|
|
|
func (rd *ISBN) Unpack(buf []byte) (int, error) {
|
|
rd.x = string(buf)
|
|
return len(buf), nil
|
|
}
|
|
|
|
func (rd *ISBN) Copy(dest dns.PrivateRdata) error {
|
|
isbn, ok := dest.(*ISBN)
|
|
if !ok {
|
|
return dns.ErrRdata
|
|
}
|
|
isbn.x = rd.x
|
|
return nil
|
|
}
|
|
|
|
var testrecord = strings.Join([]string{"example.org.", "3600", "IN", "ISBN", "12-3 456789-0-123"}, "\t")
|
|
|
|
func TestPrivateText(t *testing.T) {
|
|
dns.PrivateHandle("ISBN", TypeISBN, NewISBN)
|
|
defer dns.PrivateHandleRemove(TypeISBN)
|
|
|
|
rr, err := dns.NewRR(testrecord)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if rr.String() != testrecord {
|
|
t.Errorf("record string representation did not match original %#v != %#v", rr.String(), testrecord)
|
|
}
|
|
}
|
|
|
|
func TestPrivateByteSlice(t *testing.T) {
|
|
dns.PrivateHandle("ISBN", TypeISBN, NewISBN)
|
|
defer dns.PrivateHandleRemove(TypeISBN)
|
|
|
|
rr, err := dns.NewRR(testrecord)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
buf := make([]byte, 100)
|
|
off, err := dns.PackRR(rr, buf, 0, nil, false)
|
|
if err != nil {
|
|
t.Errorf("got error packing ISBN: %v", err)
|
|
}
|
|
|
|
custrr := rr.(*dns.PrivateRR)
|
|
if ln := custrr.Data.Len() + len(custrr.Header().Name) + 11; ln != off {
|
|
t.Errorf("offset is not matching to length of Private RR: %d!=%d", off, ln)
|
|
}
|
|
|
|
rr1, off1, err := dns.UnpackRR(buf[:off], 0)
|
|
if err != nil {
|
|
t.Errorf("got error unpacking ISBN: %v", err)
|
|
return
|
|
}
|
|
|
|
if off1 != off {
|
|
t.Errorf("offset after unpacking differs: %d != %d", off1, off)
|
|
}
|
|
|
|
if rr1.String() != testrecord {
|
|
t.Errorf("record string representation did not match original %#v != %#v", rr1.String(), testrecord)
|
|
}
|
|
}
|
|
|
|
const TypeVERSION uint16 = 0xFF01
|
|
|
|
type VERSION struct {
|
|
x string
|
|
}
|
|
|
|
func NewVersion() dns.PrivateRdata { return &VERSION{""} }
|
|
|
|
func (rd *VERSION) String() string { return rd.x }
|
|
func (rd *VERSION) Parse(txt []string) error {
|
|
rd.x = strings.TrimSpace(strings.Join(txt, " "))
|
|
return nil
|
|
}
|
|
|
|
func (rd *VERSION) Pack(buf []byte) (int, error) {
|
|
b := []byte(rd.x)
|
|
n := copy(buf, b)
|
|
if n != len(b) {
|
|
return n, dns.ErrBuf
|
|
}
|
|
return n, nil
|
|
}
|
|
|
|
func (rd *VERSION) Unpack(buf []byte) (int, error) {
|
|
rd.x = string(buf)
|
|
return len(buf), nil
|
|
}
|
|
|
|
func (rd *VERSION) Copy(dest dns.PrivateRdata) error {
|
|
isbn, ok := dest.(*VERSION)
|
|
if !ok {
|
|
return dns.ErrRdata
|
|
}
|
|
isbn.x = rd.x
|
|
return nil
|
|
}
|
|
|
|
func (rd *VERSION) Len() int {
|
|
return len([]byte(rd.x))
|
|
}
|
|
|
|
var smallzone = `$ORIGIN example.org.
|
|
@ 3600 IN SOA sns.dns.icann.org. noc.dns.icann.org. (
|
|
2014091518 7200 3600 1209600 3600
|
|
)
|
|
A 1.2.3.4
|
|
ok ISBN 1231-92110-12
|
|
go VERSION (
|
|
1.3.1 ; comment
|
|
)
|
|
www ISBN 1231-92110-16
|
|
* CNAME @
|
|
`
|
|
|
|
func TestPrivateZoneParser(t *testing.T) {
|
|
dns.PrivateHandle("ISBN", TypeISBN, NewISBN)
|
|
dns.PrivateHandle("VERSION", TypeVERSION, NewVersion)
|
|
defer dns.PrivateHandleRemove(TypeISBN)
|
|
defer dns.PrivateHandleRemove(TypeVERSION)
|
|
|
|
r := strings.NewReader(smallzone)
|
|
for x := range dns.ParseZone(r, ".", "") {
|
|
if err := x.Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
}
|