1.1 --- a/core/AdminCache.cpp Fri Mar 28 06:21:34 2008 +0000
1.2 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000
1.3 @@ -1,1732 +0,0 @@
1.4 -/**
1.5 - * vim: set ts=4 :
1.6 - * =============================================================================
1.7 - * SourceMod
1.8 - * Copyright (C) 2004-2007 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|FCVAR_PROTECTED, "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 -}
