]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/users.cpp
This won't work yet.
[user/henk/code/inspircd.git] / src / users.cpp
index f978fc4416ce0d2dbefe529db84b3e22dd29918a..3803a7a9efebeeeea00e209f689b558dd0e71ab3 100644 (file)
@@ -11,6 +11,8 @@
  * ---------------------------------------------------
  */
 
+/* $Core: libIRCDusers */
+
 #include "inspircd.h"
 #include <stdarg.h>
 #include "socketengine.h"
@@ -175,9 +177,10 @@ User::User(InspIRCd* Instance, const std::string &uid) : ServerInstance(Instance
        server = (char*)Instance->FindServerNamePtr(Instance->Config->ServerName);
        reset_due = ServerInstance->Time();
        age = ServerInstance->Time(true);
+       Penalty = 0;
        lines_in = lastping = signon = idle_lastmsg = nping = registered = 0;
-       ChannelCount = timeout = flood = bytes_in = bytes_out = cmds_in = cmds_out = 0;
-       ExemptFromPenalty = UnderPenalty = muted = exempt = haspassed = dns_done = false;
+       ChannelCount = timeout = bytes_in = bytes_out = cmds_in = cmds_out = 0;
+       OverPenalty = ExemptFromPenalty = muted = exempt = haspassed = dns_done = false;
        fd = -1;
        recvq.clear();
        sendq.clear();
@@ -185,6 +188,7 @@ User::User(InspIRCd* Instance, const std::string &uid) : ServerInstance(Instance
        res_forward = res_reverse = NULL;
        Visibility = NULL;
        ip = NULL;
+       MyClass = NULL;
        chans.clear();
        invites.clear();
        memset(modes,0,sizeof(modes));
@@ -231,6 +235,13 @@ void User::RemoveCloneCounts()
 
 User::~User()
 {
+       /* NULL for remote users :) */
+       if (this->MyClass)
+       {
+               this->MyClass->RefCount--;
+               ServerInstance->Log(DEBUG, "User destructor -- connect refcount now: %u", this->MyClass->RefCount);
+       }
+
        this->InvalidateCache();
        this->DecrementModes();
        if (operquit)
@@ -484,10 +495,10 @@ bool User::AddBuffer(std::string a)
                if (a.length())
                        recvq.append(a);
 
-               if (recvq.length() > (unsigned)this->recvqmax)
+               if (this->MyClass && (recvq.length() > this->MyClass->GetRecvqMax()))
                {
                        this->SetWriteError("RecvQ exceeded");
-                       ServerInstance->WriteOpers("*** User %s RecvQ of %d exceeds connect class maximum of %d",this->nick,recvq.length(),this->recvqmax);
+                       ServerInstance->WriteOpers("*** User %s RecvQ of %d exceeds connect class maximum of %d",this->nick,recvq.length(),this->MyClass->GetRecvqMax());
                        return false;
                }
 
@@ -556,7 +567,7 @@ void User::AddWriteBuf(const std::string &data)
        if (*this->GetWriteError())
                return;
 
-       if (sendq.length() + data.length() > (unsigned)this->sendqmax)
+       if (this->MyClass && (sendq.length() + data.length() > this->MyClass->GetSendqMax()))
        {
                /*
                 * Fix by brain - Set the error text BEFORE calling writeopers, because
@@ -564,7 +575,7 @@ void User::AddWriteBuf(const std::string &data)
                 * to repeatedly add the text to the sendq!
                 */
                this->SetWriteError("SendQ exceeded");
-               ServerInstance->WriteOpers("*** User %s SendQ of %d exceeds connect class maximum of %d",this->nick,sendq.length() + data.length(),this->sendqmax);
+               ServerInstance->WriteOpers("*** User %s SendQ of %d exceeds connect class maximum of %d",this->nick,sendq.length() + data.length(),this->MyClass->GetSendqMax());
                return;
        }
 
@@ -700,8 +711,8 @@ void User::UnOper()
 
 void User::QuitUser(InspIRCd* Instance, User *user, const std::string &quitreason, const char* operreason)
 {
-       Instance->Log(DEBUG,"QuitUser: %s", user->nick);
-       user->Write("ERROR :Closing link (%s@%s) [%s]", user->ident, user->host, operreason);
+       Instance->Log(DEBUG,"QuitUser: %s '%s'", user->nick, quitreason.c_str());
+       user->Write("ERROR :Closing link (%s@%s) [%s]", user->ident, user->host, *operreason ? operreason : quitreason.c_str());
        user->muted = true;
        Instance->GlobalCulls.AddItem(user, quitreason.c_str(), operreason);
 }
@@ -778,7 +789,7 @@ void User::AddClient(InspIRCd* Instance, int socket, int port, bool iscached, in
         * First class check. We do this again in FullConnect after DNS is done, and NICK/USER is recieved.
         * See my note down there for why this is required. DO NOT REMOVE. :) -- w00t
         */
-       ConnectClass* i = New->GetClass();
+       ConnectClass* i = New->SetClass();
 
        if (!i)
        {
@@ -819,10 +830,11 @@ void User::AddClient(InspIRCd* Instance, int socket, int port, bool iscached, in
        }
 #endif
 
-       New->exempt = (Instance->XLines->matches_exception(New) != NULL);
+       New->exempt = (Instance->XLines->MatchesLine("E",New) != NULL);
        if (!New->exempt)
        {
-               ZLine* r = Instance->XLines->matches_zline(ipaddr);
+               XLine* r = Instance->XLines->MatchesLine("Z",New);
+
                if (r)
                {
                        char reason[MAXBUF];
@@ -847,6 +859,15 @@ void User::AddClient(InspIRCd* Instance, int socket, int port, bool iscached, in
         * BOPM and other stuff requires it.
         */
        New->WriteServ("NOTICE Auth :*** Looking up your hostname...");
+
+       if (Instance->Config->NoUserDns)
+       {
+               New->dns_done = true;
+       }
+       else
+       {
+               New->StartDNSLookup();
+       }
 }
 
 unsigned long User::GlobalCloneCount()
@@ -870,9 +891,9 @@ unsigned long User::LocalCloneCount()
 /*
  * Check class restrictions
  */
-void User::CheckClass(const std::string &explicit_class)
+void User::CheckClass()
 {
-       ConnectClass* a = this->GetClass(explicit_class);
+       ConnectClass* a = this->MyClass;
 
        if ((!a) || (a->GetType() == CC_DENY))
        {
@@ -892,13 +913,8 @@ void User::CheckClass(const std::string &explicit_class)
                return;
        }
 
-       this->pingmax = a->GetPingTime();
        this->nping = ServerInstance->Time() + a->GetPingTime() + ServerInstance->Config->dns_timeout;
        this->timeout = ServerInstance->Time() + a->GetRegTimeout();
-       this->flood = a->GetFlood();
-       this->threshold = a->GetThreshold();
-       this->sendqmax = a->GetSendqMax();
-       this->recvqmax = a->GetRecvqMax();
        this->MaxChans = a->GetMaxChans();
 }
 
@@ -913,20 +929,20 @@ void User::FullConnect()
         * may put the user into a totally seperate class with different restrictions! so we *must* check again.
         * Don't remove this! -- w00t
         */
-       this->CheckClass();
+       this->SetClass();
        
        /* Check the password, if one is required by the user's connect class.
         * This CANNOT be in CheckClass(), because that is called prior to PASS as well!
         */
-       if ((!this->GetClass()->GetPass().empty()) && (!this->haspassed))
+       if (this->MyClass && !this->MyClass->GetPass().empty() && !this->haspassed)
        {
                User::QuitUser(ServerInstance, this, "Invalid password");
                return;
        }
-       
+
        if (!this->exempt)
        {
-               GLine* r = ServerInstance->XLines->matches_gline(this);
+               XLine* r = ServerInstance->XLines->MatchesLine("G",this);
 
                if (r)
                {
@@ -939,7 +955,7 @@ void User::FullConnect()
                        return;
                }
 
-               KLine* n = ServerInstance->XLines->matches_kline(this);
+               XLine* n = ServerInstance->XLines->MatchesLine("K",this);
 
                if (n)
                {
@@ -1049,7 +1065,7 @@ bool User::ForceNickChange(const char* newnick)
                        return false;
                }
 
-               if (ServerInstance->XLines->matches_qline(newnick))
+               if (ServerInstance->XLines->MatchesLine("Q",newnick))
                {
                        ServerInstance->stats->statsCollisions++;
                        return false;
@@ -1691,40 +1707,94 @@ unsigned int User::GetMaxChans()
        return this->MaxChans;
 }
 
-/* looks up a users password for their connection class (<ALLOW>/<DENY> tags)
+
+/*
+ * Sets a user's connection class.
+ * If the class name is provided, it will be used. Otherwise, the class will be guessed using host/ip/ident/etc.
  * NOTE: If the <ALLOW> or <DENY> tag specifies an ip, and this user resolves,
  * then their ip will be taken as 'priority' anyway, so for example,
  * <connect allow="127.0.0.1"> will match joe!bloggs@localhost
  */
-ConnectClass* User::GetClass(const std::string &explicit_name)
+ConnectClass* User::SetClass(const std::string &explicit_name)
 {
+       ConnectClass *found = NULL;
+
+       if (!IS_LOCAL(this))
+               return NULL;
+
        if (!explicit_name.empty())
        {
                for (ClassVector::iterator i = ServerInstance->Config->Classes.begin(); i != ServerInstance->Config->Classes.end(); i++)
                {
-                       if (explicit_name == i->GetName())
-                               return &(*i);
+                       ConnectClass* c = *i;
+
+                       if (explicit_name == c->GetName() && !c->GetDisabled())
+                       {
+                               found = c;
+                       }
                }
        }
        else
        {
                for (ClassVector::iterator i = ServerInstance->Config->Classes.begin(); i != ServerInstance->Config->Classes.end(); i++)
                {
-                       if (((match(this->GetIPString(),i->GetHost().c_str(),true)) || (match(this->host,i->GetHost().c_str()))))
+                       ConnectClass* c = *i;
+
+                       if (((match(this->GetIPString(),c->GetHost().c_str(),true)) || (match(this->host,c->GetHost().c_str()))))
                        {
-                               if (i->GetPort())
+                               if (c->GetPort())
                                {
-                                       if (this->GetPort() == i->GetPort())
-                                               return &(*i);
+                                       if (this->GetPort() == c->GetPort() && !c->GetDisabled())
+                                       {
+                                               found = c;
+                                       }
                                        else
                                                continue;
                                }
                                else
-                                       return &(*i);
+                               {
+                                       if (!c->GetDisabled())
+                                               found = c;
+                               }
                        }
                }
        }
-       return NULL;
+
+       /* ensure we don't fuck things up refcount wise, only remove them from a class if we find a new one :P */
+       if (found)
+       {
+               /* deny change if change will take class over the limit */
+               if (found->limit && (found->RefCount + 1 >= found->limit))
+               {
+                       ServerInstance->Log(DEBUG, "OOPS: Connect class limit (%u) hit, denying", found->limit);
+                       return this->MyClass;
+               }
+
+               /* should always be valid, but just in case .. */
+               if (this->MyClass)
+               {
+                       if (found == this->MyClass) // no point changing this shit :P
+                               return this->MyClass;
+                       this->MyClass->RefCount--;
+                       ServerInstance->Log(DEBUG, "Untying user from connect class -- refcount: %u", this->MyClass->RefCount);
+               }
+
+               this->MyClass = found;
+               this->MyClass->RefCount++;
+               ServerInstance->Log(DEBUG, "User tied to new class -- connect refcount now: %u", this->MyClass->RefCount);
+       }
+
+       return this->MyClass;
+}
+
+/* looks up a users password for their connection class (<ALLOW>/<DENY> tags)
+ * NOTE: If the <ALLOW> or <DENY> tag specifies an ip, and this user resolves,
+ * then their ip will be taken as 'priority' anyway, so for example,
+ * <connect allow="127.0.0.1"> will match joe!bloggs@localhost
+ */
+ConnectClass* User::GetClass()
+{
+       return this->MyClass;
 }
 
 void User::PurgeEmptyChannels()
@@ -1845,6 +1915,16 @@ const char* User::GetOperQuit()
        return operquit ? operquit : "";
 }
 
+void User::IncreasePenalty(int increase)
+{
+       this->Penalty += increase;
+}
+
+void User::DecreasePenalty(int decrease)
+{
+       this->Penalty -= decrease;
+}
+
 VisData::VisData()
 {
 }