fqdn_only_resolver.go

package resolver

import (
	"strings"

	"github.com/0xERR0R/blocky/config"
	"github.com/0xERR0R/blocky/model"
	"github.com/0xERR0R/blocky/util"
	"github.com/miekg/dns"
)

type FqdnOnlyResolver struct {
	NextResolver
	enabled bool
}

func NewFqdnOnlyResolver(cfg config.Config) ChainedResolver {
	return &FqdnOnlyResolver{
		enabled: cfg.FqdnOnly,
	}
}

func (r *FqdnOnlyResolver) Resolve(request *model.Request) (*model.Response, error) {
	if r.enabled {
		domainFromQuestion := util.ExtractDomain(request.Req.Question[0])
		if !strings.Contains(domainFromQuestion, ".") {
			response := new(dns.Msg)
			response.Rcode = dns.RcodeNameError

			return &model.Response{Res: response, RType: model.ResponseTypeNOTFQDN, Reason: "NOTFQDN"}, nil
		}
	}

	return r.next.Resolve(request)
}

func (r *FqdnOnlyResolver) Configuration() (result []string) {
	if !r.enabled {
		return configDisabled
	}

	return configEnabled
}