gcc/testsuite/g++.dg/asan/sanitizer_pthread_wrappers.h

   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
  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
  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
  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
  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