]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_spanningtree/postcommand.cpp
Fix some regressions in sending tags between servers.
[user/henk/code/inspircd.git] / src / modules / m_spanningtree / postcommand.cpp
index 44476c8c771d4611dd8c49dfb9f659213de9fc13..10cbbce82f92d8a34312549c34d530501cbdef52 100644 (file)
-/*       +------------------------------------+
- *       | Inspire Internet Relay Chat Daemon |
- *       +------------------------------------+
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
  *
- *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
- * See: http://www.inspircd.org/wiki/index.php/Credits
+ *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
+ *   Copyright (C) 2007 Craig Edwards <craigedwards@brainbox.cc>
  *
- * This program is free but copyrighted software; see
- *          the file COPYING for details.
+ * 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/>.
  */
 
-/* $ModDesc: Provides a spanning tree server link protocol */
-                
 #include "inspircd.h"
-#include "commands/cmd_whois.h"
-#include "commands/cmd_stats.h"
-#include "socket.h"
-#include "wildcard.h"
-#include "xline.h"      
-#include "transport.h"  
-                        
-#include "m_spanningtree/timesynctimer.h"
-#include "m_spanningtree/resolvers.h"
-#include "m_spanningtree/main.h"
-#include "m_spanningtree/utils.h"
-#include "m_spanningtree/treeserver.h"
-#include "m_spanningtree/link.h"
-#include "m_spanningtree/treesocket.h"
-#include "m_spanningtree/rconnect.h"
-#include "m_spanningtree/rsquit.h"
 
-/* $ModDep: m_spanningtree/timesynctimer.h m_spanningtree/resolvers.h m_spanningtree/main.h m_spanningtree/utils.h m_spanningtree/treeserver.h m_spanningtree/link.h m_spanningtree/treesocket.h m_spanningtree/rconnect.h m_spanningtree/rsquit.h */
+#include "main.h"
+#include "utils.h"
+#include "treeserver.h"
+#include "commandbuilder.h"
 
-void ModuleSpanningTree::OnPostCommand(const std::string &command, const char** parameters, int pcnt, userrec *user, CmdResult result, const std::string &original_line)
+void ModuleSpanningTree::OnPostCommand(Command* command, const CommandBase::Params& parameters, LocalUser* user, CmdResult result, bool loop)
 {
-        if ((result == CMD_SUCCESS) && (ServerInstance->IsValidModuleCommand(command, pcnt, user)))
-        {
-                /* Safe, we know its non-null because IsValidModuleCommand returned true */
-                command_t* thiscmd = ServerInstance->Parser->GetHandler(command);
-                // this bit of code cleverly routes all module commands
-                // to all remote severs *automatically* so that modules
-                // can just handle commands locally, without having
-                // to have any special provision in place for remote
-                // commands and linking protocols.
-                std::deque<std::string> params;
-                params.clear();
-                int n_translate = thiscmd->translation.size();
-                TranslateType translate_to;
+       if (result == CMD_SUCCESS)
+               Utils->RouteCommand(NULL, command, parameters, user);
+}
 
-                /* To make sure that parameters with spaces, or empty
-                 * parameters, etc, are always sent properly, *always*
-                 * prefix the last parameter with a :. This also removes
-                 * an extra strchr() */
-                for (int j = 0; j < pcnt; j++)
-                {
-                        std::string target;
+void SpanningTreeUtilities::RouteCommand(TreeServer* origin, CommandBase* thiscmd, const CommandBase::Params& parameters, User* user)
+{
+       const std::string& command = thiscmd->name;
+       RouteDescriptor routing = thiscmd->GetRouting(user, parameters);
+       if (routing.type == ROUTE_TYPE_LOCALONLY)
+               return;
 
-                        /* Map all items to UUIDs where neccessary */
-                        if (j < n_translate)
-                        {
-                                /* We have a translation mapping for this index */
-                                translate_to = thiscmd->translation[j] != TR_END ? thiscmd->translation[j] : TR_TEXT;
-                        }
-                        else
-                                translate_to = TR_TEXT;
+       const bool encap = ((routing.type == ROUTE_TYPE_OPT_BCAST) || (routing.type == ROUTE_TYPE_OPT_UCAST));
+       CmdBuilder params(user, encap ? "ENCAP" : command.c_str());
+       params.push_tags(parameters.GetTags());
+       TreeServer* sdest = NULL;
 
-                        ServerInstance->Log(DEBUG,"TRANSLATION: %s - type is %d", parameters[j], translate_to);
-                        ServerInstance->Parser->TranslateUIDs(translate_to, parameters[j], target);
-                        
-                        if (j == (pcnt - 1))
-                                params.push_back(":" + target);
-                        else
-                                params.push_back(target);
-                }
-                Utils->DoOneToMany(user->uuid, command, params);
-        }
-}
+       if (routing.type == ROUTE_TYPE_OPT_BCAST)
+       {
+               params.push('*');
+               params.push(command);
+       }
+       else if (routing.type == ROUTE_TYPE_UNICAST || routing.type == ROUTE_TYPE_OPT_UCAST)
+       {
+               sdest = static_cast<TreeServer*>(routing.server);
+               if (!sdest)
+               {
+                       // Assume the command handler already validated routing.serverdest 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.serverdest);
+                       if (!sdest)
+                       {
+                               ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "Trying to route %s%s to nonexistent server %s", (encap ? "ENCAP " : ""), command.c_str(), routing.serverdest.c_str());
+                               return;
+                       }
+               }
+
+               if (encap)
+               {
+                       params.push(sdest->GetID());
+                       params.push(command);
+               }
+       }
+       else
+       {
+               Module* srcmodule = thiscmd->creator;
+               Version ver = srcmodule->GetVersion();
 
+               if (!(ver.Flags & (VF_COMMON | VF_CORE)) && srcmodule != Creator)
+               {
+                       ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "Routed command %s from non-VF_COMMON module %s",
+                               command.c_str(), srcmodule->ModuleSourceFile.c_str());
+                       return;
+               }
+       }
+
+       std::string output_text = CommandParser::TranslateUIDs(thiscmd->translation, parameters, true, thiscmd);
+
+       params.push(output_text);
+
+       if (routing.type == ROUTE_TYPE_MESSAGE)
+       {
+               char pfx = 0;
+               std::string dest = routing.serverdest;
+               if (ServerInstance->Modes->FindPrefix(dest[0]))
+               {
+                       pfx = dest[0];
+                       dest.erase(dest.begin());
+               }
+               if (dest[0] == '#')
+               {
+                       Channel* c = ServerInstance->FindChan(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() : NULL);
+               }
+               else if (dest[0] == '$')
+               {
+                       params.Forward(origin);
+               }
+               else
+               {
+                       // user target?
+                       User* d = ServerInstance->FindNick(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 == ROUTE_TYPE_BROADCAST || routing.type == ROUTE_TYPE_OPT_BCAST)
+       {
+               params.Forward(origin);
+       }
+       else if (routing.type == ROUTE_TYPE_UNICAST || routing.type == ROUTE_TYPE_OPT_UCAST)
+       {
+               params.Unicast(sdest->ServerUser);
+       }
+}