X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fmodules%2Fm_alias.cpp;h=32fc80b64d5427638a4a7919b2d437fa34afdec3;hb=a5d110282a864fd2e91b51ce360a977cd0643657;hp=039aeed92c5a0fb3a6979ff727c9e064c4476904;hpb=bab14f0dd2345c9d7dcbc47c918563709e1ac094;p=user%2Fhenk%2Fcode%2Finspircd.git

diff --git a/src/modules/m_alias.cpp b/src/modules/m_alias.cpp
index 039aeed92..32fc80b64 100644
--- a/src/modules/m_alias.cpp
+++ b/src/modules/m_alias.cpp
@@ -1 +1,390 @@
-/*       +------------------------------------+
 *       | Inspire Internet Relay Chat Daemon |
 *       +------------------------------------+
 *
 *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
 * See: http://www.inspircd.org/wiki/index.php/Credits
 *
 * This program is free but copyrighted software; see
 *            the file COPYING for details.
 *
 * ---------------------------------------------------
 */

#include "inspircd.h"
#include "users.h"
#include "channels.h"
#include "modules.h"
#include "wildcard.h"

/* $ModDesc: Provides aliases of commands. */

/** An alias definition
 */
class Alias : public classbase
{
 public:
	/** The text of the alias command */
	irc::string text;
	/** Text to replace with */
	std::string replace_with;
	/** Nickname required to perform alias */
	std::string requires;
	/** Alias requires ulined server */
	bool uline;
	/** Requires oper? */
	bool operonly;
	/* is case sensitive params */
	bool case_sensitive;
	/** Format that must be matched for use */
	std::string format;
};

class ModuleAlias : public Module
{
 private:
	/** We cant use a map, there may be multiple aliases with the same name */
	std::vector<Alias> Aliases;
	std::map<std::string, int> AliasMap;
	std::vector<std::string> pars;

	virtual void ReadAliases()
	{
		ConfigReader MyConf(ServerInstance);

		Aliases.clear();
		AliasMap.clear();
		for (int i = 0; i < MyConf.Enumerate("alias"); i++)
		{
			Alias a;
			std::string txt;
			txt = MyConf.ReadValue("alias", "text", i);
			a.text = txt.c_str();
			a.replace_with = MyConf.ReadValue("alias", "replace", i, true);
			a.requires = MyConf.ReadValue("alias", "requires", i);
			a.uline = MyConf.ReadFlag("alias", "uline", i);
			a.operonly = MyConf.ReadFlag("alias", "operonly", i);
			a.format = MyConf.ReadValue("alias", "format", i);
			a.case_sensitive = MyConf.ReadFlag("alias", "matchcase", i);
			Aliases.push_back(a);
			AliasMap[txt] = 1;
		}
	}

 public:
	
	ModuleAlias(InspIRCd* Me)
		: Module(Me)
	{
		ReadAliases();
		pars.resize(127);
	}

	void Implements(char* List)
	{
		List[I_OnPreCommand] = List[I_OnRehash] = 1;
	}

	virtual ~ModuleAlias()
	{
	}

	virtual Version GetVersion()
	{
		return Version(1,1,0,1,VF_VENDOR,API_VERSION);
	}

	std::string GetVar(std::string varname, const std::string &original_line)
	{
		irc::spacesepstream ss(original_line);
		varname.erase(varname.begin());
		int index = *(varname.begin()) - 48;
		varname.erase(varname.begin());
		bool everything_after = (varname == "-");
		std::string word;

		for (int j = 0; j < index; j++)
			word = ss.GetToken();

		if (everything_after)
		{
			std::string more = "*";
			while ((more = ss.GetToken()) != "")
			{
				word.append(" ");
				word.append(more);
			}
		}

		return word;
	}

	void SearchAndReplace(std::string& newline, const std::string &find, const std::string &replace)
	{
		std::string::size_type x = newline.find(find);
		while (x != std::string::npos)
		{
			newline.erase(x, find.length());
			newline.insert(x, replace);
			x = newline.find(find);
		}
	}

	virtual int OnPreCommand(const std::string &command, const char** parameters, int pcnt, userrec *user, bool validated, const std::string &original_line)
	{
		userrec *u = NULL;

		/* If theyre not registered yet, we dont want
		 * to know.
		 */
		if (user->registered != REG_ALL)
			return 0;

		/* We dont have any commands looking like this, dont bother with the loop */
		if (AliasMap.find(command) == AliasMap.end())
			return 0;

		irc::string c = command.c_str();
		/* The parameters for the command in their original form, with the command stripped off */
		std::string compare = original_line.substr(command.length());
		while (*(compare.c_str()) == ' ')
			compare.erase(compare.begin());

		std::string safe(original_line);

		/* Escape out any $ symbols in the user provided text */

		SearchAndReplace(safe, "$", "\r");

		for (unsigned int i = 0; i < Aliases.size(); i++)
		{
			if (Aliases[i].text == c)
			{
				/* Does it match the pattern? */
				if (!Aliases[i].format.empty())
				{
					if (!match(Aliases[i].case_sensitive, compare.c_str(), Aliases[i].format.c_str()))
						continue;
				}

				if ((Aliases[i].operonly) && (!IS_OPER(user)))
					return 0;

				if (!Aliases[i].requires.empty())
				{
					u = ServerInstance->FindNick(Aliases[i].requires);
					if (!u)
					{
						user->WriteServ("401 "+std::string(user->nick)+" "+Aliases[i].requires+" :is currently unavailable. Please try again later.");
						return 1;
					}
				}
				if ((u != NULL) && (!Aliases[i].requires.empty()) && (Aliases[i].uline))
				{
					if (!ServerInstance->ULine(u->server))
					{
						ServerInstance->WriteOpers("*** NOTICE -- Service "+Aliases[i].requires+" required by alias "+std::string(Aliases[i].text.c_str())+" is not on a u-lined server, possibly underhanded antics detected!"); 
						user->WriteServ("401 "+std::string(user->nick)+" "+Aliases[i].requires+" :is an imposter! Please inform an IRC operator as soon as possible.");
						return 1;
					}
				}

				/* Now, search and replace in a copy of the original_line, replacing $1 through $9 and $1- etc */

				std::string::size_type crlf = Aliases[i].replace_with.find('\n');

				if (crlf == std::string::npos)
				{
					DoCommand(Aliases[i].replace_with, user, safe);
					return 1;
				}
				else
				{
					irc::sepstream commands(Aliases[i].replace_with, '\n');
					std::string command = "*";
					while ((command = commands.GetToken()) != "")
					{
						DoCommand(command, user, safe);
					}
					return 1;
				}
			}
		}
		return 0;
	}

	void DoCommand(std::string newline, userrec* user, const std::string &original_line)
	{
		for (int v = 1; v < 10; v++)
		{
			std::string var = "$";
			var.append(ConvToStr(v));
			var.append("-");
			std::string::size_type x = newline.find(var);

			while (x != std::string::npos)
			{
				newline.erase(x, var.length());
				newline.insert(x, GetVar(var, original_line));
				x = newline.find(var);
			}

			var = "$";
			var.append(ConvToStr(v));
			x = newline.find(var);

			while (x != std::string::npos)
			{
				newline.erase(x, var.length());
				newline.insert(x, GetVar(var, original_line));
				x = newline.find(var);
			}
		}

		/* Special variables */
		SearchAndReplace(newline, "$nick", user->nick);
		SearchAndReplace(newline, "$ident", user->ident);
		SearchAndReplace(newline, "$host", user->host);
		SearchAndReplace(newline, "$vhost", user->dhost);

		/* Unescape any variable names in the user text before sending */
		SearchAndReplace(newline, "\r", "$");

		irc::tokenstream ss(newline);
		const char* parv[127];
		int x = 0;

		while (ss.GetToken(pars[x]))
		{
			parv[x] = pars[x].c_str();
			x++;
		}

		ServerInstance->Parser->CallHandler(parv[0], &parv[1], x-1, user);
	}
 
	virtual void OnRehash(userrec* user, const std::string &parameter)
	{
		ReadAliases();
 	}
};

MODULE_INIT(ModuleAlias)
\ No newline at end of file
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
+ *
+ *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
+ *   Copyright (C) 2005, 2009 Robin Burchell <robin+git@viroteck.net>
+ *   Copyright (C) 2004-2007, 2009 Craig Edwards <craigedwards@brainbox.cc>
+ *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
+ *
+ * This file is part of InspIRCd.  InspIRCd is free software: you can
+ * redistribute it and/or modify it under the terms of the GNU General Public
+ * License as published by the Free Software Foundation, version 2.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+ * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
+ * details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+
+#include "inspircd.h"
+
+/* $ModDesc: Provides aliases of commands. */
+
+/** An alias definition
+ */
+class Alias
+{
+ public:
+	/** The text of the alias command */
+	irc::string AliasedCommand;
+
+	/** Text to replace with */
+	std::string ReplaceFormat;
+
+	/** Nickname required to perform alias */
+	std::string RequiredNick;
+
+	/** Alias requires ulined server */
+	bool ULineOnly;
+
+	/** Requires oper? */
+	bool OperOnly;
+
+	/* is case sensitive params */
+	bool CaseSensitive;
+
+	/* whether or not it may be executed via fantasy (default OFF) */
+	bool ChannelCommand;
+
+	/* whether or not it may be executed via /command (default ON) */
+	bool UserCommand;
+
+	/** Format that must be matched for use */
+	std::string format;
+};
+
+class ModuleAlias : public Module
+{
+ private:
+
+	char fprefix;
+
+	/* We cant use a map, there may be multiple aliases with the same name.
+	 * We can, however, use a fancy invention: the multimap. Maps a key to one or more values.
+	 *		-- w00t
+   */
+	std::multimap<irc::string, Alias> Aliases;
+
+	/* whether or not +B users are allowed to use fantasy commands */
+	bool AllowBots;
+
+	virtual void ReadAliases()
+	{
+		ConfigTag* fantasy = ServerInstance->Config->ConfValue("fantasy");
+		AllowBots = fantasy->getBool("allowbots", false);
+		std::string fpre = fantasy->getString("prefix", "!");
+		fprefix = fpre.empty() ? '!' : fpre[0];
+
+		Aliases.clear();
+		ConfigTagList tags = ServerInstance->Config->ConfTags("alias");
+		for(ConfigIter i = tags.first; i != tags.second; ++i)
+		{
+			ConfigTag* tag = i->second;
+			Alias a;
+			std::string aliastext = tag->getString("text");
+			a.AliasedCommand = aliastext.c_str();
+			tag->readString("replace", a.ReplaceFormat, true);
+			a.RequiredNick = tag->getString("requires");
+			a.ULineOnly = tag->getBool("uline");
+			a.ChannelCommand = tag->getBool("channelcommand", false);
+			a.UserCommand = tag->getBool("usercommand", true);
+			a.OperOnly = tag->getBool("operonly");
+			a.format = tag->getString("format");
+			a.CaseSensitive = tag->getBool("matchcase");
+			Aliases.insert(std::make_pair(a.AliasedCommand, a));
+		}
+	}
+
+ public:
+
+	void init()
+	{
+		ReadAliases();
+		Implementation eventlist[] = { I_OnPreCommand, I_OnRehash, I_OnUserMessage };
+		ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
+	}
+
+	virtual ~ModuleAlias()
+	{
+	}
+
+	virtual Version GetVersion()
+	{
+		return Version("Provides aliases of commands.", VF_VENDOR);
+	}
+
+	std::string GetVar(std::string varname, const std::string &original_line)
+	{
+		irc::spacesepstream ss(original_line);
+		varname.erase(varname.begin());
+		int index = *(varname.begin()) - 48;
+		varname.erase(varname.begin());
+		bool everything_after = (varname == "-");
+		std::string word;
+
+		for (int j = 0; j < index; j++)
+			ss.GetToken(word);
+
+		if (everything_after)
+		{
+			std::string more;
+			while (ss.GetToken(more))
+			{
+				word.append(" ");
+				word.append(more);
+			}
+		}
+
+		return word;
+	}
+
+	virtual ModResult OnPreCommand(std::string &command, std::vector<std::string> &parameters, LocalUser *user, bool validated, const std::string &original_line)
+	{
+		std::multimap<irc::string, Alias>::iterator i, upperbound;
+
+		/* If theyre not registered yet, we dont want
+		 * to know.
+		 */
+		if (user->registered != REG_ALL)
+			return MOD_RES_PASSTHRU;
+
+		/* We dont have any commands looking like this? Stop processing. */
+		i = Aliases.find(command.c_str());
+		if (i == Aliases.end())
+			return MOD_RES_PASSTHRU;
+		/* Avoid iterating on to different aliases if no patterns match. */
+		upperbound = Aliases.upper_bound(command.c_str());
+
+		irc::string c = command.c_str();
+		/* The parameters for the command in their original form, with the command stripped off */
+		std::string compare = original_line.substr(command.length());
+		while (*(compare.c_str()) == ' ')
+			compare.erase(compare.begin());
+
+		while (i != upperbound)
+		{
+			if (i->second.UserCommand)
+			{
+				if (DoAlias(user, NULL, &(i->second), compare, original_line))
+				{
+					return MOD_RES_DENY;
+				}
+			}
+
+			i++;
+		}
+
+		// If we made it here, no aliases actually matched.
+		return MOD_RES_PASSTHRU;
+	}
+
+	virtual void OnUserMessage(User *user, void *dest, int target_type, const std::string &text, char status, const CUList &exempt_list)
+	{
+		if (target_type != TYPE_CHANNEL)
+		{
+			return;
+		}
+
+		// fcommands are only for local users. Spanningtree will send them back out as their original cmd.
+		if (!user || !IS_LOCAL(user))
+		{
+			return;
+		}
+
+		/* Stop here if the user is +B and allowbot is set to no. */
+		if (!AllowBots && user->IsModeSet('B'))
+		{
+			return;
+		}
+
+		Channel *c = (Channel *)dest;
+		std::string scommand;
+
+		// text is like "!moo cows bite me", we want "!moo" first
+		irc::spacesepstream ss(text);
+		ss.GetToken(scommand);
+		irc::string fcommand = scommand.c_str();
+
+		if (fcommand.empty())
+		{
+			return; // wtfbbq
+		}
+
+		// we don't want to touch non-fantasy stuff
+		if (*fcommand.c_str() != fprefix)
+		{
+			return;
+		}
+
+		// nor do we give a shit about the prefix
+		fcommand.erase(fcommand.begin());
+
+		std::multimap<irc::string, Alias>::iterator i = Aliases.find(fcommand);
+
+		if (i == Aliases.end())
+			return;
+
+		/* Avoid iterating on to other aliases if no patterns match */
+		std::multimap<irc::string, Alias>::iterator upperbound = Aliases.upper_bound(fcommand);
+
+
+		/* The parameters for the command in their original form, with the command stripped off */
+		std::string compare = text.substr(fcommand.length() + 1);
+		while (*(compare.c_str()) == ' ')
+			compare.erase(compare.begin());
+
+		while (i != upperbound)
+		{
+			if (i->second.ChannelCommand)
+			{
+				// We use substr(1) here to remove the fantasy prefix
+				if (DoAlias(user, c, &(i->second), compare, text.substr(1)))
+					return;
+			}
+
+			i++;
+		}
+	}
+
+
+	int DoAlias(User *user, Channel *c, Alias *a, const std::string& compare, const std::string& safe)
+	{
+		User *u = NULL;
+
+		/* Does it match the pattern? */
+		if (!a->format.empty())
+		{
+			if (a->CaseSensitive)
+			{
+				if (!InspIRCd::Match(compare, a->format, rfc_case_sensitive_map))
+					return 0;
+			}
+			else
+			{
+				if (!InspIRCd::Match(compare, a->format))
+					return 0;
+			}
+		}
+
+		if ((a->OperOnly) && (!IS_OPER(user)))
+			return 0;
+
+		if (!a->RequiredNick.empty())
+		{
+			u = ServerInstance->FindNick(a->RequiredNick);
+			if (!u)
+			{
+				user->WriteNumeric(401, ""+user->nick+" "+a->RequiredNick+" :is currently unavailable. Please try again later.");
+				return 1;
+			}
+		}
+		if ((u != NULL) && (!a->RequiredNick.empty()) && (a->ULineOnly))
+		{
+			if (!ServerInstance->ULine(u->server))
+			{
+				ServerInstance->SNO->WriteToSnoMask('a', "NOTICE -- Service "+a->RequiredNick+" required by alias "+std::string(a->AliasedCommand.c_str())+" is not on a u-lined server, possibly underhanded antics detected!");
+				user->WriteNumeric(401, ""+user->nick+" "+a->RequiredNick+" :is an imposter! Please inform an IRC operator as soon as possible.");
+				return 1;
+			}
+		}
+
+		/* Now, search and replace in a copy of the original_line, replacing $1 through $9 and $1- etc */
+
+		std::string::size_type crlf = a->ReplaceFormat.find('\n');
+
+		if (crlf == std::string::npos)
+		{
+			DoCommand(a->ReplaceFormat, user, c, safe);
+			return 1;
+		}
+		else
+		{
+			irc::sepstream commands(a->ReplaceFormat, '\n');
+			std::string scommand;
+			while (commands.GetToken(scommand))
+			{
+				DoCommand(scommand, user, c, safe);
+			}
+			return 1;
+		}
+	}
+
+	void DoCommand(const std::string& newline, User* user, Channel *chan, const std::string &original_line)
+	{
+		std::string result;
+		result.reserve(MAXBUF);
+		for (unsigned int i = 0; i < newline.length(); i++)
+		{
+			char c = newline[i];
+			if ((c == '$') && (i + 1 < newline.length()))
+			{
+				if (isdigit(newline[i+1]))
+				{
+					int len = ((i + 2 < newline.length()) && (newline[i+2] == '-')) ? 3 : 2;
+					std::string var = newline.substr(i, len);
+					result.append(GetVar(var, original_line));
+					i += len - 1;
+				}
+				else if (newline.substr(i, 5) == "$nick")
+				{
+					result.append(user->nick);
+					i += 4;
+				}
+				else if (newline.substr(i, 5) == "$host")
+				{
+					result.append(user->host);
+					i += 4;
+				}
+				else if (newline.substr(i, 5) == "$chan")
+				{
+					if (chan)
+						result.append(chan->name);
+					i += 4;
+				}
+				else if (newline.substr(i, 6) == "$ident")
+				{
+					result.append(user->ident);
+					i += 5;
+				}
+				else if (newline.substr(i, 6) == "$vhost")
+				{
+					result.append(user->dhost);
+					i += 5;
+				}
+				else
+					result.push_back(c);
+			}
+			else
+				result.push_back(c);
+		}
+
+		irc::tokenstream ss(result);
+		std::vector<std::string> pars;
+		std::string command, token;
+
+		ss.GetToken(command);
+		while (ss.GetToken(token) && (pars.size() <= MAXPARAMETERS))
+		{
+			pars.push_back(token);
+		}
+		ServerInstance->Parser->CallHandler(command, pars, user);
+	}
+
+	virtual void OnRehash(User* user)
+	{
+		ReadAliases();
+ 	}
+
+	virtual void Prioritize()
+	{
+		// Prioritise after spanningtree so that channel aliases show the alias before the effects.
+		Module* linkmod = ServerInstance->Modules->Find("m_spanningtree.so");
+		ServerInstance->Modules->SetPriority(this, I_OnUserMessage, PRIORITY_AFTER, &linkmod);
+	}
+};
+
+MODULE_INIT(ModuleAlias)