FreeTDS CDR logger. More...

#include "asterisk.h"
#include <time.h>
#include <math.h>
#include "asterisk/config.h"
#include "asterisk/channel.h"
#include "asterisk/cdr.h"
#include "asterisk/module.h"
#include <sqlfront.h>
#include <sybdb.h>

Include dependency graph for cdr_tds.c:

Go to the source code of this file.

Data Structures
struct	cdr_tds_config
Defines
#define	DATE_FORMAT "%Y/%m/%d %T"
Functions
static void	__reg_module (void)
static void	__unreg_module (void)
static char *	anti_injection (const char *, int)
static int	execute_and_consume (DBPROCESS dbproc, const char fmt,...)
static void	get_date (char *, size_t len, struct timeval)
static int	load_module (void)
static int	mssql_connect (void)
static int	mssql_disconnect (void)
static int	reload (void)
static int	tds_error_handler (DBPROCESS dbproc, int severity, int dberr, int oserr, char dberrstr, char *oserrstr)
static int	tds_load_module (int reload)
static int	tds_log (struct ast_cdr *cdr)
static int	tds_message_handler (DBPROCESS dbproc, DBINT msgno, int msgstate, int severity, char msgtext, char srvname, char procname, int line)
static int	tds_unload_module (void)
static int	unload_module (void)
Variables
static struct ast_module_info __MODULE_INFO_SECTION	__mod_info = { __MODULE_INFO_GLOBALS .name = AST_MODULE, .flags = AST_MODFLAG_DEFAULT , .description = "FreeTDS CDR Backend" , .key = ASTERISK_GPL_KEY , .buildopt_sum = AST_BUILDOPT_SUM, .load = load_module, .unload = unload_module, .reload = reload, }
static struct ast_module_info *	ast_module_info = &__mod_info
static char *	config = "cdr_tds.conf"
static char *	name = "FreeTDS (MSSQL)"
static struct cdr_tds_config *	settings
static ast_mutex_t	tds_lock = AST_MUTEX_INIT_VALUE

Detailed Description

FreeTDS CDR logger.

Define Documentation

#define DATE_FORMAT "%Y/%m/%d %T"

 *
 * Table Structure for `cdr`
 *
 * Created on: 05/20/2004 16:16
 * Last changed on: 07/27/2004 20:01

CREATE TABLE [dbo].[cdr] (
	[accountcode] [varchar] (20) NULL ,
	[src] [varchar] (80) NULL ,
	[dst] [varchar] (80) NULL ,
	[dcontext] [varchar] (80) NULL ,
	[clid] [varchar] (80) NULL ,
	[channel] [varchar] (80) NULL ,
	[dstchannel] [varchar] (80) NULL ,
	[lastapp] [varchar] (80) NULL ,
	[lastdata] [varchar] (80) NULL ,
	[start] [datetime] NULL ,
	[answer] [datetime] NULL ,
	[end] [datetime] NULL ,
	[duration] [int] NULL ,
	[billsec] [int] NULL ,
	[disposition] [varchar] (20) NULL ,
	[amaflags] [varchar] (16) NULL ,
	[uniqueid] [varchar] (32) NULL ,
	[userfield] [varchar] (256) NULL
) ON [PRIMARY]

Definition at line 78 of file cdr_tds.c.

Referenced by get_date().

Function Documentation

static void __reg_module ( void ) [static]

Definition at line 567 of file cdr_tds.c.

static void __unreg_module ( void ) [static]

Definition at line 567 of file cdr_tds.c.

static char * anti_injection	(	const char *	str,
		int	len
	)		`[static]`

Definition at line 245 of file cdr_tds.c.

References ast_calloc, ast_log(), buf, len(), LOG_ERROR, str, and strcasestr().

Referenced by tds_log().

{
   /* Reference to http://www.nextgenss.com/papers/advanced_sql_injection.pdf */
   char *buf;
   char *buf_ptr, *srh_ptr;
   char *known_bad[] = {"select", "insert", "update", "delete", "drop", ";", "--", "\0"};
   int idx;

   if (!(buf = ast_calloc(1, len + 1))) {
      ast_log(LOG_ERROR, "Out of memory\n");
      return NULL;
   }

   buf_ptr = buf;

   /* Escape single quotes */
   for (; *str && strlen(buf) < len; str++) {
      if (*str == '\'') {
         *buf_ptr++ = '\'';
      }
      *buf_ptr++ = *str;
   }
   *buf_ptr = '\0';

   /* Erase known bad input */
   for (idx = 0; *known_bad[idx]; idx++) {
      while ((srh_ptr = strcasestr(buf, known_bad[idx]))) {
         memmove(srh_ptr, srh_ptr + strlen(known_bad[idx]), strlen(srh_ptr + strlen(known_bad[idx])) + 1);
      }
   }

   return buf;
}

static int execute_and_consume	(	DBPROCESS *	dbproc,
		const char *	fmt,
			...
	)		`[static]`

Definition at line 291 of file cdr_tds.c.

References ast_vasprintf, and free.

Referenced by mssql_connect().

{
   va_list ap;
   char *buffer;

   va_start(ap, fmt);
   if (ast_vasprintf(&buffer, fmt, ap) < 0) {
      va_end(ap);
      return 1;
   }
   va_end(ap);

   if (dbfcmd(dbproc, buffer) == FAIL) {
      free(buffer);
      return 1;
   }

   free(buffer);

   if (dbsqlexec(dbproc) == FAIL) {
      return 1;
   }

   /* Consume the result set (we don't really care about the result, though) */
   while (dbresults(dbproc) != NO_MORE_RESULTS) {
      while (dbnextrow(dbproc) != NO_MORE_ROWS);
   }

   return 0;
}

static void get_date	(	char *	dateField,
		size_t	len,
		struct timeval	when
	)		`[static]`

Definition at line 279 of file cdr_tds.c.

References ast_copy_string(), ast_localtime(), ast_strftime(), ast_tvzero(), and DATE_FORMAT.

Referenced by tds_log().

{
   /* To make sure we have date variable if not insert null to SQL */
   if (!ast_tvzero(when)) {
      struct ast_tm tm;
      ast_localtime(&when, &tm, NULL);
      ast_strftime(dateField, len, "'" DATE_FORMAT "'", &tm);
   } else {
      ast_copy_string(dateField, "null", len);
   }
}

static int load_module ( void ) [static]

Definition at line 524 of file cdr_tds.c.

References ast_calloc, ast_cdr_register(), ast_free, ast_log(), AST_MODULE_LOAD_DECLINE, AST_MODULE_LOAD_SUCCESS, ast_string_field_free_memory, ast_string_field_init, dbinit(), ast_module_info::description, LOG_ERROR, tds_error_handler(), tds_load_module(), tds_log(), and tds_message_handler().

{
   if (dbinit() == FAIL) {
      ast_log(LOG_ERROR, "Failed to initialize FreeTDS db-lib\n");
      return AST_MODULE_LOAD_DECLINE;
   }

   dberrhandle(tds_error_handler);
   dbmsghandle(tds_message_handler);

   settings = ast_calloc(1, sizeof(*settings));

   if (!settings || ast_string_field_init(settings, 256)) {
      if (settings) {
         ast_free(settings);
         settings = NULL;
      }
      dbexit();
      return AST_MODULE_LOAD_DECLINE;
   }

   if (!tds_load_module(0)) {
      ast_string_field_free_memory(settings);
      ast_free(settings);
      settings = NULL;
      dbexit();
      return AST_MODULE_LOAD_DECLINE;
   }

   ast_cdr_register(name, ast_module_info->description, tds_log);

   return AST_MODULE_LOAD_SUCCESS;
}

static int mssql_connect ( void ) [static]

Definition at line 334 of file cdr_tds.c.

References ast_log(), cdr_tds_config::charset, cdr_tds_config::connected, cdr_tds_config::database, cdr_tds_config::dbproc, execute_and_consume(), cdr_tds_config::has_userfield, cdr_tds_config::hostname, cdr_tds_config::language, LOG_ERROR, LOG_NOTICE, cdr_tds_config::password, cdr_tds_config::table, and cdr_tds_config::username.

Referenced by tds_load_module(), and tds_log().

{
   LOGINREC *login;

   if ((login = dblogin()) == NULL) {
      ast_log(LOG_ERROR, "Unable to allocate login structure for db-lib\n");
      return -1;
   }

   DBSETLAPP(login,     "TSQL");
   DBSETLUSER(login,    (char *) settings->username);
   DBSETLPWD(login,     (char *) settings->password);
   DBSETLCHARSET(login, (char *) settings->charset);
   DBSETLNATLANG(login, (char *) settings->language);

   if ((settings->dbproc = dbopen(login, (char *) settings->hostname)) == NULL) {
      ast_log(LOG_ERROR, "Unable to connect to %s\n", settings->hostname);
      dbloginfree(login);
      return -1;
   }

   dbloginfree(login);

   if (dbuse(settings->dbproc, (char *) settings->database) == FAIL) {
      ast_log(LOG_ERROR, "Unable to select database %s\n", settings->database);
      goto failed;
   }

   if (execute_and_consume(settings->dbproc, "SELECT 1 FROM [%s] WHERE 1 = 0", settings->table)) {
      ast_log(LOG_ERROR, "Unable to find table '%s'\n", settings->table);
      goto failed;
   }

   /* Check to see if we have a userfield column in the table */
   if (execute_and_consume(settings->dbproc, "SELECT userfield FROM [%s] WHERE 1 = 0", settings->table)) {
      ast_log(LOG_NOTICE, "Unable to find 'userfield' column in table '%s'\n", settings->table);
      settings->has_userfield = 0;
   } else {
      settings->has_userfield = 1;
   }

   settings->connected = 1;

   return 0;

failed:
   dbclose(settings->dbproc);
   settings->dbproc = NULL;
   return -1;
}

static int mssql_disconnect ( void ) [static]

Definition at line 322 of file cdr_tds.c.

References cdr_tds_config::connected, and cdr_tds_config::dbproc.

Referenced by tds_load_module(), tds_log(), and tds_unload_module().

{
   if (settings->dbproc) {
      dbclose(settings->dbproc);
      settings->dbproc = NULL;
   }

   settings->connected = 0;

   return 0;
}

static int reload ( void ) [static]

Definition at line 519 of file cdr_tds.c.

References tds_load_module().

{
   return tds_load_module(1);
}

static int tds_error_handler	(	DBPROCESS *	dbproc,
		int	severity,
		int	dberr,
		int	oserr,
		char *	dberrstr,
		char *	oserrstr
	)		`[static]`

Definition at line 403 of file cdr_tds.c.

References ast_log(), and LOG_ERROR.

Referenced by load_module().

{
   ast_log(LOG_ERROR, "%s (%d)\n", dberrstr, dberr);

   if (oserr != DBNOERR) {
      ast_log(LOG_ERROR, "%s (%d)\n", oserrstr, oserr);
   }

   return INT_CANCEL;
}

static int tds_load_module ( int reload ) [static]

Definition at line 422 of file cdr_tds.c.

References ast_config_destroy(), ast_config_load, ast_log(), ast_mutex_lock(), ast_mutex_unlock(), ast_string_field_init, ast_string_field_set, ast_variable_browse(), ast_variable_retrieve(), charset, CONFIG_FLAG_FILEUNCHANGED, CONFIG_STATUS_FILEINVALID, CONFIG_STATUS_FILEUNCHANGED, hostname, language, LOG_ERROR, LOG_NOTICE, mssql_connect(), mssql_disconnect(), table, and tds_lock.

Referenced by load_module(), and reload().

{
   struct ast_config *cfg;
   const char *ptr = NULL;
   struct ast_flags config_flags = { reload ? CONFIG_FLAG_FILEUNCHANGED : 0 };

   cfg = ast_config_load(config, config_flags);
   if (!cfg || cfg == CONFIG_STATUS_FILEINVALID) {
      ast_log(LOG_NOTICE, "Unable to load TDS config for CDRs: %s\n", config);
      return 0;
   } else if (cfg == CONFIG_STATUS_FILEUNCHANGED)
      return 0;

   if (!ast_variable_browse(cfg, "global")) {
      /* nothing configured */
      ast_config_destroy(cfg);
      return 0;
   }

   ast_mutex_lock(&tds_lock);

   /* Clear out any existing settings */
   ast_string_field_init(settings, 0);

   ptr = ast_variable_retrieve(cfg, "global", "hostname");
   if (ptr) {
      ast_string_field_set(settings, hostname, ptr);
   } else {
      ast_log(LOG_ERROR, "Failed to connect: Database server hostname not specified.\n");
      goto failed;
   }

   ptr = ast_variable_retrieve(cfg, "global", "dbname");
   if (ptr) {
      ast_string_field_set(settings, database, ptr);
   } else {
      ast_log(LOG_ERROR, "Failed to connect: Database dbname not specified.\n");
      goto failed;
   }

   ptr = ast_variable_retrieve(cfg, "global", "user");
   if (ptr) {
      ast_string_field_set(settings, username, ptr);
   } else {
      ast_log(LOG_ERROR, "Failed to connect: Database dbuser not specified.\n");
      goto failed;
   }

   ptr = ast_variable_retrieve(cfg, "global", "password");
   if (ptr) {
      ast_string_field_set(settings, password, ptr);
   } else {
      ast_log(LOG_ERROR, "Failed to connect: Database password not specified.\n");
      goto failed;
   }

   ptr = ast_variable_retrieve(cfg, "global", "charset");
   if (ptr) {
      ast_string_field_set(settings, charset, ptr);
   } else {
      ast_string_field_set(settings, charset, "iso_1");
   }

   ptr = ast_variable_retrieve(cfg, "global", "language");
   if (ptr) {
      ast_string_field_set(settings, language, ptr);
   } else {
      ast_string_field_set(settings, language, "us_english");
   }

   ptr = ast_variable_retrieve(cfg, "global", "table");
   if (ptr) {
      ast_string_field_set(settings, table, ptr);
   } else {
      ast_log(LOG_NOTICE, "Table name not specified, using 'cdr' by default.\n");
      ast_string_field_set(settings, table, "cdr");
   }

   mssql_disconnect();

   if (mssql_connect()) {
      /* We failed to connect (mssql_connect takes care of logging it) */
      goto failed;
   }

   ast_mutex_unlock(&tds_lock);
   ast_config_destroy(cfg);

   return 1;

failed:
   ast_mutex_unlock(&tds_lock);
   ast_config_destroy(cfg);

   return 0;
}

static int tds_log ( struct ast_cdr * cdr ) [static]

Definition at line 111 of file cdr_tds.c.

References accountcode, anti_injection(), ast_cdr_disp2str(), ast_cdr_flags2str(), ast_free, ast_log(), AST_MAX_USER_FIELD, ast_mutex_lock(), ast_mutex_unlock(), cdr_tds_config::connected, cdr_tds_config::dbproc, get_date(), cdr_tds_config::has_userfield, cdr_tds_config::hostname, LOG_ERROR, LOG_NOTICE, mssql_connect(), mssql_disconnect(), cdr_tds_config::table, and tds_lock.

Referenced by load_module().

{
   char start[80], answer[80], end[80];
   char *accountcode, *src, *dst, *dcontext, *clid, *channel, *dstchannel, *lastapp, *lastdata, *uniqueid, *userfield = NULL;
   RETCODE erc;
   int res = -1;
   int attempt = 1;

   accountcode = anti_injection(cdr->accountcode, 20);
   src         = anti_injection(cdr->src, 80);
   dst         = anti_injection(cdr->dst, 80);
   dcontext    = anti_injection(cdr->dcontext, 80);
   clid        = anti_injection(cdr->clid, 80);
   channel     = anti_injection(cdr->channel, 80);
   dstchannel  = anti_injection(cdr->dstchannel, 80);
   lastapp     = anti_injection(cdr->lastapp, 80);
   lastdata    = anti_injection(cdr->lastdata, 80);
   uniqueid    = anti_injection(cdr->uniqueid, 32);

   get_date(start, sizeof(start), cdr->start);
   get_date(answer, sizeof(answer), cdr->answer);
   get_date(end, sizeof(end), cdr->end);

   ast_mutex_lock(&tds_lock);

   if (settings->has_userfield) {
      userfield = anti_injection(cdr->userfield, AST_MAX_USER_FIELD);
   }

retry:
   /* Ensure that we are connected */
   if (!settings->connected) {
      ast_log(LOG_NOTICE, "Attempting to reconnect to %s (Attempt %d)\n", settings->hostname, attempt);
      if (mssql_connect()) {
         /* Connect failed */
         if (attempt++ < 3) {
            goto retry;
         }
         goto done;
      }
   }

   if (settings->has_userfield) {
      erc = dbfcmd(settings->dbproc,
                "INSERT INTO %s "
                "("
                "accountcode, src, dst, dcontext, clid, channel, "
                "dstchannel, lastapp, lastdata, start, answer, [end], duration, "
                "billsec, disposition, amaflags, uniqueid, userfield"
                ") "
                "VALUES "
                "("
                "'%s', '%s', '%s', '%s', '%s', '%s', "
                "'%s', '%s', '%s', %s, %s, %s, %ld, "
                "%ld, '%s', '%s', '%s', '%s'"
                ")",
                settings->table,
                accountcode, src, dst, dcontext, clid, channel,
                dstchannel, lastapp, lastdata, start, answer, end, cdr->duration,
                cdr->billsec, ast_cdr_disp2str(cdr->disposition), ast_cdr_flags2str(cdr->amaflags), uniqueid,
                userfield
         );
   } else {
      erc = dbfcmd(settings->dbproc,
                "INSERT INTO %s "
                "("
                "accountcode, src, dst, dcontext, clid, channel, "
                "dstchannel, lastapp, lastdata, start, answer, [end], duration, "
                "billsec, disposition, amaflags, uniqueid"
                ") "
                "VALUES "
                "("
                "'%s', '%s', '%s', '%s', '%s', '%s', "
                "'%s', '%s', '%s', %s, %s, %s, %ld, "
                "%ld, '%s', '%s', '%s'"
                ")",
                settings->table,
                accountcode, src, dst, dcontext, clid, channel,
                dstchannel, lastapp, lastdata, start, answer, end, cdr->duration,
                cdr->billsec, ast_cdr_disp2str(cdr->disposition), ast_cdr_flags2str(cdr->amaflags), uniqueid
         );
   }

   if (erc == FAIL) {
      if (attempt++ < 3) {
         ast_log(LOG_NOTICE, "Failed to build INSERT statement, retrying...\n");
         mssql_disconnect();
         goto retry;
      } else {
         ast_log(LOG_ERROR, "Failed to build INSERT statement, no CDR was logged.\n");
         goto done;
      }
   }

   if (dbsqlexec(settings->dbproc) == FAIL) {
      if (attempt++ < 3) {
         ast_log(LOG_NOTICE, "Failed to execute INSERT statement, retrying...\n");
         mssql_disconnect();
         goto retry;
      } else {
         ast_log(LOG_ERROR, "Failed to execute INSERT statement, no CDR was logged.\n");
         goto done;
      }
   }

   /* Consume any results we might get back (this is more of a sanity check than
    * anything else, since an INSERT shouldn't return results). */
   while (dbresults(settings->dbproc) != NO_MORE_RESULTS) {
      while (dbnextrow(settings->dbproc) != NO_MORE_ROWS);
   }

   res = 0;

done:
   ast_mutex_unlock(&tds_lock);

   ast_free(accountcode);
   ast_free(src);
   ast_free(dst);
   ast_free(dcontext);
   ast_free(clid);
   ast_free(channel);
   ast_free(dstchannel);
   ast_free(lastapp);
   ast_free(lastdata);
   ast_free(uniqueid);

   if (userfield) {
      ast_free(userfield);
   }

   return res;
}

static int tds_message_handler	(	DBPROCESS *	dbproc,
		DBINT	msgno,
		int	msgstate,
		int	severity,
		char *	msgtext,
		char *	srvname,
		char *	procname,
		int	line
	)		`[static]`

Definition at line 414 of file cdr_tds.c.

References ast_debug, ast_log(), and LOG_NOTICE.

Referenced by load_module().

{
   ast_debug(1, "Msg %d, Level %d, State %d, Line %d\n", msgno, severity, msgstate, line);
   ast_log(LOG_NOTICE, "%s\n", msgtext);

   return 0;
}

static int tds_unload_module ( void ) [static]

Definition at line 385 of file cdr_tds.c.

References ast_cdr_unregister(), ast_free, ast_mutex_lock(), ast_mutex_unlock(), ast_string_field_free_memory, mssql_disconnect(), and tds_lock.

Referenced by unload_module().

{
   if (settings) {
      ast_mutex_lock(&tds_lock);
      mssql_disconnect();
      ast_mutex_unlock(&tds_lock);

      ast_string_field_free_memory(settings);
      ast_free(settings);
   }

   ast_cdr_unregister(name);

   dbexit();

   return 0;
}

static int unload_module ( void ) [static]

Definition at line 558 of file cdr_tds.c.

References tds_unload_module().

{
   return tds_unload_module();
}

Variable Documentation

struct ast_module_info __MODULE_INFO_SECTION __mod_info = { __MODULE_INFO_GLOBALS .name = AST_MODULE, .flags = AST_MODFLAG_DEFAULT , .description = "FreeTDS CDR Backend" , .key = ASTERISK_GPL_KEY , .buildopt_sum = AST_BUILDOPT_SUM, .load = load_module, .unload = unload_module, .reload = reload, } [static]

Definition at line 567 of file cdr_tds.c.

struct ast_module_info* ast_module_info = &__mod_info [static]

Definition at line 567 of file cdr_tds.c.

char* config = "cdr_tds.conf" [static]

Definition at line 81 of file cdr_tds.c.

char* name = "FreeTDS (MSSQL)" [static]

Definition at line 80 of file cdr_tds.c.

struct cdr_tds_config* settings [static]

Definition at line 100 of file cdr_tds.c.

ast_mutex_t tds_lock = AST_MUTEX_INIT_VALUE [static]

Definition at line 98 of file cdr_tds.c.

Referenced by tds_load_module(), tds_log(), and tds_unload_module().

Asterisk developer's documentation

cdr_tds.c File Reference
Module: CDR Drivers

Data Structures

Defines

Functions

Variables

Detailed Description

Define Documentation

Function Documentation

Variable Documentation

Asterisk developer's documentation

cdr_tds.c File ReferenceModule: CDR Drivers

Data Structures

Defines

Functions

Variables

Detailed Description

Define Documentation

Function Documentation

Variable Documentation

cdr_tds.c File Reference
Module: CDR Drivers