Skip to content

Commit

Permalink
proxy: imp code
Browse files Browse the repository at this point in the history
  • Loading branch information
Mizzick committed Jun 21, 2024
1 parent 483e911 commit 7bd99c3
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 7 deletions.
10 changes: 4 additions & 6 deletions proxy/proxy.go
Original file line number Diff line number Diff line change
Expand Up @@ -553,8 +553,6 @@ func (p *Proxy) replyFromUpstream(d *DNSContext) (ok bool, err error) {
p.recDetector.add(d.Req)
}

l := p.logger.With(slogutil.KeyPrefix, "resolving from upstream")

start := time.Now()
src := "upstream"

Expand All @@ -563,12 +561,12 @@ func (p *Proxy) replyFromUpstream(d *DNSContext) (ok bool, err error) {
if dns64Ups := p.performDNS64(req, resp, upstreams); dns64Ups != nil {
u = dns64Ups
} else if p.isBogusNXDomain(resp) {
l.Debug("response contains bogus-nxdomain ip")
p.logger.Debug("response contains bogus-nxdomain ip")
resp = p.messages.NewMsgNXDOMAIN(req)
}

if err != nil && !isPrivate && p.Fallbacks != nil {
l.Debug("using fallback", slogutil.KeyError, err)
p.logger.Debug("using fallback", slogutil.KeyError, err)

// Reset the timer.
start = time.Now()
Expand All @@ -582,12 +580,12 @@ func (p *Proxy) replyFromUpstream(d *DNSContext) (ok bool, err error) {
}

if err != nil {
l.Debug("resolving err", "upstream", src, slogutil.KeyError, err)
p.logger.Debug("resolving err", "src", src, slogutil.KeyError, err)
}

if resp != nil {
d.QueryDuration = time.Since(start)
l.Debug("resolved", "upstream", src, "rtt", d.QueryDuration)
p.logger.Debug("resolved", "src", src, "rtt", d.QueryDuration)
}

p.handleExchangeResult(d, req, resp, u)
Expand Down
2 changes: 1 addition & 1 deletion proxy/server.go
Original file line number Diff line number Diff line change
Expand Up @@ -143,7 +143,7 @@ func (p *Proxy) validateRequest(d *DNSContext) (resp *dns.Msg) {
return p.messages.NewMsgSERVFAIL(d.Req)
case p.RefuseAny && d.Req.Question[0].Qtype == dns.TypeANY:
// Refuse requests of type ANY (anti-DDOS measure).
p.logger.Debug("refusing type=ANY request")
p.logger.Debug("refusing dns type any request")

return p.messages.NewMsgNOTIMPLEMENTED(d.Req)
case p.recDetector.check(d.Req):
Expand Down

0 comments on commit 7bd99c3

Please sign in to comment.