5N/A

5N/A#include <talloc.h>

5N/A#include <tevent.h>

5N/A#include <errno.h>

5N/A#include <popt.h>

5N/A#include <unistd.h>

5N/A#include <sys/types.h>

#include <ifaddrs.h>

#include <arpa/inet.h>

#include "providers/dp_dyndns.c"

#include "tests/cmocka/common_mock.h"

#include "src/providers/dp_dyndns.h"

#define TESTS_PATH "tests_dyndns"

#define TEST_CONF_DB "test_dyndns_conf.ldb"

#define TEST_DOM_NAME "dyndns_test"

#define TEST_SYSDB_FILE "cache_"TEST_DOM_NAME".ldb"

#define TEST_ID_PROVIDER "ldap"

enum mock_nsupdate_states {

MOCK_NSUPDATE_OK,

MOCK_NSUPDATE_ERR,

MOCK_NSUPDATE_TIMEOUT,

};

struct dyndns_test_ctx {

struct sss_test_ctx *tctx;

struct be_ctx *be_ctx;

struct be_nsupdate_ctx *update_ctx;

enum mock_nsupdate_states state;

int child_status;

int child_retval;

};

static struct dyndns_test_ctx *dyndns_test_ctx;

void __wrap_execv(const char *path, char *const argv[])

{

int err;

switch (dyndns_test_ctx->state) {

case MOCK_NSUPDATE_OK:

DEBUG(SSSDBG_FUNC_DATA, "nsupdate success test case\n");

err = 0;

break;

case MOCK_NSUPDATE_ERR:

DEBUG(SSSDBG_FUNC_DATA, "nsupdate error test case\n");

err = 1;

break;

case MOCK_NSUPDATE_TIMEOUT:

DEBUG(SSSDBG_FUNC_DATA, "nsupdate timeout test case\n");

err = 2;

sleep(3);

break;

default:

DEBUG(SSSDBG_CRIT_FAILURE, "unknown test case\n");

err = 255;

break;

}

DEBUG(SSSDBG_TRACE_LIBS, "Child exiting with status %d\n", err);

_exit(err);

}

int __wrap_getifaddrs(struct ifaddrs **_ifap)

{

struct ifaddrs *ifap = NULL;

struct ifaddrs *ifap_prev = NULL;

struct ifaddrs *ifap_head = NULL;

char *name;

char *straddr;

struct sockaddr_in *sa;

while ((name = sss_mock_ptr_type(char *)) != NULL) {

straddr = sss_mock_ptr_type(char *);

if (straddr == NULL) {

errno = EINVAL;

goto fail;

}

ifap = talloc_zero(global_talloc_context, struct ifaddrs);

if (ifap == NULL) {

errno = ENOMEM; /* getifaddrs sets errno, too */

goto fail;

}

if (ifap_prev) {

ifap_prev->ifa_next = ifap;

} else {

ifap_head = ifap;

}

ifap_prev = ifap;

ifap->ifa_name = talloc_strdup(ifap, name);

if (ifap->ifa_name == NULL) {

errno = ENOMEM;

goto fail;

}

/* Do not alocate directly on ifap->ifa_addr to

sa = talloc(ifap, struct sockaddr_in);

if (sa == NULL) {

errno = ENOMEM;

goto fail;

}

sa->sin_family = AF_INET;

/* convert straddr into ifa_addr */

if (inet_pton(AF_INET, straddr, &sa->sin_addr) != 1) {

goto fail;

}

ifap->ifa_addr = (struct sockaddr *) sa;

}

*_ifap = ifap_head;

return 0;

fail:

talloc_free(ifap);

return -1;

}

void __wrap_freeifaddrs(struct ifaddrs *ifap)

{

talloc_free(ifap);

}

static void dyndns_test_done(struct tevent_req *req)

{

struct dyndns_test_ctx *ctx =

tevent_req_callback_data(req, struct dyndns_test_ctx);

ctx->child_retval = -1;

ctx->tctx->error = be_nsupdate_recv(req, &ctx->child_status);

talloc_zfree(req);

ctx->tctx->done = true;

}

void will_return_getifaddrs(const char *ifname, const char *straddr)

{

will_return(__wrap_getifaddrs, ifname);

if (ifname) {

will_return(__wrap_getifaddrs, straddr);

}

}

void dyndns_test_get_ifaddr(void **state)

{

errno_t ret;

struct sss_iface_addr *addrlist;

char straddr[128];

check_leaks_push(dyndns_test_ctx);

will_return_getifaddrs("eth0", "192.168.0.1");

will_return_getifaddrs("eth1", "192.168.0.2");

will_return_getifaddrs(NULL, NULL); /* sentinel */

ret = sss_iface_addr_list_get(dyndns_test_ctx, "eth0", &addrlist);

assert_int_equal(ret, EOK);

/* There must be only one address with the correct value */

assert_non_null(addrlist);

assert_non_null(addrlist->addr);

assert_null(addrlist->next);

assert_null(addrlist->prev);

assert_non_null(inet_ntop(AF_INET,

&((struct sockaddr_in *) addrlist->addr)->sin_addr,

straddr, INET6_ADDRSTRLEN));

assert_string_equal(straddr, "192.168.0.1");

talloc_free(addrlist);

assert_true(check_leaks_pop(dyndns_test_ctx) == true);

}

void dyndns_test_ok(void **state)

{

struct tevent_req *req;

errno_t ret;

TALLOC_CTX *tmp_ctx;

tmp_ctx = talloc_new(global_talloc_context);

assert_non_null(tmp_ctx);

check_leaks_push(tmp_ctx);

dyndns_test_ctx->state = MOCK_NSUPDATE_OK;

req = be_nsupdate_send(tmp_ctx, dyndns_test_ctx->tctx->ev,

BE_NSUPDATE_AUTH_GSS_TSIG,

discard_const("test message"), false);

assert_non_null(req);

tevent_req_set_callback(req, dyndns_test_done, dyndns_test_ctx);

/* Wait until the test finishes with EOK */

ret = test_ev_loop(dyndns_test_ctx->tctx);

DEBUG(SSSDBG_TRACE_LIBS,

"Child request returned [%d]: %s\n", ret, strerror(ret));

assert_int_equal(ret, EOK);

assert_true(WIFEXITED(dyndns_test_ctx->child_status));

assert_int_equal(WEXITSTATUS(dyndns_test_ctx->child_status), 0);

assert_true(check_leaks_pop(tmp_ctx) == true);

talloc_free(tmp_ctx);

}

void dyndns_test_error(void **state)

{

struct tevent_req *req;

errno_t ret;

TALLOC_CTX *tmp_ctx;

tmp_ctx = talloc_new(global_talloc_context);

assert_non_null(tmp_ctx);

check_leaks_push(tmp_ctx);

dyndns_test_ctx->state = MOCK_NSUPDATE_ERR;

req = be_nsupdate_send(tmp_ctx, dyndns_test_ctx->tctx->ev,

BE_NSUPDATE_AUTH_GSS_TSIG,

discard_const("test message"), false);

assert_non_null(req);

tevent_req_set_callback(req, dyndns_test_done, dyndns_test_ctx);

/* Wait until the test finishes with EIO (child error) */

ret = test_ev_loop(dyndns_test_ctx->tctx);

DEBUG(SSSDBG_TRACE_LIBS,

"Child request returned [%d]: %s\n", ret, strerror(ret));

assert_int_equal(ret, ERR_DYNDNS_FAILED);

assert_true(WIFEXITED(dyndns_test_ctx->child_status));

assert_int_equal(WEXITSTATUS(dyndns_test_ctx->child_status), 1);

assert_true(check_leaks_pop(tmp_ctx) == true);

talloc_free(tmp_ctx);

}

void dyndns_test_timeout(void **state)

{

struct tevent_req *req;

errno_t ret;

TALLOC_CTX *tmp_ctx;

tmp_ctx = talloc_new(global_talloc_context);

assert_non_null(tmp_ctx);

check_leaks_push(tmp_ctx);

dyndns_test_ctx->state = MOCK_NSUPDATE_TIMEOUT;

req = be_nsupdate_send(tmp_ctx, dyndns_test_ctx->tctx->ev,

BE_NSUPDATE_AUTH_GSS_TSIG,

discard_const("test message"), false);

assert_non_null(req);

tevent_req_set_callback(req, dyndns_test_done, dyndns_test_ctx);

/* Wait until the test finishes with EIO (child error) */

ret = test_ev_loop(dyndns_test_ctx->tctx);

/* The event queue may not be empty. We need to make sure that all events

tevent_loop_once(dyndns_test_ctx->tctx->ev); /* SIGCHLD handler */

tevent_loop_once(dyndns_test_ctx->tctx->ev); /* nsupdate_child_handler */

DEBUG(SSSDBG_TRACE_LIBS,

"Child request returned [%d]: %s\n", ret, strerror(ret));

assert_int_equal(ret, ERR_DYNDNS_TIMEOUT);

assert_true(check_leaks_pop(tmp_ctx) == true);

talloc_free(tmp_ctx);

}

void dyndns_test_timer(void *pvt)

{

struct dyndns_test_ctx *ctx = talloc_get_type(pvt, struct dyndns_test_ctx);

static int ncalls = 0;

ncalls++;

if (ncalls == 1) {

be_nsupdate_timer_schedule(ctx->tctx->ev, ctx->update_ctx);

} else if (ncalls == 2) {

ctx->tctx->done = true;

}

ctx->tctx->error = ERR_OK;

}

void dyndns_test_interval(void **state)

{

errno_t ret;

TALLOC_CTX *tmp_ctx;

tmp_ctx = talloc_new(global_talloc_context);

assert_non_null(tmp_ctx);

check_leaks_push(tmp_ctx);

ret = be_nsupdate_init(tmp_ctx, dyndns_test_ctx->be_ctx, NULL,

&dyndns_test_ctx->update_ctx);

assert_int_equal(ret, EOK);

ret = be_nsupdate_init_timer(dyndns_test_ctx->update_ctx,

dyndns_test_ctx->be_ctx->ev,

dyndns_test_timer, dyndns_test_ctx);

assert_int_equal(ret, EOK);

/* Wait until the timer hits */

ret = test_ev_loop(dyndns_test_ctx->tctx);

DEBUG(SSSDBG_TRACE_LIBS,

"Child request returned [%d]: %s\n", ret, strerror(ret));

assert_int_equal(ret, ERR_OK);

talloc_free(dyndns_test_ctx->update_ctx);

assert_true(check_leaks_pop(tmp_ctx) == true);

talloc_free(tmp_ctx);

}

void dyndns_test_setup(void **state)

{

struct sss_test_conf_param params[] = {

{ "dyndns_update", "true" },

{ "dyndns_refresh_interval", "2" },

{ NULL, NULL }, /* Sentinel */

};

assert_true(leak_check_setup());

dyndns_test_ctx = talloc_zero(global_talloc_context, struct dyndns_test_ctx);

assert_non_null(dyndns_test_ctx);

dyndns_test_ctx->tctx = create_dom_test_ctx(dyndns_test_ctx, TESTS_PATH,

TEST_CONF_DB, TEST_DOM_NAME,

TEST_ID_PROVIDER, params);

assert_non_null(dyndns_test_ctx->tctx);

dyndns_test_ctx->be_ctx = talloc_zero(dyndns_test_ctx, struct be_ctx);

assert_non_null(dyndns_test_ctx->be_ctx);

dyndns_test_ctx->be_ctx->cdb = dyndns_test_ctx->tctx->confdb;

dyndns_test_ctx->be_ctx->ev = dyndns_test_ctx->tctx->ev;

dyndns_test_ctx->be_ctx->conf_path = talloc_asprintf(dyndns_test_ctx,

CONFDB_DOMAIN_PATH_TMPL,

TEST_DOM_NAME);

assert_non_null(dyndns_test_ctx->be_ctx->conf_path);

}

void dyndns_test_teardown(void **state)

{

talloc_free(dyndns_test_ctx);

assert_true(leak_check_teardown());

}

int main(int argc, const char *argv[])

{

int rv;

int no_cleanup = 0;

poptContext pc;

int opt;

struct poptOption long_options[] = {

POPT_AUTOHELP

SSSD_DEBUG_OPTS

{"no-cleanup", 'n', POPT_ARG_NONE, &no_cleanup, 0,

_("Do not delete the test database after a test run"), NULL },

POPT_TABLEEND

};

const UnitTest tests[] = {

/* Utility functions unit test */

unit_test(dyndns_test_get_ifaddr),

/* Dynamic DNS update unit tests*/

unit_test_setup_teardown(dyndns_test_ok,

dyndns_test_setup, dyndns_test_teardown),

unit_test_setup_teardown(dyndns_test_error,

dyndns_test_setup, dyndns_test_teardown),

unit_test_setup_teardown(dyndns_test_timeout,

dyndns_test_setup, dyndns_test_teardown),

unit_test_setup_teardown(dyndns_test_interval,

dyndns_test_setup, dyndns_test_teardown),

};

/* Set debug level to invalid value so we can deside if -d 0 was used. */

debug_level = SSSDBG_INVALID;

pc = poptGetContext(argv[0], argc, argv, long_options, 0);

while((opt = poptGetNextOpt(pc)) != -1) {

switch(opt) {

default:

fprintf(stderr, "\nInvalid option %s: %s\n\n",

poptBadOption(pc, 0), poptStrerror(opt));

poptPrintUsage(pc, stderr, 0);

return 1;

}

}

poptFreeContext(pc);

DEBUG_INIT(debug_level);

/* Even though normally the tests should clean up after themselves

tests_set_cwd();

test_dom_suite_cleanup(TESTS_PATH, TEST_CONF_DB, TEST_SYSDB_FILE);

test_dom_suite_setup(TESTS_PATH);

rv = run_tests(tests);

if (rv == 0 && !no_cleanup) {

test_dom_suite_cleanup(TESTS_PATH, TEST_CONF_DB, TEST_SYSDB_FILE);

}

return rv;

}