xref: /freebsd/tools/regression/sysvshm/shmtest.c (revision 1e413cf93298b5b97441a21d9a50fdcd0ee9945e)
1 /*-
2  * Copyright (c) 1999 The NetBSD Foundation, Inc.
3  * All rights reserved.
4  *
5  * This code is derived from software contributed to The NetBSD Foundation
6  * by Jason R. Thorpe of the Numerical Aerospace Simulation Facility,
7  * NASA Ames Research Center.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer.
14  * 2. Redistributions in binary form must reproduce the above copyright
15  *    notice, this list of conditions and the following disclaimer in the
16  *    documentation and/or other materials provided with the distribution.
17  * 3. All advertising materials mentioning features or use of this software
18  *    must display the following acknowledgement:
19  *	This product includes software developed by the NetBSD
20  *	Foundation, Inc. and its contributors.
21  * 4. Neither the name of The NetBSD Foundation nor the names of its
22  *    contributors may be used to endorse or promote products derived
23  *    from this software without specific prior written permission.
24  *
25  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
26  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
27  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
28  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
29  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
30  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
31  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
32  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
33  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
34  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
35  * POSSIBILITY OF SUCH DAMAGE.
36  *
37  * Obtained from: $NetBSD: shmtest.c,v 1.3 2002/07/20 08:36:26 grant Exp $
38  * $FreeBSD$
39  */
40 
41 /*
42  * Test the SVID-compatible Shared Memory facility.
43  */
44 
45 #include <sys/param.h>
46 #include <sys/ipc.h>
47 #include <sys/shm.h>
48 #include <sys/wait.h>
49 
50 #include <err.h>
51 #include <errno.h>
52 #include <signal.h>
53 #include <stdio.h>
54 #include <stdlib.h>
55 #include <string.h>
56 #include <time.h>
57 #include <unistd.h>
58 
59 int	main __P((int, char *[]));
60 void	print_shmid_ds __P((struct shmid_ds *, mode_t));
61 void	sigsys_handler __P((int));
62 void	sigchld_handler __P((int));
63 void	cleanup __P((void));
64 void	receiver __P((void));
65 void	usage __P((void));
66 
67 const char *m_str = "The quick brown fox jumped over the lazy dog.";
68 
69 int	sender_shmid = -1;
70 pid_t	child_pid;
71 
72 key_t	shmkey;
73 
74 size_t	pgsize;
75 
76 int
77 main(argc, argv)
78 	int argc;
79 	char *argv[];
80 {
81 	struct sigaction sa;
82 	struct shmid_ds s_ds;
83 	sigset_t sigmask;
84 	char *shm_buf;
85 
86 	if (argc != 2)
87 		usage();
88 
89 	/*
90 	 * Install a SIGSYS handler so that we can exit gracefully if
91 	 * System V Shared Memory support isn't in the kernel.
92 	 */
93 	sa.sa_handler = sigsys_handler;
94 	sigemptyset(&sa.sa_mask);
95 	sa.sa_flags = 0;
96 	if (sigaction(SIGSYS, &sa, NULL) == -1)
97 		err(1, "sigaction SIGSYS");
98 
99 	/*
100 	 * Install and SIGCHLD handler to deal with all possible exit
101 	 * conditions of the receiver.
102 	 */
103 	sa.sa_handler = sigchld_handler;
104 	sigemptyset(&sa.sa_mask);
105 	sa.sa_flags = 0;
106 	if (sigaction(SIGCHLD, &sa, NULL) == -1)
107 		err(1, "sigaction SIGCHLD");
108 
109 	pgsize = sysconf(_SC_PAGESIZE);
110 
111 	shmkey = ftok(argv[1], 4160);
112 
113 	/*
114 	 * Initialize child_pid to ourselves to that the cleanup function
115 	 * works before we create the receiver.
116 	 */
117 	child_pid = getpid();
118 
119 	/*
120 	 * Make sure that when the sender exits, the message queue is
121 	 * removed.
122 	 */
123 	if (atexit(cleanup) == -1)
124 		err(1, "atexit");
125 
126 	if ((sender_shmid = shmget(shmkey, pgsize, IPC_CREAT | 0640)) == -1)
127 		err(1, "shmget");
128 
129 	if (shmctl(sender_shmid, IPC_STAT, &s_ds) == -1)
130 		err(1, "shmctl IPC_STAT");
131 
132 	print_shmid_ds(&s_ds, 0640);
133 
134 	s_ds.shm_perm.mode = (s_ds.shm_perm.mode & ~0777) | 0600;
135 
136 	if (shmctl(sender_shmid, IPC_SET, &s_ds) == -1)
137 		err(1, "shmctl IPC_SET");
138 
139 	memset(&s_ds, 0, sizeof(s_ds));
140 
141 	if (shmctl(sender_shmid, IPC_STAT, &s_ds) == -1)
142 		err(1, "shmctl IPC_STAT");
143 
144 	if ((s_ds.shm_perm.mode & 0777) != 0600)
145 		err(1, "IPC_SET of mode didn't hold");
146 
147 	print_shmid_ds(&s_ds, 0600);
148 
149 	if ((shm_buf = shmat(sender_shmid, NULL, 0)) == (void *) -1)
150 		err(1, "sender: shmat");
151 
152 	/*
153 	 * Write the test pattern into the shared memory buffer.
154 	 */
155 	strcpy(shm_buf, m_str);
156 
157 	switch ((child_pid = fork())) {
158 	case -1:
159 		err(1, "fork");
160 		/* NOTREACHED */
161 
162 	case 0:
163 		receiver();
164 		break;
165 
166 	default:
167 		break;
168 	}
169 
170 	/*
171 	 * Suspend forever; when we get SIGCHLD, the handler will exit.
172 	 */
173 	sigemptyset(&sigmask);
174 	(void) sigsuspend(&sigmask);
175 
176 	/*
177 	 * ...and any other signal is an unexpected error.
178 	 */
179 	errx(1, "sender: received unexpected signal");
180 }
181 
182 void
183 sigsys_handler(signo)
184 	int signo;
185 {
186 
187 	errx(1, "System V Shared Memory support is not present in the kernel");
188 }
189 
190 void
191 sigchld_handler(signo)
192 	int signo;
193 {
194 	struct shmid_ds s_ds;
195 	int cstatus;
196 
197 	/*
198 	 * Reap the child; if it exited successfully, then the test passed!
199 	 */
200 	if (waitpid(child_pid, &cstatus, 0) != child_pid)
201 		err(1, "waitpid");
202 
203 	if (WIFEXITED(cstatus) == 0)
204 		errx(1, "receiver exited abnormally");
205 
206 	if (WEXITSTATUS(cstatus) != 0)
207 		errx(1, "receiver exited with status %d",
208 		    WEXITSTATUS(cstatus));
209 
210 	/*
211 	 * If we get here, the child has exited normally, and thus
212 	 * we should exit normally too.  First, tho, we print out
213 	 * the final stats for the message queue.
214 	 */
215 
216 	if (shmctl(sender_shmid, IPC_STAT, &s_ds) == -1)
217 		err(1, "shmctl IPC_STAT");
218 
219 	print_shmid_ds(&s_ds, 0600);
220 
221 	exit(0);
222 }
223 
224 void
225 cleanup()
226 {
227 
228 	/*
229 	 * If we're the sender, and it exists, remove the shared memory area.
230 	 */
231 	if (child_pid != 0 && sender_shmid != -1) {
232 		if (shmctl(sender_shmid, IPC_RMID, NULL) == -1)
233 			warn("shmctl IPC_RMID");
234 	}
235 }
236 
237 void
238 print_shmid_ds(sp, mode)
239 	struct shmid_ds *sp;
240 	mode_t mode;
241 {
242 	uid_t uid = geteuid();
243 	gid_t gid = getegid();
244 
245 	printf("PERM: uid %d, gid %d, cuid %d, cgid %d, mode 0%o\n",
246 	    sp->shm_perm.uid, sp->shm_perm.gid,
247 	    sp->shm_perm.cuid, sp->shm_perm.cgid,
248 	    sp->shm_perm.mode & 0777);
249 
250 	printf("segsz %lu, lpid %d, cpid %d, nattch %u\n",
251 	    (u_long)sp->shm_segsz, sp->shm_lpid, sp->shm_cpid,
252 	    sp->shm_nattch);
253 
254 	printf("atime: %s", ctime(&sp->shm_atime));
255 	printf("dtime: %s", ctime(&sp->shm_dtime));
256 	printf("ctime: %s", ctime(&sp->shm_ctime));
257 
258 	/*
259 	 * Sanity check a few things.
260 	 */
261 
262 	if (sp->shm_perm.uid != uid || sp->shm_perm.cuid != uid)
263 		errx(1, "uid mismatch");
264 
265 	if (sp->shm_perm.gid != gid || sp->shm_perm.cgid != gid)
266 		errx(1, "gid mismatch");
267 
268 	if ((sp->shm_perm.mode & 0777) != mode)
269 		errx(1, "mode mismatch");
270 }
271 
272 void
273 usage()
274 {
275 
276 	fprintf(stderr, "usage: %s keypath\n", getprogname());
277 	exit(1);
278 }
279 
280 void
281 receiver()
282 {
283 	int shmid;
284 	void *shm_buf;
285 
286 	if ((shmid = shmget(shmkey, pgsize, 0)) == -1)
287 		err(1, "receiver: shmget");
288 
289 	if ((shm_buf = shmat(shmid, NULL, 0)) == (void *) -1)
290 		err(1, "receiver: shmat");
291 
292 	printf("%s\n", (const char *)shm_buf);
293 	if (strcmp((const char *)shm_buf, m_str) != 0)
294 		err(1, "receiver: data isn't correct");
295 
296 	exit(0);
297 }
298