thread-utils.h 5.32 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__

10 11 12 13 14 15
#if defined(__GNUC__) && defined(GIT_THREADS)
# if (__GNUC__ < 4 || (__GNUC__ == 4 && __GNUC_MINOR__ < 1))
#  error Atomic primitives do not exist on this version of gcc; configure libgit2 with -DTHREADSAFE=OFF
# endif
#endif

Vicent Marti committed
16 17
/* Common operations even if threading has been disabled */
typedef struct {
18
#if defined(GIT_WIN32)
19 20
	volatile long val;
#else
Vicent Marti committed
21
	volatile int val;
22
#endif
Vicent Marti committed
23 24
} git_atomic;

25 26
#ifdef GIT_ARCH_64

27 28 29 30 31 32 33 34
typedef struct {
#if defined(GIT_WIN32)
	__int64 val;
#else
	int64_t val;
#endif
} git_atomic64;

35 36 37 38 39 40 41 42 43 44 45 46
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
47 48
#ifdef GIT_THREADS

49
#ifdef GIT_WIN32
50
#   include "win32/thread.h"
51
#else
52
#   include "unix/pthread.h"
53
#endif
54

55 56 57 58 59 60 61 62 63 64 65
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
66
GIT_INLINE(int) git_atomic_inc(git_atomic *a)
Vicent Marti committed
67
{
68
#if defined(GIT_WIN32)
Vicent Marti committed
69
	return InterlockedIncrement(&a->val);
70 71
#elif defined(__GNUC__)
	return __sync_add_and_fetch(&a->val, 1);
Vicent Marti committed
72 73 74 75 76
#else
#	error "Unsupported architecture for atomic operations"
#endif
}

77 78 79
GIT_INLINE(int) git_atomic_add(git_atomic *a, int32_t addend)
{
#if defined(GIT_WIN32)
80
	return InterlockedExchangeAdd(&a->val, addend);
81 82 83 84 85 86 87
#elif defined(__GNUC__)
	return __sync_add_and_fetch(&a->val, addend);
#else
#	error "Unsupported architecture for atomic operations"
#endif
}

Vicent Marti committed
88
GIT_INLINE(int) git_atomic_dec(git_atomic *a)
Vicent Marti committed
89
{
90
#if defined(GIT_WIN32)
Vicent Marti committed
91
	return InterlockedDecrement(&a->val);
92 93
#elif defined(__GNUC__)
	return __sync_sub_and_fetch(&a->val, 1);
Vicent Marti committed
94 95 96 97 98
#else
#	error "Unsupported architecture for atomic operations"
#endif
}

99
GIT_INLINE(void *) git___compare_and_swap(
100
	void * volatile *ptr, void *oldval, void *newval)
101
{
102
	volatile void *foundval;
103
#if defined(GIT_WIN32)
nulltoken committed
104
	foundval = InterlockedCompareExchangePointer((volatile PVOID *)ptr, newval, oldval);
105
#elif defined(__GNUC__)
106
	foundval = __sync_val_compare_and_swap(ptr, oldval, newval);
107 108 109
#else
#	error "Unsupported architecture for atomic operations"
#endif
110
	return (foundval == oldval) ? oldval : newval;
111 112
}

113 114 115 116 117 118 119 120 121 122
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
}

123 124 125
#ifdef GIT_ARCH_64

GIT_INLINE(int64_t) git_atomic64_add(git_atomic64 *a, int64_t addend)
126 127
{
#if defined(GIT_WIN32)
128
	return InterlockedExchangeAdd64(&a->val, addend);
129 130 131 132 133 134 135
#elif defined(__GNUC__)
	return __sync_add_and_fetch(&a->val, addend);
#else
#	error "Unsupported architecture for atomic operations"
#endif
}

136 137
#endif

Vicent Marti committed
138 139 140
#else

#define git_thread unsigned int
141
#define git_thread_create(thread, start_routine, arg) 0
Vicent Marti committed
142 143 144 145
#define git_thread_join(id, status) (void)0

/* Pthreads Mutex */
#define git_mutex unsigned int
146 147 148 149
GIT_INLINE(int) git_mutex_init(git_mutex *mutex) \
	{ GIT_UNUSED(mutex); return 0; }
GIT_INLINE(int) git_mutex_lock(git_mutex *mutex) \
	{ GIT_UNUSED(mutex); return 0; }
Vicent Marti committed
150 151 152
#define git_mutex_unlock(a) (void)0
#define git_mutex_free(a) (void)0

153
/* Pthreads condition vars */
Vicent Marti committed
154 155 156 157 158 159 160
#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

161 162 163 164 165 166 167 168 169 170 171
/* 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


172 173 174 175 176
GIT_INLINE(void) git_atomic_set(git_atomic *a, int val)
{
	a->val = val;
}

Vicent Marti committed
177
GIT_INLINE(int) git_atomic_inc(git_atomic *a)
Vicent Marti committed
178 179 180 181
{
	return ++a->val;
}

182 183 184 185 186 187
GIT_INLINE(int) git_atomic_add(git_atomic *a, int32_t addend)
{
	a->val += addend;
	return a->val;
}

Vicent Marti committed
188
GIT_INLINE(int) git_atomic_dec(git_atomic *a)
Vicent Marti committed
189 190 191 192
{
	return --a->val;
}

193
GIT_INLINE(void *) git___compare_and_swap(
194
	void * volatile *ptr, void *oldval, void *newval)
195 196 197 198 199 200 201 202
{
	if (*ptr == oldval)
		*ptr = newval;
	else
		oldval = newval;
	return oldval;
}

203 204 205 206 207 208 209 210
GIT_INLINE(volatile void *) git___swap(
	void * volatile *ptr, void *newval)
{
	volatile void *old = *ptr;
	*ptr = newval;
	return old;
}

211 212 213
#ifdef GIT_ARCH_64

GIT_INLINE(int64_t) git_atomic64_add(git_atomic64 *a, int64_t addend)
214 215 216 217 218
{
	a->val += addend;
	return a->val;
}

Vicent Marti committed
219
#endif
220

221 222
#endif

223 224 225 226 227
GIT_INLINE(int) git_atomic_get(git_atomic *a)
{
	return (int)a->val;
}

228 229 230 231
/* 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) \
232
	git___compare_and_swap((void * volatile *)P, O, N)
233

234
#define git__swap(ptr, val) (void *)git___swap((void * volatile *)&ptr, val)
235

236
extern int git_online_cpus(void);
237

238
#if defined(GIT_THREADS) && defined(_MSC_VER)
239 240 241 242 243 244 245
# define GIT_MEMORY_BARRIER MemoryBarrier()
#elif defined(GIT_THREADS)
# define GIT_MEMORY_BARRIER __sync_synchronize()
#else
# define GIT_MEMORY_BARRIER /* noop */
#endif

246
#endif