Navigation


source: freeDiameter/extensions/test_app/ta_bench.c @ 706:4ffbc9f1e922

Last change on this file since 706:4ffbc9f1e922 was 706:4ffbc9f1e922, checked in by Sebastien Decugis <sdecugis@nict.go.jp>, 11 years ago

Large UNTESTED commit with the following changes:

  • Improved DiameterIdentity? handling (esp. interationalization issues), and improve efficiency of some string operations in peers, sessions, and dictionary modules (closes #7)
  • Cleanup in the session module to free only unreferenced sessions (#16)
  • Removed fd_cpu_flush_cache(), replaced by more robust alternatives.
  • Improved peer state machine algorithm to counter SCTP multistream race condition.
File size: 10.3 KB
Line 
1/*********************************************************************************************************
2* Software License Agreement (BSD License)                                                               *
3* Author: Sebastien Decugis <sdecugis@nict.go.jp>                                                        *
4*                                                                                                        *
5* Copyright (c) 2011, WIDE Project and NICT                                                              *
6* All rights reserved.                                                                                   *
7*                                                                                                        *
8* Redistribution and use of this software in source and binary forms, with or without modification, are  *
9* permitted provided that the following conditions are met:                                              *
10*                                                                                                        *
11* * Redistributions of source code must retain the above                                                 *
12*   copyright notice, this list of conditions and the                                                    *
13*   following disclaimer.                                                                                *
14*                                                                                                        *
15* * Redistributions in binary form must reproduce the above                                              *
16*   copyright notice, this list of conditions and the                                                    *
17*   following disclaimer in the documentation and/or other                                               *
18*   materials provided with the distribution.                                                            *
19*                                                                                                        *
20* * Neither the name of the WIDE Project or NICT nor the                                                 *
21*   names of its contributors may be used to endorse or                                                  *
22*   promote products derived from this software without                                                  *
23*   specific prior written permission of WIDE Project and                                                *
24*   NICT.                                                                                                *
25*                                                                                                        *
26* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED *
27* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A *
28* PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR *
29* ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT     *
30* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS    *
31* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR *
32* TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF   *
33* ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.                                                             *
34*********************************************************************************************************/
35
36/* Create and send a message, and receive it */
37
38#include "test_app.h"
39
40#include <semaphore.h>
41#include <stdio.h>
42
43struct ta_mess_info {
44        int32_t         randval;        /* a random value to store in Test-AVP */
45        struct timespec ts;             /* Time of sending the message */
46};
47
48static sem_t ta_sem; /* To handle the concurrency */
49
50/* Cb called when an answer is received */
51static void ta_cb_ans(void * data, struct msg ** msg)
52{
53        struct ta_mess_info * mi = (struct ta_mess_info *)data;
54        struct timespec ts;
55        struct avp * avp;
56        struct avp_hdr * hdr;
57        unsigned long dur;
58       
59        CHECK_SYS_DO( clock_gettime(CLOCK_REALTIME, &ts), return );
60
61        /* Value of Result Code */
62        CHECK_FCT_DO( fd_msg_search_avp ( *msg, ta_res_code, &avp), return );
63        if (avp) {
64                CHECK_FCT_DO( fd_msg_avp_hdr( avp, &hdr ), return );
65        }
66        if (!avp || !hdr || hdr->avp_value->i32 != 2001) {
67                /* error */
68                CHECK_POSIX_DO( pthread_mutex_lock(&ta_conf->stats_lock), );
69                ta_conf->stats.nb_errs++;
70                CHECK_POSIX_DO( pthread_mutex_unlock(&ta_conf->stats_lock), );
71                goto end;
72        }
73       
74        /* Check value of Test-AVP */
75        CHECK_FCT_DO( fd_msg_search_avp ( *msg, ta_avp, &avp), return );
76        if (avp) {
77                CHECK_FCT_DO( fd_msg_avp_hdr( avp, &hdr ), return );
78                ASSERT(hdr->avp_value->i32 == mi->randval);
79        }
80       
81        /* Compute how long it took */
82        dur = ((ts.tv_sec - mi->ts.tv_sec) * 1000000) + ((ts.tv_nsec - mi->ts.tv_nsec) / 1000);
83       
84        /* Add this value to the stats */
85        CHECK_POSIX_DO( pthread_mutex_lock(&ta_conf->stats_lock), );
86       
87        if (ta_conf->stats.nb_recv) {
88                /* Ponderate in the avg */
89                ta_conf->stats.avg = (ta_conf->stats.avg * ta_conf->stats.nb_recv + dur) / (ta_conf->stats.nb_recv + 1);
90                /* Min, max */
91                if (dur < ta_conf->stats.shortest)
92                        ta_conf->stats.shortest = dur;
93                if (dur > ta_conf->stats.longest)
94                        ta_conf->stats.longest = dur;
95        } else {
96                ta_conf->stats.shortest = dur;
97                ta_conf->stats.longest = dur;
98                ta_conf->stats.avg = dur;
99        }
100        ta_conf->stats.nb_recv++;
101       
102        CHECK_POSIX_DO( pthread_mutex_unlock(&ta_conf->stats_lock), );
103       
104end:   
105        /* Free the message */
106        CHECK_FCT_DO(fd_msg_free(*msg), );
107        *msg = NULL;
108       
109        free(mi);
110       
111        /* Post the semaphore */
112        CHECK_SYS_DO( sem_post(&ta_sem), );
113       
114        return;
115}
116
117/* Create a test message */
118static void ta_bench_test_message()
119{
120        struct msg * req = NULL;
121        struct avp * avp;
122        union avp_value val;
123        struct ta_mess_info * mi = NULL;
124        struct session *sess = NULL;
125       
126        TRACE_DEBUG(FULL, "Creating a new message for sending.");
127       
128        /* Create the request */
129        CHECK_FCT_DO( fd_msg_new( ta_cmd_r, MSGFL_ALLOC_ETEID, &req ), goto out );
130       
131        /* Create a new session */
132        #define TEST_APP_SID_OPT  "app_testb"
133        CHECK_FCT_DO( fd_sess_new( &sess, fd_g_config->cnf_diamid, fd_g_config->cnf_diamid_len, (os0_t)TEST_APP_SID_OPT, CONSTSTRLEN(TEST_APP_SID_OPT) ), goto out );
134       
135        /* Create the random value to store with the session */
136        mi = malloc(sizeof(struct ta_mess_info));
137        if (mi == NULL) {
138                fd_log_debug("malloc failed: %s", strerror(errno));
139                goto out;
140        }
141       
142        mi->randval = (int32_t)random();
143       
144        /* Now set all AVPs values */
145       
146        /* Session-Id */
147        {
148                os0_t sid;
149                size_t sidlen;
150                CHECK_FCT_DO( fd_sess_getsid ( sess, &sid, &sidlen ), goto out );
151                CHECK_FCT_DO( fd_msg_avp_new ( ta_sess_id, 0, &avp ), goto out );
152                val.os.data = sid;
153                val.os.len  = sidlen;
154                CHECK_FCT_DO( fd_msg_avp_setvalue( avp, &val ), goto out );
155                CHECK_FCT_DO( fd_msg_avp_add( req, MSG_BRW_FIRST_CHILD, avp ), goto out );
156               
157        }
158       
159        /* Set the Destination-Realm AVP */
160        {
161                CHECK_FCT_DO( fd_msg_avp_new ( ta_dest_realm, 0, &avp ), goto out  );
162                val.os.data = (unsigned char *)(ta_conf->dest_realm);
163                val.os.len  = strlen(ta_conf->dest_realm);
164                CHECK_FCT_DO( fd_msg_avp_setvalue( avp, &val ), goto out  );
165                CHECK_FCT_DO( fd_msg_avp_add( req, MSG_BRW_LAST_CHILD, avp ), goto out  );
166        }
167       
168        /* Set the Destination-Host AVP if needed*/
169        if (ta_conf->dest_host) {
170                CHECK_FCT_DO( fd_msg_avp_new ( ta_dest_host, 0, &avp ), goto out  );
171                val.os.data = (unsigned char *)(ta_conf->dest_host);
172                val.os.len  = strlen(ta_conf->dest_host);
173                CHECK_FCT_DO( fd_msg_avp_setvalue( avp, &val ), goto out  );
174                CHECK_FCT_DO( fd_msg_avp_add( req, MSG_BRW_LAST_CHILD, avp ), goto out  );
175        }
176       
177        /* Set Origin-Host & Origin-Realm */
178        CHECK_FCT_DO( fd_msg_add_origin ( req, 0 ), goto out  );
179       
180        /* Set the User-Name AVP if needed*/
181        if (ta_conf->user_name) {
182                CHECK_FCT_DO( fd_msg_avp_new ( ta_user_name, 0, &avp ), goto out  );
183                val.os.data = (unsigned char *)(ta_conf->user_name);
184                val.os.len  = strlen(ta_conf->user_name);
185                CHECK_FCT_DO( fd_msg_avp_setvalue( avp, &val ), goto out  );
186                CHECK_FCT_DO( fd_msg_avp_add( req, MSG_BRW_LAST_CHILD, avp ), goto out  );
187        }
188       
189        /* Set the Test-AVP AVP */
190        {
191                CHECK_FCT_DO( fd_msg_avp_new ( ta_avp, 0, &avp ), goto out  );
192                val.i32 = mi->randval;
193                CHECK_FCT_DO( fd_msg_avp_setvalue( avp, &val ), goto out  );
194                CHECK_FCT_DO( fd_msg_avp_add( req, MSG_BRW_LAST_CHILD, avp ), goto out  );
195        }
196       
197        CHECK_SYS_DO( clock_gettime(CLOCK_REALTIME, &mi->ts), goto out );
198       
199        /* Send the request */
200        CHECK_FCT_DO( fd_msg_send( &req, ta_cb_ans, mi ), goto out );
201       
202        /* Increment the counter */
203        CHECK_POSIX_DO( pthread_mutex_lock(&ta_conf->stats_lock), );
204        ta_conf->stats.nb_sent++;
205        CHECK_POSIX_DO( pthread_mutex_unlock(&ta_conf->stats_lock), );
206
207out:
208        return;
209}
210
211/* The function called when the signal is received */
212static void ta_bench_start() {
213        struct timespec end_time, now;
214        struct ta_stats start, end;
215       
216        /* Save the initial stats */
217        CHECK_POSIX_DO( pthread_mutex_lock(&ta_conf->stats_lock), );
218        memcpy(&start, &ta_conf->stats, sizeof(struct ta_stats));
219        CHECK_POSIX_DO( pthread_mutex_unlock(&ta_conf->stats_lock), );
220       
221        /* We will run for ta_conf->bench_duration seconds */
222        CHECK_SYS_DO( clock_gettime(CLOCK_REALTIME, &end_time), );
223        end_time.tv_sec += ta_conf->bench_duration;
224       
225        /* Now loop until timeout is reached */
226        do {
227                /* Do not create more that NB_CONCURRENT_MESSAGES in paralel */
228                int ret = sem_timedwait(&ta_sem, &end_time);
229                if (ret == -1) {
230                        ret = errno;
231                        if (ret != ETIMEDOUT) {
232                                CHECK_POSIX_DO(ret, ); /* Just to log it */
233                        }
234                        break;
235                }
236               
237                /* Create and send a new test message */
238                ta_bench_test_message();
239               
240                /* Update the current time */
241                CHECK_SYS_DO( clock_gettime(CLOCK_REALTIME, &now), );
242        } while (TS_IS_INFERIOR(&now, &end_time));
243       
244        /* Save the stats now */
245        CHECK_POSIX_DO( pthread_mutex_lock(&ta_conf->stats_lock), );
246        CHECK_SYS_DO( clock_gettime(CLOCK_REALTIME, &now), ); /* Re-read the time because we might have spent some time wiating for the mutex */
247        memcpy(&end, &ta_conf->stats, sizeof(struct ta_stats));
248        CHECK_POSIX_DO( pthread_mutex_unlock(&ta_conf->stats_lock), );
249       
250        /* Now, display the statistics */
251        fd_log_debug( "------- app_test Benchmark result ---------\n");
252        if (now.tv_nsec >= end_time.tv_nsec) {
253                fd_log_debug( " Executing for: %d.%06ld sec\n",
254                                (int)(now.tv_sec + ta_conf->bench_duration - end_time.tv_sec),
255                                (long)(now.tv_nsec - end_time.tv_nsec) / 1000);
256        } else {
257                fd_log_debug( " Executing for: %d.%06ld sec\n",
258                                (int)(now.tv_sec + ta_conf->bench_duration - 1 - end_time.tv_sec),
259                                (long)(now.tv_nsec + 1000000000 - end_time.tv_nsec) / 1000);
260        }
261        fd_log_debug( "   %llu messages sent\n", end.nb_sent - start.nb_sent);
262        fd_log_debug( "   %llu error(s) received\n", end.nb_errs - start.nb_errs);
263        fd_log_debug( "   %llu answer(s) received\n", end.nb_recv - start.nb_recv);
264        fd_log_debug( "   Overall:\n");
265        fd_log_debug( "     fastest: %ld.%06ld sec.\n", end.shortest / 1000000, end.shortest % 1000000);
266        fd_log_debug( "     slowest: %ld.%06ld sec.\n", end.longest / 1000000, end.longest % 1000000);
267        fd_log_debug( "     Average: %ld.%06ld sec.\n", end.avg / 1000000, end.avg % 1000000);
268        fd_log_debug( "   Throughput: %llu messages / sec\n", (end.nb_recv - start.nb_recv) / (( now.tv_sec + ta_conf->bench_duration - end_time.tv_sec ) + ((now.tv_nsec - end_time.tv_nsec) / 1000000000)));
269        fd_log_debug( "-------------------------------------\n");
270
271}
272
273
274int ta_bench_init(void)
275{
276        CHECK_SYS( sem_init( &ta_sem, 0, ta_conf->bench_concur) );
277
278        CHECK_FCT( fd_event_trig_regcb(ta_conf->signal, "test_app.bench", ta_bench_start ) );
279       
280        return 0;
281}
282
283void ta_bench_fini(void)
284{
285        // CHECK_FCT_DO( fd_sig_unregister(ta_conf->signal), /* continue */ );
286       
287        CHECK_SYS_DO( sem_destroy(&ta_sem), );
288       
289        return;
290};
Note: See TracBrowser for help on using the repository browser.