1 | /* Test for shm_open cancellation handling: BZ #18243. |
2 | Copyright (C) 2016-2022 Free Software Foundation, Inc. |
3 | This file is part of the GNU C Library. |
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, see |
17 | <https://www.gnu.org/licenses/>. */ |
18 | |
19 | #include <pthread.h> |
20 | #include <sys/mman.h> |
21 | #include <semaphore.h> |
22 | #include <stdio.h> |
23 | #include <fcntl.h> |
24 | #include <unistd.h> |
25 | #include <errno.h> |
26 | #include <stdlib.h> |
27 | |
28 | static sem_t sem; /* Use to sync with thread start. */ |
29 | static char shm_name[sizeof "/glibc-shm_open-cancel-" + sizeof (pid_t) * 3]; |
30 | |
31 | static void |
32 | init_shm_name (void) |
33 | { |
34 | snprintf (s: shm_name, maxlen: sizeof (shm_name), format: "/glibc-shm_open-cancel-%u" , |
35 | getpid ()); |
36 | } |
37 | |
38 | static void |
39 | remove_shm (int status, void *arg) |
40 | { |
41 | shm_unlink (name: shm_name); |
42 | } |
43 | |
44 | static void * |
45 | tf (void *arg) |
46 | { |
47 | pthread_setcancelstate (PTHREAD_CANCEL_DISABLE, oldstate: 0); |
48 | |
49 | if (sem_wait (sem: &sem) != 0) |
50 | { |
51 | printf (format: "error: sem_wait failed: %m" ); |
52 | exit (1); |
53 | } |
54 | |
55 | if (pthread_setcancelstate (PTHREAD_CANCEL_ENABLE, oldstate: 0) != 0) |
56 | { |
57 | printf (format: "error: pthread_setcancelstate failed: %m" ); |
58 | exit (1); |
59 | } |
60 | |
61 | /* Neither sem_unlink or sem_open should act on thread cancellation. */ |
62 | shm_unlink (name: shm_name); |
63 | on_exit (func: remove_shm, NULL); |
64 | |
65 | int fd = shm_open (name: shm_name, O_CREAT, mode: 0600); |
66 | if (fd == -1) |
67 | { |
68 | int exit_code; |
69 | if (errno == ENOSYS || errno == EACCES) |
70 | exit_code = 77; |
71 | else |
72 | exit_code = 1; |
73 | exit (exit_code); |
74 | } |
75 | |
76 | if (pthread_setcancelstate (PTHREAD_CANCEL_DISABLE, oldstate: 0) != 0) |
77 | { |
78 | printf (format: "error: pthread_setcancelstate failed: %m" ); |
79 | exit (1); |
80 | } |
81 | |
82 | if (close (fd: fd) != 0) |
83 | { |
84 | printf (format: "error: pthread_setcancelstate failed: %m" ); |
85 | exit (1); |
86 | } |
87 | |
88 | return NULL; |
89 | } |
90 | |
91 | static int |
92 | do_test (void) |
93 | { |
94 | pthread_t td; |
95 | |
96 | init_shm_name (); |
97 | |
98 | if (sem_init (sem: &sem, pshared: 0, value: 0)) |
99 | { |
100 | printf (format: "error: sem_init failed: %m\n" ); |
101 | exit (1); |
102 | } |
103 | |
104 | if (pthread_create (newthread: &td, NULL, start_routine: tf, NULL) != 0) |
105 | { |
106 | printf (format: "error: pthread_create failed: %m\n" ); |
107 | exit (1); |
108 | } |
109 | |
110 | if (pthread_cancel (th: td) != 0) |
111 | { |
112 | printf (format: "error: pthread_cancel failed: %m\n" ); |
113 | exit (1); |
114 | } |
115 | |
116 | if (sem_post (sem: &sem) != 0) |
117 | { |
118 | printf (format: "error: sem_post failed: %m\n" ); |
119 | exit (1); |
120 | } |
121 | |
122 | void *r; |
123 | if (pthread_join (th: td, thread_return: &r) != 0) |
124 | { |
125 | printf (format: "error: pthread_join failed: %m\n" ); |
126 | exit (1); |
127 | } |
128 | |
129 | if (r == PTHREAD_CANCELED) |
130 | { |
131 | puts (s: "error: pthread_join returned PTHREAD_CANCELED" ); |
132 | exit (1); |
133 | } |
134 | |
135 | return 0; |
136 | } |
137 | |
138 | #define TEST_FUNCTION do_test () |
139 | #include <test-skeleton.c> |
140 | |