b08f5b9ceb
1. Use arch_prctl to replace RDFSBASE/WRFSBASE Ptrace can't get right value if WRFSBASE is called which will make debugger fail in simulation mode. Use arch_prctl to replace these instructions in simulation mode. 2. Disable the busy thread in exit_group test exit_group doesn't have a real implementation yet but test under SGX simulation mode give core dump for exit_group test. Disable the busy loop thread and the core dump disappear. 3. Add SDK lib path to LD_LIBRARY_PATH Linker sometims can't find urts_sim and uae_service_sim when running. Explicitly add path to LD_LIBRARY_PATH when running occlum command. Signed-off-by: sanqian.hcy <sanqian.hcy@antfin.com>
86 lines
2.7 KiB
C
86 lines
2.7 KiB
C
#define _GNU_SOURCE
|
|
#include <sys/types.h>
|
|
#include <sys/syscall.h>
|
|
#include <unistd.h>
|
|
#include <pthread.h>
|
|
#include <stdio.h>
|
|
#include <linux/futex.h>
|
|
#include "test.h"
|
|
|
|
// ============================================================================
|
|
// Test case
|
|
// ============================================================================
|
|
|
|
//
|
|
// Three types of threads that will not exit voluntarily
|
|
//
|
|
|
|
// FIXME: Disable this test for NOW because exit_group does not have a real implementation yet
|
|
// and SGX simlulation mode will fail this test.
|
|
// Type 1: a busy loop thread
|
|
// static void* busyloop_thread_func(void* _) {
|
|
// while (1) {
|
|
// // By calling getpid, we give the LibOS a chance to force the thread
|
|
// // to terminate if exit_group is called by any thread in a thread group
|
|
// getpid();
|
|
// }
|
|
// return NULL;
|
|
// }
|
|
|
|
// Type 2: a sleeping thread
|
|
static void* sleeping_thread_func(void* _) {
|
|
unsigned int a_year_in_sec = 365 * 24 * 60 * 60;
|
|
sleep(a_year_in_sec);
|
|
return NULL;
|
|
}
|
|
|
|
// Type 3: a thead that keeps waiting on a futex
|
|
static void* futex_wait_thread_func(void* _) {
|
|
// Wait on a futex forever
|
|
int my_private_futex = 0;
|
|
syscall(SYS_futex, &my_private_futex, FUTEX_WAIT, my_private_futex);
|
|
return NULL;
|
|
}
|
|
|
|
|
|
// exit_group syscall should terminate all threads in a thread group.
|
|
int test_exit_group_to_force_threads_terminate(void) {
|
|
// Create three types of threads that will not exit voluntarily
|
|
// pthread_t busyloop_thread;
|
|
// if (pthread_create(&busyloop_thread, NULL, busyloop_thread_func, NULL) < 0) {
|
|
// printf("ERROR: pthread_create failed\n");
|
|
// return -1;
|
|
// }
|
|
pthread_t sleeping_thread;
|
|
if (pthread_create(&sleeping_thread, NULL, sleeping_thread_func, NULL) < 0) {
|
|
printf("ERROR: pthread_create failed\n");
|
|
return -1;
|
|
}
|
|
pthread_t futex_wait_thread;
|
|
if (pthread_create(&futex_wait_thread, NULL, futex_wait_thread_func, NULL) < 0) {
|
|
printf("ERROR: pthread_create failed\n");
|
|
return -1;
|
|
}
|
|
|
|
// Sleep for a while to make sure all three threads are running
|
|
useconds_t _200ms = 200 * 1000;
|
|
usleep(_200ms);
|
|
|
|
// exit_group syscall will be called eventually by libc's exit, after the
|
|
// main function returns. If Occlum can terminate normally, this means
|
|
// exit_group syscall taking effect.
|
|
return 0;
|
|
}
|
|
|
|
// ============================================================================
|
|
// Test suite
|
|
// ============================================================================
|
|
|
|
static test_case_t test_cases[] = {
|
|
TEST_CASE(test_exit_group_to_force_threads_terminate)
|
|
};
|
|
|
|
int main() {
|
|
return test_suite_run(test_cases, ARRAY_SIZE(test_cases));
|
|
}
|