91 lines
2.1 KiB
Go
91 lines
2.1 KiB
Go
|
package dns
|
||
|
|
|
||
|
|
type CustomRData interface {
|
||
|
|
String() string
|
||
|
|
ReadText([]string) error
|
||
|
|
Write([]byte) (int, error)
|
||
|
|
Read([]byte) (int, error)
|
||
|
|
CopyTo(CustomRData) error
|
||
|
|
}
|
||
|
|
|
||
|
|
// CopyTo needs to be here to avoid write/read pass that would require make([]byte, xxxxxx)
|
||
|
|
|
||
|
|
type CustomRR struct {
|
||
|
|
Hdr RR_Header
|
||
|
|
Data CustomRData
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *CustomRR) Header() *RR_Header { return &r.Hdr }
|
||
|
|
func (r *CustomRR) String() string { return r.Hdr.String() + r.Data.String() }
|
||
|
|
func (r *CustomRR) copy() RR {
|
||
|
|
// make new RR like this:
|
||
|
|
rrfunc, ok := typeToRR[r.Hdr.Rrtype]
|
||
|
|
if !ok {
|
||
|
|
panic("dns: invalid operation with custom RR " + r.Hdr.String())
|
||
|
|
}
|
||
|
|
rr := rrfunc()
|
||
|
|
r.Header().CopyTo(rr)
|
||
|
|
|
||
|
|
rrcust, ok := rr.(*CustomRR)
|
||
|
|
if !ok {
|
||
|
|
panic("dns: custom RR generator returned wrong interface value")
|
||
|
|
}
|
||
|
|
|
||
|
|
err := r.Data.CopyTo(rrcust.Data)
|
||
|
|
if err != nil {
|
||
|
|
panic("dns: got value that could not be used to copy custom rdata")
|
||
|
|
}
|
||
|
|
|
||
|
|
return rr
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *CustomRR) len() int { panic("TODO: WHERE THIS IS USED?"); return 0 }
|
||
|
|
|
||
|
|
func RegisterCustomRR(rtypestr string, rtype uint16, generator func() CustomRData) {
|
||
|
|
typeToRR[rtype] = func() RR { return &CustomRR{RR_Header{}, generator()} }
|
||
|
|
TypeToString[rtype] = rtypestr
|
||
|
|
StringToType[rtypestr] = rtype
|
||
|
|
|
||
|
|
setCustomRR := func(h RR_Header, c chan lex, o, f string) (RR, *ParseError, string) {
|
||
|
|
rrfunc := typeToRR[h.Rrtype]
|
||
|
|
rr, ok := rrfunc().(*CustomRR)
|
||
|
|
if !ok {
|
||
|
|
panic("dns: invalid handler registered for custom RR " + rtypestr)
|
||
|
|
}
|
||
|
|
h.CopyTo(rr)
|
||
|
|
|
||
|
|
var l lex
|
||
|
|
text := make([]string, 0)
|
||
|
|
for end := false; !end; {
|
||
|
|
switch l = <-c; l.value {
|
||
|
|
case _NEWLINE, _EOF:
|
||
|
|
end = true
|
||
|
|
case _STRING:
|
||
|
|
text = append(text, l.token)
|
||
|
|
case _BLANK:
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
err := rr.Data.ReadText(text)
|
||
|
|
if err != nil {
|
||
|
|
return nil, &ParseError{f, err.Error(), l}, ""
|
||
|
|
}
|
||
|
|
|
||
|
|
return rr, nil, ""
|
||
|
|
}
|
||
|
|
|
||
|
|
typeToparserFunc[rtype] = parserFunc{setCustomRR, false}
|
||
|
|
}
|
||
|
|
|
||
|
|
func UnregisterCustomRR(rtype uint16) {
|
||
|
|
rtypestr, ok := TypeToString[rtype]
|
||
|
|
if ok {
|
||
|
|
delete(typeToRR, rtype)
|
||
|
|
delete(TypeToString, rtype)
|
||
|
|
delete(typeToparserFunc, rtype)
|
||
|
|
delete(StringToType, rtypestr)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|