gdns/server_tcp.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
		}
	}
}