Merge pull request #1018 from SaberUK/insp20+hidekills

[user/henk/code/inspircd.git] / include / threadengine.h

diff --git a/include/threadengine.h b/include/threadengine.h
index 7d27b867ac8b911c34131bd0d066e132eb57c4e6..4bf5a48f38399a94d367989f405f9bff213f5936 100644 (file)
--- a/include/threadengine.h
+++ b/include/threadengine.h
@@ -1,18 +1,25 @@
-/*       +------------------------------------+
- *       | Inspire Internet Relay Chat Daemon |
- *       +------------------------------------+
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
  *
- *  InspIRCd: (C) 2002-2008 InspIRCd Development Team
- * See: http://www.inspircd.org/wiki/index.php/Credits
+ *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
+ *   Copyright (C) 2008 Craig Edwards <craigedwards@brainbox.cc>
  *
- * This program is free but copyrighted software; see
- *            the file COPYING for details.
+ * This file is part of InspIRCd.  InspIRCd is free software: you can
+ * redistribute it and/or modify it under the terms of the GNU General Public
+ * License as published by the Free Software Foundation, version 2.
  *
- * ---------------------------------------------------
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+ * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
+ * details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
-#ifndef __THREADENGINE__
-#define __THREADENGINE__
+
+#ifndef THREADENGINE_H
+#define THREADENGINE_H
 
 #include <vector>
 #include <string>
@@ -20,34 +27,151 @@
 #include "inspircd_config.h"
 #include "base.h"
 
-class InspIRCd;
+class ThreadData;
 
-class CoreExport Thread : public Extensible
+/** Derive from this class to implement your own threaded sections of
+ * code. Be sure to keep your code thread-safe and not prone to deadlocks
+ * and race conditions if you MUST use threading!
+ */
+class CoreExport Thread
 {
+ private:
+       /** Set to true when the thread is to exit
+        */
+       bool ExitFlag;
+ protected:
+       /** Get thread's current exit status.
+        * (are we being asked to exit?)
+        */
+       bool GetExitFlag()
+       {
+               return ExitFlag;
+       }
  public:
-       Thread() { };
-       virtual ~Thread() { };
-       virtual void Run() = 0;
-};
+       /** Opaque thread state managed by threading engine
+        */
+       ThreadData* state;
 
-class CoreExport ThreadEngine : public Extensible
-{
- protected:
+       /** Set Creator to NULL at this point
+        */
+       Thread() : ExitFlag(false), state(NULL)
+       {
+       }
 
-        InspIRCd* ServerInstance;
-        Thread* NewThread;
+       /* If the thread is running, you MUST join BEFORE deletion */
+       virtual ~Thread();
 
- public:
+       /** Override this method to put your actual
+        * threaded code here.
+        */
+       virtual void Run() = 0;
 
-       ThreadEngine(InspIRCd* Instance);
+       /** Signal the thread to exit gracefully.
+        */
+       virtual void SetExitFlag();
 
-       virtual ~ThreadEngine();
+       /** Join the thread (calls SetExitFlag and waits for exit)
+        */
+       void join();
+};
 
-       virtual bool Mutex(bool enable) = 0;
 
-       virtual void Run() = 0;
+class CoreExport QueuedThread : public Thread
+{
+       ThreadQueueData queue;
+ protected:
+       /** Waits for an enqueue operation to complete
+        * You MUST hold the queue lock when you call this.
+        * It will be unlocked while you wait, and will be relocked
+        * before the function returns
+        */
+       void WaitForQueue()
+       {
+               queue.Wait();
+       }
+ public:
+       /** Lock queue.
+        */
+       void LockQueue()
+       {
+               queue.Lock();
+       }
+       /** Unlock queue.
+        */
+       void UnlockQueue()
+       {
+               queue.Unlock();
+       }
+       /** Unlock queue and wake up worker
+        */
+       void UnlockQueueWakeup()
+       {
+               queue.Wakeup();
+               queue.Unlock();
+       }
+       virtual void SetExitFlag()
+       {
+               queue.Lock();
+               Thread::SetExitFlag();
+               queue.Wakeup();
+               queue.Unlock();
+       }
+};
 
-       virtual void Create(Thread* thread_to_init) = 0;
+class CoreExport SocketThread : public Thread
+{
+       ThreadQueueData queue;
+       ThreadSignalData signal;
+ protected:
+       /** Waits for an enqueue operation to complete
+        * You MUST hold the queue lock when you call this.
+        * It will be unlocked while you wait, and will be relocked
+        * before the function returns
+        */
+       void WaitForQueue()
+       {
+               queue.Wait();
+       }
+ public:
+       /** Notifies parent by making the SignalFD ready to read
+        * No requirements on locking
+        */
+       void NotifyParent();
+       SocketThread();
+       virtual ~SocketThread();
+       /** Lock queue.
+        */
+       void LockQueue()
+       {
+               queue.Lock();
+       }
+       /** Unlock queue.
+        */
+       void UnlockQueue()
+       {
+               queue.Unlock();
+       }
+       /** Unlock queue and send wakeup to worker
+        */
+       void UnlockQueueWakeup()
+       {
+               queue.Wakeup();
+               queue.Unlock();
+       }
+       virtual void SetExitFlag()
+       {
+               queue.Lock();
+               Thread::SetExitFlag();
+               queue.Wakeup();
+               queue.Unlock();
+       }
+
+       /**
+        * Called in the context of the parent thread after a notification
+        * has passed through the socket
+        */
+       virtual void OnNotify() = 0;
 };
 
 #endif
+