dvander@370: /**
ds@653:  * vim: set ts=4 :
dvander@1231:  * ================================================================
dvander@1231:  * SourceMod
dvander@1231:  * Copyright (C) 2004-2007 AlliedModders LLC.  All rights reserved.
dvander@1231:  * ================================================================
dvander@370:  *
dvander@1231:  * This program is free software; you can redistribute it and/or 
dvander@1231:  * modify it under the terms of the GNU General Public License, 
dvander@1231:  * version 3.0, as published by the Free Software Foundation.
dvander@1231:  * 
dvander@1231:  * This program is distributed in the hope that it will be useful,
dvander@1231:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
dvander@1231:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
dvander@1231:  * GNU General Public License for more details.
dvander@1231:  *
dvander@1231:  * You should have received a copy of the GNU General Public License
dvander@1231:  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@1231:  *
dvander@1231:  * As a special exception, AlliedModders LLC gives you permission to 
dvander@1231:  * link the code of this program (as well as its derivative works) to 
dvander@1231:  * "Half-Life 2," the "Source Engine," the "SourcePawn JIT," and any 
dvander@1231:  * Game MODs that run on software by the Valve Corporation.  You must 
dvander@1231:  * obey the GNU General Public License in all respects for all other 
dvander@1231:  * code used. Additionally, AlliedModders LLC grants this exception 
dvander@1231:  * to all derivative works. AlliedModders LLC defines further 
dvander@1231:  * exceptions, found in LICENSE.txt (as of this writing, version 
dvander@1231:  * JULY-31-2007), or <http://www.sourcemod.net/license.php>.
dvander@370:  *
dvander@370:  * Version: $Id$
dvander@370:  */
dvander@370: 
dvander@335: #include <string.h>
dvander@335: #include <assert.h>
dvander@335: #include "AdminCache.h"
dvander@335: #include "ShareSys.h"
dvander@336: #include "ForwardSys.h"
dvander@591: #include "PlayerManager.h"
dvander@583: #include "ConCmdManager.h"
dvander@335: 
dvander@335: AdminCache g_Admins;
dvander@335: 
dvander@335: AdminCache::AdminCache()
dvander@335: {
dvander@433: 	m_pCmdOverrides = sm_trie_create();
dvander@433: 	m_pCmdGrpOverrides = sm_trie_create();
dvander@335: 	m_pStrings = new BaseStringTable(1024);
dvander@335: 	m_pMemory = m_pStrings->GetMemTable();
dvander@335: 	m_FreeGroupList = m_FirstGroup = m_LastGroup = INVALID_GROUP_ID;
dvander@404: 	m_FreeUserList = m_FirstUser = m_LastUser = INVALID_ADMIN_ID;
dvander@335: 	m_pGroups = sm_trie_create();
dvander@336: 	m_pCacheFwd = NULL;
dvander@404: 	m_FirstGroup = -1;
dvander@405: 	m_pAuthTables = sm_trie_create();
dvander@458: 	m_InvalidatingAdmins = false;
dvander@520: 	m_destroying = false;
dvander@335: }
dvander@335: 
dvander@335: AdminCache::~AdminCache()
dvander@335: {
dvander@520: 	m_destroying = true;
dvander@438: 	DumpAdminCache(AdminCache_Overrides, false);
dvander@438: 	DumpAdminCache(AdminCache_Groups, false);
dvander@438: 
dvander@434: 	sm_trie_destroy(m_pCmdGrpOverrides);
dvander@434: 	sm_trie_destroy(m_pCmdOverrides);
dvander@335: 
dvander@335: 	if (m_pGroups)
dvander@335: 	{
dvander@335: 		sm_trie_destroy(m_pGroups);
dvander@335: 	}
dvander@335: 
dvander@405: 	List<AuthMethod>::iterator iter;
dvander@405: 	for (iter=m_AuthMethods.begin();
dvander@405: 		 iter!=m_AuthMethods.end();
dvander@405: 		 iter++)
dvander@405: 	{
dvander@405: 		sm_trie_destroy((*iter).table);
dvander@405: 	}
dvander@405: 
dvander@405: 	sm_trie_destroy(m_pAuthTables);
dvander@405: 
dvander@335: 	delete m_pStrings;
dvander@335: }
dvander@335: 
dvander@404: void AdminCache::OnSourceModStartup(bool late)
dvander@404: {
dvander@404: 	RegisterAuthIdentType("steam");
dvander@404: 	RegisterAuthIdentType("name");
dvander@404: 	RegisterAuthIdentType("ip");
dvander@404: }
dvander@404: 
dvander@335: void AdminCache::OnSourceModAllInitialized()
dvander@335: {
dvander@336: 	m_pCacheFwd = g_Forwards.CreateForward("OnRebuildAdminCache", ET_Ignore, 1, NULL, Param_Cell);
dvander@335: 	g_ShareSys.AddInterface(NULL, this);
dvander@335: }
dvander@335: 
dvander@336: void AdminCache::OnSourceModShutdown()
dvander@336: {
dvander@336: 	g_Forwards.ReleaseForward(m_pCacheFwd);
dvander@336: 	m_pCacheFwd = NULL;
dvander@336: }
dvander@336: 
dvander@673: void AdminCache::OnSourceModPluginsLoaded()
dvander@673: {
dvander@673: 	DumpAdminCache(AdminCache_Overrides, true);
dvander@673: 	DumpAdminCache(AdminCache_Groups, true);
dvander@673: }
dvander@673: 
dvander@433: void AdminCache::AddCommandOverride(const char *cmd, OverrideType type, FlagBits flags)
dvander@335: {
dvander@433: 	Trie *pTrie = NULL;
dvander@335: 	if (type == Override_Command)
dvander@335: 	{
dvander@433: 		pTrie = m_pCmdOverrides;
dvander@335: 	} else if (type == Override_CommandGroup) {
dvander@433: 		pTrie = m_pCmdGrpOverrides;
dvander@433: 	} else {
dvander@433: 		return;
dvander@335: 	}
dvander@433: 
dvander@433: 	sm_trie_insert(pTrie, cmd, (void *)(unsigned int)flags);
dvander@514: 
dvander@514: 	g_ConCmds.UpdateAdminCmdFlags(cmd, type, flags);
dvander@335: }
dvander@335: 
dvander@433: bool AdminCache::GetCommandOverride(const char *cmd, OverrideType type, FlagBits *pFlags)
dvander@335: {
dvander@433: 	Trie *pTrie = NULL;
dvander@433: 
dvander@335: 	if (type == Override_Command)
dvander@335: 	{
dvander@433: 		pTrie = m_pCmdOverrides;
dvander@335: 	} else if (type == Override_CommandGroup) {
dvander@433: 		pTrie = m_pCmdGrpOverrides;
dvander@433: 	} else {
dvander@433: 		return false;
dvander@433: 	}
dvander@433: 
dvander@433: 	void *object;
dvander@433: 	if (sm_trie_retrieve(pTrie, cmd, &object))
dvander@433: 	{
dvander@433: 		if (pFlags)
dvander@433: 		{
dvander@433: 			*pFlags = (FlagBits)object;
dvander@433: 		}
dvander@433: 		return true;
dvander@335: 	}
dvander@335: 
dvander@335: 	return false;
dvander@335: }
dvander@335: 
dvander@335: void AdminCache::UnsetCommandOverride(const char *cmd, OverrideType type)
dvander@335: {
dvander@335: 	if (type == Override_Command)
dvander@335: 	{
dvander@335: 		return _UnsetCommandOverride(cmd);
dvander@335: 	} else if (type == Override_CommandGroup) {
dvander@335: 		return _UnsetCommandGroupOverride(cmd);
dvander@335: 	}
dvander@335: }
dvander@335: 
dvander@335: void AdminCache::_UnsetCommandGroupOverride(const char *group)
dvander@335: {
dvander@335: 	if (!m_pCmdGrpOverrides)
dvander@335: 	{
dvander@335: 		return;
dvander@335: 	}
dvander@335: 
dvander@514: 	sm_trie_delete(m_pCmdGrpOverrides, group);
dvander@335: 
dvander@514: 	g_ConCmds.UpdateAdminCmdFlags(group, Override_CommandGroup, 0);
dvander@335: }
dvander@335: 
dvander@335: void AdminCache::_UnsetCommandOverride(const char *cmd)
dvander@335: {
dvander@335: 	if (!m_pCmdOverrides)
dvander@335: 	{
dvander@335: 		return;
dvander@335: 	}
dvander@335: 
dvander@514: 	sm_trie_delete(m_pCmdOverrides, cmd);
dvander@335: 
dvander@514: 	g_ConCmds.UpdateAdminCmdFlags(cmd, Override_Command, 0);
dvander@335: }
dvander@335: 
dvander@335: void AdminCache::DumpCommandOverrideCache(OverrideType type)
dvander@335: {
dvander@335: 	if (type == Override_Command && m_pCmdOverrides)
dvander@335: 	{
dvander@335: 		sm_trie_clear(m_pCmdOverrides);
dvander@335: 	} else if (type == Override_CommandGroup && m_pCmdGrpOverrides) {
dvander@335: 		sm_trie_clear(m_pCmdGrpOverrides);
dvander@335: 	}
dvander@335: }
dvander@335: 
dvander@404: AdminId AdminCache::CreateAdmin(const char *name)
dvander@404: {
dvander@404: 	AdminId id;
dvander@404: 	AdminUser *pUser;
dvander@404: 
dvander@404: 	if (m_FreeUserList != INVALID_ADMIN_ID)
dvander@404: 	{
dvander@404: 		pUser = (AdminUser *)m_pMemory->GetAddress(m_FreeUserList);
dvander@404: 		assert(pUser->magic == USR_MAGIC_UNSET);
dvander@404: 		id = m_FreeUserList;
dvander@404: 		m_FreeUserList = pUser->next_user;
dvander@404: 	} else {
dvander@404: 		id = m_pMemory->CreateMem(sizeof(AdminUser), (void **)&pUser);
dvander@404: 		pUser->grp_size = 0;
dvander@404: 		pUser->grp_table = -1;
dvander@404: 	}
dvander@404: 
dvander@433: 	pUser->flags = 0;
dvander@433: 	pUser->eflags = 0;
dvander@404: 	pUser->grp_count = 0;
dvander@404: 	pUser->password = -1;
dvander@404: 	pUser->magic = USR_MAGIC_SET;
dvander@405: 	pUser->auth.identidx = -1;
dvander@405: 	pUser->auth.index = 0;
dvander@434: 	pUser->immune_default = false;
dvander@434: 	pUser->immune_global = false;
dvander@404: 
dvander@404: 	if (m_FirstUser == INVALID_ADMIN_ID)
dvander@404: 	{
dvander@404: 		m_FirstUser = id;
dvander@404: 		m_LastUser = id;
dvander@404: 	} else {
dvander@404: 		AdminUser *pPrev = (AdminUser *)m_pMemory->GetAddress(m_LastUser);
dvander@404: 		pPrev->next_user = id;
dvander@404: 		pUser->prev_user = m_LastUser;
dvander@404: 		m_LastUser = id;
dvander@404: 	}
dvander@404: 
dvander@404: 	if (name && name[0] != '\0')
dvander@404: 	{
dvander@404: 		pUser->nameidx = m_pStrings->AddString(name);
dvander@404: 	}
dvander@404: 
dvander@404: 	return id;
dvander@404: }
dvander@404: 
dvander@335: GroupId AdminCache::AddGroup(const char *group_name)
dvander@335: {
dvander@335: 	if (sm_trie_retrieve(m_pGroups, group_name, NULL))
dvander@335: 	{
dvander@335: 		return INVALID_GROUP_ID;
dvander@335: 	}
dvander@335: 
dvander@335: 	GroupId id;
dvander@335: 	AdminGroup *pGroup;
dvander@335: 	if (m_FreeGroupList != INVALID_GROUP_ID)
dvander@335: 	{
dvander@335: 		pGroup = (AdminGroup *)m_pMemory->GetAddress(m_FreeGroupList);
dvander@335: 		assert(pGroup->magic == GRP_MAGIC_UNSET);
dvander@335: 		id = m_FreeGroupList;
dvander@335: 		m_FreeGroupList = pGroup->next_grp;
dvander@335: 	} else {
dvander@335: 		id = m_pMemory->CreateMem(sizeof(AdminGroup), (void **)&pGroup);
dvander@335: 	}
dvander@335: 
dvander@335: 	pGroup->immune_default = false;
dvander@335: 	pGroup->immune_global = false;
dvander@335: 	pGroup->immune_table = -1;
dvander@335: 	pGroup->magic = GRP_MAGIC_SET;
dvander@335: 	pGroup->next_grp = INVALID_GROUP_ID;
dvander@335: 	pGroup->pCmdGrpTable = NULL;
dvander@335: 	pGroup->pCmdTable = NULL;
dvander@433: 	pGroup->addflags = 0;
dvander@335: 
dvander@335: 	if (m_FirstGroup == INVALID_GROUP_ID)
dvander@335: 	{
dvander@335: 		m_FirstGroup = id;
dvander@335: 		m_LastGroup = id;
dvander@335: 		pGroup->prev_grp = INVALID_GROUP_ID;
dvander@335: 	} else {
dvander@335: 		AdminGroup *pPrev = (AdminGroup *)m_pMemory->GetAddress(m_LastGroup);
dvander@335: 		assert(pPrev->magic == GRP_MAGIC_SET);
dvander@335: 		pPrev->next_grp = id;
dvander@335: 		pGroup->prev_grp = m_LastGroup;
dvander@335: 		m_LastGroup = id;
dvander@335: 	}
dvander@335: 
dvander@404: 	pGroup->nameidx = m_pStrings->AddString(group_name);
dvander@404: 
dvander@335: 	sm_trie_insert(m_pGroups, group_name, (void *)id);
dvander@335: 
dvander@335: 	return id;
dvander@335: }
dvander@335: 
dvander@335: GroupId AdminCache::FindGroupByName(const char *group_name)
dvander@335: {
dvander@335: 	void *object;
dvander@335: 
dvander@335: 	if (!sm_trie_retrieve(m_pGroups, group_name, &object))
dvander@335: 	{
dvander@335: 		return INVALID_GROUP_ID;
dvander@335: 	}
dvander@335: 
dvander@335: 	GroupId id = (GroupId)object;
dvander@335: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335: 
dvander@335: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335: 	{
dvander@335: 		return INVALID_GROUP_ID;
dvander@335: 	}
dvander@335: 
dvander@335: 	return id;
dvander@335: }
dvander@335: 
dvander@335: void AdminCache::SetGroupAddFlag(GroupId id, AdminFlag flag, bool enabled)
dvander@335: {
dvander@335: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335: 	{
dvander@335: 		return;
dvander@335: 	}
dvander@335: 
dvander@433: 	if (flag < Admin_Reservation || flag >= AdminFlags_TOTAL)
dvander@335: 	{
dvander@335: 		return;
dvander@335: 	}
dvander@335: 
dvander@436: 	FlagBits bits = (1<<(FlagBits)flag);
dvander@436: 
dvander@436: 	if (enabled)
dvander@436: 	{
dvander@436: 		pGroup->addflags |= bits;
dvander@436: 	} else {
dvander@436: 		pGroup->addflags &= ~bits;
dvander@436: 	}
dvander@335: }
dvander@335: 
dvander@335: bool AdminCache::GetGroupAddFlag(GroupId id, AdminFlag flag)
dvander@335: {
dvander@335: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335: 	{
dvander@335: 		return false;
dvander@335: 	}
dvander@335: 
dvander@433: 	if (flag < Admin_Reservation || flag >= AdminFlags_TOTAL)
dvander@335: 	{
dvander@335: 		return false;
dvander@335: 	}
dvander@335: 
dvander@433: 	FlagBits bit = 1<<(FlagBits)flag;
dvander@433: 	return ((pGroup->addflags & bit) == bit);
dvander@335: }
dvander@335: 
dvander@433: FlagBits AdminCache::GetGroupAddFlags(GroupId id)
dvander@335: {
dvander@335: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335: 	{
dvander@335: 		return 0;
dvander@335: 	}
dvander@335: 
dvander@433: 	return pGroup->addflags;
dvander@335: }
dvander@335: 
dvander@335: void AdminCache::SetGroupGenericImmunity(GroupId id, ImmunityType type, bool enabled)
dvander@335: {
dvander@335: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335: 	{
dvander@335: 		return;
dvander@335: 	}
dvander@335: 
dvander@335: 	if (type == Immunity_Default)
dvander@335: 	{
dvander@335: 		pGroup->immune_default = enabled;
dvander@335: 	} else if (type == Immunity_Global) {
dvander@335: 		pGroup->immune_global = enabled;
dvander@335: 	} 
dvander@335: }
dvander@335: 
dvander@335: bool AdminCache::GetGroupGenericImmunity(GroupId id, ImmunityType type)
dvander@335: {
dvander@335: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335: 	{
dvander@335: 		return false;
dvander@335: 	}
dvander@335: 
dvander@335: 	if (type == Immunity_Default)
dvander@335: 	{
dvander@335: 		return pGroup->immune_default;
dvander@335: 	} else if (type == Immunity_Global) {
dvander@335: 		return pGroup->immune_global;
dvander@335: 	}
dvander@335: 
dvander@335: 	return false;
dvander@335: }
dvander@335: 
dvander@335: void AdminCache::AddGroupImmunity(GroupId id, GroupId other_id)
dvander@335: {
dvander@335: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335: 	{
dvander@335: 		return;
dvander@335: 	}
dvander@335: 
dvander@335: 	AdminGroup *pOther = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335: 	if (!pOther || pOther->magic != GRP_MAGIC_SET)
dvander@335: 	{
dvander@335: 		return;
dvander@335: 	}
dvander@335: 
dvander@335: 	/* We always need to resize the immunity table */
dvander@335: 	int *table, tblidx;
dvander@335: 	if (pOther->immune_table == -1)
dvander@335: 	{
dvander@335: 		tblidx = m_pMemory->CreateMem(sizeof(int) * 2, (void **)&table);
dvander@335: 		table[0] = 0;
dvander@335: 	} else {
dvander@335: 		int *old_table = (int *)m_pMemory->GetAddress(pOther->immune_table);
dvander@433: 		/* Break out if this group is already in the list */
dvander@433: 		for (int i=0; i<old_table[0]; i++)
dvander@433: 		{
dvander@433: 			if (old_table[1+i] == other_id)
dvander@433: 			{
dvander@433: 				return;
dvander@433: 			}
dvander@433: 		}
dvander@335: 		tblidx = m_pMemory->CreateMem(sizeof(int) * (old_table[0] + 2), (void **)&table);
dvander@433: 		/* Get the old address again in case of resize */
dvander@335: 		old_table = (int *)m_pMemory->GetAddress(pOther->immune_table);
dvander@335: 		table[0] = old_table[0];
dvander@335: 		for (unsigned int i=1; i<=(unsigned int)old_table[0]; i++)
dvander@335: 		{
dvander@335: 			table[i] = old_table[i];
dvander@335: 		}
dvander@335: 	}
dvander@335: 
dvander@335: 	/* Assign */
dvander@335: 	pOther->immune_table = tblidx;
dvander@335: 
dvander@335: 	/* Add to the array */
dvander@335: 	table[0]++;
dvander@335: 	table[table[0]] = other_id;
dvander@335: }
dvander@335: 
dvander@335: unsigned int AdminCache::GetGroupImmunityCount(GroupId id)
dvander@335: {
dvander@335: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335: 	{
dvander@335: 		return 0;
dvander@335: 	}
dvander@335: 
dvander@335: 	if (pGroup->immune_table == -1)
dvander@335: 	{
dvander@335: 		return 0;
dvander@335: 	}
dvander@335: 
dvander@335: 	int *table = (int *)m_pMemory->GetAddress(pGroup->immune_table);
dvander@335: 
dvander@335: 	return table[0];
dvander@335: }
dvander@335: 
dvander@335: GroupId AdminCache::GetGroupImmunity(GroupId id, unsigned int number)
dvander@335: {
dvander@335: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335: 	{
dvander@335: 		return INVALID_GROUP_ID;
dvander@335: 	}
dvander@335: 
dvander@335: 	if (pGroup->immune_table == -1)
dvander@335: 	{
dvander@335: 		return INVALID_GROUP_ID;
dvander@335: 	}
dvander@335: 
dvander@335: 	int *table = (int *)m_pMemory->GetAddress(pGroup->immune_table);
dvander@335: 	if (number >= (unsigned int)table[0])
dvander@335: 	{
dvander@335: 		return INVALID_GROUP_ID;
dvander@335: 	}
dvander@335: 
dvander@335: 	return table[1 + number];
dvander@335: }
dvander@335: 
dvander@335: void AdminCache::AddGroupCommandOverride(GroupId id, const char *name, OverrideType type, OverrideRule rule)
dvander@335: {
dvander@335: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335: 	{
dvander@335: 		return;
dvander@335: 	}
dvander@335: 
dvander@335: 	Trie *pTrie = NULL;
dvander@335: 	if (type == Override_Command)
dvander@335: 	{
dvander@335: 		if (pGroup->pCmdTable == NULL)
dvander@335: 		{
dvander@335: 			pGroup->pCmdTable = sm_trie_create();
dvander@335: 		}
dvander@335: 		pTrie = pGroup->pCmdTable;
dvander@335: 	} else if (type == Override_CommandGroup) {
dvander@335: 		if (pGroup->pCmdGrpTable == NULL)
dvander@335: 		{
dvander@335: 			pGroup->pCmdGrpTable = sm_trie_create();
dvander@335: 		}
dvander@335: 		pTrie = pGroup->pCmdGrpTable;
dvander@335: 	} else {
dvander@335: 		return;
dvander@335: 	}
dvander@335: 
dvander@335: 	sm_trie_insert(pTrie, name, (void *)(int)rule);
dvander@335: }
dvander@335: 
dvander@335: bool AdminCache::GetGroupCommandOverride(GroupId id, const char *name, OverrideType type, OverrideRule *pRule)
dvander@335: {
dvander@335: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335: 	{
dvander@335: 		return false;
dvander@335: 	}
dvander@335: 
dvander@335: 	Trie *pTrie = NULL;
dvander@335: 	if (type == Override_Command)
dvander@335: 	{
dvander@335: 		if (pGroup->pCmdTable == NULL)
dvander@335: 		{
dvander@335: 			return false;
dvander@335: 		}
dvander@335: 		pTrie = pGroup->pCmdTable;
dvander@335: 	} else if (type == Override_CommandGroup) {
dvander@335: 		if (pGroup->pCmdGrpTable == NULL)
dvander@335: 		{
dvander@335: 			return false;
dvander@335: 		}
dvander@335: 		pTrie = pGroup->pCmdGrpTable;
dvander@335: 	} else {
dvander@335: 		return false;
dvander@335: 	}
dvander@335: 
dvander@335: 	void *object;
dvander@335: 	if (!sm_trie_retrieve(pTrie, name, &object))
dvander@335: 	{
dvander@335: 		return false;
dvander@335: 	}
dvander@335: 
dvander@335: 	if (pRule)
dvander@335: 	{
dvander@335: 		*pRule = (OverrideRule)(int)object;
dvander@335: 	}
dvander@335: 	
dvander@335: 	return true;
dvander@335: }
dvander@335: 
dvander@404: Trie *AdminCache::GetMethodByIndex(unsigned int index)
dvander@404: {
dvander@404: 	List<AuthMethod>::iterator iter;
dvander@404: 	for (iter=m_AuthMethods.begin();
dvander@404: 		 iter!=m_AuthMethods.end();
dvander@404: 		 iter++)
dvander@404: 	{
dvander@404: 		if (index-- == 0)
dvander@404: 		{
dvander@404: 			return (*iter).table;
dvander@404: 		}
dvander@404: 	}
dvander@404: 
dvander@404: 	return NULL;
dvander@404: }
dvander@404: 
dvander@404: bool AdminCache::InvalidateAdmin(AdminId id)
dvander@404: {
dvander@404: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404: 	AdminUser *pOther;
dvander@404: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404: 	{
dvander@404: 		return false;
dvander@404: 	}
dvander@404: 
dvander@520: 	if (!m_InvalidatingAdmins && !m_destroying)
dvander@458: 	{
dvander@458: 		g_Players.ClearAdminId(id);
dvander@458: 	}
dvander@458: 
dvander@404: 	/* Unlink from the dbl link list */
dvander@404: 	if (id == m_FirstUser && id == m_LastUser)
dvander@404: 	{
dvander@404: 		m_FirstUser = INVALID_ADMIN_ID;
dvander@404: 		m_LastUser = INVALID_ADMIN_ID;
dvander@404: 	} else if (id == m_FirstUser) {
dvander@404: 		m_FirstUser = pUser->next_user;
dvander@404: 		pOther = (AdminUser *)m_pMemory->GetAddress(m_FirstUser);
dvander@404: 		pOther->prev_user = INVALID_ADMIN_ID;
dvander@404: 	} else if (id == m_LastUser) {
dvander@404: 		m_LastUser = pUser->prev_user;
dvander@404: 		pOther = (AdminUser *)m_pMemory->GetAddress(m_LastUser);
dvander@404: 		pOther->next_user = INVALID_ADMIN_ID;
dvander@404: 	} else {
dvander@404: 		pOther = (AdminUser *)m_pMemory->GetAddress(pUser->prev_user);
dvander@404: 		pOther->next_user = pUser->next_user;
dvander@404: 		pOther = (AdminUser *)m_pMemory->GetAddress(pUser->next_user);
dvander@404: 		pOther->prev_user = pUser->prev_user;
dvander@404: 	}
dvander@404: 
dvander@404: 	/* Unlink from auth tables */
dvander@405: 	if (pUser->auth.identidx != -1)
dvander@404: 	{
dvander@405: 		Trie *pTrie = GetMethodByIndex(pUser->auth.index);
dvander@405: 		if (pTrie)
dvander@404: 		{
dvander@405: 			sm_trie_delete(pTrie, m_pStrings->GetString(pUser->auth.identidx));
dvander@404: 		}
dvander@404: 	}
dvander@404: 
dvander@404: 	/* Clear table counts */
dvander@404: 	pUser->grp_count = 0;
dvander@405: 	
dvander@404: 	/* Link into free list */
dvander@404: 	pUser->magic = USR_MAGIC_UNSET;
dvander@404: 	pUser->next_user = m_FreeUserList;
dvander@404: 	m_FreeUserList = id;
dvander@404: 
dvander@404: 	return true;
dvander@404: }
dvander@404: 
dvander@404: 
dvander@335: void AdminCache::InvalidateGroup(GroupId id)
dvander@335: {
dvander@335: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335: 	AdminGroup *pOther;
dvander@335: 
dvander@335: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335: 	{
dvander@335: 		return;
dvander@335: 	}
dvander@335: 
dvander@335: 	const char *str = m_pStrings->GetString(pGroup->nameidx);
dvander@335: 	sm_trie_delete(m_pGroups, str);
dvander@335: 
dvander@335: 	/* Unlink from the live dbllink list */
dvander@335: 	if (id == m_FirstGroup && id == m_LastGroup)
dvander@335: 	{
dvander@335: 		m_LastGroup = INVALID_GROUP_ID;
dvander@335: 		m_FirstGroup = INVALID_GROUP_ID;
dvander@335: 	} else if (id == m_FirstGroup) {
dvander@335: 		m_FirstGroup = pGroup->next_grp;
dvander@335: 		pOther = (AdminGroup *)m_pMemory->GetAddress(m_FirstGroup);
dvander@335: 		pOther->prev_grp = INVALID_GROUP_ID;
dvander@335: 	} else if (id == m_LastGroup) {
dvander@335: 		m_LastGroup = pGroup->prev_grp;
dvander@335: 		pOther = (AdminGroup *)m_pMemory->GetAddress(m_LastGroup);
dvander@335: 		pOther->next_grp = INVALID_GROUP_ID;
dvander@335: 	} else {
dvander@335: 		pOther = (AdminGroup *)m_pMemory->GetAddress(pGroup->prev_grp);
dvander@335: 		pOther->next_grp = pGroup->next_grp;
dvander@335: 		pOther = (AdminGroup *)m_pMemory->GetAddress(pGroup->next_grp);
dvander@335: 		pOther->prev_grp = pGroup->prev_grp;
dvander@335: 	}
dvander@335: 
dvander@335: 	/* Free any used memory to be safe */
dvander@335: 	if (pGroup->pCmdGrpTable)
dvander@335: 	{
dvander@335: 		sm_trie_destroy(pGroup->pCmdGrpTable);
dvander@335: 		pGroup->pCmdGrpTable = NULL;
dvander@335: 	}
dvander@335: 	if (pGroup->pCmdTable)
dvander@335: 	{
dvander@335: 		sm_trie_destroy(pGroup->pCmdTable);
dvander@335: 		pGroup->pCmdTable = NULL;
dvander@335: 	}
dvander@335: 
dvander@335: 	/* Link into the free list */
dvander@335: 	pGroup->magic = GRP_MAGIC_UNSET;
dvander@335: 	pGroup->next_grp = m_FreeGroupList;
dvander@335: 	m_FreeGroupList = id;
dvander@335: 
dvander@404: 	int idx = m_FirstUser;
dvander@404: 	AdminUser *pUser;
dvander@404: 	int *table;
dvander@404: 	while (idx != INVALID_ADMIN_ID)
dvander@404: 	{
dvander@404: 		pUser = (AdminUser *)m_pMemory->GetAddress(idx);
dvander@404: 		if (pUser->grp_count > 0)
dvander@404: 		{
dvander@404: 			table = (int *)m_pMemory->GetAddress(pUser->grp_table);
dvander@404: 			for (unsigned int i=0; i<pUser->grp_count; i++)
dvander@404: 			{
dvander@404: 				if (table[i] == id)
dvander@404: 				{
dvander@404: 					/* We have to remove this entry */
dvander@404: 					for (unsigned int j=i+1; j<pUser->grp_count; j++)
dvander@404: 					{
dvander@404: 						/* Move everything down by one */
dvander@404: 						table[j-1] = table[j];
dvander@404: 					}
dvander@404: 					/* Decrease count */
dvander@404: 					pUser->grp_count--;
dvander@404: 					/* Recalculate effective flags */
dvander@433: 					pUser->eflags = pUser->flags;
dvander@404: 					for (unsigned int j=0; j<pUser->grp_count; j++)
dvander@404: 					{
dvander@404: 						pOther = (AdminGroup *)m_pMemory->GetAddress(table[j]);
dvander@433: 						pUser->eflags |= pOther->addflags;
dvander@404: 					}
dvander@404: 					/* Break now, duplicates aren't allowed */
dvander@404: 					break;
dvander@404: 				}
dvander@404: 			}
dvander@404: 		}
dvander@404: 		idx = pUser->next_user;
dvander@404: 	}
dvander@335: }
dvander@335: 
dvander@335: void AdminCache::InvalidateGroupCache()
dvander@335: {
dvander@335: 	/* Nuke the free list */
dvander@335: 	m_FreeGroupList = -1;
dvander@335: 
dvander@335: 	/* Nuke reverse lookups */
dvander@335: 	sm_trie_clear(m_pGroups);
dvander@335: 
dvander@335: 	/* Free memory on groups */
dvander@335: 	GroupId cur = m_FirstGroup;
dvander@335: 	AdminGroup *pGroup;
dvander@335: 	while (cur != INVALID_GROUP_ID)
dvander@335: 	{
dvander@335: 		pGroup = (AdminGroup *)m_pMemory->GetAddress(cur);
dvander@335: 		assert(pGroup->magic == GRP_MAGIC_SET);
dvander@335: 		if (pGroup->pCmdGrpTable)
dvander@335: 		{
dvander@335: 			sm_trie_destroy(pGroup->pCmdGrpTable);
dvander@335: 		}
dvander@335: 		if (pGroup->pCmdTable)
dvander@335: 		{
dvander@335: 			sm_trie_destroy(pGroup->pCmdTable);
dvander@335: 		}
dvander@335: 		cur = pGroup->next_grp;
dvander@335: 	}
dvander@335: 
dvander@335: 	m_FirstGroup = INVALID_GROUP_ID;
dvander@335: 	m_LastGroup = INVALID_GROUP_ID;
dvander@335: 
dvander@404: 	InvalidateAdminCache(false);
dvander@335: 
dvander@335: 	/* Reset the memory table */
dvander@335: 	m_pMemory->Reset();
dvander@335: }
dvander@336: 
dvander@336: void AdminCache::AddAdminListener(IAdminListener *pListener)
dvander@336: {
dvander@336: 	m_hooks.push_back(pListener);
dvander@336: }
dvander@336: 
dvander@336: void AdminCache::RemoveAdminListener(IAdminListener *pListener)
dvander@336: {
dvander@336: 	m_hooks.remove(pListener);
dvander@336: }
dvander@336: 
dvander@404: void AdminCache::RegisterAuthIdentType(const char *name)
dvander@404: {
dvander@404: 	if (sm_trie_retrieve(m_pAuthTables, name, NULL))
dvander@404: 	{
dvander@404: 		return;
dvander@404: 	}
dvander@404: 
dvander@404: 	Trie *pAuth = sm_trie_create();
dvander@404: 
dvander@404: 	AuthMethod method;
dvander@404: 	method.name.assign(name);
dvander@404: 	method.table = pAuth;
dvander@404: 
dvander@404: 	m_AuthMethods.push_back(method);
dvander@404: 
dvander@404: 	sm_trie_insert(m_pAuthTables, name, pAuth);
dvander@404: }
dvander@404: 
dvander@404: void AdminCache::InvalidateAdminCache(bool unlink_admins)
dvander@404: {
dvander@458: 	m_InvalidatingAdmins = true;
dvander@520: 	if (!m_destroying)
dvander@520: 	{
dvander@520: 		g_Players.ClearAllAdmins();
dvander@520: 	}
dvander@404: 	/* Wipe the identity cache first */
dvander@404: 	List<AuthMethod>::iterator iter;
dvander@404: 	for (iter=m_AuthMethods.begin();
dvander@404: 		 iter!=m_AuthMethods.end();
dvander@404: 		 iter++)
dvander@404: 	{
dvander@404: 		sm_trie_clear((*iter).table);
dvander@404: 	}
dvander@404: 	
dvander@405: 	if (unlink_admins)
dvander@404: 	{
dvander@405: 		while (m_FirstUser != INVALID_ADMIN_ID)
dvander@405: 		{
dvander@405: 			InvalidateAdmin(m_FirstUser);
dvander@405: 		}
dvander@405: 	} else {
dvander@405: 		m_FirstUser = -1;
dvander@405: 		m_LastUser = -1;
dvander@405: 		m_FreeUserList = -1;
dvander@404: 	}
dvander@458: 	m_InvalidatingAdmins = false;
dvander@404: }
dvander@404: 
dvander@438: void AdminCache::DumpAdminCache(AdminCachePart part, bool rebuild)
dvander@336: {
dvander@438: 	List<IAdminListener *>::iterator iter;
dvander@438: 	IAdminListener *pListener;
dvander@438: 	cell_t result;
dvander@438: 
dvander@438: 	if (part == AdminCache_Overrides)
dvander@336: 	{
dvander@336: 		DumpCommandOverrideCache(Override_Command);
dvander@336: 		DumpCommandOverrideCache(Override_CommandGroup);
dvander@520: 		if (rebuild && !m_destroying)
dvander@438: 		{
dvander@438: 			for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
dvander@438: 			{
dvander@438: 				pListener = (*iter);
dvander@438: 				pListener->OnRebuildOverrideCache();
dvander@438: 			}
dvander@438: 			m_pCacheFwd->PushCell(part);
dvander@438: 			m_pCacheFwd->Execute(&result);
dvander@438: 		}
dvander@438: 	} else if (part == AdminCache_Groups || part == AdminCache_Admins) {
dvander@438: 		if (part == AdminCache_Groups)
dvander@438: 		{
dvander@438: 			InvalidateGroupCache();
dvander@520: 			if (rebuild && !m_destroying)
dvander@438: 			{
dvander@438: 				for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
dvander@438: 				{
dvander@438: 					pListener = (*iter);
dvander@438: 					pListener->OnRebuildGroupCache();
dvander@438: 				}
dvander@438: 				m_pCacheFwd->PushCell(part);
dvander@438: 				m_pCacheFwd->Execute(&result);
dvander@438: 			}
dvander@438: 		}
dvander@404: 		InvalidateAdminCache(true);
dvander@520: 		if (rebuild && !m_destroying)
dvander@336: 		{
dvander@438: 			for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
dvander@438: 			{
dvander@438: 				pListener = (*iter);
dvander@438: 				pListener->OnRebuildAdminCache((part == AdminCache_Groups));
dvander@438: 			}
dvander@438: 			m_pCacheFwd->PushCell(AdminCache_Admins);
dvander@438: 			m_pCacheFwd->Execute(&result);
dvander@1165: 			g_Players.RecheckAnyAdmins();
dvander@336: 		}
dvander@336: 	}
dvander@336: }
dvander@404: 
dvander@404: const char *AdminCache::GetAdminName(AdminId id)
dvander@404: {
dvander@404: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404: 	{
dvander@404: 		return NULL;
dvander@404: 	}
dvander@404: 
dvander@404: 	return m_pStrings->GetString(pUser->nameidx);
dvander@404: }
dvander@404: 
dvander@405: bool AdminCache::GetMethodIndex(const char *name, unsigned int *_index)
dvander@405: {
dvander@405: 	List<AuthMethod>::iterator iter;
dvander@405: 	unsigned int index = 0;
dvander@405: 	for (iter=m_AuthMethods.begin();
dvander@405: 		 iter!=m_AuthMethods.end();
dvander@405: 		 iter++,index++)
dvander@405: 	{
dvander@405: 		if ((*iter).name.compare(name) == 0)
dvander@405: 		{
dvander@405: 			*_index = index;
dvander@405: 			return true;
dvander@405: 		}
dvander@405: 	}
dvander@405: 
dvander@405: 	return false;
dvander@405: }
dvander@405: 
dvander@404: bool AdminCache::BindAdminIdentity(AdminId id, const char *auth, const char *ident)
dvander@404: {
dvander@404: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404: 	{
dvander@404: 		return false;
dvander@404: 	}
dvander@404: 
dvander@404: 	Trie *pTable;
dvander@404: 	if (!sm_trie_retrieve(m_pAuthTables, auth, (void **)&pTable))
dvander@404: 	{
dvander@404: 		return false;
dvander@404: 	}
dvander@404: 
dvander@404: 	if (sm_trie_retrieve(pTable, ident, NULL))
dvander@404: 	{
dvander@404: 		return false;
dvander@404: 	}
dvander@404: 
dvander@405: 	pUser->auth.identidx = m_pStrings->AddString(ident);
dvander@405: 	GetMethodIndex(auth, &pUser->auth.index);
dvander@405: 
dvander@404: 	return sm_trie_insert(pTable, ident, (void **)id);
dvander@404: }
dvander@404: 
dvander@404: AdminId AdminCache::FindAdminByIdentity(const char *auth, const char *identity)
dvander@404: {
dvander@404: 	Trie *pTable;
dvander@404: 	if (!sm_trie_retrieve(m_pAuthTables, auth, (void **)&pTable))
dvander@404: 	{
dvander@404: 		return INVALID_ADMIN_ID;
dvander@404: 	}
dvander@404: 
dvander@404: 	void *object;
dvander@404: 	if (!sm_trie_retrieve(pTable, identity, &object))
dvander@404: 	{
dvander@404: 		return INVALID_ADMIN_ID;
dvander@404: 	}
dvander@404: 
dvander@404: 	return (AdminId)object;
dvander@404: }
dvander@404: 
dvander@404: void AdminCache::SetAdminFlag(AdminId id, AdminFlag flag, bool enabled)
dvander@404: {
dvander@404: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404: 	{
dvander@404: 		return;
dvander@404: 	}
dvander@404: 
dvander@433: 	if (flag < Admin_Reservation
dvander@404: 		|| flag >= AdminFlags_TOTAL)
dvander@404: 	{
dvander@404: 		return;
dvander@404: 	}
dvander@436: 	
dvander@436: 	FlagBits bits = (1<<(FlagBits)flag);
dvander@404: 
dvander@436: 	if (enabled)
dvander@436: 	{
dvander@436: 		pUser->flags |= bits;
dvander@436: 		pUser->eflags |= bits;
dvander@436: 	} else {
dvander@436: 		pUser->flags &= ~bits;
dvander@436: 		pUser->eflags &= ~bits;
dvander@436: 	}
dvander@404: }
dvander@404: 
dvander@404: bool AdminCache::GetAdminFlag(AdminId id, AdminFlag flag, AccessMode mode)
dvander@404: {
dvander@404: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404: 	{
dvander@404: 		return false;
dvander@404: 	}
dvander@404: 
dvander@433: 	if (flag < Admin_Reservation
dvander@404: 		|| flag >= AdminFlags_TOTAL)
dvander@404: 	{
dvander@404: 		return false;
dvander@404: 	}
dvander@404: 
dvander@433: 	FlagBits bit = (1<<(FlagBits)flag);
dvander@433: 
dvander@404: 	if (mode == Access_Real)
dvander@404: 	{
dvander@433: 		return ((pUser->flags & bit) == bit);
dvander@404: 	} else if (mode == Access_Effective) {
dvander@907: 		bool has_bit = ((pUser->eflags & bit) == bit);
dvander@907: 		if (!has_bit && flag != Admin_Root && ((pUser->eflags & ADMFLAG_ROOT) == ADMFLAG_ROOT))
dvander@907: 		{
dvander@907: 			has_bit = true;
dvander@907: 		}
dvander@907: 		return has_bit;
dvander@404: 	}
dvander@404: 
dvander@404: 	return false;
dvander@404: }
dvander@404: 
dvander@433: FlagBits AdminCache::GetAdminFlags(AdminId id, AccessMode mode)
dvander@404: {
dvander@404: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404: 	{
dvander@404: 		return 0;
dvander@404: 	}
dvander@404: 
dvander@404: 	if (mode == Access_Real)
dvander@404: 	{
dvander@433: 		return pUser->flags;
dvander@404: 	} else if (mode == Access_Effective) {
dvander@433: 		return pUser->eflags;
dvander@404: 	}
dvander@404: 
dvander@433: 	return 0;
dvander@404: }
dvander@404: 
dvander@458: void AdminCache::SetAdminFlags(AdminId id, AccessMode mode, FlagBits bits)
dvander@458: {
dvander@458: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@458: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@458: 	{
dvander@458: 		return;
dvander@458: 	}
dvander@458: 
dvander@458: 	if (mode == Access_Real)
dvander@458: 	{
dvander@458: 		pUser->flags = bits;
dvander@458: 		pUser->eflags = bits;
dvander@458: 	} else if (mode == Access_Effective) {
dvander@458: 		pUser->eflags = bits;
dvander@458: 	}
dvander@458: }
dvander@458: 
dvander@404: bool AdminCache::AdminInheritGroup(AdminId id, GroupId gid)
dvander@404: {
dvander@404: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404: 	{
dvander@404: 		return false;
dvander@404: 	}
dvander@404: 
dvander@404: 	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid);
dvander@405: 	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@404: 	{
dvander@404: 		return false;
dvander@404: 	}
dvander@404: 
dvander@433: 	/* First check for duplicates */
dvander@433: 	if (pUser->grp_count != 0)
dvander@433: 	{
dvander@433: 		int *temp_table = (int *)m_pMemory->GetAddress(pUser->grp_table);
dvander@433: 		for (unsigned int i=0; i<pUser->grp_count; i++)
dvander@433: 		{
dvander@433: 			if (temp_table[i] == gid)
dvander@433: 			{
dvander@433: 				return false;
dvander@433: 			}
dvander@433: 		}
dvander@433: 	}
dvander@433: 
dvander@404: 	int *table;
dvander@404: 	if (pUser->grp_count + 1 > pUser->grp_size)
dvander@404: 	{
dvander@404: 		int new_size = 0;
dvander@404: 		int tblidx;
dvander@404: 		
dvander@404: 		if (pUser->grp_size == 0)
dvander@404: 		{
dvander@404: 			new_size = 2;
dvander@404: 		} else {
dvander@404: 			new_size = pUser->grp_size * 2;
dvander@404: 		}
dvander@404: 		
dvander@404: 		/* Create and refresh pointers */
dvander@404: 		tblidx = m_pMemory->CreateMem(new_size * sizeof(int), (void **)&table);
dvander@404: 		pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404: 		pGroup = (AdminGroup *)m_pMemory->GetAddress(gid);
dvander@404: 
dvander@404: 		/* Copy old data if necessary */
dvander@404: 		if (pUser->grp_table != -1)
dvander@404: 		{
dvander@404: 			int *old_table = (int *)m_pMemory->GetAddress(pUser->grp_table);
dvander@404: 			memcpy(table, old_table, sizeof(int) * pUser->grp_count);
dvander@404: 		}
dvander@404: 		pUser->grp_table = tblidx;
dvander@404: 		pUser->grp_size = new_size;
dvander@404: 	} else {
dvander@404: 		table = (int *)m_pMemory->GetAddress(pUser->grp_table);
dvander@404: 	}
dvander@404: 
dvander@404: 	table[pUser->grp_count] = gid;
dvander@404: 	pUser->grp_count++;
dvander@404: 
dvander@434: 	/* Compute new effective permissions */
dvander@433: 	pUser->eflags |= pGroup->addflags;
dvander@404: 
dvander@434: 	if (pGroup->immune_default)
dvander@434: 	{
dvander@434: 		pUser->immune_default = true;
dvander@434: 	}
dvander@434: 	if (pGroup->immune_global)
dvander@434: 	{
dvander@434: 		pUser->immune_global = true;
dvander@434: 	}
dvander@434: 
dvander@404: 	return true;
dvander@404: }
dvander@404: 
dvander@458: bool AdminCache::IsValidAdmin(AdminId id)
dvander@458: {
dvander@458: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@458: 	return (pUser != NULL && pUser->magic == USR_MAGIC_SET);
dvander@458: }
dvander@458: 
dvander@404: unsigned int AdminCache::GetAdminGroupCount(AdminId id)
dvander@404: {
dvander@404: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404: 	{
dvander@404: 		return 0;
dvander@404: 	}
dvander@404: 
dvander@404: 	return pUser->grp_count;
dvander@404: }
dvander@404: 
dvander@404: GroupId AdminCache::GetAdminGroup(AdminId id, unsigned int index, const char **name)
dvander@404: {
dvander@404: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404: 	if (!pUser || pUser->magic != USR_MAGIC_SET || index >= pUser->grp_count)
dvander@404: 	{
dvander@404: 		return INVALID_GROUP_ID;
dvander@404: 	}
dvander@404: 
dvander@404: 	int *table = (int *)m_pMemory->GetAddress(pUser->grp_table);
dvander@404: 
dvander@404: 	return table[index];
dvander@404: }
dvander@404: 
dvander@404: const char *AdminCache::GetAdminPassword(AdminId id)
dvander@404: {
dvander@404: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404: 	{
dvander@404: 		return NULL;
dvander@404: 	}
dvander@404: 
dvander@404: 	return m_pStrings->GetString(pUser->password);
dvander@404: }
dvander@404: 
dvander@404: void AdminCache::SetAdminPassword(AdminId id, const char *password)
dvander@404: {	
dvander@404: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404: 	{
dvander@404: 		return;
dvander@404: 	}
dvander@404: 
dvander@404: 	pUser->password = m_pStrings->AddString(password);
dvander@404: }
dvander@404: 
dvander@433: unsigned int AdminCache::FlagBitsToBitArray(FlagBits bits, bool array[], unsigned int maxSize)
dvander@433: {
dvander@433: 	unsigned int i;
dvander@433: 	for (i=0; i<maxSize && i<AdminFlags_TOTAL; i++)
dvander@433: 	{
dvander@464: 		array[i] = ((bits & (1<<i)) == (unsigned)(1<<i));
dvander@433: 	}
dvander@433: 
dvander@433: 	return i;
dvander@433: }
dvander@433: 
dvander@433: FlagBits AdminCache::FlagBitArrayToBits(const bool array[], unsigned int maxSize)
dvander@433: {
dvander@433: 	FlagBits bits = 0;
dvander@433: 	for (unsigned int i=0; i<maxSize && i<AdminFlags_TOTAL; i++)
dvander@433: 	{
dvander@433: 		if (array[i])
dvander@433: 		{
dvander@433: 			bits |= (1<<i);
dvander@433: 		}
dvander@433: 	}
dvander@433: 
dvander@433: 	return bits;
dvander@433: }
dvander@433: 
dvander@433: FlagBits AdminCache::FlagArrayToBits(const AdminFlag array[], unsigned int numFlags)
dvander@433: {
dvander@433: 	FlagBits bits = 0;
dvander@433: 	for (unsigned int i=0; i<numFlags && i<AdminFlags_TOTAL; i++)
dvander@433: 	{
dvander@433: 		bits |= (1 << (FlagBits)array[i]);
dvander@433: 	}
dvander@433: 
dvander@433: 	return bits;
dvander@433: }
dvander@433: 
dvander@433: unsigned int AdminCache::FlagBitsToArray(FlagBits bits, AdminFlag array[], unsigned int maxSize)
dvander@433: {
dvander@433: 	unsigned int i, num=0;
dvander@1050: 	for (i=0; num<maxSize && i<AdminFlags_TOTAL; i++)
dvander@433: 	{
dvander@464: 		if ((bits & (1<<i)) == (unsigned)(1<<i))
dvander@433: 		{
dvander@433: 			array[num++] = (AdminFlag)i;
dvander@433: 		}
dvander@433: 	}
dvander@433: 
dvander@433: 	return num;
dvander@433: }
dvander@434: 
dvander@434: bool AdminCache::CheckAdminFlags(AdminId id, FlagBits bits)
dvander@434: {
dvander@434: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@434: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@434: 	{
dvander@434: 		return false;
dvander@434: 	}
dvander@434: 
dvander@434: 	return ((pUser->eflags & bits) == bits);
dvander@434: }
dvander@434: 
dvander@434: bool AdminCache::CanAdminTarget(AdminId id, AdminId target)
dvander@434: {
dvander@434: 	/** 
dvander@434: 	 * Zeroth, if the targeting AdminId is INVALID_ADMIN_ID, targeting fails.
ds@729: 	 * First, if the targeted AdminId is INVALID_ADMIN_ID, targeting succeeds.
dvander@434: 	 */
dvander@434: 
dvander@434: 	if (id == INVALID_ADMIN_ID)
dvander@434: 	{
dvander@434: 		return false;
dvander@434: 	}
dvander@434: 
dvander@434: 	if (target == INVALID_ADMIN_ID)
dvander@434: 	{
dvander@434: 		return true;
dvander@434: 	}
dvander@434: 
dvander@434: 	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@434: 	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@434: 	{
dvander@434: 		return false;
dvander@434: 	}
dvander@434: 
dvander@434: 	AdminUser *pTarget = (AdminUser *)m_pMemory->GetAddress(target);
dvander@434: 	if (!pTarget || pTarget->magic != USR_MAGIC_SET)
dvander@434: 	{
dvander@434: 		return false;
dvander@434: 	}
dvander@434: 
dvander@434: 	/** 
ds@729: 	 * Second, if the targeting admin is root, targeting succeeds.
dvander@434: 	 */
dvander@434: 	if (pUser->eflags & ADMFLAG_ROOT)
dvander@434: 	{
dvander@434: 		return true;
dvander@434: 	}
dvander@434: 
ds@729: 	/** Fourth, if the targeted admin has global immunity, targeting fails. */
dvander@434: 	if (pTarget->immune_global)
dvander@434: 	{
dvander@434: 		return false;
dvander@434: 	}
dvander@434: 
dvander@434: 	/** 
ds@729: 	 * Fifth, if the targeted admin has default immunity 
dvander@434: 	 *  and the admin belongs to no groups, targeting fails.
dvander@434: 	 */
dvander@434: 	if (pTarget->immune_default && pUser->grp_count < 1)
dvander@434: 	{
dvander@434: 		return false;
dvander@434: 	}
dvander@434: 
dvander@434: 	/** 
ds@729: 	 * Sixth, if the targeted admin has specific immunity from the
dvander@434: 	 *  targeting admin via group immunities, targeting fails.
dvander@434: 	 */
dvander@434: 	//:TODO: speed this up... maybe with trie hacks.
dvander@434: 	//idea is to insert %d.%d in the trie after computing this and use it as a cache lookup.
dvander@434: 	//problem is the trie cannot delete prefixes, so we'd have a problem with invalidations.
dvander@434: 	if (pTarget->grp_count > 0 && pUser->grp_count > 0)
dvander@434: 	{
dvander@434: 		int *grp_table = (int *)m_pMemory->GetAddress(pTarget->grp_table);
dvander@434: 		int *src_table = (int *)m_pMemory->GetAddress(pUser->grp_table);
dvander@434: 		GroupId id, other;
dvander@434: 		unsigned int num;
dvander@434: 		for (unsigned int i=0; i<pTarget->grp_count; i++)
dvander@434: 		{
dvander@434: 			id = grp_table[i];
dvander@434: 			num = GetGroupImmunityCount(id);
dvander@434: 			for (unsigned int j=0; j<num; i++)
dvander@434: 			{
dvander@434: 				other = GetGroupImmunity(id, j);
dvander@434: 				for (unsigned int k=0; k<pUser->grp_count; k++)
dvander@434: 				{
dvander@434: 					if (other == src_table[k])
dvander@434: 					{
dvander@434: 						return false;
dvander@434: 					}
dvander@434: 				}
dvander@434: 			}
dvander@434: 		}
dvander@434: 	}
dvander@434: 	
dvander@434: 	return true;
dvander@434: }