1 files changed, 150 insertions, 0 deletions
diff --git a/lib/mlibc/tests/posix/pthread_cancel.c b/lib/mlibc/tests/posix/pthread_cancel.c
new file mode 100644
index 0000000..ca4ca34
--- /dev/null
+++ b/lib/mlibc/tests/posix/pthread_cancel.c
@@ -0,0 +1,150 @@
+#include <assert.h>
+#include <pthread.h>
+#include <unistd.h>
+
+_Atomic int worker_ready = 0;
+_Atomic int main_ready = 0;
+
+static void *worker1(void *arg) {
+	(void)arg;
+	assert(!pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL));
+	assert(!pthread_setcanceltype(PTHREAD_CANCEL_ASYNCHRONOUS, NULL));
+
+	worker_ready = 1;
+
+	while (1) sleep(1);
+
+	return NULL;
+}
+
+static void *worker2(void *arg) {
+	(void) arg;
+	assert(!pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL));
+
+	worker_ready = 1;
+
+	while(!main_ready);
+
+	// Cancellation point - we should cancel right now
+	sleep(1);
+
+	assert(!"Expected to be cancelled!");
+	__builtin_unreachable();
+}
+
+static void *worker3(void *arg) {
+	(void) arg;
+	assert(!pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL));
+
+	worker_ready = 1;
+
+	while(!main_ready);
+
+	// Cancellation point - we should cancel right now
+	pthread_testcancel();
+
+	assert(!"Expected to be cancelled!");
+	__builtin_unreachable();
+}
+
+static void *worker4(void *arg) {
+	(void) arg;
+	assert(!pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL));
+
+	worker_ready = 1;
+	sleep(1);
+
+	// We expect to be canceled during the sleep
+
+	assert(!"Expected to be cancelled!");
+	__builtin_unreachable();
+}
+
+static void *worker5(void *arg) {
+	assert(!pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL));
+
+	worker_ready = 1;
+
+	while(!main_ready);
+
+	// Cancellation point - we should NOT cancel right now
+	pthread_testcancel();
+
+	int *arg_int = (int*)arg;
+	*arg_int = 1;
+
+	return NULL;
+}
+
+static void check_result(pthread_t thread) {
+
+	void *ret_val = NULL;
+	int ret = pthread_join(thread, &ret_val);
+	assert(!ret);
+	assert(ret_val == PTHREAD_CANCELED);
+}
+
+int main() {
+	pthread_t thread;
+	int ret = pthread_create(&thread, NULL, &worker1, NULL);
+	assert(!ret);
+
+	while (!worker_ready);
+	ret = pthread_cancel(thread);
+	assert(!ret);
+	check_result(thread);
+
+	main_ready = 0;
+	worker_ready = 0;
+	main_ready = 0;
+	ret = pthread_create(&thread, NULL, &worker2, NULL);
+	assert(!ret);
+
+	while(!worker_ready);
+	ret = pthread_cancel(thread);
+	assert(!ret);
+	main_ready = 1;
+	check_result(thread);
+
+	main_ready = 0;
+	worker_ready = 0;
+	main_ready = 0;
+	ret = pthread_create(&thread, NULL, &worker3, NULL);
+	assert(!ret);
+
+	while(!worker_ready);
+	ret = pthread_cancel(thread);
+	assert(!ret);
+	main_ready = 1;
+	check_result(thread);
+
+	worker_ready = 0;
+	main_ready = 0;
+	ret = pthread_create(&thread, NULL, &worker4, NULL);
+	assert(!ret);
+
+	while(!worker_ready);
+	ret = pthread_cancel(thread);
+	assert(!ret);
+	main_ready = 1;
+	check_result(thread);
+
+	// Test for bug where pthread_testcancel() was not checking if
+	// cancellation was triggered properly.
+	worker_ready = 0;
+	int pthread_arg = 0;
+	main_ready = 0;
+	ret = pthread_create(&thread, NULL, &worker5, &pthread_arg);
+	assert(!ret);
+
+	while(!worker_ready);
+	main_ready = 1;
+
+	void *ret_val = NULL;
+	ret = pthread_join(thread, &ret_val);
+	assert(!ret);
+	assert(!ret_val);
+	assert(pthread_arg == 1);
+
+	return 0;
+}