2 * Asterisk -- A telephony toolkit for Linux.
4 * Copyright (C) 1999-2004, Digium, Inc.
6 * Mark Spencer <markster@digium.com>
8 * res_odbc.c <ODBC resource manager>
9 * Copyright (C) 2004 Anthony Minessale II <anthmct@yahoo.com>
12 #include <asterisk/file.h>
13 #include <asterisk/logger.h>
14 #include <asterisk/channel.h>
15 #include <asterisk/config.h>
16 #include <asterisk/options.h>
17 #include <asterisk/pbx.h>
18 #include <asterisk/module.h>
19 #include <asterisk/cli.h>
20 #include <asterisk/lock.h>
24 #include <asterisk/res_odbc.h>
25 #define MAX_ODBC_HANDLES 25
34 static struct odbc_list ODBC_REGISTRY[MAX_ODBC_HANDLES];
36 static void odbc_destroy(void)
40 for (x = 0; x < MAX_ODBC_HANDLES; x++) {
41 if (ODBC_REGISTRY[x].obj) {
42 destroy_obdc_obj(&ODBC_REGISTRY[x].obj);
43 ODBC_REGISTRY[x].obj = NULL;
48 static odbc_obj *odbc_read(struct odbc_list *registry, const char *name)
51 for (x = 0; x < MAX_ODBC_HANDLES; x++) {
52 if (registry[x].used && !strcmp(registry[x].name, name)) {
53 return registry[x].obj;
59 static int odbc_write(struct odbc_list *registry, char *name, odbc_obj * obj)
62 for (x = 0; x < MAX_ODBC_HANDLES; x++) {
63 if (!registry[x].used) {
64 strncpy(registry[x].name, name, sizeof(registry[x].name) - 1);
65 registry[x].obj = obj;
73 static void odbc_init(void)
76 for (x = 0; x < MAX_ODBC_HANDLES; x++) {
77 memset(&ODBC_REGISTRY[x], 0, sizeof(struct odbc_list));
81 static char *tdesc = "ODBC Resource";
84 static int load_odbc_config(void)
86 static char *cfg = "res_odbc.conf";
87 struct ast_config *config;
88 struct ast_variable *v;
89 char *cat, *dsn, *username, *password;
96 config = ast_load(cfg);
98 for (cat = ast_category_browse(config, NULL); cat; cat=ast_category_browse(config, cat)) {
99 if (!strcmp(cat, "ENV")) {
100 for (v = ast_variable_browse(config, cat); v; v = v->next) {
101 env_var = malloc(strlen(v->name) + strlen(v->value) + 2);
102 sprintf(env_var, "%s=%s", v->name, v->value);
103 ast_log(LOG_NOTICE, "Adding ENV var: %s=%s\n", v->name, v->value);
108 cat = ast_category_browse(config, cat);
111 dsn = username = password = NULL;
114 for (v = ast_variable_browse(config, cat); v; v = v->next) {
115 if (!strcmp(v->name, "enabled"))
116 enabled = ast_true(v->value);
117 if (!strcmp(v->name, "pre-connect"))
118 connect = ast_true(v->value);
119 if (!strcmp(v->name, "dsn"))
121 if (!strcmp(v->name, "username"))
123 if (!strcmp(v->name, "password"))
127 if (enabled && dsn) {
128 obj = new_odbc_obj(cat, dsn, username, password);
130 register_odbc_obj(cat, obj);
131 ast_log(LOG_NOTICE, "registered database handle '%s' dsn->[%s]\n", cat, obj->dsn);
133 odbc_obj_connect(obj);
136 ast_log(LOG_WARNING, "Addition of obj %s failed.\n", cat);
146 int odbc_dump_fd(int fd, odbc_obj * obj)
148 ast_cli(fd, "\n\nName: %s\nDSN: %s\nConnected: %s\n\n", obj->name, obj->dsn, obj->up ? "yes" : "no");
152 static int odbc_usage(int fd)
154 ast_cli(fd, "\n\nusage odbc <command> <arg1> .. <argn>\n\n");
158 static int odbc_command(int fd, int argc, char **argv)
163 return odbc_usage(fd);
167 if (!strcmp(argv[1], "connect") || !strcmp(argv[1], "disconnect")) {
169 return odbc_usage(fd);
171 obj = odbc_read(ODBC_REGISTRY, argv[2]);
173 if (!strcmp(argv[1], "connect"))
174 odbc_obj_connect(obj);
176 if (!strcmp(argv[1], "disconnect"))
177 odbc_obj_disconnect(obj);
180 } else if (!strcmp(argv[1], "show")) {
181 if (!argv[2] || (argv[2] && !strcmp(argv[2], "all"))) {
182 for (x = 0; x < MAX_ODBC_HANDLES; x++) {
183 if (!ODBC_REGISTRY[x].used)
185 if (ODBC_REGISTRY[x].obj)
186 odbc_dump_fd(fd, ODBC_REGISTRY[x].obj);
189 obj = odbc_read(ODBC_REGISTRY, argv[2]);
191 odbc_dump_fd(fd, obj);
195 return odbc_usage(fd);
201 static struct ast_cli_entry odbc_command_struct = {
202 {"odbc", NULL}, odbc_command,
203 "Execute ODBC Command", "obdc <command> <arg1> .. <argn>", NULL
208 int register_odbc_obj(char *name, odbc_obj * obj)
211 return odbc_write(ODBC_REGISTRY, name, obj);
215 odbc_obj *fetch_odbc_obj(const char *name)
217 return (odbc_obj *) odbc_read(ODBC_REGISTRY, name);
220 odbc_obj *new_odbc_obj(char *name, char *dsn, char *username, char *password)
222 static odbc_obj *new;
224 new = malloc(sizeof(odbc_obj));
225 memset(new, 0, sizeof(odbc_obj));
226 new->env = SQL_NULL_HANDLE;
228 new->name = malloc(strlen(name) + 1);
229 if (new->name == NULL)
232 new->dsn = malloc(strlen(dsn) + 1);
233 if (new->dsn == NULL)
237 new->username = malloc(strlen(username) + 1);
238 if (new->username == NULL)
240 strcpy(new->username, username);
244 new->password = malloc(strlen(password) + 1);
245 if (new->password == NULL)
247 strcpy(new->password, password);
250 strcpy(new->name, name);
251 strcpy(new->dsn, dsn);
253 ast_mutex_init(&new->lock);
257 void destroy_obdc_obj(odbc_obj ** obj)
259 odbc_obj_disconnect(*obj);
261 ast_mutex_lock(&(*obj)->lock);
262 SQLFreeHandle(SQL_HANDLE_STMT, (*obj)->stmt);
263 SQLFreeHandle(SQL_HANDLE_DBC, (*obj)->con);
264 SQLFreeHandle(SQL_HANDLE_ENV, (*obj)->env);
268 if ((*obj)->username)
269 free((*obj)->username);
270 if ((*obj)->password)
271 free((*obj)->password);
272 ast_mutex_unlock(&(*obj)->lock);
276 odbc_status odbc_obj_disconnect(odbc_obj * obj)
279 ast_mutex_lock(&obj->lock);
282 res = SQLDisconnect(obj->con);
287 if (res == ODBC_SUCCESS) {
288 ast_log(LOG_WARNING, "res_odbc: disconnected %d from %s [%s]\n", res, obj->name, obj->dsn);
291 ast_log(LOG_WARNING, "res_odbc: %s [%s] already disconnected\n",
292 obj->name, obj->dsn);
294 ast_mutex_unlock(&obj->lock);
298 odbc_status odbc_obj_connect(odbc_obj * obj)
303 char msg[200], stat[10];
305 ast_mutex_lock(&obj->lock);
307 if (obj->env == SQL_NULL_HANDLE) {
308 res = SQLAllocHandle(SQL_HANDLE_ENV, SQL_NULL_HANDLE, &obj->env);
310 if ((res != SQL_SUCCESS) && (res != SQL_SUCCESS_WITH_INFO)) {
311 if (option_verbose > 3)
312 ast_log(LOG_WARNING, "res_odbc: Error AllocHandle\n");
313 ast_mutex_unlock(&obj->lock);
317 res = SQLSetEnvAttr(obj->env, SQL_ATTR_ODBC_VERSION, (void *) SQL_OV_ODBC3, 0);
319 if ((res != SQL_SUCCESS) && (res != SQL_SUCCESS_WITH_INFO)) {
320 if (option_verbose > 3)
321 ast_log(LOG_WARNING, "res_odbc: Error SetEnv\n");
322 SQLFreeHandle(SQL_HANDLE_ENV, obj->env);
323 ast_mutex_unlock(&obj->lock);
327 res = SQLAllocHandle(SQL_HANDLE_DBC, obj->env, &obj->con);
329 if ((res != SQL_SUCCESS) && (res != SQL_SUCCESS_WITH_INFO)) {
331 if (option_verbose > 3)
332 ast_log(LOG_WARNING, "res_odbc: Error AllocHDB %d\n", res);
333 SQLFreeHandle(SQL_HANDLE_ENV, obj->env);
335 ast_mutex_unlock(&obj->lock);
338 SQLSetConnectAttr(obj->con, SQL_LOGIN_TIMEOUT, (SQLPOINTER *) 10, 0);
341 ast_log(LOG_NOTICE, "Calling %p/%p\n", obj->username, obj->password);
342 res = SQLConnect(obj->con,
343 (SQLCHAR *) obj->dsn, SQL_NTS,
344 (SQLCHAR *) obj->username, SQL_NTS,
345 (SQLCHAR *) obj->password, SQL_NTS);
347 if ((res != SQL_SUCCESS) && (res != SQL_SUCCESS_WITH_INFO)) {
348 SQLGetDiagRec(SQL_HANDLE_DBC, obj->con, 1, stat, &err, msg, 100, &mlen);
349 SQLFreeHandle(SQL_HANDLE_ENV, obj->env);
350 ast_log(LOG_WARNING, "res_odbc: Error SQLConnect=%d errno=%ld %s\n", res, err, msg);
354 ast_log(LOG_NOTICE, "res_odbc: Connected to %s [%s]\n", obj->name, obj->dsn);
358 ast_mutex_unlock(&obj->lock);
366 int unload_module(void)
368 STANDARD_HANGUP_LOCALUSERS;
370 ast_cli_unregister(&odbc_command_struct);
371 ast_log(LOG_NOTICE, "res_odbc unloaded.\n");
375 int load_module(void)
379 ast_cli_register(&odbc_command_struct);
380 ast_log(LOG_NOTICE, "res_odbc loaded.\n");
384 char *description(void)
392 STANDARD_USECOUNT(res);
398 return ASTERISK_GPL_KEY;