2 * Asterisk -- An open source telephony toolkit.
4 * Copyright (C) 2003 - 2006
6 * Matthew D. Hardeman <mhardemn@papersoft.com>
7 * Adapted from the MySQL CDR logger originally by James Sharp
9 * Modified September 2003
10 * Matthew D. Hardeman <mhardemn@papersoft.com>
12 * See http://www.asterisk.org for more information about
13 * the Asterisk project. Please do not directly contact
14 * any of the maintainers of this project for assistance;
15 * the project provides a web site, mailing lists and IRC
16 * channels for your use.
18 * This program is free software, distributed under the terms of
19 * the GNU General Public License Version 2. See the LICENSE file
20 * at the top of the source tree.
25 * \brief PostgreSQL CDR logger
27 * \author Matthew D. Hardeman <mhardemn@papersoft.com>
30 * \arg \ref Config_cdr
31 * \arg http://www.postgresql.org/
32 * \ingroup cdr_drivers
36 <depend>pgsql</depend>
41 ASTERISK_FILE_VERSION(__FILE__
, "$Revision$")
43 #include <sys/types.h>
52 #include "asterisk/config.h"
53 #include "asterisk/options.h"
54 #include "asterisk/channel.h"
55 #include "asterisk/cdr.h"
56 #include "asterisk/module.h"
57 #include "asterisk/logger.h"
60 #define DATE_FORMAT "%Y-%m-%d %T"
62 static char *name
= "pgsql";
63 static char *config
= "cdr_pgsql.conf";
64 static char *pghostname
= NULL
, *pgdbname
= NULL
, *pgdbuser
= NULL
, *pgpassword
= NULL
, *pgdbport
= NULL
, *table
= NULL
;
65 static int connected
= 0;
67 AST_MUTEX_DEFINE_STATIC(pgsql_lock
);
70 static PGresult
*result
;
72 static int pgsql_log(struct ast_cdr
*cdr
)
75 char sqlcmd
[2048] = "", timestr
[128];
78 ast_mutex_lock(&pgsql_lock
);
80 localtime_r(&cdr
->start
.tv_sec
,&tm
);
81 strftime(timestr
, sizeof(timestr
), DATE_FORMAT
, &tm
);
83 if ((!connected
) && pghostname
&& pgdbuser
&& pgpassword
&& pgdbname
) {
84 conn
= PQsetdbLogin(pghostname
, pgdbport
, NULL
, NULL
, pgdbname
, pgdbuser
, pgpassword
);
85 if (PQstatus(conn
) != CONNECTION_BAD
) {
88 pgerror
= PQerrorMessage(conn
);
89 ast_log(LOG_ERROR
, "cdr_pgsql: Unable to connect to database server %s. Calls will not be logged!\n", pghostname
);
90 ast_log(LOG_ERROR
, "cdr_pgsql: Reason: %s\n", pgerror
);
95 char *clid
=NULL
, *dcontext
=NULL
, *channel
=NULL
, *dstchannel
=NULL
, *lastapp
=NULL
, *lastdata
=NULL
;
96 char *uniqueid
=NULL
, *userfield
=NULL
;
98 /* Maximum space needed would be if all characters needed to be escaped, plus a trailing NULL */
99 if ((clid
= alloca(strlen(cdr
->clid
) * 2 + 1)) != NULL
)
100 PQescapeString(clid
, cdr
->clid
, strlen(cdr
->clid
));
101 if ((dcontext
= alloca(strlen(cdr
->dcontext
) * 2 + 1)) != NULL
)
102 PQescapeString(dcontext
, cdr
->dcontext
, strlen(cdr
->dcontext
));
103 if ((channel
= alloca(strlen(cdr
->channel
) * 2 + 1)) != NULL
)
104 PQescapeString(channel
, cdr
->channel
, strlen(cdr
->channel
));
105 if ((dstchannel
= alloca(strlen(cdr
->dstchannel
) * 2 + 1)) != NULL
)
106 PQescapeString(dstchannel
, cdr
->dstchannel
, strlen(cdr
->dstchannel
));
107 if ((lastapp
= alloca(strlen(cdr
->lastapp
) * 2 + 1)) != NULL
)
108 PQescapeString(lastapp
, cdr
->lastapp
, strlen(cdr
->lastapp
));
109 if ((lastdata
= alloca(strlen(cdr
->lastdata
) * 2 + 1)) != NULL
)
110 PQescapeString(lastdata
, cdr
->lastdata
, strlen(cdr
->lastdata
));
111 if ((uniqueid
= alloca(strlen(cdr
->uniqueid
) * 2 + 1)) != NULL
)
112 PQescapeString(uniqueid
, cdr
->uniqueid
, strlen(cdr
->uniqueid
));
113 if ((userfield
= alloca(strlen(cdr
->userfield
) * 2 + 1)) != NULL
)
114 PQescapeString(userfield
, cdr
->userfield
, strlen(cdr
->userfield
));
116 /* Check for all alloca failures above at once */
117 if ((!clid
) || (!dcontext
) || (!channel
) || (!dstchannel
) || (!lastapp
) || (!lastdata
) || (!uniqueid
) || (!userfield
)) {
118 ast_log(LOG_ERROR
, "cdr_pgsql: Out of memory error (insert fails)\n");
119 ast_mutex_unlock(&pgsql_lock
);
123 if (option_debug
> 1)
124 ast_log(LOG_DEBUG
, "cdr_pgsql: inserting a CDR record.\n");
126 snprintf(sqlcmd
,sizeof(sqlcmd
),"INSERT INTO %s (calldate,clid,src,dst,dcontext,channel,dstchannel,"
127 "lastapp,lastdata,duration,billsec,disposition,amaflags,accountcode,uniqueid,userfield) VALUES"
128 " ('%s','%s','%s','%s','%s', '%s','%s','%s','%s',%ld,%ld,'%s',%ld,'%s','%s','%s')",
129 table
,timestr
,clid
,cdr
->src
, cdr
->dst
, dcontext
,channel
, dstchannel
, lastapp
, lastdata
,
130 cdr
->duration
,cdr
->billsec
,ast_cdr_disp2str(cdr
->disposition
),cdr
->amaflags
, cdr
->accountcode
, uniqueid
, userfield
);
132 if (option_debug
> 2)
133 ast_log(LOG_DEBUG
, "cdr_pgsql: SQL command executed: %s\n",sqlcmd
);
135 /* Test to be sure we're still connected... */
136 /* If we're connected, and connection is working, good. */
137 /* Otherwise, attempt reconnect. If it fails... sorry... */
138 if (PQstatus(conn
) == CONNECTION_OK
) {
141 ast_log(LOG_ERROR
, "cdr_pgsql: Connection was lost... attempting to reconnect.\n");
143 if (PQstatus(conn
) == CONNECTION_OK
) {
144 ast_log(LOG_ERROR
, "cdr_pgsql: Connection reestablished.\n");
147 pgerror
= PQerrorMessage(conn
);
148 ast_log(LOG_ERROR
, "cdr_pgsql: Unable to reconnect to database server %s. Calls will not be logged!\n", pghostname
);
149 ast_log(LOG_ERROR
, "cdr_pgsql: Reason: %s\n", pgerror
);
151 ast_mutex_unlock(&pgsql_lock
);
155 result
= PQexec(conn
, sqlcmd
);
156 if ( PQresultStatus(result
) != PGRES_COMMAND_OK
) {
157 pgerror
= PQresultErrorMessage(result
);
158 ast_log(LOG_ERROR
,"cdr_pgsql: Failed to insert call detail record into database!\n");
159 ast_log(LOG_ERROR
,"cdr_pgsql: Reason: %s\n", pgerror
);
160 ast_log(LOG_ERROR
,"cdr_pgsql: Connection may have been lost... attempting to reconnect.\n");
162 if (PQstatus(conn
) == CONNECTION_OK
) {
163 ast_log(LOG_ERROR
, "cdr_pgsql: Connection reestablished.\n");
165 result
= PQexec(conn
, sqlcmd
);
166 if ( PQresultStatus(result
) != PGRES_COMMAND_OK
)
168 pgerror
= PQresultErrorMessage(result
);
169 ast_log(LOG_ERROR
,"cdr_pgsql: HARD ERROR! Attempted reconnection failed. DROPPING CALL RECORD!\n");
170 ast_log(LOG_ERROR
,"cdr_pgsql: Reason: %s\n", pgerror
);
173 ast_mutex_unlock(&pgsql_lock
);
177 ast_mutex_unlock(&pgsql_lock
);
181 static int my_unload_module(void)
197 ast_cdr_unregister(name
);
201 static int process_my_load_module(struct ast_config
*cfg
)
203 struct ast_variable
*var
;
207 if (!(var
= ast_variable_browse(cfg
, "global")))
210 if (!(tmp
= ast_variable_retrieve(cfg
,"global","hostname"))) {
211 ast_log(LOG_WARNING
,"PostgreSQL server hostname not specified. Assuming unix socket connection\n");
212 tmp
= ""; /* connect via UNIX-socket by default */
215 if (!(pghostname
= ast_strdup(tmp
)))
218 if (!(tmp
= ast_variable_retrieve(cfg
, "global", "dbname"))) {
219 ast_log(LOG_WARNING
,"PostgreSQL database not specified. Assuming asterisk\n");
220 tmp
= "asteriskcdrdb";
223 if (!(pgdbname
= ast_strdup(tmp
)))
226 if (!(tmp
= ast_variable_retrieve(cfg
, "global", "user"))) {
227 ast_log(LOG_WARNING
,"PostgreSQL database user not specified. Assuming asterisk\n");
231 if (!(pgdbuser
= ast_strdup(tmp
)))
234 if (!(tmp
= ast_variable_retrieve(cfg
, "global", "password"))) {
235 ast_log(LOG_WARNING
,"PostgreSQL database password not specified. Assuming blank\n");
239 if (!(pgpassword
= ast_strdup(tmp
)))
242 if (!(tmp
= ast_variable_retrieve(cfg
,"global","port"))) {
243 ast_log(LOG_WARNING
,"PostgreSQL database port not specified. Using default 5432.\n");
247 if (!(pgdbport
= ast_strdup(tmp
)))
250 if (!(tmp
= ast_variable_retrieve(cfg
, "global", "table"))) {
251 ast_log(LOG_WARNING
,"CDR table not specified. Assuming cdr\n");
255 if (!(table
= ast_strdup(tmp
)))
259 if (ast_strlen_zero(pghostname
))
260 ast_log(LOG_DEBUG
, "cdr_pgsql: using default unix socket\n");
262 ast_log(LOG_DEBUG
, "cdr_pgsql: got hostname of %s\n", pghostname
);
263 ast_log(LOG_DEBUG
, "cdr_pgsql: got port of %s\n", pgdbport
);
264 ast_log(LOG_DEBUG
, "cdr_pgsql: got user of %s\n", pgdbuser
);
265 ast_log(LOG_DEBUG
, "cdr_pgsql: got dbname of %s\n", pgdbname
);
266 ast_log(LOG_DEBUG
, "cdr_pgsql: got password of %s\n", pgpassword
);
267 ast_log(LOG_DEBUG
, "cdr_pgsql: got sql table name of %s\n", table
);
270 conn
= PQsetdbLogin(pghostname
, pgdbport
, NULL
, NULL
, pgdbname
, pgdbuser
, pgpassword
);
271 if (PQstatus(conn
) != CONNECTION_BAD
) {
273 ast_log(LOG_DEBUG
, "Successfully connected to PostgreSQL database.\n");
276 pgerror
= PQerrorMessage(conn
);
277 ast_log(LOG_ERROR
, "cdr_pgsql: Unable to connect to database server %s. CALLS WILL NOT BE LOGGED!!\n", pghostname
);
278 ast_log(LOG_ERROR
, "cdr_pgsql: Reason: %s\n", pgerror
);
282 return ast_cdr_register(name
, ast_module_info
->description
, pgsql_log
);
285 static int my_load_module(void)
287 struct ast_config
*cfg
;
290 if (!(cfg
= ast_config_load(config
))) {
291 ast_log(LOG_WARNING
, "Unable to load config for PostgreSQL CDR's: %s\n", config
);
292 return AST_MODULE_LOAD_DECLINE
;
295 res
= process_my_load_module(cfg
);
296 ast_config_destroy(cfg
);
301 static int load_module(void)
303 return my_load_module();
306 static int unload_module(void)
308 return my_unload_module();
311 static int reload(void)
314 return my_load_module();
317 AST_MODULE_INFO(ASTERISK_GPL_KEY
, AST_MODFLAG_DEFAULT
, "PostgreSQL CDR Backend",
319 .unload
= unload_module
,