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;

    24

    25

public:

    26

	/**

    27

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

    28

	 */

    30

		m_id_thr(0),

    31

		m_h_thr(NULL),

    32

		m_proc(proc),

    33

		m_param(param),

    36

	{

    37

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

    38

		if (m_h_thr == NULL) return;

    39

		ResumeThread(m_h_thr);

    40

	}

    41

    42

	/**

    43

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

    44

	 */

    45

	ThreadObject_Win32() :

    46

		m_id_thr(0),

    47

		m_h_thr(NULL),

    48

		m_proc(NULL),

    49

		m_param(NULL),

    52

	{

    53

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

    54

		if (!ret) return;

    55

		m_id_thr = GetCurrentThreadId();

    56

	}

    57

    58

	/* virtual */ ~ThreadObject_Win32()

    59

	{

    60

		if (m_h_thr != NULL) {

    61

			CloseHandle(m_h_thr);

    62

			m_h_thr = NULL;

    63

		}

    64

	}

    65

    66

	/* virtual */ bool IsRunning()

    67

	{

    68

		if (m_h_thr == NULL) return false;

    69

		DWORD exit_code = 0;

    70

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

    71

		return (exit_code == STILL_ACTIVE);

    72

	}

    73

    74

	/* virtual */ bool WaitForStop()

    75

	{

    76

		/* You can't wait on yourself */

    77

		assert(!IsCurrent());

    78

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

    79

		if (!IsRunning()) return true;

    80

    81

		DWORD res = WaitForSingleObject(m_h_thr, INFINITE);

    82

		return res == WAIT_OBJECT_0;

    83

	}

    84

    85

	/* virtual */ bool Exit()

    86

	{

    87

		/* You can only exit yourself */

    88

		assert(IsCurrent());

    89

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

    90

		if (!IsRunning()) return false;

    91

    92

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

    93

		throw 0;

    94

	}

    95

    96

	/* virtual */ void Join()

    97

	{

    98

		/* You cannot join yourself */

    99

		assert(!IsCurrent());

   100

   101

		WaitForSingleObject(m_h_thr, INFINITE);

   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

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

   123

		return 0;

   124

	}

   125

   126

	/**

   127

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

   128

	 *  function of the creator of the thread.

   129

	 */

   130

	void ThreadProc()

   131

	{

   132

		try {

   133

			m_proc(m_param);

   134

		} catch (...) {

   135

		}

   138

	}

   139

};

   140

   142

{

   144

}

   145

   146

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

   147

{

   148

	return new ThreadObject_Win32();

   149

}

   150

   151

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

   152

{

   153

	return GetCurrentThreadId();

   154

}

   155

   156

   157

/**

   158

 * Win32 thread version of ThreadSemaphore.

   159

 */

   160

class ThreadSemaphore_Win32 : public ThreadSemaphore {

   161

private:

   162

	HANDLE m_handle;

   163

   164

public:

   165

	ThreadSemaphore_Win32()

   166

	{

   167

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

   168

	}

   169

   170

	/* virtual */ ~ThreadSemaphore_Win32()

   171

	{

   172

		::CloseHandle(m_handle);

   173

	}

   174

   175

	/* virtual */ void Set()

   176

	{

   177

		::SetEvent(m_handle);

   178

	}

   179

   180

	/* virtual */ void Wait()

   181

	{

   182

		::WaitForSingleObject(m_handle, INFINITE);

   183

	}

   184

};

   185

   186

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

   187

{

   188

	return new ThreadSemaphore_Win32();

   189

}