c/pthread_extra/pthread_extra.h

   1 #ifndef __PTHREAD_EXTRA_H__
   2 #define __PTHREAD_EXTRA_H__
   3
   4 #include <pthread.h>
   5 #include <stdbool.h>
   6 #include <stdint.h>
   7 #include <signal.h>
   8 //#include <time.h>
   9
  10 #define PTHREAD_XTIME_NOBLOCK (&(struct timespec){ .tv_sec = 0, .tv_nsec = 0 })
  11 #define PTHREAD_XTIME_FOREVER NULL
  12
  13 // User data
  14
  15 #define PTHREAD_XTHREADS_MAX 65535
  16 #define PTHREAD_XNULL ((pthread_t)NULL)
  17
  18 #ifdef __PTHREAD_EXTRA_INTERNAL
  19 typedef struct pthread_user_data_internal_t {
  20         pthread_t tid; //Thread ID
  21         sig_atomic_t running; //Internaly used by pthread_pause
  22         void *usr; //User pointer
  23 } pthread_user_data_internal_t;
  24
  25 pthread_user_data_internal_t* pthread_user_data_internal(pthread_t thread);
  26 #endif //__PTHREAD_EXTRA_INTERNAL
  27
  28 void** pthread_user_data_ptr(pthread_t thread);
  29 void*  pthread_user_data_get(pthread_t thread);
  30 void   pthread_user_data_set(pthread_t thread, void *usr);
  31 void   pthread_user_data_cleanup(void * arg);
  32
  33 // Pausing
  34
  35 //GDB: handle SIG34 nostop noprint
  36 #define PTHREAD_XSIG_STOP (SIGRTMIN+0)
  37 #define PTHREAD_XSIG_CONT (SIGRTMIN+1)
  38 #define PTHREAD_XSIGRTMIN (SIGRTMIN+2) //First unused RT signal
  39
  40 int pthread_extra_create(pthread_t *restrict thread,
  41                           const pthread_attr_t *restrict attr,
  42                           void *(*start_routine)(void *),
  43                           void *restrict arg);
  44 void pthread_unpause_handler();
  45 void pthread_pause_handler();
  46 void pthread_pause_enable();
  47 int pthread_pause(pthread_t thread);
  48 int pthread_unpause(pthread_t thread);
  49 int pthread_pause_reschedule(pthread_t thread);
  50
  51 // Message queues
  52
  53 //Flags
  54 typedef uint8_t pthread_mq_flags_t;
  55 #define PTHREAD_XMQ_NONE        0                       ///< No flags specified (default behaviour)
  56 #define PTHREAD_XMQ_FRONT       1                       ///< Send to front of the queue (scheduled for next receive)
  57 #define PTHREAD_XMQ_BACK        PTHREAD_XMQ_NONE        ///< Send to back of the queue (default)
  58 #define PTHREAD_XMQ_PEEK        2                       ///< Only peek, do not remove received item
  59 #define PTHREAD_XMQ_RECV        PTHREAD_XMQ_NONE        ///< Remove received item from queue (default)
  60 #define PTHREAD_XMQ_OVERW       4                       ///< Overwrite item if queue full
  61
  62 typedef struct pthread_mq_t {
  63         pthread_mutex_t lock;
  64         pthread_cond_t cond_readable;
  65         pthread_cond_t cond_writable;
  66         uint8_t * data;
  67         size_t msg_size;
  68         size_t msg_count;
  69         size_t msg_count_max;
  70         size_t head_idx;
  71         char * name;
  72 } pthread_mq_t;
  73
  74 bool pthread_mq_init(pthread_mq_t *mq, size_t msg_size, size_t msg_count_max);
  75 void pthread_mq_free(pthread_mq_t *mq);
  76
  77 bool pthread_mq_reset(pthread_mq_t *mq);
  78 bool pthread_mq_send_generic(pthread_mq_t *mq, void * data, pthread_mq_flags_t flags, const struct timespec *restrict abs_timeout);
  79 bool pthread_mq_receive_generic(pthread_mq_t *mq, void * data, pthread_mq_flags_t flags, const struct timespec *restrict abs_timeout);
  80
  81 size_t pthread_mq_waiting(pthread_mq_t *mq);
  82 size_t pthread_mq_vacant(pthread_mq_t *mq);
  83
  84 // Multi mutex locking
  85
  86 #define pthread_mutex_swap(a, b) { pthread_mutex_t *s; s = (a); a = (b); b = s; }
  87
  88 #define pthread_mutex_lock_two(a,b) pthread_mutex_timedlock_multi_generic((pthread_mutex_t *[2]){(a), (b)}, 2, true, NULL)
  89 #define pthread_mutex_timedlock_two(a,b,tm) pthread_mutex_timedlock_multi_generic((pthread_mutex_t *[2]){(a), (b)}, 2, true, (tm))
  90 #define pthread_mutex_trylock_two(a,b) pthread_mutex_timedlock_multi_generic((pthread_mutex_t *[2]){(a), (b)}, 2, false, NULL)
  91
  92 #define pthread_mutex_lock_multi(lcks,cnt) pthread_mutex_timedlock_multi_generic((lcks),(cnt),true,NULL)
  93 #define pthread_mutex_timedlock_multi(lcks,cnt,tm) pthread_mutex_timedlock_multi_generic((lcks),(cnt),true,(tm))
  94 #define pthread_mutex_trylock_multi(lcks,cnt) pthread_mutex_timedlock_multi_generic((lcks),(cnt),false,NULL)
  95
  96 int pthread_mutex_timedlock_multi_generic(pthread_mutex_t **lck, int cnt, bool block, const struct timespec *restrict abs_timeout);
  97
  98
  99 #endif //__PTHREAD_EXTRA_H__