]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_spanningtree.cpp
Fixes :P
[user/henk/code/inspircd.git] / src / modules / m_spanningtree.cpp
index e99bd814d4efcb2c78e7c6a0adcdf105085db295..78c37c5848f22354897eefe7427d0772afa4cea7 100644 (file)
@@ -169,6 +169,7 @@ class TreeServer
        TreeSocket* Socket;                     /* For directly connected servers this points at the socket object */
        time_t NextPing;                        /* After this time, the server should be PINGed*/
        bool LastPingWasGood;                   /* True if the server responded to the last PING with a PONG */
+       std::map<userrec*,userrec*> Users;      /* Users on this server */
        
  public:
 
@@ -265,6 +266,38 @@ class TreeServer
                this->AddHashEntry();
        }
 
+       void AddUser(userrec* user)
+       {
+               log(DEBUG,"Add user %s to server %s",user->nick,this->ServerName.c_str());
+               std::map<userrec*,userrec*>::iterator iter;
+               iter = Users.find(user);
+               if (iter == Users.end())
+                       Users[user] = user;
+       }
+
+       void DelUser(userrec* user)
+       {
+               log(DEBUG,"Remove user %s from server %s",user->nick,this->ServerName.c_str());
+               std::map<userrec*,userrec*>::iterator iter;
+               iter = Users.find(user);
+               if (iter != Users.end())
+                       Users.erase(iter);
+       }
+
+       int QuitUsers(const std::string &reason)
+       {
+               int x = Users.size();
+               log(DEBUG,"Removing all users from server %s",this->ServerName.c_str());
+               const char* reason_s = reason.c_str();
+               for (std::map<userrec*,userrec*>::iterator n = Users.begin(); n != Users.end(); n++)
+               {
+                       log(DEBUG,"Kill %s",n->second->nick);
+                       kill_link(n->second,reason_s);
+               }
+               Users.clear();
+               return x;
+       }
+
        /* This method is used to add the structure to the
         * hash_map for linear searches. It is only called
         * by the constructors.
@@ -468,13 +501,6 @@ class Link
 ConfigReader *Conf;
 std::vector<Link> LinkBlocks;
 
-template<typename T> inline string ConvToStr(const T &in)
-{
-       stringstream tmp;
-       if (!(tmp << in)) return string();
-       return tmp.str();
-}
-
 /* Yay for fast searches!
  * This is hundreds of times faster than recursion
  * or even scanning a linked list, especially when
@@ -809,7 +835,7 @@ class TreeSocket : public InspSocket
         * is having a REAL bad hair day, this function shouldnt be called
         * too many times a month ;-)
         */
-       void SquitServer(std::string &from, TreeServer* Current, CullList* Goners)
+       void SquitServer(std::string &from, TreeServer* Current)
        {
                /* recursively squit the servers attached to 'Current'.
                 * We're going backwards so we don't remove users
@@ -818,18 +844,19 @@ class TreeSocket : public InspSocket
                for (unsigned int q = 0; q < Current->ChildCount(); q++)
                {
                        TreeServer* recursive_server = Current->GetChild(q);
-                       this->SquitServer(from,recursive_server,Goners);
+                       this->SquitServer(from,recursive_server);
                }
                /* Now we've whacked the kids, whack self */
                num_lost_servers++;
-               for (user_hash::iterator u = clientlist.begin(); u != clientlist.end(); u++)
+               num_lost_users += Current->QuitUsers(from);
+               /*for (user_hash::iterator u = clientlist.begin(); u != clientlist.end(); u++)
                {
                        if (!strcasecmp(u->second->server,Current->GetName().c_str()))
                        {
                                Goners->AddItem(u->second,from);
                                num_lost_users++;
                        }
-               }
+               }*/
        }
 
        /* This is a wrapper function for SquitServer above, which
@@ -854,14 +881,11 @@ class TreeSocket : public InspSocket
                        }
                        num_lost_servers = 0;
                        num_lost_users = 0;
-                       CullList* Goners = new CullList();
                        std::string from = Current->GetParent()->GetName()+" "+Current->GetName();
-                       SquitServer(from, Current, Goners);
-                       Goners->Apply();
+                       SquitServer(from, Current);
                        Current->Tidy();
                        Current->GetParent()->DelChild(Current);
                        delete Current;
-                       delete Goners;
                        WriteOpers("Netsplit complete, lost \002%d\002 users on \002%d\002 servers.", num_lost_users, num_lost_servers);
                }
                else
@@ -1109,6 +1133,23 @@ class TreeSocket : public InspSocket
                clientlist[tempnick]->registered = 7;
                clientlist[tempnick]->signon = age;
                strlcpy(clientlist[tempnick]->modes, modes.c_str(),53);
+               for (char *v = clientlist[tempnick]->modes; *v; v++)
+               {
+                       switch (*v)
+                       {
+                               case 'i':
+                                       clientlist[tempnick]->modebits |= UM_INVISIBLE;
+                               break;
+                               case 'w':
+                                       clientlist[tempnick]->modebits |= UM_WALLOPS;
+                               break;
+                               case 's':
+                                       clientlist[tempnick]->modebits |= UM_SERVERNOTICE;
+                               break;
+                               default:
+                               break;
+                       }
+               }
                inet_aton(params[6].c_str(),&clientlist[tempnick]->ip4);
 
                WriteOpers("*** Client connecting at %s: %s!%s@%s [%s]",clientlist[tempnick]->server,clientlist[tempnick]->nick,clientlist[tempnick]->ident,clientlist[tempnick]->host,(char*)inet_ntoa(clientlist[tempnick]->ip4));
@@ -1120,6 +1161,7 @@ class TreeSocket : public InspSocket
                TreeServer* SourceServer = FindServer(source);
                if (SourceServer)
                {
+                       SourceServer->AddUser(clientlist[tempnick]);
                        SourceServer->AddUserCount();
                }
 
@@ -1135,8 +1177,11 @@ class TreeSocket : public InspSocket
                log(DEBUG,"Sending FJOINs to other server for %s",c->name);
                char list[MAXBUF];
                std::string individual_halfops = ":"+Srv->GetServerName()+" FMODE "+c->name;
-               size_t counter = snprintf(list,MAXBUF,":%s FJOIN %s %lu",Srv->GetServerName().c_str(),c->name,(unsigned long)c->age);
-               size_t initial = counter;
+               
+               size_t dlen, curlen;
+               dlen = curlen = snprintf(list,MAXBUF,":%s FJOIN %s %lu",Srv->GetServerName().c_str(),c->name,(unsigned long)c->age);
+               int numusers = 0;
+               char* ptr = list + dlen;
 
                CUList *ulist = c->GetUsers();
                std::vector<userrec*> specific_halfop;
@@ -1144,8 +1189,6 @@ class TreeSocket : public InspSocket
 
                for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)
                {
-                       charlcat(list,' ',MAXBUF);
-                       counter++;
                        int x = cflags(i->second,c);
                        if ((x & UCMODE_HOP) && (x & UCMODE_OP))
                        {
@@ -1156,33 +1199,34 @@ class TreeSocket : public InspSocket
                                specific_voice.push_back(i->second);
                        }
 
-                       char n = 0;
+                       const char* n = "";
                        if (x & UCMODE_OP)
                        {
-                               n = '@';
+                               n = "@";
                        }
                        else if (x & UCMODE_HOP)
                        {
-                               n = '%';
+                               n = "%";
                        }
                        else if (x & UCMODE_VOICE)
                        {
-                               n = '+';
+                               n = "+";
                        }
 
-                       if (n)
-                       {
-                               charlcat(list,n,MAXBUF);
-                               counter++;
-                       }
+                       size_t ptrlen = snprintf(ptr, MAXBUF, " %s%s", n, i->second->nick);
 
-                       counter += strlcat(list,i->second->nick,MAXBUF);
+                       curlen += ptrlen;
+                       ptr += ptrlen;
 
-                       if (counter > (480-NICKMAX))
+                       numusers++;
+
+                       if (curlen > (480-NICKMAX))
                        {
-                               log(DEBUG,"FJOIN line wrapped");
                                this->WriteLine(list);
-                               counter = snprintf(list,MAXBUF,":%s FJOIN %s %lu",Srv->GetServerName().c_str(),c->name,(unsigned long)c->age);
+                               dlen = curlen = snprintf(list,MAXBUF,":%s FJOIN %s %lu",Srv->GetServerName().c_str(),c->name,(unsigned long)c->age);
+                               ptr = list + dlen;
+                               ptrlen = 0;
+                               numusers = 0;
                                for (unsigned int y = 0; y < specific_voice.size(); y++)
                                {
                                        this->WriteLine(":"+Srv->GetServerName()+" FMODE "+c->name+" +v "+specific_voice[y]->nick);
@@ -1193,9 +1237,8 @@ class TreeSocket : public InspSocket
                                }
                        }
                }
-               if (counter != initial)
+               if (numusers)
                {
-                       log(DEBUG,"Final FJOIN line");
                        this->WriteLine(list);
                        for (unsigned int y = 0; y < specific_voice.size(); y++)
                        {
@@ -1367,26 +1410,32 @@ class TreeSocket : public InspSocket
                 */
                std::string burst = "BURST "+ConvToStr(time(NULL));
                std::string endburst = "ENDBURST";
-               Srv->SendOpers("*** Bursting to \2"+s->GetName()+"\2.");
+               // Because by the end of the netburst, it  could be gone!
+               std::string name = s->GetName();
+               Srv->SendOpers("*** Bursting to \2"+name+"\2.");
                this->WriteLine(burst);
                ServerInstance->DoOneIteration(false);
                /* send our version string */
                this->WriteLine(":"+Srv->GetServerName()+" VERSION :"+Srv->GetVersion());
                /* Send server tree */
-               this->SendServers(TreeRoot,s,1);
+               if (FindServer(name))
+                       this->SendServers(TreeRoot,s,1);
                ServerInstance->DoOneIteration(false);
                /* Send users and their oper status */
-               this->SendUsers(s);
+               if (FindServer(name))
+                       this->SendUsers(s);
                ServerInstance->DoOneIteration(false);
                /* Send everything else (channel modes, xlines etc) */
-               this->SendChannelModes(s);
+               if (FindServer(name))
+                       this->SendChannelModes(s);
                ServerInstance->DoOneIteration(false);
-               this->SendXLines(s);
+               if (FindServer(name))
+                       this->SendXLines(s);
                ServerInstance->DoOneIteration(false);
                FOREACH_MOD(I_OnSyncOtherMetaData,OnSyncOtherMetaData((Module*)TreeProtocolModule,(void*)this));
                ServerInstance->DoOneIteration(false);
                this->WriteLine(endburst);
-               Srv->SendOpers("*** Finished bursting to \2"+s->GetName()+"\2.");
+               Srv->SendOpers("*** Finished bursting to \2"+name+"\2.");
        }
 
        /* This function is called when we receive data from a remote
@@ -2533,7 +2582,15 @@ class TreeSocket : public InspSocket
                                        }
                                        if (who)
                                        {
-                                               if ((command == "NICK") && (params.size() > 0))
+                                               if (command == "QUIT")
+                                               {
+                                                       TreeServer* s = FindServer(who->server);
+                                                       if (s)
+                                                       {
+                                                               s->DelUser(who);
+                                                       }
+                                               }
+                                               else if ((command == "NICK") && (params.size() > 0))
                                                {
                                                        /* On nick messages, check that the nick doesnt
                                                         * already exist here. If it does, kill their copy,
@@ -3310,7 +3367,7 @@ class ModuleSpanningTree : public Module
                return 0;
        }
 
-       virtual int OnPreCommand(std::string command, char **parameters, int pcnt, userrec *user, bool validated)
+       virtual int OnPreCommand(const std::string &command, char **parameters, int pcnt, userrec *user, bool validated)
        {
                /* If the command doesnt appear to be valid, we dont want to mess with it. */
                if (!validated)
@@ -3382,7 +3439,7 @@ class ModuleSpanningTree : public Module
                return 0;
        }
 
-       virtual void OnGetServerDescription(std::string servername,std::string &description)
+       virtual void OnGetServerDescription(const std::string &servername,std::string &description)
        {
                TreeServer* s = FindServer(servername);
                if (s)
@@ -3402,7 +3459,7 @@ class ModuleSpanningTree : public Module
                }
        }
 
-       virtual void OnPostLocalTopicChange(userrec* user, chanrec* chan, std::string topic)
+       virtual void OnPostLocalTopicChange(userrec* user, chanrec* chan, const std::string &topic)
        {
                std::deque<std::string> params;
                params.push_back(chan->name);
@@ -3410,7 +3467,7 @@ class ModuleSpanningTree : public Module
                DoOneToMany(user->nick,"TOPIC",params);
        }
 
-       virtual void OnWallops(userrec* user, std::string text)
+       virtual void OnWallops(userrec* user, const std::string &text)
        {
                if (user->fd > -1)
                {
@@ -3420,7 +3477,7 @@ class ModuleSpanningTree : public Module
                }
        }
 
-       virtual void OnUserNotice(userrec* user, void* dest, int target_type, std::string text, char status)
+       virtual void OnUserNotice(userrec* user, void* dest, int target_type, const std::string &text, char status)
        {
                if (target_type == TYPE_USER)
                {
@@ -3455,7 +3512,7 @@ class ModuleSpanningTree : public Module
                }
        }
 
-       virtual void OnUserMessage(userrec* user, void* dest, int target_type, std::string text, char status)
+       virtual void OnUserMessage(userrec* user, void* dest, int target_type, const std::string &text, char status)
        {
                if (target_type == TYPE_USER)
                {
@@ -3531,7 +3588,7 @@ class ModuleSpanningTree : public Module
                }
        }
 
-       virtual void OnChangeHost(userrec* user, std::string newhost)
+       virtual void OnChangeHost(userrec* user, const std::string &newhost)
        {
                // only occurs for local clients
                if (user->registered != 7)
@@ -3541,7 +3598,7 @@ class ModuleSpanningTree : public Module
                DoOneToMany(user->nick,"FHOST",params);
        }
 
-       virtual void OnChangeName(userrec* user, std::string gecos)
+       virtual void OnChangeName(userrec* user, const std::string &gecos)
        {
                // only occurs for local clients
                if (user->registered != 7)
@@ -3551,7 +3608,7 @@ class ModuleSpanningTree : public Module
                DoOneToMany(user->nick,"FNAME",params);
        }
 
-       virtual void OnUserPart(userrec* user, chanrec* channel, std::string partmessage)
+       virtual void OnUserPart(userrec* user, chanrec* channel, const std::string &partmessage)
        {
                if (user->fd > -1)
                {
@@ -3590,7 +3647,7 @@ class ModuleSpanningTree : public Module
                }
        }
 
-       virtual void OnUserQuit(userrec* user, std::string reason)
+       virtual void OnUserQuit(userrec* user, const std::string &reason)
        {
                if ((user->fd > -1) && (user->registered == 7))
                {
@@ -3607,7 +3664,7 @@ class ModuleSpanningTree : public Module
 
        }
 
-       virtual void OnUserPostNick(userrec* user, std::string oldnick)
+       virtual void OnUserPostNick(userrec* user, const std::string &oldnick)
        {
                if (user->fd > -1)
                {
@@ -3617,7 +3674,7 @@ class ModuleSpanningTree : public Module
                }
        }
 
-       virtual void OnUserKick(userrec* source, userrec* user, chanrec* chan, std::string reason)
+       virtual void OnUserKick(userrec* source, userrec* user, chanrec* chan, const std::string &reason)
        {
                if ((source) && (source->fd > -1))
                {
@@ -3637,7 +3694,7 @@ class ModuleSpanningTree : public Module
                }
        }
 
-       virtual void OnRemoteKill(userrec* source, userrec* dest, std::string reason)
+       virtual void OnRemoteKill(userrec* source, userrec* dest, const std::string &reason)
        {
                std::deque<std::string> params;
                params.push_back(dest->nick);
@@ -3645,7 +3702,7 @@ class ModuleSpanningTree : public Module
                DoOneToMany(source->nick,"KILL",params);
        }
 
-       virtual void OnRehash(std::string parameter)
+       virtual void OnRehash(const std::string &parameter)
        {
                if (parameter != "")
                {
@@ -3665,7 +3722,7 @@ class ModuleSpanningTree : public Module
        // note: the protocol does not allow direct umode +o except
        // via NICK with 8 params. sending OPERTYPE infers +o modechange
        // locally.
-       virtual void OnOper(userrec* user, std::string opertype)
+       virtual void OnOper(userrec* user, const std::string &opertype)
        {
                if (user->fd > -1)
                {
@@ -3675,7 +3732,7 @@ class ModuleSpanningTree : public Module
                }
        }
 
-       void OnLine(userrec* source, std::string host, bool adding, char linetype, long duration, std::string reason)
+       void OnLine(userrec* source, const std::string &host, bool adding, char linetype, long duration, const std::string &reason)
        {
                if (source->fd > -1)
                {
@@ -3701,47 +3758,47 @@ class ModuleSpanningTree : public Module
                }
        }
 
-       virtual void OnAddGLine(long duration, userrec* source, std::string reason, std::string hostmask)
+       virtual void OnAddGLine(long duration, userrec* source, const std::string &reason, const std::string &hostmask)
        {
                OnLine(source,hostmask,true,'G',duration,reason);
        }
        
-       virtual void OnAddZLine(long duration, userrec* source, std::string reason, std::string ipmask)
+       virtual void OnAddZLine(long duration, userrec* source, const std::string &reason, const std::string &ipmask)
        {
                OnLine(source,ipmask,true,'Z',duration,reason);
        }
 
-       virtual void OnAddQLine(long duration, userrec* source, std::string reason, std::string nickmask)
+       virtual void OnAddQLine(long duration, userrec* source, const std::string &reason, const std::string &nickmask)
        {
                OnLine(source,nickmask,true,'Q',duration,reason);
        }
 
-       virtual void OnAddELine(long duration, userrec* source, std::string reason, std::string hostmask)
+       virtual void OnAddELine(long duration, userrec* source, const std::string &reason, const std::string &hostmask)
        {
                OnLine(source,hostmask,true,'E',duration,reason);
        }
 
-       virtual void OnDelGLine(userrec* source, std::string hostmask)
+       virtual void OnDelGLine(userrec* source, const std::string &hostmask)
        {
                OnLine(source,hostmask,false,'G',0,"");
        }
 
-       virtual void OnDelZLine(userrec* source, std::string ipmask)
+       virtual void OnDelZLine(userrec* source, const std::string &ipmask)
        {
                OnLine(source,ipmask,false,'Z',0,"");
        }
 
-       virtual void OnDelQLine(userrec* source, std::string nickmask)
+       virtual void OnDelQLine(userrec* source, const std::string &nickmask)
        {
                OnLine(source,nickmask,false,'Q',0,"");
        }
 
-       virtual void OnDelELine(userrec* source, std::string hostmask)
+       virtual void OnDelELine(userrec* source, const std::string &hostmask)
        {
                OnLine(source,hostmask,false,'E',0,"");
        }
 
-       virtual void OnMode(userrec* user, void* dest, int target_type, std::string text)
+       virtual void OnMode(userrec* user, void* dest, int target_type, const std::string &text)
        {
                if ((user->fd > -1) && (user->registered == 7))
                {
@@ -3784,7 +3841,7 @@ class ModuleSpanningTree : public Module
                }
        }
 
-       virtual void ProtoSendMode(void* opaque, int target_type, void* target, std::string modeline)
+       virtual void ProtoSendMode(void* opaque, int target_type, void* target, const std::string &modeline)
        {
                TreeSocket* s = (TreeSocket*)opaque;
                if (target)
@@ -3802,7 +3859,7 @@ class ModuleSpanningTree : public Module
                }
        }
 
-       virtual void ProtoSendMetaData(void* opaque, int target_type, void* target, std::string extname, std::string extdata)
+       virtual void ProtoSendMetaData(void* opaque, int target_type, void* target, const std::string &extname, const std::string &extdata)
        {
                TreeSocket* s = (TreeSocket*)opaque;
                if (target)