xmengnet/mihomo
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Chore: clean up code

Browse source
This commit is contained in:
Dreamacro 2019-05-14 21:35:34 +08:00
parent 0d4a999707
commit 71a08ad8e2
2 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
dns/client.go
View file

@ -99,7 +99,7 @@ func (r *Resolver) exchange(servers []*nameserver, m *D.Msg) (msg *D.Msg, err er
if err != nil || msg.Rcode != D.RcodeSuccess { if err != nil || msg.Rcode != D.RcodeSuccess {
return return
} }
in <- &result{Msg: msg, Error: err} in <- msg
}(server) }(server)
} }
@ -114,8 +114,7 @@ func (r *Resolver) exchange(servers []*nameserver, m *D.Msg) (msg *D.Msg, err er
return nil, errors.New("All DNS requests failed") return nil, errors.New("All DNS requests failed")
} }
resp := elm.(*result) msg = elm.(*D.Msg)
msg, err = resp.Msg, resp.Error
return return
} }

5
dns/server.go
View file

@ -57,10 +57,9 @@ func (s *Server) handleFakeIP(r *D.Msg) (msg *D.Msg, err error) {
q := r.Question[0] q := r.Question[0]
cache, _ := s.r.cache.GetWithExpire("fakeip:" + q.String()) cache := s.r.cache.Get("fakeip:" + q.String())
if cache != nil { if cache != nil {
msg = cache.(*D.Msg).Copy() msg = cache.(*D.Msg).Copy()
setMsgTTL(msg, 1)
return return
} }
@ -72,6 +71,8 @@ func (s *Server) handleFakeIP(r *D.Msg) (msg *D.Msg, err error) {
putMsgToCache(s.r.cache, "fakeip:"+q.String(), msg) putMsgToCache(s.r.cache, "fakeip:"+q.String(), msg)
putMsgToCache(s.r.cache, ip.String(), msg) putMsgToCache(s.r.cache, ip.String(), msg)
// putMsgToCache depend on msg ttl to set cache expired time, then set msg ref ttl to 1
setMsgTTL(msg, 1) setMsgTTL(msg, 1)
}() }()