cares_wrap.cc

// Copyright Joyent, Inc. and other Node contributors.
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to permit
// persons to whom the Software is furnished to do so, subject to the
// following conditions:
//
// The above copyright notice and this permission notice shall be included
// in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
// USE OR OTHER DEALINGS IN THE SOFTWARE.

#include "cares_wrap.h"
#include "ada.h"
#include "async_wrap-inl.h"
#include "base_object-inl.h"
#include "env-inl.h"
#include "memory_tracker-inl.h"
#include "nbytes.h"
#include "node.h"
#include "node_errors.h"
#include "node_external_reference.h"
#include "req_wrap-inl.h"
#include "util-inl.h"
#include "uv.h"
#include "v8.h"

#include <cerrno>
#include <cstdlib>
#include <cstring>
#include <memory>
#include <string>
#include <string_view>
#include <unordered_set>
#include <utility>
#include <vector>

#ifndef T_TLSA
#define T_TLSA 52 /* TLSA certificate association */
#endif

#ifndef T_CAA
# define T_CAA    257 /* Certification Authority Authorization */
#endif

// OpenBSD does not define these
#ifndef AI_ALL
# define AI_ALL 0
#endif
#ifndef AI_V4MAPPED
# define AI_V4MAPPED 0
#endif


namespace node {
namespace cares_wrap {

using v8::Array;
using v8::ArrayBuffer;
using v8::Context;
using v8::DictionaryTemplate;
using v8::EscapableHandleScope;
using v8::Exception;
using v8::FunctionCallbackInfo;
using v8::FunctionTemplate;
using v8::HandleScope;
using v8::Int32;
using v8::Integer;
using v8::Isolate;
using v8::Just;
using v8::JustVoid;
using v8::Local;
using v8::LocalVector;
using v8::Maybe;
using v8::MaybeLocal;
using v8::Nothing;
using v8::Null;
using v8::Object;
using v8::String;
using v8::Uint32;
using v8::Undefined;
using v8::Value;

namespace {

Mutex ares_library_mutex;

inline uint16_t cares_get_16bit(const unsigned char* p) {
  return static_cast<uint32_t>(p[0] << 8U) | (static_cast<uint32_t>(p[1]));
}

void ares_poll_cb(uv_poll_t* watcher, int status, int events) {
  NodeAresTask* task = ContainerOf(&NodeAresTask::poll_watcher, watcher);
  ChannelWrap* channel = task->channel;

  /* Reset the idle timer */
  uv_timer_again(channel->timer_handle());

  if (status < 0) {
    /* An error happened. Just pretend that the socket is both readable and */
    /* writable. */
    ares_process_fd(channel->cares_channel(), task->sock, task->sock);
    return;
  }

  /* Process DNS responses */
  ares_process_fd(channel->cares_channel(),
                  events & UV_READABLE ? task->sock : ARES_SOCKET_BAD,
                  events & UV_WRITABLE ? task->sock : ARES_SOCKET_BAD);
}


void ares_poll_close_cb(uv_poll_t* watcher) {
  std::unique_ptr<NodeAresTask> free_me(
        ContainerOf(&NodeAresTask::poll_watcher, watcher));
}


/* Callback from ares when socket operation is started */
void ares_sockstate_cb(void* data, ares_socket_t sock, int read, int write) {
  ChannelWrap* channel = static_cast<ChannelWrap*>(data);
  NodeAresTask* task;

  NodeAresTask lookup_task;
  lookup_task.sock = sock;
  auto it = channel->task_list()->find(&lookup_task);

  task = (it == channel->task_list()->end()) ? nullptr : *it;

  if (read || write) {
    if (!task) {
      /* New socket */
      channel->StartTimer();

      task = NodeAresTask::Create(channel, sock);
      if (task == nullptr) {
        /* This should never happen unless we're out of memory or something */
        /* is seriously wrong. The socket won't be polled, but the query will */
        /* eventually time out. */
        return;
      }

      channel->task_list()->insert(task);
    }

    /* This should never fail. If it fails anyway, the query will eventually */
    /* time out. */
    uv_poll_start(&task->poll_watcher,
                  (read ? UV_READABLE : 0) | (write ? UV_WRITABLE : 0),
                  ares_poll_cb);

  } else {
    if (task != nullptr) {
      channel->task_list()->erase(it);
      channel->env()->CloseHandle(&task->poll_watcher, ares_poll_close_cb);
    }

    if (channel->task_list()->empty()) {
      channel->CloseTimer();
    }
  }
}

MaybeLocal<Array> HostentToNames(Environment* env, struct hostent* host) {
  EscapableHandleScope scope(env->isolate());

  LocalVector<Value> names(env->isolate());

  for (uint32_t i = 0; host->h_aliases[i] != nullptr; ++i) {
    names.emplace_back(OneByteString(env->isolate(), host->h_aliases[i]));
  }

  return scope.Escape(Array::New(env->isolate(), names.data(), names.size()));
}

template <typename T>
Local<Array> AddrTTLToArray(
    Environment* env,
    const T* addrttls,
    size_t naddrttls) {
  MaybeStackBuffer<Value, 8> ttls(env->isolate(), naddrttls);
  for (size_t i = 0; i < naddrttls; i++) {
    ttls[i] = Integer::NewFromUnsigned(env->isolate(), addrttls[i].ttl);
  }

  return ttls.ToArray();
}

// Parse the CSV produced by ares_get_servers_csv() back into (ip, port)
// pairs. Each entry is "ipv4:port" or "[ipv6]:port"; entries whose UDP and
// TCP ports differ are emitted in the "dns://host:port?tcpport=..." URI form.
// A trailing "%interface" is ignored.
std::vector<std::pair<std::string, int>> ParseServersCsv(const char* csv) {
  std::vector<std::pair<std::string, int>> servers;
  if (csv == nullptr) return servers;

  std::string_view all(csv);
  size_t pos = 0;
  while (pos <= all.size()) {
    size_t comma = all.find(',', pos);
    std::string_view entry = all.substr(
        pos,
        comma == std::string_view::npos ? std::string_view::npos : comma - pos);
    pos = comma == std::string_view::npos ? all.size() + 1 : comma + 1;

    // Drop a URI scheme prefix (only present when udp/tcp ports differ),
    // a URI query string, and/or a trailing interface name.
    size_t scheme = entry.find("://");
    if (scheme != std::string_view::npos) entry = entry.substr(scheme + 3);
    size_t query = entry.find('?');
    if (query != std::string_view::npos) entry = entry.substr(0, query);
    size_t iface = entry.find('%');
    if (iface != std::string_view::npos) entry = entry.substr(0, iface);
    if (entry.empty()) continue;

    std::string host;
    int port = 53;
    std::string_view port_str;
    if (entry.front() == '[') {
      size_t close = entry.find(']');
      if (close == std::string_view::npos) continue;
      host = std::string(entry.substr(1, close - 1));
      size_t colon = entry.find(':', close);
      if (colon != std::string_view::npos) port_str = entry.substr(colon + 1);
    } else {
      size_t colon = entry.rfind(':');
      if (colon != std::string_view::npos) {
        host = std::string(entry.substr(0, colon));
        port_str = entry.substr(colon + 1);
      } else {
        host = std::string(entry);
      }
    }
    if (!port_str.empty()) {
      std::string s(port_str);
      port = static_cast<int>(strtol(s.c_str(), nullptr, 10));
    }
    servers.emplace_back(std::move(host), port);
  }
  return servers;
}

int GetAnswerCountForTTLBuffer(const unsigned char* buf, int len) {
  static constexpr int kDNSAnswerCountOffset = 6;
  static constexpr int kAresDefaultTTLBufferLength = 256;
  if (len <= kDNSAnswerCountOffset + 1) {
    return kAresDefaultTTLBufferLength;
  }

  const int answer_count = (static_cast<int>(buf[kDNSAnswerCountOffset]) << 8) |
                           static_cast<int>(buf[kDNSAnswerCountOffset + 1]);
  return answer_count == 0 ? 1 : answer_count;
}

template <typename T>
std::vector<T> MakeAddrTTLBuffer(const unsigned char* buf, int len) {
  return std::vector<T>(GetAnswerCountForTTLBuffer(buf, len));
}

Maybe<int> ParseGeneralReply(Environment* env,
                             const unsigned char* buf,
                             int len,
                             int* type,
                             Local<Array> ret,
                             void* addrttls = nullptr,
                             int* naddrttls = nullptr) {
  HandleScope handle_scope(env->isolate());

  ares_dns_record_t* dnsrec = nullptr;
  int status = ares_dns_parse(buf, len, 0, &dnsrec);
  if (status != ARES_SUCCESS) {
    ares_dns_record_destroy(dnsrec);
    return Just<int>(status);
  }
  DeleteFnPtr<ares_dns_record_t, ares_dns_record_destroy> free_me(dnsrec);

  const size_t rr_count = ares_dns_record_rr_cnt(dnsrec, ARES_SECTION_ANSWER);

  /* If it's `CNAME_OR_A`, a response that carries a CNAME record is reported
   * as a CNAME; otherwise it is reported as an A record. */
  if (*type == ns_t_cname_or_a) {
    bool has_cname = false;
    for (size_t i = 0; i < rr_count; i++) {
      const ares_dns_rr_t* rr =
          ares_dns_record_rr_get(dnsrec, ARES_SECTION_ANSWER, i);
      if (ares_dns_rr_get_type(rr) == ARES_REC_TYPE_CNAME) {
        has_cname = true;
        break;
      }
    }
    *type = has_cname ? ns_t_cname : ns_t_a;
  }

  const uint32_t offset = ret->Length();
  uint32_t count = 0;

  auto append_name = [&](ares_dns_rec_type_t rec_type,
                         ares_dns_rr_key_t key) -> Maybe<bool> {
    for (size_t i = 0; i < rr_count; i++) {
      const ares_dns_rr_t* rr =
          ares_dns_record_rr_get(dnsrec, ARES_SECTION_ANSWER, i);
      if (ares_dns_rr_get_type(rr) != rec_type) continue;
      const char* name = ares_dns_rr_get_str(rr, key);
      if (name == nullptr) continue;
      if (ret->Set(env->context(),
                   offset + count++,
                   OneByteString(env->isolate(), name))
              .IsNothing()) {
        return Nothing<bool>();
      }
    }
    return Just(true);
  };

  switch (*type) {
    case ns_t_cname:
      // A cname lookup always returns a single record in practice but we
      // follow the common API here and return every CNAME target.
      if (append_name(ARES_REC_TYPE_CNAME, ARES_RR_CNAME_CNAME).IsNothing()) {
        return Nothing<int>();
      }
      break;
    case ns_t_ns:
      if (append_name(ARES_REC_TYPE_NS, ARES_RR_NS_NSDNAME).IsNothing()) {
        return Nothing<int>();
      }
      break;
    case ns_t_ptr:
      if (append_name(ARES_REC_TYPE_PTR, ARES_RR_PTR_DNAME).IsNothing()) {
        return Nothing<int>();
      }
      break;
    case ns_t_a: {
      auto* ttls = static_cast<ares_addrttl*>(addrttls);
      const int max = naddrttls != nullptr ? *naddrttls : 0;
      char ip[INET6_ADDRSTRLEN];
      for (size_t i = 0; i < rr_count; i++) {
        const ares_dns_rr_t* rr =
            ares_dns_record_rr_get(dnsrec, ARES_SECTION_ANSWER, i);
        if (ares_dns_rr_get_type(rr) != ARES_REC_TYPE_A) continue;
        if (ttls != nullptr && static_cast<int>(count) >= max) break;
        const struct in_addr* addr = ares_dns_rr_get_addr(rr, ARES_RR_A_ADDR);
        if (addr == nullptr) continue;
        uv_inet_ntop(AF_INET, addr, ip, sizeof(ip));
        if (ttls != nullptr) {
          ttls[count].ipaddr = *addr;
          ttls[count].ttl = ares_dns_rr_get_ttl(rr);
        }
        if (ret->Set(env->context(),
                     offset + count++,
                     OneByteString(env->isolate(), ip))
                .IsNothing()) {
          return Nothing<int>();
        }
      }
      if (naddrttls != nullptr) *naddrttls = count;
      break;
    }
    case ns_t_aaaa: {
      auto* ttls = static_cast<ares_addr6ttl*>(addrttls);
      const int max = naddrttls != nullptr ? *naddrttls : 0;
      char ip[INET6_ADDRSTRLEN];
      for (size_t i = 0; i < rr_count; i++) {
        const ares_dns_rr_t* rr =
            ares_dns_record_rr_get(dnsrec, ARES_SECTION_ANSWER, i);
        if (ares_dns_rr_get_type(rr) != ARES_REC_TYPE_AAAA) continue;
        if (ttls != nullptr && static_cast<int>(count) >= max) break;
        const struct ares_in6_addr* addr =
            ares_dns_rr_get_addr6(rr, ARES_RR_AAAA_ADDR);
        if (addr == nullptr) continue;
        uv_inet_ntop(AF_INET6, addr, ip, sizeof(ip));
        if (ttls != nullptr) {
          ttls[count].ip6addr = *addr;
          ttls[count].ttl = ares_dns_rr_get_ttl(rr);
        }
        if (ret->Set(env->context(),
                     offset + count++,
                     OneByteString(env->isolate(), ip))
                .IsNothing()) {
          return Nothing<int>();
        }
      }
      if (naddrttls != nullptr) *naddrttls = count;
      break;
    }
    default:
      UNREACHABLE("Bad NS type");
  }

  // Preserve the behavior of the previous ares_parse_*_reply() helpers, which
  // reported an empty answer section as ARES_ENODATA.
  if (count == 0) return Just<int>(ARES_ENODATA);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> ParseMxReply(Environment* env,
                        const unsigned char* buf,
                        int len,
                        Local<Array> ret,
                        bool need_type = false) {
  HandleScope handle_scope(env->isolate());

  auto tmpl = env->mx_record_template();
  if (tmpl.IsEmpty()) {
    static constexpr std::string_view names[] = {
        "exchange",
        "priority",
        "type",
    };
    tmpl = DictionaryTemplate::New(env->isolate(), names);
    env->set_mx_record_template(tmpl);
  }

  ares_dns_record_t* dnsrec = nullptr;
  int status = ares_dns_parse(buf, len, 0, &dnsrec);
  if (status != ARES_SUCCESS) {
    ares_dns_record_destroy(dnsrec);
    return Just<int>(status);
  }
  DeleteFnPtr<ares_dns_record_t, ares_dns_record_destroy> free_me(dnsrec);

  const size_t rr_count = ares_dns_record_rr_cnt(dnsrec, ARES_SECTION_ANSWER);
  uint32_t offset = ret->Length();
  uint32_t count = 0;

  MaybeLocal<Value> values[] = {
      Undefined(env->isolate()),  // exchange
      Undefined(env->isolate()),  // priority
      Undefined(env->isolate()),  // type
  };

  for (size_t i = 0; i < rr_count; i++) {
    const ares_dns_rr_t* rr =
        ares_dns_record_rr_get(dnsrec, ARES_SECTION_ANSWER, i);
    if (ares_dns_rr_get_type(rr) != ARES_REC_TYPE_MX) continue;

    const char* exchange = ares_dns_rr_get_str(rr, ARES_RR_MX_EXCHANGE);
    unsigned short priority =  // NOLINT(runtime/int)
        ares_dns_rr_get_u16(rr, ARES_RR_MX_PREFERENCE);
    values[0] = OneByteString(env->isolate(), exchange);
    values[1] = Integer::New(env->isolate(), priority);
    values[2] = env->dns_mx_string();
    Local<Value> record;
    if (!NewDictionaryInstance(env->context(), tmpl, values).ToLocal(&record) ||
        ret->Set(env->context(), offset + count++, record).IsNothing()) {
      return Nothing<int>();
    }
  }

  if (count == 0) return Just<int>(ARES_ENODATA);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> ParseCaaReply(Environment* env,
                         const unsigned char* buf,
                         int len,
                         Local<Array> ret,
                         bool need_type = false) {
  HandleScope handle_scope(env->isolate());

  auto tmpl = env->caa_record_template();
  if (tmpl.IsEmpty()) {
    static constexpr std::string_view names[] = {
        "critical",
        "type",
    };
    tmpl = DictionaryTemplate::New(env->isolate(), names);
    env->set_caa_record_template(tmpl);
  }

  ares_dns_record_t* dnsrec = nullptr;
  int status = ares_dns_parse(buf, len, 0, &dnsrec);
  if (status != ARES_SUCCESS) {
    ares_dns_record_destroy(dnsrec);
    return Just<int>(status);
  }
  DeleteFnPtr<ares_dns_record_t, ares_dns_record_destroy> free_me(dnsrec);

  MaybeLocal<Value> values[] = {
      Undefined(env->isolate()),  // critical
      Undefined(env->isolate()),  // type
  };

  const size_t rr_count = ares_dns_record_rr_cnt(dnsrec, ARES_SECTION_ANSWER);
  uint32_t offset = ret->Length();
  uint32_t count = 0;
  for (size_t i = 0; i < rr_count; i++) {
    const ares_dns_rr_t* rr =
        ares_dns_record_rr_get(dnsrec, ARES_SECTION_ANSWER, i);
    if (ares_dns_rr_get_type(rr) != ARES_REC_TYPE_CAA) continue;

    unsigned char critical = ares_dns_rr_get_u8(rr, ARES_RR_CAA_CRITICAL);
    values[0] = Integer::New(env->isolate(), critical);
    values[1] = env->dns_caa_string();
    Local<Object> caa_record;
    if (!NewDictionaryInstance(env->context(), tmpl, values)
             .ToLocal(&caa_record)) {
      return Nothing<int>();
    }

    const char* tag = ares_dns_rr_get_str(rr, ARES_RR_CAA_TAG);
    size_t value_len = 0;
    const unsigned char* value =
        ares_dns_rr_get_bin(rr, ARES_RR_CAA_VALUE, &value_len);

    // This additional property is not part of the template as it is
    // variable based on the record.
    if (caa_record
            ->Set(env->context(),
                  OneByteString(env->isolate(), tag),
                  OneByteString(env->isolate(), value, value_len))
            .IsNothing()) {
      return Nothing<int>();
    }

    if (ret->Set(env->context(), offset + count++, caa_record).IsNothing()) {
      return Nothing<int>();
    }
  }

  if (count == 0) return Just<int>(ARES_ENODATA);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> ParseTlsaReply(Environment* env,
                          unsigned char* buf,
                          int len,
                          Local<Array> ret) {
  EscapableHandleScope handle_scope(env->isolate());

  auto tmpl = env->tlsa_record_template();
  if (tmpl.IsEmpty()) {
    static constexpr std::string_view names[] = {
        "certUsage",
        "selector",
        "match",
        "data",
    };
    tmpl = DictionaryTemplate::New(env->isolate(), names);
    env->set_tlsa_record_template(tmpl);
  }

  ares_dns_record_t* dnsrec = nullptr;

  int status = ares_dns_parse(buf, len, 0, &dnsrec);
  if (status != ARES_SUCCESS) {
    ares_dns_record_destroy(dnsrec);
    return Just<int>(status);
  }

  DeleteFnPtr<ares_dns_record_t, ares_dns_record_destroy> free_me(dnsrec);

  uint32_t offset = ret->Length();
  size_t rr_count = ares_dns_record_rr_cnt(dnsrec, ARES_SECTION_ANSWER);

  MaybeLocal<Value> values[] = {
      Undefined(env->isolate()),  // certUsage
      Undefined(env->isolate()),  // selector
      Undefined(env->isolate()),  // match
      Undefined(env->isolate()),  // data
  };

  for (size_t i = 0; i < rr_count; i++) {
    const ares_dns_rr_t* rr =
        ares_dns_record_rr_get(dnsrec, ARES_SECTION_ANSWER, i);

    if (ares_dns_rr_get_type(rr) != ARES_REC_TYPE_TLSA) continue;

    unsigned char certusage = ares_dns_rr_get_u8(rr, ARES_RR_TLSA_CERT_USAGE);
    unsigned char selector = ares_dns_rr_get_u8(rr, ARES_RR_TLSA_SELECTOR);
    unsigned char match = ares_dns_rr_get_u8(rr, ARES_RR_TLSA_MATCH);
    size_t data_len;
    const unsigned char* data =
        ares_dns_rr_get_bin(rr, ARES_RR_TLSA_DATA, &data_len);
    if (!data || data_len == 0) continue;

    Local<ArrayBuffer> data_ab = ArrayBuffer::New(env->isolate(), data_len);
    memcpy(data_ab->Data(), data, data_len);

    values[0] = Integer::NewFromUnsigned(env->isolate(), certusage);
    values[1] = Integer::NewFromUnsigned(env->isolate(), selector);
    values[2] = Integer::NewFromUnsigned(env->isolate(), match);
    values[3] = data_ab;

    Local<Object> tlsa_rec;
    if (!NewDictionaryInstance(env->context(), tmpl, values)
             .ToLocal(&tlsa_rec) ||
        ret->Set(env->context(), offset + i, tlsa_rec).IsNothing()) {
      return Nothing<int>();
    }
  }

  return Just<int>(ARES_SUCCESS);
}

Maybe<int> ParseTxtReply(Environment* env,
                         const unsigned char* buf,
                         int len,
                         Local<Array> ret,
                         bool need_type = false) {
  HandleScope handle_scope(env->isolate());

  auto tmpl = env->txt_record_template();
  if (tmpl.IsEmpty()) {
    static constexpr std::string_view names[] = {
        "entries",
        "type",
    };
    tmpl = DictionaryTemplate::New(env->isolate(), names);
    env->set_txt_record_template(tmpl);
  }

  ares_dns_record_t* dnsrec = nullptr;
  int status = ares_dns_parse(buf, len, 0, &dnsrec);
  if (status != ARES_SUCCESS) {
    ares_dns_record_destroy(dnsrec);
    return Just<int>(status);
  }
  DeleteFnPtr<ares_dns_record_t, ares_dns_record_destroy> free_me(dnsrec);

  const size_t rr_count = ares_dns_record_rr_cnt(dnsrec, ARES_SECTION_ANSWER);
  uint32_t offset = ret->Length();
  uint32_t count = 0;

  MaybeLocal<Value> values[] = {
      Undefined(env->isolate()),  // entries
      Undefined(env->isolate()),  // type
  };

  for (size_t i = 0; i < rr_count; i++) {
    const ares_dns_rr_t* rr =
        ares_dns_record_rr_get(dnsrec, ARES_SECTION_ANSWER, i);
    if (ares_dns_rr_get_type(rr) != ARES_REC_TYPE_TXT) continue;

    // Each TXT record is a chunk consisting of one or more character-strings.
    LocalVector<Value> chunks(env->isolate());
    size_t str_count = ares_dns_rr_get_abin_cnt(rr, ARES_RR_TXT_DATA);
    chunks.reserve(str_count);
    for (size_t j = 0; j < str_count; j++) {
      size_t str_len = 0;
      const unsigned char* str =
          ares_dns_rr_get_abin(rr, ARES_RR_TXT_DATA, j, &str_len);
      chunks.push_back(OneByteString(env->isolate(), str, str_len));
    }

    Local<Array> txt_chunk =
        Array::New(env->isolate(), chunks.data(), chunks.size());
    if (need_type) {
      values[0] = txt_chunk;
      values[1] = env->dns_txt_string();
      Local<Object> elem;
      if (!NewDictionaryInstance(env->context(), tmpl, values).ToLocal(&elem) ||
          ret->Set(env->context(), offset + count++, elem).IsNothing()) {
        return Nothing<int>();
      }
    } else if (ret->Set(env->context(), offset + count++, txt_chunk)
                   .IsNothing()) {
      return Nothing<int>();
    }
  }

  if (count == 0) return Just<int>(ARES_ENODATA);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> ParseSrvReply(Environment* env,
                         const unsigned char* buf,
                         int len,
                         Local<Array> ret,
                         bool need_type = false) {
  HandleScope handle_scope(env->isolate());

  auto tmpl = env->srv_record_template();
  if (tmpl.IsEmpty()) {
    static constexpr std::string_view names[] = {
        "name",
        "port",
        "priority",
        "weight",
        "type",
    };
    tmpl = DictionaryTemplate::New(env->isolate(), names);
    env->set_srv_record_template(tmpl);
  }

  ares_dns_record_t* dnsrec = nullptr;
  int status = ares_dns_parse(buf, len, 0, &dnsrec);
  if (status != ARES_SUCCESS) {
    ares_dns_record_destroy(dnsrec);
    return Just<int>(status);
  }
  DeleteFnPtr<ares_dns_record_t, ares_dns_record_destroy> free_me(dnsrec);

  MaybeLocal<Value> values[] = {
      Undefined(env->isolate()),  // name
      Undefined(env->isolate()),  // port
      Undefined(env->isolate()),  // priority
      Undefined(env->isolate()),  // weight
      Undefined(env->isolate()),  // type
  };

  const size_t rr_count = ares_dns_record_rr_cnt(dnsrec, ARES_SECTION_ANSWER);
  uint32_t offset = ret->Length();
  uint32_t count = 0;
  for (size_t i = 0; i < rr_count; i++) {
    const ares_dns_rr_t* rr =
        ares_dns_record_rr_get(dnsrec, ARES_SECTION_ANSWER, i);
    if (ares_dns_rr_get_type(rr) != ARES_REC_TYPE_SRV) continue;

    const char* target = ares_dns_rr_get_str(rr, ARES_RR_SRV_TARGET);
    values[0] = OneByteString(env->isolate(), target);
    values[1] =
        Integer::New(env->isolate(), ares_dns_rr_get_u16(rr, ARES_RR_SRV_PORT));
    values[2] = Integer::New(env->isolate(),
                             ares_dns_rr_get_u16(rr, ARES_RR_SRV_PRIORITY));
    values[3] = Integer::New(env->isolate(),
                             ares_dns_rr_get_u16(rr, ARES_RR_SRV_WEIGHT));
    values[4] = env->dns_srv_string();

    Local<Object> srv_record;
    if (!NewDictionaryInstance(env->context(), tmpl, values)
             .ToLocal(&srv_record) ||
        ret->Set(env->context(), offset + count++, srv_record).IsNothing()) {
      return Nothing<int>();
    }
  }

  if (count == 0) return Just<int>(ARES_ENODATA);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> ParseNaptrReply(Environment* env,
                           const unsigned char* buf,
                           int len,
                           Local<Array> ret,
                           bool need_type = false) {
  HandleScope handle_scope(env->isolate());

  auto tmpl = env->naptr_record_template();
  if (tmpl.IsEmpty()) {
    static constexpr std::string_view names[] = {
        "flags",
        "service",
        "regexp",
        "replacement",
        "order",
        "preference",
        "type",
    };
    tmpl = DictionaryTemplate::New(env->isolate(), names);
    env->set_naptr_record_template(tmpl);
  }

  ares_dns_record_t* dnsrec = nullptr;
  int status = ares_dns_parse(buf, len, 0, &dnsrec);
  if (status != ARES_SUCCESS) {
    ares_dns_record_destroy(dnsrec);
    return Just<int>(status);
  }
  DeleteFnPtr<ares_dns_record_t, ares_dns_record_destroy> free_me(dnsrec);

  MaybeLocal<Value> values[] = {
      Undefined(env->isolate()),  // flags
      Undefined(env->isolate()),  // service
      Undefined(env->isolate()),  // regexp
      Undefined(env->isolate()),  // replacement
      Undefined(env->isolate()),  // order
      Undefined(env->isolate()),  // preference
      Undefined(env->isolate()),  // type
  };

  const size_t rr_count = ares_dns_record_rr_cnt(dnsrec, ARES_SECTION_ANSWER);
  uint32_t offset = ret->Length();
  uint32_t count = 0;
  for (size_t i = 0; i < rr_count; i++) {
    const ares_dns_rr_t* rr =
        ares_dns_record_rr_get(dnsrec, ARES_SECTION_ANSWER, i);
    if (ares_dns_rr_get_type(rr) != ARES_REC_TYPE_NAPTR) continue;

    values[0] = OneByteString(env->isolate(),
                              ares_dns_rr_get_str(rr, ARES_RR_NAPTR_FLAGS));
    values[1] = OneByteString(env->isolate(),
                              ares_dns_rr_get_str(rr, ARES_RR_NAPTR_SERVICES));
    values[2] = OneByteString(env->isolate(),
                              ares_dns_rr_get_str(rr, ARES_RR_NAPTR_REGEXP));
    values[3] = OneByteString(
        env->isolate(), ares_dns_rr_get_str(rr, ARES_RR_NAPTR_REPLACEMENT));
    values[4] = Integer::New(env->isolate(),
                             ares_dns_rr_get_u16(rr, ARES_RR_NAPTR_ORDER));
    values[5] = Integer::New(env->isolate(),
                             ares_dns_rr_get_u16(rr, ARES_RR_NAPTR_PREFERENCE));
    if (need_type) {
      values[6] = env->dns_naptr_string();
    }

    Local<Object> naptr_record;
    if (!NewDictionaryInstance(env->context(), tmpl, values)
             .ToLocal(&naptr_record) ||
        ret->Set(env->context(), offset + count++, naptr_record).IsNothing()) {
      return Nothing<int>();
    }
  }

  if (count == 0) return Just<int>(ARES_ENODATA);
  return Just<int>(ARES_SUCCESS);
}

Local<DictionaryTemplate> getSoaRecordTemplate(Environment* env) {
  auto tmpl = env->soa_record_template();
  if (tmpl.IsEmpty()) {
    static constexpr std::string_view names[] = {
        "nsname",
        "hostmaster",
        "serial",
        "refresh",
        "retry",
        "expire",
        "minttl",
        "type",
    };
    tmpl = DictionaryTemplate::New(env->isolate(), names);
    env->set_soa_record_template(tmpl);
  }
  return tmpl;
}

Maybe<int> ParseSoaReply(Environment* env,
                         unsigned char* buf,
                         int len,
                         Local<Object>* ret) {
  EscapableHandleScope handle_scope(env->isolate());

  auto tmpl = getSoaRecordTemplate(env);

  // Manage memory using standardard smart pointer std::unique_tr
  struct AresDeleter {
    void operator()(char* ptr) const noexcept { ares_free_string(ptr); }
  };
  using ares_unique_ptr = std::unique_ptr<char[], AresDeleter>;

  // Can't use ares_parse_soa_reply() here which can only parse single record
  const unsigned int ancount = cares_get_16bit(buf + 6);
  unsigned char* ptr = buf + NS_HFIXEDSZ;
  char* name_temp = nullptr;
  long temp_len;  // NOLINT(runtime/int)
  int status = ares_expand_name(ptr, buf, len, &name_temp, &temp_len);
  if (status != ARES_SUCCESS) {
    // returns EBADRESP in case of invalid input
    return Just<int>(status == ARES_EBADNAME ? ARES_EBADRESP : status);
  }

  const ares_unique_ptr name(name_temp);

  if (ptr + temp_len + NS_QFIXEDSZ > buf + len) {
    return Just<int>(ARES_EBADRESP);
  }
  ptr += temp_len + NS_QFIXEDSZ;

  MaybeLocal<Value> values[] = {
      Undefined(env->isolate()),  // nsname
      Undefined(env->isolate()),  // hostmaster
      Undefined(env->isolate()),  // serial
      Undefined(env->isolate()),  // refresh
      Undefined(env->isolate()),  // retry
      Undefined(env->isolate()),  // expire
      Undefined(env->isolate()),  // minttl
      Undefined(env->isolate()),  // type
  };

  for (unsigned int i = 0; i < ancount; i++) {
    char* rr_name_temp = nullptr;
    long rr_temp_len;  // NOLINT(runtime/int)
    int status2 = ares_expand_name(ptr, buf, len, &rr_name_temp, &rr_temp_len);

    if (status2 != ARES_SUCCESS)
      return Just<int>(status2 == ARES_EBADNAME ? ARES_EBADRESP : status2);

    const ares_unique_ptr rr_name(rr_name_temp);

    ptr += rr_temp_len;
    if (ptr + NS_RRFIXEDSZ > buf + len) {
      return Just<int>(ARES_EBADRESP);
    }

    const int rr_type = cares_get_16bit(ptr);
    const int rr_len = cares_get_16bit(ptr + 8);
    ptr += NS_RRFIXEDSZ;

    // only need SOA
    if (rr_type == ns_t_soa) {
      char* nsname_temp = nullptr;
      long nsname_temp_len;  // NOLINT(runtime/int)

      int status3 = ares_expand_name(ptr, buf, len,
                                     &nsname_temp,
                                     &nsname_temp_len);
      if (status3 != ARES_SUCCESS) {
        return Just<int>(status3 == ARES_EBADNAME ? ARES_EBADRESP : status3);
      }
      const ares_unique_ptr nsname(nsname_temp);
      ptr += nsname_temp_len;

      char* hostmaster_temp = nullptr;
      long hostmaster_temp_len;  // NOLINT(runtime/int)
      int status4 = ares_expand_name(ptr, buf, len,
                                     &hostmaster_temp,
                                     &hostmaster_temp_len);
      if (status4 != ARES_SUCCESS) {
        return Just<int>(status4 == ARES_EBADNAME ? ARES_EBADRESP : status4);
      }
      const ares_unique_ptr hostmaster(hostmaster_temp);
      ptr += hostmaster_temp_len;

      if (ptr + 5 * 4 > buf + len) {
        return Just<int>(ARES_EBADRESP);
      }

      const unsigned int serial = nbytes::ReadUint32BE(ptr + 0 * 4);
      const unsigned int refresh = nbytes::ReadUint32BE(ptr + 1 * 4);
      const unsigned int retry = nbytes::ReadUint32BE(ptr + 2 * 4);
      const unsigned int expire = nbytes::ReadUint32BE(ptr + 3 * 4);
      const unsigned int minttl = nbytes::ReadUint32BE(ptr + 4 * 4);

      values[0] = OneByteString(env->isolate(), nsname.get());
      values[1] = OneByteString(env->isolate(), hostmaster.get());
      values[2] = Integer::NewFromUnsigned(env->isolate(), serial);
      values[3] = Integer::New(env->isolate(), refresh);
      values[4] = Integer::New(env->isolate(), retry);
      values[5] = Integer::New(env->isolate(), expire);
      values[6] = Integer::NewFromUnsigned(env->isolate(), minttl);
      values[7] = env->dns_soa_string();
      Local<Object> soa_record;
      if (!NewDictionaryInstance(env->context(), tmpl, values)
               .ToLocal(&soa_record)) {
        return Nothing<int>();
      }

      *ret = handle_scope.Escape(soa_record);
      break;
    }

    ptr += rr_len;
  }

  return Just<int>(ARES_SUCCESS);
}
}  // anonymous namespace

ChannelWrap::ChannelWrap(Environment* env,
                         Local<Object> object,
                         int timeout,
                         int tries,
                         int max_timeout)
    : AsyncWrap(env, object, PROVIDER_DNSCHANNEL),
      timeout_(timeout),
      tries_(tries),
      max_timeout_(max_timeout) {
  MakeWeak();

  Setup();
}

void ChannelWrap::MemoryInfo(MemoryTracker* tracker) const {
  if (timer_handle_ != nullptr)
    tracker->TrackField("timer_handle", *timer_handle_);
  tracker->TrackField("task_list", task_list_, "NodeAresTask::List");
}

void ChannelWrap::New(const FunctionCallbackInfo<Value>& args) {
  CHECK(args.IsConstructCall());
  CHECK_EQ(args.Length(), 3);
  CHECK(args[0]->IsInt32());
  CHECK(args[1]->IsInt32());
  CHECK(args[2]->IsInt32());
  const int timeout = args[0].As<Int32>()->Value();
  const int tries = args[1].As<Int32>()->Value();
  const int max_timeout = args[2].As<Int32>()->Value();
  Environment* env = Environment::GetCurrent(args);
  new ChannelWrap(env, args.This(), timeout, tries, max_timeout);
}

GetAddrInfoReqWrap::GetAddrInfoReqWrap(Environment* env,
                                       Local<Object> req_wrap_obj,
                                       uint8_t order)
    : ReqWrap(env, req_wrap_obj, AsyncWrap::PROVIDER_GETADDRINFOREQWRAP),
      order_(order) {}

GetNameInfoReqWrap::GetNameInfoReqWrap(
    Environment* env,
    Local<Object> req_wrap_obj)
    : ReqWrap(env, req_wrap_obj, AsyncWrap::PROVIDER_GETNAMEINFOREQWRAP) {}

/* This is called once per second by loop->timer. It is used to constantly */
/* call back into c-ares for possibly processing timeouts. */
void ChannelWrap::AresTimeout(uv_timer_t* handle) {
  ChannelWrap* channel = static_cast<ChannelWrap*>(handle->data);
  CHECK_EQ(channel->timer_handle(), handle);
  ares_process_fd(channel->cares_channel(), ARES_SOCKET_BAD, ARES_SOCKET_BAD);
}


void NodeAresTask::MemoryInfo(MemoryTracker* tracker) const {
  tracker->TrackField("channel", channel);
}

/* Allocates and returns a new NodeAresTask */
NodeAresTask* NodeAresTask::Create(ChannelWrap* channel, ares_socket_t sock) {
  auto task = std::make_unique<NodeAresTask>();

  task->channel = channel;
  task->sock = sock;

  if (uv_poll_init_socket(channel->env()->event_loop(),
                          &task->poll_watcher, sock) < 0) {
    /* This should never happen. */
    return nullptr;
  }

  return task.release();
}

void ChannelWrap::Setup() {
  struct ares_options options;
  memset(&options, 0, sizeof(options));
  options.flags = ARES_FLAG_NOCHECKRESP;
  options.sock_state_cb = ares_sockstate_cb;
  options.sock_state_cb_data = this;
  options.timeout = timeout_;
  options.tries = tries_;
  options.qcache_max_ttl = 0;
  // Resolver APIs always perform DNS queries and must not consult hosts files.
  options.lookups = const_cast<char*>("b");

  int r;
  if (!library_inited_) {
    Mutex::ScopedLock lock(ares_library_mutex);
    // Multiple calls to ares_library_init() increase a reference counter,
    // so this is a no-op except for the first call to it.
    r = ares_library_init(ARES_LIB_INIT_ALL);
    if (r != ARES_SUCCESS)
      return env()->ThrowError(ToErrorCodeString(r));
  }

  /* We do the call to ares_init_option for caller. */
  int optmask = ARES_OPT_FLAGS | ARES_OPT_TIMEOUTMS | ARES_OPT_SOCK_STATE_CB |
                ARES_OPT_TRIES | ARES_OPT_QUERY_CACHE | ARES_OPT_LOOKUPS;

  if (max_timeout_ > 0) {
    options.maxtimeout = max_timeout_;
    optmask |= ARES_OPT_MAXTIMEOUTMS;
  }

  r = ares_init_options(&channel_, &options, optmask);

  if (r != ARES_SUCCESS) {
    Mutex::ScopedLock lock(ares_library_mutex);
    ares_library_cleanup();
    return env()->ThrowError(ToErrorCodeString(r));
  }

  library_inited_ = true;
}

void ChannelWrap::StartTimer() {
  if (timer_handle_ == nullptr) {
    timer_handle_ = new uv_timer_t();
    timer_handle_->data = static_cast<void*>(this);
    uv_timer_init(env()->event_loop(), timer_handle_);
  } else if (uv_is_active(reinterpret_cast<uv_handle_t*>(timer_handle_))) {
    return;
  }
  int timeout = timeout_;
  if (timeout <= 0 || timeout > 1000) timeout = 1000;
  uv_timer_start(timer_handle_, AresTimeout, timeout, timeout);
}

void ChannelWrap::CloseTimer() {
  if (timer_handle_ == nullptr)
    return;

  env()->CloseHandle(timer_handle_, [](uv_timer_t* handle) { delete handle; });
  timer_handle_ = nullptr;
}

ChannelWrap::~ChannelWrap() {
  ares_destroy(channel_);

  if (library_inited_) {
    Mutex::ScopedLock lock(ares_library_mutex);
    // This decreases the reference counter increased by ares_library_init().
    ares_library_cleanup();
  }

  CloseTimer();
}


void ChannelWrap::ModifyActivityQueryCount(int count) {
  active_query_count_ += count;
  CHECK_GE(active_query_count_, 0);
}


/**
 * This function is to check whether current servers are fallback servers
 * when cares initialized.
 *
 * The fallback servers of cares is [ "127.0.0.1" ] with no user additional
 * setting.
 */
void ChannelWrap::EnsureServers() {
  /* if last query is OK or servers are set by user self, do not check */
  if (query_last_ok_ || !is_servers_default_) {
    return;
  }

  char* csv = ares_get_servers_csv(channel_);
  auto cleanup = OnScopeLeave([&]() { ares_free_string(csv); });
  std::vector<std::pair<std::string, int>> servers = ParseServersCsv(csv);

  /* if no server, ignore */
  if (servers.empty()) return;

  /* if multi-servers, mark as non-default and ignore */
  if (servers.size() > 1) {
    is_servers_default_ = false;
    return;
  }

  /* Check if the only server is a loopback address (IPv4 127.0.0.1 or IPv6
   * ::1). Newer c-ares versions may set tcp_port/udp_port to 53 instead of 0,
   * so we no longer check port values. */
  const std::string& host = servers[0].first;
  if (host != "127.0.0.1" && host != "::1") {
    is_servers_default_ = false;
    return;
  }

  /* destroy channel and reset channel */
  ares_destroy(channel_);

  CloseTimer();
  Setup();
}

int AnyTraits::Send(QueryWrap<AnyTraits>* wrap, const char* name) {
  wrap->AresQuery(name, ARES_CLASS_IN, ARES_REC_TYPE_ANY);
  return ARES_SUCCESS;
}

int ATraits::Send(QueryWrap<ATraits>* wrap, const char* name) {
  wrap->AresQuery(name, ARES_CLASS_IN, ARES_REC_TYPE_A);
  return ARES_SUCCESS;
}

int AaaaTraits::Send(QueryWrap<AaaaTraits>* wrap, const char* name) {
  wrap->AresQuery(name, ARES_CLASS_IN, ARES_REC_TYPE_AAAA);
  return ARES_SUCCESS;
}

int CaaTraits::Send(QueryWrap<CaaTraits>* wrap, const char* name) {
  wrap->AresQuery(name, ARES_CLASS_IN, ARES_REC_TYPE_CAA);
  return ARES_SUCCESS;
}

int CnameTraits::Send(QueryWrap<CnameTraits>* wrap, const char* name) {
  wrap->AresQuery(name, ARES_CLASS_IN, ARES_REC_TYPE_CNAME);
  return ARES_SUCCESS;
}

int MxTraits::Send(QueryWrap<MxTraits>* wrap, const char* name) {
  wrap->AresQuery(name, ARES_CLASS_IN, ARES_REC_TYPE_MX);
  return ARES_SUCCESS;
}

int NsTraits::Send(QueryWrap<NsTraits>* wrap, const char* name) {
  wrap->AresQuery(name, ARES_CLASS_IN, ARES_REC_TYPE_NS);
  return ARES_SUCCESS;
}

int TlsaTraits::Send(QueryWrap<TlsaTraits>* wrap, const char* name) {
  wrap->AresQuery(name, ARES_CLASS_IN, ARES_REC_TYPE_TLSA);
  return ARES_SUCCESS;
}

int TxtTraits::Send(QueryWrap<TxtTraits>* wrap, const char* name) {
  wrap->AresQuery(name, ARES_CLASS_IN, ARES_REC_TYPE_TXT);
  return ARES_SUCCESS;
}

int SrvTraits::Send(QueryWrap<SrvTraits>* wrap, const char* name) {
  wrap->AresQuery(name, ARES_CLASS_IN, ARES_REC_TYPE_SRV);
  return ARES_SUCCESS;
}

int PtrTraits::Send(QueryWrap<PtrTraits>* wrap, const char* name) {
  wrap->AresQuery(name, ARES_CLASS_IN, ARES_REC_TYPE_PTR);
  return ARES_SUCCESS;
}

int NaptrTraits::Send(QueryWrap<NaptrTraits>* wrap, const char* name) {
  wrap->AresQuery(name, ARES_CLASS_IN, ARES_REC_TYPE_NAPTR);
  return ARES_SUCCESS;
}

int SoaTraits::Send(QueryWrap<SoaTraits>* wrap, const char* name) {
  wrap->AresQuery(name, ARES_CLASS_IN, ARES_REC_TYPE_SOA);
  return ARES_SUCCESS;
}

Maybe<int> AnyTraits::Parse(QueryAnyWrap* wrap,
                            const std::unique_ptr<ResponseData>& response) {
  if (response->is_host) [[unlikely]] {
    return Just<int>(ARES_EBADRESP);
  }

  unsigned char* buf = response->buf.data;
  int len = response->buf.size;

  Environment* env = wrap->env();
  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());

  Local<Array> ret = Array::New(env->isolate());
  int type, status, old_count;

  /* Parse A records or CNAME records */
  std::vector<ares_addrttl> addrttls =
      MakeAddrTTLBuffer<ares_addrttl>(buf, len);
  int naddrttls = static_cast<int>(addrttls.size());

  type = ns_t_cname_or_a;
  if (!ParseGeneralReply(env, buf, len, &type, ret, addrttls.data(), &naddrttls)
           .To(&status)) {
    return Nothing<int>();
  }
  uint32_t a_count = ret->Length();
  if (status != ARES_SUCCESS && status != ARES_ENODATA) {
    return Just<int>(status);
  }

  if (type == ns_t_a) {
    CHECK_EQ(static_cast<uint32_t>(naddrttls), a_count);

    auto tmpl = env->a_record_template();
    if (tmpl.IsEmpty()) {
      static constexpr std::string_view names[] = {
          "address",
          "ttl",
          "type",
      };
      tmpl = DictionaryTemplate::New(env->isolate(), names);
      env->set_a_record_template(tmpl);
    }
    MaybeLocal<Value> values[] = {
        Undefined(env->isolate()),  // address
        Undefined(env->isolate()),  // ttl
        Undefined(env->isolate()),  // type
    };

    for (uint32_t i = 0; i < a_count; i++) {
      Local<Value> address;
      if (!ret->Get(env->context(), i).ToLocal(&address)) {
        return Nothing<int>();
      }
      values[0] = address;
      values[1] = Integer::NewFromUnsigned(env->isolate(), addrttls[i].ttl);
      values[2] = env->dns_a_string();

      Local<Object> obj;
      if (!NewDictionaryInstance(env->context(), tmpl, values).ToLocal(&obj) ||
          ret->Set(env->context(), i, obj).IsNothing()) {
        return Nothing<int>();
      }
    }
  } else {
    auto tmpl = env->cname_record_template();
    if (tmpl.IsEmpty()) {
      static constexpr std::string_view names[] = {
          "value",
          "type",
      };
      tmpl = DictionaryTemplate::New(env->isolate(), names);
      env->set_cname_record_template(tmpl);
    }
    MaybeLocal<Value> values[] = {
        Undefined(env->isolate()),  // value
        Undefined(env->isolate()),  // type
    };
    for (uint32_t i = 0; i < a_count; i++) {
      Local<Value> value;
      if (!ret->Get(env->context(), i).ToLocal(&value)) {
        return Nothing<int>();
      }
      values[0] = value;
      values[1] = env->dns_cname_string();
      Local<Object> obj;
      if (!NewDictionaryInstance(env->context(), tmpl, values).ToLocal(&obj) ||
          ret->Set(env->context(), i, obj).IsNothing()) {
        return Nothing<int>();
      }
    }
  }

  /* Parse AAAA records */
  std::vector<ares_addr6ttl> addr6ttls =
      MakeAddrTTLBuffer<ares_addr6ttl>(buf, len);
  int naddr6ttls = static_cast<int>(addr6ttls.size());

  type = ns_t_aaaa;
  if (!ParseGeneralReply(
           env, buf, len, &type, ret, addr6ttls.data(), &naddr6ttls)
           .To(&status)) {
    return Nothing<int>();
  }
  uint32_t aaaa_count = ret->Length() - a_count;
  if (status != ARES_SUCCESS && status != ARES_ENODATA)
    return Just<int>(status);

  CHECK_EQ(aaaa_count, static_cast<uint32_t>(naddr6ttls));
  CHECK_EQ(ret->Length(), a_count + aaaa_count);

  auto tmpl = env->aaaa_record_template();
  if (tmpl.IsEmpty()) {
    static constexpr std::string_view names[] = {
        "address",
        "ttl",
        "type",
    };
    tmpl = DictionaryTemplate::New(env->isolate(), names);
    env->set_aaaa_record_template(tmpl);
  }

  MaybeLocal<Value> values[] = {
      Undefined(env->isolate()),  // address
      Undefined(env->isolate()),  // ttl
      Undefined(env->isolate()),  // type
  };

  for (uint32_t i = a_count; i < ret->Length(); i++) {
    Local<Value> address;
    if (!ret->Get(env->context(), i).ToLocal(&address)) {
      return Nothing<int>();
    }
    values[0] = address;
    values[1] =
        Integer::NewFromUnsigned(env->isolate(), addr6ttls[i - a_count].ttl);
    values[2] = env->dns_aaaa_string();
    Local<Object> obj;
    if (!NewDictionaryInstance(env->context(), tmpl, values).ToLocal(&obj) ||
        ret->Set(env->context(), i, obj).IsNothing()) {
      return Nothing<int>();
    }
  }

  /* Parse MX records */
  if (!ParseMxReply(env, buf, len, ret, true).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS && status != ARES_ENODATA) {
    return Just<int>(status);
  }

  /* Parse NS records */
  type = ns_t_ns;
  old_count = ret->Length();
  if (!ParseGeneralReply(env, buf, len, &type, ret).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS && status != ARES_ENODATA) {
    return Just<int>(status);
  }

  auto dns_ns_tmpl = env->dns_ns_record_template();
  if (dns_ns_tmpl.IsEmpty()) {
    static constexpr std::string_view names[] = {
        "value",
        "type",
    };
    dns_ns_tmpl = DictionaryTemplate::New(env->isolate(), names);
    env->set_dns_ns_record_template(dns_ns_tmpl);
  }

  MaybeLocal<Value> values_ns[] = {
      Undefined(env->isolate()),  // value
      Undefined(env->isolate()),  // type
  };

  for (uint32_t i = old_count; i < ret->Length(); i++) {
    Local<Value> value;
    if (!ret->Get(env->context(), i).ToLocal(&value)) {
      return Nothing<int>();
    }
    values_ns[0] = value;
    values_ns[1] = env->dns_ns_string();
    Local<Object> obj;
    if (!NewDictionaryInstance(env->context(), dns_ns_tmpl, values_ns)
             .ToLocal(&obj) ||
        ret->Set(env->context(), i, obj).IsNothing()) {
      return Nothing<int>();
    }
  }

  /* Parse TXT records */
  if (!ParseTxtReply(env, buf, len, ret, true).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS && status != ARES_ENODATA) {
    return Just<int>(status);
  }

  /* Parse SRV records */
  if (!ParseSrvReply(env, buf, len, ret, true).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS && status != ARES_ENODATA) {
    return Just<int>(status);
  }

  /* Parse PTR records */
  type = ns_t_ptr;
  old_count = ret->Length();
  if (!ParseGeneralReply(env, buf, len, &type, ret).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS && status != ARES_ENODATA)
    return Just<int>(status);

  for (uint32_t i = old_count; i < ret->Length(); i++) {
    Local<Value> value;
    if (!ret->Get(env->context(), i).ToLocal(&value)) {
      return Nothing<int>();
    }
    values_ns[0] = value;
    values_ns[1] = env->dns_ptr_string();
    Local<Object> obj;
    if (!NewDictionaryInstance(env->context(), dns_ns_tmpl, values_ns)
             .ToLocal(&obj) ||
        ret->Set(env->context(), i, obj).IsNothing()) {
      return Nothing<int>();
    }
  }

  /* Parse NAPTR records */
  if (!ParseNaptrReply(env, buf, len, ret, true).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS && status != ARES_ENODATA) {
    return Just<int>(status);
  }

  /* Parse SOA records */
  Local<Object> soa_record = Local<Object>();
  if (!ParseSoaReply(env, buf, len, &soa_record).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS && status != ARES_ENODATA) {
    return Just<int>(status);
  }

  if (!soa_record.IsEmpty()) {
    if (ret->Set(env->context(), ret->Length(), soa_record).IsNothing()) {
      return Just<int>(ARES_ENOMEM);
    }
  }

  /* Parse TLSA records */
  if (!ParseTlsaReply(env, buf, len, ret).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS && status != ARES_ENODATA) {
    return Just<int>(status);
  }

  /* Parse CAA records */
  if (!ParseCaaReply(env, buf, len, ret, true).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS && status != ARES_ENODATA) {
    return Just<int>(status);
  }

  wrap->CallOnComplete(ret);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> ATraits::Parse(QueryAWrap* wrap,
                          const std::unique_ptr<ResponseData>& response) {
  if (response->is_host) [[unlikely]] {
    return Just<int>(ARES_EBADRESP);
  }

  unsigned char* buf = response->buf.data;
  int len = response->buf.size;

  Environment* env = wrap->env();
  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());

  std::vector<ares_addrttl> addrttls =
      MakeAddrTTLBuffer<ares_addrttl>(buf, len);
  int naddrttls = static_cast<int>(addrttls.size()), status;
  Local<Array> ret = Array::New(env->isolate());

  int type = ns_t_a;
  if (!ParseGeneralReply(env, buf, len, &type, ret, addrttls.data(), &naddrttls)
           .To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS) {
    return Just<int>(status);
  }

  Local<Array> ttls =
      AddrTTLToArray<ares_addrttl>(env, addrttls.data(), naddrttls);

  wrap->CallOnComplete(ret, ttls);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> AaaaTraits::Parse(QueryAaaaWrap* wrap,
                             const std::unique_ptr<ResponseData>& response) {
  if (response->is_host) [[unlikely]] {
    return Just<int>(ARES_EBADRESP);
  }

  unsigned char* buf = response->buf.data;
  int len = response->buf.size;

  Environment* env = wrap->env();
  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());

  std::vector<ares_addr6ttl> addrttls =
      MakeAddrTTLBuffer<ares_addr6ttl>(buf, len);
  int naddrttls = static_cast<int>(addrttls.size()), status;
  Local<Array> ret = Array::New(env->isolate());

  int type = ns_t_aaaa;
  if (!ParseGeneralReply(env, buf, len, &type, ret, addrttls.data(), &naddrttls)
           .To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS) {
    return Just<int>(status);
  }

  Local<Array> ttls =
      AddrTTLToArray<ares_addr6ttl>(env, addrttls.data(), naddrttls);

  wrap->CallOnComplete(ret, ttls);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> CaaTraits::Parse(QueryCaaWrap* wrap,
                            const std::unique_ptr<ResponseData>& response) {
  if (response->is_host) [[unlikely]] {
    return Just<int>(ARES_EBADRESP);
  }

  unsigned char* buf = response->buf.data;
  int len = response->buf.size;

  Environment* env = wrap->env();
  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());

  Local<Array> ret = Array::New(env->isolate());
  int status;
  if (!ParseCaaReply(env, buf, len, ret).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS) {
    return Just<int>(status);
  }

  wrap->CallOnComplete(ret);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> CnameTraits::Parse(QueryCnameWrap* wrap,
                              const std::unique_ptr<ResponseData>& response) {
  if (response->is_host) [[unlikely]] {
    return Just<int>(ARES_EBADRESP);
  }

  unsigned char* buf = response->buf.data;
  int len = response->buf.size;

  Environment* env = wrap->env();
  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());

  Local<Array> ret = Array::New(env->isolate());
  int type = ns_t_cname;
  int status;
  if (!ParseGeneralReply(env, buf, len, &type, ret).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS) {
    return Just<int>(status);
  }

  wrap->CallOnComplete(ret);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> MxTraits::Parse(QueryMxWrap* wrap,
                           const std::unique_ptr<ResponseData>& response) {
  if (response->is_host) [[unlikely]] {
    return Just<int>(ARES_EBADRESP);
  }

  unsigned char* buf = response->buf.data;
  int len = response->buf.size;

  Environment* env = wrap->env();
  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());

  Local<Array> mx_records = Array::New(env->isolate());
  int status;
  if (!ParseMxReply(env, buf, len, mx_records).To(&status)) {
    return Nothing<int>();
  }

  if (status != ARES_SUCCESS) {
    return Just<int>(status);
  }

  wrap->CallOnComplete(mx_records);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> NsTraits::Parse(QueryNsWrap* wrap,
                           const std::unique_ptr<ResponseData>& response) {
  if (response->is_host) [[unlikely]] {
    return Just<int>(ARES_EBADRESP);
  }

  unsigned char* buf = response->buf.data;
  int len = response->buf.size;

  Environment* env = wrap->env();
  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());

  int type = ns_t_ns;
  Local<Array> names = Array::New(env->isolate());
  int status;
  if (!ParseGeneralReply(env, buf, len, &type, names).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS) {
    return Just<int>(status);
  }

  wrap->CallOnComplete(names);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> TlsaTraits::Parse(QueryTlsaWrap* wrap,
                             const std::unique_ptr<ResponseData>& response) {
  if (response->is_host) [[unlikely]] {
    return Just<int>(ARES_EBADRESP);
  }

  unsigned char* buf = response->buf.data;
  int len = response->buf.size;

  Environment* env = wrap->env();
  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());

  Local<Array> tlsa_records = Array::New(env->isolate());
  int status;
  if (!ParseTlsaReply(env, buf, len, tlsa_records).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS) {
    return Just<int>(status);
  }

  wrap->CallOnComplete(tlsa_records);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> TxtTraits::Parse(QueryTxtWrap* wrap,
                            const std::unique_ptr<ResponseData>& response) {
  if (response->is_host) [[unlikely]] {
    return Just<int>(ARES_EBADRESP);
  }

  unsigned char* buf = response->buf.data;
  int len = response->buf.size;

  Environment* env = wrap->env();
  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());

  Local<Array> txt_records = Array::New(env->isolate());
  int status;
  if (!ParseTxtReply(env, buf, len, txt_records).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS) {
    return Just<int>(status);
  }

  wrap->CallOnComplete(txt_records);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> SrvTraits::Parse(QuerySrvWrap* wrap,
                            const std::unique_ptr<ResponseData>& response) {
  if (response->is_host) [[unlikely]] {
    return Just<int>(ARES_EBADRESP);
  }

  unsigned char* buf = response->buf.data;
  int len = response->buf.size;

  Environment* env = wrap->env();
  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());

  Local<Array> srv_records = Array::New(env->isolate());
  int status;
  if (!ParseSrvReply(env, buf, len, srv_records).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS) return Just<int>(status);

  wrap->CallOnComplete(srv_records);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> PtrTraits::Parse(QueryPtrWrap* wrap,
                            const std::unique_ptr<ResponseData>& response) {
  if (response->is_host) [[unlikely]] {
    return Just<int>(ARES_EBADRESP);
  }
  unsigned char* buf = response->buf.data;
  int len = response->buf.size;

  Environment* env = wrap->env();
  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());

  int type = ns_t_ptr;
  Local<Array> aliases = Array::New(env->isolate());

  int status;
  if (!ParseGeneralReply(env, buf, len, &type, aliases).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS) {
    return Just<int>(status);
  }

  wrap->CallOnComplete(aliases);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> NaptrTraits::Parse(QueryNaptrWrap* wrap,
                              const std::unique_ptr<ResponseData>& response) {
  if (response->is_host) [[unlikely]] {
    return Just<int>(ARES_EBADRESP);
  }
  unsigned char* buf = response->buf.data;
  int len = response->buf.size;

  Environment* env = wrap->env();
  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());

  Local<Array> naptr_records = Array::New(env->isolate());
  int status;
  if (!ParseNaptrReply(env, buf, len, naptr_records).To(&status)) {
    return Nothing<int>();
  }
  if (status != ARES_SUCCESS) {
    return Just<int>(status);
  }

  wrap->CallOnComplete(naptr_records);
  return Just<int>(ARES_SUCCESS);
}

Maybe<int> SoaTraits::Parse(QuerySoaWrap* wrap,
                            const std::unique_ptr<ResponseData>& response) {
  if (response->is_host) [[unlikely]] {
    return Just<int>(ARES_EBADRESP);
  }
  unsigned char* buf = response->buf.data;
  int len = response->buf.size;

  Environment* env = wrap->env();
  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());

  auto tmpl = getSoaRecordTemplate(env);
  MaybeLocal<Value> values[] = {
      Undefined(env->isolate()),  // nsname
      Undefined(env->isolate()),  // hostmaster
      Undefined(env->isolate()),  // serial
      Undefined(env->isolate()),  // refresh
      Undefined(env->isolate()),  // retry
      Undefined(env->isolate()),  // expire
      Undefined(env->isolate()),  // minttl
      Undefined(env->isolate()),  // type
  };

  ares_dns_record_t* dnsrec = nullptr;
  int status = ares_dns_parse(buf, len, 0, &dnsrec);
  if (status != ARES_SUCCESS) {
    ares_dns_record_destroy(dnsrec);
    return Just<int>(status);
  }
  DeleteFnPtr<ares_dns_record_t, ares_dns_record_destroy> free_me(dnsrec);

  const size_t rr_count = ares_dns_record_rr_cnt(dnsrec, ARES_SECTION_ANSWER);
  const ares_dns_rr_t* soa = nullptr;
  for (size_t i = 0; i < rr_count; i++) {
    const ares_dns_rr_t* rr =
        ares_dns_record_rr_get(dnsrec, ARES_SECTION_ANSWER, i);
    if (ares_dns_rr_get_type(rr) == ARES_REC_TYPE_SOA) {
      soa = rr;
      break;
    }
  }

  if (soa == nullptr) return Just<int>(ARES_ENODATA);

  const char* nsname = ares_dns_rr_get_str(soa, ARES_RR_SOA_MNAME);
  const char* hostmaster = ares_dns_rr_get_str(soa, ARES_RR_SOA_RNAME);
  values[0] = OneByteString(env->isolate(), nsname);
  values[1] = OneByteString(env->isolate(), hostmaster);
  values[2] = Integer::NewFromUnsigned(
      env->isolate(), ares_dns_rr_get_u32(soa, ARES_RR_SOA_SERIAL));
  values[3] = Integer::New(env->isolate(),
                           ares_dns_rr_get_u32(soa, ARES_RR_SOA_REFRESH));
  values[4] =
      Integer::New(env->isolate(), ares_dns_rr_get_u32(soa, ARES_RR_SOA_RETRY));
  values[5] = Integer::New(env->isolate(),
                           ares_dns_rr_get_u32(soa, ARES_RR_SOA_EXPIRE));
  values[6] = Integer::NewFromUnsigned(
      env->isolate(), ares_dns_rr_get_u32(soa, ARES_RR_SOA_MINIMUM));
  Local<Object> soa_record;
  if (!NewDictionaryInstance(env->context(), tmpl, values)
           .ToLocal(&soa_record)) {
    return Nothing<int>();
  }

  wrap->CallOnComplete(soa_record);
  return Just<int>(ARES_SUCCESS);
}

int ReverseTraits::Send(QueryReverseWrap* wrap, const char* name) {
  permission::PermissionScope scope = permission::PermissionScope::kNet;
  Environment* env_holder = wrap->env();

  if (!env_holder->permission()->is_granted(env_holder, scope, name))
      [[unlikely]] {
    wrap->QueuePermissionModelResponseCallback(name);
    // Error will be returned in the callback
    return ARES_SUCCESS;
  }

  int length, family;
  char address_buffer[sizeof(struct in6_addr)];

  if (uv_inet_pton(AF_INET, name, &address_buffer) == 0) {
    length = sizeof(struct in_addr);
    family = AF_INET;
  } else if (uv_inet_pton(AF_INET6, name, &address_buffer) == 0) {
    length = sizeof(struct in6_addr);
    family = AF_INET6;
  } else {
    return UV_EINVAL;  // So errnoException() reports a proper error.
  }

  TRACE_EVENT_NESTABLE_ASYNC_BEGIN2(
      TRACING_CATEGORY_NODE2(dns, native), "reverse", wrap,
      "name", TRACE_STR_COPY(name),
      "family", family == AF_INET ? "ipv4" : "ipv6");

  ares_gethostbyaddr(wrap->channel()->cares_channel(),
                     address_buffer,
                     length,
                     family,
                     QueryReverseWrap::Callback,
                     wrap->MakeCallbackPointer());
  return ARES_SUCCESS;
}

Maybe<int> ReverseTraits::Parse(QueryReverseWrap* wrap,
                                const std::unique_ptr<ResponseData>& response) {
  if (!response->is_host) [[unlikely]] {
    return Just<int>(ARES_EBADRESP);
  }
  struct hostent* host = response->host.get();

  Environment* env = wrap->env();
  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());
  Local<Array> names;
  if (!HostentToNames(env, host).ToLocal(&names)) {
    return Nothing<int>();
  }
  wrap->CallOnComplete(names);
  return Just<int>(ARES_SUCCESS);
}

namespace {
template <class Wrap>
static void Query(const FunctionCallbackInfo<Value>& args) {
  ChannelWrap* channel;
  ASSIGN_OR_RETURN_UNWRAP(&channel, args.This());

  CHECK_EQ(false, args.IsConstructCall());
  CHECK(args[0]->IsObject());
  CHECK(args[1]->IsString());

  Local<Object> req_wrap_obj = args[0].As<Object>();
  Local<String> string = args[1].As<String>();
  auto wrap = std::make_unique<Wrap>(channel, req_wrap_obj);

  node::Utf8Value utf8name(args.GetIsolate(), string);
  auto plain_name = utf8name.ToStringView();
  std::string name = ada::idna::to_ascii(plain_name);

  // Ensure c-ares did not fall back to loopback resolver.
  channel->EnsureServers();

  channel->ModifyActivityQueryCount(1);
  int err = wrap->Send(name.c_str());
  if (err) {
    channel->ModifyActivityQueryCount(-1);
  } else {
    // Release ownership of the pointer allowing the ownership to be transferred
    USE(wrap.release());
  }

  args.GetReturnValue().Set(err);
}

void AfterGetAddrInfo(uv_getaddrinfo_t* req, int status, struct addrinfo* res) {
  auto cleanup = OnScopeLeave([&]() { uv_freeaddrinfo(res); });
  BaseObjectPtr<GetAddrInfoReqWrap> req_wrap{
      static_cast<GetAddrInfoReqWrap*>(req->data)};
  Environment* env = req_wrap->env();

  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());

  Local<Value> argv[] = {
    Integer::New(env->isolate(), status),
    Null(env->isolate())
  };

  uint32_t n = 0;
  const uint8_t order = req_wrap->order();

  if (status == 0) {
    Local<Array> results = Array::New(env->isolate());

    auto add = [&](bool want_ipv4, bool want_ipv6) -> Maybe<void> {
      for (auto p = res; p != nullptr; p = p->ai_next) {
        CHECK_EQ(p->ai_socktype, SOCK_STREAM);

        const char* addr;
        if (want_ipv4 && p->ai_family == AF_INET) {
          addr = reinterpret_cast<char*>(
              &(reinterpret_cast<struct sockaddr_in*>(p->ai_addr)->sin_addr));
        } else if (want_ipv6 && p->ai_family == AF_INET6) {
          addr = reinterpret_cast<char*>(
              &(reinterpret_cast<struct sockaddr_in6*>(p->ai_addr)->sin6_addr));
        } else {
          continue;
        }

        char ip[INET6_ADDRSTRLEN];
        if (uv_inet_ntop(p->ai_family, addr, ip, sizeof(ip)))
          continue;

        Local<String> s = OneByteString(env->isolate(), ip);
        if (results->Set(env->context(), n, s).IsNothing())
          return Nothing<void>();
        n++;
      }
      return JustVoid();
    };

    switch (order) {
      case DNS_ORDER_IPV4_FIRST:
        if (add(true, false).IsNothing() || add(false, true).IsNothing())
          return;

        break;
      case DNS_ORDER_IPV6_FIRST:
        if (add(false, true).IsNothing() || add(true, false).IsNothing())
          return;

        break;
      default:
        if (add(true, true).IsNothing()) return;

        break;
    }

    // No responses were found to return
    if (n == 0) {
      argv[0] = Integer::New(env->isolate(), UV_EAI_NODATA);
    }

    argv[1] = results;
  }

  TRACE_EVENT_NESTABLE_ASYNC_END2(TRACING_CATEGORY_NODE2(dns, native),
                                  "lookup",
                                  req_wrap.get(),
                                  "count",
                                  n,
                                  "order",
                                  order);

  // Make the callback into JavaScript
  req_wrap->MakeCallback(env->oncomplete_string(), arraysize(argv), argv);
}


void AfterGetNameInfo(uv_getnameinfo_t* req,
                      int status,
                      const char* hostname,
                      const char* service) {
  BaseObjectPtr<GetNameInfoReqWrap> req_wrap{
      static_cast<GetNameInfoReqWrap*>(req->data)};
  Environment* env = req_wrap->env();

  HandleScope handle_scope(env->isolate());
  Context::Scope context_scope(env->context());

  Local<Value> argv[] = {
    Integer::New(env->isolate(), status),
    Null(env->isolate()),
    Null(env->isolate())
  };

  if (status == 0) {
    // Success
    Local<String> js_hostname = OneByteString(env->isolate(), hostname);
    Local<String> js_service = OneByteString(env->isolate(), service);
    argv[1] = js_hostname;
    argv[2] = js_service;
  }

  TRACE_EVENT_NESTABLE_ASYNC_END2(
      TRACING_CATEGORY_NODE2(dns, native), "lookupService", req_wrap.get(),
      "hostname", TRACE_STR_COPY(hostname),
      "service", TRACE_STR_COPY(service));

  // Make the callback into JavaScript
  req_wrap->MakeCallback(env->oncomplete_string(), arraysize(argv), argv);
}

void CanonicalizeIP(const FunctionCallbackInfo<Value>& args) {
  Isolate* isolate = args.GetIsolate();
  node::Utf8Value ip(isolate, args[0]);

  int af;
  unsigned char result[sizeof(ares_addr_port_node::addr)];
  if (uv_inet_pton(af = AF_INET, *ip, result) != 0 &&
      uv_inet_pton(af = AF_INET6, *ip, result) != 0)
    return;

  char canonical_ip[INET6_ADDRSTRLEN]{};
  CHECK_EQ(0, uv_inet_ntop(af, result, canonical_ip, sizeof(canonical_ip)));
  args.GetReturnValue().Set(OneByteString(isolate, canonical_ip));
}

void ConvertIpv6StringToBuffer(const FunctionCallbackInfo<Value>& args) {
  Isolate* isolate = args.GetIsolate();
  node::Utf8Value ip(isolate, args[0]);
  unsigned char dst[16];  // IPv6 addresses are 128 bits (16 bytes)

  if (uv_inet_pton(AF_INET6, *ip, dst) != 0) {
    isolate->ThrowException(Exception::Error(
        FIXED_ONE_BYTE_STRING(isolate, "Invalid IPv6 address")));
    return;
  }

  Local<Object> buffer;
  if (node::Buffer::Copy(
          isolate, reinterpret_cast<const char*>(dst), sizeof(dst))
          .ToLocal(&buffer)) {
    args.GetReturnValue().Set(buffer);
  }
}

void GetAddrInfo(const FunctionCallbackInfo<Value>& args) {
  Environment* env = Environment::GetCurrent(args);

  CHECK(args[0]->IsObject());
  CHECK(args[1]->IsString());
  CHECK(args[2]->IsInt32());
  CHECK(args[4]->IsUint32());
  Local<Object> req_wrap_obj = args[0].As<Object>();
  node::Utf8Value hostname(env->isolate(), args[1]);

  ERR_ACCESS_DENIED_IF_INSUFFICIENT_PERMISSIONS(
      env, permission::PermissionScope::kNet, hostname.ToStringView(), args);

  std::string ascii_hostname = ada::idna::to_ascii(hostname.ToStringView());

  int32_t flags = 0;
  if (args[3]->IsInt32()) {
    flags = args[3].As<Int32>()->Value();
  }

  int family;

  switch (args[2].As<Int32>()->Value()) {
    case 0:
      family = AF_UNSPEC;
      break;
    case 4:
      family = AF_INET;
      break;
    case 6:
      family = AF_INET6;
      break;
    default:
      UNREACHABLE("bad address family");
  }

  Local<Uint32> order = args[4].As<Uint32>();

  auto req_wrap =
      std::make_unique<GetAddrInfoReqWrap>(env, req_wrap_obj, order->Value());

  struct addrinfo hints;
  memset(&hints, 0, sizeof(hints));
  hints.ai_family = family;
  hints.ai_socktype = SOCK_STREAM;
  hints.ai_flags = flags;

  TRACE_EVENT_NESTABLE_ASYNC_BEGIN2(TRACING_CATEGORY_NODE2(dns, native),
                                    "lookup",
                                    req_wrap.get(),
                                    "hostname",
                                    TRACE_STR_COPY(ascii_hostname.data()),
                                    "family",
                                    family == AF_INET    ? "ipv4"
                                    : family == AF_INET6 ? "ipv6"
                                                         : "unspec");

  int err = req_wrap->Dispatch(
      uv_getaddrinfo, AfterGetAddrInfo, ascii_hostname.data(), nullptr, &hints);
  if (err == 0)
    // Release ownership of the pointer allowing the ownership to be transferred
    USE(req_wrap.release());

  args.GetReturnValue().Set(err);
}


void GetNameInfo(const FunctionCallbackInfo<Value>& args) {
  Environment* env = Environment::GetCurrent(args);

  CHECK(args[0]->IsObject());
  CHECK(args[1]->IsString());
  CHECK(args[2]->IsUint32());
  Local<Object> req_wrap_obj = args[0].As<Object>();
  node::Utf8Value ip(env->isolate(), args[1]);
  const unsigned port = args[2]->Uint32Value(env->context()).FromJust();
  struct sockaddr_storage addr;

  CHECK(uv_ip4_addr(*ip, port, reinterpret_cast<sockaddr_in*>(&addr)) == 0 ||
        uv_ip6_addr(*ip, port, reinterpret_cast<sockaddr_in6*>(&addr)) == 0);

  auto req_wrap = std::make_unique<GetNameInfoReqWrap>(env, req_wrap_obj);

  TRACE_EVENT_NESTABLE_ASYNC_BEGIN2(
      TRACING_CATEGORY_NODE2(dns, native), "lookupService", req_wrap.get(),
      "ip", TRACE_STR_COPY(*ip), "port", port);

  int err = 0;
  if (!env->permission()->is_granted(
          env, permission::PermissionScope::kNet, ip.ToStringView()))
      [[unlikely]] {
    req_wrap->InsufficientPermissionError(*ip);
  } else {
    err = req_wrap->Dispatch(uv_getnameinfo,
                             AfterGetNameInfo,
                             reinterpret_cast<struct sockaddr*>(&addr),
                             NI_NAMEREQD);
  }

  if (err == 0)
    // Release ownership of the pointer allowing the ownership to be transferred
    USE(req_wrap.release());

  args.GetReturnValue().Set(err);
}


void GetServers(const FunctionCallbackInfo<Value>& args) {
  Environment* env = Environment::GetCurrent(args);
  ChannelWrap* channel;
  ASSIGN_OR_RETURN_UNWRAP(&channel, args.This());

  Local<Array> server_array = Array::New(env->isolate());

  char* csv = ares_get_servers_csv(channel->cares_channel());
  auto cleanup = OnScopeLeave([&]() { ares_free_string(csv); });
  std::vector<std::pair<std::string, int>> servers = ParseServersCsv(csv);

  for (uint32_t i = 0; i < servers.size(); i++) {
    Local<Value> ret[] = {
        OneByteString(env->isolate(), servers[i].first.c_str()),
        Integer::New(env->isolate(), servers[i].second)};

    if (server_array->Set(env->context(), i,
                          Array::New(env->isolate(), ret, arraysize(ret)))
          .IsNothing()) {
      return;
    }
  }

  args.GetReturnValue().Set(server_array);
}


void SetServers(const FunctionCallbackInfo<Value>& args) {
  Environment* env = Environment::GetCurrent(args);
  ChannelWrap* channel;
  ASSIGN_OR_RETURN_UNWRAP(&channel, args.This());

  if (channel->active_query_count()) {
    return args.GetReturnValue().Set(DNS_ESETSRVPENDING);
  }

  CHECK(args[0]->IsArray());

  Local<Array> arr = args[0].As<Array>();

  uint32_t len = arr->Length();

  // An empty list clears all configured servers. ares_set_servers_ports_csv()
  // treats an empty string as "blank all servers".
  std::string csv;

  for (uint32_t i = 0; i < len; i++) {
    Local<Value> val;
    if (!arr->Get(env->context(), i).ToLocal(&val)) return;
    CHECK(val->IsArray());

    Local<Array> elm = val.As<Array>();

    Local<Value> familyValue;
    Local<Value> ipValue;
    Local<Value> portValue;

    if (!elm->Get(env->context(), 0).ToLocal(&familyValue)) return;
    if (!elm->Get(env->context(), 1).ToLocal(&ipValue)) return;
    if (!elm->Get(env->context(), 2).ToLocal(&portValue)) return;

    CHECK(familyValue->IsInt32());
    CHECK(ipValue->IsString());
    CHECK(portValue->IsInt32());

    int32_t fam = familyValue.As<Int32>()->Value();
    node::Utf8Value ip(env->isolate(), ipValue);
    int32_t port = portValue.As<Int32>()->Value();

    if (!csv.empty()) csv += ',';

    // Incoming CSV format expected by c-ares: host[:port][,host[:port]]...
    // IPv6 addresses must be wrapped in square brackets.
    switch (fam) {
      case 4:
        csv += *ip;
        break;
      case 6:
        csv += '[';
        csv += *ip;
        csv += ']';
        break;
      default:
        UNREACHABLE("Bad address family");
    }
    csv += ':';
    csv += std::to_string(port);
  }

  int err = ares_set_servers_ports_csv(channel->cares_channel(), csv.c_str());

  if (err == ARES_SUCCESS)
    channel->set_is_servers_default(false);

  args.GetReturnValue().Set(err);
}

void SetLocalAddress(const FunctionCallbackInfo<Value>& args) {
  Environment* env = Environment::GetCurrent(args);
  ChannelWrap* channel;
  ASSIGN_OR_RETURN_UNWRAP(&channel, args.This());

  CHECK_EQ(args.Length(), 2);
  CHECK(args[0]->IsString());

  Isolate* isolate = args.GetIsolate();
  node::Utf8Value ip0(isolate, args[0]);

  unsigned char addr0[sizeof(struct in6_addr)];
  unsigned char addr1[sizeof(struct in6_addr)];
  int type0 = 0;

  // This function accepts 2 arguments.  The first may be either an IPv4
  // address or an IPv6 address.  If present, the second argument must be the
  // other type of address.  Otherwise, the unspecified type of IP is set
  // to 0 (any).

  if (uv_inet_pton(AF_INET, *ip0, &addr0) == 0) {
    ares_set_local_ip4(channel->cares_channel(), nbytes::ReadUint32BE(addr0));
    type0 = 4;
  } else if (uv_inet_pton(AF_INET6, *ip0, &addr0) == 0) {
    ares_set_local_ip6(channel->cares_channel(), addr0);
    type0 = 6;
  } else {
    THROW_ERR_INVALID_ARG_VALUE(env, "Invalid IP address.");
    return;
  }

  if (!args[1]->IsUndefined()) {
    CHECK(args[1]->IsString());
    node::Utf8Value ip1(isolate, args[1]);

    if (uv_inet_pton(AF_INET, *ip1, &addr1) == 0) {
      if (type0 == 4) {
        THROW_ERR_INVALID_ARG_VALUE(env, "Cannot specify two IPv4 addresses.");
        return;
      } else {
        ares_set_local_ip4(channel->cares_channel(),
                           nbytes::ReadUint32BE(addr1));
      }
    } else if (uv_inet_pton(AF_INET6, *ip1, &addr1) == 0) {
      if (type0 == 6) {
        THROW_ERR_INVALID_ARG_VALUE(env, "Cannot specify two IPv6 addresses.");
        return;
      } else {
        ares_set_local_ip6(channel->cares_channel(), addr1);
      }
    } else {
      THROW_ERR_INVALID_ARG_VALUE(env, "Invalid IP address.");
      return;
    }
  } else {
    // No second arg specified
    if (type0 == 4) {
      memset(&addr1, 0, sizeof(addr1));
      ares_set_local_ip6(channel->cares_channel(), addr1);
    } else {
      ares_set_local_ip4(channel->cares_channel(), 0);
    }
  }
}

void Cancel(const FunctionCallbackInfo<Value>& args) {
  ChannelWrap* channel;
  ASSIGN_OR_RETURN_UNWRAP(&channel, args.This());

  TRACE_EVENT_INSTANT0(TRACING_CATEGORY_NODE2(dns, native),
      "cancel", TRACE_EVENT_SCOPE_THREAD);

  ares_cancel(channel->cares_channel());
}

const char EMSG_ESETSRVPENDING[] = "There are pending queries.";
void StrError(const FunctionCallbackInfo<Value>& args) {
  Environment* env = Environment::GetCurrent(args);
  int code = args[0]->Int32Value(env->context()).FromJust();
  const char* errmsg = (code == DNS_ESETSRVPENDING) ?
    EMSG_ESETSRVPENDING :
    ares_strerror(code);
  args.GetReturnValue().Set(OneByteString(env->isolate(), errmsg));
}

}  // namespace

inline void safe_free_hostent(struct hostent* host) {
  int idx;

  if (host->h_addr_list != nullptr) {
    idx = 0;
    while (host->h_addr_list[idx]) {
      free(host->h_addr_list[idx++]);
    }
    free(host->h_addr_list);
    host->h_addr_list = nullptr;
  }

  if (host->h_aliases != nullptr) {
    idx = 0;
    while (host->h_aliases[idx]) {
      free(host->h_aliases[idx++]);
    }
    free(host->h_aliases);
    host->h_aliases = nullptr;
  }

  free(host->h_name);
  free(host);
}

void Initialize(Local<Object> target,
                Local<Value> unused,
                Local<Context> context,
                void* priv) {
  Environment* env = Environment::GetCurrent(context);
  Isolate* isolate = env->isolate();

  SetMethod(context, target, "getaddrinfo", GetAddrInfo);
  SetMethod(context, target, "getnameinfo", GetNameInfo);
  SetMethodNoSideEffect(context, target, "canonicalizeIP", CanonicalizeIP);
  SetMethodNoSideEffect(
      context, target, "convertIpv6StringToBuffer", ConvertIpv6StringToBuffer);

  SetMethod(context, target, "strerror", StrError);

  NODE_DEFINE_CONSTANT(target, AF_INET);
  NODE_DEFINE_CONSTANT(target, AF_INET6);
  NODE_DEFINE_CONSTANT(target, AF_UNSPEC);
  NODE_DEFINE_CONSTANT(target, AI_ADDRCONFIG);
  NODE_DEFINE_CONSTANT(target, AI_ALL);
  NODE_DEFINE_CONSTANT(target, AI_V4MAPPED);
  NODE_DEFINE_CONSTANT(target, DNS_ORDER_VERBATIM);
  NODE_DEFINE_CONSTANT(target, DNS_ORDER_IPV4_FIRST);
  NODE_DEFINE_CONSTANT(target, DNS_ORDER_IPV6_FIRST);

  Local<FunctionTemplate> aiw = AsyncWrap::MakeLazilyInitializedJSTemplate(env);
  SetConstructorFunction(context, target, "GetAddrInfoReqWrap", aiw);

  Local<FunctionTemplate> niw = AsyncWrap::MakeLazilyInitializedJSTemplate(env);
  SetConstructorFunction(context, target, "GetNameInfoReqWrap", niw);

  Local<FunctionTemplate> qrw = AsyncWrap::MakeLazilyInitializedJSTemplate(env);
  SetConstructorFunction(context, target, "QueryReqWrap", qrw);

  Local<FunctionTemplate> channel_wrap =
      NewFunctionTemplate(isolate, ChannelWrap::New);
  channel_wrap->InstanceTemplate()->SetInternalFieldCount(
      ChannelWrap::kInternalFieldCount);
  channel_wrap->Inherit(AsyncWrap::GetConstructorTemplate(env));

#define V(Name, _, JS)                                                         \
  SetProtoMethod(isolate, channel_wrap, #JS, Query<Query##Name##Wrap>);
  QUERY_TYPES(V)
#undef V

  SetProtoMethodNoSideEffect(isolate, channel_wrap, "getServers", GetServers);
  SetProtoMethod(isolate, channel_wrap, "setServers", SetServers);
  SetProtoMethod(isolate, channel_wrap, "setLocalAddress", SetLocalAddress);
  SetProtoMethod(isolate, channel_wrap, "cancel", Cancel);

  SetConstructorFunction(context, target, "ChannelWrap", channel_wrap);
}

void RegisterExternalReferences(ExternalReferenceRegistry* registry) {
  registry->Register(GetAddrInfo);
  registry->Register(GetNameInfo);
  registry->Register(CanonicalizeIP);
  registry->Register(ConvertIpv6StringToBuffer);
  registry->Register(StrError);
  registry->Register(ChannelWrap::New);

#define V(Name, _, __) registry->Register(Query<Query##Name##Wrap>);
  QUERY_TYPES(V)
#undef V

  registry->Register(GetServers);
  registry->Register(SetServers);
  registry->Register(SetLocalAddress);
  registry->Register(Cancel);
}

}  // namespace cares_wrap
}  // namespace node

NODE_BINDING_CONTEXT_AWARE_INTERNAL(cares_wrap, node::cares_wrap::Initialize)
NODE_BINDING_EXTERNAL_REFERENCE(cares_wrap,
                                node::cares_wrap::RegisterExternalReferences)