dvander@370: /** ds@653: * vim: set ts=4 : ds@1330: * ============================================================================= dvander@1231: * SourceMod dvander@1231: * Copyright (C) 2004-2007 AlliedModders LLC. All rights reserved. ds@1330: * ============================================================================= dvander@370: * ds@1330: * This program is free software; you can redistribute it and/or modify it under ds@1330: * the terms of the GNU General Public License, version 3.0, as published by the ds@1330: * Free Software Foundation. dvander@1231: * ds@1330: * This program is distributed in the hope that it will be useful, but WITHOUT ds@1330: * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS ds@1330: * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more ds@1330: * details. dvander@1231: * ds@1330: * You should have received a copy of the GNU General Public License along with ds@1330: * this program. If not, see . dvander@1231: * ds@1330: * As a special exception, AlliedModders LLC gives you permission to link the ds@1330: * code of this program (as well as its derivative works) to "Half-Life 2," the ds@1330: * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software ds@1330: * by the Valve Corporation. You must obey the GNU General Public License in ds@1330: * all respects for all other code used. Additionally, AlliedModders LLC grants ds@1330: * this exception to all derivative works. AlliedModders LLC defines further ds@1330: * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007), ds@1330: * or . dvander@370: * dvander@370: * Version: $Id$ dvander@370: */ dvander@370: dvander@335: #include dvander@335: #include dvander@1406: #include 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@1298: #include "Logger.h" dvander@1298: #include "sourcemod.h" dvander@1298: #include "sm_stringutil.h" dvander@1403: #include "sourcemm_api.h" dvander@1298: dvander@1298: #define LEVEL_STATE_NONE 0 dvander@1298: #define LEVEL_STATE_LEVELS 1 dvander@1298: #define LEVEL_STATE_FLAGS 2 dvander@335: dvander@335: AdminCache g_Admins; dvander@1298: AdminFlag g_FlagLetters[26]; dvander@1298: bool g_FlagSet[26]; dvander@1298: dvander@1403: ConVar sm_immunity_mode("sm_immunity_mode", "1", FCVAR_SPONLY|FCVAR_PROTECTED, "Mode for deciding immunity protection"); dvander@1403: dvander@1298: /* Default flags */ dvander@1298: AdminFlag g_DefaultFlags[26] = dvander@1298: { dvander@1298: Admin_Reservation, Admin_Generic, Admin_Kick, Admin_Ban, Admin_Unban, dvander@1298: Admin_Slay, Admin_Changemap, Admin_Convars, Admin_Config, Admin_Chat, dvander@1298: Admin_Vote, Admin_Password, Admin_RCON, Admin_Cheats, Admin_Custom1, dvander@1298: Admin_Custom2, Admin_Custom3, Admin_Custom4, Admin_Custom5, Admin_Custom6, dvander@1298: Admin_Generic, Admin_Generic, Admin_Generic, Admin_Generic, Admin_Generic, dvander@1298: Admin_Root dvander@1298: }; dvander@1298: dvander@1298: dvander@1298: class FlagReader : public ITextListener_SMC dvander@1298: { dvander@1298: public: dvander@1298: void LoadLevels() dvander@1298: { dvander@1298: if (!Parse()) dvander@1298: { dvander@1298: memcpy(g_FlagLetters, g_DefaultFlags, sizeof(AdminFlag) * 26); dvander@1298: for (unsigned int i=0; i<20; i++) dvander@1298: { dvander@1298: g_FlagSet[i] = true; dvander@1298: } dvander@1298: g_FlagSet[25] = true; dvander@1298: } dvander@1298: } dvander@1298: private: dvander@1298: bool Parse() dvander@1298: { dvander@1298: unsigned int line = 0; dvander@1298: SMCParseError error; dvander@1298: dvander@1298: m_Line = 0; dvander@1298: m_bFileNameLogged = false; dvander@1298: g_SourceMod.BuildPath(Path_SM, m_File, sizeof(m_File), "configs/admin_levels.cfg"); dvander@1298: dvander@1406: if ((error = textparsers->ParseFile_SMC(m_File, this, &line, NULL)) dvander@1298: != SMCParse_Okay) dvander@1298: { dvander@1406: const char *err_string = textparsers->GetSMCErrorString(error); dvander@1298: if (!err_string) dvander@1298: { dvander@1298: err_string = "Unknown error"; dvander@1298: } dvander@1298: ParseError("Error %d (%s)", error, err_string); dvander@1298: return false; dvander@1298: } dvander@1298: dvander@1298: return true; dvander@1298: } dvander@1298: void ReadSMC_ParseStart() dvander@1298: { dvander@1298: m_LevelState = LEVEL_STATE_NONE; dvander@1298: m_IgnoreLevel = 0; dvander@1298: memset(g_FlagSet, 0, sizeof(g_FlagSet)); dvander@1298: } dvander@1298: SMCParseResult ReadSMC_NewSection(const char *name, bool opt_quotes) dvander@1298: { dvander@1298: if (m_IgnoreLevel) dvander@1298: { dvander@1298: m_IgnoreLevel++; dvander@1298: return SMCParse_Continue; dvander@1298: } dvander@1298: dvander@1298: if (m_LevelState == LEVEL_STATE_NONE) dvander@1298: { dvander@1298: if (strcmp(name, "Levels") == 0) dvander@1298: { dvander@1299: m_LevelState = LEVEL_STATE_LEVELS; dvander@1298: } else { dvander@1298: m_IgnoreLevel++; dvander@1298: } dvander@1299: } else if (m_LevelState == LEVEL_STATE_LEVELS) { dvander@1298: if (strcmp(name, "Flags") == 0) dvander@1298: { dvander@1299: m_LevelState = LEVEL_STATE_FLAGS; dvander@1298: } else { dvander@1298: m_IgnoreLevel++; dvander@1298: } dvander@1298: } else { dvander@1298: m_IgnoreLevel++; dvander@1298: } dvander@1298: dvander@1298: return SMCParse_Continue; dvander@1298: } dvander@1298: SMCParseResult ReadSMC_KeyValue(const char *key, const char *value, bool key_quotes, bool value_quotes) dvander@1298: { dvander@1299: if (m_LevelState != LEVEL_STATE_FLAGS || m_IgnoreLevel) dvander@1298: { dvander@1298: return SMCParse_Continue; dvander@1298: } dvander@1298: dvander@1298: unsigned char c = (unsigned)value[0]; dvander@1298: dvander@1298: if (c < (unsigned)'a' || c > (unsigned)'z') dvander@1298: { dvander@1298: ParseError("Flag \"%c\" is not a lower-case ASCII letter", c); dvander@1298: return SMCParse_Continue; dvander@1298: } dvander@1298: dvander@1298: c -= (unsigned)'a'; dvander@1298: dvander@1298: assert(c >= 0 && c < 26); dvander@1298: dvander@1298: if (!g_Admins.FindFlag(key, &g_FlagLetters[c])) dvander@1298: { dvander@1298: ParseError("Unrecognized admin level \"%s\"", key); dvander@1298: return SMCParse_Continue; dvander@1298: } dvander@1298: dvander@1298: g_FlagSet[c] = true; dvander@1298: dvander@1298: return SMCParse_Continue; dvander@1298: } dvander@1298: SMCParseResult ReadSMC_LeavingSection() dvander@1298: { dvander@1298: if (m_IgnoreLevel) dvander@1298: { dvander@1298: m_IgnoreLevel--; dvander@1298: return SMCParse_Continue; dvander@1298: } dvander@1298: dvander@1298: if (m_LevelState == LEVEL_STATE_FLAGS) dvander@1298: { dvander@1298: m_LevelState = LEVEL_STATE_LEVELS; dvander@1298: return SMCParse_Halt; dvander@1298: } else if (m_LevelState == LEVEL_STATE_LEVELS) { dvander@1298: m_LevelState = LEVEL_STATE_NONE; dvander@1298: } dvander@1298: dvander@1298: return SMCParse_Continue; dvander@1298: } dvander@1298: SMCParseResult ReadSMC_RawLine(const char *line, unsigned int curline) dvander@1298: { dvander@1298: m_Line = curline; dvander@1298: return SMCParse_Continue; dvander@1298: } dvander@1298: void ParseError(const char *message, ...) dvander@1298: { dvander@1298: va_list ap; dvander@1298: char buffer[256]; dvander@1298: dvander@1298: va_start(ap, message); dvander@1298: UTIL_FormatArgs(buffer, sizeof(buffer), message, ap); dvander@1298: va_end(ap); dvander@1298: dvander@1298: if (!m_bFileNameLogged) dvander@1298: { dvander@1298: g_Logger.LogError("[SM] Parse error(s) detected in file \"%s\":", m_File); dvander@1298: m_bFileNameLogged = true; dvander@1298: } dvander@1298: dvander@1298: g_Logger.LogError("[SM] (Line %d): %s", m_Line, buffer); dvander@1298: } dvander@1298: private: dvander@1298: bool m_bFileNameLogged; dvander@1298: char m_File[PLATFORM_MAX_PATH]; dvander@1298: int m_LevelState; dvander@1298: int m_IgnoreLevel; dvander@1298: unsigned int m_Line; dvander@1298: } s_FlagReader; 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@1298: m_pLevelNames = sm_trie_create(); 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::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@1298: dvander@1298: sm_trie_destroy(m_pLevelNames); 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@1298: dvander@1298: NameFlag("reservation", Admin_Reservation); dvander@1298: NameFlag("kick", Admin_Kick); dvander@1298: NameFlag("generic", Admin_Generic); dvander@1298: NameFlag("ban", Admin_Ban); dvander@1298: NameFlag("unban", Admin_Unban); dvander@1298: NameFlag("slay", Admin_Slay); dvander@1298: NameFlag("changemap", Admin_Changemap); dvander@1298: NameFlag("cvars", Admin_Convars); dvander@1298: NameFlag("config", Admin_Config); dvander@1298: NameFlag("chat", Admin_Chat); dvander@1298: NameFlag("vote", Admin_Vote); dvander@1298: NameFlag("password", Admin_Password); dvander@1298: NameFlag("rcon", Admin_RCON); dvander@1298: NameFlag("cheats", Admin_Cheats); dvander@1298: NameFlag("root", Admin_Root); dvander@1298: NameFlag("custom1", Admin_Custom1); dvander@1298: NameFlag("custom2", Admin_Custom2); dvander@1298: NameFlag("custom3", Admin_Custom3); dvander@1298: NameFlag("custom4", Admin_Custom4); dvander@1298: NameFlag("custom5", Admin_Custom5); dvander@1298: NameFlag("custom6", Admin_Custom6); 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@1298: void AdminCache::OnSourceModLevelChange(const char *mapName) dvander@1298: { dvander@1298: /* For now, we only read these once per level. */ dvander@1298: s_FlagReader.LoadLevels(); dvander@1298: } dvander@1298: 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@1298: void AdminCache::NameFlag(const char *str, AdminFlag flag) dvander@1298: { dvander@1298: sm_trie_insert(m_pLevelNames, str, (void *)flag); dvander@1298: } dvander@1298: dvander@1298: bool AdminCache::FindFlag(const char *str, AdminFlag *pFlag) dvander@1298: { dvander@1298: void *obj; dvander@1298: if (!sm_trie_retrieve(m_pLevelNames, str, &obj)) dvander@1298: { dvander@1298: return false; dvander@1298: } dvander@1298: dvander@1298: if (pFlag) dvander@1298: { dvander@1298: *pFlag = (AdminFlag)(int)obj; dvander@1298: } dvander@1298: dvander@1298: return true; dvander@1298: } dvander@1298: 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@1403: pUser->immunity_level = 0; dvander@1349: pUser->serialchange = 1; 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@1356: int nameidx = m_pStrings->AddString(name); dvander@1356: pUser = (AdminUser *)m_pMemory->GetAddress(id); dvander@1356: pUser->nameidx = nameidx; 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@1403: pGroup->immunity_level = 0; 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@1356: int nameidx = m_pStrings->AddString(group_name); dvander@1356: pGroup = (AdminGroup *)m_pMemory->GetAddress(id); dvander@1356: pGroup->nameidx = nameidx; 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@1356: const char *AdminCache::GetGroupName(GroupId gid) dvander@1356: { dvander@1356: AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid); dvander@1356: if (!pGroup || pGroup->magic != GRP_MAGIC_SET) dvander@1356: { dvander@1403: return NULL; dvander@1356: } dvander@1356: dvander@1356: return m_pStrings->GetString(pGroup->nameidx); dvander@1356: } dvander@1356: 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@1403: unsigned int level = 0; dvander@1403: dvander@1403: if (enabled) dvander@335: { dvander@1403: if (type == Immunity_Default) dvander@1403: { dvander@1403: level = 1; dvander@1403: } else if (type == Immunity_Global) { dvander@1403: level = 2; dvander@1403: } dvander@1403: if (level > pGroup->immunity_level) dvander@1403: { dvander@1403: pGroup->immunity_level = level; dvander@1403: } dvander@1403: } else { dvander@1403: pGroup->immunity_level = 0; dvander@1403: } 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@1403: if (type == Immunity_Default && pGroup->immunity_level >= 1) dvander@335: { dvander@1403: return true; dvander@1403: } else if (type == Immunity_Global && pGroup->immunity_level >= 2) { dvander@1403: return true; 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@1356: AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(other_id); dvander@335: if (!pGroup || pGroup->magic != GRP_MAGIC_SET) dvander@335: { dvander@335: return; dvander@335: } dvander@335: dvander@1356: pGroup = (AdminGroup *)m_pMemory->GetAddress(id); dvander@1356: if (!pGroup || pGroup->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@1356: if (pGroup->immune_table == -1) dvander@335: { dvander@335: tblidx = m_pMemory->CreateMem(sizeof(int) * 2, (void **)&table); dvander@1356: pGroup = (AdminGroup *)m_pMemory->GetAddress(id); dvander@335: table[0] = 0; dvander@335: } else { dvander@1356: int *old_table = (int *)m_pMemory->GetAddress(pGroup->immune_table); dvander@433: /* Break out if this group is already in the list */ dvander@433: for (int i=0; iCreateMem(sizeof(int) * (old_table[0] + 2), (void **)&table); dvander@433: /* Get the old address again in case of resize */ dvander@1356: pGroup = (AdminGroup *)m_pMemory->GetAddress(id); dvander@1356: old_table = (int *)m_pMemory->GetAddress(pGroup->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@1356: pGroup->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::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@1349: /* Unset serial change */ dvander@1349: pUser->serialchange = 0; dvander@1349: 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; igrp_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; jgrp_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; jgrp_count; j++) dvander@404: { dvander@404: pOther = (AdminGroup *)m_pMemory->GetAddress(table[j]); dvander@433: pUser->eflags |= pOther->addflags; dvander@404: } dvander@1349: /* Mark as changed */ dvander@1349: pUser->serialchange++; 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::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::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::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@1356: int i_ident = m_pStrings->AddString(ident); dvander@1356: dvander@1356: pUser = (AdminUser *)m_pMemory->GetAddress(id); dvander@1356: pUser->auth.identidx = i_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@1349: dvander@1349: pUser->serialchange++; 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@1349: dvander@1349: pUser->serialchange++; 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; igrp_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@1403: if (pGroup->immunity_level > pUser->immunity_level) dvander@434: { dvander@1403: pUser->immunity_level = pGroup->immunity_level; dvander@434: } dvander@434: dvander@1349: pUser->serialchange++; dvander@1349: 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@1356: GroupId gid = table[index]; dvander@1356: dvander@1356: if (name) dvander@1356: { dvander@1356: *name = GetGroupName(gid); dvander@1356: } dvander@1356: dvander@1356: return gid; 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@1356: int i_password = m_pStrings->AddString(password); dvander@1356: pUser = (AdminUser *)m_pMemory->GetAddress(id); dvander@1356: pUser->password = i_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; iGetAddress(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@1403: if (id == target) dvander@1403: { dvander@1403: return true; dvander@1403: } dvander@1403: 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: dvander@1403: /** Fourth, if the targeted admin is immune from targeting admin. */ dvander@1403: int mode = sm_immunity_mode.GetInt(); dvander@1403: switch (mode) dvander@434: { dvander@1403: case 1: dvander@1403: { dvander@1403: if (pTarget->immunity_level > pUser->immunity_level) dvander@1403: { dvander@1403: return false; dvander@1403: } dvander@1403: break; dvander@1403: } dvander@1403: case 3: dvander@1403: { dvander@1403: /* If neither has any immunity, let this pass. */ dvander@1403: if (!pUser->immunity_level && !pTarget->immunity_level) dvander@1403: { dvander@1403: return true; dvander@1403: } dvander@1403: /* Don't break, go to the next case. */ dvander@1403: } dvander@1403: case 2: dvander@1403: { dvander@1403: if (pTarget->immunity_level >= pUser->immunity_level) dvander@1403: { dvander@1403: return false; dvander@1403: } dvander@1403: break; dvander@1403: } dvander@434: } dvander@434: dvander@434: /** dvander@1403: * Fifth, 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; igrp_count; i++) dvander@434: { dvander@434: id = grp_table[i]; dvander@434: num = GetGroupImmunityCount(id); dvander@434: for (unsigned int j=0; jgrp_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: } dvander@1298: dvander@1298: bool AdminCache::FindFlag(char c, AdminFlag *pAdmFlag) dvander@1298: { dvander@1298: if (c < 'a' || c > 'z') dvander@1298: { dvander@1298: return false; dvander@1298: } dvander@1298: dvander@1305: if (pAdmFlag) dvander@1298: { dvander@1298: *pAdmFlag = g_FlagLetters[(unsigned)c - (unsigned)'a']; dvander@1298: } dvander@1298: dvander@1298: return true; dvander@1298: } dvander@1298: dvander@1298: FlagBits AdminCache::ReadFlagString(const char *flags, const char **end) dvander@1298: { dvander@1298: FlagBits bits = 0; dvander@1298: dvander@1298: while (flags && (*flags != '\0')) dvander@1298: { dvander@1298: AdminFlag flag; dvander@1298: if (!FindFlag(*flags, &flag)) dvander@1298: { dvander@1298: break; dvander@1298: } dvander@1298: bits |= FlagArrayToBits(&flag, 1); dvander@1298: flags++; dvander@1298: } dvander@1298: dvander@1298: if (end) dvander@1298: { dvander@1298: *end = flags; dvander@1298: } dvander@1298: dvander@1298: return bits; dvander@1298: } dvander@1349: dvander@1349: unsigned int AdminCache::GetAdminSerialChange(AdminId id) dvander@1349: { dvander@1349: AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); dvander@1349: if (!pUser || pUser->magic != USR_MAGIC_SET) dvander@1349: { dvander@1349: return 0; dvander@1349: } dvander@1349: dvander@1349: return pUser->serialchange; dvander@1349: } dvander@1349: dvander@1349: bool AdminCache::CanAdminUseCommand(int client, const char *cmd) dvander@1349: { dvander@1349: FlagBits bits; dvander@1349: OverrideType otype = Override_Command; dvander@1349: dvander@1349: if (cmd[0] == '@') dvander@1349: { dvander@1349: otype = Override_CommandGroup; dvander@1349: cmd++; dvander@1349: } dvander@1349: dvander@1349: if (!g_ConCmds.LookForCommandAdminFlags(cmd, &bits)) dvander@1349: { dvander@1349: if (!GetCommandOverride(cmd, otype, &bits)) dvander@1349: { dvander@1349: bits = 0; dvander@1349: } dvander@1349: } dvander@1349: dvander@1349: return g_ConCmds.CheckCommandAccess(client, cmd, bits); dvander@1349: } dvander@1403: dvander@1403: unsigned int AdminCache::SetGroupImmunityLevel(GroupId gid, unsigned int level) dvander@1403: { dvander@1403: AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid); dvander@1403: if (!pGroup || pGroup->magic != GRP_MAGIC_SET) dvander@1403: { dvander@1403: return 0; dvander@1403: } dvander@1403: dvander@1403: unsigned int old_level = pGroup->immunity_level; dvander@1403: dvander@1403: pGroup->immunity_level = level; dvander@1403: dvander@1403: return old_level; dvander@1403: } dvander@1403: dvander@1403: unsigned int AdminCache::GetGroupImmunityLevel(GroupId gid) dvander@1403: { dvander@1403: AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid); dvander@1403: if (!pGroup || pGroup->magic != GRP_MAGIC_SET) dvander@1403: { dvander@1403: return 0; dvander@1403: } dvander@1403: dvander@1403: return pGroup->immunity_level; dvander@1403: } dvander@1403: dvander@1403: unsigned int AdminCache::SetAdminImmunityLevel(AdminId id, unsigned int level) dvander@1403: { dvander@1403: AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); dvander@1403: if (!pUser || pUser->magic != USR_MAGIC_SET) dvander@1403: { dvander@1403: return 0; dvander@1403: } dvander@1403: dvander@1403: unsigned int old_level = pUser->immunity_level; dvander@1403: dvander@1403: pUser->immunity_level = level; dvander@1403: dvander@1403: return old_level; dvander@1403: } dvander@1403: dvander@1403: unsigned int AdminCache::GetAdminImmunityLevel(AdminId id) dvander@1403: { dvander@1403: AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); dvander@1403: if (!pUser || pUser->magic != USR_MAGIC_SET) dvander@1403: { dvander@1403: return 0; dvander@1403: } dvander@1403: dvander@1403: return pUser->immunity_level; dvander@1403: } dvander@1403: