ef315ef3e2
* Add edns0 code rewrite * check arg count * change `new`; set EDNS0 if request doesn't have it set * change set to replace_or_append * change to append_or_replace * return error in new * update documents * fixt UT * return error * go fmt * Rework for more general EDNS0 use Also changed how rules are created and validated. Implements EDNS0 NSID in addition to local. * go fmt * README updates, NSID tests and fixes * gofmt -s -w * Fix tests for rewrite syntax change * Add tests, fix error message * Review nits * Missed on nit * More tests, integration test, fix edns0 parse issue * Fix README, use RewriteIgnored * go fmt
24 lines
496 B
Go
24 lines
496 B
Go
package rewrite
|
|
|
|
import (
|
|
"github.com/coredns/coredns/middleware"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
type nameRule struct {
|
|
From, To string
|
|
}
|
|
|
|
func newNameRule(from, to string) (Rule, error) {
|
|
return &nameRule{middleware.Name(from).Normalize(), middleware.Name(to).Normalize()}, nil
|
|
}
|
|
|
|
// Rewrite rewrites the the current request.
|
|
func (rule *nameRule) Rewrite(r *dns.Msg) Result {
|
|
if rule.From == r.Question[0].Name {
|
|
r.Question[0].Name = rule.To
|
|
return RewriteDone
|
|
}
|
|
return RewriteIgnored
|
|
}
|