diff options
Diffstat (limited to 'cdr')
| -rw-r--r-- | cdr/cdr_sqlite.c | 5 | ||||
| -rw-r--r-- | cdr/cdr_sqlite3_custom.c | 264 |
2 files changed, 269 insertions, 0 deletions
diff --git a/cdr/cdr_sqlite.c b/cdr/cdr_sqlite.c index bcc50aa0e..bd232b450 100644 --- a/cdr/cdr_sqlite.c +++ b/cdr/cdr_sqlite.c @@ -30,6 +30,8 @@ * * Creates the database and table on-the-fly * \ingroup cdr_drivers + * + * \note This module has been marked deprecated in favor for cdr_sqlite3_custom */ /*** MODULEINFO @@ -180,6 +182,9 @@ static int load_module(void) char fn[PATH_MAX]; int res; + ast_log(LOG_WARNING, "This module has been marked deprecated in favor of " + "using cdr_sqlite3_custom. (May be removed after Asterisk 1.6)\n"); + /* is the database there? */ snprintf(fn, sizeof(fn), "%s/cdr.db", ast_config_AST_LOG_DIR); db = sqlite_open(fn, AST_FILE_MODE, &zErr); diff --git a/cdr/cdr_sqlite3_custom.c b/cdr/cdr_sqlite3_custom.c new file mode 100644 index 000000000..c3dcbd269 --- /dev/null +++ b/cdr/cdr_sqlite3_custom.c @@ -0,0 +1,264 @@ +/* + * Asterisk -- An open source telephony toolkit. + * + * Copyright (C) 1999 - 2007, Digium, Inc. + * + * Mark Spencer <markster@digium.com> and others. + * + * 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 Custom SQLite3 CDR records. + * + * \author Adapted by Alejandro Rios <alejandro.rios@avatar.com.co> and + * Russell Bryant <russell@digium.com> from + * cdr_mysql_custom by Edward Eastman <ed@dm3.co.uk>, + * and cdr_sqlite by Holger Schurig <hs4233@mail.mn-solutions.de> + * + * + * \arg See also \ref AstCDR + * + * + * \ingroup cdr_drivers + */ + +/*** MODULEINFO + <depend>sqlite3</depend> + ***/ + +#include "asterisk.h" + +ASTERISK_FILE_VERSION(__FILE__, "$Revision$") + +#include <stdlib.h> +#include <stdio.h> +#include <string.h> +#include <errno.h> +#include <unistd.h> +#include <time.h> +#include <sys/types.h> +#include <sqlite3.h> + +#include "asterisk/channel.h" +#include "asterisk/cdr.h" +#include "asterisk/module.h" +#include "asterisk/config.h" +#include "asterisk/pbx.h" +#include "asterisk/logger.h" +#include "asterisk/utils.h" +#include "asterisk/cli.h" +#include "asterisk/options.h" + +AST_MUTEX_DEFINE_STATIC(lock); + +static const char config_file[] = "cdr_sqlite3_custom.conf"; + +static char *desc = "Customizable SQLite3 CDR Backend"; +static char *name = "cdr_sqlite3_custom"; +static sqlite3 *db = NULL; + +static char table[80]; +static char columns[1024]; +static char values[1024]; + +static int load_config(int reload) +{ + struct ast_config *cfg; + struct ast_variable *mappingvar; + const char *tmp; + + if (!(cfg = ast_config_load(config_file))) { + if (reload) + ast_log(LOG_WARNING, "%s: Failed to reload configuration file.\n", name); + else { + ast_log(LOG_WARNING, + "%s: Failed to load configuration file. Module not activated.\n", + name); + } + return -1; + } + + if (!reload) + ast_mutex_lock(&lock); + + if (!(mappingvar = ast_variable_browse(cfg, "master"))) { + /* nothing configured */ + ast_config_destroy(cfg); + return 0; + } + + /* Mapping must have a table name */ + tmp = ast_variable_retrieve(cfg, "master", "table"); + if (!ast_strlen_zero(tmp)) + ast_copy_string(table, tmp, sizeof(table)); + else { + ast_log(LOG_WARNING, "%s: Table name not specified. Assuming cdr.\n", name); + strcpy(table, "cdr"); + } + + tmp = ast_variable_retrieve(cfg, "master", "columns"); + if (!ast_strlen_zero(tmp)) + ast_copy_string(columns, tmp, sizeof(columns)); + else { + ast_log(LOG_WARNING, "%s: Column names not specified. Module not loaded.\n", + name); + ast_config_destroy(cfg); + return -1; + } + + tmp = ast_variable_retrieve(cfg, "master", "values"); + if (!ast_strlen_zero(tmp)) + ast_copy_string(values, tmp, sizeof(values)); + else { + ast_log(LOG_WARNING, "%s: Values not specified. Module not loaded.\n", name); + ast_config_destroy(cfg); + return -1; + } + + if (!reload) + ast_mutex_unlock(&lock); + + ast_config_destroy(cfg); + + return 0; +} + +/* assumues 'to' buffer is at least strlen(from) * 2 + 1 bytes */ +static int do_escape(char *to, const char *from) +{ + char *out = to; + + for (; *from; from++) { + if (*from == '\'' || *from == '\\') + *out++ = *from; + *out++ = *from; + } + *out = '\0'; + + return 0; +} + +static int sqlite3_log(struct ast_cdr *cdr) +{ + int res = 0; + char *zErr = 0; + char *sql_cmd; + struct ast_channel dummy = { 0, }; + int count; + + { /* Make it obvious that only sql_cmd should be used outside of this block */ + char *sql_tmp_cmd; + char sql_insert_cmd[2048] = ""; + sql_tmp_cmd = sqlite3_mprintf("INSERT INTO %q (%q) VALUES (%q)", table, columns, values); + dummy.cdr = cdr; + pbx_substitute_variables_helper(&dummy, sql_tmp_cmd, sql_insert_cmd, sizeof(sql_insert_cmd) - 1); + sqlite3_free(sql_tmp_cmd); + sql_cmd = alloca(strlen(sql_insert_cmd) * 2 + 1); + do_escape(sql_cmd, sql_insert_cmd); + } + + ast_mutex_lock(&lock); + + for (count = 0; count < 5; count++) { + res = sqlite3_exec(db, sql_cmd, NULL, NULL, &zErr); + if (res != SQLITE_BUSY && res != SQLITE_LOCKED) + break; + usleep(200); + } + + if (zErr) { + ast_log(LOG_ERROR, "%s: %s. sentence: %s.\n", name, zErr, sql_cmd); + sqlite3_free(zErr); + } + + ast_mutex_unlock(&lock); + + return res; +} + +static int unload_module(void) +{ + if (db) + sqlite3_close(db); + + ast_cdr_unregister(name); + + return 0; +} + +static int load_module(void) +{ + char *zErr; + char fn[PATH_MAX]; + int res; + char *sql_cmd; + + if (!load_config(0)) { + res = ast_cdr_register(name, desc, sqlite3_log); + if (res) { + ast_log(LOG_ERROR, "%s: Unable to register custom SQLite3 CDR handling\n", name); + return AST_MODULE_LOAD_DECLINE; + } + } + + /* is the database there? */ + snprintf(fn, sizeof(fn), "%s/master.db", ast_config_AST_LOG_DIR); + res = sqlite3_open(fn, &db); + if (!db) { + ast_log(LOG_ERROR, "%s: Could not open database %s.\n", name, fn); + sqlite3_free(zErr); + return AST_MODULE_LOAD_DECLINE; + } + + /* is the table there? */ + sql_cmd = sqlite3_mprintf("SELECT COUNT(AcctId) FROM %q;", table); + res = sqlite3_exec(db, sql_cmd, NULL, NULL, NULL); + sqlite3_free(sql_cmd); + if (res) { + sql_cmd = sqlite3_mprintf("CREATE TABLE %q (AcctId INTEGER PRIMARY KEY,%q)", table, columns); + res = sqlite3_exec(db, sql_cmd, NULL, NULL, &zErr); + sqlite3_free(sql_cmd); + if (zErr) { + ast_log(LOG_WARNING, "%s: %s.\n", name, zErr); + sqlite3_free(zErr); + return 0; + } + + if (res) { + ast_log(LOG_ERROR, "%s: Unable to create table '%s': %s.\n", name, table, zErr); + sqlite3_free(zErr); + if (db) + sqlite3_close(db); + return AST_MODULE_LOAD_DECLINE; + } + } + + return 0; +} + +static int reload(void) +{ + int res; + + ast_mutex_lock(&lock); + res = load_config(1); + ast_mutex_unlock(&lock); + + return res; +} + +AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_DEFAULT, "SQLite3 Custom CDR Module", + .load = load_module, + .unload = unload_module, + .reload = reload, +); |