Initial commit
Change-Id: I6a4444e3c193dae437cd7929f4c39aba7b749efa
diff --git a/extensions/test_app/test_app.c b/extensions/test_app/test_app.c
new file mode 100644
index 0000000..a89f281
--- /dev/null
+++ b/extensions/test_app/test_app.c
@@ -0,0 +1,223 @@
+/*********************************************************************************************************
+* 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. *
+*********************************************************************************************************/
+
+/*
+ * Test application for freeDiameter.
+ */
+
+#include "test_app.h"
+
+/* Initialize the configuration */
+struct ta_conf * ta_conf = NULL;
+static struct ta_conf _conf;
+static pthread_t ta_stats_th = (pthread_t)NULL;
+
+static int ta_conf_init(void)
+{
+ ta_conf = &_conf;
+ memset(ta_conf, 0, sizeof(struct ta_conf));
+
+ /* Set the default values */
+ ta_conf->vendor_id = 999999; /* Dummy value */
+ ta_conf->appli_id = 0xffffff; /* dummy value */
+ ta_conf->cmd_id = 0xfffffe; /* Experimental */
+ ta_conf->avp_id = 0xffffff; /* dummy value */
+ ta_conf->long_avp_len = 5000;
+ ta_conf->mode = MODE_SERV | MODE_CLI;
+ ta_conf->dest_realm = strdup(fd_g_config->cnf_diamrlm);
+ ta_conf->dest_host = NULL;
+ ta_conf->signal = TEST_APP_DEFAULT_SIGNAL;
+ ta_conf->bench_concur = 100;
+ ta_conf->bench_duration = 10;
+
+ /* Initialize the mutex */
+ CHECK_POSIX( pthread_mutex_init(&ta_conf->stats_lock, NULL) );
+
+ return 0;
+}
+
+static void ta_conf_dump(void)
+{
+ if (!TRACE_BOOL(INFO))
+ return;
+ fd_log_debug( "------- app_test configuration dump: ---------");
+ fd_log_debug( " Vendor Id .......... : %u", ta_conf->vendor_id);
+ fd_log_debug( " Application Id ..... : %u", ta_conf->appli_id);
+ fd_log_debug( " Command Id ......... : %u", ta_conf->cmd_id);
+ fd_log_debug( " AVP Id ............. : %u", ta_conf->avp_id);
+ fd_log_debug( " Long AVP Id ........ : %u", ta_conf->long_avp_id);
+ fd_log_debug( " Long AVP len ....... : %zu", ta_conf->long_avp_len);
+ fd_log_debug( " Mode ............... : %s%s%s", ta_conf->mode & MODE_SERV ? "Serv" : "", ta_conf->mode & MODE_CLI ? "Cli" : "", ta_conf->mode & MODE_BENCH ? " (Benchmark)" : "");
+ fd_log_debug( " Destination Realm .. : %s", ta_conf->dest_realm ?: "- none -");
+ fd_log_debug( " Destination Host ... : %s", ta_conf->dest_host ?: "- none -");
+ fd_log_debug( " Signal ............. : %i", ta_conf->signal);
+ fd_log_debug( "------- /app_test configuration dump ---------");
+}
+
+/* Function to display statistics periodically */
+static void * ta_stats(void * arg) {
+
+ struct timespec start, now;
+ struct ta_stats copy;
+
+ /* Get the start time */
+ CHECK_SYS_DO( clock_gettime(CLOCK_REALTIME, &start), );
+
+ /* Now, loop until canceled */
+ while (1) {
+ /* Display statistics every XX seconds */
+ sleep(ta_conf->bench_duration + 3);
+
+ /* Now, get the current stats */
+ CHECK_POSIX_DO( pthread_mutex_lock(&ta_conf->stats_lock), );
+ memcpy(©, &ta_conf->stats, sizeof(struct ta_stats));
+ CHECK_POSIX_DO( pthread_mutex_unlock(&ta_conf->stats_lock), );
+
+ /* Get the current execution time */
+ CHECK_SYS_DO( clock_gettime(CLOCK_REALTIME, &now), );
+
+ /* Now, display everything */
+ fd_log_debug( "------- app_test statistics ---------");
+ if (now.tv_nsec >= start.tv_nsec) {
+ fd_log_debug( " Executing for: %d.%06ld sec",
+ (int)(now.tv_sec - start.tv_sec),
+ (long)(now.tv_nsec - start.tv_nsec) / 1000);
+ } else {
+ fd_log_debug( " Executing for: %d.%06ld sec",
+ (int)(now.tv_sec - 1 - start.tv_sec),
+ (long)(now.tv_nsec + 1000000000 - start.tv_nsec) / 1000);
+ }
+
+ if (ta_conf->mode & MODE_SERV) {
+ fd_log_debug( " Server: %llu message(s) echoed", copy.nb_echoed);
+ }
+ if (ta_conf->mode & MODE_CLI) {
+ fd_log_debug( " Client:");
+ fd_log_debug( " %llu message(s) sent", copy.nb_sent);
+ fd_log_debug( " %llu error(s) received", copy.nb_errs);
+ fd_log_debug( " %llu answer(s) received", copy.nb_recv);
+ fd_log_debug( " fastest: %ld.%06ld sec.", copy.shortest / 1000000, copy.shortest % 1000000);
+ fd_log_debug( " slowest: %ld.%06ld sec.", copy.longest / 1000000, copy.longest % 1000000);
+ fd_log_debug( " Average: %ld.%06ld sec.", copy.avg / 1000000, copy.avg % 1000000);
+ }
+ fd_log_debug( "-------------------------------------");
+ }
+
+ return NULL; /* never called */
+}
+
+static struct fd_hook_hdl * hookhdl[2] = { NULL, NULL };
+static void ta_hook_cb_silent(enum fd_hook_type type, struct msg * msg, struct peer_hdr * peer, void * other, struct fd_hook_permsgdata *pmd, void * regdata) {
+}
+static void ta_hook_cb_oneline(enum fd_hook_type type, struct msg * msg, struct peer_hdr * peer, void * other, struct fd_hook_permsgdata *pmd, void * regdata) {
+ char * buf = NULL;
+ size_t len;
+
+ CHECK_MALLOC_DO( fd_msg_dump_summary(&buf, &len, NULL, msg, NULL, 0, 0),
+ { LOG_E("Error while dumping a message"); return; } );
+
+ LOG_N("{%d} %s: %s", type, (char *)other ?:"<nil>", buf ?:"<nil>");
+
+ free(buf);
+}
+
+
+/* entry point */
+static int ta_entry(char * conffile)
+{
+ TRACE_ENTRY("%p", conffile);
+
+ /* Initialize configuration */
+ CHECK_FCT( ta_conf_init() );
+
+ /* Parse configuration file */
+ if (conffile != NULL) {
+ CHECK_FCT( ta_conf_handle(conffile) );
+ }
+
+ TRACE_DEBUG(INFO, "Extension Test_App initialized with configuration: '%s'", conffile);
+ ta_conf_dump();
+
+ /* Install objects definitions for this test application */
+ CHECK_FCT( ta_dict_init() );
+
+ /* Install the handlers for incoming messages */
+ if (ta_conf->mode & MODE_SERV) {
+ CHECK_FCT( ta_serv_init() );
+ }
+
+ /* Start the signal handler thread */
+ if (ta_conf->mode & MODE_CLI) {
+ if (ta_conf->mode & MODE_BENCH) {
+ CHECK_FCT( ta_bench_init() );
+ } else {
+ CHECK_FCT( ta_cli_init() );
+ }
+ }
+
+ /* Advertise the support for the test application in the peer */
+ CHECK_FCT( fd_disp_app_support ( ta_appli, ta_vendor, 1, 0 ) );
+
+ if (ta_conf->mode & MODE_BENCH) {
+ /* Register an empty hook to disable the default handling */
+ CHECK_FCT( fd_hook_register( HOOK_MASK( HOOK_DATA_RECEIVED, HOOK_MESSAGE_RECEIVED, HOOK_MESSAGE_LOCAL, HOOK_MESSAGE_SENT, HOOK_MESSAGE_ROUTING_FORWARD, HOOK_MESSAGE_ROUTING_LOCAL ),
+ ta_hook_cb_silent, NULL, NULL, &hookhdl[0]) );
+ CHECK_FCT( fd_hook_register( HOOK_MASK( HOOK_MESSAGE_ROUTING_ERROR, HOOK_MESSAGE_DROPPED ),
+ ta_hook_cb_oneline, NULL, NULL, &hookhdl[1]) );
+
+ }
+
+ /* Start the statistics thread */
+ CHECK_POSIX( pthread_create(&ta_stats_th, NULL, ta_stats, NULL) );
+
+ return 0;
+}
+
+/* Unload */
+void fd_ext_fini(void)
+{
+ if (ta_conf->mode & MODE_CLI)
+ ta_cli_fini();
+ if (ta_conf->mode & MODE_SERV)
+ ta_serv_fini();
+ if (hookhdl[0])
+ fd_hook_unregister( hookhdl[0] );
+ if (hookhdl[1])
+ fd_hook_unregister( hookhdl[1] );
+ CHECK_FCT_DO( fd_thr_term(&ta_stats_th), );
+ CHECK_POSIX_DO( pthread_mutex_destroy(&ta_conf->stats_lock), );
+}
+
+EXTENSION_ENTRY("test_app", ta_entry);