Mercurial > hg > freeDiameter
view extensions/loadtest_cc/loadtest_cc.c @ 1364:a37692441c4d
Snapshot commit of loadtest_cc extension, does not work.
author | Thomas Klausner <tk@giga.or.at> |
---|---|
date | Fri, 07 Jun 2019 14:27:30 +0200 |
parents | |
children | 9294ed1c7ac8 |
line wrap: on
line source
/********************************************************************************************************** * Software License Agreement(BSD License) * * Author: Thomas Klausner <tk@giga.or.at> * * * * Copyright(c) 2019, Thomas Klausner * * All rights reserved. * * * * Written under contract by nfotex IT GmbH, http://nfotex.com/ * * * * 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. * * * * 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. * **********************************************************************************************************/ /* This extension waits for a signal (SIGUSR2). When it gets it, it * generates messages as quickly as possible to the configured target * host; a second SIGUSR2 signal will stop this */ #include <freeDiameter/extension.h> #include <pthread.h> #include <signal.h> #define MODULE_NAME "loadtest_cc" static pthread_t gen_thr = (pthread_t)NULL; struct disp_hdl *ccr_handler_hdl; volatile int do_generate = 0; const char *target = NULL; struct dict_object * aai_avp_do; /* cache the Auth-Application-Id dictionary object */ struct dict_object * dh_avp_do; /* cache the Destination-Host dictionary object */ struct dict_object * dr_avp_do; /* cache the Destination-Realm dictionary object */ struct dict_object * rc_avp_do; /* cache the Result-Code dictionary object */ struct dict_object * si_avp_do; /* cache the Session-Id dictionary object */ struct dict_object * pi_avp_do; /* cache the Proxy-Info dictionary object */ struct dict_object * ph_avp_do; /* cache the Proxy-Host dictionary object */ struct dict_object * ps_avp_do; /* cache the Proxy-State dictionary object */ static int ccr_handler(struct msg ** msg, struct avp * avp, struct session * sess, void * data, enum disp_action * act) { struct msg_hdr *hdr = NULL; TRACE_ENTRY("%p %p %p %p", msg, avp, sess, act); if (msg == NULL) return EINVAL; CHECK_FCT(fd_msg_hdr(*msg, &hdr)); if (hdr->msg_flags & CMD_FLAG_REQUEST) { fd_log_error("received Credit-Control-Request, dropping it"); CHECK_FCT(fd_msg_free(*msg)); *msg = NULL; return 0; } /* Answer received, check it */ fd_log_notice("Credit-Control-Answer received, this code should do something about it but doesn't yet"); /* TODO */ CHECK_FCT(fd_msg_free(*msg)); *msg = NULL; return 0; } /* create message to send */ struct msg *create_message(const char *destination) { struct dict_object *command; struct msg *msg; struct avp *avp, *avp1; union avp_value val; struct msg_hdr *msg_hdr; const char *realm; const char *session_id = "fixed-session-id-for-now"; const char *proxy_host = "Dummy-Proxy-Host-to-Increase-Package-Size"; const char *proxy_state = "This is just date to increase the package size\nXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX"; if (fd_dict_search(fd_g_config->cnf_dict, DICT_COMMAND, CMD_BY_NAME, "Credit-Control-Request", &command, ENOENT) != 0) { fd_log_error("can't find template for 'Credit-Control-Request'"); return NULL; } if (fd_msg_new(command, MSGFL_ALLOC_ETEID, &msg) != 0) { fd_log_error("can't create new 'Credit-Control-Request' message"); return NULL; } /* Application Id in header needs to be set to for since this Credit-Control-Request is for Diameter Credit Control */ if (fd_msg_hdr(msg, &msg_hdr) != 0) { fd_log_error("can't get message header for 'Credit-Control-Request' message"); fd_msg_free(msg); return NULL; } msg_hdr->msg_appl = 4; if (fd_msg_add_origin(msg, 0) != 0) { fd_log_error("can't set Origin for 'Credit-Control-Request' message"); fd_msg_free(msg); return NULL; } /* Destination-Host */ fd_msg_avp_new(dh_avp_do, 0, &avp); memset(&val, 0, sizeof(val)); val.os.data = (uint8_t *)target; val.os.len = strlen(target); if (fd_msg_avp_setvalue(avp, &val) != 0) { fd_msg_free(msg); fd_log_error("can't set value for 'Destination-Host' for 'Credit-Control-Request' message"); return NULL; } fd_msg_avp_add(msg, MSG_BRW_LAST_CHILD, avp); if ((realm = strchr(target, '.')) == NULL) { fd_msg_free(msg); fd_log_error("can't extract realm from host '%s'", target); return NULL; } /* skip dot */ realm++; /* Destination-Realm */ fd_msg_avp_new(dr_avp_do, 0, &avp); memset(&val, 0, sizeof(val)); val.os.data = (uint8_t *)realm; val.os.len = strlen(realm); if (fd_msg_avp_setvalue(avp, &val) != 0) { fd_msg_free(msg); fd_log_error("can't set value for 'Destination-Realm' for 'Credit-Control-Request' message"); return NULL; } fd_msg_avp_add(msg, MSG_BRW_LAST_CHILD, avp); /* Session-Id */ fd_msg_avp_new(si_avp_do, 0, &avp); memset(&val, 0, sizeof(val)); val.os.data = (uint8_t *)session_id; val.os.len = strlen(session_id); if (fd_msg_avp_setvalue(avp, &val) != 0) { fd_msg_free(msg); fd_log_error("can't set value for 'Session-Id' for 'Credit-Control-Request' message"); return NULL; } fd_msg_avp_add(msg, MSG_BRW_FIRST_CHILD, avp); /* Auth-Application-Id */ fd_msg_avp_new(aai_avp_do, 0, &avp); memset(&val, 0, sizeof(val)); val.i32 = 4; if (fd_msg_avp_setvalue(avp, &val) != 0) { fd_msg_free(msg); fd_log_error("can't set value for 'Auth-Application-Id' for 'Credit-Control-Request' message"); return NULL; } fd_msg_avp_add(msg, MSG_BRW_LAST_CHILD, avp); /* Proxy-Info */ fd_msg_avp_new(pi_avp_do, 0, &avp); /* Proxy-Host */ fd_msg_avp_new(ph_avp_do, 0, &avp1); memset(&val, 0, sizeof(val)); val.os.data = (uint8_t *)proxy_host; val.os.len = strlen(proxy_host); if (fd_msg_avp_setvalue(avp1, &val) != 0) { fd_msg_free(msg); fd_log_error("can't set value for 'Proxy-Host' for 'Credit-Control-Request' message"); return NULL; } fd_msg_avp_add(avp, MSG_BRW_LAST_CHILD, avp1); /* Proxy-State */ fd_msg_avp_new(ps_avp_do, 0, &avp1); memset(&val, 0, sizeof(val)); val.os.data = (uint8_t *)proxy_state; val.os.len = strlen(proxy_state); if (fd_msg_avp_setvalue(avp1, &val) != 0) { fd_msg_free(msg); fd_log_error("can't set value for 'Proxy-State' for 'Credit-Control-Request' message"); return NULL; } fd_msg_avp_add(avp, MSG_BRW_LAST_CHILD, avp1); fd_msg_avp_add(msg, MSG_BRW_LAST_CHILD, avp); return msg; } /* The thread that handles expired entries cleanup. */ void * gen_thr_fct(void * arg) { struct msg *msg; fd_log_threadname ( "Loadtest/Generator" ); do { if (do_generate) { fd_msg_send(&msg, NULL, NULL); fd_log_notice("sent message"); sleep(1); } else { sleep(1); } } while (1); } /* signal handler */ static void sig_hdlr(void) { if (do_generate) { do_generate = 0; } else { do_generate = 1; } fd_log_notice("%s: switched generation of CCRs %s", MODULE_NAME, do_generate ? "on" : "off"); } /* entry hook: register callback */ static int cc_entry(char * conffile) { struct disp_when data; if ((target = conffile) == NULL) { fd_log_error("invalid conffile"); return EINVAL; } memset(&data, 0, sizeof(data)); /* Advertise the support for the Diameter Credit Control application in the peer */ CHECK_FCT( fd_dict_search( fd_g_config->cnf_dict, DICT_APPLICATION, APPLICATION_BY_NAME, "Diameter Credit Control Application", &data.app, ENOENT) ); CHECK_FCT( fd_disp_app_support ( data.app, NULL, 1, 0 ) ); /* register handler for CCR */ CHECK_FCT( fd_dict_search( fd_g_config->cnf_dict, DICT_COMMAND, CMD_BY_NAME, "Credit-Control-Request", &data.command, ENOENT) ); CHECK_FCT( fd_disp_register( ccr_handler, DISP_HOW_CC, &data, NULL, &ccr_handler_hdl ) ); CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Auth-Application-Id", &aai_avp_do, ENOENT), { LOG_E("Unable to find 'Auth-Application-Id' AVP in the loaded dictionaries."); }); CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Destination-Host", &dh_avp_do, ENOENT), { LOG_E("Unable to find 'Destination-Host' AVP in the loaded dictionaries."); }); CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Destination-Realm", &dr_avp_do, ENOENT), { LOG_E("Unable to find 'Destination-Realm' AVP in the loaded dictionaries."); }); CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Result-Code", &rc_avp_do, ENOENT), { LOG_E("Unable to find 'Result-Code' AVP in the loaded dictionaries."); }); CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Session-Id", &si_avp_do, ENOENT), { LOG_E("Unable to find 'Session-Id' AVP in the loaded dictionaries."); }); CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Proxy-Info", &pi_avp_do, ENOENT), { LOG_E("Unable to find 'Proxy-Info' AVP in the loaded dictionaries."); }); CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Proxy-Host", &ph_avp_do, ENOENT), { LOG_E("Unable to find 'Proxy-Host' AVP in the loaded dictionaries."); }); CHECK_FCT_DO(fd_dict_search(fd_g_config->cnf_dict, DICT_AVP, AVP_BY_NAME, "Proxy-State", &ps_avp_do, ENOENT), { LOG_E("Unable to find 'Proxy-State' AVP in the loaded dictionaries."); }); /* Start the generator thread */ CHECK_POSIX( pthread_create( &gen_thr, NULL, gen_thr_fct, NULL ) ); /* Register generator callback */ CHECK_FCT(fd_event_trig_regcb(SIGUSR2, MODULE_NAME, sig_hdlr)); return 0; } /* And terminate it */ void fd_ext_fini(void) { /* stop sending */ do_generate = 0; /* Stop the expiry thread */ CHECK_FCT_DO( fd_thr_term(&gen_thr), ); /* Unregister the callbacks */ if (ccr_handler_hdl) { CHECK_FCT_DO( fd_disp_unregister(&ccr_handler_hdl, NULL), ); ccr_handler_hdl = NULL; } return; } EXTENSION_ENTRY(MODULE_NAME, cc_entry);