dvander@667: /**
dvander@667:  * vim: set ts=4 :
dvander@667:  * ===============================================================
dvander@667:  * SourceMod (C)2004-2007 AlliedModders LLC.  All rights reserved.
dvander@667:  * ===============================================================
dvander@667:  *
dvander@667:  * This file is not open source and may not be copied without explicit
dvander@667:  * written permission of AlliedModders LLC.  This file may not be redistributed 
dvander@667:  * in whole or significant part.
dvander@667:  * For information, see LICENSE.txt or http://www.sourcemod.net/license.php
dvander@667:  *
dvander@667:  * Version: $Id$
dvander@667:  */
dvander@667: 
dvander@666: #include "Database.h"
dvander@666: #include "HandleSys.h"
dvander@862: #include "ShareSys.h"
dvander@666: #include "sourcemod.h"
dvander@868: #include "sm_stringutil.h"
dvander@868: #include "TextParsers.h"
dvander@868: #include "Logger.h"
dvander@868: #include "ExtensionSys.h"
dvander@868: #include <stdlib.h>
dvander@868: 
dvander@868: #define DBPARSE_LEVEL_NONE		0
dvander@868: #define DBPARSE_LEVEL_MAIN		1
dvander@868: #define DBPARSE_LEVEL_DATABASE	2
dvander@666: 
dvander@666: DBManager g_DBMan;
dvander@666: 
dvander@868: DBManager::DBManager() 
dvander@868: : m_StrTab(512), m_ParseLevel(0), m_ParseState(0), m_pDefault(NULL)
dvander@868: {
dvander@868: }
dvander@868: 
dvander@666: void DBManager::OnSourceModAllInitialized()
dvander@666: {
dvander@666: 	HandleAccess sec;
dvander@666: 
dvander@666: 	g_HandleSys.InitAccessDefaults(NULL, &sec);
dvander@666: 	sec.access[HandleAccess_Delete] |= HANDLE_RESTRICT_IDENTITY;
dvander@666: 	sec.access[HandleAccess_Clone] |= HANDLE_RESTRICT_IDENTITY;
dvander@666: 	
dvander@666: 	m_DriverType = g_HandleSys.CreateType("IDriver", this, 0, NULL, &sec, g_pCoreIdent, NULL);
dvander@666: 	m_DatabaseType = g_HandleSys.CreateType("IDatabase", this, 0, NULL, NULL, g_pCoreIdent, NULL);
dvander@862: 
dvander@862: 	g_ShareSys.AddInterface(NULL, this);
dvander@868: 
dvander@868: 	g_SourceMod.BuildPath(Path_SM, m_Filename, sizeof(m_Filename), "configs/databases.cfg");
dvander@868: }
dvander@868: 
dvander@868: void DBManager::OnSourceModLevelChange(const char *mapName)
dvander@868: {
dvander@868: 	SMCParseError err;
dvander@868: 	unsigned int line = 0;
dvander@868: 	if ((err = g_TextParser.ParseFile_SMC(m_Filename, this, &line, NULL)) != SMCParse_Okay)
dvander@868: 	{
dvander@868: 		g_Logger.LogError("[SM] Detected parse error(s) in file \"%s\"", m_Filename);
dvander@868: 		if (err != SMCParse_Custom)
dvander@868: 		{
dvander@868: 			const char *txt = g_TextParser.GetSMCErrorString(err);
dvander@868: 			g_Logger.LogError("[SM] Line %d: %s", line, txt);
dvander@868: 		}
dvander@868: 	}
dvander@858: }
dvander@858: 
dvander@858: void DBManager::OnSourceModShutdown()
dvander@858: {
dvander@858: 	g_HandleSys.RemoveType(m_DatabaseType, g_pCoreIdent);
dvander@858: 	g_HandleSys.RemoveType(m_DriverType, g_pCoreIdent);
dvander@858: }
dvander@858: 
dvander@858: unsigned int DBManager::GetInterfaceVersion()
dvander@858: {
dvander@858: 	return SMINTERFACE_DBI_VERSION;
dvander@858: }
dvander@858: 
dvander@858: const char *DBManager::GetInterfaceName()
dvander@858: {
dvander@858: 	return SMINTERFACE_DBI_NAME;
dvander@666: }
dvander@666: 
dvander@666: void DBManager::OnHandleDestroy(HandleType_t type, void *object)
dvander@666: {
dvander@666: 	if (type == m_DriverType)
dvander@666: 	{
dvander@666: 		/* Ignore */
dvander@666: 		return;
dvander@666: 	}
dvander@666: 
dvander@858: 	if (g_HandleSys.TypeCheck(type, m_DatabaseType))
dvander@666: 	{
dvander@666: 		IDatabase *pdb = (IDatabase *)object;
dvander@666: 		pdb->Close();
dvander@666: 	}
dvander@666: }
dvander@666: 
dvander@868: void DBManager::ReadSMC_ParseStart()
dvander@666: {
dvander@868: 	m_confs.clear();
dvander@868: 	m_ParseLevel = 0;
dvander@868: 	m_ParseState = DBPARSE_LEVEL_NONE;
dvander@868: 	m_StrTab.Reset();
dvander@868: 	m_DefDriver.clear();
dvander@868: }
dvander@666: 
dvander@868: ConfDbInfo s_CurInfo;
dvander@868: SMCParseResult DBManager::ReadSMC_NewSection(const char *name, bool opt_quotes)
dvander@868: {
dvander@868: 	if (m_ParseLevel)
dvander@868: 	{
dvander@868: 		m_ParseLevel++;
dvander@868: 		return SMCParse_Continue;
dvander@868: 	}
dvander@868: 
dvander@868: 	if (m_ParseState == DBPARSE_LEVEL_NONE)
dvander@666: 	{
dvander@868: 		if (strcmp(name, "Databases") == 0)
dvander@666: 		{
dvander@868: 			m_ParseState = DBPARSE_LEVEL_MAIN;
dvander@868: 		} else {
dvander@868: 			m_ParseLevel++;
dvander@868: 		}
dvander@868: 	} else if (m_ParseState == DBPARSE_LEVEL_MAIN) {
dvander@868: 		s_CurInfo = ConfDbInfo();
dvander@869: 		s_CurInfo.name = m_StrTab.AddString(name);
dvander@868: 		m_ParseState = DBPARSE_LEVEL_DATABASE;
dvander@868: 	} else if (m_ParseState == DBPARSE_LEVEL_DATABASE) {
dvander@868: 		m_ParseLevel++;
dvander@868: 	}
dvander@868: 
dvander@868: 	return SMCParse_Continue;
dvander@868: }
dvander@868: 
dvander@868: SMCParseResult DBManager::ReadSMC_KeyValue(const char *key, const char *value, bool key_quotes, bool value_quotes)
dvander@868: {
dvander@868: 	if (m_ParseLevel)
dvander@868: 	{
dvander@868: 		return SMCParse_Continue;
dvander@868: 	}
dvander@868: 
dvander@868: 	if (m_ParseState == DBPARSE_LEVEL_MAIN)
dvander@868: 	{
dvander@868: 		if (strcmp(key, "driver_default") == 0)
dvander@868: 		{
dvander@868: 			m_DefDriver.assign(value);
dvander@868: 		}
dvander@868: 	} else if (m_ParseState == DBPARSE_LEVEL_DATABASE) {
dvander@868: 		if (strcmp(key, "driver") == 0)
dvander@868: 		{
dvander@868: 			if (strcmp(value, "default") != 0)
dvander@868: 			{
dvander@868: 				s_CurInfo.driver = m_StrTab.AddString(value);
dvander@868: 			}
dvander@868: 		} else if (strcmp(key, "database") == 0) {
dvander@868: 			s_CurInfo.database = m_StrTab.AddString(value);
dvander@868: 		} else if (strcmp(key, "host") == 0) {
dvander@868: 			s_CurInfo.host = m_StrTab.AddString(value);
dvander@868: 		} else if (strcmp(key, "user") == 0) {
dvander@868: 			s_CurInfo.user = m_StrTab.AddString(value);
dvander@868: 		} else if (strcmp(key, "pass") == 0) {
dvander@868: 			s_CurInfo.pass = m_StrTab.AddString(value);
dvander@868: 		} else if (strcmp(key, "timeout") == 0) {
dvander@868: 			s_CurInfo.info.maxTimeout = atoi(value);
dvander@868: 		} else if (strcmp(key, "port") == 0) {
dvander@868: 			s_CurInfo.info.port = atoi(value);
dvander@666: 		}
dvander@666: 	}
dvander@666: 
dvander@868: 	return SMCParse_Continue;
dvander@868: }
dvander@868: 
dvander@868: #define ASSIGN_VAR(var) \
dvander@868: 	if (s_CurInfo.var == -1) { \
dvander@868: 		s_CurInfo.info.var = ""; \
dvander@868: 	} else { \
dvander@868: 		s_CurInfo.info.var = m_StrTab.GetString(s_CurInfo.var); \
dvander@868: 	}
dvander@868: 
dvander@868: SMCParseResult DBManager::ReadSMC_LeavingSection()
dvander@868: {
dvander@868: 	if (m_ParseLevel)
dvander@868: 	{
dvander@868: 		m_ParseLevel--;
dvander@868: 		return SMCParse_Continue;
dvander@868: 	}
dvander@868: 
dvander@868: 	if (m_ParseState == DBPARSE_LEVEL_DATABASE)
dvander@868: 	{
dvander@868: 		/* Set all of the info members to either a blank string
dvander@868: 		 * or the string pointer from the string table.
dvander@868: 		 */
dvander@868: 		ASSIGN_VAR(driver);
dvander@868: 		ASSIGN_VAR(database);
dvander@868: 		ASSIGN_VAR(host);
dvander@868: 		ASSIGN_VAR(user);
dvander@868: 		ASSIGN_VAR(pass);
dvander@868: 		
dvander@868: 		/* Save it.. */
dvander@868: 		m_confs.push_back(s_CurInfo);
dvander@868: 
dvander@868: 		/* Go up one level */
dvander@868: 		m_ParseState = DBPARSE_LEVEL_MAIN;
dvander@868: 	} else if (m_ParseState == DBPARSE_LEVEL_MAIN) {
dvander@868: 		m_ParseState = DBPARSE_LEVEL_NONE;
dvander@868: 		return SMCParse_Halt;
dvander@868: 	}
dvander@868: 
dvander@868: 	return SMCParse_Continue;
dvander@868: }
dvander@868: #undef ASSIGN_VAR
dvander@868: 
dvander@868: void DBManager::ReadSMC_ParseEnd(bool halted, bool failed)
dvander@868: {
dvander@868: }
dvander@868: 
dvander@868: bool DBManager::Connect(const char *name, IDBDriver **pdr, IDatabase **pdb, bool persistent, char *error, size_t maxlength)
dvander@868: {
dvander@868: 	ConfDbInfo *pInfo = GetDatabaseConf(name);
dvander@868: 
dvander@868: 	if (!pInfo)
dvander@868: 	{
dvander@868: 		if (pdr)
dvander@868: 		{
dvander@868: 			*pdr = NULL;
dvander@868: 		}
dvander@868: 		*pdb = NULL;
dvander@868: 		UTIL_Format(error, maxlength, "Configuration \"%s\" not found", name);
dvander@868: 		return false;
dvander@868: 	}
dvander@868: 
dvander@869: 	const char *dname = pInfo->info.driver;
dvander@868: 	if (!pInfo->realDriver)
dvander@868: 	{
dvander@868: 		/* Try to assign a real driver pointer */
dvander@868: 		if (pInfo->info.driver[0] == '\0')
dvander@868: 		{
dvander@868: 			if (!m_pDefault && m_DefDriver.size() > 0)
dvander@868: 			{
dvander@868: 				m_pDefault = FindOrLoadDriver(m_DefDriver.c_str());
dvander@868: 			}
dvander@869: 			dname = m_DefDriver.size() ? m_DefDriver.c_str() : "default";
dvander@868: 			pInfo->realDriver = m_pDefault;
dvander@868: 		} else {
dvander@868: 			pInfo->realDriver = FindOrLoadDriver(pInfo->info.driver);
dvander@868: 		}
dvander@868: 	}
dvander@868: 
dvander@868: 	if (pInfo->realDriver)
dvander@868: 	{
dvander@868: 		if (pdr)
dvander@868: 		{
dvander@868: 			*pdr = pInfo->realDriver;
dvander@868: 		}
dvander@868: 		*pdb = pInfo->realDriver->Connect(&pInfo->info, persistent, error, maxlength);
dvander@868: 		return (*pdb != NULL);
dvander@868: 	}
dvander@868: 
dvander@868: 	if (pdr)
dvander@868: 	{
dvander@868: 		*pdr = NULL;
dvander@868: 	}
dvander@666: 	*pdb = NULL;
dvander@666: 
dvander@869: 	UTIL_Format(error, maxlength, "Driver \"%s\" not found", dname);
dvander@868: 
dvander@666: 	return false;
dvander@666: }
dvander@858: 
dvander@858: void DBManager::AddDriver(IDBDriver *pDriver)
dvander@858: {
dvander@858: 	 m_drivers.push_back(pDriver);
dvander@858: }
dvander@858: 
dvander@858: void DBManager::RemoveDriver(IDBDriver *pDriver)
dvander@858: {
dvander@858: 	for (size_t i=0; i<m_drivers.size(); i++)
dvander@858: 	{
dvander@858: 		if (m_drivers[i] == pDriver)
dvander@858: 		{
dvander@858: 			m_drivers.erase(m_drivers.iterAt(i));
dvander@858: 			return;
dvander@858: 		}
dvander@858: 	}
dvander@868: 
dvander@868: 	/* Make sure NOTHING references this! */
dvander@868: 	List<ConfDbInfo>::iterator iter;
dvander@868: 	for (iter=m_confs.begin(); iter!=m_confs.end(); iter++)
dvander@868: 	{
dvander@868: 		ConfDbInfo &db = (*iter);
dvander@868: 		if (db.realDriver == pDriver)
dvander@868: 		{
dvander@868: 			db.realDriver = NULL;
dvander@868: 		}
dvander@868: 	}
dvander@868: }
dvander@868: 
dvander@868: IDBDriver *DBManager::GetDefaultDriver()
dvander@868: {
dvander@868: 	if (!m_pDefault && m_DefDriver.size() > 0)
dvander@868: 	{
dvander@868: 		m_pDefault = FindOrLoadDriver(m_DefDriver.c_str());
dvander@868: 	}
dvander@868: 
dvander@868: 	return m_pDefault;
dvander@858: }
dvander@858: 
dvander@858: Handle_t DBManager::CreateHandle(DBHandleType dtype, void *ptr, IdentityToken_t *pToken)
dvander@858: {
dvander@858: 	HandleType_t type = 0;
dvander@858: 
dvander@858: 	if (dtype == DBHandle_Driver)
dvander@858: 	{
dvander@858: 		type = m_DriverType;
dvander@858: 	} else if (dtype == DBHandle_Database) {
dvander@858: 		type = m_DatabaseType;
dvander@858: 	} else {
dvander@858: 		return BAD_HANDLE;
dvander@858: 	}
dvander@858: 
dvander@858: 	return g_HandleSys.CreateHandle(type, ptr, pToken, g_pCoreIdent, NULL);
dvander@858: }
dvander@858: 
dvander@858: HandleError DBManager::ReadHandle(Handle_t hndl, DBHandleType dtype, void **ptr)
dvander@858: {
dvander@858: 	HandleType_t type = 0;
dvander@858: 
dvander@858: 	if (dtype == DBHandle_Driver)
dvander@858: 	{
dvander@858: 		type = m_DriverType;
dvander@858: 	} else if (dtype == DBHandle_Database) {
dvander@858: 		type = m_DatabaseType;
dvander@858: 	} else {
dvander@858: 		return HandleError_Type;
dvander@858: 	}
dvander@858: 
dvander@858: 	HandleSecurity sec(NULL, g_pCoreIdent);
dvander@858: 
dvander@869: 	return g_HandleSys.ReadHandle(hndl, type, &sec, ptr);
dvander@858: }
dvander@858: 
dvander@858: HandleError DBManager::ReleaseHandle(Handle_t hndl, DBHandleType type, IdentityToken_t *token)
dvander@858: {
dvander@858: 	HandleSecurity sec(token, g_pCoreIdent);
dvander@858: 	return g_HandleSys.FreeHandle(hndl, &sec);
dvander@858: }
dvander@858: 
dvander@858: unsigned int DBManager::GetDriverCount()
dvander@858: {
dvander@858: 	return (unsigned int)m_drivers.size();
dvander@858: }
dvander@858: 
dvander@858: IDBDriver *DBManager::GetDriver(unsigned int index)
dvander@858: {
dvander@858: 	if (index >= m_drivers.size())
dvander@858: 	{
dvander@858: 		return NULL;
dvander@858: 	}
dvander@858: 
dvander@858: 	return m_drivers[index];
dvander@858: }
dvander@858: 
dvander@858: const DatabaseInfo *DBManager::FindDatabaseConf(const char *name)
dvander@858: {
dvander@868: 	ConfDbInfo *info = GetDatabaseConf(name);
dvander@868: 	if (!info)
dvander@868: 	{
dvander@868: 		return NULL;
dvander@868: 	}
dvander@868: 
dvander@868: 	return &info->info;
dvander@868: }
dvander@868: 
dvander@868: ConfDbInfo *DBManager::GetDatabaseConf(const char *name)
dvander@868: {
dvander@868: 	List<ConfDbInfo>::iterator iter;
dvander@868: 
dvander@868: 	for (iter=m_confs.begin(); iter!=m_confs.end(); iter++)
dvander@868: 	{
dvander@868: 		ConfDbInfo &conf = (*iter);
dvander@868: 		if (strcmp(m_StrTab.GetString(conf.name), name) == 0)
dvander@868: 		{
dvander@868: 			return &conf;
dvander@868: 		}
dvander@868: 	}
dvander@868: 
dvander@858: 	return NULL;
dvander@858: }
dvander@868: 
dvander@868: IDBDriver *DBManager::FindOrLoadDriver(const char *name)
dvander@868: {
dvander@868: 	size_t last_size = m_drivers.size();
dvander@868: 	for (size_t i=0; i<last_size; i++)
dvander@868: 	{
dvander@868: 		if (strcmp(m_drivers[i]->GetIdentifier(), name) == 0)
dvander@868: 		{
dvander@868: 			return m_drivers[i];
dvander@868: 		}
dvander@868: 	}
dvander@868: 
dvander@868: 	char filename[PLATFORM_MAX_PATH];
dvander@869: 	UTIL_Format(filename, sizeof(filename), "dbi.%s.ext", name);
dvander@868: 
dvander@868: 	IExtension *pExt = g_Extensions.LoadAutoExtension(filename);
dvander@868: 	if (!pExt || !pExt->IsLoaded() || m_drivers.size() <= last_size)
dvander@868: 	{
dvander@868: 		return NULL;
dvander@868: 	}
dvander@868: 
dvander@868: 	/* last_size is now gauranteed to be a valid index.
dvander@868: 	 * The identifier must match the name.
dvander@868: 	 */
dvander@868: 	if (strcmp(m_drivers[last_size]->GetIdentifier(), name) == 0)
dvander@868: 	{
dvander@868: 		return m_drivers[last_size];
dvander@868: 	}
dvander@868: 
dvander@868: 	return NULL;
dvander@868: }