xref: /linux/kernel/sysctl-test.c (revision 97733180fafbeb7cc3fd1c8be60d05980615f5d6)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * KUnit test of proc sysctl.
4  */
5 
6 #include <kunit/test.h>
7 #include <linux/sysctl.h>
8 
9 #define KUNIT_PROC_READ 0
10 #define KUNIT_PROC_WRITE 1
11 
12 static int i_zero;
13 static int i_one_hundred = 100;
14 
15 /*
16  * Test that proc_dointvec will not try to use a NULL .data field even when the
17  * length is non-zero.
18  */
19 static void sysctl_test_api_dointvec_null_tbl_data(struct kunit *test)
20 {
21 	struct ctl_table null_data_table = {
22 		.procname = "foo",
23 		/*
24 		 * Here we are testing that proc_dointvec behaves correctly when
25 		 * we give it a NULL .data field. Normally this would point to a
26 		 * piece of memory where the value would be stored.
27 		 */
28 		.data		= NULL,
29 		.maxlen		= sizeof(int),
30 		.mode		= 0644,
31 		.proc_handler	= proc_dointvec,
32 		.extra1		= &i_zero,
33 		.extra2         = &i_one_hundred,
34 	};
35 	/*
36 	 * proc_dointvec expects a buffer in user space, so we allocate one. We
37 	 * also need to cast it to __user so sparse doesn't get mad.
38 	 */
39 	void __user *buffer = (void __user *)kunit_kzalloc(test, sizeof(int),
40 							   GFP_USER);
41 	size_t len;
42 	loff_t pos;
43 
44 	/*
45 	 * We don't care what the starting length is since proc_dointvec should
46 	 * not try to read because .data is NULL.
47 	 */
48 	len = 1234;
49 	KUNIT_EXPECT_EQ(test, 0, proc_dointvec(&null_data_table,
50 					       KUNIT_PROC_READ, buffer, &len,
51 					       &pos));
52 	KUNIT_EXPECT_EQ(test, 0, len);
53 
54 	/*
55 	 * See above.
56 	 */
57 	len = 1234;
58 	KUNIT_EXPECT_EQ(test, 0, proc_dointvec(&null_data_table,
59 					       KUNIT_PROC_WRITE, buffer, &len,
60 					       &pos));
61 	KUNIT_EXPECT_EQ(test, 0, len);
62 }
63 
64 /*
65  * Similar to the previous test, we create a struct ctrl_table that has a .data
66  * field that proc_dointvec cannot do anything with; however, this time it is
67  * because we tell proc_dointvec that the size is 0.
68  */
69 static void sysctl_test_api_dointvec_table_maxlen_unset(struct kunit *test)
70 {
71 	int data = 0;
72 	struct ctl_table data_maxlen_unset_table = {
73 		.procname = "foo",
74 		.data		= &data,
75 		/*
76 		 * So .data is no longer NULL, but we tell proc_dointvec its
77 		 * length is 0, so it still shouldn't try to use it.
78 		 */
79 		.maxlen		= 0,
80 		.mode		= 0644,
81 		.proc_handler	= proc_dointvec,
82 		.extra1		= &i_zero,
83 		.extra2         = &i_one_hundred,
84 	};
85 	void __user *buffer = (void __user *)kunit_kzalloc(test, sizeof(int),
86 							   GFP_USER);
87 	size_t len;
88 	loff_t pos;
89 
90 	/*
91 	 * As before, we don't care what buffer length is because proc_dointvec
92 	 * cannot do anything because its internal .data buffer has zero length.
93 	 */
94 	len = 1234;
95 	KUNIT_EXPECT_EQ(test, 0, proc_dointvec(&data_maxlen_unset_table,
96 					       KUNIT_PROC_READ, buffer, &len,
97 					       &pos));
98 	KUNIT_EXPECT_EQ(test, 0, len);
99 
100 	/*
101 	 * See previous comment.
102 	 */
103 	len = 1234;
104 	KUNIT_EXPECT_EQ(test, 0, proc_dointvec(&data_maxlen_unset_table,
105 					       KUNIT_PROC_WRITE, buffer, &len,
106 					       &pos));
107 	KUNIT_EXPECT_EQ(test, 0, len);
108 }
109 
110 /*
111  * Here we provide a valid struct ctl_table, but we try to read and write from
112  * it using a buffer of zero length, so it should still fail in a similar way as
113  * before.
114  */
115 static void sysctl_test_api_dointvec_table_len_is_zero(struct kunit *test)
116 {
117 	int data = 0;
118 	/* Good table. */
119 	struct ctl_table table = {
120 		.procname = "foo",
121 		.data		= &data,
122 		.maxlen		= sizeof(int),
123 		.mode		= 0644,
124 		.proc_handler	= proc_dointvec,
125 		.extra1		= &i_zero,
126 		.extra2         = &i_one_hundred,
127 	};
128 	void __user *buffer = (void __user *)kunit_kzalloc(test, sizeof(int),
129 							   GFP_USER);
130 	/*
131 	 * However, now our read/write buffer has zero length.
132 	 */
133 	size_t len = 0;
134 	loff_t pos;
135 
136 	KUNIT_EXPECT_EQ(test, 0, proc_dointvec(&table, KUNIT_PROC_READ, buffer,
137 					       &len, &pos));
138 	KUNIT_EXPECT_EQ(test, 0, len);
139 
140 	KUNIT_EXPECT_EQ(test, 0, proc_dointvec(&table, KUNIT_PROC_WRITE, buffer,
141 					       &len, &pos));
142 	KUNIT_EXPECT_EQ(test, 0, len);
143 }
144 
145 /*
146  * Test that proc_dointvec refuses to read when the file position is non-zero.
147  */
148 static void sysctl_test_api_dointvec_table_read_but_position_set(
149 		struct kunit *test)
150 {
151 	int data = 0;
152 	/* Good table. */
153 	struct ctl_table table = {
154 		.procname = "foo",
155 		.data		= &data,
156 		.maxlen		= sizeof(int),
157 		.mode		= 0644,
158 		.proc_handler	= proc_dointvec,
159 		.extra1		= &i_zero,
160 		.extra2         = &i_one_hundred,
161 	};
162 	void __user *buffer = (void __user *)kunit_kzalloc(test, sizeof(int),
163 							   GFP_USER);
164 	/*
165 	 * We don't care about our buffer length because we start off with a
166 	 * non-zero file position.
167 	 */
168 	size_t len = 1234;
169 	/*
170 	 * proc_dointvec should refuse to read into the buffer since the file
171 	 * pos is non-zero.
172 	 */
173 	loff_t pos = 1;
174 
175 	KUNIT_EXPECT_EQ(test, 0, proc_dointvec(&table, KUNIT_PROC_READ, buffer,
176 					       &len, &pos));
177 	KUNIT_EXPECT_EQ(test, 0, len);
178 }
179 
180 /*
181  * Test that we can read a two digit number in a sufficiently size buffer.
182  * Nothing fancy.
183  */
184 static void sysctl_test_dointvec_read_happy_single_positive(struct kunit *test)
185 {
186 	int data = 0;
187 	/* Good table. */
188 	struct ctl_table table = {
189 		.procname = "foo",
190 		.data		= &data,
191 		.maxlen		= sizeof(int),
192 		.mode		= 0644,
193 		.proc_handler	= proc_dointvec,
194 		.extra1		= &i_zero,
195 		.extra2         = &i_one_hundred,
196 	};
197 	size_t len = 4;
198 	loff_t pos = 0;
199 	char *buffer = kunit_kzalloc(test, len, GFP_USER);
200 	char __user *user_buffer = (char __user *)buffer;
201 	/* Store 13 in the data field. */
202 	*((int *)table.data) = 13;
203 
204 	KUNIT_EXPECT_EQ(test, 0, proc_dointvec(&table, KUNIT_PROC_READ,
205 					       user_buffer, &len, &pos));
206 	KUNIT_ASSERT_EQ(test, 3, len);
207 	buffer[len] = '\0';
208 	/* And we read 13 back out. */
209 	KUNIT_EXPECT_STREQ(test, "13\n", buffer);
210 }
211 
212 /*
213  * Same as previous test, just now with negative numbers.
214  */
215 static void sysctl_test_dointvec_read_happy_single_negative(struct kunit *test)
216 {
217 	int data = 0;
218 	/* Good table. */
219 	struct ctl_table table = {
220 		.procname = "foo",
221 		.data		= &data,
222 		.maxlen		= sizeof(int),
223 		.mode		= 0644,
224 		.proc_handler	= proc_dointvec,
225 		.extra1		= &i_zero,
226 		.extra2         = &i_one_hundred,
227 	};
228 	size_t len = 5;
229 	loff_t pos = 0;
230 	char *buffer = kunit_kzalloc(test, len, GFP_USER);
231 	char __user *user_buffer = (char __user *)buffer;
232 	*((int *)table.data) = -16;
233 
234 	KUNIT_EXPECT_EQ(test, 0, proc_dointvec(&table, KUNIT_PROC_READ,
235 					       user_buffer, &len, &pos));
236 	KUNIT_ASSERT_EQ(test, 4, len);
237 	buffer[len] = '\0';
238 	KUNIT_EXPECT_STREQ(test, "-16\n", buffer);
239 }
240 
241 /*
242  * Test that a simple positive write works.
243  */
244 static void sysctl_test_dointvec_write_happy_single_positive(struct kunit *test)
245 {
246 	int data = 0;
247 	/* Good table. */
248 	struct ctl_table table = {
249 		.procname = "foo",
250 		.data		= &data,
251 		.maxlen		= sizeof(int),
252 		.mode		= 0644,
253 		.proc_handler	= proc_dointvec,
254 		.extra1		= &i_zero,
255 		.extra2         = &i_one_hundred,
256 	};
257 	char input[] = "9";
258 	size_t len = sizeof(input) - 1;
259 	loff_t pos = 0;
260 	char *buffer = kunit_kzalloc(test, len, GFP_USER);
261 	char __user *user_buffer = (char __user *)buffer;
262 
263 	memcpy(buffer, input, len);
264 
265 	KUNIT_EXPECT_EQ(test, 0, proc_dointvec(&table, KUNIT_PROC_WRITE,
266 					       user_buffer, &len, &pos));
267 	KUNIT_EXPECT_EQ(test, sizeof(input) - 1, len);
268 	KUNIT_EXPECT_EQ(test, sizeof(input) - 1, pos);
269 	KUNIT_EXPECT_EQ(test, 9, *((int *)table.data));
270 }
271 
272 /*
273  * Same as previous test, but now with negative numbers.
274  */
275 static void sysctl_test_dointvec_write_happy_single_negative(struct kunit *test)
276 {
277 	int data = 0;
278 	struct ctl_table table = {
279 		.procname = "foo",
280 		.data		= &data,
281 		.maxlen		= sizeof(int),
282 		.mode		= 0644,
283 		.proc_handler	= proc_dointvec,
284 		.extra1		= &i_zero,
285 		.extra2         = &i_one_hundred,
286 	};
287 	char input[] = "-9";
288 	size_t len = sizeof(input) - 1;
289 	loff_t pos = 0;
290 	char *buffer = kunit_kzalloc(test, len, GFP_USER);
291 	char __user *user_buffer = (char __user *)buffer;
292 
293 	memcpy(buffer, input, len);
294 
295 	KUNIT_EXPECT_EQ(test, 0, proc_dointvec(&table, KUNIT_PROC_WRITE,
296 					       user_buffer, &len, &pos));
297 	KUNIT_EXPECT_EQ(test, sizeof(input) - 1, len);
298 	KUNIT_EXPECT_EQ(test, sizeof(input) - 1, pos);
299 	KUNIT_EXPECT_EQ(test, -9, *((int *)table.data));
300 }
301 
302 /*
303  * Test that writing a value smaller than the minimum possible value is not
304  * allowed.
305  */
306 static void sysctl_test_api_dointvec_write_single_less_int_min(
307 		struct kunit *test)
308 {
309 	int data = 0;
310 	struct ctl_table table = {
311 		.procname = "foo",
312 		.data		= &data,
313 		.maxlen		= sizeof(int),
314 		.mode		= 0644,
315 		.proc_handler	= proc_dointvec,
316 		.extra1		= &i_zero,
317 		.extra2         = &i_one_hundred,
318 	};
319 	size_t max_len = 32, len = max_len;
320 	loff_t pos = 0;
321 	char *buffer = kunit_kzalloc(test, max_len, GFP_USER);
322 	char __user *user_buffer = (char __user *)buffer;
323 	unsigned long abs_of_less_than_min = (unsigned long)INT_MAX
324 					     - (INT_MAX + INT_MIN) + 1;
325 
326 	/*
327 	 * We use this rigmarole to create a string that contains a value one
328 	 * less than the minimum accepted value.
329 	 */
330 	KUNIT_ASSERT_LT(test,
331 			(size_t)snprintf(buffer, max_len, "-%lu",
332 					 abs_of_less_than_min),
333 			max_len);
334 
335 	KUNIT_EXPECT_EQ(test, -EINVAL, proc_dointvec(&table, KUNIT_PROC_WRITE,
336 						     user_buffer, &len, &pos));
337 	KUNIT_EXPECT_EQ(test, max_len, len);
338 	KUNIT_EXPECT_EQ(test, 0, *((int *)table.data));
339 }
340 
341 /*
342  * Test that writing the maximum possible value works.
343  */
344 static void sysctl_test_api_dointvec_write_single_greater_int_max(
345 		struct kunit *test)
346 {
347 	int data = 0;
348 	struct ctl_table table = {
349 		.procname = "foo",
350 		.data		= &data,
351 		.maxlen		= sizeof(int),
352 		.mode		= 0644,
353 		.proc_handler	= proc_dointvec,
354 		.extra1		= &i_zero,
355 		.extra2         = &i_one_hundred,
356 	};
357 	size_t max_len = 32, len = max_len;
358 	loff_t pos = 0;
359 	char *buffer = kunit_kzalloc(test, max_len, GFP_USER);
360 	char __user *user_buffer = (char __user *)buffer;
361 	unsigned long greater_than_max = (unsigned long)INT_MAX + 1;
362 
363 	KUNIT_ASSERT_GT(test, greater_than_max, (unsigned long)INT_MAX);
364 	KUNIT_ASSERT_LT(test, (size_t)snprintf(buffer, max_len, "%lu",
365 					       greater_than_max),
366 			max_len);
367 	KUNIT_EXPECT_EQ(test, -EINVAL, proc_dointvec(&table, KUNIT_PROC_WRITE,
368 						     user_buffer, &len, &pos));
369 	KUNIT_ASSERT_EQ(test, max_len, len);
370 	KUNIT_EXPECT_EQ(test, 0, *((int *)table.data));
371 }
372 
373 static struct kunit_case sysctl_test_cases[] = {
374 	KUNIT_CASE(sysctl_test_api_dointvec_null_tbl_data),
375 	KUNIT_CASE(sysctl_test_api_dointvec_table_maxlen_unset),
376 	KUNIT_CASE(sysctl_test_api_dointvec_table_len_is_zero),
377 	KUNIT_CASE(sysctl_test_api_dointvec_table_read_but_position_set),
378 	KUNIT_CASE(sysctl_test_dointvec_read_happy_single_positive),
379 	KUNIT_CASE(sysctl_test_dointvec_read_happy_single_negative),
380 	KUNIT_CASE(sysctl_test_dointvec_write_happy_single_positive),
381 	KUNIT_CASE(sysctl_test_dointvec_write_happy_single_negative),
382 	KUNIT_CASE(sysctl_test_api_dointvec_write_single_less_int_min),
383 	KUNIT_CASE(sysctl_test_api_dointvec_write_single_greater_int_max),
384 	{}
385 };
386 
387 static struct kunit_suite sysctl_test_suite = {
388 	.name = "sysctl_test",
389 	.test_cases = sysctl_test_cases,
390 };
391 
392 kunit_test_suites(&sysctl_test_suite);
393 
394 MODULE_LICENSE("GPL v2");
395