Merge "AST_MODULE_INFO: Format corrections to the usages of AST_MODULE_INFO macro."
[asterisk/asterisk.git] / cel / cel_pgsql.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 2008
5  *
6  * Steve Murphy - adapted to CEL, from:
7  * Matthew D. Hardeman <mhardemn@papersoft.com>
8  * Adapted from the MySQL CDR logger originally by James Sharp
9  *
10  * Modified April, 2007; Dec, 2008
11  * Steve Murphy <murf@digium.com>
12
13  * Modified September 2003
14  * Matthew D. Hardeman <mhardemn@papersoft.com>
15  *
16  * See http://www.asterisk.org for more information about
17  * the Asterisk project. Please do not directly contact
18  * any of the maintainers of this project for assistance;
19  * the project provides a web site, mailing lists and IRC
20  * channels for your use.
21  *
22  * This program is free software, distributed under the terms of
23  * the GNU General Public License Version 2. See the LICENSE file
24  * at the top of the source tree.
25  */
26
27 /*! \file
28  *
29  * \brief PostgreSQL CEL logger
30  *
31  * \author Steve Murphy <murf@digium.com>
32  * PostgreSQL http://www.postgresql.org/
33  *
34  * See also
35  * \arg \ref Config_cel
36  * PostgreSQL http://www.postgresql.org/
37  * \ingroup cel_drivers
38  */
39
40 /*** MODULEINFO
41         <depend>pgsql</depend>
42         <support_level>extended</support_level>
43  ***/
44
45 #include "asterisk.h"
46
47 ASTERISK_REGISTER_FILE()
48
49 #include <libpq-fe.h>
50
51 #include "asterisk/config.h"
52 #include "asterisk/options.h"
53 #include "asterisk/channel.h"
54 #include "asterisk/cel.h"
55 #include "asterisk/module.h"
56 #include "asterisk/logger.h"
57 #include "asterisk.h"
58
59 #define DATE_FORMAT "%Y-%m-%d %T.%6q"
60
61 #define PGSQL_BACKEND_NAME "CEL PGSQL backend"
62
63 #define PGSQL_MIN_VERSION_SCHEMA 70300
64
65 static char *config = "cel_pgsql.conf";
66
67 static char *pghostname;
68 static char *pgdbname;
69 static char *pgdbuser;
70 static char *pgpassword;
71 static char *pgappname;
72 static char *pgdbport;
73 static char *table;
74 static char *schema;
75
76 static int connected = 0;
77 /* Optimization to reduce number of memory allocations */
78 static int maxsize = 512, maxsize2 = 512;
79 static int usegmtime = 0;
80
81 /*! \brief show_user_def is off by default */
82 #define CEL_SHOW_USERDEF_DEFAULT        0
83
84 /*! TRUE if we should set the eventtype field to USER_DEFINED on user events. */
85 static unsigned char cel_show_user_def;
86
87 AST_MUTEX_DEFINE_STATIC(pgsql_lock);
88
89 static PGconn   *conn = NULL;
90 static PGresult *result = NULL;
91
92 struct columns {
93         char *name;
94         char *type;
95         int len;
96         unsigned int notnull:1;
97         unsigned int hasdefault:1;
98         AST_RWLIST_ENTRY(columns) list;
99 };
100
101 static AST_RWLIST_HEAD_STATIC(psql_columns, columns);
102
103 #define LENGTHEN_BUF1(size) \
104         do { \
105                 /* Lengthen buffer, if necessary */ \
106                 if (ast_str_strlen(sql) + size + 1 > ast_str_size(sql)) { \
107                         if (ast_str_make_space(&sql, ((ast_str_size(sql) + size + 3) / 512 + 1) * 512) != 0) { \
108                                 ast_log(LOG_ERROR, "Unable to allocate sufficient memory.  Insert CEL '%s:%s' failed.\n", pghostname, table); \
109                                 ast_free(sql); \
110                                 ast_free(sql2); \
111                                 AST_RWLIST_UNLOCK(&psql_columns); \
112                                 return; \
113                         } \
114                 } \
115         } while (0)
116
117 #define LENGTHEN_BUF2(size) \
118         do { \
119                 if (ast_str_strlen(sql2) + size + 1 > ast_str_size(sql2)) { \
120                         if (ast_str_make_space(&sql2, ((ast_str_size(sql2) + size + 3) / 512 + 1) * 512) != 0) { \
121                                 ast_log(LOG_ERROR, "Unable to allocate sufficient memory.  Insert CEL '%s:%s' failed.\n", pghostname, table); \
122                                 ast_free(sql); \
123                                 ast_free(sql2); \
124                                 AST_RWLIST_UNLOCK(&psql_columns); \
125                                 return; \
126                         } \
127                 } \
128         } while (0)
129
130 static void pgsql_reconnect(void)
131 {
132         struct ast_str *conn_info = ast_str_create(128);
133         if (!conn_info) {
134                 ast_log(LOG_ERROR, "Failed to allocate memory for connection string.\n");
135                 return;
136         }
137
138         if (conn) {
139                 PQfinish(conn);
140                 conn = NULL;
141         }
142
143         ast_str_set(&conn_info, 0, "host=%s port=%s dbname=%s user=%s",
144                 pghostname, pgdbport, pgdbname, pgdbuser);
145
146         if (!ast_strlen_zero(pgappname)) {
147                 ast_str_append(&conn_info, 0, " application_name=%s", pgappname);
148         }
149
150         if (!ast_strlen_zero(pgpassword)) {
151                 ast_str_append(&conn_info, 0, " password=%s", pgpassword);
152         }
153
154         conn = PQconnectdb(ast_str_buffer(conn_info));
155         ast_free(conn_info);
156 }
157
158
159 static void pgsql_log(struct ast_event *event)
160 {
161         struct ast_tm tm;
162         char timestr[128];
163         char *pgerror;
164         struct ast_cel_event_record record = {
165                 .version = AST_CEL_EVENT_RECORD_VERSION,
166         };
167
168         if (ast_cel_fill_record(event, &record)) {
169                 return;
170         }
171
172         ast_mutex_lock(&pgsql_lock);
173
174         ast_localtime(&record.event_time, &tm, usegmtime ? "GMT" : NULL);
175         ast_strftime(timestr, sizeof(timestr), DATE_FORMAT, &tm);
176
177         if ((!connected) && pghostname && pgdbuser && pgpassword && pgdbname) {
178                 pgsql_reconnect();
179                 if (PQstatus(conn) != CONNECTION_BAD) {
180                         connected = 1;
181                 } else {
182                         pgerror = PQerrorMessage(conn);
183                         ast_log(LOG_ERROR, "cel_pgsql: Unable to connect to database server %s.  Calls will not be logged!\n", pghostname);
184                         ast_log(LOG_ERROR, "cel_pgsql: Reason: %s\n", pgerror);
185                         PQfinish(conn);
186                         conn = NULL;
187                 }
188         }
189         if (connected) {
190                 struct columns *cur;
191                 struct ast_str *sql = ast_str_create(maxsize), *sql2 = ast_str_create(maxsize2);
192                 char buf[257], escapebuf[513];
193                 const char *value;
194                 int first = 1;
195
196                 if (!sql || !sql2) {
197                         goto ast_log_cleanup;
198                 }
199
200                 ast_str_set(&sql, 0, "INSERT INTO %s (", table);
201                 ast_str_set(&sql2, 0, " VALUES (");
202
203 #define SEP (first ? "" : ",")
204
205                 AST_RWLIST_RDLOCK(&psql_columns);
206                 AST_RWLIST_TRAVERSE(&psql_columns, cur, list) {
207                         LENGTHEN_BUF1(strlen(cur->name) + 2);
208                         ast_str_append(&sql, 0, "%s\"%s\"", SEP, cur->name);
209
210                         if (strcmp(cur->name, "eventtime") == 0) {
211                                 if (strncmp(cur->type, "int", 3) == 0) {
212                                         LENGTHEN_BUF2(13);
213                                         ast_str_append(&sql2, 0, "%s%ld", SEP, (long) record.event_time.tv_sec);
214                                 } else if (strncmp(cur->type, "float", 5) == 0) {
215                                         LENGTHEN_BUF2(31);
216                                         ast_str_append(&sql2, 0, "%s%f",
217                                                 SEP,
218                                                 (double) record.event_time.tv_sec +
219                                                 (double) record.event_time.tv_usec / 1000000.0);
220                                 } else {
221                                         /* char, hopefully */
222                                         LENGTHEN_BUF2(31);
223                                         ast_localtime(&record.event_time, &tm, usegmtime ? "GMT" : NULL);
224                                         ast_strftime(buf, sizeof(buf), DATE_FORMAT, &tm);
225                                         ast_str_append(&sql2, 0, "%s'%s'", SEP, buf);
226                                 }
227                         } else if (strcmp(cur->name, "eventtype") == 0) {
228                                 if (cur->type[0] == 'i') {
229                                         /* Get integer, no need to escape anything */
230                                         LENGTHEN_BUF2(5);
231                                         ast_str_append(&sql2, 0, "%s%d", SEP, (int) record.event_type);
232                                 } else if (strncmp(cur->type, "float", 5) == 0) {
233                                         LENGTHEN_BUF2(31);
234                                         ast_str_append(&sql2, 0, "%s%f", SEP, (double) record.event_type);
235                                 } else {
236                                         /* Char field, probably */
237                                         const char *event_name;
238
239                                         event_name = (!cel_show_user_def
240                                                 && record.event_type == AST_CEL_USER_DEFINED)
241                                                 ? record.user_defined_name : record.event_name;
242                                         LENGTHEN_BUF2(strlen(event_name) + 1);
243                                         ast_str_append(&sql2, 0, "%s'%s'", SEP, event_name);
244                                 }
245                         } else if (strcmp(cur->name, "amaflags") == 0) {
246                                 if (strncmp(cur->type, "int", 3) == 0) {
247                                         /* Integer, no need to escape anything */
248                                         LENGTHEN_BUF2(13);
249                                         ast_str_append(&sql2, 0, "%s%u", SEP, record.amaflag);
250                                 } else {
251                                         /* Although this is a char field, there are no special characters in the values for these fields */
252                                         LENGTHEN_BUF2(31);
253                                         ast_str_append(&sql2, 0, "%s'%u'", SEP, record.amaflag);
254                                 }
255                         } else {
256                                 /* Arbitrary field, could be anything */
257                                 if (strcmp(cur->name, "userdeftype") == 0) {
258                                         value = record.user_defined_name;
259                                 } else if (strcmp(cur->name, "cid_name") == 0) {
260                                         value = record.caller_id_name;
261                                 } else if (strcmp(cur->name, "cid_num") == 0) {
262                                         value = record.caller_id_num;
263                                 } else if (strcmp(cur->name, "cid_ani") == 0) {
264                                         value = record.caller_id_ani;
265                                 } else if (strcmp(cur->name, "cid_rdnis") == 0) {
266                                         value = record.caller_id_rdnis;
267                                 } else if (strcmp(cur->name, "cid_dnid") == 0) {
268                                         value = record.caller_id_dnid;
269                                 } else if (strcmp(cur->name, "exten") == 0) {
270                                         value = record.extension;
271                                 } else if (strcmp(cur->name, "context") == 0) {
272                                         value = record.context;
273                                 } else if (strcmp(cur->name, "channame") == 0) {
274                                         value = record.channel_name;
275                                 } else if (strcmp(cur->name, "appname") == 0) {
276                                         value = record.application_name;
277                                 } else if (strcmp(cur->name, "appdata") == 0) {
278                                         value = record.application_data;
279                                 } else if (strcmp(cur->name, "accountcode") == 0) {
280                                         value = record.account_code;
281                                 } else if (strcmp(cur->name, "peeraccount") == 0) {
282                                         value = record.peer_account;
283                                 } else if (strcmp(cur->name, "uniqueid") == 0) {
284                                         value = record.unique_id;
285                                 } else if (strcmp(cur->name, "linkedid") == 0) {
286                                         value = record.linked_id;
287                                 } else if (strcmp(cur->name, "userfield") == 0) {
288                                         value = record.user_field;
289                                 } else if (strcmp(cur->name, "peer") == 0) {
290                                         value = record.peer;
291                                 } else if (strcmp(cur->name, "extra") == 0) {
292                                         value = record.extra;
293                                 } else {
294                                         value = NULL;
295                                 }
296
297                                 if (value == NULL) {
298                                         ast_str_append(&sql2, 0, "%sDEFAULT", SEP);
299                                 } else if (strncmp(cur->type, "int", 3) == 0) {
300                                         long long whatever;
301                                         if (value && sscanf(value, "%30lld", &whatever) == 1) {
302                                                 LENGTHEN_BUF2(26);
303                                                 ast_str_append(&sql2, 0, "%s%lld", SEP, whatever);
304                                         } else {
305                                                 LENGTHEN_BUF2(2);
306                                                 ast_str_append(&sql2, 0, "%s0", SEP);
307                                         }
308                                 } else if (strncmp(cur->type, "float", 5) == 0) {
309                                         long double whatever;
310                                         if (value && sscanf(value, "%30Lf", &whatever) == 1) {
311                                                 LENGTHEN_BUF2(51);
312                                                 ast_str_append(&sql2, 0, "%s%30Lf", SEP, whatever);
313                                         } else {
314                                                 LENGTHEN_BUF2(2);
315                                                 ast_str_append(&sql2, 0, "%s0", SEP);
316                                         }
317                                         /* XXX Might want to handle dates, times, and other misc fields here XXX */
318                                 } else {
319                                         if (value) {
320                                                 PQescapeStringConn(conn, escapebuf, value, strlen(value), NULL);
321                                         } else {
322                                                 escapebuf[0] = '\0';
323                                         }
324                                         LENGTHEN_BUF2(strlen(escapebuf) + 3);
325                                         ast_str_append(&sql2, 0, "%s'%s'", SEP, escapebuf);
326                                 }
327                         }
328                         first = 0;
329                 }
330                 AST_RWLIST_UNLOCK(&psql_columns);
331                 LENGTHEN_BUF1(ast_str_strlen(sql2) + 2);
332                 ast_str_append(&sql, 0, ")%s)", ast_str_buffer(sql2));
333
334                 ast_debug(3, "Inserting a CEL record: [%s].\n", ast_str_buffer(sql));
335                 /* Test to be sure we're still connected... */
336                 /* If we're connected, and connection is working, good. */
337                 /* Otherwise, attempt reconnect.  If it fails... sorry... */
338                 if (PQstatus(conn) == CONNECTION_OK) {
339                         connected = 1;
340                 } else {
341                         ast_log(LOG_WARNING, "Connection was lost... attempting to reconnect.\n");
342                         PQreset(conn);
343                         if (PQstatus(conn) == CONNECTION_OK) {
344                                 ast_log(LOG_NOTICE, "Connection reestablished.\n");
345                                 connected = 1;
346                         } else {
347                                 pgerror = PQerrorMessage(conn);
348                                 ast_log(LOG_ERROR, "Unable to reconnect to database server %s. Calls will not be logged!\n", pghostname);
349                                 ast_log(LOG_ERROR, "Reason: %s\n", pgerror);
350                                 PQfinish(conn);
351                                 conn = NULL;
352                                 connected = 0;
353                                 goto ast_log_cleanup;
354                         }
355                 }
356                 result = PQexec(conn, ast_str_buffer(sql));
357                 if (PQresultStatus(result) != PGRES_COMMAND_OK) {
358                         pgerror = PQresultErrorMessage(result);
359                         ast_log(LOG_WARNING, "Failed to insert call detail record into database!\n");
360                         ast_log(LOG_WARNING, "Reason: %s\n", pgerror);
361                         ast_log(LOG_WARNING, "Connection may have been lost... attempting to reconnect.\n");
362                         PQreset(conn);
363                         if (PQstatus(conn) == CONNECTION_OK) {
364                                 ast_log(LOG_NOTICE, "Connection reestablished.\n");
365                                 connected = 1;
366                                 PQclear(result);
367                                 result = PQexec(conn, ast_str_buffer(sql));
368                                 if (PQresultStatus(result) != PGRES_COMMAND_OK) {
369                                         pgerror = PQresultErrorMessage(result);
370                                         ast_log(LOG_ERROR, "HARD ERROR!  Attempted reconnection failed.  DROPPING CALL RECORD!\n");
371                                         ast_log(LOG_ERROR, "Reason: %s\n", pgerror);
372                                 }
373                         }
374                         PQclear(result);
375                         goto ast_log_cleanup;
376                 }
377                 PQclear(result);
378
379                 /* Next time, just allocate buffers that are that big to start with. */
380                 if (ast_str_strlen(sql) > maxsize) {
381                         maxsize = ast_str_strlen(sql);
382                 }
383                 if (ast_str_strlen(sql2) > maxsize2) {
384                         maxsize2 = ast_str_strlen(sql2);
385                 }
386
387 ast_log_cleanup:
388                 ast_free(sql);
389                 ast_free(sql2);
390         }
391
392         ast_mutex_unlock(&pgsql_lock);
393 }
394
395 static int my_unload_module(void)
396 {
397         struct columns *current;
398
399         ast_cel_backend_unregister(PGSQL_BACKEND_NAME);
400         AST_RWLIST_WRLOCK(&psql_columns);
401         if (conn) {
402                 PQfinish(conn);
403                 conn = NULL;
404         }
405         if (pghostname) {
406                 ast_free(pghostname);
407                 pghostname = NULL;
408         }
409         if (pgdbname) {
410                 ast_free(pgdbname);
411                 pgdbname = NULL;
412         }
413         if (pgdbuser) {
414                 ast_free(pgdbuser);
415                 pgdbuser = NULL;
416         }
417         if (pgpassword) {
418                 ast_free(pgpassword);
419                 pgpassword = NULL;
420         }
421         if (pgappname) {
422                 ast_free(pgappname);
423                 pgappname = NULL;
424         }
425         if (pgdbport) {
426                 ast_free(pgdbport);
427                 pgdbport = NULL;
428         }
429         if (table) {
430                 ast_free(table);
431                 table = NULL;
432         }
433         if (schema) {
434                 ast_free(schema);
435                 schema = NULL;
436         }
437         while ((current = AST_RWLIST_REMOVE_HEAD(&psql_columns, list))) {
438                 ast_free(current);
439         }
440         AST_RWLIST_UNLOCK(&psql_columns);
441         return 0;
442 }
443
444 static int unload_module(void)
445 {
446         return my_unload_module();
447 }
448
449 static int process_my_load_module(struct ast_config *cfg)
450 {
451         struct ast_variable *var;
452         char *pgerror;
453         const char *tmp;
454         PGresult *result;
455         struct columns *cur;
456
457         if (!(var = ast_variable_browse(cfg, "global"))) {
458                 ast_log(LOG_WARNING,"CEL pgsql config file missing global section.\n");
459                 return AST_MODULE_LOAD_DECLINE;
460         }
461         if (!(tmp = ast_variable_retrieve(cfg,"global","hostname"))) {
462                 ast_log(LOG_WARNING,"PostgreSQL server hostname not specified.  Assuming unix socket connection\n");
463                 tmp = "";       /* connect via UNIX-socket by default */
464         }
465         if (pghostname)
466                 ast_free(pghostname);
467         if (!(pghostname = ast_strdup(tmp))) {
468                 ast_log(LOG_WARNING,"PostgreSQL Ran out of memory copying host info\n");
469                 return AST_MODULE_LOAD_DECLINE;
470         }
471         if (!(tmp = ast_variable_retrieve(cfg, "global", "dbname"))) {
472                 ast_log(LOG_WARNING,"PostgreSQL database not specified.  Assuming asterisk\n");
473                 tmp = "asteriskceldb";
474         }
475         if (pgdbname)
476                 ast_free(pgdbname);
477         if (!(pgdbname = ast_strdup(tmp))) {
478                 ast_log(LOG_WARNING,"PostgreSQL Ran out of memory copying dbname info\n");
479                 return AST_MODULE_LOAD_DECLINE;
480         }
481         if (!(tmp = ast_variable_retrieve(cfg, "global", "user"))) {
482                 ast_log(LOG_WARNING,"PostgreSQL database user not specified.  Assuming asterisk\n");
483                 tmp = "asterisk";
484         }
485         if (pgdbuser)
486                 ast_free(pgdbuser);
487         if (!(pgdbuser = ast_strdup(tmp))) {
488                 ast_log(LOG_WARNING,"PostgreSQL Ran out of memory copying user info\n");
489                 return AST_MODULE_LOAD_DECLINE;
490         }
491         if (!(tmp = ast_variable_retrieve(cfg, "global", "password"))) {
492                 ast_log(LOG_WARNING, "PostgreSQL database password not specified.  Assuming blank\n");
493                 tmp = "";
494         }
495         if (pgpassword)
496                 ast_free(pgpassword);
497         if (!(pgpassword = ast_strdup(tmp))) {
498                 ast_log(LOG_WARNING,"PostgreSQL Ran out of memory copying password info\n");
499                 return AST_MODULE_LOAD_DECLINE;
500         }
501         if (!(tmp = ast_variable_retrieve(cfg, "global", "appname"))) {
502                 tmp = "";
503         }
504         if (pgappname) {
505                 ast_free(pgappname);
506         }
507         if (!(pgappname = ast_strdup(tmp))) {
508                 ast_log(LOG_WARNING,"PostgreSQL Ran out of memory copying appname info\n");
509                 return AST_MODULE_LOAD_DECLINE;
510         }
511
512         if (!(tmp = ast_variable_retrieve(cfg,"global","port"))) {
513                 ast_log(LOG_WARNING,"PostgreSQL database port not specified.  Using default 5432.\n");
514                 tmp = "5432";
515         }
516         if (pgdbport)
517                 ast_free(pgdbport);
518         if (!(pgdbport = ast_strdup(tmp))) {
519                 ast_log(LOG_WARNING,"PostgreSQL Ran out of memory copying port info\n");
520                 return AST_MODULE_LOAD_DECLINE;
521         }
522         if (!(tmp = ast_variable_retrieve(cfg, "global", "table"))) {
523                 ast_log(LOG_WARNING,"CEL table not specified.  Assuming cel\n");
524                 tmp = "cel";
525         }
526         if (table)
527                 ast_free(table);
528         if (!(table = ast_strdup(tmp))) {
529                 return AST_MODULE_LOAD_DECLINE;
530         }
531         cel_show_user_def = CEL_SHOW_USERDEF_DEFAULT;
532         if ((tmp = ast_variable_retrieve(cfg, "global", "show_user_defined"))) {
533                 cel_show_user_def = ast_true(tmp) ? 1 : 0;
534         }
535         if ((tmp = ast_variable_retrieve(cfg, "global", "usegmtime"))) {
536                 usegmtime = ast_true(tmp);
537         } else {
538                 usegmtime = 0;
539         }
540         if (!(tmp = ast_variable_retrieve(cfg, "global", "schema"))) {
541                 tmp = "";
542         }
543         if (schema) {
544                 ast_free(schema);
545         }
546         if (!(schema = ast_strdup(tmp))) {
547                 ast_log(LOG_WARNING,"PostgreSQL Ran out of memory copying schema info\n");
548                 return AST_MODULE_LOAD_DECLINE;
549         }
550         if (option_debug) {
551                 if (ast_strlen_zero(pghostname)) {
552                         ast_debug(3, "cel_pgsql: using default unix socket\n");
553                 } else {
554                         ast_debug(3, "cel_pgsql: got hostname of %s\n", pghostname);
555                 }
556                 ast_debug(3, "cel_pgsql: got port of %s\n", pgdbport);
557                 ast_debug(3, "cel_pgsql: got user of %s\n", pgdbuser);
558                 ast_debug(3, "cel_pgsql: got dbname of %s\n", pgdbname);
559                 ast_debug(3, "cel_pgsql: got password of %s\n", pgpassword);
560                 ast_debug(3, "cel_pgsql: got sql table name of %s\n", table);
561                 ast_debug(3, "cel_pgsql: got show_user_defined of %s\n",
562                         cel_show_user_def ? "Yes" : "No");
563         }
564
565         pgsql_reconnect();
566         if (PQstatus(conn) != CONNECTION_BAD) {
567                 char sqlcmd[768];
568                 char *fname, *ftype, *flen, *fnotnull, *fdef, *tablename, *tmp_tablename;
569                 int i, rows, version;
570
571                 ast_debug(1, "Successfully connected to PostgreSQL database.\n");
572                 connected = 1;
573
574                 version = PQserverVersion(conn);
575                 /* Remove any schema name from the table */
576                 if ((tmp_tablename = strrchr(table, '.'))) {
577                         tmp_tablename++;
578                 } else {
579                         tmp_tablename = table;
580                 }
581                 tablename = ast_alloca(strlen(tmp_tablename) * 2 + 1);
582                 PQescapeStringConn(conn, tablename, tmp_tablename, strlen(tmp_tablename), NULL);
583                 if (version >= PGSQL_MIN_VERSION_SCHEMA) {
584                         char *schemaname;
585                         int lenschema;
586                         lenschema = strlen(schema);
587                         schemaname = ast_alloca(lenschema * 2 + 1);
588                         PQescapeStringConn(conn, schemaname, schema, lenschema, NULL);
589
590                         snprintf(sqlcmd, sizeof(sqlcmd),
591                         "SELECT a.attname, t.typname, a.attlen, a.attnotnull, d.adsrc, a.atttypmod "
592                         "FROM (((pg_catalog.pg_class c INNER JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace "
593                                  "AND c.relname = '%s' AND n.nspname = %s%s%s) "
594                                "INNER JOIN pg_catalog.pg_attribute a ON ("
595                                    "NOT a.attisdropped) AND a.attnum > 0 AND a.attrelid = c.oid) "
596                             "INNER JOIN pg_catalog.pg_type t ON t.oid = a.atttypid) "
597                         "LEFT OUTER JOIN pg_attrdef d ON a.atthasdef AND d.adrelid = a.attrelid "
598                           "AND d.adnum = a.attnum "
599                         "ORDER BY n.nspname, c.relname, attnum",
600                                 tablename,
601                                 lenschema == 0 ? "" : "'", lenschema == 0 ? "current_schema()" : schemaname, lenschema == 0 ? "" : "'");
602                 } else {
603                         snprintf(sqlcmd, sizeof(sqlcmd),
604                         "SELECT a.attname, t.typname, a.attlen, a.attnotnull, d.adsrc, a.atttypmod "
605                         "FROM pg_class c, pg_type t, pg_attribute a "
606                         "LEFT OUTER JOIN pg_attrdef d ON a.atthasdef AND d.adrelid = a.attrelid "
607                         "AND d.adnum = a.attnum WHERE c.oid = a.attrelid AND a.atttypid = t.oid "
608                         "AND (a.attnum > 0) AND c.relname = '%s' ORDER BY c.relname, attnum", tablename);
609                 }
610                 /* Query the columns */
611                 result = PQexec(conn, sqlcmd);
612                 if (PQresultStatus(result) != PGRES_TUPLES_OK) {
613                         pgerror = PQresultErrorMessage(result);
614                         ast_log(LOG_ERROR, "Failed to query database columns: %s\n", pgerror);
615                         PQclear(result);
616                         unload_module();
617                         return AST_MODULE_LOAD_DECLINE;
618                 }
619
620                 rows = PQntuples(result);
621                 for (i = 0; i < rows; i++) {
622                         fname = PQgetvalue(result, i, 0);
623                         ftype = PQgetvalue(result, i, 1);
624                         flen = PQgetvalue(result, i, 2);
625                         fnotnull = PQgetvalue(result, i, 3);
626                         fdef = PQgetvalue(result, i, 4);
627                         ast_verb(4, "Found column '%s' of type '%s'\n", fname, ftype);
628                         cur = ast_calloc(1, sizeof(*cur) + strlen(fname) + strlen(ftype) + 2);
629                         if (cur) {
630                                 sscanf(flen, "%30d", &cur->len);
631                                 cur->name = (char *)cur + sizeof(*cur);
632                                 cur->type = (char *)cur + sizeof(*cur) + strlen(fname) + 1;
633                                 strcpy(cur->name, fname);
634                                 strcpy(cur->type, ftype);
635                                 if (*fnotnull == 't') {
636                                         cur->notnull = 1;
637                                 } else {
638                                         cur->notnull = 0;
639                                 }
640                                 if (!ast_strlen_zero(fdef)) {
641                                         cur->hasdefault = 1;
642                                 } else {
643                                         cur->hasdefault = 0;
644                                 }
645                                 AST_RWLIST_INSERT_TAIL(&psql_columns, cur, list);
646                         }
647                 }
648                 PQclear(result);
649         } else {
650                 pgerror = PQerrorMessage(conn);
651                 ast_log(LOG_ERROR, "cel_pgsql: Unable to connect to database server %s.  CALLS WILL NOT BE LOGGED!!\n", pghostname);
652                 ast_log(LOG_ERROR, "cel_pgsql: Reason: %s\n", pgerror);
653                 connected = 0;
654                 PQfinish(conn);
655                 conn = NULL;
656         }
657         return AST_MODULE_LOAD_SUCCESS;
658 }
659
660 static int my_load_module(int reload)
661 {
662         struct ast_config *cfg;
663         struct ast_flags config_flags = { reload ? CONFIG_FLAG_FILEUNCHANGED : 0 };
664
665         if ((cfg = ast_config_load(config, config_flags)) == NULL || cfg == CONFIG_STATUS_FILEINVALID) {
666                 ast_log(LOG_WARNING, "Unable to load config for PostgreSQL CEL's: %s\n", config);
667                 return AST_MODULE_LOAD_DECLINE;
668         } else if (cfg == CONFIG_STATUS_FILEUNCHANGED) {
669                 return AST_MODULE_LOAD_SUCCESS;
670         }
671
672         if (reload) {
673                 my_unload_module();
674         }
675
676         process_my_load_module(cfg);
677         ast_config_destroy(cfg);
678
679         if (ast_cel_backend_register(PGSQL_BACKEND_NAME, pgsql_log)) {
680                 ast_log(LOG_WARNING, "Unable to subscribe to CEL events for pgsql\n");
681                 return AST_MODULE_LOAD_DECLINE;
682         }
683
684         return AST_MODULE_LOAD_SUCCESS;
685 }
686
687 static int load_module(void)
688 {
689         return my_load_module(0);
690 }
691
692 static int reload(void)
693 {
694         return my_load_module(1);
695 }
696
697 AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "PostgreSQL CEL Backend",
698         .support_level = AST_MODULE_SUPPORT_EXTENDED,
699         .load = load_module,
700         .unload = unload_module,
701         .reload = reload,
702         .load_pri = AST_MODPRI_CDR_DRIVER,
703 );