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 <http://www.gnu.org/licenses/>.
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 <http://www.sourcemod.net/license.php>.
dvander@1231:  *
dvander@1231:  * Version: $Id$
dvander@1231:  */
dvander@1231: 
dvander@1406: #include <ITextParsers.h>
dvander@893: #include "ChatTriggers.h"
dvander@893: #include "sm_stringutil.h"
dvander@893: #include "ConCmdManager.h"
dvander@1794: #include "PlayerManager.h"
dvander@1925: #include "Translator.h"
dvander@1925: #include "HalfLife2.h"
dvander@893: 
dvander@1547: /* :HACKHACK: We can't SH_DECL here because ConCmdManager.cpp does.
dvander@1547:  * While the OB build only runs on MM:S 1.6.0+ (SH 5+), the older one 
dvander@1547:  * can technically be compiled against any MM:S version after 1.4.2.
dvander@1547:  */
ds@2400: #if SOURCE_ENGINE >= SE_ORANGEBOX
faluco@1541: extern bool __SourceHook_FHRemoveConCommandDispatch(void *, bool, class fastdelegate::FastDelegate1<const CCommand &, void>);
faluco@1541: extern int __SourceHook_FHAddConCommandDispatch(void *, ISourceHook::AddHookMode, bool, class fastdelegate::FastDelegate1<const CCommand &, void>);
faluco@1541: #else
dvander@893: extern bool __SourceHook_FHRemoveConCommandDispatch(void *, bool, class fastdelegate::FastDelegate0<void>);
dvander@1547: #if SH_IMPL_VERSION >= 5
dvander@1547: extern int __SourceHook_FHAddConCommandDispatch(void *, ISourceHook::AddHookMode, bool, class fastdelegate::FastDelegate0<void>);
dvander@1547: #elif SH_IMPL_VERSION == 4
dvander@893: extern int __SourceHook_FHAddConCommandDispatch(void *, bool, class fastdelegate::FastDelegate0<void>);
dvander@1547: #elif SH_IMPL_VERSION == 3
dvander@904: extern bool __SourceHook_FHAddConCommandDispatch(void *, bool, class fastdelegate::FastDelegate0<void>);
dvander@1547: #endif //SH_IMPL_VERSION
ds@2400: #endif //SE_ORANGEBOX
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;
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);
dvander@1925: }
dvander@1925: 
dvander@1925: void ChatTriggers::OnSourceModAllInitialized_Post()
dvander@1925: {
dvander@2276: 	g_pCorePhrases->AddPhraseFile("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: 	{
dvander@893: 		SH_ADD_HOOK_MEMFUNC(ConCommand, Dispatch, m_pSayCmd, this, &ChatTriggers::OnSayCommand_Pre, false);
dvander@893: 		SH_ADD_HOOK_MEMFUNC(ConCommand, Dispatch, m_pSayCmd, this, &ChatTriggers::OnSayCommand_Post, true);
dvander@893: 	}
dvander@897: 	if (m_pSayTeamCmd)
dvander@897: 	{
dvander@897: 		SH_ADD_HOOK_MEMFUNC(ConCommand, Dispatch, m_pSayTeamCmd, this, &ChatTriggers::OnSayCommand_Pre, false);
dvander@897: 		SH_ADD_HOOK_MEMFUNC(ConCommand, Dispatch, m_pSayTeamCmd, this, &ChatTriggers::OnSayCommand_Post, true);
dvander@897: 	}
dvander@893: }
dvander@893: 
dvander@893: void ChatTriggers::OnSourceModShutdown()
dvander@893: {
dvander@897: 	if (m_pSayTeamCmd)
dvander@897: 	{
dvander@897: 		SH_REMOVE_HOOK_MEMFUNC(ConCommand, Dispatch, m_pSayTeamCmd, this, &ChatTriggers::OnSayCommand_Post, true);
dvander@897: 		SH_REMOVE_HOOK_MEMFUNC(ConCommand, Dispatch, m_pSayTeamCmd, this, &ChatTriggers::OnSayCommand_Pre, false);
dvander@897: 	}
dvander@893: 	if (m_pSayCmd)
dvander@893: 	{
dvander@893: 		SH_REMOVE_HOOK_MEMFUNC(ConCommand, Dispatch, m_pSayCmd, this, &ChatTriggers::OnSayCommand_Post, true);
dvander@893: 		SH_REMOVE_HOOK_MEMFUNC(ConCommand, Dispatch, m_pSayCmd, this, &ChatTriggers::OnSayCommand_Pre, false);
dvander@893: 	}
dvander@1925: 
dvander@1925: 	g_Forwards.ReleaseForward(m_pShouldFloodBlock);
dvander@1925: 	g_Forwards.ReleaseForward(m_pDidFloodBlock);
dvander@893: }
dvander@893: 
ds@2400: #if 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
dvander@1934: 	int client;
dvander@1934: 	CPlayer *pPlayer;
dvander@1934: 	
dvander@1934: 	client = g_ConCmds.GetCommandClient();
dvander@1469: 	m_bIsChatTrigger = false;
dvander@1925: 	m_bWasFloodedMessage = false;
dvander@893: 
dvander@893: 	/* The server console cannot do this */
dvander@1934: 	if (client == 0 || (pPlayer = g_Players.GetPlayerByIndex(client)) == NULL)
dvander@1934: 	{
dvander@1934: 		RETURN_META(MRES_IGNORED);
dvander@1934: 	}
dvander@1934: 
dvander@1934: 	/* We guarantee the client is connected */
dvander@1934: 	if (!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@2042: 		if (!CoreTranslate(
dvander@2042: 			buffer,
dvander@1925: 			sizeof(buffer),
dvander@2042: 			"%T",
dvander@2042: 			2,
dvander@1925: 			NULL,
dvander@2042: 			"Flooding the server",
dvander@2042: 			&client))
dvander@1925: 		{
dvander@1925: 			UTIL_Format(buffer, sizeof(buffer), "You are flooding the server!");
dvander@1925: 		}
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: 
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@1925: 	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@893: 	if (!is_trigger)
dvander@893: 	{
dvander@893: 		RETURN_META(MRES_IGNORED);
dvander@893: 	}
dvander@893: 
dvander@1103: 	/**
dvander@1103: 	 * Test if this is actually a command!
dvander@1103: 	 */
ds@2400: 	if (!PreProcessTrigger(PEntityOfEntIndex(client), args, is_quoted))
dvander@893: 	{
dvander@1794: 		CPlayer *pPlayer;
dvander@1794: 		if (is_silent 
dvander@1794: 			&& g_bSupressSilentFails 
dvander@1794: 			&& client != 0
dvander@1794: 			&& (pPlayer = g_Players.GetPlayerByIndex(client)) != NULL
dvander@1794: 			&& pPlayer->GetAdminId() != INVALID_ADMIN_ID)
dvander@1794: 		{
dvander@1794: 			RETURN_META(MRES_SUPERCEDE);
dvander@1794: 		}
dvander@893: 		RETURN_META(MRES_IGNORED);
dvander@893: 	}
dvander@893: 
dvander@1469: 	m_bIsChatTrigger = true;
dvander@1469: 
dvander@1103: 	/**
dvander@1103: 	 * We'll execute it in post.
dvander@1103: 	 */
dvander@1103: 	m_bWillProcessInPost = true;
dvander@1103: 	m_bTriggerWasSilent = is_silent;
dvander@1103: 
dvander@1103: 	/* If we're silent, block */
dvander@1103: 	if (is_silent)
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: 
ds@2400: #if SOURCE_ENGINE >= SE_ORANGEBOX
faluco@1541: void ChatTriggers::OnSayCommand_Post(const CCommand &command)
faluco@1541: #else
dvander@893: void ChatTriggers::OnSayCommand_Post()
faluco@1541: #endif
dvander@893: {
dvander@1469: 	m_bIsChatTrigger = false;
dvander@1925: 	m_bWasFloodedMessage = false;
dvander@893: 	if (m_bWillProcessInPost)
dvander@893: 	{
dvander@893: 		/* Reset this for re-entrancy */
dvander@893: 		m_bWillProcessInPost = false;
dvander@1103: 		
dvander@1103: 		/* Execute the cached command */
dvander@893: 		int client = g_ConCmds.GetCommandClient();
dvander@1103: 		unsigned int old = SetReplyTo(SM_REPLY_CHAT);
ds@2400: 		serverpluginhelpers->ClientCommand(PEntityOfEntIndex(client), m_ToExecute);
dvander@1103: 		SetReplyTo(old);
dvander@893: 	}
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: }