tests/common/thread-id.h

   1 #ifndef _TEST_THREAD_ID_H
   2 #define _TEST_THREAD_ID_H
   3
   4 /*
   5  * thread-id.h
   6  *
   7  * Userspace RCU library - thread ID
   8  *
   9  * Copyright 2013 - Mathieu Desnoyers <mathieu.desnoyers@polymtl.ca>
  10  *
  11  * THIS MATERIAL IS PROVIDED AS IS, WITH ABSOLUTELY NO WARRANTY EXPRESSED
  12  * OR IMPLIED.  ANY USE IS AT YOUR OWN RISK.
  13  *
  14  * Permission is hereby granted to use or copy this program
  15  * for any purpose,  provided the above notices are retained on all copies.
  16  * Permission to modify the code and to distribute modified code is granted,
  17  * provided the above notices are retained, and a notice that the code was
  18  * modified is included with the above copyright notice.
  19  */
  20
  21 #ifdef __linux__
  22 # include <urcu/syscall-compat.h>
  23
  24 # if defined(HAVE_GETTID)
  25 /*
  26  * Do not redefine gettid() as it is already included
  27  * in bionic through <unistd.h>. Some other libc
  28  * may also already contain an implementation of gettid.
  29  */
  30 # elif defined(_syscall0)
  31 _syscall0(pid_t, gettid)
  32 # elif defined(__NR_gettid)
  33 static inline pid_t gettid(void)
  34 {
  35         return syscall(__NR_gettid);
  36 }
  37 # endif
  38
  39 static inline
  40 unsigned long urcu_get_thread_id(void)
  41 {
  42         return (unsigned long) gettid();
  43 }
  44 #elif defined(__FreeBSD__)
  45 # include <pthread_np.h>
  46
  47 static inline
  48 unsigned long urcu_get_thread_id(void)
  49 {
  50         return (unsigned long) pthread_getthreadid_np();
  51 }
  52 #elif defined(__sun__) || defined(__APPLE__)
  53 #include <pthread.h>
  54
  55 static inline
  56 unsigned long urcu_get_thread_id(void)
  57 {
  58         return (unsigned long) pthread_self();
  59 }
  60 #else
  61 # warning "use pid as thread ID"
  62 static inline
  63 unsigned long urcu_get_thread_id(void)
  64 {
  65         return (unsigned long) getpid();
  66 }
  67 #endif
  68
  69 #endif /* _TEST_THREAD_ID_H */