2014-09-19 13:04:53 +10:00
|
|
|
package dns
|
|
|
|
|
2014-09-19 23:40:24 +10:00
|
|
|
import (
|
2014-09-21 08:04:02 +10:00
|
|
|
"fmt"
|
2014-09-19 23:40:24 +10:00
|
|
|
"strings"
|
|
|
|
)
|
|
|
|
|
2014-09-21 08:04:02 +10:00
|
|
|
// PrivateRdata is an interface to implement non-RFC dictated resource records. See also dns.PrivateRR, dns.NewPrivateRR and dns.DelPrivateRR
|
|
|
|
type PrivateRdata interface {
|
2014-09-19 13:04:53 +10:00
|
|
|
String() string
|
2014-09-21 08:04:02 +10:00
|
|
|
ParseTextSlice([]string) error
|
|
|
|
WriteByteSlice([]byte) (int, error)
|
|
|
|
ParseByteSlice([]byte) (int, error)
|
|
|
|
PasteRdata(PrivateRdata) error
|
2014-09-19 23:40:24 +10:00
|
|
|
RdataLen() int
|
2014-09-19 13:04:53 +10:00
|
|
|
}
|
|
|
|
|
2014-09-21 08:04:02 +10:00
|
|
|
// PrivateRR represents RR that uses PrivateRdata user-defined type. It mocks normal RRs and implements dns.RR interface.
|
2014-09-19 23:40:24 +10:00
|
|
|
type PrivateRR struct {
|
2014-09-19 13:04:53 +10:00
|
|
|
Hdr RR_Header
|
2014-09-21 08:04:02 +10:00
|
|
|
Data PrivateRdata
|
2014-09-19 13:04:53 +10:00
|
|
|
}
|
|
|
|
|
2014-09-21 08:04:02 +10:00
|
|
|
// Panics if RR is not an instance of PrivateRR
|
|
|
|
func mkPrivateRR(rrtype uint16) *PrivateRR {
|
|
|
|
rrfunc, ok := typeToRR[rrtype]
|
|
|
|
if !ok {
|
|
|
|
panic(fmt.Sprintf("dns: invalid operation with Private RR type %d", rrtype))
|
|
|
|
}
|
2014-09-19 23:40:24 +10:00
|
|
|
|
2014-09-21 08:04:02 +10:00
|
|
|
anyrr := rrfunc()
|
|
|
|
switch rr := anyrr.(type) {
|
|
|
|
case *PrivateRR:
|
|
|
|
return rr
|
|
|
|
}
|
|
|
|
panic(fmt.Sprintf("dns: RR is not a PrivateRR, typeToRR[%d] generator returned %T", rrtype, anyrr))
|
|
|
|
}
|
|
|
|
|
|
|
|
func (r *PrivateRR) Header() *RR_Header { return &r.Hdr }
|
|
|
|
func (r *PrivateRR) String() string { return r.Hdr.String() + r.Data.String() }
|
2014-09-19 23:40:24 +10:00
|
|
|
|
|
|
|
// Private len and copy parts to satisfy RR interface.
|
|
|
|
func (r *PrivateRR) len() int { return r.Hdr.len() + r.Data.RdataLen() }
|
|
|
|
func (r *PrivateRR) copy() RR {
|
2014-09-19 13:04:53 +10:00
|
|
|
// make new RR like this:
|
2014-09-21 08:04:02 +10:00
|
|
|
rr := mkPrivateRR(r.Hdr.Rrtype)
|
|
|
|
newh := r.Hdr.copyHeader()
|
|
|
|
rr.Hdr = *newh
|
2014-09-19 13:04:53 +10:00
|
|
|
|
2014-09-21 08:04:02 +10:00
|
|
|
err := r.Data.PasteRdata(rr.Data)
|
2014-09-19 13:04:53 +10:00
|
|
|
if err != nil {
|
2014-09-19 23:40:24 +10:00
|
|
|
panic("dns: got value that could not be used to copy Private rdata")
|
2014-09-19 13:04:53 +10:00
|
|
|
}
|
|
|
|
|
|
|
|
return rr
|
|
|
|
}
|
|
|
|
|
2014-09-21 08:04:02 +10:00
|
|
|
// NewPrivateRR adds support for user-defined resource record type to internals of dns library. Requires
|
2014-09-19 23:40:24 +10:00
|
|
|
// string and numeric representation of RR type and generator function as argument.
|
2014-09-21 08:04:02 +10:00
|
|
|
func NewPrivateRR(rtypestr string, rtype uint16, generator func() PrivateRdata) {
|
2014-09-19 23:40:24 +10:00
|
|
|
rtypestr = strings.ToUpper(rtypestr)
|
2014-09-19 13:04:53 +10:00
|
|
|
|
2014-09-19 23:40:24 +10:00
|
|
|
typeToRR[rtype] = func() RR { return &PrivateRR{RR_Header{}, generator()} }
|
2014-09-19 13:04:53 +10:00
|
|
|
TypeToString[rtype] = rtypestr
|
|
|
|
StringToType[rtypestr] = rtype
|
|
|
|
|
2014-09-19 23:40:24 +10:00
|
|
|
setPrivateRR := func(h RR_Header, c chan lex, o, f string) (RR, *ParseError, string) {
|
2014-09-21 08:04:02 +10:00
|
|
|
rr := mkPrivateRR(h.Rrtype)
|
|
|
|
rr.Hdr = h
|
2014-09-19 13:04:53 +10:00
|
|
|
|
|
|
|
var l lex
|
2014-09-21 08:04:02 +10:00
|
|
|
text := make([]string, 0, 2) // could be 0..N elements, median is probably 1
|
|
|
|
FETCH:
|
|
|
|
for {
|
2014-09-19 13:04:53 +10:00
|
|
|
switch l = <-c; l.value {
|
|
|
|
case _NEWLINE, _EOF:
|
2014-09-21 08:04:02 +10:00
|
|
|
break FETCH
|
2014-09-19 13:04:53 +10:00
|
|
|
case _STRING:
|
|
|
|
text = append(text, l.token)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-09-21 08:04:02 +10:00
|
|
|
err := rr.Data.ParseTextSlice(text)
|
2014-09-19 13:04:53 +10:00
|
|
|
if err != nil {
|
|
|
|
return nil, &ParseError{f, err.Error(), l}, ""
|
|
|
|
}
|
|
|
|
|
|
|
|
return rr, nil, ""
|
|
|
|
}
|
|
|
|
|
2014-09-19 23:40:24 +10:00
|
|
|
typeToparserFunc[rtype] = parserFunc{setPrivateRR, false}
|
2014-09-19 13:04:53 +10:00
|
|
|
}
|
|
|
|
|
2014-09-21 08:04:02 +10:00
|
|
|
// DelPrivateRR removes defenitions required to support user RR type.
|
|
|
|
func DelPrivateRR(rtype uint16) {
|
2014-09-19 13:04:53 +10:00
|
|
|
rtypestr, ok := TypeToString[rtype]
|
|
|
|
if ok {
|
|
|
|
delete(typeToRR, rtype)
|
|
|
|
delete(TypeToString, rtype)
|
|
|
|
delete(typeToparserFunc, rtype)
|
|
|
|
delete(StringToType, rtypestr)
|
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|