thread-utils.h 6.99 KB
Newer Older
Vicent Marti committed
1
/*
Edward Thomson committed
2
 * Copyright (C) the libgit2 contributors. All rights reserved.
Vicent Marti committed
3 4 5 6
 *
 * This file is part of libgit2, distributed under the GNU GPL v2 with
 * a Linking Exception. For full terms see the included COPYING file.
 */
7 8 9
#ifndef INCLUDE_thread_utils_h__
#define INCLUDE_thread_utils_h__

Vicent Marti committed
10 11
/* Common operations even if threading has been disabled */
typedef struct {
12
#if defined(GIT_WIN32)
13 14
	volatile long val;
#else
Vicent Marti committed
15
	volatile int val;
16
#endif
Vicent Marti committed
17 18
} git_atomic;

19 20
#ifdef GIT_ARCH_64

21 22 23 24 25 26 27 28
typedef struct {
#if defined(GIT_WIN32)
	__int64 val;
#else
	int64_t val;
#endif
} git_atomic64;

29 30 31 32 33 34 35 36 37 38 39 40
typedef git_atomic64 git_atomic_ssize;

#define git_atomic_ssize_add git_atomic64_add

#else

typedef git_atomic git_atomic_ssize;

#define git_atomic_ssize_add git_atomic_add

#endif

Vicent Marti committed
41 42
#ifdef GIT_THREADS

43
#define git_thread pthread_t
44 45
#define git_thread_create(thread, attr, start_routine, arg) \
	pthread_create(thread, attr, start_routine, arg)
46 47 48 49
#define git_thread_kill(thread) pthread_cancel(thread)
#define git_thread_exit(status)	pthread_exit(status)
#define git_thread_join(id, status) pthread_join(id, status)

50 51 52 53 54 55
#if defined(GIT_WIN32)
#define git_thread_yield() Sleep(0)
#else
#define git_thread_yield() sched_yield()
#endif

56 57
/* Pthreads Mutex */
#define git_mutex pthread_mutex_t
Vicent Marti committed
58 59
#define git_mutex_init(a)	pthread_mutex_init(a, NULL)
#define git_mutex_lock(a)	pthread_mutex_lock(a)
60
#define git_mutex_unlock(a) pthread_mutex_unlock(a)
Vicent Marti committed
61
#define git_mutex_free(a)	pthread_mutex_destroy(a)
62

63 64 65 66 67 68 69
/* Pthreads condition vars */
#define git_cond pthread_cond_t
#define git_cond_init(c)	pthread_cond_init(c, NULL)
#define git_cond_free(c) 	pthread_cond_destroy(c)
#define git_cond_wait(c, l)	pthread_cond_wait(c, l)
#define git_cond_signal(c)	pthread_cond_signal(c)
#define git_cond_broadcast(c)	pthread_cond_broadcast(c)
Vicent Marti committed
70

71 72 73 74 75 76 77 78 79
/* Pthread (-ish) rwlock
 *
 * This differs from normal pthreads rwlocks in two ways:
 * 1. Separate APIs for releasing read locks and write locks (as
 *    opposed to the pure POSIX API which only has one unlock fn)
 * 2. You should not use recursive read locks (i.e. grabbing a read
 *    lock in a thread that already holds a read lock) because the
 *    Windows implementation doesn't support it
 */
80 81 82
#define git_rwlock pthread_rwlock_t
#define git_rwlock_init(a)		pthread_rwlock_init(a, NULL)
#define git_rwlock_rdlock(a)	pthread_rwlock_rdlock(a)
83
#define git_rwlock_rdunlock(a)	pthread_rwlock_rdunlock(a)
84
#define git_rwlock_wrlock(a)	pthread_rwlock_wrlock(a)
85
#define git_rwlock_wrunlock(a)	pthread_rwlock_wrunlock(a)
86 87 88
#define git_rwlock_free(a)		pthread_rwlock_destroy(a)
#define GIT_RWLOCK_STATIC_INIT	PTHREAD_RWLOCK_INITIALIZER

89 90 91 92 93
#ifndef GIT_WIN32
#define pthread_rwlock_rdunlock pthread_rwlock_unlock
#define pthread_rwlock_wrunlock pthread_rwlock_unlock
#endif

94

95 96 97 98 99 100 101 102 103 104 105
GIT_INLINE(void) git_atomic_set(git_atomic *a, int val)
{
#if defined(GIT_WIN32)
	InterlockedExchange(&a->val, (LONG)val);
#elif defined(__GNUC__)
	__sync_lock_test_and_set(&a->val, val);
#else
#	error "Unsupported architecture for atomic operations"
#endif
}

Vicent Marti committed
106
GIT_INLINE(int) git_atomic_inc(git_atomic *a)
Vicent Marti committed
107
{
108
#if defined(GIT_WIN32)
Vicent Marti committed
109
	return InterlockedIncrement(&a->val);
110 111
#elif defined(__GNUC__)
	return __sync_add_and_fetch(&a->val, 1);
Vicent Marti committed
112 113 114 115 116
#else
#	error "Unsupported architecture for atomic operations"
#endif
}

117 118 119
GIT_INLINE(int) git_atomic_add(git_atomic *a, int32_t addend)
{
#if defined(GIT_WIN32)
120
	return InterlockedExchangeAdd(&a->val, addend);
121 122 123 124 125 126 127
#elif defined(__GNUC__)
	return __sync_add_and_fetch(&a->val, addend);
#else
#	error "Unsupported architecture for atomic operations"
#endif
}

Vicent Marti committed
128
GIT_INLINE(int) git_atomic_dec(git_atomic *a)
Vicent Marti committed
129
{
130
#if defined(GIT_WIN32)
Vicent Marti committed
131
	return InterlockedDecrement(&a->val);
132 133
#elif defined(__GNUC__)
	return __sync_sub_and_fetch(&a->val, 1);
Vicent Marti committed
134 135 136 137 138
#else
#	error "Unsupported architecture for atomic operations"
#endif
}

139
GIT_INLINE(void *) git___compare_and_swap(
140
	void * volatile *ptr, void *oldval, void *newval)
141
{
142
	volatile void *foundval;
143
#if defined(GIT_WIN32)
nulltoken committed
144
	foundval = InterlockedCompareExchangePointer((volatile PVOID *)ptr, newval, oldval);
145
#elif defined(__GNUC__)
146
	foundval = __sync_val_compare_and_swap(ptr, oldval, newval);
147 148 149
#else
#	error "Unsupported architecture for atomic operations"
#endif
150
	return (foundval == oldval) ? oldval : newval;
151 152
}

153 154 155 156 157 158 159 160 161 162
GIT_INLINE(volatile void *) git___swap(
	void * volatile *ptr, void *newval)
{
#if defined(GIT_WIN32)
	return InterlockedExchangePointer(ptr, newval);
#else
	return __sync_lock_test_and_set(ptr, newval);
#endif
}

163 164 165
#ifdef GIT_ARCH_64

GIT_INLINE(int64_t) git_atomic64_add(git_atomic64 *a, int64_t addend)
166 167
{
#if defined(GIT_WIN32)
168
	return InterlockedExchangeAdd64(&a->val, addend);
169 170 171 172 173 174 175
#elif defined(__GNUC__)
	return __sync_add_and_fetch(&a->val, addend);
#else
#	error "Unsupported architecture for atomic operations"
#endif
}

176 177
#endif

Vicent Marti committed
178 179 180
#else

#define git_thread unsigned int
181
#define git_thread_create(thread, attr, start_routine, arg) 0
Vicent Marti committed
182 183 184
#define git_thread_kill(thread) (void)0
#define git_thread_exit(status) (void)0
#define git_thread_join(id, status) (void)0
185
#define git_thread_yield() (void)0
Vicent Marti committed
186 187 188

/* Pthreads Mutex */
#define git_mutex unsigned int
Russell Belfer committed
189
#define git_mutex_init(a) 0
190
#define git_mutex_lock(a) 0
Vicent Marti committed
191 192 193
#define git_mutex_unlock(a) (void)0
#define git_mutex_free(a) (void)0

194
/* Pthreads condition vars */
Vicent Marti committed
195 196 197 198 199 200 201
#define git_cond unsigned int
#define git_cond_init(c, a)	(void)0
#define git_cond_free(c) (void)0
#define git_cond_wait(c, l)	(void)0
#define git_cond_signal(c) (void)0
#define git_cond_broadcast(c) (void)0

202 203 204 205 206 207 208 209 210 211 212
/* Pthreads rwlock */
#define git_rwlock unsigned int
#define git_rwlock_init(a)		0
#define git_rwlock_rdlock(a)	0
#define git_rwlock_rdunlock(a)	(void)0
#define git_rwlock_wrlock(a)	0
#define git_rwlock_wrunlock(a)	(void)0
#define git_rwlock_free(a)		(void)0
#define GIT_RWLOCK_STATIC_INIT	0


213 214 215 216 217
GIT_INLINE(void) git_atomic_set(git_atomic *a, int val)
{
	a->val = val;
}

Vicent Marti committed
218
GIT_INLINE(int) git_atomic_inc(git_atomic *a)
Vicent Marti committed
219 220 221 222
{
	return ++a->val;
}

223 224 225 226 227 228
GIT_INLINE(int) git_atomic_add(git_atomic *a, int32_t addend)
{
	a->val += addend;
	return a->val;
}

Vicent Marti committed
229
GIT_INLINE(int) git_atomic_dec(git_atomic *a)
Vicent Marti committed
230 231 232 233
{
	return --a->val;
}

234
GIT_INLINE(void *) git___compare_and_swap(
235
	void * volatile *ptr, void *oldval, void *newval)
236 237 238 239 240 241 242 243
{
	if (*ptr == oldval)
		*ptr = newval;
	else
		oldval = newval;
	return oldval;
}

244 245 246 247 248 249 250 251
GIT_INLINE(volatile void *) git___swap(
	void * volatile *ptr, void *newval)
{
	volatile void *old = *ptr;
	*ptr = newval;
	return old;
}

252 253 254
#ifdef GIT_ARCH_64

GIT_INLINE(int64_t) git_atomic64_add(git_atomic64 *a, int64_t addend)
255 256 257 258 259
{
	a->val += addend;
	return a->val;
}

Vicent Marti committed
260
#endif
261

262 263
#endif

264 265 266 267 268
GIT_INLINE(int) git_atomic_get(git_atomic *a)
{
	return (int)a->val;
}

269 270 271 272
/* Atomically replace oldval with newval
 * @return oldval if it was replaced or newval if it was not
 */
#define git__compare_and_swap(P,O,N) \
273
	git___compare_and_swap((void * volatile *)P, O, N)
274

275
#define git__swap(ptr, val) (void *)git___swap((void * volatile *)&ptr, val)
276

277
extern int git_online_cpus(void);
278

279 280 281 282 283 284 285 286
#if defined(GIT_THREADS) && defined(GIT_WIN32)
# define GIT_MEMORY_BARRIER MemoryBarrier()
#elif defined(GIT_THREADS)
# define GIT_MEMORY_BARRIER __sync_synchronize()
#else
# define GIT_MEMORY_BARRIER /* noop */
#endif

287
#endif /* INCLUDE_thread_utils_h__ */