diff options
Diffstat (limited to 'cel/cel_pgsql.c')
-rw-r--r-- | cel/cel_pgsql.c | 565 |
1 files changed, 565 insertions, 0 deletions
diff --git a/cel/cel_pgsql.c b/cel/cel_pgsql.c new file mode 100644 index 000000000..8b10261ee --- /dev/null +++ b/cel/cel_pgsql.c @@ -0,0 +1,565 @@ +/* + * Asterisk -- An open source telephony toolkit. + * + * Copyright (C) 2008 + * + * Steve Murphy - adapted to CEL, from: + * Matthew D. Hardeman <mhardemn@papersoft.com> + * Adapted from the MySQL CDR logger originally by James Sharp + * + * Modified April, 2007; Dec, 2008 + * Steve Murphy <murf@digium.com> + + * Modified September 2003 + * Matthew D. Hardeman <mhardemn@papersoft.com> + * + * 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 PostgreSQL CEL logger + * + * \author Steve Murphy <murf@digium.com> + * \extref PostgreSQL http://www.postgresql.org/ + * + * See also + * \arg \ref Config_cel + * \arg http://www.postgresql.org/ + * \ingroup cel_drivers + */ + +/*** MODULEINFO + <depend>pgsql</depend> + ***/ + +#include "asterisk.h" + +ASTERISK_FILE_VERSION(__FILE__, "$Revision$") + +#include <libpq-fe.h> + +#include "asterisk/config.h" +#include "asterisk/options.h" +#include "asterisk/channel.h" +#include "asterisk/cel.h" +#include "asterisk/module.h" +#include "asterisk/logger.h" +#include "asterisk.h" + +#define DATE_FORMAT "%Y-%m-%d %T" + +static char *config = "cel_pgsql.conf"; +static char *pghostname = NULL, *pgdbname = NULL, *pgdbuser = NULL, *pgpassword = NULL, *pgdbport = NULL, *table = NULL; +static int connected = 0; +static int maxsize = 512, maxsize2 = 512; + +AST_MUTEX_DEFINE_STATIC(pgsql_lock); + +static PGconn *conn = NULL; +static PGresult *result = NULL; +static struct ast_event_sub *event_sub = NULL; + +struct columns { + char *name; + char *type; + int len; + unsigned int notnull:1; + unsigned int hasdefault:1; + AST_RWLIST_ENTRY(columns) list; +}; + +static AST_RWLIST_HEAD_STATIC(psql_columns, columns); + +#define LENGTHEN_BUF1(size) \ + do { \ + /* Lengthen buffer, if necessary */ \ + if (ast_str_strlen(sql) + size + 1 > ast_str_size(sql)) { \ + if (ast_str_make_space(&sql, ((ast_str_size(sql) + size + 3) / 512 + 1) * 512) != 0) { \ + ast_log(LOG_ERROR, "Unable to allocate sufficient memory. Insert CDR failed.\n"); \ + ast_free(sql); \ + ast_free(sql2); \ + AST_RWLIST_UNLOCK(&psql_columns); \ + return; \ + } \ + } \ + } while (0) + +#define LENGTHEN_BUF2(size) \ + do { \ + if (ast_str_strlen(sql2) + size + 1 > ast_str_size(sql2)) { \ + if (ast_str_make_space(&sql2, ((ast_str_size(sql2) + size + 3) / 512 + 1) * 512) != 0) { \ + ast_log(LOG_ERROR, "Unable to allocate sufficient memory. Insert CDR failed.\n"); \ + ast_free(sql); \ + ast_free(sql2); \ + AST_RWLIST_UNLOCK(&psql_columns); \ + return; \ + } \ + } \ + } while (0) + +static void pgsql_log(const struct ast_event *event, void *userdata) +{ + struct ast_tm tm; + char timestr[128]; + char *pgerror; + struct ast_cel_event_record record = { + .version = AST_CEL_EVENT_RECORD_VERSION, + }; + + if (ast_cel_fill_record(event, &record)) { + return; + } + + ast_mutex_lock(&pgsql_lock); + + ast_localtime(&record.event_time, &tm, NULL); + ast_strftime(timestr, sizeof(timestr), DATE_FORMAT, &tm); + + if ((!connected) && pghostname && pgdbuser && pgpassword && pgdbname) { + conn = PQsetdbLogin(pghostname, pgdbport, NULL, NULL, pgdbname, pgdbuser, pgpassword); + if (PQstatus(conn) != CONNECTION_BAD) { + connected = 1; + } else { + pgerror = PQerrorMessage(conn); + ast_log(LOG_ERROR, "cel_pgsql: Unable to connect to database server %s. Calls will not be logged!\n", pghostname); + ast_log(LOG_ERROR, "cel_pgsql: Reason: %s\n", pgerror); + PQfinish(conn); + conn = NULL; + } + } + if (connected) { + struct columns *cur; + struct ast_str *sql = ast_str_create(maxsize), *sql2 = ast_str_create(maxsize2); + char buf[257], escapebuf[513]; + const char *value; + int first = 1; + + if (!sql || !sql2) { + if (sql) { + ast_free(sql); + } + if (sql2) { + ast_free(sql2); + } + return; + } + + ast_str_set(&sql, 0, "INSERT INTO %s (", table); + ast_str_set(&sql2, 0, " VALUES ("); + +#define SEP (first ? "" : ",") + + AST_RWLIST_RDLOCK(&psql_columns); + AST_RWLIST_TRAVERSE(&psql_columns, cur, list) { + LENGTHEN_BUF1(strlen(cur->name) + 2); + ast_str_append(&sql, 0, "%s\"%s\"", first ? "" : ",", cur->name); + + if (strcmp(cur->name, "eventtime") == 0) { + if (strncmp(cur->type, "int", 3) == 0) { + LENGTHEN_BUF2(13); + ast_str_append(&sql2, 0, "%s%ld", SEP, record.event_time.tv_sec); + } else if (strncmp(cur->type, "float", 5) == 0) { + LENGTHEN_BUF2(31); + ast_str_append(&sql2, 0, "%s%f", + SEP, + (double) record.event_time.tv_sec + + (double) record.event_time.tv_usec / 1000000.0); + } else { + /* char, hopefully */ + LENGTHEN_BUF2(31); + ast_localtime(&record.event_time, &tm, NULL); + ast_strftime(buf, sizeof(buf), DATE_FORMAT, &tm); + ast_str_append(&sql2, 0, "%s'%s'", SEP, buf); + } + } else if (strcmp(cur->name, "eventtype") == 0) { + if (cur->type[0] == 'i') { + /* Get integer, no need to escape anything */ + LENGTHEN_BUF2(5); + ast_str_append(&sql2, 0, "%s%d", SEP, (int) record.event_type); + } else if (strncmp(cur->type, "float", 5) == 0) { + LENGTHEN_BUF2(31); + ast_str_append(&sql2, 0, "%s%f", SEP, (double) record.event_type); + } else { + /* Char field, probably */ + LENGTHEN_BUF2(strlen(record.event_name) + 1); + ast_str_append(&sql2, 0, "%s'%s'", SEP, record.event_name); + } + } else if (strcmp(cur->name, "amaflags") == 0) { + if (strncmp(cur->type, "int", 3) == 0) { + /* Integer, no need to escape anything */ + LENGTHEN_BUF2(13); + ast_str_append(&sql2, 0, "%s%d", SEP, record.amaflag); + } else { + /* Although this is a char field, there are no special characters in the values for these fields */ + LENGTHEN_BUF2(31); + ast_str_append(&sql2, 0, "%s'%d'", SEP, record.amaflag); + } + } else { + /* Arbitrary field, could be anything */ + if (strcmp(cur->name, "userdeftype") == 0) { + value = record.user_defined_name; + } else if (strcmp(cur->name, "cid_name") == 0) { + value = record.caller_id_name; + } else if (strcmp(cur->name, "cid_num") == 0) { + value = record.caller_id_num; + } else if (strcmp(cur->name, "cid_ani") == 0) { + value = record.caller_id_ani; + } else if (strcmp(cur->name, "cid_rdnis") == 0) { + value = record.caller_id_rdnis; + } else if (strcmp(cur->name, "cid_dnid") == 0) { + value = record.caller_id_dnid; + } else if (strcmp(cur->name, "exten") == 0) { + value = record.extension; + } else if (strcmp(cur->name, "context") == 0) { + value = record.context; + } else if (strcmp(cur->name, "channame") == 0) { + value = record.channel_name; + } else if (strcmp(cur->name, "appname") == 0) { + value = record.application_name; + } else if (strcmp(cur->name, "appdata") == 0) { + value = record.application_data; + } else if (strcmp(cur->name, "accountcode") == 0) { + value = record.account_code; + } else if (strcmp(cur->name, "peeraccount") == 0) { + value = record.peer_account; + } else if (strcmp(cur->name, "uniqueid") == 0) { + value = record.unique_id; + } else if (strcmp(cur->name, "linkedid") == 0) { + value = record.linked_id; + } else if (strcmp(cur->name, "userfield") == 0) { + value = record.user_field; + } else if (strcmp(cur->name, "peer") == 0) { + value = record.peer; + } else { + value = ""; + } + if (strncmp(cur->type, "int", 3) == 0) { + long long whatever; + if (value && sscanf(value, "%lld", &whatever) == 1) { + LENGTHEN_BUF2(26); + ast_str_append(&sql2, 0, "%s%lld", SEP, whatever); + } else { + LENGTHEN_BUF2(2); + ast_str_append(&sql2, 0, "%s0", SEP); + } + } else if (strncmp(cur->type, "float", 5) == 0) { + long double whatever; + if (value && sscanf(value, "%Lf", &whatever) == 1) { + LENGTHEN_BUF2(51); + ast_str_append(&sql2, 0, "%s%30Lf", SEP, whatever); + } else { + LENGTHEN_BUF2(2); + ast_str_append(&sql2, 0, "%s0", SEP); + } + /* XXX Might want to handle dates, times, and other misc fields here XXX */ + } else { + if (value) { + PQescapeStringConn(conn, escapebuf, value, strlen(value), NULL); + } else { + escapebuf[0] = '\0'; + } + LENGTHEN_BUF2(strlen(escapebuf) + 3); + ast_str_append(&sql2, 0, "%s'%s'", SEP, escapebuf); + } + } + first = 0; + } + AST_RWLIST_UNLOCK(&psql_columns); + LENGTHEN_BUF1(ast_str_strlen(sql2) + 2); + ast_str_append(&sql, 0, ")%s)", ast_str_buffer(sql2)); + ast_verb(11, "[%s]\n", ast_str_buffer(sql)); + + ast_debug(2, "inserting a CEL record.\n"); + /* Test to be sure we're still connected... */ + /* If we're connected, and connection is working, good. */ + /* Otherwise, attempt reconnect. If it fails... sorry... */ + if (PQstatus(conn) == CONNECTION_OK) { + connected = 1; + } else { + ast_log(LOG_ERROR, "Connection was lost... attempting to reconnect.\n"); + PQreset(conn); + if (PQstatus(conn) == CONNECTION_OK) { + ast_log(LOG_ERROR, "Connection reestablished.\n"); + connected = 1; + } else { + pgerror = PQerrorMessage(conn); + ast_log(LOG_ERROR, "Unable to reconnect to database server %s. Calls will not be logged!\n", pghostname); + ast_log(LOG_ERROR, "Reason: %s\n", pgerror); + PQfinish(conn); + conn = NULL; + connected = 0; + ast_mutex_unlock(&pgsql_lock); + ast_free(sql); + ast_free(sql2); + return; + } + } + result = PQexec(conn, ast_str_buffer(sql)); + if (PQresultStatus(result) != PGRES_COMMAND_OK) { + pgerror = PQresultErrorMessage(result); + ast_log(LOG_ERROR, "Failed to insert call detail record into database!\n"); + ast_log(LOG_ERROR, "Reason: %s\n", pgerror); + ast_log(LOG_ERROR, "Connection may have been lost... attempting to reconnect.\n"); + PQreset(conn); + if (PQstatus(conn) == CONNECTION_OK) { + ast_log(LOG_ERROR, "Connection reestablished.\n"); + connected = 1; + PQclear(result); + result = PQexec(conn, ast_str_buffer(sql)); + if (PQresultStatus(result) != PGRES_COMMAND_OK) { + pgerror = PQresultErrorMessage(result); + ast_log(LOG_ERROR, "HARD ERROR! Attempted reconnection failed. DROPPING CALL RECORD!\n"); + ast_log(LOG_ERROR, "Reason: %s\n", pgerror); + } + } + ast_mutex_unlock(&pgsql_lock); + PQclear(result); + ast_free(sql); + ast_free(sql2); + return; + } + ast_mutex_unlock(&pgsql_lock); + } +} + +static int my_unload_module(void) +{ + struct columns *current; + if (event_sub) { + event_sub = ast_event_unsubscribe(event_sub); + } + if (conn) { + PQfinish(conn); + } + if (pghostname) { + ast_free(pghostname); + } + if (pgdbname) { + ast_free(pgdbname); + } + if (pgdbuser) { + ast_free(pgdbuser); + } + if (pgpassword) { + ast_free(pgpassword); + } + if (pgdbport) { + ast_free(pgdbport); + } + if (table) { + ast_free(table); + } + AST_RWLIST_WRLOCK(&psql_columns); + while ((current = AST_RWLIST_REMOVE_HEAD(&psql_columns, list))) { + ast_free(current); + } + AST_RWLIST_UNLOCK(&psql_columns); + return 0; +} + +static int unload_module(void) +{ + return my_unload_module(); +} + +static int process_my_load_module(struct ast_config *cfg) +{ + struct ast_variable *var; + char *pgerror; + const char *tmp; + PGresult *result; + struct columns *cur; + + if (!(var = ast_variable_browse(cfg, "global"))) { + ast_log(LOG_WARNING,"CEL pgsql config file missing global section.\n"); + return AST_MODULE_LOAD_DECLINE; + } + if (!(tmp = ast_variable_retrieve(cfg,"global","hostname"))) { + ast_log(LOG_WARNING,"PostgreSQL server hostname not specified. Assuming unix socket connection\n"); + tmp = ""; /* connect via UNIX-socket by default */ + } + if (pghostname) + ast_free(pghostname); + if (!(pghostname = ast_strdup(tmp))) { + ast_log(LOG_WARNING,"PostgreSQL Ran out of memory copying host info\n"); + return AST_MODULE_LOAD_DECLINE; + } + if (!(tmp = ast_variable_retrieve(cfg, "global", "dbname"))) { + ast_log(LOG_WARNING,"PostgreSQL database not specified. Assuming asterisk\n"); + tmp = "asteriskceldb"; + } + if (pgdbname) + ast_free(pgdbname); + if (!(pgdbname = ast_strdup(tmp))) { + ast_log(LOG_WARNING,"PostgreSQL Ran out of memory copying dbname info\n"); + return AST_MODULE_LOAD_DECLINE; + } + if (!(tmp = ast_variable_retrieve(cfg, "global", "user"))) { + ast_log(LOG_WARNING,"PostgreSQL database user not specified. Assuming asterisk\n"); + tmp = "asterisk"; + } + if (pgdbuser) + ast_free(pgdbuser); + if (!(pgdbuser = ast_strdup(tmp))) { + ast_log(LOG_WARNING,"PostgreSQL Ran out of memory copying user info\n"); + return AST_MODULE_LOAD_DECLINE; + } + if (!(tmp = ast_variable_retrieve(cfg, "global", "password"))) { + ast_log(LOG_WARNING, "PostgreSQL database password not specified. Assuming blank\n"); + tmp = ""; + } + if (pgpassword) + ast_free(pgpassword); + if (!(pgpassword = ast_strdup(tmp))) { + ast_log(LOG_WARNING,"PostgreSQL Ran out of memory copying password info\n"); + return AST_MODULE_LOAD_DECLINE; + } + if (!(tmp = ast_variable_retrieve(cfg,"global","port"))) { + ast_log(LOG_WARNING,"PostgreSQL database port not specified. Using default 5432.\n"); + tmp = "5432"; + } + if (pgdbport) + ast_free(pgdbport); + if (!(pgdbport = ast_strdup(tmp))) { + ast_log(LOG_WARNING,"PostgreSQL Ran out of memory copying port info\n"); + return AST_MODULE_LOAD_DECLINE; + } + if (!(tmp = ast_variable_retrieve(cfg, "global", "table"))) { + ast_log(LOG_WARNING,"CEL table not specified. Assuming cel\n"); + tmp = "cel"; + } + if (table) + ast_free(table); + if (!(table = ast_strdup(tmp))) { + return AST_MODULE_LOAD_DECLINE; + } + if (option_debug) { + if (ast_strlen_zero(pghostname)) { + ast_debug(3, "cel_pgsql: using default unix socket\n"); + } else { + ast_debug(3, "cel_pgsql: got hostname of %s\n", pghostname); + } + ast_debug(3, "cel_pgsql: got port of %s\n", pgdbport); + ast_debug(3, "cel_pgsql: got user of %s\n", pgdbuser); + ast_debug(3, "cel_pgsql: got dbname of %s\n", pgdbname); + ast_debug(3, "cel_pgsql: got password of %s\n", pgpassword); + ast_debug(3, "cel_pgsql: got sql table name of %s\n", table); + } + + conn = PQsetdbLogin(pghostname, pgdbport, NULL, NULL, pgdbname, pgdbuser, pgpassword); + if (PQstatus(conn) != CONNECTION_BAD) { + char sqlcmd[512]; + char *fname, *ftype, *flen, *fnotnull, *fdef; + char *tableptr; + int i, rows; + + ast_debug(1, "Successfully connected to PostgreSQL database.\n"); + connected = 1; + + /* Remove any schema name from the table */ + if ((tableptr = strrchr(table, '.'))) { + tableptr++; + } else { + tableptr = table; + } + + /* Query the columns */ + snprintf(sqlcmd, sizeof(sqlcmd), "select a.attname, t.typname, a.attlen, a.attnotnull, d.adsrc from pg_class c, pg_type t, pg_attribute a left outer join pg_attrdef d on a.atthasdef and d.adrelid = a.attrelid and d.adnum = a.attnum where c.oid = a.attrelid and a.atttypid = t.oid and (a.attnum > 0) and c.relname = '%s' order by c.relname, attnum", tableptr); + result = PQexec(conn, sqlcmd); + if (PQresultStatus(result) != PGRES_TUPLES_OK) { + pgerror = PQresultErrorMessage(result); + ast_log(LOG_ERROR, "Failed to query database columns: %s\n", pgerror); + PQclear(result); + unload_module(); + return AST_MODULE_LOAD_DECLINE; + } + + rows = PQntuples(result); + for (i = 0; i < rows; i++) { + fname = PQgetvalue(result, i, 0); + ftype = PQgetvalue(result, i, 1); + flen = PQgetvalue(result, i, 2); + fnotnull = PQgetvalue(result, i, 3); + fdef = PQgetvalue(result, i, 4); + ast_verb(4, "Found column '%s' of type '%s'\n", fname, ftype); + cur = ast_calloc(1, sizeof(*cur) + strlen(fname) + strlen(ftype) + 2); + if (cur) { + sscanf(flen, "%d", &cur->len); + cur->name = (char *)cur + sizeof(*cur); + cur->type = (char *)cur + sizeof(*cur) + strlen(fname) + 1; + strcpy(cur->name, fname); + strcpy(cur->type, ftype); + if (*fnotnull == 't') { + cur->notnull = 1; + } else { + cur->notnull = 0; + } + if (!ast_strlen_zero(fdef)) { + cur->hasdefault = 1; + } else { + cur->hasdefault = 0; + } + AST_RWLIST_INSERT_TAIL(&psql_columns, cur, list); + } + } + PQclear(result); + } else { + pgerror = PQerrorMessage(conn); + ast_log(LOG_ERROR, "cel_pgsql: Unable to connect to database server %s. CALLS WILL NOT BE LOGGED!!\n", pghostname); + ast_log(LOG_ERROR, "cel_pgsql: Reason: %s\n", pgerror); + connected = 0; + } + return AST_MODULE_LOAD_SUCCESS; +} + +static int my_load_module(int reload) +{ + struct ast_config *cfg; + int res; + struct ast_flags config_flags = { reload ? CONFIG_FLAG_FILEUNCHANGED : 0 }; + + if ((cfg = ast_config_load(config, config_flags)) == NULL || cfg == CONFIG_STATUS_FILEINVALID) { + ast_log(LOG_WARNING, "Unable to load config for PostgreSQL CEL's: %s\n", config); + return AST_MODULE_LOAD_DECLINE; + } else if (cfg == CONFIG_STATUS_FILEUNCHANGED) { + return AST_MODULE_LOAD_SUCCESS; + } + + res = process_my_load_module(cfg); + ast_config_destroy(cfg); + + event_sub = ast_event_subscribe(AST_EVENT_CEL, pgsql_log, "CEL PGSQL backend", NULL, AST_EVENT_IE_END); + + if (!event_sub) { + ast_log(LOG_WARNING, "Unable to subscribe to CEL events for pgsql\n"); + return AST_MODULE_LOAD_DECLINE; + } + + return AST_MODULE_LOAD_SUCCESS; +} + +static int load_module(void) +{ + return my_load_module(0); +} + +static int reload(void) +{ + my_unload_module(); + return my_load_module(1); +} + +AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_DEFAULT, "PostgreSQL CEL Backend", + .load = load_module, + .unload = unload_module, + .reload = reload, +); |