dvander@2168: /**
dvander@2168:  * vim: set ts=4 :
dvander@2168:  * =============================================================================
dvander@2168:  * SourceMod SQL Admins Plugin (Prefetch)
dvander@2168:  * Prefetches admins from an SQL database without threading.
dvander@2168:  *
dvander@2168:  * SourceMod (C)2004-2008 AlliedModders LLC.  All rights reserved.
dvander@2168:  * =============================================================================
dvander@2168:  *
dvander@2168:  * This program is free software; you can redistribute it and/or modify it under
dvander@2168:  * the terms of the GNU General Public License, version 3.0, as published by the
dvander@2168:  * Free Software Foundation.
dvander@2168:  * 
dvander@2168:  * This program is distributed in the hope that it will be useful, but WITHOUT
dvander@2168:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
dvander@2168:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
dvander@2168:  * details.
dvander@2168:  *
dvander@2168:  * You should have received a copy of the GNU General Public License along with
dvander@2168:  * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@2168:  *
dvander@2168:  * As a special exception, AlliedModders LLC gives you permission to link the
dvander@2168:  * code of this program (as well as its derivative works) to "Half-Life 2," the
dvander@2168:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
dvander@2168:  * by the Valve Corporation.  You must obey the GNU General Public License in
dvander@2168:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
dvander@2168:  * this exception to all derivative works.  AlliedModders LLC defines further
dvander@2168:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
dvander@2168:  * or <http://www.sourcemod.net/license.php>.
dvander@2168:  *
dvander@2168:  * Version: $Id$
dvander@2168:  */
dvander@2168: 
dvander@2168: /* We like semicolons */
dvander@2168: #pragma semicolon 1
dvander@2168: 
dvander@2168: #include <sourcemod>
dvander@2168: 
dvander@2168: public Plugin:myinfo = 
dvander@2168: {
dvander@2168: 	name = "SQL Admins (Prefetch)",
dvander@2168: 	author = "AlliedModders LLC",
dvander@2168: 	description = "Reads all admins from SQL",
dvander@2168: 	version = SOURCEMOD_VERSION,
dvander@2168: 	url = "http://www.sourcemod.net/"
dvander@2168: };
dvander@2168: 
dvander@2168: public OnRebuildAdminCache(AdminCachePart:part)
dvander@2168: {
dvander@2168: 	/* First try to get a database connection */
dvander@2168: 	decl String:error[255];
dvander@2168: 	new Handle:db;
dvander@2168: 	
dvander@2168: 	if (SQL_CheckConfig("admins"))
dvander@2168: 	{
dvander@2168: 		db = SQL_Connect("admins", true, error, sizeof(error));
dvander@2168: 	} else {
dvander@2168: 		db = SQL_Connect("default", true, error, sizeof(error));
dvander@2168: 	}
dvander@2168: 	
dvander@2168: 	if (db == INVALID_HANDLE)
dvander@2168: 	{
dvander@2168: 		LogError("Could not connect to database \"default\": %s", error);
dvander@2168: 		return;
dvander@2168: 	}
dvander@2168: 	
dvander@2168: 	if (part == AdminCache_Overrides)
dvander@2168: 	{
dvander@2168: 		FetchOverrides(db);
dvander@2168: 	} else if (part == AdminCache_Groups) {
dvander@2168: 		FetchGroups(db);
dvander@2168: 	} else if (part == AdminCache_Admins) {
dvander@2168: 		FetchUsers(db);
dvander@2168: 	}
dvander@2168: 	
dvander@2168: 	CloseHandle(db);
dvander@2168: }
dvander@2168: 
dvander@2168: FetchUserGroups(Handle:hQuery, AdminId:adm, id)
dvander@2168: {
dvander@2168: 	SQL_BindParamInt(hQuery, 0, id);
dvander@2168: 	if (!SQL_Execute(hQuery))
dvander@2168: 	{
dvander@2168: 		decl String:error[255];
dvander@2168: 		SQL_GetError(hQuery, error, sizeof(error));
dvander@2168: 		LogError("FetchUserGroups() statement execute failed: %s", error);
dvander@2168: 		return;
dvander@2168: 	}
dvander@2168: 	
dvander@2168: 	decl String:name[80];
dvander@2168: 	new GroupId:gid;
dvander@2168: 	while (SQL_FetchRow(hQuery))
dvander@2168: 	{
dvander@2168: 		SQL_FetchString(hQuery, 0, name, sizeof(name));
dvander@2168: 		
dvander@2168: 		if ((gid = FindAdmGroup(name)) == INVALID_GROUP_ID)
dvander@2168: 		{
dvander@2168: 			/* Group wasn't found, don't bother with it.  */
dvander@2168: 			continue;
dvander@2168: 		}
dvander@2168: 		
dvander@2168: 		#if defined _DEBUG
dvander@2168: 		PrintToServer("Found admin group (%d, %d, %d)", id, gid, adm);
dvander@2168: 		#endif
dvander@2168: 		
dvander@2168: 		AdminInheritGroup(adm, gid);
dvander@2168: 	}
dvander@2168: }
dvander@2168: 
dvander@2168: FetchUsers(Handle:db)
dvander@2168: {
dvander@2168: 	decl String:query[255], String:error[255];
dvander@2168: 	new Handle:hQuery, Handle:hGroupQuery;
dvander@2168: 	
dvander@2168: 	Format(query, sizeof(query), "SELECT id, authtype, identity, password, flags, name, immunity FROM sm_admins");
dvander@2168: 	if ((hQuery = SQL_Query(db, query)) == INVALID_HANDLE)
dvander@2168: 	{
dvander@2168: 		SQL_GetError(db, error, sizeof(error));
dvander@2168: 		LogError("FetchUsers() query failed: %s", query);
dvander@2168: 		LogError("Query error: %s", error);
dvander@2168: 		return;
dvander@2168: 	}
dvander@2168: 	
dvander@2168: 	/* If this fails it's non-fatal */
dvander@2168: 	Format(query,
dvander@2168: 		sizeof(query), 
dvander@2168: 		"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@2168: 	if ((hGroupQuery = SQL_PrepareQuery(db, query, error, sizeof(error))) == INVALID_HANDLE)
dvander@2168: 	{
dvander@2168: 		LogError("FetchUsers() group query preparation failed: %s", query);
dvander@2168: 		LogError("Query error: %s", error);
dvander@2168: 	}
dvander@2168: 	
dvander@2168: 	decl String:authtype[16];
dvander@2168: 	decl String:identity[80];
dvander@2168: 	decl String:password[80];
dvander@2168: 	decl String:flags[32];
dvander@2168: 	decl String:name[80];
dvander@2168: 	new immunity;
dvander@2168: 	new AdminId:adm, id;
dvander@2168: 	
dvander@2168: 	while (SQL_FetchRow(hQuery))
dvander@2168: 	{
dvander@2168: 		id = SQL_FetchInt(hQuery, 0);
dvander@2168: 		SQL_FetchString(hQuery, 1, authtype, sizeof(authtype));
dvander@2168: 		SQL_FetchString(hQuery, 2, identity, sizeof(identity));
dvander@2168: 		SQL_FetchString(hQuery, 3, password, sizeof(password));
dvander@2168: 		SQL_FetchString(hQuery, 4, flags, sizeof(flags));
dvander@2168: 		SQL_FetchString(hQuery, 5, name, sizeof(name));
dvander@2168: 		immunity = SQL_FetchInt(hQuery, 6);
dvander@2168: 		
dvander@2168: 		/* Use a pre-existing admin if we can */
dvander@2168: 		if ((adm = FindAdminByIdentity(authtype, identity)) == INVALID_ADMIN_ID)
dvander@2168: 		{
dvander@2168: 			adm = CreateAdmin(name);
dvander@2168: 			if (!BindAdminIdentity(adm, authtype, identity))
dvander@2168: 			{
dvander@2168: 				LogError("Could not bind prefetched SQL admin (authtype \"%s\") (identity \"%s\")", authtype, identity);
dvander@2168: 				continue;
dvander@2168: 			}
dvander@2168: 		}
dvander@2168: 		
dvander@2168: 		#if defined _DEBUG
dvander@2168: 		PrintToServer("Found SQL admin (%d,%s,%s,%s,%s,%s,%d):%d", id, authtype, identity, password, flags, name, immunity, adm);
dvander@2168: 		#endif
dvander@2168: 		
dvander@2168: 		/* See if this admin wants a password */
dvander@2168: 		if (password[0] != '\0')
dvander@2168: 		{
dvander@2168: 			SetAdminPassword(adm, password);
dvander@2168: 		}
dvander@2168: 		
dvander@2168: 		/* Apply each flag */
dvander@2168: 		new len = strlen(flags);
dvander@2168: 		new AdminFlag:flag;
dvander@2168: 		for (new i=0; i<len; i++)
dvander@2168: 		{
dvander@2168: 			if (!FindFlagByChar(flags[i], flag))
dvander@2168: 			{
dvander@2168: 				continue;
dvander@2168: 			}
dvander@2168: 			SetAdminFlag(adm, flag, true);
dvander@2168: 		}
dvander@2168: 
dvander@2168: 		SetAdminImmunityLevel(adm, immunity);
dvander@2168: 		
dvander@2168: 		/* Look up groups */
dvander@2168: 		if (hGroupQuery != INVALID_HANDLE)
dvander@2168: 		{
dvander@2168: 			FetchUserGroups(hGroupQuery, adm, id);
dvander@2168: 		}
dvander@2168: 	}
dvander@2168: 	
dvander@2168: 	CloseHandle(hGroupQuery);
dvander@2168: 	CloseHandle(hQuery);
dvander@2168: }
dvander@2168: 
dvander@2168: FetchGroups(Handle:db)
dvander@2168: {
dvander@2168: 	decl String:query[255];
dvander@2168: 	new Handle:hQuery;
dvander@2168: 	
dvander@2168: 	Format(query, sizeof(query), "SELECT flags, name, immunity_level FROM sm_groups");
dvander@2168: 
dvander@2168: 	if ((hQuery = SQL_Query(db, query)) == INVALID_HANDLE)
dvander@2168: 	{
dvander@2168: 		decl String:error[255];
dvander@2168: 		SQL_GetError(db, error, sizeof(error));
dvander@2168: 		LogError("FetchGroups() query failed: %s", query);
dvander@2168: 		LogError("Query error: %s", error);
dvander@2168: 		return;
dvander@2168: 	}
dvander@2168: 	
dvander@2168: 	/* Now start fetching groups */
dvander@2168: 	decl String:flags[32];
dvander@2168: 	decl String:name[128];
dvander@2168: 	new immunity;
dvander@2168: 	while (SQL_FetchRow(hQuery))
dvander@2168: 	{
dvander@2168: 		SQL_FetchString(hQuery, 0, flags, sizeof(flags));
dvander@2168: 		SQL_FetchString(hQuery, 1, name, sizeof(name));
dvander@2168: 		immunity = SQL_FetchInt(hQuery, 2);
dvander@2168: 		
dvander@2168: #if defined _DEBUG
dvander@2168: 		PrintToServer("Adding group (%d, %s, %s)", immunity, flags, name);
dvander@2168: #endif
dvander@2168: 		
dvander@2168: 		/* Find or create the group */
dvander@2168: 		new GroupId:gid;
dvander@2168: 		if ((gid = FindAdmGroup(name)) == INVALID_GROUP_ID)
dvander@2168: 		{
dvander@2168: 			gid = CreateAdmGroup(name);
dvander@2168: 		}
dvander@2168: 		
dvander@2168: 		/* Add flags from the database to the group */
dvander@2168: 		new num_flag_chars = strlen(flags);
dvander@2168: 		for (new i=0; i<num_flag_chars; i++)
dvander@2168: 		{
dvander@2168: 			decl AdminFlag:flag;
dvander@2168: 			if (!FindFlagByChar(flags[i], flag))
dvander@2168: 			{
dvander@2168: 				continue;
dvander@2168: 			}
dvander@2168: 			SetAdmGroupAddFlag(gid, flag, true);
dvander@2168: 		}
dvander@2168: 		
dvander@2168: 		/* Set the immunity level this group has */
dvander@2168: 		SetAdmGroupImmunityLevel(gid, immunity);
dvander@2168: 	}
dvander@2168: 	
dvander@2168: 	CloseHandle(hQuery);
dvander@2168: 	
dvander@2168: 	/** 
dvander@2168: 	 * Get immunity in a big lump.  This is a nasty query but it gets the job done.
dvander@2168: 	 */
dvander@2168: 	new len = 0;
dvander@2168: 	len += Format(query[len], sizeof(query)-len, "SELECT g1.name, g2.name FROM sm_group_immunity gi");
dvander@2168: 	len += Format(query[len], sizeof(query)-len, " LEFT JOIN sm_groups g1 ON g1.id = gi.group_id ");
dvander@2168: 	len += Format(query[len], sizeof(query)-len, " LEFT JOIN sm_groups g2 ON g2.id = gi.other_id");
dvander@2168: 	
dvander@2168: 	if ((hQuery = SQL_Query(db, query)) == INVALID_HANDLE)
dvander@2168: 	{
dvander@2168: 		decl String:error[255];
dvander@2168: 		SQL_GetError(db, error, sizeof(error));
dvander@2168: 		LogError("FetchGroups() query failed: %s", query);
dvander@2168: 		LogError("Query error: %s", error);
dvander@2168: 		return;
dvander@2168: 	}
dvander@2168: 	
dvander@2168: 	while (SQL_FetchRow(hQuery))
dvander@2168: 	{
dvander@2168: 		decl String:group1[80];
dvander@2168: 		decl String:group2[80];
dvander@2168: 		new GroupId:gid1, GroupId:gid2;
dvander@2168: 		
dvander@2168: 		SQL_FetchString(hQuery, 0, group1, sizeof(group1));
dvander@2168: 		SQL_FetchString(hQuery, 1, group2, sizeof(group2));
dvander@2168: 		
dvander@2168: 		if (((gid1 = FindAdmGroup(group1)) == INVALID_GROUP_ID)
dvander@2168: 			|| (gid2 = FindAdmGroup(group2)) == INVALID_GROUP_ID)
dvander@2168: 		{
dvander@2168: 			continue;
dvander@2168: 		}
dvander@2168: 		
dvander@2168: 		SetAdmGroupImmuneFrom(gid1, gid2);
dvander@2168: #if defined _DEBUG
dvander@2168: 		PrintToServer("SetAdmGroupImmuneFrom(%d, %d)", gid1, gid2);
dvander@2168: #endif
dvander@2168: 	}
dvander@2168: 	
dvander@2168: 	CloseHandle(hQuery);
dvander@2168: 	
dvander@2168: 	/**
dvander@2168: 	 * Fetch overrides in a lump query.
dvander@2168: 	 */
dvander@2168: 	Format(query, sizeof(query), "SELECT g.name, go.type, go.name, go.access FROM sm_group_overrides go LEFT JOIN sm_groups g ON go.group_id = g.id");
dvander@2168: 	
dvander@2168: 	if ((hQuery = SQL_Query(db, query)) == INVALID_HANDLE)
dvander@2168: 	{
dvander@2168: 		decl String:error[255];
dvander@2168: 		SQL_GetError(db, error, sizeof(error));
dvander@2168: 		LogError("FetchGroups() query failed: %s", query);
dvander@2168: 		LogError("Query error: %s", error);
dvander@2168: 		return;
dvander@2168: 	}
dvander@2168: 	
dvander@2168: 	decl String:type[16];
dvander@2168: 	decl String:cmd[64];
dvander@2168: 	decl String:access[16];
dvander@2168: 	while (SQL_FetchRow(hQuery))
dvander@2168: 	{
dvander@2168: 		SQL_FetchString(hQuery, 0, name, sizeof(name));
dvander@2168: 		SQL_FetchString(hQuery, 1, type, sizeof(type));
dvander@2168: 		SQL_FetchString(hQuery, 2, cmd, sizeof(cmd));
dvander@2168: 		SQL_FetchString(hQuery, 3, access, sizeof(access));
dvander@2168: 		
dvander@2168: 		new GroupId:gid;
dvander@2168: 		if ((gid = FindAdmGroup(name)) == INVALID_GROUP_ID)
dvander@2168: 		{
dvander@2168: 			continue;
dvander@2168: 		}
dvander@2168: 				
dvander@2168: 		new OverrideType:o_type = Override_Command;
dvander@2168: 		if (StrEqual(type, "group"))
dvander@2168: 		{
dvander@2168: 			o_type = Override_CommandGroup;
dvander@2168: 		}
dvander@2168: 		
dvander@2168: 		new OverrideRule:o_rule = Command_Deny;
dvander@2168: 		if (StrEqual(access, "allow"))
dvander@2168: 		{
dvander@2168: 			o_rule = Command_Allow;
dvander@2168: 		}
dvander@2168: 				
dvander@2168: 		#if defined _DEBUG
dvander@2168: 		PrintToServer("AddAdmGroupCmdOverride(%d, %s, %d, %d)", gid, cmd, o_type, o_rule);
dvander@2168: 		#endif
dvander@2168: 		
dvander@2168: 		AddAdmGroupCmdOverride(gid, cmd, o_type, o_rule);
dvander@2168: 	}
dvander@2168: 	
dvander@2168: 	CloseHandle(hQuery);
dvander@2168: }
dvander@2168: 
dvander@2168: FetchOverrides(Handle:db)
dvander@2168: {
dvander@2168: 	decl String:query[255];
dvander@2168: 	new Handle:hQuery;
dvander@2168: 	
dvander@2168: 	Format(query, sizeof(query), "SELECT type, name, flags FROM sm_overrides");
dvander@2168: 
dvander@2168: 	if ((hQuery = SQL_Query(db, query)) == INVALID_HANDLE)
dvander@2168: 	{
dvander@2168: 		decl String:error[255];
dvander@2168: 		SQL_GetError(db, error, sizeof(error));
dvander@2168: 		LogError("FetchOverrides() query failed: %s", query);
dvander@2168: 		LogError("Query error: %s", error);
dvander@2168: 		return;
dvander@2168: 	}
dvander@2168: 	
dvander@2168: 	decl String:type[64];
dvander@2168: 	decl String:name[64];
dvander@2168: 	decl String:flags[32];
dvander@2168: 	new flag_bits;
dvander@2168: 	while (SQL_FetchRow(hQuery))
dvander@2168: 	{
dvander@2168: 		SQL_FetchString(hQuery, 0, type, sizeof(type));
dvander@2168: 		SQL_FetchString(hQuery, 1, name, sizeof(name));
dvander@2168: 		SQL_FetchString(hQuery, 2, flags, sizeof(flags));
dvander@2168: 		
dvander@2168: #if defined _DEBUG
dvander@2168: 		PrintToServer("Adding override (%s, %s, %s)", type, name, flags);
dvander@2168: #endif
dvander@2168: 		
dvander@2168: 		flag_bits = ReadFlagString(flags);
dvander@2168: 		if (StrEqual(type, "command"))
dvander@2168: 		{
dvander@2168: 			AddCommandOverride(name, Override_Command, flag_bits);
dvander@2168: 		} else if (StrEqual(type, "group")) {
dvander@2168: 			AddCommandOverride(name, Override_CommandGroup, flag_bits);
dvander@2168: 		}
dvander@2168: 	}
dvander@2168: 	
dvander@2168: 	CloseHandle(hQuery);
dvander@2168: }
dvander@2168: