]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/inspircd.cpp
Align names that are output on startup using \t htabs
[user/henk/code/inspircd.git] / src / inspircd.cpp
index cf4872fbebc841d0d92f4f5126d128d410fbf188..9b48d83ef4dc85b5efa2af17c96490f3a1169aba 100644 (file)
@@ -72,6 +72,8 @@ std::vector<userrec*> local_users;
 extern int MODCOUNT;
 extern char LOG_FILE[MAXBUF];
 int openSockfd[MAXSOCKS];
+int yield_depth;
+int iterations = 0;
 sockaddr_in client,server;
 socklen_t length;
 
@@ -89,7 +91,7 @@ ServerConfig *Config = new ServerConfig;
 
 user_hash clientlist;
 chan_hash chanlist;
-whowas_hash whowas;
+
 servernamelist servernames;
 char lowermap[255];
 
@@ -127,21 +129,7 @@ bool FindServerName(std::string servername)
 
 std::string InspIRCd::GetRevision()
 {
-       /* w00t got me to replace a bunch of strtok_r
-        * with something nicer, so i did this. Its the
-        * same thing really, only in C++. It places the
-        * text into a std::stringstream which is a readable
-        * and writeable buffer stream, and then pops two
-        * words off it, space delimited. Because it reads
-        * into the same variable twice, the first word
-        * is discarded, and the second one returned.
-        */
-
-       /* XXX - this revision ID is NOT bumping automatically -- w00t */
-       std::stringstream Revision("$Revision$");
-       std::string single;
-       Revision >> single >> single;
-       return single;
+       return REVISION;
 }
 
 void InspIRCd::MakeLowerMap()
@@ -251,11 +239,11 @@ std::string InspIRCd::GetVersionString()
 #endif
        if (*Config->CustomVersion)
        {
-               snprintf(versiondata,MAXBUF,"%s Rev. %s %s :%s",VERSION,GetRevision().c_str(),Config->ServerName,Config->CustomVersion);
+               snprintf(versiondata,MAXBUF,"%s %s :%s",VERSION,Config->ServerName,Config->CustomVersion);
        }
        else
        {
-               snprintf(versiondata,MAXBUF,"%s Rev. %s %s :%s [FLAGS=%lu,%s,%s]",VERSION,GetRevision().c_str(),Config->ServerName,SYSTEM,(unsigned long)OPTIMISATION,SE->GetName().c_str(),dnsengine);
+               snprintf(versiondata,MAXBUF,"%s %s :%s [FLAGS=%lu,%s,%s]",VERSION,Config->ServerName,SYSTEM,(unsigned long)OPTIMISATION,SE->GetName().c_str(),dnsengine);
        }
        return versiondata;
 }
@@ -604,19 +592,218 @@ bool InspIRCd::LoadModule(const char* filename)
        return true;
 }
 
-int InspIRCd::Run()
+void InspIRCd::DoOneIteration(bool process_module_sockets)
 {
-       bool expire_run = false;
-       int activefds[MAX_DESCRIPTORS];
-       int incomingSockfd;
-       int in_port;
-       userrec* cu = NULL;
-       InspSocket* s = NULL;
-       InspSocket* s_del = NULL;
-       unsigned int numberactive;
+        int activefds[MAX_DESCRIPTORS];
+        int incomingSockfd;
+        int in_port;
+        userrec* cu = NULL;
+        InspSocket* s = NULL;
+        InspSocket* s_del = NULL;
+        unsigned int numberactive;
         sockaddr_in sock_us;     // our port number
-       socklen_t uslen;         // length of our port number
+        socklen_t uslen;         // length of our port number
+
+       if (yield_depth > 3)
+               return;
+
+       yield_depth++;
+
+        /* time() seems to be a pretty expensive syscall, so avoid calling it too much.
+         * Once per loop iteration is pleanty.
+         */
+        OLDTIME = TIME;
+        TIME = time(NULL);
+        
+        /* Run background module timers every few seconds
+         * (the docs say modules shouldnt rely on accurate
+         * timing using this event, so we dont have to
+         * time this exactly).
+         */
+        if (((TIME % 5) == 0) && (!expire_run))
+        {
+                expire_lines();
+               if (process_module_sockets)
+               {
+                       /* Fix by brain - the addition of DoOneIteration means that this
+                        * can end up getting called recursively in the following pattern:
+                        *
+                        * m_spanningtree DoPingChecks
+                        * (server pings out and is squit)
+                        * (squit causes call to DoOneIteration)
+                        * DoOneIteration enters here
+                        * calls DoBackground timer
+                        * enters m_spanningtree DoPingChecks... see step 1.
+                        *
+                        * This should do the job and fix the bug.
+                        */
+                       FOREACH_MOD(I_OnBackgroundTimer,OnBackgroundTimer(TIME));
+               }
+                TickMissedTimers(TIME);
+                expire_run = true;
+               yield_depth--;
+                return;
+        }   
+        else if ((TIME % 5) == 1)
+        {
+                expire_run = false;
+        }
+
+       if (iterations++ == 15)
+       {
+               iterations = 0;
+               DoBackgroundUserStuff(TIME);
+       }
+        /* Once a second, do the background processing */
+        if (TIME != OLDTIME)
+        {
+                if (TIME < OLDTIME)
+                        WriteOpers("*** \002EH?!\002 -- Time is flowing BACKWARDS in this dimension! Clock drifted backwards %d secs.",abs(OLDTIME-TIME));
+               if ((TIME % 3600) == 0)
+               {
+                       MaintainWhoWas(TIME);
+               }
+        }
+
+        /* Process timeouts on module sockets each time around
+         * the loop. There shouldnt be many module sockets, at
+         * most, 20 or so, so this won't be much of a performance
+         * hit at all.   
+         */ 
+       if (process_module_sockets)
+               DoSocketTimeouts(TIME);  
+         
+        TickTimers(TIME);
+         
+        /* Call the socket engine to wait on the active
+         * file descriptors. The socket engine has everything's
+         * descriptors in its list... dns, modules, users,
+         * servers... so its nice and easy, just one call.
+         */
+        if (!(numberactive = SE->Wait(activefds)))
+       {
+               yield_depth--;
+                return;
+       }
+
+        /**
+         * Now process each of the fd's. For users, we have a fast
+         * lookup table which can find a user by file descriptor, so
+         * processing them by fd isnt expensive. If we have a lot of
+         * listening ports or module sockets though, things could get
+         * ugly.
+         */
+        for (unsigned int activefd = 0; activefd < numberactive; activefd++)
+        {
+                int socket_type = SE->GetType(activefds[activefd]);
+                switch (socket_type)
+                {
+                        case X_ESTAB_CLIENT:
+
+                                cu = fd_ref_table[activefds[activefd]];
+                                if (cu)
+                                        ProcessUser(cu);  
+        
+                        break;
+        
+                        case X_ESTAB_MODULE:
+
+                               if (!process_module_sockets)
+                                       break;
+
+                                /* Process module-owned sockets.
+                                 * Modules are encouraged to inherit their sockets from
+                                 * InspSocket so we can process them neatly like this.
+                                 */
+                                s = socket_ref[activefds[activefd]]; 
+              
+                                if ((s) && (!s->Poll()))
+                                {
+                                        log(DEBUG,"Socket poll returned false, close and bail");
+                                        SE->DelFd(s->GetFd());
+                                        for (std::vector<InspSocket*>::iterator a = module_sockets.begin(); a < module_sockets.end(); a++)
+                                        {
+                                                s_del = (InspSocket*)*a;
+                                                if ((s_del) && (s_del->GetFd() == activefds[activefd]))
+                                                {
+                                                        module_sockets.erase(a);
+                                                        break;
+                                                }
+                                        }
+                                        s->Close();
+                                        delete s;
+                                }
+                        break;
+
+                        case X_ESTAB_DNS:
+                                /* When we are using single-threaded dns,
+                                 * the sockets for dns end up in our mainloop.
+                                 * When we are using multi-threaded dns,
+                                 * each thread has its own basic poll() loop
+                                 * within it, making them 'fire and forget'
+                                 * and independent of the mainloop.
+                                 */
+#ifndef THREADED_DNS
+                                dns_poll(activefds[activefd]);
+#endif
+                        break;
+
+                       case X_LISTEN:
+
+                                /* It's a listener */
+                                uslen = sizeof(sock_us);
+                                length = sizeof(client);
+                                incomingSockfd = accept (activefds[activefd],(struct sockaddr*)&client,&length);
+        
+                                if ((incomingSockfd > -1) && (!getsockname(incomingSockfd,(sockaddr*)&sock_us,&uslen)))
+                                {
+                                        in_port = ntohs(sock_us.sin_port);
+                                        log(DEBUG,"Accepted socket %d",incomingSockfd);
+                                        /* Years and years ago, we used to resolve here
+                                         * using gethostbyaddr(). That is sucky and we
+                                         * don't do that any more...
+                                         */
+                                        NonBlocking(incomingSockfd);
+                                        if (Config->GetIOHook(in_port))
+                                        {
+                                                try
+                                                {
+                                                        Config->GetIOHook(in_port)->OnRawSocketAccept(incomingSockfd, (char*)inet_ntoa(client.sin_addr), in_port);
+                                                }
+                                                catch (ModuleException& modexcept)
+                                                {
+                                                        log(DEBUG,"Module exception cought: %s",modexcept.GetReason());
+                                                }
+                                        }
+                                        stats->statsAccept++;
+                                        AddClient(incomingSockfd, in_port, false, client.sin_addr);
+                                        log(DEBUG,"Adding client on port %lu fd=%lu",(unsigned long)in_port,(unsigned long)incomingSockfd);
+                                }
+                                else
+                                {
+                                        log(DEBUG,"Accept failed on fd %lu: %s",(unsigned long)incomingSockfd,strerror(errno));
+                                        shutdown(incomingSockfd,2);
+                                        close(incomingSockfd);
+                                        stats->statsRefused++;
+                                }
+                        break;
+
+                        default:
+                                /* Something went wrong if we're in here.
+                                 * In fact, so wrong, im not quite sure
+                                 * what we would do, so for now, its going
+                                 * to safely do bugger all.
+                                 */
+                               SE->DelFd(activefds[activefd]);
+                        break;
+                }
+        }
+       yield_depth--;
+}
 
+int InspIRCd::Run()
+{
        /* Until THIS point, ServerInstance == NULL */
        
         LoadAllModules(this);
@@ -625,8 +812,9 @@ int InspIRCd::Run()
        
        if (!Config->nofork)
        {
-               freopen("/dev/null","w",stdout);
-               freopen("/dev/null","w",stderr);
+               fclose(stdout);
+               fclose(stderr);
+               fclose(stdin);
        }
 
        /* Add the listening sockets used for client inbound connections
@@ -638,172 +826,13 @@ int InspIRCd::Run()
        WritePID(Config->PID);
 
        /* main loop, this never returns */
-       for (;;)
-       {
-               /* time() seems to be a pretty expensive syscall, so avoid calling it too much.
-                * Once per loop iteration is pleanty.
-                */
-               OLDTIME = TIME;
-               TIME = time(NULL);
-
-               /* Run background module timers every few seconds
-                * (the docs say modules shouldnt rely on accurate
-                * timing using this event, so we dont have to
-                * time this exactly).
-                */
-               if (((TIME % 5) == 0) && (!expire_run))
-               {
-                       expire_lines();
-                       FOREACH_MOD(I_OnBackgroundTimer,OnBackgroundTimer(TIME));
-                       TickMissedTimers(TIME);
-                       expire_run = true;
-                       continue;
-               }
-               else if ((TIME % 5) == 1)
-               {
-                       expire_run = false;
-               }
-               
-               /* Once a second, do the background processing */
-               if (TIME != OLDTIME)
-               {
-                       if (TIME < OLDTIME)
-                               WriteOpers("*** \002EH?!\002 -- Time is flowing BACKWARDS in this dimension! Clock drifted backwards %d secs.",abs(OLDTIME-TIME));
-                       DoBackgroundUserStuff(TIME);
-
-                       /*
-                        * Trigger all InspTimers that are pending
-                        */
-               }
-
-               /* Process timeouts on module sockets each time around
-                * the loop. There shouldnt be many module sockets, at
-                * most, 20 or so, so this won't be much of a performance
-                * hit at all.
-                */
-               DoSocketTimeouts(TIME);
-
-               TickTimers(TIME);
-
-               /* Call the socket engine to wait on the active
-                * file descriptors. The socket engine has everything's
-                * descriptors in its list... dns, modules, users,
-                * servers... so its nice and easy, just one call.
-                */
-               if (!(numberactive = SE->Wait(activefds)))
-                       continue;
-
-               /**
-                * Now process each of the fd's. For users, we have a fast
-                * lookup table which can find a user by file descriptor, so
-                * processing them by fd isnt expensive. If we have a lot of
-                * listening ports or module sockets though, things could get
-                * ugly.
-                */
-               for (unsigned int activefd = 0; activefd < numberactive; activefd++)
-               {
-                       int socket_type = SE->GetType(activefds[activefd]);
-                       switch (socket_type)
-                       {
-                               case X_ESTAB_CLIENT:
-
-                                       cu = fd_ref_table[activefds[activefd]];
-                                       if (cu)
-                                               ProcessUser(cu);
-
-                               break;
-
-                               case X_ESTAB_MODULE:
-
-                                       /* Process module-owned sockets.
-                                        * Modules are encouraged to inherit their sockets from
-                                        * InspSocket so we can process them neatly like this.
-                                        */
-                                       s = socket_ref[activefds[activefd]];
-
-                                       if ((s) && (!s->Poll()))
-                                       {
-                                               log(DEBUG,"Socket poll returned false, close and bail");
-                                               SE->DelFd(s->GetFd());
-                                               for (std::vector<InspSocket*>::iterator a = module_sockets.begin(); a < module_sockets.end(); a++)
-                                               {
-                                                       s_del = (InspSocket*)*a;
-                                                       if ((s_del) && (s_del->GetFd() == activefds[activefd]))
-                                                       {
-                                                               module_sockets.erase(a);
-                                                               break;
-                                                       }
-                                               }
-                                               s->Close();
-                                               delete s;
-                                       }
-
-                               break;
-
-                               case X_ESTAB_DNS:
-
-                                       /* When we are using single-threaded dns,
-                                        * the sockets for dns end up in our mainloop.
-                                        * When we are using multi-threaded dns,
-                                        * each thread has its own basic poll() loop
-                                        * within it, making them 'fire and forget'
-                                        * and independent of the mainloop.
-                                        */
-#ifndef THREADED_DNS
-                                       dns_poll(activefds[activefd]);
-#endif
-                               break;
-                               
-                               case X_LISTEN:
-
-                                       /* It's a listener */
-                                       uslen = sizeof(sock_us);
-                                       length = sizeof(client);
-                                       incomingSockfd = accept (activefds[activefd],(struct sockaddr*)&client,&length);
-                                       
-                                       if ((incomingSockfd > -1) && (!getsockname(incomingSockfd,(sockaddr*)&sock_us,&uslen)))
-                                       {
-                                               in_port = ntohs(sock_us.sin_port);
-                                               log(DEBUG,"Accepted socket %d",incomingSockfd);
-                                               /* Years and years ago, we used to resolve here
-                                                * using gethostbyaddr(). That is sucky and we
-                                                * don't do that any more...
-                                                */
-                                               NonBlocking(incomingSockfd);
-                                               if (Config->GetIOHook(in_port))
-                                               {
-                                                       try
-                                                       {
-                                                               Config->GetIOHook(in_port)->OnRawSocketAccept(incomingSockfd, (char*)inet_ntoa(client.sin_addr), in_port);
-                                                       }
-                                                       catch (ModuleException& modexcept)
-                                                       {
-                                                               log(DEBUG,"Module exception cought: %s",modexcept.GetReason()); \
-                                                       }
-                                               }
-                                               stats->statsAccept++;
-                                               AddClient(incomingSockfd, in_port, false, client.sin_addr);
-                                               log(DEBUG,"Adding client on port %lu fd=%lu",(unsigned long)in_port,(unsigned long)incomingSockfd);
-                                       }
-                                       else
-                                       {
-                                               log(DEBUG,"Accept failed on fd %lu: %s",(unsigned long)incomingSockfd,strerror(errno));
-                                               shutdown(incomingSockfd,2);
-                                               close(incomingSockfd);
-                                               stats->statsRefused++;
-                                       }
-                               break;
-
-                               default:
-                                       /* Something went wrong if we're in here.
-                                        * In fact, so wrong, im not quite sure
-                                        * what we would do, so for now, its going
-                                        * to safely do bugger all.
-                                        */
-                               break;
-                       }
-               }
+       expire_run = false;
+       yield_depth = 0;
+       iterations = 0;
 
+       while (true)
+       {
+               DoOneIteration(true);
        }
        /* This is never reached -- we hope! */
        return 0;