Home page | Mailing list | Docs

Last updated: Sat Feb 3 05:00:53 2007

Asterisk developer's documentation :: Codename Pineapple

---

res_config_pgsql.c

Go to the documentation of this file.

/*
 * Asterisk -- A telephony toolkit for Linux.
 *
 * Copyright (C) 1999-2005, Digium, Inc.
 * 
 * Manuel Guesdon <mguesdon@oxymium.net> - Postgresql RealTime Driver Author/Adaptor
 * Mark Spencer <markster@digium.com>  - Asterisk Author
 * Matthew Boehm <mboehm@cytelcom.com> - MySQL RealTime Driver Author
 *
 * res_config_pgsql.c <Postgresql plugin for RealTime configuration engine>
 *
 * v1.0   - (07-11-05) - Initial version based on res_config_mysql v2.0
 */

/*! \file
 *
 * \brief Postgresql plugin for Asterisk RealTime Architecture
 *
 * \author Mark Spencer <markster@digium.com>
 * \author Manuel Guesdon <mguesdon@oxymium.net> - Postgresql RealTime Driver Author/Adaptor
 *
 * \arg http://www.postgresql.org
 */

/*** MODULEINFO
   <depend>pgsql</depend>
 ***/

#include "asterisk.h"

ASTERISK_FILE_VERSION(__FILE__, "$Revision: 48395 $")

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <libpq-fe.h>         /* PostgreSQL */

#include "asterisk/file.h"
#include "asterisk/logger.h"
#include "asterisk/channel.h"
#include "asterisk/pbx.h"
#include "asterisk/config.h"
#include "asterisk/module.h"
#include "asterisk/lock.h"
#include "asterisk/options.h"
#include "asterisk/utils.h"
#include "asterisk/cli.h"

AST_MUTEX_DEFINE_STATIC(pgsql_lock);

#define RES_CONFIG_PGSQL_CONF "res_pgsql.conf"

PGconn *pgsqlConn = NULL;

#define MAX_DB_OPTION_SIZE 64

static char dbhost[MAX_DB_OPTION_SIZE] = "";
static char dbuser[MAX_DB_OPTION_SIZE] = "";
static char dbpass[MAX_DB_OPTION_SIZE] = "";
static char dbname[MAX_DB_OPTION_SIZE] = "";
static char dbsock[MAX_DB_OPTION_SIZE] = "";
static int dbport = 5432;
static time_t connect_time = 0;

static int parse_config(void);
static int pgsql_reconnect(const char *database);
static int realtime_pgsql_status(int fd, int argc, char **argv);

static const char cli_realtime_pgsql_status_usage[] =
   "Usage: realtime pgsql status\n"
   "       Shows connection information for the Postgresql RealTime driver\n";

static struct ast_cli_entry cli_realtime[] = {
   { { "realtime", "pgsql", "status", NULL },
   realtime_pgsql_status, "Shows connection information for the Postgresql RealTime driver",
   cli_realtime_pgsql_status_usage },
};

static struct ast_variable *realtime_pgsql(const char *database, const char *table, va_list ap)
{
   PGresult *result = NULL;
   int num_rows = 0;
   char sql[256];
   char *stringp;
   char *chunk;
   char *op;
   const char *newparam, *newval;
   struct ast_variable *var = NULL, *prev = NULL;

   if (!table) {
      ast_log(LOG_WARNING, "Postgresql RealTime: No table specified.\n");
      return NULL;
   }

   /* Get the first parameter and first value in our list of passed paramater/value pairs */
   newparam = va_arg(ap, const char *);
   newval = va_arg(ap, const char *);
   if (!newparam || !newval) {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: Realtime retrieval requires at least 1 parameter and 1 value to search on.\n");
      if (pgsqlConn) {
         PQfinish(pgsqlConn);
         pgsqlConn = NULL;
      };
      return NULL;
   }

   /* Create the first part of the query using the first parameter/value pairs we just extracted
      If there is only 1 set, then we have our query. Otherwise, loop thru the list and concat */
   op = strchr(newparam, ' ') ? "" : " =";

   snprintf(sql, sizeof(sql), "SELECT * FROM %s WHERE %s%s '%s'", table, newparam, op,
          newval);
   while ((newparam = va_arg(ap, const char *))) {
      newval = va_arg(ap, const char *);
      if (!strchr(newparam, ' '))
         op = " =";
      else
         op = "";
      snprintf(sql + strlen(sql), sizeof(sql) - strlen(sql), " AND %s%s '%s'", newparam,
             op, newval);
   }
   va_end(ap);

   /* We now have our complete statement; Lets connect to the server and execute it. */
   ast_mutex_lock(&pgsql_lock);
   if (!pgsql_reconnect(database)) {
      ast_mutex_unlock(&pgsql_lock);
      return NULL;
   }

   if (!(result = PQexec(pgsqlConn, sql))) {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: Failed to query database. Check debug for more info.\n");
      if (option_debug) {
         ast_log(LOG_DEBUG, "Postgresql RealTime: Query: %s\n", sql);
         ast_log(LOG_DEBUG, "Postgresql RealTime: Query Failed because: %s\n",
               PQerrorMessage(pgsqlConn));
      }
      ast_mutex_unlock(&pgsql_lock);
      return NULL;
   } else {
      ExecStatusType result_status = PQresultStatus(result);
      if (result_status != PGRES_COMMAND_OK
         && result_status != PGRES_TUPLES_OK
         && result_status != PGRES_NONFATAL_ERROR) {
         ast_log(LOG_WARNING,
               "Postgresql RealTime: Failed to query database. Check debug for more info.\n");
         if (option_debug) {
            ast_log(LOG_DEBUG, "Postgresql RealTime: Query: %s\n", sql);
            ast_log(LOG_DEBUG, "Postgresql RealTime: Query Failed because: %s (%s)\n",
                  PQresultErrorMessage(result), PQresStatus(result_status));
         }
         ast_mutex_unlock(&pgsql_lock);
         return NULL;
      }
   }

   if (option_debug)
      ast_log(LOG_DEBUG, "1Postgresql RealTime: Result=%p Query: %s\n", result, sql);

   if ((num_rows = PQntuples(result)) > 0) {
      int i = 0;
      int rowIndex = 0;
      int numFields = PQnfields(result);
      char **fieldnames = NULL;

      if (option_debug)
         ast_log(LOG_DEBUG, "Postgresql RealTime: Found %d rows.\n", num_rows);

      if (!(fieldnames = ast_calloc(1, numFields * sizeof(char *)))) {
         ast_mutex_unlock(&pgsql_lock);
         PQclear(result);
         return NULL;
      }
      for (i = 0; i < numFields; i++)
         fieldnames[i] = PQfname(result, i);
      for (rowIndex = 0; rowIndex < num_rows; rowIndex++) {
         for (i = 0; i < numFields; i++) {
            stringp = PQgetvalue(result, rowIndex, i);
            while (stringp) {
               chunk = strsep(&stringp, ";");
               if (chunk && !ast_strlen_zero(ast_strip(chunk))) {
                  if (prev) {
                     prev->next = ast_variable_new(fieldnames[i], chunk);
                     if (prev->next) {
                        prev = prev->next;
                     }
                  } else {
                     prev = var = ast_variable_new(fieldnames[i], chunk);
                  }
               }
            }
         }
      }
      free(fieldnames);
   } else {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: Could not find any rows in table %s.\n", table);
   }

   ast_mutex_unlock(&pgsql_lock);
   PQclear(result);

   return var;
}

static struct ast_config *realtime_multi_pgsql(const char *database, const char *table, va_list ap)
{
   PGresult *result = NULL;
   int num_rows = 0;
   char sql[256];
   const char *initfield = NULL;
   char *stringp;
   char *chunk;
   char *op;
   const char *newparam, *newval;
   struct ast_realloca ra;
   struct ast_variable *var = NULL;
   struct ast_config *cfg = NULL;
   struct ast_category *cat = NULL;

   if (!table) {
      ast_log(LOG_WARNING, "Postgresql RealTime: No table specified.\n");
      return NULL;
   }

   memset(&ra, 0, sizeof(ra));

   if (!(cfg = ast_config_new()))
      return NULL;

   /* Get the first parameter and first value in our list of passed paramater/value pairs */
   newparam = va_arg(ap, const char *);
   newval = va_arg(ap, const char *);
   if (!newparam || !newval) {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: Realtime retrieval requires at least 1 parameter and 1 value to search on.\n");
      if (pgsqlConn) {
         PQfinish(pgsqlConn);
         pgsqlConn = NULL;
      };
      return NULL;
   }

   initfield = ast_strdupa(newparam);
   if ((op = strchr(initfield, ' '))) {
      *op = '\0';
   }

   /* Create the first part of the query using the first parameter/value pairs we just extracted
      If there is only 1 set, then we have our query. Otherwise, loop thru the list and concat */

   if (!strchr(newparam, ' '))
      op = " =";
   else
      op = "";

   snprintf(sql, sizeof(sql), "SELECT * FROM %s WHERE %s%s '%s'", table, newparam, op,
          newval);
   while ((newparam = va_arg(ap, const char *))) {
      newval = va_arg(ap, const char *);
      if (!strchr(newparam, ' '))
         op = " =";
      else
         op = "";
      snprintf(sql + strlen(sql), sizeof(sql) - strlen(sql), " AND %s%s '%s'", newparam,
             op, newval);
   }

   if (initfield) {
      snprintf(sql + strlen(sql), sizeof(sql) - strlen(sql), " ORDER BY %s", initfield);
   }

   va_end(ap);

   /* We now have our complete statement; Lets connect to the server and execute it. */
   ast_mutex_lock(&pgsql_lock);
   if (!pgsql_reconnect(database)) {
      ast_mutex_unlock(&pgsql_lock);
      return NULL;
   }

   if (!(result = PQexec(pgsqlConn, sql))) {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: Failed to query database. Check debug for more info.\n");
      if (option_debug) {
         ast_log(LOG_DEBUG, "Postgresql RealTime: Query: %s\n", sql);
         ast_log(LOG_DEBUG, "Postgresql RealTime: Query Failed because: %s\n",
               PQerrorMessage(pgsqlConn));
      }
      ast_mutex_unlock(&pgsql_lock);
      return NULL;
   } else {
      ExecStatusType result_status = PQresultStatus(result);
      if (result_status != PGRES_COMMAND_OK
         && result_status != PGRES_TUPLES_OK
         && result_status != PGRES_NONFATAL_ERROR) {
         ast_log(LOG_WARNING,
               "Postgresql RealTime: Failed to query database. Check debug for more info.\n");
         if (option_debug) {
            ast_log(LOG_DEBUG, "Postgresql RealTime: Query: %s\n", sql);
            ast_log(LOG_DEBUG, "Postgresql RealTime: Query Failed because: %s (%s)\n",
                  PQresultErrorMessage(result), PQresStatus(result_status));
         }
         ast_mutex_unlock(&pgsql_lock);
         return NULL;
      }
   }

   if (option_debug)
      ast_log(LOG_DEBUG, "2Postgresql RealTime: Result=%p Query: %s\n", result, sql);

   if ((num_rows = PQntuples(result)) > 0) {
      int numFields = PQnfields(result);
      int i = 0;
      int rowIndex = 0;
      char **fieldnames = NULL;

      if (option_debug)
         ast_log(LOG_DEBUG, "Postgresql RealTime: Found %d rows.\n", num_rows);

      if (!(fieldnames = ast_calloc(1, numFields * sizeof(char *)))) {
         ast_mutex_unlock(&pgsql_lock);
         PQclear(result);
         return NULL;
      }
      for (i = 0; i < numFields; i++)
         fieldnames[i] = PQfname(result, i);

      for (rowIndex = 0; rowIndex < num_rows; rowIndex++) {
         var = NULL;
         if (!(cat = ast_category_new("")))
            continue;
         for (i = 0; i < numFields; i++) {
            stringp = PQgetvalue(result, rowIndex, i);
            while (stringp) {
               chunk = strsep(&stringp, ";");
               if (chunk && !ast_strlen_zero(ast_strip(chunk))) {
                  if (initfield && !strcmp(initfield, fieldnames[i])) {
                     ast_category_rename(cat, chunk);
                  }
                  var = ast_variable_new(fieldnames[i], chunk);
                  ast_variable_append(cat, var);
               }
            }
         }
         ast_category_append(cfg, cat);
      }
      free(fieldnames);
   } else {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: Could not find any rows in table %s.\n", table);
   }

   ast_mutex_unlock(&pgsql_lock);
   PQclear(result);

   return cfg;
}

static int update_pgsql(const char *database, const char *table, const char *keyfield,
                  const char *lookup, va_list ap)
{
   PGresult *result = NULL;
   int numrows = 0;
   char sql[256];
   const char *newparam, *newval;

   if (!table) {
      ast_log(LOG_WARNING, "Postgresql RealTime: No table specified.\n");
      return -1;
   }

   /* Get the first parameter and first value in our list of passed paramater/value pairs */
   newparam = va_arg(ap, const char *);
   newval = va_arg(ap, const char *);
   if (!newparam || !newval) {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: Realtime retrieval requires at least 1 parameter and 1 value to search on.\n");
      if (pgsqlConn) {
         PQfinish(pgsqlConn);
         pgsqlConn = NULL;
      };
      return -1;
   }

   /* Create the first part of the query using the first parameter/value pairs we just extracted
      If there is only 1 set, then we have our query. Otherwise, loop thru the list and concat */

   snprintf(sql, sizeof(sql), "UPDATE %s SET %s = '%s'", table, newparam, newval);
   while ((newparam = va_arg(ap, const char *))) {
      newval = va_arg(ap, const char *);
      snprintf(sql + strlen(sql), sizeof(sql) - strlen(sql), ", %s = '%s'", newparam,
             newval);
   }
   va_end(ap);
   snprintf(sql + strlen(sql), sizeof(sql) - strlen(sql), " WHERE %s = '%s'", keyfield,
          lookup);

   if (option_debug)
      ast_log(LOG_DEBUG, "Postgresql RealTime: Update SQL: %s\n", sql);

   /* We now have our complete statement; Lets connect to the server and execute it. */
   ast_mutex_lock(&pgsql_lock);
   if (!pgsql_reconnect(database)) {
      ast_mutex_unlock(&pgsql_lock);
      return -1;
   }

   if (!(result = PQexec(pgsqlConn, sql))) {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: Failed to query database. Check debug for more info.\n");
      if (option_debug) {
         ast_log(LOG_DEBUG, "Postgresql RealTime: Query: %s\n", sql);
         ast_log(LOG_DEBUG, "Postgresql RealTime: Query Failed because: %s\n",
               PQerrorMessage(pgsqlConn));
      }
      ast_mutex_unlock(&pgsql_lock);
      return -1;
   } else {
      ExecStatusType result_status = PQresultStatus(result);
      if (result_status != PGRES_COMMAND_OK
         && result_status != PGRES_TUPLES_OK
         && result_status != PGRES_NONFATAL_ERROR) {
         ast_log(LOG_WARNING,
               "Postgresql RealTime: Failed to query database. Check debug for more info.\n");
         if (option_debug) {
            ast_log(LOG_DEBUG, "Postgresql RealTime: Query: %s\n", sql);
            ast_log(LOG_DEBUG, "Postgresql RealTime: Query Failed because: %s (%s)\n",
                  PQresultErrorMessage(result), PQresStatus(result_status));
         }
         ast_mutex_unlock(&pgsql_lock);
         return -1;
      }
   }

   numrows = atoi(PQcmdTuples(result));
   ast_mutex_unlock(&pgsql_lock);

   if (option_debug)
      ast_log(LOG_DEBUG, "Postgresql RealTime: Updated %d rows on table: %s\n", numrows,
            table);

   /* From http://dev.pgsql.com/doc/pgsql/en/pgsql-affected-rows.html
    * An integer greater than zero indicates the number of rows affected
    * Zero indicates that no records were updated
    * -1 indicates that the query returned an error (although, if the query failed, it should have been caught above.)
    */

   if (numrows >= 0)
      return (int) numrows;

   return -1;
}

static struct ast_config *config_pgsql(const char *database, const char *table,
                  const char *file, struct ast_config *cfg,
                  int withcomments)
{
   PGresult *result = NULL;
   long num_rows;
   struct ast_variable *new_v;
   struct ast_category *cur_cat = NULL;
   char sql[250] = "";
   char last[80] = "";
   int last_cat_metric = 0;

   last[0] = '\0';

   if (!file || !strcmp(file, RES_CONFIG_PGSQL_CONF)) {
      ast_log(LOG_WARNING, "Postgresql RealTime: Cannot configure myself.\n");
      return NULL;
   }

   snprintf(sql, sizeof(sql),
          "SELECT category, var_name, var_val, cat_metric FROM %s WHERE filename='%s' and commented=0 ORDER BY filename, cat_metric desc, var_metric asc, category, var_name, var_val, id",
          table, file);

   if (option_debug)
      ast_log(LOG_DEBUG, "Postgresql RealTime: Static SQL: %s\n", sql);

   /* We now have our complete statement; Lets connect to the server and execute it. */
   ast_mutex_lock(&pgsql_lock);
   if (!pgsql_reconnect(database)) {
      ast_mutex_unlock(&pgsql_lock);
      return NULL;
   }

   if (!(result = PQexec(pgsqlConn, sql))) {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: Failed to query database. Check debug for more info.\n");
      if (option_debug) {
         ast_log(LOG_DEBUG, "Postgresql RealTime: Query: %s\n", sql);
         ast_log(LOG_DEBUG, "Postgresql RealTime: Query Failed because: %s\n",
               PQerrorMessage(pgsqlConn));
      }
      ast_mutex_unlock(&pgsql_lock);
      return NULL;
   } else {
      ExecStatusType result_status = PQresultStatus(result);
      if (result_status != PGRES_COMMAND_OK
         && result_status != PGRES_TUPLES_OK
         && result_status != PGRES_NONFATAL_ERROR) {
         ast_log(LOG_WARNING,
               "Postgresql RealTime: Failed to query database. Check debug for more info.\n");
         if (option_debug) {
            ast_log(LOG_DEBUG, "Postgresql RealTime: Query: %s\n", sql);
            ast_log(LOG_DEBUG, "Postgresql RealTime: Query Failed because: %s (%s)\n",
                  PQresultErrorMessage(result), PQresStatus(result_status));
         }
         ast_mutex_unlock(&pgsql_lock);
         return NULL;
      }
   }

   if ((num_rows = PQntuples(result)) > 0) {
      int numFields = PQnfields(result);
      int i = 0;
      int rowIndex = 0;
      char **fieldnames = NULL;

      if (option_debug)
         ast_log(LOG_DEBUG, "Postgresql RealTime: Found %ld rows.\n", num_rows);

      if (!(fieldnames = ast_calloc(1, numFields * sizeof(char *)))) {
         ast_mutex_unlock(&pgsql_lock);
         PQclear(result);
         return NULL;
      }
      for (i = 0; i < numFields; i++)
         fieldnames[i] = PQfname(result, i);

      for (rowIndex = 0; rowIndex < num_rows; rowIndex++) {
         char *field_category = PQgetvalue(result, rowIndex, 0);
         char *field_var_name = PQgetvalue(result, rowIndex, 1);
         char *field_var_val = PQgetvalue(result, rowIndex, 2);
         char *field_cat_metric = PQgetvalue(result, rowIndex, 3);
         if (!strcmp(field_var_name, "#include")) {
            if (!ast_config_internal_load(field_var_val, cfg, 0)) {
               PQclear(result);
               ast_mutex_unlock(&pgsql_lock);
               return NULL;
            }
            continue;
         }

         if (strcmp(last, field_category) || last_cat_metric != atoi(field_cat_metric)) {
            cur_cat = ast_category_new(field_category);
            if (!cur_cat)
               break;
            strcpy(last, field_category);
            last_cat_metric = atoi(field_cat_metric);
            ast_category_append(cfg, cur_cat);
         }
         new_v = ast_variable_new(field_var_name, field_var_val);
         ast_variable_append(cur_cat, new_v);
      }
   } else {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: Could not find config '%s' in database.\n", file);
   }

   PQclear(result);
   ast_mutex_unlock(&pgsql_lock);

   return cfg;
}

static struct ast_config_engine pgsql_engine = {
   .name = "pgsql",
   .load_func = config_pgsql,
   .realtime_func = realtime_pgsql,
   .realtime_multi_func = realtime_multi_pgsql,
   .update_func = update_pgsql
};

static int load_module(void)
{
   if(!parse_config())
      return AST_MODULE_LOAD_DECLINE;

   ast_mutex_lock(&pgsql_lock);

   if (!pgsql_reconnect(NULL)) {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: Couldn't establish connection. Check debug.\n");
      if (option_debug)
         ast_log(LOG_DEBUG, "Postgresql RealTime: Cannot Connect: %s\n",
               PQerrorMessage(pgsqlConn));
   }

   ast_config_engine_register(&pgsql_engine);
   if (option_verbose) {
      ast_verbose("Postgresql RealTime driver loaded.\n");
   }
   ast_cli_register_multiple(cli_realtime, sizeof(cli_realtime) / sizeof(struct ast_cli_entry));

   ast_mutex_unlock(&pgsql_lock);

   return 0;
}

static int unload_module(void)
{
   /* Aquire control before doing anything to the module itself. */
   ast_mutex_lock(&pgsql_lock);

   if (pgsqlConn) {
      PQfinish(pgsqlConn);
      pgsqlConn = NULL;
   };
   ast_cli_unregister_multiple(cli_realtime, sizeof(cli_realtime) / sizeof(struct ast_cli_entry));
   ast_config_engine_deregister(&pgsql_engine);
   if (option_verbose) {
      ast_verbose("Postgresql RealTime unloaded.\n");
   }

   ast_module_user_hangup_all();

   /* Unlock so something else can destroy the lock. */
   ast_mutex_unlock(&pgsql_lock);

   return 0;
}

static int reload(void)
{
   /* Aquire control before doing anything to the module itself. */
   ast_mutex_lock(&pgsql_lock);

   if (pgsqlConn) {
      PQfinish(pgsqlConn);
      pgsqlConn = NULL;
   };
   parse_config();

   if (!pgsql_reconnect(NULL)) {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: Couldn't establish connection. Check debug.\n");
      if (option_debug)
         ast_log(LOG_DEBUG, "Postgresql RealTime: Cannot Connect: %s\n",
               PQerrorMessage(pgsqlConn));
   }

   ast_verbose(VERBOSE_PREFIX_2 "Postgresql RealTime reloaded.\n");

   /* Done reloading. Release lock so others can now use driver. */
   ast_mutex_unlock(&pgsql_lock);

   return 0;
}

static int parse_config(void)
{
   struct ast_config *config;
   const char *s;

   config = ast_config_load(RES_CONFIG_PGSQL_CONF);

   if (!config) {
      ast_log(LOG_WARNING, "Unable to load config %s\n",RES_CONFIG_PGSQL_CONF);
      return 0;
   }
   if (!(s = ast_variable_retrieve(config, "general", "dbuser"))) {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: No database user found, using 'asterisk' as default.\n");
      strcpy(dbuser, "asterisk");
   } else {
      ast_copy_string(dbuser, s, sizeof(dbuser));
   }

   if (!(s = ast_variable_retrieve(config, "general", "dbpass"))) {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: No database password found, using 'asterisk' as default.\n");
      strcpy(dbpass, "asterisk");
   } else {
      ast_copy_string(dbpass, s, sizeof(dbpass));
   }

   if (!(s = ast_variable_retrieve(config, "general", "dbhost"))) {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: No database host found, using localhost via socket.\n");
      dbhost[0] = '\0';
   } else {
      ast_copy_string(dbhost, s, sizeof(dbhost));
   }

   if (!(s = ast_variable_retrieve(config, "general", "dbname"))) {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: No database name found, using 'asterisk' as default.\n");
      strcpy(dbname, "asterisk");
   } else {
      ast_copy_string(dbname, s, sizeof(dbname));
   }

   if (!(s = ast_variable_retrieve(config, "general", "dbport"))) {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: No database port found, using 5432 as default.\n");
      dbport = 5432;
   } else {
      dbport = atoi(s);
   }

   if (dbhost && !(s = ast_variable_retrieve(config, "general", "dbsock"))) {
      ast_log(LOG_WARNING,
            "Postgresql RealTime: No database socket found, using '/tmp/pgsql.sock' as default.\n");
      strcpy(dbsock, "/tmp/pgsql.sock");
   } else {
      ast_copy_string(dbsock, s, sizeof(dbsock));
   }
   ast_config_destroy(config);

   if (option_debug) {
      if (dbhost) {
         ast_log(LOG_DEBUG, "Postgresql RealTime Host: %s\n", dbhost);
         ast_log(LOG_DEBUG, "Postgresql RealTime Port: %i\n", dbport);
      } else {
         ast_log(LOG_DEBUG, "Postgresql RealTime Socket: %s\n", dbsock);
      }
      ast_log(LOG_DEBUG, "Postgresql RealTime User: %s\n", dbuser);
      ast_log(LOG_DEBUG, "Postgresql RealTime Password: %s\n", dbpass);
      ast_log(LOG_DEBUG, "Postgresql RealTime DBName: %s\n", dbname);
   }

   return 1;
}

static int pgsql_reconnect(const char *database)
{
   char my_database[50];

   ast_copy_string(my_database, S_OR(database, dbname), sizeof(my_database));

   /* mutex lock should have been locked before calling this function. */

   if (pgsqlConn && PQstatus(pgsqlConn) != CONNECTION_OK) {
      PQfinish(pgsqlConn);
      pgsqlConn = NULL;
   }

   if ((!pgsqlConn) && (dbhost || dbsock) && dbuser && dbpass && my_database) {
      char *connInfo = NULL;
      unsigned int size = 100 + strlen(dbhost)
         + strlen(dbuser)
         + strlen(dbpass)
         + strlen(my_database);
      
      if (!(connInfo = ast_malloc(size)))
         return 0;
      
      sprintf(connInfo, "host=%s port=%d dbname=%s user=%s password=%s",
               dbhost, dbport, my_database, dbuser, dbpass);
      if (option_debug)
         ast_log(LOG_DEBUG, "%u connInfo=%s\n", size, connInfo);
      pgsqlConn = PQconnectdb(connInfo);
      if (option_debug)
         ast_log(LOG_DEBUG, "%u connInfo=%s\n", size, connInfo);
      free(connInfo);
      connInfo = NULL;
      if (option_debug)
         ast_log(LOG_DEBUG, "pgsqlConn=%p\n", pgsqlConn);
      if (pgsqlConn && PQstatus(pgsqlConn) == CONNECTION_OK) {
         if (option_debug)
            ast_log(LOG_DEBUG, "Postgresql RealTime: Successfully connected to database.\n");
         connect_time = time(NULL);
         return 1;
      } else {
         ast_log(LOG_ERROR,
               "Postgresql RealTime: Failed to connect database server %s on %s. Check debug for more info.\n",
               dbname, dbhost);
         if (option_debug)
            ast_log(LOG_DEBUG, "Postgresql RealTime: Cannot Connect: %s\n",
                  PQresultErrorMessage(NULL));
         return 0;
      }
   } else {
      if (option_debug)
         ast_log(LOG_DEBUG, "Postgresql RealTime: Everything is fine.\n");
      return 1;
   }
}

static int realtime_pgsql_status(int fd, int argc, char **argv)
{
   char status[256], status2[100] = "";
   int ctime = time(NULL) - connect_time;

   if (pgsqlConn && PQstatus(pgsqlConn) == CONNECTION_OK) {
      if (dbhost) {
         snprintf(status, 255, "Connected to %s@%s, port %d", dbname, dbhost, dbport);
      } else if (dbsock) {
         snprintf(status, 255, "Connected to %s on socket file %s", dbname, dbsock);
      } else {
         snprintf(status, 255, "Connected to %s@%s", dbname, dbhost);
      }

      if (dbuser && *dbuser) {
         snprintf(status2, 99, " with username %s", dbuser);
      }

      if (ctime > 31536000) {
         ast_cli(fd, "%s%s for %d years, %d days, %d hours, %d minutes, %d seconds.\n",
               status, status2, ctime / 31536000, (ctime % 31536000) / 86400,
               (ctime % 86400) / 3600, (ctime % 3600) / 60, ctime % 60);
      } else if (ctime > 86400) {
         ast_cli(fd, "%s%s for %d days, %d hours, %d minutes, %d seconds.\n", status,
               status2, ctime / 86400, (ctime % 86400) / 3600, (ctime % 3600) / 60,
               ctime % 60);
      } else if (ctime > 3600) {
         ast_cli(fd, "%s%s for %d hours, %d minutes, %d seconds.\n", status, status2,
               ctime / 3600, (ctime % 3600) / 60, ctime % 60);
      } else if (ctime > 60) {
         ast_cli(fd, "%s%s for %d minutes, %d seconds.\n", status, status2, ctime / 60,
               ctime % 60);
      } else {
         ast_cli(fd, "%s%s for %d seconds.\n", status, status2, ctime);
      }

      return RESULT_SUCCESS;
   } else {
      return RESULT_FAILURE;
   }
}

/* needs usecount semantics defined */
AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_GLOBAL_SYMBOLS, "PostgreSQL RealTime Configuration Driver",
      .load = load_module,
      .unload = unload_module,
      .reload = reload
          );

---

Asterisk is a trademark for Digium, inc.. | Edvina.net | Asterisk.org | This documentation was generated with Doxygen