1 //===-- sanitizer_pthread_wrappers.h ----------------------------*- C++ -*-===//
2 //
3 // This file is distributed under the University of Illinois Open Source
4 // License. See LICENSE.TXT for details.
5 //
6 //===----------------------------------------------------------------------===//
7 //
8 // This file is a part of *Sanitizer runtime.
9 // It provides handy wrappers for thread manipulation, that:
10 // a) assert on any failure rather than returning an error code
11 // b) defines pthread-like interface on platforms where where <pthread.h>
12 // is not supplied by default.
13 //
14 //===----------------------------------------------------------------------===//
15
16 #ifndef SANITIZER_PTHREAD_WRAPPERS_H
17 #define SANITIZER_PTHREAD_WRAPPERS_H
18
19 #include "sanitizer_test_utils.h"
20
21 #if !defined(_WIN32)
22 # include <pthread.h>
23 // Simply forward the arguments and check that the pthread functions succeed.
24 # define PTHREAD_CREATE(a, b, c, d) ASSERT_EQ(0, pthread_create(a, b, c, d))
25 # define PTHREAD_JOIN(a, b) ASSERT_EQ(0, pthread_join(a, b))
26 #else
27 typedef HANDLE pthread_t;
28
29 struct PthreadHelperCreateThreadInfo {
30 void *(*start_routine)(void *);
31 void *arg;
32 };
33
PthreadHelperThreadProc(void * arg)34 inline DWORD WINAPI PthreadHelperThreadProc(void *arg) {
35 PthreadHelperCreateThreadInfo *start_data =
36 reinterpret_cast<PthreadHelperCreateThreadInfo*>(arg);
37 void *ret = (start_data->start_routine)(start_data->arg);
38 delete start_data;
39 return (DWORD)ret;
40 }
41
PTHREAD_CREATE(pthread_t * thread,void * attr,void * (* start_routine)(void *),void * arg)42 inline void PTHREAD_CREATE(pthread_t *thread, void *attr,
43 void *(*start_routine)(void *), void *arg) {
44 ASSERT_EQ(0, attr) << "Thread attributes are not supported yet.";
45 PthreadHelperCreateThreadInfo *data = new PthreadHelperCreateThreadInfo;
46 data->start_routine = start_routine;
47 data->arg = arg;
48 *thread = CreateThread(0, 0, PthreadHelperThreadProc, data, 0, 0);
49 ASSERT_NE(nullptr, *thread) << "Failed to create a thread.";
50 }
51
PTHREAD_JOIN(pthread_t thread,void ** value_ptr)52 inline void PTHREAD_JOIN(pthread_t thread, void **value_ptr) {
53 ASSERT_EQ(0, value_ptr) << "Nonzero value_ptr is not supported yet.";
54 ASSERT_EQ(WAIT_OBJECT_0, WaitForSingleObject(thread, INFINITE));
55 ASSERT_NE(0, CloseHandle(thread));
56 }
57
pthread_exit(void * retval)58 inline void pthread_exit(void *retval) {
59 ASSERT_EQ(0, retval) << "Nonzero retval is not supported yet.";
60 ExitThread((DWORD)retval);
61 }
62 #endif // _WIN32
63
64 #endif // SANITIZER_PTHREAD_WRAPPERS_H
65