Class: Resolv::DNS

# File 'lib/resolv.rb', line 639

def self.bind_random_port(udpsock, bind_host="0.0.0.0") # :nodoc:
  begin
    port = random(1024..65535)
    udpsock.bind(bind_host, port)
  rescue Errno::EADDRINUSE, # POSIX
         Errno::EACCES, # SunOS: See PRIV_SYS_NFS in privileges(5)
         Errno::EPERM # FreeBSD: security.mac.portacl.port_high is configurable.  See mac_portacl(4).
    retry
  end
end

# File 'lib/resolv.rb', line 627

def self.free_request_id(host, port, id) # :nodoc:
  RequestIDMutex.synchronize {
    key = [host, port]
    if h = RequestID[key]
      h.delete id
      if h.empty?
        RequestID.delete key
      end
    end
  }
end

# File 'lib/resolv.rb', line 296

def self.open(*args)
  dns = new(*args)
  return dns unless block_given?
  begin
    yield dns
  ensure
    dns.close
  end
end

# File 'lib/resolv.rb', line 599

def self.random(arg) # :nodoc:
  begin
    SecureRandom.random_number(arg)
  rescue NotImplementedError
    rand(arg)
  end
end

# File 'lib/resolv.rb', line 410

def use_ipv6? # :nodoc:
  begin
    list = Socket.ip_address_list
  rescue NotImplementedError
    return true
  end
  list.any? {|a| a.ipv6? && !a.ipv6_loopback? && !a.ipv6_linklocal? }
end

# File 'lib/resolv.rb', line 403

def each_address(name)
  each_resource(name, Resource::IN::A) {|resource| yield resource.address}
  if use_ipv6?
    each_resource(name, Resource::IN::AAAA) {|resource| yield resource.address}
  end
end

# File 'lib/resolv.rb', line 450

def each_name(address)
  case address
  when Name
    ptr = address
  when IPv4::Regex
    ptr = IPv4.create(address).to_name
  when IPv6::Regex
    ptr = IPv6.create(address).to_name
  else
    raise ResolvError.new("cannot interpret as address: #{address}")
  end
  each_resource(ptr, Resource::IN::PTR) {|resource| yield resource.name}
end

# File 'lib/resolv.rb', line 506

def each_resource(name, typeclass, &proc)
  fetch_resource(name, typeclass) {|reply, reply_name|
    extract_resources(reply, reply_name, typeclass, &proc)
  }
end

# File 'lib/resolv.rb', line 567

def extract_resources(msg, name, typeclass) # :nodoc:
  if typeclass < Resource::ANY
    n0 = Name.create(name)
    msg.each_resource {|n, ttl, data|
      yield data if n0 == n
    }
  end
  yielded = false
  n0 = Name.create(name)
  msg.each_resource {|n, ttl, data|
    if n0 == n
      case data
      when typeclass
        yield data
        yielded = true
      when Resource::CNAME
        n0 = data.name
      end
    end
  }
  return if yielded
  msg.each_resource {|n, ttl, data|
    if n0 == n
      case data
      when typeclass
        yield data
      end
    end
  }
end

# File 'lib/resolv.rb', line 512

def fetch_resource(name, typeclass)
  lazy_initialize
  requester = make_udp_requester
  senders = {}
  begin
    @config.resolv(name) {|candidate, tout, nameserver, port|
      msg = Message.new
      msg.rd = 1
      msg.add_question(candidate, typeclass)
      unless sender = senders[[candidate, nameserver, port]]
        sender = requester.sender(msg, candidate, nameserver, port)
        next if !sender
        senders[[candidate, nameserver, port]] = sender
      end
      reply, reply_name = requester.request(sender, tout)
      case reply.rcode
      when RCode::NoError
        if reply.tc == 1 and not Requester::TCP === requester
          requester.close
          # Retry via TCP:
          requester = make_tcp_requester(nameserver, port)
          senders = {}
          # This will use TCP for all remaining candidates (assuming the
          # current candidate does not already respond successfully via
          # TCP).  This makes sense because we already know the full
          # response will not fit in an untruncated UDP packet.
          redo
        else
          yield(reply, reply_name)
        end
        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

# File 'lib/resolv.rb', line 379

def getaddress(name)
  each_address(name) {|address| return address}
  raise ResolvError.new("DNS result has no information for #{name}")
end

# File 'lib/resolv.rb', line 390

def getaddresses(name)
  ret = []
  each_address(name) {|address| ret << address}
  return ret
end

# File 'lib/resolv.rb', line 426

def getname(address)
  each_name(address) {|name| return name}
  raise ResolvError.new("DNS result has no information for #{address}")
end

# File 'lib/resolv.rb', line 437

def getnames(address)
  ret = []
  each_name(address) {|name| ret << name}
  return ret
end

# File 'lib/resolv.rb', line 487

def getresource(name, typeclass)
  each_resource(name, typeclass) {|resource| return resource}
  raise ResolvError.new("DNS result has no information for #{name}")
end

# File 'lib/resolv.rb', line 496

def getresources(name, typeclass)
  ret = []
  each_resource(name, typeclass) {|resource| ret << resource}
  return ret
end

# File 'lib/resolv.rb', line 352

def lazy_initialize # :nodoc:
  @mutex.synchronize {
    unless @initialized
      @config.lazy_initialize
      @initialized = true
    end
  }
  self
end

# File 'lib/resolv.rb', line 563

def make_tcp_requester(host, port) # :nodoc:
  return Requester::TCP.new(host, port)
end

# File 'lib/resolv.rb', line 554

def make_udp_requester # :nodoc:
  nameserver_port = @config.nameserver_port
  if nameserver_port.length == 1
    Requester::ConnectedUDP.new(*nameserver_port[0])
  else
    Requester::UnconnectedUDP.new(*nameserver_port)
  end
end