1

/* $Id$ */

     2

     3

/** @file thread_win32.cpp Win32 thread implementation of Threads. */

     4

     5

#include "stdafx.h"

     6

#include "thread.h"

     7

#include "debug.h"

     8

#include "core/alloc_func.hpp"

     9

#include <stdlib.h>

    10

#include <windows.h>

    11

#include <process.h>

    12

    13

/**

    14

 * Win32 thread version for ThreadObject.

    15

 */

    16

class ThreadObject_Win32 : public ThreadObject {

    17

private:

    18

	uint     m_id_thr;

    19

	HANDLE   m_h_thr;

    20

	OTTDThreadFunc m_proc;

    21

	void     *m_param;

    22

	bool     m_attached;

    23

	void     *ret;

    24

    25

public:

    26

	/**

    27

	 * Create a win32 thread and start it, calling proc(param).

    28

	 */

    29

	ThreadObject_Win32(OTTDThreadFunc proc, void *param) :

    30

		m_id_thr(0),

    31

		m_h_thr(NULL),

    32

		m_proc(proc),

    33

		m_param(param),

    34

		m_attached(false)

    35

	{

    36

		m_h_thr = (HANDLE)_beginthreadex(NULL, 0, &stThreadProc, this, CREATE_SUSPENDED, &m_id_thr);

    37

		if (m_h_thr == NULL) return;

    38

		ResumeThread(m_h_thr);

    39

	}

    40

    41

	/**

    42

	 * Create a win32 thread and attach current thread to it.

    43

	 */

    44

	ThreadObject_Win32() :

    45

		m_id_thr(0),

    46

		m_h_thr(NULL),

    47

		m_proc(NULL),

    48

		m_param(NULL),

    49

		m_attached(false)

    50

	{

    51

		BOOL ret = DuplicateHandle(GetCurrentProcess(), GetCurrentThread(), GetCurrentProcess(), &m_h_thr, 0, FALSE, DUPLICATE_SAME_ACCESS);

    52

		if (!ret) return;

    53

		m_id_thr = GetCurrentThreadId();

    54

	}

    55

    56

	/* virtual */ ~ThreadObject_Win32()

    57

	{

    58

		if (m_h_thr != NULL) {

    59

			CloseHandle(m_h_thr);

    60

			m_h_thr = NULL;

    61

		}

    62

	}

    63

    64

	/* virtual */ bool IsRunning()

    65

	{

    66

		if (m_h_thr == NULL) return false;

    67

		DWORD exit_code = 0;

    68

		if (!GetExitCodeThread(m_h_thr, &exit_code)) return false;

    69

		return (exit_code == STILL_ACTIVE);

    70

	}

    71

    72

	/* virtual */ bool WaitForStop()

    73

	{

    74

		/* You can't wait on yourself */

    75

		assert(!IsCurrent());

    76

		/* If the thread is not running, waiting is over */

    77

		if (!IsRunning()) return true;

    78

    79

		DWORD res = WaitForSingleObject(m_h_thr, INFINITE);

    80

		return res == WAIT_OBJECT_0;

    81

	}

    82

    83

	/* virtual */ bool Exit()

    84

	{

    85

		/* You can only exit yourself */

    86

		assert(IsCurrent());

    87

		/* If the thread is not running, we are already closed */

    88

		if (!IsRunning()) return false;

    89

    90

		/* For now we terminate by throwing an error, gives much cleaner cleanup */

    91

		throw 0;

    92

	}

    93

    94

	/* virtual */ void *Join()

    95

	{

    96

		/* You cannot join yourself */

    97

		assert(!IsCurrent());

    98

    99

		WaitForSingleObject(m_h_thr, INFINITE);

   100

   101

		return this->ret;

   102

	}

   103

   104

	/* virtual */ bool IsCurrent()

   105

	{

   106

		DWORD id_cur = GetCurrentThreadId();

   107

		return id_cur == m_id_thr;

   108

	}

   109

   110

	/* virtual */ uint GetId()

   111

	{

   112

		return m_id_thr;

   113

	}

   114

   115

private:

   116

	/**

   117

	 * On thread creation, this function is called, which calls the real startup

   118

	 *  function. This to get back into the correct instance again.

   119

	 */

   120

	static uint CALLBACK stThreadProc(void *thr)

   121

	{

   122

		return ((ThreadObject_Win32 *)thr)->ThreadProc();

   123

	}

   124

   125

	/**

   126

	 * A new thread is created, and this function is called. Call the custom

   127

	 *  function of the creator of the thread.

   128

	 */

   129

	uint ThreadProc()

   130

	{

   131

		try {

   132

			this->ret = m_proc(m_param);

   133

		} catch (...) {

   134

		}

   135

   136

		return 0;

   137

	}

   138

};

   139

   140

/* static */ ThreadObject *ThreadObject::New(OTTDThreadFunc proc, void *param)

   141

{

   142

	return new ThreadObject_Win32(proc, param);

   143

}

   144

   145

/* static */ ThreadObject* ThreadObject::AttachCurrent()

   146

{

   147

	return new ThreadObject_Win32();

   148

}

   149

   150

/* static */ uint ThreadObject::CurrentId()

   151

{

   152

	return GetCurrentThreadId();

   153

}

   154

   155

   156

/**

   157

 * Win32 thread version of ThreadSemaphore.

   158

 */

   159

class ThreadSemaphore_Win32 : public ThreadSemaphore {

   160

private:

   161

	HANDLE m_handle;

   162

   163

public:

   164

	ThreadSemaphore_Win32()

   165

	{

   166

		m_handle = ::CreateEvent(NULL, FALSE, FALSE, NULL);

   167

	}

   168

   169

	/* virtual */ ~ThreadSemaphore_Win32()

   170

	{

   171

		::CloseHandle(m_handle);

   172

	}

   173

   174

	/* virtual */ void Set()

   175

	{

   176

		::SetEvent(m_handle);

   177

	}

   178

   179

	/* virtual */ void Wait()

   180

	{

   181

		::WaitForSingleObject(m_handle, INFINITE);

   182

	}

   183

};

   184

   185

/* static */ ThreadSemaphore *ThreadSemaphore::New()

   186

{

   187

	return new ThreadSemaphore_Win32();

   188

}