Removed pointless "systems" folder from core.
3 * =============================================================================
5 * Copyright (C) 2004-2008 AlliedModders LLC. All rights reserved.
6 * =============================================================================
8 * This program is free software; you can redistribute it and/or modify it under
9 * the terms of the GNU General Public License, version 3.0, as published by the
10 * Free Software Foundation.
12 * This program is distributed in the hope that it will be useful, but WITHOUT
13 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
14 * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
17 * You should have received a copy of the GNU General Public License along with
18 * this program. If not, see <http://www.gnu.org/licenses/>.
20 * As a special exception, AlliedModders LLC gives you permission to link the
21 * code of this program (as well as its derivative works) to "Half-Life 2," the
22 * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
23 * by the Valve Corporation. You must obey the GNU General Public License in
24 * all respects for all other code used. Additionally, AlliedModders LLC grants
25 * this exception to all derivative works. AlliedModders LLC defines further
26 * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
27 * or <http://www.sourcemod.net/license.php>.
32 #include "ConCmdManager.h"
33 #include "sm_srvcmds.h"
34 #include "AdminCache.h"
35 #include "sm_stringutil.h"
36 #include "PlayerManager.h"
37 #include "Translator.h"
38 #include "HalfLife2.h"
39 #include "ChatTriggers.h"
41 ConCmdManager g_ConCmds;
43 #if defined ORANGEBOX_BUILD
44 SH_DECL_HOOK1_void(ConCommand, Dispatch, SH_NOATTRIB, false, const CCommand &);
46 SH_DECL_HOOK0_void(ConCommand, Dispatch, SH_NOATTRIB, false);
49 SH_DECL_HOOK1_void(IServerGameClients, SetCommandClient, SH_NOATTRIB, false, int);
57 typedef List<PlCmdInfo> CmdList;
59 void AddToPlCmdList(CmdList *pList, const PlCmdInfo &info);
61 ConCmdManager::ConCmdManager() : m_Strings(1024)
63 m_pCmds = sm_trie_create();
64 m_pCmdGrps = sm_trie_create();
68 ConCmdManager::~ConCmdManager()
70 sm_trie_destroy(m_pCmds);
71 sm_trie_destroy(m_pCmdGrps);
74 void ConCmdManager::OnSourceModAllInitialized()
76 g_PluginSys.AddPluginsListener(this);
77 g_RootMenu.AddRootConsoleCommand("cmds", "List console commands", this);
78 SH_ADD_HOOK_MEMFUNC(IServerGameClients, SetCommandClient, serverClients, this, &ConCmdManager::SetCommandClient, false);
81 void ConCmdManager::OnSourceModShutdown()
83 /* All commands should already be removed by the time we're done */
84 SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, SetCommandClient, serverClients, this, &ConCmdManager::SetCommandClient, false);
85 g_RootMenu.RemoveRootConsoleCommand("cmds", this);
88 void ConCmdManager::OnUnlinkConCommandBase(ConCommandBase *pBase, const char *name, bool is_read_safe)
90 /* Whoa, first get its information struct */
93 if (!sm_trie_retrieve(m_pCmds, name, (void **)&pInfo))
98 RemoveConCmds(pInfo->srvhooks);
99 RemoveConCmds(pInfo->conhooks);
101 RemoveConCmd(pInfo, name, is_read_safe, false);
104 void ConCmdManager::RemoveConCmds(List<CmdHook *> &cmdlist)
106 List<CmdHook *>::iterator iter = cmdlist.begin();
108 while (iter != cmdlist.end())
110 CmdHook *pHook = (*iter);
111 IPluginContext *pContext = pHook->pf->GetParentContext();
112 IPlugin *pPlugin = g_PluginSys.GetPluginByCtx(pContext->GetContext());
113 CmdList *pList = NULL;
116 if (!pPlugin->GetProperty("CommandList", (void **)&pList, false) || !pList)
121 CmdList::iterator p_iter = pList->begin();
122 while (p_iter != pList->end())
124 PlCmdInfo &cmd = (*p_iter);
125 if (cmd.pHook == pHook)
127 p_iter = pList->erase(p_iter);
135 delete pHook->pAdmin;
138 iter = cmdlist.erase(iter);
142 void ConCmdManager::RemoveConCmds(List<CmdHook *> &cmdlist, IPluginContext *pContext)
144 List<CmdHook *>::iterator iter = cmdlist.begin();
147 while (iter != cmdlist.end())
150 if (pHook->pf->GetParentContext() == pContext)
152 delete pHook->pAdmin;
154 iter = cmdlist.erase(iter);
163 void ConCmdManager::OnPluginDestroyed(IPlugin *plugin)
166 List<ConCmdInfo *> removed;
167 if (plugin->GetProperty("CommandList", (void **)&pList, true))
169 IPluginContext *pContext = plugin->GetBaseContext();
170 CmdList::iterator iter;
172 * Only bother if there's an actual command list on this plugin...
174 for (iter=pList->begin();
178 PlCmdInfo &cmd = (*iter);
179 ConCmdInfo *pInfo = cmd.pInfo;
181 /* Has this chain already been fully cleaned/removed? */
182 if (removed.find(pInfo) != removed.end())
187 /* Remove any hooks from us on this command */
188 RemoveConCmds(pInfo->conhooks, pContext);
189 RemoveConCmds(pInfo->srvhooks, pContext);
191 /* See if there are still hooks */
192 if (pInfo->srvhooks.size())
196 if (pInfo->conhooks.size())
201 /* Remove the command, it should be safe now */
202 RemoveConCmd(pInfo, pInfo->pCmd->GetName(), true, true);
203 removed.push_back(pInfo);
208 #if defined ORANGEBOX_BUILD
209 void CommandCallback(const CCommand &command)
212 void CommandCallback()
217 g_HL2.PushCommandStack(&command);
219 g_ConCmds.InternalDispatch(command);
221 g_HL2.PopCommandStack();
224 void ConCmdManager::SetCommandClient(int client)
226 m_CmdClient = client + 1;
229 ResultType ConCmdManager::DispatchClientCommand(int client, const char *cmd, int args, ResultType type)
232 if (sm_trie_retrieve(m_pCmds, cmd, (void **)&pInfo))
234 cell_t result = type;
235 cell_t tempres = result;
236 List<CmdHook *>::iterator iter;
238 for (iter=pInfo->conhooks.begin();
239 iter!=pInfo->conhooks.end();
243 if (!pHook->pf->IsRunnable())
247 if (pHook->pAdmin && !CheckAccess(client, cmd, pHook->pAdmin))
249 if (result < Pl_Handled)
255 pHook->pf->PushCell(client);
256 pHook->pf->PushCell(args);
257 if (pHook->pf->Execute(&tempres) == SP_ERROR_NONE)
259 if (tempres > result)
263 if (result == Pl_Stop)
269 type = (ResultType)result;
275 void ConCmdManager::InternalDispatch(const CCommand &command)
277 int client = m_CmdClient;
281 CPlayer *pPlayer = g_Players.GetPlayerByIndex(client);
282 if (!pPlayer || !pPlayer->IsConnected())
289 * Note: Console commands will EITHER go through IServerGameDLL::ClientCommand,
290 * OR this dispatch. They will NEVER go through both.
292 * Whether or not it goes through the callback is determined by FCVAR_GAMEDLL
294 const char *cmd = g_HL2.CurrentCommandName();
297 if (!sm_trie_retrieve(m_pCmds, cmd, (void **)&pInfo))
299 /* Unfortunately, we now have to do a slow lookup because Valve made client commands
300 * case-insensitive. We can't even use our sortedness.
302 if (client == 0 && !engine->IsDedicatedServer())
307 List<ConCmdInfo *>::iterator iter;
310 iter = m_CmdList.begin();
311 while (iter != m_CmdList.end())
313 if (strcasecmp((*iter)->pCmd->GetName(), cmd) == 0)
327 /* This is a hack to prevent say triggers from firing on messages that were
328 * blocked because of flooding. We won't remove this, but the hack will get
329 * "nicer" when we expose explicit say hooks.
331 if (g_ChatTriggers.WasFloodedMessage())
336 cell_t result = Pl_Continue;
337 int args = command.ArgC() - 1;
339 List<CmdHook *>::iterator iter;
342 /* Execute server-only commands if viable */
343 if (client == 0 && pInfo->srvhooks.size())
345 cell_t tempres = result;
346 for (iter=pInfo->srvhooks.begin();
347 iter!=pInfo->srvhooks.end();
351 if (!pHook->pf->IsRunnable())
355 pHook->pf->PushCell(args);
356 if (pHook->pf->Execute(&tempres) == SP_ERROR_NONE)
358 if (tempres > result)
362 if (result == Pl_Stop)
369 /* Check if there's an early stop */
370 if (result >= Pl_Stop)
372 if (!pInfo->sourceMod)
374 RETURN_META(MRES_SUPERCEDE);
380 /* Execute console commands */
381 if (pInfo->conhooks.size())
383 cell_t tempres = result;
384 for (iter=pInfo->conhooks.begin();
385 iter!=pInfo->conhooks.end();
389 if (!pHook->pf->IsRunnable())
395 && !CheckAccess(client, cmd, pHook->pAdmin))
397 if (result < Pl_Handled)
404 /* On a listen server, sometimes the server host's client index can be set as 0.
405 * So index 1 is passed to the command callback to correct this potential problem.
407 if (!engine->IsDedicatedServer())
409 client = g_Players.ListenClient();
412 pHook->pf->PushCell(client);
413 pHook->pf->PushCell(args);
415 if (pHook->pf->Execute(&tempres) == SP_ERROR_NONE)
417 if (tempres > result)
421 if (result == Pl_Stop)
429 if (result >= Pl_Handled)
431 if (!pInfo->sourceMod)
433 RETURN_META(MRES_SUPERCEDE);
439 bool ConCmdManager::CheckCommandAccess(int client, const char *cmd, FlagBits cmdflags)
441 if (cmdflags == 0 || client == 0)
446 /* If running listen server, then client 1 is the server host and should have 'root' access */
447 if (client == 1 && !engine->IsDedicatedServer())
452 CPlayer *player = g_Players.GetPlayerByIndex(client);
454 || player->GetEdict() == NULL
455 || player->IsFakeClient())
460 AdminId adm = player->GetAdminId();
461 if (adm != INVALID_ADMIN_ID)
463 FlagBits bits = g_Admins.GetAdminFlags(adm, Access_Effective);
465 /* root knows all, WHOA */
466 if ((bits & ADMFLAG_ROOT) == ADMFLAG_ROOT)
471 /* Check for overrides
472 * :TODO: is it worth optimizing this?
474 unsigned int groups = g_Admins.GetAdminGroupCount(adm);
477 bool override = false;
478 for (unsigned int i=0; i<groups; i++)
480 gid = g_Admins.GetAdminGroup(adm, i, NULL);
481 /* First get group-level override */
482 override = g_Admins.GetGroupCommandOverride(gid, cmd, Override_CommandGroup, &rule);
483 /* Now get the specific command override */
484 if (g_Admins.GetGroupCommandOverride(gid, cmd, Override_Command, &rule))
490 if (rule == Command_Allow)
494 else if (rule == Command_Deny)
501 /* See if our other flags match */
502 if ((bits & cmdflags) == cmdflags)
511 bool ConCmdManager::CheckAccess(int client, const char *cmd, AdminCmdInfo *pAdmin)
513 if (CheckCommandAccess(client, cmd, pAdmin->eflags))
518 edict_t *pEdict = engine->PEntityOfEntIndex(client);
520 /* If we got here, the command failed... */
522 if (!CoreTranslate(buffer, sizeof(buffer), "%T", 2, NULL, "No Access", &client))
524 UTIL_Format(buffer, sizeof(buffer), "You do not have access to this command");
527 unsigned int replyto = g_ChatTriggers.GetReplyTo();
528 if (replyto == SM_REPLY_CONSOLE)
530 char fullbuffer[192];
531 UTIL_Format(fullbuffer, sizeof(fullbuffer), "[SM] %s.\n", buffer);
532 engine->ClientPrintf(pEdict, fullbuffer);
534 else if (replyto == SM_REPLY_CHAT)
536 char fullbuffer[192];
537 UTIL_Format(fullbuffer, sizeof(fullbuffer), "[SM] %s.", buffer);
538 g_HL2.TextMsg(client, HUD_PRINTTALK, fullbuffer);
544 bool ConCmdManager::AddConsoleCommand(IPluginFunction *pFunction,
546 const char *description,
549 ConCmdInfo *pInfo = AddOrFindCommand(name, description, flags);
556 CmdHook *pHook = new CmdHook();
558 pHook->pf = pFunction;
559 if (description && description[0])
561 pHook->helptext.assign(description);
563 pInfo->conhooks.push_back(pHook);
565 /* Add to the plugin */
567 IPlugin *pPlugin = g_PluginSys.GetPluginByCtx(pFunction->GetParentContext()->GetContext());
568 if (!pPlugin->GetProperty("CommandList", (void **)&pList))
570 pList = new CmdList();
571 pPlugin->SetProperty("CommandList", pList);
575 info.type = Cmd_Console;
577 AddToPlCmdList(pList, info);
582 bool ConCmdManager::AddAdminCommand(IPluginFunction *pFunction,
586 const char *description,
589 ConCmdInfo *pInfo = AddOrFindCommand(name, description, flags);
596 CmdHook *pHook = new CmdHook();
597 AdminCmdInfo *pAdmin = new AdminCmdInfo();
599 pHook->pf = pFunction;
600 if (description && description[0])
602 pHook->helptext.assign(description);
604 pHook->pAdmin = pAdmin;
608 if (!sm_trie_retrieve(m_pCmdGrps, group, (void **)&object))
610 grpid = m_Strings.AddString(group);
611 sm_trie_insert(m_pCmdGrps, group, (void *)grpid);
618 pAdmin->cmdGrpId = grpid;
619 pAdmin->flags = adminflags;
621 /* First get the command group override, if any */
622 bool override = g_Admins.GetCommandOverride(group,
623 Override_CommandGroup,
626 /* Next get the command override, if any */
627 if (g_Admins.GetCommandOverride(name,
634 /* Assign normal flags if there were no overrides */
637 pAdmin->eflags = pAdmin->flags;
640 /* Finally, add the hook */
641 pInfo->conhooks.push_back(pHook);
642 pInfo->admin = *(pHook->pAdmin);
643 pInfo->is_admin_set = true;
645 /* Now add to the plugin */
647 IPlugin *pPlugin = g_PluginSys.GetPluginByCtx(pFunction->GetParentContext()->GetContext());
648 if (!pPlugin->GetProperty("CommandList", (void **)&pList))
650 pList = new CmdList();
651 pPlugin->SetProperty("CommandList", pList);
655 info.type = Cmd_Admin;
657 AddToPlCmdList(pList, info);
662 bool ConCmdManager::AddServerCommand(IPluginFunction *pFunction,
664 const char *description,
668 ConCmdInfo *pInfo = AddOrFindCommand(name, description, flags);
675 CmdHook *pHook = new CmdHook();
677 pHook->pf = pFunction;
678 if (description && description[0])
680 pHook->helptext.assign(description);
683 pInfo->srvhooks.push_back(pHook);
685 /* Add to the plugin */
687 IPlugin *pPlugin = g_PluginSys.GetPluginByCtx(pFunction->GetParentContext()->GetContext());
688 if (!pPlugin->GetProperty("CommandList", (void **)&pList))
690 pList = new CmdList();
691 pPlugin->SetProperty("CommandList", pList);
695 info.type = Cmd_Server;
697 AddToPlCmdList(pList, info);
702 void AddToPlCmdList(CmdList *pList, const PlCmdInfo &info)
704 CmdList::iterator iter = pList->begin();
705 bool inserted = false;
706 const char *orig = NULL;
708 orig = info.pInfo->pCmd->GetName();
710 /* Insert this into the help list, SORTED alphabetically. */
711 while (iter != pList->end())
713 PlCmdInfo &obj = (*iter);
714 const char *cmd = obj.pInfo->pCmd->GetName();
715 if (strcmp(orig, cmd) < 0)
717 pList->insert(iter, info);
726 pList->push_back(info);
730 void ConCmdManager::AddToCmdList(ConCmdInfo *info)
732 List<ConCmdInfo *>::iterator iter = m_CmdList.begin();
734 bool inserted = false;
735 const char *orig = NULL;
737 orig = info->pCmd->GetName();
739 /* Insert this into the help list, SORTED alphabetically. */
740 while (iter != m_CmdList.end())
742 const char *cmd = NULL;
744 cmd = pInfo->pCmd->GetName();
745 if (strcmp(orig, cmd) < 0)
747 m_CmdList.insert(iter, info);
756 m_CmdList.push_back(info);
760 void ConCmdManager::UpdateAdminCmdFlags(const char *cmd, OverrideType type, FlagBits bits, bool remove)
764 if (type == Override_Command)
766 if (!sm_trie_retrieve(m_pCmds, cmd, (void **)&pInfo))
771 List<CmdHook *>::iterator iter;
774 for (iter=pInfo->conhooks.begin(); iter!=pInfo->conhooks.end(); iter++)
781 pHook->pAdmin->eflags = bits;
783 pHook->pAdmin->eflags = pHook->pAdmin->flags;
785 pInfo->admin = *(pHook->pAdmin);
788 pInfo->is_admin_set = true;
790 else if (type == Override_CommandGroup)
793 if (!sm_trie_retrieve(m_pCmdGrps, cmd, &object))
797 int grpid = (int)object;
799 /* This is bad :( loop through all commands */
800 List<ConCmdInfo *>::iterator iter;
802 for (iter=m_CmdList.begin(); iter!=m_CmdList.end(); iter++)
805 for (List<CmdHook *>::iterator citer=pInfo->conhooks.begin();
806 citer!=pInfo->conhooks.end();
810 if (pHook->pAdmin && pHook->pAdmin->cmdGrpId == grpid)
814 pHook->pAdmin->eflags = bits;
816 pHook->pAdmin->eflags = pHook->pAdmin->flags;
818 pInfo->admin = *(pHook->pAdmin);
822 pInfo->is_admin_set = true;
826 void ConCmdManager::RemoveConCmd(ConCmdInfo *info, const char *name, bool is_read_safe, bool untrack)
828 /* Remove from the trie */
829 sm_trie_delete(m_pCmds, name);
831 /* Remove console-specific information
832 * This should always be true as of right now
838 /* Unlink from SourceMM */
839 g_SMAPI->UnregisterConCommandBase(g_PLAPI, info->pCmd);
840 /* Delete the command's memory */
841 char *new_help = const_cast<char *>(info->pCmd->GetHelpText());
842 char *new_name = const_cast<char *>(info->pCmd->GetName());
851 /* Remove the external hook */
852 SH_REMOVE_HOOK_STATICFUNC(ConCommand, Dispatch, info->pCmd, CommandCallback, false);
856 UntrackConCommandBase(info->pCmd, this);
861 /* Remove from list */
862 m_CmdList.remove(info);
867 bool ConCmdManager::LookForSourceModCommand(const char *cmd)
871 if (!sm_trie_retrieve(m_pCmds, cmd, (void **)&pInfo))
876 return pInfo->sourceMod && (pInfo->conhooks.size() > 0);
879 bool ConCmdManager::LookForCommandAdminFlags(const char *cmd, FlagBits *pFlags)
883 if (!sm_trie_retrieve(m_pCmds, cmd, (void **)&pInfo))
888 *pFlags = pInfo->admin.eflags;
890 return pInfo->is_admin_set;
893 ConCmdInfo *ConCmdManager::AddOrFindCommand(const char *name, const char *description, int flags)
896 if (!sm_trie_retrieve(m_pCmds, name, (void **)&pInfo))
898 pInfo = new ConCmdInfo();
899 /* Find the commandopan */
900 ConCommandBase *pBase = icvar->GetCommands();
901 ConCommand *pCmd = NULL;
904 if (strcmp(pBase->GetName(), name) == 0)
906 /* Don't want to return convar with same name */
907 if (!pBase->IsCommand())
912 pCmd = (ConCommand *)pBase;
915 pBase = const_cast<ConCommandBase *>(pBase->GetNext());
920 /* Note that we have to duplicate because the source might not be
921 * a static string, and these expect static memory.
927 char *new_name = sm_strdup(name);
928 char *new_help = sm_strdup(description);
929 pCmd = new ConCommand(new_name, CommandCallback, new_help, flags);
930 pInfo->sourceMod = true;
934 TrackConCommandBase(pCmd, this);
935 SH_ADD_HOOK_STATICFUNC(ConCommand, Dispatch, pCmd, CommandCallback, false);
939 pInfo->is_admin_set = false;
941 sm_trie_insert(m_pCmds, name, pInfo);
948 void ConCmdManager::OnRootConsoleCommand(const char *cmdname, const CCommand &command)
950 if (command.ArgC() >= 3)
952 const char *text = command.Arg(2);
954 CPlugin *pPlugin = g_PluginSys.FindPluginByConsoleArg(text);
958 g_RootMenu.ConsolePrint("[SM] Plugin \"%s\" was not found.", text);
962 const sm_plugininfo_t *plinfo = pPlugin->GetPublicInfo();
963 const char *plname = IS_STR_FILLED(plinfo->name) ? plinfo->name : pPlugin->GetFilename();
966 if (!pPlugin->GetProperty("CommandList", (void **)&pList))
968 g_RootMenu.ConsolePrint("[SM] No commands found for: %s", plname);
973 g_RootMenu.ConsolePrint("[SM] No commands found for: %s", plname);
977 CmdList::iterator iter;
978 const char *type = NULL;
981 g_RootMenu.ConsolePrint("[SM] Listing %d commands for: %s", pList->size(), plname);
982 g_RootMenu.ConsolePrint(" %-17.16s %-8.7s %s", "[Name]", "[Type]", "[Help]");
983 for (iter=pList->begin();
987 PlCmdInfo &cmd = (*iter);
988 if (cmd.type == Cmd_Server)
992 else if (cmd.type == Cmd_Console)
996 else if (cmd.type == Cmd_Admin)
1000 name = cmd.pInfo->pCmd->GetName();
1001 if (cmd.pHook->helptext.size())
1003 help = cmd.pHook->helptext.c_str();
1007 help = cmd.pInfo->pCmd->GetHelpText();
1009 g_RootMenu.ConsolePrint(" %-17.16s %-12.11s %s", name, type, help);
1015 g_RootMenu.ConsolePrint("[SM] Usage: sm cmds <plugin #>");