diff options
Diffstat (limited to 'modules/spanningtree/postcommand.cpp')
| -rw-r--r-- | modules/spanningtree/postcommand.cpp | 135 |
1 files changed, 135 insertions, 0 deletions
diff --git a/modules/spanningtree/postcommand.cpp b/modules/spanningtree/postcommand.cpp new file mode 100644 index 000000000..95cb5832a --- /dev/null +++ b/modules/spanningtree/postcommand.cpp @@ -0,0 +1,135 @@ +/* + * InspIRCd -- Internet Relay Chat Daemon + * + * Copyright (C) 2018-2024 Sadie Powell <sadie@witchery.services> + * Copyright (C) 2013-2016 Attila Molnar <attilamolnar@hush.com> + * Copyright (C) 2012 Robby <robby@chatbelgie.be> + * Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org> + * Copyright (C) 2007 Craig Edwards <brain@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" + +#include "main.h" +#include "utils.h" +#include "treeserver.h" +#include "commandbuilder.h" + +void ModuleSpanningTree::OnPostCommand(Command* command, const CommandBase::Params& parameters, LocalUser* user, CmdResult result, bool loop) +{ + if (result == CmdResult::SUCCESS) + Utils->RouteCommand(nullptr, command, parameters, user); +} + +void SpanningTreeUtilities::RouteCommand(TreeServer* origin, CommandBase* thiscmd, const CommandBase::Params& parameters, User* user) +{ + RouteDescriptor routing = thiscmd->GetRouting(user, parameters); + if (routing.type == RouteType::LOCAL) + return; + + const std::string& command = thiscmd->name; + const bool encap = ((routing.type == RouteType::OPTIONAL_BROADCAST) || (routing.type == RouteType::OPTIONAL_UNICAST)); + CmdBuilder params(user, encap ? "ENCAP" : command.c_str()); + params.push_tags(parameters.GetTags()); + const TreeServer* sdest = nullptr; + + if (routing.type == RouteType::OPTIONAL_BROADCAST) + { + params.push('*'); + params.push(command); + } + else if (routing.type == RouteType::UNICAST || routing.type == RouteType::OPTIONAL_UNICAST) + { + sdest = static_cast<const TreeServer*>(routing.server); + if (!sdest) + { + // Assume the command handler already validated routing.target and have only returned success if the target is something that the + // user executing the command is allowed to look up e.g. target is not an uuid if user is local. + sdest = FindRouteTarget(routing.target); + if (!sdest) + { + ServerInstance->Logs.Debug(MODNAME, "Trying to route {}{} to nonexistent server {}", (encap ? "ENCAP " : ""), command, routing.target); + return; + } + } + + if (encap) + { + params.push(sdest->GetId()); + params.push(command); + } + } + else + { + Module* srcmodule = thiscmd->creator; + if (!(srcmodule->properties & (VF_COMMON | VF_CORE)) && srcmodule != Creator) + { + ServerInstance->Logs.Normal(MODNAME, "Routed command {} from non-VF_COMMON module {}", + command, srcmodule->ModuleFile); + return; + } + } + + std::string output_text = CommandParser::TranslateUIDs(thiscmd->translation, parameters, true, thiscmd); + + params.push(output_text); + + if (routing.type == RouteType::MESSAGE) + { + char pfx = 0; + std::string dest = routing.target; + if (ServerInstance->Modes.FindPrefix(dest[0])) + { + pfx = dest[0]; + dest.erase(dest.begin()); + } + if (ServerInstance->Channels.IsPrefix(dest[0])) + { + auto* c = ServerInstance->Channels.Find(dest); + if (!c) + return; + // TODO OnBuildExemptList hook was here + CUList exempts; + std::string message; + if (parameters.size() >= 2) + message.assign(parameters[1]); + SendChannelMessage(user, c, message, pfx, parameters.GetTags(), exempts, command.c_str(), origin ? origin->GetSocket() : nullptr); + } + else if (dest[0] == '$') + { + params.Forward(origin); + } + else + { + // user target? + auto* d = ServerInstance->Users.Find(dest); + if (!d || IS_LOCAL(d)) + return; + TreeServer* tsd = TreeServer::Get(d)->GetRoute(); + if (tsd == origin) + // huh? no routing stuff around in a circle, please. + return; + params.Unicast(d); + } + } + else if (routing.type == RouteType::BROADCAST || routing.type == RouteType::OPTIONAL_BROADCAST) + { + params.Forward(origin); + } + else if (routing.type == RouteType::UNICAST || routing.type == RouteType::OPTIONAL_UNICAST) + { + params.Unicast(sdest->ServerUser); + } +} |