1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/core/AdminCache.cpp Sun Mar 30 07:00:22 2008 +0000
1.3 @@ -0,0 +1,1732 @@
1.4 +/**
1.5 + * vim: set ts=4 :
1.6 + * =============================================================================
1.7 + * SourceMod
1.8 + * Copyright (C) 2004-2008 AlliedModders LLC. All rights reserved.
1.9 + * =============================================================================
1.10 + *
1.11 + * This program is free software; you can redistribute it and/or modify it under
1.12 + * the terms of the GNU General Public License, version 3.0, as published by the
1.13 + * Free Software Foundation.
1.14 + *
1.15 + * This program is distributed in the hope that it will be useful, but WITHOUT
1.16 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
1.17 + * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
1.18 + * details.
1.19 + *
1.20 + * You should have received a copy of the GNU General Public License along with
1.21 + * this program. If not, see <http://www.gnu.org/licenses/>.
1.22 + *
1.23 + * As a special exception, AlliedModders LLC gives you permission to link the
1.24 + * code of this program (as well as its derivative works) to "Half-Life 2," the
1.25 + * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
1.26 + * by the Valve Corporation. You must obey the GNU General Public License in
1.27 + * all respects for all other code used. Additionally, AlliedModders LLC grants
1.28 + * this exception to all derivative works. AlliedModders LLC defines further
1.29 + * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
1.30 + * or <http://www.sourcemod.net/license.php>.
1.31 + *
1.32 + * Version: $Id$
1.33 + */
1.34 +
1.35 +#include <string.h>
1.36 +#include <assert.h>
1.37 +#include <ITextParsers.h>
1.38 +#include "AdminCache.h"
1.39 +#include "ShareSys.h"
1.40 +#include "ForwardSys.h"
1.41 +#include "PlayerManager.h"
1.42 +#include "ConCmdManager.h"
1.43 +#include "Logger.h"
1.44 +#include "sourcemod.h"
1.45 +#include "sm_stringutil.h"
1.46 +#include "sourcemm_api.h"
1.47 +
1.48 +#define LEVEL_STATE_NONE 0
1.49 +#define LEVEL_STATE_LEVELS 1
1.50 +#define LEVEL_STATE_FLAGS 2
1.51 +
1.52 +AdminCache g_Admins;
1.53 +AdminFlag g_FlagLetters[26];
1.54 +bool g_FlagSet[26];
1.55 +
1.56 +ConVar sm_immunity_mode("sm_immunity_mode", "1", FCVAR_SPONLY, "Mode for deciding immunity protection");
1.57 +
1.58 +/* Default flags */
1.59 +AdminFlag g_DefaultFlags[26] =
1.60 +{
1.61 + Admin_Reservation, Admin_Generic, Admin_Kick, Admin_Ban, Admin_Unban,
1.62 + Admin_Slay, Admin_Changemap, Admin_Convars, Admin_Config, Admin_Chat,
1.63 + Admin_Vote, Admin_Password, Admin_RCON, Admin_Cheats, Admin_Custom1,
1.64 + Admin_Custom2, Admin_Custom3, Admin_Custom4, Admin_Custom5, Admin_Custom6,
1.65 + Admin_Generic, Admin_Generic, Admin_Generic, Admin_Generic, Admin_Generic,
1.66 + Admin_Root
1.67 +};
1.68 +
1.69 +
1.70 +class FlagReader : public ITextListener_SMC
1.71 +{
1.72 +public:
1.73 + void LoadLevels()
1.74 + {
1.75 + if (!Parse())
1.76 + {
1.77 + memcpy(g_FlagLetters, g_DefaultFlags, sizeof(AdminFlag) * 26);
1.78 + for (unsigned int i=0; i<20; i++)
1.79 + {
1.80 + g_FlagSet[i] = true;
1.81 + }
1.82 + g_FlagSet[25] = true;
1.83 + }
1.84 + }
1.85 +private:
1.86 + bool Parse()
1.87 + {
1.88 + SMCStates states;
1.89 + SMCError error;
1.90 +
1.91 + m_bFileNameLogged = false;
1.92 + g_SourceMod.BuildPath(Path_SM, m_File, sizeof(m_File), "configs/admin_levels.cfg");
1.93 +
1.94 + if ((error = textparsers->ParseFile_SMC(m_File, this, &states))
1.95 + != SMCError_Okay)
1.96 + {
1.97 + const char *err_string = textparsers->GetSMCErrorString(error);
1.98 + if (!err_string)
1.99 + {
1.100 + err_string = "Unknown error";
1.101 + }
1.102 + ParseError(NULL, "Error %d (%s)", error, err_string);
1.103 + return false;
1.104 + }
1.105 +
1.106 + return true;
1.107 + }
1.108 + void ReadSMC_ParseStart()
1.109 + {
1.110 + m_LevelState = LEVEL_STATE_NONE;
1.111 + m_IgnoreLevel = 0;
1.112 + memset(g_FlagSet, 0, sizeof(g_FlagSet));
1.113 + }
1.114 + SMCResult ReadSMC_NewSection(const SMCStates *states, const char *name)
1.115 + {
1.116 + if (m_IgnoreLevel)
1.117 + {
1.118 + m_IgnoreLevel++;
1.119 + return SMCResult_Continue;
1.120 + }
1.121 +
1.122 + if (m_LevelState == LEVEL_STATE_NONE)
1.123 + {
1.124 + if (strcmp(name, "Levels") == 0)
1.125 + {
1.126 + m_LevelState = LEVEL_STATE_LEVELS;
1.127 + }
1.128 + else
1.129 + {
1.130 + m_IgnoreLevel++;
1.131 + }
1.132 + } else if (m_LevelState == LEVEL_STATE_LEVELS) {
1.133 + if (strcmp(name, "Flags") == 0)
1.134 + {
1.135 + m_LevelState = LEVEL_STATE_FLAGS;
1.136 + }
1.137 + else
1.138 + {
1.139 + m_IgnoreLevel++;
1.140 + }
1.141 + }
1.142 + else
1.143 + {
1.144 + m_IgnoreLevel++;
1.145 + }
1.146 +
1.147 + return SMCResult_Continue;
1.148 + }
1.149 + SMCResult ReadSMC_KeyValue(const SMCStates *states, const char *key, const char *value)
1.150 + {
1.151 + if (m_LevelState != LEVEL_STATE_FLAGS || m_IgnoreLevel)
1.152 + {
1.153 + return SMCResult_Continue;
1.154 + }
1.155 +
1.156 + unsigned char c = (unsigned)value[0];
1.157 +
1.158 + if (c < (unsigned)'a' || c > (unsigned)'z')
1.159 + {
1.160 + ParseError(states, "Flag \"%c\" is not a lower-case ASCII letter", c);
1.161 + return SMCResult_Continue;
1.162 + }
1.163 +
1.164 + c -= (unsigned)'a';
1.165 +
1.166 + assert(c >= 0 && c < 26);
1.167 +
1.168 + if (!g_Admins.FindFlag(key, &g_FlagLetters[c]))
1.169 + {
1.170 + ParseError(states, "Unrecognized admin level \"%s\"", key);
1.171 + return SMCResult_Continue;
1.172 + }
1.173 +
1.174 + g_FlagSet[c] = true;
1.175 +
1.176 + return SMCResult_Continue;
1.177 + }
1.178 + SMCResult ReadSMC_LeavingSection(const SMCStates *states)
1.179 + {
1.180 + if (m_IgnoreLevel)
1.181 + {
1.182 + m_IgnoreLevel--;
1.183 + return SMCResult_Continue;
1.184 + }
1.185 +
1.186 + if (m_LevelState == LEVEL_STATE_FLAGS)
1.187 + {
1.188 + m_LevelState = LEVEL_STATE_LEVELS;
1.189 + return SMCResult_Halt;
1.190 + }
1.191 + else if (m_LevelState == LEVEL_STATE_LEVELS)
1.192 + {
1.193 + m_LevelState = LEVEL_STATE_NONE;
1.194 + }
1.195 +
1.196 + return SMCResult_Continue;
1.197 + }
1.198 + void ParseError(const SMCStates *states, const char *message, ...)
1.199 + {
1.200 + va_list ap;
1.201 + char buffer[256];
1.202 +
1.203 + va_start(ap, message);
1.204 + UTIL_FormatArgs(buffer, sizeof(buffer), message, ap);
1.205 + va_end(ap);
1.206 +
1.207 + if (!m_bFileNameLogged)
1.208 + {
1.209 + g_Logger.LogError("[SM] Parse error(s) detected in file \"%s\":", m_File);
1.210 + m_bFileNameLogged = true;
1.211 + }
1.212 +
1.213 + g_Logger.LogError("[SM] (Line %d): %s", states ? states->line : 0, buffer);
1.214 + }
1.215 +private:
1.216 + bool m_bFileNameLogged;
1.217 + char m_File[PLATFORM_MAX_PATH];
1.218 + int m_LevelState;
1.219 + int m_IgnoreLevel;
1.220 +} s_FlagReader;
1.221 +
1.222 +AdminCache::AdminCache()
1.223 +{
1.224 + m_pCmdOverrides = sm_trie_create();
1.225 + m_pCmdGrpOverrides = sm_trie_create();
1.226 + m_pStrings = new BaseStringTable(1024);
1.227 + m_pMemory = m_pStrings->GetMemTable();
1.228 + m_FreeGroupList = m_FirstGroup = m_LastGroup = INVALID_GROUP_ID;
1.229 + m_FreeUserList = m_FirstUser = m_LastUser = INVALID_ADMIN_ID;
1.230 + m_pGroups = sm_trie_create();
1.231 + m_pCacheFwd = NULL;
1.232 + m_FirstGroup = -1;
1.233 + m_pAuthTables = sm_trie_create();
1.234 + m_InvalidatingAdmins = false;
1.235 + m_destroying = false;
1.236 + m_pLevelNames = sm_trie_create();
1.237 +}
1.238 +
1.239 +AdminCache::~AdminCache()
1.240 +{
1.241 + m_destroying = true;
1.242 + DumpAdminCache(AdminCache_Overrides, false);
1.243 + DumpAdminCache(AdminCache_Groups, false);
1.244 +
1.245 + sm_trie_destroy(m_pCmdGrpOverrides);
1.246 + sm_trie_destroy(m_pCmdOverrides);
1.247 +
1.248 + if (m_pGroups)
1.249 + {
1.250 + sm_trie_destroy(m_pGroups);
1.251 + }
1.252 +
1.253 + List<AuthMethod>::iterator iter;
1.254 + for (iter=m_AuthMethods.begin();
1.255 + iter!=m_AuthMethods.end();
1.256 + iter++)
1.257 + {
1.258 + sm_trie_destroy((*iter).table);
1.259 + }
1.260 +
1.261 + sm_trie_destroy(m_pAuthTables);
1.262 +
1.263 + delete m_pStrings;
1.264 +
1.265 + sm_trie_destroy(m_pLevelNames);
1.266 +}
1.267 +
1.268 +void AdminCache::OnSourceModStartup(bool late)
1.269 +{
1.270 + RegisterAuthIdentType("steam");
1.271 + RegisterAuthIdentType("name");
1.272 + RegisterAuthIdentType("ip");
1.273 +
1.274 + NameFlag("reservation", Admin_Reservation);
1.275 + NameFlag("kick", Admin_Kick);
1.276 + NameFlag("generic", Admin_Generic);
1.277 + NameFlag("ban", Admin_Ban);
1.278 + NameFlag("unban", Admin_Unban);
1.279 + NameFlag("slay", Admin_Slay);
1.280 + NameFlag("changemap", Admin_Changemap);
1.281 + NameFlag("cvars", Admin_Convars);
1.282 + NameFlag("config", Admin_Config);
1.283 + NameFlag("chat", Admin_Chat);
1.284 + NameFlag("vote", Admin_Vote);
1.285 + NameFlag("password", Admin_Password);
1.286 + NameFlag("rcon", Admin_RCON);
1.287 + NameFlag("cheats", Admin_Cheats);
1.288 + NameFlag("root", Admin_Root);
1.289 + NameFlag("custom1", Admin_Custom1);
1.290 + NameFlag("custom2", Admin_Custom2);
1.291 + NameFlag("custom3", Admin_Custom3);
1.292 + NameFlag("custom4", Admin_Custom4);
1.293 + NameFlag("custom5", Admin_Custom5);
1.294 + NameFlag("custom6", Admin_Custom6);
1.295 +}
1.296 +
1.297 +void AdminCache::OnSourceModAllInitialized()
1.298 +{
1.299 + m_pCacheFwd = g_Forwards.CreateForward("OnRebuildAdminCache", ET_Ignore, 1, NULL, Param_Cell);
1.300 + g_ShareSys.AddInterface(NULL, this);
1.301 +}
1.302 +
1.303 +void AdminCache::OnSourceModLevelChange(const char *mapName)
1.304 +{
1.305 + /* For now, we only read these once per level. */
1.306 + s_FlagReader.LoadLevels();
1.307 +}
1.308 +
1.309 +void AdminCache::OnSourceModShutdown()
1.310 +{
1.311 + g_Forwards.ReleaseForward(m_pCacheFwd);
1.312 + m_pCacheFwd = NULL;
1.313 +}
1.314 +
1.315 +void AdminCache::OnSourceModPluginsLoaded()
1.316 +{
1.317 + DumpAdminCache(AdminCache_Overrides, true);
1.318 + DumpAdminCache(AdminCache_Groups, true);
1.319 +}
1.320 +
1.321 +void AdminCache::NameFlag(const char *str, AdminFlag flag)
1.322 +{
1.323 + sm_trie_insert(m_pLevelNames, str, (void *)flag);
1.324 +}
1.325 +
1.326 +bool AdminCache::FindFlag(const char *str, AdminFlag *pFlag)
1.327 +{
1.328 + void *obj;
1.329 + if (!sm_trie_retrieve(m_pLevelNames, str, &obj))
1.330 + {
1.331 + return false;
1.332 + }
1.333 +
1.334 + if (pFlag)
1.335 + {
1.336 + *pFlag = (AdminFlag)(int)obj;
1.337 + }
1.338 +
1.339 + return true;
1.340 +}
1.341 +
1.342 +void AdminCache::AddCommandOverride(const char *cmd, OverrideType type, FlagBits flags)
1.343 +{
1.344 + Trie *pTrie = NULL;
1.345 + if (type == Override_Command)
1.346 + {
1.347 + pTrie = m_pCmdOverrides;
1.348 + } else if (type == Override_CommandGroup) {
1.349 + pTrie = m_pCmdGrpOverrides;
1.350 + } else {
1.351 + return;
1.352 + }
1.353 +
1.354 + sm_trie_insert(pTrie, cmd, (void *)(unsigned int)flags);
1.355 +
1.356 + g_ConCmds.UpdateAdminCmdFlags(cmd, type, flags, false);
1.357 +}
1.358 +
1.359 +bool AdminCache::GetCommandOverride(const char *cmd, OverrideType type, FlagBits *pFlags)
1.360 +{
1.361 + Trie *pTrie = NULL;
1.362 +
1.363 + if (type == Override_Command)
1.364 + {
1.365 + pTrie = m_pCmdOverrides;
1.366 + } else if (type == Override_CommandGroup) {
1.367 + pTrie = m_pCmdGrpOverrides;
1.368 + } else {
1.369 + return false;
1.370 + }
1.371 +
1.372 + void *object;
1.373 + if (sm_trie_retrieve(pTrie, cmd, &object))
1.374 + {
1.375 + if (pFlags)
1.376 + {
1.377 + *pFlags = (FlagBits)object;
1.378 + }
1.379 + return true;
1.380 + }
1.381 +
1.382 + return false;
1.383 +}
1.384 +
1.385 +void AdminCache::UnsetCommandOverride(const char *cmd, OverrideType type)
1.386 +{
1.387 + if (type == Override_Command)
1.388 + {
1.389 + return _UnsetCommandOverride(cmd);
1.390 + } else if (type == Override_CommandGroup) {
1.391 + return _UnsetCommandGroupOverride(cmd);
1.392 + }
1.393 +}
1.394 +
1.395 +void AdminCache::_UnsetCommandGroupOverride(const char *group)
1.396 +{
1.397 + if (!m_pCmdGrpOverrides)
1.398 + {
1.399 + return;
1.400 + }
1.401 +
1.402 + sm_trie_delete(m_pCmdGrpOverrides, group);
1.403 +
1.404 + g_ConCmds.UpdateAdminCmdFlags(group, Override_CommandGroup, 0, true);
1.405 +}
1.406 +
1.407 +void AdminCache::_UnsetCommandOverride(const char *cmd)
1.408 +{
1.409 + if (!m_pCmdOverrides)
1.410 + {
1.411 + return;
1.412 + }
1.413 +
1.414 + sm_trie_delete(m_pCmdOverrides, cmd);
1.415 +
1.416 + g_ConCmds.UpdateAdminCmdFlags(cmd, Override_Command, 0, true);
1.417 +}
1.418 +
1.419 +void AdminCache::DumpCommandOverrideCache(OverrideType type)
1.420 +{
1.421 + if (type == Override_Command && m_pCmdOverrides)
1.422 + {
1.423 + sm_trie_clear(m_pCmdOverrides);
1.424 + } else if (type == Override_CommandGroup && m_pCmdGrpOverrides) {
1.425 + sm_trie_clear(m_pCmdGrpOverrides);
1.426 + }
1.427 +}
1.428 +
1.429 +AdminId AdminCache::CreateAdmin(const char *name)
1.430 +{
1.431 + AdminId id;
1.432 + AdminUser *pUser;
1.433 +
1.434 + if (m_FreeUserList != INVALID_ADMIN_ID)
1.435 + {
1.436 + pUser = (AdminUser *)m_pMemory->GetAddress(m_FreeUserList);
1.437 + assert(pUser->magic == USR_MAGIC_UNSET);
1.438 + id = m_FreeUserList;
1.439 + m_FreeUserList = pUser->next_user;
1.440 + } else {
1.441 + id = m_pMemory->CreateMem(sizeof(AdminUser), (void **)&pUser);
1.442 + pUser->grp_size = 0;
1.443 + pUser->grp_table = -1;
1.444 + }
1.445 +
1.446 + pUser->flags = 0;
1.447 + pUser->eflags = 0;
1.448 + pUser->grp_count = 0;
1.449 + pUser->password = -1;
1.450 + pUser->magic = USR_MAGIC_SET;
1.451 + pUser->auth.identidx = -1;
1.452 + pUser->auth.index = 0;
1.453 + pUser->immunity_level = 0;
1.454 + pUser->serialchange = 1;
1.455 +
1.456 + if (m_FirstUser == INVALID_ADMIN_ID)
1.457 + {
1.458 + m_FirstUser = id;
1.459 + m_LastUser = id;
1.460 + } else {
1.461 + AdminUser *pPrev = (AdminUser *)m_pMemory->GetAddress(m_LastUser);
1.462 + pPrev->next_user = id;
1.463 + pUser->prev_user = m_LastUser;
1.464 + m_LastUser = id;
1.465 + }
1.466 +
1.467 + if (name && name[0] != '\0')
1.468 + {
1.469 + int nameidx = m_pStrings->AddString(name);
1.470 + pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.471 + pUser->nameidx = nameidx;
1.472 + }
1.473 +
1.474 + return id;
1.475 +}
1.476 +
1.477 +GroupId AdminCache::AddGroup(const char *group_name)
1.478 +{
1.479 + if (sm_trie_retrieve(m_pGroups, group_name, NULL))
1.480 + {
1.481 + return INVALID_GROUP_ID;
1.482 + }
1.483 +
1.484 + GroupId id;
1.485 + AdminGroup *pGroup;
1.486 + if (m_FreeGroupList != INVALID_GROUP_ID)
1.487 + {
1.488 + pGroup = (AdminGroup *)m_pMemory->GetAddress(m_FreeGroupList);
1.489 + assert(pGroup->magic == GRP_MAGIC_UNSET);
1.490 + id = m_FreeGroupList;
1.491 + m_FreeGroupList = pGroup->next_grp;
1.492 + } else {
1.493 + id = m_pMemory->CreateMem(sizeof(AdminGroup), (void **)&pGroup);
1.494 + }
1.495 +
1.496 + pGroup->immunity_level = 0;
1.497 + pGroup->immune_table = -1;
1.498 + pGroup->magic = GRP_MAGIC_SET;
1.499 + pGroup->next_grp = INVALID_GROUP_ID;
1.500 + pGroup->pCmdGrpTable = NULL;
1.501 + pGroup->pCmdTable = NULL;
1.502 + pGroup->addflags = 0;
1.503 +
1.504 + if (m_FirstGroup == INVALID_GROUP_ID)
1.505 + {
1.506 + m_FirstGroup = id;
1.507 + m_LastGroup = id;
1.508 + pGroup->prev_grp = INVALID_GROUP_ID;
1.509 + } else {
1.510 + AdminGroup *pPrev = (AdminGroup *)m_pMemory->GetAddress(m_LastGroup);
1.511 + assert(pPrev->magic == GRP_MAGIC_SET);
1.512 + pPrev->next_grp = id;
1.513 + pGroup->prev_grp = m_LastGroup;
1.514 + m_LastGroup = id;
1.515 + }
1.516 +
1.517 + int nameidx = m_pStrings->AddString(group_name);
1.518 + pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
1.519 + pGroup->nameidx = nameidx;
1.520 +
1.521 + sm_trie_insert(m_pGroups, group_name, (void *)id);
1.522 +
1.523 + return id;
1.524 +}
1.525 +
1.526 +GroupId AdminCache::FindGroupByName(const char *group_name)
1.527 +{
1.528 + void *object;
1.529 +
1.530 + if (!sm_trie_retrieve(m_pGroups, group_name, &object))
1.531 + {
1.532 + return INVALID_GROUP_ID;
1.533 + }
1.534 +
1.535 + GroupId id = (GroupId)object;
1.536 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
1.537 +
1.538 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.539 + {
1.540 + return INVALID_GROUP_ID;
1.541 + }
1.542 +
1.543 + return id;
1.544 +}
1.545 +
1.546 +void AdminCache::SetGroupAddFlag(GroupId id, AdminFlag flag, bool enabled)
1.547 +{
1.548 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
1.549 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.550 + {
1.551 + return;
1.552 + }
1.553 +
1.554 + if (flag < Admin_Reservation || flag >= AdminFlags_TOTAL)
1.555 + {
1.556 + return;
1.557 + }
1.558 +
1.559 + FlagBits bits = (1<<(FlagBits)flag);
1.560 +
1.561 + if (enabled)
1.562 + {
1.563 + pGroup->addflags |= bits;
1.564 + } else {
1.565 + pGroup->addflags &= ~bits;
1.566 + }
1.567 +}
1.568 +
1.569 +bool AdminCache::GetGroupAddFlag(GroupId id, AdminFlag flag)
1.570 +{
1.571 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
1.572 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.573 + {
1.574 + return false;
1.575 + }
1.576 +
1.577 + if (flag < Admin_Reservation || flag >= AdminFlags_TOTAL)
1.578 + {
1.579 + return false;
1.580 + }
1.581 +
1.582 + FlagBits bit = 1<<(FlagBits)flag;
1.583 + return ((pGroup->addflags & bit) == bit);
1.584 +}
1.585 +
1.586 +FlagBits AdminCache::GetGroupAddFlags(GroupId id)
1.587 +{
1.588 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
1.589 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.590 + {
1.591 + return 0;
1.592 + }
1.593 +
1.594 + return pGroup->addflags;
1.595 +}
1.596 +
1.597 +const char *AdminCache::GetGroupName(GroupId gid)
1.598 +{
1.599 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid);
1.600 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.601 + {
1.602 + return NULL;
1.603 + }
1.604 +
1.605 + return m_pStrings->GetString(pGroup->nameidx);
1.606 +}
1.607 +
1.608 +void AdminCache::SetGroupGenericImmunity(GroupId id, ImmunityType type, bool enabled)
1.609 +{
1.610 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
1.611 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.612 + {
1.613 + return;
1.614 + }
1.615 +
1.616 + unsigned int level = 0;
1.617 +
1.618 + if (enabled)
1.619 + {
1.620 + if (type == Immunity_Default)
1.621 + {
1.622 + level = 1;
1.623 + } else if (type == Immunity_Global) {
1.624 + level = 2;
1.625 + }
1.626 + if (level > pGroup->immunity_level)
1.627 + {
1.628 + pGroup->immunity_level = level;
1.629 + }
1.630 + } else {
1.631 + pGroup->immunity_level = 0;
1.632 + }
1.633 +}
1.634 +
1.635 +bool AdminCache::GetGroupGenericImmunity(GroupId id, ImmunityType type)
1.636 +{
1.637 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
1.638 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.639 + {
1.640 + return false;
1.641 + }
1.642 +
1.643 + if (type == Immunity_Default && pGroup->immunity_level >= 1)
1.644 + {
1.645 + return true;
1.646 + } else if (type == Immunity_Global && pGroup->immunity_level >= 2) {
1.647 + return true;
1.648 + }
1.649 +
1.650 + return false;
1.651 +}
1.652 +
1.653 +void AdminCache::AddGroupImmunity(GroupId id, GroupId other_id)
1.654 +{
1.655 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(other_id);
1.656 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.657 + {
1.658 + return;
1.659 + }
1.660 +
1.661 + pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
1.662 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.663 + {
1.664 + return;
1.665 + }
1.666 +
1.667 + /* We always need to resize the immunity table */
1.668 + int *table, tblidx;
1.669 + if (pGroup->immune_table == -1)
1.670 + {
1.671 + tblidx = m_pMemory->CreateMem(sizeof(int) * 2, (void **)&table);
1.672 + pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
1.673 + table[0] = 0;
1.674 + } else {
1.675 + int *old_table = (int *)m_pMemory->GetAddress(pGroup->immune_table);
1.676 + /* Break out if this group is already in the list */
1.677 + for (int i=0; i<old_table[0]; i++)
1.678 + {
1.679 + if (old_table[1+i] == other_id)
1.680 + {
1.681 + return;
1.682 + }
1.683 + }
1.684 + tblidx = m_pMemory->CreateMem(sizeof(int) * (old_table[0] + 2), (void **)&table);
1.685 + /* Get the old address again in case of resize */
1.686 + pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
1.687 + old_table = (int *)m_pMemory->GetAddress(pGroup->immune_table);
1.688 + table[0] = old_table[0];
1.689 + for (unsigned int i=1; i<=(unsigned int)old_table[0]; i++)
1.690 + {
1.691 + table[i] = old_table[i];
1.692 + }
1.693 + }
1.694 +
1.695 + /* Assign */
1.696 + pGroup->immune_table = tblidx;
1.697 +
1.698 + /* Add to the array */
1.699 + table[0]++;
1.700 + table[table[0]] = other_id;
1.701 +}
1.702 +
1.703 +unsigned int AdminCache::GetGroupImmunityCount(GroupId id)
1.704 +{
1.705 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
1.706 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.707 + {
1.708 + return 0;
1.709 + }
1.710 +
1.711 + if (pGroup->immune_table == -1)
1.712 + {
1.713 + return 0;
1.714 + }
1.715 +
1.716 + int *table = (int *)m_pMemory->GetAddress(pGroup->immune_table);
1.717 +
1.718 + return table[0];
1.719 +}
1.720 +
1.721 +GroupId AdminCache::GetGroupImmunity(GroupId id, unsigned int number)
1.722 +{
1.723 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
1.724 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.725 + {
1.726 + return INVALID_GROUP_ID;
1.727 + }
1.728 +
1.729 + if (pGroup->immune_table == -1)
1.730 + {
1.731 + return INVALID_GROUP_ID;
1.732 + }
1.733 +
1.734 + int *table = (int *)m_pMemory->GetAddress(pGroup->immune_table);
1.735 + if (number >= (unsigned int)table[0])
1.736 + {
1.737 + return INVALID_GROUP_ID;
1.738 + }
1.739 +
1.740 + return table[1 + number];
1.741 +}
1.742 +
1.743 +void AdminCache::AddGroupCommandOverride(GroupId id, const char *name, OverrideType type, OverrideRule rule)
1.744 +{
1.745 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
1.746 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.747 + {
1.748 + return;
1.749 + }
1.750 +
1.751 + Trie *pTrie = NULL;
1.752 + if (type == Override_Command)
1.753 + {
1.754 + if (pGroup->pCmdTable == NULL)
1.755 + {
1.756 + pGroup->pCmdTable = sm_trie_create();
1.757 + }
1.758 + pTrie = pGroup->pCmdTable;
1.759 + } else if (type == Override_CommandGroup) {
1.760 + if (pGroup->pCmdGrpTable == NULL)
1.761 + {
1.762 + pGroup->pCmdGrpTable = sm_trie_create();
1.763 + }
1.764 + pTrie = pGroup->pCmdGrpTable;
1.765 + } else {
1.766 + return;
1.767 + }
1.768 +
1.769 + sm_trie_insert(pTrie, name, (void *)(int)rule);
1.770 +}
1.771 +
1.772 +bool AdminCache::GetGroupCommandOverride(GroupId id, const char *name, OverrideType type, OverrideRule *pRule)
1.773 +{
1.774 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
1.775 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.776 + {
1.777 + return false;
1.778 + }
1.779 +
1.780 + Trie *pTrie = NULL;
1.781 + if (type == Override_Command)
1.782 + {
1.783 + if (pGroup->pCmdTable == NULL)
1.784 + {
1.785 + return false;
1.786 + }
1.787 + pTrie = pGroup->pCmdTable;
1.788 + } else if (type == Override_CommandGroup) {
1.789 + if (pGroup->pCmdGrpTable == NULL)
1.790 + {
1.791 + return false;
1.792 + }
1.793 + pTrie = pGroup->pCmdGrpTable;
1.794 + } else {
1.795 + return false;
1.796 + }
1.797 +
1.798 + void *object;
1.799 + if (!sm_trie_retrieve(pTrie, name, &object))
1.800 + {
1.801 + return false;
1.802 + }
1.803 +
1.804 + if (pRule)
1.805 + {
1.806 + *pRule = (OverrideRule)(int)object;
1.807 + }
1.808 +
1.809 + return true;
1.810 +}
1.811 +
1.812 +Trie *AdminCache::GetMethodByIndex(unsigned int index)
1.813 +{
1.814 + List<AuthMethod>::iterator iter;
1.815 + for (iter=m_AuthMethods.begin();
1.816 + iter!=m_AuthMethods.end();
1.817 + iter++)
1.818 + {
1.819 + if (index-- == 0)
1.820 + {
1.821 + return (*iter).table;
1.822 + }
1.823 + }
1.824 +
1.825 + return NULL;
1.826 +}
1.827 +
1.828 +bool AdminCache::InvalidateAdmin(AdminId id)
1.829 +{
1.830 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.831 + AdminUser *pOther;
1.832 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.833 + {
1.834 + return false;
1.835 + }
1.836 +
1.837 + if (!m_InvalidatingAdmins && !m_destroying)
1.838 + {
1.839 + g_Players.ClearAdminId(id);
1.840 + }
1.841 +
1.842 + /* Unlink from the dbl link list */
1.843 + if (id == m_FirstUser && id == m_LastUser)
1.844 + {
1.845 + m_FirstUser = INVALID_ADMIN_ID;
1.846 + m_LastUser = INVALID_ADMIN_ID;
1.847 + } else if (id == m_FirstUser) {
1.848 + m_FirstUser = pUser->next_user;
1.849 + pOther = (AdminUser *)m_pMemory->GetAddress(m_FirstUser);
1.850 + pOther->prev_user = INVALID_ADMIN_ID;
1.851 + } else if (id == m_LastUser) {
1.852 + m_LastUser = pUser->prev_user;
1.853 + pOther = (AdminUser *)m_pMemory->GetAddress(m_LastUser);
1.854 + pOther->next_user = INVALID_ADMIN_ID;
1.855 + } else {
1.856 + pOther = (AdminUser *)m_pMemory->GetAddress(pUser->prev_user);
1.857 + pOther->next_user = pUser->next_user;
1.858 + pOther = (AdminUser *)m_pMemory->GetAddress(pUser->next_user);
1.859 + pOther->prev_user = pUser->prev_user;
1.860 + }
1.861 +
1.862 + /* Unlink from auth tables */
1.863 + if (pUser->auth.identidx != -1)
1.864 + {
1.865 + Trie *pTrie = GetMethodByIndex(pUser->auth.index);
1.866 + if (pTrie)
1.867 + {
1.868 + sm_trie_delete(pTrie, m_pStrings->GetString(pUser->auth.identidx));
1.869 + }
1.870 + }
1.871 +
1.872 + /* Clear table counts */
1.873 + pUser->grp_count = 0;
1.874 +
1.875 + /* Link into free list */
1.876 + pUser->magic = USR_MAGIC_UNSET;
1.877 + pUser->next_user = m_FreeUserList;
1.878 + m_FreeUserList = id;
1.879 +
1.880 + /* Unset serial change */
1.881 + pUser->serialchange = 0;
1.882 +
1.883 + return true;
1.884 +}
1.885 +
1.886 +
1.887 +void AdminCache::InvalidateGroup(GroupId id)
1.888 +{
1.889 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
1.890 + AdminGroup *pOther;
1.891 +
1.892 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.893 + {
1.894 + return;
1.895 + }
1.896 +
1.897 + const char *str = m_pStrings->GetString(pGroup->nameidx);
1.898 + sm_trie_delete(m_pGroups, str);
1.899 +
1.900 + /* Unlink from the live dbllink list */
1.901 + if (id == m_FirstGroup && id == m_LastGroup)
1.902 + {
1.903 + m_LastGroup = INVALID_GROUP_ID;
1.904 + m_FirstGroup = INVALID_GROUP_ID;
1.905 + } else if (id == m_FirstGroup) {
1.906 + m_FirstGroup = pGroup->next_grp;
1.907 + pOther = (AdminGroup *)m_pMemory->GetAddress(m_FirstGroup);
1.908 + pOther->prev_grp = INVALID_GROUP_ID;
1.909 + } else if (id == m_LastGroup) {
1.910 + m_LastGroup = pGroup->prev_grp;
1.911 + pOther = (AdminGroup *)m_pMemory->GetAddress(m_LastGroup);
1.912 + pOther->next_grp = INVALID_GROUP_ID;
1.913 + } else {
1.914 + pOther = (AdminGroup *)m_pMemory->GetAddress(pGroup->prev_grp);
1.915 + pOther->next_grp = pGroup->next_grp;
1.916 + pOther = (AdminGroup *)m_pMemory->GetAddress(pGroup->next_grp);
1.917 + pOther->prev_grp = pGroup->prev_grp;
1.918 + }
1.919 +
1.920 + /* Free any used memory to be safe */
1.921 + if (pGroup->pCmdGrpTable)
1.922 + {
1.923 + sm_trie_destroy(pGroup->pCmdGrpTable);
1.924 + pGroup->pCmdGrpTable = NULL;
1.925 + }
1.926 + if (pGroup->pCmdTable)
1.927 + {
1.928 + sm_trie_destroy(pGroup->pCmdTable);
1.929 + pGroup->pCmdTable = NULL;
1.930 + }
1.931 +
1.932 + /* Link into the free list */
1.933 + pGroup->magic = GRP_MAGIC_UNSET;
1.934 + pGroup->next_grp = m_FreeGroupList;
1.935 + m_FreeGroupList = id;
1.936 +
1.937 + int idx = m_FirstUser;
1.938 + AdminUser *pUser;
1.939 + int *table;
1.940 + while (idx != INVALID_ADMIN_ID)
1.941 + {
1.942 + pUser = (AdminUser *)m_pMemory->GetAddress(idx);
1.943 + if (pUser->grp_count > 0)
1.944 + {
1.945 + table = (int *)m_pMemory->GetAddress(pUser->grp_table);
1.946 + for (unsigned int i=0; i<pUser->grp_count; i++)
1.947 + {
1.948 + if (table[i] == id)
1.949 + {
1.950 + /* We have to remove this entry */
1.951 + for (unsigned int j=i+1; j<pUser->grp_count; j++)
1.952 + {
1.953 + /* Move everything down by one */
1.954 + table[j-1] = table[j];
1.955 + }
1.956 + /* Decrease count */
1.957 + pUser->grp_count--;
1.958 + /* Recalculate effective flags */
1.959 + pUser->eflags = pUser->flags;
1.960 + for (unsigned int j=0; j<pUser->grp_count; j++)
1.961 + {
1.962 + pOther = (AdminGroup *)m_pMemory->GetAddress(table[j]);
1.963 + pUser->eflags |= pOther->addflags;
1.964 + }
1.965 + /* Mark as changed */
1.966 + pUser->serialchange++;
1.967 + /* Break now, duplicates aren't allowed */
1.968 + break;
1.969 + }
1.970 + }
1.971 + }
1.972 + idx = pUser->next_user;
1.973 + }
1.974 +}
1.975 +
1.976 +void AdminCache::InvalidateGroupCache()
1.977 +{
1.978 + /* Nuke the free list */
1.979 + m_FreeGroupList = -1;
1.980 +
1.981 + /* Nuke reverse lookups */
1.982 + sm_trie_clear(m_pGroups);
1.983 +
1.984 + /* Free memory on groups */
1.985 + GroupId cur = m_FirstGroup;
1.986 + AdminGroup *pGroup;
1.987 + while (cur != INVALID_GROUP_ID)
1.988 + {
1.989 + pGroup = (AdminGroup *)m_pMemory->GetAddress(cur);
1.990 + assert(pGroup->magic == GRP_MAGIC_SET);
1.991 + if (pGroup->pCmdGrpTable)
1.992 + {
1.993 + sm_trie_destroy(pGroup->pCmdGrpTable);
1.994 + }
1.995 + if (pGroup->pCmdTable)
1.996 + {
1.997 + sm_trie_destroy(pGroup->pCmdTable);
1.998 + }
1.999 + cur = pGroup->next_grp;
1.1000 + }
1.1001 +
1.1002 + m_FirstGroup = INVALID_GROUP_ID;
1.1003 + m_LastGroup = INVALID_GROUP_ID;
1.1004 +
1.1005 + InvalidateAdminCache(false);
1.1006 +
1.1007 + /* Reset the memory table */
1.1008 + m_pMemory->Reset();
1.1009 +}
1.1010 +
1.1011 +void AdminCache::AddAdminListener(IAdminListener *pListener)
1.1012 +{
1.1013 + m_hooks.push_back(pListener);
1.1014 +}
1.1015 +
1.1016 +void AdminCache::RemoveAdminListener(IAdminListener *pListener)
1.1017 +{
1.1018 + m_hooks.remove(pListener);
1.1019 +}
1.1020 +
1.1021 +void AdminCache::RegisterAuthIdentType(const char *name)
1.1022 +{
1.1023 + if (sm_trie_retrieve(m_pAuthTables, name, NULL))
1.1024 + {
1.1025 + return;
1.1026 + }
1.1027 +
1.1028 + Trie *pAuth = sm_trie_create();
1.1029 +
1.1030 + AuthMethod method;
1.1031 + method.name.assign(name);
1.1032 + method.table = pAuth;
1.1033 +
1.1034 + m_AuthMethods.push_back(method);
1.1035 +
1.1036 + sm_trie_insert(m_pAuthTables, name, pAuth);
1.1037 +}
1.1038 +
1.1039 +void AdminCache::InvalidateAdminCache(bool unlink_admins)
1.1040 +{
1.1041 + m_InvalidatingAdmins = true;
1.1042 + if (!m_destroying)
1.1043 + {
1.1044 + g_Players.ClearAllAdmins();
1.1045 + }
1.1046 + /* Wipe the identity cache first */
1.1047 + List<AuthMethod>::iterator iter;
1.1048 + for (iter=m_AuthMethods.begin();
1.1049 + iter!=m_AuthMethods.end();
1.1050 + iter++)
1.1051 + {
1.1052 + sm_trie_clear((*iter).table);
1.1053 + }
1.1054 +
1.1055 + if (unlink_admins)
1.1056 + {
1.1057 + while (m_FirstUser != INVALID_ADMIN_ID)
1.1058 + {
1.1059 + InvalidateAdmin(m_FirstUser);
1.1060 + }
1.1061 + } else {
1.1062 + m_FirstUser = -1;
1.1063 + m_LastUser = -1;
1.1064 + m_FreeUserList = -1;
1.1065 + }
1.1066 + m_InvalidatingAdmins = false;
1.1067 +}
1.1068 +
1.1069 +void AdminCache::DumpAdminCache(AdminCachePart part, bool rebuild)
1.1070 +{
1.1071 + List<IAdminListener *>::iterator iter;
1.1072 + IAdminListener *pListener;
1.1073 + cell_t result;
1.1074 +
1.1075 + if (part == AdminCache_Overrides)
1.1076 + {
1.1077 + DumpCommandOverrideCache(Override_Command);
1.1078 + DumpCommandOverrideCache(Override_CommandGroup);
1.1079 + if (rebuild && !m_destroying)
1.1080 + {
1.1081 + for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
1.1082 + {
1.1083 + pListener = (*iter);
1.1084 + pListener->OnRebuildOverrideCache();
1.1085 + }
1.1086 + m_pCacheFwd->PushCell(part);
1.1087 + m_pCacheFwd->Execute(&result);
1.1088 + }
1.1089 + } else if (part == AdminCache_Groups || part == AdminCache_Admins) {
1.1090 + if (part == AdminCache_Groups)
1.1091 + {
1.1092 + InvalidateGroupCache();
1.1093 + if (rebuild && !m_destroying)
1.1094 + {
1.1095 + for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
1.1096 + {
1.1097 + pListener = (*iter);
1.1098 + pListener->OnRebuildGroupCache();
1.1099 + }
1.1100 + m_pCacheFwd->PushCell(part);
1.1101 + m_pCacheFwd->Execute(&result);
1.1102 + }
1.1103 + }
1.1104 + InvalidateAdminCache(true);
1.1105 + if (rebuild && !m_destroying)
1.1106 + {
1.1107 + for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
1.1108 + {
1.1109 + pListener = (*iter);
1.1110 + pListener->OnRebuildAdminCache((part == AdminCache_Groups));
1.1111 + }
1.1112 + m_pCacheFwd->PushCell(AdminCache_Admins);
1.1113 + m_pCacheFwd->Execute(&result);
1.1114 + g_Players.RecheckAnyAdmins();
1.1115 + }
1.1116 + }
1.1117 +}
1.1118 +
1.1119 +const char *AdminCache::GetAdminName(AdminId id)
1.1120 +{
1.1121 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1122 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.1123 + {
1.1124 + return NULL;
1.1125 + }
1.1126 +
1.1127 + return m_pStrings->GetString(pUser->nameidx);
1.1128 +}
1.1129 +
1.1130 +bool AdminCache::GetMethodIndex(const char *name, unsigned int *_index)
1.1131 +{
1.1132 + List<AuthMethod>::iterator iter;
1.1133 + unsigned int index = 0;
1.1134 + for (iter=m_AuthMethods.begin();
1.1135 + iter!=m_AuthMethods.end();
1.1136 + iter++,index++)
1.1137 + {
1.1138 + if ((*iter).name.compare(name) == 0)
1.1139 + {
1.1140 + *_index = index;
1.1141 + return true;
1.1142 + }
1.1143 + }
1.1144 +
1.1145 + return false;
1.1146 +}
1.1147 +
1.1148 +bool AdminCache::BindAdminIdentity(AdminId id, const char *auth, const char *ident)
1.1149 +{
1.1150 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1151 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.1152 + {
1.1153 + return false;
1.1154 + }
1.1155 +
1.1156 + Trie *pTable;
1.1157 + if (!sm_trie_retrieve(m_pAuthTables, auth, (void **)&pTable))
1.1158 + {
1.1159 + return false;
1.1160 + }
1.1161 +
1.1162 + if (sm_trie_retrieve(pTable, ident, NULL))
1.1163 + {
1.1164 + return false;
1.1165 + }
1.1166 +
1.1167 + int i_ident = m_pStrings->AddString(ident);
1.1168 +
1.1169 + pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1170 + pUser->auth.identidx = i_ident;
1.1171 + GetMethodIndex(auth, &pUser->auth.index);
1.1172 +
1.1173 + return sm_trie_insert(pTable, ident, (void **)id);
1.1174 +}
1.1175 +
1.1176 +AdminId AdminCache::FindAdminByIdentity(const char *auth, const char *identity)
1.1177 +{
1.1178 + Trie *pTable;
1.1179 + if (!sm_trie_retrieve(m_pAuthTables, auth, (void **)&pTable))
1.1180 + {
1.1181 + return INVALID_ADMIN_ID;
1.1182 + }
1.1183 +
1.1184 + void *object;
1.1185 + if (!sm_trie_retrieve(pTable, identity, &object))
1.1186 + {
1.1187 + return INVALID_ADMIN_ID;
1.1188 + }
1.1189 +
1.1190 + return (AdminId)object;
1.1191 +}
1.1192 +
1.1193 +void AdminCache::SetAdminFlag(AdminId id, AdminFlag flag, bool enabled)
1.1194 +{
1.1195 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1196 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.1197 + {
1.1198 + return;
1.1199 + }
1.1200 +
1.1201 + if (flag < Admin_Reservation
1.1202 + || flag >= AdminFlags_TOTAL)
1.1203 + {
1.1204 + return;
1.1205 + }
1.1206 +
1.1207 + FlagBits bits = (1<<(FlagBits)flag);
1.1208 +
1.1209 + if (enabled)
1.1210 + {
1.1211 + pUser->flags |= bits;
1.1212 + pUser->eflags |= bits;
1.1213 + } else {
1.1214 + pUser->flags &= ~bits;
1.1215 + pUser->eflags &= ~bits;
1.1216 + }
1.1217 +
1.1218 + pUser->serialchange++;
1.1219 +}
1.1220 +
1.1221 +bool AdminCache::GetAdminFlag(AdminId id, AdminFlag flag, AccessMode mode)
1.1222 +{
1.1223 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1224 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.1225 + {
1.1226 + return false;
1.1227 + }
1.1228 +
1.1229 + if (flag < Admin_Reservation
1.1230 + || flag >= AdminFlags_TOTAL)
1.1231 + {
1.1232 + return false;
1.1233 + }
1.1234 +
1.1235 + FlagBits bit = (1<<(FlagBits)flag);
1.1236 +
1.1237 + if (mode == Access_Real)
1.1238 + {
1.1239 + return ((pUser->flags & bit) == bit);
1.1240 + } else if (mode == Access_Effective) {
1.1241 + bool has_bit = ((pUser->eflags & bit) == bit);
1.1242 + if (!has_bit && flag != Admin_Root && ((pUser->eflags & ADMFLAG_ROOT) == ADMFLAG_ROOT))
1.1243 + {
1.1244 + has_bit = true;
1.1245 + }
1.1246 + return has_bit;
1.1247 + }
1.1248 +
1.1249 + return false;
1.1250 +}
1.1251 +
1.1252 +FlagBits AdminCache::GetAdminFlags(AdminId id, AccessMode mode)
1.1253 +{
1.1254 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1255 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.1256 + {
1.1257 + return 0;
1.1258 + }
1.1259 +
1.1260 + if (mode == Access_Real)
1.1261 + {
1.1262 + return pUser->flags;
1.1263 + } else if (mode == Access_Effective) {
1.1264 + return pUser->eflags;
1.1265 + }
1.1266 +
1.1267 + return 0;
1.1268 +}
1.1269 +
1.1270 +void AdminCache::SetAdminFlags(AdminId id, AccessMode mode, FlagBits bits)
1.1271 +{
1.1272 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1273 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.1274 + {
1.1275 + return;
1.1276 + }
1.1277 +
1.1278 + if (mode == Access_Real)
1.1279 + {
1.1280 + pUser->flags = bits;
1.1281 + pUser->eflags = bits;
1.1282 + } else if (mode == Access_Effective) {
1.1283 + pUser->eflags = bits;
1.1284 + }
1.1285 +
1.1286 + pUser->serialchange++;
1.1287 +}
1.1288 +
1.1289 +bool AdminCache::AdminInheritGroup(AdminId id, GroupId gid)
1.1290 +{
1.1291 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1292 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.1293 + {
1.1294 + return false;
1.1295 + }
1.1296 +
1.1297 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid);
1.1298 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.1299 + {
1.1300 + return false;
1.1301 + }
1.1302 +
1.1303 + /* First check for duplicates */
1.1304 + if (pUser->grp_count != 0)
1.1305 + {
1.1306 + int *temp_table = (int *)m_pMemory->GetAddress(pUser->grp_table);
1.1307 + for (unsigned int i=0; i<pUser->grp_count; i++)
1.1308 + {
1.1309 + if (temp_table[i] == gid)
1.1310 + {
1.1311 + return false;
1.1312 + }
1.1313 + }
1.1314 + }
1.1315 +
1.1316 + int *table;
1.1317 + if (pUser->grp_count + 1 > pUser->grp_size)
1.1318 + {
1.1319 + int new_size = 0;
1.1320 + int tblidx;
1.1321 +
1.1322 + if (pUser->grp_size == 0)
1.1323 + {
1.1324 + new_size = 2;
1.1325 + } else {
1.1326 + new_size = pUser->grp_size * 2;
1.1327 + }
1.1328 +
1.1329 + /* Create and refresh pointers */
1.1330 + tblidx = m_pMemory->CreateMem(new_size * sizeof(int), (void **)&table);
1.1331 + pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1332 + pGroup = (AdminGroup *)m_pMemory->GetAddress(gid);
1.1333 +
1.1334 + /* Copy old data if necessary */
1.1335 + if (pUser->grp_table != -1)
1.1336 + {
1.1337 + int *old_table = (int *)m_pMemory->GetAddress(pUser->grp_table);
1.1338 + memcpy(table, old_table, sizeof(int) * pUser->grp_count);
1.1339 + }
1.1340 + pUser->grp_table = tblidx;
1.1341 + pUser->grp_size = new_size;
1.1342 + } else {
1.1343 + table = (int *)m_pMemory->GetAddress(pUser->grp_table);
1.1344 + }
1.1345 +
1.1346 + table[pUser->grp_count] = gid;
1.1347 + pUser->grp_count++;
1.1348 +
1.1349 + /* Compute new effective permissions */
1.1350 + pUser->eflags |= pGroup->addflags;
1.1351 +
1.1352 + if (pGroup->immunity_level > pUser->immunity_level)
1.1353 + {
1.1354 + pUser->immunity_level = pGroup->immunity_level;
1.1355 + }
1.1356 +
1.1357 + pUser->serialchange++;
1.1358 +
1.1359 + return true;
1.1360 +}
1.1361 +
1.1362 +bool AdminCache::IsValidAdmin(AdminId id)
1.1363 +{
1.1364 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1365 + return (pUser != NULL && pUser->magic == USR_MAGIC_SET);
1.1366 +}
1.1367 +
1.1368 +unsigned int AdminCache::GetAdminGroupCount(AdminId id)
1.1369 +{
1.1370 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1371 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.1372 + {
1.1373 + return 0;
1.1374 + }
1.1375 +
1.1376 + return pUser->grp_count;
1.1377 +}
1.1378 +
1.1379 +GroupId AdminCache::GetAdminGroup(AdminId id, unsigned int index, const char **name)
1.1380 +{
1.1381 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1382 + if (!pUser || pUser->magic != USR_MAGIC_SET || index >= pUser->grp_count)
1.1383 + {
1.1384 + return INVALID_GROUP_ID;
1.1385 + }
1.1386 +
1.1387 + int *table = (int *)m_pMemory->GetAddress(pUser->grp_table);
1.1388 +
1.1389 + GroupId gid = table[index];
1.1390 +
1.1391 + if (name)
1.1392 + {
1.1393 + *name = GetGroupName(gid);
1.1394 + }
1.1395 +
1.1396 + return gid;
1.1397 +}
1.1398 +
1.1399 +const char *AdminCache::GetAdminPassword(AdminId id)
1.1400 +{
1.1401 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1402 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.1403 + {
1.1404 + return NULL;
1.1405 + }
1.1406 +
1.1407 + return m_pStrings->GetString(pUser->password);
1.1408 +}
1.1409 +
1.1410 +void AdminCache::SetAdminPassword(AdminId id, const char *password)
1.1411 +{
1.1412 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1413 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.1414 + {
1.1415 + return;
1.1416 + }
1.1417 +
1.1418 + int i_password = m_pStrings->AddString(password);
1.1419 + pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1420 + pUser->password = i_password;
1.1421 +}
1.1422 +
1.1423 +unsigned int AdminCache::FlagBitsToBitArray(FlagBits bits, bool array[], unsigned int maxSize)
1.1424 +{
1.1425 + unsigned int i;
1.1426 + for (i=0; i<maxSize && i<AdminFlags_TOTAL; i++)
1.1427 + {
1.1428 + array[i] = ((bits & (1<<i)) == (unsigned)(1<<i));
1.1429 + }
1.1430 +
1.1431 + return i;
1.1432 +}
1.1433 +
1.1434 +FlagBits AdminCache::FlagBitArrayToBits(const bool array[], unsigned int maxSize)
1.1435 +{
1.1436 + FlagBits bits = 0;
1.1437 + for (unsigned int i=0; i<maxSize && i<AdminFlags_TOTAL; i++)
1.1438 + {
1.1439 + if (array[i])
1.1440 + {
1.1441 + bits |= (1<<i);
1.1442 + }
1.1443 + }
1.1444 +
1.1445 + return bits;
1.1446 +}
1.1447 +
1.1448 +FlagBits AdminCache::FlagArrayToBits(const AdminFlag array[], unsigned int numFlags)
1.1449 +{
1.1450 + FlagBits bits = 0;
1.1451 + for (unsigned int i=0; i<numFlags && i<AdminFlags_TOTAL; i++)
1.1452 + {
1.1453 + bits |= (1 << (FlagBits)array[i]);
1.1454 + }
1.1455 +
1.1456 + return bits;
1.1457 +}
1.1458 +
1.1459 +unsigned int AdminCache::FlagBitsToArray(FlagBits bits, AdminFlag array[], unsigned int maxSize)
1.1460 +{
1.1461 + unsigned int i, num=0;
1.1462 + for (i=0; num<maxSize && i<AdminFlags_TOTAL; i++)
1.1463 + {
1.1464 + if ((bits & (1<<i)) == (unsigned)(1<<i))
1.1465 + {
1.1466 + array[num++] = (AdminFlag)i;
1.1467 + }
1.1468 + }
1.1469 +
1.1470 + return num;
1.1471 +}
1.1472 +
1.1473 +bool AdminCache::CheckAdminFlags(AdminId id, FlagBits bits)
1.1474 +{
1.1475 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1476 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.1477 + {
1.1478 + return false;
1.1479 + }
1.1480 +
1.1481 + return ((pUser->eflags & bits) == bits);
1.1482 +}
1.1483 +
1.1484 +bool AdminCache::CanAdminTarget(AdminId id, AdminId target)
1.1485 +{
1.1486 + /**
1.1487 + * Zeroth, if the targeting AdminId is INVALID_ADMIN_ID, targeting fails.
1.1488 + * First, if the targeted AdminId is INVALID_ADMIN_ID, targeting succeeds.
1.1489 + */
1.1490 +
1.1491 + if (id == INVALID_ADMIN_ID)
1.1492 + {
1.1493 + return false;
1.1494 + }
1.1495 +
1.1496 + if (target == INVALID_ADMIN_ID)
1.1497 + {
1.1498 + return true;
1.1499 + }
1.1500 +
1.1501 + if (id == target)
1.1502 + {
1.1503 + return true;
1.1504 + }
1.1505 +
1.1506 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1507 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.1508 + {
1.1509 + return false;
1.1510 + }
1.1511 +
1.1512 + AdminUser *pTarget = (AdminUser *)m_pMemory->GetAddress(target);
1.1513 + if (!pTarget || pTarget->magic != USR_MAGIC_SET)
1.1514 + {
1.1515 + return false;
1.1516 + }
1.1517 +
1.1518 + /**
1.1519 + * Second, if the targeting admin is root, targeting succeeds.
1.1520 + */
1.1521 + if (pUser->eflags & ADMFLAG_ROOT)
1.1522 + {
1.1523 + return true;
1.1524 + }
1.1525 +
1.1526 + /** Fourth, if the targeted admin is immune from targeting admin. */
1.1527 + int mode = sm_immunity_mode.GetInt();
1.1528 + switch (mode)
1.1529 + {
1.1530 + case 1:
1.1531 + {
1.1532 + if (pTarget->immunity_level > pUser->immunity_level)
1.1533 + {
1.1534 + return false;
1.1535 + }
1.1536 + break;
1.1537 + }
1.1538 + case 3:
1.1539 + {
1.1540 + /* If neither has any immunity, let this pass. */
1.1541 + if (!pUser->immunity_level && !pTarget->immunity_level)
1.1542 + {
1.1543 + return true;
1.1544 + }
1.1545 + /* Don't break, go to the next case. */
1.1546 + }
1.1547 + case 2:
1.1548 + {
1.1549 + if (pTarget->immunity_level >= pUser->immunity_level)
1.1550 + {
1.1551 + return false;
1.1552 + }
1.1553 + break;
1.1554 + }
1.1555 + }
1.1556 +
1.1557 + /**
1.1558 + * Fifth, if the targeted admin has specific immunity from the
1.1559 + * targeting admin via group immunities, targeting fails.
1.1560 + */
1.1561 + //:TODO: speed this up... maybe with trie hacks.
1.1562 + //idea is to insert %d.%d in the trie after computing this and use it as a cache lookup.
1.1563 + //problem is the trie cannot delete prefixes, so we'd have a problem with invalidations.
1.1564 + if (pTarget->grp_count > 0 && pUser->grp_count > 0)
1.1565 + {
1.1566 + int *grp_table = (int *)m_pMemory->GetAddress(pTarget->grp_table);
1.1567 + int *src_table = (int *)m_pMemory->GetAddress(pUser->grp_table);
1.1568 + GroupId id, other;
1.1569 + unsigned int num;
1.1570 + for (unsigned int i=0; i<pTarget->grp_count; i++)
1.1571 + {
1.1572 + id = grp_table[i];
1.1573 + num = GetGroupImmunityCount(id);
1.1574 + for (unsigned int j=0; j<num; i++)
1.1575 + {
1.1576 + other = GetGroupImmunity(id, j);
1.1577 + for (unsigned int k=0; k<pUser->grp_count; k++)
1.1578 + {
1.1579 + if (other == src_table[k])
1.1580 + {
1.1581 + return false;
1.1582 + }
1.1583 + }
1.1584 + }
1.1585 + }
1.1586 + }
1.1587 +
1.1588 + return true;
1.1589 +}
1.1590 +
1.1591 +bool AdminCache::FindFlag(char c, AdminFlag *pAdmFlag)
1.1592 +{
1.1593 + if (c < 'a' || c > 'z')
1.1594 + {
1.1595 + return false;
1.1596 + }
1.1597 +
1.1598 + if (pAdmFlag)
1.1599 + {
1.1600 + *pAdmFlag = g_FlagLetters[(unsigned)c - (unsigned)'a'];
1.1601 + }
1.1602 +
1.1603 + return true;
1.1604 +}
1.1605 +
1.1606 +FlagBits AdminCache::ReadFlagString(const char *flags, const char **end)
1.1607 +{
1.1608 + FlagBits bits = 0;
1.1609 +
1.1610 + while (flags && (*flags != '\0'))
1.1611 + {
1.1612 + AdminFlag flag;
1.1613 + if (!FindFlag(*flags, &flag))
1.1614 + {
1.1615 + break;
1.1616 + }
1.1617 + bits |= FlagArrayToBits(&flag, 1);
1.1618 + flags++;
1.1619 + }
1.1620 +
1.1621 + if (end)
1.1622 + {
1.1623 + *end = flags;
1.1624 + }
1.1625 +
1.1626 + return bits;
1.1627 +}
1.1628 +
1.1629 +unsigned int AdminCache::GetAdminSerialChange(AdminId id)
1.1630 +{
1.1631 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1632 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.1633 + {
1.1634 + return 0;
1.1635 + }
1.1636 +
1.1637 + return pUser->serialchange;
1.1638 +}
1.1639 +
1.1640 +bool AdminCache::CanAdminUseCommand(int client, const char *cmd)
1.1641 +{
1.1642 + FlagBits bits;
1.1643 + OverrideType otype = Override_Command;
1.1644 +
1.1645 + if (cmd[0] == '@')
1.1646 + {
1.1647 + otype = Override_CommandGroup;
1.1648 + cmd++;
1.1649 + }
1.1650 +
1.1651 + if (!g_ConCmds.LookForCommandAdminFlags(cmd, &bits))
1.1652 + {
1.1653 + if (!GetCommandOverride(cmd, otype, &bits))
1.1654 + {
1.1655 + bits = 0;
1.1656 + }
1.1657 + }
1.1658 +
1.1659 + return g_ConCmds.CheckCommandAccess(client, cmd, bits);
1.1660 +}
1.1661 +
1.1662 +unsigned int AdminCache::SetGroupImmunityLevel(GroupId gid, unsigned int level)
1.1663 +{
1.1664 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid);
1.1665 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.1666 + {
1.1667 + return 0;
1.1668 + }
1.1669 +
1.1670 + unsigned int old_level = pGroup->immunity_level;
1.1671 +
1.1672 + pGroup->immunity_level = level;
1.1673 +
1.1674 + return old_level;
1.1675 +}
1.1676 +
1.1677 +unsigned int AdminCache::GetGroupImmunityLevel(GroupId gid)
1.1678 +{
1.1679 + AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid);
1.1680 + if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
1.1681 + {
1.1682 + return 0;
1.1683 + }
1.1684 +
1.1685 + return pGroup->immunity_level;
1.1686 +}
1.1687 +
1.1688 +unsigned int AdminCache::SetAdminImmunityLevel(AdminId id, unsigned int level)
1.1689 +{
1.1690 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1691 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.1692 + {
1.1693 + return 0;
1.1694 + }
1.1695 +
1.1696 + unsigned int old_level = pUser->immunity_level;
1.1697 +
1.1698 + pUser->immunity_level = level;
1.1699 +
1.1700 + return old_level;
1.1701 +}
1.1702 +
1.1703 +unsigned int AdminCache::GetAdminImmunityLevel(AdminId id)
1.1704 +{
1.1705 + AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
1.1706 + if (!pUser || pUser->magic != USR_MAGIC_SET)
1.1707 + {
1.1708 + return 0;
1.1709 + }
1.1710 +
1.1711 + return pUser->immunity_level;
1.1712 +}
1.1713 +
1.1714 +bool AdminCache::CheckAccess(int client, const char *cmd, FlagBits flags, bool override_only)
1.1715 +{
1.1716 + if (client == 0)
1.1717 + {
1.1718 + return true;
1.1719 + }
1.1720 +
1.1721 + /* Auto-detect a command if we can */
1.1722 + FlagBits bits = flags;
1.1723 + bool found_command = false;
1.1724 + if (!override_only)
1.1725 + {
1.1726 + found_command = g_ConCmds.LookForCommandAdminFlags(cmd, &bits);
1.1727 + }
1.1728 +
1.1729 + if (!found_command)
1.1730 + {
1.1731 + GetCommandOverride(cmd, Override_Command, &bits);
1.1732 + }
1.1733 +
1.1734 + return g_ConCmds.CheckCommandAccess(client, cmd, bits) ? 1 : 0;
1.1735 +}
