Mercurial > hg > freeDiameter
annotate extensions/loadtest_cc/loadtest_cc.c @ 1371:d1c40627673c
loadtest_cc: randomize session ids
author | Thomas Klausner <tk@giga.or.at> |
---|---|
date | Mon, 10 Jun 2019 16:56:40 +0200 |
parents | 9294ed1c7ac8 |
children | 6b1a2405cebb |
rev | line source |
---|---|
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
1 /********************************************************************************************************** |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
2 * Software License Agreement(BSD License) * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
3 * Author: Thomas Klausner <tk@giga.or.at> * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
4 * * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
5 * Copyright(c) 2019, Thomas Klausner * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
6 * All rights reserved. * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
7 * * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
8 * Written under contract by nfotex IT GmbH, http://nfotex.com/ * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
9 * * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
10 * Redistribution and use of this software in source and binary forms, with or without modification, are * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
11 * permitted provided that the following conditions are met: * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
12 * * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
13 * * Redistributions of source code must retain the above * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
14 * copyright notice, this list of conditions and the * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
15 * following disclaimer. * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
16 * * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
17 * * Redistributions in binary form must reproduce the above * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
18 * copyright notice, this list of conditions and the * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
19 * following disclaimer in the documentation and/or other * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
20 * materials provided with the distribution. * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
21 * * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
22 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
23 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
24 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
25 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES(INCLUDING, BUT NOT * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
26 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
27 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
28 * TORT(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
29 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. * |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
30 **********************************************************************************************************/ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
31 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
32 /* This extension waits for a signal (SIGUSR2). When it gets it, it |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
33 * generates messages as quickly as possible to the configured target |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
34 * host; a second SIGUSR2 signal will stop this */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
35 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
36 #include <freeDiameter/extension.h> |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
37 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
38 #include <pthread.h> |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
39 #include <signal.h> |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
40 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
41 #define MODULE_NAME "loadtest_cc" |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
42 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
43 static pthread_t gen_thr = (pthread_t)NULL; |
1366 | 44 struct disp_hdl *ccr_local_hdl; |
45 struct fd_rt_fwd_hdl *ccr_fwd_hdl; | |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
46 volatile int do_generate = 0; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
47 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
48 const char *target = NULL; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
49 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
50 struct dict_object * aai_avp_do; /* cache the Auth-Application-Id dictionary object */ |
1366 | 51 struct dict_object * crn_avp_do; /* cache the CC-Request-Number dictionary object */ |
52 struct dict_object * crt_avp_do; /* cache the CC-Request-Type dictionary object */ | |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
53 struct dict_object * dh_avp_do; /* cache the Destination-Host dictionary object */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
54 struct dict_object * dr_avp_do; /* cache the Destination-Realm dictionary object */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
55 struct dict_object * rc_avp_do; /* cache the Result-Code dictionary object */ |
1366 | 56 struct dict_object * sci_avp_do; /* cache the Service-Context-Id dictionary object */ |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
57 struct dict_object * si_avp_do; /* cache the Session-Id dictionary object */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
58 struct dict_object * pi_avp_do; /* cache the Proxy-Info dictionary object */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
59 struct dict_object * ph_avp_do; /* cache the Proxy-Host dictionary object */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
60 struct dict_object * ps_avp_do; /* cache the Proxy-State dictionary object */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
61 |
1366 | 62 struct statistics { |
63 uint64_t sent; | |
64 uint64_t success; | |
65 uint64_t error; | |
66 time_t first; | |
67 time_t last; | |
68 } statistics; | |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
69 |
1366 | 70 static int handle_message(struct msg **msg) { |
71 struct msg_hdr *hdr = NULL; | |
72 struct avp_hdr *ahdr = NULL; | |
73 struct avp *rc; | |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
74 |
1366 | 75 if (msg == NULL) { |
76 fd_log_error("[%s] NULL CCA message", MODULE_NAME); | |
77 return -1; | |
78 } | |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
79 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
80 CHECK_FCT(fd_msg_hdr(*msg, &hdr)); |
1366 | 81 /* handle CCAs only */ |
82 if ((hdr->msg_code != 272) || (hdr->msg_flags & CMD_FLAG_REQUEST)) { | |
83 fd_log_error("[%s] invalid message type (type %d, flags 0x%x)", MODULE_NAME, hdr->msg_code, hdr->msg_flags); | |
84 return -1; | |
85 } | |
86 | |
87 /* Answer received, check it */ | |
88 if (fd_msg_search_avp(*msg, rc_avp_do, &rc) < 0 || rc == NULL) { | |
89 fd_log_error("[%s] Result-Code not found in CCA", MODULE_NAME); | |
90 return -1; | |
91 } | |
92 | |
93 if (fd_msg_avp_hdr(rc, &ahdr) < 0) { | |
94 fd_log_error("[%s] error parsing Result-Code in CCA", MODULE_NAME); | |
95 return -1; | |
96 } | |
97 fd_log_debug("Credit-Control-Answer with Result-Code %d received", ahdr->avp_value->i32); | |
98 switch (ahdr->avp_value->i32/1000) { | |
99 case 2: | |
100 statistics.success++; | |
101 break; | |
102 default: | |
103 statistics.error++; | |
104 break; | |
105 } | |
106 | |
107 return 0; | |
108 } | |
109 | |
110 static int ccr_local_handler(struct msg ** msg, struct avp * avp, struct session * sess, void * data, enum disp_action * act) | |
111 { | |
112 fd_log_debug("[%s] CCR local handler called", MODULE_NAME); | |
113 | |
114 if (handle_message(msg) < 0) { | |
115 fd_log_error("dropping message"); | |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
116 CHECK_FCT(fd_msg_free(*msg)); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
117 *msg = NULL; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
118 return 0; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
119 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
120 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
121 CHECK_FCT(fd_msg_free(*msg)); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
122 *msg = NULL; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
123 return 0; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
124 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
125 |
1366 | 126 static int ccr_fwd_handler(void *cb_data, struct msg **msg) |
127 { | |
128 fd_log_debug("[%s] CCR FWD handler called", MODULE_NAME); | |
129 | |
130 if (handle_message(msg) < 0) { | |
131 return 0; | |
132 } | |
133 | |
134 return 0; | |
135 } | |
136 | |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
137 /* create message to send */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
138 struct msg *create_message(const char *destination) |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
139 { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
140 struct dict_object *command; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
141 struct msg *msg; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
142 struct avp *avp, *avp1; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
143 union avp_value val; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
144 struct msg_hdr *msg_hdr; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
145 const char *realm; |
1371
d1c40627673c
loadtest_cc: randomize session ids
Thomas Klausner <tk@giga.or.at>
parents:
1366
diff
changeset
|
146 char session_id[800]; |
1366 | 147 const char *service_context_id = "version2.clci.ipc@vodafone.com"; |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
148 const char *proxy_host = "Dummy-Proxy-Host-to-Increase-Package-Size"; |
1371
d1c40627673c
loadtest_cc: randomize session ids
Thomas Klausner <tk@giga.or.at>
parents:
1366
diff
changeset
|
149 const char *proxy_state = "This is just data to increase the package size\nXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX"; |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
150 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
151 if (fd_dict_search(fd_g_config->cnf_dict, DICT_COMMAND, CMD_BY_NAME, "Credit-Control-Request", &command, ENOENT) != 0) { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
152 fd_log_error("can't find template for 'Credit-Control-Request'"); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
153 return NULL; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
154 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
155 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
156 if (fd_msg_new(command, MSGFL_ALLOC_ETEID, &msg) != 0) { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
157 fd_log_error("can't create new 'Credit-Control-Request' message"); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
158 return NULL; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
159 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
160 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
161 /* Application Id in header needs to be set to for since this Credit-Control-Request is for Diameter Credit Control */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
162 if (fd_msg_hdr(msg, &msg_hdr) != 0) { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
163 fd_log_error("can't get message header for 'Credit-Control-Request' message"); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
164 fd_msg_free(msg); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
165 return NULL; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
166 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
167 msg_hdr->msg_appl = 4; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
168 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
169 if (fd_msg_add_origin(msg, 0) != 0) { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
170 fd_log_error("can't set Origin for 'Credit-Control-Request' message"); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
171 fd_msg_free(msg); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
172 return NULL; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
173 } |
1366 | 174 |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
175 /* Destination-Host */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
176 fd_msg_avp_new(dh_avp_do, 0, &avp); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
177 memset(&val, 0, sizeof(val)); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
178 val.os.data = (uint8_t *)target; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
179 val.os.len = strlen(target); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
180 if (fd_msg_avp_setvalue(avp, &val) != 0) { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
181 fd_msg_free(msg); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
182 fd_log_error("can't set value for 'Destination-Host' for 'Credit-Control-Request' message"); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
183 return NULL; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
184 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
185 fd_msg_avp_add(msg, MSG_BRW_LAST_CHILD, avp); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
186 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
187 if ((realm = strchr(target, '.')) == NULL) { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
188 fd_msg_free(msg); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
189 fd_log_error("can't extract realm from host '%s'", target); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
190 return NULL; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
191 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
192 /* skip dot */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
193 realm++; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
194 /* Destination-Realm */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
195 fd_msg_avp_new(dr_avp_do, 0, &avp); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
196 memset(&val, 0, sizeof(val)); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
197 val.os.data = (uint8_t *)realm; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
198 val.os.len = strlen(realm); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
199 if (fd_msg_avp_setvalue(avp, &val) != 0) { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
200 fd_msg_free(msg); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
201 fd_log_error("can't set value for 'Destination-Realm' for 'Credit-Control-Request' message"); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
202 return NULL; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
203 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
204 fd_msg_avp_add(msg, MSG_BRW_LAST_CHILD, avp); |
1366 | 205 |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
206 /* Session-Id */ |
1371
d1c40627673c
loadtest_cc: randomize session ids
Thomas Klausner <tk@giga.or.at>
parents:
1366
diff
changeset
|
207 snprintf(session_id, sizeof(session_id), "session %ld", random()); |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
208 fd_msg_avp_new(si_avp_do, 0, &avp); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
209 memset(&val, 0, sizeof(val)); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
210 val.os.data = (uint8_t *)session_id; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
211 val.os.len = strlen(session_id); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
212 if (fd_msg_avp_setvalue(avp, &val) != 0) { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
213 fd_msg_free(msg); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
214 fd_log_error("can't set value for 'Session-Id' for 'Credit-Control-Request' message"); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
215 return NULL; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
216 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
217 fd_msg_avp_add(msg, MSG_BRW_FIRST_CHILD, avp); |
1366 | 218 |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
219 /* Auth-Application-Id */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
220 fd_msg_avp_new(aai_avp_do, 0, &avp); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
221 memset(&val, 0, sizeof(val)); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
222 val.i32 = 4; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
223 if (fd_msg_avp_setvalue(avp, &val) != 0) { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
224 fd_msg_free(msg); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
225 fd_log_error("can't set value for 'Auth-Application-Id' for 'Credit-Control-Request' message"); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
226 return NULL; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
227 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
228 fd_msg_avp_add(msg, MSG_BRW_LAST_CHILD, avp); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
229 |
1366 | 230 /* Service-Context-Id */ |
231 fd_msg_avp_new(sci_avp_do, 0, &avp); | |
232 memset(&val, 0, sizeof(val)); | |
233 val.os.data = (uint8_t *)service_context_id; | |
234 val.os.len = strlen(service_context_id); | |
235 if (fd_msg_avp_setvalue(avp, &val) != 0) { | |
236 fd_msg_free(msg); | |
237 fd_log_error("can't set value for 'Service-Context-Id' for 'Credit-Control-Request' message"); | |
238 return NULL; | |
239 } | |
240 fd_msg_avp_add(msg, MSG_BRW_LAST_CHILD, avp); | |
241 | |
242 /* CC-Request-Type */ | |
243 fd_msg_avp_new(crt_avp_do, 0, &avp); | |
244 memset(&val, 0, sizeof(val)); | |
245 val.i32 = 1; /* Initial */ | |
246 if (fd_msg_avp_setvalue(avp, &val) != 0) { | |
247 fd_msg_free(msg); | |
248 fd_log_error("can't set value for 'CC-Request-Type' for 'Credit-Control-Request' message"); | |
249 return NULL; | |
250 } | |
251 fd_msg_avp_add(msg, MSG_BRW_LAST_CHILD, avp); | |
252 | |
253 /* CC-Request-Number */ | |
254 fd_msg_avp_new(crn_avp_do, 0, &avp); | |
255 memset(&val, 0, sizeof(val)); | |
256 val.i32 = 1; | |
257 if (fd_msg_avp_setvalue(avp, &val) != 0) { | |
258 fd_msg_free(msg); | |
259 fd_log_error("can't set value for 'CC-Request-Number' for 'Credit-Control-Request' message"); | |
260 return NULL; | |
261 } | |
262 fd_msg_avp_add(msg, MSG_BRW_LAST_CHILD, avp); | |
263 | |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
264 /* Proxy-Info */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
265 fd_msg_avp_new(pi_avp_do, 0, &avp); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
266 /* Proxy-Host */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
267 fd_msg_avp_new(ph_avp_do, 0, &avp1); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
268 memset(&val, 0, sizeof(val)); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
269 val.os.data = (uint8_t *)proxy_host; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
270 val.os.len = strlen(proxy_host); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
271 if (fd_msg_avp_setvalue(avp1, &val) != 0) { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
272 fd_msg_free(msg); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
273 fd_log_error("can't set value for 'Proxy-Host' for 'Credit-Control-Request' message"); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
274 return NULL; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
275 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
276 fd_msg_avp_add(avp, MSG_BRW_LAST_CHILD, avp1); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
277 /* Proxy-State */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
278 fd_msg_avp_new(ps_avp_do, 0, &avp1); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
279 memset(&val, 0, sizeof(val)); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
280 val.os.data = (uint8_t *)proxy_state; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
281 val.os.len = strlen(proxy_state); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
282 if (fd_msg_avp_setvalue(avp1, &val) != 0) { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
283 fd_msg_free(msg); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
284 fd_log_error("can't set value for 'Proxy-State' for 'Credit-Control-Request' message"); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
285 return NULL; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
286 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
287 fd_msg_avp_add(avp, MSG_BRW_LAST_CHILD, avp1); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
288 fd_msg_avp_add(msg, MSG_BRW_LAST_CHILD, avp); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
289 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
290 return msg; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
291 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
292 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
293 /* The thread that handles expired entries cleanup. */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
294 void * gen_thr_fct(void * arg) |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
295 { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
296 struct msg *msg; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
297 fd_log_threadname ( "Loadtest/Generator" ); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
298 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
299 do { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
300 if (do_generate) { |
1366 | 301 if (statistics.first == 0) { |
302 statistics.first = time(NULL); | |
303 } | |
304 msg = create_message(target); | |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
305 fd_msg_send(&msg, NULL, NULL); |
1366 | 306 fd_log_debug("[%s] sent message", MODULE_NAME); |
307 statistics.last = time(NULL); | |
308 statistics.sent++; | |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
309 } else { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
310 sleep(1); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
311 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
312 } while (1); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
313 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
314 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
315 /* signal handler */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
316 static void sig_hdlr(void) |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
317 { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
318 if (do_generate) { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
319 do_generate = 0; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
320 } else { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
321 do_generate = 1; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
322 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
323 fd_log_notice("%s: switched generation of CCRs %s", MODULE_NAME, do_generate ? "on" : "off"); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
324 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
325 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
326 /* entry hook: register callback */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
327 static int cc_entry(char * conffile) |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
328 { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
329 struct disp_when data; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
330 |
1371
d1c40627673c
loadtest_cc: randomize session ids
Thomas Klausner <tk@giga.or.at>
parents:
1366
diff
changeset
|
331 /* initialize random number generator for session IDs */ |
d1c40627673c
loadtest_cc: randomize session ids
Thomas Klausner <tk@giga.or.at>
parents:
1366
diff
changeset
|
332 srandom(time(NULL) | (unsigned int)getpid()); |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
333 if ((target = conffile) == NULL) { |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
334 fd_log_error("invalid conffile"); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
335 return EINVAL; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
336 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
337 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
338 memset(&data, 0, sizeof(data)); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
339 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
340 /* Advertise the support for the Diameter Credit Control application in the peer */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
341 CHECK_FCT( fd_dict_search( fd_g_config->cnf_dict, DICT_APPLICATION, APPLICATION_BY_NAME, "Diameter Credit Control Application", &data.app, ENOENT) ); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
342 CHECK_FCT( fd_disp_app_support ( data.app, NULL, 1, 0 ) ); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
343 |
1366 | 344 /* the handling of requests is different if it might be locally handled or not */ |
345 /* for possibly locally handled requests, fd_rt_fwd_register is not called, so we need to add handlers for both cases */ | |
346 /* register forward handler for CCR -- needs to look at all requests */ | |
347 CHECK_FCT(fd_rt_fwd_register(ccr_fwd_handler, NULL, RT_FWD_REQ, &ccr_fwd_hdl)); | |
348 /* register local handler for CCR - if not Destination-Host is set, or the local host is used */ | |
349 memset(&data, 0, sizeof(data)); | |
350 CHECK_FCT(fd_dict_search(fd_g_config->cnf_dict, DICT_COMMAND, CMD_BY_NAME, "Credit-Control-Answer", &data.command, ENOENT)); | |
351 CHECK_FCT(fd_disp_register(ccr_local_handler, DISP_HOW_CC, &data, NULL, &ccr_local_hdl)); | |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
352 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
353 CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Auth-Application-Id", &aai_avp_do, ENOENT), |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
354 { LOG_E("Unable to find 'Auth-Application-Id' AVP in the loaded dictionaries."); }); |
1366 | 355 CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "CC-Request-Number", &crn_avp_do, ENOENT), |
356 { LOG_E("Unable to find 'CC-Request-Number' AVP in the loaded dictionaries."); }); | |
357 CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "CC-Request-Type", &crt_avp_do, ENOENT), | |
358 | |
359 { LOG_E("Unable to find 'CC-Request-Type' AVP in the loaded dictionaries."); }); | |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
360 CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Destination-Host", &dh_avp_do, ENOENT), |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
361 { LOG_E("Unable to find 'Destination-Host' AVP in the loaded dictionaries."); }); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
362 CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Destination-Realm", &dr_avp_do, ENOENT), |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
363 { LOG_E("Unable to find 'Destination-Realm' AVP in the loaded dictionaries."); }); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
364 CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Result-Code", &rc_avp_do, ENOENT), |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
365 { LOG_E("Unable to find 'Result-Code' AVP in the loaded dictionaries."); }); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
366 CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Session-Id", &si_avp_do, ENOENT), |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
367 { LOG_E("Unable to find 'Session-Id' AVP in the loaded dictionaries."); }); |
1366 | 368 CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Service-Context-Id", &sci_avp_do, ENOENT), |
369 { LOG_E("Unable to find 'Service-Context-Id' AVP in the loaded dictionaries."); }); | |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
370 CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Proxy-Info", &pi_avp_do, ENOENT), |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
371 { LOG_E("Unable to find 'Proxy-Info' AVP in the loaded dictionaries."); }); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
372 CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Proxy-Host", &ph_avp_do, ENOENT), |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
373 { LOG_E("Unable to find 'Proxy-Host' AVP in the loaded dictionaries."); }); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
374 CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Proxy-State", &ps_avp_do, ENOENT), |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
375 { LOG_E("Unable to find 'Proxy-State' AVP in the loaded dictionaries."); }); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
376 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
377 /* Start the generator thread */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
378 CHECK_POSIX( pthread_create( &gen_thr, NULL, gen_thr_fct, NULL ) ); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
379 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
380 /* Register generator callback */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
381 CHECK_FCT(fd_event_trig_regcb(SIGUSR2, MODULE_NAME, sig_hdlr)); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
382 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
383 return 0; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
384 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
385 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
386 /* And terminate it */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
387 void fd_ext_fini(void) |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
388 { |
1366 | 389 uint64_t missing; |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
390 /* stop sending */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
391 do_generate = 0; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
392 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
393 /* Stop the expiry thread */ |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
394 CHECK_FCT_DO( fd_thr_term(&gen_thr), ); |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
395 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
396 /* Unregister the callbacks */ |
1366 | 397 if (ccr_local_hdl) { |
398 CHECK_FCT_DO( fd_disp_unregister(&ccr_local_hdl, NULL), ); | |
399 ccr_local_hdl = NULL; | |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
400 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
401 |
1366 | 402 missing = statistics.sent - statistics.error - statistics.success; |
403 | |
404 fd_log_error("%lld messages sent in %llds (%.2f messages/second), %lld success (%.2f%%), %lld errors (%.2f%%), %lld missing (%.2f%%)", | |
405 (long long)statistics.sent, (long long)(statistics.last-statistics.first), (float)statistics.sent / (statistics.last-statistics.first), | |
406 (long long)statistics.success, | |
407 100*(float)statistics.success/statistics.sent, (long long)statistics.error, 100*(float)statistics.error/statistics.sent, | |
408 missing, 100*(float)missing/statistics.sent); | |
409 | |
1364
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
410 return; |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
411 } |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
412 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
413 |
a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
Thomas Klausner <tk@giga.or.at>
parents:
diff
changeset
|
414 EXTENSION_ENTRY(MODULE_NAME, cc_entry); |