1 /*
2 * Copyright (c) 2000-2002, 2004, 2005 Sendmail, Inc. and its suppliers.
3 * All rights reserved.
4 *
5 * By using this file, you agree to the terms and conditions set
6 * forth in the LICENSE file which can be found at the top level of
7 * the sendmail distribution.
8 */
9
10 #include <sm/gen.h>
11 SM_RCSID("@(#)$Id: t-shm.c,v 1.22 2005/01/14 02:14:10 ca Exp $")
12
13 #include <stdio.h>
14
15 #if SM_CONF_SHM
16 # include <stdlib.h>
17 # include <unistd.h>
18 # include <sys/wait.h>
19
20 # include <sm/heap.h>
21 # include <sm/string.h>
22 # include <sm/test.h>
23 # include <sm/shm.h>
24
25 # define SHMSIZE 1024
26 # define SHM_MAX 6400000
27 # define T_SHMKEY 21
28
29
30 /*
31 ** SHMINTER -- interactive testing of shared memory
32 **
33 ** Parameters:
34 ** owner -- create segment.
35 **
36 ** Returns:
37 ** 0 on success
38 ** < 0 on failure.
39 */
40
41 int shminter __P((bool));
42
43 int
shminter(owner)44 shminter(owner)
45 bool owner;
46 {
47 int *shm, shmid;
48 int i, t;
49
50 shm = (int *) sm_shmstart(T_SHMKEY, SHMSIZE, 0, &shmid, owner);
51 if (shm == (int *) 0)
52 {
53 perror("shminit failed");
54 return -1;
55 }
56
57 while ((t = getchar()) != EOF)
58 {
59 switch (t)
60 {
61 case 'c':
62 *shm = 0;
63 break;
64 case 'i':
65 ++*shm;
66 break;
67 case 'd':
68 --*shm;
69 break;
70 case 's':
71 sleep(1);
72 break;
73 case 'l':
74 t = *shm;
75 for (i = 0; i < SHM_MAX; i++)
76 {
77 ++*shm;
78 }
79 if (*shm != SHM_MAX + t)
80 fprintf(stderr, "error: %d != %d\n",
81 *shm, SHM_MAX + t);
82 break;
83 case 'v':
84 printf("shmval: %d\n", *shm);
85 break;
86 case 'S':
87 i = sm_shmsetowner(shmid, getuid(), getgid(), 0644);
88 printf("sm_shmsetowner=%d\n", i);
89 break;
90 }
91 }
92 return sm_shmstop((void *) shm, shmid, owner);
93 }
94
95
96 /*
97 ** SHMBIG -- testing of shared memory
98 **
99 ** Parameters:
100 ** owner -- create segment.
101 ** size -- size of segment.
102 **
103 ** Returns:
104 ** 0 on success
105 ** < 0 on failure.
106 */
107
108 int shmbig __P((bool, int));
109
110 int
shmbig(owner,size)111 shmbig(owner, size)
112 bool owner;
113 int size;
114 {
115 int *shm, shmid;
116 int i;
117
118 shm = (int *) sm_shmstart(T_SHMKEY, size, 0, &shmid, owner);
119 if (shm == (int *) 0)
120 {
121 perror("shminit failed");
122 return -1;
123 }
124
125 for (i = 0; i < size / sizeof(int); i++)
126 shm[i] = i;
127 for (i = 0; i < size / sizeof(int); i++)
128 {
129 if (shm[i] != i)
130 {
131 fprintf(stderr, "failed at %d: %d", i, shm[i]);
132 }
133 }
134
135 return sm_shmstop((void *) shm, shmid, owner);
136 }
137
138
139 /*
140 ** SHMTEST -- test of shared memory
141 **
142 ** Parameters:
143 ** owner -- create segment.
144 **
145 ** Returns:
146 ** 0 on success
147 ** < 0 on failure.
148 */
149
150 # define MAX_CNT 10
151
152 int shmtest __P((int));
153
154 int
shmtest(owner)155 shmtest(owner)
156 int owner;
157 {
158 int *shm, shmid;
159 int cnt = 0;
160
161 shm = (int *) sm_shmstart(T_SHMKEY, SHMSIZE, 0, &shmid, owner);
162 if (shm == (int *) 0)
163 {
164 perror("shminit failed");
165 return -1;
166 }
167
168 if (owner)
169 {
170 int r;
171
172 r = sm_shmsetowner(shmid, getuid(), getgid(), 0660);
173 SM_TEST(r == 0);
174 *shm = 1;
175 while (*shm == 1 && cnt++ < MAX_CNT)
176 sleep(1);
177 SM_TEST(cnt <= MAX_CNT);
178
179 /* release and re-acquire the segment */
180 r = sm_shmstop((void *) shm, shmid, owner);
181 SM_TEST(r == 0);
182 shm = (int *) sm_shmstart(T_SHMKEY, SHMSIZE, 0, &shmid, owner);
183 SM_TEST(shm != (int *) 0);
184 }
185 else
186 {
187 while (*shm != 1 && cnt++ < MAX_CNT)
188 sleep(1);
189 SM_TEST(cnt <= MAX_CNT);
190 *shm = 2;
191
192 /* wait a momemt so the segment is still in use */
193 sleep(2);
194 }
195 return sm_shmstop((void *) shm, shmid, owner);
196 }
197
198 int
main(argc,argv)199 main(argc, argv)
200 int argc;
201 char *argv[];
202 {
203 bool interactive = false;
204 bool owner = false;
205 int big = -1;
206 int ch;
207 int r = 0;
208 int status;
209 extern char *optarg;
210
211 # define OPTIONS "b:io"
212 while ((ch = getopt(argc, argv, OPTIONS)) != -1)
213 {
214 switch ((char) ch)
215 {
216 case 'b':
217 big = atoi(optarg);
218 break;
219
220 case 'i':
221 interactive = true;
222 break;
223
224 case 'o':
225 owner = true;
226 break;
227
228 default:
229 break;
230 }
231 }
232
233 if (interactive)
234 r = shminter(owner);
235 else if (big > 0)
236 r = shmbig(true, big);
237 else
238 {
239 pid_t pid;
240 extern int SmTestNumErrors;
241
242 if ((pid = fork()) < 0)
243 {
244 perror("fork failed\n");
245 return -1;
246 }
247
248 sm_test_begin(argc, argv, "test shared memory");
249 if (pid == 0)
250 {
251 /* give the parent the chance to setup data */
252 sleep(1);
253 r = shmtest(false);
254 }
255 else
256 {
257 r = shmtest(true);
258 (void) wait(&status);
259 }
260 SM_TEST(r == 0);
261 if (SmTestNumErrors > 0)
262 printf("add -DSM_CONF_SHM=0 to confENVDEF in devtools/Site/site.config.m4\nand start over.\n");
263 return sm_test_end();
264 }
265 return r;
266 }
267 #else /* SM_CONF_SHM */
268 int
269 main(argc, argv)
270 int argc;
271 char *argv[];
272 {
273 printf("No support for shared memory configured on this machine\n");
274 return 0;
275 }
276 #endif /* SM_CONF_SHM */
277