123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273 |
- /*
- * Asterisk -- An open source telephony toolkit.
- *
- * Copyright (C) 2017, Greenfield Technologies Ltd.
- *
- * Nir Simionovich <nirs@greenfieldtech.net>
- * who freely borrowed code from the cel manager equivalents
- * (see cel/cel_manager.c)
- *
- * See http://www.asterisk.org for more information about
- * the Asterisk project. Please do not directly contact
- * any of the maintainers of this project for assistance;
- * the project provides a web site, mailing lists and IRC
- * channels for your use.
- *
- * This program is free software, distributed under the terms of
- * the GNU General Public License Version 2. See the LICENSE file
- * at the top of the source tree.
- */
- /*! \file
- *
- * \brief Asterisk Channel Event Beanstalkd backend
- *
- * This module requires the beanstalk-client library, avaialble from
- * https://github.com/deepfryed/beanstalk-client
- * \ingroup cel_drivers
- */
- /*! \li \ref cek_beanstalkd.c uses the configuration file \ref cel.conf
- * \addtogroup configuration_file Configuration Files
- */
- /*!
- * \page cel.conf cel.conf
- * \verbinclude cel.conf.sample
- */
- /*** MODULEINFO
- <depend>beanstalk</depend>
- <support_level>extended</support_level>
- ***/
- #include "asterisk.h"
- #include "asterisk/channel.h"
- #include "asterisk/cel.h"
- #include "asterisk/module.h"
- #include "asterisk/logger.h"
- #include "asterisk/utils.h"
- #include "asterisk/manager.h"
- #include "asterisk/config.h"
- #include "asterisk/json.h"
- #include "beanstalk.h"
- static const char DATE_FORMAT[] = "%Y-%m-%d %T";
- static const char CONF_FILE[] = "cel_beanstalkd.conf";
- /*! \brief Beanstalk CEL is off by default */
- #define CEL_BEANSTALK_ENABLED_DEFAULT 0
- static int enablecel;
- /*! \brief show_user_def is off by default */
- #define CEL_SHOW_USERDEF_DEFAULT 0
- #define CEL_BACKEND_NAME "Beanstalk Event Logging"
- #define BEANSTALK_JOB_SIZE 4096
- #define BEANSTALK_JOB_PRIORITY 99
- #define BEANSTALK_JOB_TTR 60
- #define BEANSTALK_JOB_DELAY 0
- #define DEFAULT_BEANSTALK_HOST "127.0.0.1"
- #define DEFAULT_BEANSTALK_PORT 11300
- #define DEFAULT_BEANSTALK_TUBE "asterisk-cel"
- static char *bs_host;
- static int bs_port;
- static char *bs_tube;
- static int priority;
- AST_RWLOCK_DEFINE_STATIC(config_lock);
- static void cel_bs_put(struct ast_event *event)
- {
- struct ast_tm timeresult;
- char start_time[80];
- char *cel_buffer;
- int bs_id;
- int bs_socket;
- struct ast_json *t_cel_json;
- struct ast_cel_event_record record = {
- .version = AST_CEL_EVENT_RECORD_VERSION,
- };
- if (!enablecel) {
- return;
- }
- if (ast_cel_fill_record(event, &record)) {
- return;
- }
- ast_rwlock_rdlock(&config_lock);
- bs_socket = bs_connect(bs_host, bs_port);
- if (bs_use(bs_socket, bs_tube) != BS_STATUS_OK) {
- ast_log(LOG_ERROR, "Connection to Beanstalk tube %s @ %s:%d had failed", bs_tube, bs_host, bs_port);
- ast_rwlock_unlock(&config_lock);
- return;
- }
- ast_localtime(&record.event_time, &timeresult, NULL);
- ast_strftime(start_time, sizeof(start_time), DATE_FORMAT, &timeresult);
- ast_rwlock_unlock(&config_lock);
- t_cel_json = ast_json_pack("{s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s}",
- "EventName", S_OR(record.event_name, ""),
- "AccountCode", S_OR(record.account_code, ""),
- "CallerIDnum", S_OR(record.caller_id_num, ""),
- "CallerIDname", S_OR(record.caller_id_name, ""),
- "CallerIDani", S_OR(record.caller_id_ani, ""),
- "CallerIDrdnis", S_OR(record.caller_id_rdnis, ""),
- "CallerIDdnid", S_OR(record.caller_id_dnid, ""),
- "Exten", S_OR(record.extension, ""),
- "Context", S_OR(record.context, ""),
- "Channel", S_OR(record.channel_name, ""),
- "Application", S_OR(record.application_name, ""),
- "AppData", S_OR(record.application_data, ""),
- "EventTime", S_OR(start_time, ""),
- "AMAFlags", S_OR(ast_channel_amaflags2string(record.amaflag), ""),
- "UniqueID", S_OR(record.unique_id, ""),
- "LinkedID", S_OR(record.linked_id, ""),
- "Userfield", S_OR(record.user_field, ""),
- "Peer", S_OR(record.peer_account, ""),
- "PeerAccount", S_OR(record.peer_account, ""),
- "Extra", S_OR(record.extra, "")
- );
- cel_buffer = ast_json_dump_string(t_cel_json);
- ast_json_unref(t_cel_json);
- bs_id = bs_put(bs_socket, priority, BEANSTALK_JOB_DELAY, BEANSTALK_JOB_TTR, cel_buffer, strlen(cel_buffer));
- if (bs_id > 0) {
- ast_log(LOG_DEBUG, "Successfully created job %d with %s\n", bs_id, cel_buffer);
- } else {
- ast_log(LOG_ERROR, "CDR job creation failed for %s\n", cel_buffer);
- }
- bs_disconnect(bs_socket);
- ast_json_free(cel_buffer);
- }
- static int load_config(int reload)
- {
- const char *cat = NULL;
- struct ast_config *cfg;
- struct ast_flags config_flags = { reload ? CONFIG_FLAG_FILEUNCHANGED : 0 };
- struct ast_variable *v;
- int newenablecel = CEL_BEANSTALK_ENABLED_DEFAULT;
- cfg = ast_config_load(CONF_FILE, config_flags);
- if (cfg == CONFIG_STATUS_FILEUNCHANGED) {
- return 0;
- }
- if (cfg == CONFIG_STATUS_FILEINVALID) {
- ast_log(LOG_WARNING, "Configuration file '%s' is invalid. CEL Beanstalkd Module not activated.\n",
- CONF_FILE);
- return -1;
- } else if (!cfg) {
- ast_log(LOG_WARNING, "Failed to load configuration file. CEL Beanstalkd Module not activated.\n");
- if (enablecel) {
- ast_cel_backend_unregister(CEL_BACKEND_NAME);
- }
- enablecel = 0;
- return -1;
- }
- if (reload) {
- ast_rwlock_wrlock(&config_lock);
- ast_free(bs_host);
- ast_free(bs_tube);
- }
- /* Bootstrap the default configuration */
- bs_host = ast_strdup(DEFAULT_BEANSTALK_HOST);
- bs_port = DEFAULT_BEANSTALK_PORT;
- bs_tube = ast_strdup(DEFAULT_BEANSTALK_TUBE);
- priority = BEANSTALK_JOB_PRIORITY;
- while ((cat = ast_category_browse(cfg, cat))) {
- if (strcasecmp(cat, "general")) {
- continue;
- }
- for (v = ast_variable_browse(cfg, cat); v; v = v->next) {
- if (!strcasecmp(v->name, "enabled")) {
- newenablecel = ast_true(v->value) ? 1 : 0;
- } else if (!strcasecmp(v->name, "host")) {
- ast_free(bs_host);
- bs_host = ast_strdup(v->value);
- } else if (!strcasecmp(v->name, "port")) {
- bs_port = atoi(v->value);
- } else if (!strcasecmp(v->name, "tube")) {
- ast_free(bs_tube);
- bs_tube = ast_strdup(v->value);
- } else if (!strcasecmp(v->name, "priority")) {
- priority = atoi(v->value);
- } else {
- ast_log(LOG_NOTICE, "Unknown option '%s' specified "
- "for CEL beanstalk backend.\n", v->name);
- }
- }
- }
- if (reload) {
- ast_rwlock_unlock(&config_lock);
- }
- ast_config_destroy(cfg);
- if (enablecel && !newenablecel) {
- ast_cel_backend_unregister(CEL_BACKEND_NAME);
- } else if (!enablecel && newenablecel) {
- if (ast_cel_backend_register(CEL_BACKEND_NAME, cel_bs_put)) {
- ast_log(LOG_ERROR, "Unable to register Beanstalkd CEL handling\n");
- }
- }
- enablecel = newenablecel;
- return 0;
- }
- static int unload_module(void)
- {
- ast_cel_backend_unregister(CEL_BACKEND_NAME);
- ast_free(bs_host);
- ast_free(bs_tube);
- return 0;
- }
- static int load_module(void)
- {
- if (load_config(0)) {
- return AST_MODULE_LOAD_DECLINE;
- }
- return AST_MODULE_LOAD_SUCCESS;
- }
- static int reload(void)
- {
- return load_config(1);
- }
- AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "Beanstalkd CEL Backend",
- .support_level = AST_MODULE_SUPPORT_EXTENDED,
- .load = load_module,
- .unload = unload_module,
- .reload = reload,
- .load_pri = AST_MODPRI_CDR_DRIVER,
- .requires = "cel",
- );
|