nptl/tst-rwlock4.c

   1 /* Copyright (C) 2002, 2003 Free Software Foundation, Inc.
   2    This file is part of the GNU C Library.
   3    Contributed by Ulrich Drepper <drepper@redhat.com>, 2002.
   4
   5    The GNU C Library is free software; you can redistribute it and/or
   6    modify it under the terms of the GNU Lesser General Public
   7    License as published by the Free Software Foundation; either
   8    version 2.1 of the License, or (at your option) any later version.
   9
  10    The GNU C Library is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13    Lesser General Public License for more details.
  14
  15    You should have received a copy of the GNU Lesser General Public
  16    License along with the GNU C Library; if not, write to the Free
  17    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  18    02111-1307 USA.  */
  19
  20 #include <errno.h>
  21 #include <pthread.h>
  22 #include <stdint.h>
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26 #include <unistd.h>
  27 #include <sys/mman.h>
  28 #include <sys/wait.h>
  29
  30
  31 static int
  32 do_test (void)
  33 {
  34 #if ! _POSIX_THREAD_PROCESS_SHARED
  35
  36   puts ("_POSIX_THREAD_PROCESS_SHARED not supported, test skipped");
  37   return 0;
  38
  39 #else
  40
  41   size_t ps = sysconf (_SC_PAGESIZE);
  42   char tmpfname[] = "/tmp/tst-rwlock4.XXXXXX";
  43   char data[ps];
  44   void *mem;
  45   int fd;
  46   pthread_rwlock_t *r;
  47   pthread_rwlockattr_t a;
  48   pid_t pid;
  49   char *p;
  50   int err;
  51   int s;
  52
  53   fd = mkstemp (tmpfname);
  54   if (fd == -1)
  55     {
  56       printf ("cannot open temporary file: %m\n");
  57       return 1;
  58     }
  59
  60   /* Make sure it is always removed.  */
  61   unlink (tmpfname);
  62
  63   /* Create one page of data.  */
  64   memset (data, '\0', ps);
  65
  66   /* Write the data to the file.  */
  67   if (write (fd, data, ps) != ps)
  68     {
  69       puts ("short write");
  70       return 1;
  71     }
  72
  73   mem = mmap (NULL, ps, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
  74   if (mem == MAP_FAILED)
  75     {
  76       printf ("mmap failed: %m\n");
  77       return 1;
  78     }
  79
  80   r = (pthread_rwlock_t *) (((uintptr_t) mem + __alignof (pthread_rwlock_t))
  81                             & ~(__alignof (pthread_rwlock_t) - 1));
  82   p = (char *) (r + 1);
  83
  84   if (pthread_rwlockattr_init (&a) != 0)
  85     {
  86       puts ("rwlockattr_init failed");
  87       return 1;
  88     }
  89
  90   if (pthread_rwlockattr_getpshared (&a, &s) != 0)
  91     {
  92       puts ("1st rwlockattr_getpshared failed");
  93       return 1;
  94     }
  95
  96   if (s != PTHREAD_PROCESS_PRIVATE)
  97     {
  98       puts ("default pshared value wrong");
  99       return 1;
 100     }
 101
 102   if (pthread_rwlockattr_setpshared (&a, PTHREAD_PROCESS_SHARED) != 0)
 103     {
 104       puts ("rwlockattr_setpshared failed");
 105       return 1;
 106     }
 107
 108   if (pthread_rwlockattr_getpshared (&a, &s) != 0)
 109     {
 110       puts ("2nd rwlockattr_getpshared failed");
 111       return 1;
 112     }
 113
 114   if (s != PTHREAD_PROCESS_SHARED)
 115     {
 116       puts ("pshared value after setpshared call wrong");
 117       return 1;
 118     }
 119
 120   if (pthread_rwlock_init (r, &a) != 0)
 121     {
 122       puts ("rwlock_init failed");
 123       return 1;
 124     }
 125
 126   if (pthread_rwlock_rdlock (r) != 0)
 127     {
 128       puts ("rwlock_rdlock failed");
 129       return 1;
 130     }
 131
 132   if (pthread_rwlockattr_destroy (&a) != 0)
 133     {
 134       puts ("rwlockattr_destroy failed");
 135       return 1;
 136     }
 137
 138   err = pthread_rwlock_trywrlock (r);
 139   if (err == 0)
 140     {
 141       puts ("rwlock_trywrlock succeeded");
 142       return 1;
 143     }
 144   else if (err != EBUSY)
 145     {
 146       puts ("rwlock_trywrlock didn't return EBUSY");
 147       return 1;
 148     }
 149
 150   *p = 0;
 151
 152   puts ("going to fork now");
 153   pid = fork ();
 154   if (pid == -1)
 155     {
 156       puts ("fork failed");
 157       return 1;
 158     }
 159   else if (pid == 0)
 160     {
 161       /* Play some lock ping-pong.  It's our turn to unlock first.  */
 162       if ((*p)++ != 0)
 163         {
 164           puts ("child: *p != 0");
 165           return 1;
 166         }
 167
 168       if (pthread_rwlock_unlock (r) != 0)
 169         {
 170           puts ("child: 1st rwlock_unlock failed");
 171           return 1;
 172         }
 173
 174       puts ("child done");
 175     }
 176   else
 177     {
 178       if (pthread_rwlock_wrlock (r) != 0)
 179         {
 180           puts ("parent: rwlock_wrlock failed");
 181           return 1;
 182         }
 183
 184       if (*p != 1)
 185         {
 186           puts ("*p != 1");
 187           return 1;
 188         }
 189
 190       puts ("parent done");
 191     }
 192
 193   return 0;
 194 #endif
 195 }
 196
 197 #define TEST_FUNCTION do_test ()
 198 #include "../test-skeleton.c"