128 lines
3.3 KiB
C
128 lines
3.3 KiB
C
|
|
/* Copyright 1998 by the Massachusetts Institute of Technology.
|
|
* Copyright (C) 2008-2010 by Daniel Stenberg
|
|
*
|
|
* Permission to use, copy, modify, and distribute this
|
|
* software and its documentation for any purpose and without
|
|
* fee is hereby granted, provided that the above copyright
|
|
* notice appear in all copies and that both that copyright
|
|
* notice and this permission notice appear in supporting
|
|
* documentation, and that the name of M.I.T. not be used in
|
|
* advertising or publicity pertaining to distribution of the
|
|
* software without specific, written prior permission.
|
|
* M.I.T. makes no representations about the suitability of
|
|
* this software for any purpose. It is provided "as is"
|
|
* without express or implied warranty.
|
|
*/
|
|
|
|
|
|
#include "ares_setup.h"
|
|
|
|
#include "ares.h"
|
|
#include "ares_data.h"
|
|
#include "ares_private.h"
|
|
|
|
|
|
int ares_get_servers(ares_channel channel,
|
|
struct ares_addr_node **servers)
|
|
{
|
|
struct ares_addr_node *srvr_head = NULL;
|
|
struct ares_addr_node *srvr_last = NULL;
|
|
struct ares_addr_node *srvr_curr;
|
|
int status = ARES_SUCCESS;
|
|
int i;
|
|
|
|
if (!channel)
|
|
return ARES_ENODATA;
|
|
|
|
for (i = 0; i < channel->nservers; i++)
|
|
{
|
|
/* Allocate storage for this server node appending it to the list */
|
|
srvr_curr = ares_malloc_data(ARES_DATATYPE_ADDR_NODE);
|
|
if (!srvr_curr)
|
|
{
|
|
status = ARES_ENOMEM;
|
|
break;
|
|
}
|
|
if (srvr_last)
|
|
{
|
|
srvr_last->next = srvr_curr;
|
|
}
|
|
else
|
|
{
|
|
srvr_head = srvr_curr;
|
|
}
|
|
srvr_last = srvr_curr;
|
|
|
|
/* Fill this server node data */
|
|
srvr_curr->family = channel->servers[i].addr.family;
|
|
if (srvr_curr->family == AF_INET)
|
|
memcpy(&srvr_curr->addrV4, &channel->servers[i].addr.addrV4,
|
|
sizeof(srvr_curr->addrV4));
|
|
else
|
|
memcpy(&srvr_curr->addrV6, &channel->servers[i].addr.addrV6,
|
|
sizeof(srvr_curr->addrV6));
|
|
}
|
|
|
|
if (status != ARES_SUCCESS)
|
|
{
|
|
if (srvr_head)
|
|
{
|
|
ares_free_data(srvr_head);
|
|
srvr_head = NULL;
|
|
}
|
|
}
|
|
|
|
*servers = srvr_head;
|
|
|
|
return status;
|
|
}
|
|
|
|
|
|
int ares_set_servers(ares_channel channel,
|
|
struct ares_addr_node *servers)
|
|
{
|
|
struct ares_addr_node *srvr;
|
|
int num_srvrs = 0;
|
|
int i;
|
|
|
|
if (ares_library_initialized() != ARES_SUCCESS)
|
|
return ARES_ENOTINITIALIZED;
|
|
|
|
if (!channel)
|
|
return ARES_ENODATA;
|
|
|
|
ares__destroy_servers_state(channel);
|
|
|
|
for (srvr = servers; srvr; srvr = srvr->next)
|
|
{
|
|
num_srvrs++;
|
|
}
|
|
|
|
if (num_srvrs > 0)
|
|
{
|
|
/* Allocate storage for servers state */
|
|
channel->servers = malloc(num_srvrs * sizeof(struct server_state));
|
|
if (!channel->servers)
|
|
{
|
|
return ARES_ENOMEM;
|
|
}
|
|
channel->nservers = num_srvrs;
|
|
/* Fill servers state address data */
|
|
for (i = 0, srvr = servers; srvr; i++, srvr = srvr->next)
|
|
{
|
|
channel->servers[i].addr.family = srvr->family;
|
|
if (srvr->family == AF_INET)
|
|
memcpy(&channel->servers[i].addr.addrV4, &srvr->addrV4,
|
|
sizeof(srvr->addrV4));
|
|
else
|
|
memcpy(&channel->servers[i].addr.addrV6, &srvr->addrV6,
|
|
sizeof(srvr->addrV6));
|
|
}
|
|
/* Initialize servers state remaining data */
|
|
ares__init_servers_state(channel);
|
|
}
|
|
|
|
return ARES_SUCCESS;
|
|
}
|