xref: /linux/tools/testing/selftests/proc/proc-loadavg-001.c (revision 0526b56cbc3c489642bd6a5fe4b718dea7ef0ee8)
1 /*
2  * Copyright © 2018 Alexey Dobriyan <adobriyan@gmail.com>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 /* Test that /proc/loadavg correctly reports last pid in pid namespace. */
17 #include <errno.h>
18 #include <sched.h>
19 #include <sys/types.h>
20 #include <sys/stat.h>
21 #include <fcntl.h>
22 #include <unistd.h>
23 #include <sys/wait.h>
24 
25 int main(void)
26 {
27 	pid_t pid;
28 	int wstatus;
29 
30 	if (unshare(CLONE_NEWPID) == -1) {
31 		if (errno == ENOSYS || errno == EPERM)
32 			return 4;
33 		return 1;
34 	}
35 
36 	pid = fork();
37 	if (pid == -1)
38 		return 1;
39 	if (pid == 0) {
40 		char buf[128], *p;
41 		int fd;
42 		ssize_t rv;
43 
44 		fd = open("/proc/loadavg" , O_RDONLY);
45 		if (fd == -1)
46 			return 1;
47 		rv = read(fd, buf, sizeof(buf));
48 		if (rv < 3)
49 			return 1;
50 		p = buf + rv;
51 
52 		/* pid 1 */
53 		if (!(p[-3] == ' ' && p[-2] == '1' && p[-1] == '\n'))
54 			return 1;
55 
56 		pid = fork();
57 		if (pid == -1)
58 			return 1;
59 		if (pid == 0)
60 			return 0;
61 		if (waitpid(pid, NULL, 0) == -1)
62 			return 1;
63 
64 		lseek(fd, 0, SEEK_SET);
65 		rv = read(fd, buf, sizeof(buf));
66 		if (rv < 3)
67 			return 1;
68 		p = buf + rv;
69 
70 		/* pid 2 */
71 		if (!(p[-3] == ' ' && p[-2] == '2' && p[-1] == '\n'))
72 			return 1;
73 
74 		return 0;
75 	}
76 
77 	if (waitpid(pid, &wstatus, 0) == -1)
78 		return 1;
79 	if (WIFEXITED(wstatus) && WEXITSTATUS(wstatus) == 0)
80 		return 0;
81 	return 1;
82 }
83