mirror of https://github.com/getdnsapi/getdns.git
Add timeout test.
This commit is contained in:
Neel Goyal
parent
0419c4673c
commit
f6760fc94c
1148
src/context.c
1148
src/context.c
File diff suppressed because it is too large
Load Diff
|
|
@ -45,8 +45,8 @@ tests_stub_sync: tests_stub_sync.o
|
|||
check_getdns_common: check_getdns_common.o
|
||||
$(LIBTOOL) --tag=CC --mode=link $(CC) $(CFLAGS) $(LDFLAGS) $(LDLIBS) -o $@ check_getdns_common.o
|
||||
|
||||
check_getdns: check_getdns.o check_getdns_common.o check_getdns_libev.o
|
||||
$(LIBTOOL) --tag=CC --mode=link $(CC) $(CFLAGS) $(LDFLAGS) $(LDLIBS) -o $@ check_getdns.o check_getdns_common.o check_getdns_libev.o
|
||||
check_getdns: check_getdns.o check_getdns_common.o check_getdns_libev.o check_getdns_context_set_timeout.o
|
||||
$(LIBTOOL) --tag=CC --mode=link $(CC) $(CFLAGS) $(LDFLAGS) $(LDLIBS) -o $@ check_getdns.o check_getdns_common.o check_getdns_libev.o check_getdns_context_set_timeout.o
|
||||
|
||||
tests_dnssec: tests_dnssec.o testmessages.o
|
||||
$(LIBTOOL) --tag=CC --mode=link $(CC) $(CFLAGS) $(LDFLAGS) $(LDLIBS) -o $@ tests_dnssec.o testmessages.o
|
||||
|
|
|
|||
|
|
@ -38,6 +38,7 @@
|
|||
#include "check_getdns_pretty_print_dict.h"
|
||||
#include "check_getdns_display_ip_address.h"
|
||||
#include "check_getdns_context_set_context_update_callback.h"
|
||||
#include "check_getdns_context_set_timeout.h"
|
||||
|
||||
|
||||
int
|
||||
|
|
@ -46,39 +47,6 @@ main (int argc, char** argv)
|
|||
int number_failed;
|
||||
SRunner *sr ;
|
||||
|
||||
Suite *getdns_general_suite(void);
|
||||
Suite *getdns_general_sync_suite(void);
|
||||
Suite *getdns_address_suite(void);
|
||||
Suite *getdns_address_sync_suite(void);
|
||||
Suite *getdns_hostname_suite(void);
|
||||
Suite *getdns_hostname_sync_suite(void);
|
||||
Suite *getdns_context_create_suite(void);
|
||||
Suite *getdns_context_destroy_suite(void);
|
||||
Suite *getdns_cancel_callback_suite(void);
|
||||
Suite *getdns_list_get_length_suite(void);
|
||||
Suite *getdns_list_get_dict_suite(void);
|
||||
Suite *getdns_list_get_list_suite(void);
|
||||
Suite *getdns_list_get_int_suite(void);
|
||||
Suite *getdns_list_get_data_type_suite(void);
|
||||
Suite *getdns_list_get_bindata_suite(void);
|
||||
Suite *getdns_dict_get_names_suite(void);
|
||||
Suite *getdns_dict_get_data_type_suite(void);
|
||||
Suite *getdns_dict_get_dict_suite(void);
|
||||
Suite *getdns_dict_get_list_suite(void);
|
||||
Suite *getdns_dict_get_dict_suite(void);
|
||||
Suite *getdns_dict_get_bindata_suite(void);
|
||||
Suite *getdns_dict_get_int_suite(void);
|
||||
Suite *getdns_dict_destroy_suite(void);
|
||||
Suite *getdns_dict_set_dict_suite(void);
|
||||
Suite *getdns_dict_set_list_suite(void);
|
||||
Suite *getdns_dict_set_bindata_suite(void);
|
||||
Suite *getdns_dict_set_int_suite(void);
|
||||
Suite *getdns_convert_ulabel_to_alabel_suite(void);
|
||||
Suite *getdns_convert_alabel_to_ulabel_suite(void);
|
||||
Suite *getdns_pretty_print_dict_suite(void);
|
||||
Suite *getdns_display_ip_address_suite(void);
|
||||
Suite *getdns_context_set_context_update_callback_suite(void);
|
||||
|
||||
sr = srunner_create(getdns_general_suite());
|
||||
srunner_add_suite(sr, getdns_general_sync_suite());
|
||||
srunner_add_suite(sr, getdns_address_suite());
|
||||
|
|
@ -110,6 +78,7 @@ main (int argc, char** argv)
|
|||
srunner_add_suite(sr, getdns_pretty_print_dict_suite());
|
||||
srunner_add_suite(sr,getdns_display_ip_address_suite());
|
||||
srunner_add_suite(sr,getdns_context_set_context_update_callback_suite());
|
||||
srunner_add_suite(sr,getdns_context_set_timeout_suite());
|
||||
|
||||
srunner_set_log(sr, "check_getdns.log");
|
||||
srunner_run_all(sr, CK_NORMAL);
|
||||
|
|
|
|||
|
|
@ -1,7 +1,7 @@
|
|||
#ifndef _check_getdns_common_h_
|
||||
#define _check_getdns_common_h_
|
||||
|
||||
|
||||
#include <getdns/getdns.h>
|
||||
|
||||
#define TRUE 1
|
||||
#define FALSE 0
|
||||
|
|
|
|||
|
|
@ -0,0 +1,282 @@
|
|||
/*
|
||||
* Copyright (c) 2013, NLNet Labs, Versign, 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 name of the <organization> 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_context_set_timeout.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>
|
||||
|
||||
|
||||
/*
|
||||
**************************************************************************
|
||||
* *
|
||||
* T E S T S F O R G E T D N S _ C O N T E X T _ S E T _ TIMEOUT *
|
||||
* *
|
||||
**************************************************************************
|
||||
*/
|
||||
|
||||
START_TEST (getdns_context_set_timeout_1)
|
||||
{
|
||||
/*
|
||||
* context is NULL
|
||||
* expect: GETDNS_RETURN_INVALID_PARAMETER
|
||||
*/
|
||||
|
||||
struct getdns_context *context = NULL;
|
||||
|
||||
ASSERT_RC(getdns_context_set_timeout(context, 1000),
|
||||
GETDNS_RETURN_INVALID_PARAMETER, "Return code from getdns_context_set_timeout()");
|
||||
|
||||
}
|
||||
END_TEST
|
||||
|
||||
START_TEST (getdns_context_set_timeout_2)
|
||||
{
|
||||
/*
|
||||
* timeout is 0
|
||||
* expect: GETDNS_RETURN_INVALID_PARAMETER
|
||||
*/
|
||||
|
||||
struct getdns_context *context = NULL;
|
||||
CONTEXT_CREATE(TRUE);
|
||||
|
||||
ASSERT_RC(getdns_context_set_timeout(context, 0),
|
||||
GETDNS_RETURN_INVALID_PARAMETER, "Return code from getdns_context_set_timeout()");
|
||||
|
||||
CONTEXT_DESTROY;
|
||||
|
||||
}
|
||||
END_TEST
|
||||
|
||||
#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"
|
||||
|
||||
/* utilities to start a junk udp listener */
|
||||
typedef struct timeout_thread_data {
|
||||
uint16_t port;
|
||||
volatile int running;
|
||||
int num_callbacks;
|
||||
int num_timeouts;
|
||||
} timeout_thread_data;
|
||||
|
||||
typedef struct queued_response {
|
||||
struct sockaddr_in client_addr;
|
||||
ldns_pkt* pkt;
|
||||
} queued_response;
|
||||
|
||||
void* run_server(void* data) {
|
||||
timeout_thread_data* tdata = (timeout_thread_data*)data;
|
||||
int fd;
|
||||
struct sockaddr_in serv_addr;
|
||||
uint8_t mesg[65536];
|
||||
fd_set read_fds;
|
||||
ldns_rdf* answerfrom;
|
||||
ldns_resolver* resolver;
|
||||
int num_received = 0;
|
||||
queued_response responses[10];
|
||||
|
||||
ldns_resolver_new_frm_file(&resolver, NULL);
|
||||
|
||||
|
||||
fd=socket(AF_INET,SOCK_DGRAM,0);
|
||||
|
||||
bzero(&serv_addr,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(fd,(struct sockaddr *)&serv_addr,sizeof(serv_addr));
|
||||
|
||||
/* signal that it's listening */
|
||||
/* dirty timing hack to yield */
|
||||
sleep(1);
|
||||
tdata->running = 1;
|
||||
/* queue up query responses to send out, and delay sending them
|
||||
* for a second */
|
||||
while (tdata->running) {
|
||||
struct sockaddr_in client_addr;
|
||||
FD_ZERO(&read_fds);
|
||||
FD_SET(fd, &read_fds);
|
||||
struct timeval tv;
|
||||
tv.tv_sec = 1;
|
||||
tv.tv_usec = 0;
|
||||
int r = select(fd + 1, &read_fds, NULL, NULL, &tv);
|
||||
if (r > 0 && num_received < 10) {
|
||||
ldns_pkt* query;
|
||||
socklen_t len = sizeof(client_addr);
|
||||
int n = recvfrom(fd,mesg,65536,0,(struct sockaddr *)&(responses[num_received].client_addr),&len);
|
||||
ldns_wire2pkt(&query, mesg, n);
|
||||
ldns_resolver_send_pkt(&(responses[num_received].pkt), resolver, query);
|
||||
ldns_str2rdf_a(&answerfrom, "127.0.0.1");
|
||||
ldns_pkt_set_answerfrom(responses[num_received].pkt, answerfrom);
|
||||
ldns_pkt_free(query);
|
||||
++num_received;
|
||||
} else if (r == 0 && num_received > 0) {
|
||||
int i = 0;
|
||||
/* timeout - see if we have anything to send */
|
||||
for (i = 0; i < num_received; ++i) {
|
||||
uint8_t* pkt_data;
|
||||
size_t pkt_len;
|
||||
ldns_pkt* answer = responses[i].pkt;
|
||||
ldns_pkt2wire(&pkt_data, answer, &pkt_len);
|
||||
sendto(fd,pkt_data,pkt_len,0,(struct sockaddr *)&(responses[i].client_addr),sizeof(client_addr));
|
||||
free(pkt_data);
|
||||
ldns_pkt_free(answer);
|
||||
}
|
||||
num_received = 0;
|
||||
}
|
||||
}
|
||||
ldns_resolver_deep_free(resolver);
|
||||
|
||||
return NULL;
|
||||
|
||||
}
|
||||
|
||||
void timeout_3_cb(struct getdns_context *context,
|
||||
uint16_t callback_type,
|
||||
struct getdns_dict * response,
|
||||
void *userarg, getdns_transaction_t transaction_id) {
|
||||
timeout_thread_data *tdata = (timeout_thread_data*)userarg;
|
||||
tdata->num_callbacks++;
|
||||
if (callback_type == GETDNS_CALLBACK_TIMEOUT) {
|
||||
tdata->num_timeouts++;
|
||||
}
|
||||
if (tdata->num_callbacks == 1) {
|
||||
/* set timeout to 2 seconds and then issue request */
|
||||
getdns_context_set_timeout(context, 500);
|
||||
getdns_general(context, "getdnsapi.org", GETDNS_RRTYPE_A, NULL,
|
||||
tdata, NULL, timeout_3_cb);
|
||||
}
|
||||
}
|
||||
|
||||
START_TEST (getdns_context_set_timeout_3)
|
||||
{
|
||||
/*
|
||||
* Create a context by calling getdns_context_create()
|
||||
* Create listener thread
|
||||
* Set upstream to localhost:port
|
||||
*
|
||||
* getdns_context_set_resolution_type() to GETDNS_CONTEXT_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;
|
||||
|
||||
timeout_thread_data t_data;
|
||||
t_data.running = 0;
|
||||
t_data.num_callbacks = 0;
|
||||
t_data.num_timeouts = 0;
|
||||
t_data.port = 43210;
|
||||
|
||||
pthread_create(&thread, NULL, run_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, 43210),
|
||||
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");
|
||||
|
||||
/* stub */
|
||||
ASSERT_RC(getdns_context_set_resolution_type(context, GETDNS_CONTEXT_STUB),
|
||||
GETDNS_RETURN_GOOD, "Return code from getdns_context_set_resolution_type()");
|
||||
|
||||
EVENT_BASE_CREATE;
|
||||
|
||||
getdns_general(context, "getdnsapi.net", GETDNS_RRTYPE_A, NULL,
|
||||
&t_data, NULL, timeout_3_cb);
|
||||
|
||||
RUN_EVENT_LOOP;
|
||||
|
||||
CONTEXT_DESTROY;
|
||||
|
||||
t_data.running = 0;
|
||||
pthread_join(thread, NULL);
|
||||
ck_assert_msg(t_data.num_callbacks == 2, "callbacks != 2");
|
||||
ck_assert_msg(t_data.num_timeouts == 1, "timeouts != 1");
|
||||
|
||||
}
|
||||
END_TEST
|
||||
|
||||
|
||||
|
||||
|
||||
Suite *
|
||||
getdns_context_set_timeout_suite (void)
|
||||
{
|
||||
Suite *s = suite_create ("getdns_context_set_timeout()");
|
||||
|
||||
/* Negative test caseis */
|
||||
TCase *tc_neg = tcase_create("Negative");
|
||||
tcase_add_test(tc_neg, getdns_context_set_timeout_1);
|
||||
tcase_add_test(tc_neg, getdns_context_set_timeout_2);
|
||||
suite_add_tcase(s, tc_neg);
|
||||
|
||||
/* Positive test cases */
|
||||
TCase *tc_pos = tcase_create("Positive");
|
||||
tcase_set_timeout(tc_pos, 15.0);
|
||||
tcase_add_test(tc_pos, getdns_context_set_timeout_3);
|
||||
suite_add_tcase(s, tc_pos);
|
||||
|
||||
return s;
|
||||
|
||||
}
|
||||
|
||||
|
|
@ -0,0 +1,9 @@
|
|||
#ifndef _check_getdns_context_set_timeout_h_
|
||||
#define _check_getdns_context_set_timeout_h_
|
||||
|
||||
#include <check.h>
|
||||
|
||||
Suite *
|
||||
getdns_context_set_timeout_suite (void);
|
||||
|
||||
#endif
|
||||
Loading…
Reference in New Issue