handle server going down (bug #3597)

git-svn-id: https://origsvn.digium.com/svn/asterisk/branches/v1-0@5063 65c4cc65-6c06-0410-ace0-fbb531ad65f3

cdr/cdr_tds.c

@@ -65,6 +65,11 @@ static char *desc = "MSSQL CDR Backend";
 static char *name = "mssql";
 static char *config = "cdr_tds.conf";
 
+static char *hostname = NULL, *dbname = NULL, *dbuser = NULL, *password = NULL, *charset = NULL, *language = NULL;
+
+static int connected = 0;
+static time_t connect_time = 0;
+
 AST_MUTEX_DEFINE_STATIC(tds_lock);
 
 static TDSSOCKET *tds;
@@ -75,11 +80,15 @@ static char *stristr(const char*, const char*);
 static char *anti_injection(const char *, int);
 static void get_date(char *, struct timeval);
 
+static int mssql_connect(void);
+static int mssql_disconnect(void);
+
 static int tds_log(struct ast_cdr *cdr)
 {
 	char sqlcmd[2048], start[80], answer[80], end[80];
 	char *accountcode, *src, *dst, *dcontext, *clid, *channel, *dstchannel, *lastapp, *lastdata, *uniqueid;
 	int res = 0;
+	int retried = 0;
 #ifdef TDS_PRE_0_62
 	TDS_INT result_type;
 #endif
@@ -164,16 +173,27 @@ static int tds_log(struct ast_cdr *cdr)
 		uniqueid
 	);
 
+	do {
+		if (!connected) {
+			if (mssql_connect())
+				ast_log(LOG_ERROR, "Failed to reconnect to SQL database.\n");
+			else
+				ast_log(LOG_WARNING, "Reconnected to SQL database.\n");
+
+			retried = 1;	/* note that we have now tried */
+		}
+
 #ifdef TDS_PRE_0_62
-	if ((tds_submit_query(tds, sqlcmd) != TDS_SUCCEED) || (tds_process_simple_query(tds, &result_type) != TDS_SUCCEED || result_type != TDS_CMD_SUCCEED))
+		if (!connected || (tds_submit_query(tds, sqlcmd) != TDS_SUCCEED) || (tds_process_simple_query(tds, &result_type) != TDS_SUCCEED || result_type != TDS_CMD_SUCCEED))
 #else
-	if ((tds_submit_query(tds, sqlcmd) != TDS_SUCCEED) || (tds_process_simple_query(tds) != TDS_SUCCEED))
+		if (!connected || (tds_submit_query(tds, sqlcmd) != TDS_SUCCEED) || (tds_process_simple_query(tds) != TDS_SUCCEED))
 #endif
 		{
-		ast_log(LOG_ERROR, "Failed to insert record into database.\n");
+			ast_log(LOG_ERROR, "Failed to insert Call Data Record into SQL database.\n");
 
-		res = -1;
+			mssql_disconnect();	/* this is ok even if we are already disconnected */
 		}
+	} while (!connected && !retried);
 
 	free(accountcode);
 	free(src);
@@ -368,110 +388,40 @@ char *description(void)
 	return desc;
 }
 
-int unload_module(void)
+static int mssql_disconnect(void)
 {
+	if (tds) {
 		tds_free_socket(tds);
-	tds_free_login(login);
+		tds = NULL;
+	}
+
+	if (context) {
 		tds_free_context(context);
+		context = NULL;
+	}
 
-	ast_cdr_unregister(name);
+	if (login) {
+		tds_free_login(login);
+		login = NULL;
+	}
+
+	connected = 0;
 
 	return 0;
 }
 
-int load_module(void)
+static int mssql_connect(void)
 {
-	TDSCONNECTINFO *connection;
+	TDSCONNECTINFO *connection = NULL;
-	int res = 0;
+	char query[128];
-	struct ast_config *cfg;
-	struct ast_variable *var;
-	char query[1024], *ptr = NULL;
-	char *hostname = NULL, *dbname = NULL, *dbuser = NULL, *password = NULL, *charset = NULL, *language = NULL;
-#ifdef TDS_PRE_0_62
-	TDS_INT result_type;
-#endif
-
-	cfg = ast_load(config);
-	if (!cfg)
-	{
-		ast_log(LOG_NOTICE, "Unable to load config for MSSQL CDR's: %s\n", config);
-		return 0;
-	}
-
-	var = ast_variable_browse(cfg, "global");
-	if (!var) /* nothing configured */
-		return 0;
-
-	ptr = ast_variable_retrieve(cfg, "global", "hostname");
-	if (ptr)
-	{
-		hostname = strdupa(ptr);
-	}
-	else
-	{
-		ast_log(LOG_ERROR,"Database server hostname not specified.\n");
-	}
-
-	ptr = ast_variable_retrieve(cfg, "global", "dbname");
-	if (ptr)
-	{
-		dbname = strdupa(ptr);
-	}
-	else
-	{
-		ast_log(LOG_ERROR,"Database dbname not specified.\n");
-	}
-
-	ptr = ast_variable_retrieve(cfg, "global", "user");
-	if (ptr)
-	{
-		dbuser = strdupa(ptr);
-	}
-	else
-	{
-		ast_log(LOG_ERROR,"Database dbuser not specified.\n");
-	}
-
-	ptr = ast_variable_retrieve(cfg, "global", "password");
-	if (ptr)
-	{
-		password = strdupa(ptr);
-	}
-	else
-	{
-		ast_log(LOG_ERROR,"Database password not specified.\n");
-	}
-
-	ptr = ast_variable_retrieve(cfg, "global", "charset");
-	if (ptr)
-	{
-		charset = strdupa(ptr);
-	}
-	else
-	{
-		charset = strdupa("iso_1");
-	}
-
-	ptr = ast_variable_retrieve(cfg, "global", "language");
-	if (ptr)
-	{
-		language = strdupa(ptr);
-	}
-	else
-	{
-		language = strdupa("us_english");
-	}
-
-	ast_destroy(cfg);
 
 	/* Connect to M$SQL Server */
 	if (!(login = tds_alloc_login()))
 	{
 		ast_log(LOG_ERROR, "tds_alloc_login() failed.\n");
-		res = -1;
+		return -1;
 	}
-	else
+	
-	{
 	tds_set_server(login, hostname);
 	tds_set_user(login, dbuser);
 	tds_set_passwd(login, password);
@@ -484,21 +434,36 @@ int load_module(void)
 	tds_set_packet(login, 512);
 	tds_set_version(login, 7, 0);
 
-		context = tds_alloc_context();
+	if (!(context = tds_alloc_context()))
-		tds = tds_alloc_socket(context, 512);
+	{
+		ast_log(LOG_ERROR, "tds_alloc_context() failed.\n");
+		goto connect_fail;
+	}
+
+	if (!(tds = tds_alloc_socket(context, 512))) {
+		ast_log(LOG_ERROR, "tds_alloc_socket() failed.\n");
+		goto connect_fail;
+	}
 
 	tds_set_parent(tds, NULL);
-		connection = tds_read_config_info(NULL, login, context->locale);
+	connection = tds_read_config_info(tds, login, context->locale);
-		if (!connection || tds_connect(tds, connection) == TDS_FAIL)
+	if (!connection)
+	{
+		ast_log(LOG_ERROR, "tds_read_config() failed.\n");
+		goto connect_fail;
+	}
+
+	if (tds_connect(tds, connection) == TDS_FAIL)
 	{
 		ast_log(LOG_ERROR, "Failed to connect to MSSQL server.\n");
-			res = -1;
+		tds = NULL;	/* freed by tds_connect() on error */
+		tds_free_connect(connection);
+		connection = NULL;
+		goto connect_fail;
 	}
 	tds_free_connect(connection);
+	connection = NULL;
 
-		if (!res)
-		{
-			memset(query, 0, sizeof(query));
 	sprintf(query, "USE %s", dbname);
 #ifdef TDS_PRE_0_62
 	if ((tds_submit_query(tds, query) != TDS_SUCCEED) || (tds_process_simple_query(tds, &result_type) != TDS_SUCCEED || result_type != TDS_CMD_SUCCEED))
@@ -507,31 +472,119 @@ int load_module(void)
 #endif
 	{
 		ast_log(LOG_ERROR, "Could not change database (%s)\n", dbname);
-				res = -1;
+		goto connect_fail;
 	}
-			else
+
+	connected = 1;
+	return 0;
+
+connect_fail:
+	mssql_disconnect();
+	return -1;
+}
+
+int unload_module(void)
 {
+	mssql_disconnect();
+
+	ast_cdr_unregister(name);
+
+	if (hostname) free(hostname);
+	if (dbname) free(dbname);
+	if (dbuser) free(dbuser);
+	if (password) free(password);
+	if (charset) free(charset);
+	if (language) free(language);
+
+	return 0;
+}
+
+int load_module(void)
+{
+	int res = 0;
+	struct ast_config *cfg;
+	struct ast_variable *var;
+	char *ptr = NULL;
+#ifdef TDS_PRE_0_62
+	TDS_INT result_type;
+#endif
+
+	cfg = ast_load(config);
+	if (!cfg) {
+		ast_log(LOG_NOTICE, "Unable to load config for MSSQL CDR's: %s\n", config);
+		return 0;
+	}
+
+	var = ast_variable_browse(cfg, "global");
+	if (!var) /* nothing configured */
+		return 0;
+
+	ptr = ast_variable_retrieve(cfg, "global", "hostname");
+	if (ptr)
+		hostname = strdup(ptr);
+	else
+		ast_log(LOG_ERROR,"Database server hostname not specified.\n");
+
+	ptr = ast_variable_retrieve(cfg, "global", "dbname");
+	if (ptr)
+		dbname = strdup(ptr);
+	else
+		ast_log(LOG_ERROR,"Database dbname not specified.\n");
+
+	ptr = ast_variable_retrieve(cfg, "global", "user");
+	if (ptr)
+		dbuser = strdup(ptr);
+	else
+		ast_log(LOG_ERROR,"Database dbuser not specified.\n");
+
+	ptr = ast_variable_retrieve(cfg, "global", "password");
+	if (ptr)
+		password = strdup(ptr);
+	else
+		ast_log(LOG_ERROR,"Database password not specified.\n");
+
+	ptr = ast_variable_retrieve(cfg, "global", "charset");
+	if (ptr)
+		charset = strdup(ptr);
+	else
+		charset = strdup("iso_1");
+
+	ptr = ast_variable_retrieve(cfg, "global", "language");
+	if (ptr)
+		language = strdup(ptr);
+	else
+		language = strdup("us_english");
+
+	ast_destroy(cfg);
+
+	mssql_connect();
+
 	/* Register MSSQL CDR handler */
 	res = ast_cdr_register(name, desc, tds_log);
 	if (res)
 	{
 		ast_log(LOG_ERROR, "Unable to register MSSQL CDR handling\n");
 	}
-			}
+
-		}
-	}
 	return res;
 }
 
 int reload(void)
 {
-	return 0;
+	unload_module();
+	return load_module();
 }
 
 int usecount(void)
 {
+	/* Simplistic use count */
+	if (ast_mutex_trylock(&tds_lock)) {
+		return 1;
+	} else {
+		ast_mutex_unlock(&tds_lock);
 		return 0;
 	}
+}
 
 char *key()
 {