03ea2ae955
* Make request.Request smaller This makes the request struct smaller and removes the pointer to the do boolean (tri-bool) as size == 0 will indicate if we have cached it. Family can be a int8 because it only carries 3 values, Size itself is just a uint16 under the covers. This is a more comprehensive fix than #3292 Closes #3292 Signed-off-by: Miek Gieben <miek@miek.nl> * cache: fix test this now needs a valid response writter Signed-off-by: Miek Gieben <miek@miek.nl>
74 lines
1.9 KiB
Go
74 lines
1.9 KiB
Go
// Package edns provides function useful for adding/inspecting OPT records to/in messages.
|
|
package edns
|
|
|
|
import (
|
|
"errors"
|
|
"sync"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
var sup = &supported{m: make(map[uint16]struct{})}
|
|
|
|
type supported struct {
|
|
m map[uint16]struct{}
|
|
sync.RWMutex
|
|
}
|
|
|
|
// SetSupportedOption adds a new supported option the set of EDNS0 options that we support. Plugins typically call
|
|
// this in their setup code to signal support for a new option.
|
|
// By default we support:
|
|
// dns.EDNS0NSID, dns.EDNS0EXPIRE, dns.EDNS0COOKIE, dns.EDNS0TCPKEEPALIVE, dns.EDNS0PADDING. These
|
|
// values are not in this map and checked directly in the server.
|
|
func SetSupportedOption(option uint16) {
|
|
sup.Lock()
|
|
sup.m[option] = struct{}{}
|
|
sup.Unlock()
|
|
}
|
|
|
|
// SupportedOption returns true if the option code is supported as an extra EDNS0 option.
|
|
func SupportedOption(option uint16) bool {
|
|
sup.RLock()
|
|
_, ok := sup.m[option]
|
|
sup.RUnlock()
|
|
return ok
|
|
}
|
|
|
|
// Version checks the EDNS version in the request. If error
|
|
// is nil everything is OK and we can invoke the plugin. If non-nil, the
|
|
// returned Msg is valid to be returned to the client (and should). For some
|
|
// reason this response should not contain a question RR in the question section.
|
|
func Version(req *dns.Msg) (*dns.Msg, error) {
|
|
opt := req.IsEdns0()
|
|
if opt == nil {
|
|
return nil, nil
|
|
}
|
|
if opt.Version() == 0 {
|
|
return nil, nil
|
|
}
|
|
m := new(dns.Msg)
|
|
m.SetReply(req)
|
|
// zero out question section, wtf.
|
|
m.Question = nil
|
|
|
|
o := new(dns.OPT)
|
|
o.Hdr.Name = "."
|
|
o.Hdr.Rrtype = dns.TypeOPT
|
|
o.SetVersion(0)
|
|
m.Rcode = dns.RcodeBadVers
|
|
o.SetExtendedRcode(dns.RcodeBadVers)
|
|
m.Extra = []dns.RR{o}
|
|
|
|
return m, errors.New("EDNS0 BADVERS")
|
|
}
|
|
|
|
// Size returns a normalized size based on proto.
|
|
func Size(proto string, size uint16) uint16 {
|
|
if proto == "tcp" {
|
|
return dns.MaxMsgSize
|
|
}
|
|
if size < dns.MinMsgSize {
|
|
return dns.MinMsgSize
|
|
}
|
|
return size
|
|
}
|