1 /*
2 * Copyright (c) 2015-2017 Dmitry V. Levin <ldv@altlinux.org>
3 * 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. The name of the author may not be used to endorse or promote products
14 * derived from this software without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
17 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27
28 #include "tests.h"
29 #include <asm/unistd.h>
30
31 #if defined __NR_sched_getattr && defined __NR_sched_setattr
32
33 # include <inttypes.h>
34 # include <stdio.h>
35 # include <sched.h>
36 # include <unistd.h>
37 # include "sched_attr.h"
38 # include "xlat.h"
39 # include "xlat/schedulers.h"
40
41 static const char *errstr;
42
43 static long
sys_sched_getattr(kernel_ulong_t pid,kernel_ulong_t attr,kernel_ulong_t size,kernel_ulong_t flags)44 sys_sched_getattr(kernel_ulong_t pid, kernel_ulong_t attr,
45 kernel_ulong_t size, kernel_ulong_t flags)
46 {
47 long rc = syscall(__NR_sched_getattr, pid, attr, size, flags);
48 errstr = sprintrc(rc);
49 return rc;
50 }
51
52 static long
sys_sched_setattr(kernel_ulong_t pid,kernel_ulong_t attr,kernel_ulong_t flags)53 sys_sched_setattr(kernel_ulong_t pid, kernel_ulong_t attr, kernel_ulong_t flags)
54 {
55 long rc = syscall(__NR_sched_setattr, pid, attr, flags);
56 errstr = sprintrc(rc);
57 return rc;
58 }
59
60 int
main(void)61 main(void)
62 {
63 static const kernel_ulong_t bogus_pid =
64 (kernel_ulong_t) 0xdefacedfacefeedULL;
65 static const kernel_ulong_t bogus_size =
66 (kernel_ulong_t) 0xdefacedcafef00dULL;
67 static const kernel_ulong_t bogus_flags =
68 (kernel_ulong_t) 0xdefaceddeadc0deULL;
69
70 TAIL_ALLOC_OBJECT_CONST_PTR(struct sched_attr, attr);
71 TAIL_ALLOC_OBJECT_CONST_PTR(unsigned int, psize);
72 void *const efault = attr + 1;
73
74 sys_sched_getattr(0, 0, 0, 0);
75 printf("sched_getattr(0, NULL, 0, 0) = %s\n", errstr);
76
77 sys_sched_getattr(0, (unsigned long) attr, 0, 0);
78 printf("sched_getattr(0, %p, 0, 0) = %s\n", attr, errstr);
79
80 sys_sched_getattr(bogus_pid, 0, 0, 0);
81 printf("sched_getattr(%d, NULL, 0, 0) = %s\n", (int) bogus_pid, errstr);
82
83 sys_sched_getattr(-1U, (unsigned long) attr, bogus_size, bogus_flags);
84 printf("sched_getattr(-1, %p, %s%u, %u) = %s\n",
85 attr,
86 # if defined __arm64__ || defined __aarch64__
87 "0xdefaced<<32|",
88 # else
89 "",
90 # endif
91 (unsigned) bogus_size, (unsigned) bogus_flags, errstr);
92
93 sys_sched_getattr(0, (unsigned long) efault, sizeof(*attr), 0);
94 printf("sched_getattr(0, %p, %u, 0) = %s\n",
95 efault, (unsigned) sizeof(*attr), errstr);
96
97 if (sys_sched_getattr(0, (unsigned long) attr, sizeof(*attr), 0))
98 perror_msg_and_skip("sched_getattr");
99 printf("sched_getattr(0, {size=%u, sched_policy=", attr->size);
100 printxval(schedulers, attr->sched_policy, NULL);
101 printf(", sched_flags=%s, sched_nice=%d, sched_priority=%u"
102 ", sched_runtime=%" PRIu64 ", sched_deadline=%" PRIu64
103 ", sched_period=%" PRIu64 "}, %u, 0) = 0\n",
104 attr->sched_flags ? "SCHED_FLAG_RESET_ON_FORK" : "0",
105 attr->sched_nice,
106 attr->sched_priority,
107 attr->sched_runtime,
108 attr->sched_deadline,
109 attr->sched_period,
110 (unsigned) sizeof(*attr));
111
112 # if defined __arm64__ || defined __aarch64__
113 long rc =
114 # endif
115 sys_sched_getattr(F8ILL_KULONG_MASK, (unsigned long) attr,
116 F8ILL_KULONG_MASK | sizeof(*attr), F8ILL_KULONG_MASK);
117 # if defined __arm64__ || defined __aarch64__
118 if (rc) {
119 printf("sched_getattr(0, %p, 0xffffffff<<32|%u, 0) = %s\n",
120 attr, (unsigned) sizeof(*attr), errstr);
121 } else
122 # endif
123 {
124 printf("sched_getattr(0, {size=%u, sched_policy=", attr->size);
125 printxval(schedulers, attr->sched_policy, NULL);
126 printf(", sched_flags=%s, sched_nice=%d, sched_priority=%u"
127 ", sched_runtime=%" PRIu64 ", sched_deadline=%" PRIu64
128 ", sched_period=%" PRIu64 "}, %u, 0) = 0\n",
129 attr->sched_flags ? "SCHED_FLAG_RESET_ON_FORK" : "0",
130 attr->sched_nice,
131 attr->sched_priority,
132 attr->sched_runtime,
133 attr->sched_deadline,
134 attr->sched_period,
135 (unsigned) sizeof(*attr));
136 }
137
138 sys_sched_setattr(bogus_pid, 0, 0);
139 printf("sched_setattr(%d, NULL, 0) = %s\n", (int) bogus_pid, errstr);
140
141 attr->sched_flags |= 1;
142
143 if (sys_sched_setattr(0, (unsigned long) attr, 0))
144 perror_msg_and_skip("sched_setattr");
145 printf("sched_setattr(0, {size=%u, sched_policy=", attr->size);
146 printxval(schedulers, attr->sched_policy, NULL);
147 printf(", sched_flags=%s, sched_nice=%d, sched_priority=%u"
148 ", sched_runtime=%" PRIu64 ", sched_deadline=%" PRIu64
149 ", sched_period=%" PRIu64 "}, 0) = 0\n",
150 "SCHED_FLAG_RESET_ON_FORK",
151 attr->sched_nice,
152 attr->sched_priority,
153 attr->sched_runtime,
154 attr->sched_deadline,
155 attr->sched_period);
156
157 sys_sched_setattr(F8ILL_KULONG_MASK, (unsigned long) attr,
158 F8ILL_KULONG_MASK);
159 printf("sched_setattr(0, {size=%u, sched_policy=", attr->size);
160 printxval(schedulers, attr->sched_policy, NULL);
161 printf(", sched_flags=%s, sched_nice=%d, sched_priority=%u"
162 ", sched_runtime=%" PRIu64 ", sched_deadline=%" PRIu64
163 ", sched_period=%" PRIu64 "}, 0) = 0\n",
164 "SCHED_FLAG_RESET_ON_FORK",
165 attr->sched_nice,
166 attr->sched_priority,
167 attr->sched_runtime,
168 attr->sched_deadline,
169 attr->sched_period);
170
171 *psize = attr->size;
172
173 sys_sched_setattr(0, (unsigned long) psize, 0);
174 printf("sched_setattr(0, %p, 0) = %s\n", psize, errstr);
175
176 attr->size = 0;
177
178 sys_sched_setattr(0, (unsigned long) attr, 0);
179 printf("sched_setattr(0, {size=%u, sched_policy=", attr->size);
180 printxval(schedulers, attr->sched_policy, NULL);
181 printf(", sched_flags=%s, sched_nice=%d, sched_priority=%u"
182 ", sched_runtime=%" PRIu64 ", sched_deadline=%" PRIu64
183 ", sched_period=%" PRIu64 "}, 0) = 0\n",
184 "SCHED_FLAG_RESET_ON_FORK",
185 attr->sched_nice,
186 attr->sched_priority,
187 attr->sched_runtime,
188 attr->sched_deadline,
189 attr->sched_period);
190
191 attr->size = 1;
192
193 sys_sched_setattr(0, (unsigned long) attr, 0);
194 printf("sched_setattr(0, {size=%u} => {size=%u}, 0) = %s\n",
195 1, attr->size, errstr);
196
197 attr->size = SCHED_ATTR_MIN_SIZE - 1;
198
199 sys_sched_setattr(0, (unsigned long) attr, 0);
200 printf("sched_setattr(0, {size=%u} => {size=%u}, 0) = %s\n",
201 SCHED_ATTR_MIN_SIZE - 1, attr->size, errstr);
202
203 attr->size = 0x90807060;
204 attr->sched_policy = 0xca7faced;
205 attr->sched_flags = 0xbadc0ded1057da7aULL;
206 attr->sched_nice = 0xafbfcfdf;
207 attr->sched_priority = 0xb8c8d8e8;
208 attr->sched_runtime = 0xbadcaffedeadf157ULL;
209 attr->sched_deadline = 0xc0de70a57badac75ULL;
210 attr->sched_period = 0xded1ca7edda7aca7ULL;
211
212 sys_sched_setattr(bogus_pid, (unsigned long) attr, bogus_flags);
213 printf("sched_setattr(%d, {size=%u, sched_policy=%#x /* SCHED_??? */, "
214 "sched_flags=%#" PRIx64 " /* SCHED_FLAG_??? */, "
215 "sched_nice=%d, sched_priority=%u, sched_runtime=%" PRIu64 ", "
216 "sched_deadline=%" PRIu64 ", sched_period=%" PRIu64 ", ...}, %u)"
217 " = %s\n",
218 (int) bogus_pid,
219 attr->size,
220 attr->sched_policy,
221 attr->sched_flags,
222 attr->sched_nice,
223 attr->sched_priority,
224 attr->sched_runtime,
225 attr->sched_deadline,
226 attr->sched_period,
227 (unsigned) bogus_flags, errstr);
228
229 if (F8ILL_KULONG_SUPPORTED) {
230 const kernel_ulong_t ill = f8ill_ptr_to_kulong(attr);
231
232 sys_sched_getattr(0, ill, sizeof(*attr), 0);
233 printf("sched_getattr(0, %#llx, %u, 0) = %s\n",
234 (unsigned long long) ill, (unsigned) sizeof(*attr),
235 errstr);
236
237 sys_sched_setattr(0, ill, 0);
238 printf("sched_setattr(0, %#llx, 0) = %s\n",
239 (unsigned long long) ill, errstr);
240 }
241
242 puts("+++ exited with 0 +++");
243 return 0;
244 }
245
246 #else
247
248 SKIP_MAIN_UNDEFINED("__NR_sched_getattr && __NR_sched_setattr")
249
250 #endif
251