From 69986e77dd8cf53de35980b889bced836a9783bc Mon Sep 17 00:00:00 2001 From: Matthias Radestock Date: Thu, 20 Aug 2015 18:46:55 +0100 Subject: [PATCH] cosmetic --- nameserver/dns.go | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/nameserver/dns.go b/nameserver/dns.go index 849881a1..23aefda7 100644 --- a/nameserver/dns.go +++ b/nameserver/dns.go @@ -166,8 +166,8 @@ func (h *handler) handleLocal(w dns.ResponseWriter, req *dns.Msg) { ip := addr.IP4() response.Answer[i] = &dns.A{Hdr: header, A: ip} } - shuffleAnswers(&response.Answer) + h.truncateResponse(req, &response) h.ns.debugf("response: %+v", response) @@ -255,17 +255,6 @@ func (h *handler) handleRecursive(w dns.ResponseWriter, req *dns.Msg) { h.errorResponse(req, dns.RcodeServerFailure, w) } -func shuffleAnswers(answers *[]dns.RR) { - if len(*answers) <= 1 { - return - } - - for i := range *answers { - j := rand.Intn(i + 1) - (*answers)[i], (*answers)[j] = (*answers)[j], (*answers)[i] - } -} - func (h *handler) truncateResponse(req, response *dns.Msg) { maxSize := h.getMaxResponseSize(req) if len(response.Answer) <= 1 || maxSize <= 0 { @@ -296,3 +285,14 @@ func (h *handler) getMaxResponseSize(req *dns.Msg) int { } return h.maxResponseSize } + +func shuffleAnswers(answers *[]dns.RR) { + if len(*answers) <= 1 { + return + } + + for i := range *answers { + j := rand.Intn(i + 1) + (*answers)[i], (*answers)[j] = (*answers)[j], (*answers)[i] + } +}