xref: /freebsd/contrib/netbsd-tests/lib/libc/gen/posix_spawn/t_spawnattr.c (revision 98e0ffaefb0f241cda3a72395d3be04192ae0d47)
1 /* $NetBSD: t_spawnattr.c,v 1.1 2012/02/13 21:03:08 martin Exp $ */
2 
3 /*-
4  * Copyright (c) 2012 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * This code is derived from software contributed to The NetBSD Foundation
8  * by Charles Zhang <charles@NetBSD.org> and
9  * Martin Husemann <martin@NetBSD.org>.
10  *
11  * Redistribution and use in source and binary forms, with or without
12  * modification, are permitted provided that the following conditions
13  * are met:
14  * 1. Redistributions of source code must retain the above copyright
15  *    notice, this list of conditions and the following disclaimer.
16  * 2. Redistributions in binary form must reproduce the above copyright
17  *    notice, this list of conditions and the following disclaimer in the
18  *    documentation and/or other materials provided with the distribution.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
21  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
22  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
23  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
24  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
25  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
26  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
27  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
28  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
29  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
30  * POSSIBILITY OF SUCH DAMAGE.
31  */
32 
33 #include <atf-c.h>
34 #include <stdio.h>
35 #include <stdlib.h>
36 #include <string.h>
37 #include <errno.h>
38 #include <fcntl.h>
39 #include <sched.h>
40 #include <signal.h>
41 #include <spawn.h>
42 #include <unistd.h>
43 #include <sys/wait.h>
44 
45 #define MAX(a, b)	(a) > (b) ? (a) : (b)
46 #define MIN(a, b)	(a) > (b) ? (b) : (a)
47 
48 static int get_different_scheduler(void);
49 static int get_different_priority(void);
50 
51 static int
52 get_different_scheduler()
53 {
54 	int scheduler, max, min, new;
55 
56 	max = MAX(MAX(SCHED_FIFO, SCHED_OTHER), SCHED_RR);
57 	min = MIN(MIN(SCHED_FIFO, SCHED_OTHER), SCHED_RR);
58 
59 	/* get current schedule policy */
60 	scheduler = sched_getscheduler(0);
61 
62 	/* new scheduler */
63 	new = (scheduler + 1);
64 	if (new > max)
65 		new = min;
66 
67 	return new;
68 }
69 
70 static int
71 get_different_priority()
72 {
73 	int scheduler, max, min, new, priority;
74 	struct sched_param param;
75 
76 	/* get current schedule policy */
77 	scheduler = sched_getscheduler(0);
78 
79 	max = sched_get_priority_max(scheduler);
80 	min = sched_get_priority_min(scheduler);
81 
82 	sched_getparam(0, &param);
83 	priority = param.sched_priority;
84 
85 	/* new schedule policy */
86 	new = (priority + 1);
87 	if (new > max)
88 		new = min;
89 
90 	return new;
91 }
92 
93 ATF_TC(t_spawnattr);
94 
95 ATF_TC_HEAD(t_spawnattr, tc)
96 {
97 	atf_tc_set_md_var(tc, "require.user", "root");
98 	atf_tc_set_md_var(tc, "descr",
99 	    "Tests posix_spawn with scheduler attributes");
100 }
101 
102 ATF_TC_BODY(t_spawnattr, tc)
103 {
104 	int pid, scheduler, child_scheduler, priority, status, err, pfd[2];
105 	char helper_arg[128];
106 	char * const args[] = { __UNCONST("h_spawnattr"), helper_arg, NULL };
107 	struct sched_param sp, child_sp;
108 	sigset_t sig;
109 	posix_spawnattr_t attr;
110 	char helper[FILENAME_MAX];
111 
112 	/*
113 	 * create a pipe to controll the child
114 	 */
115 	err = pipe(pfd);
116 	ATF_REQUIRE_MSG(err == 0, "could not create pipe, errno %d", errno);
117 	sprintf(helper_arg, "%d", pfd[0]);
118 
119 	posix_spawnattr_init(&attr);
120 
121 	scheduler = get_different_scheduler();
122 	priority = get_different_priority();
123 	sp.sched_priority = priority;
124 
125 	sigemptyset(&sig);
126 	sigaddset(&sig, SIGUSR1);
127 
128 	posix_spawnattr_setflags(&attr, POSIX_SPAWN_SETSCHEDULER |
129 		POSIX_SPAWN_SETSCHEDPARAM | POSIX_SPAWN_SETPGROUP |
130 		POSIX_SPAWN_SETSIGMASK | POSIX_SPAWN_SETSIGDEF |
131 		POSIX_SPAWN_SETSIGDEF);
132 	posix_spawnattr_setpgroup(&attr, 0);
133 	posix_spawnattr_setschedparam(&attr, &sp);
134 	posix_spawnattr_setschedpolicy(&attr, scheduler);
135 	posix_spawnattr_setsigmask(&attr, &sig);
136 	posix_spawnattr_setsigdefault(&attr, &sig);
137 
138 	sprintf(helper, "%s/h_spawnattr",
139 	    atf_tc_get_config_var(tc, "srcdir"));
140 	err = posix_spawn(&pid, helper, NULL, &attr, args, NULL);
141 	ATF_REQUIRE_MSG(err == 0, "error %d", err);
142 
143 	child_scheduler = sched_getscheduler(pid);
144 	ATF_REQUIRE_MSG(scheduler == child_scheduler,
145 	    "scheduler = %d, child_scheduler = %d, pid %d, errno %d",
146 	    scheduler, child_scheduler, pid, errno);
147 
148 	sched_getparam(pid, &child_sp);
149 	ATF_REQUIRE_MSG(child_sp.sched_priority == sp.sched_priority,
150 	    "priority is: %d, but we requested: %d",
151 	    child_sp.sched_priority, sp.sched_priority);
152 
153 	ATF_REQUIRE_MSG(pid == getpgid(pid), "child pid: %d, child pgid: %d",
154 	    pid, getpgid(pid));
155 
156 	/* ready, let child go */
157 	write(pfd[1], "q", 1);
158 	close(pfd[0]);
159 	close(pfd[1]);
160 
161 	/* wait and check result from child */
162 	waitpid(pid, &status, 0);
163 	ATF_REQUIRE(WIFEXITED(status) && WEXITSTATUS(status) == EXIT_SUCCESS);
164 
165 	posix_spawnattr_destroy(&attr);
166 }
167 
168 ATF_TP_ADD_TCS(tp)
169 {
170 	ATF_TP_ADD_TC(tp, t_spawnattr);
171 
172 	return atf_no_error();
173 }
174