tst-spin2.c 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159
  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, write to the Free
  14. Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  15. 02111-1307 USA. */
  16. #include <errno.h>
  17. #include <pthread.h>
  18. #include <stdint.h>
  19. #include <stdio.h>
  20. #include <stdlib.h>
  21. #include <string.h>
  22. #include <unistd.h>
  23. #include <sys/mman.h>
  24. #include <sys/wait.h>
  25. static int
  26. do_test (void)
  27. {
  28. size_t ps = sysconf (_SC_PAGESIZE);
  29. char tmpfname[] = "/tmp/tst-spin2.XXXXXX";
  30. char data[ps];
  31. void *mem;
  32. int fd;
  33. pthread_spinlock_t *s;
  34. pid_t pid;
  35. char *p;
  36. int err;
  37. fd = mkstemp (tmpfname);
  38. if (fd == -1)
  39. {
  40. printf ("cannot open temporary file: %m\n");
  41. return 1;
  42. }
  43. /* Make sure it is always removed. */
  44. unlink (tmpfname);
  45. /* Create one page of data. */
  46. memset (data, '\0', ps);
  47. /* Write the data to the file. */
  48. if (write (fd, data, ps) != (ssize_t) ps)
  49. {
  50. puts ("short write");
  51. return 1;
  52. }
  53. mem = mmap (NULL, ps, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
  54. if (mem == MAP_FAILED)
  55. {
  56. printf ("mmap failed: %m\n");
  57. return 1;
  58. }
  59. s = (pthread_spinlock_t *) (((uintptr_t) mem
  60. + __alignof (pthread_spinlock_t))
  61. & ~(__alignof (pthread_spinlock_t) - 1));
  62. p = (char *) (s + 1);
  63. if (pthread_spin_init (s, PTHREAD_PROCESS_SHARED) != 0)
  64. {
  65. puts ("spin_init failed");
  66. return 1;
  67. }
  68. if (pthread_spin_lock (s) != 0)
  69. {
  70. puts ("spin_lock failed");
  71. return 1;
  72. }
  73. err = pthread_spin_trylock (s);
  74. if (err == 0)
  75. {
  76. puts ("1st spin_trylock succeeded");
  77. return 1;
  78. }
  79. else if (err != EBUSY)
  80. {
  81. puts ("1st spin_trylock didn't return EBUSY");
  82. return 1;
  83. }
  84. err = pthread_spin_unlock (s);
  85. if (err != 0)
  86. {
  87. puts ("parent: spin_unlock failed");
  88. return 1;
  89. }
  90. err = pthread_spin_trylock (s);
  91. if (err != 0)
  92. {
  93. puts ("2nd spin_trylock failed");
  94. return 1;
  95. }
  96. *p = 0;
  97. puts ("going to fork now");
  98. pid = fork ();
  99. if (pid == -1)
  100. {
  101. puts ("fork failed");
  102. return 1;
  103. }
  104. else if (pid == 0)
  105. {
  106. /* Play some lock ping-pong. It's our turn to unlock first. */
  107. if ((*p)++ != 0)
  108. {
  109. puts ("child: *p != 0");
  110. return 1;
  111. }
  112. if (pthread_spin_unlock (s) != 0)
  113. {
  114. puts ("child: 1st spin_unlock failed");
  115. return 1;
  116. }
  117. puts ("child done");
  118. }
  119. else
  120. {
  121. if (pthread_spin_lock (s) != 0)
  122. {
  123. puts ("parent: 2nd spin_lock failed");
  124. return 1;
  125. }
  126. puts ("waiting for child");
  127. waitpid (pid, NULL, 0);
  128. puts ("parent done");
  129. }
  130. return 0;
  131. }
  132. #define TEST_FUNCTION do_test ()
  133. #include "../test-skeleton.c"