/* ares_dup() duplicates a channel handle with all its options and returns a
new channel handle */
int ares_dup(ares_channel *dest, ares_channel src)
{
struct ares_options opts;
struct ares_addr_node *servers;
int ipv6_nservers = 0;
int i, rc;
int optmask;
*dest = NULL; /* in case of failure return NULL explicitly */
/* First get the options supported by the old ares_save_options() function,
which is most of them */
rc = ares_save_options(src, &opts, &optmask);
if(rc)
return rc;
/* Then create the new channel with those options */
rc = ares_init_options(dest, &opts, optmask);
/* destroy the options copy to not leak any memory */
ares_destroy_options(&opts);
if(rc)
return rc;
/* Now clone the options that ares_save_options() doesn't support. */
(*dest)->sock_create_cb = src->sock_create_cb;
(*dest)->sock_create_cb_data = src->sock_create_cb_data;
strncpy((*dest)->local_dev_name, src->local_dev_name, sizeof(src->local_dev_name));
(*dest)->local_ip4 = src->local_ip4;
memcpy((*dest)->local_ip6, src->local_ip6, sizeof(src->local_ip6));
/* Full name server cloning required when not all are IPv4 */
for (i = 0; i < src->nservers; i++)
{
if (src->servers[i].addr.family != AF_INET) {
ipv6_nservers++;
break;
}
}
if (ipv6_nservers) {
rc = ares_get_servers(src, &servers);
if (rc != ARES_SUCCESS)
return rc;
rc = ares_set_servers(*dest, servers);
ares_free_data(servers);
if (rc != ARES_SUCCESS)
return rc;
}
return ARES_SUCCESS; /* everything went fine */
}
static PyObject *
Channel_servers_get(Channel *self, void *closure)
{
int r;
char ip[INET6_ADDRSTRLEN];
struct ares_addr_node *server, *servers;
PyObject *server_list;
PyObject *tmp;
UNUSED_ARG(closure);
if (!self->channel) {
PyErr_SetString(PyExc_AresError, "Channel has already been destroyed");
return NULL;
}
server_list = PyList_New(0);
if (!server_list) {
PyErr_NoMemory();
return NULL;
}
r = ares_get_servers(self->channel, &servers);
if (r != ARES_SUCCESS) {
RAISE_ARES_EXCEPTION(r);
return NULL;
}
for (server = servers; server != NULL; server = server->next) {
if (server->family == AF_INET) {
ares_inet_ntop(AF_INET, &(server->addr.addr4), ip, INET_ADDRSTRLEN);
tmp = Py_BuildValue("s", ip);
} else {
ares_inet_ntop(AF_INET6, &(server->addr.addr6), ip, INET6_ADDRSTRLEN);
tmp = Py_BuildValue("s", ip);
}
if (tmp == NULL) {
break;
}
r = PyList_Append(server_list, tmp);
Py_DECREF(tmp);
if (r != 0) {
break;
}
}
return server_list;
}
static PyObject *
DNSResolver_servers_get(DNSResolver *self, void *closure)
{
int r;
char ip[INET6_ADDRSTRLEN];
struct ares_addr_node *server, *servers;
PyObject *server_list;
PyObject *tmp;
UNUSED_ARG(closure);
server_list = PyList_New(0);
if (!server_list) {
PyErr_NoMemory();
return NULL;
}
r = ares_get_servers(self->channel, &servers);
if (r != 0) {
PyErr_SetString(PyExc_DNSError, "error getting c-ares nameservers");
return NULL;
}
for (server = servers; server != NULL; server = server->next) {
if (server->family == AF_INET) {
uv_inet_ntop(AF_INET, &(server->addr.addr4), ip, INET_ADDRSTRLEN);
tmp = PyString_FromString(ip);
} else {
uv_inet_ntop(AF_INET6, &(server->addr.addr6), ip, INET6_ADDRSTRLEN);
tmp = PyString_FromString(ip);
}
if (tmp == NULL) {
break;
}
r = PyList_Append(server_list, tmp);
Py_DECREF(tmp);
if (r != 0) {
break;
}
}
return server_list;
}
