]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/users.cpp
Add Module::init() for correct exception handling during hook registration
[user/henk/code/inspircd.git] / src / users.cpp
index 36b41fb189d2b711ab8dec822a6533f502560197..c2890f9f723e1b7ffe071a277347075d7bf3913c 100644 (file)
@@ -2,7 +2,7 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
+ *  InspIRCd: (C) 2002-2010 InspIRCd Development Team
  * See: http://wiki.inspircd.org/Credits
  *
  * This program is free but copyrighted software; see
@@ -104,7 +104,7 @@ std::string User::ProcessNoticeMasks(const char *sm)
                        break;
                }
 
-               *c++;
+               c++;
        }
 
        std::string s = this->FormatNoticeMasks();
@@ -203,32 +203,13 @@ const char* User::FormatModes(bool showparameters)
        return data;
 }
 
-void User::DecrementModes()
-{
-       ServerInstance->Logs->Log("USERS", DEBUG, "DecrementModes()");
-       for (unsigned char n = 'A'; n <= 'z'; n++)
-       {
-               if (modes[n-65])
-               {
-                       ServerInstance->Logs->Log("USERS", DEBUG,"DecrementModes() found mode %c", n);
-                       ModeHandler* mh = ServerInstance->Modes->FindMode(n, MODETYPE_USER);
-                       if (mh)
-                       {
-                               ServerInstance->Logs->Log("USERS", DEBUG,"Found handler %c and call ChangeCount", n);
-                               mh->ChangeCount(-1);
-                       }
-               }
-       }
-}
-
-User::User(const std::string &uid, const std::string& sid)
-       : uuid(uid), server(sid)
+User::User(const std::string &uid, const std::string& sid, int type)
+       : uuid(uid), server(sid), usertype(type)
 {
        age = ServerInstance->Time();
        signon = idle_lastmsg = 0;
        registered = 0;
        quietquit = quitting = exempt = dns_done = false;
-       fd = -1;
        client_sa.sa.sa_family = AF_UNSPEC;
 
        ServerInstance->Logs->Log("USERS", DEBUG, "New UUID for user: %s", uuid.c_str());
@@ -240,12 +221,16 @@ User::User(const std::string &uid, const std::string& sid)
                throw CoreException("Duplicate UUID "+std::string(uuid)+" in User constructor");
 }
 
-LocalUser::LocalUser() : User(ServerInstance->GetUID(), ServerInstance->Config->ServerName)
+LocalUser::LocalUser(int myfd, irc::sockets::sockaddrs* client, irc::sockets::sockaddrs* servaddr)
+       : User(ServerInstance->GetUID(), ServerInstance->Config->ServerName, USERTYPE_LOCAL), eh(this)
 {
        bytes_in = bytes_out = cmds_in = cmds_out = 0;
        server_sa.sa.sa_family = AF_UNSPEC;
-       Penalty = 0;
+       CommandFloodPenalty = 0;
        lastping = nping = 0;
+       eh.SetFd(myfd);
+       memcpy(&client_sa, client, sizeof(irc::sockets::sockaddrs));
+       memcpy(&server_sa, servaddr, sizeof(irc::sockets::sockaddrs));
 }
 
 User::~User()
@@ -492,29 +477,25 @@ bool LocalUser::HasPrivPermission(const std::string &privstr, bool noisy)
        return false;
 }
 
-void User::OnDataReady()
+void UserIOHandler::OnDataReady()
 {
-}
-
-void LocalUser::OnDataReady()
-{
-       if (quitting)
+       if (user->quitting)
                return;
 
-       if (recvq.length() > MyClass->GetRecvqMax() && !HasPrivPermission("users/flood/increased-buffers"))
+       if (recvq.length() > user->MyClass->GetRecvqMax() && !user->HasPrivPermission("users/flood/increased-buffers"))
        {
-               ServerInstance->Users->QuitUser(this, "RecvQ exceeded");
+               ServerInstance->Users->QuitUser(user, "RecvQ exceeded");
                ServerInstance->SNO->WriteToSnoMask('a', "User %s RecvQ of %lu exceeds connect class maximum of %lu",
-                       nick.c_str(), (unsigned long)recvq.length(), MyClass->GetRecvqMax());
+                       user->nick.c_str(), (unsigned long)recvq.length(), user->MyClass->GetRecvqMax());
        }
        unsigned long sendqmax = ULONG_MAX;
-       if (!HasPrivPermission("users/flood/increased-buffers"))
-               sendqmax = MyClass->GetSendqSoftMax();
-       int penaltymax = MyClass->GetPenaltyThreshold();
-       if (penaltymax == 0 || HasPrivPermission("users/flood/no-fakelag"))
-               penaltymax = INT_MAX;
+       if (!user->HasPrivPermission("users/flood/increased-buffers"))
+               sendqmax = user->MyClass->GetSendqSoftMax();
+       unsigned long penaltymax = ULONG_MAX;
+       if (!user->HasPrivPermission("users/flood/no-fakelag"))
+               penaltymax = user->MyClass->GetPenaltyThreshold() * 1000;
 
-       while (Penalty < penaltymax && getSendQSize() < sendqmax)
+       while (user->CommandFloodPenalty < penaltymax && getSendQSize() < sendqmax)
        {
                std::string line;
                line.reserve(MAXBUF);
@@ -543,29 +524,32 @@ eol_found:
 
                // TODO should this be moved to when it was inserted in recvq?
                ServerInstance->stats->statsRecv += qpos;
-               this->bytes_in += qpos;
-               this->cmds_in++;
+               user->bytes_in += qpos;
+               user->cmds_in++;
 
-               ServerInstance->Parser->ProcessBuffer(line, this);
-               if (quitting)
+               ServerInstance->Parser->ProcessBuffer(line, user);
+               if (user->quitting)
                        return;
        }
        // Add pseudo-penalty so that we continue processing after sendq recedes
-       if (Penalty == 0 && getSendQSize() >= sendqmax)
-               Penalty++;
+       if (user->CommandFloodPenalty == 0 && getSendQSize() >= sendqmax)
+               user->CommandFloodPenalty++;
+       if (user->CommandFloodPenalty >= penaltymax && !user->MyClass->fakelag)
+               ServerInstance->Users->QuitUser(user, "Excess Flood");
 }
 
-void LocalUser::AddWriteBuf(const std::string &data)
+void UserIOHandler::AddWriteBuf(const std::string &data)
 {
-       if (!quitting && getSendQSize() + data.length() > MyClass->GetSendqHardMax() && !HasPrivPermission("users/flood/increased-buffers"))
+       if (!user->quitting && getSendQSize() + data.length() > user->MyClass->GetSendqHardMax() &&
+               !user->HasPrivPermission("users/flood/increased-buffers"))
        {
                /*
                 * Quit the user FIRST, because otherwise we could recurse
                 * here and hit the same limit.
                 */
-               ServerInstance->Users->QuitUser(this, "SendQ exceeded");
+               ServerInstance->Users->QuitUser(user, "SendQ exceeded");
                ServerInstance->SNO->WriteToSnoMask('a', "User %s SendQ exceeds connect class maximum of %lu",
-                       nick.c_str(), MyClass->GetSendqHardMax());
+                       user->nick.c_str(), user->MyClass->GetSendqHardMax());
                return;
        }
 
@@ -575,9 +559,9 @@ void LocalUser::AddWriteBuf(const std::string &data)
        WriteData(data);
 }
 
-void User::OnError(BufferedSocketError)
+void UserIOHandler::OnError(BufferedSocketError)
 {
-       ServerInstance->Users->QuitUser(this, getError());
+       ServerInstance->Users->QuitUser(user, getError());
 }
 
 CullResult User::cull()
@@ -585,11 +569,11 @@ CullResult User::cull()
        if (!quitting)
                ServerInstance->Users->QuitUser(this, "Culled without QuitUser");
        PurgeEmptyChannels();
-       if (IS_LOCAL(this) && fd != INT_MAX)
-               Close();
 
        this->InvalidateCache();
-       this->DecrementModes();
+
+       if (client_sa.sa.sa_family != AF_UNSPEC)
+               ServerInstance->Users->RemoveCloneCounts(this);
 
        ServerInstance->Users->uuidlist->erase(uuid);
        return Extensible::cull();
@@ -603,8 +587,7 @@ CullResult LocalUser::cull()
        else
                ServerInstance->Logs->Log("USERS", DEBUG, "Failed to remove user from vector");
 
-       if (client_sa.sa.sa_family != AF_UNSPEC)
-               ServerInstance->Users->RemoveCloneCounts(this);
+       eh.cull();
        return User::cull();
 }
 
@@ -965,7 +948,7 @@ const char* User::GetIPString()
        return cachedip.c_str();
 }
 
-irc::string User::GetCIDRMask()
+irc::sockets::cidr_mask User::GetCIDRMask()
 {
        int range = 0;
        switch (client_sa.sa.sa_family)
@@ -977,7 +960,7 @@ irc::string User::GetCIDRMask()
                        range = ServerInstance->Config->c_ipv4_range;
                        break;
        }
-       return assign(irc::sockets::mask(client_sa, range));
+       return irc::sockets::cidr_mask(client_sa, range);
 }
 
 bool User::SetClientIP(const char* sip)
@@ -998,7 +981,7 @@ void User::Write(const char *text, ...)
 
 void LocalUser::Write(const std::string& text)
 {
-       if (!ServerInstance->SE->BoundsCheckFd(this))
+       if (!ServerInstance->SE->BoundsCheckFd(&eh))
                return;
 
        if (text.length() > MAXBUF - 2)
@@ -1009,10 +992,10 @@ void LocalUser::Write(const std::string& text)
                return;
        }
 
-       ServerInstance->Logs->Log("USEROUTPUT", DEBUG,"C[%d] O %s", this->GetFd(), text.c_str());
+       ServerInstance->Logs->Log("USEROUTPUT", DEBUG,"C[%s] O %s", uuid.c_str(), text.c_str());
 
-       this->AddWriteBuf(text);
-       this->AddWriteBuf(wide_newline);
+       eh.AddWriteBuf(text);
+       eh.AddWriteBuf(wide_newline);
 
        ServerInstance->stats->statsSent += text.length() + 2;
        this->bytes_out += text.length() + 2;
@@ -1174,10 +1157,10 @@ void User::WriteCommonRaw(const std::string &line, bool include_self)
 
        for (std::map<User*,bool>::iterator i = exceptions.begin(); i != exceptions.end(); ++i)
        {
-               User* u = i->first;
-               if (IS_LOCAL(u) && !u->quitting)
+               LocalUser* u = IS_LOCAL(i->first);
+               if (u && !u->quitting)
                {
-                       already_sent[u->fd] = uniq_id;
+                       already_sent[u->GetFd()] = uniq_id;
                        if (i->second)
                                u->Write(line);
                }
@@ -1188,10 +1171,10 @@ void User::WriteCommonRaw(const std::string &line, bool include_self)
                const UserMembList* ulist = c->GetUsers();
                for (UserMembList::const_iterator i = ulist->begin(); i != ulist->end(); i++)
                {
-                       User* u = i->first;
-                       if (IS_LOCAL(u) && !u->quitting && already_sent[u->fd] != uniq_id)
+                       LocalUser* u = IS_LOCAL(i->first);
+                       if (u && !u->quitting && already_sent[u->GetFd()] != uniq_id)
                        {
-                               already_sent[u->fd] = uniq_id;
+                               already_sent[u->GetFd()] = uniq_id;
                                u->Write(line);
                        }
                }
@@ -1220,10 +1203,10 @@ void User::WriteCommonQuit(const std::string &normal_text, const std::string &op
 
        for (std::map<User*,bool>::iterator i = exceptions.begin(); i != exceptions.end(); ++i)
        {
-               User* u = i->first;
-               if (IS_LOCAL(u) && !u->quitting)
+               LocalUser* u = IS_LOCAL(i->first);
+               if (u && !u->quitting)
                {
-                       already_sent[u->fd] = uniq_id;
+                       already_sent[u->GetFd()] = uniq_id;
                        if (i->second)
                                u->Write(IS_OPER(u) ? out2 : out1);
                }
@@ -1233,10 +1216,10 @@ void User::WriteCommonQuit(const std::string &normal_text, const std::string &op
                const UserMembList* ulist = (*v)->GetUsers();
                for (UserMembList::const_iterator i = ulist->begin(); i != ulist->end(); i++)
                {
-                       User* u = i->first;
-                       if (IS_LOCAL(u) && !u->quitting && (already_sent[u->fd] != uniq_id))
+                       LocalUser* u = IS_LOCAL(i->first);
+                       if (u && !u->quitting && (already_sent[u->GetFd()] != uniq_id))
                        {
-                               already_sent[u->fd] = uniq_id;
+                               already_sent[u->GetFd()] = uniq_id;
                                u->Write(IS_OPER(u) ? out2 : out1);
                        }
                }
@@ -1357,17 +1340,17 @@ void User::DoHostCycle(const std::string &quitline)
 
        for (std::map<User*,bool>::iterator i = exceptions.begin(); i != exceptions.end(); ++i)
        {
-               User* u = i->first;
-               if (IS_LOCAL(u) && !u->quitting)
+               LocalUser* u = IS_LOCAL(i->first);
+               if (u && !u->quitting)
                {
                        if (i->second)
                        {
-                               already_sent[u->fd] = seen_id;
+                               already_sent[u->GetFd()] = seen_id;
                                u->Write(quitline);
                        }
                        else
                        {
-                               already_sent[u->fd] = silent_id;
+                               already_sent[u->GetFd()] = silent_id;
                        }
                }
        }
@@ -1376,9 +1359,12 @@ void User::DoHostCycle(const std::string &quitline)
                Channel* c = *v;
                snprintf(buffer, MAXBUF, ":%s JOIN %s", GetFullHost().c_str(), c->name.c_str());
                std::string joinline(buffer);
-               std::string modeline = ServerInstance->Modes->ModeString(this, c);
+               Membership* memb = c->GetUser(this);
+               std::string modeline = memb->modes;
                if (modeline.length() > 0)
                {
+                       for(unsigned int i=0; i < memb->modes.length(); i++)
+                               modeline.append(" ").append(nick);
                        snprintf(buffer, MAXBUF, ":%s MODE %s +%s", GetFullHost().c_str(), c->name.c_str(), modeline.c_str());
                        modeline = buffer;
                }
@@ -1386,16 +1372,16 @@ void User::DoHostCycle(const std::string &quitline)
                const UserMembList *ulist = c->GetUsers();
                for (UserMembList::const_iterator i = ulist->begin(); i != ulist->end(); i++)
                {
-                       User* u = i->first;
-                       if (u == this || !IS_LOCAL(u))
+                       LocalUser* u = IS_LOCAL(i->first);
+                       if (u == NULL || u == this)
                                continue;
-                       if (already_sent[u->fd] == silent_id)
+                       if (already_sent[u->GetFd()] == silent_id)
                                continue;
 
-                       if (already_sent[u->fd] != seen_id)
+                       if (already_sent[u->GetFd()] != seen_id)
                        {
                                u->Write(quitline);
-                               already_sent[i->first->fd] = seen_id;
+                               already_sent[u->GetFd()] = seen_id;
                        }
                        u->Write(joinline);
                        if (modeline.length() > 0)
@@ -1686,19 +1672,19 @@ const std::string& FakeUser::GetFullRealHost()
 }
 
 ConnectClass::ConnectClass(ConfigTag* tag, char t, const std::string& mask)
-       : config(tag), type(t), name("unnamed"), registration_timeout(0), host(mask),
-       pingtime(0), pass(""), hash(""), softsendqmax(0), hardsendqmax(0),
-       recvqmax(0), penaltythreshold(0), maxlocal(0), maxglobal(0), maxchans(0), port(0), limit(0)
+       : config(tag), type(t), fakelag(true), name("unnamed"), registration_timeout(0), host(mask),
+       pingtime(0), pass(""), hash(""), softsendqmax(0), hardsendqmax(0), recvqmax(0),
+       penaltythreshold(0), commandrate(0), maxlocal(0), maxglobal(0), maxchans(0), port(0), limit(0)
 {
 }
 
 ConnectClass::ConnectClass(ConfigTag* tag, char t, const std::string& mask, const ConnectClass& parent)
-       : config(tag), type(t), name("unnamed"),
-       registration_timeout(parent.registration_timeout), host(mask),
-       pingtime(parent.pingtime), pass(parent.pass), hash(parent.hash),
-       softsendqmax(parent.softsendqmax), hardsendqmax(parent.hardsendqmax),
-       recvqmax(parent.recvqmax), penaltythreshold(parent.penaltythreshold), maxlocal(parent.maxlocal),
-       maxglobal(parent.maxglobal), maxchans(parent.maxchans),
+       : config(tag), type(t), fakelag(parent.fakelag), name("unnamed"),
+       registration_timeout(parent.registration_timeout), host(mask), pingtime(parent.pingtime),
+       pass(parent.pass), hash(parent.hash), softsendqmax(parent.softsendqmax),
+       hardsendqmax(parent.hardsendqmax), recvqmax(parent.recvqmax),
+       penaltythreshold(parent.penaltythreshold), commandrate(parent.commandrate),
+       maxlocal(parent.maxlocal), maxglobal(parent.maxglobal), maxchans(parent.maxchans),
        port(parent.port), limit(parent.limit)
 {
 }