(svn r13412) -Add: OTTDThreadTerminateFunc, for all thread systems, which is called when a thread is terminated. Now GenWorld- and SaveLoad-thread cleanup theirselves correctly, while Fibers don't (as that causes access-violations)

This commit is contained in:
truebrain 2008-06-08 12:06:27 +00:00
parent 96d7f87cc9
commit b95042b05c
9 changed files with 44 additions and 27 deletions

View File

@ -32,7 +32,7 @@ public:
{ {
this->m_sem = ThreadSemaphore::New(); this->m_sem = ThreadSemaphore::New();
/* Create a thread and start stFiberProc */ /* Create a thread and start stFiberProc */
this->m_thread = ThreadObject::New(&stFiberProc, this); this->m_thread = ThreadObject::New(&stFiberProc, this, NULL);
} }
/** /**

View File

@ -288,7 +288,7 @@ void GenerateWorld(GenerateWorldMode mode, uint size_x, uint size_y)
SetupColorsAndInitialWindow(); SetupColorsAndInitialWindow();
if (_network_dedicated || if (_network_dedicated ||
(_gw.thread = ThreadObject::New(&_GenerateWorld, NULL)) == NULL) { (_gw.thread = ThreadObject::New(&_GenerateWorld, NULL, &ThreadObject::TerminateCleanup)) == NULL) {
DEBUG(misc, 1, "Cannot create genworld thread, reverting to single-threaded mode"); DEBUG(misc, 1, "Cannot create genworld thread, reverting to single-threaded mode");
_gw.threaded = false; _gw.threaded = false;
_GenerateWorld(NULL); _GenerateWorld(NULL);

View File

@ -1695,7 +1695,7 @@ SaveOrLoadResult SaveOrLoad(const char *filename, int mode, Subdirectory sb)
SaveFileStart(); SaveFileStart();
if (_network_server || if (_network_server ||
(_save_thread = ThreadObject::New(&SaveFileToDiskThread, NULL)) == NULL) { (_save_thread = ThreadObject::New(&SaveFileToDiskThread, NULL, &ThreadObject::TerminateCleanup)) == NULL) {
if (!_network_server) DEBUG(sl, 1, "Cannot create savegame thread, reverting to single-threaded mode..."); if (!_network_server) DEBUG(sl, 1, "Cannot create savegame thread, reverting to single-threaded mode...");
SaveOrLoadResult result = SaveFileToDisk(false); SaveOrLoadResult result = SaveFileToDisk(false);

View File

@ -6,6 +6,7 @@
#define THREAD_H #define THREAD_H
typedef void (*OTTDThreadFunc)(void *); typedef void (*OTTDThreadFunc)(void *);
typedef void (*OTTDThreadTerminateFunc)(class ThreadObject *self);
/** /**
* A Thread Object which works on all our supported OSes. * A Thread Object which works on all our supported OSes.
@ -56,9 +57,10 @@ public:
* with optinal params. * with optinal params.
* @param proc The procedure to call inside the thread. * @param proc The procedure to call inside the thread.
* @param param The params to give with 'proc'. * @param param The params to give with 'proc'.
* @param terminate_func The function (or NULL) to call when the thread terminates.
* @return True if the thread was started correctly. * @return True if the thread was started correctly.
*/ */
static ThreadObject *New(OTTDThreadFunc proc, void *param); static ThreadObject *New(OTTDThreadFunc proc, void *param, OTTDThreadTerminateFunc terminate_func);
/** /**
* Convert the current thread to a new ThreadObject. * Convert the current thread to a new ThreadObject.
@ -71,6 +73,12 @@ public:
* @return The thread ID of the current active thread. * @return The thread ID of the current active thread.
*/ */
static uint CurrentId(); static uint CurrentId();
/**
* A OTTDThreadTerminateFunc, which cleans up the thread itself
* at termination of the thread (so it becomes self-managed).
*/
static void TerminateCleanup(ThreadObject *self) { delete self; }
}; };
/** /**

View File

@ -57,6 +57,7 @@ void KPutStr(CONST_STRPTR format)
class ThreadObject_MorphOS : public ThreadObject { class ThreadObject_MorphOS : public ThreadObject {
private: private:
APTR m_thr; ///< System thread identifier. APTR m_thr; ///< System thread identifier.
OTTDThreadTerminateFunc m_terminate_func; ///< Function to call on thread termination.
struct MsgPort *m_replyport; struct MsgPort *m_replyport;
struct OTTDThreadStartupMessage m_msg; struct OTTDThreadStartupMessage m_msg;
@ -64,7 +65,9 @@ public:
/** /**
* Create a sub process and start it, calling proc(param). * Create a sub process and start it, calling proc(param).
*/ */
ThreadObject_MorphOS(OTTDThreadFunc proc, void *param) : m_thr(0) ThreadObject_MorphOS(OTTDThreadFunc proc, void *param, OTTDThreadTerminateFunc terminate_func) :
m_thr(0),
m_terminate_func(terminate_func)
{ {
struct Task *parent; struct Task *parent;
@ -111,7 +114,9 @@ public:
/** /**
* Create a thread and attach current thread to it. * Create a thread and attach current thread to it.
*/ */
ThreadObject_MorphOS() : m_thr(0) ThreadObject_MorphOS() :
m_thr(0),
m_terminate_func(NULL)
{ {
m_thr = FindTask(NULL); m_thr = FindTask(NULL);
} }
@ -210,12 +215,14 @@ private:
/* Quit the child, exec.library will reply the startup msg internally. */ /* Quit the child, exec.library will reply the startup msg internally. */
KPutStr("[Child] Done.\n"); KPutStr("[Child] Done.\n");
if (this->terminate_func != NULL) this->terminate_func(this);
} }
}; };
/* static */ ThreadObject *ThreadObject::New(OTTDThreadFunc proc, void *param) /* static */ ThreadObject *ThreadObject::New(OTTDThreadFunc proc, void *param, OTTDThreadTerminateFunc terminate_func)
{ {
return new ThreadObject_MorphOS(proc, param); return new ThreadObject_MorphOS(proc, param, terminate_func);
} }
/* static */ ThreadObject *ThreadObject::AttachCurrent() /* static */ ThreadObject *ThreadObject::AttachCurrent()

View File

@ -6,7 +6,7 @@
#include "thread.h" #include "thread.h"
#include "fiber.hpp" #include "fiber.hpp"
/* static */ ThreadObject *ThreadObject::New(OTTDThreadFunc proc, void *param) /* static */ ThreadObject *ThreadObject::New(OTTDThreadFunc proc, void *param, OTTDThreadTerminateFunc terminate_func)
{ {
return NULL; return NULL;
} }

View File

@ -59,7 +59,7 @@ void OTTDExitThread()
#endif #endif
/* static */ ThreadObject *ThreadObject::New(OTTDThreadFunc proc, void *param) /* static */ ThreadObject *ThreadObject::New(OTTDThreadFunc proc, void *param, OTTDThreadTerminateFunc terminate_func)
{ {
return NULL; return NULL;
} }

View File

@ -22,16 +22,18 @@ private:
bool m_attached; ///< True if the ThreadObject was attached to an existing thread. bool m_attached; ///< True if the ThreadObject was attached to an existing thread.
sem_t m_sem_start; ///< Here the new thread waits before it starts. sem_t m_sem_start; ///< Here the new thread waits before it starts.
sem_t m_sem_stop; ///< Here the other thread can wait for this thread to end. sem_t m_sem_stop; ///< Here the other thread can wait for this thread to end.
OTTDThreadTerminateFunc m_terminate_func; ///< Function to call on thread termination.
public: public:
/** /**
* Create a pthread and start it, calling proc(param). * Create a pthread and start it, calling proc(param).
*/ */
ThreadObject_pthread(OTTDThreadFunc proc, void *param) : ThreadObject_pthread(OTTDThreadFunc proc, void *param, OTTDThreadTerminateFunc terminate_func) :
m_thr(0), m_thr(0),
m_proc(proc), m_proc(proc),
m_param(param), m_param(param),
m_attached(false) m_attached(false),
m_terminate_func(terminate_func)
{ {
sem_init(&m_sem_start, 0, 0); sem_init(&m_sem_start, 0, 0);
sem_init(&m_sem_stop, 0, 0); sem_init(&m_sem_stop, 0, 0);
@ -47,7 +49,8 @@ public:
m_thr(0), m_thr(0),
m_proc(NULL), m_proc(NULL),
m_param(0), m_param(0),
m_attached(true) m_attached(true),
m_terminate_func(NULL)
{ {
sem_init(&m_sem_start, 0, 0); sem_init(&m_sem_start, 0, 0);
sem_init(&m_sem_stop, 0, 0); sem_init(&m_sem_stop, 0, 0);
@ -102,8 +105,6 @@ public:
pthread_join(m_thr, NULL); pthread_join(m_thr, NULL);
m_thr = 0; m_thr = 0;
delete this;
} }
/* virtual */ bool IsCurrent() /* virtual */ bool IsCurrent()
@ -136,26 +137,22 @@ private:
/* The new thread stops here so the calling thread can complete pthread_create() call */ /* The new thread stops here so the calling thread can complete pthread_create() call */
sem_wait(&m_sem_start); sem_wait(&m_sem_start);
/* Did this thread die naturally/via exit, or did it join? */
bool exit = false;
/* Call the proc of the creator to continue this thread */ /* Call the proc of the creator to continue this thread */
try { try {
m_proc(m_param); m_proc(m_param);
} catch (...) { } catch (...) {
exit = true;
} }
/* Notify threads waiting for our completion */ /* Notify threads waiting for our completion */
sem_post(&m_sem_stop); sem_post(&m_sem_stop);
if (exit) delete this; if (this->m_terminate_func != NULL) this->m_terminate_func(this);
} }
}; };
/* static */ ThreadObject *ThreadObject::New(OTTDThreadFunc proc, void *param) /* static */ ThreadObject *ThreadObject::New(OTTDThreadFunc proc, void *param, OTTDThreadTerminateFunc terminate_func)
{ {
return new ThreadObject_pthread(proc, param); return new ThreadObject_pthread(proc, param, terminate_func);
} }
/* static */ ThreadObject *ThreadObject::AttachCurrent() /* static */ ThreadObject *ThreadObject::AttachCurrent()

View File

@ -20,17 +20,19 @@ private:
OTTDThreadFunc m_proc; OTTDThreadFunc m_proc;
void *m_param; void *m_param;
bool m_attached; bool m_attached;
OTTDThreadTerminateFunc m_terminate_func;
public: public:
/** /**
* Create a win32 thread and start it, calling proc(param). * Create a win32 thread and start it, calling proc(param).
*/ */
ThreadObject_Win32(OTTDThreadFunc proc, void *param) : ThreadObject_Win32(OTTDThreadFunc proc, void *param, OTTDThreadTerminateFunc terminate_func) :
m_id_thr(0), m_id_thr(0),
m_h_thr(NULL), m_h_thr(NULL),
m_proc(proc), m_proc(proc),
m_param(param), m_param(param),
m_attached(false) m_attached(false),
m_terminate_func(terminate_func)
{ {
m_h_thr = (HANDLE)_beginthreadex(NULL, 0, &stThreadProc, this, CREATE_SUSPENDED, &m_id_thr); m_h_thr = (HANDLE)_beginthreadex(NULL, 0, &stThreadProc, this, CREATE_SUSPENDED, &m_id_thr);
if (m_h_thr == NULL) return; if (m_h_thr == NULL) return;
@ -45,7 +47,8 @@ public:
m_h_thr(NULL), m_h_thr(NULL),
m_proc(NULL), m_proc(NULL),
m_param(NULL), m_param(NULL),
m_attached(false) m_attached(false),
m_terminate_func(NULL)
{ {
BOOL ret = DuplicateHandle(GetCurrentProcess(), GetCurrentThread(), GetCurrentProcess(), &m_h_thr, 0, FALSE, DUPLICATE_SAME_ACCESS); BOOL ret = DuplicateHandle(GetCurrentProcess(), GetCurrentThread(), GetCurrentProcess(), &m_h_thr, 0, FALSE, DUPLICATE_SAME_ACCESS);
if (!ret) return; if (!ret) return;
@ -130,12 +133,14 @@ private:
m_proc(m_param); m_proc(m_param);
} catch (...) { } catch (...) {
} }
if (this->m_terminate_func != NULL) this->m_terminate_func(this);
} }
}; };
/* static */ ThreadObject *ThreadObject::New(OTTDThreadFunc proc, void *param) /* static */ ThreadObject *ThreadObject::New(OTTDThreadFunc proc, void *param, OTTDThreadTerminateFunc terminate_func)
{ {
return new ThreadObject_Win32(proc, param); return new ThreadObject_Win32(proc, param, terminate_func);
} }
/* static */ ThreadObject* ThreadObject::AttachCurrent() /* static */ ThreadObject* ThreadObject::AttachCurrent()