| /* Copyright (c) 2015, Google Inc. |
| * |
| * Permission to use, copy, modify, and/or distribute this software for any |
| * purpose with or without fee is hereby granted, provided that the above |
| * copyright notice and this permission notice appear in all copies. |
| * |
| * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES |
| * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF |
| * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY |
| * SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES |
| * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION |
| * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN |
| * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */ |
| |
| #include "internal.h" |
| |
| #include <gtest/gtest.h> |
| |
| #include <openssl/crypto.h> |
| #include <openssl/rand.h> |
| |
| #include "test/test_util.h" |
| |
| |
| #if !defined(OPENSSL_NO_THREADS) |
| |
| #if defined(OPENSSL_WINDOWS) |
| |
| OPENSSL_MSVC_PRAGMA(warning(push, 3)) |
| #include <windows.h> |
| OPENSSL_MSVC_PRAGMA(warning(pop)) |
| |
| typedef HANDLE thread_t; |
| |
| static DWORD WINAPI thread_run(LPVOID arg) { |
| void (*thread_func)(void); |
| /* VC really doesn't like casting between data and function pointers. */ |
| OPENSSL_memcpy(&thread_func, &arg, sizeof(thread_func)); |
| thread_func(); |
| return 0; |
| } |
| |
| static int run_thread(thread_t *out_thread, void (*thread_func)(void)) { |
| void *arg; |
| /* VC really doesn't like casting between data and function pointers. */ |
| OPENSSL_memcpy(&arg, &thread_func, sizeof(arg)); |
| |
| *out_thread = CreateThread(NULL /* security attributes */, |
| 0 /* default stack size */, thread_run, arg, |
| 0 /* run immediately */, NULL /* ignore id */); |
| return *out_thread != NULL; |
| } |
| |
| static int wait_for_thread(thread_t thread) { |
| return WaitForSingleObject(thread, INFINITE) == 0; |
| } |
| |
| #else |
| |
| #include <pthread.h> |
| #include <string.h> |
| #include <time.h> |
| |
| typedef pthread_t thread_t; |
| |
| static void *thread_run(void *arg) { |
| void (*thread_func)(void) = reinterpret_cast<void (*)(void)>(arg); |
| thread_func(); |
| return NULL; |
| } |
| |
| static int run_thread(thread_t *out_thread, void (*thread_func)(void)) { |
| return pthread_create(out_thread, NULL /* default attributes */, thread_run, |
| reinterpret_cast<void *>(thread_func)) == 0; |
| } |
| |
| static int wait_for_thread(thread_t thread) { |
| return pthread_join(thread, NULL) == 0; |
| } |
| |
| #endif /* OPENSSL_WINDOWS */ |
| |
| static unsigned g_once_init_called = 0; |
| |
| static void once_init(void) { |
| g_once_init_called++; |
| |
| /* Sleep briefly so one |call_once_thread| instance will call |CRYPTO_once| |
| * while the other is running this function. */ |
| #if defined(OPENSSL_WINDOWS) |
| Sleep(1 /* milliseconds */); |
| #else |
| struct timespec req; |
| OPENSSL_memset(&req, 0, sizeof(req)); |
| req.tv_nsec = 1000000; |
| nanosleep(&req, NULL); |
| #endif |
| } |
| |
| static CRYPTO_once_t g_test_once = CRYPTO_ONCE_INIT; |
| |
| static void call_once_thread(void) { |
| CRYPTO_once(&g_test_once, once_init); |
| } |
| |
| static CRYPTO_once_t once_init_value = CRYPTO_ONCE_INIT; |
| static CRYPTO_once_t once_bss; |
| |
| static struct CRYPTO_STATIC_MUTEX mutex_init_value = CRYPTO_STATIC_MUTEX_INIT; |
| static struct CRYPTO_STATIC_MUTEX mutex_bss; |
| |
| static CRYPTO_EX_DATA_CLASS ex_data_class_value = CRYPTO_EX_DATA_CLASS_INIT; |
| static CRYPTO_EX_DATA_CLASS ex_data_class_bss; |
| |
| TEST(ThreadTest, Once) { |
| ASSERT_EQ(0u, g_once_init_called) |
| << "g_once_init_called was non-zero at start."; |
| |
| thread_t thread1, thread2; |
| ASSERT_TRUE(run_thread(&thread1, call_once_thread)); |
| ASSERT_TRUE(run_thread(&thread2, call_once_thread)); |
| ASSERT_TRUE(wait_for_thread(thread1)); |
| ASSERT_TRUE(wait_for_thread(thread2)); |
| |
| CRYPTO_once(&g_test_once, once_init); |
| |
| EXPECT_EQ(1u, g_once_init_called); |
| } |
| |
| TEST(ThreadTest, InitZeros) { |
| if (FIPS_mode()) { |
| /* Our FIPS tooling currently requires that |CRYPTO_ONCE_INIT|, |
| * |CRYPTO_STATIC_MUTEX_INIT| and |CRYPTO_EX_DATA_CLASS| are all zeros and |
| * so can be placed in the BSS section. */ |
| EXPECT_EQ(Bytes((uint8_t *)&once_bss, sizeof(once_bss)), |
| Bytes((uint8_t *)&once_init_value, sizeof(once_init_value))); |
| EXPECT_EQ(Bytes((uint8_t *)&mutex_bss, sizeof(mutex_bss)), |
| Bytes((uint8_t *)&mutex_init_value, sizeof(mutex_init_value))); |
| EXPECT_EQ( |
| Bytes((uint8_t *)&ex_data_class_bss, sizeof(ex_data_class_bss)), |
| Bytes((uint8_t *)&ex_data_class_value, sizeof(ex_data_class_value))); |
| } |
| } |
| |
| static int g_test_thread_ok = 0; |
| static unsigned g_destructor_called_count = 0; |
| |
| static void thread_local_destructor(void *arg) { |
| if (arg == NULL) { |
| return; |
| } |
| |
| unsigned *count = reinterpret_cast<unsigned*>(arg); |
| (*count)++; |
| } |
| |
| TEST(ThreadTest, ThreadLocal) { |
| ASSERT_EQ(nullptr, CRYPTO_get_thread_local(OPENSSL_THREAD_LOCAL_TEST)) |
| << "Thread-local data was non-NULL at start."; |
| |
| thread_t thread; |
| ASSERT_TRUE(run_thread(&thread, []() { |
| if (CRYPTO_get_thread_local(OPENSSL_THREAD_LOCAL_TEST) != NULL || |
| !CRYPTO_set_thread_local(OPENSSL_THREAD_LOCAL_TEST, |
| &g_destructor_called_count, |
| thread_local_destructor) || |
| CRYPTO_get_thread_local(OPENSSL_THREAD_LOCAL_TEST) != |
| &g_destructor_called_count) { |
| return; |
| } |
| |
| g_test_thread_ok = 1; |
| })); |
| ASSERT_TRUE(wait_for_thread(thread)); |
| |
| EXPECT_TRUE(g_test_thread_ok) << "Thread-local data didn't work in thread."; |
| EXPECT_EQ(1u, g_destructor_called_count); |
| |
| // Create a no-op thread to test test that the thread destructor function |
| // works even if thread-local storage wasn't used for a thread. |
| ASSERT_TRUE(run_thread(&thread, []() {})); |
| ASSERT_TRUE(wait_for_thread(thread)); |
| } |
| |
| TEST(ThreadTest, RandState) { |
| /* In FIPS mode, rand.c maintains a linked-list of thread-local data because |
| * we're required to clear it on process exit. This test exercises removing a |
| * value from that list. */ |
| uint8_t buf[1]; |
| RAND_bytes(buf, sizeof(buf)); |
| |
| thread_t thread; |
| ASSERT_TRUE(run_thread(&thread, []() { |
| uint8_t buf2[1]; |
| RAND_bytes(buf2, sizeof(buf2)); |
| })); |
| ASSERT_TRUE(wait_for_thread(thread)); |
| } |
| |
| #endif /* !OPENSSL_NO_THREADS */ |