michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: /*
michael@0:  * RWLock rank tests
michael@0:  */
michael@0: 
michael@0: #include "nspr.h"
michael@0: #include "plgetopt.h"
michael@0: 
michael@0: static int _debug_on;
michael@0: static PRRWLock *rwlock0;
michael@0: static PRRWLock *rwlock1;
michael@0: static PRRWLock *rwlock2;
michael@0: 
michael@0: static void rwtest(void *args)
michael@0: {
michael@0:     PR_RWLock_Rlock(rwlock1);
michael@0:     PR_RWLock_Unlock(rwlock1);
michael@0: 
michael@0:     PR_RWLock_Rlock(rwlock1);
michael@0:     PR_RWLock_Unlock(rwlock1);
michael@0: 
michael@0:     // Test correct lock rank.
michael@0:     PR_RWLock_Rlock(rwlock1);
michael@0:     PR_RWLock_Rlock(rwlock2);
michael@0:     PR_RWLock_Unlock(rwlock2);
michael@0:     PR_RWLock_Unlock(rwlock1);
michael@0: 
michael@0:     PR_RWLock_Rlock(rwlock1);
michael@0:     PR_RWLock_Rlock(rwlock2);
michael@0:     PR_RWLock_Unlock(rwlock1);
michael@0:     PR_RWLock_Unlock(rwlock2);
michael@0: 
michael@0:     PR_RWLock_Rlock(rwlock1);
michael@0:     PR_RWLock_Rlock(rwlock0);
michael@0:     PR_RWLock_Rlock(rwlock2);
michael@0:     PR_RWLock_Unlock(rwlock2);
michael@0:     PR_RWLock_Unlock(rwlock0);
michael@0:     PR_RWLock_Unlock(rwlock1);
michael@0: 
michael@0: #if 0
michael@0:     // Test incorrect lock rank.
michael@0:     PR_RWLock_Rlock(rwlock2);
michael@0:     PR_RWLock_Rlock(rwlock1);
michael@0:     PR_RWLock_Unlock(rwlock1);
michael@0:     PR_RWLock_Unlock(rwlock2);
michael@0: 
michael@0:     PR_RWLock_Rlock(rwlock2);
michael@0:     PR_RWLock_Rlock(rwlock0);
michael@0:     PR_RWLock_Rlock(rwlock1);
michael@0:     PR_RWLock_Unlock(rwlock1);
michael@0:     PR_RWLock_Unlock(rwlock0);
michael@0:     PR_RWLock_Unlock(rwlock2);
michael@0: #endif
michael@0: }
michael@0: 
michael@0: int main(int argc, char **argv)
michael@0: {
michael@0:     PRStatus rc;
michael@0:     PRThread *thread;
michael@0:     PLOptStatus os;
michael@0:     PLOptState *opt = PL_CreateOptState(argc, argv, "d");
michael@0: 
michael@0:     while (PL_OPT_EOL != (os = PL_GetNextOpt(opt))) {
michael@0:         if (PL_OPT_BAD == os) continue;
michael@0:         switch (opt->option) {
michael@0:         case 'd':  /* debug mode */
michael@0:             _debug_on = 1;
michael@0:             break;
michael@0:          default:
michael@0:             break;
michael@0:         }
michael@0:     }
michael@0:     PL_DestroyOptState(opt);
michael@0: 
michael@0:     rwlock0 = PR_NewRWLock(PR_RWLOCK_RANK_NONE, "Lock 0");
michael@0:     if (rwlock0 == NULL) {
michael@0:         fprintf(stderr, "PR_NewRWLock failed - error %d\n",
michael@0:                 (int)PR_GetError());
michael@0:         return 1;
michael@0:     }
michael@0:     rwlock1 = PR_NewRWLock(1, "Lock 1");
michael@0:     if (rwlock1 == NULL) {
michael@0:         fprintf(stderr, "PR_NewRWLock failed - error %d\n",
michael@0:                 (int)PR_GetError());
michael@0:         return 1;
michael@0:     }
michael@0:     rwlock2 = PR_NewRWLock(2, "Lock 2");
michael@0:     if (rwlock2 == NULL) {
michael@0:         fprintf(stderr, "PR_NewRWLock failed - error %d\n",
michael@0:                 (int)PR_GetError());
michael@0:         return 1;
michael@0:     }
michael@0: 
michael@0:     thread = PR_CreateThread(PR_USER_THREAD, rwtest, NULL, PR_PRIORITY_NORMAL,
michael@0:                              PR_GLOBAL_THREAD, PR_JOINABLE_THREAD, 0);
michael@0:     if (thread == NULL) {
michael@0:         fprintf(stderr, "PR_CreateThread failed - error %d\n",
michael@0:                 (int)PR_GetError());
michael@0:         PR_ProcessExit(2);
michael@0:     }
michael@0:     if (_debug_on) {
michael@0:         printf("%s: created thread = %p\n", argv[0], thread);
michael@0:     }
michael@0: 
michael@0:     rc = PR_JoinThread(thread);
michael@0:     PR_ASSERT(rc == PR_SUCCESS);
michael@0: 
michael@0:     PR_DestroyRWLock(rwlock0);
michael@0:     rwlock0 = NULL;
michael@0:     PR_DestroyRWLock(rwlock1);
michael@0:     rwlock1 = NULL;
michael@0:     PR_DestroyRWLock(rwlock2);
michael@0:     rwlock2 = NULL;
michael@0: 
michael@0:     printf("PASS\n");
michael@0:     return 0;
michael@0: }