user: implement mlibc as the libc, finally.
It's finally done.. Signed-off-by: kaguya <vpshinomiya@protonmail.com>
This commit is contained in:
@@ -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;
|
||||
}
|
||||
Reference in New Issue
Block a user