[20be782] | 1 | |
---|
[428adbc] | 2 | /* test pthread_key_create/set_specific/get_specific |
---|
[20be782] | 3 | get specific == set specific |
---|
| 4 | dtor is invoked (no mem leak) |
---|
| 5 | */ |
---|
| 6 | |
---|
| 7 | |
---|
| 8 | extern "C"{ |
---|
| 9 | |
---|
| 10 | #include <stdio.h> |
---|
| 11 | #include <stdlib.h> |
---|
| 12 | #include <errno.h> |
---|
| 13 | #include <pthread.h> |
---|
| 14 | |
---|
| 15 | #define THREADS 5 |
---|
| 16 | #define BUFFSZ 48 |
---|
| 17 | pthread_key_t key; |
---|
| 18 | volatile int total_value,total_value_getspec; |
---|
| 19 | pthread_mutex_t value_mutex; |
---|
| 20 | |
---|
| 21 | void *threadfunc(void *parm) |
---|
| 22 | { |
---|
| 23 | int status; |
---|
| 24 | void *value; |
---|
| 25 | int threadnum; |
---|
| 26 | int *tnum; |
---|
| 27 | void *getvalue; |
---|
| 28 | char Buffer[BUFFSZ]; |
---|
| 29 | |
---|
| 30 | tnum = (int*)parm; |
---|
| 31 | threadnum = *tnum; |
---|
| 32 | |
---|
| 33 | //printf("Thread %d executing\n", threadnum); |
---|
| 34 | value = (void *)(rand()%100); |
---|
| 35 | status = pthread_setspecific(key, (void *) value); |
---|
| 36 | if ( status != 0) { |
---|
| 37 | printf("pthread_setspecific failed, thread %d, errno %d", |
---|
| 38 | threadnum, errno); |
---|
| 39 | return (void*)12; |
---|
| 40 | } |
---|
| 41 | pthread_mutex_lock(&value_mutex); |
---|
| 42 | total_value_getspec += (int)value; |
---|
| 43 | total_value += (int)pthread_getspecific(key); |
---|
| 44 | pthread_mutex_unlock(&value_mutex); |
---|
| 45 | |
---|
| 46 | |
---|
| 47 | if (!(value = malloc(sizeof(Buffer)))) |
---|
| 48 | printf("Thread %d could not allocate storage, errno = %d\n", |
---|
| 49 | threadnum, errno); |
---|
| 50 | status = pthread_setspecific(key, (void *) value); |
---|
| 51 | if ( status != 0) { |
---|
| 52 | printf("pthread_setspecific failed, thread %d, errno %d", |
---|
| 53 | threadnum, errno); |
---|
| 54 | return (void*)12; |
---|
| 55 | } |
---|
| 56 | //printf("Thread %d setspecific value: %d\n", threadnum, value); |
---|
| 57 | |
---|
| 58 | getvalue = 0; |
---|
| 59 | getvalue = pthread_getspecific(key); |
---|
| 60 | |
---|
| 61 | if (getvalue != value) { |
---|
| 62 | printf("getvalue not valid, getvalue=%d", (u_int64_t)getvalue); |
---|
| 63 | return (void*)68; |
---|
| 64 | } |
---|
| 65 | |
---|
| 66 | pthread_exit((void *)0); |
---|
| 67 | } |
---|
| 68 | |
---|
| 69 | void destr_fn(void *parm) |
---|
| 70 | { |
---|
| 71 | |
---|
| 72 | printf("Destructor function invoked\n"); |
---|
| 73 | free(parm); |
---|
| 74 | } |
---|
| 75 | |
---|
| 76 | |
---|
| 77 | int main() { |
---|
| 78 | int getvalue; |
---|
| 79 | int status; |
---|
| 80 | int i; |
---|
| 81 | int threadparm[THREADS]; |
---|
| 82 | pthread_t threadid[THREADS]; |
---|
| 83 | void* thread_stat[THREADS]; |
---|
| 84 | |
---|
| 85 | // rand seed for testing |
---|
| 86 | srand(1003); |
---|
| 87 | pthread_mutex_init(&value_mutex, NULL); |
---|
| 88 | |
---|
| 89 | // testing getspec and setspec |
---|
| 90 | total_value = 0; |
---|
| 91 | total_value_getspec = 0; |
---|
| 92 | |
---|
| 93 | if ((status = pthread_key_create(&key, destr_fn )) < 0) { |
---|
| 94 | printf("pthread_key_create failed, errno=%d", errno); |
---|
| 95 | exit(1); |
---|
| 96 | } |
---|
| 97 | |
---|
| 98 | // create 3 THREADS, pass each its number |
---|
| 99 | for (i=0; i<THREADS; i++) { |
---|
| 100 | threadparm[i] = i+1; |
---|
| 101 | status = pthread_create( &threadid[i], |
---|
| 102 | NULL, |
---|
| 103 | threadfunc, |
---|
| 104 | (void *)&threadparm[i]); |
---|
| 105 | if ( status < 0) { |
---|
| 106 | printf("pthread_create failed, errno=%d", errno); |
---|
| 107 | exit(2); |
---|
| 108 | } |
---|
| 109 | } |
---|
| 110 | |
---|
| 111 | for ( i=0; i<THREADS; i++) { |
---|
| 112 | status = pthread_join( threadid[i], (void **)&thread_stat[i]); |
---|
| 113 | if ( status < 0) { |
---|
| 114 | printf("pthread_join failed, thread %d, errno=%d\n", i+1, errno); |
---|
| 115 | } |
---|
| 116 | |
---|
| 117 | if (thread_stat[i] != 0) { |
---|
| 118 | printf("bad thread status, thread %d, status=%d\n", i+1, |
---|
| 119 | (u_int64_t)thread_stat[i]); |
---|
| 120 | } |
---|
| 121 | } |
---|
| 122 | printf("total value is %d, total value by pthread_getspecific is %d\n", total_value, total_value_getspec); |
---|
| 123 | exit(0); |
---|
| 124 | } // main |
---|
| 125 | } |
---|
| 126 | |
---|