diff options
Diffstat (limited to '3rdparty/plibsys/tests/puthread_test.cpp')
| -rw-r--r-- | 3rdparty/plibsys/tests/puthread_test.cpp | 481 |
1 files changed, 481 insertions, 0 deletions
diff --git a/3rdparty/plibsys/tests/puthread_test.cpp b/3rdparty/plibsys/tests/puthread_test.cpp new file mode 100644 index 0000000..5c31cd1 --- /dev/null +++ b/3rdparty/plibsys/tests/puthread_test.cpp @@ -0,0 +1,481 @@ +/* + * The MIT License + * + * Copyright (C) 2013-2019 Alexander Saprykin <saprykin.spb@gmail.com> + * + * Permission is hereby granted, free of charge, to any person obtaining + * a copy of this software and associated documentation files (the + * 'Software'), to deal in the Software without restriction, including + * without limitation the rights to use, copy, modify, merge, publish, + * distribute, sublicense, and/or sell copies of the Software, and to + * permit persons to whom the Software is furnished to do so, subject to + * the following conditions: + * + * The above copyright notice and this permission notice shall be + * included in all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND, + * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF + * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. + * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY + * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, + * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE + * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. + */ + +#include "plibsys.h" +#include "ptestmacros.h" + +P_TEST_MODULE_INIT (); + +static pint thread_wakes_1 = 0; +static pint thread_wakes_2 = 0; +static pint thread_to_wakes = 0; +static volatile pboolean is_threads_working = FALSE; + +static P_HANDLE thread1_id = (P_HANDLE) NULL; +static P_HANDLE thread2_id = (P_HANDLE) NULL; +static PUThread * thread1_obj = NULL; +static PUThread * thread2_obj = NULL; + +static PUThreadKey * tls_key = NULL; +static PUThreadKey * tls_key_2 = NULL; +static volatile pint free_counter = 0; + +extern "C" ppointer pmem_alloc (psize nbytes) +{ + P_UNUSED (nbytes); + return (ppointer) NULL; +} + +extern "C" ppointer pmem_realloc (ppointer block, psize nbytes) +{ + P_UNUSED (block); + P_UNUSED (nbytes); + return (ppointer) NULL; +} + +extern "C" void pmem_free (ppointer block) +{ + P_UNUSED (block); +} + +extern "C" void free_with_check (ppointer mem) +{ + p_free (mem); + p_atomic_int_inc (&free_counter); +} + +static void * test_thread_func (void *data) +{ + pint *counter = static_cast < pint * > (data); + + if ((*counter) == 1) { + thread1_id = p_uthread_current_id (); + thread1_obj = p_uthread_current (); + } else { + thread2_id = p_uthread_current_id (); + thread2_obj = p_uthread_current (); + } + + p_uthread_set_local (tls_key, (ppointer) p_uthread_current_id ()); + + *counter = 0; + + while (is_threads_working == TRUE) { + p_uthread_sleep (10); + ++(*counter); + p_uthread_yield (); + + if (p_uthread_get_local (tls_key) != (ppointer) p_uthread_current_id ()) + p_uthread_exit (-1); + } + + p_uthread_exit (*counter); + + return NULL; +} + +static void * test_thread_nonjoinable_func (void *data) +{ + pint *counter = static_cast < pint * > (data); + + is_threads_working = TRUE; + + for (int i = thread_to_wakes; i > 0; --i) { + p_uthread_sleep (10); + ++(*counter); + p_uthread_yield (); + } + + is_threads_working = FALSE; + + p_uthread_exit (0); + + return NULL; +} + +static void * test_thread_tls_func (void *data) +{ + pint self_thread_free = *((pint *) data); + + pint *tls_value = (pint *) p_malloc0 (sizeof (pint)); + *tls_value = 0; + p_uthread_set_local (tls_key, (ppointer) tls_value); + + pint prev_tls = 0; + pint counter = 0; + + while (is_threads_working == TRUE) { + p_uthread_sleep (10); + + pint *last_tls = (pint *) p_uthread_get_local (tls_key); + + if ((*last_tls) != prev_tls) + p_uthread_exit (-1); + + pint *tls_new_value = (pint *) p_malloc0 (sizeof (pint)); + + *tls_new_value = (*last_tls) + 1; + prev_tls = (*last_tls) + 1; + + p_uthread_replace_local (tls_key, (ppointer) tls_new_value); + + if (self_thread_free) + p_free (last_tls); + + ++counter; + + p_uthread_yield (); + } + + if (self_thread_free) { + pint *last_tls = (pint *) p_uthread_get_local (tls_key); + + if ((*last_tls) != prev_tls) + p_uthread_exit (-1); + + p_free (last_tls); + + p_uthread_replace_local (tls_key, (ppointer) NULL); + } + + p_uthread_exit (counter); + + return NULL; +} + +static void * test_thread_tls_create_func (void *data) +{ + P_UNUSED (data); + + pint *tls_value = (pint *) p_malloc0 (sizeof (pint)); + *tls_value = 0; + p_uthread_set_local (tls_key, (ppointer) tls_value); + + pint *tls_value_2 = (pint *) p_malloc0 (sizeof (pint)); + *tls_value_2 = 0; + p_uthread_set_local (tls_key_2, (ppointer) tls_value_2); + + return NULL; +} + +P_TEST_CASE_BEGIN (puthread_nomem_test) +{ + p_libsys_init (); + + PUThreadKey *thread_key = p_uthread_local_new (p_free); + P_TEST_CHECK (thread_key != NULL); + + PMemVTable vtable; + + vtable.free = pmem_free; + vtable.malloc = pmem_alloc; + vtable.realloc = pmem_realloc; + + P_TEST_CHECK (p_mem_set_vtable (&vtable) == TRUE); + + thread_wakes_1 = 0; + thread_wakes_2 = 0; + + P_TEST_CHECK (p_uthread_create ((PUThreadFunc) test_thread_func, + (ppointer) &thread_wakes_1, + TRUE, + NULL) == NULL); + + P_TEST_CHECK (p_uthread_create_full ((PUThreadFunc) test_thread_func, + (ppointer) &thread_wakes_2, + TRUE, + P_UTHREAD_PRIORITY_NORMAL, + 0, + NULL) == NULL); + + P_TEST_CHECK (p_uthread_current () == NULL); + P_TEST_CHECK (p_uthread_local_new (NULL) == NULL); + + p_uthread_exit (0); + + p_uthread_set_local (thread_key, PINT_TO_POINTER (10)); + + ppointer tls_value = p_uthread_get_local (thread_key); + + if (tls_value != NULL) { + P_TEST_CHECK (tls_value == PINT_TO_POINTER (10)); + p_uthread_set_local (thread_key, NULL); + } + + p_uthread_replace_local (thread_key, PINT_TO_POINTER (12)); + + tls_value = p_uthread_get_local (thread_key); + + if (tls_value != NULL) { + P_TEST_CHECK (tls_value == PINT_TO_POINTER (12)); + p_uthread_set_local (thread_key, NULL); + } + + p_mem_restore_vtable (); + + p_uthread_local_free (thread_key); + + p_libsys_shutdown (); +} +P_TEST_CASE_END () + +P_TEST_CASE_BEGIN (puthread_bad_input_test) +{ + p_libsys_init (); + + P_TEST_CHECK (p_uthread_create (NULL, NULL, FALSE, NULL) == NULL); + P_TEST_CHECK (p_uthread_create_full (NULL, NULL, FALSE, P_UTHREAD_PRIORITY_NORMAL, 0, NULL) == NULL); + P_TEST_CHECK (p_uthread_join (NULL) == -1); + P_TEST_CHECK (p_uthread_set_priority (NULL, P_UTHREAD_PRIORITY_NORMAL) == FALSE); + P_TEST_CHECK (p_uthread_get_local (NULL) == NULL); + p_uthread_set_local (NULL, NULL); + p_uthread_replace_local (NULL, NULL); + p_uthread_ref (NULL); + p_uthread_unref (NULL); + p_uthread_local_free (NULL); + p_uthread_exit (0); + + p_libsys_shutdown (); +} +P_TEST_CASE_END () + +P_TEST_CASE_BEGIN (puthread_general_test) +{ + p_libsys_init (); + + thread_wakes_1 = 1; + thread_wakes_2 = 2; + thread1_id = (P_HANDLE) NULL; + thread2_id = (P_HANDLE) NULL; + thread1_obj = NULL; + thread2_obj = NULL; + + tls_key = p_uthread_local_new (NULL); + P_TEST_CHECK (tls_key != NULL); + + /* Threre is no guarantee that we wouldn't get one of the IDs + * of the finished test threads */ + + P_HANDLE main_id = p_uthread_current_id (); + + is_threads_working = TRUE; + + PUThread *thr1 = p_uthread_create_full ((PUThreadFunc) test_thread_func, + (ppointer) &thread_wakes_1, + TRUE, + P_UTHREAD_PRIORITY_NORMAL, + 64 * 1024, + "thread_name"); + + PUThread *thr2 = p_uthread_create_full ((PUThreadFunc) test_thread_func, + (ppointer) &thread_wakes_2, + TRUE, + P_UTHREAD_PRIORITY_NORMAL, + 64 * 1024, + "very_long_name_for_thread_testing"); + + p_uthread_ref (thr1); + + p_uthread_set_priority (thr1, P_UTHREAD_PRIORITY_NORMAL); + + P_TEST_REQUIRE (thr1 != NULL); + P_TEST_REQUIRE (thr2 != NULL); + + p_uthread_sleep (5000); + + is_threads_working = FALSE; + + P_TEST_CHECK (p_uthread_join (thr1) == thread_wakes_1); + P_TEST_CHECK (p_uthread_join (thr2) == thread_wakes_2); + + P_TEST_REQUIRE (thread1_id != thread2_id); + P_TEST_CHECK (thread1_id != main_id && thread2_id != main_id); + + P_TEST_CHECK (thread1_obj == thr1); + P_TEST_CHECK (thread2_obj == thr2); + + p_uthread_local_free (tls_key); + p_uthread_unref (thr1); + p_uthread_unref (thr2); + + p_uthread_unref (thr1); + + PUThread *cur_thr = p_uthread_current (); + P_TEST_CHECK (cur_thr != NULL); + + P_TEST_CHECK (p_uthread_ideal_count () > 0); + + p_libsys_shutdown (); +} +P_TEST_CASE_END () + +P_TEST_CASE_BEGIN (puthread_nonjoinable_test) +{ + p_libsys_init (); + + thread_wakes_1 = 0; + thread_to_wakes = 100; + is_threads_working = TRUE; + + PUThread *thr1 = p_uthread_create ((PUThreadFunc) test_thread_nonjoinable_func, + (ppointer) &thread_wakes_1, + FALSE, + NULL); + + P_TEST_REQUIRE (thr1 != NULL); + + p_uthread_sleep (3000); + + P_TEST_CHECK (p_uthread_join (thr1) == -1); + + while (is_threads_working == TRUE) + p_uthread_sleep (10); + + P_TEST_CHECK (thread_wakes_1 == thread_to_wakes); + + p_uthread_unref (thr1); + + p_libsys_shutdown (); +} +P_TEST_CASE_END () + +P_TEST_CASE_BEGIN (puthread_tls_test) +{ + p_libsys_init (); + + /* With destroy notification */ + tls_key = p_uthread_local_new (free_with_check); + + is_threads_working = TRUE; + free_counter = 0; + + pint self_thread_free = 0; + + PUThread *thr1 = p_uthread_create ((PUThreadFunc) test_thread_tls_func, + (ppointer) &self_thread_free, + TRUE, + NULL); + + PUThread *thr2 = p_uthread_create ((PUThreadFunc) test_thread_tls_func, + (ppointer) &self_thread_free, + TRUE, + NULL); + + P_TEST_REQUIRE (thr1 != NULL); + P_TEST_REQUIRE (thr2 != NULL); + + p_uthread_sleep (5000); + + is_threads_working = FALSE; + + pint total_counter = 0; + + total_counter += (p_uthread_join (thr1) + 1); + total_counter += (p_uthread_join (thr2) + 1); + + P_TEST_CHECK (total_counter == free_counter); + + p_uthread_local_free (tls_key); + p_uthread_unref (thr1); + p_uthread_unref (thr2); + + /* Without destroy notification */ + tls_key = p_uthread_local_new (NULL); + + free_counter = 0; + is_threads_working = TRUE; + self_thread_free = 1; + + thr1 = p_uthread_create ((PUThreadFunc) test_thread_tls_func, + (ppointer) &self_thread_free, + TRUE, + NULL); + + thr2 = p_uthread_create ((PUThreadFunc) test_thread_tls_func, + (ppointer) &self_thread_free, + TRUE, + NULL); + + P_TEST_REQUIRE (thr1 != NULL); + P_TEST_REQUIRE (thr2 != NULL); + + p_uthread_sleep (5000); + + is_threads_working = FALSE; + + total_counter = 0; + + total_counter += (p_uthread_join (thr1) + 1); + total_counter += (p_uthread_join (thr2) + 1); + + P_TEST_CHECK (total_counter > 0); + P_TEST_CHECK (free_counter == 0); + + p_uthread_local_free (tls_key); + p_uthread_unref (thr1); + p_uthread_unref (thr2); + + /* With implicit thread exit */ + tls_key = p_uthread_local_new (free_with_check); + tls_key_2 = p_uthread_local_new (free_with_check); + + free_counter = 0; + + thr1 = p_uthread_create ((PUThreadFunc) test_thread_tls_create_func, + NULL, + TRUE, + NULL); + + thr2 = p_uthread_create ((PUThreadFunc) test_thread_tls_create_func, + NULL, + TRUE, + NULL); + + P_TEST_REQUIRE (thr1 != NULL); + P_TEST_REQUIRE (thr2 != NULL); + + p_uthread_join (thr1); + p_uthread_join (thr2); + + P_TEST_CHECK (free_counter == 4); + + p_uthread_local_free (tls_key); + p_uthread_local_free (tls_key_2); + p_uthread_unref (thr1); + p_uthread_unref (thr2); + + p_libsys_shutdown (); +} +P_TEST_CASE_END () + +P_TEST_SUITE_BEGIN() +{ + P_TEST_SUITE_RUN_CASE (puthread_nomem_test); + P_TEST_SUITE_RUN_CASE (puthread_bad_input_test); + P_TEST_SUITE_RUN_CASE (puthread_general_test); + P_TEST_SUITE_RUN_CASE (puthread_nonjoinable_test); + P_TEST_SUITE_RUN_CASE (puthread_tls_test); +} +P_TEST_SUITE_END() |