forked from Mirrors/freeswitch
816f984722
* sres_cache: added sres_cache_copy_answers() git-svn-id: http://svn.freeswitch.org/svn/freeswitch/trunk@11949 d0543943-73ff-0310-b7d9-9358b9ac24b2
456 lines
12 KiB
C
456 lines
12 KiB
C
/*
|
|
* This file is part of the Sofia-SIP package
|
|
*
|
|
* Copyright (C) 2006 Nokia Corporation.
|
|
*
|
|
* Contact: Pekka Pessi <pekka.pessi@nokia.com>
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public License
|
|
* as published by the Free Software Foundation; either version 2.1 of
|
|
* the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful, but
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
|
|
* 02110-1301 USA
|
|
*
|
|
*/
|
|
|
|
/**@internal
|
|
*
|
|
* @CFILE torture_sresolv.c Torture tests for Sofia resolver.
|
|
*
|
|
* @author Mikko Haataja
|
|
* @author Pekka Pessi <Pekka.Pessi@nokia.com>.
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#if HAVE_STDINT_H
|
|
#include <stdint.h>
|
|
#elif HAVE_INTTYPES_H
|
|
#include <inttypes.h>
|
|
#else
|
|
#if defined(_WIN32)
|
|
typedef unsigned _int8 uint8_t;
|
|
typedef unsigned _int16 uint16_t;
|
|
typedef unsigned _int32 uint32_t;
|
|
#endif
|
|
#endif
|
|
|
|
#if HAVE_FCNTL_H
|
|
#include <fcntl.h>
|
|
#endif
|
|
|
|
#if HAVE_NETINET_IN_H
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
#endif
|
|
|
|
#if HAVE_WINSOCK2_H
|
|
#include <winsock2.h>
|
|
#include <ws2tcpip.h>
|
|
#endif
|
|
|
|
#include <sofia-resolv/sres.h>
|
|
#include <sofia-resolv/sres_async.h>
|
|
#include <sofia-resolv/sres_record.h>
|
|
#include <sofia-resolv/sres_cache.h>
|
|
|
|
#include <sofia-sip/su_alloc.h>
|
|
|
|
#include <time.h>
|
|
#include <assert.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#include <errno.h>
|
|
#include <unistd.h>
|
|
|
|
#if HAVE_ALARM
|
|
#include <signal.h>
|
|
#endif
|
|
|
|
#define TSTFLAGS tstflags
|
|
int tstflags, o_timing;
|
|
|
|
#include <sofia-sip/tstdef.h>
|
|
|
|
char const name[] = "torture_sresolv";
|
|
|
|
struct sres_context_s
|
|
{
|
|
su_home_t home[1];
|
|
};
|
|
|
|
static void test_answer(sres_context_t *ctx,
|
|
sres_query_t *q,
|
|
sres_record_t **answer)
|
|
{
|
|
}
|
|
|
|
static char name2048[2049] =
|
|
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
|
|
"bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb"
|
|
"cccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccc"
|
|
"dddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd"
|
|
"eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee"
|
|
"ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff"
|
|
"gggggggggggggggggggggggggggggggggggggggggggggggggggggggggggggggg"
|
|
"hhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhh"
|
|
|
|
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
|
|
"bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb"
|
|
"cccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccc"
|
|
"dddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd"
|
|
"eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee"
|
|
"ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff"
|
|
"gggggggggggggggggggggggggggggggggggggggggggggggggggggggggggggggg"
|
|
"hhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhh"
|
|
|
|
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
|
|
"bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb"
|
|
"cccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccc"
|
|
"dddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd"
|
|
"eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee"
|
|
"ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff"
|
|
"gggggggggggggggggggggggggggggggggggggggggggggggggggggggggggggggg"
|
|
"hhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhh"
|
|
|
|
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
|
|
"bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb"
|
|
"cccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccc"
|
|
"dddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd"
|
|
"eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee"
|
|
"ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff"
|
|
"gggggggggggggggggggggggggggggggggggggggggggggggggggggggggggggggg"
|
|
"hhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhh";
|
|
|
|
/* Test API function argument validation */
|
|
static
|
|
int test_api_errors(void)
|
|
{
|
|
sres_context_t ctx[1];
|
|
sres_resolver_t *res;
|
|
int s, fd;
|
|
int sockets[20];
|
|
struct sockaddr sa[1] = {{ 0 }};
|
|
char *template = NULL;
|
|
FILE *f;
|
|
|
|
BEGIN();
|
|
|
|
memset(ctx, 0, sizeof ctx);
|
|
|
|
template = su_sprintf(ctx->home, ".torture_sresolv_api.conf.XXXXXX");
|
|
TEST_1(template);
|
|
|
|
TEST_1(res = sres_resolver_new(NULL));
|
|
TEST(su_home_threadsafe((su_home_t *)res), 0);
|
|
TEST_VOID(sres_resolver_unref(res));
|
|
|
|
#ifndef _WIN32
|
|
fd = mkstemp(template); TEST_1(fd != -1);
|
|
#else
|
|
fd = open(template, O_WRONLY); TEST_1(fd != -1);
|
|
#endif
|
|
|
|
f = fdopen(fd, "w"); TEST_1(f);
|
|
fprintf(f, "domain example.com\n");
|
|
fclose(f);
|
|
|
|
/* Test also LOCALDOMAIN handling */
|
|
putenv("LOCALDOMAIN=localdomain");
|
|
|
|
TEST_1(res = sres_resolver_new(template));
|
|
TEST(su_home_threadsafe((su_home_t *)res), 0);
|
|
|
|
unlink(template);
|
|
|
|
s = sockets[0];
|
|
|
|
TEST_P(sres_resolver_ref(NULL), NULL);
|
|
TEST(errno, EFAULT);
|
|
sres_resolver_unref(NULL);
|
|
|
|
TEST_P(sres_resolver_set_userdata(NULL, NULL), NULL);
|
|
TEST(errno, EFAULT);
|
|
|
|
TEST_P(sres_resolver_get_userdata(NULL), NULL);
|
|
|
|
TEST_P(sres_resolver_get_userdata(res), NULL);
|
|
TEST_P(sres_resolver_set_userdata(res, sa), NULL);
|
|
TEST_P(sres_resolver_get_userdata(res), sa);
|
|
TEST_P(sres_resolver_set_userdata(res, NULL), sa);
|
|
TEST_P(sres_resolver_get_userdata(res), NULL);
|
|
|
|
errno = 0;
|
|
TEST_P(sres_query(NULL, test_answer, ctx, sres_type_a, "com"), NULL);
|
|
TEST(errno, EFAULT); errno = 0;
|
|
TEST_P(sres_query(res, test_answer, ctx, sres_type_a, NULL), NULL);
|
|
TEST(errno, EFAULT); errno = 0;
|
|
TEST_P(sres_query_sockaddr(res, test_answer, ctx,
|
|
sres_qtype_any, sa), NULL);
|
|
TEST(errno, EAFNOSUPPORT); errno = 0;
|
|
|
|
TEST_P(sres_cached_answers(NULL, sres_qtype_any, "example.com"), NULL);
|
|
TEST(errno, EFAULT); errno = 0;
|
|
TEST_P(sres_cached_answers(res, sres_qtype_any, NULL), NULL);
|
|
TEST(errno, EFAULT); errno = 0;
|
|
TEST_P(sres_cached_answers(res, sres_qtype_any, name2048), NULL);
|
|
TEST(errno, ENAMETOOLONG); errno = 0;
|
|
TEST_P(sres_cached_answers_sockaddr(res, sres_qtype_any, sa), NULL);
|
|
TEST(errno, EAFNOSUPPORT); errno = 0;
|
|
|
|
sres_free_answer(res, NULL);
|
|
sres_free_answers(res, NULL);
|
|
sres_sort_answers(res, NULL);
|
|
|
|
sres_free_answer(NULL, NULL);
|
|
sres_free_answers(NULL, NULL);
|
|
sres_sort_answers(NULL, NULL);
|
|
|
|
sres_resolver_unref(res);
|
|
|
|
END();
|
|
}
|
|
|
|
extern void sres_cache_clean(sres_cache_t *cache, time_t now);
|
|
|
|
#ifndef CLOCK_PROCESS_CPUTIME_ID
|
|
#define CLOCK_PROCESS_CPUTIME_ID CLOCK_REALTIME
|
|
#endif
|
|
|
|
static
|
|
int test_cache(void)
|
|
{
|
|
BEGIN();
|
|
|
|
sres_a_record_t *a, a0[1], **all;
|
|
sres_record_t **copy;
|
|
char host[128];
|
|
sres_cache_t *cache;
|
|
time_t now, base;
|
|
struct timespec t0, t1, t2;
|
|
|
|
size_t i, N, N1 = 1000, N3 = 1000000;
|
|
|
|
time(&base);
|
|
|
|
cache = sres_cache_new(N1);
|
|
TEST_1(cache);
|
|
|
|
all = calloc(N3, sizeof *all); if (!all) perror("calloc"), exit(2);
|
|
|
|
memset(a0, 0, sizeof a0);
|
|
|
|
a0->a_record->r_refcount = 1;
|
|
a0->a_record->r_size = sizeof *a;
|
|
a0->a_record->r_type = sres_type_a;
|
|
a0->a_record->r_class = sres_class_in;
|
|
a0->a_record->r_ttl = 3600;
|
|
a0->a_record->r_rdlen = sizeof a->a_addr;
|
|
a0->a_record->r_parsed = 1;
|
|
|
|
for (i = 0, N = N3; i < N; i++) {
|
|
a0->a_record->r_name = host;
|
|
|
|
snprintf(host, sizeof host, "%u.example.com.", (unsigned)i);
|
|
|
|
a = (sres_a_record_t *)
|
|
sres_cache_alloc_record(cache, (sres_record_t *)a0, 0);
|
|
|
|
if (!a)
|
|
perror("sres_cache_alloc_record"), exit(2);
|
|
|
|
all[i] = a, a->a_record->r_refcount = 1;
|
|
}
|
|
|
|
clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &t0);
|
|
|
|
for (i = 0, N = N3; i < N; i++) {
|
|
now = base + (3600 * i + N / 2) / N;
|
|
a->a_record->r_ttl = 60 + (i * 60) % 3600;
|
|
sres_cache_store(cache, (sres_record_t *)all[i], now);
|
|
}
|
|
|
|
clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &t1);
|
|
if (o_timing) {
|
|
t2.tv_sec = t1.tv_sec - t0.tv_sec, t2.tv_nsec = t1.tv_nsec - t0.tv_nsec;
|
|
if (t1.tv_nsec < t0.tv_nsec)
|
|
t2.tv_sec--, t2.tv_nsec += 1000000000;
|
|
printf("sres_cache: stored %u entries: %lu.%09lu sec\n",
|
|
(unsigned)N, (long unsigned)t2.tv_sec, t2.tv_nsec);
|
|
}
|
|
|
|
for (i = 0, N; i < N; i++)
|
|
TEST(all[i]->a_record->r_refcount, 2);
|
|
|
|
TEST_1(copy = sres_cache_copy_answers(cache, (sres_record_t **)all));
|
|
|
|
for (i = 0, N; i < N; i++)
|
|
TEST(all[i]->a_record->r_refcount, 3);
|
|
|
|
clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &t0);
|
|
|
|
for (now = base; now <= base + 3660; now += 30)
|
|
sres_cache_clean(cache, now + 3600);
|
|
|
|
clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &t1);
|
|
if (o_timing) {
|
|
t2.tv_sec = t1.tv_sec - t0.tv_sec, t2.tv_nsec = t1.tv_nsec - t0.tv_nsec;
|
|
if (t1.tv_nsec < t0.tv_nsec)
|
|
t2.tv_sec--, t2.tv_nsec += 1000000000;
|
|
printf("sres_cache: cleaned %u entries: %lu.%09lu sec\n",
|
|
(unsigned)N, (long unsigned)t2.tv_sec, t2.tv_nsec);
|
|
}
|
|
|
|
for (i = 0, N; i < N; i++)
|
|
TEST(all[i]->a_record->r_refcount, 2);
|
|
|
|
sres_cache_free_answers(cache, copy), copy = NULL;
|
|
|
|
for (i = 0, N; i < N; i++)
|
|
TEST(all[i]->a_record->r_refcount, 1);
|
|
|
|
base += 24 * 3600;
|
|
|
|
clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &t0);
|
|
|
|
for (i = 0, N; i < N; i++) {
|
|
now = base + (3600 * i + N / 2) / N;
|
|
a->a_record->r_ttl = 60 + (i * 60) % 3600;
|
|
sres_cache_store(cache, (sres_record_t *)all[i], now);
|
|
}
|
|
|
|
clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &t1);
|
|
if (o_timing) {
|
|
t2.tv_sec = t1.tv_sec - t0.tv_sec, t2.tv_nsec = t1.tv_nsec - t0.tv_nsec;
|
|
if (t1.tv_nsec < t0.tv_nsec)
|
|
t2.tv_sec--, t2.tv_nsec += 1000000000;
|
|
printf("sres_cache: stored %u entries: %lu.%09lu sec\n",
|
|
(unsigned)N, (long unsigned)t2.tv_sec, t2.tv_nsec);
|
|
}
|
|
|
|
for (i = 0, N; i < N; i++)
|
|
TEST(all[i]->a_record->r_refcount, 2);
|
|
|
|
clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &t0);
|
|
|
|
for (now = base; now <= base + 3660; now += 1)
|
|
sres_cache_clean(cache, now + 3600);
|
|
|
|
clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &t1);
|
|
|
|
if (o_timing) {
|
|
t2.tv_sec = t1.tv_sec - t0.tv_sec, t2.tv_nsec = t1.tv_nsec - t0.tv_nsec;
|
|
if (t1.tv_nsec < t0.tv_nsec)
|
|
t2.tv_sec--, t2.tv_nsec += 1000000000;
|
|
printf("sres_cache: cleaned %u entries: %lu.%09lu sec\n",
|
|
(unsigned)N, (long unsigned)t2.tv_sec, t2.tv_nsec);
|
|
}
|
|
|
|
for (i = 0, N; i < N; i++) {
|
|
TEST(all[i]->a_record->r_refcount, 1);
|
|
sres_cache_free_one(cache, (sres_record_t *)all[i]);
|
|
}
|
|
|
|
sres_cache_unref(cache);
|
|
|
|
free(all);
|
|
|
|
END();
|
|
}
|
|
|
|
|
|
#if HAVE_ALARM
|
|
static RETSIGTYPE sig_alarm(int s)
|
|
{
|
|
fprintf(stderr, "%s: FAIL! test timeout!\n", name);
|
|
exit(1);
|
|
}
|
|
#endif
|
|
|
|
void usage(int exitcode)
|
|
{
|
|
fprintf(stderr,
|
|
"usage: %s OPTIONS [-]\n"
|
|
"\twhere OPTIONS are\n"
|
|
"\t -v be verbose\n"
|
|
"\t -a abort on error\n"
|
|
"\t -t show timing\n"
|
|
"\r --no-alarm disable timeout\n"
|
|
"\t -llevel set debugging level\n",
|
|
name);
|
|
exit(exitcode);
|
|
}
|
|
|
|
#include <sofia-sip/su_log.h>
|
|
|
|
extern su_log_t sresolv_log[];
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int i;
|
|
int error = 0;
|
|
int o_alarm = 1;
|
|
|
|
for (i = 1; argv[i]; i++) {
|
|
if (argv[i][0] != '-')
|
|
break;
|
|
else if (strcmp(argv[i], "-") == 0) {
|
|
i++; break;
|
|
}
|
|
else if (strcmp(argv[i], "-v") == 0)
|
|
tstflags |= tst_verbatim;
|
|
else if (strcmp(argv[i], "-a") == 0)
|
|
tstflags |= tst_abort;
|
|
else if (strcmp(argv[i], "-t") == 0)
|
|
o_timing = 1;
|
|
else if (strcmp(argv[i], "--no-alarm") == 0) {
|
|
o_alarm = 0;
|
|
}
|
|
else if (strncmp(argv[i], "-l", 2) == 0) {
|
|
int level = 3;
|
|
char *rest = NULL;
|
|
|
|
if (argv[i][2])
|
|
level = strtol(argv[i] + 2, &rest, 10);
|
|
else if (argv[i + 1])
|
|
level = strtol(argv[i + 1], &rest, 10), i++;
|
|
else
|
|
level = 3, rest = "";
|
|
|
|
if (rest == NULL || *rest)
|
|
usage(1);
|
|
|
|
su_log_set_level(sresolv_log, level);
|
|
}
|
|
else
|
|
usage(1);
|
|
}
|
|
|
|
#if HAVE_ALARM
|
|
if (o_alarm) {
|
|
alarm(60);
|
|
signal(SIGALRM, sig_alarm);
|
|
}
|
|
#endif
|
|
|
|
if (!(TSTFLAGS & tst_verbatim)) {
|
|
su_log_soft_set_level(sresolv_log, 0);
|
|
}
|
|
|
|
error |= test_api_errors();
|
|
error |= test_cache();
|
|
|
|
return error;
|
|
}
|