You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
819 B
Go
41 lines
819 B
Go
package main
|
|
|
|
import (
|
|
"net"
|
|
|
|
"github.com/fangdingjun/go-log"
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
func (srv *server) handleTCP(c net.Conn) {
|
|
defer c.Close()
|
|
log.Debugln("tcp from", c.RemoteAddr())
|
|
conn := dns.Conn{Conn: c}
|
|
for {
|
|
msg, err := conn.ReadMsg()
|
|
if err != nil {
|
|
log.Debugln("tcp read message", err)
|
|
break
|
|
}
|
|
reply := false
|
|
for _, up := range srv.upstreams {
|
|
log.Debugf("from %s query upstream %s", conn.RemoteAddr(), up.String())
|
|
log.Debugln("query", msg.Question[0].String())
|
|
m, err := queryUpstream(msg, up)
|
|
if err == nil {
|
|
for _, a := range m.Answer {
|
|
log.Debugln("result", a.String())
|
|
}
|
|
log.Debugln("got reply", m.String())
|
|
conn.WriteMsg(m)
|
|
reply = true
|
|
break
|
|
}
|
|
log.Debugln("query upstream", up.String(), err)
|
|
}
|
|
if !reply {
|
|
break
|
|
}
|
|
}
|
|
}
|