b2a6f2440bdd1a9678d0049adb48d6ce17dd6292
[dragonfly.git] / tools / regression / p1003_1b / sched.c
1 /*
2  * Copyright (c) 1996-1999
3  *      HD Associates, Inc.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *      This product includes software developed by HD Associates, Inc
16  * 4. Neither the name of the author nor the names of any co-contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY HD ASSOCIATES AND CONTRIBUTORS ``AS IS'' AND
21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23  * ARE DISCLAIMED.  IN NO EVENT SHALL HD ASSOCIATES OR CONTRIBUTORS BE LIABLE
24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30  * SUCH DAMAGE.
31  * $FreeBSD: src/tools/regression/p1003_1b/sched.c,v 1.1 2000/02/16 14:28:42 dufault Exp $
32  * $DragonFly: src/tools/regression/p1003_1b/sched.c,v 1.2 2003/06/17 04:29:11 dillon Exp $
33  *
34  */
35
36 /* XXX: The spec says that if _POSIX_C_SOURCE is defined then
37  *      _POSIX_SOURCE is ignored.  However, this is similar to
38  *      the code in the O'Reilly "POSIX.4" book
39  */
40
41 #define _POSIX_VERSION 199309L
42 #define _POSIX_SOURCE
43 #define _POSIX_C_SOURCE 199309L
44
45 #include <unistd.h>
46 #include <stdlib.h>
47
48 #include <stdio.h>
49 #include <string.h>
50 #include <errno.h>
51 #include <fcntl.h>
52 #include <sys/mman.h>
53
54 #include <sched.h>
55
56 #include "prutil.h"
57
58 static FILE *verbose;
59
60 static void
61 checkpris(int sched)
62 {
63         int smin;
64         int smax;
65
66         errno = 0;
67
68         if ( (smin = sched_get_priority_min(sched)) == -1 && errno)
69                 quit("sched_get_priority_min");
70
71         if ( (smax = sched_get_priority_max(sched)) == -1 && errno)
72                 quit("sched_get_priority_max");
73
74         if (smax - smin + 1 < 32 || smax < smin) {
75                 fprintf(stderr, "Illegal priority range for %s: %d to %d\n",
76                 sched_text(sched), smin, smax);
77                 exit(-1);
78         }
79
80         if (verbose)
81                 fprintf(verbose, "%12s: sched_min %2d sched_max %2d\n",
82                 sched_text(sched), smin, smax);
83 }
84
85 /* Set "try_anyway" to quit if you don't want to go on when
86  * it doesn't look like something should work.
87  */
88 static void try_anyway(const char *s)
89 {
90         fputs(s, stderr);
91         fprintf(stderr, "(trying anyway)\n");
92         errno = 0;
93 }
94
95 static void q(int line, int code, const char *text)
96 {
97         if (code == -1)
98         {
99                 fprintf(stderr, "Error at line %d:\n", line);
100                 perror(text);
101                 exit(errno);
102         }
103 }
104
105 int sched(int ac, char *av[])
106 {
107         int fifo_schedmin, fifo_schedmax;
108         int i;
109         struct sched_param rt_param;
110         int n_instances = 10;
111         int sched;
112
113         verbose = 0;
114
115 #if _POSIX_VERSION < 199309
116         try_anyway("The _POSIX_VERSION predates P1003.1B\n");
117 #endif
118
119 #if !defined(_POSIX_PRIORITY_SCHEDULING)
120         try_anyway(
121         "The environment does not claim to support Posix scheduling.\n");
122 #endif
123
124         /* Is priority scheduling configured?
125          */
126         errno = 0;
127         if (sysconf(_SC_PRIORITY_SCHEDULING) == -1) {
128                 if (errno != 0) {
129                         /* This isn't valid - may be a standard violation
130                          */
131                         quit("(should not happen) sysconf(_SC_PRIORITY_SCHEDULING)");
132                 }
133                 else {
134                         try_anyway(
135                         "The environment does not have run-time "
136                         "support for Posix scheduling.\n");
137                 }
138         }
139
140         /* Check that the priorities seem reasonable.
141          */
142
143         checkpris(SCHED_FIFO);
144         checkpris(SCHED_RR);
145         checkpris(SCHED_OTHER);
146
147 /* BSD extensions?
148  */
149 #if defined(SCHED_IDLE)
150         checkpris(SCHED_IDLE);
151 #endif
152
153         fifo_schedmin = sched_get_priority_min(SCHED_FIFO);
154         fifo_schedmax = sched_get_priority_max(SCHED_FIFO);
155
156         /* Make sure we can do some basic schedule switching:
157          */
158         {
159                 struct sched_param orig_param, shouldbe;
160                 int orig_scheduler = sched_is(__LINE__, &orig_param, -1);
161
162                 if (verbose)
163                         fprintf(verbose,
164                         "The original scheduler is %s and the priority is %d.\n",
165                         sched_text(orig_scheduler), orig_param.sched_priority);
166
167                 /* Basic check: Try to set current settings:
168                  */
169                 q(__LINE__, sched_setscheduler(0, orig_scheduler, &orig_param),
170                         "sched_setscheduler: Can't set original scheduler");
171
172                 rt_param.sched_priority = fifo_schedmin;
173
174                 q(__LINE__, sched_setscheduler(0, SCHED_FIFO, &rt_param),
175                 "sched_setscheduler SCHED_FIFO");
176
177                 (void)sched_is(__LINE__, 0, SCHED_FIFO);
178
179                 q(__LINE__, sched_getparam(0, &shouldbe), "sched_getparam");
180
181                 if (shouldbe.sched_priority != fifo_schedmin)
182                         quit("sched_setscheduler wrong priority (min)");
183
184                 rt_param.sched_priority = fifo_schedmin;
185
186                 q(__LINE__, sched_setparam(0, &rt_param),
187                         "sched_setparam to fifo_schedmin");
188
189                 rt_param.sched_priority = fifo_schedmin + 1;
190
191                 q(__LINE__, sched_setparam(0, &rt_param),
192                         "sched_setparam to fifo_schedmin + 1");
193
194                 q(__LINE__, sched_getparam(0, &shouldbe),
195                         "sched_getparam");
196
197                 if (shouldbe.sched_priority != fifo_schedmin + 1)
198                         quit("sched_setscheduler wrong priority (min + 1)");
199
200                 q(__LINE__, sched_setscheduler(0, SCHED_RR, &rt_param),
201                         "sched_setscheduler SCHED_RR");
202
203                 (void)sched_is(__LINE__, 0, SCHED_RR);
204
205                 q(__LINE__, sched_setscheduler(0, orig_scheduler, &orig_param),
206                         "sched_setscheduler restoring original scheduler");
207
208                 (void)sched_is(__LINE__, 0, orig_scheduler);
209         }
210
211
212         {
213 #define NAM "P1003_1b_schedXXXX"
214                 char nam[L_tmpnam];
215                 int fd;
216                 pid_t p;
217                 pid_t *lastrun;
218
219                 strcpy(nam, NAM);
220                 if (tmpnam(nam) != nam)
221                         q(__LINE__, errno, "tmpnam " NAM);
222                 q(__LINE__, (fd = open(nam, O_RDWR|O_CREAT, 0666)),
223                         "open " NAM);
224
225                 (void)unlink(nam);
226
227                 p = (pid_t)0;
228
229                 write(fd, &p, sizeof(p));
230
231                 q(__LINE__,  (int)(lastrun = mmap(0, sizeof(*lastrun), PROT_READ|PROT_WRITE,
232                 MAP_SHARED, fd, 0)), "mmap");
233
234                 /* Set our priority at the highest:
235                  */
236                 sched = SCHED_FIFO;
237                 rt_param.sched_priority = fifo_schedmax;
238                 q(__LINE__, sched_setscheduler(0, sched, &rt_param),
239                 "sched_setscheduler sched");
240
241                 for (i = 0; i < n_instances; i++)
242                 {
243                         pid_t me;
244
245                         /* XXX This is completely bogus.  The children never run.
246                          */
247                         if ((me = fork()) != 0)
248                         {
249                                 /* Parent.
250                                  */
251                                 (void)sched_is(__LINE__, 0, sched);
252
253                                 /* Lower our priority:
254                                  */
255                                 rt_param.sched_priority--;
256
257                                 q(__LINE__, sched_setscheduler(0, sched, &rt_param),
258                                 "sched_setscheduler sched");
259
260                                 while (1)
261                                 {
262                                         q(__LINE__, sched_getparam(0, &rt_param), "sched_getparam");
263
264                                         rt_param.sched_priority--;
265
266
267                                         if (rt_param.sched_priority < fifo_schedmin)
268                                                 exit(0);
269
270                                         *lastrun = me;
271                                         q(__LINE__, sched_setparam(0, &rt_param), "sched_setparam");
272
273                                         if (*lastrun == me)
274                                         {
275                                                 /* The child will run twice
276                                                  * at  the end:
277                                                  */
278                                                 if (!me || rt_param.sched_priority != 0)
279                                                 {
280                                                         fprintf(stderr,
281                                                         "ran process %ld twice at priority %d\n",
282                                                         (long)me, rt_param.sched_priority + 1);
283                                                         exit(-1);
284                                                 }
285                                         }
286                                 }
287                         }
288                 }
289         }
290
291         return 0;
292 }
293 #ifdef STANDALONE_TESTS
294 int main(int argc, char *argv[]) { return sched(argc, argv); }
295 #endif