mirror of https://github.com/getdnsapi/getdns.git
Added a check for transport
This commit is contained in:
John Dickinson
parent
44736501fb
commit
372c980453
|
|
@ -86,18 +86,18 @@ tests_stub_sync: tests_stub_sync.lo
|
|||
|
||||
check_getdns_common: check_getdns_common.lo
|
||||
$(LIBTOOL) --tag=CC --mode=link $(CC) $(CFLAGS) $(LDFLAGS) $(LDLIBS) -o $@ check_getdns_common.lo
|
||||
|
||||
check_getdns: check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_transport.lo check_getdns_selectloop.lo
|
||||
$(LIBTOOL) --tag=CC --mode=link $(CC) $(CFLAGS) $(LDFLAGS) $(LDLIBS) -lpthread -o $@ check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_transport.lo check_getdns_selectloop.lo
|
||||
|
||||
check_getdns: check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_selectloop.lo
|
||||
$(LIBTOOL) --tag=CC --mode=link $(CC) $(CFLAGS) $(LDFLAGS) $(LDLIBS) -lpthread -o $@ check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_selectloop.lo
|
||||
check_getdns_event: check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_transport.lo check_getdns_libevent.lo
|
||||
$(LIBTOOL) --tag=CC --mode=link $(CC) $(CFLAGS) $(LDFLAGS) -lpthread -lgetdns_ext_event $(EXTENSION_LIBEVENT_LDFLAGS) $(EXTENSION_LIBEVENT_EXT_LIBS) $(LDLIBS) -o $@ check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_transport.lo check_getdns_libevent.lo
|
||||
|
||||
check_getdns_event: check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_libevent.lo
|
||||
$(LIBTOOL) --tag=CC --mode=link $(CC) $(CFLAGS) $(LDFLAGS) -lpthread -lgetdns_ext_event $(EXTENSION_LIBEVENT_LDFLAGS) $(EXTENSION_LIBEVENT_EXT_LIBS) $(LDLIBS) -o $@ check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_libevent.lo
|
||||
check_getdns_uv: check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_transport.lo check_getdns_libuv.lo
|
||||
$(LIBTOOL) --tag=CC --mode=link $(CC) $(CFLAGS) $(LDFLAGS) -lpthread -lgetdns_ext_uv $(EXTENSION_LIBUV_LDFLAGS) $(EXTENSION_LIBUV_EXT_LIBS) $(LDLIBS) -o $@ check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_transport.lo check_getdns_libuv.lo
|
||||
|
||||
check_getdns_uv: check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_libuv.lo
|
||||
$(LIBTOOL) --tag=CC --mode=link $(CC) $(CFLAGS) $(LDFLAGS) -lpthread -lgetdns_ext_uv $(EXTENSION_LIBUV_LDFLAGS) $(EXTENSION_LIBUV_EXT_LIBS) $(LDLIBS) -o $@ check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_libuv.lo
|
||||
|
||||
check_getdns_ev: check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_libev.lo
|
||||
$(LIBTOOL) --tag=CC --mode=link $(CC) $(CFLAGS) $(LDFLAGS) -lpthread -lgetdns_ext_ev $(EXTENSION_LIBEV_LDFLAGS) $(EXTENSION_LIBEV_EXT_LIBS) $(LDLIBS) -o $@ check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_libev.lo
|
||||
check_getdns_ev: check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_transport.lo check_getdns_libev.lo
|
||||
$(LIBTOOL) --tag=CC --mode=link $(CC) $(CFLAGS) $(LDFLAGS) -lpthread -lgetdns_ext_ev $(EXTENSION_LIBEV_LDFLAGS) $(EXTENSION_LIBEV_EXT_LIBS) $(LDLIBS) -o $@ check_getdns.lo check_getdns_common.lo check_getdns_context_set_timeout.lo check_getdns_transport.lo check_getdns_libev.lo
|
||||
|
||||
tests_dnssec: tests_dnssec.lo testmessages.lo
|
||||
$(LIBTOOL) --tag=CC --mode=link $(CC) $(CFLAGS) $(LDFLAGS) $(LDLIBS) -o $@ tests_dnssec.lo testmessages.lo
|
||||
|
|
|
|||
|
|
@ -68,6 +68,7 @@
|
|||
#include "check_getdns_context_set_upstream_recursive_servers.h"
|
||||
#include "check_getdns_service.h"
|
||||
#include "check_getdns_service_sync.h"
|
||||
#include "check_getdns_transport.h"
|
||||
|
||||
|
||||
int
|
||||
|
|
@ -112,6 +113,7 @@ main (int argc, char** argv)
|
|||
Suite *getdns_service_suite(void);
|
||||
Suite *getdns_service_sync_suite(void);
|
||||
Suite *getdns_context_set_timeout_suite(void);
|
||||
Suite *getdns_transport_suite(void);
|
||||
|
||||
sr = srunner_create(getdns_general_suite());
|
||||
srunner_add_suite(sr, getdns_general_sync_suite());
|
||||
|
|
@ -148,7 +150,8 @@ main (int argc, char** argv)
|
|||
srunner_add_suite(sr,getdns_context_set_upstream_recursive_servers_suite());
|
||||
srunner_add_suite(sr,getdns_service_suite());
|
||||
srunner_add_suite(sr,getdns_service_sync_suite());
|
||||
|
||||
srunner_add_suite(sr,getdns_transport_suite());
|
||||
|
||||
srunner_set_log(sr, "check_getdns.log");
|
||||
srunner_run_all(sr, CK_NORMAL);
|
||||
number_failed = srunner_ntests_failed(sr);
|
||||
|
|
|
|||
|
|
@ -0,0 +1,469 @@
|
|||
/*
|
||||
* Copyright (c) 2013, NLNet Labs, Verisign, Inc.
|
||||
* All rights reserved.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions are met:
|
||||
* * Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
* * Redistributions in binary form must reproduce the above copyright
|
||||
* notice, this list of conditions and the following disclaimer in the
|
||||
* documentation and/or other materials provided with the distribution.
|
||||
* * Neither the names of the copyright holders nor the
|
||||
* names of its contributors may be used to endorse or promote products
|
||||
* derived from this software without specific prior written permission.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||||
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||
* DISCLAIMED. IN NO EVENT SHALL Verisign, Inc. BE LIABLE FOR ANY
|
||||
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||||
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||||
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
||||
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
|
||||
#include "check_getdns_transport.h"
|
||||
#include "check_getdns_common.h"
|
||||
#include <pthread.h>
|
||||
#include <unistd.h>
|
||||
#include <sys/socket.h>
|
||||
#include <netinet/in.h>
|
||||
#include <stdio.h>
|
||||
#include <ldns/ldns.h>
|
||||
#include <sys/param.h>
|
||||
|
||||
|
||||
#define GETDNS_STR_IPV4 "IPv4"
|
||||
#define GETDNS_STR_IPV6 "IPv6"
|
||||
#define GETDNS_STR_ADDRESS_TYPE "address_type"
|
||||
#define GETDNS_STR_ADDRESS_DATA "address_data"
|
||||
#define GETDNS_STR_PORT "port"
|
||||
#define TEST_PORT 43210
|
||||
|
||||
/* utilities to start a junk listener */
|
||||
typedef struct transport_thread_data {
|
||||
uint16_t port;
|
||||
volatile int running;
|
||||
int udp_count;
|
||||
int tcp_count;
|
||||
} transport_thread_data;
|
||||
|
||||
void* run_transport_server(void* data) {
|
||||
transport_thread_data* tdata = (transport_thread_data*) data;
|
||||
int udp, tcp, conn = 0;
|
||||
struct sockaddr_in serv_addr;
|
||||
uint8_t mesg[65536], tcplength[2];
|
||||
fd_set read_fds;
|
||||
ldns_rdf* answerfrom;
|
||||
ldns_resolver* resolver;
|
||||
int udp_count = 0;
|
||||
int tcp_count = 0;
|
||||
ldns_resolver_new_frm_file(&resolver, NULL);
|
||||
|
||||
udp = socket(AF_INET, SOCK_DGRAM, 0);
|
||||
tcp = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
|
||||
|
||||
memset(&serv_addr, 0, sizeof (serv_addr));
|
||||
serv_addr.sin_family = AF_INET;
|
||||
serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
|
||||
serv_addr.sin_port = htons(tdata->port);
|
||||
bind(udp, (struct sockaddr *) &serv_addr, sizeof (serv_addr));
|
||||
bind(tcp, (struct sockaddr *) &serv_addr, sizeof (serv_addr));
|
||||
listen(tcp, 5);
|
||||
/* signal that it's listening */
|
||||
/* dirty timing hack to yield */
|
||||
sleep(1);
|
||||
tdata->running = 1;
|
||||
while (tdata->running) {
|
||||
struct sockaddr_in client_addr;
|
||||
FD_ZERO(&read_fds);
|
||||
FD_SET(udp, &read_fds);
|
||||
FD_SET(tcp, &read_fds);
|
||||
ldns_pkt* pkt;
|
||||
int n = 0;
|
||||
struct timeval tv;
|
||||
tv.tv_sec = 1;
|
||||
tv.tv_usec = 0;
|
||||
int maxfdp1 = MAX(udp, tcp) + 1;
|
||||
int r = select(maxfdp1, &read_fds, NULL, NULL, &tv);
|
||||
if (r > 0) {
|
||||
ldns_pkt* query;
|
||||
socklen_t len = sizeof (client_addr);
|
||||
if (FD_ISSET(udp, &read_fds) == 1) {
|
||||
n = recvfrom(udp, mesg, 65536, 0, (struct sockaddr *) &client_addr, &len);
|
||||
udp_count++;
|
||||
} else if (FD_ISSET(tcp, &read_fds) == 1) {
|
||||
conn = accept(tcp, (struct sockaddr *) &client_addr, &len);
|
||||
/* throw away the length */
|
||||
n = read(conn, tcplength, 2);
|
||||
n = read(conn, mesg, 65536);
|
||||
tcp_count++;
|
||||
}
|
||||
ldns_wire2pkt(&query, mesg, n);
|
||||
ldns_resolver_send_pkt(&pkt, resolver, query);
|
||||
ldns_str2rdf_a(&answerfrom, "127.0.0.1");
|
||||
ldns_pkt_set_answerfrom(pkt, answerfrom);
|
||||
ldns_pkt_free(query);
|
||||
uint8_t* pkt_data;
|
||||
size_t pkt_len;
|
||||
ldns_pkt* answer = pkt;
|
||||
ldns_pkt2wire(&pkt_data, answer, &pkt_len);
|
||||
if (udp_count > 0) {
|
||||
sendto(udp, pkt_data, pkt_len, 0, (struct sockaddr *) &client_addr, sizeof (client_addr));
|
||||
} else if (conn > 0) {
|
||||
int wcount = write(conn, pkt_data, pkt_len);
|
||||
if (wcount != pkt_len) {
|
||||
/* For now ignore this */
|
||||
continue;
|
||||
}
|
||||
}
|
||||
free(pkt_data);
|
||||
ldns_pkt_free(answer);
|
||||
} /* End of if */
|
||||
} /* end of while loop */
|
||||
close(udp);
|
||||
close(tcp);
|
||||
if (conn > 0) close(conn);
|
||||
ldns_resolver_deep_free(resolver);
|
||||
tdata->udp_count = udp_count;
|
||||
tdata->tcp_count = tcp_count;
|
||||
return NULL;
|
||||
|
||||
}
|
||||
|
||||
void transport_cb(struct getdns_context *context,
|
||||
getdns_callback_type_t callback_type,
|
||||
struct getdns_dict * response,
|
||||
void *userarg, getdns_transaction_t transaction_id) {
|
||||
/* Don't really care about the answer*/
|
||||
return;
|
||||
}
|
||||
|
||||
START_TEST(getdns_transport_udp_sync) {
|
||||
/*
|
||||
* Create a context by calling getdns_context_create()
|
||||
* Create listener thread
|
||||
* Set upstream to localhost:port
|
||||
*
|
||||
* getdns_context_set_resolution_type() to GETDNS_RESOLUTION_STUB
|
||||
* expect: GETDNS_CONTEXT_CODE_RESOLUTION_TYPE
|
||||
*/
|
||||
|
||||
struct getdns_context *context = NULL;
|
||||
struct getdns_dict* server_dict;
|
||||
getdns_dict* response = NULL;
|
||||
struct getdns_list* upstream_list;
|
||||
struct getdns_bindata bindata;
|
||||
uint32_t local_addr = htonl(0x7F000001);
|
||||
pthread_t thread;
|
||||
transport_thread_data t_data;
|
||||
t_data.running = 0;
|
||||
t_data.udp_count = 0;
|
||||
t_data.tcp_count = 0;
|
||||
t_data.port = TEST_PORT;
|
||||
|
||||
pthread_create(&thread, NULL, run_transport_server, (void *) &t_data);
|
||||
|
||||
while (!t_data.running) {
|
||||
sleep(1);
|
||||
}
|
||||
|
||||
/* set up */
|
||||
CONTEXT_CREATE(TRUE);
|
||||
server_dict = getdns_dict_create_with_context(context);
|
||||
ck_assert_msg(server_dict != NULL, "Allocate IP dictionary failed");
|
||||
bindata.size = strlen(GETDNS_STR_IPV4) + 1;
|
||||
bindata.data = (uint8_t*) GETDNS_STR_IPV4;
|
||||
ASSERT_RC(getdns_dict_set_bindata(server_dict, GETDNS_STR_ADDRESS_TYPE, &bindata),
|
||||
GETDNS_RETURN_GOOD, "set ip bindata");
|
||||
bindata.size = 4;
|
||||
bindata.data = (uint8_t*) & local_addr;
|
||||
ASSERT_RC(getdns_dict_set_bindata(server_dict, GETDNS_STR_ADDRESS_DATA, &bindata),
|
||||
GETDNS_RETURN_GOOD, "set addr bindata");
|
||||
ASSERT_RC(getdns_dict_set_int(server_dict, GETDNS_STR_PORT, t_data.port),
|
||||
GETDNS_RETURN_GOOD, "set addr port");
|
||||
|
||||
upstream_list = getdns_list_create_with_context(context);
|
||||
ck_assert_msg(upstream_list != NULL, "Allocate lists");
|
||||
|
||||
ASSERT_RC(getdns_list_set_dict(upstream_list, 0, server_dict),
|
||||
GETDNS_RETURN_GOOD, "set upstream");
|
||||
|
||||
ASSERT_RC(getdns_context_set_upstream_recursive_servers(context, upstream_list),
|
||||
GETDNS_RETURN_GOOD, "set rec servers");
|
||||
|
||||
ASSERT_RC(getdns_context_set_dns_transport(context, GETDNS_TRANSPORT_UDP_ONLY),
|
||||
GETDNS_RETURN_GOOD, "Bad return code from setting udp transport");
|
||||
|
||||
/* stub */
|
||||
ASSERT_RC(getdns_context_set_resolution_type(context, GETDNS_RESOLUTION_STUB),
|
||||
GETDNS_RETURN_GOOD, "Return code from getdns_context_set_resolution_type()");
|
||||
|
||||
ASSERT_RC(getdns_general_sync(context, "getdnsapi.net", GETDNS_RRTYPE_A, NULL, &response),
|
||||
GETDNS_RETURN_GOOD, "Bad return code from getdns_general_sync");
|
||||
|
||||
CONTEXT_DESTROY;
|
||||
|
||||
t_data.running = 0;
|
||||
pthread_join(thread, NULL);
|
||||
ck_assert_msg(t_data.udp_count == 1, "udp_count != 1");
|
||||
ck_assert_msg(t_data.tcp_count == 0, "tcp_count != 0");
|
||||
|
||||
}
|
||||
|
||||
END_TEST
|
||||
|
||||
START_TEST(getdns_transport_tcp_sync) {
|
||||
/*
|
||||
* Create a context by calling getdns_context_create()
|
||||
* Create listener thread
|
||||
* Set upstream to localhost:port
|
||||
*
|
||||
* getdns_context_set_resolution_type() to GETDNS_RESOLUTION_STUB
|
||||
* expect: GETDNS_CONTEXT_CODE_RESOLUTION_TYPE
|
||||
*/
|
||||
|
||||
struct getdns_context *context = NULL;
|
||||
struct getdns_dict* server_dict;
|
||||
getdns_dict* response = NULL;
|
||||
struct getdns_list* upstream_list;
|
||||
struct getdns_bindata bindata;
|
||||
uint32_t local_addr = htonl(0x7F000001);
|
||||
pthread_t thread;
|
||||
transport_thread_data t_data;
|
||||
t_data.running = 0;
|
||||
t_data.udp_count = 0;
|
||||
t_data.tcp_count = 0;
|
||||
t_data.port = TEST_PORT;
|
||||
|
||||
pthread_create(&thread, NULL, run_transport_server, (void *) &t_data);
|
||||
|
||||
while (!t_data.running) {
|
||||
sleep(1);
|
||||
}
|
||||
|
||||
/* set up */
|
||||
CONTEXT_CREATE(TRUE);
|
||||
server_dict = getdns_dict_create_with_context(context);
|
||||
ck_assert_msg(server_dict != NULL, "Allocate IP dictionary failed");
|
||||
bindata.size = strlen(GETDNS_STR_IPV4) + 1;
|
||||
bindata.data = (uint8_t*) GETDNS_STR_IPV4;
|
||||
ASSERT_RC(getdns_dict_set_bindata(server_dict, GETDNS_STR_ADDRESS_TYPE, &bindata),
|
||||
GETDNS_RETURN_GOOD, "set ip bindata");
|
||||
bindata.size = 4;
|
||||
bindata.data = (uint8_t*) & local_addr;
|
||||
ASSERT_RC(getdns_dict_set_bindata(server_dict, GETDNS_STR_ADDRESS_DATA, &bindata),
|
||||
GETDNS_RETURN_GOOD, "set addr bindata");
|
||||
ASSERT_RC(getdns_dict_set_int(server_dict, GETDNS_STR_PORT, t_data.port),
|
||||
GETDNS_RETURN_GOOD, "set addr port");
|
||||
|
||||
upstream_list = getdns_list_create_with_context(context);
|
||||
ck_assert_msg(upstream_list != NULL, "Allocate lists");
|
||||
|
||||
ASSERT_RC(getdns_list_set_dict(upstream_list, 0, server_dict),
|
||||
GETDNS_RETURN_GOOD, "set upstream");
|
||||
|
||||
ASSERT_RC(getdns_context_set_upstream_recursive_servers(context, upstream_list),
|
||||
GETDNS_RETURN_GOOD, "set rec servers");
|
||||
|
||||
ASSERT_RC(getdns_context_set_dns_transport(context, GETDNS_TRANSPORT_TCP_ONLY),
|
||||
GETDNS_RETURN_GOOD, "Bad return code from setting tcp transport");
|
||||
|
||||
/* stub */
|
||||
ASSERT_RC(getdns_context_set_resolution_type(context, GETDNS_RESOLUTION_STUB),
|
||||
GETDNS_RETURN_GOOD, "Return code from getdns_context_set_resolution_type()");
|
||||
|
||||
ASSERT_RC(getdns_general_sync(context, "getdnsapi.net", GETDNS_RRTYPE_A, NULL, &response),
|
||||
GETDNS_RETURN_GOOD, "Bad return code from getdns_general_sync");
|
||||
|
||||
CONTEXT_DESTROY;
|
||||
|
||||
t_data.running = 0;
|
||||
pthread_join(thread, NULL);
|
||||
ck_assert_msg(t_data.udp_count == 0, "udp_count != 0");
|
||||
ck_assert_msg(t_data.tcp_count == 1, "tcp_count != 1");
|
||||
|
||||
}
|
||||
|
||||
END_TEST
|
||||
|
||||
START_TEST(getdns_transport_udp_async) {
|
||||
/*
|
||||
* Create a context by calling getdns_context_create()
|
||||
* Create listener thread
|
||||
* Set upstream to localhost:port
|
||||
*
|
||||
* getdns_context_set_resolution_type() to GETDNS_RESOLUTION_STUB
|
||||
* expect: GETDNS_CONTEXT_CODE_RESOLUTION_TYPE
|
||||
*/
|
||||
|
||||
struct getdns_context *context = NULL;
|
||||
void* eventloop = NULL;
|
||||
struct getdns_dict* server_dict;
|
||||
struct getdns_list* upstream_list;
|
||||
struct getdns_bindata bindata;
|
||||
uint32_t local_addr = htonl(0x7F000001);
|
||||
pthread_t thread;
|
||||
transport_thread_data t_data;
|
||||
t_data.running = 0;
|
||||
t_data.udp_count = 0;
|
||||
t_data.tcp_count = 0;
|
||||
t_data.port = TEST_PORT;
|
||||
|
||||
pthread_create(&thread, NULL, run_transport_server, (void *) &t_data);
|
||||
|
||||
while (!t_data.running) {
|
||||
sleep(1);
|
||||
}
|
||||
|
||||
/* set up */
|
||||
CONTEXT_CREATE(TRUE);
|
||||
server_dict = getdns_dict_create_with_context(context);
|
||||
ck_assert_msg(server_dict != NULL, "Allocate IP dictionary failed");
|
||||
bindata.size = strlen(GETDNS_STR_IPV4) + 1;
|
||||
bindata.data = (uint8_t*) GETDNS_STR_IPV4;
|
||||
ASSERT_RC(getdns_dict_set_bindata(server_dict, GETDNS_STR_ADDRESS_TYPE, &bindata),
|
||||
GETDNS_RETURN_GOOD, "set ip bindata");
|
||||
bindata.size = 4;
|
||||
bindata.data = (uint8_t*) & local_addr;
|
||||
ASSERT_RC(getdns_dict_set_bindata(server_dict, GETDNS_STR_ADDRESS_DATA, &bindata),
|
||||
GETDNS_RETURN_GOOD, "set addr bindata");
|
||||
ASSERT_RC(getdns_dict_set_int(server_dict, GETDNS_STR_PORT, t_data.port),
|
||||
GETDNS_RETURN_GOOD, "set addr port");
|
||||
|
||||
upstream_list = getdns_list_create_with_context(context);
|
||||
ck_assert_msg(upstream_list != NULL, "Allocate lists");
|
||||
|
||||
ASSERT_RC(getdns_list_set_dict(upstream_list, 0, server_dict),
|
||||
GETDNS_RETURN_GOOD, "set upstream");
|
||||
|
||||
ASSERT_RC(getdns_context_set_upstream_recursive_servers(context, upstream_list),
|
||||
GETDNS_RETURN_GOOD, "set rec servers");
|
||||
|
||||
ASSERT_RC(getdns_context_set_dns_transport(context, GETDNS_TRANSPORT_UDP_ONLY),
|
||||
GETDNS_RETURN_GOOD, "Bad return code from setting udp transport");
|
||||
|
||||
/* stub */
|
||||
ASSERT_RC(getdns_context_set_resolution_type(context, GETDNS_RESOLUTION_STUB),
|
||||
GETDNS_RETURN_GOOD, "Return code from getdns_context_set_resolution_type()");
|
||||
|
||||
EVENT_BASE_CREATE;
|
||||
|
||||
ASSERT_RC(getdns_general(context, "getdnsapi.net", GETDNS_RRTYPE_A, NULL, &t_data, NULL, transport_cb),
|
||||
GETDNS_RETURN_GOOD, "Bad return code from getdns_general_sync");
|
||||
|
||||
RUN_EVENT_LOOP;
|
||||
|
||||
CONTEXT_DESTROY;
|
||||
|
||||
t_data.running = 0;
|
||||
pthread_join(thread, NULL);
|
||||
ck_assert_msg(t_data.udp_count == 1, "udp_count != 1");
|
||||
ck_assert_msg(t_data.tcp_count == 0, "tcp_count != 0");
|
||||
|
||||
}
|
||||
|
||||
END_TEST
|
||||
|
||||
START_TEST(getdns_transport_tcp_async) {
|
||||
/*
|
||||
* Create a context by calling getdns_context_create()
|
||||
* Create listener thread
|
||||
* Set upstream to localhost:port
|
||||
*
|
||||
* getdns_context_set_resolution_type() to GETDNS_RESOLUTION_STUB
|
||||
* expect: GETDNS_CONTEXT_CODE_RESOLUTION_TYPE
|
||||
*/
|
||||
|
||||
struct getdns_context *context = NULL;
|
||||
void* eventloop = NULL;
|
||||
struct getdns_dict* server_dict;
|
||||
struct getdns_list* upstream_list;
|
||||
struct getdns_bindata bindata;
|
||||
uint32_t local_addr = htonl(0x7F000001);
|
||||
pthread_t thread;
|
||||
transport_thread_data t_data;
|
||||
t_data.running = 0;
|
||||
t_data.udp_count = 0;
|
||||
t_data.tcp_count = 0;
|
||||
t_data.port = TEST_PORT;
|
||||
|
||||
pthread_create(&thread, NULL, run_transport_server, (void *) &t_data);
|
||||
|
||||
while (!t_data.running) {
|
||||
sleep(1);
|
||||
}
|
||||
|
||||
/* set up */
|
||||
CONTEXT_CREATE(TRUE);
|
||||
server_dict = getdns_dict_create_with_context(context);
|
||||
ck_assert_msg(server_dict != NULL, "Allocate IP dictionary failed");
|
||||
bindata.size = strlen(GETDNS_STR_IPV4) + 1;
|
||||
bindata.data = (uint8_t*) GETDNS_STR_IPV4;
|
||||
ASSERT_RC(getdns_dict_set_bindata(server_dict, GETDNS_STR_ADDRESS_TYPE, &bindata),
|
||||
GETDNS_RETURN_GOOD, "set ip bindata");
|
||||
bindata.size = 4;
|
||||
bindata.data = (uint8_t*) & local_addr;
|
||||
ASSERT_RC(getdns_dict_set_bindata(server_dict, GETDNS_STR_ADDRESS_DATA, &bindata),
|
||||
GETDNS_RETURN_GOOD, "set addr bindata");
|
||||
ASSERT_RC(getdns_dict_set_int(server_dict, GETDNS_STR_PORT, t_data.port),
|
||||
GETDNS_RETURN_GOOD, "set addr port");
|
||||
|
||||
upstream_list = getdns_list_create_with_context(context);
|
||||
ck_assert_msg(upstream_list != NULL, "Allocate lists");
|
||||
|
||||
ASSERT_RC(getdns_list_set_dict(upstream_list, 0, server_dict),
|
||||
GETDNS_RETURN_GOOD, "set upstream");
|
||||
|
||||
ASSERT_RC(getdns_context_set_upstream_recursive_servers(context, upstream_list),
|
||||
GETDNS_RETURN_GOOD, "set rec servers");
|
||||
|
||||
ASSERT_RC(getdns_context_set_dns_transport(context, GETDNS_TRANSPORT_TCP_ONLY),
|
||||
GETDNS_RETURN_GOOD, "Bad return code from setting tcp transport");
|
||||
|
||||
/* stub */
|
||||
ASSERT_RC(getdns_context_set_resolution_type(context, GETDNS_RESOLUTION_STUB),
|
||||
GETDNS_RETURN_GOOD, "Return code from getdns_context_set_resolution_type()");
|
||||
|
||||
EVENT_BASE_CREATE;
|
||||
|
||||
ASSERT_RC(getdns_general(context, "getdnsapi.net", GETDNS_RRTYPE_A, NULL, &t_data, NULL, transport_cb),
|
||||
GETDNS_RETURN_GOOD, "Bad return code from getdns_general_sync");
|
||||
|
||||
RUN_EVENT_LOOP;
|
||||
|
||||
CONTEXT_DESTROY;
|
||||
|
||||
t_data.running = 0;
|
||||
pthread_join(thread, NULL);
|
||||
ck_assert_msg(t_data.udp_count == 0, "udp_count != 0");
|
||||
ck_assert_msg(t_data.tcp_count == 1, "tcp_count != 1");
|
||||
|
||||
}
|
||||
|
||||
END_TEST
|
||||
|
||||
|
||||
|
||||
|
||||
Suite *
|
||||
getdns_transport_suite(void) {
|
||||
Suite *s = suite_create("getdns_transport()");
|
||||
|
||||
/* Positive test cases */
|
||||
TCase *tc_pos = tcase_create("Positive");
|
||||
tcase_set_timeout(tc_pos, 15.0);
|
||||
tcase_add_test(tc_pos, getdns_transport_udp_sync);
|
||||
tcase_add_test(tc_pos, getdns_transport_tcp_sync);
|
||||
tcase_add_test(tc_pos, getdns_transport_udp_async);
|
||||
tcase_add_test(tc_pos, getdns_transport_tcp_async);
|
||||
suite_add_tcase(s, tc_pos);
|
||||
|
||||
return s;
|
||||
|
||||
}
|
||||
|
||||
|
|
@ -0,0 +1,35 @@
|
|||
/*
|
||||
* Copyright (c) 2013, NLNet Labs, Verisign, Inc.
|
||||
* All rights reserved.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions are met:
|
||||
* * Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
* * Redistributions in binary form must reproduce the above copyright
|
||||
* notice, this list of conditions and the following disclaimer in the
|
||||
* documentation and/or other materials provided with the distribution.
|
||||
* * Neither the names of the copyright holders nor the
|
||||
* names of its contributors may be used to endorse or promote products
|
||||
* derived from this software without specific prior written permission.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||||
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||
* DISCLAIMED. IN NO EVENT SHALL Verisign, Inc. BE LIABLE FOR ANY
|
||||
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||||
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||||
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
||||
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
#ifndef _check_getdns_transport_h_
|
||||
#define _check_getdns_transport_h_
|
||||
|
||||
#include <check.h>
|
||||
|
||||
Suite *
|
||||
getdns_transport_suite (void);
|
||||
|
||||
#endif
|
||||
Loading…
Reference in New Issue