ladybird/Tests/LibC/TestPthreadCleanup.cpp

87 lines
2.2 KiB
C++

/*
* Copyright (c) 2022, Tim Schumacher <timschumi@serenityos.org>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <LibTest/TestCase.h>
#include <pthread.h>
static size_t exit_count = 0;
static void exit_count_test_handler(void* data)
{
EXPECT_EQ(exit_count, reinterpret_cast<size_t>(data));
exit_count++;
}
static void immediate_fail_handler(void*)
{
FAIL("Called a cleanup handler");
}
static void* cleanup_pthread_exit_inner(void*)
{
// Push handlers in reverse order as they are taken from the top of the stack on cleanup.
pthread_cleanup_push(exit_count_test_handler, reinterpret_cast<void*>(2));
pthread_cleanup_push(exit_count_test_handler, reinterpret_cast<void*>(1));
pthread_cleanup_push(exit_count_test_handler, reinterpret_cast<void*>(0));
pthread_exit(nullptr);
}
TEST_CASE(cleanup_pthread_exit)
{
pthread_t thread;
exit_count = 0;
pthread_create(&thread, nullptr, cleanup_pthread_exit_inner, nullptr);
pthread_join(thread, nullptr);
// Ensure that all exit handlers have been called.
EXPECT_EQ(exit_count, 3ul);
}
static void* cleanup_return_inner(void*)
{
// Returning from the main function should not call any cleanup handlers.
pthread_cleanup_push(immediate_fail_handler, nullptr);
return nullptr;
}
TEST_CASE(cleanup_return)
{
pthread_t thread;
pthread_create(&thread, nullptr, cleanup_return_inner, nullptr);
pthread_join(thread, nullptr);
}
static void* cleanup_pop_inner(void*)
{
pthread_cleanup_push(exit_count_test_handler, reinterpret_cast<void*>(1));
pthread_cleanup_push(immediate_fail_handler, nullptr);
pthread_cleanup_push(exit_count_test_handler, reinterpret_cast<void*>(0));
pthread_cleanup_push(immediate_fail_handler, nullptr);
// Popping a cleanup handler shouldn't run the callback unless `execute` is given.
pthread_cleanup_pop(0);
pthread_cleanup_pop(1);
pthread_cleanup_pop(0);
pthread_cleanup_pop(1);
return nullptr;
}
TEST_CASE(cleanup_pop)
{
pthread_t thread;
exit_count = 0;
pthread_create(&thread, nullptr, cleanup_pop_inner, nullptr);
pthread_join(thread, nullptr);
// Ensure that all exit handlers have been called.
EXPECT_EQ(exit_count, 2ul);
}