Merged revisions 291038 via svnmerge from
[asterisk/asterisk.git] / cdr / cdr_odbc.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 2003-2005, Digium, Inc.
5  *
6  * Brian K. West <brian@bkw.org>
7  *
8  * See http://www.asterisk.org for more information about
9  * the Asterisk project. Please do not directly contact
10  * any of the maintainers of this project for assistance;
11  * the project provides a web site, mailing lists and IRC
12  * channels for your use.
13  *
14  * This program is free software, distributed under the terms of
15  * the GNU General Public License Version 2. See the LICENSE file
16  * at the top of the source tree.
17  */
18
19 /*!
20  * \file
21  * \brief ODBC CDR Backend
22  *
23  * \author Brian K. West <brian@bkw.org>
24  *
25  * See also:
26  * \arg http://www.unixodbc.org
27  * \arg \ref Config_cdr
28  * \ingroup cdr_drivers
29  */
30
31 /*** MODULEINFO
32         <depend>res_odbc</depend>
33  ***/
34
35 #include "asterisk.h"
36
37 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
38
39 #include "asterisk/config.h"
40 #include "asterisk/channel.h"
41 #include "asterisk/cdr.h"
42 #include "asterisk/module.h"
43 #include "asterisk/res_odbc.h"
44
45 #define DATE_FORMAT "%Y-%m-%d %T"
46
47 static const char name[] = "ODBC";
48 static const char config_file[] = "cdr_odbc.conf";
49 static char *dsn = NULL, *table = NULL;
50
51 enum {
52         CONFIG_LOGUNIQUEID =       1 << 0,
53         CONFIG_USEGMTIME =         1 << 1,
54         CONFIG_DISPOSITIONSTRING = 1 << 2,
55         CONFIG_HRTIME =            1 << 3,
56         CONFIG_REGISTERED =        1 << 4,
57 };
58
59 static struct ast_flags config = { 0 };
60
61 static SQLHSTMT execute_cb(struct odbc_obj *obj, void *data)
62 {
63         struct ast_cdr *cdr = data;
64         SQLRETURN ODBC_res;
65         char sqlcmd[2048] = "", timestr[128];
66         struct ast_tm tm;
67         SQLHSTMT stmt;
68
69         ast_localtime(&cdr->start, &tm, ast_test_flag(&config, CONFIG_USEGMTIME) ? "GMT" : NULL);
70         ast_strftime(timestr, sizeof(timestr), DATE_FORMAT, &tm);
71
72         if (ast_test_flag(&config, CONFIG_LOGUNIQUEID)) {
73                 snprintf(sqlcmd,sizeof(sqlcmd),"INSERT INTO %s "
74                 "(calldate,clid,src,dst,dcontext,channel,dstchannel,lastapp,"
75                 "lastdata,duration,billsec,disposition,amaflags,accountcode,uniqueid,userfield) "
76                 "VALUES ({ts '%s'},?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)", table, timestr);
77         } else {
78                 snprintf(sqlcmd,sizeof(sqlcmd),"INSERT INTO %s "
79                 "(calldate,clid,src,dst,dcontext,channel,dstchannel,lastapp,lastdata,"
80                 "duration,billsec,disposition,amaflags,accountcode) "
81                 "VALUES ({ts '%s'},?,?,?,?,?,?,?,?,?,?,?,?,?)", table, timestr);
82         }
83
84         ODBC_res = SQLAllocHandle(SQL_HANDLE_STMT, obj->con, &stmt);
85
86         if ((ODBC_res != SQL_SUCCESS) && (ODBC_res != SQL_SUCCESS_WITH_INFO)) {
87                 ast_verb(11, "cdr_odbc: Failure in AllocStatement %d\n", ODBC_res);
88                 SQLFreeHandle(SQL_HANDLE_STMT, stmt);
89                 return NULL;
90         }
91
92         SQLBindParameter(stmt, 1, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->clid), 0, cdr->clid, 0, NULL);
93         SQLBindParameter(stmt, 2, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->src), 0, cdr->src, 0, NULL);
94         SQLBindParameter(stmt, 3, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->dst), 0, cdr->dst, 0, NULL);
95         SQLBindParameter(stmt, 4, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->dcontext), 0, cdr->dcontext, 0, NULL);
96         SQLBindParameter(stmt, 5, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->channel), 0, cdr->channel, 0, NULL);
97         SQLBindParameter(stmt, 6, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->dstchannel), 0, cdr->dstchannel, 0, NULL);
98         SQLBindParameter(stmt, 7, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->lastapp), 0, cdr->lastapp, 0, NULL);
99         SQLBindParameter(stmt, 8, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->lastdata), 0, cdr->lastdata, 0, NULL);
100
101         if (ast_test_flag(&config, CONFIG_HRTIME)) {
102                 double hrbillsec = 0.0;
103                 double hrduration;
104
105                 if (!ast_tvzero(cdr->answer)) {
106                         hrbillsec = (double) ast_tvdiff_us(cdr->end, cdr->answer) / 1000000.0;
107                 }
108                 hrduration = (double) ast_tvdiff_us(cdr->end, cdr->start) / 1000000.0;
109
110                 SQLBindParameter(stmt, 9, SQL_PARAM_INPUT, SQL_C_DOUBLE, SQL_FLOAT, 0, 0, &hrbillsec, 0, NULL);
111                 SQLBindParameter(stmt, 10, SQL_PARAM_INPUT, SQL_C_DOUBLE, SQL_FLOAT, 0, 0, &hrduration, 0, NULL);
112         } else {
113                 SQLBindParameter(stmt, 9, SQL_PARAM_INPUT, SQL_C_SLONG, SQL_INTEGER, 0, 0, &cdr->duration, 0, NULL);
114                 SQLBindParameter(stmt, 10, SQL_PARAM_INPUT, SQL_C_SLONG, SQL_INTEGER, 0, 0, &cdr->billsec, 0, NULL);
115         }
116
117         if (ast_test_flag(&config, CONFIG_DISPOSITIONSTRING))
118                 SQLBindParameter(stmt, 11, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, strlen(ast_cdr_disp2str(cdr->disposition)) + 1, 0, ast_cdr_disp2str(cdr->disposition), 0, NULL);
119         else
120                 SQLBindParameter(stmt, 11, SQL_PARAM_INPUT, SQL_C_SLONG, SQL_INTEGER, 0, 0, &cdr->disposition, 0, NULL);
121         SQLBindParameter(stmt, 12, SQL_PARAM_INPUT, SQL_C_SLONG, SQL_INTEGER, 0, 0, &cdr->amaflags, 0, NULL);
122         SQLBindParameter(stmt, 13, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->accountcode), 0, cdr->accountcode, 0, NULL);
123
124         if (ast_test_flag(&config, CONFIG_LOGUNIQUEID)) {
125                 SQLBindParameter(stmt, 14, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->uniqueid), 0, cdr->uniqueid, 0, NULL);
126                 SQLBindParameter(stmt, 15, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, sizeof(cdr->userfield), 0, cdr->userfield, 0, NULL);
127         }
128
129         ODBC_res = SQLExecDirect(stmt, (unsigned char *)sqlcmd, SQL_NTS);
130
131         if ((ODBC_res != SQL_SUCCESS) && (ODBC_res != SQL_SUCCESS_WITH_INFO)) {
132                 ast_verb(11, "cdr_odbc: Error in ExecDirect: %d\n", ODBC_res);
133                 SQLFreeHandle(SQL_HANDLE_STMT, stmt);
134                 return NULL;
135         }
136
137         return stmt;
138 }
139
140
141 static int odbc_log(struct ast_cdr *cdr)
142 {
143         struct odbc_obj *obj = ast_odbc_request_obj(dsn, 0);
144         SQLHSTMT stmt;
145
146         if (!obj) {
147                 ast_log(LOG_ERROR, "Unable to retrieve database handle.  CDR failed.\n");
148                 return -1;
149         }
150
151         stmt = ast_odbc_direct_execute(obj, execute_cb, cdr);
152         if (stmt) {
153                 SQLLEN rows = 0;
154
155                 SQLRowCount(stmt, &rows);
156                 SQLFreeHandle(SQL_HANDLE_STMT, stmt);
157
158                 if (rows == 0)
159                         ast_log(LOG_WARNING, "CDR successfully ran, but inserted 0 rows?\n");
160         } else
161                 ast_log(LOG_ERROR, "CDR direct execute failed\n");
162         ast_odbc_release_obj(obj);
163         return 0;
164 }
165
166 static int odbc_load_module(int reload)
167 {
168         int res = 0;
169         struct ast_config *cfg;
170         struct ast_variable *var;
171         const char *tmp;
172         struct ast_flags config_flags = { reload ? CONFIG_FLAG_FILEUNCHANGED : 0 };
173
174         do {
175                 cfg = ast_config_load(config_file, config_flags);
176                 if (!cfg || cfg == CONFIG_STATUS_FILEINVALID) {
177                         ast_log(LOG_WARNING, "cdr_odbc: Unable to load config for ODBC CDR's: %s\n", config_file);
178                         res = AST_MODULE_LOAD_DECLINE;
179                         break;
180                 } else if (cfg == CONFIG_STATUS_FILEUNCHANGED)
181                         break;
182
183                 var = ast_variable_browse(cfg, "global");
184                 if (!var) {
185                         /* nothing configured */
186                         break;
187                 }
188
189                 if ((tmp = ast_variable_retrieve(cfg, "global", "dsn")) == NULL) {
190                         ast_log(LOG_WARNING, "cdr_odbc: dsn not specified.  Assuming asteriskdb\n");
191                         tmp = "asteriskdb";
192                 }
193                 if (dsn)
194                         ast_free(dsn);
195                 dsn = ast_strdup(tmp);
196                 if (dsn == NULL) {
197                         res = -1;
198                         break;
199                 }
200
201                 if (((tmp = ast_variable_retrieve(cfg, "global", "dispositionstring"))) && ast_true(tmp))
202                         ast_set_flag(&config, CONFIG_DISPOSITIONSTRING);
203                 else
204                         ast_clear_flag(&config, CONFIG_DISPOSITIONSTRING);
205
206                 if (((tmp = ast_variable_retrieve(cfg, "global", "loguniqueid"))) && ast_true(tmp)) {
207                         ast_set_flag(&config, CONFIG_LOGUNIQUEID);
208                         ast_debug(1, "cdr_odbc: Logging uniqueid\n");
209                 } else {
210                         ast_clear_flag(&config, CONFIG_LOGUNIQUEID);
211                         ast_debug(1, "cdr_odbc: Not logging uniqueid\n");
212                 }
213
214                 if (((tmp = ast_variable_retrieve(cfg, "global", "usegmtime"))) && ast_true(tmp)) {
215                         ast_set_flag(&config, CONFIG_USEGMTIME);
216                         ast_debug(1, "cdr_odbc: Logging in GMT\n");
217                 } else {
218                         ast_clear_flag(&config, CONFIG_USEGMTIME);
219                         ast_debug(1, "cdr_odbc: Logging in local time\n");
220                 }
221
222                 if (((tmp = ast_variable_retrieve(cfg, "global", "hrtime"))) && ast_true(tmp)) {
223                         ast_set_flag(&config, CONFIG_HRTIME);
224                         ast_debug(1, "cdr_odbc: Logging billsec and duration fields as floats\n");
225                 } else {
226                         ast_clear_flag(&config, CONFIG_HRTIME);
227                         ast_debug(1, "cdr_odbc: Logging billsec and duration fields as integers\n");
228                 }
229
230                 if ((tmp = ast_variable_retrieve(cfg, "global", "table")) == NULL) {
231                         ast_log(LOG_WARNING, "cdr_odbc: table not specified.  Assuming cdr\n");
232                         tmp = "cdr";
233                 }
234                 if (table)
235                         ast_free(table);
236                 table = ast_strdup(tmp);
237                 if (table == NULL) {
238                         res = -1;
239                         break;
240                 }
241
242                 ast_verb(3, "cdr_odbc: dsn is %s\n", dsn);
243                 ast_verb(3, "cdr_odbc: table is %s\n", table);
244
245                 if (!ast_test_flag(&config, CONFIG_REGISTERED)) {
246                         res = ast_cdr_register(name, ast_module_info->description, odbc_log);
247                         if (res) {
248                                 ast_log(LOG_ERROR, "cdr_odbc: Unable to register ODBC CDR handling\n");
249                         } else {
250                                 ast_set_flag(&config, CONFIG_REGISTERED);
251                         }
252                 }
253         } while (0);
254
255         if (ast_test_flag(&config, CONFIG_REGISTERED) && (!cfg || dsn == NULL || table == NULL)) {
256                 ast_cdr_unregister(name);
257                 ast_clear_flag(&config, CONFIG_REGISTERED);
258         }
259
260         if (cfg && cfg != CONFIG_STATUS_FILEUNCHANGED && cfg != CONFIG_STATUS_FILEINVALID) {
261                 ast_config_destroy(cfg);
262         }
263         return res;
264 }
265
266 static int load_module(void)
267 {
268         return odbc_load_module(0);
269 }
270
271 static int unload_module(void)
272 {
273         ast_cdr_unregister(name);
274
275         if (dsn) {
276                 ast_verb(11, "cdr_odbc: free dsn\n");
277                 ast_free(dsn);
278         }
279         if (table) {
280                 ast_verb(11, "cdr_odbc: free table\n");
281                 ast_free(table);
282         }
283
284         return 0;
285 }
286
287 static int reload(void)
288 {
289         return odbc_load_module(1);
290 }
291
292 AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "ODBC CDR Backend",
293                 .load = load_module,
294                 .unload = unload_module,
295                 .reload = reload,
296                 .load_pri = AST_MODPRI_CDR_DRIVER,
297                );