tst-spin2.c 3.3 KB

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