Merge res_odbc and res_config

git-svn-id: http://svn.digium.com/svn/asterisk/trunk@3186 f38db490-d61c-443f-a65b-d21fe96a405b

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(&reg1);
+	ast_log(LOG_NOTICE,"res_config_odbc unloaded.\n");
+	STANDARD_HANGUP_LOCALUSERS;
+	return 0;
+}
+
+int load_module(void)
+{
+
+	memset(&reg1,0,sizeof(struct ast_config_reg));
+	strcpy(reg1.name,"odbc");
+	reg1.func = config_odbc;
+	ast_cust_config_register(&reg1);
+	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;
+}
+
+
+
+
+
+
+
+
+
+
+
+