Mercurial > hg > freeDiameter
view extensions/test_app/test_app.c @ 1214:c2fbaf2985f4
New options to test_app extension to generate long Diameter messages
author | Sebastien Decugis <sdecugis@freediameter.net> |
---|---|
date | Tue, 18 Jun 2013 16:27:45 +0800 |
parents | 2c62bba0ef4b |
children |
line wrap: on
line source
/********************************************************************************************************* * 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);