1/*
2 * Copyright (c) 2015 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#ifdef HAVE_CONFIG_H
29# include "config.h"
30#endif
31
32#include <stdio.h>
33#include <inttypes.h>
34#include <unistd.h>
35#include <sys/syscall.h>
36
37#if defined __NR_sched_getattr && defined __NR_sched_setattr
38
39int
40main(void)
41{
42	static union {
43		struct {
44			uint32_t size;
45			uint32_t sched_policy;
46			uint64_t sched_flags;
47			uint32_t sched_nice;
48			uint32_t sched_priority;
49			uint64_t sched_runtime;
50			uint64_t sched_deadline;
51			uint64_t sched_period;
52		} attr;
53		char buf[256];
54	} sched;
55
56	if (syscall(__NR_sched_getattr, 0, &sched, sizeof(sched), 0))
57		return 77;
58
59	printf("sched_getattr\\(0, \\{size=%u, sched_policy=SCHED_[A-Z]+, sched_flags=%s, sched_nice=%u, sched_priority=%u, sched_runtime=%" PRIu64 ", sched_deadline=%" PRIu64 ", sched_period=%" PRIu64 "\\}, 256, 0\\) += 0\n",
60		sched.attr.size,
61		sched.attr.sched_flags ? "SCHED_FLAG_RESET_ON_FORK" : "0",
62		sched.attr.sched_nice,
63		sched.attr.sched_priority,
64		sched.attr.sched_runtime,
65		sched.attr.sched_deadline,
66		sched.attr.sched_period);
67
68	sched.attr.sched_flags |= 1;
69	if (syscall(__NR_sched_setattr, 0, &sched, 0))
70		return 77;
71
72	printf("sched_setattr\\(0, \\{size=%u, sched_policy=SCHED_[A-Z]+, sched_flags=%s, sched_nice=%u, sched_priority=%u, sched_runtime=%" PRIu64 ", sched_deadline=%" PRIu64 ", sched_period=%" PRIu64 "\\}, 0\\) += 0\n",
73		sched.attr.size,
74		"SCHED_FLAG_RESET_ON_FORK",
75		sched.attr.sched_nice,
76		sched.attr.sched_priority,
77		sched.attr.sched_runtime,
78		sched.attr.sched_deadline,
79		sched.attr.sched_period);
80
81	return 0;
82}
83
84#else
85
86int
87main(void)
88{
89	return 77;
90}
91
92#endif
93