1 #define __PTHREAD_EXTRA_INTERNAL
4 #define _GNU_SOURCE //PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP
10 #include <pthread_extra.h>
12 //Static array with user data for all thread handles
13 //TODO: perhaps use something more sophisticated like linked list?
14 pthread_user_data_internal_t pthread_user_data
[PTHREAD_XTHREADS_MAX
+1] = {{.tid
=PTHREAD_XNULL
}};
15 pthread_mutex_t pthread_user_data_mutex
= PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP
;
17 int pthread_user_data_lock() {
18 //pthread_pause_disable();
19 return pthread_mutex_lock(&pthread_user_data_mutex
);
22 int pthread_user_data_unlock() {
23 pthread_mutex_unlock(&pthread_user_data_mutex
);
24 //pthread_pause_enable();
28 //Get pointer to internal record tied to specified thread
29 pthread_user_data_internal_t
* pthread_user_data_internal(pthread_t thread
) {
30 //Return NULL if requested thread handle is NULL
31 if(pthread_equal(thread
, PTHREAD_XNULL
)) return NULL
;
34 pthread_user_data_lock(); //TODO: maybe use R/W locking scheme?
36 //Find if the thread is already registered, add it if not
37 //FIXME: detect overflow of array and cause assert fail!
38 //FIXME: recycle slots of destroyed threads!!! signaled using pthread_user_data_remove();
40 for(i
= 0; i
<PTHREAD_XTHREADS_MAX
; i
++) {
41 if(pthread_equal(pthread_user_data
[i
].tid
, PTHREAD_XNULL
)) {
42 assert(((i
+1)<PTHREAD_XTHREADS_MAX
) && "FIXME: Pthread_extra user data table overflow!");
43 pthread_user_data
[i
+1].tid
= PTHREAD_XNULL
;
44 pthread_user_data
[i
].tid
= thread
;
45 pthread_user_data
[i
].running
= 1; //NEW THREADS ARE RUNNING UNLESS PAUSED!
46 //printf("ADDED %lu = %lu\n", pthread_user_data[i].tid, thread);
49 if(pthread_equal(pthread_user_data
[i
].tid
, thread
)) break;
53 pthread_user_data_unlock();
56 return &pthread_user_data
[i
];
59 //Iterate specified callback over all registered threads
60 int pthread_user_data_internal_iterate(int (*routine
)(pthread_t
), void *arg
) {
63 pthread_user_data_lock();
64 for(i
= 0; i
<PTHREAD_XTHREADS_MAX
; i
++) {
65 if(pthread_equal(pthread_user_data
[i
].tid
, PTHREAD_XNULL
)) break;
66 routine(pthread_user_data
[i
].tid
);
68 pthread_user_data_unlock();
72 //Get pointer to user specified pointer of that thread
73 void** pthread_user_data_ptr(pthread_t thread
) {
74 return &pthread_user_data_internal(thread
)->usr
;
77 //Set user specified pointer for thread
78 void pthread_user_data_set(pthread_t thread
, void *usr
) {
79 *(pthread_user_data_ptr(thread
)) = usr
;
82 //Get user specified pointer for thread
83 void* pthread_user_data_get(pthread_t thread
) {
84 return *(pthread_user_data_ptr(thread
));
87 //Remove canceled thread from user data registry
88 int pthread_user_data_remove(pthread_t thread
) {
89 //FIXME: not implemented yet!
94 //User data cleanup handler to be registered with pthread_cleanup_push()
95 void pthread_user_data_cleanup( void * arg
) {
96 pthread_t t
= (pthread_t
)arg
;
97 assert(pthread_equal(t
, pthread_self()) && "Pthread_self() is not working in cleanup context!");
98 pthread_user_data_remove(t
);