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 0;

    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 (...) {

   132

		}

   133

	}

   134

};

   135

   136

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

   137

{

   138

	return new ThreadObject_Win32(proc, param);

   139

}

   140

   141

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

   142

{

   143

	return new ThreadObject_Win32();

   144

}

   145

   146

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

   147

{

   148

	return GetCurrentThreadId();

   149

}

   150

   151

   152

/**

   153

 * Win32 thread version of ThreadSemaphore.

   154

 */

   155

class ThreadSemaphore_Win32 : public ThreadSemaphore {

   156

private:

   157

	HANDLE m_handle;

   158

   159

public:

   160

	ThreadSemaphore_Win32()

   161

	{

   162

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

   163

	}

   164

   165

	/* virtual */ ~ThreadSemaphore_Win32()

   166

	{

   167

		::CloseHandle(m_handle);

   168

	}

   169

   170

	/* virtual */ void Set()

   171

	{

   172

		::SetEvent(m_handle);

   173

	}

   174

   175

	/* virtual */ void Wait()

   176

	{

   177

		::WaitForSingleObject(m_handle, INFINITE);

   178

	}

   179

};

   180

   181

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

   182

{

   183

	return new ThreadSemaphore_Win32();

   184

}