dvander@1316: /**
dvander@1316:  * admin-sql-prefetch.sp
dvander@1316:  * Prefetches admins from an SQL database without threading.
dvander@1316:  * This file is part of SourceMod, Copyright (C) 2004-2007 AlliedModders LLC
dvander@1316:  *
dvander@1316:  * This program is free software; you can redistribute it and/or
dvander@1316:  * modify it under the terms of the GNU General Public License
dvander@1316:  * as published by the Free Software Foundation; either version 2
dvander@1316:  * of the License, or (at your option) any later version.
dvander@1316:  *
dvander@1316:  * This program is distributed in the hope that it will be useful,
dvander@1316:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
dvander@1316:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
dvander@1316:  * GNU General Public License for more details.
dvander@1316:  * 
dvander@1316:  * You should have received a copy of the GNU General Public License
dvander@1316:  * along with this program; if not, write to the Free Software
dvander@1316:  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
dvander@1316:  *
dvander@1316:  * Version: $Id$
dvander@1316:  */
dvander@1316: 
dvander@1316: /* We like semicolons */
dvander@1316: #pragma semicolon 1
dvander@1316: 
dvander@1316: #include <sourcemod>
dvander@1316: 
dvander@1316: public Plugin:myinfo = 
dvander@1316: {
dvander@1317: 	name = "SQL Admins (Prefetch)",
dvander@1316: 	author = "AlliedModders LLC",
dvander@1316: 	description = "Reads all admins from SQL",
dvander@1316: 	version = SOURCEMOD_VERSION,
dvander@1316: 	url = "http://www.sourcemod.net/"
dvander@1316: };
dvander@1316: 
dvander@1316: public OnRebuildAdminCache(AdminCachePart:part)
dvander@1316: {
dvander@1316: 	/* First try to get a database connection */
dvander@1316: 	decl String:error[255];
dvander@1324: 	new Handle:db;
dvander@1324: 	
dvander@1324: 	if (SQL_CheckConfig("admins"))
dvander@1324: 	{
dvander@1324: 		db = SQL_Connect("admins", true, error, sizeof(error));
dvander@1324: 	} else {
dvander@1324: 		db = SQL_Connect("default", true, error, sizeof(error));
dvander@1324: 	}
dvander@1324: 	
dvander@1316: 	if (db == INVALID_HANDLE)
dvander@1316: 	{
dvander@1316: 		LogError("Could not connect to database \"default\": %s", error);
dvander@1316: 		return;
dvander@1316: 	}
dvander@1316: 	
dvander@1316: 	if (part == AdminCache_Overrides)
dvander@1316: 	{
dvander@1316: 		FetchOverrides(db);
dvander@1316: 	} else if (part == AdminCache_Groups) {
dvander@1316: 		FetchGroups(db);
dvander@1316: 	} else if (part == AdminCache_Admins) {
dvander@1316: 		FetchUsers(db);
dvander@1316: 	}
dvander@1316: 	
dvander@1316: 	CloseHandle(db);
dvander@1316: }
dvander@1316: 
dvander@1316: FetchUserGroups(Handle:hQuery, AdminId:adm, id)
dvander@1316: {
dvander@1316: 	SQL_BindParamInt(hQuery, 0, id);
dvander@1316: 	if (!SQL_Execute(hQuery))
dvander@1316: 	{
dvander@1316: 		decl String:error[255];
dvander@1316: 		SQL_GetError(hQuery, error, sizeof(error));
dvander@1316: 		LogError("FetchUserGroups() statement execute failed: %s", error);
dvander@1316: 		return;
dvander@1316: 	}
dvander@1316: 	
dvander@1316: 	decl String:name[80];
dvander@1316: 	new GroupId:gid;
dvander@1316: 	while (SQL_FetchRow(hQuery))
dvander@1316: 	{
dvander@1316: 		SQL_FetchString(hQuery, 0, name, sizeof(name));
dvander@1316: 		
dvander@1316: 		if ((gid = FindAdmGroup(name)) == INVALID_GROUP_ID)
dvander@1316: 		{
dvander@1316: 			/* Group wasn't found, don't bother with it.  */
dvander@1316: 			continue;
dvander@1316: 		}
dvander@1316: 		
dvander@1316: 		#if defined _DEBUG
dvander@1316: 		PrintToServer("Found admin group (%d, %d, %d)", id, gid, adm);
dvander@1316: 		#endif
dvander@1316: 		
dvander@1316: 		AdminInheritGroup(adm, gid);
dvander@1316: 	}
dvander@1316: }
dvander@1316: 
dvander@1316: FetchUsers(Handle:db)
dvander@1316: {
dvander@1316: 	decl String:query[255], String:error[255];
dvander@1316: 	new Handle:hQuery, Handle:hGroupQuery;
dvander@1316: 	
dvander@1316: 	Format(query, sizeof(query), "SELECT id, authtype, identity, passwd, flags, name FROM sm_admins");
dvander@1316: 	if ((hQuery = SQL_Query(db, query)) == INVALID_HANDLE)
dvander@1316: 	{
dvander@1316: 		SQL_GetError(db, error, sizeof(error));
dvander@1316: 		LogError("FetchUsers() query failed: %s", query);
dvander@1316: 		LogError("Query error: %s", error);
dvander@1316: 		return;
dvander@1316: 	}
dvander@1316: 	
dvander@1316: 	/* If this fails it's non-fatal */
dvander@1316: 	Format(query,
dvander@1316: 		sizeof(query), 
dvander@1316: 		"SELECT g.name FROM sm_admins_groups ag JOIN sm_groups g ON ag.group_id = g.id WHERE ag.admin_id = ? ORDER BY inherit_order ASC");
dvander@1316: 	if ((hGroupQuery = SQL_PrepareQuery(db, query, error, sizeof(error))) == INVALID_HANDLE)
dvander@1316: 	{
dvander@1316: 		LogError("FetchUsers() group query preparation failed: %s", query);
dvander@1316: 		LogError("Query error: %s", error);
dvander@1316: 	}
dvander@1316: 	
dvander@1316: 	decl String:authtype[16];
dvander@1316: 	decl String:identity[80];
dvander@1316: 	decl String:password[80];
dvander@1316: 	decl String:flags[32];
dvander@1316: 	decl String:name[80];
dvander@1316: 	new AdminId:adm, id;
dvander@1316: 	
dvander@1316: 	while (SQL_FetchRow(hQuery))
dvander@1316: 	{
dvander@1316: 		id = SQL_FetchInt(hQuery, 0);
dvander@1316: 		SQL_FetchString(hQuery, 1, authtype, sizeof(authtype));
dvander@1316: 		SQL_FetchString(hQuery, 2, identity, sizeof(identity));
dvander@1316: 		SQL_FetchString(hQuery, 3, password, sizeof(password));
dvander@1316: 		SQL_FetchString(hQuery, 4, flags, sizeof(flags));
dvander@1316: 		SQL_FetchString(hQuery, 5, name, sizeof(name));
dvander@1316: 		
dvander@1316: 		/* Use a pre-existing admin if we can */
dvander@1316: 		if ((adm = FindAdminByIdentity(authtype, identity)) == INVALID_ADMIN_ID)
dvander@1316: 		{
dvander@1316: 			adm = CreateAdmin(name);
dvander@1316: 			if (!BindAdminIdentity(adm, authtype, identity))
dvander@1316: 			{
dvander@1316: 				LogError("Could not bind prefetched SQL admin (authtype \"%s\") (identity \"%s\")", authtype, identity);
dvander@1316: 				continue;
dvander@1316: 			}
dvander@1316: 		}
dvander@1316: 		
dvander@1316: 		#if defined _DEBUG
dvander@1316: 		PrintToServer("Found SQL admin (%d,%s,%s,%s,%s,%s):%d", id, authtype, identity, password, flags, name, adm);
dvander@1316: 		#endif
dvander@1316: 		
dvander@1316: 		/* See if this admin wants a password */
dvander@1316: 		if (password[0] != '\0')
dvander@1316: 		{
dvander@1316: 			SetAdminPassword(adm, password);
dvander@1316: 		}
dvander@1316: 		
dvander@1316: 		/* Apply each flag */
dvander@1316: 		new len = strlen(flags);
dvander@1316: 		new AdminFlag:flag;
dvander@1316: 		for (new i=0; i<len; i++)
dvander@1316: 		{
dvander@1316: 			if (!FindFlagByChar(flags[i], flag))
dvander@1316: 			{
dvander@1316: 				continue;
dvander@1316: 			}
dvander@1316: 			SetAdminFlag(adm, flag, true);
dvander@1316: 		}
dvander@1316: 		
dvander@1316: 		/* Look up groups */
dvander@1316: 		if (hGroupQuery != INVALID_HANDLE)
dvander@1316: 		{
dvander@1316: 			FetchUserGroups(hGroupQuery, adm, id);
dvander@1316: 		}
dvander@1316: 	}
dvander@1316: 	
dvander@1316: 	CloseHandle(hGroupQuery);
dvander@1316: 	CloseHandle(hQuery);
dvander@1316: }
dvander@1316: 
dvander@1316: FetchGroups(Handle:db)
dvander@1316: {
dvander@1316: 	decl String:query[255];
dvander@1316: 	new Handle:hQuery;
dvander@1316: 	
dvander@1316: 	Format(query, sizeof(query), "SELECT id, immunity, flags, name, groups_immune FROM sm_groups");
dvander@1316: 
dvander@1316: 	if ((hQuery = SQL_Query(db, query)) == INVALID_HANDLE)
dvander@1316: 	{
dvander@1316: 		decl String:error[255];
dvander@1316: 		SQL_GetError(db, error, sizeof(error));
dvander@1316: 		LogError("FetchGroups() query failed: %s", query);
dvander@1316: 		LogError("Query error: %s", error);
dvander@1316: 		return;
dvander@1316: 	}
dvander@1316: 	
dvander@1316: 	/* We cache basic group info so we can do reverse lookups */
dvander@1316: 	new Handle:groups = CreateArray(3);
dvander@1316: 	
dvander@1316: 	/* Now start fetching groups */
dvander@1316: 	decl String:immunity[16];
dvander@1316: 	decl String:flags[32];
dvander@1316: 	decl String:name[128];
dvander@1316: 	decl String:grp_immunity[256];
dvander@1316: 	while (SQL_FetchRow(hQuery))
dvander@1316: 	{
dvander@1316: 		new id = SQL_FetchInt(hQuery, 0);
dvander@1316: 		SQL_FetchString(hQuery, 1, immunity, sizeof(immunity));
dvander@1316: 		SQL_FetchString(hQuery, 2, flags, sizeof(flags));
dvander@1316: 		SQL_FetchString(hQuery, 3, name, sizeof(name));
dvander@1316: 		SQL_FetchString(hQuery, 4, grp_immunity, sizeof(grp_immunity));
dvander@1316: 		
dvander@1316: #if defined _DEBUG
dvander@1316: 		PrintToServer("Adding group (%d, %s, %s, %s, %s)", id, immunity, flags, name, grp_immunity);
dvander@1316: #endif
dvander@1316: 		
dvander@1316: 		/* Find or create the group */
dvander@1316: 		new GroupId:gid;
dvander@1316: 		if ((gid = FindAdmGroup(name)) == INVALID_GROUP_ID)
dvander@1316: 		{
dvander@1316: 			gid = CreateAdmGroup(name);
dvander@1316: 		}
dvander@1316: 		
dvander@1316: 		/* Add flags from the database to the group */
dvander@1316: 		new num_flag_chars = strlen(flags);
dvander@1316: 		for (new i=0; i<num_flag_chars; i++)
dvander@1316: 		{
dvander@1316: 			decl AdminFlag:flag;
dvander@1316: 			if (!FindFlagByChar(flags[i], flag))
dvander@1316: 			{
dvander@1316: 				continue;
dvander@1316: 			}
dvander@1316: 			SetAdmGroupAddFlag(gid, flag, true);
dvander@1316: 		}
dvander@1316: 		
dvander@1316: 		/* Set the immunity level this group has */
dvander@1316: 		if (StrEqual(immunity, "all"))
dvander@1316: 		{
dvander@1316: 			SetAdmGroupImmunity(gid, Immunity_Default, true);
dvander@1316: 			SetAdmGroupImmunity(gid, Immunity_Global, true);
dvander@1316: 		} else if (StrEqual(immunity, "default")) {
dvander@1316: 			SetAdmGroupImmunity(gid, Immunity_Default, true);
dvander@1316: 		} else if (StrEqual(immunity, "global")) {
dvander@1316: 			SetAdmGroupImmunity(gid, Immunity_Global, true);
dvander@1316: 		}
dvander@1316: 		
dvander@1316: 		new Handle:immunity_list = UTIL_ExplodeNumberString(grp_immunity);
dvander@1316: 		
dvander@1316: 		/* Now, save all this for later */
dvander@1316: 		decl data[3];
dvander@1316: 		data[0] = id;
dvander@1316: 		data[1] = _:gid;
dvander@1316: 		data[2] = _:immunity_list;
dvander@1316: 		
dvander@1316: 		PushArrayArray(groups, data);
dvander@1316: 	}
dvander@1316: 	
dvander@1316: 	CloseHandle(hQuery);
dvander@1316: 	
dvander@1316: 	/* Second pass - resolve immunity and group overrides */
dvander@1316: 	new num_groups = GetArraySize(groups);
dvander@1316: 	for (new i=0; i<num_groups; i++)
dvander@1316: 	{
dvander@1316: 		decl data[3];
dvander@1316: 		GetArrayArray(groups, i, data);
dvander@1316: 		
dvander@1316: 		new id = data[0];
dvander@1316: 		new GroupId:gid = GroupId:data[1];
dvander@1316: 		new Handle:immunity_list = Handle:data[2];
dvander@1316: 		
dvander@1316: 		/* Query to get per-group override list */
dvander@1316: 		Format(query, 
dvander@1316: 			sizeof(query),
dvander@1316: 			"SELECT type, name, access FROM sm_group_overrides WHERE group_id = %d",
dvander@1316: 			id);
dvander@1316: 		
dvander@1316: 		/* Fetch the override query results */
dvander@1316: 		if ((hQuery = SQL_Query(db, query)) == INVALID_HANDLE)
dvander@1316: 		{
dvander@1316: 			decl String:error[255];
dvander@1316: 			SQL_GetError(db, error, sizeof(error));
dvander@1316: 			LogError("FetchOverrides() query failed: %s", query);
dvander@1316: 			LogError("Query error: %s", error);
dvander@1316: 		} else {
dvander@1316: 			decl String:type[16];
dvander@1316: 			decl String:cmd[64];
dvander@1316: 			decl String:access[16];
dvander@1316: 			while (SQL_FetchRow(hQuery))
dvander@1316: 			{
dvander@1316: 				SQL_FetchString(hQuery, 0, type, sizeof(type));
dvander@1316: 				SQL_FetchString(hQuery, 1, cmd, sizeof(cmd));
dvander@1316: 				SQL_FetchString(hQuery, 2, access, sizeof(access));
dvander@1316: 				
dvander@1316: 				new OverrideType:o_type = Override_Command;
dvander@1316: 				if (StrEqual(type, "group"))
dvander@1316: 				{
dvander@1316: 					o_type = Override_CommandGroup;
dvander@1316: 				}
dvander@1316: 				
dvander@1316: 				new OverrideRule:o_rule = Command_Deny;
dvander@1316: 				if (StrEqual(access, "allow"))
dvander@1316: 				{
dvander@1316: 					o_rule = Command_Allow;
dvander@1316: 				}
dvander@1316: 				
dvander@1316: 				#if defined _DEBUG
dvander@1316: 				PrintToServer("AddAdmGroupCmdOverride(%d, %s, %d, %d)", gid, cmd, o_type, o_rule);
dvander@1316: 				#endif
dvander@1316: 				
dvander@1316: 				AddAdmGroupCmdOverride(gid, cmd, o_type, o_rule);
dvander@1316: 			}
dvander@1316: 			CloseHandle(hQuery);
dvander@1316: 		}
dvander@1316: 		
dvander@1316: 		/* Lastly, resolve the immunity list if any */
dvander@1316: 		new immunity_count;
dvander@1316: 		if (immunity_list != INVALID_HANDLE 
dvander@1316: 			&& ((immunity_count = GetArraySize(immunity_list)) > 0))
dvander@1316: 		{
dvander@1316: 			for (new j=0; j<immunity_count; j++)
dvander@1316: 			{
dvander@1316: 				new other_id = GetArrayCell(immunity_list, j);
dvander@1316: 				
dvander@1316: 				if (other_id == id)
dvander@1316: 				{
dvander@1316: 					continue;
dvander@1316: 				}
dvander@1316: 				
dvander@1316: 				/* See if we can find this other ID */
dvander@1316: 				new GroupId:other_gid;
dvander@1316: 				if ((other_gid = UTIL_FindGroupInCache(groups, other_id)) != INVALID_GROUP_ID)
dvander@1316: 				{
dvander@1316: 					#if defined _DEBUG
dvander@1316: 					PrintToServer("SetAdmGroupImmuneFrom(%d, %d)", gid, other_gid);
dvander@1316: 					#endif
dvander@1316: 					SetAdmGroupImmuneFrom(gid, other_gid);
dvander@1316: 				}
dvander@1316: 			}
dvander@1316: 		}
dvander@1319: 		
dvander@1319: 		if (immunity_list != INVALID_HANDLE)
dvander@1319: 		{
dvander@1319: 			CloseHandle(immunity_list);
dvander@1319: 		}
dvander@1316: 	}
dvander@1318: 	
dvander@1318: 	CloseHandle(groups);
dvander@1316: }
dvander@1316: 
dvander@1316: FetchOverrides(Handle:db)
dvander@1316: {
dvander@1316: 	decl String:query[255];
dvander@1316: 	new Handle:hQuery;
dvander@1316: 	
dvander@1316: 	Format(query, sizeof(query), "SELECT type, name, flags FROM sm_overrides");
dvander@1316: 
dvander@1316: 	if ((hQuery = SQL_Query(db, query)) == INVALID_HANDLE)
dvander@1316: 	{
dvander@1316: 		decl String:error[255];
dvander@1316: 		SQL_GetError(db, error, sizeof(error));
dvander@1316: 		LogError("FetchOverrides() query failed: %s", query);
dvander@1316: 		LogError("Query error: %s", error);
dvander@1316: 		return;
dvander@1316: 	}
dvander@1316: 	
dvander@1316: 	decl String:type[64];
dvander@1316: 	decl String:name[64];
dvander@1316: 	decl String:flags[32];
dvander@1316: 	new flag_bits;
dvander@1316: 	while (SQL_FetchRow(hQuery))
dvander@1316: 	{
dvander@1316: 		SQL_FetchString(hQuery, 0, type, sizeof(type));
dvander@1316: 		SQL_FetchString(hQuery, 1, name, sizeof(name));
dvander@1316: 		SQL_FetchString(hQuery, 2, flags, sizeof(flags));
dvander@1316: 		
dvander@1316: #if defined _DEBUG
dvander@1316: 		PrintToServer("Adding override (%s, %s, %s)", type, name, flags);
dvander@1316: #endif
dvander@1316: 		
dvander@1316: 		flag_bits = ReadFlagString(flags);
dvander@1316: 		if (StrEqual(type, "command"))
dvander@1316: 		{
dvander@1316: 			AddCommandOverride(name, Override_Command, flag_bits);
dvander@1316: 		} else if (StrEqual(type, "group")) {
dvander@1316: 			AddCommandOverride(name, Override_CommandGroup, flag_bits);
dvander@1316: 		}
dvander@1316: 	}
dvander@1316: 	
dvander@1316: 	CloseHandle(hQuery);
dvander@1316: }
dvander@1316: 
dvander@1316: stock GroupId:UTIL_FindGroupInCache(Handle:group_list, id)
dvander@1316: {
dvander@1316: 	new size = GetArraySize(group_list);
dvander@1316: 	
dvander@1316: 	for (new i=0; i<size; i++)
dvander@1316: 	{
dvander@1316: 		decl data[3];
dvander@1316: 		GetArrayArray(group_list, i, data);
dvander@1316: 		
dvander@1316: 		if (data[0] == id)
dvander@1316: 		{
dvander@1316: 			return GroupId:data[1];
dvander@1316: 		}
dvander@1316: 	}
dvander@1316: 	
dvander@1316: 	return INVALID_GROUP_ID;
dvander@1316: }
dvander@1316: 
dvander@1316: /**
dvander@1316:  * Breaks a comma-delimited string into a list of numbers, returning a new 
dvander@1316:  * dynamic array.
dvander@1316:  *
dvander@1316:  * @param text				The string to split.
dvander@1316:  * @return					A new array Handle which must be closed, or 
dvander@1316:  *							INVALID_HANDLE on no strings found.
dvander@1316:  */
dvander@1316: stock Handle:UTIL_ExplodeNumberString(const String:text[])
dvander@1316: {
dvander@1316: 	new Handle:array = INVALID_HANDLE;
dvander@1316: 	decl String:buffer[32];
dvander@1316: 	new reloc_idx, idx;
dvander@1316: 	
dvander@1316: 	while ((idx = SplitString(text[reloc_idx], ",", buffer, sizeof(buffer))) != -1)
dvander@1316: 	{
dvander@1316: 		reloc_idx += idx;
dvander@1316: 		if (text[reloc_idx] == '\0')
dvander@1316: 		{
dvander@1316: 			break;
dvander@1316: 		}
dvander@1316: 		if (array == INVALID_HANDLE)
dvander@1316: 		{
dvander@1316: 			array = CreateArray();
dvander@1316: 		}
dvander@1316: 		PushArrayCell(array, StringToInt(buffer));
dvander@1316: 	}
dvander@1316: 	
dvander@1316: 	if (text[reloc_idx] != '\0')
dvander@1316: 	{
dvander@1316: 		if (array == INVALID_HANDLE)
dvander@1316: 		{
dvander@1316: 			array = CreateArray();
dvander@1316: 		}
dvander@1316: 		PushArrayCell(array, StringToInt(text[reloc_idx]));
dvander@1316: 	}
dvander@1316: 	
dvander@1316: 	return array;
dvander@1316: }
dvander@1316: