X-Git-Url: https://git.saurik.com/apple/libpthread.git/blobdiff_plain/3404ec80a22ac38c97e573f77f7cbdf4bddc8ca9..964d3577b041867f776d8eb940bf4a1108ffb97c:/tests/mutex.c

diff --git a/tests/mutex.c b/tests/mutex.c
new file mode 100644
index 0000000..b80d58f
--- /dev/null
+++ b/tests/mutex.c
@@ -0,0 +1,87 @@
+#include <assert.h>
+#include <pthread.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <stdbool.h>
+#include <errno.h>
+
+struct context {
+	pthread_mutex_t mutex;
+	long value;
+	long count;
+};
+
+void *test_thread(void *ptr) {
+	int res;
+	long old;
+	struct context *context = ptr;
+
+	int i = 0;
+	char *str;
+
+	do {
+		bool try = i & 1;
+
+		switch (i++ & 1) {
+			case 0:
+				str = "pthread_mutex_lock";
+				res = pthread_mutex_lock(&context->mutex);
+				break;
+			case 1:
+				str = "pthread_mutex_trylock";
+				res = pthread_mutex_trylock(&context->mutex);
+				break;
+		}
+		if (res != 0) {
+			if (try && res == EBUSY) {
+				continue;
+			}
+			fprintf(stderr, "[%ld] %s: %s\n", context->count, str, strerror(res));
+			abort();
+		}
+		
+		old = __sync_fetch_and_or(&context->value, 1);
+		if ((old & 1) != 0) {
+			fprintf(stderr, "[%ld] OR %lx\n", context->count, old);
+			abort();
+		}
+
+		old = __sync_fetch_and_and(&context->value, 0);
+		if ((old & 1) == 0) {
+			fprintf(stderr, "[%ld] AND %lx\n", context->count, old);
+			abort();
+		}
+	
+		res = pthread_mutex_unlock(&context->mutex);
+		if (res) {
+			fprintf(stderr, "[%ld] pthread_mutex_lock: %s\n", context->count, strerror(res));
+			abort();
+		}
+	} while (__sync_fetch_and_sub(&context->count, 1) > 0);
+	exit(0);
+}
+
+int main(int argc, char *argv[])
+{
+	struct context context = {
+		.mutex = PTHREAD_MUTEX_INITIALIZER,
+		.value = 0,
+		.count = 5000000,
+	};
+	int i;
+	int res;
+	int threads = 16;
+	pthread_t p[threads];
+	for (i = 0; i < threads; ++i) {
+		res = pthread_create(&p[i], NULL, test_thread, &context);
+		assert(res == 0);
+	}
+	for (i = 0; i < threads; ++i) {
+		res = pthread_join(p[i], NULL);
+		assert(res == 0);
+	}
+	
+	return 0;
+}