core/AdminCache.cpp
author Scott Ehlert <ds@alliedmods.net>
Thu Mar 22 21:50:20 2007 +0000 (2007-03-22)
changeset 653 ef60870067ff
parent 591 86e97f3eda1f
child 673 fc1d50579d06
permissions -rw-r--r--
More consistency of some sort...

Oh god, what has possessed me to do this?
dvander@370
     1
/**
ds@653
     2
 * vim: set ts=4 :
dvander@370
     3
 * ===============================================================
dvander@370
     4
 * SourceMod (C)2004-2007 AlliedModders LLC.  All rights reserved.
dvander@370
     5
 * ===============================================================
dvander@370
     6
 *
dvander@370
     7
 * This file is not open source and may not be copied without explicit
dvander@370
     8
 * written permission of AlliedModders LLC.  This file may not be redistributed 
dvander@370
     9
 * in whole or significant part.
dvander@370
    10
 * For information, see LICENSE.txt or http://www.sourcemod.net/license.php
dvander@370
    11
 *
dvander@370
    12
 * Version: $Id$
dvander@370
    13
 */
dvander@370
    14
dvander@335
    15
#include <string.h>
dvander@335
    16
#include <assert.h>
dvander@335
    17
#include "AdminCache.h"
dvander@335
    18
#include "ShareSys.h"
dvander@336
    19
#include "ForwardSys.h"
dvander@591
    20
#include "PlayerManager.h"
dvander@583
    21
#include "ConCmdManager.h"
dvander@335
    22
dvander@335
    23
AdminCache g_Admins;
dvander@335
    24
dvander@335
    25
AdminCache::AdminCache()
dvander@335
    26
{
dvander@433
    27
	m_pCmdOverrides = sm_trie_create();
dvander@433
    28
	m_pCmdGrpOverrides = sm_trie_create();
dvander@335
    29
	m_pStrings = new BaseStringTable(1024);
dvander@335
    30
	m_pMemory = m_pStrings->GetMemTable();
dvander@335
    31
	m_FreeGroupList = m_FirstGroup = m_LastGroup = INVALID_GROUP_ID;
dvander@404
    32
	m_FreeUserList = m_FirstUser = m_LastUser = INVALID_ADMIN_ID;
dvander@335
    33
	m_pGroups = sm_trie_create();
dvander@336
    34
	m_pCacheFwd = NULL;
dvander@404
    35
	m_FirstGroup = -1;
dvander@405
    36
	m_pAuthTables = sm_trie_create();
dvander@458
    37
	m_InvalidatingAdmins = false;
dvander@520
    38
	m_destroying = false;
dvander@335
    39
}
dvander@335
    40
dvander@335
    41
AdminCache::~AdminCache()
dvander@335
    42
{
dvander@520
    43
	m_destroying = true;
dvander@438
    44
	DumpAdminCache(AdminCache_Overrides, false);
dvander@438
    45
	DumpAdminCache(AdminCache_Groups, false);
dvander@438
    46
dvander@434
    47
	sm_trie_destroy(m_pCmdGrpOverrides);
dvander@434
    48
	sm_trie_destroy(m_pCmdOverrides);
dvander@335
    49
dvander@335
    50
	if (m_pGroups)
dvander@335
    51
	{
dvander@335
    52
		sm_trie_destroy(m_pGroups);
dvander@335
    53
	}
dvander@335
    54
dvander@405
    55
	List<AuthMethod>::iterator iter;
dvander@405
    56
	for (iter=m_AuthMethods.begin();
dvander@405
    57
		 iter!=m_AuthMethods.end();
dvander@405
    58
		 iter++)
dvander@405
    59
	{
dvander@405
    60
		sm_trie_destroy((*iter).table);
dvander@405
    61
	}
dvander@405
    62
dvander@405
    63
	sm_trie_destroy(m_pAuthTables);
dvander@405
    64
dvander@335
    65
	delete m_pStrings;
dvander@335
    66
}
dvander@335
    67
dvander@404
    68
void AdminCache::OnSourceModStartup(bool late)
dvander@404
    69
{
dvander@404
    70
	RegisterAuthIdentType("steam");
dvander@404
    71
	RegisterAuthIdentType("name");
dvander@404
    72
	RegisterAuthIdentType("ip");
dvander@404
    73
}
dvander@404
    74
dvander@335
    75
void AdminCache::OnSourceModAllInitialized()
dvander@335
    76
{
dvander@336
    77
	m_pCacheFwd = g_Forwards.CreateForward("OnRebuildAdminCache", ET_Ignore, 1, NULL, Param_Cell);
dvander@335
    78
	g_ShareSys.AddInterface(NULL, this);
dvander@335
    79
}
dvander@335
    80
dvander@336
    81
void AdminCache::OnSourceModShutdown()
dvander@336
    82
{
dvander@336
    83
	g_Forwards.ReleaseForward(m_pCacheFwd);
dvander@336
    84
	m_pCacheFwd = NULL;
dvander@336
    85
}
dvander@336
    86
dvander@433
    87
void AdminCache::AddCommandOverride(const char *cmd, OverrideType type, FlagBits flags)
dvander@335
    88
{
dvander@433
    89
	Trie *pTrie = NULL;
dvander@335
    90
	if (type == Override_Command)
dvander@335
    91
	{
dvander@433
    92
		pTrie = m_pCmdOverrides;
dvander@335
    93
	} else if (type == Override_CommandGroup) {
dvander@433
    94
		pTrie = m_pCmdGrpOverrides;
dvander@433
    95
	} else {
dvander@433
    96
		return;
dvander@335
    97
	}
dvander@433
    98
dvander@433
    99
	sm_trie_insert(pTrie, cmd, (void *)(unsigned int)flags);
dvander@514
   100
dvander@514
   101
	g_ConCmds.UpdateAdminCmdFlags(cmd, type, flags);
dvander@335
   102
}
dvander@335
   103
dvander@433
   104
bool AdminCache::GetCommandOverride(const char *cmd, OverrideType type, FlagBits *pFlags)
dvander@335
   105
{
dvander@433
   106
	Trie *pTrie = NULL;
dvander@433
   107
dvander@335
   108
	if (type == Override_Command)
dvander@335
   109
	{
dvander@433
   110
		pTrie = m_pCmdOverrides;
dvander@335
   111
	} else if (type == Override_CommandGroup) {
dvander@433
   112
		pTrie = m_pCmdGrpOverrides;
dvander@433
   113
	} else {
dvander@433
   114
		return false;
dvander@433
   115
	}
dvander@433
   116
dvander@433
   117
	void *object;
dvander@433
   118
	if (sm_trie_retrieve(pTrie, cmd, &object))
dvander@433
   119
	{
dvander@433
   120
		if (pFlags)
dvander@433
   121
		{
dvander@433
   122
			*pFlags = (FlagBits)object;
dvander@433
   123
		}
dvander@433
   124
		return true;
dvander@335
   125
	}
dvander@335
   126
dvander@335
   127
	return false;
dvander@335
   128
}
dvander@335
   129
dvander@335
   130
void AdminCache::UnsetCommandOverride(const char *cmd, OverrideType type)
dvander@335
   131
{
dvander@335
   132
	if (type == Override_Command)
dvander@335
   133
	{
dvander@335
   134
		return _UnsetCommandOverride(cmd);
dvander@335
   135
	} else if (type == Override_CommandGroup) {
dvander@335
   136
		return _UnsetCommandGroupOverride(cmd);
dvander@335
   137
	}
dvander@335
   138
}
dvander@335
   139
dvander@335
   140
void AdminCache::_UnsetCommandGroupOverride(const char *group)
dvander@335
   141
{
dvander@335
   142
	if (!m_pCmdGrpOverrides)
dvander@335
   143
	{
dvander@335
   144
		return;
dvander@335
   145
	}
dvander@335
   146
dvander@514
   147
	sm_trie_delete(m_pCmdGrpOverrides, group);
dvander@335
   148
dvander@514
   149
	g_ConCmds.UpdateAdminCmdFlags(group, Override_CommandGroup, 0);
dvander@335
   150
}
dvander@335
   151
dvander@335
   152
void AdminCache::_UnsetCommandOverride(const char *cmd)
dvander@335
   153
{
dvander@335
   154
	if (!m_pCmdOverrides)
dvander@335
   155
	{
dvander@335
   156
		return;
dvander@335
   157
	}
dvander@335
   158
dvander@514
   159
	sm_trie_delete(m_pCmdOverrides, cmd);
dvander@335
   160
dvander@514
   161
	g_ConCmds.UpdateAdminCmdFlags(cmd, Override_Command, 0);
dvander@335
   162
}
dvander@335
   163
dvander@335
   164
void AdminCache::DumpCommandOverrideCache(OverrideType type)
dvander@335
   165
{
dvander@335
   166
	if (type == Override_Command && m_pCmdOverrides)
dvander@335
   167
	{
dvander@335
   168
		sm_trie_clear(m_pCmdOverrides);
dvander@335
   169
	} else if (type == Override_CommandGroup && m_pCmdGrpOverrides) {
dvander@335
   170
		sm_trie_clear(m_pCmdGrpOverrides);
dvander@335
   171
	}
dvander@335
   172
}
dvander@335
   173
dvander@404
   174
AdminId AdminCache::CreateAdmin(const char *name)
dvander@404
   175
{
dvander@404
   176
	AdminId id;
dvander@404
   177
	AdminUser *pUser;
dvander@404
   178
dvander@404
   179
	if (m_FreeUserList != INVALID_ADMIN_ID)
dvander@404
   180
	{
dvander@404
   181
		pUser = (AdminUser *)m_pMemory->GetAddress(m_FreeUserList);
dvander@404
   182
		assert(pUser->magic == USR_MAGIC_UNSET);
dvander@404
   183
		id = m_FreeUserList;
dvander@404
   184
		m_FreeUserList = pUser->next_user;
dvander@404
   185
	} else {
dvander@404
   186
		id = m_pMemory->CreateMem(sizeof(AdminUser), (void **)&pUser);
dvander@404
   187
		pUser->grp_size = 0;
dvander@404
   188
		pUser->grp_table = -1;
dvander@404
   189
	}
dvander@404
   190
dvander@433
   191
	pUser->flags = 0;
dvander@433
   192
	pUser->eflags = 0;
dvander@404
   193
	pUser->grp_count = 0;
dvander@404
   194
	pUser->password = -1;
dvander@404
   195
	pUser->magic = USR_MAGIC_SET;
dvander@405
   196
	pUser->auth.identidx = -1;
dvander@405
   197
	pUser->auth.index = 0;
dvander@434
   198
	pUser->immune_default = false;
dvander@434
   199
	pUser->immune_global = false;
dvander@404
   200
dvander@404
   201
	if (m_FirstUser == INVALID_ADMIN_ID)
dvander@404
   202
	{
dvander@404
   203
		m_FirstUser = id;
dvander@404
   204
		m_LastUser = id;
dvander@404
   205
	} else {
dvander@404
   206
		AdminUser *pPrev = (AdminUser *)m_pMemory->GetAddress(m_LastUser);
dvander@404
   207
		pPrev->next_user = id;
dvander@404
   208
		pUser->prev_user = m_LastUser;
dvander@404
   209
		m_LastUser = id;
dvander@404
   210
	}
dvander@404
   211
dvander@404
   212
	if (name && name[0] != '\0')
dvander@404
   213
	{
dvander@404
   214
		pUser->nameidx = m_pStrings->AddString(name);
dvander@404
   215
	}
dvander@404
   216
dvander@404
   217
	return id;
dvander@404
   218
}
dvander@404
   219
dvander@335
   220
GroupId AdminCache::AddGroup(const char *group_name)
dvander@335
   221
{
dvander@335
   222
	if (sm_trie_retrieve(m_pGroups, group_name, NULL))
dvander@335
   223
	{
dvander@335
   224
		return INVALID_GROUP_ID;
dvander@335
   225
	}
dvander@335
   226
dvander@335
   227
	GroupId id;
dvander@335
   228
	AdminGroup *pGroup;
dvander@335
   229
	if (m_FreeGroupList != INVALID_GROUP_ID)
dvander@335
   230
	{
dvander@335
   231
		pGroup = (AdminGroup *)m_pMemory->GetAddress(m_FreeGroupList);
dvander@335
   232
		assert(pGroup->magic == GRP_MAGIC_UNSET);
dvander@335
   233
		id = m_FreeGroupList;
dvander@335
   234
		m_FreeGroupList = pGroup->next_grp;
dvander@335
   235
	} else {
dvander@335
   236
		id = m_pMemory->CreateMem(sizeof(AdminGroup), (void **)&pGroup);
dvander@335
   237
	}
dvander@335
   238
dvander@335
   239
	pGroup->immune_default = false;
dvander@335
   240
	pGroup->immune_global = false;
dvander@335
   241
	pGroup->immune_table = -1;
dvander@335
   242
	pGroup->magic = GRP_MAGIC_SET;
dvander@335
   243
	pGroup->next_grp = INVALID_GROUP_ID;
dvander@335
   244
	pGroup->pCmdGrpTable = NULL;
dvander@335
   245
	pGroup->pCmdTable = NULL;
dvander@433
   246
	pGroup->addflags = 0;
dvander@335
   247
dvander@335
   248
	if (m_FirstGroup == INVALID_GROUP_ID)
dvander@335
   249
	{
dvander@335
   250
		m_FirstGroup = id;
dvander@335
   251
		m_LastGroup = id;
dvander@335
   252
		pGroup->prev_grp = INVALID_GROUP_ID;
dvander@335
   253
	} else {
dvander@335
   254
		AdminGroup *pPrev = (AdminGroup *)m_pMemory->GetAddress(m_LastGroup);
dvander@335
   255
		assert(pPrev->magic == GRP_MAGIC_SET);
dvander@335
   256
		pPrev->next_grp = id;
dvander@335
   257
		pGroup->prev_grp = m_LastGroup;
dvander@335
   258
		m_LastGroup = id;
dvander@335
   259
	}
dvander@335
   260
dvander@404
   261
	pGroup->nameidx = m_pStrings->AddString(group_name);
dvander@404
   262
dvander@335
   263
	sm_trie_insert(m_pGroups, group_name, (void *)id);
dvander@335
   264
dvander@335
   265
	return id;
dvander@335
   266
}
dvander@335
   267
dvander@335
   268
GroupId AdminCache::FindGroupByName(const char *group_name)
dvander@335
   269
{
dvander@335
   270
	void *object;
dvander@335
   271
dvander@335
   272
	if (!sm_trie_retrieve(m_pGroups, group_name, &object))
dvander@335
   273
	{
dvander@335
   274
		return INVALID_GROUP_ID;
dvander@335
   275
	}
dvander@335
   276
dvander@335
   277
	GroupId id = (GroupId)object;
dvander@335
   278
	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335
   279
dvander@335
   280
	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335
   281
	{
dvander@335
   282
		return INVALID_GROUP_ID;
dvander@335
   283
	}
dvander@335
   284
dvander@335
   285
	return id;
dvander@335
   286
}
dvander@335
   287
dvander@335
   288
void AdminCache::SetGroupAddFlag(GroupId id, AdminFlag flag, bool enabled)
dvander@335
   289
{
dvander@335
   290
	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335
   291
	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335
   292
	{
dvander@335
   293
		return;
dvander@335
   294
	}
dvander@335
   295
dvander@433
   296
	if (flag < Admin_Reservation || flag >= AdminFlags_TOTAL)
dvander@335
   297
	{
dvander@335
   298
		return;
dvander@335
   299
	}
dvander@335
   300
dvander@436
   301
	FlagBits bits = (1<<(FlagBits)flag);
dvander@436
   302
dvander@436
   303
	if (enabled)
dvander@436
   304
	{
dvander@436
   305
		pGroup->addflags |= bits;
dvander@436
   306
	} else {
dvander@436
   307
		pGroup->addflags &= ~bits;
dvander@436
   308
	}
dvander@335
   309
}
dvander@335
   310
dvander@335
   311
bool AdminCache::GetGroupAddFlag(GroupId id, AdminFlag flag)
dvander@335
   312
{
dvander@335
   313
	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335
   314
	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335
   315
	{
dvander@335
   316
		return false;
dvander@335
   317
	}
dvander@335
   318
dvander@433
   319
	if (flag < Admin_Reservation || flag >= AdminFlags_TOTAL)
dvander@335
   320
	{
dvander@335
   321
		return false;
dvander@335
   322
	}
dvander@335
   323
dvander@433
   324
	FlagBits bit = 1<<(FlagBits)flag;
dvander@433
   325
	return ((pGroup->addflags & bit) == bit);
dvander@335
   326
}
dvander@335
   327
dvander@433
   328
FlagBits AdminCache::GetGroupAddFlags(GroupId id)
dvander@335
   329
{
dvander@335
   330
	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335
   331
	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335
   332
	{
dvander@335
   333
		return 0;
dvander@335
   334
	}
dvander@335
   335
dvander@433
   336
	return pGroup->addflags;
dvander@335
   337
}
dvander@335
   338
dvander@335
   339
void AdminCache::SetGroupGenericImmunity(GroupId id, ImmunityType type, bool enabled)
dvander@335
   340
{
dvander@335
   341
	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335
   342
	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335
   343
	{
dvander@335
   344
		return;
dvander@335
   345
	}
dvander@335
   346
dvander@335
   347
	if (type == Immunity_Default)
dvander@335
   348
	{
dvander@335
   349
		pGroup->immune_default = enabled;
dvander@335
   350
	} else if (type == Immunity_Global) {
dvander@335
   351
		pGroup->immune_global = enabled;
dvander@335
   352
	} 
dvander@335
   353
}
dvander@335
   354
dvander@335
   355
bool AdminCache::GetGroupGenericImmunity(GroupId id, ImmunityType type)
dvander@335
   356
{
dvander@335
   357
	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335
   358
	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335
   359
	{
dvander@335
   360
		return false;
dvander@335
   361
	}
dvander@335
   362
dvander@335
   363
	if (type == Immunity_Default)
dvander@335
   364
	{
dvander@335
   365
		return pGroup->immune_default;
dvander@335
   366
	} else if (type == Immunity_Global) {
dvander@335
   367
		return pGroup->immune_global;
dvander@335
   368
	}
dvander@335
   369
dvander@335
   370
	return false;
dvander@335
   371
}
dvander@335
   372
dvander@335
   373
void AdminCache::AddGroupImmunity(GroupId id, GroupId other_id)
dvander@335
   374
{
dvander@335
   375
	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335
   376
	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335
   377
	{
dvander@335
   378
		return;
dvander@335
   379
	}
dvander@335
   380
dvander@335
   381
	AdminGroup *pOther = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335
   382
	if (!pOther || pOther->magic != GRP_MAGIC_SET)
dvander@335
   383
	{
dvander@335
   384
		return;
dvander@335
   385
	}
dvander@335
   386
dvander@335
   387
	/* We always need to resize the immunity table */
dvander@335
   388
	int *table, tblidx;
dvander@335
   389
	if (pOther->immune_table == -1)
dvander@335
   390
	{
dvander@335
   391
		tblidx = m_pMemory->CreateMem(sizeof(int) * 2, (void **)&table);
dvander@335
   392
		table[0] = 0;
dvander@335
   393
	} else {
dvander@335
   394
		int *old_table = (int *)m_pMemory->GetAddress(pOther->immune_table);
dvander@433
   395
		/* Break out if this group is already in the list */
dvander@433
   396
		for (int i=0; i<old_table[0]; i++)
dvander@433
   397
		{
dvander@433
   398
			if (old_table[1+i] == other_id)
dvander@433
   399
			{
dvander@433
   400
				return;
dvander@433
   401
			}
dvander@433
   402
		}
dvander@335
   403
		tblidx = m_pMemory->CreateMem(sizeof(int) * (old_table[0] + 2), (void **)&table);
dvander@433
   404
		/* Get the old address again in case of resize */
dvander@335
   405
		old_table = (int *)m_pMemory->GetAddress(pOther->immune_table);
dvander@335
   406
		table[0] = old_table[0];
dvander@335
   407
		for (unsigned int i=1; i<=(unsigned int)old_table[0]; i++)
dvander@335
   408
		{
dvander@335
   409
			table[i] = old_table[i];
dvander@335
   410
		}
dvander@335
   411
	}
dvander@335
   412
dvander@335
   413
	/* Assign */
dvander@335
   414
	pOther->immune_table = tblidx;
dvander@335
   415
dvander@335
   416
	/* Add to the array */
dvander@335
   417
	table[0]++;
dvander@335
   418
	table[table[0]] = other_id;
dvander@335
   419
}
dvander@335
   420
dvander@335
   421
unsigned int AdminCache::GetGroupImmunityCount(GroupId id)
dvander@335
   422
{
dvander@335
   423
	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335
   424
	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335
   425
	{
dvander@335
   426
		return 0;
dvander@335
   427
	}
dvander@335
   428
dvander@335
   429
	if (pGroup->immune_table == -1)
dvander@335
   430
	{
dvander@335
   431
		return 0;
dvander@335
   432
	}
dvander@335
   433
dvander@335
   434
	int *table = (int *)m_pMemory->GetAddress(pGroup->immune_table);
dvander@335
   435
dvander@335
   436
	return table[0];
dvander@335
   437
}
dvander@335
   438
dvander@335
   439
GroupId AdminCache::GetGroupImmunity(GroupId id, unsigned int number)
dvander@335
   440
{
dvander@335
   441
	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335
   442
	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335
   443
	{
dvander@335
   444
		return INVALID_GROUP_ID;
dvander@335
   445
	}
dvander@335
   446
dvander@335
   447
	if (pGroup->immune_table == -1)
dvander@335
   448
	{
dvander@335
   449
		return INVALID_GROUP_ID;
dvander@335
   450
	}
dvander@335
   451
dvander@335
   452
	int *table = (int *)m_pMemory->GetAddress(pGroup->immune_table);
dvander@335
   453
	if (number >= (unsigned int)table[0])
dvander@335
   454
	{
dvander@335
   455
		return INVALID_GROUP_ID;
dvander@335
   456
	}
dvander@335
   457
dvander@335
   458
	return table[1 + number];
dvander@335
   459
}
dvander@335
   460
dvander@335
   461
void AdminCache::AddGroupCommandOverride(GroupId id, const char *name, OverrideType type, OverrideRule rule)
dvander@335
   462
{
dvander@335
   463
	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335
   464
	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335
   465
	{
dvander@335
   466
		return;
dvander@335
   467
	}
dvander@335
   468
dvander@335
   469
	Trie *pTrie = NULL;
dvander@335
   470
	if (type == Override_Command)
dvander@335
   471
	{
dvander@335
   472
		if (pGroup->pCmdTable == NULL)
dvander@335
   473
		{
dvander@335
   474
			pGroup->pCmdTable = sm_trie_create();
dvander@335
   475
		}
dvander@335
   476
		pTrie = pGroup->pCmdTable;
dvander@335
   477
	} else if (type == Override_CommandGroup) {
dvander@335
   478
		if (pGroup->pCmdGrpTable == NULL)
dvander@335
   479
		{
dvander@335
   480
			pGroup->pCmdGrpTable = sm_trie_create();
dvander@335
   481
		}
dvander@335
   482
		pTrie = pGroup->pCmdGrpTable;
dvander@335
   483
	} else {
dvander@335
   484
		return;
dvander@335
   485
	}
dvander@335
   486
dvander@335
   487
	sm_trie_insert(pTrie, name, (void *)(int)rule);
dvander@335
   488
}
dvander@335
   489
dvander@335
   490
bool AdminCache::GetGroupCommandOverride(GroupId id, const char *name, OverrideType type, OverrideRule *pRule)
dvander@335
   491
{
dvander@335
   492
	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335
   493
	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335
   494
	{
dvander@335
   495
		return false;
dvander@335
   496
	}
dvander@335
   497
dvander@335
   498
	Trie *pTrie = NULL;
dvander@335
   499
	if (type == Override_Command)
dvander@335
   500
	{
dvander@335
   501
		if (pGroup->pCmdTable == NULL)
dvander@335
   502
		{
dvander@335
   503
			return false;
dvander@335
   504
		}
dvander@335
   505
		pTrie = pGroup->pCmdTable;
dvander@335
   506
	} else if (type == Override_CommandGroup) {
dvander@335
   507
		if (pGroup->pCmdGrpTable == NULL)
dvander@335
   508
		{
dvander@335
   509
			return false;
dvander@335
   510
		}
dvander@335
   511
		pTrie = pGroup->pCmdGrpTable;
dvander@335
   512
	} else {
dvander@335
   513
		return false;
dvander@335
   514
	}
dvander@335
   515
dvander@335
   516
	void *object;
dvander@335
   517
	if (!sm_trie_retrieve(pTrie, name, &object))
dvander@335
   518
	{
dvander@335
   519
		return false;
dvander@335
   520
	}
dvander@335
   521
dvander@335
   522
	if (pRule)
dvander@335
   523
	{
dvander@335
   524
		*pRule = (OverrideRule)(int)object;
dvander@335
   525
	}
dvander@335
   526
	
dvander@335
   527
	return true;
dvander@335
   528
}
dvander@335
   529
dvander@404
   530
Trie *AdminCache::GetMethodByIndex(unsigned int index)
dvander@404
   531
{
dvander@404
   532
	List<AuthMethod>::iterator iter;
dvander@404
   533
	for (iter=m_AuthMethods.begin();
dvander@404
   534
		 iter!=m_AuthMethods.end();
dvander@404
   535
		 iter++)
dvander@404
   536
	{
dvander@404
   537
		if (index-- == 0)
dvander@404
   538
		{
dvander@404
   539
			return (*iter).table;
dvander@404
   540
		}
dvander@404
   541
	}
dvander@404
   542
dvander@404
   543
	return NULL;
dvander@404
   544
}
dvander@404
   545
dvander@404
   546
bool AdminCache::InvalidateAdmin(AdminId id)
dvander@404
   547
{
dvander@404
   548
	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404
   549
	AdminUser *pOther;
dvander@404
   550
	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404
   551
	{
dvander@404
   552
		return false;
dvander@404
   553
	}
dvander@404
   554
dvander@520
   555
	if (!m_InvalidatingAdmins && !m_destroying)
dvander@458
   556
	{
dvander@458
   557
		g_Players.ClearAdminId(id);
dvander@458
   558
	}
dvander@458
   559
dvander@404
   560
	/* Unlink from the dbl link list */
dvander@404
   561
	if (id == m_FirstUser && id == m_LastUser)
dvander@404
   562
	{
dvander@404
   563
		m_FirstUser = INVALID_ADMIN_ID;
dvander@404
   564
		m_LastUser = INVALID_ADMIN_ID;
dvander@404
   565
	} else if (id == m_FirstUser) {
dvander@404
   566
		m_FirstUser = pUser->next_user;
dvander@404
   567
		pOther = (AdminUser *)m_pMemory->GetAddress(m_FirstUser);
dvander@404
   568
		pOther->prev_user = INVALID_ADMIN_ID;
dvander@404
   569
	} else if (id == m_LastUser) {
dvander@404
   570
		m_LastUser = pUser->prev_user;
dvander@404
   571
		pOther = (AdminUser *)m_pMemory->GetAddress(m_LastUser);
dvander@404
   572
		pOther->next_user = INVALID_ADMIN_ID;
dvander@404
   573
	} else {
dvander@404
   574
		pOther = (AdminUser *)m_pMemory->GetAddress(pUser->prev_user);
dvander@404
   575
		pOther->next_user = pUser->next_user;
dvander@404
   576
		pOther = (AdminUser *)m_pMemory->GetAddress(pUser->next_user);
dvander@404
   577
		pOther->prev_user = pUser->prev_user;
dvander@404
   578
	}
dvander@404
   579
dvander@404
   580
	/* Unlink from auth tables */
dvander@405
   581
	if (pUser->auth.identidx != -1)
dvander@404
   582
	{
dvander@405
   583
		Trie *pTrie = GetMethodByIndex(pUser->auth.index);
dvander@405
   584
		if (pTrie)
dvander@404
   585
		{
dvander@405
   586
			sm_trie_delete(pTrie, m_pStrings->GetString(pUser->auth.identidx));
dvander@404
   587
		}
dvander@404
   588
	}
dvander@404
   589
dvander@404
   590
	/* Clear table counts */
dvander@404
   591
	pUser->grp_count = 0;
dvander@405
   592
	
dvander@404
   593
	/* Link into free list */
dvander@404
   594
	pUser->magic = USR_MAGIC_UNSET;
dvander@404
   595
	pUser->next_user = m_FreeUserList;
dvander@404
   596
	m_FreeUserList = id;
dvander@404
   597
dvander@404
   598
	return true;
dvander@404
   599
}
dvander@404
   600
dvander@404
   601
dvander@335
   602
void AdminCache::InvalidateGroup(GroupId id)
dvander@335
   603
{
dvander@335
   604
	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(id);
dvander@335
   605
	AdminGroup *pOther;
dvander@335
   606
dvander@335
   607
	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@335
   608
	{
dvander@335
   609
		return;
dvander@335
   610
	}
dvander@335
   611
dvander@335
   612
	const char *str = m_pStrings->GetString(pGroup->nameidx);
dvander@335
   613
	sm_trie_delete(m_pGroups, str);
dvander@335
   614
dvander@335
   615
	/* Unlink from the live dbllink list */
dvander@335
   616
	if (id == m_FirstGroup && id == m_LastGroup)
dvander@335
   617
	{
dvander@335
   618
		m_LastGroup = INVALID_GROUP_ID;
dvander@335
   619
		m_FirstGroup = INVALID_GROUP_ID;
dvander@335
   620
	} else if (id == m_FirstGroup) {
dvander@335
   621
		m_FirstGroup = pGroup->next_grp;
dvander@335
   622
		pOther = (AdminGroup *)m_pMemory->GetAddress(m_FirstGroup);
dvander@335
   623
		pOther->prev_grp = INVALID_GROUP_ID;
dvander@335
   624
	} else if (id == m_LastGroup) {
dvander@335
   625
		m_LastGroup = pGroup->prev_grp;
dvander@335
   626
		pOther = (AdminGroup *)m_pMemory->GetAddress(m_LastGroup);
dvander@335
   627
		pOther->next_grp = INVALID_GROUP_ID;
dvander@335
   628
	} else {
dvander@335
   629
		pOther = (AdminGroup *)m_pMemory->GetAddress(pGroup->prev_grp);
dvander@335
   630
		pOther->next_grp = pGroup->next_grp;
dvander@335
   631
		pOther = (AdminGroup *)m_pMemory->GetAddress(pGroup->next_grp);
dvander@335
   632
		pOther->prev_grp = pGroup->prev_grp;
dvander@335
   633
	}
dvander@335
   634
dvander@335
   635
	/* Free any used memory to be safe */
dvander@335
   636
	if (pGroup->pCmdGrpTable)
dvander@335
   637
	{
dvander@335
   638
		sm_trie_destroy(pGroup->pCmdGrpTable);
dvander@335
   639
		pGroup->pCmdGrpTable = NULL;
dvander@335
   640
	}
dvander@335
   641
	if (pGroup->pCmdTable)
dvander@335
   642
	{
dvander@335
   643
		sm_trie_destroy(pGroup->pCmdTable);
dvander@335
   644
		pGroup->pCmdTable = NULL;
dvander@335
   645
	}
dvander@335
   646
dvander@335
   647
	/* Link into the free list */
dvander@335
   648
	pGroup->magic = GRP_MAGIC_UNSET;
dvander@335
   649
	pGroup->next_grp = m_FreeGroupList;
dvander@335
   650
	m_FreeGroupList = id;
dvander@335
   651
dvander@404
   652
	int idx = m_FirstUser;
dvander@404
   653
	AdminUser *pUser;
dvander@404
   654
	int *table;
dvander@404
   655
	while (idx != INVALID_ADMIN_ID)
dvander@404
   656
	{
dvander@404
   657
		pUser = (AdminUser *)m_pMemory->GetAddress(idx);
dvander@404
   658
		if (pUser->grp_count > 0)
dvander@404
   659
		{
dvander@404
   660
			table = (int *)m_pMemory->GetAddress(pUser->grp_table);
dvander@404
   661
			for (unsigned int i=0; i<pUser->grp_count; i++)
dvander@404
   662
			{
dvander@404
   663
				if (table[i] == id)
dvander@404
   664
				{
dvander@404
   665
					/* We have to remove this entry */
dvander@404
   666
					for (unsigned int j=i+1; j<pUser->grp_count; j++)
dvander@404
   667
					{
dvander@404
   668
						/* Move everything down by one */
dvander@404
   669
						table[j-1] = table[j];
dvander@404
   670
					}
dvander@404
   671
					/* Decrease count */
dvander@404
   672
					pUser->grp_count--;
dvander@404
   673
					/* Recalculate effective flags */
dvander@433
   674
					pUser->eflags = pUser->flags;
dvander@404
   675
					for (unsigned int j=0; j<pUser->grp_count; j++)
dvander@404
   676
					{
dvander@404
   677
						pOther = (AdminGroup *)m_pMemory->GetAddress(table[j]);
dvander@433
   678
						pUser->eflags |= pOther->addflags;
dvander@404
   679
					}
dvander@404
   680
					/* Break now, duplicates aren't allowed */
dvander@404
   681
					break;
dvander@404
   682
				}
dvander@404
   683
			}
dvander@404
   684
		}
dvander@404
   685
		idx = pUser->next_user;
dvander@404
   686
	}
dvander@335
   687
}
dvander@335
   688
dvander@335
   689
void AdminCache::InvalidateGroupCache()
dvander@335
   690
{
dvander@335
   691
	/* Nuke the free list */
dvander@335
   692
	m_FreeGroupList = -1;
dvander@335
   693
dvander@335
   694
	/* Nuke reverse lookups */
dvander@335
   695
	sm_trie_clear(m_pGroups);
dvander@335
   696
dvander@335
   697
	/* Free memory on groups */
dvander@335
   698
	GroupId cur = m_FirstGroup;
dvander@335
   699
	AdminGroup *pGroup;
dvander@335
   700
	while (cur != INVALID_GROUP_ID)
dvander@335
   701
	{
dvander@335
   702
		pGroup = (AdminGroup *)m_pMemory->GetAddress(cur);
dvander@335
   703
		assert(pGroup->magic == GRP_MAGIC_SET);
dvander@335
   704
		if (pGroup->pCmdGrpTable)
dvander@335
   705
		{
dvander@335
   706
			sm_trie_destroy(pGroup->pCmdGrpTable);
dvander@335
   707
		}
dvander@335
   708
		if (pGroup->pCmdTable)
dvander@335
   709
		{
dvander@335
   710
			sm_trie_destroy(pGroup->pCmdTable);
dvander@335
   711
		}
dvander@335
   712
		cur = pGroup->next_grp;
dvander@335
   713
	}
dvander@335
   714
dvander@335
   715
	m_FirstGroup = INVALID_GROUP_ID;
dvander@335
   716
	m_LastGroup = INVALID_GROUP_ID;
dvander@335
   717
dvander@404
   718
	InvalidateAdminCache(false);
dvander@335
   719
dvander@335
   720
	/* Reset the memory table */
dvander@335
   721
	m_pMemory->Reset();
dvander@335
   722
}
dvander@336
   723
dvander@336
   724
void AdminCache::AddAdminListener(IAdminListener *pListener)
dvander@336
   725
{
dvander@336
   726
	m_hooks.push_back(pListener);
dvander@336
   727
}
dvander@336
   728
dvander@336
   729
void AdminCache::RemoveAdminListener(IAdminListener *pListener)
dvander@336
   730
{
dvander@336
   731
	m_hooks.remove(pListener);
dvander@336
   732
}
dvander@336
   733
dvander@404
   734
void AdminCache::RegisterAuthIdentType(const char *name)
dvander@404
   735
{
dvander@404
   736
	if (sm_trie_retrieve(m_pAuthTables, name, NULL))
dvander@404
   737
	{
dvander@404
   738
		return;
dvander@404
   739
	}
dvander@404
   740
dvander@404
   741
	Trie *pAuth = sm_trie_create();
dvander@404
   742
dvander@404
   743
	AuthMethod method;
dvander@404
   744
	method.name.assign(name);
dvander@404
   745
	method.table = pAuth;
dvander@404
   746
dvander@404
   747
	m_AuthMethods.push_back(method);
dvander@404
   748
dvander@404
   749
	sm_trie_insert(m_pAuthTables, name, pAuth);
dvander@404
   750
}
dvander@404
   751
dvander@404
   752
void AdminCache::InvalidateAdminCache(bool unlink_admins)
dvander@404
   753
{
dvander@458
   754
	m_InvalidatingAdmins = true;
dvander@520
   755
	if (!m_destroying)
dvander@520
   756
	{
dvander@520
   757
		g_Players.ClearAllAdmins();
dvander@520
   758
	}
dvander@404
   759
	/* Wipe the identity cache first */
dvander@404
   760
	List<AuthMethod>::iterator iter;
dvander@404
   761
	for (iter=m_AuthMethods.begin();
dvander@404
   762
		 iter!=m_AuthMethods.end();
dvander@404
   763
		 iter++)
dvander@404
   764
	{
dvander@404
   765
		sm_trie_clear((*iter).table);
dvander@404
   766
	}
dvander@404
   767
	
dvander@405
   768
	if (unlink_admins)
dvander@404
   769
	{
dvander@405
   770
		while (m_FirstUser != INVALID_ADMIN_ID)
dvander@405
   771
		{
dvander@405
   772
			InvalidateAdmin(m_FirstUser);
dvander@405
   773
		}
dvander@405
   774
	} else {
dvander@405
   775
		m_FirstUser = -1;
dvander@405
   776
		m_LastUser = -1;
dvander@405
   777
		m_FreeUserList = -1;
dvander@404
   778
	}
dvander@458
   779
	m_InvalidatingAdmins = false;
dvander@404
   780
}
dvander@404
   781
dvander@438
   782
void AdminCache::DumpAdminCache(AdminCachePart part, bool rebuild)
dvander@336
   783
{
dvander@438
   784
	List<IAdminListener *>::iterator iter;
dvander@438
   785
	IAdminListener *pListener;
dvander@438
   786
	cell_t result;
dvander@438
   787
dvander@438
   788
	if (part == AdminCache_Overrides)
dvander@336
   789
	{
dvander@336
   790
		DumpCommandOverrideCache(Override_Command);
dvander@336
   791
		DumpCommandOverrideCache(Override_CommandGroup);
dvander@520
   792
		if (rebuild && !m_destroying)
dvander@438
   793
		{
dvander@438
   794
			for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
dvander@438
   795
			{
dvander@438
   796
				pListener = (*iter);
dvander@438
   797
				pListener->OnRebuildOverrideCache();
dvander@438
   798
			}
dvander@438
   799
			m_pCacheFwd->PushCell(part);
dvander@438
   800
			m_pCacheFwd->Execute(&result);
dvander@438
   801
		}
dvander@438
   802
	} else if (part == AdminCache_Groups || part == AdminCache_Admins) {
dvander@438
   803
		if (part == AdminCache_Groups)
dvander@438
   804
		{
dvander@438
   805
			InvalidateGroupCache();
dvander@520
   806
			if (rebuild && !m_destroying)
dvander@438
   807
			{
dvander@438
   808
				for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
dvander@438
   809
				{
dvander@438
   810
					pListener = (*iter);
dvander@438
   811
					pListener->OnRebuildGroupCache();
dvander@438
   812
				}
dvander@438
   813
				m_pCacheFwd->PushCell(part);
dvander@438
   814
				m_pCacheFwd->Execute(&result);
dvander@438
   815
			}
dvander@438
   816
		}
dvander@404
   817
		InvalidateAdminCache(true);
dvander@520
   818
		if (rebuild && !m_destroying)
dvander@336
   819
		{
dvander@438
   820
			for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
dvander@438
   821
			{
dvander@438
   822
				pListener = (*iter);
dvander@438
   823
				pListener->OnRebuildAdminCache((part == AdminCache_Groups));
dvander@438
   824
			}
dvander@438
   825
			m_pCacheFwd->PushCell(AdminCache_Admins);
dvander@438
   826
			m_pCacheFwd->Execute(&result);
dvander@336
   827
		}
dvander@336
   828
	}
dvander@336
   829
}
dvander@404
   830
dvander@404
   831
const char *AdminCache::GetAdminName(AdminId id)
dvander@404
   832
{
dvander@404
   833
	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404
   834
	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404
   835
	{
dvander@404
   836
		return NULL;
dvander@404
   837
	}
dvander@404
   838
dvander@404
   839
	return m_pStrings->GetString(pUser->nameidx);
dvander@404
   840
}
dvander@404
   841
dvander@405
   842
bool AdminCache::GetMethodIndex(const char *name, unsigned int *_index)
dvander@405
   843
{
dvander@405
   844
	List<AuthMethod>::iterator iter;
dvander@405
   845
	unsigned int index = 0;
dvander@405
   846
	for (iter=m_AuthMethods.begin();
dvander@405
   847
		 iter!=m_AuthMethods.end();
dvander@405
   848
		 iter++,index++)
dvander@405
   849
	{
dvander@405
   850
		if ((*iter).name.compare(name) == 0)
dvander@405
   851
		{
dvander@405
   852
			*_index = index;
dvander@405
   853
			return true;
dvander@405
   854
		}
dvander@405
   855
	}
dvander@405
   856
dvander@405
   857
	return false;
dvander@405
   858
}
dvander@405
   859
dvander@404
   860
bool AdminCache::BindAdminIdentity(AdminId id, const char *auth, const char *ident)
dvander@404
   861
{
dvander@404
   862
	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404
   863
	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404
   864
	{
dvander@404
   865
		return false;
dvander@404
   866
	}
dvander@404
   867
dvander@404
   868
	Trie *pTable;
dvander@404
   869
	if (!sm_trie_retrieve(m_pAuthTables, auth, (void **)&pTable))
dvander@404
   870
	{
dvander@404
   871
		return false;
dvander@404
   872
	}
dvander@404
   873
dvander@404
   874
	if (sm_trie_retrieve(pTable, ident, NULL))
dvander@404
   875
	{
dvander@404
   876
		return false;
dvander@404
   877
	}
dvander@404
   878
dvander@405
   879
	pUser->auth.identidx = m_pStrings->AddString(ident);
dvander@405
   880
	GetMethodIndex(auth, &pUser->auth.index);
dvander@405
   881
dvander@404
   882
	return sm_trie_insert(pTable, ident, (void **)id);
dvander@404
   883
}
dvander@404
   884
dvander@404
   885
AdminId AdminCache::FindAdminByIdentity(const char *auth, const char *identity)
dvander@404
   886
{
dvander@404
   887
	Trie *pTable;
dvander@404
   888
	if (!sm_trie_retrieve(m_pAuthTables, auth, (void **)&pTable))
dvander@404
   889
	{
dvander@404
   890
		return INVALID_ADMIN_ID;
dvander@404
   891
	}
dvander@404
   892
dvander@404
   893
	void *object;
dvander@404
   894
	if (!sm_trie_retrieve(pTable, identity, &object))
dvander@404
   895
	{
dvander@404
   896
		return INVALID_ADMIN_ID;
dvander@404
   897
	}
dvander@404
   898
dvander@404
   899
	return (AdminId)object;
dvander@404
   900
}
dvander@404
   901
dvander@404
   902
void AdminCache::SetAdminFlag(AdminId id, AdminFlag flag, bool enabled)
dvander@404
   903
{
dvander@404
   904
	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404
   905
	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404
   906
	{
dvander@404
   907
		return;
dvander@404
   908
	}
dvander@404
   909
dvander@433
   910
	if (flag < Admin_Reservation
dvander@404
   911
		|| flag >= AdminFlags_TOTAL)
dvander@404
   912
	{
dvander@404
   913
		return;
dvander@404
   914
	}
dvander@436
   915
	
dvander@436
   916
	FlagBits bits = (1<<(FlagBits)flag);
dvander@404
   917
dvander@436
   918
	if (enabled)
dvander@436
   919
	{
dvander@436
   920
		pUser->flags |= bits;
dvander@436
   921
		pUser->eflags |= bits;
dvander@436
   922
	} else {
dvander@436
   923
		pUser->flags &= ~bits;
dvander@436
   924
		pUser->eflags &= ~bits;
dvander@436
   925
	}
dvander@404
   926
}
dvander@404
   927
dvander@404
   928
bool AdminCache::GetAdminFlag(AdminId id, AdminFlag flag, AccessMode mode)
dvander@404
   929
{
dvander@404
   930
	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404
   931
	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404
   932
	{
dvander@404
   933
		return false;
dvander@404
   934
	}
dvander@404
   935
dvander@433
   936
	if (flag < Admin_Reservation
dvander@404
   937
		|| flag >= AdminFlags_TOTAL)
dvander@404
   938
	{
dvander@404
   939
		return false;
dvander@404
   940
	}
dvander@404
   941
dvander@433
   942
	FlagBits bit = (1<<(FlagBits)flag);
dvander@433
   943
dvander@404
   944
	if (mode == Access_Real)
dvander@404
   945
	{
dvander@433
   946
		return ((pUser->flags & bit) == bit);
dvander@404
   947
	} else if (mode == Access_Effective) {
dvander@433
   948
		return ((pUser->eflags & bit) == bit);
dvander@404
   949
	}
dvander@404
   950
dvander@404
   951
	return false;
dvander@404
   952
}
dvander@404
   953
dvander@433
   954
FlagBits AdminCache::GetAdminFlags(AdminId id, AccessMode mode)
dvander@404
   955
{
dvander@404
   956
	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404
   957
	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404
   958
	{
dvander@404
   959
		return 0;
dvander@404
   960
	}
dvander@404
   961
dvander@404
   962
	if (mode == Access_Real)
dvander@404
   963
	{
dvander@433
   964
		return pUser->flags;
dvander@404
   965
	} else if (mode == Access_Effective) {
dvander@433
   966
		return pUser->eflags;
dvander@404
   967
	}
dvander@404
   968
dvander@433
   969
	return 0;
dvander@404
   970
}
dvander@404
   971
dvander@458
   972
void AdminCache::SetAdminFlags(AdminId id, AccessMode mode, FlagBits bits)
dvander@458
   973
{
dvander@458
   974
	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@458
   975
	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@458
   976
	{
dvander@458
   977
		return;
dvander@458
   978
	}
dvander@458
   979
dvander@458
   980
	if (mode == Access_Real)
dvander@458
   981
	{
dvander@458
   982
		pUser->flags = bits;
dvander@458
   983
		pUser->eflags = bits;
dvander@458
   984
	} else if (mode == Access_Effective) {
dvander@458
   985
		pUser->eflags = bits;
dvander@458
   986
	}
dvander@458
   987
}
dvander@458
   988
dvander@404
   989
bool AdminCache::AdminInheritGroup(AdminId id, GroupId gid)
dvander@404
   990
{
dvander@404
   991
	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404
   992
	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404
   993
	{
dvander@404
   994
		return false;
dvander@404
   995
	}
dvander@404
   996
dvander@404
   997
	AdminGroup *pGroup = (AdminGroup *)m_pMemory->GetAddress(gid);
dvander@405
   998
	if (!pGroup || pGroup->magic != GRP_MAGIC_SET)
dvander@404
   999
	{
dvander@404
  1000
		return false;
dvander@404
  1001
	}
dvander@404
  1002
dvander@433
  1003
	/* First check for duplicates */
dvander@433
  1004
	if (pUser->grp_count != 0)
dvander@433
  1005
	{
dvander@433
  1006
		int *temp_table = (int *)m_pMemory->GetAddress(pUser->grp_table);
dvander@433
  1007
		for (unsigned int i=0; i<pUser->grp_count; i++)
dvander@433
  1008
		{
dvander@433
  1009
			if (temp_table[i] == gid)
dvander@433
  1010
			{
dvander@433
  1011
				return false;
dvander@433
  1012
			}
dvander@433
  1013
		}
dvander@433
  1014
	}
dvander@433
  1015
dvander@404
  1016
	int *table;
dvander@404
  1017
	if (pUser->grp_count + 1 > pUser->grp_size)
dvander@404
  1018
	{
dvander@404
  1019
		int new_size = 0;
dvander@404
  1020
		int tblidx;
dvander@404
  1021
		
dvander@404
  1022
		if (pUser->grp_size == 0)
dvander@404
  1023
		{
dvander@404
  1024
			new_size = 2;
dvander@404
  1025
		} else {
dvander@404
  1026
			new_size = pUser->grp_size * 2;
dvander@404
  1027
		}
dvander@404
  1028
		
dvander@404
  1029
		/* Create and refresh pointers */
dvander@404
  1030
		tblidx = m_pMemory->CreateMem(new_size * sizeof(int), (void **)&table);
dvander@404
  1031
		pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404
  1032
		pGroup = (AdminGroup *)m_pMemory->GetAddress(gid);
dvander@404
  1033
dvander@404
  1034
		/* Copy old data if necessary */
dvander@404
  1035
		if (pUser->grp_table != -1)
dvander@404
  1036
		{
dvander@404
  1037
			int *old_table = (int *)m_pMemory->GetAddress(pUser->grp_table);
dvander@404
  1038
			memcpy(table, old_table, sizeof(int) * pUser->grp_count);
dvander@404
  1039
		}
dvander@404
  1040
		pUser->grp_table = tblidx;
dvander@404
  1041
		pUser->grp_size = new_size;
dvander@404
  1042
	} else {
dvander@404
  1043
		table = (int *)m_pMemory->GetAddress(pUser->grp_table);
dvander@404
  1044
	}
dvander@404
  1045
dvander@404
  1046
	table[pUser->grp_count] = gid;
dvander@404
  1047
	pUser->grp_count++;
dvander@404
  1048
dvander@434
  1049
	/* Compute new effective permissions */
dvander@433
  1050
	pUser->eflags |= pGroup->addflags;
dvander@404
  1051
dvander@434
  1052
	if (pGroup->immune_default)
dvander@434
  1053
	{
dvander@434
  1054
		pUser->immune_default = true;
dvander@434
  1055
	}
dvander@434
  1056
	if (pGroup->immune_global)
dvander@434
  1057
	{
dvander@434
  1058
		pUser->immune_global = true;
dvander@434
  1059
	}
dvander@434
  1060
dvander@404
  1061
	return true;
dvander@404
  1062
}
dvander@404
  1063
dvander@458
  1064
bool AdminCache::IsValidAdmin(AdminId id)
dvander@458
  1065
{
dvander@458
  1066
	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@458
  1067
	return (pUser != NULL && pUser->magic == USR_MAGIC_SET);
dvander@458
  1068
}
dvander@458
  1069
dvander@404
  1070
unsigned int AdminCache::GetAdminGroupCount(AdminId id)
dvander@404
  1071
{
dvander@404
  1072
	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404
  1073
	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404
  1074
	{
dvander@404
  1075
		return 0;
dvander@404
  1076
	}
dvander@404
  1077
dvander@404
  1078
	return pUser->grp_count;
dvander@404
  1079
}
dvander@404
  1080
dvander@404
  1081
GroupId AdminCache::GetAdminGroup(AdminId id, unsigned int index, const char **name)
dvander@404
  1082
{
dvander@404
  1083
	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404
  1084
	if (!pUser || pUser->magic != USR_MAGIC_SET || index >= pUser->grp_count)
dvander@404
  1085
	{
dvander@404
  1086
		return INVALID_GROUP_ID;
dvander@404
  1087
	}
dvander@404
  1088
dvander@404
  1089
	int *table = (int *)m_pMemory->GetAddress(pUser->grp_table);
dvander@404
  1090
dvander@404
  1091
	return table[index];
dvander@404
  1092
}
dvander@404
  1093
dvander@404
  1094
const char *AdminCache::GetAdminPassword(AdminId id)
dvander@404
  1095
{
dvander@404
  1096
	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404
  1097
	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404
  1098
	{
dvander@404
  1099
		return NULL;
dvander@404
  1100
	}
dvander@404
  1101
dvander@404
  1102
	return m_pStrings->GetString(pUser->password);
dvander@404
  1103
}
dvander@404
  1104
dvander@404
  1105
void AdminCache::SetAdminPassword(AdminId id, const char *password)
dvander@404
  1106
{	
dvander@404
  1107
	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@404
  1108
	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@404
  1109
	{
dvander@404
  1110
		return;
dvander@404
  1111
	}
dvander@404
  1112
dvander@404
  1113
	pUser->password = m_pStrings->AddString(password);
dvander@404
  1114
}
dvander@404
  1115
dvander@433
  1116
unsigned int AdminCache::FlagBitsToBitArray(FlagBits bits, bool array[], unsigned int maxSize)
dvander@433
  1117
{
dvander@433
  1118
	unsigned int i;
dvander@433
  1119
	for (i=0; i<maxSize && i<AdminFlags_TOTAL; i++)
dvander@433
  1120
	{
dvander@464
  1121
		array[i] = ((bits & (1<<i)) == (unsigned)(1<<i));
dvander@433
  1122
	}
dvander@433
  1123
dvander@433
  1124
	return i;
dvander@433
  1125
}
dvander@433
  1126
dvander@433
  1127
FlagBits AdminCache::FlagBitArrayToBits(const bool array[], unsigned int maxSize)
dvander@433
  1128
{
dvander@433
  1129
	FlagBits bits = 0;
dvander@433
  1130
	for (unsigned int i=0; i<maxSize && i<AdminFlags_TOTAL; i++)
dvander@433
  1131
	{
dvander@433
  1132
		if (array[i])
dvander@433
  1133
		{
dvander@433
  1134
			bits |= (1<<i);
dvander@433
  1135
		}
dvander@433
  1136
	}
dvander@433
  1137
dvander@433
  1138
	return bits;
dvander@433
  1139
}
dvander@433
  1140
dvander@433
  1141
FlagBits AdminCache::FlagArrayToBits(const AdminFlag array[], unsigned int numFlags)
dvander@433
  1142
{
dvander@433
  1143
	FlagBits bits = 0;
dvander@433
  1144
	for (unsigned int i=0; i<numFlags && i<AdminFlags_TOTAL; i++)
dvander@433
  1145
	{
dvander@433
  1146
		bits |= (1 << (FlagBits)array[i]);
dvander@433
  1147
	}
dvander@433
  1148
dvander@433
  1149
	return bits;
dvander@433
  1150
}
dvander@433
  1151
dvander@433
  1152
unsigned int AdminCache::FlagBitsToArray(FlagBits bits, AdminFlag array[], unsigned int maxSize)
dvander@433
  1153
{
dvander@433
  1154
	unsigned int i, num=0;
dvander@433
  1155
	for (i=0; i<maxSize && i<AdminFlags_TOTAL; i++)
dvander@433
  1156
	{
dvander@464
  1157
		if ((bits & (1<<i)) == (unsigned)(1<<i))
dvander@433
  1158
		{
dvander@433
  1159
			array[num++] = (AdminFlag)i;
dvander@433
  1160
		}
dvander@433
  1161
	}
dvander@433
  1162
dvander@433
  1163
	return num;
dvander@433
  1164
}
dvander@434
  1165
dvander@434
  1166
bool AdminCache::CheckAdminFlags(AdminId id, FlagBits bits)
dvander@434
  1167
{
dvander@434
  1168
	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@434
  1169
	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@434
  1170
	{
dvander@434
  1171
		return false;
dvander@434
  1172
	}
dvander@434
  1173
dvander@434
  1174
	return ((pUser->eflags & bits) == bits);
dvander@434
  1175
}
dvander@434
  1176
dvander@434
  1177
bool AdminCache::CanAdminTarget(AdminId id, AdminId target)
dvander@434
  1178
{
dvander@434
  1179
	/** 
dvander@434
  1180
	 * Zeroth, if the targeting AdminId is INVALID_ADMIN_ID, targeting fails.
dvander@434
  1181
	 * First, if the targetted AdminId is INVALID_ADMIN_ID, targeting succeeds.
dvander@434
  1182
	 */
dvander@434
  1183
dvander@434
  1184
	if (id == INVALID_ADMIN_ID)
dvander@434
  1185
	{
dvander@434
  1186
		return false;
dvander@434
  1187
	}
dvander@434
  1188
dvander@434
  1189
	if (target == INVALID_ADMIN_ID)
dvander@434
  1190
	{
dvander@434
  1191
		return true;
dvander@434
  1192
	}
dvander@434
  1193
dvander@434
  1194
	AdminUser *pUser = (AdminUser *)m_pMemory->GetAddress(id);
dvander@434
  1195
	if (!pUser || pUser->magic != USR_MAGIC_SET)
dvander@434
  1196
	{
dvander@434
  1197
		return false;
dvander@434
  1198
	}
dvander@434
  1199
dvander@434
  1200
	AdminUser *pTarget = (AdminUser *)m_pMemory->GetAddress(target);
dvander@434
  1201
	if (!pTarget || pTarget->magic != USR_MAGIC_SET)
dvander@434
  1202
	{
dvander@434
  1203
		return false;
dvander@434
  1204
	}
dvander@434
  1205
dvander@434
  1206
	/** 
dvander@434
  1207
	 * Second, if the targeting admin is root, targeting suceeds.
dvander@434
  1208
	 */
dvander@434
  1209
	if (pUser->eflags & ADMFLAG_ROOT)
dvander@434
  1210
	{
dvander@434
  1211
		return true;
dvander@434
  1212
	}
dvander@434
  1213
dvander@434
  1214
	/** Fourth, if the targetted admin has global immunity, targeting fails. */
dvander@434
  1215
	if (pTarget->immune_global)
dvander@434
  1216
	{
dvander@434
  1217
		return false;
dvander@434
  1218
	}
dvander@434
  1219
dvander@434
  1220
	/** 
dvander@434
  1221
	 * Fifth, if the targetted admin has default immunity 
dvander@434
  1222
	 *  and the admin belongs to no groups, targeting fails.
dvander@434
  1223
	 */
dvander@434
  1224
	if (pTarget->immune_default && pUser->grp_count < 1)
dvander@434
  1225
	{
dvander@434
  1226
		return false;
dvander@434
  1227
	}
dvander@434
  1228
dvander@434
  1229
	/** 
dvander@434
  1230
	 * Sixth, if the targetted admin has specific immunity from the
dvander@434
  1231
	 *  targeting admin via group immunities, targeting fails.
dvander@434
  1232
	 */
dvander@434
  1233
	//:TODO: speed this up... maybe with trie hacks.
dvander@434
  1234
	//idea is to insert %d.%d in the trie after computing this and use it as a cache lookup.
dvander@434
  1235
	//problem is the trie cannot delete prefixes, so we'd have a problem with invalidations.
dvander@434
  1236
	if (pTarget->grp_count > 0 && pUser->grp_count > 0)
dvander@434
  1237
	{
dvander@434
  1238
		int *grp_table = (int *)m_pMemory->GetAddress(pTarget->grp_table);
dvander@434
  1239
		int *src_table = (int *)m_pMemory->GetAddress(pUser->grp_table);
dvander@434
  1240
		GroupId id, other;
dvander@434
  1241
		unsigned int num;
dvander@434
  1242
		for (unsigned int i=0; i<pTarget->grp_count; i++)
dvander@434
  1243
		{
dvander@434
  1244
			id = grp_table[i];
dvander@434
  1245
			num = GetGroupImmunityCount(id);
dvander@434
  1246
			for (unsigned int j=0; j<num; i++)
dvander@434
  1247
			{
dvander@434
  1248
				other = GetGroupImmunity(id, j);
dvander@434
  1249
				for (unsigned int k=0; k<pUser->grp_count; k++)
dvander@434
  1250
				{
dvander@434
  1251
					if (other == src_table[k])
dvander@434
  1252
					{
dvander@434
  1253
						return false;
dvander@434
  1254
					}
dvander@434
  1255
				}
dvander@434
  1256
			}
dvander@434
  1257
		}
dvander@434
  1258
	}
dvander@434
  1259
	
dvander@434
  1260
	return true;
dvander@434
  1261
}