// 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.

#define CARES_STATICLIB

#include "ares.h"

#include "async-wrap.h"

#include "async-wrap-inl.h"

#include "env.h"

#include "env-inl.h"

#include "node.h"

#include "req-wrap.h"

#include "req-wrap-inl.h"

#include "tree.h"

#include "util.h"

#include "util-inl.h"

#include "uv.h"

#include <errno.h>

#include <stdlib.h>

#include <string.h>

#if defined(__ANDROID__) || \

defined(__MINGW32__) || \

defined(__OpenBSD__) || \

defined(_MSC_VER)

# include <nameser.h>

#else

# include <arpa/nameser.h>

#endif

#if defined(__OpenBSD__)

# define AI_V4MAPPED 0

#endif

namespace node {

namespace cares_wrap {

using v8::Array;

using v8::Context;

using v8::EscapableHandleScope;

using v8::FunctionCallbackInfo;

using v8::FunctionTemplate;

using v8::HandleScope;

using v8::Integer;

using v8::Local;

using v8::Null;

using v8::Object;

using v8::String;

using v8::Value;

namespace {

inline uint16_t cares_get_16bit(const unsigned char* p) {

return static_cast<uint32_t>(p[0] << 8U) | (static_cast<uint32_t>(p[1]));

}

inline uint32_t cares_get_32bit(const unsigned char* p) {

return static_cast<uint32_t>(p[0] << 24U) |

static_cast<uint32_t>(p[1] << 16U) |

static_cast<uint32_t>(p[2] << 8U) |

static_cast<uint32_t>(p[3]);

}

const int ns_t_cname_or_a = -1;

inline const char* ToErrorCodeString(int status) {

switch (status) {

#define V(code) case ARES_##code: return #code;

V(EADDRGETNETWORKPARAMS)

V(EBADFAMILY)

V(EBADFLAGS)

V(EBADHINTS)

V(EBADNAME)

V(EBADQUERY)

V(EBADRESP)

V(EBADSTR)

V(ECANCELLED)

V(ECONNREFUSED)

V(EDESTRUCTION)

V(EFILE)

V(EFORMERR)

V(ELOADIPHLPAPI)

V(ENODATA)

V(ENOMEM)

V(ENONAME)

V(ENOTFOUND)

V(ENOTIMP)

V(ENOTINITIALIZED)

V(EOF)

V(EREFUSED)

V(ESERVFAIL)

V(ETIMEOUT)

#undef V

}

return "UNKNOWN_ARES_ERROR";

}

class GetAddrInfoReqWrap : public ReqWrap<uv_getaddrinfo_t> {

public:

GetAddrInfoReqWrap(Environment* env, Local<Object> req_wrap_obj);

~GetAddrInfoReqWrap();

size_t self_size() const override { return sizeof(*this); }

};

GetAddrInfoReqWrap::GetAddrInfoReqWrap(Environment* env,

Local<Object> req_wrap_obj)

: ReqWrap(env, req_wrap_obj, AsyncWrap::PROVIDER_GETADDRINFOREQWRAP) {

Wrap(req_wrap_obj, this);

}

GetAddrInfoReqWrap::~GetAddrInfoReqWrap() {

ClearWrap(object());

}

class GetNameInfoReqWrap : public ReqWrap<uv_getnameinfo_t> {

public:

GetNameInfoReqWrap(Environment* env, Local<Object> req_wrap_obj);

~GetNameInfoReqWrap();

size_t self_size() const override { return sizeof(*this); }

};

GetNameInfoReqWrap::GetNameInfoReqWrap(Environment* env,

Local<Object> req_wrap_obj)

: ReqWrap(env, req_wrap_obj, AsyncWrap::PROVIDER_GETNAMEINFOREQWRAP) {

Wrap(req_wrap_obj, this);

}

GetNameInfoReqWrap::~GetNameInfoReqWrap() {

ClearWrap(object());

}

int cmp_ares_tasks(const node_ares_task* a, const node_ares_task* b) {

if (a->sock < b->sock)

return -1;

if (a->sock > b->sock)

return 1;

return 0;

}

RB_GENERATE_STATIC(node_ares_task_list, node_ares_task, node, cmp_ares_tasks)

/* This is called once per second by loop->timer. It is used to constantly */

/* call back into c-ares for possibly processing timeouts. */

void ares_timeout(uv_timer_t* handle) {

Environment* env = Environment::from_cares_timer_handle(handle);

CHECK_EQ(false, RB_EMPTY(env->cares_task_list()));

ares_process_fd(env->cares_channel(), ARES_SOCKET_BAD, ARES_SOCKET_BAD);

}

void ares_poll_cb(uv_poll_t* watcher, int status, int events) {

node_ares_task* task = ContainerOf(&node_ares_task::poll_watcher, watcher);

Environment* env = task->env;

/* Reset the idle timer */

uv_timer_again(env->cares_timer_handle());

if (status < 0) {

/* An error happened. Just pretend that the socket is both readable and */

/* writable. */

ares_process_fd(env->cares_channel(), task->sock, task->sock);

return;

}

/* Process DNS responses */

ares_process_fd(env->cares_channel(),

events & UV_READABLE ? task->sock : ARES_SOCKET_BAD,

events & UV_WRITABLE ? task->sock : ARES_SOCKET_BAD);

}

void ares_poll_close_cb(uv_handle_t* watcher) {

node_ares_task* task = ContainerOf(&node_ares_task::poll_watcher,

reinterpret_cast<uv_poll_t*>(watcher));

free(task);

}

/* Allocates and returns a new node_ares_task */

node_ares_task* ares_task_create(Environment* env, ares_socket_t sock) {

auto task = node::UncheckedMalloc<node_ares_task>(1);

if (task == nullptr) {

/* Out of memory. */

return nullptr;

}

task->env = env;

task->sock = sock;

if (uv_poll_init_socket(env->event_loop(), &task->poll_watcher, sock) < 0) {

/* This should never happen. */

free(task);

return nullptr;

}

return task;

}

/* Callback from ares when socket operation is started */

void ares_sockstate_cb(void* data,

ares_socket_t sock,

int read,

int write) {

Environment* env = static_cast<Environment*>(data);

node_ares_task* task;

node_ares_task lookup_task;

lookup_task.sock = sock;

task = RB_FIND(node_ares_task_list, env->cares_task_list(), &lookup_task);

if (read || write) {

if (!task) {

/* New socket */

/* If this is the first socket then start the timer. */

uv_timer_t* timer_handle = env->cares_timer_handle();

if (!uv_is_active(reinterpret_cast<uv_handle_t*>(timer_handle))) {

CHECK(RB_EMPTY(env->cares_task_list()));

uv_timer_start(timer_handle, ares_timeout, 1000, 1000);

}

task = ares_task_create(env, 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;

}

RB_INSERT(node_ares_task_list, env->cares_task_list(), 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 {

/* read == 0 and write == 0 this is c-ares's way of notifying us that */

/* the socket is now closed. We must free the data associated with */

/* socket. */

CHECK(task &&

"When an ares socket is closed we should have a handle for it");

RB_REMOVE(node_ares_task_list, env->cares_task_list(), task);

uv_close(reinterpret_cast<uv_handle_t*>(&task->poll_watcher),

ares_poll_close_cb);

if (RB_EMPTY(env->cares_task_list())) {

uv_timer_stop(env->cares_timer_handle());

}

}

}

Local<Array> HostentToAddresses(Environment* env,

struct hostent* host,

Local<Array> append_to = Local<Array>()) {

EscapableHandleScope scope(env->isolate());

auto context = env->context();

bool append = !append_to.IsEmpty();

Local<Array> addresses = append ? append_to : Array::New(env->isolate());

size_t offset = addresses->Length();

char ip[INET6_ADDRSTRLEN];

for (uint32_t i = 0; host->h_addr_list[i] != nullptr; ++i) {

uv_inet_ntop(host->h_addrtype, host->h_addr_list[i], ip, sizeof(ip));

Local<String> address = OneByteString(env->isolate(), ip);

addresses->Set(context, i + offset, address).FromJust();

}

return append ? addresses : scope.Escape(addresses);

}

Local<Array> HostentToNames(Environment* env,

struct hostent* host,

Local<Array> append_to = Local<Array>()) {

EscapableHandleScope scope(env->isolate());

auto context = env->context();

bool append = !append_to.IsEmpty();

Local<Array> names = append ? append_to : Array::New(env->isolate());

size_t offset = names->Length();

for (uint32_t i = 0; host->h_aliases[i] != nullptr; ++i) {

Local<String> address = OneByteString(env->isolate(), host->h_aliases[i]);

names->Set(context, i + offset, address).FromJust();

}

return append ? names : scope.Escape(names);

}

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 = 0;

}

if (host->h_aliases != nullptr) {

idx = 0;

while (host->h_aliases[idx]) {

free(host->h_aliases[idx++]);

}

free(host->h_aliases);

host->h_aliases = 0;

}

if (host->h_name != nullptr) {

free(host->h_name);

}

host->h_addrtype = host->h_length = 0;

}

void cares_wrap_hostent_cpy(struct hostent* dest, struct hostent* src) {

dest->h_addr_list = nullptr;

dest->h_addrtype = 0;

dest->h_aliases = nullptr;

dest->h_length = 0;

dest->h_name = nullptr;

/* copy `h_name` */

size_t name_size = strlen(src->h_name) + 1;

dest->h_name = node::Malloc<char>(name_size);

memcpy(dest->h_name, src->h_name, name_size);

/* copy `h_aliases` */

size_t alias_count;

size_t cur_alias_length;

for (alias_count = 0;

src->h_aliases[alias_count] != nullptr;

alias_count++) {

}

dest->h_aliases = node::Malloc<char*>(alias_count + 1);

for (size_t i = 0; i < alias_count; i++) {

cur_alias_length = strlen(src->h_aliases[i]);

dest->h_aliases[i] = node::Malloc(cur_alias_length + 1);

memcpy(dest->h_aliases[i], src->h_aliases[i], cur_alias_length + 1);

}

dest->h_aliases[alias_count] = nullptr;

/* copy `h_addr_list` */

size_t list_count;

for (list_count = 0;

src->h_addr_list[list_count] != nullptr;

list_count++) {

}

dest->h_addr_list = node::Malloc<char*>(list_count + 1);

for (size_t i = 0; i < list_count; i++) {

dest->h_addr_list[i] = node::Malloc(src->h_length);

memcpy(dest->h_addr_list[i], src->h_addr_list[i], src->h_length);

}

dest->h_addr_list[list_count] = nullptr;

/* work after work */

dest->h_length = src->h_length;

dest->h_addrtype = src->h_addrtype;

}

class QueryWrap;

struct CaresAsyncData {

QueryWrap* wrap;

int status;

bool is_host;

union {

hostent* host;

unsigned char* buf;

} data;

int len;

uv_async_t async_handle;

};

void SetupCaresChannel(Environment* env) {

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 = env;

/* We do the call to ares_init_option for caller. */

int r = ares_init_options(env->cares_channel_ptr(),

&options,

ARES_OPT_FLAGS | ARES_OPT_SOCK_STATE_CB);

if (r != ARES_SUCCESS) {

ares_library_cleanup();

return env->ThrowError(ToErrorCodeString(r));

}

}

/**

* 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 AresEnsureServers(Environment* env) {

/* if last query is OK or servers are set by user self, do not check */

if (env->cares_query_last_ok() || !env->cares_is_servers_default()) {

return;

}

ares_channel channel = env->cares_channel();

ares_addr_port_node* servers = nullptr;

ares_get_servers_ports(channel, &servers);

/* if no server or multi-servers, ignore */

if (servers == nullptr) return;

if (servers->next != nullptr) {

ares_free_data(servers);

env->set_cares_is_servers_default(false);

return;

}

/* if the only server is not 127.0.0.1, ignore */

if (servers[0].family != AF_INET ||

servers[0].addr.addr4.s_addr != htonl(INADDR_LOOPBACK) ||

servers[0].tcp_port != 0 ||

servers[0].udp_port != 0) {

ares_free_data(servers);

env->set_cares_is_servers_default(false);

return;

}

ares_free_data(servers);

servers = nullptr;

/* destroy channel and reset channel */

ares_destroy(channel);

SetupCaresChannel(env);

}

class QueryWrap : public AsyncWrap {

public:

QueryWrap(Environment* env, Local<Object> req_wrap_obj)

: AsyncWrap(env, req_wrap_obj, AsyncWrap::PROVIDER_QUERYWRAP) {

if (env->in_domain())

req_wrap_obj->Set(env->domain_string(), env->domain_array()->Get(0));

Wrap(req_wrap_obj, this);

}

~QueryWrap() override {

CHECK_EQ(false, persistent().IsEmpty());

ClearWrap(object());

persistent().Reset();

}

// Subclasses should implement the appropriate Send method.

virtual int Send(const char* name) {

UNREACHABLE();

return 0;

}

virtual int Send(const char* name, int family) {

UNREACHABLE();

return 0;

}

protected:

void* GetQueryArg() {

return static_cast<void*>(this);

}

static void AresQuery(Environment* env, const char* name,

int dnsclass, int type, ares_callback callback,

void* arg) {

AresEnsureServers(env);

ares_query(env->cares_channel(), name, dnsclass, type, callback, arg);

}

static void CaresAsyncClose(uv_handle_t* handle) {

uv_async_t* async = reinterpret_cast<uv_async_t*>(handle);

auto data = static_cast<struct CaresAsyncData*>(async->data);

delete data->wrap;

delete data;

}

static void CaresAsyncCb(uv_async_t* handle) {

auto data = static_cast<struct CaresAsyncData*>(handle->data);

QueryWrap* wrap = data->wrap;

int status = data->status;

if (status != ARES_SUCCESS) {

wrap->ParseError(status);

} else if (!data->is_host) {

unsigned char* buf = data->data.buf;

wrap->Parse(buf, data->len);

free(buf);

} else {

hostent* host = data->data.host;

wrap->Parse(host);

safe_free_hostent(host);

free(host);

}

uv_close(reinterpret_cast<uv_handle_t*>(handle), CaresAsyncClose);

}

static void Callback(void *arg, int status, int timeouts,

unsigned char* answer_buf, int answer_len) {

QueryWrap* wrap = static_cast<QueryWrap*>(arg);

unsigned char* buf_copy = nullptr;

if (status == ARES_SUCCESS) {

buf_copy = node::Malloc<unsigned char>(answer_len);

memcpy(buf_copy, answer_buf, answer_len);

}

CaresAsyncData* data = new CaresAsyncData();

data->status = status;

data->wrap = wrap;

data->is_host = false;

data->data.buf = buf_copy;

data->len = answer_len;

uv_async_t* async_handle = &data->async_handle;

CHECK_EQ(0, uv_async_init(wrap->env()->event_loop(),

async_handle,

CaresAsyncCb));

wrap->env()->set_cares_query_last_ok(status != ARES_ECONNREFUSED);

async_handle->data = data;

uv_async_send(async_handle);

}

static void Callback(void *arg, int status, int timeouts,

struct hostent* host) {

QueryWrap* wrap = static_cast<QueryWrap*>(arg);

struct hostent* host_copy = nullptr;

if (status == ARES_SUCCESS) {

host_copy = node::Malloc<hostent>(1);

cares_wrap_hostent_cpy(host_copy, host);

}

CaresAsyncData* data = new CaresAsyncData();

data->status = status;

data->data.host = host_copy;

data->wrap = wrap;

data->is_host = true;

uv_async_t* async_handle = &data->async_handle;

CHECK_EQ(0, uv_async_init(wrap->env()->event_loop(),

async_handle,

CaresAsyncCb));

wrap->env()->set_cares_query_last_ok(status != ARES_ECONNREFUSED);

async_handle->data = data;

uv_async_send(async_handle);

}

void CallOnComplete(Local<Value> answer,

Local<Value> extra = Local<Value>()) {

HandleScope handle_scope(env()->isolate());

Context::Scope context_scope(env()->context());

Local<Value> argv[] = {

Integer::New(env()->isolate(), 0),

answer,

extra

};

const int argc = arraysize(argv) - extra.IsEmpty();

MakeCallback(env()->oncomplete_string(), argc, argv);

}

void ParseError(int status) {

CHECK_NE(status, ARES_SUCCESS);

HandleScope handle_scope(env()->isolate());

Context::Scope context_scope(env()->context());

const char* code = ToErrorCodeString(status);

Local<Value> arg = OneByteString(env()->isolate(), code);

MakeCallback(env()->oncomplete_string(), 1, &arg);

}

// Subclasses should implement the appropriate Parse method.

virtual void Parse(unsigned char* buf, int len) {

UNREACHABLE();

}

virtual void Parse(struct hostent* host) {

UNREACHABLE();

}

};

template<typename T>

Local<Array> AddrTTLToArray(Environment* env,

const T* addrttls,

size_t naddrttls) {

auto isolate = env->isolate();

EscapableHandleScope escapable_handle_scope(isolate);

auto context = env->context();

Local<Array> ttls = Array::New(isolate, naddrttls);

for (size_t i = 0; i < naddrttls; i++) {

auto value = Integer::New(isolate, addrttls[i].ttl);

ttls->Set(context, i, value).FromJust();

}

return escapable_handle_scope.Escape(ttls);

}

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());

auto context = env->context();

hostent* host;

int status;

switch (*type) {

case ns_t_a:

case ns_t_cname:

case ns_t_cname_or_a:

status = ares_parse_a_reply(buf,

len,

&host,

static_cast<ares_addrttl*>(addrttls),

naddrttls);

break;

case ns_t_aaaa:

status = ares_parse_aaaa_reply(buf,

len,

&host,

static_cast<ares_addr6ttl*>(addrttls),

naddrttls);

break;

case ns_t_ns:

status = ares_parse_ns_reply(buf, len, &host);

break;

case ns_t_ptr:

status = ares_parse_ptr_reply(buf, len, NULL, 0, AF_INET, &host);

break;

default:

CHECK(0 && "Bad NS type");

break;

}

if (status != ARES_SUCCESS)

return status;

/* If it's `CNAME`, return the CNAME value;

* And if it's `CNAME_OR_A` and it has value in `h_name` and `h_aliases[0]`,

* we consider it's a CNAME record, otherwise we consider it's an A record. */

if ((*type == ns_t_cname_or_a && host->h_name && host->h_aliases[0]) ||

*type == ns_t_cname) {

// A cname lookup always returns a single record but we follow the

// common API here.

*type = ns_t_cname;

ret->Set(context,

ret->Length(),

OneByteString(env->isolate(), host->h_name)).FromJust();

ares_free_hostent(host);

return ARES_SUCCESS;

}

if (*type == ns_t_cname_or_a)

*type = ns_t_a;

if (*type == ns_t_ns) {

HostentToNames(env, host, ret);

} else if (*type == ns_t_ptr) {

uint32_t offset = ret->Length();

for (uint32_t i = 0; host->h_aliases[i] != NULL; i++) {

ret->Set(context,

i + offset,

OneByteString(env->isolate(), host->h_aliases[i])).FromJust();

}

} else {

HostentToAddresses(env, host, ret);

}

ares_free_hostent(host);

return ARES_SUCCESS;

}

int ParseMxReply(Environment* env,

const unsigned char* buf,

int len,

Local<Array> ret,

bool need_type = false) {

HandleScope handle_scope(env->isolate());

auto context = env->context();

struct ares_mx_reply* mx_start;

int status = ares_parse_mx_reply(buf, len, &mx_start);

if (status != ARES_SUCCESS) {

return status;

}

Local<String> exchange_symbol = env->exchange_string();

Local<String> priority_symbol = env->priority_string();

Local<String> type_symbol = env->type_string();

Local<String> mx_symbol = env->dns_mx_string();

uint32_t offset = ret->Length();

ares_mx_reply* current = mx_start;

for (uint32_t i = 0; current != nullptr; ++i, current = current->next) {

Local<Object> mx_record = Object::New(env->isolate());

mx_record->Set(context,

exchange_symbol,

OneByteString(env->isolate(), current->host)).FromJust();

mx_record->Set(context,

priority_symbol,

Integer::New(env->isolate(), current->priority)).FromJust();

if (need_type)

mx_record->Set(context, type_symbol, mx_symbol).FromJust();

ret->Set(context, i + offset, mx_record).FromJust();

}

ares_free_data(mx_start);

return ARES_SUCCESS;

}

int ParseTxtReply(Environment* env,

const unsigned char* buf,

int len,

Local<Array> ret,

bool need_type = false) {

HandleScope handle_scope(env->isolate());

auto context = env->context();

struct ares_txt_ext* txt_out;

int status = ares_parse_txt_reply_ext(buf, len, &txt_out);

if (status != ARES_SUCCESS) {

return status;

}

Local<Array> txt_chunk;

struct ares_txt_ext* current = txt_out;

uint32_t i = 0, j;

uint32_t offset = ret->Length();

for (j = 0; current != nullptr; current = current->next) {

Local<String> txt = OneByteString(env->isolate(), current->txt);

// New record found - write out the current chunk

if (current->record_start) {

if (!txt_chunk.IsEmpty()) {

if (need_type) {

Local<Object> elem = Object::New(env->isolate());

elem->Set(context, env->entries_string(), txt_chunk).FromJust();

elem->Set(context,

env->type_string(),

env->dns_txt_string()).FromJust();

ret->Set(context, offset + i++, elem).FromJust();

} else {

ret->Set(context, offset + i++, txt_chunk).FromJust();

}

}

txt_chunk = Array::New(env->isolate());

j = 0;

}

txt_chunk->Set(context, j++, txt).FromJust();

}

// Push last chunk if it isn't empty

if (!txt_chunk.IsEmpty()) {

if (need_type) {

Local<Object> elem = Object::New(env->isolate());

elem->Set(context, env->entries_string(), txt_chunk).FromJust();

elem->Set(context,

env->type_string(),

env->dns_txt_string()).FromJust();

ret->Set(context, offset + i, elem).FromJust();

} else {

ret->Set(context, offset + i, txt_chunk).FromJust();

}

}

ares_free_data(txt_out);

return ARES_SUCCESS;

}

int ParseSrvReply(Environment* env,

const unsigned char* buf,

int len,

Local<Array> ret,

bool need_type = false) {

HandleScope handle_scope(env->isolate());

auto context = env->context();

struct ares_srv_reply* srv_start;

int status = ares_parse_srv_reply(buf, len, &srv_start);

if (status != ARES_SUCCESS) {

return status;

}

Local<String> name_symbol = env->name_string();

Local<String> port_symbol = env->port_string();

Local<String> priority_symbol = env->priority_string();

Local<String> weight_symbol = env->weight_string();

Local<String> type_symbol = env->type_string();

Local<String> srv_symbol = env->dns_srv_string();

ares_srv_reply* current = srv_start;

int offset = ret->Length();

for (uint32_t i = 0; current != nullptr; ++i, current = current->next) {

Local<Object> srv_record = Object::New(env->isolate());

srv_record->Set(context,

name_symbol,

OneByteString(env->isolate(), current->host)).FromJust();

srv_record->Set(context,

port_symbol,

Integer::New(env->isolate(), current->port)).FromJust();

srv_record->Set(context,

priority_symbol,

Integer::New(env->isolate(), current->priority)).FromJust();

srv_record->Set(context,

weight_symbol,

Integer::New(env->isolate(), current->weight)).FromJust();

if (need_type)

srv_record->Set(context, type_symbol, srv_symbol).FromJust();

ret->Set(context, i + offset, srv_record).FromJust();

}

ares_free_data(srv_start);

return ARES_SUCCESS;

}

int ParseNaptrReply(Environment* env,

const unsigned char* buf,

int len,

Local<Array> ret,

bool need_type = false) {

HandleScope handle_scope(env->isolate());

auto context = env->context();

ares_naptr_reply* naptr_start;

int status = ares_parse_naptr_reply(buf, len, &naptr_start);

if (status != ARES_SUCCESS) {

return status;

}

Local<String> flags_symbol = env->flags_string();

Local<String> service_symbol = env->service_string();

Local<String> regexp_symbol = env->regexp_string();

Local<String> replacement_symbol = env->replacement_string();

Local<String> order_symbol = env->order_string();

Local<String> preference_symbol = env->preference_string();

Local<String> type_symbol = env->type_string();

Local<String> naptr_symbol = env->dns_naptr_string();

ares_naptr_reply* current = naptr_start;

int offset = ret->Length();

for (uint32_t i = 0; current != nullptr; ++i, current = current->next) {

Local<Object> naptr_record = Object::New(env->isolate());

naptr_record->Set(context,

flags_symbol,

OneByteString(env->isolate(), current->flags)).FromJust();

naptr_record->Set(context,

service_symbol,

OneByteString(env->isolate(),

current->service)).FromJust();

naptr_record->Set(context,

regexp_symbol,

OneByteString(env->isolate(),

current->regexp)).FromJust();

naptr_record->Set(context,

replacement_symbol,

OneByteString(env->isolate(),

current->replacement)).FromJust();

naptr_record->Set(context,

order_symbol,

Integer::New(env->isolate(), current->order)).FromJust();

naptr_record->Set(context,

preference_symbol,

Integer::New(env->isolate(),

current->preference)).FromJust();

if (need_type)

naptr_record->Set(context, type_symbol, naptr_symbol).FromJust();

ret->Set(context, i + offset, naptr_record).FromJust();

}

ares_free_data(naptr_start);

return ARES_SUCCESS;

}

int ParseSoaReply(Environment* env,

unsigned char* buf,

int len,

Local<Object>* ret) {

EscapableHandleScope handle_scope(env->isolate());

auto context = env->context();

/* Can't use ares_parse_soa_reply() here which can only parse single record */

unsigned int ancount = cares_get_16bit(buf + 6);

unsigned char* ptr = buf + NS_HFIXEDSZ;

int rr_type, rr_len;

char* name;

char* rr_name;

long temp_len; // NOLINT(runtime/int)

int status = ares_expand_name(ptr, buf, len, &name, &temp_len);

if (status != ARES_SUCCESS) {

/* returns EBADRESP in case of invalid input */

return status == ARES_EBADNAME ? ARES_EBADRESP : status;

}

if (ptr + temp_len + NS_QFIXEDSZ > buf + len) {

free(name);

return ARES_EBADRESP;

}

ptr += temp_len + NS_QFIXEDSZ;

for (unsigned int i = 0; i < ancount; i++) {

status = ares_expand_name(ptr, buf, len, &rr_name, &temp_len);

if (status != ARES_SUCCESS)

break;

ptr += temp_len;

if (ptr + NS_RRFIXEDSZ > buf + len) {

free(rr_name);

status = ARES_EBADRESP;

break;

}

rr_type = cares_get_16bit(ptr);

rr_len = cares_get_16bit(ptr + 8);

ptr += NS_RRFIXEDSZ;

/* only need SOA */

if (rr_type == ns_t_soa) {

ares_soa_reply soa;

status = ares_expand_name(ptr, buf, len, &soa.nsname, &temp_len);

if (status != ARES_SUCCESS) {

free(rr_name);

break;

}

ptr += temp_len;

status = ares_expand_name(ptr, buf, len, &soa.hostmaster, &temp_len);

if (status != ARES_SUCCESS) {

free(rr_name);

free(soa.nsname);

break;

}

ptr += temp_len;