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

    24

public:

    25

	/**

    26

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

    27

	 */

    28

	ThreadObject_Win32(OTTDThreadFunc proc, void *param) :

    29

		m_id_thr(0),

    30

		m_h_thr(NULL),

    31

		m_proc(proc),

    32

		m_param(param),

    33

		m_attached(false)

    34

	{

    35

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

    36

		if (m_h_thr == NULL) return;

    37

		ResumeThread(m_h_thr);

    38

	}

    39

    40

	/**

    41

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

    42

	 */

    43

	ThreadObject_Win32() :

    44

		m_id_thr(0),

    45

		m_h_thr(NULL),

    46

		m_proc(NULL),

    47

		m_param(NULL),

    48

		m_attached(false)

    49

	{

    50

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

    51

		if (!ret) return;

    52

		m_id_thr = GetCurrentThreadId();

    53

	}

    54

    55

	/* virtual */ ~ThreadObject_Win32()

    56

	{

    57

		if (m_h_thr != NULL) {

    58

			CloseHandle(m_h_thr);

    59

			m_h_thr = NULL;

    60

		}

    61

	}

    62

    63

	/* virtual */ bool IsRunning()

    64

	{

    65

		if (m_h_thr == NULL) return false;

    66

		DWORD exit_code = 0;

    67

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

    68

		return (exit_code == STILL_ACTIVE);

    69

	}

    70

    71

	/* virtual */ bool WaitForStop()

    72

	{

    73

		/* You can't wait on yourself */

    74

		assert(!IsCurrent());

    75

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

    76

		if (!IsRunning()) return true;

    77

    78

		DWORD res = WaitForSingleObject(m_h_thr, INFINITE);

    79

		return res == WAIT_OBJECT_0;

    80

	}

    81

    82

	/* virtual */ bool Exit()

    83

	{

    84

		/* You can only exit yourself */

    85

		assert(IsCurrent());

    86

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

    87

		if (!IsRunning()) return false;

    88

    89

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

    90

		throw OTTDThreadExitSignal();

    91

	}

    92

    93

	/* virtual */ void Join()

    94

	{

    95

		/* You cannot join yourself */

    96

		assert(!IsCurrent());

    97

    98

		WaitForSingleObject(m_h_thr, INFINITE);

    99

	}

   100

   101

	/* virtual */ bool IsCurrent()

   102

	{

   103

		DWORD id_cur = GetCurrentThreadId();

   104

		return id_cur == m_id_thr;

   105

	}

   106

   107

	/* virtual */ uint GetId()

   108

	{

   109

		return m_id_thr;

   110

	}

   111

   112

private:

   113

	/**

   114

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

   115

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

   116

	 */

   117

	static uint CALLBACK stThreadProc(void *thr)

   118

	{

   119

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

   120

		return 0;

   121

	}

   122

   123

	/**

   124

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

   125

	 *  function of the creator of the thread.

   126

	 */

   127

	void ThreadProc()

   128

	{

   129

		try {

   130

			m_proc(m_param);

   131

		} catch (OTTDThreadExitSignal e) {

   132

		} catch (...) {

   133

			NOT_REACHED();

   134

		}

   135

	}

   136

};

   137

   138

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

   139

{

   140

	return new ThreadObject_Win32(proc, param);

   141

}

   142

   143

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

   144

{

   145

	return new ThreadObject_Win32();

   146

}

   147

   148

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

   149

{

   150

	return GetCurrentThreadId();

   151

}

   152

   153

   154

/**

   155

 * Win32 thread version of ThreadSemaphore.

   156

 */

   157

class ThreadSemaphore_Win32 : public ThreadSemaphore {

   158

private:

   159

	HANDLE m_handle;

   160

   161

public:

   162

	ThreadSemaphore_Win32()

   163

	{

   164

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

   165

	}

   166

   167

	/* virtual */ ~ThreadSemaphore_Win32()

   168

	{

   169

		::CloseHandle(m_handle);

   170

	}

   171

   172

	/* virtual */ void Set()

   173

	{

   174

		::SetEvent(m_handle);

   175

	}

   176

   177

	/* virtual */ void Wait()

   178

	{

   179

		::WaitForSingleObject(m_handle, INFINITE);

   180

	}

   181

};

   182

   183

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

   184

{

   185

	return new ThreadSemaphore_Win32();

   186

}