1/*
2 * Check decoding of set_mempolicy syscall.
3 *
4 * Copyright (c) 2016 Dmitry V. Levin <ldv@altlinux.org>
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 * 3. The name of the author may not be used to endorse or promote products
16 *    derived from this software without specific prior written permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
19 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
20 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
21 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
22 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
23 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
27 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 */
29
30#include "tests.h"
31#include <asm/unistd.h>
32
33#ifdef __NR_set_mempolicy
34
35# include <errno.h>
36# include <stdio.h>
37# include <string.h>
38# include <unistd.h>
39
40# include "xlat.h"
41# include "xlat/policies.h"
42
43# define MAX_STRLEN 3
44# define NLONGS(n) ((n + 8 * sizeof(long) - 2) \
45		      / (8 * sizeof(long)))
46
47static void
48print_nodes(const unsigned long maxnode, unsigned int offset)
49{
50	unsigned int nlongs = NLONGS(maxnode);
51	if (nlongs <= offset)
52		nlongs = 0;
53	else
54		nlongs -= offset;
55	const unsigned int size = nlongs * sizeof(long);
56	unsigned long *const nodemask =
57		tail_alloc(size ? size : (offset ? 1 : 0));
58	memset(nodemask, 0, size);
59
60	long rc = syscall(__NR_set_mempolicy, 0, nodemask, maxnode);
61	const char *errstr = sprintrc(rc);
62
63	fputs("set_mempolicy(MPOL_DEFAULT, ", stdout);
64
65	if (nlongs) {
66		putc('[', stdout);
67		unsigned int i;
68		for (i = 0; i < nlongs + offset; ++i) {
69			if (i)
70				fputs(", ", stdout);
71			if (i < nlongs) {
72				if (i >= MAX_STRLEN) {
73					fputs("...", stdout);
74					break;
75				}
76				printf("%#0*lx", (int) sizeof(long) * 2 + 2,
77				       nodemask[i]);
78			} else {
79				printf("%p", nodemask + i);
80				break;
81			}
82		}
83		putc(']', stdout);
84	} else {
85		if (maxnode)
86			printf("%p", nodemask);
87		else
88			printf("[]");
89	}
90
91	printf(", %lu) = %s\n", maxnode, errstr);
92}
93
94static void
95test_offset(const unsigned int offset)
96{
97	unsigned long maxnode = get_page_size() * 8;
98
99	print_nodes(maxnode, offset);
100	print_nodes(maxnode + 1, offset);
101	print_nodes(maxnode + 2, offset);
102
103	maxnode = sizeof(long) * 8;
104	print_nodes(0, offset);
105	print_nodes(1, offset);
106	print_nodes(2, offset);
107	print_nodes(maxnode - 1, offset);
108	print_nodes(maxnode    , offset);
109	print_nodes(maxnode + 1, offset);
110	print_nodes(maxnode + 2, offset);
111	print_nodes(maxnode * 2 - 1, offset);
112	print_nodes(maxnode * 2    , offset);
113	print_nodes(maxnode * 2 + 1, offset);
114	print_nodes(maxnode * 2 + 2, offset);
115	print_nodes(maxnode * 3 - 1, offset);
116	print_nodes(maxnode * 3    , offset);
117	print_nodes(maxnode * 3 + 1, offset);
118	print_nodes(maxnode * 3 + 2, offset);
119	print_nodes(maxnode * 4 + 2, offset);
120}
121
122int
123main(void)
124{
125	if (syscall(__NR_set_mempolicy, 0, 0, 0))
126		perror_msg_and_skip("set_mempolicy");
127	puts("set_mempolicy(MPOL_DEFAULT, NULL, 0) = 0");
128
129	const unsigned long *nodemask = (void *) 0xfacefeedfffffffeULL;
130	const unsigned long maxnode = (unsigned long) 0xcafef00dbadc0dedULL;
131	long rc = syscall(__NR_set_mempolicy, 1, nodemask, maxnode);
132	printf("set_mempolicy(MPOL_PREFERRED, %p, %lu) = %s\n",
133	       nodemask, maxnode, sprintrc(rc));
134
135	test_offset(0);
136	test_offset(1);
137
138	puts("+++ exited with 0 +++");
139	return 0;
140}
141
142#else
143
144SKIP_MAIN_UNDEFINED("__NR_set_mempolicy")
145
146#endif
147