68 lines · c
1//===-- sanitizer_pthread_wrappers.h ----------------------------*- C++ -*-===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8//9// This file is a part of *Sanitizer runtime.10// It provides handy wrappers for thread manipulation, that:11// a) assert on any failure rather than returning an error code12// b) defines pthread-like interface on platforms where where <pthread.h>13// is not supplied by default.14//15//===----------------------------------------------------------------------===//16 17#ifndef SANITIZER_PTHREAD_WRAPPERS_H18#define SANITIZER_PTHREAD_WRAPPERS_H19 20#include "sanitizer_test_utils.h"21 22#if !defined(_WIN32)23# include <pthread.h>24// Simply forward the arguments and check that the pthread functions succeed.25# define PTHREAD_CREATE(a, b, c, d) ASSERT_EQ(0, pthread_create(a, b, c, d))26# define PTHREAD_JOIN(a, b) ASSERT_EQ(0, pthread_join(a, b))27#else28typedef HANDLE pthread_t;29 30struct PthreadHelperCreateThreadInfo {31 void *(*start_routine)(void *);32 void *arg;33};34 35inline DWORD WINAPI PthreadHelperThreadProc(void *arg) {36 PthreadHelperCreateThreadInfo *start_data =37 reinterpret_cast<PthreadHelperCreateThreadInfo*>(arg);38 (start_data->start_routine)(start_data->arg);39 delete start_data;40 return 0;41}42 43inline void PTHREAD_CREATE(pthread_t *thread, void *attr,44 void *(*start_routine)(void *), void *arg) {45 ASSERT_EQ(0, attr) << "Thread attributes are not supported yet.";46 PthreadHelperCreateThreadInfo *data = new PthreadHelperCreateThreadInfo;47 data->start_routine = start_routine;48 data->arg = arg;49 *thread = CreateThread(0, 0, PthreadHelperThreadProc, data, 0, 0);50 DWORD err = GetLastError();51 ASSERT_NE(nullptr, *thread) << "Failed to create a thread, got error 0x"52 << std::hex << err;53}54 55inline void PTHREAD_JOIN(pthread_t thread, void **value_ptr) {56 ASSERT_EQ(0, value_ptr) << "Nonzero value_ptr is not supported yet.";57 ASSERT_EQ(WAIT_OBJECT_0, WaitForSingleObject(thread, INFINITE));58 ASSERT_NE(0, CloseHandle(thread));59}60 61inline void pthread_exit(void *retval) {62 ASSERT_EQ(0, retval) << "Nonzero retval is not supported yet.";63 ExitThread(0);64}65#endif // _WIN3266 67#endif // SANITIZER_PTHREAD_WRAPPERS_H68