dvander@1231: /** dvander@1231: * vim: set ts=4 : ds@1330: * ============================================================================= dvander@1231: * SourceMod ds@1947: * Copyright (C) 2004-2008 AlliedModders LLC. All rights reserved. ds@1330: * ============================================================================= dvander@1231: * ds@1330: * This program is free software; you can redistribute it and/or modify it under ds@1330: * the terms of the GNU General Public License, version 3.0, as published by the ds@1330: * Free Software Foundation. dvander@1231: * ds@1330: * This program is distributed in the hope that it will be useful, but WITHOUT ds@1330: * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS ds@1330: * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more ds@1330: * details. dvander@1231: * ds@1330: * You should have received a copy of the GNU General Public License along with ds@1330: * this program. If not, see . dvander@1231: * ds@1330: * As a special exception, AlliedModders LLC gives you permission to link the ds@1330: * code of this program (as well as its derivative works) to "Half-Life 2," the ds@1330: * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software ds@1330: * by the Valve Corporation. You must obey the GNU General Public License in ds@1330: * all respects for all other code used. Additionally, AlliedModders LLC grants ds@1330: * this exception to all derivative works. AlliedModders LLC defines further ds@1330: * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007), ds@1330: * or . dvander@1231: * dvander@1231: * Version: $Id$ dvander@1231: */ dvander@1231: dvander@1406: #include dvander@893: #include "ChatTriggers.h" dvander@893: #include "sm_stringutil.h" dvander@893: #include "ConCmdManager.h" dvander@1794: #include "PlayerManager.h" dvander@1925: #include "HalfLife2.h" dvander@3004: #include "logic_bridge.h" dvander@893: skamonkey@3899: #if SOURCE_ENGINE == SE_DOTA skamonkey@3942: SH_DECL_EXTERN2_void(ConCommand, Dispatch, SH_NOATTRIB, false, const CCommandContext &, const CCommand &); skamonkey@3899: #elif SOURCE_ENGINE >= SE_ORANGEBOX skamonkey@3899: SH_DECL_EXTERN1_void(ConCommand, Dispatch, SH_NOATTRIB, false, const CCommand &); skamonkey@3899: #elif SOURCE_ENGINE == SE_DARKMESSIAH skamonkey@3899: SH_DECL_EXTERN0_void(ConCommand, Dispatch, SH_NOATTRIB, false); faluco@1541: #else skamonkey@3899: # if SH_IMPL_VERSION >= 4 skamonkey@3899: extern int __SourceHook_FHAddConCommandDispatch(void *,bool,class fastdelegate::FastDelegate0); skamonkey@3899: # else skamonkey@3899: extern bool __SourceHook_FHAddConCommandDispatch(void *,bool,class fastdelegate::FastDelegate0); skamonkey@3899: # endif skamonkey@3899: extern bool __SourceHook_FHRemoveConCommandDispatch(void *,bool,class fastdelegate::FastDelegate0); skamonkey@3899: #endif dvander@893: dvander@893: ChatTriggers g_ChatTriggers; dvander@1794: bool g_bSupressSilentFails = false; dvander@893: dvander@893: ChatTriggers::ChatTriggers() : m_pSayCmd(NULL), m_bWillProcessInPost(false), dvander@1103: m_bTriggerWasSilent(false), m_ReplyTo(SM_REPLY_CONSOLE) dvander@893: { dvander@893: m_PubTrigger = sm_strdup("!"); dvander@893: m_PrivTrigger = sm_strdup("/"); dvander@893: m_PubTriggerSize = 1; dvander@893: m_PrivTriggerSize = 1; dvander@1469: m_bIsChatTrigger = false; kyle@3974: m_bPluginIgnored = false; kyle@3974: #if SOURCE_ENGINE == SE_EPISODEONE kyle@3974: m_bIsINS = false; kyle@3974: #endif dvander@893: } dvander@893: dvander@897: ChatTriggers::~ChatTriggers() dvander@897: { dvander@897: delete [] m_PubTrigger; dvander@897: m_PubTrigger = NULL; dvander@897: delete [] m_PrivTrigger; dvander@897: m_PrivTrigger = NULL; dvander@897: } dvander@896: dvander@896: ConfigResult ChatTriggers::OnSourceModConfigChanged(const char *key, dvander@896: const char *value, dvander@896: ConfigSource source, dvander@896: char *error, dvander@896: size_t maxlength) dvander@896: { dvander@896: if (strcmp(key, "PublicChatTrigger") == 0) dvander@896: { dvander@896: delete [] m_PubTrigger; dvander@896: m_PubTrigger = sm_strdup(value); dvander@896: m_PubTriggerSize = strlen(m_PubTrigger); dvander@896: return ConfigResult_Accept; dvander@1794: } dvander@1794: else if (strcmp(key, "SilentChatTrigger") == 0) dvander@1794: { dvander@896: delete [] m_PrivTrigger; dvander@896: m_PrivTrigger = sm_strdup(value); dvander@896: m_PrivTriggerSize = strlen(m_PrivTrigger); dvander@896: return ConfigResult_Accept; dvander@896: } dvander@1794: else if (strcmp(key, "SilentFailSuppress") == 0) dvander@1794: { dvander@1794: g_bSupressSilentFails = strcmp(value, "yes") == 0; dvander@1794: return ConfigResult_Accept; dvander@1794: } dvander@896: dvander@896: return ConfigResult_Ignore; dvander@896: } dvander@896: dvander@1925: void ChatTriggers::OnSourceModAllInitialized() dvander@1925: { dvander@1925: m_pShouldFloodBlock = g_Forwards.CreateForward("OnClientFloodCheck", ET_Event, 1, NULL, Param_Cell); dvander@1925: m_pDidFloodBlock = g_Forwards.CreateForward("OnClientFloodResult", ET_Event, 2, NULL, Param_Cell, Param_Cell); kyle@3974: m_pOnClientSayCmd = g_Forwards.CreateForward("OnClientSayCommand", ET_Event, 3, NULL, Param_Cell, Param_String, Param_String); kyle@3974: m_pOnClientSayCmd_Post = g_Forwards.CreateForward("OnClientSayCommand_Post", ET_Ignore, 3, NULL, Param_Cell, Param_String, Param_String); dvander@1925: } dvander@1925: dvander@1925: void ChatTriggers::OnSourceModAllInitialized_Post() dvander@1925: { dvander@3004: logicore.AddCorePhraseFile("antiflood.phrases"); dvander@1925: } dvander@1925: dvander@893: void ChatTriggers::OnSourceModGameInitialized() dvander@893: { ds@2400: m_pSayCmd = FindCommand("say"); ds@2400: m_pSayTeamCmd = FindCommand("say_team"); dvander@893: dvander@893: if (m_pSayCmd) dvander@893: { kyle@3819: SH_ADD_HOOK(ConCommand, Dispatch, m_pSayCmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Pre), false); kyle@3819: SH_ADD_HOOK(ConCommand, Dispatch, m_pSayCmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Post), true); dvander@893: } dvander@897: if (m_pSayTeamCmd) dvander@897: { kyle@3819: SH_ADD_HOOK(ConCommand, Dispatch, m_pSayTeamCmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Pre), false); kyle@3819: SH_ADD_HOOK(ConCommand, Dispatch, m_pSayTeamCmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Post), true); dvander@897: } kyle@3974: kyle@3974: #if SOURCE_ENGINE == SE_EPISODEONE skamonkey@3975: m_bIsINS = (strcmp(g_SourceMod.GetGameFolderName(), "insurgency") == 0); kyle@3974: kyle@3974: if (m_bIsINS) kyle@3974: { kyle@3974: m_pSay2Cmd = FindCommand("say2"); kyle@3974: if (m_pSay2Cmd) kyle@3974: { kyle@3974: SH_ADD_HOOK(ConCommand, Dispatch, m_pSay2Cmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Pre), false); kyle@3974: SH_ADD_HOOK(ConCommand, Dispatch, m_pSay2Cmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Post), true); kyle@3974: } kyle@3974: } kyle@3974: #elif SOURCE_ENGINE == SE_NUCLEARDAWN kyle@3974: m_pSaySquadCmd = FindCommand("say_squad"); kyle@3974: kyle@3974: if (m_pSaySquadCmd) kyle@3974: { kyle@3974: SH_ADD_HOOK(ConCommand, Dispatch, m_pSaySquadCmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Pre), false); kyle@3974: SH_ADD_HOOK(ConCommand, Dispatch, m_pSaySquadCmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Post), true); kyle@3974: } kyle@3974: #endif dvander@893: } dvander@893: dvander@893: void ChatTriggers::OnSourceModShutdown() dvander@893: { dvander@893: if (m_pSayCmd) dvander@893: { kyle@3819: SH_REMOVE_HOOK(ConCommand, Dispatch, m_pSayCmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Post), true); kyle@3819: SH_REMOVE_HOOK(ConCommand, Dispatch, m_pSayCmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Pre), false); dvander@893: } dvander@1925: kyle@3974: if (m_pSayTeamCmd) kyle@3974: { kyle@3974: SH_REMOVE_HOOK(ConCommand, Dispatch, m_pSayTeamCmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Post), true); kyle@3974: SH_REMOVE_HOOK(ConCommand, Dispatch, m_pSayTeamCmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Pre), false); kyle@3974: } kyle@3974: kyle@3974: #if SOURCE_ENGINE == SE_EPISODEONE kyle@3974: if (m_bIsINS && m_pSay2Cmd) kyle@3974: { kyle@3974: SH_REMOVE_HOOK(ConCommand, Dispatch, m_pSay2Cmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Pre), false); kyle@3974: SH_REMOVE_HOOK(ConCommand, Dispatch, m_pSay2Cmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Post), true); kyle@3974: } kyle@3974: #elif SOURCE_ENGINE == SE_NUCLEARDAWN kyle@3974: if (m_pSaySquadCmd) kyle@3974: { kyle@3974: SH_REMOVE_HOOK(ConCommand, Dispatch, m_pSaySquadCmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Pre), false); kyle@3974: SH_REMOVE_HOOK(ConCommand, Dispatch, m_pSaySquadCmd, SH_MEMBER(this, &ChatTriggers::OnSayCommand_Post), true); kyle@3974: } kyle@3974: #endif kyle@3974: dvander@1925: g_Forwards.ReleaseForward(m_pShouldFloodBlock); dvander@1925: g_Forwards.ReleaseForward(m_pDidFloodBlock); kyle@3974: g_Forwards.ReleaseForward(m_pOnClientSayCmd); kyle@3974: g_Forwards.ReleaseForward(m_pOnClientSayCmd_Post); dvander@893: } dvander@893: skamonkey@3899: #if SOURCE_ENGINE == SE_DOTA skamonkey@3942: void ChatTriggers::OnSayCommand_Pre(const CCommandContext &context, const CCommand &command) skamonkey@3899: { skamonkey@3899: #elif SOURCE_ENGINE >= SE_ORANGEBOX faluco@1541: void ChatTriggers::OnSayCommand_Pre(const CCommand &command) faluco@1541: { faluco@1541: #else dvander@893: void ChatTriggers::OnSayCommand_Pre() dvander@893: { faluco@1541: CCommand command; faluco@1541: #endif kyle@3974: int client = g_ConCmds.GetCommandClient(); dvander@1469: m_bIsChatTrigger = false; dvander@1925: m_bWasFloodedMessage = false; kyle@3974: m_bPluginIgnored = false; dvander@893: dvander@893: /* The server console cannot do this */ kyle@3974: if (client == 0) dvander@1934: { dvander@1934: RETURN_META(MRES_IGNORED); dvander@1934: } dvander@1934: kyle@3974: CPlayer *pPlayer = g_Players.GetPlayerByIndex(client); kyle@3974: dvander@1934: /* We guarantee the client is connected */ kyle@3974: if (!pPlayer || !pPlayer->IsConnected()) dvander@893: { dvander@893: RETURN_META(MRES_IGNORED); dvander@893: } dvander@893: faluco@1541: const char *args = command.ArgS(); ds@1139: ds@1139: if (!args) ds@1139: { ds@1139: RETURN_META(MRES_IGNORED); ds@1139: } dvander@893: dvander@1925: /* Check if we need to block this message from being sent */ dvander@1925: if (ClientIsFlooding(client)) dvander@1925: { dvander@1925: char buffer[128]; dvander@1925: dvander@3004: if (!logicore.CoreTranslate(buffer, sizeof(buffer), "%T", 2, NULL, "Flooding the server", &client)) dvander@1925: UTIL_Format(buffer, sizeof(buffer), "You are flooding the server!"); dvander@1925: dvander@1925: /* :TODO: we should probably kick people who spam too much. */ dvander@1925: dvander@1925: char fullbuffer[192]; dvander@1925: UTIL_Format(fullbuffer, sizeof(fullbuffer), "[SM] %s", buffer); dvander@1925: g_HL2.TextMsg(client, HUD_PRINTTALK, fullbuffer); dvander@1925: dvander@1925: m_bWasFloodedMessage = true; dvander@1925: dvander@1925: RETURN_META(MRES_SUPERCEDE); dvander@1925: } dvander@1925: dvander@893: /* Handle quoted string sets */ dvander@893: bool is_quoted = false; dvander@893: if (args[0] == '"') dvander@893: { dvander@893: args++; dvander@893: is_quoted = true; dvander@893: } dvander@893: skamonkey@4015: /* Save these off for post hook as the command data returned from the engine in older engine versions skamonkey@4015: * can be NULL, despite the data still being there and valid. */ skamonkey@4015: m_Arg0Backup = command.Arg(0); skamonkey@4015: m_ArgSBackup = command.ArgS(); skamonkey@4015: kyle@3974: #if SOURCE_ENGINE == SE_EPISODEONE skamonkey@4015: if (m_bIsINS && strcmp(m_Arg0Backup, "say2") == 0 && strlen(args) >= 4) kyle@3974: { kyle@3974: args += 4; kyle@3974: } kyle@3974: #endif kyle@3974: dvander@893: bool is_trigger = false; dvander@893: bool is_silent = false; dvander@893: dvander@893: /* Check for either trigger */ dvander@893: if (m_PubTriggerSize && strncmp(args, m_PubTrigger, m_PubTriggerSize) == 0) dvander@893: { dvander@893: is_trigger = true; dvander@1103: args = &args[m_PubTriggerSize]; dvander@1925: } dvander@2607: else if (m_PrivTriggerSize && strncmp(args, m_PrivTrigger, m_PrivTriggerSize) == 0) dvander@1925: { dvander@893: is_trigger = true; dvander@893: is_silent = true; dvander@1103: args = &args[m_PrivTriggerSize]; dvander@893: } dvander@893: dvander@1103: /** dvander@1103: * Test if this is actually a command! dvander@1103: */ kyle@3974: if (is_trigger && PreProcessTrigger(PEntityOfEntIndex(client), args, is_quoted)) dvander@893: { kyle@3974: m_bIsChatTrigger = true; kyle@3974: kyle@3974: /** kyle@3974: * We'll execute it in post. kyle@3974: */ kyle@3974: m_bWillProcessInPost = true; kyle@3974: m_bTriggerWasSilent = is_silent; kyle@3974: } kyle@3974: kyle@3974: if (m_pOnClientSayCmd->GetFunctionCount() != 0) kyle@3974: { kyle@3974: cell_t res = Pl_Continue; kyle@3974: m_pOnClientSayCmd->PushCell(client); skamonkey@4015: m_pOnClientSayCmd->PushString(m_Arg0Backup); skamonkey@4015: m_pOnClientSayCmd->PushString(m_ArgSBackup); kyle@3974: m_pOnClientSayCmd->Execute(&res); kyle@3974: kyle@3974: if (res >= Pl_Handled) dvander@1794: { kyle@3974: m_bPluginIgnored = (res >= Pl_Stop); dvander@1794: RETURN_META(MRES_SUPERCEDE); dvander@1794: } dvander@893: } dvander@893: kyle@3974: if (m_bWillProcessInPost || \ kyle@3974: (is_silent && g_bSupressSilentFails && pPlayer->GetAdminId() != INVALID_ADMIN_ID)) dvander@893: { dvander@893: RETURN_META(MRES_SUPERCEDE); dvander@893: } dvander@893: dvander@1103: /* Otherwise, let the command continue */ dvander@893: RETURN_META(MRES_IGNORED); dvander@893: } dvander@893: skamonkey@3899: #if SOURCE_ENGINE == SE_DOTA skamonkey@3942: void ChatTriggers::OnSayCommand_Post(const CCommandContext &context, const CCommand &command) skamonkey@3899: #elif SOURCE_ENGINE >= SE_ORANGEBOX faluco@1541: void ChatTriggers::OnSayCommand_Post(const CCommand &command) faluco@1541: #else dvander@893: void ChatTriggers::OnSayCommand_Post() skamonkey@4015: #endif skamonkey@3975: { kyle@3974: int client = g_ConCmds.GetCommandClient(); kyle@3974: skamonkey@4041: if (m_bWillProcessInPost) dvander@893: { skamonkey@4041: /* Reset this for re-entrancy */ skamonkey@4041: m_bWillProcessInPost = false; skamonkey@4041: dvander@1103: /* Execute the cached command */ dvander@1103: unsigned int old = SetReplyTo(SM_REPLY_CHAT); skamonkey@3854: #if SOURCE_ENGINE == SE_DOTA skamonkey@3913: engine->ClientCommand(client, "%s", m_ToExecute); skamonkey@3854: #else ds@2400: serverpluginhelpers->ClientCommand(PEntityOfEntIndex(client), m_ToExecute); skamonkey@3854: #endif dvander@1103: SetReplyTo(old); dvander@893: } kyle@3974: kyle@3974: if (m_bPluginIgnored) kyle@3974: { kyle@3974: m_bPluginIgnored = false; kyle@3974: } kyle@3974: else if (!m_bWasFloodedMessage && m_pOnClientSayCmd_Post->GetFunctionCount() != 0) kyle@3974: { kyle@3974: m_pOnClientSayCmd_Post->PushCell(client); skamonkey@4015: m_pOnClientSayCmd_Post->PushString(m_Arg0Backup); skamonkey@4015: m_pOnClientSayCmd_Post->PushString(m_ArgSBackup); kyle@3974: m_pOnClientSayCmd_Post->Execute(NULL); kyle@3974: } kyle@3974: kyle@3974: m_bIsChatTrigger = false; kyle@3974: m_bWasFloodedMessage = false; dvander@893: } dvander@893: dvander@1103: bool ChatTriggers::PreProcessTrigger(edict_t *pEdict, const char *args, bool is_quoted) dvander@893: { dvander@893: /* Extract a command. This is kind of sloppy. */ dvander@893: char cmd_buf[64]; dvander@893: size_t cmd_len = 0; dvander@893: const char *inptr = args; dvander@893: while (*inptr != '\0' dvander@1406: && !textparsers->IsWhitespace(inptr) dvander@935: && *inptr != '"' dvander@893: && cmd_len < sizeof(cmd_buf) - 1) dvander@893: { dvander@893: cmd_buf[cmd_len++] = *inptr++; dvander@893: } dvander@893: cmd_buf[cmd_len] = '\0'; dvander@893: dvander@1377: if (cmd_len == 0) dvander@1377: { dvander@1377: return false; dvander@1377: } dvander@1377: dvander@893: /* See if we have this registered */ dvander@893: bool prepended = false; dvander@893: if (!g_ConCmds.LookForSourceModCommand(cmd_buf)) dvander@893: { dvander@893: /* Check if we had an "sm_" prefix */ dvander@893: if (strncmp(cmd_buf, "sm_", 3) == 0) dvander@893: { dvander@893: return false; dvander@893: } dvander@893: dvander@893: /* Now, prepend. Don't worry about the buffers. This will dvander@893: * work because the sizes are limited from earlier. dvander@893: */ dvander@893: char new_buf[80]; dvander@893: strcpy(new_buf, "sm_"); dvander@893: strncopy(&new_buf[3], cmd_buf, sizeof(new_buf)-3); dvander@893: dvander@893: /* Recheck */ dvander@893: if (!g_ConCmds.LookForSourceModCommand(new_buf)) dvander@893: { dvander@893: return false; dvander@893: } dvander@893: dvander@893: prepended = true; dvander@893: } dvander@893: dvander@893: /* See if we need to do extra string manipulation */ dvander@893: if (is_quoted || prepended) dvander@893: { dvander@893: size_t len; dvander@893: dvander@893: /* Check if we need to prepend sm_ */ dvander@893: if (prepended) dvander@893: { dvander@1103: len = UTIL_Format(m_ToExecute, sizeof(m_ToExecute), "sm_%s", args); dvander@893: } else { dvander@1103: len = strncopy(m_ToExecute, args, sizeof(m_ToExecute)); dvander@893: } dvander@893: dvander@893: /* Check if we need to strip a quote */ dvander@893: if (is_quoted) dvander@893: { dvander@1103: if (m_ToExecute[len-1] == '"') dvander@893: { dvander@1103: m_ToExecute[--len] = '\0'; dvander@893: } dvander@893: } dvander@1103: } else { dvander@1103: strncopy(m_ToExecute, args, sizeof(m_ToExecute)); dvander@893: } dvander@893: dvander@893: return true; dvander@893: } dvander@893: dvander@893: unsigned int ChatTriggers::SetReplyTo(unsigned int reply) dvander@893: { dvander@893: unsigned int old = m_ReplyTo; dvander@893: dvander@893: m_ReplyTo = reply; dvander@893: dvander@893: return old; dvander@893: } dvander@893: dvander@893: unsigned int ChatTriggers::GetReplyTo() dvander@893: { dvander@893: return m_ReplyTo; dvander@893: } dvander@1469: dvander@1469: bool ChatTriggers::IsChatTrigger() dvander@1469: { dvander@1469: return m_bIsChatTrigger; dvander@1469: } dvander@1925: dvander@1925: bool ChatTriggers::ClientIsFlooding(int client) dvander@1925: { dvander@1925: bool is_flooding = false; dvander@1925: dvander@1925: if (m_pShouldFloodBlock->GetFunctionCount() != 0) dvander@1925: { dvander@1925: cell_t res = 0; dvander@1925: dvander@1925: m_pShouldFloodBlock->PushCell(client); dvander@1925: m_pShouldFloodBlock->Execute(&res); dvander@1925: dvander@1925: if (res != 0) dvander@1925: { dvander@1925: is_flooding = true; dvander@1925: } dvander@1925: } dvander@1925: dvander@1925: if (m_pDidFloodBlock->GetFunctionCount() != 0) dvander@1925: { dvander@1925: m_pDidFloodBlock->PushCell(client); dvander@1925: m_pDidFloodBlock->PushCell(is_flooding ? 1 : 0); dvander@1925: m_pDidFloodBlock->Execute(NULL); dvander@1925: } dvander@1925: dvander@1925: return is_flooding; dvander@1925: } dvander@1925: dvander@1925: bool ChatTriggers::WasFloodedMessage() dvander@1925: { dvander@1925: return m_bWasFloodedMessage; dvander@1925: }