]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/timer.cpp
Add and properly test the ability for an InspTimer to DelTimer itself from within...
[user/henk/code/inspircd.git] / src / timer.cpp
index c172f2fa5bac2f2c35b38ce6a00c880f8ca79810..e7635c52811d1f806ef2ee5cfdc9c18f756b1484 100644 (file)
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
- *                       E-mail:
- *                <brain@chatspike.net>
- *               <Craig@chatspike.net>
- *     
- * Written by Craig Edwards, Craig McLure, and others.
+ *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
+ *
  * This program is free but copyrighted software; see
  *            the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 
-using namespace std;
-
-#include "inspircd_config.h"
-#include <sys/time.h>
-#include <sys/resource.h>
-#include <sys/types.h>
-#include <sys/socket.h>
-#include <netinet/in.h>
-#include <string>
-#include <unistd.h>
-#include <fcntl.h>
-#include <poll.h>
-#include <sstream>
-#include <iostream>
-#include <fstream>
-#include <stdexcept>
 #include "inspircd.h"
-#include "inspircd_io.h"
-#include "inspstring.h"
-#include "helperfuncs.h"
-
-extern InspIRCd* ServerInstance;
-extern ServerConfig* Config;
-extern time_t TIME;
+#include "timer.h"
 
-typedef std::vector<InspTimer*> timergroup;
-typedef std::map<time_t, timergroup*> timerlist;
-timerlist Timers;
+TimerManager::TimerManager() : CantDeleteHere(false)
+{
+}
 
-void TickTimers(time_t TIME)
+void TimerManager::TickTimers(time_t TIME)
 {
+       this->CantDeleteHere = true;
        timerlist::iterator found = Timers.find(TIME);
 
        if (found != Timers.end())
        {
                timergroup* x = found->second;
-               /*
-                * There are pending timers to trigger
+               /* There are pending timers to trigger.
+                * WARNING: Timers may delete themselves from within
+                * their own Tick methods! see the comment below in
+                * the DelTimer method.
                 */
                for (timergroup::iterator y = x->begin(); y != x->end(); y++)
                {
-                       InspTimer* n = (InspTimer*)*y;
+                       InspTimer* n = *y;
                        n->Tick(TIME);
-                       delete n;
+                       if (n->GetRepeat())
+                       {
+                               AddTimer(n, n->GetSecs());
+                       }
+                       else
+                       {
+                               DELETE(n);
+                       }
                }
 
                Timers.erase(found);
-               delete x;
+               DELETE(x);
        }
+
+       this->CantDeleteHere = false;
 }
 
-void TickMissedTimers(time_t TIME)
+void TimerManager::DelTimer(InspTimer* T)
 {
+       if (this->CantDeleteHere)
+       {
+               /* If a developer tries to delete a timer from within its own Tick method,
+                * then chances are this is just going to totally fuck over the timergroup
+                * and timerlist iterators and cause a crash. Thanks to peavey and Bricker
+                * for noticing this bug.
+                * If we're within the tick loop when the DelTimer is called (signified
+                * by the var 'CantDeleteHere') then we simply return for non-repeating
+                * timers, and cancel the repeat on repeating timers. We can do this because
+                * we know that the timer tick loop will safely delete the timer for us
+                * anyway and therefore we avoid stack corruption.
+                */
+               if (T->GetRepeat())
+                       T->CancelRepeat();
+               else
+                       return;
+       }
+
+       timerlist::iterator found = Timers.find(T->GetTimer());
+
+       if (found != Timers.end())
+       {
+               timergroup* x = found->second;
+               for (timergroup::iterator y = x->begin(); y != x->end(); y++)
+               {
+                       InspTimer* n = *y;
+                       if (n == T)
+                       {
+                               DELETE(n);
+                               x->erase(y);
+                               if (!x->size())
+                               {
+                                       Timers.erase(found);
+                               }
+                               return;
+                       }
+               }
+       }
+}
+
+/** Because some muppets may do odd things, and their ircd may lock up due
+ * to crappy 3rd party modules, or they may change their system time a bit,
+ * this accounts for shifts of up to 120 secs by looking behind for missed
+ * timers and executing them. This is only executed once every 5 secs.
+ * If you move your clock BACK, and your timers move further ahead as a result,
+ * then tough titty you'll just have to wait.
+ */
+void TimerManager::TickMissedTimers(time_t TIME)
+{
+       /** See comment above in TickTimers
+        */
+       this->CantDeleteHere = true;
+
        for (time_t n = TIME-1; n > TIME-120; n--)
        {
                timerlist::iterator found = Timers.find(n);
@@ -75,23 +115,38 @@ void TickMissedTimers(time_t TIME)
                        timergroup* x = found->second;
                        for (timergroup::iterator y = x->begin(); y != x->end(); y++)
                        {
-                               InspTimer* z = (InspTimer*)*y;
+                               InspTimer* z = *y;
                                z->Tick(TIME);
-                               delete z;
+                               if (z->GetRepeat())
+                               {
+                                       AddTimer(z, z->GetSecs());
+                               }
+                               else
+                               {
+                                       DELETE(z);
+                               }
                        }
 
                        Timers.erase(found);
-                       delete x;
+                       DELETE(x);
                }
        }
+
+       this->CantDeleteHere = false;
 }
 
-void AddTimer(InspTimer* T)
+void TimerManager::AddTimer(InspTimer* T, long secs_from_now)
 {
        timergroup* x = NULL;
 
-       timerlist::iterator found = Timers.find(T->GetTimer());
-       
+       int time_to_trigger = 0;
+       if (!secs_from_now)
+               time_to_trigger = T->GetTimer();
+       else
+               time_to_trigger = secs_from_now + time(NULL);
+
+       timerlist::iterator found = Timers.find(time_to_trigger);
+
        if (found != Timers.end())
        {
                x = found->second;
@@ -99,8 +154,9 @@ void AddTimer(InspTimer* T)
        else
        {
                x = new timergroup;
-               Timers[T->GetTimer()] = x;
+               Timers[time_to_trigger] = x;
        }
 
        x->push_back(T);
 }
+