1 /*
2  * Copyright (C) 2012 Linux Test Project, Inc.
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of version 2 of the GNU General Public
6  * License as published by the Free Software Foundation.
7  *
8  * This program is distributed in the hope that it would be useful,
9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
11  *
12  * Further, this software is distributed without any warranty that it
13  * is free of the rightful claim of any third person regarding
14  * infringement or the like.  Any license provided herein, whether
15  * implied or otherwise, applies only to this software file.  Patent
16  * licenses, if any, provided herein do not apply to combinations of
17  * this program with other software, or any other product whatsoever.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
22  * 02110-1301, USA.
23  */
24 
25 /*
26  * errno tests for migrate_pages() syscall
27  */
28 #include <sys/types.h>
29 #include <sys/syscall.h>
30 #include <sys/wait.h>
31 #include <sys/mman.h>
32 #include <errno.h>
33 #include <stdio.h>
34 #include <stdlib.h>
35 #include <string.h>
36 #include <unistd.h>
37 #include <pwd.h>
38 
39 #include "test.h"
40 #include "safe_macros.h"
41 #include "lapi/syscalls.h"
42 #include "numa_helper.h"
43 #include "migrate_pages_common.h"
44 
45 char *TCID = "migrate_pages01";
46 int TST_TOTAL = 1;
47 
48 option_t options[] = {
49 	{NULL, NULL, NULL}
50 };
51 
52 #ifdef HAVE_NUMA_V2
53 
54 static unsigned long *sane_old_nodes;
55 static unsigned long *sane_new_nodes;
56 static int sane_nodemask_size;
57 static int sane_max_node;
58 
59 static void setup(void);
60 static void cleanup(void);
61 
test_sane_nodes(void)62 static void test_sane_nodes(void)
63 {
64 	tst_resm(TINFO, "test_empty_mask");
65 	TEST(ltp_syscall(__NR_migrate_pages, 0, sane_max_node,
66 		     sane_old_nodes, sane_new_nodes));
67 	check_ret(0);
68 }
69 
test_invalid_pid(void)70 static void test_invalid_pid(void)
71 {
72 	pid_t invalid_pid = -1;
73 
74 	tst_resm(TINFO, "test_invalid_pid -1");
75 	TEST(ltp_syscall(__NR_migrate_pages, invalid_pid, sane_max_node,
76 		     sane_old_nodes, sane_new_nodes));
77 	check_ret(-1);
78 	check_errno(ESRCH);
79 
80 	tst_resm(TINFO, "test_invalid_pid unused pid");
81 	invalid_pid = tst_get_unused_pid(cleanup);
82 	TEST(ltp_syscall(__NR_migrate_pages, invalid_pid, sane_max_node,
83 		     sane_old_nodes, sane_new_nodes));
84 	check_ret(-1);
85 	check_errno(ESRCH);
86 }
87 
test_invalid_masksize(void)88 static void test_invalid_masksize(void)
89 {
90 	tst_resm(TINFO, "test_invalid_masksize");
91 	TEST(ltp_syscall(__NR_migrate_pages, 0, -1, sane_old_nodes,
92 		     sane_new_nodes));
93 	check_ret(-1);
94 	check_errno(EINVAL);
95 }
96 
test_invalid_mem(void)97 static void test_invalid_mem(void)
98 {
99 	unsigned long *p;
100 
101 	tst_resm(TINFO, "test_invalid_mem -1");
102 	TEST(ltp_syscall(__NR_migrate_pages, 0, sane_max_node, -1, -1));
103 	check_ret(-1);
104 	check_errno(EFAULT);
105 
106 	tst_resm(TINFO, "test_invalid_mem invalid prot");
107 	p = mmap(NULL, getpagesize(), PROT_NONE,
108 		 MAP_PRIVATE | MAP_ANONYMOUS, 0, 0);
109 	if (p == MAP_FAILED)
110 		tst_brkm(TBROK | TERRNO, cleanup, "mmap");
111 	TEST(ltp_syscall(__NR_migrate_pages, 0, sane_max_node, p, p));
112 	check_ret(-1);
113 	check_errno(EFAULT);
114 
115 	SAFE_MUNMAP(cleanup, p, getpagesize());
116 	tst_resm(TINFO, "test_invalid_mem unmmaped");
117 	TEST(ltp_syscall(__NR_migrate_pages, 0, sane_max_node, p, p));
118 	check_ret(-1);
119 	check_errno(EFAULT);
120 }
121 
test_invalid_nodes(void)122 static void test_invalid_nodes(void)
123 {
124 	int *nodes;
125 	int num_nodes, ret, i;
126 	int invalid_node = 0;
127 	unsigned long *old_nodes, *new_nodes;
128 
129 	tst_resm(TINFO, "test_invalid_nodes");
130 	ret = get_allowed_nodes_arr(NH_MEMS, &num_nodes, &nodes);
131 	if (ret < 0)
132 		tst_brkm(TBROK | TERRNO, cleanup,
133 			 "get_allowed_nodes_arr: %d", ret);
134 
135 	/* get first node which is not in nodes */
136 	for (i = 0; i < num_nodes; i++, invalid_node++)
137 		if (invalid_node != nodes[i])
138 			break;
139 	if (invalid_node < sane_max_node) {
140 		old_nodes = SAFE_MALLOC(NULL, sane_nodemask_size);
141 		new_nodes = SAFE_MALLOC(NULL, sane_nodemask_size);
142 		memcpy(old_nodes, sane_old_nodes, sane_nodemask_size);
143 		memset(new_nodes, 0, sane_nodemask_size);
144 		set_bit(new_nodes, invalid_node, 1);
145 
146 		TEST(ltp_syscall(__NR_migrate_pages, 0, sane_max_node,
147 			     old_nodes, new_nodes));
148 		check_ret(-1);
149 		check_errno(EINVAL);
150 		free(old_nodes);
151 		free(new_nodes);
152 	} else {
153 		tst_resm(TCONF, "All possible nodes are present");
154 	}
155 
156 	free(nodes);
157 }
158 
test_invalid_perm(void)159 static void test_invalid_perm(void)
160 {
161 	char nobody_uid[] = "nobody";
162 	struct passwd *ltpuser;
163 	int status;
164 	pid_t child_pid;
165 	pid_t parent_pid;
166 	int ret = 0;
167 
168 	tst_resm(TINFO, "test_invalid_perm");
169 	parent_pid = getpid();
170 	fflush(stdout);
171 	child_pid = fork();
172 	switch (child_pid) {
173 	case -1:
174 		tst_brkm(TBROK | TERRNO, cleanup, "fork");
175 		break;
176 	case 0:
177 		ltpuser = getpwnam(nobody_uid);
178 		if (ltpuser == NULL)
179 			tst_brkm(TBROK | TERRNO, NULL, "getpwnam failed");
180 		SAFE_SETUID(NULL, ltpuser->pw_uid);
181 		TEST(ltp_syscall(__NR_migrate_pages, parent_pid,
182 			     sane_max_node, sane_old_nodes, sane_new_nodes));
183 		ret |= check_ret(-1);
184 		ret |= check_errno(EPERM);
185 		exit(ret);
186 	default:
187 		SAFE_WAITPID(cleanup, child_pid, &status, 0);
188 		if (!WIFEXITED(status) || WEXITSTATUS(status) != 0)
189 			tst_resm(TFAIL, "child returns %d", status);
190 	}
191 }
192 
main(int argc,char * argv[])193 int main(int argc, char *argv[])
194 {
195 	int lc;
196 
197 	tst_parse_opts(argc, argv, options, NULL);
198 
199 	setup();
200 	for (lc = 0; TEST_LOOPING(lc); lc++) {
201 		tst_count = 0;
202 		test_sane_nodes();
203 		test_invalid_pid();
204 		test_invalid_masksize();
205 		test_invalid_mem();
206 		test_invalid_nodes();
207 		test_invalid_perm();
208 	}
209 	cleanup();
210 	tst_exit();
211 }
212 
setup(void)213 static void setup(void)
214 {
215 	int node, ret;
216 
217 	tst_require_root();
218 	TEST(ltp_syscall(__NR_migrate_pages, 0, 0, NULL, NULL));
219 
220 	if (!is_numa(NULL, NH_MEMS, 1))
221 		tst_brkm(TCONF, NULL, "requires NUMA with at least 1 node");
222 
223 	ret = get_allowed_nodes(NH_MEMS, 1, &node);
224 	if (ret < 0)
225 		tst_brkm(TBROK | TERRNO, NULL, "get_allowed_nodes_arr: %d",
226 			 ret);
227 
228 	sane_max_node = LTP_ALIGN(get_max_node(), sizeof(unsigned long)*8);
229 	sane_nodemask_size = sane_max_node / 8;
230 	sane_old_nodes = SAFE_MALLOC(NULL, sane_nodemask_size);
231 	sane_new_nodes = SAFE_MALLOC(NULL, sane_nodemask_size);
232 	memset(sane_old_nodes, 0, sane_nodemask_size);
233 	memset(sane_new_nodes, 0, sane_nodemask_size);
234 
235 	set_bit(sane_old_nodes, node, 1);
236 	set_bit(sane_new_nodes, node, 1);
237 
238 	TEST_PAUSE;
239 }
240 
cleanup(void)241 static void cleanup(void)
242 {
243 	free(sane_old_nodes);
244 	free(sane_new_nodes);
245 }
246 
247 #else
main(void)248 int main(void)
249 {
250 	tst_brkm(TCONF, NULL, NUMA_ERROR_MSG);
251 }
252 #endif
253