Changeset 1195:16f2d2f15e5b in freeDiameter for extensions/test_app/test_app.c

Timestamp:

Jun 11, 2013, 3:05:33 PM (11 years ago)

Author:

Sebastien Decugis <sdecugis@freediameter.net>

Branch:

default

Phase:

public

Message:

Allow better usage in benchmark mode

File:

• extensions/test_app/test_app.c (modified) (5 diffs)

extensions/test_app/test_app.c

@@ -44 +44 @@
 static struct ta_conf _conf;
 static pthread_t ta_stats_th = (pthread_t)NULL;
+static struct fd_hook_hdl * hookhdl = NULL;
 
 static int ta_conf_init(void)
@@ -96 +97 @@
         while (1) {
                 /* Display statistics every XX seconds */
-                sleep(ta_conf->bench_duration * 3);
+                sleep(ta_conf->bench_duration + 3);
                 
                 /* Now, get the current stats */
@@ -136 +137 @@
 }
 
+static void ta_hook_cb(enum fd_hook_type type, struct msg * msg, struct peer_hdr * peer, void * other, struct fd_hook_permsgdata *pmd, void * regdata) {
+
+}
+
+
 /* entry point */
 static int ta_entry(char * conffile)
@@ -172 +178 @@
         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( (1<<HOOK_LAST) - 1, 
+                                        ta_hook_cb, NULL, NULL, &hookhdl) );
+                
+        }
+        
         /* Start the statistics thread */
         CHECK_POSIX( pthread_create(&ta_stats_th, NULL, ta_stats, NULL) );
@@ -185 +198 @@
         if (ta_conf->mode & MODE_SERV)
                 ta_serv_fini();
+        if (hookhdl)
+                fd_hook_unregister( hookhdl );
         CHECK_FCT_DO( fd_thr_term(&ta_stats_th), );
         CHECK_POSIX_DO( pthread_mutex_destroy(&ta_conf->stats_lock), );