ds@1952: /**
ds@1952:  * vim: set ts=4 :
ds@1952:  * =============================================================================
ds@1952:  * SourceMod
ds@1952:  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
ds@1952:  * =============================================================================
ds@1952:  *
ds@1952:  * This program is free software; you can redistribute it and/or modify it under
ds@1952:  * the terms of the GNU General Public License, version 3.0, as published by the
ds@1952:  * Free Software Foundation.
ds@1952:  * 
ds@1952:  * This program is distributed in the hope that it will be useful, but WITHOUT
ds@1952:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
ds@1952:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
ds@1952:  * details.
ds@1952:  *
ds@1952:  * You should have received a copy of the GNU General Public License along with
ds@1952:  * this program.  If not, see <http://www.gnu.org/licenses/>.
ds@1952:  *
ds@1952:  * As a special exception, AlliedModders LLC gives you permission to link the
ds@1952:  * code of this program (as well as its derivative works) to "Half-Life 2," the
ds@1952:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
ds@1952:  * by the Valve Corporation.  You must obey the GNU General Public License in
ds@1952:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
ds@1952:  * this exception to all derivative works.  AlliedModders LLC defines further
ds@1952:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
ds@1952:  * or <http://www.sourcemod.net/license.php>.
ds@1952:  *
ds@1952:  * Version: $Id$
ds@1952:  */
ds@1952: 
ds@1952: #include <string.h>
ds@1952: #include <assert.h>
ds@1952: #include <ITextParsers.h>
ds@1952: #include "AdminCache.h"
ds@1952: #include "ShareSys.h"
ds@1952: #include "ForwardSys.h"
ds@1952: #include "PlayerManager.h"
ds@1952: #include "ConCmdManager.h"
ds@1952: #include "Logger.h"
ds@1952: #include "sourcemod.h"
ds@1952: #include "sm_stringutil.h"
ds@1952: #include "sourcemm_api.h"
dvander@1965: #include "sm_srvcmds.h"
ds@1952: 
ds@1952: #define LEVEL_STATE_NONE		0
ds@1952: #define LEVEL_STATE_LEVELS		1
ds@1952: #define LEVEL_STATE_FLAGS		2
ds@1952: 
ds@1952: AdminCache g_Admins;
dvander@1965: char g_ReverseFlags[26];
ds@1952: AdminFlag g_FlagLetters[26];
ds@1952: bool g_FlagSet[26];
ds@1952: 
ds@1952: ConVar sm_immunity_mode("sm_immunity_mode", "1", FCVAR_SPONLY, "Mode for deciding immunity protection");
ds@1952: 
ds@1952: /* Default flags */
ds@1952: AdminFlag g_DefaultFlags[26] = 
ds@1952: {
ds@1952: 	Admin_Reservation, Admin_Generic, Admin_Kick, Admin_Ban, Admin_Unban,
ds@1952: 	Admin_Slay, Admin_Changemap, Admin_Convars, Admin_Config, Admin_Chat,
ds@1952: 	Admin_Vote, Admin_Password, Admin_RCON, Admin_Cheats, Admin_Custom1, 
ds@1952: 	Admin_Custom2, Admin_Custom3, Admin_Custom4, Admin_Custom5, Admin_Custom6,
ds@1952: 	Admin_Generic, Admin_Generic, Admin_Generic, Admin_Generic, Admin_Generic,
ds@1952: 	Admin_Root
ds@1952: };
ds@1952: 
ds@1952: 
ds@1952: class FlagReader : public ITextListener_SMC
ds@1952: {
ds@1952: public:
ds@1952: 	void LoadLevels()
ds@1952: 	{
ds@1952: 		if (!Parse())
ds@1952: 		{
ds@1952: 			memcpy(g_FlagLetters, g_DefaultFlags, sizeof(AdminFlag) * 26);
ds@1952: 			for (unsigned int i=0; i<20; i++)
ds@1952: 			{
ds@1952: 				g_FlagSet[i] = true;
ds@1952: 			}
ds@1952: 			g_FlagSet[25] = true;
ds@1952: 		}
ds@1952: 	}
ds@1952: private:
ds@1952: 	bool Parse()
ds@1952: 	{
ds@1952: 		SMCStates states;
ds@1952: 		SMCError error;
ds@1952: 
ds@1952: 		m_bFileNameLogged = false;
ds@1952: 		g_SourceMod.BuildPath(Path_SM, m_File, sizeof(m_File), "configs/admin_levels.cfg");
ds@1952: 
ds@1952: 		if ((error = textparsers->ParseFile_SMC(m_File, this, &states))
ds@1952: 			!= SMCError_Okay)
ds@1952: 		{
ds@1952: 			const char *err_string = textparsers->GetSMCErrorString(error);
ds@1952: 			if (!err_string)
ds@1952: 			{
ds@1952: 				err_string = "Unknown error";
ds@1952: 			}
ds@1952: 			ParseError(NULL, "Error %d (%s)", error, err_string);
ds@1952: 			return false;
ds@1952: 		}
ds@1952: 
ds@1952: 		return true;
ds@1952: 	}
ds@1952: 	void ReadSMC_ParseStart()
ds@1952: 	{
ds@1952: 		m_LevelState = LEVEL_STATE_NONE;
ds@1952: 		m_IgnoreLevel = 0;
ds@1952: 		memset(g_FlagSet, 0, sizeof(g_FlagSet));
ds@1952: 	}
ds@1952: 	SMCResult ReadSMC_NewSection(const SMCStates *states, const char *name)
ds@1952: 	{
ds@1952: 		if (m_IgnoreLevel)
ds@1952: 		{
ds@1952: 			m_IgnoreLevel++;
ds@1952: 			return SMCResult_Continue;
ds@1952: 		}
ds@1952: 
ds@1952: 		if (m_LevelState == LEVEL_STATE_NONE)
ds@1952: 		{
ds@1952: 			if (strcmp(name, "Levels") == 0)
ds@1952: 			{
ds@1952: 				m_LevelState = LEVEL_STATE_LEVELS;
ds@1952: 			} 
ds@1952: 			else 
ds@1952: 			{
ds@1952: 				m_IgnoreLevel++;
ds@1952: 			}
ds@1952: 		} else if (m_LevelState == LEVEL_STATE_LEVELS) {
ds@1952: 			if (strcmp(name, "Flags") == 0)
ds@1952: 			{
ds@1952: 				m_LevelState = LEVEL_STATE_FLAGS;
ds@1952: 			} 
ds@1952: 			else 
ds@1952: 			{
ds@1952: 				m_IgnoreLevel++;
ds@1952: 			}
ds@1952: 		} 
ds@1952: 		else 
ds@1952: 		{
ds@1952: 			m_IgnoreLevel++;
ds@1952: 		}
ds@1952: 
ds@1952: 		return SMCResult_Continue;
ds@1952: 	}
ds@1952: 	SMCResult ReadSMC_KeyValue(const SMCStates *states, const char *key, const char *value)
ds@1952: 	{
ds@1952: 		if (m_LevelState != LEVEL_STATE_FLAGS || m_IgnoreLevel)
ds@1952: 		{
ds@1952: 			return SMCResult_Continue;
ds@1952: 		}
ds@1952: 
ds@1952: 		unsigned char c = (unsigned)value[0];
ds@1952: 
ds@1952: 		if (c < (unsigned)'a' || c > (unsigned)'z')
ds@1952: 		{
ds@1952: 			ParseError(states, "Flag \"%c\" is not a lower-case ASCII letter", c);
ds@1952: 			return SMCResult_Continue;
ds@1952: 		}
ds@1952: 
ds@1952: 		c -= (unsigned)'a';
ds@1952: 
ds@1952: 		if (!g_Admins.FindFlag(key, &g_FlagLetters[c]))
ds@1952: 		{
ds@1952: 			ParseError(states, "Unrecognized admin level \"%s\"", key);
ds@1952: 			return SMCResult_Continue;
ds@1952: 		}
ds@1952: 
ds@1952: 		g_FlagSet[c] = true;
ds@1952: 
ds@1952: 		return SMCResult_Continue;
ds@1952: 	}
ds@1952: 	SMCResult ReadSMC_LeavingSection(const SMCStates *states)
ds@1952: 	{
ds@1952: 		if (m_IgnoreLevel)
ds@1952: 		{
ds@1952: 			m_IgnoreLevel--;
ds@1952: 			return SMCResult_Continue;
ds@1952: 		}
ds@1952: 
ds@1952: 		if (m_LevelState == LEVEL_STATE_FLAGS)
ds@1952: 		{
ds@1952: 			m_LevelState = LEVEL_STATE_LEVELS;
ds@1952: 			return SMCResult_Halt;
ds@1952: 		} 
ds@1952: 		else if (m_LevelState == LEVEL_STATE_LEVELS) 
ds@1952: 		{
ds@1952: 			m_LevelState = LEVEL_STATE_NONE;
ds@1952: 		}
ds@1952: 
ds@1952: 		return SMCResult_Continue;
ds@1952: 	}
ds@1952: 	void ParseError(const SMCStates *states, const char *message, ...)
ds@1952: 	{
ds@1952: 		va_list ap;
ds@1952: 		char buffer[256];
ds@1952: 
ds@1952: 		va_start(ap, message);
ds@1952: 		UTIL_FormatArgs(buffer, sizeof(buffer), message, ap);
ds@1952: 		va_end(ap);
ds@1952: 
ds@1952: 		if (!m_bFileNameLogged)
ds@1952: 		{
ds@1952: 			g_Logger.LogError("[SM] Parse error(s) detected in file \"%s\":", m_File);
ds@1952: 			m_bFileNameLogged = true;
ds@1952: 		}
ds@1952: 
ds@1952: 		g_Logger.LogError("[SM] (Line %d): %s", states ? states->line : 0, buffer);
ds@1952: 	}
ds@1952: private:
ds@1952: 	bool m_bFileNameLogged;
ds@1952: 	char m_File[PLATFORM_MAX_PATH];
ds@1952: 	int m_LevelState;
ds@1952: 	int m_IgnoreLevel;
ds@1952: } s_FlagReader;
ds@1952: 
ds@1952: AdminCache::AdminCache()
ds@1952: {
ds@1952: 	m_pCmdOverrides = sm_trie_create();
ds@1952: 	m_pCmdGrpOverrides = sm_trie_create();
ds@1952: 	m_pStrings = new BaseStringTable(1024);
ds@1952: 	m_pMemory = m_pStrings->GetMemTable();
ds@1952: 	m_FreeGroupList = m_FirstGroup = m_LastGroup = INVALID_GROUP_ID;
ds@1952: 	m_FreeUserList = m_FirstUser = m_LastUser = INVALID_ADMIN_ID;
ds@1952: 	m_pGroups = sm_trie_create();
ds@1952: 	m_pCacheFwd = NULL;
ds@1952: 	m_FirstGroup = -1;
ds@1952: 	m_pAuthTables = sm_trie_create();
ds@1952: 	m_InvalidatingAdmins = false;
ds@1952: 	m_destroying = false;
ds@1952: 	m_pLevelNames = sm_trie_create();
ds@1952: }
ds@1952: 
ds@1952: AdminCache::~AdminCache()
ds@1952: {
ds@1952: 	m_destroying = true;
ds@1952: 	DumpAdminCache(AdminCache_Overrides, false);
ds@1952: 	DumpAdminCache(AdminCache_Groups, false);
ds@1952: 
ds@1952: 	sm_trie_destroy(m_pCmdGrpOverrides);
ds@1952: 	sm_trie_destroy(m_pCmdOverrides);
ds@1952: 
ds@1952: 	if (m_pGroups)
ds@1952: 	{
ds@1952: 		sm_trie_destroy(m_pGroups);
ds@1952: 	}
ds@1952: 
ds@1952: 	List<AuthMethod>::iterator iter;
ds@1952: 	for (iter=m_AuthMethods.begin();
ds@1952: 		 iter!=m_AuthMethods.end();
ds@1952: 		 iter++)
ds@1952: 	{
ds@1952: 		sm_trie_destroy((*iter).table);
ds@1952: 	}
ds@1952: 
ds@1952: 	sm_trie_destroy(m_pAuthTables);
ds@1952: 
ds@1952: 	delete m_pStrings;
ds@1952: 
ds@1952: 	sm_trie_destroy(m_pLevelNames);
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::OnSourceModStartup(bool late)
ds@1952: {
ds@1952: 	RegisterAuthIdentType("steam");
ds@1952: 	RegisterAuthIdentType("name");
ds@1952: 	RegisterAuthIdentType("ip");
ds@1952: 
ds@1952: 	NameFlag("reservation", Admin_Reservation);
ds@1952: 	NameFlag("kick", Admin_Kick);
ds@1952: 	NameFlag("generic", Admin_Generic);
ds@1952: 	NameFlag("ban", Admin_Ban);
ds@1952: 	NameFlag("unban", Admin_Unban);
ds@1952: 	NameFlag("slay", Admin_Slay);
ds@1952: 	NameFlag("changemap", Admin_Changemap);
ds@1952: 	NameFlag("cvars", Admin_Convars);
ds@1952: 	NameFlag("config", Admin_Config);
ds@1952: 	NameFlag("chat", Admin_Chat);
ds@1952: 	NameFlag("vote", Admin_Vote);
ds@1952: 	NameFlag("password", Admin_Password);
ds@1952: 	NameFlag("rcon", Admin_RCON);
ds@1952: 	NameFlag("cheats", Admin_Cheats);
ds@1952: 	NameFlag("root", Admin_Root);
ds@1952: 	NameFlag("custom1", Admin_Custom1);
ds@1952: 	NameFlag("custom2", Admin_Custom2);
ds@1952: 	NameFlag("custom3", Admin_Custom3);
ds@1952: 	NameFlag("custom4", Admin_Custom4);
ds@1952: 	NameFlag("custom5", Admin_Custom5);
ds@1952: 	NameFlag("custom6", Admin_Custom6);
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::OnSourceModAllInitialized()
ds@1952: {
ds@1952: 	m_pCacheFwd = g_Forwards.CreateForward("OnRebuildAdminCache", ET_Ignore, 1, NULL, Param_Cell);
ds@1952: 	g_ShareSys.AddInterface(NULL, this);
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::OnSourceModLevelChange(const char *mapName)
ds@1952: {
dvander@1965: 	int i;
dvander@1965: 	AdminFlag flag;
dvander@1965: 
ds@1952: 	/* For now, we only read these once per level. */
ds@1952: 	s_FlagReader.LoadLevels();
dvander@1965: 
dvander@1965: 	for (i = 0; i < 26; i++)
dvander@1965: 	{
dvander@1965: 		if (FindFlag('a' + i, &flag))
dvander@1965: 		{
dvander@1965: 			g_ReverseFlags[flag] = 'a' + i;
dvander@1965: 		}
dvander@1965: 		else
dvander@1965: 		{
dvander@1965: 			g_ReverseFlags[flag] = '?';
dvander@1965: 		}
dvander@1965: 	}
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::OnSourceModShutdown()
ds@1952: {
ds@1952: 	g_Forwards.ReleaseForward(m_pCacheFwd);
ds@1952: 	m_pCacheFwd = NULL;
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::OnSourceModPluginsLoaded()
ds@1952: {
ds@1952: 	DumpAdminCache(AdminCache_Overrides, true);
ds@1952: 	DumpAdminCache(AdminCache_Groups, true);
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::NameFlag(const char *str, AdminFlag flag)
ds@1952: {
ds@1952: 	sm_trie_insert(m_pLevelNames, str, (void *)flag);
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::FindFlag(const char *str, AdminFlag *pFlag)
ds@1952: {
ds@1952: 	void *obj;
ds@1952: 	if (!sm_trie_retrieve(m_pLevelNames, str, &obj))
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (pFlag)
ds@1952: 	{
ds@1952: 		*pFlag = (AdminFlag)(int)obj;
ds@1952: 	}
ds@1952: 
ds@1952: 	return true;
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::AddCommandOverride(const char *cmd, OverrideType type, FlagBits flags)
ds@1952: {
ds@1952: 	Trie *pTrie = NULL;
ds@1952: 	if (type == Override_Command)
ds@1952: 	{
ds@1952: 		pTrie = m_pCmdOverrides;
ds@1952: 	} else if (type == Override_CommandGroup) {
ds@1952: 		pTrie = m_pCmdGrpOverrides;
ds@1952: 	} else {
ds@1952: 		return;
ds@1952: 	}
ds@1952: 
ds@1952: 	sm_trie_insert(pTrie, cmd, (void *)(unsigned int)flags);
ds@1952: 
ds@1952: 	g_ConCmds.UpdateAdminCmdFlags(cmd, type, flags, false);
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::GetCommandOverride(const char *cmd, OverrideType type, FlagBits *pFlags)
ds@1952: {
ds@1952: 	Trie *pTrie = NULL;
ds@1952: 
ds@1952: 	if (type == Override_Command)
ds@1952: 	{
ds@1952: 		pTrie = m_pCmdOverrides;
ds@1952: 	} else if (type == Override_CommandGroup) {
ds@1952: 		pTrie = m_pCmdGrpOverrides;
ds@1952: 	} else {
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	void *object;
ds@1952: 	if (sm_trie_retrieve(pTrie, cmd, &object))
ds@1952: 	{
ds@1952: 		if (pFlags)
ds@1952: 		{
ds@1952: 			*pFlags = (FlagBits)object;
ds@1952: 		}
ds@1952: 		return true;
ds@1952: 	}
ds@1952: 
ds@1952: 	return false;
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::UnsetCommandOverride(const char *cmd, OverrideType type)
ds@1952: {
ds@1952: 	if (type == Override_Command)
ds@1952: 	{
ds@1952: 		return _UnsetCommandOverride(cmd);
ds@1952: 	} else if (type == Override_CommandGroup) {
ds@1952: 		return _UnsetCommandGroupOverride(cmd);
ds@1952: 	}
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::_UnsetCommandGroupOverride(const char *group)
ds@1952: {
ds@1952: 	if (!m_pCmdGrpOverrides)
ds@1952: 	{
ds@1952: 		return;
ds@1952: 	}
ds@1952: 
ds@1952: 	sm_trie_delete(m_pCmdGrpOverrides, group);
ds@1952: 
ds@1952: 	g_ConCmds.UpdateAdminCmdFlags(group, Override_CommandGroup, 0, true);
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::_UnsetCommandOverride(const char *cmd)
ds@1952: {
ds@1952: 	if (!m_pCmdOverrides)
ds@1952: 	{
ds@1952: 		return;
ds@1952: 	}
ds@1952: 
ds@1952: 	sm_trie_delete(m_pCmdOverrides, cmd);
ds@1952: 
ds@1952: 	g_ConCmds.UpdateAdminCmdFlags(cmd, Override_Command, 0, true);
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::DumpCommandOverrideCache(OverrideType type)
ds@1952: {
ds@1952: 	if (type == Override_Command && m_pCmdOverrides)
ds@1952: 	{
ds@1952: 		sm_trie_clear(m_pCmdOverrides);
ds@1952: 	} else if (type == Override_CommandGroup && m_pCmdGrpOverrides) {
ds@1952: 		sm_trie_clear(m_pCmdGrpOverrides);
ds@1952: 	}
ds@1952: }
ds@1952: 
ds@1952: AdminId AdminCache::CreateAdmin(const char *name)
ds@1952: {
ds@1952: 	AdminId id;
ds@1952: 	AdminUser *pUser;
ds@1952: 
ds@1952: 	if (m_FreeUserList != INVALID_ADMIN_ID)
ds@1952: 	{
ds@1952: 		pUser = (AdminUser *)m_pMemory->GetAddress(m_FreeUserList);
ds@1952: 		assert(pUser->magic == USR_MAGIC_UNSET);
ds@1952: 		id = m_FreeUserList;
ds@1952: 		m_FreeUserList = pUser->next_user;
dvander@1965: 	}
dvander@1965: 	else
dvander@1965: 	{
ds@1952: 		id = m_pMemory->CreateMem(sizeof(AdminUser), (void **)&pUser);
ds@1952: 		pUser->grp_size = 0;
ds@1952: 		pUser->grp_table = -1;
ds@1952: 	}
ds@1952: 
ds@1952: 	pUser->flags = 0;
ds@1952: 	pUser->eflags = 0;
ds@1952: 	pUser->grp_count = 0;
ds@1952: 	pUser->password = -1;
ds@1952: 	pUser->magic = USR_MAGIC_SET;
ds@1952: 	pUser->auth.identidx = -1;
ds@1952: 	pUser->auth.index = 0;
ds@1952: 	pUser->immunity_level = 0;
ds@1952: 	pUser->serialchange = 1;
ds@1952: 
ds@1952: 	if (m_FirstUser == INVALID_ADMIN_ID)
ds@1952: 	{
ds@1952: 		m_FirstUser = id;
ds@1952: 		m_LastUser = id;
dvander@1965: 	}
dvander@1965: 	else
dvander@1965: 	{
ds@1952: 		AdminUser *pPrev = (AdminUser *)m_pMemory->GetAddress(m_LastUser);
ds@1952: 		pPrev->next_user = id;
ds@1952: 		pUser->prev_user = m_LastUser;
ds@1952: 		m_LastUser = id;
ds@1952: 	}
ds@1952: 
dvander@1965: 	/* Since we always append to the tail, we should invalidate their next */
dvander@1965: 	pUser->next_user = -1;
dvander@1965: 
ds@1952: 	if (name && name[0] != '\0')
ds@1952: 	{
ds@1952: 		int nameidx = m_pStrings->AddString(name);
ds@1952: 		pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 		pUser->nameidx = nameidx;
ds@1952: 	}
dvander@1965: 	else
dvander@1965: 	{
dvander@1965: 		pUser->nameidx = -1;
dvander@1965: 	}
ds@1952: 
ds@1952: 	return id;
ds@1952: }
ds@1952: 
ds@1952: GroupId AdminCache::AddGroup(const char *group_name)
ds@1952: {
ds@1952: 	if (sm_trie_retrieve(m_pGroups, group_name, NULL))
ds@1952: 	{
ds@1952: 		return INVALID_GROUP_ID;
ds@1952: 	}
ds@1952: 
ds@1952: 	GroupId id;
ds@1952: 	AdminGroup *pGroup;
ds@1952: 	if (m_FreeGroupList != INVALID_GROUP_ID)
ds@1952: 	{
ds@1952: 		pGroup = (AdminGroup *)m_pMemory->GetAddress(m_FreeGroupList);
ds@1952: 		assert(pGroup->magic == GRP_MAGIC_UNSET);
ds@1952: 		id = m_FreeGroupList;
ds@1952: 		m_FreeGroupList = pGroup->next_grp;
ds@1952: 	} else {
ds@1952: 		id = m_pMemory->CreateMem(sizeof(AdminGroup), (void **)&pGroup);
ds@1952: 	}
ds@1952: 
ds@1952: 	pGroup->immunity_level = 0;
ds@1952: 	pGroup->immune_table = -1;
ds@1952: 	pGroup->magic = GRP_MAGIC_SET;
ds@1952: 	pGroup->next_grp = INVALID_GROUP_ID;
ds@1952: 	pGroup->pCmdGrpTable = NULL;
ds@1952: 	pGroup->pCmdTable = NULL;
ds@1952: 	pGroup->addflags = 0;
ds@1952: 
ds@1952: 	if (m_FirstGroup == INVALID_GROUP_ID)
ds@1952: 	{
ds@1952: 		m_FirstGroup = id;
ds@1952: 		m_LastGroup = id;
ds@1952: 		pGroup->prev_grp = INVALID_GROUP_ID;
ds@1952: 	} else {
ds@1952: 		AdminGroup *pPrev = (AdminGroup *)m_pMemory->GetAddress(m_LastGroup);
ds@1952: 		assert(pPrev->magic == GRP_MAGIC_SET);
ds@1952: 		pPrev->next_grp = id;
ds@1952: 		pGroup->prev_grp = m_LastGroup;
ds@1952: 		m_LastGroup = id;
ds@1952: 	}
ds@1952: 
ds@1952: 	int nameidx = m_pStrings->AddString(group_name);
ds@1952: 	pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
ds@1952: 	pGroup->nameidx = nameidx;
ds@1952: 
ds@1952: 	sm_trie_insert(m_pGroups, group_name, (void *)id);
ds@1952: 
ds@1952: 	return id;
ds@1952: }
ds@1952: 
ds@1952: GroupId AdminCache::FindGroupByName(const char *group_name)
ds@1952: {
ds@1952: 	void *object;
ds@1952: 
ds@1952: 	if (!sm_trie_retrieve(m_pGroups, group_name, &object))
ds@1952: 	{
ds@1952: 		return INVALID_GROUP_ID;
ds@1952: 	}
ds@1952: 
ds@1952: 	GroupId id = (GroupId)object;
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
ds@1952: 
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return INVALID_GROUP_ID;
ds@1952: 	}
ds@1952: 
ds@1952: 	return id;
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::SetGroupAddFlag(GroupId id, AdminFlag flag, bool enabled)
ds@1952: {
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (flag < Admin_Reservation || flag >= AdminFlags_TOTAL)
ds@1952: 	{
ds@1952: 		return;
ds@1952: 	}
ds@1952: 
ds@1952: 	FlagBits bits = (1<<(FlagBits)flag);
ds@1952: 
ds@1952: 	if (enabled)
ds@1952: 	{
ds@1952: 		pGroup->addflags |= bits;
ds@1952: 	} else {
ds@1952: 		pGroup->addflags &= ~bits;
ds@1952: 	}
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::GetGroupAddFlag(GroupId id, AdminFlag flag)
ds@1952: {
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (flag < Admin_Reservation || flag >= AdminFlags_TOTAL)
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	FlagBits bit = 1<<(FlagBits)flag;
ds@1952: 	return ((pGroup->addflags & bit) == bit);
ds@1952: }
ds@1952: 
ds@1952: FlagBits AdminCache::GetGroupAddFlags(GroupId id)
ds@1952: {
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return 0;
ds@1952: 	}
ds@1952: 
ds@1952: 	return pGroup->addflags;
ds@1952: }
ds@1952: 
ds@1952: const char *AdminCache::GetGroupName(GroupId gid)
ds@1952: {
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid);
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return NULL;
ds@1952: 	}
ds@1952: 
ds@1952: 	return m_pStrings->GetString(pGroup->nameidx);
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::SetGroupGenericImmunity(GroupId id, ImmunityType type, bool enabled)
ds@1952: {
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return;
ds@1952: 	}
ds@1952: 
ds@1952: 	unsigned int level = 0;
ds@1952: 
ds@1952: 	if (enabled)
ds@1952: 	{
ds@1952: 		if (type == Immunity_Default)
ds@1952: 		{
ds@1952: 			level = 1;
ds@1952: 		} else if (type == Immunity_Global) {
ds@1952: 			level = 2;
ds@1952: 		}
ds@1952: 		if (level > pGroup->immunity_level)
ds@1952: 		{
ds@1952: 			pGroup->immunity_level = level;
ds@1952: 		}
ds@1952: 	} else {
ds@1952: 		pGroup->immunity_level = 0;
ds@1952: 	}
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::GetGroupGenericImmunity(GroupId id, ImmunityType type)
ds@1952: {
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (type == Immunity_Default && pGroup->immunity_level >= 1)
ds@1952: 	{
ds@1952: 		return true;
ds@1952: 	} else if (type == Immunity_Global && pGroup->immunity_level >= 2) {
ds@1952: 		return true;
ds@1952: 	}
ds@1952: 
ds@1952: 	return false;
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::AddGroupImmunity(GroupId id, GroupId other_id)
ds@1952: {
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(other_id);
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return;
ds@1952: 	}
ds@1952: 
ds@1952: 	pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return;
ds@1952: 	}
ds@1952: 
ds@1952: 	/* We always need to resize the immunity table */
ds@1952: 	int *table, tblidx;
ds@1952: 	if (pGroup->immune_table == -1)
ds@1952: 	{
ds@1952: 		tblidx = m_pMemory->CreateMem(sizeof(int) * 2, (void **)&table);
ds@1952: 		pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
ds@1952: 		table[0] = 0;
ds@1952: 	} else {
ds@1952: 		int *old_table = (int *)m_pMemory->GetAddress(pGroup->immune_table);
ds@1952: 		/* Break out if this group is already in the list */
ds@1952: 		for (int i=0; i<old_table[0]; i++)
ds@1952: 		{
ds@1952: 			if (old_table[1+i] == other_id)
ds@1952: 			{
ds@1952: 				return;
ds@1952: 			}
ds@1952: 		}
ds@1952: 		tblidx = m_pMemory->CreateMem(sizeof(int) * (old_table[0] + 2), (void **)&table);
ds@1952: 		/* Get the old address again in case of resize */
ds@1952: 		pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
ds@1952: 		old_table = (int *)m_pMemory->GetAddress(pGroup->immune_table);
ds@1952: 		table[0] = old_table[0];
ds@1952: 		for (unsigned int i=1; i<=(unsigned int)old_table[0]; i++)
ds@1952: 		{
ds@1952: 			table[i] = old_table[i];
ds@1952: 		}
ds@1952: 	}
ds@1952: 
ds@1952: 	/* Assign */
ds@1952: 	pGroup->immune_table = tblidx;
ds@1952: 
ds@1952: 	/* Add to the array */
ds@1952: 	table[0]++;
ds@1952: 	table[table[0]] = other_id;
ds@1952: }
ds@1952: 
ds@1952: unsigned int AdminCache::GetGroupImmunityCount(GroupId id)
ds@1952: {
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return 0;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (pGroup->immune_table == -1)
ds@1952: 	{
ds@1952: 		return 0;
ds@1952: 	}
ds@1952: 
ds@1952: 	int *table = (int *)m_pMemory->GetAddress(pGroup->immune_table);
ds@1952: 
ds@1952: 	return table[0];
ds@1952: }
ds@1952: 
ds@1952: GroupId AdminCache::GetGroupImmunity(GroupId id, unsigned int number)
ds@1952: {
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return INVALID_GROUP_ID;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (pGroup->immune_table == -1)
ds@1952: 	{
ds@1952: 		return INVALID_GROUP_ID;
ds@1952: 	}
ds@1952: 
ds@1952: 	int *table = (int *)m_pMemory->GetAddress(pGroup->immune_table);
ds@1952: 	if (number >= (unsigned int)table[0])
ds@1952: 	{
ds@1952: 		return INVALID_GROUP_ID;
ds@1952: 	}
ds@1952: 
ds@1952: 	return table[1 + number];
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::AddGroupCommandOverride(GroupId id, const char *name, OverrideType type, OverrideRule rule)
ds@1952: {
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return;
ds@1952: 	}
ds@1952: 
ds@1952: 	Trie *pTrie = NULL;
ds@1952: 	if (type == Override_Command)
ds@1952: 	{
ds@1952: 		if (pGroup->pCmdTable == NULL)
ds@1952: 		{
ds@1952: 			pGroup->pCmdTable = sm_trie_create();
ds@1952: 		}
ds@1952: 		pTrie = pGroup->pCmdTable;
ds@1952: 	} else if (type == Override_CommandGroup) {
ds@1952: 		if (pGroup->pCmdGrpTable == NULL)
ds@1952: 		{
ds@1952: 			pGroup->pCmdGrpTable = sm_trie_create();
ds@1952: 		}
ds@1952: 		pTrie = pGroup->pCmdGrpTable;
ds@1952: 	} else {
ds@1952: 		return;
ds@1952: 	}
ds@1952: 
ds@1952: 	sm_trie_insert(pTrie, name, (void *)(int)rule);
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::GetGroupCommandOverride(GroupId id, const char *name, OverrideType type, OverrideRule *pRule)
ds@1952: {
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	Trie *pTrie = NULL;
ds@1952: 	if (type == Override_Command)
ds@1952: 	{
ds@1952: 		if (pGroup->pCmdTable == NULL)
ds@1952: 		{
ds@1952: 			return false;
ds@1952: 		}
ds@1952: 		pTrie = pGroup->pCmdTable;
ds@1952: 	} else if (type == Override_CommandGroup) {
ds@1952: 		if (pGroup->pCmdGrpTable == NULL)
ds@1952: 		{
ds@1952: 			return false;
ds@1952: 		}
ds@1952: 		pTrie = pGroup->pCmdGrpTable;
ds@1952: 	} else {
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	void *object;
ds@1952: 	if (!sm_trie_retrieve(pTrie, name, &object))
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (pRule)
ds@1952: 	{
ds@1952: 		*pRule = (OverrideRule)(int)object;
ds@1952: 	}
ds@1952: 	
ds@1952: 	return true;
ds@1952: }
ds@1952: 
ds@1952: Trie *AdminCache::GetMethodByIndex(unsigned int index)
ds@1952: {
ds@1952: 	List<AuthMethod>::iterator iter;
ds@1952: 	for (iter=m_AuthMethods.begin();
ds@1952: 		 iter!=m_AuthMethods.end();
ds@1952: 		 iter++)
ds@1952: 	{
ds@1952: 		if (index-- == 0)
ds@1952: 		{
ds@1952: 			return (*iter).table;
ds@1952: 		}
ds@1952: 	}
ds@1952: 
ds@1952: 	return NULL;
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::InvalidateAdmin(AdminId id)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	AdminUser *pOther;
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (!m_InvalidatingAdmins && !m_destroying)
ds@1952: 	{
ds@1952: 		g_Players.ClearAdminId(id);
ds@1952: 	}
ds@1952: 
ds@1952: 	/* Unlink from the dbl link list */
ds@1952: 	if (id == m_FirstUser && id == m_LastUser)
ds@1952: 	{
ds@1952: 		m_FirstUser = INVALID_ADMIN_ID;
ds@1952: 		m_LastUser = INVALID_ADMIN_ID;
ds@1952: 	} else if (id == m_FirstUser) {
ds@1952: 		m_FirstUser = pUser->next_user;
ds@1952: 		pOther = (AdminUser *)m_pMemory->GetAddress(m_FirstUser);
ds@1952: 		pOther->prev_user = INVALID_ADMIN_ID;
ds@1952: 	} else if (id == m_LastUser) {
ds@1952: 		m_LastUser = pUser->prev_user;
ds@1952: 		pOther = (AdminUser *)m_pMemory->GetAddress(m_LastUser);
ds@1952: 		pOther->next_user = INVALID_ADMIN_ID;
ds@1952: 	} else {
ds@1952: 		pOther = (AdminUser *)m_pMemory->GetAddress(pUser->prev_user);
ds@1952: 		pOther->next_user = pUser->next_user;
ds@1952: 		pOther = (AdminUser *)m_pMemory->GetAddress(pUser->next_user);
ds@1952: 		pOther->prev_user = pUser->prev_user;
ds@1952: 	}
ds@1952: 
ds@1952: 	/* Unlink from auth tables */
ds@1952: 	if (pUser->auth.identidx != -1)
ds@1952: 	{
ds@1952: 		Trie *pTrie = GetMethodByIndex(pUser->auth.index);
ds@1952: 		if (pTrie)
ds@1952: 		{
ds@1952: 			sm_trie_delete(pTrie, m_pStrings->GetString(pUser->auth.identidx));
ds@1952: 		}
ds@1952: 	}
ds@1952: 
ds@1952: 	/* Clear table counts */
ds@1952: 	pUser->grp_count = 0;
ds@1952: 	
ds@1952: 	/* Link into free list */
ds@1952: 	pUser->magic = USR_MAGIC_UNSET;
ds@1952: 	pUser->next_user = m_FreeUserList;
ds@1952: 	m_FreeUserList = id;
ds@1952: 
ds@1952: 	/* Unset serial change */
ds@1952: 	pUser->serialchange = 0;
ds@1952: 
ds@1952: 	return true;
ds@1952: }
ds@1952: 
ds@1952: 
ds@1952: void AdminCache::InvalidateGroup(GroupId id)
ds@1952: {
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
ds@1952: 	AdminGroup *pOther;
ds@1952: 
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return;
ds@1952: 	}
ds@1952: 
ds@1952: 	const char *str = m_pStrings->GetString(pGroup->nameidx);
ds@1952: 	sm_trie_delete(m_pGroups, str);
ds@1952: 
ds@1952: 	/* Unlink from the live dbllink list */
ds@1952: 	if (id == m_FirstGroup && id == m_LastGroup)
ds@1952: 	{
ds@1952: 		m_LastGroup = INVALID_GROUP_ID;
ds@1952: 		m_FirstGroup = INVALID_GROUP_ID;
ds@1952: 	} else if (id == m_FirstGroup) {
ds@1952: 		m_FirstGroup = pGroup->next_grp;
ds@1952: 		pOther = (AdminGroup *)m_pMemory->GetAddress(m_FirstGroup);
ds@1952: 		pOther->prev_grp = INVALID_GROUP_ID;
ds@1952: 	} else if (id == m_LastGroup) {
ds@1952: 		m_LastGroup = pGroup->prev_grp;
ds@1952: 		pOther = (AdminGroup *)m_pMemory->GetAddress(m_LastGroup);
ds@1952: 		pOther->next_grp = INVALID_GROUP_ID;
ds@1952: 	} else {
ds@1952: 		pOther = (AdminGroup *)m_pMemory->GetAddress(pGroup->prev_grp);
ds@1952: 		pOther->next_grp = pGroup->next_grp;
ds@1952: 		pOther = (AdminGroup *)m_pMemory->GetAddress(pGroup->next_grp);
ds@1952: 		pOther->prev_grp = pGroup->prev_grp;
ds@1952: 	}
ds@1952: 
ds@1952: 	/* Free any used memory to be safe */
ds@1952: 	if (pGroup->pCmdGrpTable)
ds@1952: 	{
ds@1952: 		sm_trie_destroy(pGroup->pCmdGrpTable);
ds@1952: 		pGroup->pCmdGrpTable = NULL;
ds@1952: 	}
ds@1952: 	if (pGroup->pCmdTable)
ds@1952: 	{
ds@1952: 		sm_trie_destroy(pGroup->pCmdTable);
ds@1952: 		pGroup->pCmdTable = NULL;
ds@1952: 	}
ds@1952: 
ds@1952: 	/* Link into the free list */
ds@1952: 	pGroup->magic = GRP_MAGIC_UNSET;
ds@1952: 	pGroup->next_grp = m_FreeGroupList;
ds@1952: 	m_FreeGroupList = id;
ds@1952: 
ds@1952: 	int idx = m_FirstUser;
ds@1952: 	AdminUser *pUser;
ds@1952: 	int *table;
ds@1952: 	while (idx != INVALID_ADMIN_ID)
ds@1952: 	{
ds@1952: 		pUser = (AdminUser *)m_pMemory->GetAddress(idx);
ds@1952: 		if (pUser->grp_count > 0)
ds@1952: 		{
ds@1952: 			table = (int *)m_pMemory->GetAddress(pUser->grp_table);
ds@1952: 			for (unsigned int i=0; i<pUser->grp_count; i++)
ds@1952: 			{
ds@1952: 				if (table[i] == id)
ds@1952: 				{
ds@1952: 					/* We have to remove this entry */
ds@1952: 					for (unsigned int j=i+1; j<pUser->grp_count; j++)
ds@1952: 					{
ds@1952: 						/* Move everything down by one */
ds@1952: 						table[j-1] = table[j];
ds@1952: 					}
ds@1952: 					/* Decrease count */
ds@1952: 					pUser->grp_count--;
ds@1952: 					/* Recalculate effective flags */
ds@1952: 					pUser->eflags = pUser->flags;
ds@1952: 					for (unsigned int j=0; j<pUser->grp_count; j++)
ds@1952: 					{
ds@1952: 						pOther = (AdminGroup *)m_pMemory->GetAddress(table[j]);
ds@1952: 						pUser->eflags |= pOther->addflags;
ds@1952: 					}
ds@1952: 					/* Mark as changed */
ds@1952: 					pUser->serialchange++;
ds@1952: 					/* Break now, duplicates aren't allowed */
ds@1952: 					break;
ds@1952: 				}
ds@1952: 			}
ds@1952: 		}
ds@1952: 		idx = pUser->next_user;
ds@1952: 	}
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::InvalidateGroupCache()
ds@1952: {
ds@1952: 	/* Nuke the free list */
ds@1952: 	m_FreeGroupList = -1;
ds@1952: 
ds@1952: 	/* Nuke reverse lookups */
ds@1952: 	sm_trie_clear(m_pGroups);
ds@1952: 
ds@1952: 	/* Free memory on groups */
ds@1952: 	GroupId cur = m_FirstGroup;
ds@1952: 	AdminGroup *pGroup;
ds@1952: 	while (cur != INVALID_GROUP_ID)
ds@1952: 	{
ds@1952: 		pGroup = (AdminGroup *)m_pMemory->GetAddress(cur);
ds@1952: 		assert(pGroup->magic == GRP_MAGIC_SET);
ds@1952: 		if (pGroup->pCmdGrpTable)
ds@1952: 		{
ds@1952: 			sm_trie_destroy(pGroup->pCmdGrpTable);
ds@1952: 		}
ds@1952: 		if (pGroup->pCmdTable)
ds@1952: 		{
ds@1952: 			sm_trie_destroy(pGroup->pCmdTable);
ds@1952: 		}
ds@1952: 		cur = pGroup->next_grp;
ds@1952: 	}
ds@1952: 
ds@1952: 	m_FirstGroup = INVALID_GROUP_ID;
ds@1952: 	m_LastGroup = INVALID_GROUP_ID;
ds@1952: 
ds@1952: 	InvalidateAdminCache(false);
ds@1952: 
ds@1952: 	/* Reset the memory table */
ds@1952: 	m_pMemory->Reset();
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::AddAdminListener(IAdminListener *pListener)
ds@1952: {
ds@1952: 	m_hooks.push_back(pListener);
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::RemoveAdminListener(IAdminListener *pListener)
ds@1952: {
ds@1952: 	m_hooks.remove(pListener);
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::RegisterAuthIdentType(const char *name)
ds@1952: {
ds@1952: 	if (sm_trie_retrieve(m_pAuthTables, name, NULL))
ds@1952: 	{
ds@1952: 		return;
ds@1952: 	}
ds@1952: 
ds@1952: 	Trie *pAuth = sm_trie_create();
ds@1952: 
ds@1952: 	AuthMethod method;
ds@1952: 	method.name.assign(name);
ds@1952: 	method.table = pAuth;
ds@1952: 
ds@1952: 	m_AuthMethods.push_back(method);
ds@1952: 
ds@1952: 	sm_trie_insert(m_pAuthTables, name, pAuth);
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::InvalidateAdminCache(bool unlink_admins)
ds@1952: {
ds@1952: 	m_InvalidatingAdmins = true;
ds@1952: 	if (!m_destroying)
ds@1952: 	{
ds@1952: 		g_Players.ClearAllAdmins();
ds@1952: 	}
ds@1952: 	/* Wipe the identity cache first */
ds@1952: 	List<AuthMethod>::iterator iter;
ds@1952: 	for (iter=m_AuthMethods.begin();
ds@1952: 		 iter!=m_AuthMethods.end();
ds@1952: 		 iter++)
ds@1952: 	{
ds@1952: 		sm_trie_clear((*iter).table);
ds@1952: 	}
ds@1952: 	
ds@1952: 	if (unlink_admins)
ds@1952: 	{
ds@1952: 		while (m_FirstUser != INVALID_ADMIN_ID)
ds@1952: 		{
ds@1952: 			InvalidateAdmin(m_FirstUser);
ds@1952: 		}
ds@1952: 	} else {
ds@1952: 		m_FirstUser = -1;
ds@1952: 		m_LastUser = -1;
ds@1952: 		m_FreeUserList = -1;
ds@1952: 	}
ds@1952: 	m_InvalidatingAdmins = false;
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::DumpAdminCache(AdminCachePart part, bool rebuild)
ds@1952: {
ds@1952: 	List<IAdminListener *>::iterator iter;
ds@1952: 	IAdminListener *pListener;
ds@1952: 	cell_t result;
ds@1952: 
ds@1952: 	if (part == AdminCache_Overrides)
ds@1952: 	{
ds@1952: 		DumpCommandOverrideCache(Override_Command);
ds@1952: 		DumpCommandOverrideCache(Override_CommandGroup);
ds@1952: 		if (rebuild && !m_destroying)
ds@1952: 		{
ds@1952: 			for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
ds@1952: 			{
ds@1952: 				pListener = (*iter);
ds@1952: 				pListener->OnRebuildOverrideCache();
ds@1952: 			}
ds@1952: 			m_pCacheFwd->PushCell(part);
ds@1952: 			m_pCacheFwd->Execute(&result);
ds@1952: 		}
ds@1952: 	} else if (part == AdminCache_Groups || part == AdminCache_Admins) {
ds@1952: 		if (part == AdminCache_Groups)
ds@1952: 		{
ds@1952: 			InvalidateGroupCache();
ds@1952: 			if (rebuild && !m_destroying)
ds@1952: 			{
ds@1952: 				for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
ds@1952: 				{
ds@1952: 					pListener = (*iter);
ds@1952: 					pListener->OnRebuildGroupCache();
ds@1952: 				}
ds@1952: 				m_pCacheFwd->PushCell(part);
ds@1952: 				m_pCacheFwd->Execute(&result);
ds@1952: 			}
ds@1952: 		}
ds@1952: 		InvalidateAdminCache(true);
ds@1952: 		if (rebuild && !m_destroying)
ds@1952: 		{
ds@1952: 			for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
ds@1952: 			{
ds@1952: 				pListener = (*iter);
ds@1952: 				pListener->OnRebuildAdminCache((part == AdminCache_Groups));
ds@1952: 			}
ds@1952: 			m_pCacheFwd->PushCell(AdminCache_Admins);
ds@1952: 			m_pCacheFwd->Execute(&result);
ds@1952: 			g_Players.RecheckAnyAdmins();
ds@1952: 		}
ds@1952: 	}
ds@1952: }
ds@1952: 
ds@1952: const char *AdminCache::GetAdminName(AdminId id)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return NULL;
ds@1952: 	}
ds@1952: 
ds@1952: 	return m_pStrings->GetString(pUser->nameidx);
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::GetMethodIndex(const char *name, unsigned int *_index)
ds@1952: {
ds@1952: 	List<AuthMethod>::iterator iter;
ds@1952: 	unsigned int index = 0;
ds@1952: 	for (iter=m_AuthMethods.begin();
ds@1952: 		 iter!=m_AuthMethods.end();
ds@1952: 		 iter++,index++)
ds@1952: 	{
ds@1952: 		if ((*iter).name.compare(name) == 0)
ds@1952: 		{
ds@1952: 			*_index = index;
ds@1952: 			return true;
ds@1952: 		}
ds@1952: 	}
ds@1952: 
ds@1952: 	return false;
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::BindAdminIdentity(AdminId id, const char *auth, const char *ident)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	Trie *pTable;
ds@1952: 	if (!sm_trie_retrieve(m_pAuthTables, auth, (void **)&pTable))
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (sm_trie_retrieve(pTable, ident, NULL))
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	int i_ident = m_pStrings->AddString(ident);
ds@1952: 
ds@1952: 	pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	pUser->auth.identidx = i_ident;
ds@1952: 	GetMethodIndex(auth, &pUser->auth.index);
ds@1952: 
ds@1952: 	return sm_trie_insert(pTable, ident, (void **)id);
ds@1952: }
ds@1952: 
ds@1952: AdminId AdminCache::FindAdminByIdentity(const char *auth, const char *identity)
ds@1952: {
ds@1952: 	Trie *pTable;
ds@1952: 	if (!sm_trie_retrieve(m_pAuthTables, auth, (void **)&pTable))
ds@1952: 	{
ds@1952: 		return INVALID_ADMIN_ID;
ds@1952: 	}
ds@1952: 
ds@1952: 	void *object;
ds@1952: 	if (!sm_trie_retrieve(pTable, identity, &object))
ds@1952: 	{
ds@1952: 		return INVALID_ADMIN_ID;
ds@1952: 	}
ds@1952: 
ds@1952: 	return (AdminId)object;
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::SetAdminFlag(AdminId id, AdminFlag flag, bool enabled)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (flag < Admin_Reservation
ds@1952: 		|| flag >= AdminFlags_TOTAL)
ds@1952: 	{
ds@1952: 		return;
ds@1952: 	}
ds@1952: 	
ds@1952: 	FlagBits bits = (1<<(FlagBits)flag);
ds@1952: 
ds@1952: 	if (enabled)
ds@1952: 	{
ds@1952: 		pUser->flags |= bits;
ds@1952: 		pUser->eflags |= bits;
ds@1952: 	} else {
ds@1952: 		pUser->flags &= ~bits;
ds@1952: 		pUser->eflags &= ~bits;
ds@1952: 	}
ds@1952: 
ds@1952: 	pUser->serialchange++;
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::GetAdminFlag(AdminId id, AdminFlag flag, AccessMode mode)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (flag < Admin_Reservation
ds@1952: 		|| flag >= AdminFlags_TOTAL)
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	FlagBits bit = (1<<(FlagBits)flag);
ds@1952: 
ds@1952: 	if (mode == Access_Real)
ds@1952: 	{
ds@1952: 		return ((pUser->flags & bit) == bit);
ds@1952: 	} else if (mode == Access_Effective) {
ds@1952: 		bool has_bit = ((pUser->eflags & bit) == bit);
ds@1952: 		if (!has_bit && flag != Admin_Root && ((pUser->eflags & ADMFLAG_ROOT) == ADMFLAG_ROOT))
ds@1952: 		{
ds@1952: 			has_bit = true;
ds@1952: 		}
ds@1952: 		return has_bit;
ds@1952: 	}
ds@1952: 
ds@1952: 	return false;
ds@1952: }
ds@1952: 
ds@1952: FlagBits AdminCache::GetAdminFlags(AdminId id, AccessMode mode)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return 0;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (mode == Access_Real)
ds@1952: 	{
ds@1952: 		return pUser->flags;
ds@1952: 	} else if (mode == Access_Effective) {
ds@1952: 		return pUser->eflags;
ds@1952: 	}
ds@1952: 
ds@1952: 	return 0;
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::SetAdminFlags(AdminId id, AccessMode mode, FlagBits bits)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (mode == Access_Real)
ds@1952: 	{
ds@1952: 		pUser->flags = bits;
ds@1952: 		pUser->eflags = bits;
ds@1952: 	} else if (mode == Access_Effective) {
ds@1952: 		pUser->eflags = bits;
ds@1952: 	}
ds@1952: 
ds@1952: 	pUser->serialchange++;
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::AdminInheritGroup(AdminId id, GroupId gid)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid);
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	/* First check for duplicates */
ds@1952: 	if (pUser->grp_count != 0)
ds@1952: 	{
ds@1952: 		int *temp_table = (int *)m_pMemory->GetAddress(pUser->grp_table);
ds@1952: 		for (unsigned int i=0; i<pUser->grp_count; i++)
ds@1952: 		{
ds@1952: 			if (temp_table[i] == gid)
ds@1952: 			{
ds@1952: 				return false;
ds@1952: 			}
ds@1952: 		}
ds@1952: 	}
ds@1952: 
ds@1952: 	int *table;
ds@1952: 	if (pUser->grp_count + 1 > pUser->grp_size)
ds@1952: 	{
ds@1952: 		int new_size = 0;
ds@1952: 		int tblidx;
ds@1952: 		
ds@1952: 		if (pUser->grp_size == 0)
ds@1952: 		{
ds@1952: 			new_size = 2;
ds@1952: 		} else {
ds@1952: 			new_size = pUser->grp_size * 2;
ds@1952: 		}
ds@1952: 		
ds@1952: 		/* Create and refresh pointers */
ds@1952: 		tblidx = m_pMemory->CreateMem(new_size * sizeof(int), (void **)&table);
ds@1952: 		pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 		pGroup = (AdminGroup *)m_pMemory->GetAddress(gid);
ds@1952: 
ds@1952: 		/* Copy old data if necessary */
ds@1952: 		if (pUser->grp_table != -1)
ds@1952: 		{
ds@1952: 			int *old_table = (int *)m_pMemory->GetAddress(pUser->grp_table);
ds@1952: 			memcpy(table, old_table, sizeof(int) * pUser->grp_count);
ds@1952: 		}
ds@1952: 		pUser->grp_table = tblidx;
ds@1952: 		pUser->grp_size = new_size;
ds@1952: 	} else {
ds@1952: 		table = (int *)m_pMemory->GetAddress(pUser->grp_table);
ds@1952: 	}
ds@1952: 
ds@1952: 	table[pUser->grp_count] = gid;
ds@1952: 	pUser->grp_count++;
ds@1952: 
ds@1952: 	/* Compute new effective permissions */
ds@1952: 	pUser->eflags |= pGroup->addflags;
ds@1952: 
ds@1952: 	if (pGroup->immunity_level > pUser->immunity_level)
ds@1952: 	{
ds@1952: 		pUser->immunity_level = pGroup->immunity_level;
ds@1952: 	}
ds@1952: 
ds@1952: 	pUser->serialchange++;
ds@1952: 
ds@1952: 	return true;
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::IsValidAdmin(AdminId id)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	return (pUser != NULL && pUser->magic == USR_MAGIC_SET);
ds@1952: }
ds@1952: 
ds@1952: unsigned int AdminCache::GetAdminGroupCount(AdminId id)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return 0;
ds@1952: 	}
ds@1952: 
ds@1952: 	return pUser->grp_count;
ds@1952: }
ds@1952: 
ds@1952: GroupId AdminCache::GetAdminGroup(AdminId id, unsigned int index, const char **name)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET || index >= pUser->grp_count)
ds@1952: 	{
ds@1952: 		return INVALID_GROUP_ID;
ds@1952: 	}
ds@1952: 
ds@1952: 	int *table = (int *)m_pMemory->GetAddress(pUser->grp_table);
ds@1952: 
ds@1952: 	GroupId gid = table[index];
ds@1952: 
ds@1952: 	if (name)
ds@1952: 	{
ds@1952: 		*name = GetGroupName(gid);
ds@1952: 	}
ds@1952: 
ds@1952: 	return gid;
ds@1952: }
ds@1952: 
ds@1952: const char *AdminCache::GetAdminPassword(AdminId id)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return NULL;
ds@1952: 	}
ds@1952: 
ds@1952: 	return m_pStrings->GetString(pUser->password);
ds@1952: }
ds@1952: 
ds@1952: void AdminCache::SetAdminPassword(AdminId id, const char *password)
ds@1952: {	
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return;
ds@1952: 	}
ds@1952: 
ds@1952: 	int i_password = m_pStrings->AddString(password);
ds@1952: 	pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	pUser->password = i_password;
ds@1952: }
ds@1952: 
ds@1952: unsigned int AdminCache::FlagBitsToBitArray(FlagBits bits, bool array[], unsigned int maxSize)
ds@1952: {
ds@1952: 	unsigned int i;
ds@1952: 	for (i=0; i<maxSize && i<AdminFlags_TOTAL; i++)
ds@1952: 	{
ds@1952: 		array[i] = ((bits & (1<<i)) == (unsigned)(1<<i));
ds@1952: 	}
ds@1952: 
ds@1952: 	return i;
ds@1952: }
ds@1952: 
ds@1952: FlagBits AdminCache::FlagBitArrayToBits(const bool array[], unsigned int maxSize)
ds@1952: {
ds@1952: 	FlagBits bits = 0;
ds@1952: 	for (unsigned int i=0; i<maxSize && i<AdminFlags_TOTAL; i++)
ds@1952: 	{
ds@1952: 		if (array[i])
ds@1952: 		{
ds@1952: 			bits |= (1<<i);
ds@1952: 		}
ds@1952: 	}
ds@1952: 
ds@1952: 	return bits;
ds@1952: }
ds@1952: 
ds@1952: FlagBits AdminCache::FlagArrayToBits(const AdminFlag array[], unsigned int numFlags)
ds@1952: {
ds@1952: 	FlagBits bits = 0;
ds@1952: 	for (unsigned int i=0; i<numFlags && i<AdminFlags_TOTAL; i++)
ds@1952: 	{
ds@1952: 		bits |= (1 << (FlagBits)array[i]);
ds@1952: 	}
ds@1952: 
ds@1952: 	return bits;
ds@1952: }
ds@1952: 
ds@1952: unsigned int AdminCache::FlagBitsToArray(FlagBits bits, AdminFlag array[], unsigned int maxSize)
ds@1952: {
ds@1952: 	unsigned int i, num=0;
ds@1952: 	for (i=0; num<maxSize && i<AdminFlags_TOTAL; i++)
ds@1952: 	{
ds@1952: 		if ((bits & (1<<i)) == (unsigned)(1<<i))
ds@1952: 		{
ds@1952: 			array[num++] = (AdminFlag)i;
ds@1952: 		}
ds@1952: 	}
ds@1952: 
ds@1952: 	return num;
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::CheckAdminFlags(AdminId id, FlagBits bits)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	return ((pUser->eflags & bits) == bits);
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::CanAdminTarget(AdminId id, AdminId target)
ds@1952: {
ds@1952: 	/** 
ds@1952: 	 * Zeroth, if the targeting AdminId is INVALID_ADMIN_ID, targeting fails.
ds@1952: 	 * First, if the targeted AdminId is INVALID_ADMIN_ID, targeting succeeds.
ds@1952: 	 */
ds@1952: 
ds@1952: 	if (id == INVALID_ADMIN_ID)
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (target == INVALID_ADMIN_ID)
ds@1952: 	{
ds@1952: 		return true;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (id == target)
ds@1952: 	{
ds@1952: 		return true;
ds@1952: 	}
ds@1952: 
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	AdminUser *pTarget = (AdminUser *)m_pMemory->GetAddress(target);
ds@1952: 	if (!pTarget || pTarget->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	/** 
ds@1952: 	 * Second, if the targeting admin is root, targeting succeeds.
ds@1952: 	 */
ds@1952: 	if (pUser->eflags & ADMFLAG_ROOT)
ds@1952: 	{
ds@1952: 		return true;
ds@1952: 	}
ds@1952: 
ds@1952: 	/** Fourth, if the targeted admin is immune from targeting admin. */
ds@1952: 	int mode = sm_immunity_mode.GetInt();
ds@1952: 	switch (mode)
ds@1952: 	{
ds@1952: 		case 1:
ds@1952: 		{
ds@1952: 			if (pTarget->immunity_level > pUser->immunity_level)
ds@1952: 			{
ds@1952: 				return false;
ds@1952: 			}
ds@1952: 			break;
ds@1952: 		}
ds@1952: 		case 3:
ds@1952: 		{
ds@1952: 			/* If neither has any immunity, let this pass. */
ds@1952: 			if (!pUser->immunity_level && !pTarget->immunity_level)
ds@1952: 			{
ds@1952: 				return true;
ds@1952: 			}
ds@1952: 			/* Don't break, go to the next case. */
ds@1952: 		}
ds@1952: 		case 2:
ds@1952: 		{
ds@1952: 			if (pTarget->immunity_level >= pUser->immunity_level)
ds@1952: 			{
ds@1952: 				return false;
ds@1952: 			}
ds@1952: 			break;
ds@1952: 		}
ds@1952: 	}
ds@1952: 
ds@1952: 	/** 
ds@1952: 	 * Fifth, if the targeted admin has specific immunity from the
ds@1952: 	 *  targeting admin via group immunities, targeting fails.
ds@1952: 	 */
ds@1952: 	//:TODO: speed this up... maybe with trie hacks.
ds@1952: 	//idea is to insert %d.%d in the trie after computing this and use it as a cache lookup.
ds@1952: 	//problem is the trie cannot delete prefixes, so we'd have a problem with invalidations.
ds@1952: 	if (pTarget->grp_count > 0 && pUser->grp_count > 0)
ds@1952: 	{
ds@1952: 		int *grp_table = (int *)m_pMemory->GetAddress(pTarget->grp_table);
ds@1952: 		int *src_table = (int *)m_pMemory->GetAddress(pUser->grp_table);
ds@1952: 		GroupId id, other;
ds@1952: 		unsigned int num;
ds@1952: 		for (unsigned int i=0; i<pTarget->grp_count; i++)
ds@1952: 		{
ds@1952: 			id = grp_table[i];
ds@1952: 			num = GetGroupImmunityCount(id);
ds@1952: 			for (unsigned int j=0; j<num; i++)
ds@1952: 			{
ds@1952: 				other = GetGroupImmunity(id, j);
ds@1952: 				for (unsigned int k=0; k<pUser->grp_count; k++)
ds@1952: 				{
ds@1952: 					if (other == src_table[k])
ds@1952: 					{
ds@1952: 						return false;
ds@1952: 					}
ds@1952: 				}
ds@1952: 			}
ds@1952: 		}
ds@1952: 	}
ds@1952: 	
ds@1952: 	return true;
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::FindFlag(char c, AdminFlag *pAdmFlag)
ds@1952: {
dvander@1965: 	if (c < 'a' 
dvander@1965: 		|| c > 'z'
dvander@1965: 		|| !g_FlagSet[(unsigned)c - (unsigned)'a'])
ds@1952: 	{
ds@1952: 		return false;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (pAdmFlag)
ds@1952: 	{
ds@1952: 		*pAdmFlag = g_FlagLetters[(unsigned)c - (unsigned)'a'];
ds@1952: 	}
ds@1952: 
ds@1952: 	return true;
ds@1952: }
ds@1952: 
dvander@1965: bool AdminCache::FindFlagChar(AdminFlag flag, char *c)
dvander@1965: {
dvander@1965: 	if (!g_FlagSet[flag])
dvander@1965: 	{
dvander@1965: 		return false;
dvander@1965: 	}
dvander@1965: 
dvander@1965: 	if (c)
dvander@1965: 	{
dvander@1965: 		*c = g_ReverseFlags[flag];
dvander@1965: 	}
dvander@1965: 
dvander@1965: 	return true;
dvander@1965: }
dvander@1965: 
ds@1952: FlagBits AdminCache::ReadFlagString(const char *flags, const char **end)
ds@1952: {
ds@1952: 	FlagBits bits = 0;
ds@1952: 
ds@1952: 	while (flags && (*flags != '\0'))
ds@1952: 	{
ds@1952: 		AdminFlag flag;
ds@1952: 		if (!FindFlag(*flags, &flag))
ds@1952: 		{
ds@1952: 			break;
ds@1952: 		}
ds@1952: 		bits |= FlagArrayToBits(&flag, 1);
ds@1952: 		flags++;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (end)
ds@1952: 	{
ds@1952: 		*end = flags;
ds@1952: 	}
ds@1952: 
ds@1952: 	return bits;
ds@1952: }
ds@1952: 
ds@1952: unsigned int AdminCache::GetAdminSerialChange(AdminId id)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return 0;
ds@1952: 	}
ds@1952: 
ds@1952: 	return pUser->serialchange;
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::CanAdminUseCommand(int client, const char *cmd)
ds@1952: {
ds@1952: 	FlagBits bits;
ds@1952: 	OverrideType otype = Override_Command;
ds@1952: 
ds@1952: 	if (cmd[0] == '@')
ds@1952: 	{
ds@1952: 		otype = Override_CommandGroup;
ds@1952: 		cmd++;
ds@1952: 	}
ds@1952: 
ds@1952: 	if (!g_ConCmds.LookForCommandAdminFlags(cmd, &bits))
ds@1952: 	{
ds@1952: 		if (!GetCommandOverride(cmd, otype, &bits))
ds@1952: 		{
ds@1952: 			bits = 0;
ds@1952: 		}
ds@1952: 	}
ds@1952: 
ds@1952: 	return g_ConCmds.CheckCommandAccess(client, cmd, bits);
ds@1952: }
ds@1952: 
ds@1952: unsigned int AdminCache::SetGroupImmunityLevel(GroupId gid, unsigned int level)
ds@1952: {
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid);
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return 0;
ds@1952: 	}
ds@1952: 
ds@1952: 	unsigned int old_level = pGroup->immunity_level;
ds@1952: 
ds@1952: 	pGroup->immunity_level = level;
ds@1952: 
ds@1952: 	return old_level;
ds@1952: }
ds@1952: 
ds@1952: unsigned int AdminCache::GetGroupImmunityLevel(GroupId gid)
ds@1952: {
ds@1952: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid);
ds@1952: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return 0;
ds@1952: 	}
ds@1952: 
ds@1952: 	return pGroup->immunity_level;
ds@1952: }
ds@1952: 
ds@1952: unsigned int AdminCache::SetAdminImmunityLevel(AdminId id, unsigned int level)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return 0;
ds@1952: 	}
ds@1952: 
ds@1952: 	unsigned int old_level = pUser->immunity_level;
ds@1952: 
ds@1952: 	pUser->immunity_level = level;
ds@1952: 
ds@1952: 	return old_level;
ds@1952: }
ds@1952: 
ds@1952: unsigned int AdminCache::GetAdminImmunityLevel(AdminId id)
ds@1952: {
ds@1952: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
ds@1952: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
ds@1952: 	{
ds@1952: 		return 0;
ds@1952: 	}
ds@1952: 
ds@1952: 	return pUser->immunity_level;
ds@1952: }
ds@1952: 
ds@1952: bool AdminCache::CheckAccess(int client, const char *cmd, FlagBits flags, bool override_only)
ds@1952: {
ds@1952: 	if (client == 0)
ds@1952: 	{
ds@1952: 		return true;
ds@1952: 	}
ds@1952: 
ds@1952: 	/* Auto-detect a command if we can */
ds@1952: 	FlagBits bits = flags;
ds@1952: 	bool found_command = false;
ds@1952: 	if (!override_only)
ds@1952: 	{
ds@1952: 		found_command = g_ConCmds.LookForCommandAdminFlags(cmd, &bits);
ds@1952: 	}
ds@1952: 
ds@1952: 	if (!found_command)
ds@1952: 	{
ds@1952: 		GetCommandOverride(cmd, Override_Command, &bits);
ds@1952: 	}
ds@1952: 
ds@1952: 	return g_ConCmds.CheckCommandAccess(client, cmd, bits) ? 1 : 0;
ds@1952: }
dvander@1965: 
dvander@1965: void iterator_glob_basic_override(Trie *pTrie, const char *key, void **value, void *data)
dvander@1965: {
dvander@1965: 	FILE *fp;
dvander@1965: 	int flags;
dvander@1965: 	char flagstr[64];
dvander@1965: 
dvander@1965: 	fp = (FILE *)data;
dvander@1965: 	flags = (int)*value;
dvander@1965: 	g_Admins.FillFlagString(flags, flagstr, sizeof(flagstr));
dvander@1965: 
dvander@1965: 	fprintf(fp, "\t\"%s\"\t\t\"%s\"\n", key, flagstr);
dvander@1965: }
dvander@1965: 
dvander@1965: void iterator_glob_grp_override(Trie *pTrie, const char *key, void **value, void *data)
dvander@1965: {
dvander@1965: 	FILE *fp;
dvander@1965: 	int flags;
dvander@1965: 	char flagstr[64];
dvander@1965: 
dvander@1965: 	fp = (FILE *)data;
dvander@1965: 	flags = (int)*value;
dvander@1965: 	g_Admins.FillFlagString(flags, flagstr, sizeof(flagstr));
dvander@1965: 
dvander@1965: 	fprintf(fp, "\t\"@%s\"\t\t\"%s\"\n", key, flagstr);
dvander@1965: }
dvander@1965: 
dvander@1965: void iterator_group_basic_override(Trie *pTrie, const char *key, void **value, void *data)
dvander@1965: {
dvander@1965: 	FILE *fp;
dvander@1965: 	int flags;
dvander@1965: 	char flagstr[64];
dvander@1965: 
dvander@1965: 	fp = (FILE *)data;
dvander@1965: 	flags = (int)*value;
dvander@1965: 	g_Admins.FillFlagString(flags, flagstr, sizeof(flagstr));
dvander@1965: 
dvander@1965: 	fprintf(fp, "\t\t\t\"%s\"\t\t\"%s\"\n", key, flagstr);
dvander@1965: }
dvander@1965: 
dvander@1965: void iterator_group_grp_override(Trie *pTrie, const char *key, void **value, void *data)
dvander@1965: {
dvander@1965: 	FILE *fp;
dvander@1965: 	int flags;
dvander@1965: 	char flagstr[64];
dvander@1965: 
dvander@1965: 	fp = (FILE *)data;
dvander@1965: 	flags = (int)*value;
dvander@1965: 	g_Admins.FillFlagString(flags, flagstr, sizeof(flagstr));
dvander@1965: 
dvander@1965: 	fprintf(fp, "\t\t\t\"@%s\"\t\t\"%s\"\n", key, flagstr);
dvander@1965: }
dvander@1965: 
dvander@1965: void AdminCache::DumpCache(FILE *fp)
dvander@1965: {
dvander@1965: 	int *itable;
dvander@1965: 	AdminId aid;
dvander@1965: 	GroupId gid;
dvander@1965: 	char flagstr[64];
dvander@1965: 	unsigned int num;
dvander@1965: 	AdminUser *pAdmin;
dvander@1965: 	AdminGroup *pGroup;
dvander@1965: 	char name_buffer[512];
dvander@1965: 
dvander@1965: 	fprintf(fp, "\"Groups\"\n{\n");
dvander@1965: 	
dvander@1965: 	num = 0;
dvander@1965: 	gid = m_FirstGroup;
dvander@1965: 	while (gid != INVALID_GROUP_ID
dvander@1965: 			&& (pGroup = GetGroup(gid)) != NULL)
dvander@1965: 	{
dvander@1965: 		num++;
dvander@1965: 		FillFlagString(pGroup->addflags, flagstr, sizeof(flagstr));
dvander@1965: 
dvander@1965: 		fprintf(fp, "\t/* num = %d, gid = 0x%X */\n", num, gid);
dvander@1965: 		fprintf(fp, "\t\"%s\"\n\t{\n", GetString(pGroup->nameidx));
dvander@1965: 		fprintf(fp, "\t\t\"flags\"\t\t\t\"%s\"\n", flagstr);
dvander@1965: 		fprintf(fp, "\t\t\"immunity\"\t\t\"%d\"\n", pGroup->immunity_level);
dvander@1965: 		
dvander@1965: 		if (pGroup->immune_table != -1
dvander@1965: 			&& (itable = (int *)m_pMemory->GetAddress(pGroup->immune_table)) != NULL)
dvander@1965: 		{
dvander@1965: 			AdminGroup *pAltGroup;
dvander@1965: 			const char *gname, *mod;
dvander@1965: 
dvander@1965: 			for (int i = 1; i <= itable[0]; i++)
dvander@1965: 			{
dvander@1965: 				if ((pAltGroup = GetGroup(itable[i])) == NULL)
dvander@1965: 				{
dvander@1965: 					/* Assume the rest of the table is corrupt */
dvander@1965: 					break;
dvander@1965: 				}
dvander@1965: 				gname = GetString(pAltGroup->nameidx);
dvander@1965: 				if (atoi(gname) != 0)
dvander@1965: 				{
dvander@1965: 					mod = "@";
dvander@1965: 				}
dvander@1965: 				else
dvander@1965: 				{
dvander@1965: 					mod = "";
dvander@1965: 				}
dvander@1965: 				fprintf(fp, "\t\t\"immunity\"\t\t\"%s%s\"\n", mod, gname);
dvander@1965: 			}
dvander@1965: 		}
dvander@1965: 
dvander@1965: 		fprintf(fp, "\n\t\t\"Overrides\"\n\t\t{\n");
dvander@1965: 		if (pGroup->pCmdGrpTable != NULL)
dvander@1965: 		{
dvander@1965: 			sm_trie_bad_iterator(pGroup->pCmdGrpTable, 
dvander@1965: 				name_buffer,
dvander@1965: 				sizeof(name_buffer),
dvander@1965: 				iterator_group_grp_override,
dvander@1965: 				fp);
dvander@1965: 		}
dvander@1965: 		if (pGroup->pCmdTable != NULL)
dvander@1965: 		{
dvander@1965: 			sm_trie_bad_iterator(pGroup->pCmdTable, 
dvander@1965: 				name_buffer,
dvander@1965: 				sizeof(name_buffer),
dvander@1965: 				iterator_group_basic_override,
dvander@1965: 				fp);
dvander@1965: 		}
dvander@1965: 		fprintf(fp, "\t\t}\n");
dvander@1965: 
dvander@1965: 		fprintf(fp, "\t}\n");
dvander@1965: 
dvander@1965: 		if ((gid = pGroup->next_grp) != INVALID_GROUP_ID)
dvander@1965: 		{
dvander@1965: 			fprintf(fp, "\n");
dvander@1965: 		}
dvander@1965: 	}
dvander@1965: 
dvander@1965: 	fprintf(fp, "}\n\n");
dvander@1965: 	fprintf(fp, "\"Admins\"\n{\n");
dvander@1965: 
dvander@1965: 	num = 0;
dvander@1965: 	aid = m_FirstUser;
dvander@1965: 	while (aid != INVALID_ADMIN_ID
dvander@1965: 			&& (pAdmin = GetUser(aid)) != NULL)
dvander@1965: 	{
dvander@1965: 		num++;
dvander@1965: 		FillFlagString(pAdmin->flags, flagstr, sizeof(flagstr));
dvander@1965: 
dvander@1965: 		fprintf(fp, "\t/* num = %d, aid = 0x%X, serialno = 0x%X*/\n", num, aid, pAdmin->serialchange);
dvander@1965: 
dvander@1965: 		if (pAdmin->nameidx != -1)
dvander@1965: 		{
dvander@1965: 			fprintf(fp, "\t\"%s\"\n\t{\n", GetString(pAdmin->nameidx));
dvander@1965: 		}
dvander@1965: 		else
dvander@1965: 		{
dvander@1965: 			fprintf(fp, "\t\"\"\n\t{\n");
dvander@1965: 		}
dvander@1965: 
dvander@1965: 		if (pAdmin->auth.identidx != -1)
dvander@1965: 		{
dvander@1965: 			fprintf(fp, "\t\t\"auth\"\t\t\t\"%s\"\n", GetMethodName(pAdmin->auth.index));
dvander@1965: 			fprintf(fp, "\t\t\"identity\"\t\t\"%s\"\n", GetString(pAdmin->auth.identidx));
dvander@1965: 		}
dvander@1965: 		if (pAdmin->password != -1)
dvander@1965: 		{
dvander@1965: 			fprintf(fp, "\t\t\"password\"\t\t\"%s\"\n", GetString(pAdmin->password));
dvander@1965: 		}
dvander@1965: 		fprintf(fp, "\t\t\"flags\"\t\t\t\"%s\"\n", flagstr);
dvander@1965: 		fprintf(fp, "\t\t\"immunity\"\t\t\"%d\"\n", pAdmin->immunity_level);
dvander@1965: 
dvander@1965: 		if (pAdmin->grp_count != 0 
dvander@1965: 			&& pAdmin->grp_table != -1 
dvander@1965: 			&& (itable = (int *)m_pMemory->GetAddress(pAdmin->grp_table)) != NULL)
dvander@1965: 		{
dvander@1965: 			unsigned int i;
dvander@1965: 
dvander@1965: 			for (i = 0; i < pAdmin->grp_count; i++)
dvander@1965: 			{
dvander@1965: 				if ((pGroup = GetGroup(itable[i])) == NULL)
dvander@1965: 				{
dvander@1965: 					/* Assume the rest of the table is corrupt */
dvander@1965: 					break;
dvander@1965: 				}
dvander@1965: 				fprintf(fp, "\t\t\"group\"\t\t\t\"%s\"\n", GetString(pGroup->nameidx));
dvander@1965: 			}
dvander@1965: 		}
dvander@1965: 
dvander@1965: 		fprintf(fp, "\t}\n");
dvander@1965: 
dvander@1965: 		if ((aid = pAdmin->next_user) != INVALID_ADMIN_ID)
dvander@1965: 		{
dvander@1965: 			fprintf(fp, "\n");
dvander@1965: 		}
dvander@1965: 	}
dvander@1965: 
dvander@1965: 	fprintf(fp, "}\n\n");
dvander@1965: 
dvander@1965: 	fprintf(fp, "\"Overrides\"\n{\n");
dvander@1965: 	if (m_pCmdGrpOverrides != NULL)
dvander@1965: 	{
dvander@1965: 		sm_trie_bad_iterator(m_pCmdGrpOverrides,
dvander@1965: 			name_buffer,
dvander@1965: 			sizeof(name_buffer),
dvander@1965: 			iterator_glob_grp_override,
dvander@1965: 			fp);
dvander@1965: 	}
dvander@1965: 	if (m_pCmdOverrides != NULL)
dvander@1965: 	{
dvander@1965: 		sm_trie_bad_iterator(m_pCmdOverrides,
dvander@1965: 			name_buffer, 
dvander@1965: 			sizeof(name_buffer), 
dvander@1965: 			iterator_glob_basic_override, 
dvander@1965: 			fp);
dvander@1965: 	}
dvander@1965: 	fprintf(fp, "}\n");
dvander@1965: }
dvander@1965: 
dvander@1965: AdminGroup *AdminCache::GetGroup(GroupId gid)
dvander@1965: {
dvander@1965: 	AdminGroup *pGroup;
dvander@1965: 	
dvander@1965: 	pGroup = (AdminGroup *)m_pMemory->GetAddress(gid);
dvander@1965: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@1965: 	{
dvander@1965: 		return NULL;
dvander@1965: 	}
dvander@1965: 
dvander@1965: 	return pGroup;
dvander@1965: }
dvander@1965: 
dvander@1965: AdminUser *AdminCache::GetUser(AdminId aid)
dvander@1965: {
dvander@1965: 	AdminUser *pAdmin;
dvander@1965: 
dvander@1965: 	pAdmin = (AdminUser *)m_pMemory->GetAddress(aid);
dvander@1965: 	if (!pAdmin || pAdmin->magic != USR_MAGIC_SET)
dvander@1965: 	{
dvander@1965: 		return NULL;
dvander@1965: 	}
dvander@1965: 
dvander@1965: 	return pAdmin;
dvander@1965: }
dvander@1965: 
dvander@1965: const char *AdminCache::GetMethodName(unsigned int index)
dvander@1965: {
dvander@1965: 	List<AuthMethod>::iterator iter;
dvander@1965: 	for (iter=m_AuthMethods.begin();
dvander@1965: 		iter!=m_AuthMethods.end();
dvander@1965: 		iter++)
dvander@1965: 	{
dvander@1965: 		if (index-- == 0)
dvander@1965: 		{
dvander@1965: 			return (*iter).name.c_str();
dvander@1965: 		}
dvander@1965: 	}
dvander@1965: 
dvander@1965: 	return NULL;
dvander@1965: }
dvander@1965: 
dvander@1965: const char *AdminCache::GetString(int idx)
dvander@1965: {
dvander@1965: 	return m_pStrings->GetString(idx);
dvander@1965: }
dvander@1965: 
dvander@1965: size_t AdminCache::FillFlagString(FlagBits bits, char *buffer, size_t maxlen)
dvander@1965: {
dvander@1965: 	size_t pos;
dvander@1965: 	unsigned int i, num_flags;
dvander@1965: 	AdminFlag flags[AdminFlags_TOTAL];
dvander@1965: 
dvander@1965: 	num_flags = FlagBitsToArray(bits, flags, AdminFlags_TOTAL);
dvander@1965: 
dvander@1965: 	pos = 0;
dvander@1965: 	for (i = 0; pos < maxlen && i < num_flags; i++)
dvander@1965: 	{
dvander@1965: 		if (FindFlagChar(flags[i], &buffer[pos]))
dvander@1965: 		{
dvander@1965: 			pos++;
dvander@1965: 		}
dvander@1965: 	}
dvander@1965: 	buffer[pos] = '\0';
dvander@1965: 
dvander@1965: 	return pos;
dvander@1965: }
dvander@1965: 
dvander@1965: CON_COMMAND(sm_dump_admcache, "Dumps the admin cache for debugging")
dvander@1965: {
dvander@1965: 	FILE *fp;
dvander@1965: 	char buffer[PLATFORM_MAX_PATH];
dvander@1965: 
dvander@1965: 	g_SourceMod.BuildPath(Path_SM, buffer, sizeof(buffer), "data/admin_cache_dump.txt");
dvander@1965: 
dvander@1965: 	if ((fp = fopen(buffer, "wt")) == NULL)
dvander@1965: 	{
dvander@1965: 		g_RootMenu.ConsolePrint("Could not open file for writing: %s", buffer);
dvander@1965: 		return;
dvander@1965: 	}
dvander@1965: 
dvander@1965: 	g_Admins.DumpCache(fp);
dvander@1965: 
dvander@1965: 	g_RootMenu.ConsolePrint("Admin cache dumped to: %s", buffer);
dvander@1965: 
dvander@1965: 	fclose(fp);
dvander@1965: }