/********************************************************************************************************* * Software License Agreement (BSD License) * * Author: Sebastien Decugis * * * * Copyright (c) 2013, WIDE Project and NICT * * All rights reserved. * * * * Redistribution and use of this software 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 WIDE Project or NICT nor the * * names of its contributors may be used to endorse or * * promote products derived from this software without * * specific prior written permission of WIDE Project and * * NICT. * * * * 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 THE COPYRIGHT OWNER OR CONTRIBUTORS 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. * *********************************************************************************************************/ /* This file contains the definition of our test harness. * The harness is very simple yet. * It may be interessant to go to dejagnu later... * */ #ifndef _TESTS_H #define _TESTS_H #include "fdproto-internal.h" #include "fdcore-internal.h" #include #include GCC_DIAG_OFF("-Wdeprecated-declarations") #include GCC_DIAG_ON("-Wdeprecated-declarations") /* Test timeout duration, unless -n is passed on the command line */ #ifndef TEST_TIMEOUT #define TEST_TIMEOUT 120 /* in seconds */ #endif /* TEST_TIMEOUT */ /* Standard includes */ #include #include #include #include /* Define the return code values */ #define PASS 0 #define FAIL 1 /* Define the macro to fail a test with a message */ #define FAILTEST( message... ){ \ LOG_F(message); \ LOG_F("FAILED: %s ", __STRIPPED_FILE__); \ free(tbuf); \ exit(FAIL); \ } /* Define the macro to pass a test */ #define PASSTEST( ){ \ LOG_N("PASS: %s", __STRIPPED_FILE__); \ (void)fd_core_shutdown(); \ (void)fd_core_wait_shutdown_complete(); \ (void)fd_thr_term(&signal_thr); \ free(tbuf); \ exit(PASS); \ } static struct fd_config conf; extern struct fd_config * fd_g_config; /* for dumps */ static char * tbuf = NULL; size_t tbuflen = 0; #define FD_DUMP_TEST_PARAMS &tbuf, &tbuflen, NULL /* Define the standard check routines */ #define CHECK( _val, _assert ){ \ LOG_D("CHECK( %s == %s )", \ #_assert, \ #_val); \ { \ __typeof__ (_val) __ret = (_assert); \ if (__ret != (_val)) { \ FAILTEST( "%s:%d: CHECK FAILED : %s == %lx != %lx", \ __STRIPPED_FILE__, \ __LINE__, \ #_assert, \ (unsigned long)__ret, \ (unsigned long)(_val)); \ }} \ } static pthread_t signal_thr; static void * signal_catch(void * arg) { int sig; sigset_t ss; fd_log_threadname ( "Signal catcher" ); sigemptyset(&ss); /* We use SIGALRM */ sigaddset(&ss, SIGALRM); /* Unblock any other signal for this thread, so that default handler is enabled */ CHECK_SYS_DO( pthread_sigmask( SIG_SETMASK, &ss, NULL ), ); /* Now wait for sigwait or cancelation */ CHECK_POSIX_DO( sigwait(&ss, &sig), ); FAILTEST("The timeout (" _stringize(TEST_TIMEOUT) " sec) was reached. Use -n or change TEST_TIMEOUT if the test needs more time to execute."); return NULL; } GCRY_THREAD_OPTION_PTHREAD_IMPL; /* gnutls debug. */ static void fd_gnutls_debug(int level, const char * str) { const char * __thn = ((char *)pthread_getspecific(fd_log_thname) ?: "unnamed"); fd_log_debug("tid:%-20s[gnutls:%d] %s", __thn, level, str); } static int gnutls_debug = 0; static int test_parameter = 0; static inline void parse_cmdline(int argc, char * argv[]) { int c; int no_timeout = 0; while ((c = getopt (argc, argv, "dqnf:F:g:p:")) != -1) { switch (c) { case 'd': /* Increase verbosity of debug messages. */ fd_g_debug_lvl--; break; case 'q': /* Decrease verbosity. */ fd_g_debug_lvl++; break; case 'n': /* Disable the timeout of the test. */ no_timeout = 1; break; case 'f': /* Full debug for the function with this name. */ #ifdef DEBUG fd_debug_one_function = optarg; #else /* DEBUG */ TRACE_DEBUG(INFO, "Error: must compile with DEBUG support to use this feature"); #endif /* DEBUG */ break; case 'F': /* Full debug for the functions in file with this name. */ #ifdef DEBUG fd_debug_one_file = optarg; #else /* DEBUG */ TRACE_DEBUG(INFO, "Error: must compile with DEBUG support to use this feature"); #endif /* DEBUG */ break; case 'g': /* Set a debug level and function for GNU TLS calls. */ gnutls_debug = (int)atoi(optarg); break; case 'p': /* Set a debug level and function for GNU TLS calls. */ test_parameter = (int)atoi(optarg); break; default: /* bug: option not considered. */ return; } } if (!no_timeout) { alarm(TEST_TIMEOUT); } CHECK( 0, pthread_create(&signal_thr, NULL, signal_catch, NULL) ); } static inline void test_init(int argc, char * argv[], char *fname) { sigset_t sig_all; sigfillset(&sig_all); CHECK( 0, pthread_sigmask(SIG_BLOCK, &sig_all, NULL)); fd_g_config = &conf; memset(fd_g_config, 0, sizeof(struct fd_config)); CHECK( 0, fd_libproto_init() ); CHECK( 0, fd_hooks_init() ); fd_log_threadname(fname); /* Parse the command line */ parse_cmdline(argc, argv); /* Initialize gcrypt and gnutls */ (void) gcry_control (GCRYCTL_SET_THREAD_CBS, &gcry_threads_pthread); (void) gcry_control (GCRYCTL_ENABLE_QUICK_RANDOM, 0); CHECK( 0, gnutls_global_init()); /* Set gnutls debug level ? */ if (gnutls_debug) { gnutls_global_set_log_function((gnutls_log_func)fd_gnutls_debug); gnutls_global_set_log_level (gnutls_debug); TRACE_DEBUG(INFO, "Enabled GNUTLS debug at level %d", gnutls_debug); } /* Initialize the config */ CHECK( 0, fd_conf_init() ); /* Add definitions of the base protocol */ CHECK( 0, fd_dict_base_protocol(fd_g_config->cnf_dict) ); /* Initialize only the sessions */ CHECK( 0, fd_sess_start() ); return; } #define INIT_FD() test_init(argc, argv, __STRIPPED_FILE__) #endif /* _TESTS_H */