diff options
Diffstat (limited to 'cdr/cdr_odbc.c')
| -rw-r--r-- | cdr/cdr_odbc.c | 481 |
1 files changed, 481 insertions, 0 deletions
diff --git a/cdr/cdr_odbc.c b/cdr/cdr_odbc.c new file mode 100644 index 000000000..da8327375 --- /dev/null +++ b/cdr/cdr_odbc.c @@ -0,0 +1,481 @@ +/* + * Asterisk -- An open source telephony toolkit. + * + * Copyright (C) 2003-2005, Digium, Inc. + * + * Brian K. West <brian@bkw.org> + * + * 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 ODBC CDR Backend + * + * \author Brian K. West <brian@bkw.org> + * + * See also: + * \arg http://www.unixodbc.org + * \arg \ref Config_cdr + * \ingroup cdr_drivers + */ + +/*** MODULEINFO + <depend>unixodbc</depend> + <depend>ltdl</depend> + ***/ + +#include "asterisk.h" + +ASTERISK_FILE_VERSION(__FILE__, "$Revision$") + +#include <sys/types.h> +#include <stdio.h> +#include <string.h> + +#include <stdlib.h> +#include <unistd.h> +#include <time.h> + +#ifndef __CYGWIN__ +#include <sql.h> +#include <sqlext.h> +#include <sqltypes.h> +#else +#include <windows.h> +#include <w32api/sql.h> +#include <w32api/sqlext.h> +#include <w32api/sqltypes.h> +#endif + +#include "asterisk/config.h" +#include "asterisk/options.h" +#include "asterisk/channel.h" +#include "asterisk/cdr.h" +#include "asterisk/module.h" +#include "asterisk/logger.h" + +#define DATE_FORMAT "%Y-%m-%d %T" + +static char *name = "ODBC"; +static char *config = "cdr_odbc.conf"; +static char *dsn = NULL, *username = NULL, *password = NULL, *table = NULL; +static int loguniqueid = 0; +static int usegmtime = 0; +static int dispositionstring = 0; +static int connected = 0; + +AST_MUTEX_DEFINE_STATIC(odbc_lock); + +static int odbc_do_query(void); +static int odbc_init(void); + +static SQLHENV ODBC_env = SQL_NULL_HANDLE; /* global ODBC Environment */ +static SQLHDBC ODBC_con; /* global ODBC Connection Handle */ +static SQLHSTMT ODBC_stmt; /* global ODBC Statement Handle */ + +static void odbc_disconnect(void) +{ + ODBC_stmt = SQL_NULL_HANDLE; + SQLDisconnect(ODBC_con); + SQLFreeHandle(SQL_HANDLE_DBC, ODBC_con); + ODBC_con = SQL_NULL_HANDLE; + SQLFreeHandle(SQL_HANDLE_ENV, ODBC_env); + ODBC_env = SQL_NULL_HANDLE; + connected = 0; +} + +static void build_query(struct ast_cdr *cdr, char *timestr, int timesize) +{ + int ODBC_res; + char sqlcmd[2048] = ""; + int res = 0; + struct tm tm; + + if (usegmtime) + gmtime_r(&cdr->start.tv_sec,&tm); + else + ast_localtime(&cdr->start.tv_sec, &tm, NULL); + + strftime(timestr, timesize, DATE_FORMAT, &tm); + memset(sqlcmd,0,2048); + if (loguniqueid) { + snprintf(sqlcmd,sizeof(sqlcmd),"INSERT INTO %s " + "(calldate,clid,src,dst,dcontext,channel,dstchannel,lastapp," + "lastdata,duration,billsec,disposition,amaflags,accountcode,uniqueid,userfield) " + "VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)", table); + } else { + snprintf(sqlcmd,sizeof(sqlcmd),"INSERT INTO %s " + "(calldate,clid,src,dst,dcontext,channel,dstchannel,lastapp,lastdata," + "duration,billsec,disposition,amaflags,accountcode) " + "VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?)", table); + } + if (!connected) { + res = odbc_init(); + if (res < 0) { + odbc_disconnect(); + return; + } + } + + ODBC_res = SQLAllocHandle(SQL_HANDLE_STMT, ODBC_con, &ODBC_stmt); + + if ((ODBC_res != SQL_SUCCESS) && (ODBC_res != SQL_SUCCESS_WITH_INFO)) { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: Failure in AllocStatement %d\n", ODBC_res); + SQLFreeHandle(SQL_HANDLE_STMT, ODBC_stmt); + ODBC_stmt = SQL_NULL_HANDLE; + odbc_disconnect(); + return; + } + + /* We really should only have to do this once. But for some + strange reason if I don't it blows holes in memory like + like a shotgun. So we just do this so its safe. */ + + ODBC_res = SQLPrepare(ODBC_stmt, (unsigned char *)sqlcmd, SQL_NTS); + + if ((ODBC_res != SQL_SUCCESS) && (ODBC_res != SQL_SUCCESS_WITH_INFO)) { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: Error in PREPARE %d\n", ODBC_res); + SQLFreeHandle(SQL_HANDLE_STMT, ODBC_stmt); + ODBC_stmt = SQL_NULL_HANDLE; + odbc_disconnect(); + return; + } + + SQLBindParameter(ODBC_stmt, 1, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, timesize, 0, timestr, 0, NULL); + SQLBindParameter(ODBC_stmt, 2, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->clid), 0, cdr->clid, 0, NULL); + SQLBindParameter(ODBC_stmt, 3, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->src), 0, cdr->src, 0, NULL); + SQLBindParameter(ODBC_stmt, 4, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->dst), 0, cdr->dst, 0, NULL); + SQLBindParameter(ODBC_stmt, 5, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->dcontext), 0, cdr->dcontext, 0, NULL); + SQLBindParameter(ODBC_stmt, 6, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->channel), 0, cdr->channel, 0, NULL); + SQLBindParameter(ODBC_stmt, 7, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->dstchannel), 0, cdr->dstchannel, 0, NULL); + SQLBindParameter(ODBC_stmt, 8, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->lastapp), 0, cdr->lastapp, 0, NULL); + SQLBindParameter(ODBC_stmt, 9, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->lastdata), 0, cdr->lastdata, 0, NULL); + SQLBindParameter(ODBC_stmt, 10, SQL_PARAM_INPUT, SQL_C_SLONG, SQL_INTEGER, 0, 0, &cdr->duration, 0, NULL); + SQLBindParameter(ODBC_stmt, 11, SQL_PARAM_INPUT, SQL_C_SLONG, SQL_INTEGER, 0, 0, &cdr->billsec, 0, NULL); + if (dispositionstring) + SQLBindParameter(ODBC_stmt, 12, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, strlen(ast_cdr_disp2str(cdr->disposition)) + 1, 0, ast_cdr_disp2str(cdr->disposition), 0, NULL); + else + SQLBindParameter(ODBC_stmt, 12, SQL_PARAM_INPUT, SQL_C_SLONG, SQL_INTEGER, 0, 0, &cdr->disposition, 0, NULL); + SQLBindParameter(ODBC_stmt, 13, SQL_PARAM_INPUT, SQL_C_SLONG, SQL_INTEGER, 0, 0, &cdr->amaflags, 0, NULL); + SQLBindParameter(ODBC_stmt, 14, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->accountcode), 0, cdr->accountcode, 0, NULL); + + if (loguniqueid) { + SQLBindParameter(ODBC_stmt, 15, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->uniqueid), 0, cdr->uniqueid, 0, NULL); + SQLBindParameter(ODBC_stmt, 16, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->userfield), 0, cdr->userfield, 0, NULL); + } +} + +static int odbc_log(struct ast_cdr *cdr) +{ + int res = 0; + char timestr[150]; + + ast_mutex_lock(&odbc_lock); + build_query(cdr, timestr, sizeof(timestr)); + + if (connected) { + res = odbc_do_query(); + if (res < 0) { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: Reconnecting to dsn %s\n", dsn); + odbc_disconnect(); + res = odbc_init(); + if (res < 0) { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: %s has gone away!\n", dsn); + odbc_disconnect(); + } else { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: Trying Query again!\n"); + SQLFreeHandle(SQL_HANDLE_STMT, ODBC_stmt); + ODBC_stmt = SQL_NULL_HANDLE; + build_query(cdr, timestr, sizeof(timestr)); /* what a waste. If we have to reconnect, we have to build a new query */ + res = odbc_do_query(); + if (res < 0) { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: Query FAILED Call not logged!\n"); + } + } + } + } else { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: Query FAILED Call not logged!\n"); + } + SQLFreeHandle(SQL_HANDLE_STMT, ODBC_stmt); + ODBC_stmt = SQL_NULL_HANDLE; + ast_mutex_unlock(&odbc_lock); + return 0; +} + +static int odbc_unload_module(void) +{ + ast_mutex_lock(&odbc_lock); + if (connected) { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: Disconnecting from %s\n", dsn); + SQLFreeHandle(SQL_HANDLE_STMT, ODBC_stmt); + ODBC_stmt = SQL_NULL_HANDLE; + odbc_disconnect(); + } + if (dsn) { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: free dsn\n"); + free(dsn); + } + if (username) { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: free username\n"); + free(username); + } + if (password) { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: free password\n"); + free(password); + } + if (table) { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: free table\n"); + free(table); + } + + ast_cdr_unregister(name); + ast_mutex_unlock(&odbc_lock); + return 0; +} + +static int odbc_load_module(void) +{ + int res = 0; + struct ast_config *cfg; + struct ast_variable *var; + const char *tmp; + + ast_mutex_lock(&odbc_lock); + + cfg = ast_config_load(config); + if (!cfg) { + ast_log(LOG_WARNING, "cdr_odbc: Unable to load config for ODBC CDR's: %s\n", config); + res = AST_MODULE_LOAD_DECLINE; + goto out; + } + + var = ast_variable_browse(cfg, "global"); + if (!var) { + /* nothing configured */ + goto out; + } + + tmp = ast_variable_retrieve(cfg,"global","dsn"); + if (tmp == NULL) { + ast_log(LOG_WARNING,"cdr_odbc: dsn not specified. Assuming asteriskdb\n"); + tmp = "asteriskdb"; + } + dsn = strdup(tmp); + if (dsn == NULL) { + ast_log(LOG_ERROR,"cdr_odbc: Out of memory error.\n"); + res = -1; + goto out; + } + + tmp = ast_variable_retrieve(cfg,"global","dispositionstring"); + if (tmp) { + dispositionstring = ast_true(tmp); + } else { + dispositionstring = 0; + } + + tmp = ast_variable_retrieve(cfg,"global","username"); + if (tmp) { + username = strdup(tmp); + if (username == NULL) { + ast_log(LOG_ERROR,"cdr_odbc: Out of memory error.\n"); + res = -1; + goto out; + } + } + + tmp = ast_variable_retrieve(cfg,"global","password"); + if (tmp) { + password = strdup(tmp); + if (password == NULL) { + ast_log(LOG_ERROR,"cdr_odbc: Out of memory error.\n"); + res = -1; + goto out; + } + } + + tmp = ast_variable_retrieve(cfg,"global","loguniqueid"); + if (tmp) { + loguniqueid = ast_true(tmp); + if (loguniqueid) { + ast_log(LOG_DEBUG,"cdr_odbc: Logging uniqueid\n"); + } else { + ast_log(LOG_DEBUG,"cdr_odbc: Not logging uniqueid\n"); + } + } else { + ast_log(LOG_DEBUG,"cdr_odbc: Not logging uniqueid\n"); + loguniqueid = 0; + } + + tmp = ast_variable_retrieve(cfg,"global","usegmtime"); + if (tmp) { + usegmtime = ast_true(tmp); + if (usegmtime) { + ast_log(LOG_DEBUG,"cdr_odbc: Logging in GMT\n"); + } else { + ast_log(LOG_DEBUG,"cdr_odbc: Not logging in GMT\n"); + } + } else { + ast_log(LOG_DEBUG,"cdr_odbc: Not logging in GMT\n"); + usegmtime = 0; + } + + tmp = ast_variable_retrieve(cfg,"global","table"); + if (tmp == NULL) { + ast_log(LOG_WARNING,"cdr_odbc: table not specified. Assuming cdr\n"); + tmp = "cdr"; + } + table = strdup(tmp); + if (table == NULL) { + ast_log(LOG_ERROR,"cdr_odbc: Out of memory error.\n"); + res = -1; + goto out; + } + + if (option_verbose > 2) { + ast_verbose( VERBOSE_PREFIX_3 "cdr_odbc: dsn is %s\n",dsn); + if (username) + { + ast_verbose( VERBOSE_PREFIX_3 "cdr_odbc: username is %s\n",username); + ast_verbose( VERBOSE_PREFIX_3 "cdr_odbc: password is [secret]\n"); + } + else + ast_verbose( VERBOSE_PREFIX_3 "cdr_odbc: retreiving username and password from odbc config\n"); + ast_verbose( VERBOSE_PREFIX_3 "cdr_odbc: table is %s\n",table); + } + + res = odbc_init(); + if (res < 0) { + ast_log(LOG_ERROR, "cdr_odbc: Unable to connect to datasource: %s\n", dsn); + if (option_verbose > 2) { + ast_verbose( VERBOSE_PREFIX_3 "cdr_odbc: Unable to connect to datasource: %s\n", dsn); + } + } + res = ast_cdr_register(name, ast_module_info->description, odbc_log); + if (res) { + ast_log(LOG_ERROR, "cdr_odbc: Unable to register ODBC CDR handling\n"); + } +out: + if (cfg) + ast_config_destroy(cfg); + ast_mutex_unlock(&odbc_lock); + return res; +} + +static int odbc_do_query(void) +{ + int ODBC_res; + ODBC_res = SQLExecute(ODBC_stmt); + + if ((ODBC_res != SQL_SUCCESS) && (ODBC_res != SQL_SUCCESS_WITH_INFO)) { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: Error in Query %d\n", ODBC_res); + return -1; + } else { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: Query Successful!\n"); + connected = 1; + } + return 0; +} + +static int odbc_init(void) +{ + int ODBC_res; + + if (ODBC_env == SQL_NULL_HANDLE || connected == 0) { + ODBC_res = SQLAllocHandle(SQL_HANDLE_ENV, SQL_NULL_HANDLE, &ODBC_env); + if ((ODBC_res != SQL_SUCCESS) && (ODBC_res != SQL_SUCCESS_WITH_INFO)) { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: Error AllocHandle\n"); + connected = 0; + return -1; + } + + ODBC_res = SQLSetEnvAttr(ODBC_env, SQL_ATTR_ODBC_VERSION, (void*)SQL_OV_ODBC3, 0); + + if ((ODBC_res != SQL_SUCCESS) && (ODBC_res != SQL_SUCCESS_WITH_INFO)) { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: Error SetEnv\n"); + SQLFreeHandle(SQL_HANDLE_ENV, ODBC_env); + ODBC_env = SQL_NULL_HANDLE; + connected = 0; + return -1; + } + + ODBC_res = SQLAllocHandle(SQL_HANDLE_DBC, ODBC_env, &ODBC_con); + + if ((ODBC_res != SQL_SUCCESS) && (ODBC_res != SQL_SUCCESS_WITH_INFO)) { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: Error AllocHDB %d\n", ODBC_res); + SQLFreeHandle(SQL_HANDLE_ENV, ODBC_env); + ODBC_env = SQL_NULL_HANDLE; + connected = 0; + return -1; + } + SQLSetConnectAttr(ODBC_con, SQL_LOGIN_TIMEOUT, (SQLPOINTER *)10, 0); + } + + /* Note that the username and password could be NULL here, but that is allowed in ODBC. + In this case, the default username and password will be used from odbc.conf */ + ODBC_res = SQLConnect(ODBC_con, (SQLCHAR*)dsn, SQL_NTS, (SQLCHAR*)username, SQL_NTS, (SQLCHAR*)password, SQL_NTS); + + if ((ODBC_res != SQL_SUCCESS) && (ODBC_res != SQL_SUCCESS_WITH_INFO)) { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: Error SQLConnect %d\n", ODBC_res); + SQLFreeHandle(SQL_HANDLE_DBC, ODBC_con); + ODBC_con = SQL_NULL_HANDLE; + SQLFreeHandle(SQL_HANDLE_ENV, ODBC_env); + ODBC_env = SQL_NULL_HANDLE; + connected = 0; + return -1; + } else { + if (option_verbose > 10) + ast_verbose( VERBOSE_PREFIX_4 "cdr_odbc: Connected to %s\n", dsn); + connected = 1; + } + return 0; +} + +static int load_module(void) +{ + return odbc_load_module(); +} + +static int unload_module(void) +{ + return odbc_unload_module(); +} + +static int reload(void) +{ + odbc_unload_module(); + return odbc_load_module(); +} + +AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_DEFAULT, "ODBC CDR Backend", + .load = load_module, + .unload = unload_module, + .reload = reload, + ); |