Initial commit
Change-Id: I6a4444e3c193dae437cd7929f4c39aba7b749efa
diff --git a/tests/tests.h b/tests/tests.h
new file mode 100644
index 0000000..85e6d9b
--- /dev/null
+++ b/tests/tests.h
@@ -0,0 +1,239 @@
+/*********************************************************************************************************
+* Software License Agreement (BSD License) *
+* Author: Sebastien Decugis <sdecugis@freediameter.net> *
+* *
+* 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 <pthread.h>
+#include <errno.h>
+GCC_DIAG_OFF("-Wdeprecated-declarations")
+#include <gcrypt.h>
+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 <getopt.h>
+#include <time.h>
+#include <libgen.h>
+#include <signal.h>
+
+/* 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 */