123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707 |
- /*
- * Asterisk -- An open source telephony toolkit.
- *
- * James Sharp <jsharp@psychoses.org>
- *
- * Modified August 2003
- * Tilghman Lesher <asterisk__cdr__cdr_mysql__200308@the-tilghman.com>
- *
- * Modified August 6, 2005
- * Joseph Benden <joe@thrallingpenguin.com>
- * Added mysql connection timeout parameter
- * Added an automatic reconnect as to not lose a cdr record
- * Cleaned up the original code to match the coding guidelines
- *
- * Modified Juli 2006
- * Martin Portmann <map@infinitum.ch>
- * Added mysql ssl support
- *
- * 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 MySQL CDR backend
- * \ingroup cdr_drivers
- */
- /*** MODULEINFO
- <depend>mysqlclient</depend>
- <defaultenabled>no</defaultenabled>
- <support_level>deprecated</support_level>
- <replacement>cdr_adaptive_odbc</replacement>
- ***/
- #include "asterisk.h"
- ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
- #include <mysql/mysql.h>
- #include <mysql/errmsg.h>
- #include "asterisk/config.h"
- #include "asterisk/options.h"
- #include "asterisk/channel.h"
- #include "asterisk/cdr.h"
- #include "asterisk/module.h"
- #include "asterisk/logger.h"
- #include "asterisk/cli.h"
- #include "asterisk/strings.h"
- #include "asterisk/linkedlists.h"
- #include "asterisk/threadstorage.h"
- #define DATE_FORMAT "%Y-%m-%d %T"
- AST_THREADSTORAGE(sql1_buf);
- AST_THREADSTORAGE(sql2_buf);
- AST_THREADSTORAGE(escape_buf);
- static const char desc[] = "MySQL CDR Backend";
- static const char name[] = "mysql";
- static const char config[] = "cdr_mysql.conf";
- static struct ast_str *hostname = NULL, *dbname = NULL, *dbuser = NULL, *password = NULL, *dbsock = NULL, *dbtable = NULL, *dbcharset = NULL, *cdrzone = NULL;
- static struct ast_str *ssl_ca = NULL, *ssl_cert = NULL, *ssl_key = NULL;
- static int dbport = 0;
- static int connected = 0;
- static time_t connect_time = 0;
- static int records = 0;
- static int totalrecords = 0;
- static int timeout = 0;
- static int calldate_compat = 0;
- AST_MUTEX_DEFINE_STATIC(mysql_lock);
- struct unload_string {
- AST_LIST_ENTRY(unload_string) entry;
- struct ast_str *str;
- };
- static AST_LIST_HEAD_STATIC(unload_strings, unload_string);
- struct column {
- char *name;
- char *cdrname;
- char *staticvalue;
- char *type;
- AST_LIST_ENTRY(column) list;
- };
- /* Protected with mysql_lock */
- static AST_RWLIST_HEAD_STATIC(columns, column);
- static MYSQL mysql = { { NULL }, };
- static char *handle_cli_cdr_mysql_status(struct ast_cli_entry *e, int cmd, struct ast_cli_args *a)
- {
- switch (cmd) {
- case CLI_INIT:
- e->command = "cdr mysql status";
- e->usage =
- "Usage: cdr mysql status\n"
- " Shows current connection status for cdr_mysql\n";
- return NULL;
- case CLI_GENERATE:
- return NULL;
- }
- if (a->argc != 3)
- return CLI_SHOWUSAGE;
- if (connected) {
- char status[256], status2[100] = "";
- int ctime = time(NULL) - connect_time;
- if (dbport)
- snprintf(status, 255, "Connected to %s@%s, port %d", ast_str_buffer(dbname), ast_str_buffer(hostname), dbport);
- else if (dbsock)
- snprintf(status, 255, "Connected to %s on socket file %s", ast_str_buffer(dbname), S_OR(ast_str_buffer(dbsock), "default"));
- else
- snprintf(status, 255, "Connected to %s@%s", ast_str_buffer(dbname), ast_str_buffer(hostname));
- if (!ast_strlen_zero(ast_str_buffer(dbuser)))
- snprintf(status2, 99, " with username %s", ast_str_buffer(dbuser));
- if (ast_str_strlen(dbtable))
- snprintf(status2, 99, " using table %s", ast_str_buffer(dbtable));
- if (ctime > 31536000) {
- ast_cli(a->fd, "%s%s for %d years, %d days, %d hours, %d minutes, %d seconds.\n", status, status2, ctime / 31536000, (ctime % 31536000) / 86400, (ctime % 86400) / 3600, (ctime % 3600) / 60, ctime % 60);
- } else if (ctime > 86400) {
- ast_cli(a->fd, "%s%s for %d days, %d hours, %d minutes, %d seconds.\n", status, status2, ctime / 86400, (ctime % 86400) / 3600, (ctime % 3600) / 60, ctime % 60);
- } else if (ctime > 3600) {
- ast_cli(a->fd, "%s%s for %d hours, %d minutes, %d seconds.\n", status, status2, ctime / 3600, (ctime % 3600) / 60, ctime % 60);
- } else if (ctime > 60) {
- ast_cli(a->fd, "%s%s for %d minutes, %d seconds.\n", status, status2, ctime / 60, ctime % 60);
- } else {
- ast_cli(a->fd, "%s%s for %d seconds.\n", status, status2, ctime);
- }
- if (records == totalrecords)
- ast_cli(a->fd, " Wrote %d records since last restart.\n", totalrecords);
- else
- ast_cli(a->fd, " Wrote %d records since last restart and %d records since last reconnect.\n", totalrecords, records);
- } else {
- ast_cli(a->fd, "Not currently connected to a MySQL server.\n");
- }
- return CLI_SUCCESS;
- }
- static struct ast_cli_entry cdr_mysql_status_cli[] = {
- AST_CLI_DEFINE(handle_cli_cdr_mysql_status, "Show connection status of cdr_mysql"),
- };
- static int mysql_log(struct ast_cdr *cdr)
- {
- struct ast_str *sql1 = ast_str_thread_get(&sql1_buf, 1024), *sql2 = ast_str_thread_get(&sql2_buf, 1024);
- int retries = 5;
- #if MYSQL_VERSION_ID >= 50013
- my_bool my_bool_true = 1;
- #endif
- if (!sql1 || !sql2) {
- ast_log(LOG_ERROR, "Memory error\n");
- return -1;
- }
- ast_mutex_lock(&mysql_lock);
- db_reconnect:
- if ((!connected) && (hostname || dbsock) && dbuser && password && dbname && dbtable ) {
- /* Attempt to connect */
- mysql_init(&mysql);
- /* Add option to quickly timeout the connection */
- if (timeout && mysql_options(&mysql, MYSQL_OPT_CONNECT_TIMEOUT, (char *)&timeout) != 0) {
- ast_log(LOG_ERROR, "mysql_options returned (%d) %s\n", mysql_errno(&mysql), mysql_error(&mysql));
- }
- #if MYSQL_VERSION_ID >= 50013
- /* Add option for automatic reconnection */
- if (mysql_options(&mysql, MYSQL_OPT_RECONNECT, &my_bool_true) != 0) {
- ast_log(LOG_ERROR, "mysql_options returned (%d) %s\n", mysql_errno(&mysql), mysql_error(&mysql));
- }
- #endif
- if (ssl_ca || ssl_cert || ssl_key) {
- mysql_ssl_set(&mysql, ssl_key ? ast_str_buffer(ssl_key) : NULL, ssl_cert ? ast_str_buffer(ssl_cert) : NULL, ssl_ca ? ast_str_buffer(ssl_ca) : NULL, NULL, NULL);
- }
- if (mysql_real_connect(&mysql, ast_str_buffer(hostname), ast_str_buffer(dbuser), ast_str_buffer(password), ast_str_buffer(dbname), dbport, dbsock && ast_str_strlen(dbsock) ? ast_str_buffer(dbsock) : NULL, ssl_ca ? CLIENT_SSL : 0)) {
- connected = 1;
- connect_time = time(NULL);
- records = 0;
- if (dbcharset) {
- ast_str_set(&sql1, 0, "SET NAMES '%s'", ast_str_buffer(dbcharset));
- mysql_real_query(&mysql, ast_str_buffer(sql1), ast_str_strlen(sql1));
- ast_debug(1, "SQL command as follows: %s\n", ast_str_buffer(sql1));
- }
- } else {
- ast_log(LOG_ERROR, "Cannot connect to database server %s: (%d) %s\n", ast_str_buffer(hostname), mysql_errno(&mysql), mysql_error(&mysql));
- connected = 0;
- }
- } else {
- /* Long connection - ping the server */
- int error;
- if ((error = mysql_ping(&mysql))) {
- connected = 0;
- records = 0;
- switch (mysql_errno(&mysql)) {
- case CR_SERVER_GONE_ERROR:
- case CR_SERVER_LOST:
- ast_log(LOG_ERROR, "Server has gone away. Attempting to reconnect.\n");
- break;
- default:
- ast_log(LOG_ERROR, "Unknown connection error: (%d) %s\n", mysql_errno(&mysql), mysql_error(&mysql));
- }
- retries--;
- if (retries) {
- goto db_reconnect;
- } else {
- ast_log(LOG_ERROR, "Retried to connect five times, giving up.\n");
- }
- }
- }
- if (connected) {
- int column_count = 0;
- char *cdrname;
- char workspace[2048], *value = NULL;
- struct column *entry;
- struct ast_str *escape = ast_str_thread_get(&escape_buf, 16);
- ast_str_set(&sql1, 0, "INSERT INTO %s (", AS_OR(dbtable, "cdr"));
- ast_str_set(&sql2, 0, ") VALUES (");
- AST_RWLIST_RDLOCK(&columns);
- AST_RWLIST_TRAVERSE(&columns, entry, list) {
- if (!strcmp(entry->name, "calldate")) {
- /*!\note
- * For some dumb reason, "calldate" used to be formulated using
- * the datetime the record was posted, rather than the start
- * time of the call. If someone really wants the old compatible
- * behavior, it's provided here.
- */
- if (calldate_compat) {
- struct timeval tv = ast_tvnow();
- struct ast_tm tm;
- char timestr[128];
- ast_localtime(&tv, &tm, ast_str_strlen(cdrzone) ? ast_str_buffer(cdrzone) : NULL);
- ast_strftime(timestr, sizeof(timestr), "%Y-%m-%d %T", &tm);
- value = ast_strdupa(timestr);
- cdrname = "calldate";
- } else {
- cdrname = "start";
- }
- } else {
- cdrname = entry->cdrname;
- }
- /* Construct SQL */
- /* Need the type and value to determine if we want the raw value or not */
- if (entry->staticvalue) {
- value = ast_strdupa(entry->staticvalue);
- } else if ((!strcmp(cdrname, "start") ||
- !strcmp(cdrname, "answer") ||
- !strcmp(cdrname, "end") ||
- !strcmp(cdrname, "disposition") ||
- !strcmp(cdrname, "amaflags")) &&
- (strstr(entry->type, "int") ||
- strstr(entry->type, "dec") ||
- strstr(entry->type, "float") ||
- strstr(entry->type, "double") ||
- strstr(entry->type, "real") ||
- strstr(entry->type, "numeric") ||
- strstr(entry->type, "fixed"))) {
- ast_cdr_format_var(cdr, cdrname, &value, workspace, sizeof(workspace), 1);
- } else if (!strcmp(cdrname, "calldate")) {
- /* Skip calldate - the value has already been dup'd */
- } else {
- ast_cdr_format_var(cdr, cdrname, &value, workspace, sizeof(workspace), 0);
- }
- if (value) {
- size_t valsz;
- if (column_count++) {
- ast_str_append(&sql1, 0, ",");
- ast_str_append(&sql2, 0, ",");
- }
- if (!strcasecmp(cdrname, "billsec") &&
- (strstr(entry->type, "float") ||
- strstr(entry->type, "double") ||
- strstr(entry->type, "decimal") ||
- strstr(entry->type, "numeric") ||
- strstr(entry->type, "real"))) {
- if (!ast_tvzero(cdr->answer)) {
- snprintf(workspace, sizeof(workspace), "%lf",
- (double) (ast_tvdiff_us(cdr->end, cdr->answer) / 1000000.0));
- } else {
- ast_copy_string(workspace, "0", sizeof(workspace));
- }
- if (!ast_strlen_zero(workspace)) {
- value = workspace;
- }
- }
- if (!strcasecmp(cdrname, "duration") &&
- (strstr(entry->type, "float") ||
- strstr(entry->type, "double") ||
- strstr(entry->type, "decimal") ||
- strstr(entry->type, "numeric") ||
- strstr(entry->type, "real"))) {
- snprintf(workspace, sizeof(workspace), "%lf",
- (double) (ast_tvdiff_us(cdr->end, cdr->start) / 1000000.0));
- if (!ast_strlen_zero(workspace)) {
- value = workspace;
- }
- }
- ast_str_make_space(&escape, (valsz = strlen(value)) * 2 + 1);
- mysql_real_escape_string(&mysql, ast_str_buffer(escape), value, valsz);
- ast_str_append(&sql1, 0, "`%s`", entry->name);
- ast_str_append(&sql2, 0, "'%s'", ast_str_buffer(escape));
- }
- }
- AST_RWLIST_UNLOCK(&columns);
- ast_debug(1, "Inserting a CDR record.\n");
- ast_str_append(&sql1, 0, "%s)", ast_str_buffer(sql2));
- ast_debug(1, "SQL command as follows: %s\n", ast_str_buffer(sql1));
- if (mysql_real_query(&mysql, ast_str_buffer(sql1), ast_str_strlen(sql1))) {
- ast_log(LOG_ERROR, "Failed to insert into database: (%d) %s\n", mysql_errno(&mysql), mysql_error(&mysql));
- mysql_close(&mysql);
- connected = 0;
- } else {
- records++;
- totalrecords++;
- }
- }
- ast_mutex_unlock(&mysql_lock);
- return 0;
- }
- static int my_unload_module(int reload)
- {
- struct unload_string *us;
- struct column *entry;
- ast_cli_unregister_multiple(cdr_mysql_status_cli, sizeof(cdr_mysql_status_cli) / sizeof(struct ast_cli_entry));
- if (connected) {
- mysql_close(&mysql);
- connected = 0;
- records = 0;
- }
- AST_LIST_LOCK(&unload_strings);
- while ((us = AST_LIST_REMOVE_HEAD(&unload_strings, entry))) {
- ast_free(us->str);
- ast_free(us);
- }
- AST_LIST_UNLOCK(&unload_strings);
- if (!reload) {
- AST_RWLIST_WRLOCK(&columns);
- }
- while ((entry = AST_RWLIST_REMOVE_HEAD(&columns, list))) {
- ast_free(entry);
- }
- if (!reload) {
- AST_RWLIST_UNLOCK(&columns);
- }
- dbport = 0;
- if (reload) {
- return ast_cdr_backend_suspend(name);
- } else {
- return ast_cdr_unregister(name);
- }
- }
- static int my_load_config_string(struct ast_config *cfg, const char *category, const char *variable, struct ast_str **field, const char *def)
- {
- struct unload_string *us;
- const char *tmp;
- if (!(us = ast_calloc(1, sizeof(*us))))
- return -1;
- if (!(*field = ast_str_create(16))) {
- ast_free(us);
- return -1;
- }
- tmp = ast_variable_retrieve(cfg, category, variable);
- ast_str_set(field, 0, "%s", tmp ? tmp : def);
- us->str = *field;
- AST_LIST_LOCK(&unload_strings);
- AST_LIST_INSERT_HEAD(&unload_strings, us, entry);
- AST_LIST_UNLOCK(&unload_strings);
- return 0;
- }
- static int my_load_config_number(struct ast_config *cfg, const char *category, const char *variable, int *field, int def)
- {
- const char *tmp;
- tmp = ast_variable_retrieve(cfg, category, variable);
- if (!tmp || sscanf(tmp, "%30d", field) < 1)
- *field = def;
- return 0;
- }
- static int my_load_module(int reload)
- {
- int res;
- struct ast_config *cfg;
- struct ast_variable *var;
- /* CONFIG_STATUS_FILEUNCHANGED is impossible when config_flags is always 0,
- * and it has to be zero, so a reload can be sent to tell the driver to
- * rescan the table layout. */
- struct ast_flags config_flags = { 0 };
- struct column *entry;
- char *temp;
- struct ast_str *compat;
- MYSQL_ROW row;
- MYSQL_RES *result;
- char sqldesc[128];
- #if MYSQL_VERSION_ID >= 50013
- my_bool my_bool_true = 1;
- #endif
- /* Cannot use a conditionally different flag, because the table layout may
- * have changed, which is not detectable by config file change detection,
- * but should still cause the configuration to be re-parsed. */
- cfg = ast_config_load(config, config_flags);
- if (cfg == CONFIG_STATUS_FILEMISSING) {
- ast_log(LOG_WARNING, "Unable to load config for mysql CDR's: %s\n", config);
- return AST_MODULE_LOAD_SUCCESS;
- } else if (cfg == CONFIG_STATUS_FILEINVALID) {
- ast_log(LOG_ERROR, "Unable to load configuration file '%s'\n", config);
- return AST_MODULE_LOAD_DECLINE;
- }
- if (reload) {
- AST_RWLIST_WRLOCK(&columns);
- my_unload_module(1);
- }
- var = ast_variable_browse(cfg, "global");
- if (!var) {
- /* nothing configured */
- if (reload) {
- AST_RWLIST_UNLOCK(&columns);
- }
- ast_config_destroy(cfg);
- return AST_MODULE_LOAD_SUCCESS;
- }
- res = 0;
- res |= my_load_config_string(cfg, "global", "hostname", &hostname, "localhost");
- res |= my_load_config_string(cfg, "global", "dbname", &dbname, "astriskcdrdb");
- res |= my_load_config_string(cfg, "global", "user", &dbuser, "root");
- res |= my_load_config_string(cfg, "global", "sock", &dbsock, "");
- res |= my_load_config_string(cfg, "global", "table", &dbtable, "cdr");
- res |= my_load_config_string(cfg, "global", "password", &password, "");
- res |= my_load_config_string(cfg, "global", "charset", &dbcharset, "");
- res |= my_load_config_string(cfg, "global", "ssl_ca", &ssl_ca, "");
- res |= my_load_config_string(cfg, "global", "ssl_cert", &ssl_cert, "");
- res |= my_load_config_string(cfg, "global", "ssl_key", &ssl_key, "");
- res |= my_load_config_number(cfg, "global", "port", &dbport, 0);
- res |= my_load_config_number(cfg, "global", "timeout", &timeout, 0);
- res |= my_load_config_string(cfg, "global", "compat", &compat, "no");
- res |= my_load_config_string(cfg, "global", "cdrzone", &cdrzone, "");
- if (ast_str_strlen(cdrzone) == 0) {
- for (; var; var = var->next) {
- if (!strcasecmp(var->name, "usegmtime") && ast_true(var->value)) {
- ast_str_set(&cdrzone, 0, "UTC");
- }
- }
- }
- if (ast_true(ast_str_buffer(compat))) {
- calldate_compat = 1;
- } else {
- calldate_compat = 0;
- }
- if (res < 0) {
- if (reload) {
- AST_RWLIST_UNLOCK(&columns);
- }
- ast_config_destroy(cfg);
- return AST_MODULE_LOAD_FAILURE;
- }
- /* Check for any aliases */
- if (!reload) {
- /* Lock, if not already */
- AST_RWLIST_WRLOCK(&columns);
- }
- while ((entry = AST_LIST_REMOVE_HEAD(&columns, list))) {
- ast_free(entry);
- }
- ast_debug(1, "Got hostname of %s\n", ast_str_buffer(hostname));
- ast_debug(1, "Got port of %d\n", dbport);
- ast_debug(1, "Got a timeout of %d\n", timeout);
- if (dbsock)
- ast_debug(1, "Got sock file of %s\n", ast_str_buffer(dbsock));
- ast_debug(1, "Got user of %s\n", ast_str_buffer(dbuser));
- ast_debug(1, "Got dbname of %s\n", ast_str_buffer(dbname));
- ast_debug(1, "Got password of %s\n", ast_str_buffer(password));
- ast_debug(1, "%sunning in calldate compatibility mode\n", calldate_compat ? "R" : "Not r");
- ast_debug(1, "Dates and times are localized to %s\n", S_OR(ast_str_buffer(cdrzone), "local timezone"));
- if (dbcharset) {
- ast_debug(1, "Got DB charset of %s\n", ast_str_buffer(dbcharset));
- }
- mysql_init(&mysql);
- if (timeout && mysql_options(&mysql, MYSQL_OPT_CONNECT_TIMEOUT, (char *)&timeout) != 0) {
- ast_log(LOG_ERROR, "cdr_mysql: mysql_options returned (%d) %s\n", mysql_errno(&mysql), mysql_error(&mysql));
- }
- #if MYSQL_VERSION_ID >= 50013
- /* Add option for automatic reconnection */
- if (mysql_options(&mysql, MYSQL_OPT_RECONNECT, &my_bool_true) != 0) {
- ast_log(LOG_ERROR, "cdr_mysql: mysql_options returned (%d) %s\n", mysql_errno(&mysql), mysql_error(&mysql));
- }
- #endif
- if ((ssl_ca && ast_str_strlen(ssl_ca)) || (ssl_cert && ast_str_strlen(ssl_cert)) || (ssl_key && ast_str_strlen(ssl_key))) {
- mysql_ssl_set(&mysql,
- ssl_key ? ast_str_buffer(ssl_key) : NULL,
- ssl_cert ? ast_str_buffer(ssl_cert) : NULL,
- ssl_ca ? ast_str_buffer(ssl_ca) : NULL,
- NULL, NULL);
- }
- temp = dbsock && ast_str_strlen(dbsock) ? ast_str_buffer(dbsock) : NULL;
- if (!mysql_real_connect(&mysql, ast_str_buffer(hostname), ast_str_buffer(dbuser), ast_str_buffer(password), ast_str_buffer(dbname), dbport, temp, ssl_ca && ast_str_strlen(ssl_ca) ? CLIENT_SSL : 0)) {
- ast_log(LOG_ERROR, "Failed to connect to mysql database %s on %s.\n", ast_str_buffer(dbname), ast_str_buffer(hostname));
- connected = 0;
- records = 0;
- } else {
- ast_debug(1, "Successfully connected to MySQL database.\n");
- connected = 1;
- records = 0;
- connect_time = time(NULL);
- if (dbcharset) {
- snprintf(sqldesc, sizeof(sqldesc), "SET NAMES '%s'", ast_str_buffer(dbcharset));
- mysql_real_query(&mysql, sqldesc, strlen(sqldesc));
- ast_debug(1, "SQL command as follows: %s\n", sqldesc);
- }
- /* Get table description */
- snprintf(sqldesc, sizeof(sqldesc), "DESC %s", dbtable ? ast_str_buffer(dbtable) : "cdr");
- if (mysql_query(&mysql, sqldesc)) {
- ast_log(LOG_ERROR, "Unable to query table description!! Logging disabled.\n");
- mysql_close(&mysql);
- connected = 0;
- AST_RWLIST_UNLOCK(&columns);
- ast_config_destroy(cfg);
- return AST_MODULE_LOAD_FAILURE;
- }
- if (!(result = mysql_store_result(&mysql))) {
- ast_log(LOG_ERROR, "Unable to query table description!! Logging disabled.\n");
- mysql_close(&mysql);
- connected = 0;
- AST_RWLIST_UNLOCK(&columns);
- ast_config_destroy(cfg);
- return AST_MODULE_LOAD_FAILURE;
- }
- while ((row = mysql_fetch_row(result))) {
- struct column *entry;
- char *cdrvar = "", *staticvalue = "";
- ast_debug(1, "Got a field '%s' of type '%s'\n", row[0], row[1]);
- /* Check for an alias or a static value */
- for (var = ast_variable_browse(cfg, "columns"); var; var = var->next) {
- if (strncmp(var->name, "alias", 5) == 0 && strcasecmp(var->value, row[0]) == 0 ) {
- char *alias = ast_strdupa(var->name + 5);
- cdrvar = ast_strip(alias);
- ast_verb(3, "Found alias %s for column %s\n", cdrvar, row[0]);
- break;
- } else if (strncmp(var->name, "static", 6) == 0 && strcasecmp(var->value, row[0]) == 0) {
- char *item = ast_strdupa(var->name + 6);
- item = ast_strip(item);
- if (item[0] == '"' && item[strlen(item) - 1] == '"') {
- /* Remove surrounding quotes */
- item[strlen(item) - 1] = '\0';
- item++;
- }
- staticvalue = item;
- }
- }
- entry = ast_calloc(sizeof(char), sizeof(*entry) + strlen(row[0]) + 1 + strlen(cdrvar) + 1 + strlen(staticvalue) + 1 + strlen(row[1]) + 1);
- if (!entry) {
- ast_log(LOG_ERROR, "Out of memory creating entry for column '%s'\n", row[0]);
- res = -1;
- break;
- }
- entry->name = (char *)entry + sizeof(*entry);
- strcpy(entry->name, row[0]);
- if (!ast_strlen_zero(cdrvar)) {
- entry->cdrname = entry->name + strlen(row[0]) + 1;
- strcpy(entry->cdrname, cdrvar);
- } else { /* Point to same place as the column name */
- entry->cdrname = (char *)entry + sizeof(*entry);
- }
- if (!ast_strlen_zero(staticvalue)) {
- entry->staticvalue = entry->cdrname + strlen(entry->cdrname) + 1;
- strcpy(entry->staticvalue, staticvalue);
- ast_debug(1, "staticvalue length: %d\n", (int) strlen(staticvalue) );
- entry->type = entry->staticvalue + strlen(entry->staticvalue) + 1;
- } else {
- entry->type = entry->cdrname + strlen(entry->cdrname) + 1;
- }
- strcpy(entry->type, row[1]);
- ast_debug(1, "Entry name '%s'\n", entry->name);
- ast_debug(1, " cdrname '%s'\n", entry->cdrname);
- ast_debug(1, " static '%s'\n", entry->staticvalue);
- ast_debug(1, " type '%s'\n", entry->type);
- AST_LIST_INSERT_TAIL(&columns, entry, list);
- }
- mysql_free_result(result);
- }
- AST_RWLIST_UNLOCK(&columns);
- ast_config_destroy(cfg);
- if (res < 0) {
- return AST_MODULE_LOAD_FAILURE;
- }
- if (!reload) {
- res = ast_cdr_register(name, desc, mysql_log);
- } else {
- res = ast_cdr_backend_unsuspend(name);
- }
- if (res) {
- ast_log(LOG_ERROR, "Unable to register MySQL CDR handling\n");
- } else {
- res = ast_cli_register_multiple(cdr_mysql_status_cli, sizeof(cdr_mysql_status_cli) / sizeof(struct ast_cli_entry));
- }
- return res;
- }
- static int load_module(void)
- {
- return my_load_module(0);
- }
- static int unload_module(void)
- {
- return my_unload_module(0);
- }
- static int reload(void)
- {
- int ret;
- ast_mutex_lock(&mysql_lock);
- ret = my_load_module(1);
- ast_mutex_unlock(&mysql_lock);
- return ret;
- }
- AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_DEFAULT, "MySQL CDR Backend",
- .support_level = AST_MODULE_SUPPORT_DEPRECATED,
- .load = load_module,
- .unload = unload_module,
- .reload = reload,
- );
|