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 #include <config.h>
  21
  22 #ifdef __linux__
  23 # include <urcu/syscall-compat.h>
  24
  25 # if defined(HAVE_GETTID)
  26 /*
  27  * Do not redefine gettid() as it is already included
  28  * in bionic through <unistd.h>. Some other libc
  29  * may also already contain an implementation of gettid.
  30  */
  31 # elif defined(_syscall0)
  32 _syscall0(pid_t, gettid)
  33 # elif defined(__NR_gettid)
  34 static inline pid_t gettid(void)
  35 {
  36         return syscall(__NR_gettid);
  37 }
  38 # endif
  39
  40 static inline
  41 unsigned long urcu_get_thread_id(void)
  42 {
  43         return (unsigned long) gettid();
  44 }
  45 #elif defined(__FreeBSD__)
  46 # include <pthread_np.h>
  47
  48 static inline
  49 unsigned long urcu_get_thread_id(void)
  50 {
  51         return (unsigned long) pthread_getthreadid_np();
  52 }
  53 #else
  54 # warning "use pid as thread ID"
  55 static inline
  56 unsigned long urcu_get_thread_id(void)
  57 {
  58         return (unsigned long) getpid();
  59 }
  60 #endif
  61
  62 #endif /* _TEST_THREAD_ID_H */