diff options
author | markster <markster@f38db490-d61c-443f-a65b-d21fe96a405b> | 2004-06-11 00:12:35 +0000 |
---|---|---|
committer | markster <markster@f38db490-d61c-443f-a65b-d21fe96a405b> | 2004-06-11 00:12:35 +0000 |
commit | a5e941b56ea72044c4dadd9bdb8221cb919bea85 (patch) | |
tree | 938a7378ae06da558d7c6b35c4597db2a14162cf /res | |
parent | 894fdbaf3ff8e922f2872750383d2f3b34914ba0 (diff) |
Merge res_odbc and res_config
git-svn-id: http://svn.digium.com/svn/asterisk/trunk@3186 f38db490-d61c-443f-a65b-d21fe96a405b
Diffstat (limited to 'res')
-rwxr-xr-x | res/Makefile | 5 | ||||
-rwxr-xr-x | res/res_config_odbc.c | 253 | ||||
-rwxr-xr-x | res/res_odbc.c | 430 |
3 files changed, 688 insertions, 0 deletions
diff --git a/res/Makefile b/res/Makefile index 32e42a2e2..fa6e3cf81 100755 --- a/res/Makefile +++ b/res/Makefile @@ -12,6 +12,8 @@ # MODS=res_adsi.so res_parking.so res_crypto.so res_musiconhold.so res_indications.so res_monitor.so +MODS+=$(shell if [ -f "/usr/include/odbcinst.h" ]; then echo "res_odbc.so res_config_odbc.so"; fi) +MODS+=$(shell if [ -f "/usr/local/include/odbcinst.h" ]; then echo "res_odbc.so res_config_odbc.so"; fi) CRYPTO_LIBS=-lssl -lcrypto @@ -34,6 +36,9 @@ res_crypto.so: res_crypto.o clean: rm -f *.so *.o .depend +res_odbc.so: res_odbc.o + $(CC) $(SOLINK) -o $@ $< -lodbc + %.so : %.o $(CC) $(SOLINK) -o $@ $< diff --git a/res/res_config_odbc.c b/res/res_config_odbc.c new file mode 100755 index 000000000..ffaccb8cb --- /dev/null +++ b/res/res_config_odbc.c @@ -0,0 +1,253 @@ +/* + * Asterisk -- A telephony toolkit for Linux. + * + * Copyright (C) 1999, Mark Spencer + * + * Mark Spencer <markster@linux-support.net> + * + * res_config_odbc.c <odbc+odbc plugin for portable configuration engine > + * Copyright (C) 2004 Anthony Minessale II <anthmct@yahoo.com> + */ + + +#include <asterisk/file.h> +#include <asterisk/logger.h> +#include <asterisk/channel.h> +#include <asterisk/pbx.h> +#include <asterisk/config.h> +#include <asterisk/config_pvt.h> +#include <asterisk/module.h> +#include <stdlib.h> +#include <unistd.h> +#include <string.h> +#include <pthread.h> +#include <asterisk/res_odbc.h> + + + +static char *tdesc = "Odbc Configuration"; +static struct ast_config_reg reg1; + + +STANDARD_LOCAL_USER; + +LOCAL_USER_DECL; + + +static struct ast_config *config_odbc(char *file, struct ast_config *new_config_s,struct ast_category **new_cat_p,struct ast_variable **new_v_p,int recur +#ifdef PRESERVE_COMMENTS + ,struct ast_comment_struct *acs +#endif + + ) { + + + + + struct ast_config *config,*new; + struct ast_variable *v,*cur_v,*new_v; + struct ast_category *cur_cat,*new_cat; + char table[128]; + char connection[128]; + int configured=0, res=0; + odbc_obj *obj; + SQLINTEGER err,commented,cat_metric,var_metric; + SQLBIGINT id; + char sql[255],filename[128],category[128],var_name[128],var_val[128]; + SQLSMALLINT rowcount; + SQLHSTMT stmt; + char last[80]; + int cat_started=0; + int var_started=0; + + + if(new_config_s) { + new = new_config_s; + cat_started++; + } + else + new = ast_new_config(); + + last[0] = '\0'; + + if(!file || !strcmp(file,"res_config_odbc.conf")) + return NULL; // cant configure myself with myself ! + + config = ast_load("res_config_odbc.conf"); + + if(config) { + for(v = ast_variable_browse(config,"settings");v;v=v->next) { + if(!strcmp(v->name,"table")) { + strncpy(table,v->value,sizeof(table)); + configured++; + } + else if(!strcmp(v->name,"connection")) { + strncpy(connection,v->value,sizeof(connection)); + configured++; + } + } + ast_destroy(config); + } + + if(configured < 2) + return NULL; + + obj = fetch_odbc_obj(connection); + if(!obj) + return NULL; + + + + + + res=SQLAllocHandle(SQL_HANDLE_STMT,obj->con, &stmt); + + SQLBindCol(stmt,1,SQL_C_ULONG,&id,sizeof(id),&err); + SQLBindCol(stmt,2,SQL_C_ULONG,&cat_metric,sizeof(cat_metric),&err); + SQLBindCol(stmt,3,SQL_C_ULONG,&var_metric,sizeof(var_metric),&err); + SQLBindCol(stmt,4,SQL_C_ULONG,&commented,sizeof(commented),&err); + SQLBindCol(stmt,5,SQL_C_CHAR,&filename,sizeof(filename),&err); + SQLBindCol(stmt,6,SQL_C_CHAR,&category,sizeof(category),&err); + SQLBindCol(stmt,7,SQL_C_CHAR,&var_name,sizeof(var_name),&err); + SQLBindCol(stmt,8,SQL_C_CHAR,&var_val,sizeof(var_val),&err); + + sprintf(sql,"select * from %s where filename='%s' and commented=0 order by filename,cat_metric,var_metric,id",table,file); + res = SQLExecDirect(stmt,sql,SQL_NTS); + + if ((res != SQL_SUCCESS) && (res != SQL_SUCCESS_WITH_INFO)) { + ast_log(LOG_WARNING,"SQL select error!\n[%s]\n\n",sql); + return NULL; + } + + res=SQLNumResultCols(stmt,&rowcount); + + if ((res != SQL_SUCCESS) && (res != SQL_SUCCESS_WITH_INFO)) { + ast_log(LOG_WARNING,"SQL select error!\n[%s]\n\n",sql); + return NULL; + } + + + if(rowcount) { + + res=SQLFetch(stmt); + cat_started=0; + + + + cur_cat = *new_cat_p; + cur_v = *new_v_p; + + if(cur_cat) + cat_started=1; + if(cur_v) + var_started=1; + + + + while(res != SQL_NO_DATA) { + //ast_log(LOG_NOTICE,"found %s %s %s\n",filename,var_name,var_val); + // build the config sql table probably needs to be improved! + + if(!strcmp(var_name,"#include") && recur < MAX_INCLUDE_LEVEL) { + + config_odbc(var_val,new,&cur_cat,&cur_v,recur + 1 +#ifdef PRESERVE_COMMENTS + ,acs +#endif + ); + + } + else { + + + + if(strcmp(last,category)) { + strcpy(last,category); + new_cat = (struct ast_category *) ast_new_category(category); + + if(!cat_started) { + cat_started++; + new->root = new_cat; + cur_cat=new->root; + } + else { + cur_cat->next = new_cat; + cur_cat = cur_cat->next; + } + var_started=0; + + } + + new_v = ast_new_variable(var_name,var_val); + + + if(!var_started) { + var_started++; + cur_cat->root = new_v; + cur_v = cur_cat->root; + } + else { + cur_v->next = new_v; + cur_v = cur_v->next; + } + } + + // next row + res=SQLFetch(stmt); + } + + SQLFreeHandle(SQL_HANDLE_STMT,stmt); + } + else + ast_log(LOG_NOTICE,"found nothing\n"); + + + + + + + + + return new; + +} + + + + +int unload_module(void) +{ + ast_cust_config_deregister(®1); + ast_log(LOG_NOTICE,"res_config_odbc unloaded.\n"); + STANDARD_HANGUP_LOCALUSERS; + return 0; +} + +int load_module(void) +{ + + memset(®1,0,sizeof(struct ast_config_reg)); + strcpy(reg1.name,"odbc"); + reg1.func = config_odbc; + ast_cust_config_register(®1); + ast_log(LOG_NOTICE,"res_config_odbc loaded.\n"); + return 0; +} + +char *description(void) +{ + return tdesc; +} + +int usecount(void) +{ + /* never unload a config module */ + return 1; +} + +char *key() +{ + return ASTERISK_GPL_KEY; +} + + diff --git a/res/res_odbc.c b/res/res_odbc.c new file mode 100755 index 000000000..24c28dd3f --- /dev/null +++ b/res/res_odbc.c @@ -0,0 +1,430 @@ +/* + * Asterisk -- A telephony toolkit for Linux. + * + * Copyright (C) 1999, Mark Spencer + * + * Mark Spencer <markster@linux-support.net> + * + * res_odbc.c <ODBC resource manager> + * Copyright (C) 2004 Anthony Minessale II <anthmct@yahoo.com> + */ + + + +#include <asterisk/file.h> +#include <asterisk/logger.h> +#include <asterisk/channel.h> +#include <asterisk/config.h> +#include <asterisk/options.h> +#include <asterisk/pbx.h> +#include <asterisk/module.h> +#include <asterisk/cli.h> +#include <stdlib.h> +#include <unistd.h> +#include <string.h> +#include <pthread.h> +#include <asterisk/res_odbc.h> +#define MAX_ODBC_HANDLES 25 + +struct odbc_list { + char name[80]; + odbc_obj *obj; + int used; +}; + +static struct odbc_list ODBC_REGISTRY[MAX_ODBC_HANDLES]; + +static void odbc_destroy(void) { + int x=0; + + for(x=0;x<MAX_ODBC_HANDLES;x++) { + if(ODBC_REGISTRY[x].obj) { + destroy_obdc_obj(&ODBC_REGISTRY[x].obj); + ODBC_REGISTRY[x].obj = NULL; + } + } + +} + +static odbc_obj *odbc_read(struct odbc_list *registry,char *name) { + int x=0; + for(x=0;x<MAX_ODBC_HANDLES;x++) { + if(registry[x].used && !strcmp(registry[x].name,name)) { + return registry[x].obj; + } + } + return NULL; +} + +static int odbc_write(struct odbc_list *registry,char *name,odbc_obj *obj) { + int x=0; + for(x=0;x<MAX_ODBC_HANDLES;x++) { + if(!registry[x].used) { + strncpy(registry[x].name,name,sizeof(registry[x].name)); + registry[x].obj = obj; + registry[x].used=1; + return 1; + } + } + return 0; +} + +static void odbc_init(void) { + int x=0; + for(x=0;x<MAX_ODBC_HANDLES;x++) { + memset(&ODBC_REGISTRY[x],0,sizeof(struct odbc_list)); + } +} + + +static char *tdesc = "ODBC Resource"; +/* internal stuff */ + + +static int load_odbc_config(void) { + static char *cfg = "res_odbc.conf"; + struct ast_config *config; + struct ast_variable *v; + char *cat,*dsn,*username,*password; + int enabled; + int connect = 0; + char *env_var; + + odbc_obj *obj; + + config = ast_load(cfg); + if(config) { + for(cat = ast_category_browse(config,NULL);cat;cat=ast_category_browse(config,cat)) { + + if(!strcmp(cat,"ENV")) { + for(v = ast_variable_browse(config,cat);v;v=v->next) { + env_var = malloc( strlen(v->name) + strlen(v->value) +2); + sprintf(env_var,"%s=%s",v->name,v->value); + ast_log(LOG_NOTICE,"Adding ENV var: %s=%s\n",v->name,v->value); + putenv(env_var); + free(env_var); + } + + cat=ast_category_browse(config,cat); + } + + + dsn = username = password = NULL; + enabled = 1; + connect = 0; + for(v = ast_variable_browse(config,cat);v;v=v->next) { + if(!strcmp(v->name,"enabled")) + enabled = ast_true(v->value); + if(!strcmp(v->name,"pre-connect")) + connect = ast_true(v->value); + if(!strcmp(v->name,"dsn")) + dsn = v->value; + if(!strcmp(v->name,"username")) + username = v->value; + if(!strcmp(v->name,"password")) + password = v->value; + } + + if(enabled && dsn && username && password) { + obj = new_odbc_obj(cat,dsn,username,password); + if(obj) { + register_odbc_obj(cat,obj); + ast_log(LOG_NOTICE,"registered database handle '%s' dsn->[%s]\n",cat,obj->dsn); + if(connect) { + odbc_obj_connect(obj); + } + + } + else + ast_log(LOG_WARNING,"Addition of obj %s failed.\n",cat); + } + + + } + ast_destroy(config); + } + + + return 0; +} + + +int odbc_dump_fd(int fd,odbc_obj *obj) { + + ast_cli(fd,"\n\nName: %s\nDSN: %s\nConnected: %s\n\n",obj->name,obj->dsn,obj->up ? "yes" : "no"); + + return 0; +} + +static int odbc_usage(int fd) { + ast_cli(fd,"\n\nusage odbc <command> <arg1> .. <argn>\n\n"); + + return 0; +} + +static int odbc_command(int fd, int argc, char **argv) { + odbc_obj *obj; + int x=0; + if(!argv[1]) + return odbc_usage(fd); + + ast_cli(fd,"\n\n"); + + if(!strcmp(argv[1],"connect") || !strcmp(argv[1],"disconnect")) { + + if(!argv[2]) + return odbc_usage(fd); + + obj=odbc_read(ODBC_REGISTRY,argv[2]); + if(obj) { + if(!strcmp(argv[1],"connect")) + odbc_obj_connect(obj); + + if(!strcmp(argv[1],"disconnect")) + odbc_obj_disconnect(obj); + + } + + + + } + else if(!strcmp(argv[1],"show")) { + if(!argv[2] || (argv[2] && !strcmp(argv[2],"all"))) { + for(x=0;x<MAX_ODBC_HANDLES;x++) { + if(!ODBC_REGISTRY[x].used) + break; + if(ODBC_REGISTRY[x].obj) + odbc_dump_fd(fd,ODBC_REGISTRY[x].obj); + } + } + else { + obj=odbc_read(ODBC_REGISTRY,argv[2]); + if(obj) + odbc_dump_fd(fd,obj); + } + + + } + else + return odbc_usage(fd); + + ast_cli(fd,"\n\n"); + return 0; +} + + +static struct ast_cli_entry odbc_command_struct = { + { "odbc", NULL }, odbc_command, + "Execute ODBC Command","obdc <command> <arg1> .. <argn>", NULL }; + +/* api calls */ + +int register_odbc_obj(char *name,odbc_obj *obj) { + if(obj != NULL) + return odbc_write(ODBC_REGISTRY,name,obj); + + return 0; +} + +odbc_obj *fetch_odbc_obj(char *name) { + return (odbc_obj *) odbc_read(ODBC_REGISTRY,name); +} + + +odbc_obj *new_odbc_obj(char *name,char *dsn,char *username, char *password) { + static odbc_obj *new; + + new=malloc(sizeof(odbc_obj)); + memset(new,0,sizeof(odbc_obj)); + new->env = SQL_NULL_HANDLE; + + new->name = malloc(strlen(name)+1); + if(new->name == NULL) + return NULL; + + new->dsn = malloc(strlen(dsn)+1); + if(new->dsn == NULL) + return NULL; + + new->username = malloc(strlen(username)+1); + if(new->username == NULL) + return NULL; + + new->password = malloc(strlen(password)+1); + if(new->password == NULL) + return NULL; + + strcpy(new->name,name); + strcpy(new->dsn,dsn); + strcpy(new->username,username); + strcpy(new->password,password); + new->up=0; + ast_mutex_init(&new->lock); + return new; +} + +void destroy_obdc_obj(odbc_obj **obj) { + + odbc_obj_disconnect(*obj); + + ast_mutex_lock(&(*obj)->lock); + SQLFreeHandle(SQL_HANDLE_STMT, (*obj)->stmt); + SQLFreeHandle(SQL_HANDLE_DBC,(*obj)->con); + SQLFreeHandle(SQL_HANDLE_ENV,(*obj)->env); + + free((*obj)->name); + free((*obj)->dsn); + free((*obj)->username); + free((*obj)->password); + ast_mutex_unlock(&(*obj)->lock); + free(*obj); +} + +odbc_status odbc_obj_disconnect(odbc_obj *obj) { + ast_mutex_lock(&obj->lock); + int res; + + if(obj->up) + res = SQLDisconnect(obj->con); + else + res = -1; + + if(res == ODBC_SUCCESS) { + ast_log(LOG_WARNING, "res_odbc: disconnected %d from %s [%s]\n",res,obj->name,obj->dsn); + obj->up=0; + } + else + ast_log(LOG_WARNING, "res_odbc: %s [%s] already disconnected\n",obj->name,obj->dsn); + + ast_mutex_unlock(&obj->lock); + return ODBC_SUCCESS; +} + + +odbc_status odbc_obj_connect(odbc_obj *obj) { + int res; + long int err; + short int mlen; + char msg[200], stat[10]; + + ast_mutex_lock(&obj->lock); + + if(obj->env == SQL_NULL_HANDLE) { + + res = SQLAllocHandle(SQL_HANDLE_ENV, SQL_NULL_HANDLE, &obj->env); + + if((res != SQL_SUCCESS) && (res != SQL_SUCCESS_WITH_INFO)) { + if(option_verbose > 3) + ast_log(LOG_WARNING, "res_odbc: Error AllocHandle\n"); + ast_mutex_unlock(&obj->lock); + return ODBC_FAIL; + } + + res = SQLSetEnvAttr(obj->env, SQL_ATTR_ODBC_VERSION, (void*)SQL_OV_ODBC3, 0); + + if((res != SQL_SUCCESS) && (res != SQL_SUCCESS_WITH_INFO)) { + if(option_verbose > 3) + ast_log(LOG_WARNING, "res_odbc: Error SetEnv\n"); + SQLFreeHandle(SQL_HANDLE_ENV, obj->env); + ast_mutex_unlock(&obj->lock); + return ODBC_FAIL; + } + + res = SQLAllocHandle(SQL_HANDLE_DBC, obj->env, &obj->con); + + if((res != SQL_SUCCESS) && (res != SQL_SUCCESS_WITH_INFO)) { + + if(option_verbose > 3) + ast_log(LOG_WARNING, "res_odbc: Error AllocHDB %d\n", res); + SQLFreeHandle(SQL_HANDLE_ENV, obj->env); + + ast_mutex_unlock(&obj->lock); + return ODBC_FAIL; + } + SQLSetConnectAttr(obj->con, SQL_LOGIN_TIMEOUT, (SQLPOINTER *)10, 0); + } + + + + res = SQLConnect( + + obj->con, + (SQLCHAR*) obj->dsn,SQL_NTS, + (SQLCHAR*) obj->username,SQL_NTS, + (SQLCHAR*) obj->password,SQL_NTS + + ); + + + if((res != SQL_SUCCESS) && (res != SQL_SUCCESS_WITH_INFO)) { + SQLGetDiagRec(SQL_HANDLE_DBC, obj->con, 1, stat, &err, msg, 100, &mlen); + SQLFreeHandle(SQL_HANDLE_ENV, obj->env); + if(option_verbose > 3) + ast_log(LOG_WARNING,"res_odbc: Error SQLConnect=%d errno=%ld %s\n", res,err,msg); + return ODBC_FAIL; + } + else { + + if(option_verbose > 3) + ast_log(LOG_NOTICE, "res_odbc: Connected to %s [%s]\n",obj->name,obj->dsn); + obj->up=1; + } + + ast_mutex_unlock(&obj->lock); + return ODBC_SUCCESS; + +} + + + + + + +STANDARD_LOCAL_USER; + +LOCAL_USER_DECL; + + +int unload_module(void) { + STANDARD_HANGUP_LOCALUSERS; + + odbc_destroy(); + ast_cli_unregister(&odbc_command_struct); + ast_log(LOG_NOTICE,"res_odbc unloaded.\n"); + + return 0; +} + +int load_module(void) { + odbc_init(); + load_odbc_config(); + ast_cli_register(&odbc_command_struct); + ast_log(LOG_NOTICE,"res_odbc loaded.\n"); + return 0; +} + +char *description(void) { + return tdesc; +} + +int usecount(void) { + int res; + STANDARD_USECOUNT(res); + return res; +} + +char *key() { + return ASTERISK_GPL_KEY; +} + + + + + + + + + + + + |