pthread.c 5.88 KB
Newer Older
1
/*
Edward Thomson committed
2
 * Copyright (C) the libgit2 contributors. All rights reserved.
3
 *
Vicent Marti committed
4 5
 * This file is part of libgit2, distributed under the GNU GPL v2 with
 * a Linking Exception. For full terms see the included COPYING file.
6 7 8
 */

#include "pthread.h"
Russell Belfer committed
9
#include "../global.h"
10

11 12 13 14 15 16 17 18 19 20
#define CLEAN_THREAD_EXIT 0x6F012842

/* The thread procedure stub used to invoke the caller's procedure
 * and capture the return value for later collection. Windows will
 * only hold a DWORD, but we need to be able to store an entire
 * void pointer. This requires the indirection. */
static DWORD WINAPI git_win32__threadproc(LPVOID lpParameter)
{
	git_win32_thread *thread = lpParameter;

Philip Kelley committed
21
	thread->result = thread->proc(thread->param);
22

23
	git__free_tls_data();
24

25 26 27 28 29
	return CLEAN_THREAD_EXIT;
}

int git_win32__thread_create(
	git_win32_thread *GIT_RESTRICT thread,
30 31 32
	const pthread_attr_t *GIT_RESTRICT attr,
	void *(*start_routine)(void*),
	void *GIT_RESTRICT arg)
33
{
34
	GIT_UNUSED(attr);
35

Philip Kelley committed
36 37
	thread->result = NULL;
	thread->param = arg;
38 39 40 41 42
	thread->proc = start_routine;
	thread->thread = CreateThread(
		NULL, 0, git_win32__threadproc, thread, 0, NULL);

	return thread->thread ? 0 : -1;
43 44
}

45 46 47
int git_win32__thread_join(
	git_win32_thread *thread,
	void **value_ptr)
48
{
49 50 51 52 53 54 55 56 57
	DWORD exit;

	if (WaitForSingleObject(thread->thread, INFINITE) != WAIT_OBJECT_0)
		return -1;

	if (!GetExitCodeThread(thread->thread, &exit)) {
		CloseHandle(thread->thread);
		return -1;
	}
58

59 60 61 62
	/* Check for the thread having exited uncleanly. If exit was unclean,
	 * then we don't have a return value to give back to the caller. */
	if (exit != CLEAN_THREAD_EXIT) {
		assert(false);
Philip Kelley committed
63
		thread->result = NULL;
64 65
	}

66
	if (value_ptr)
Philip Kelley committed
67
		*value_ptr = thread->result;
68 69 70

	CloseHandle(thread->thread);
	return 0;
71 72
}

73 74 75
int pthread_mutex_init(
	pthread_mutex_t *GIT_RESTRICT mutex,
	const pthread_mutexattr_t *GIT_RESTRICT mutexattr)
76
{
77
	GIT_UNUSED(mutexattr);
Vicent Marti committed
78 79
	InitializeCriticalSection(mutex);
	return 0;
80 81 82 83
}

int pthread_mutex_destroy(pthread_mutex_t *mutex)
{
Vicent Marti committed
84 85
	DeleteCriticalSection(mutex);
	return 0;
86 87
}

Vicent Marti committed
88
int pthread_mutex_lock(pthread_mutex_t *mutex)
89
{
Vicent Marti committed
90 91
	EnterCriticalSection(mutex);
	return 0;
92 93
}

Vicent Marti committed
94
int pthread_mutex_unlock(pthread_mutex_t *mutex)
95
{
Vicent Marti committed
96 97
	LeaveCriticalSection(mutex);
	return 0;
98 99
}

100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123
int pthread_cond_init(pthread_cond_t *cond, const pthread_condattr_t *attr)
{
	/* We don't support non-default attributes. */
	if (attr)
		return EINVAL;

	/* This is an auto-reset event. */
	*cond = CreateEventW(NULL, FALSE, FALSE, NULL);
	assert(*cond);

	/* If we can't create the event, claim that the reason was out-of-memory.
	 * The actual reason can be fetched with GetLastError(). */
	return *cond ? 0 : ENOMEM;
}

int pthread_cond_destroy(pthread_cond_t *cond)
{
	BOOL closed;

	if (!cond)
		return EINVAL;

	closed = CloseHandle(*cond);
	assert(closed);
124
	GIT_UNUSED(closed);
125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145

	*cond = NULL;
	return 0;
}

int pthread_cond_wait(pthread_cond_t *cond, pthread_mutex_t *mutex)
{
	int error;
	DWORD wait_result;

	if (!cond || !mutex)
		return EINVAL;

	/* The caller must be holding the mutex. */
	error = pthread_mutex_unlock(mutex);

	if (error)
		return error;

	wait_result = WaitForSingleObject(*cond, INFINITE);
	assert(WAIT_OBJECT_0 == wait_result);
146
	GIT_UNUSED(wait_result);
147 148 149 150 151 152 153 154 155 156 157 158 159

	return pthread_mutex_lock(mutex);
}

int pthread_cond_signal(pthread_cond_t *cond)
{
	BOOL signaled;

	if (!cond)
		return EINVAL;

	signaled = SetEvent(*cond);
	assert(signaled);
160
	GIT_UNUSED(signaled);
161 162 163 164

	return 0;
}

165 166 167 168
/* pthread_cond_broadcast is not implemented because doing so with just
 * Win32 events is quite complicated, and no caller in libgit2 uses it
 * yet.
 */
169 170
int pthread_num_processors_np(void)
{
Vicent Marti committed
171 172
	DWORD_PTR p, s;
	int n = 0;
173

Vicent Marti committed
174 175 176
	if (GetProcessAffinityMask(GetCurrentProcess(), &p, &s))
		for (; p; p >>= 1)
			n += p&1;
177

Vicent Marti committed
178
	return n ? n : 1;
179 180
}

181
typedef void (WINAPI *win32_srwlock_fn)(GIT_SRWLOCK *);
182 183 184 185 186 187 188

static win32_srwlock_fn win32_srwlock_initialize;
static win32_srwlock_fn win32_srwlock_acquire_shared;
static win32_srwlock_fn win32_srwlock_release_shared;
static win32_srwlock_fn win32_srwlock_acquire_exclusive;
static win32_srwlock_fn win32_srwlock_release_exclusive;

189 190 191 192
int pthread_rwlock_init(
	pthread_rwlock_t *GIT_RESTRICT lock,
	const pthread_rwlockattr_t *GIT_RESTRICT attr)
{
193
	GIT_UNUSED(attr);
194 195 196 197 198 199

	if (win32_srwlock_initialize)
		win32_srwlock_initialize(&lock->native.srwl);
	else
		InitializeCriticalSection(&lock->native.csec);

200 201 202 203 204
	return 0;
}

int pthread_rwlock_rdlock(pthread_rwlock_t *lock)
{
205 206 207 208 209
	if (win32_srwlock_acquire_shared)
		win32_srwlock_acquire_shared(&lock->native.srwl);
	else
		EnterCriticalSection(&lock->native.csec);

210 211 212 213 214
	return 0;
}

int pthread_rwlock_rdunlock(pthread_rwlock_t *lock)
{
215 216 217 218 219
	if (win32_srwlock_release_shared)
		win32_srwlock_release_shared(&lock->native.srwl);
	else
		LeaveCriticalSection(&lock->native.csec);

220 221 222 223 224
	return 0;
}

int pthread_rwlock_wrlock(pthread_rwlock_t *lock)
{
225 226 227 228 229
	if (win32_srwlock_acquire_exclusive)
		win32_srwlock_acquire_exclusive(&lock->native.srwl);
	else
		EnterCriticalSection(&lock->native.csec);

230 231 232 233 234
	return 0;
}

int pthread_rwlock_wrunlock(pthread_rwlock_t *lock)
{
235 236 237 238 239
	if (win32_srwlock_release_exclusive)
		win32_srwlock_release_exclusive(&lock->native.srwl);
	else
		LeaveCriticalSection(&lock->native.csec);

240 241 242 243 244
	return 0;
}

int pthread_rwlock_destroy(pthread_rwlock_t *lock)
{
245 246 247 248 249 250 251 252
	if (!win32_srwlock_initialize)
		DeleteCriticalSection(&lock->native.csec);
	git__memzero(lock, sizeof(*lock));
	return 0;
}

int win32_pthread_initialize(void)
{
253 254 255 256 257 258 259 260 261 262 263 264 265
	HMODULE hModule = GetModuleHandleW(L"kernel32");

	if (hModule) {
		win32_srwlock_initialize = (win32_srwlock_fn)
			GetProcAddress(hModule, "InitializeSRWLock");
		win32_srwlock_acquire_shared = (win32_srwlock_fn)
			GetProcAddress(hModule, "AcquireSRWLockShared");
		win32_srwlock_release_shared = (win32_srwlock_fn)
			GetProcAddress(hModule, "ReleaseSRWLockShared");
		win32_srwlock_acquire_exclusive = (win32_srwlock_fn)
			GetProcAddress(hModule, "AcquireSRWLockExclusive");
		win32_srwlock_release_exclusive = (win32_srwlock_fn)
			GetProcAddress(hModule, "ReleaseSRWLockExclusive");
266 267
	}

268 269
	return 0;
}