|
ds@1952
|
1 |
/** |
|
ds@1952
|
2 |
* vim: set ts=4 : |
|
ds@1952
|
3 |
* ============================================================================= |
|
ds@1952
|
4 |
* SourceMod |
|
ds@1952
|
5 |
* Copyright (C) 2004-2008 AlliedModders LLC. All rights reserved. |
|
ds@1952
|
6 |
* ============================================================================= |
|
ds@1952
|
7 |
* |
|
ds@1952
|
8 |
* This program is free software; you can redistribute it and/or modify it under |
|
ds@1952
|
9 |
* the terms of the GNU General Public License, version 3.0, as published by the |
|
ds@1952
|
10 |
* Free Software Foundation. |
|
ds@1952
|
11 |
* |
|
ds@1952
|
12 |
* This program is distributed in the hope that it will be useful, but WITHOUT |
|
ds@1952
|
13 |
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS |
|
ds@1952
|
14 |
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more |
|
ds@1952
|
15 |
* details. |
|
ds@1952
|
16 |
* |
|
ds@1952
|
17 |
* You should have received a copy of the GNU General Public License along with |
|
ds@1952
|
18 |
* this program. If not, see <http://www.gnu.org/licenses/>. |
|
ds@1952
|
19 |
* |
|
ds@1952
|
20 |
* As a special exception, AlliedModders LLC gives you permission to link the |
|
ds@1952
|
21 |
* code of this program (as well as its derivative works) to "Half-Life 2," the |
|
ds@1952
|
22 |
* "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software |
|
ds@1952
|
23 |
* by the Valve Corporation. You must obey the GNU General Public License in |
|
ds@1952
|
24 |
* all respects for all other code used. Additionally, AlliedModders LLC grants |
|
ds@1952
|
25 |
* this exception to all derivative works. AlliedModders LLC defines further |
|
ds@1952
|
26 |
* exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007), |
|
ds@1952
|
27 |
* or <http://www.sourcemod.net/license.php>. |
|
ds@1952
|
28 |
* |
|
ds@1952
|
29 |
* Version: $Id$ |
|
ds@1952
|
30 |
*/ |
|
ds@1952
|
31 |
|
|
ds@1952
|
32 |
#include <string.h> |
|
ds@1952
|
33 |
#include <assert.h> |
|
ds@1952
|
34 |
#include <ITextParsers.h> |
|
ds@1952
|
35 |
#include "AdminCache.h" |
|
ds@1952
|
36 |
#include "ShareSys.h" |
|
ds@1952
|
37 |
#include "ForwardSys.h" |
|
ds@1952
|
38 |
#include "PlayerManager.h" |
|
ds@1952
|
39 |
#include "ConCmdManager.h" |
|
ds@1952
|
40 |
#include "Logger.h" |
|
ds@1952
|
41 |
#include "sourcemod.h" |
|
ds@1952
|
42 |
#include "sm_stringutil.h" |
|
ds@1952
|
43 |
#include "sourcemm_api.h" |
|
dvander@1965
|
44 |
#include "sm_srvcmds.h" |
|
ds@1952
|
45 |
|
|
ds@1952
|
46 |
#define LEVEL_STATE_NONE 0 |
|
ds@1952
|
47 |
#define LEVEL_STATE_LEVELS 1 |
|
ds@1952
|
48 |
#define LEVEL_STATE_FLAGS 2 |
|
ds@1952
|
49 |
|
|
ds@1952
|
50 |
AdminCache g_Admins; |
|
dvander@1965
|
51 |
char g_ReverseFlags[26]; |
|
ds@1952
|
52 |
AdminFlag g_FlagLetters[26]; |
|
ds@1952
|
53 |
bool g_FlagSet[26]; |
|
ds@1952
|
54 |
|
|
ds@1952
|
55 |
ConVar sm_immunity_mode("sm_immunity_mode", "1", FCVAR_SPONLY, "Mode for deciding immunity protection"); |
|
ds@1952
|
56 |
|
|
ds@1952
|
57 |
/* Default flags */ |
|
ds@1952
|
58 |
AdminFlag g_DefaultFlags[26] = |
|
ds@1952
|
59 |
{ |
|
ds@1952
|
60 |
Admin_Reservation, Admin_Generic, Admin_Kick, Admin_Ban, Admin_Unban, |
|
ds@1952
|
61 |
Admin_Slay, Admin_Changemap, Admin_Convars, Admin_Config, Admin_Chat, |
|
ds@1952
|
62 |
Admin_Vote, Admin_Password, Admin_RCON, Admin_Cheats, Admin_Custom1, |
|
ds@1952
|
63 |
Admin_Custom2, Admin_Custom3, Admin_Custom4, Admin_Custom5, Admin_Custom6, |
|
ds@1952
|
64 |
Admin_Generic, Admin_Generic, Admin_Generic, Admin_Generic, Admin_Generic, |
|
ds@1952
|
65 |
Admin_Root |
|
ds@1952
|
66 |
}; |
|
ds@1952
|
67 |
|
|
ds@1952
|
68 |
|
|
ds@1952
|
69 |
class FlagReader : public ITextListener_SMC |
|
ds@1952
|
70 |
{ |
|
ds@1952
|
71 |
public: |
|
ds@1952
|
72 |
void LoadLevels() |
|
ds@1952
|
73 |
{ |
|
ds@1952
|
74 |
if (!Parse()) |
|
ds@1952
|
75 |
{ |
|
ds@1952
|
76 |
memcpy(g_FlagLetters, g_DefaultFlags, sizeof(AdminFlag) * 26); |
|
ds@1952
|
77 |
for (unsigned int i=0; i<20; i++) |
|
ds@1952
|
78 |
{ |
|
ds@1952
|
79 |
g_FlagSet[i] = true; |
|
ds@1952
|
80 |
} |
|
ds@1952
|
81 |
g_FlagSet[25] = true; |
|
ds@1952
|
82 |
} |
|
ds@1952
|
83 |
} |
|
ds@1952
|
84 |
private: |
|
ds@1952
|
85 |
bool Parse() |
|
ds@1952
|
86 |
{ |
|
ds@1952
|
87 |
SMCStates states; |
|
ds@1952
|
88 |
SMCError error; |
|
ds@1952
|
89 |
|
|
ds@1952
|
90 |
m_bFileNameLogged = false; |
|
ds@1952
|
91 |
g_SourceMod.BuildPath(Path_SM, m_File, sizeof(m_File), "configs/admin_levels.cfg"); |
|
ds@1952
|
92 |
|
|
ds@1952
|
93 |
if ((error = textparsers->ParseFile_SMC(m_File, this, &states)) |
|
ds@1952
|
94 |
!= SMCError_Okay) |
|
ds@1952
|
95 |
{ |
|
ds@1952
|
96 |
const char *err_string = textparsers->GetSMCErrorString(error); |
|
ds@1952
|
97 |
if (!err_string) |
|
ds@1952
|
98 |
{ |
|
ds@1952
|
99 |
err_string = "Unknown error"; |
|
ds@1952
|
100 |
} |
|
ds@1952
|
101 |
ParseError(NULL, "Error %d (%s)", error, err_string); |
|
ds@1952
|
102 |
return false; |
|
ds@1952
|
103 |
} |
|
ds@1952
|
104 |
|
|
ds@1952
|
105 |
return true; |
|
ds@1952
|
106 |
} |
|
ds@1952
|
107 |
void ReadSMC_ParseStart() |
|
ds@1952
|
108 |
{ |
|
ds@1952
|
109 |
m_LevelState = LEVEL_STATE_NONE; |
|
ds@1952
|
110 |
m_IgnoreLevel = 0; |
|
ds@1952
|
111 |
memset(g_FlagSet, 0, sizeof(g_FlagSet)); |
|
ds@1952
|
112 |
} |
|
ds@1952
|
113 |
SMCResult ReadSMC_NewSection(const SMCStates *states, const char *name) |
|
ds@1952
|
114 |
{ |
|
ds@1952
|
115 |
if (m_IgnoreLevel) |
|
ds@1952
|
116 |
{ |
|
ds@1952
|
117 |
m_IgnoreLevel++; |
|
ds@1952
|
118 |
return SMCResult_Continue; |
|
ds@1952
|
119 |
} |
|
ds@1952
|
120 |
|
|
ds@1952
|
121 |
if (m_LevelState == LEVEL_STATE_NONE) |
|
ds@1952
|
122 |
{ |
|
ds@1952
|
123 |
if (strcmp(name, "Levels") == 0) |
|
ds@1952
|
124 |
{ |
|
ds@1952
|
125 |
m_LevelState = LEVEL_STATE_LEVELS; |
|
ds@1952
|
126 |
} |
|
ds@1952
|
127 |
else |
|
ds@1952
|
128 |
{ |
|
ds@1952
|
129 |
m_IgnoreLevel++; |
|
ds@1952
|
130 |
} |
|
ds@1952
|
131 |
} else if (m_LevelState == LEVEL_STATE_LEVELS) { |
|
ds@1952
|
132 |
if (strcmp(name, "Flags") == 0) |
|
ds@1952
|
133 |
{ |
|
ds@1952
|
134 |
m_LevelState = LEVEL_STATE_FLAGS; |
|
ds@1952
|
135 |
} |
|
ds@1952
|
136 |
else |
|
ds@1952
|
137 |
{ |
|
ds@1952
|
138 |
m_IgnoreLevel++; |
|
ds@1952
|
139 |
} |
|
ds@1952
|
140 |
} |
|
ds@1952
|
141 |
else |
|
ds@1952
|
142 |
{ |
|
ds@1952
|
143 |
m_IgnoreLevel++; |
|
ds@1952
|
144 |
} |
|
ds@1952
|
145 |
|
|
ds@1952
|
146 |
return SMCResult_Continue; |
|
ds@1952
|
147 |
} |
|
ds@1952
|
148 |
SMCResult ReadSMC_KeyValue(const SMCStates *states, const char *key, const char *value) |
|
ds@1952
|
149 |
{ |
|
ds@1952
|
150 |
if (m_LevelState != LEVEL_STATE_FLAGS || m_IgnoreLevel) |
|
ds@1952
|
151 |
{ |
|
ds@1952
|
152 |
return SMCResult_Continue; |
|
ds@1952
|
153 |
} |
|
ds@1952
|
154 |
|
|
ds@1952
|
155 |
unsigned char c = (unsigned)value[0]; |
|
ds@1952
|
156 |
|
|
ds@1952
|
157 |
if (c < (unsigned)'a' || c > (unsigned)'z') |
|
ds@1952
|
158 |
{ |
|
ds@1952
|
159 |
ParseError(states, "Flag \"%c\" is not a lower-case ASCII letter", c); |
|
ds@1952
|
160 |
return SMCResult_Continue; |
|
ds@1952
|
161 |
} |
|
ds@1952
|
162 |
|
|
ds@1952
|
163 |
c -= (unsigned)'a'; |
|
ds@1952
|
164 |
|
|
ds@1952
|
165 |
if (!g_Admins.FindFlag(key, &g_FlagLetters[c])) |
|
ds@1952
|
166 |
{ |
|
ds@1952
|
167 |
ParseError(states, "Unrecognized admin level \"%s\"", key); |
|
ds@1952
|
168 |
return SMCResult_Continue; |
|
ds@1952
|
169 |
} |
|
ds@1952
|
170 |
|
|
ds@1952
|
171 |
g_FlagSet[c] = true; |
|
ds@1952
|
172 |
|
|
ds@1952
|
173 |
return SMCResult_Continue; |
|
ds@1952
|
174 |
} |
|
ds@1952
|
175 |
SMCResult ReadSMC_LeavingSection(const SMCStates *states) |
|
ds@1952
|
176 |
{ |
|
ds@1952
|
177 |
if (m_IgnoreLevel) |
|
ds@1952
|
178 |
{ |
|
ds@1952
|
179 |
m_IgnoreLevel--; |
|
ds@1952
|
180 |
return SMCResult_Continue; |
|
ds@1952
|
181 |
} |
|
ds@1952
|
182 |
|
|
ds@1952
|
183 |
if (m_LevelState == LEVEL_STATE_FLAGS) |
|
ds@1952
|
184 |
{ |
|
ds@1952
|
185 |
m_LevelState = LEVEL_STATE_LEVELS; |
|
ds@1952
|
186 |
return SMCResult_Halt; |
|
ds@1952
|
187 |
} |
|
ds@1952
|
188 |
else if (m_LevelState == LEVEL_STATE_LEVELS) |
|
ds@1952
|
189 |
{ |
|
ds@1952
|
190 |
m_LevelState = LEVEL_STATE_NONE; |
|
ds@1952
|
191 |
} |
|
ds@1952
|
192 |
|
|
ds@1952
|
193 |
return SMCResult_Continue; |
|
ds@1952
|
194 |
} |
|
ds@1952
|
195 |
void ParseError(const SMCStates *states, const char *message, ...) |
|
ds@1952
|
196 |
{ |
|
ds@1952
|
197 |
va_list ap; |
|
ds@1952
|
198 |
char buffer[256]; |
|
ds@1952
|
199 |
|
|
ds@1952
|
200 |
va_start(ap, message); |
|
ds@1952
|
201 |
UTIL_FormatArgs(buffer, sizeof(buffer), message, ap); |
|
ds@1952
|
202 |
va_end(ap); |
|
ds@1952
|
203 |
|
|
ds@1952
|
204 |
if (!m_bFileNameLogged) |
|
ds@1952
|
205 |
{ |
|
ds@1952
|
206 |
g_Logger.LogError("[SM] Parse error(s) detected in file \"%s\":", m_File); |
|
ds@1952
|
207 |
m_bFileNameLogged = true; |
|
ds@1952
|
208 |
} |
|
ds@1952
|
209 |
|
|
ds@1952
|
210 |
g_Logger.LogError("[SM] (Line %d): %s", states ? states->line : 0, buffer); |
|
ds@1952
|
211 |
} |
|
ds@1952
|
212 |
private: |
|
ds@1952
|
213 |
bool m_bFileNameLogged; |
|
ds@1952
|
214 |
char m_File[PLATFORM_MAX_PATH]; |
|
ds@1952
|
215 |
int m_LevelState; |
|
ds@1952
|
216 |
int m_IgnoreLevel; |
|
ds@1952
|
217 |
} s_FlagReader; |
|
ds@1952
|
218 |
|
|
ds@1952
|
219 |
AdminCache::AdminCache() |
|
ds@1952
|
220 |
{ |
|
ds@1952
|
221 |
m_pCmdOverrides = sm_trie_create(); |
|
ds@1952
|
222 |
m_pCmdGrpOverrides = sm_trie_create(); |
|
ds@1952
|
223 |
m_pStrings = new BaseStringTable(1024); |
|
ds@1952
|
224 |
m_pMemory = m_pStrings->GetMemTable(); |
|
ds@1952
|
225 |
m_FreeGroupList = m_FirstGroup = m_LastGroup = INVALID_GROUP_ID; |
|
ds@1952
|
226 |
m_FreeUserList = m_FirstUser = m_LastUser = INVALID_ADMIN_ID; |
|
ds@1952
|
227 |
m_pGroups = sm_trie_create(); |
|
ds@1952
|
228 |
m_pCacheFwd = NULL; |
|
ds@1952
|
229 |
m_FirstGroup = -1; |
|
ds@1952
|
230 |
m_pAuthTables = sm_trie_create(); |
|
ds@1952
|
231 |
m_InvalidatingAdmins = false; |
|
ds@1952
|
232 |
m_destroying = false; |
|
ds@1952
|
233 |
m_pLevelNames = sm_trie_create(); |
|
ds@1952
|
234 |
} |
|
ds@1952
|
235 |
|
|
ds@1952
|
236 |
AdminCache::~AdminCache() |
|
ds@1952
|
237 |
{ |
|
ds@1952
|
238 |
m_destroying = true; |
|
ds@1952
|
239 |
DumpAdminCache(AdminCache_Overrides, false); |
|
ds@1952
|
240 |
DumpAdminCache(AdminCache_Groups, false); |
|
ds@1952
|
241 |
|
|
ds@1952
|
242 |
sm_trie_destroy(m_pCmdGrpOverrides); |
|
ds@1952
|
243 |
sm_trie_destroy(m_pCmdOverrides); |
|
ds@1952
|
244 |
|
|
ds@1952
|
245 |
if (m_pGroups) |
|
ds@1952
|
246 |
{ |
|
ds@1952
|
247 |
sm_trie_destroy(m_pGroups); |
|
ds@1952
|
248 |
} |
|
ds@1952
|
249 |
|
|
ds@1952
|
250 |
List<AuthMethod>::iterator iter; |
|
ds@1952
|
251 |
for (iter=m_AuthMethods.begin(); |
|
ds@1952
|
252 |
iter!=m_AuthMethods.end(); |
|
ds@1952
|
253 |
iter++) |
|
ds@1952
|
254 |
{ |
|
ds@1952
|
255 |
sm_trie_destroy((*iter).table); |
|
ds@1952
|
256 |
} |
|
ds@1952
|
257 |
|
|
ds@1952
|
258 |
sm_trie_destroy(m_pAuthTables); |
|
ds@1952
|
259 |
|
|
ds@1952
|
260 |
delete m_pStrings; |
|
ds@1952
|
261 |
|
|
ds@1952
|
262 |
sm_trie_destroy(m_pLevelNames); |
|
ds@1952
|
263 |
} |
|
ds@1952
|
264 |
|
|
ds@1952
|
265 |
void AdminCache::OnSourceModStartup(bool late) |
|
ds@1952
|
266 |
{ |
|
ds@1952
|
267 |
RegisterAuthIdentType("steam"); |
|
ds@1952
|
268 |
RegisterAuthIdentType("name"); |
|
ds@1952
|
269 |
RegisterAuthIdentType("ip"); |
|
ds@1952
|
270 |
|
|
ds@1952
|
271 |
NameFlag("reservation", Admin_Reservation); |
|
ds@1952
|
272 |
NameFlag("kick", Admin_Kick); |
|
ds@1952
|
273 |
NameFlag("generic", Admin_Generic); |
|
ds@1952
|
274 |
NameFlag("ban", Admin_Ban); |
|
ds@1952
|
275 |
NameFlag("unban", Admin_Unban); |
|
ds@1952
|
276 |
NameFlag("slay", Admin_Slay); |
|
ds@1952
|
277 |
NameFlag("changemap", Admin_Changemap); |
|
ds@1952
|
278 |
NameFlag("cvars", Admin_Convars); |
|
ds@1952
|
279 |
NameFlag("config", Admin_Config); |
|
ds@1952
|
280 |
NameFlag("chat", Admin_Chat); |
|
ds@1952
|
281 |
NameFlag("vote", Admin_Vote); |
|
ds@1952
|
282 |
NameFlag("password", Admin_Password); |
|
ds@1952
|
283 |
NameFlag("rcon", Admin_RCON); |
|
ds@1952
|
284 |
NameFlag("cheats", Admin_Cheats); |
|
ds@1952
|
285 |
NameFlag("root", Admin_Root); |
|
ds@1952
|
286 |
NameFlag("custom1", Admin_Custom1); |
|
ds@1952
|
287 |
NameFlag("custom2", Admin_Custom2); |
|
ds@1952
|
288 |
NameFlag("custom3", Admin_Custom3); |
|
ds@1952
|
289 |
NameFlag("custom4", Admin_Custom4); |
|
ds@1952
|
290 |
NameFlag("custom5", Admin_Custom5); |
|
ds@1952
|
291 |
NameFlag("custom6", Admin_Custom6); |
|
ds@1952
|
292 |
} |
|
ds@1952
|
293 |
|
|
ds@1952
|
294 |
void AdminCache::OnSourceModAllInitialized() |
|
ds@1952
|
295 |
{ |
|
ds@1952
|
296 |
m_pCacheFwd = g_Forwards.CreateForward("OnRebuildAdminCache", ET_Ignore, 1, NULL, Param_Cell); |
|
ds@1952
|
297 |
g_ShareSys.AddInterface(NULL, this); |
|
ds@1952
|
298 |
} |
|
ds@1952
|
299 |
|
|
ds@1952
|
300 |
void AdminCache::OnSourceModLevelChange(const char *mapName) |
|
ds@1952
|
301 |
{ |
|
dvander@1965
|
302 |
int i; |
|
dvander@1965
|
303 |
AdminFlag flag; |
|
dvander@1965
|
304 |
|
|
ds@1952
|
305 |
/* For now, we only read these once per level. */ |
|
ds@1952
|
306 |
s_FlagReader.LoadLevels(); |
|
dvander@1965
|
307 |
|
|
dvander@1965
|
308 |
for (i = 0; i < 26; i++) |
|
dvander@1965
|
309 |
{ |
|
dvander@1965
|
310 |
if (FindFlag('a' + i, &flag)) |
|
dvander@1965
|
311 |
{ |
|
dvander@1965
|
312 |
g_ReverseFlags[flag] = 'a' + i; |
|
dvander@1965
|
313 |
} |
|
dvander@1965
|
314 |
else |
|
dvander@1965
|
315 |
{ |
|
dvander@1965
|
316 |
g_ReverseFlags[flag] = '?'; |
|
dvander@1965
|
317 |
} |
|
dvander@1965
|
318 |
} |
|
ds@1952
|
319 |
} |
|
ds@1952
|
320 |
|
|
ds@1952
|
321 |
void AdminCache::OnSourceModShutdown() |
|
ds@1952
|
322 |
{ |
|
ds@1952
|
323 |
g_Forwards.ReleaseForward(m_pCacheFwd); |
|
ds@1952
|
324 |
m_pCacheFwd = NULL; |
|
ds@1952
|
325 |
} |
|
ds@1952
|
326 |
|
|
ds@1952
|
327 |
void AdminCache::OnSourceModPluginsLoaded() |
|
ds@1952
|
328 |
{ |
|
ds@1952
|
329 |
DumpAdminCache(AdminCache_Overrides, true); |
|
ds@1952
|
330 |
DumpAdminCache(AdminCache_Groups, true); |
|
ds@1952
|
331 |
} |
|
ds@1952
|
332 |
|
|
ds@1952
|
333 |
void AdminCache::NameFlag(const char *str, AdminFlag flag) |
|
ds@1952
|
334 |
{ |
|
ds@1952
|
335 |
sm_trie_insert(m_pLevelNames, str, (void *)flag); |
|
ds@1952
|
336 |
} |
|
ds@1952
|
337 |
|
|
ds@1952
|
338 |
bool AdminCache::FindFlag(const char *str, AdminFlag *pFlag) |
|
ds@1952
|
339 |
{ |
|
ds@1952
|
340 |
void *obj; |
|
ds@1952
|
341 |
if (!sm_trie_retrieve(m_pLevelNames, str, &obj)) |
|
ds@1952
|
342 |
{ |
|
ds@1952
|
343 |
return false; |
|
ds@1952
|
344 |
} |
|
ds@1952
|
345 |
|
|
ds@1952
|
346 |
if (pFlag) |
|
ds@1952
|
347 |
{ |
|
ds@1952
|
348 |
*pFlag = (AdminFlag)(int)obj; |
|
ds@1952
|
349 |
} |
|
ds@1952
|
350 |
|
|
ds@1952
|
351 |
return true; |
|
ds@1952
|
352 |
} |
|
ds@1952
|
353 |
|
|
ds@1952
|
354 |
void AdminCache::AddCommandOverride(const char *cmd, OverrideType type, FlagBits flags) |
|
ds@1952
|
355 |
{ |
|
ds@1952
|
356 |
Trie *pTrie = NULL; |
|
ds@1952
|
357 |
if (type == Override_Command) |
|
ds@1952
|
358 |
{ |
|
ds@1952
|
359 |
pTrie = m_pCmdOverrides; |
|
ds@1952
|
360 |
} else if (type == Override_CommandGroup) { |
|
ds@1952
|
361 |
pTrie = m_pCmdGrpOverrides; |
|
ds@1952
|
362 |
} else { |
|
ds@1952
|
363 |
return; |
|
ds@1952
|
364 |
} |
|
ds@1952
|
365 |
|
|
ds@1952
|
366 |
sm_trie_insert(pTrie, cmd, (void *)(unsigned int)flags); |
|
ds@1952
|
367 |
|
|
ds@1952
|
368 |
g_ConCmds.UpdateAdminCmdFlags(cmd, type, flags, false); |
|
ds@1952
|
369 |
} |
|
ds@1952
|
370 |
|
|
ds@1952
|
371 |
bool AdminCache::GetCommandOverride(const char *cmd, OverrideType type, FlagBits *pFlags) |
|
ds@1952
|
372 |
{ |
|
ds@1952
|
373 |
Trie *pTrie = NULL; |
|
ds@1952
|
374 |
|
|
ds@1952
|
375 |
if (type == Override_Command) |
|
ds@1952
|
376 |
{ |
|
ds@1952
|
377 |
pTrie = m_pCmdOverrides; |
|
ds@1952
|
378 |
} else if (type == Override_CommandGroup) { |
|
ds@1952
|
379 |
pTrie = m_pCmdGrpOverrides; |
|
ds@1952
|
380 |
} else { |
|
ds@1952
|
381 |
return false; |
|
ds@1952
|
382 |
} |
|
ds@1952
|
383 |
|
|
ds@1952
|
384 |
void *object; |
|
ds@1952
|
385 |
if (sm_trie_retrieve(pTrie, cmd, &object)) |
|
ds@1952
|
386 |
{ |
|
ds@1952
|
387 |
if (pFlags) |
|
ds@1952
|
388 |
{ |
|
ds@1952
|
389 |
*pFlags = (FlagBits)object; |
|
ds@1952
|
390 |
} |
|
ds@1952
|
391 |
return true; |
|
ds@1952
|
392 |
} |
|
ds@1952
|
393 |
|
|
ds@1952
|
394 |
return false; |
|
ds@1952
|
395 |
} |
|
ds@1952
|
396 |
|
|
ds@1952
|
397 |
void AdminCache::UnsetCommandOverride(const char *cmd, OverrideType type) |
|
ds@1952
|
398 |
{ |
|
ds@1952
|
399 |
if (type == Override_Command) |
|
ds@1952
|
400 |
{ |
|
ds@1952
|
401 |
return _UnsetCommandOverride(cmd); |
|
ds@1952
|
402 |
} else if (type == Override_CommandGroup) { |
|
ds@1952
|
403 |
return _UnsetCommandGroupOverride(cmd); |
|
ds@1952
|
404 |
} |
|
ds@1952
|
405 |
} |
|
ds@1952
|
406 |
|
|
ds@1952
|
407 |
void AdminCache::_UnsetCommandGroupOverride(const char *group) |
|
ds@1952
|
408 |
{ |
|
ds@1952
|
409 |
if (!m_pCmdGrpOverrides) |
|
ds@1952
|
410 |
{ |
|
ds@1952
|
411 |
return; |
|
ds@1952
|
412 |
} |
|
ds@1952
|
413 |
|
|
ds@1952
|
414 |
sm_trie_delete(m_pCmdGrpOverrides, group); |
|
ds@1952
|
415 |
|
|
ds@1952
|
416 |
g_ConCmds.UpdateAdminCmdFlags(group, Override_CommandGroup, 0, true); |
|
ds@1952
|
417 |
} |
|
ds@1952
|
418 |
|
|
ds@1952
|
419 |
void AdminCache::_UnsetCommandOverride(const char *cmd) |
|
ds@1952
|
420 |
{ |
|
ds@1952
|
421 |
if (!m_pCmdOverrides) |
|
ds@1952
|
422 |
{ |
|
ds@1952
|
423 |
return; |
|
ds@1952
|
424 |
} |
|
ds@1952
|
425 |
|
|
ds@1952
|
426 |
sm_trie_delete(m_pCmdOverrides, cmd); |
|
ds@1952
|
427 |
|
|
ds@1952
|
428 |
g_ConCmds.UpdateAdminCmdFlags(cmd, Override_Command, 0, true); |
|
ds@1952
|
429 |
} |
|
ds@1952
|
430 |
|
|
ds@1952
|
431 |
void AdminCache::DumpCommandOverrideCache(OverrideType type) |
|
ds@1952
|
432 |
{ |
|
ds@1952
|
433 |
if (type == Override_Command && m_pCmdOverrides) |
|
ds@1952
|
434 |
{ |
|
ds@1952
|
435 |
sm_trie_clear(m_pCmdOverrides); |
|
ds@1952
|
436 |
} else if (type == Override_CommandGroup && m_pCmdGrpOverrides) { |
|
ds@1952
|
437 |
sm_trie_clear(m_pCmdGrpOverrides); |
|
ds@1952
|
438 |
} |
|
ds@1952
|
439 |
} |
|
ds@1952
|
440 |
|
|
ds@1952
|
441 |
AdminId AdminCache::CreateAdmin(const char *name) |
|
ds@1952
|
442 |
{ |
|
ds@1952
|
443 |
AdminId id; |
|
ds@1952
|
444 |
AdminUser *pUser; |
|
ds@1952
|
445 |
|
|
ds@1952
|
446 |
if (m_FreeUserList != INVALID_ADMIN_ID) |
|
ds@1952
|
447 |
{ |
|
ds@1952
|
448 |
pUser = (AdminUser *)m_pMemory->GetAddress(m_FreeUserList); |
|
ds@1952
|
449 |
assert(pUser->magic == USR_MAGIC_UNSET); |
|
ds@1952
|
450 |
id = m_FreeUserList; |
|
ds@1952
|
451 |
m_FreeUserList = pUser->next_user; |
|
dvander@1965
|
452 |
} |
|
dvander@1965
|
453 |
else |
|
dvander@1965
|
454 |
{ |
|
ds@1952
|
455 |
id = m_pMemory->CreateMem(sizeof(AdminUser), (void **)&pUser); |
|
ds@1952
|
456 |
pUser->grp_size = 0; |
|
ds@1952
|
457 |
pUser->grp_table = -1; |
|
ds@1952
|
458 |
} |
|
ds@1952
|
459 |
|
|
ds@1952
|
460 |
pUser->flags = 0; |
|
ds@1952
|
461 |
pUser->eflags = 0; |
|
ds@1952
|
462 |
pUser->grp_count = 0; |
|
ds@1952
|
463 |
pUser->password = -1; |
|
ds@1952
|
464 |
pUser->magic = USR_MAGIC_SET; |
|
ds@1952
|
465 |
pUser->auth.identidx = -1; |
|
ds@1952
|
466 |
pUser->auth.index = 0; |
|
ds@1952
|
467 |
pUser->immunity_level = 0; |
|
ds@1952
|
468 |
pUser->serialchange = 1; |
|
ds@1952
|
469 |
|
|
ds@1952
|
470 |
if (m_FirstUser == INVALID_ADMIN_ID) |
|
ds@1952
|
471 |
{ |
|
ds@1952
|
472 |
m_FirstUser = id; |
|
ds@1952
|
473 |
m_LastUser = id; |
|
dvander@1965
|
474 |
} |
|
dvander@1965
|
475 |
else |
|
dvander@1965
|
476 |
{ |
|
ds@1952
|
477 |
AdminUser *pPrev = (AdminUser *)m_pMemory->GetAddress(m_LastUser); |
|
ds@1952
|
478 |
pPrev->next_user = id; |
|
ds@1952
|
479 |
pUser->prev_user = m_LastUser; |
|
ds@1952
|
480 |
m_LastUser = id; |
|
ds@1952
|
481 |
} |
|
ds@1952
|
482 |
|
|
dvander@1965
|
483 |
/* Since we always append to the tail, we should invalidate their next */ |
|
dvander@1965
|
484 |
pUser->next_user = -1; |
|
dvander@1965
|
485 |
|
|
ds@1952
|
486 |
if (name && name[0] != '\0') |
|
ds@1952
|
487 |
{ |
|
ds@1952
|
488 |
int nameidx = m_pStrings->AddString(name); |
|
ds@1952
|
489 |
pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
490 |
pUser->nameidx = nameidx; |
|
ds@1952
|
491 |
} |
|
dvander@1965
|
492 |
else |
|
dvander@1965
|
493 |
{ |
|
dvander@1965
|
494 |
pUser->nameidx = -1; |
|
dvander@1965
|
495 |
} |
|
ds@1952
|
496 |
|
|
ds@1952
|
497 |
return id; |
|
ds@1952
|
498 |
} |
|
ds@1952
|
499 |
|
|
ds@1952
|
500 |
GroupId AdminCache::AddGroup(const char *group_name) |
|
ds@1952
|
501 |
{ |
|
ds@1952
|
502 |
if (sm_trie_retrieve(m_pGroups, group_name, NULL)) |
|
ds@1952
|
503 |
{ |
|
ds@1952
|
504 |
return INVALID_GROUP_ID; |
|
ds@1952
|
505 |
} |
|
ds@1952
|
506 |
|
|
ds@1952
|
507 |
GroupId id; |
|
ds@1952
|
508 |
AdminGroup *pGroup; |
|
ds@1952
|
509 |
if (m_FreeGroupList != INVALID_GROUP_ID) |
|
ds@1952
|
510 |
{ |
|
ds@1952
|
511 |
pGroup = (AdminGroup *)m_pMemory->GetAddress(m_FreeGroupList); |
|
ds@1952
|
512 |
assert(pGroup->magic == GRP_MAGIC_UNSET); |
|
ds@1952
|
513 |
id = m_FreeGroupList; |
|
ds@1952
|
514 |
m_FreeGroupList = pGroup->next_grp; |
|
ds@1952
|
515 |
} else { |
|
ds@1952
|
516 |
id = m_pMemory->CreateMem(sizeof(AdminGroup), (void **)&pGroup); |
|
ds@1952
|
517 |
} |
|
ds@1952
|
518 |
|
|
ds@1952
|
519 |
pGroup->immunity_level = 0; |
|
ds@1952
|
520 |
pGroup->immune_table = -1; |
|
ds@1952
|
521 |
pGroup->magic = GRP_MAGIC_SET; |
|
ds@1952
|
522 |
pGroup->next_grp = INVALID_GROUP_ID; |
|
ds@1952
|
523 |
pGroup->pCmdGrpTable = NULL; |
|
ds@1952
|
524 |
pGroup->pCmdTable = NULL; |
|
ds@1952
|
525 |
pGroup->addflags = 0; |
|
ds@1952
|
526 |
|
|
ds@1952
|
527 |
if (m_FirstGroup == INVALID_GROUP_ID) |
|
ds@1952
|
528 |
{ |
|
ds@1952
|
529 |
m_FirstGroup = id; |
|
ds@1952
|
530 |
m_LastGroup = id; |
|
ds@1952
|
531 |
pGroup->prev_grp = INVALID_GROUP_ID; |
|
ds@1952
|
532 |
} else { |
|
ds@1952
|
533 |
AdminGroup *pPrev = (AdminGroup *)m_pMemory->GetAddress(m_LastGroup); |
|
ds@1952
|
534 |
assert(pPrev->magic == GRP_MAGIC_SET); |
|
ds@1952
|
535 |
pPrev->next_grp = id; |
|
ds@1952
|
536 |
pGroup->prev_grp = m_LastGroup; |
|
ds@1952
|
537 |
m_LastGroup = id; |
|
ds@1952
|
538 |
} |
|
ds@1952
|
539 |
|
|
ds@1952
|
540 |
int nameidx = m_pStrings->AddString(group_name); |
|
ds@1952
|
541 |
pGroup = (AdminGroup *)m_pMemory->GetAddress(id); |
|
ds@1952
|
542 |
pGroup->nameidx = nameidx; |
|
ds@1952
|
543 |
|
|
ds@1952
|
544 |
sm_trie_insert(m_pGroups, group_name, (void *)id); |
|
ds@1952
|
545 |
|
|
ds@1952
|
546 |
return id; |
|
ds@1952
|
547 |
} |
|
ds@1952
|
548 |
|
|
ds@1952
|
549 |
GroupId AdminCache::FindGroupByName(const char *group_name) |
|
ds@1952
|
550 |
{ |
|
ds@1952
|
551 |
void *object; |
|
ds@1952
|
552 |
|
|
ds@1952
|
553 |
if (!sm_trie_retrieve(m_pGroups, group_name, &object)) |
|
ds@1952
|
554 |
{ |
|
ds@1952
|
555 |
return INVALID_GROUP_ID; |
|
ds@1952
|
556 |
} |
|
ds@1952
|
557 |
|
|
ds@1952
|
558 |
GroupId id = (GroupId)object; |
|
ds@1952
|
559 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id); |
|
ds@1952
|
560 |
|
|
ds@1952
|
561 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
562 |
{ |
|
ds@1952
|
563 |
return INVALID_GROUP_ID; |
|
ds@1952
|
564 |
} |
|
ds@1952
|
565 |
|
|
ds@1952
|
566 |
return id; |
|
ds@1952
|
567 |
} |
|
ds@1952
|
568 |
|
|
ds@1952
|
569 |
void AdminCache::SetGroupAddFlag(GroupId id, AdminFlag flag, bool enabled) |
|
ds@1952
|
570 |
{ |
|
ds@1952
|
571 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id); |
|
ds@1952
|
572 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
573 |
{ |
|
ds@1952
|
574 |
return; |
|
ds@1952
|
575 |
} |
|
ds@1952
|
576 |
|
|
ds@1952
|
577 |
if (flag < Admin_Reservation || flag >= AdminFlags_TOTAL) |
|
ds@1952
|
578 |
{ |
|
ds@1952
|
579 |
return; |
|
ds@1952
|
580 |
} |
|
ds@1952
|
581 |
|
|
ds@1952
|
582 |
FlagBits bits = (1<<(FlagBits)flag); |
|
ds@1952
|
583 |
|
|
ds@1952
|
584 |
if (enabled) |
|
ds@1952
|
585 |
{ |
|
ds@1952
|
586 |
pGroup->addflags |= bits; |
|
ds@1952
|
587 |
} else { |
|
ds@1952
|
588 |
pGroup->addflags &= ~bits; |
|
ds@1952
|
589 |
} |
|
ds@1952
|
590 |
} |
|
ds@1952
|
591 |
|
|
ds@1952
|
592 |
bool AdminCache::GetGroupAddFlag(GroupId id, AdminFlag flag) |
|
ds@1952
|
593 |
{ |
|
ds@1952
|
594 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id); |
|
ds@1952
|
595 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
596 |
{ |
|
ds@1952
|
597 |
return false; |
|
ds@1952
|
598 |
} |
|
ds@1952
|
599 |
|
|
ds@1952
|
600 |
if (flag < Admin_Reservation || flag >= AdminFlags_TOTAL) |
|
ds@1952
|
601 |
{ |
|
ds@1952
|
602 |
return false; |
|
ds@1952
|
603 |
} |
|
ds@1952
|
604 |
|
|
ds@1952
|
605 |
FlagBits bit = 1<<(FlagBits)flag; |
|
ds@1952
|
606 |
return ((pGroup->addflags & bit) == bit); |
|
ds@1952
|
607 |
} |
|
ds@1952
|
608 |
|
|
ds@1952
|
609 |
FlagBits AdminCache::GetGroupAddFlags(GroupId id) |
|
ds@1952
|
610 |
{ |
|
ds@1952
|
611 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id); |
|
ds@1952
|
612 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
613 |
{ |
|
ds@1952
|
614 |
return 0; |
|
ds@1952
|
615 |
} |
|
ds@1952
|
616 |
|
|
ds@1952
|
617 |
return pGroup->addflags; |
|
ds@1952
|
618 |
} |
|
ds@1952
|
619 |
|
|
ds@1952
|
620 |
const char *AdminCache::GetGroupName(GroupId gid) |
|
ds@1952
|
621 |
{ |
|
ds@1952
|
622 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid); |
|
ds@1952
|
623 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
624 |
{ |
|
ds@1952
|
625 |
return NULL; |
|
ds@1952
|
626 |
} |
|
ds@1952
|
627 |
|
|
ds@1952
|
628 |
return m_pStrings->GetString(pGroup->nameidx); |
|
ds@1952
|
629 |
} |
|
ds@1952
|
630 |
|
|
ds@1952
|
631 |
void AdminCache::SetGroupGenericImmunity(GroupId id, ImmunityType type, bool enabled) |
|
ds@1952
|
632 |
{ |
|
ds@1952
|
633 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id); |
|
ds@1952
|
634 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
635 |
{ |
|
ds@1952
|
636 |
return; |
|
ds@1952
|
637 |
} |
|
ds@1952
|
638 |
|
|
ds@1952
|
639 |
unsigned int level = 0; |
|
ds@1952
|
640 |
|
|
ds@1952
|
641 |
if (enabled) |
|
ds@1952
|
642 |
{ |
|
ds@1952
|
643 |
if (type == Immunity_Default) |
|
ds@1952
|
644 |
{ |
|
ds@1952
|
645 |
level = 1; |
|
ds@1952
|
646 |
} else if (type == Immunity_Global) { |
|
ds@1952
|
647 |
level = 2; |
|
ds@1952
|
648 |
} |
|
ds@1952
|
649 |
if (level > pGroup->immunity_level) |
|
ds@1952
|
650 |
{ |
|
ds@1952
|
651 |
pGroup->immunity_level = level; |
|
ds@1952
|
652 |
} |
|
ds@1952
|
653 |
} else { |
|
ds@1952
|
654 |
pGroup->immunity_level = 0; |
|
ds@1952
|
655 |
} |
|
ds@1952
|
656 |
} |
|
ds@1952
|
657 |
|
|
ds@1952
|
658 |
bool AdminCache::GetGroupGenericImmunity(GroupId id, ImmunityType type) |
|
ds@1952
|
659 |
{ |
|
ds@1952
|
660 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id); |
|
ds@1952
|
661 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
662 |
{ |
|
ds@1952
|
663 |
return false; |
|
ds@1952
|
664 |
} |
|
ds@1952
|
665 |
|
|
ds@1952
|
666 |
if (type == Immunity_Default && pGroup->immunity_level >= 1) |
|
ds@1952
|
667 |
{ |
|
ds@1952
|
668 |
return true; |
|
ds@1952
|
669 |
} else if (type == Immunity_Global && pGroup->immunity_level >= 2) { |
|
ds@1952
|
670 |
return true; |
|
ds@1952
|
671 |
} |
|
ds@1952
|
672 |
|
|
ds@1952
|
673 |
return false; |
|
ds@1952
|
674 |
} |
|
ds@1952
|
675 |
|
|
ds@1952
|
676 |
void AdminCache::AddGroupImmunity(GroupId id, GroupId other_id) |
|
ds@1952
|
677 |
{ |
|
ds@1952
|
678 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(other_id); |
|
ds@1952
|
679 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
680 |
{ |
|
ds@1952
|
681 |
return; |
|
ds@1952
|
682 |
} |
|
ds@1952
|
683 |
|
|
ds@1952
|
684 |
pGroup = (AdminGroup *)m_pMemory->GetAddress(id); |
|
ds@1952
|
685 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
686 |
{ |
|
ds@1952
|
687 |
return; |
|
ds@1952
|
688 |
} |
|
ds@1952
|
689 |
|
|
ds@1952
|
690 |
/* We always need to resize the immunity table */ |
|
ds@1952
|
691 |
int *table, tblidx; |
|
ds@1952
|
692 |
if (pGroup->immune_table == -1) |
|
ds@1952
|
693 |
{ |
|
ds@1952
|
694 |
tblidx = m_pMemory->CreateMem(sizeof(int) * 2, (void **)&table); |
|
ds@1952
|
695 |
pGroup = (AdminGroup *)m_pMemory->GetAddress(id); |
|
ds@1952
|
696 |
table[0] = 0; |
|
ds@1952
|
697 |
} else { |
|
ds@1952
|
698 |
int *old_table = (int *)m_pMemory->GetAddress(pGroup->immune_table); |
|
ds@1952
|
699 |
/* Break out if this group is already in the list */ |
|
ds@1952
|
700 |
for (int i=0; i<old_table[0]; i++) |
|
ds@1952
|
701 |
{ |
|
ds@1952
|
702 |
if (old_table[1+i] == other_id) |
|
ds@1952
|
703 |
{ |
|
ds@1952
|
704 |
return; |
|
ds@1952
|
705 |
} |
|
ds@1952
|
706 |
} |
|
ds@1952
|
707 |
tblidx = m_pMemory->CreateMem(sizeof(int) * (old_table[0] + 2), (void **)&table); |
|
ds@1952
|
708 |
/* Get the old address again in case of resize */ |
|
ds@1952
|
709 |
pGroup = (AdminGroup *)m_pMemory->GetAddress(id); |
|
ds@1952
|
710 |
old_table = (int *)m_pMemory->GetAddress(pGroup->immune_table); |
|
ds@1952
|
711 |
table[0] = old_table[0]; |
|
ds@1952
|
712 |
for (unsigned int i=1; i<=(unsigned int)old_table[0]; i++) |
|
ds@1952
|
713 |
{ |
|
ds@1952
|
714 |
table[i] = old_table[i]; |
|
ds@1952
|
715 |
} |
|
ds@1952
|
716 |
} |
|
ds@1952
|
717 |
|
|
ds@1952
|
718 |
/* Assign */ |
|
ds@1952
|
719 |
pGroup->immune_table = tblidx; |
|
ds@1952
|
720 |
|
|
ds@1952
|
721 |
/* Add to the array */ |
|
ds@1952
|
722 |
table[0]++; |
|
ds@1952
|
723 |
table[table[0]] = other_id; |
|
ds@1952
|
724 |
} |
|
ds@1952
|
725 |
|
|
ds@1952
|
726 |
unsigned int AdminCache::GetGroupImmunityCount(GroupId id) |
|
ds@1952
|
727 |
{ |
|
ds@1952
|
728 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id); |
|
ds@1952
|
729 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
730 |
{ |
|
ds@1952
|
731 |
return 0; |
|
ds@1952
|
732 |
} |
|
ds@1952
|
733 |
|
|
ds@1952
|
734 |
if (pGroup->immune_table == -1) |
|
ds@1952
|
735 |
{ |
|
ds@1952
|
736 |
return 0; |
|
ds@1952
|
737 |
} |
|
ds@1952
|
738 |
|
|
ds@1952
|
739 |
int *table = (int *)m_pMemory->GetAddress(pGroup->immune_table); |
|
ds@1952
|
740 |
|
|
ds@1952
|
741 |
return table[0]; |
|
ds@1952
|
742 |
} |
|
ds@1952
|
743 |
|
|
ds@1952
|
744 |
GroupId AdminCache::GetGroupImmunity(GroupId id, unsigned int number) |
|
ds@1952
|
745 |
{ |
|
ds@1952
|
746 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id); |
|
ds@1952
|
747 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
748 |
{ |
|
ds@1952
|
749 |
return INVALID_GROUP_ID; |
|
ds@1952
|
750 |
} |
|
ds@1952
|
751 |
|
|
ds@1952
|
752 |
if (pGroup->immune_table == -1) |
|
ds@1952
|
753 |
{ |
|
ds@1952
|
754 |
return INVALID_GROUP_ID; |
|
ds@1952
|
755 |
} |
|
ds@1952
|
756 |
|
|
ds@1952
|
757 |
int *table = (int *)m_pMemory->GetAddress(pGroup->immune_table); |
|
ds@1952
|
758 |
if (number >= (unsigned int)table[0]) |
|
ds@1952
|
759 |
{ |
|
ds@1952
|
760 |
return INVALID_GROUP_ID; |
|
ds@1952
|
761 |
} |
|
ds@1952
|
762 |
|
|
ds@1952
|
763 |
return table[1 + number]; |
|
ds@1952
|
764 |
} |
|
ds@1952
|
765 |
|
|
ds@1952
|
766 |
void AdminCache::AddGroupCommandOverride(GroupId id, const char *name, OverrideType type, OverrideRule rule) |
|
ds@1952
|
767 |
{ |
|
ds@1952
|
768 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id); |
|
ds@1952
|
769 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
770 |
{ |
|
ds@1952
|
771 |
return; |
|
ds@1952
|
772 |
} |
|
ds@1952
|
773 |
|
|
ds@1952
|
774 |
Trie *pTrie = NULL; |
|
ds@1952
|
775 |
if (type == Override_Command) |
|
ds@1952
|
776 |
{ |
|
ds@1952
|
777 |
if (pGroup->pCmdTable == NULL) |
|
ds@1952
|
778 |
{ |
|
ds@1952
|
779 |
pGroup->pCmdTable = sm_trie_create(); |
|
ds@1952
|
780 |
} |
|
ds@1952
|
781 |
pTrie = pGroup->pCmdTable; |
|
ds@1952
|
782 |
} else if (type == Override_CommandGroup) { |
|
ds@1952
|
783 |
if (pGroup->pCmdGrpTable == NULL) |
|
ds@1952
|
784 |
{ |
|
ds@1952
|
785 |
pGroup->pCmdGrpTable = sm_trie_create(); |
|
ds@1952
|
786 |
} |
|
ds@1952
|
787 |
pTrie = pGroup->pCmdGrpTable; |
|
ds@1952
|
788 |
} else { |
|
ds@1952
|
789 |
return; |
|
ds@1952
|
790 |
} |
|
ds@1952
|
791 |
|
|
ds@1952
|
792 |
sm_trie_insert(pTrie, name, (void *)(int)rule); |
|
ds@1952
|
793 |
} |
|
ds@1952
|
794 |
|
|
ds@1952
|
795 |
bool AdminCache::GetGroupCommandOverride(GroupId id, const char *name, OverrideType type, OverrideRule *pRule) |
|
ds@1952
|
796 |
{ |
|
ds@1952
|
797 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id); |
|
ds@1952
|
798 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
799 |
{ |
|
ds@1952
|
800 |
return false; |
|
ds@1952
|
801 |
} |
|
ds@1952
|
802 |
|
|
ds@1952
|
803 |
Trie *pTrie = NULL; |
|
ds@1952
|
804 |
if (type == Override_Command) |
|
ds@1952
|
805 |
{ |
|
ds@1952
|
806 |
if (pGroup->pCmdTable == NULL) |
|
ds@1952
|
807 |
{ |
|
ds@1952
|
808 |
return false; |
|
ds@1952
|
809 |
} |
|
ds@1952
|
810 |
pTrie = pGroup->pCmdTable; |
|
ds@1952
|
811 |
} else if (type == Override_CommandGroup) { |
|
ds@1952
|
812 |
if (pGroup->pCmdGrpTable == NULL) |
|
ds@1952
|
813 |
{ |
|
ds@1952
|
814 |
return false; |
|
ds@1952
|
815 |
} |
|
ds@1952
|
816 |
pTrie = pGroup->pCmdGrpTable; |
|
ds@1952
|
817 |
} else { |
|
ds@1952
|
818 |
return false; |
|
ds@1952
|
819 |
} |
|
ds@1952
|
820 |
|
|
ds@1952
|
821 |
void *object; |
|
ds@1952
|
822 |
if (!sm_trie_retrieve(pTrie, name, &object)) |
|
ds@1952
|
823 |
{ |
|
ds@1952
|
824 |
return false; |
|
ds@1952
|
825 |
} |
|
ds@1952
|
826 |
|
|
ds@1952
|
827 |
if (pRule) |
|
ds@1952
|
828 |
{ |
|
ds@1952
|
829 |
*pRule = (OverrideRule)(int)object; |
|
ds@1952
|
830 |
} |
|
ds@1952
|
831 |
|
|
ds@1952
|
832 |
return true; |
|
ds@1952
|
833 |
} |
|
ds@1952
|
834 |
|
|
ds@1952
|
835 |
Trie *AdminCache::GetMethodByIndex(unsigned int index) |
|
ds@1952
|
836 |
{ |
|
ds@1952
|
837 |
List<AuthMethod>::iterator iter; |
|
ds@1952
|
838 |
for (iter=m_AuthMethods.begin(); |
|
ds@1952
|
839 |
iter!=m_AuthMethods.end(); |
|
ds@1952
|
840 |
iter++) |
|
ds@1952
|
841 |
{ |
|
ds@1952
|
842 |
if (index-- == 0) |
|
ds@1952
|
843 |
{ |
|
ds@1952
|
844 |
return (*iter).table; |
|
ds@1952
|
845 |
} |
|
ds@1952
|
846 |
} |
|
ds@1952
|
847 |
|
|
ds@1952
|
848 |
return NULL; |
|
ds@1952
|
849 |
} |
|
ds@1952
|
850 |
|
|
ds@1952
|
851 |
bool AdminCache::InvalidateAdmin(AdminId id) |
|
ds@1952
|
852 |
{ |
|
ds@1952
|
853 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
854 |
AdminUser *pOther; |
|
ds@1952
|
855 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
856 |
{ |
|
ds@1952
|
857 |
return false; |
|
ds@1952
|
858 |
} |
|
ds@1952
|
859 |
|
|
ds@1952
|
860 |
if (!m_InvalidatingAdmins && !m_destroying) |
|
ds@1952
|
861 |
{ |
|
ds@1952
|
862 |
g_Players.ClearAdminId(id); |
|
ds@1952
|
863 |
} |
|
ds@1952
|
864 |
|
|
ds@1952
|
865 |
/* Unlink from the dbl link list */ |
|
ds@1952
|
866 |
if (id == m_FirstUser && id == m_LastUser) |
|
ds@1952
|
867 |
{ |
|
ds@1952
|
868 |
m_FirstUser = INVALID_ADMIN_ID; |
|
ds@1952
|
869 |
m_LastUser = INVALID_ADMIN_ID; |
|
ds@1952
|
870 |
} else if (id == m_FirstUser) { |
|
ds@1952
|
871 |
m_FirstUser = pUser->next_user; |
|
ds@1952
|
872 |
pOther = (AdminUser *)m_pMemory->GetAddress(m_FirstUser); |
|
ds@1952
|
873 |
pOther->prev_user = INVALID_ADMIN_ID; |
|
ds@1952
|
874 |
} else if (id == m_LastUser) { |
|
ds@1952
|
875 |
m_LastUser = pUser->prev_user; |
|
ds@1952
|
876 |
pOther = (AdminUser *)m_pMemory->GetAddress(m_LastUser); |
|
ds@1952
|
877 |
pOther->next_user = INVALID_ADMIN_ID; |
|
ds@1952
|
878 |
} else { |
|
ds@1952
|
879 |
pOther = (AdminUser *)m_pMemory->GetAddress(pUser->prev_user); |
|
ds@1952
|
880 |
pOther->next_user = pUser->next_user; |
|
ds@1952
|
881 |
pOther = (AdminUser *)m_pMemory->GetAddress(pUser->next_user); |
|
ds@1952
|
882 |
pOther->prev_user = pUser->prev_user; |
|
ds@1952
|
883 |
} |
|
ds@1952
|
884 |
|
|
ds@1952
|
885 |
/* Unlink from auth tables */ |
|
ds@1952
|
886 |
if (pUser->auth.identidx != -1) |
|
ds@1952
|
887 |
{ |
|
ds@1952
|
888 |
Trie *pTrie = GetMethodByIndex(pUser->auth.index); |
|
ds@1952
|
889 |
if (pTrie) |
|
ds@1952
|
890 |
{ |
|
ds@1952
|
891 |
sm_trie_delete(pTrie, m_pStrings->GetString(pUser->auth.identidx)); |
|
ds@1952
|
892 |
} |
|
ds@1952
|
893 |
} |
|
ds@1952
|
894 |
|
|
ds@1952
|
895 |
/* Clear table counts */ |
|
ds@1952
|
896 |
pUser->grp_count = 0; |
|
ds@1952
|
897 |
|
|
ds@1952
|
898 |
/* Link into free list */ |
|
ds@1952
|
899 |
pUser->magic = USR_MAGIC_UNSET; |
|
ds@1952
|
900 |
pUser->next_user = m_FreeUserList; |
|
ds@1952
|
901 |
m_FreeUserList = id; |
|
ds@1952
|
902 |
|
|
ds@1952
|
903 |
/* Unset serial change */ |
|
ds@1952
|
904 |
pUser->serialchange = 0; |
|
ds@1952
|
905 |
|
|
ds@1952
|
906 |
return true; |
|
ds@1952
|
907 |
} |
|
ds@1952
|
908 |
|
|
ds@1952
|
909 |
|
|
ds@1952
|
910 |
void AdminCache::InvalidateGroup(GroupId id) |
|
ds@1952
|
911 |
{ |
|
ds@1952
|
912 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id); |
|
ds@1952
|
913 |
AdminGroup *pOther; |
|
ds@1952
|
914 |
|
|
ds@1952
|
915 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
916 |
{ |
|
ds@1952
|
917 |
return; |
|
ds@1952
|
918 |
} |
|
ds@1952
|
919 |
|
|
ds@1952
|
920 |
const char *str = m_pStrings->GetString(pGroup->nameidx); |
|
ds@1952
|
921 |
sm_trie_delete(m_pGroups, str); |
|
ds@1952
|
922 |
|
|
ds@1952
|
923 |
/* Unlink from the live dbllink list */ |
|
ds@1952
|
924 |
if (id == m_FirstGroup && id == m_LastGroup) |
|
ds@1952
|
925 |
{ |
|
ds@1952
|
926 |
m_LastGroup = INVALID_GROUP_ID; |
|
ds@1952
|
927 |
m_FirstGroup = INVALID_GROUP_ID; |
|
ds@1952
|
928 |
} else if (id == m_FirstGroup) { |
|
ds@1952
|
929 |
m_FirstGroup = pGroup->next_grp; |
|
ds@1952
|
930 |
pOther = (AdminGroup *)m_pMemory->GetAddress(m_FirstGroup); |
|
ds@1952
|
931 |
pOther->prev_grp = INVALID_GROUP_ID; |
|
ds@1952
|
932 |
} else if (id == m_LastGroup) { |
|
ds@1952
|
933 |
m_LastGroup = pGroup->prev_grp; |
|
ds@1952
|
934 |
pOther = (AdminGroup *)m_pMemory->GetAddress(m_LastGroup); |
|
ds@1952
|
935 |
pOther->next_grp = INVALID_GROUP_ID; |
|
ds@1952
|
936 |
} else { |
|
ds@1952
|
937 |
pOther = (AdminGroup *)m_pMemory->GetAddress(pGroup->prev_grp); |
|
ds@1952
|
938 |
pOther->next_grp = pGroup->next_grp; |
|
ds@1952
|
939 |
pOther = (AdminGroup *)m_pMemory->GetAddress(pGroup->next_grp); |
|
ds@1952
|
940 |
pOther->prev_grp = pGroup->prev_grp; |
|
ds@1952
|
941 |
} |
|
ds@1952
|
942 |
|
|
ds@1952
|
943 |
/* Free any used memory to be safe */ |
|
ds@1952
|
944 |
if (pGroup->pCmdGrpTable) |
|
ds@1952
|
945 |
{ |
|
ds@1952
|
946 |
sm_trie_destroy(pGroup->pCmdGrpTable); |
|
ds@1952
|
947 |
pGroup->pCmdGrpTable = NULL; |
|
ds@1952
|
948 |
} |
|
ds@1952
|
949 |
if (pGroup->pCmdTable) |
|
ds@1952
|
950 |
{ |
|
ds@1952
|
951 |
sm_trie_destroy(pGroup->pCmdTable); |
|
ds@1952
|
952 |
pGroup->pCmdTable = NULL; |
|
ds@1952
|
953 |
} |
|
ds@1952
|
954 |
|
|
ds@1952
|
955 |
/* Link into the free list */ |
|
ds@1952
|
956 |
pGroup->magic = GRP_MAGIC_UNSET; |
|
ds@1952
|
957 |
pGroup->next_grp = m_FreeGroupList; |
|
ds@1952
|
958 |
m_FreeGroupList = id; |
|
ds@1952
|
959 |
|
|
ds@1952
|
960 |
int idx = m_FirstUser; |
|
ds@1952
|
961 |
AdminUser *pUser; |
|
ds@1952
|
962 |
int *table; |
|
ds@1952
|
963 |
while (idx != INVALID_ADMIN_ID) |
|
ds@1952
|
964 |
{ |
|
ds@1952
|
965 |
pUser = (AdminUser *)m_pMemory->GetAddress(idx); |
|
ds@1952
|
966 |
if (pUser->grp_count > 0) |
|
ds@1952
|
967 |
{ |
|
ds@1952
|
968 |
table = (int *)m_pMemory->GetAddress(pUser->grp_table); |
|
ds@1952
|
969 |
for (unsigned int i=0; i<pUser->grp_count; i++) |
|
ds@1952
|
970 |
{ |
|
ds@1952
|
971 |
if (table[i] == id) |
|
ds@1952
|
972 |
{ |
|
ds@1952
|
973 |
/* We have to remove this entry */ |
|
ds@1952
|
974 |
for (unsigned int j=i+1; j<pUser->grp_count; j++) |
|
ds@1952
|
975 |
{ |
|
ds@1952
|
976 |
/* Move everything down by one */ |
|
ds@1952
|
977 |
table[j-1] = table[j]; |
|
ds@1952
|
978 |
} |
|
ds@1952
|
979 |
/* Decrease count */ |
|
ds@1952
|
980 |
pUser->grp_count--; |
|
ds@1952
|
981 |
/* Recalculate effective flags */ |
|
ds@1952
|
982 |
pUser->eflags = pUser->flags; |
|
ds@1952
|
983 |
for (unsigned int j=0; j<pUser->grp_count; j++) |
|
ds@1952
|
984 |
{ |
|
ds@1952
|
985 |
pOther = (AdminGroup *)m_pMemory->GetAddress(table[j]); |
|
ds@1952
|
986 |
pUser->eflags |= pOther->addflags; |
|
ds@1952
|
987 |
} |
|
ds@1952
|
988 |
/* Mark as changed */ |
|
ds@1952
|
989 |
pUser->serialchange++; |
|
ds@1952
|
990 |
/* Break now, duplicates aren't allowed */ |
|
ds@1952
|
991 |
break; |
|
ds@1952
|
992 |
} |
|
ds@1952
|
993 |
} |
|
ds@1952
|
994 |
} |
|
ds@1952
|
995 |
idx = pUser->next_user; |
|
ds@1952
|
996 |
} |
|
ds@1952
|
997 |
} |
|
ds@1952
|
998 |
|
|
ds@1952
|
999 |
void AdminCache::InvalidateGroupCache() |
|
ds@1952
|
1000 |
{ |
|
ds@1952
|
1001 |
/* Nuke the free list */ |
|
ds@1952
|
1002 |
m_FreeGroupList = -1; |
|
ds@1952
|
1003 |
|
|
ds@1952
|
1004 |
/* Nuke reverse lookups */ |
|
ds@1952
|
1005 |
sm_trie_clear(m_pGroups); |
|
ds@1952
|
1006 |
|
|
ds@1952
|
1007 |
/* Free memory on groups */ |
|
ds@1952
|
1008 |
GroupId cur = m_FirstGroup; |
|
ds@1952
|
1009 |
AdminGroup *pGroup; |
|
ds@1952
|
1010 |
while (cur != INVALID_GROUP_ID) |
|
ds@1952
|
1011 |
{ |
|
ds@1952
|
1012 |
pGroup = (AdminGroup *)m_pMemory->GetAddress(cur); |
|
ds@1952
|
1013 |
assert(pGroup->magic == GRP_MAGIC_SET); |
|
ds@1952
|
1014 |
if (pGroup->pCmdGrpTable) |
|
ds@1952
|
1015 |
{ |
|
ds@1952
|
1016 |
sm_trie_destroy(pGroup->pCmdGrpTable); |
|
ds@1952
|
1017 |
} |
|
ds@1952
|
1018 |
if (pGroup->pCmdTable) |
|
ds@1952
|
1019 |
{ |
|
ds@1952
|
1020 |
sm_trie_destroy(pGroup->pCmdTable); |
|
ds@1952
|
1021 |
} |
|
ds@1952
|
1022 |
cur = pGroup->next_grp; |
|
ds@1952
|
1023 |
} |
|
ds@1952
|
1024 |
|
|
ds@1952
|
1025 |
m_FirstGroup = INVALID_GROUP_ID; |
|
ds@1952
|
1026 |
m_LastGroup = INVALID_GROUP_ID; |
|
ds@1952
|
1027 |
|
|
ds@1952
|
1028 |
InvalidateAdminCache(false); |
|
ds@1952
|
1029 |
|
|
ds@1952
|
1030 |
/* Reset the memory table */ |
|
ds@1952
|
1031 |
m_pMemory->Reset(); |
|
ds@1952
|
1032 |
} |
|
ds@1952
|
1033 |
|
|
ds@1952
|
1034 |
void AdminCache::AddAdminListener(IAdminListener *pListener) |
|
ds@1952
|
1035 |
{ |
|
ds@1952
|
1036 |
m_hooks.push_back(pListener); |
|
ds@1952
|
1037 |
} |
|
ds@1952
|
1038 |
|
|
ds@1952
|
1039 |
void AdminCache::RemoveAdminListener(IAdminListener *pListener) |
|
ds@1952
|
1040 |
{ |
|
ds@1952
|
1041 |
m_hooks.remove(pListener); |
|
ds@1952
|
1042 |
} |
|
ds@1952
|
1043 |
|
|
ds@1952
|
1044 |
void AdminCache::RegisterAuthIdentType(const char *name) |
|
ds@1952
|
1045 |
{ |
|
ds@1952
|
1046 |
if (sm_trie_retrieve(m_pAuthTables, name, NULL)) |
|
ds@1952
|
1047 |
{ |
|
ds@1952
|
1048 |
return; |
|
ds@1952
|
1049 |
} |
|
ds@1952
|
1050 |
|
|
ds@1952
|
1051 |
Trie *pAuth = sm_trie_create(); |
|
ds@1952
|
1052 |
|
|
ds@1952
|
1053 |
AuthMethod method; |
|
ds@1952
|
1054 |
method.name.assign(name); |
|
ds@1952
|
1055 |
method.table = pAuth; |
|
ds@1952
|
1056 |
|
|
ds@1952
|
1057 |
m_AuthMethods.push_back(method); |
|
ds@1952
|
1058 |
|
|
ds@1952
|
1059 |
sm_trie_insert(m_pAuthTables, name, pAuth); |
|
ds@1952
|
1060 |
} |
|
ds@1952
|
1061 |
|
|
ds@1952
|
1062 |
void AdminCache::InvalidateAdminCache(bool unlink_admins) |
|
ds@1952
|
1063 |
{ |
|
ds@1952
|
1064 |
m_InvalidatingAdmins = true; |
|
ds@1952
|
1065 |
if (!m_destroying) |
|
ds@1952
|
1066 |
{ |
|
ds@1952
|
1067 |
g_Players.ClearAllAdmins(); |
|
ds@1952
|
1068 |
} |
|
ds@1952
|
1069 |
/* Wipe the identity cache first */ |
|
ds@1952
|
1070 |
List<AuthMethod>::iterator iter; |
|
ds@1952
|
1071 |
for (iter=m_AuthMethods.begin(); |
|
ds@1952
|
1072 |
iter!=m_AuthMethods.end(); |
|
ds@1952
|
1073 |
iter++) |
|
ds@1952
|
1074 |
{ |
|
ds@1952
|
1075 |
sm_trie_clear((*iter).table); |
|
ds@1952
|
1076 |
} |
|
ds@1952
|
1077 |
|
|
ds@1952
|
1078 |
if (unlink_admins) |
|
ds@1952
|
1079 |
{ |
|
ds@1952
|
1080 |
while (m_FirstUser != INVALID_ADMIN_ID) |
|
ds@1952
|
1081 |
{ |
|
ds@1952
|
1082 |
InvalidateAdmin(m_FirstUser); |
|
ds@1952
|
1083 |
} |
|
ds@1952
|
1084 |
} else { |
|
ds@1952
|
1085 |
m_FirstUser = -1; |
|
ds@1952
|
1086 |
m_LastUser = -1; |
|
ds@1952
|
1087 |
m_FreeUserList = -1; |
|
ds@1952
|
1088 |
} |
|
ds@1952
|
1089 |
m_InvalidatingAdmins = false; |
|
ds@1952
|
1090 |
} |
|
ds@1952
|
1091 |
|
|
ds@1952
|
1092 |
void AdminCache::DumpAdminCache(AdminCachePart part, bool rebuild) |
|
ds@1952
|
1093 |
{ |
|
ds@1952
|
1094 |
List<IAdminListener *>::iterator iter; |
|
ds@1952
|
1095 |
IAdminListener *pListener; |
|
ds@1952
|
1096 |
cell_t result; |
|
ds@1952
|
1097 |
|
|
ds@1952
|
1098 |
if (part == AdminCache_Overrides) |
|
ds@1952
|
1099 |
{ |
|
ds@1952
|
1100 |
DumpCommandOverrideCache(Override_Command); |
|
ds@1952
|
1101 |
DumpCommandOverrideCache(Override_CommandGroup); |
|
ds@1952
|
1102 |
if (rebuild && !m_destroying) |
|
ds@1952
|
1103 |
{ |
|
ds@1952
|
1104 |
for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++) |
|
ds@1952
|
1105 |
{ |
|
ds@1952
|
1106 |
pListener = (*iter); |
|
ds@1952
|
1107 |
pListener->OnRebuildOverrideCache(); |
|
ds@1952
|
1108 |
} |
|
ds@1952
|
1109 |
m_pCacheFwd->PushCell(part); |
|
ds@1952
|
1110 |
m_pCacheFwd->Execute(&result); |
|
ds@1952
|
1111 |
} |
|
ds@1952
|
1112 |
} else if (part == AdminCache_Groups || part == AdminCache_Admins) { |
|
ds@1952
|
1113 |
if (part == AdminCache_Groups) |
|
ds@1952
|
1114 |
{ |
|
ds@1952
|
1115 |
InvalidateGroupCache(); |
|
ds@1952
|
1116 |
if (rebuild && !m_destroying) |
|
ds@1952
|
1117 |
{ |
|
ds@1952
|
1118 |
for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++) |
|
ds@1952
|
1119 |
{ |
|
ds@1952
|
1120 |
pListener = (*iter); |
|
ds@1952
|
1121 |
pListener->OnRebuildGroupCache(); |
|
ds@1952
|
1122 |
} |
|
ds@1952
|
1123 |
m_pCacheFwd->PushCell(part); |
|
ds@1952
|
1124 |
m_pCacheFwd->Execute(&result); |
|
ds@1952
|
1125 |
} |
|
ds@1952
|
1126 |
} |
|
ds@1952
|
1127 |
InvalidateAdminCache(true); |
|
ds@1952
|
1128 |
if (rebuild && !m_destroying) |
|
ds@1952
|
1129 |
{ |
|
ds@1952
|
1130 |
for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++) |
|
ds@1952
|
1131 |
{ |
|
ds@1952
|
1132 |
pListener = (*iter); |
|
ds@1952
|
1133 |
pListener->OnRebuildAdminCache((part == AdminCache_Groups)); |
|
ds@1952
|
1134 |
} |
|
ds@1952
|
1135 |
m_pCacheFwd->PushCell(AdminCache_Admins); |
|
ds@1952
|
1136 |
m_pCacheFwd->Execute(&result); |
|
ds@1952
|
1137 |
g_Players.RecheckAnyAdmins(); |
|
ds@1952
|
1138 |
} |
|
ds@1952
|
1139 |
} |
|
ds@1952
|
1140 |
} |
|
ds@1952
|
1141 |
|
|
ds@1952
|
1142 |
const char *AdminCache::GetAdminName(AdminId id) |
|
ds@1952
|
1143 |
{ |
|
ds@1952
|
1144 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1145 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
1146 |
{ |
|
ds@1952
|
1147 |
return NULL; |
|
ds@1952
|
1148 |
} |
|
ds@1952
|
1149 |
|
|
ds@1952
|
1150 |
return m_pStrings->GetString(pUser->nameidx); |
|
ds@1952
|
1151 |
} |
|
ds@1952
|
1152 |
|
|
ds@1952
|
1153 |
bool AdminCache::GetMethodIndex(const char *name, unsigned int *_index) |
|
ds@1952
|
1154 |
{ |
|
ds@1952
|
1155 |
List<AuthMethod>::iterator iter; |
|
ds@1952
|
1156 |
unsigned int index = 0; |
|
ds@1952
|
1157 |
for (iter=m_AuthMethods.begin(); |
|
ds@1952
|
1158 |
iter!=m_AuthMethods.end(); |
|
ds@1952
|
1159 |
iter++,index++) |
|
ds@1952
|
1160 |
{ |
|
ds@1952
|
1161 |
if ((*iter).name.compare(name) == 0) |
|
ds@1952
|
1162 |
{ |
|
ds@1952
|
1163 |
*_index = index; |
|
ds@1952
|
1164 |
return true; |
|
ds@1952
|
1165 |
} |
|
ds@1952
|
1166 |
} |
|
ds@1952
|
1167 |
|
|
ds@1952
|
1168 |
return false; |
|
ds@1952
|
1169 |
} |
|
ds@1952
|
1170 |
|
|
ds@1952
|
1171 |
bool AdminCache::BindAdminIdentity(AdminId id, const char *auth, const char *ident) |
|
ds@1952
|
1172 |
{ |
|
ds@1952
|
1173 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1174 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
1175 |
{ |
|
ds@1952
|
1176 |
return false; |
|
ds@1952
|
1177 |
} |
|
ds@1952
|
1178 |
|
|
ds@1952
|
1179 |
Trie *pTable; |
|
ds@1952
|
1180 |
if (!sm_trie_retrieve(m_pAuthTables, auth, (void **)&pTable)) |
|
ds@1952
|
1181 |
{ |
|
ds@1952
|
1182 |
return false; |
|
ds@1952
|
1183 |
} |
|
ds@1952
|
1184 |
|
|
ds@1952
|
1185 |
if (sm_trie_retrieve(pTable, ident, NULL)) |
|
ds@1952
|
1186 |
{ |
|
ds@1952
|
1187 |
return false; |
|
ds@1952
|
1188 |
} |
|
ds@1952
|
1189 |
|
|
ds@1952
|
1190 |
int i_ident = m_pStrings->AddString(ident); |
|
ds@1952
|
1191 |
|
|
ds@1952
|
1192 |
pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1193 |
pUser->auth.identidx = i_ident; |
|
ds@1952
|
1194 |
GetMethodIndex(auth, &pUser->auth.index); |
|
ds@1952
|
1195 |
|
|
ds@1952
|
1196 |
return sm_trie_insert(pTable, ident, (void **)id); |
|
ds@1952
|
1197 |
} |
|
ds@1952
|
1198 |
|
|
ds@1952
|
1199 |
AdminId AdminCache::FindAdminByIdentity(const char *auth, const char *identity) |
|
ds@1952
|
1200 |
{ |
|
ds@1952
|
1201 |
Trie *pTable; |
|
ds@1952
|
1202 |
if (!sm_trie_retrieve(m_pAuthTables, auth, (void **)&pTable)) |
|
ds@1952
|
1203 |
{ |
|
ds@1952
|
1204 |
return INVALID_ADMIN_ID; |
|
ds@1952
|
1205 |
} |
|
ds@1952
|
1206 |
|
|
ds@1952
|
1207 |
void *object; |
|
ds@1952
|
1208 |
if (!sm_trie_retrieve(pTable, identity, &object)) |
|
ds@1952
|
1209 |
{ |
|
ds@1952
|
1210 |
return INVALID_ADMIN_ID; |
|
ds@1952
|
1211 |
} |
|
ds@1952
|
1212 |
|
|
ds@1952
|
1213 |
return (AdminId)object; |
|
ds@1952
|
1214 |
} |
|
ds@1952
|
1215 |
|
|
ds@1952
|
1216 |
void AdminCache::SetAdminFlag(AdminId id, AdminFlag flag, bool enabled) |
|
ds@1952
|
1217 |
{ |
|
ds@1952
|
1218 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1219 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
1220 |
{ |
|
ds@1952
|
1221 |
return; |
|
ds@1952
|
1222 |
} |
|
ds@1952
|
1223 |
|
|
ds@1952
|
1224 |
if (flag < Admin_Reservation |
|
ds@1952
|
1225 |
|| flag >= AdminFlags_TOTAL) |
|
ds@1952
|
1226 |
{ |
|
ds@1952
|
1227 |
return; |
|
ds@1952
|
1228 |
} |
|
ds@1952
|
1229 |
|
|
ds@1952
|
1230 |
FlagBits bits = (1<<(FlagBits)flag); |
|
ds@1952
|
1231 |
|
|
ds@1952
|
1232 |
if (enabled) |
|
ds@1952
|
1233 |
{ |
|
ds@1952
|
1234 |
pUser->flags |= bits; |
|
ds@1952
|
1235 |
pUser->eflags |= bits; |
|
ds@1952
|
1236 |
} else { |
|
ds@1952
|
1237 |
pUser->flags &= ~bits; |
|
ds@1952
|
1238 |
pUser->eflags &= ~bits; |
|
ds@1952
|
1239 |
} |
|
ds@1952
|
1240 |
|
|
ds@1952
|
1241 |
pUser->serialchange++; |
|
ds@1952
|
1242 |
} |
|
ds@1952
|
1243 |
|
|
ds@1952
|
1244 |
bool AdminCache::GetAdminFlag(AdminId id, AdminFlag flag, AccessMode mode) |
|
ds@1952
|
1245 |
{ |
|
ds@1952
|
1246 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1247 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
1248 |
{ |
|
ds@1952
|
1249 |
return false; |
|
ds@1952
|
1250 |
} |
|
ds@1952
|
1251 |
|
|
ds@1952
|
1252 |
if (flag < Admin_Reservation |
|
ds@1952
|
1253 |
|| flag >= AdminFlags_TOTAL) |
|
ds@1952
|
1254 |
{ |
|
ds@1952
|
1255 |
return false; |
|
ds@1952
|
1256 |
} |
|
ds@1952
|
1257 |
|
|
ds@1952
|
1258 |
FlagBits bit = (1<<(FlagBits)flag); |
|
ds@1952
|
1259 |
|
|
ds@1952
|
1260 |
if (mode == Access_Real) |
|
ds@1952
|
1261 |
{ |
|
ds@1952
|
1262 |
return ((pUser->flags & bit) == bit); |
|
ds@1952
|
1263 |
} else if (mode == Access_Effective) { |
|
ds@1952
|
1264 |
bool has_bit = ((pUser->eflags & bit) == bit); |
|
ds@1952
|
1265 |
if (!has_bit && flag != Admin_Root && ((pUser->eflags & ADMFLAG_ROOT) == ADMFLAG_ROOT)) |
|
ds@1952
|
1266 |
{ |
|
ds@1952
|
1267 |
has_bit = true; |
|
ds@1952
|
1268 |
} |
|
ds@1952
|
1269 |
return has_bit; |
|
ds@1952
|
1270 |
} |
|
ds@1952
|
1271 |
|
|
ds@1952
|
1272 |
return false; |
|
ds@1952
|
1273 |
} |
|
ds@1952
|
1274 |
|
|
ds@1952
|
1275 |
FlagBits AdminCache::GetAdminFlags(AdminId id, AccessMode mode) |
|
ds@1952
|
1276 |
{ |
|
ds@1952
|
1277 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1278 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
1279 |
{ |
|
ds@1952
|
1280 |
return 0; |
|
ds@1952
|
1281 |
} |
|
ds@1952
|
1282 |
|
|
ds@1952
|
1283 |
if (mode == Access_Real) |
|
ds@1952
|
1284 |
{ |
|
ds@1952
|
1285 |
return pUser->flags; |
|
ds@1952
|
1286 |
} else if (mode == Access_Effective) { |
|
ds@1952
|
1287 |
return pUser->eflags; |
|
ds@1952
|
1288 |
} |
|
ds@1952
|
1289 |
|
|
ds@1952
|
1290 |
return 0; |
|
ds@1952
|
1291 |
} |
|
ds@1952
|
1292 |
|
|
ds@1952
|
1293 |
void AdminCache::SetAdminFlags(AdminId id, AccessMode mode, FlagBits bits) |
|
ds@1952
|
1294 |
{ |
|
ds@1952
|
1295 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1296 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
1297 |
{ |
|
ds@1952
|
1298 |
return; |
|
ds@1952
|
1299 |
} |
|
ds@1952
|
1300 |
|
|
ds@1952
|
1301 |
if (mode == Access_Real) |
|
ds@1952
|
1302 |
{ |
|
ds@1952
|
1303 |
pUser->flags = bits; |
|
ds@1952
|
1304 |
pUser->eflags = bits; |
|
ds@1952
|
1305 |
} else if (mode == Access_Effective) { |
|
ds@1952
|
1306 |
pUser->eflags = bits; |
|
ds@1952
|
1307 |
} |
|
ds@1952
|
1308 |
|
|
ds@1952
|
1309 |
pUser->serialchange++; |
|
ds@1952
|
1310 |
} |
|
ds@1952
|
1311 |
|
|
ds@1952
|
1312 |
bool AdminCache::AdminInheritGroup(AdminId id, GroupId gid) |
|
ds@1952
|
1313 |
{ |
|
ds@1952
|
1314 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1315 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
1316 |
{ |
|
ds@1952
|
1317 |
return false; |
|
ds@1952
|
1318 |
} |
|
ds@1952
|
1319 |
|
|
ds@1952
|
1320 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid); |
|
ds@1952
|
1321 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
1322 |
{ |
|
ds@1952
|
1323 |
return false; |
|
ds@1952
|
1324 |
} |
|
ds@1952
|
1325 |
|
|
ds@1952
|
1326 |
/* First check for duplicates */ |
|
ds@1952
|
1327 |
if (pUser->grp_count != 0) |
|
ds@1952
|
1328 |
{ |
|
ds@1952
|
1329 |
int *temp_table = (int *)m_pMemory->GetAddress(pUser->grp_table); |
|
ds@1952
|
1330 |
for (unsigned int i=0; i<pUser->grp_count; i++) |
|
ds@1952
|
1331 |
{ |
|
ds@1952
|
1332 |
if (temp_table[i] == gid) |
|
ds@1952
|
1333 |
{ |
|
ds@1952
|
1334 |
return false; |
|
ds@1952
|
1335 |
} |
|
ds@1952
|
1336 |
} |
|
ds@1952
|
1337 |
} |
|
ds@1952
|
1338 |
|
|
ds@1952
|
1339 |
int *table; |
|
ds@1952
|
1340 |
if (pUser->grp_count + 1 > pUser->grp_size) |
|
ds@1952
|
1341 |
{ |
|
ds@1952
|
1342 |
int new_size = 0; |
|
ds@1952
|
1343 |
int tblidx; |
|
ds@1952
|
1344 |
|
|
ds@1952
|
1345 |
if (pUser->grp_size == 0) |
|
ds@1952
|
1346 |
{ |
|
ds@1952
|
1347 |
new_size = 2; |
|
ds@1952
|
1348 |
} else { |
|
ds@1952
|
1349 |
new_size = pUser->grp_size * 2; |
|
ds@1952
|
1350 |
} |
|
ds@1952
|
1351 |
|
|
ds@1952
|
1352 |
/* Create and refresh pointers */ |
|
ds@1952
|
1353 |
tblidx = m_pMemory->CreateMem(new_size * sizeof(int), (void **)&table); |
|
ds@1952
|
1354 |
pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1355 |
pGroup = (AdminGroup *)m_pMemory->GetAddress(gid); |
|
ds@1952
|
1356 |
|
|
ds@1952
|
1357 |
/* Copy old data if necessary */ |
|
ds@1952
|
1358 |
if (pUser->grp_table != -1) |
|
ds@1952
|
1359 |
{ |
|
ds@1952
|
1360 |
int *old_table = (int *)m_pMemory->GetAddress(pUser->grp_table); |
|
ds@1952
|
1361 |
memcpy(table, old_table, sizeof(int) * pUser->grp_count); |
|
ds@1952
|
1362 |
} |
|
ds@1952
|
1363 |
pUser->grp_table = tblidx; |
|
ds@1952
|
1364 |
pUser->grp_size = new_size; |
|
ds@1952
|
1365 |
} else { |
|
ds@1952
|
1366 |
table = (int *)m_pMemory->GetAddress(pUser->grp_table); |
|
ds@1952
|
1367 |
} |
|
ds@1952
|
1368 |
|
|
ds@1952
|
1369 |
table[pUser->grp_count] = gid; |
|
ds@1952
|
1370 |
pUser->grp_count++; |
|
ds@1952
|
1371 |
|
|
ds@1952
|
1372 |
/* Compute new effective permissions */ |
|
ds@1952
|
1373 |
pUser->eflags |= pGroup->addflags; |
|
ds@1952
|
1374 |
|
|
ds@1952
|
1375 |
if (pGroup->immunity_level > pUser->immunity_level) |
|
ds@1952
|
1376 |
{ |
|
ds@1952
|
1377 |
pUser->immunity_level = pGroup->immunity_level; |
|
ds@1952
|
1378 |
} |
|
ds@1952
|
1379 |
|
|
ds@1952
|
1380 |
pUser->serialchange++; |
|
ds@1952
|
1381 |
|
|
ds@1952
|
1382 |
return true; |
|
ds@1952
|
1383 |
} |
|
ds@1952
|
1384 |
|
|
ds@1952
|
1385 |
bool AdminCache::IsValidAdmin(AdminId id) |
|
ds@1952
|
1386 |
{ |
|
ds@1952
|
1387 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1388 |
return (pUser != NULL && pUser->magic == USR_MAGIC_SET); |
|
ds@1952
|
1389 |
} |
|
ds@1952
|
1390 |
|
|
ds@1952
|
1391 |
unsigned int AdminCache::GetAdminGroupCount(AdminId id) |
|
ds@1952
|
1392 |
{ |
|
ds@1952
|
1393 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1394 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
1395 |
{ |
|
ds@1952
|
1396 |
return 0; |
|
ds@1952
|
1397 |
} |
|
ds@1952
|
1398 |
|
|
ds@1952
|
1399 |
return pUser->grp_count; |
|
ds@1952
|
1400 |
} |
|
ds@1952
|
1401 |
|
|
ds@1952
|
1402 |
GroupId AdminCache::GetAdminGroup(AdminId id, unsigned int index, const char **name) |
|
ds@1952
|
1403 |
{ |
|
ds@1952
|
1404 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1405 |
if (!pUser || pUser->magic != USR_MAGIC_SET || index >= pUser->grp_count) |
|
ds@1952
|
1406 |
{ |
|
ds@1952
|
1407 |
return INVALID_GROUP_ID; |
|
ds@1952
|
1408 |
} |
|
ds@1952
|
1409 |
|
|
ds@1952
|
1410 |
int *table = (int *)m_pMemory->GetAddress(pUser->grp_table); |
|
ds@1952
|
1411 |
|
|
ds@1952
|
1412 |
GroupId gid = table[index]; |
|
ds@1952
|
1413 |
|
|
ds@1952
|
1414 |
if (name) |
|
ds@1952
|
1415 |
{ |
|
ds@1952
|
1416 |
*name = GetGroupName(gid); |
|
ds@1952
|
1417 |
} |
|
ds@1952
|
1418 |
|
|
ds@1952
|
1419 |
return gid; |
|
ds@1952
|
1420 |
} |
|
ds@1952
|
1421 |
|
|
ds@1952
|
1422 |
const char *AdminCache::GetAdminPassword(AdminId id) |
|
ds@1952
|
1423 |
{ |
|
ds@1952
|
1424 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1425 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
1426 |
{ |
|
ds@1952
|
1427 |
return NULL; |
|
ds@1952
|
1428 |
} |
|
ds@1952
|
1429 |
|
|
ds@1952
|
1430 |
return m_pStrings->GetString(pUser->password); |
|
ds@1952
|
1431 |
} |
|
ds@1952
|
1432 |
|
|
ds@1952
|
1433 |
void AdminCache::SetAdminPassword(AdminId id, const char *password) |
|
ds@1952
|
1434 |
{ |
|
ds@1952
|
1435 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1436 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
1437 |
{ |
|
ds@1952
|
1438 |
return; |
|
ds@1952
|
1439 |
} |
|
ds@1952
|
1440 |
|
|
ds@1952
|
1441 |
int i_password = m_pStrings->AddString(password); |
|
ds@1952
|
1442 |
pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1443 |
pUser->password = i_password; |
|
ds@1952
|
1444 |
} |
|
ds@1952
|
1445 |
|
|
ds@1952
|
1446 |
unsigned int AdminCache::FlagBitsToBitArray(FlagBits bits, bool array[], unsigned int maxSize) |
|
ds@1952
|
1447 |
{ |
|
ds@1952
|
1448 |
unsigned int i; |
|
ds@1952
|
1449 |
for (i=0; i<maxSize && i<AdminFlags_TOTAL; i++) |
|
ds@1952
|
1450 |
{ |
|
ds@1952
|
1451 |
array[i] = ((bits & (1<<i)) == (unsigned)(1<<i)); |
|
ds@1952
|
1452 |
} |
|
ds@1952
|
1453 |
|
|
ds@1952
|
1454 |
return i; |
|
ds@1952
|
1455 |
} |
|
ds@1952
|
1456 |
|
|
ds@1952
|
1457 |
FlagBits AdminCache::FlagBitArrayToBits(const bool array[], unsigned int maxSize) |
|
ds@1952
|
1458 |
{ |
|
ds@1952
|
1459 |
FlagBits bits = 0; |
|
ds@1952
|
1460 |
for (unsigned int i=0; i<maxSize && i<AdminFlags_TOTAL; i++) |
|
ds@1952
|
1461 |
{ |
|
ds@1952
|
1462 |
if (array[i]) |
|
ds@1952
|
1463 |
{ |
|
ds@1952
|
1464 |
bits |= (1<<i); |
|
ds@1952
|
1465 |
} |
|
ds@1952
|
1466 |
} |
|
ds@1952
|
1467 |
|
|
ds@1952
|
1468 |
return bits; |
|
ds@1952
|
1469 |
} |
|
ds@1952
|
1470 |
|
|
ds@1952
|
1471 |
FlagBits AdminCache::FlagArrayToBits(const AdminFlag array[], unsigned int numFlags) |
|
ds@1952
|
1472 |
{ |
|
ds@1952
|
1473 |
FlagBits bits = 0; |
|
ds@1952
|
1474 |
for (unsigned int i=0; i<numFlags && i<AdminFlags_TOTAL; i++) |
|
ds@1952
|
1475 |
{ |
|
ds@1952
|
1476 |
bits |= (1 << (FlagBits)array[i]); |
|
ds@1952
|
1477 |
} |
|
ds@1952
|
1478 |
|
|
ds@1952
|
1479 |
return bits; |
|
ds@1952
|
1480 |
} |
|
ds@1952
|
1481 |
|
|
ds@1952
|
1482 |
unsigned int AdminCache::FlagBitsToArray(FlagBits bits, AdminFlag array[], unsigned int maxSize) |
|
ds@1952
|
1483 |
{ |
|
ds@1952
|
1484 |
unsigned int i, num=0; |
|
ds@1952
|
1485 |
for (i=0; num<maxSize && i<AdminFlags_TOTAL; i++) |
|
ds@1952
|
1486 |
{ |
|
ds@1952
|
1487 |
if ((bits & (1<<i)) == (unsigned)(1<<i)) |
|
ds@1952
|
1488 |
{ |
|
ds@1952
|
1489 |
array[num++] = (AdminFlag)i; |
|
ds@1952
|
1490 |
} |
|
ds@1952
|
1491 |
} |
|
ds@1952
|
1492 |
|
|
ds@1952
|
1493 |
return num; |
|
ds@1952
|
1494 |
} |
|
ds@1952
|
1495 |
|
|
ds@1952
|
1496 |
bool AdminCache::CheckAdminFlags(AdminId id, FlagBits bits) |
|
ds@1952
|
1497 |
{ |
|
ds@1952
|
1498 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1499 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
1500 |
{ |
|
ds@1952
|
1501 |
return false; |
|
ds@1952
|
1502 |
} |
|
ds@1952
|
1503 |
|
|
ds@1952
|
1504 |
return ((pUser->eflags & bits) == bits); |
|
ds@1952
|
1505 |
} |
|
ds@1952
|
1506 |
|
|
ds@1952
|
1507 |
bool AdminCache::CanAdminTarget(AdminId id, AdminId target) |
|
ds@1952
|
1508 |
{ |
|
ds@1952
|
1509 |
/** |
|
ds@1952
|
1510 |
* Zeroth, if the targeting AdminId is INVALID_ADMIN_ID, targeting fails. |
|
ds@1952
|
1511 |
* First, if the targeted AdminId is INVALID_ADMIN_ID, targeting succeeds. |
|
ds@1952
|
1512 |
*/ |
|
ds@1952
|
1513 |
|
|
ds@1952
|
1514 |
if (id == INVALID_ADMIN_ID) |
|
ds@1952
|
1515 |
{ |
|
ds@1952
|
1516 |
return false; |
|
ds@1952
|
1517 |
} |
|
ds@1952
|
1518 |
|
|
ds@1952
|
1519 |
if (target == INVALID_ADMIN_ID) |
|
ds@1952
|
1520 |
{ |
|
ds@1952
|
1521 |
return true; |
|
ds@1952
|
1522 |
} |
|
ds@1952
|
1523 |
|
|
ds@1952
|
1524 |
if (id == target) |
|
ds@1952
|
1525 |
{ |
|
ds@1952
|
1526 |
return true; |
|
ds@1952
|
1527 |
} |
|
ds@1952
|
1528 |
|
|
ds@1952
|
1529 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1530 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
1531 |
{ |
|
ds@1952
|
1532 |
return false; |
|
ds@1952
|
1533 |
} |
|
ds@1952
|
1534 |
|
|
ds@1952
|
1535 |
AdminUser *pTarget = (AdminUser *)m_pMemory->GetAddress(target); |
|
ds@1952
|
1536 |
if (!pTarget || pTarget->magic != USR_MAGIC_SET) |
|
ds@1952
|
1537 |
{ |
|
ds@1952
|
1538 |
return false; |
|
ds@1952
|
1539 |
} |
|
ds@1952
|
1540 |
|
|
ds@1952
|
1541 |
/** |
|
ds@1952
|
1542 |
* Second, if the targeting admin is root, targeting succeeds. |
|
ds@1952
|
1543 |
*/ |
|
ds@1952
|
1544 |
if (pUser->eflags & ADMFLAG_ROOT) |
|
ds@1952
|
1545 |
{ |
|
ds@1952
|
1546 |
return true; |
|
ds@1952
|
1547 |
} |
|
ds@1952
|
1548 |
|
|
ds@1952
|
1549 |
/** Fourth, if the targeted admin is immune from targeting admin. */ |
|
ds@1952
|
1550 |
int mode = sm_immunity_mode.GetInt(); |
|
ds@1952
|
1551 |
switch (mode) |
|
ds@1952
|
1552 |
{ |
|
ds@1952
|
1553 |
case 1: |
|
ds@1952
|
1554 |
{ |
|
ds@1952
|
1555 |
if (pTarget->immunity_level > pUser->immunity_level) |
|
ds@1952
|
1556 |
{ |
|
ds@1952
|
1557 |
return false; |
|
ds@1952
|
1558 |
} |
|
ds@1952
|
1559 |
break; |
|
ds@1952
|
1560 |
} |
|
ds@1952
|
1561 |
case 3: |
|
ds@1952
|
1562 |
{ |
|
ds@1952
|
1563 |
/* If neither has any immunity, let this pass. */ |
|
ds@1952
|
1564 |
if (!pUser->immunity_level && !pTarget->immunity_level) |
|
ds@1952
|
1565 |
{ |
|
ds@1952
|
1566 |
return true; |
|
ds@1952
|
1567 |
} |
|
ds@1952
|
1568 |
/* Don't break, go to the next case. */ |
|
ds@1952
|
1569 |
} |
|
ds@1952
|
1570 |
case 2: |
|
ds@1952
|
1571 |
{ |
|
ds@1952
|
1572 |
if (pTarget->immunity_level >= pUser->immunity_level) |
|
ds@1952
|
1573 |
{ |
|
ds@1952
|
1574 |
return false; |
|
ds@1952
|
1575 |
} |
|
ds@1952
|
1576 |
break; |
|
ds@1952
|
1577 |
} |
|
ds@1952
|
1578 |
} |
|
ds@1952
|
1579 |
|
|
ds@1952
|
1580 |
/** |
|
ds@1952
|
1581 |
* Fifth, if the targeted admin has specific immunity from the |
|
ds@1952
|
1582 |
* targeting admin via group immunities, targeting fails. |
|
ds@1952
|
1583 |
*/ |
|
ds@1952
|
1584 |
//:TODO: speed this up... maybe with trie hacks. |
|
ds@1952
|
1585 |
//idea is to insert %d.%d in the trie after computing this and use it as a cache lookup. |
|
ds@1952
|
1586 |
//problem is the trie cannot delete prefixes, so we'd have a problem with invalidations. |
|
ds@1952
|
1587 |
if (pTarget->grp_count > 0 && pUser->grp_count > 0) |
|
ds@1952
|
1588 |
{ |
|
ds@1952
|
1589 |
int *grp_table = (int *)m_pMemory->GetAddress(pTarget->grp_table); |
|
ds@1952
|
1590 |
int *src_table = (int *)m_pMemory->GetAddress(pUser->grp_table); |
|
ds@1952
|
1591 |
GroupId id, other; |
|
ds@1952
|
1592 |
unsigned int num; |
|
ds@1952
|
1593 |
for (unsigned int i=0; i<pTarget->grp_count; i++) |
|
ds@1952
|
1594 |
{ |
|
ds@1952
|
1595 |
id = grp_table[i]; |
|
ds@1952
|
1596 |
num = GetGroupImmunityCount(id); |
|
ds@1952
|
1597 |
for (unsigned int j=0; j<num; i++) |
|
ds@1952
|
1598 |
{ |
|
ds@1952
|
1599 |
other = GetGroupImmunity(id, j); |
|
ds@1952
|
1600 |
for (unsigned int k=0; k<pUser->grp_count; k++) |
|
ds@1952
|
1601 |
{ |
|
ds@1952
|
1602 |
if (other == src_table[k]) |
|
ds@1952
|
1603 |
{ |
|
ds@1952
|
1604 |
return false; |
|
ds@1952
|
1605 |
} |
|
ds@1952
|
1606 |
} |
|
ds@1952
|
1607 |
} |
|
ds@1952
|
1608 |
} |
|
ds@1952
|
1609 |
} |
|
ds@1952
|
1610 |
|
|
ds@1952
|
1611 |
return true; |
|
ds@1952
|
1612 |
} |
|
ds@1952
|
1613 |
|
|
ds@1952
|
1614 |
bool AdminCache::FindFlag(char c, AdminFlag *pAdmFlag) |
|
ds@1952
|
1615 |
{ |
|
dvander@1965
|
1616 |
if (c < 'a' |
|
dvander@1965
|
1617 |
|| c > 'z' |
|
dvander@1965
|
1618 |
|| !g_FlagSet[(unsigned)c - (unsigned)'a']) |
|
ds@1952
|
1619 |
{ |
|
ds@1952
|
1620 |
return false; |
|
ds@1952
|
1621 |
} |
|
ds@1952
|
1622 |
|
|
ds@1952
|
1623 |
if (pAdmFlag) |
|
ds@1952
|
1624 |
{ |
|
ds@1952
|
1625 |
*pAdmFlag = g_FlagLetters[(unsigned)c - (unsigned)'a']; |
|
ds@1952
|
1626 |
} |
|
ds@1952
|
1627 |
|
|
ds@1952
|
1628 |
return true; |
|
ds@1952
|
1629 |
} |
|
ds@1952
|
1630 |
|
|
dvander@1965
|
1631 |
bool AdminCache::FindFlagChar(AdminFlag flag, char *c) |
|
dvander@1965
|
1632 |
{ |
|
dvander@1965
|
1633 |
if (!g_FlagSet[flag]) |
|
dvander@1965
|
1634 |
{ |
|
dvander@1965
|
1635 |
return false; |
|
dvander@1965
|
1636 |
} |
|
dvander@1965
|
1637 |
|
|
dvander@1965
|
1638 |
if (c) |
|
dvander@1965
|
1639 |
{ |
|
dvander@1965
|
1640 |
*c = g_ReverseFlags[flag]; |
|
dvander@1965
|
1641 |
} |
|
dvander@1965
|
1642 |
|
|
dvander@1965
|
1643 |
return true; |
|
dvander@1965
|
1644 |
} |
|
dvander@1965
|
1645 |
|
|
ds@1952
|
1646 |
FlagBits AdminCache::ReadFlagString(const char *flags, const char **end) |
|
ds@1952
|
1647 |
{ |
|
ds@1952
|
1648 |
FlagBits bits = 0; |
|
ds@1952
|
1649 |
|
|
ds@1952
|
1650 |
while (flags && (*flags != '\0')) |
|
ds@1952
|
1651 |
{ |
|
ds@1952
|
1652 |
AdminFlag flag; |
|
ds@1952
|
1653 |
if (!FindFlag(*flags, &flag)) |
|
ds@1952
|
1654 |
{ |
|
ds@1952
|
1655 |
break; |
|
ds@1952
|
1656 |
} |
|
ds@1952
|
1657 |
bits |= FlagArrayToBits(&flag, 1); |
|
ds@1952
|
1658 |
flags++; |
|
ds@1952
|
1659 |
} |
|
ds@1952
|
1660 |
|
|
ds@1952
|
1661 |
if (end) |
|
ds@1952
|
1662 |
{ |
|
ds@1952
|
1663 |
*end = flags; |
|
ds@1952
|
1664 |
} |
|
ds@1952
|
1665 |
|
|
ds@1952
|
1666 |
return bits; |
|
ds@1952
|
1667 |
} |
|
ds@1952
|
1668 |
|
|
ds@1952
|
1669 |
unsigned int AdminCache::GetAdminSerialChange(AdminId id) |
|
ds@1952
|
1670 |
{ |
|
ds@1952
|
1671 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1672 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
1673 |
{ |
|
ds@1952
|
1674 |
return 0; |
|
ds@1952
|
1675 |
} |
|
ds@1952
|
1676 |
|
|
ds@1952
|
1677 |
return pUser->serialchange; |
|
ds@1952
|
1678 |
} |
|
ds@1952
|
1679 |
|
|
ds@1952
|
1680 |
bool AdminCache::CanAdminUseCommand(int client, const char *cmd) |
|
ds@1952
|
1681 |
{ |
|
ds@1952
|
1682 |
FlagBits bits; |
|
ds@1952
|
1683 |
OverrideType otype = Override_Command; |
|
ds@1952
|
1684 |
|
|
ds@1952
|
1685 |
if (cmd[0] == '@') |
|
ds@1952
|
1686 |
{ |
|
ds@1952
|
1687 |
otype = Override_CommandGroup; |
|
ds@1952
|
1688 |
cmd++; |
|
ds@1952
|
1689 |
} |
|
ds@1952
|
1690 |
|
|
ds@1952
|
1691 |
if (!g_ConCmds.LookForCommandAdminFlags(cmd, &bits)) |
|
ds@1952
|
1692 |
{ |
|
ds@1952
|
1693 |
if (!GetCommandOverride(cmd, otype, &bits)) |
|
ds@1952
|
1694 |
{ |
|
ds@1952
|
1695 |
bits = 0; |
|
ds@1952
|
1696 |
} |
|
ds@1952
|
1697 |
} |
|
ds@1952
|
1698 |
|
|
ds@1952
|
1699 |
return g_ConCmds.CheckCommandAccess(client, cmd, bits); |
|
ds@1952
|
1700 |
} |
|
ds@1952
|
1701 |
|
|
ds@1952
|
1702 |
unsigned int AdminCache::SetGroupImmunityLevel(GroupId gid, unsigned int level) |
|
ds@1952
|
1703 |
{ |
|
ds@1952
|
1704 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid); |
|
ds@1952
|
1705 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
1706 |
{ |
|
ds@1952
|
1707 |
return 0; |
|
ds@1952
|
1708 |
} |
|
ds@1952
|
1709 |
|
|
ds@1952
|
1710 |
unsigned int old_level = pGroup->immunity_level; |
|
ds@1952
|
1711 |
|
|
ds@1952
|
1712 |
pGroup->immunity_level = level; |
|
ds@1952
|
1713 |
|
|
ds@1952
|
1714 |
return old_level; |
|
ds@1952
|
1715 |
} |
|
ds@1952
|
1716 |
|
|
ds@1952
|
1717 |
unsigned int AdminCache::GetGroupImmunityLevel(GroupId gid) |
|
ds@1952
|
1718 |
{ |
|
ds@1952
|
1719 |
AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid); |
|
ds@1952
|
1720 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
ds@1952
|
1721 |
{ |
|
ds@1952
|
1722 |
return 0; |
|
ds@1952
|
1723 |
} |
|
ds@1952
|
1724 |
|
|
ds@1952
|
1725 |
return pGroup->immunity_level; |
|
ds@1952
|
1726 |
} |
|
ds@1952
|
1727 |
|
|
ds@1952
|
1728 |
unsigned int AdminCache::SetAdminImmunityLevel(AdminId id, unsigned int level) |
|
ds@1952
|
1729 |
{ |
|
ds@1952
|
1730 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1731 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
1732 |
{ |
|
ds@1952
|
1733 |
return 0; |
|
ds@1952
|
1734 |
} |
|
ds@1952
|
1735 |
|
|
ds@1952
|
1736 |
unsigned int old_level = pUser->immunity_level; |
|
ds@1952
|
1737 |
|
|
ds@1952
|
1738 |
pUser->immunity_level = level; |
|
ds@1952
|
1739 |
|
|
ds@1952
|
1740 |
return old_level; |
|
ds@1952
|
1741 |
} |
|
ds@1952
|
1742 |
|
|
ds@1952
|
1743 |
unsigned int AdminCache::GetAdminImmunityLevel(AdminId id) |
|
ds@1952
|
1744 |
{ |
|
ds@1952
|
1745 |
AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id); |
|
ds@1952
|
1746 |
if (!pUser || pUser->magic != USR_MAGIC_SET) |
|
ds@1952
|
1747 |
{ |
|
ds@1952
|
1748 |
return 0; |
|
ds@1952
|
1749 |
} |
|
ds@1952
|
1750 |
|
|
ds@1952
|
1751 |
return pUser->immunity_level; |
|
ds@1952
|
1752 |
} |
|
ds@1952
|
1753 |
|
|
ds@1952
|
1754 |
bool AdminCache::CheckAccess(int client, const char *cmd, FlagBits flags, bool override_only) |
|
ds@1952
|
1755 |
{ |
|
ds@1952
|
1756 |
if (client == 0) |
|
ds@1952
|
1757 |
{ |
|
ds@1952
|
1758 |
return true; |
|
ds@1952
|
1759 |
} |
|
ds@1952
|
1760 |
|
|
ds@1952
|
1761 |
/* Auto-detect a command if we can */ |
|
ds@1952
|
1762 |
FlagBits bits = flags; |
|
ds@1952
|
1763 |
bool found_command = false; |
|
ds@1952
|
1764 |
if (!override_only) |
|
ds@1952
|
1765 |
{ |
|
ds@1952
|
1766 |
found_command = g_ConCmds.LookForCommandAdminFlags(cmd, &bits); |
|
ds@1952
|
1767 |
} |
|
ds@1952
|
1768 |
|
|
ds@1952
|
1769 |
if (!found_command) |
|
ds@1952
|
1770 |
{ |
|
ds@1952
|
1771 |
GetCommandOverride(cmd, Override_Command, &bits); |
|
ds@1952
|
1772 |
} |
|
ds@1952
|
1773 |
|
|
ds@1952
|
1774 |
return g_ConCmds.CheckCommandAccess(client, cmd, bits) ? 1 : 0; |
|
ds@1952
|
1775 |
} |
|
dvander@1965
|
1776 |
|
|
dvander@1965
|
1777 |
void iterator_glob_basic_override(Trie *pTrie, const char *key, void **value, void *data) |
|
dvander@1965
|
1778 |
{ |
|
dvander@1965
|
1779 |
FILE *fp; |
|
dvander@1965
|
1780 |
int flags; |
|
dvander@1965
|
1781 |
char flagstr[64]; |
|
dvander@1965
|
1782 |
|
|
dvander@1965
|
1783 |
fp = (FILE *)data; |
|
dvander@1965
|
1784 |
flags = (int)*value; |
|
dvander@1965
|
1785 |
g_Admins.FillFlagString(flags, flagstr, sizeof(flagstr)); |
|
dvander@1965
|
1786 |
|
|
dvander@1965
|
1787 |
fprintf(fp, "\t\"%s\"\t\t\"%s\"\n", key, flagstr); |
|
dvander@1965
|
1788 |
} |
|
dvander@1965
|
1789 |
|
|
dvander@1965
|
1790 |
void iterator_glob_grp_override(Trie *pTrie, const char *key, void **value, void *data) |
|
dvander@1965
|
1791 |
{ |
|
dvander@1965
|
1792 |
FILE *fp; |
|
dvander@1965
|
1793 |
int flags; |
|
dvander@1965
|
1794 |
char flagstr[64]; |
|
dvander@1965
|
1795 |
|
|
dvander@1965
|
1796 |
fp = (FILE *)data; |
|
dvander@1965
|
1797 |
flags = (int)*value; |
|
dvander@1965
|
1798 |
g_Admins.FillFlagString(flags, flagstr, sizeof(flagstr)); |
|
dvander@1965
|
1799 |
|
|
dvander@1965
|
1800 |
fprintf(fp, "\t\"@%s\"\t\t\"%s\"\n", key, flagstr); |
|
dvander@1965
|
1801 |
} |
|
dvander@1965
|
1802 |
|
|
dvander@1965
|
1803 |
void iterator_group_basic_override(Trie *pTrie, const char *key, void **value, void *data) |
|
dvander@1965
|
1804 |
{ |
|
dvander@1965
|
1805 |
FILE *fp; |
|
dvander@1965
|
1806 |
int flags; |
|
dvander@1965
|
1807 |
char flagstr[64]; |
|
dvander@1965
|
1808 |
|
|
dvander@1965
|
1809 |
fp = (FILE *)data; |
|
dvander@1965
|
1810 |
flags = (int)*value; |
|
dvander@1965
|
1811 |
g_Admins.FillFlagString(flags, flagstr, sizeof(flagstr)); |
|
dvander@1965
|
1812 |
|
|
dvander@1965
|
1813 |
fprintf(fp, "\t\t\t\"%s\"\t\t\"%s\"\n", key, flagstr); |
|
dvander@1965
|
1814 |
} |
|
dvander@1965
|
1815 |
|
|
dvander@1965
|
1816 |
void iterator_group_grp_override(Trie *pTrie, const char *key, void **value, void *data) |
|
dvander@1965
|
1817 |
{ |
|
dvander@1965
|
1818 |
FILE *fp; |
|
dvander@1965
|
1819 |
int flags; |
|
dvander@1965
|
1820 |
char flagstr[64]; |
|
dvander@1965
|
1821 |
|
|
dvander@1965
|
1822 |
fp = (FILE *)data; |
|
dvander@1965
|
1823 |
flags = (int)*value; |
|
dvander@1965
|
1824 |
g_Admins.FillFlagString(flags, flagstr, sizeof(flagstr)); |
|
dvander@1965
|
1825 |
|
|
dvander@1965
|
1826 |
fprintf(fp, "\t\t\t\"@%s\"\t\t\"%s\"\n", key, flagstr); |
|
dvander@1965
|
1827 |
} |
|
dvander@1965
|
1828 |
|
|
dvander@1965
|
1829 |
void AdminCache::DumpCache(FILE *fp) |
|
dvander@1965
|
1830 |
{ |
|
dvander@1965
|
1831 |
int *itable; |
|
dvander@1965
|
1832 |
AdminId aid; |
|
dvander@1965
|
1833 |
GroupId gid; |
|
dvander@1965
|
1834 |
char flagstr[64]; |
|
dvander@1965
|
1835 |
unsigned int num; |
|
dvander@1965
|
1836 |
AdminUser *pAdmin; |
|
dvander@1965
|
1837 |
AdminGroup *pGroup; |
|
dvander@1965
|
1838 |
char name_buffer[512]; |
|
dvander@1965
|
1839 |
|
|
dvander@1965
|
1840 |
fprintf(fp, "\"Groups\"\n{\n"); |
|
dvander@1965
|
1841 |
|
|
dvander@1965
|
1842 |
num = 0; |
|
dvander@1965
|
1843 |
gid = m_FirstGroup; |
|
dvander@1965
|
1844 |
while (gid != INVALID_GROUP_ID |
|
dvander@1965
|
1845 |
&& (pGroup = GetGroup(gid)) != NULL) |
|
dvander@1965
|
1846 |
{ |
|
dvander@1965
|
1847 |
num++; |
|
dvander@1965
|
1848 |
FillFlagString(pGroup->addflags, flagstr, sizeof(flagstr)); |
|
dvander@1965
|
1849 |
|
|
dvander@1965
|
1850 |
fprintf(fp, "\t/* num = %d, gid = 0x%X */\n", num, gid); |
|
dvander@1965
|
1851 |
fprintf(fp, "\t\"%s\"\n\t{\n", GetString(pGroup->nameidx)); |
|
dvander@1965
|
1852 |
fprintf(fp, "\t\t\"flags\"\t\t\t\"%s\"\n", flagstr); |
|
dvander@1965
|
1853 |
fprintf(fp, "\t\t\"immunity\"\t\t\"%d\"\n", pGroup->immunity_level); |
|
dvander@1965
|
1854 |
|
|
dvander@1965
|
1855 |
if (pGroup->immune_table != -1 |
|
dvander@1965
|
1856 |
&& (itable = (int *)m_pMemory->GetAddress(pGroup->immune_table)) != NULL) |
|
dvander@1965
|
1857 |
{ |
|
dvander@1965
|
1858 |
AdminGroup *pAltGroup; |
|
dvander@1965
|
1859 |
const char *gname, *mod; |
|
dvander@1965
|
1860 |
|
|
dvander@1965
|
1861 |
for (int i = 1; i <= itable[0]; i++) |
|
dvander@1965
|
1862 |
{ |
|
dvander@1965
|
1863 |
if ((pAltGroup = GetGroup(itable[i])) == NULL) |
|
dvander@1965
|
1864 |
{ |
|
dvander@1965
|
1865 |
/* Assume the rest of the table is corrupt */ |
|
dvander@1965
|
1866 |
break; |
|
dvander@1965
|
1867 |
} |
|
dvander@1965
|
1868 |
gname = GetString(pAltGroup->nameidx); |
|
dvander@1965
|
1869 |
if (atoi(gname) != 0) |
|
dvander@1965
|
1870 |
{ |
|
dvander@1965
|
1871 |
mod = "@"; |
|
dvander@1965
|
1872 |
} |
|
dvander@1965
|
1873 |
else |
|
dvander@1965
|
1874 |
{ |
|
dvander@1965
|
1875 |
mod = ""; |
|
dvander@1965
|
1876 |
} |
|
dvander@1965
|
1877 |
fprintf(fp, "\t\t\"immunity\"\t\t\"%s%s\"\n", mod, gname); |
|
dvander@1965
|
1878 |
} |
|
dvander@1965
|
1879 |
} |
|
dvander@1965
|
1880 |
|
|
dvander@1965
|
1881 |
fprintf(fp, "\n\t\t\"Overrides\"\n\t\t{\n"); |
|
dvander@1965
|
1882 |
if (pGroup->pCmdGrpTable != NULL) |
|
dvander@1965
|
1883 |
{ |
|
dvander@1965
|
1884 |
sm_trie_bad_iterator(pGroup->pCmdGrpTable, |
|
dvander@1965
|
1885 |
name_buffer, |
|
dvander@1965
|
1886 |
sizeof(name_buffer), |
|
dvander@1965
|
1887 |
iterator_group_grp_override, |
|
dvander@1965
|
1888 |
fp); |
|
dvander@1965
|
1889 |
} |
|
dvander@1965
|
1890 |
if (pGroup->pCmdTable != NULL) |
|
dvander@1965
|
1891 |
{ |
|
dvander@1965
|
1892 |
sm_trie_bad_iterator(pGroup->pCmdTable, |
|
dvander@1965
|
1893 |
name_buffer, |
|
dvander@1965
|
1894 |
sizeof(name_buffer), |
|
dvander@1965
|
1895 |
iterator_group_basic_override, |
|
dvander@1965
|
1896 |
fp); |
|
dvander@1965
|
1897 |
} |
|
dvander@1965
|
1898 |
fprintf(fp, "\t\t}\n"); |
|
dvander@1965
|
1899 |
|
|
dvander@1965
|
1900 |
fprintf(fp, "\t}\n"); |
|
dvander@1965
|
1901 |
|
|
dvander@1965
|
1902 |
if ((gid = pGroup->next_grp) != INVALID_GROUP_ID) |
|
dvander@1965
|
1903 |
{ |
|
dvander@1965
|
1904 |
fprintf(fp, "\n"); |
|
dvander@1965
|
1905 |
} |
|
dvander@1965
|
1906 |
} |
|
dvander@1965
|
1907 |
|
|
dvander@1965
|
1908 |
fprintf(fp, "}\n\n"); |
|
dvander@1965
|
1909 |
fprintf(fp, "\"Admins\"\n{\n"); |
|
dvander@1965
|
1910 |
|
|
dvander@1965
|
1911 |
num = 0; |
|
dvander@1965
|
1912 |
aid = m_FirstUser; |
|
dvander@1965
|
1913 |
while (aid != INVALID_ADMIN_ID |
|
dvander@1965
|
1914 |
&& (pAdmin = GetUser(aid)) != NULL) |
|
dvander@1965
|
1915 |
{ |
|
dvander@1965
|
1916 |
num++; |
|
dvander@1965
|
1917 |
FillFlagString(pAdmin->flags, flagstr, sizeof(flagstr)); |
|
dvander@1965
|
1918 |
|
|
dvander@1965
|
1919 |
fprintf(fp, "\t/* num = %d, aid = 0x%X, serialno = 0x%X*/\n", num, aid, pAdmin->serialchange); |
|
dvander@1965
|
1920 |
|
|
dvander@1965
|
1921 |
if (pAdmin->nameidx != -1) |
|
dvander@1965
|
1922 |
{ |
|
dvander@1965
|
1923 |
fprintf(fp, "\t\"%s\"\n\t{\n", GetString(pAdmin->nameidx)); |
|
dvander@1965
|
1924 |
} |
|
dvander@1965
|
1925 |
else |
|
dvander@1965
|
1926 |
{ |
|
dvander@1965
|
1927 |
fprintf(fp, "\t\"\"\n\t{\n"); |
|
dvander@1965
|
1928 |
} |
|
dvander@1965
|
1929 |
|
|
dvander@1965
|
1930 |
if (pAdmin->auth.identidx != -1) |
|
dvander@1965
|
1931 |
{ |
|
dvander@1965
|
1932 |
fprintf(fp, "\t\t\"auth\"\t\t\t\"%s\"\n", GetMethodName(pAdmin->auth.index)); |
|
dvander@1965
|
1933 |
fprintf(fp, "\t\t\"identity\"\t\t\"%s\"\n", GetString(pAdmin->auth.identidx)); |
|
dvander@1965
|
1934 |
} |
|
dvander@1965
|
1935 |
if (pAdmin->password != -1) |
|
dvander@1965
|
1936 |
{ |
|
dvander@1965
|
1937 |
fprintf(fp, "\t\t\"password\"\t\t\"%s\"\n", GetString(pAdmin->password)); |
|
dvander@1965
|
1938 |
} |
|
dvander@1965
|
1939 |
fprintf(fp, "\t\t\"flags\"\t\t\t\"%s\"\n", flagstr); |
|
dvander@1965
|
1940 |
fprintf(fp, "\t\t\"immunity\"\t\t\"%d\"\n", pAdmin->immunity_level); |
|
dvander@1965
|
1941 |
|
|
dvander@1965
|
1942 |
if (pAdmin->grp_count != 0 |
|
dvander@1965
|
1943 |
&& pAdmin->grp_table != -1 |
|
dvander@1965
|
1944 |
&& (itable = (int *)m_pMemory->GetAddress(pAdmin->grp_table)) != NULL) |
|
dvander@1965
|
1945 |
{ |
|
dvander@1965
|
1946 |
unsigned int i; |
|
dvander@1965
|
1947 |
|
|
dvander@1965
|
1948 |
for (i = 0; i < pAdmin->grp_count; i++) |
|
dvander@1965
|
1949 |
{ |
|
dvander@1965
|
1950 |
if ((pGroup = GetGroup(itable[i])) == NULL) |
|
dvander@1965
|
1951 |
{ |
|
dvander@1965
|
1952 |
/* Assume the rest of the table is corrupt */ |
|
dvander@1965
|
1953 |
break; |
|
dvander@1965
|
1954 |
} |
|
dvander@1965
|
1955 |
fprintf(fp, "\t\t\"group\"\t\t\t\"%s\"\n", GetString(pGroup->nameidx)); |
|
dvander@1965
|
1956 |
} |
|
dvander@1965
|
1957 |
} |
|
dvander@1965
|
1958 |
|
|
dvander@1965
|
1959 |
fprintf(fp, "\t}\n"); |
|
dvander@1965
|
1960 |
|
|
dvander@1965
|
1961 |
if ((aid = pAdmin->next_user) != INVALID_ADMIN_ID) |
|
dvander@1965
|
1962 |
{ |
|
dvander@1965
|
1963 |
fprintf(fp, "\n"); |
|
dvander@1965
|
1964 |
} |
|
dvander@1965
|
1965 |
} |
|
dvander@1965
|
1966 |
|
|
dvander@1965
|
1967 |
fprintf(fp, "}\n\n"); |
|
dvander@1965
|
1968 |
|
|
dvander@1965
|
1969 |
fprintf(fp, "\"Overrides\"\n{\n"); |
|
dvander@1965
|
1970 |
if (m_pCmdGrpOverrides != NULL) |
|
dvander@1965
|
1971 |
{ |
|
dvander@1965
|
1972 |
sm_trie_bad_iterator(m_pCmdGrpOverrides, |
|
dvander@1965
|
1973 |
name_buffer, |
|
dvander@1965
|
1974 |
sizeof(name_buffer), |
|
dvander@1965
|
1975 |
iterator_glob_grp_override, |
|
dvander@1965
|
1976 |
fp); |
|
dvander@1965
|
1977 |
} |
|
dvander@1965
|
1978 |
if (m_pCmdOverrides != NULL) |
|
dvander@1965
|
1979 |
{ |
|
dvander@1965
|
1980 |
sm_trie_bad_iterator(m_pCmdOverrides, |
|
dvander@1965
|
1981 |
name_buffer, |
|
dvander@1965
|
1982 |
sizeof(name_buffer), |
|
dvander@1965
|
1983 |
iterator_glob_basic_override, |
|
dvander@1965
|
1984 |
fp); |
|
dvander@1965
|
1985 |
} |
|
dvander@1965
|
1986 |
fprintf(fp, "}\n"); |
|
dvander@1965
|
1987 |
} |
|
dvander@1965
|
1988 |
|
|
dvander@1965
|
1989 |
AdminGroup *AdminCache::GetGroup(GroupId gid) |
|
dvander@1965
|
1990 |
{ |
|
dvander@1965
|
1991 |
AdminGroup *pGroup; |
|
dvander@1965
|
1992 |
|
|
dvander@1965
|
1993 |
pGroup = (AdminGroup *)m_pMemory->GetAddress(gid); |
|
dvander@1965
|
1994 |
if (!pGroup || pGroup->magic != GRP_MAGIC_SET) |
|
dvander@1965
|
1995 |
{ |
|
dvander@1965
|
1996 |
return NULL; |
|
dvander@1965
|
1997 |
} |
|
dvander@1965
|
1998 |
|
|
dvander@1965
|
1999 |
return pGroup; |
|
dvander@1965
|
2000 |
} |
|
dvander@1965
|
2001 |
|
|
dvander@1965
|
2002 |
AdminUser *AdminCache::GetUser(AdminId aid) |
|
dvander@1965
|
2003 |
{ |
|
dvander@1965
|
2004 |
AdminUser *pAdmin; |
|
dvander@1965
|
2005 |
|
|
dvander@1965
|
2006 |
pAdmin = (AdminUser *)m_pMemory->GetAddress(aid); |
|
dvander@1965
|
2007 |
if (!pAdmin || pAdmin->magic != USR_MAGIC_SET) |
|
dvander@1965
|
2008 |
{ |
|
dvander@1965
|
2009 |
return NULL; |
|
dvander@1965
|
2010 |
} |
|
dvander@1965
|
2011 |
|
|
dvander@1965
|
2012 |
return pAdmin; |
|
dvander@1965
|
2013 |
} |
|
dvander@1965
|
2014 |
|
|
dvander@1965
|
2015 |
const char *AdminCache::GetMethodName(unsigned int index) |
|
dvander@1965
|
2016 |
{ |
|
dvander@1965
|
2017 |
List<AuthMethod>::iterator iter; |
|
dvander@1965
|
2018 |
for (iter=m_AuthMethods.begin(); |
|
dvander@1965
|
2019 |
iter!=m_AuthMethods.end(); |
|
dvander@1965
|
2020 |
iter++) |
|
dvander@1965
|
2021 |
{ |
|
dvander@1965
|
2022 |
if (index-- == 0) |
|
dvander@1965
|
2023 |
{ |
|
dvander@1965
|
2024 |
return (*iter).name.c_str(); |
|
dvander@1965
|
2025 |
} |
|
dvander@1965
|
2026 |
} |
|
dvander@1965
|
2027 |
|
|
dvander@1965
|
2028 |
return NULL; |
|
dvander@1965
|
2029 |
} |
|
dvander@1965
|
2030 |
|
|
dvander@1965
|
2031 |
const char *AdminCache::GetString(int idx) |
|
dvander@1965
|
2032 |
{ |
|
dvander@1965
|
2033 |
return m_pStrings->GetString(idx); |
|
dvander@1965
|
2034 |
} |
|
dvander@1965
|
2035 |
|
|
dvander@1965
|
2036 |
size_t AdminCache::FillFlagString(FlagBits bits, char *buffer, size_t maxlen) |
|
dvander@1965
|
2037 |
{ |
|
dvander@1965
|
2038 |
size_t pos; |
|
dvander@1965
|
2039 |
unsigned int i, num_flags; |
|
dvander@1965
|
2040 |
AdminFlag flags[AdminFlags_TOTAL]; |
|
dvander@1965
|
2041 |
|
|
dvander@1965
|
2042 |
num_flags = FlagBitsToArray(bits, flags, AdminFlags_TOTAL); |
|
dvander@1965
|
2043 |
|
|
dvander@1965
|
2044 |
pos = 0; |
|
dvander@1965
|
2045 |
for (i = 0; pos < maxlen && i < num_flags; i++) |
|
dvander@1965
|
2046 |
{ |
|
dvander@1965
|
2047 |
if (FindFlagChar(flags[i], &buffer[pos])) |
|
dvander@1965
|
2048 |
{ |
|
dvander@1965
|
2049 |
pos++; |
|
dvander@1965
|
2050 |
} |
|
dvander@1965
|
2051 |
} |
|
dvander@1965
|
2052 |
buffer[pos] = '\0'; |
|
dvander@1965
|
2053 |
|
|
dvander@1965
|
2054 |
return pos; |
|
dvander@1965
|
2055 |
} |
|
dvander@1965
|
2056 |
|
|
dvander@1965
|
2057 |
CON_COMMAND(sm_dump_admcache, "Dumps the admin cache for debugging") |
|
dvander@1965
|
2058 |
{ |
|
dvander@1965
|
2059 |
FILE *fp; |
|
dvander@1965
|
2060 |
char buffer[PLATFORM_MAX_PATH]; |
|
dvander@1965
|
2061 |
|
|
dvander@1965
|
2062 |
g_SourceMod.BuildPath(Path_SM, buffer, sizeof(buffer), "data/admin_cache_dump.txt"); |
|
dvander@1965
|
2063 |
|
|
dvander@1965
|
2064 |
if ((fp = fopen(buffer, "wt")) == NULL) |
|
dvander@1965
|
2065 |
{ |
|
dvander@1965
|
2066 |
g_RootMenu.ConsolePrint("Could not open file for writing: %s", buffer); |
|
dvander@1965
|
2067 |
return; |
|
dvander@1965
|
2068 |
} |
|
dvander@1965
|
2069 |
|
|
dvander@1965
|
2070 |
g_Admins.DumpCache(fp); |
|
dvander@1965
|
2071 |
|
|
dvander@1965
|
2072 |
g_RootMenu.ConsolePrint("Admin cache dumped to: %s", buffer); |
|
dvander@1965
|
2073 |
|
|
dvander@1965
|
2074 |
fclose(fp); |
|
dvander@1965
|
2075 |
} |