kernel/test/synch/rwlock3.c

   1 /*
   2  * Copyright (C) 2001-2004 Jakub Jermar
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  *
   9  * - Redistributions of source code must retain the above copyright
  10  *   notice, this list of conditions and the following disclaimer.
  11  * - Redistributions in binary form must reproduce the above copyright
  12  *   notice, this list of conditions and the following disclaimer in the
  13  *   documentation and/or other materials provided with the distribution.
  14  * - The name of the author may not be used to endorse or promote products
  15  *   derived from this software without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  18  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  19  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  20  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  21  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  22  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  23  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  24  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27  */
  28
  29 #include <test.h>
  30 #include <arch.h>
  31 #include <atomic.h>
  32 #include <print.h>
  33 #include <proc/thread.h>
  34
  35 #include <synch/rwlock.h>
  36
  37 #define THREADS 4
  38
  39 static atomic_t thread_count;
  40 static rwlock_t rwlock;
  41 static bool sh_quiet;
  42
  43 static void reader(void *arg)
  44 {
  45         thread_detach(THREAD);
  46
  47         if (!sh_quiet)
  48                 printf("cpu%d, tid %d: trying to lock rwlock for reading....\n", CPU->id, THREAD->tid);
  49
  50         rwlock_read_lock(&rwlock);
  51         rwlock_read_unlock(&rwlock);
  52
  53         if (!sh_quiet) {
  54                 printf("cpu%d, tid %d: success\n", CPU->id, THREAD->tid);
  55                 printf("cpu%d, tid %d: trying to lock rwlock for writing....\n", CPU->id, THREAD->tid);
  56         }
  57
  58         rwlock_write_lock(&rwlock);
  59         rwlock_write_unlock(&rwlock);
  60
  61         if (!sh_quiet)
  62                 printf("cpu%d, tid %d: success\n", CPU->id, THREAD->tid);
  63
  64         atomic_dec(&thread_count);
  65 }
  66
  67 char * test_rwlock3(bool quiet)
  68 {
  69         int i;
  70         thread_t *thrd;
  71         sh_quiet = quiet;
  72
  73         atomic_set(&thread_count, THREADS);
  74
  75         rwlock_initialize(&rwlock);
  76         rwlock_write_lock(&rwlock);
  77
  78         for (i = 0; i < THREADS; i++) {
  79                 thrd = thread_create(reader, NULL, TASK, 0, "reader", false);
  80                 if (thrd)
  81                         thread_ready(thrd);
  82                 else if (!quiet)
  83                         printf("Could not create reader %d\n", i);
  84         }
  85
  86         thread_sleep(1);
  87         rwlock_write_unlock(&rwlock);
  88
  89         while (atomic_get(&thread_count) > 0) {
  90                 if (!quiet)
  91                         printf("Threads left: %d\n", atomic_get(&thread_count));
  92                 thread_sleep(1);
  93         }
  94
  95         return NULL;
  96 }