[svn-commits] seanbright: trunk r202417 - /trunk/cdr/cdr_sqlite3_custom.c

[SVN commits to the Digium repositories]

Author: seanbright
Date: Mon Jun 22 11:09:50 2009
New Revision: 202417

URL: http://svn.asterisk.org/svn-view/asterisk?view=rev&rev=202417
Log:
Fix lock usage in cdr_sqlite3_custom to avoid potential crashes during reload.

Pointed out by Russell while working on the CEL branch.

Modified:
    trunk/cdr/cdr_sqlite3_custom.c

Modified: trunk/cdr/cdr_sqlite3_custom.c
URL: http://svn.asterisk.org/svn-view/asterisk/trunk/cdr/cdr_sqlite3_custom.c?view=diff&rev=202417&r1=202416&r2=202417
==============================================================================
--- trunk/cdr/cdr_sqlite3_custom.c (original)
+++ trunk/cdr/cdr_sqlite3_custom.c Mon Jun 22 11:09:50 2009
@@ -61,8 +61,6 @@
 static sqlite3 *db = NULL;
 
 static char table[80];
-
-/*! XXX \bug Handling of this variable has crash potential on reload */
 static char *columns;
 
 struct values {
@@ -72,7 +70,7 @@
 
 static AST_LIST_HEAD_STATIC(sql_values, values);
 
-static int free_config(void);
+static void free_config(void);
 
 static int load_column_config(const char *tmp)
 {
@@ -168,11 +166,8 @@
 		free_config();
 	}
 
-	ast_mutex_lock(&lock);
-
 	if (!(mappingvar = ast_variable_browse(cfg, "master"))) {
 		/* Nothing configured */
-		ast_mutex_unlock(&lock);
 		ast_config_destroy(cfg);
 		return -1;
 	}
@@ -187,7 +182,6 @@
 
 	/* Columns */
 	if (load_column_config(ast_variable_retrieve(cfg, "master", "columns"))) {
-		ast_mutex_unlock(&lock);
 		ast_config_destroy(cfg);
 		free_config();
 		return -1;
@@ -195,7 +189,6 @@
 
 	/* Values */
 	if (load_values_config(ast_variable_retrieve(cfg, "master", "values"))) {
-		ast_mutex_unlock(&lock);
 		ast_config_destroy(cfg);
 		free_config();
 		return -1;
@@ -203,17 +196,14 @@
 
 	ast_verb(3, "cdr_sqlite3_custom: Logging CDR records to table '%s' in 'master.db'\n", table);
 
-	ast_mutex_unlock(&lock);
 	ast_config_destroy(cfg);
 
 	return 0;
 }
 
-static int free_config(void)
+static void free_config(void)
 {
 	struct values *value;
-
-	ast_mutex_lock(&lock);
 
 	if (db) {
 		sqlite3_close(db);
@@ -228,10 +218,6 @@
 	while ((value = AST_LIST_REMOVE_HEAD(&sql_values, list))) {
 		ast_free(value);
 	}
-
-	ast_mutex_unlock(&lock);
-
-	return 0;
 }
 
 static int sqlite3_log(struct ast_cdr *cdr)
@@ -245,6 +231,8 @@
 		/* Should not have loaded, but be failsafe. */
 		return 0;
 	}
+
+	ast_mutex_lock(&lock);
 
 	{ /* Make it obvious that only sql should be used outside of this block */
 		char *escaped;
@@ -257,6 +245,7 @@
 		if (!dummy) {
 			ast_log(LOG_ERROR, "Unable to allocate channel for variable subsitution.\n");
 			ast_free(value_string);
+			ast_mutex_unlock(&lock);
 			return 0;
 		}
 		dummy->cdr = ast_cdr_dup(cdr);
@@ -272,8 +261,6 @@
 		ast_free(value_string);
 	}
 
-	ast_mutex_lock(&lock);
-
 	/* XXX This seems awful arbitrary... */
 	for (count = 0; count < 5; count++) {
 		res = sqlite3_exec(db, sql, NULL, NULL, &error);
@@ -299,9 +286,9 @@
 
 static int unload_module(void)
 {
+	ast_cdr_unregister(name);
+
 	free_config();
-
-	ast_cdr_unregister(name);
 
 	return 0;
 }
@@ -313,14 +300,7 @@
 	int res;
 	char *sql;
 
-	if (!load_config(0)) {
-		res = ast_cdr_register(name, desc, sqlite3_log);
-		if (res) {
-			ast_log(LOG_ERROR, "Unable to register custom SQLite3 CDR handling\n");
-			free_config();
-			return AST_MODULE_LOAD_DECLINE;
-		}
-	} else {
+	if (load_config(0)) {
 		return AST_MODULE_LOAD_DECLINE;
 	}
 
@@ -350,12 +330,25 @@
 		}
 	}
 
-	return 0;
+	res = ast_cdr_register(name, desc, sqlite3_log);
+	if (res) {
+		ast_log(LOG_ERROR, "Unable to register custom SQLite3 CDR handling\n");
+		free_config();
+		return AST_MODULE_LOAD_DECLINE;
+	}
+
+	return AST_MODULE_LOAD_SUCCESS;
 }
 
 static int reload(void)
 {
-	return load_config(1);
+	int res = 0;
+
+	ast_mutex_lock(&lock);
+	res = load_config(1);
+	ast_mutex_unlock(&lock);
+
+	return res;
 }
 
 AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_DEFAULT, "SQLite3 Custom CDR Module",