PQClear is not called when the result object of a call to PQExec has a status of PGRES_COMMAND_OK. Interestingly enough, the off nominal case was handled properly, so this memory leak only occurred when CEL records were successfully written. This patch properly clears the result in the nominal code path. (closes issue ASTERISK-19991) Reported by: Etienne Lessard Tested by: Etienne Lessard patches: mem_leak_cel_pgsql.patch uploaded by Etienne Lessard (license #6394) ........ Merged revisions 372158 from http://svn.asterisk.org/svn/asterisk/branches/1.8 ........ Merged revisions 372165 from http://svn.asterisk.org/svn/asterisk/branches/10 ........ Merged revisions 372175 from http://svn.asterisk.org/svn/asterisk/branches/11 git-svn-id: https://origsvn.digium.com/svn/asterisk/trunk@372176 65c4cc65-6c06-0410-ace0-fbb531ad65f3
591 lines
18 KiB
C
591 lines
18 KiB
C
/*
|
|
* 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
|
|
* \extref PostgreSQL http://www.postgresql.org/
|
|
* \ingroup cel_drivers
|
|
*/
|
|
|
|
/*** MODULEINFO
|
|
<depend>pgsql</depend>
|
|
<support_level>extended</support_level>
|
|
***/
|
|
|
|
#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.%6q"
|
|
|
|
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;
|
|
|
|
/*! \brief show_user_def is off by default */
|
|
#define CEL_SHOW_USERDEF_DEFAULT 0
|
|
|
|
/*! TRUE if we should set the eventtype field to USER_DEFINED on user events. */
|
|
static unsigned char cel_show_user_def;
|
|
|
|
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) {
|
|
goto ast_log_cleanup;
|
|
}
|
|
|
|
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, (long) 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 */
|
|
const char *event_name;
|
|
|
|
event_name = (!cel_show_user_def
|
|
&& record.event_type == AST_CEL_USER_DEFINED)
|
|
? record.user_defined_name : record.event_name;
|
|
LENGTHEN_BUF2(strlen(event_name) + 1);
|
|
ast_str_append(&sql2, 0, "%s'%s'", SEP, 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 if (strcmp(cur->name, "extra") == 0) {
|
|
value = record.extra;
|
|
} else {
|
|
value = NULL;
|
|
}
|
|
|
|
if (value == NULL) {
|
|
ast_str_append(&sql2, 0, "%sDEFAULT", SEP);
|
|
} else if (strncmp(cur->type, "int", 3) == 0) {
|
|
long long whatever;
|
|
if (value && sscanf(value, "%30lld", &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, "%30Lf", &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_WARNING, "Connection was lost... attempting to reconnect.\n");
|
|
PQreset(conn);
|
|
if (PQstatus(conn) == CONNECTION_OK) {
|
|
ast_log(LOG_NOTICE, "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;
|
|
goto ast_log_cleanup;
|
|
}
|
|
}
|
|
result = PQexec(conn, ast_str_buffer(sql));
|
|
if (PQresultStatus(result) != PGRES_COMMAND_OK) {
|
|
pgerror = PQresultErrorMessage(result);
|
|
ast_log(LOG_WARNING, "Failed to insert call detail record into database!\n");
|
|
ast_log(LOG_WARNING, "Reason: %s\n", pgerror);
|
|
ast_log(LOG_WARNING, "Connection may have been lost... attempting to reconnect.\n");
|
|
PQreset(conn);
|
|
if (PQstatus(conn) == CONNECTION_OK) {
|
|
ast_log(LOG_NOTICE, "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);
|
|
}
|
|
}
|
|
PQclear(result);
|
|
goto ast_log_cleanup;
|
|
}
|
|
PQclear(result);
|
|
|
|
ast_log_cleanup:
|
|
ast_free(sql);
|
|
ast_free(sql2);
|
|
}
|
|
|
|
ast_mutex_unlock(&pgsql_lock);
|
|
}
|
|
|
|
static int my_unload_module(void)
|
|
{
|
|
struct columns *current;
|
|
AST_RWLIST_WRLOCK(&psql_columns);
|
|
if (event_sub) {
|
|
event_sub = ast_event_unsubscribe(event_sub);
|
|
event_sub = NULL;
|
|
}
|
|
if (conn) {
|
|
PQfinish(conn);
|
|
conn = NULL;
|
|
}
|
|
if (pghostname) {
|
|
ast_free(pghostname);
|
|
pghostname = NULL;
|
|
}
|
|
if (pgdbname) {
|
|
ast_free(pgdbname);
|
|
pgdbname = NULL;
|
|
}
|
|
if (pgdbuser) {
|
|
ast_free(pgdbuser);
|
|
pgdbuser = NULL;
|
|
}
|
|
if (pgpassword) {
|
|
ast_free(pgpassword);
|
|
pgpassword = NULL;
|
|
}
|
|
if (pgdbport) {
|
|
ast_free(pgdbport);
|
|
pgdbport = NULL;
|
|
}
|
|
if (table) {
|
|
ast_free(table);
|
|
table = NULL;
|
|
}
|
|
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;
|
|
}
|
|
cel_show_user_def = CEL_SHOW_USERDEF_DEFAULT;
|
|
if ((tmp = ast_variable_retrieve(cfg, "global", "show_user_defined"))) {
|
|
cel_show_user_def = ast_true(tmp) ? 1 : 0;
|
|
}
|
|
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);
|
|
ast_debug(3, "cel_pgsql: got show_user_defined of %s\n",
|
|
cel_show_user_def ? "Yes" : "No");
|
|
}
|
|
|
|
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, "%30d", &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;
|
|
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;
|
|
}
|
|
|
|
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_LOAD_ORDER, "PostgreSQL CEL Backend",
|
|
.load = load_module,
|
|
.unload = unload_module,
|
|
.reload = reload,
|
|
.load_pri = AST_MODPRI_CDR_DRIVER,
|
|
);
|