tst-rwlock4.c 3.9 KB

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