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

EscapableHandleScope scope(env->isolate());

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

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(i, address);

}

return scope.Escape(addresses);

}

Local<Array> HostentToNames(Environment* env, struct hostent* host) {

EscapableHandleScope scope(env->isolate());

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

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

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

names->Set(i, address);

}

return 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_node* servers = nullptr;

ares_get_servers(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)) {

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

}

};

class QueryAWrap: public QueryWrap {

public:

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

: QueryWrap(env, req_wrap_obj) {

}

int Send(const char* name) override {

AresQuery(env(), name, ns_c_in, ns_t_a, Callback, GetQueryArg());

return 0;

}

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

protected:

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

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

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

hostent* host;

ares_addrttl addrttls[256];

int naddrttls = arraysize(addrttls);

int status = ares_parse_a_reply(buf, len, &host, addrttls, &naddrttls);

if (status != ARES_SUCCESS) {

ParseError(status);

return;

}

Local<Array> addresses = HostentToAddresses(env(), host);

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

auto context = env()->context();

for (int i = 0; i < naddrttls; i += 1) {

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

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

}

ares_free_hostent(host);

CallOnComplete(addresses, ttls);

}

};

class QueryAaaaWrap: public QueryWrap {

public:

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

: QueryWrap(env, req_wrap_obj) {

}

int Send(const char* name) override {

AresQuery(env(), name, ns_c_in, ns_t_aaaa, Callback, GetQueryArg());

return 0;

}

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

protected:

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

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

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

hostent* host;

ares_addr6ttl addrttls[256];

int naddrttls = arraysize(addrttls);

int status = ares_parse_aaaa_reply(buf, len, &host, addrttls, &naddrttls);

if (status != ARES_SUCCESS) {

ParseError(status);

return;

}

Local<Array> addresses = HostentToAddresses(env(), host);

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

auto context = env()->context();

for (int i = 0; i < naddrttls; i += 1) {

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

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

}

ares_free_hostent(host);

CallOnComplete(addresses, ttls);

}

};

class QueryCnameWrap: public QueryWrap {

public:

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

: QueryWrap(env, req_wrap_obj) {

}

int Send(const char* name) override {

AresQuery(env(), name, ns_c_in, ns_t_cname, Callback, GetQueryArg());

return 0;

}

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

protected:

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

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

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

struct hostent* host;

int status = ares_parse_a_reply(buf, len, &host, nullptr, nullptr);

if (status != ARES_SUCCESS) {

ParseError(status);

return;

}

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

// common API here.

Local<Array> result = Array::New(env()->isolate(), 1);

result->Set(0, OneByteString(env()->isolate(), host->h_name));

ares_free_hostent(host);

this->CallOnComplete(result);

}

};

class QueryMxWrap: public QueryWrap {

public:

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

: QueryWrap(env, req_wrap_obj) {

}

int Send(const char* name) override {

AresQuery(env(), name, ns_c_in, ns_t_mx, Callback, GetQueryArg());

return 0;

}

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

protected:

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

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

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

struct ares_mx_reply* mx_start;

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

if (status != ARES_SUCCESS) {

ParseError(status);

return;

}

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

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

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

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(exchange_symbol,

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

mx_record->Set(priority_symbol,

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

mx_records->Set(i, mx_record);

}

ares_free_data(mx_start);

this->CallOnComplete(mx_records);

}

};

class QueryNsWrap: public QueryWrap {

public:

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

: QueryWrap(env, req_wrap_obj) {

}

int Send(const char* name) override {

AresQuery(env(), name, ns_c_in, ns_t_ns, Callback, GetQueryArg());

return 0;

}

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

protected:

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

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

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

struct hostent* host;

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

if (status != ARES_SUCCESS) {

ParseError(status);

return;

}

Local<Array> names = HostentToNames(env(), host);

ares_free_hostent(host);

this->CallOnComplete(names);

}

};

class QueryTxtWrap: public QueryWrap {

public:

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

: QueryWrap(env, req_wrap_obj) {

}

int Send(const char* name) override {

AresQuery(env(), name, ns_c_in, ns_t_txt, Callback, GetQueryArg());

return 0;

}

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

protected:

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

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

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

struct ares_txt_ext* txt_out;

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

if (status != ARES_SUCCESS) {

ParseError(status);

return;

}

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

Local<Array> txt_chunk;

struct ares_txt_ext* current = txt_out;

uint32_t i = 0;

for (uint32_t 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())

txt_records->Set(i++, txt_chunk);

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

j = 0;

}

txt_chunk->Set(j++, txt);

}

// Push last chunk if it isn't empty

if (!txt_chunk.IsEmpty())

txt_records->Set(i, txt_chunk);

ares_free_data(txt_out);

this->CallOnComplete(txt_records);

}

};

class QuerySrvWrap: public QueryWrap {

public:

explicit QuerySrvWrap(Environment* env, Local<Object> req_wrap_obj)

: QueryWrap(env, req_wrap_obj) {

}

int Send(const char* name) override {

AresQuery(env(), name, ns_c_in, ns_t_srv, Callback, GetQueryArg());

return 0;

}

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

protected:

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

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

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

struct ares_srv_reply* srv_start;

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

if (status != ARES_SUCCESS) {

ParseError(status);

return;

}

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

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

ares_srv_reply* current = srv_start;

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

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

srv_record->Set(name_symbol,

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

srv_record->Set(port_symbol,

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

srv_record->Set(priority_symbol,

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

srv_record->Set(weight_symbol,

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

srv_records->Set(i, srv_record);

}

ares_free_data(srv_start);

this->CallOnComplete(srv_records);

}

};

class QueryPtrWrap: public QueryWrap {

public:

explicit QueryPtrWrap(Environment* env, Local<Object> req_wrap_obj)

: QueryWrap(env, req_wrap_obj) {

}

int Send(const char* name) override {

AresQuery(env(), name, ns_c_in, ns_t_ptr, Callback, GetQueryArg());

return 0;

}

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

protected:

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

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

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

struct hostent* host;

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

if (status != ARES_SUCCESS) {

ParseError(status);

return;

}

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

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

aliases->Set(i, OneByteString(env()->isolate(), host->h_aliases[i]));

}

ares_free_hostent(host);

this->CallOnComplete(aliases);

}

};

class QueryNaptrWrap: public QueryWrap {

public:

explicit QueryNaptrWrap(Environment* env, Local<Object> req_wrap_obj)

: QueryWrap(env, req_wrap_obj) {

}

int Send(const char* name) override {

AresQuery(env(), name, ns_c_in, ns_t_naptr, Callback, GetQueryArg());

return 0;

}

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

protected:

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

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

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

ares_naptr_reply* naptr_start;

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

if (status != ARES_SUCCESS) {

ParseError(status);

return;

}

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

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

ares_naptr_reply* current = naptr_start;

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

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

naptr_record->Set(flags_symbol,

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

naptr_record->Set(service_symbol,

OneByteString(env()->isolate(), current->service));

naptr_record->Set(regexp_symbol,

OneByteString(env()->isolate(), current->regexp));

naptr_record->Set(replacement_symbol,

OneByteString(env()->isolate(), current->replacement));

naptr_record->Set(order_symbol,

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

naptr_record->Set(preference_symbol,

Integer::New(env()->isolate(), current->preference));

naptr_records->Set(i, naptr_record);

}

ares_free_data(naptr_start);