| /* |
| * Copyright (c) 2002, Intel Corporation. All rights reserved. |
| * This file is licensed under the GPL license. For the full content |
| * of this license, see the COPYING file at the top level of this |
| * source tree. |
| * Test pthread_rwlock_wrlock(pthread_rwlock_t * rwlock) |
| * |
| * If a signal is delivered to a thread waiting for a read-write lock for writing, upon |
| * return from the signal handler the thread resumes waiting for the read-write lock for |
| * writing as if it was not interrupted. |
| * |
| * Steps: |
| * 1. main thread create read-write lock 'rwlock', and lock it for writing |
| * 2. main thread create a thread sig_thread, the thread is set to handle SIGUSR1 |
| * 3. sig_thread try to lock 'rwlock' for writing but blocked |
| * 4. main thread send SIGUSR1 to sig_thread via pthread_kill, while sig_thread is blocking |
| * 5. test that thread handler is called |
| * 6. check that when thread handler returns, sig_thread resume block |
| * 7. main thread unlock 'rwlock', sig_thread should get the lock |
| */ |
| |
| #define _XOPEN_SOURCE 600 |
| #include <pthread.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <signal.h> |
| #include <errno.h> |
| #include <unistd.h> |
| #include "posixtest.h" |
| |
| static pthread_t sig_thread; |
| static pthread_rwlock_t rwlock; |
| |
| /* thread_state indicates child thread state: |
| 1: not in child thread yet; |
| 2: just enter child thread ; |
| 3: just before child thread exit; |
| */ |
| |
| #define NOT_CREATED_THREAD 1 |
| #define ENTERED_THREAD 2 |
| #define EXITING_THREAD 3 |
| |
| static int thread_state; |
| static int handler_called; |
| |
| static void sig_handler() { |
| if (pthread_equal(pthread_self(), sig_thread)) |
| { |
| printf("sig_handler: handled signal SIGUSR1\n"); |
| handler_called = 1; |
| } |
| else |
| { |
| printf("signal was not handled by sig_thread\n"); |
| exit(PTS_UNRESOLVED); |
| } |
| } |
| |
| static void * th_fn(void *arg) |
| { |
| struct sigaction act; |
| int rc = 0; |
| |
| /* Set up signal handler for SIGUSR1 */ |
| act.sa_flags = 0; |
| act.sa_handler = sig_handler; |
| /* block all the signal while handling SIGUSR1 */ |
| sigfillset(&act.sa_mask); |
| sigaction(SIGUSR1, &act, NULL); |
| |
| thread_state = ENTERED_THREAD; |
| printf("sig_thread: attempt write lock\n"); |
| rc = pthread_rwlock_wrlock(&rwlock); |
| if (rc != 0) |
| { |
| printf("Test FAILED: sig_thread: Error at pthread_rwlock_wrlock(), error code:%d\n", rc); |
| exit(PTS_FAIL); |
| } |
| |
| printf("sig_thread: acquired write lock\n"); |
| printf("sig_thread: unlock write lock\n"); |
| if (pthread_rwlock_unlock(&rwlock) != 0) |
| { |
| printf("sig_thread: Error at pthread_rwlock_unlock()\n"); |
| exit(PTS_UNRESOLVED); |
| } |
| thread_state = EXITING_THREAD; |
| pthread_exit(0); |
| return NULL; |
| } |
| |
| int main() |
| { |
| int cnt; |
| int rc = 0; |
| handler_called=0; |
| |
| if (pthread_rwlock_init(&rwlock, NULL) != 0) |
| { |
| printf("Error at pthread_rwlock_init()\n"); |
| return PTS_UNRESOLVED; |
| } |
| |
| printf("main: attempt write lock\n"); |
| rc = pthread_rwlock_wrlock(&rwlock); |
| if (rc != 0) |
| { |
| printf("main: Error at pthread_rwlock_wrlock(), error code:%d\n", rc); |
| return PTS_UNRESOLVED; |
| } |
| |
| thread_state = NOT_CREATED_THREAD; |
| if (pthread_create(&sig_thread, NULL, th_fn, NULL) != 0) |
| { |
| printf("Error at pthread_create()\n"); |
| return PTS_UNRESOLVED; |
| } |
| |
| /* wait at most 3 seconds for sig_thread to block*/ |
| cnt = 0; |
| do{ |
| sleep(1); |
| }while (thread_state != EXITING_THREAD && cnt++ < 3); |
| |
| if (thread_state == EXITING_THREAD) |
| { |
| /* the sig_thread is not blocking*/ |
| printf("Test FAILED: the thread should block when getting write lock\n"); |
| exit(PTS_FAIL); |
| } |
| else if (thread_state != ENTERED_THREAD) |
| { |
| printf("sig_thread in unexpected state %d\n", thread_state); |
| exit(PTS_UNRESOLVED); |
| } |
| |
| /* sig_thread is blocking */ |
| printf("main: fire SIGUSR1 to sig_thread\n"); |
| if (pthread_kill(sig_thread, SIGUSR1) != 0) |
| { |
| printf("Error at pthread_kill()\n"); |
| exit(PTS_UNRESOLVED); |
| } |
| |
| /* wait at most 3 seconds for the singal to be handled */ |
| cnt = 0; |
| do{ |
| sleep(1); |
| }while (handler_called == 0 && cnt++ < 3); |
| |
| if (handler_called != 1) |
| { |
| printf("The signal handler did not get called.\n"); |
| exit(PTS_UNRESOLVED); |
| } |
| |
| /* sig_thread resume to block? */ |
| cnt = 0; |
| do{ |
| sleep(1); |
| }while (thread_state != EXITING_THREAD && cnt++ < 3); |
| |
| if (thread_state == 3) |
| { |
| printf("Test FAILED: upon return from signal handler, sig_thread does not resume to wait\n"); |
| exit(PTS_FAIL); |
| } |
| |
| printf("sig_thread: correctly still blocking after signal handler returns\n"); |
| printf("main: unlock write lock\n"); |
| if (pthread_rwlock_unlock(&rwlock) != 0) |
| { |
| printf("main: Error releasing write lock\n"); |
| exit(PTS_UNRESOLVED); |
| } |
| |
| /* sig_thread should get write lock */ |
| cnt = 0; |
| do{ |
| sleep(1); |
| }while (thread_state != EXITING_THREAD && cnt++ < 3); |
| |
| if (thread_state != EXITING_THREAD) |
| { |
| /* sig_thread does not unblock */ |
| printf("Test FAILED: sig_thread should get the write lock and exit\n"); |
| exit(PTS_FAIL); |
| } |
| |
| if (pthread_join(sig_thread, NULL) != 0) |
| { |
| printf("Error at pthread_join()\n"); |
| exit(PTS_UNRESOLVED); |
| } |
| |
| if (pthread_rwlock_destroy(&rwlock) != 0) |
| { |
| printf("pthread_rwlock_destroy()\n"); |
| exit(PTS_UNRESOLVED); |
| } |
| |
| printf("Test PASSED\n"); |
| return PTS_PASS; |
| } |