Resolv::DNS - как мне заставить TCP?

Question

Кто-нибудь знает, как заставить TCP при использовании Resolv::DNS?
Кажется, что когда я прошу ANY записи, вывод усекается, и я получаю частичные результаты. Когда я выполняю много запросов (по одному для каждого типа записи), я получаю больше результатов. Я также получаю противоречивые результаты (разные машины, два последовательных запроса дают разные результаты,...)

Я думал, что это может быть связано с тем, что UDP ограничен размером пакета.

Любая идея, как я могу заставить его использовать TCP? Любой другой DNS-пакет, который я могу использовать?

Answer 1

У меня была такая же проблема, желая использовать Resolv для запросов только по протоколу TCP, поскольку я ожидал, что наборы результатов будут довольно большими. В итоге я перерыл исходный код Resolv и узнал, что по умолчанию TCP-запросы выполняются только в случае сбоя UDP-запроса. Я обнаружил, что я мог бы подкласс Resolv::DNS и переопределить each_resource метод. Вот мой источник:

require 'resolv'

# A TCP-only resolver built from `Resolv::DNS`. See the docs for what it's about.
# http://ruby-doc.org/stdlib-1.9.3/libdoc/resolv/rdoc/Resolv/DNS.html
class TcpDNS < Resolv::DNS
  # Override fetch_resource to use a TCP requester instead of a UDP requester. This
  # is mostly borrowed from `lib/resolv.rb` with the UDP->TCP fallback logic removed.
  def each_resource(name, typeclass, &proc)
    lazy_initialize
    senders = {}
    requester = nil
    begin
      @config.resolv(name) { |candidate, tout, nameserver, port|
        requester = make_tcp_requester(nameserver, port)
        msg = Message.new
        msg.rd = 1
        msg.add_question(candidate, typeclass)
        unless sender = senders[[candidate, nameserver, port]]
          sender = senders[[candidate, nameserver, port]] =
            requester.sender(msg, candidate, nameserver, port)
        end

        begin # HACK
          reply, reply_name = requester.request(sender, tout)
        rescue
          return
        end

        case reply.rcode
        when RCode::NoError
          extract_resources(reply, reply_name, typeclass, &proc)
          return
        when RCode::NXDomain
          raise Config::NXDomain.new(reply_name.to_s)
        else
          raise Config::OtherResolvError.new(reply_name.to_s)
        end
      }
    ensure
      requester.close
    end
  end
end

Тогда использовать его так же просто, как показано ниже:

TcpDNS.open :nameserver => ns_addrs, :search => '', :ndots => 1 do |dns|
  resp = dns.getresources target, Resolv::DNS::Resource::IN::ANY
end