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

#include <vector>

#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 ChannelWrap;

struct node_ares_task {

ChannelWrap* channel;

ares_socket_t sock;

uv_poll_t poll_watcher;

RB_ENTRY(node_ares_task) node;

};

RB_HEAD(node_ares_task_list, node_ares_task);

class ChannelWrap : public AsyncWrap {

public:

ChannelWrap(Environment* env, Local<Object> object);

~ChannelWrap();

static void New(const FunctionCallbackInfo<Value>& args);

void Setup();

void EnsureServers();

void CleanupTimer();

inline uv_timer_t* timer_handle() { return timer_handle_; }

inline ares_channel cares_channel() { return channel_; }

inline bool query_last_ok() const { return query_last_ok_; }

inline void set_query_last_ok(bool ok) { query_last_ok_ = ok; }

inline bool is_servers_default() const { return is_servers_default_; }

inline void set_is_servers_default(bool is_default) {

is_servers_default_ = is_default;

}

inline node_ares_task_list* task_list() { return &task_list_; }

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

static void AresTimeout(uv_timer_t* handle);

private:

uv_timer_t* timer_handle_;

ares_channel channel_;

bool query_last_ok_;

bool is_servers_default_;

bool library_inited_;

node_ares_task_list task_list_;

};

ChannelWrap::ChannelWrap(Environment* env,

Local<Object> object)

: AsyncWrap(env, object, PROVIDER_DNSCHANNEL),

timer_handle_(nullptr),

channel_(nullptr),

query_last_ok_(true),

is_servers_default_(true),

library_inited_(false) {

RB_INIT(&task_list_);

MakeWeak<ChannelWrap>(this);

Setup();

}

void ChannelWrap::New(const FunctionCallbackInfo<Value>& args) {

CHECK(args.IsConstructCall());

CHECK_EQ(args.Length(), 0);

Environment* env = Environment::GetCurrent(args);

new ChannelWrap(env, args.This());

}

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 ChannelWrap::AresTimeout(uv_timer_t* handle) {

ChannelWrap* channel = static_cast<ChannelWrap*>(handle->data);

CHECK_EQ(channel->timer_handle(), handle);

CHECK_EQ(false, RB_EMPTY(channel->task_list()));

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

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_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(ChannelWrap* channel, ares_socket_t sock) {

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

if (task == nullptr) {

/* Out of memory. */

return nullptr;

}

task->channel = channel;

task->sock = sock;

if (uv_poll_init_socket(channel->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) {

ChannelWrap* channel = static_cast<ChannelWrap*>(data);

node_ares_task* task;

node_ares_task lookup_task;

lookup_task.sock = sock;

task = RB_FIND(node_ares_task_list, channel->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 = channel->timer_handle();

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

CHECK(RB_EMPTY(channel->task_list()));

uv_timer_start(timer_handle, ChannelWrap::AresTimeout, 1000, 1000);

}

task = ares_task_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;

}

RB_INSERT(node_ares_task_list, channel->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, channel->task_list(), task);

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

ares_poll_close_cb);

if (RB_EMPTY(channel->task_list())) {

uv_timer_stop(channel->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 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;

int r;

if (!library_inited_) {

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

r = ares_init_options(&channel_,

&options,

ARES_OPT_FLAGS | ARES_OPT_SOCK_STATE_CB);

if (r != ARES_SUCCESS) {

ares_library_cleanup();

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

}

library_inited_ = true;

/* Initialize the timeout timer. The timer won't be started until the */

/* first socket is opened. */

CleanupTimer();

timer_handle_ = new uv_timer_t();

timer_handle_->data = static_cast<void*>(this);

uv_timer_init(env()->event_loop(), timer_handle_);

}

ChannelWrap::~ChannelWrap() {

if (library_inited_)

ares_library_cleanup();

ares_destroy(channel_);

CleanupTimer();

}

void ChannelWrap::CleanupTimer() {

if (timer_handle_ == nullptr) return;

uv_close(reinterpret_cast<uv_handle_t*>(timer_handle_),

[](uv_handle_t* handle) {

delete reinterpret_cast<uv_timer_t*>(handle);

});

timer_handle_ = nullptr;

}

/**

* 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;

}

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

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

is_servers_default_ = false;

return;

}

ares_free_data(servers);

servers = nullptr;

/* destroy channel and reset channel */

ares_destroy(channel_);

Setup();

}

class QueryWrap : public AsyncWrap {

public:

QueryWrap(ChannelWrap* channel, Local<Object> req_wrap_obj)

: AsyncWrap(channel->env(), req_wrap_obj, AsyncWrap::PROVIDER_QUERYWRAP),

channel_(channel) {

if (env()->in_domain()) {

req_wrap_obj->Set(env()->domain_string(),

env()->domain_array()->Get(env()->context(), 0)

.ToLocalChecked());

}

Wrap(req_wrap_obj, this);

// Make sure the channel object stays alive during the query lifetime.

req_wrap_obj->Set(env()->context(),

env()->channel_string(),

channel->object()).FromJust();

}

~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 AresQuery(const char* name,

int dnsclass,

int type) {

channel_->EnsureServers();

ares_query(channel_->cares_channel(), name, dnsclass, type, Callback,

static_cast<void*>(this));

}

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

}

ChannelWrap* channel_;

};

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