Loading...
1/*
2 * Copyright (C) 2016 Google, Inc.
3 *
4 * This software is licensed under the terms of the GNU General Public
5 * License version 2, as published by the Free Software Foundation, and
6 * may be copied, distributed, and modified under those terms.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
12 *
13 */
14
15#define _GNU_SOURCE
16
17#include <errno.h>
18#include <fcntl.h>
19#include <sched.h>
20#include <signal.h>
21#include <stdbool.h>
22#include <stdio.h>
23#include <string.h>
24#include <unistd.h>
25#include <sys/ptrace.h>
26#include <sys/stat.h>
27#include <sys/timerfd.h>
28#include <sys/types.h>
29#include <sys/wait.h>
30
31#include "../kselftest.h"
32
33void child(int cpu)
34{
35 cpu_set_t set;
36
37 CPU_ZERO(&set);
38 CPU_SET(cpu, &set);
39 if (sched_setaffinity(0, sizeof(set), &set) != 0) {
40 perror("sched_setaffinity() failed");
41 _exit(1);
42 }
43
44 if (ptrace(PTRACE_TRACEME, 0, NULL, NULL) != 0) {
45 perror("ptrace(PTRACE_TRACEME) failed");
46 _exit(1);
47 }
48
49 if (raise(SIGSTOP) != 0) {
50 perror("raise(SIGSTOP) failed");
51 _exit(1);
52 }
53
54 _exit(0);
55}
56
57bool run_test(int cpu)
58{
59 int status;
60 pid_t pid = fork();
61 pid_t wpid;
62
63 if (pid < 0) {
64 perror("fork() failed");
65 return false;
66 }
67 if (pid == 0)
68 child(cpu);
69
70 wpid = waitpid(pid, &status, __WALL);
71 if (wpid != pid) {
72 perror("waitpid() failed");
73 return false;
74 }
75 if (!WIFSTOPPED(status)) {
76 printf("child did not stop\n");
77 return false;
78 }
79 if (WSTOPSIG(status) != SIGSTOP) {
80 printf("child did not stop with SIGSTOP\n");
81 return false;
82 }
83
84 if (ptrace(PTRACE_SINGLESTEP, pid, NULL, NULL) < 0) {
85 if (errno == EIO) {
86 printf("ptrace(PTRACE_SINGLESTEP) not supported on this architecture\n");
87 ksft_exit_skip();
88 }
89 perror("ptrace(PTRACE_SINGLESTEP) failed");
90 return false;
91 }
92
93 wpid = waitpid(pid, &status, __WALL);
94 if (wpid != pid) {
95 perror("waitpid() failed");
96 return false;
97 }
98 if (WIFEXITED(status)) {
99 printf("child did not single-step\n");
100 return false;
101 }
102 if (!WIFSTOPPED(status)) {
103 printf("child did not stop\n");
104 return false;
105 }
106 if (WSTOPSIG(status) != SIGTRAP) {
107 printf("child did not stop with SIGTRAP\n");
108 return false;
109 }
110
111 if (ptrace(PTRACE_CONT, pid, NULL, NULL) < 0) {
112 perror("ptrace(PTRACE_CONT) failed");
113 return false;
114 }
115
116 wpid = waitpid(pid, &status, __WALL);
117 if (wpid != pid) {
118 perror("waitpid() failed");
119 return false;
120 }
121 if (!WIFEXITED(status)) {
122 printf("child did not exit after PTRACE_CONT\n");
123 return false;
124 }
125
126 return true;
127}
128
129void suspend(void)
130{
131 int power_state_fd;
132 struct sigevent event = {};
133 int timerfd;
134 int err;
135 struct itimerspec spec = {};
136
137 power_state_fd = open("/sys/power/state", O_RDWR);
138 if (power_state_fd < 0) {
139 perror("open(\"/sys/power/state\") failed (is this test running as root?)");
140 ksft_exit_fail();
141 }
142
143 timerfd = timerfd_create(CLOCK_BOOTTIME_ALARM, 0);
144 if (timerfd < 0) {
145 perror("timerfd_create() failed");
146 ksft_exit_fail();
147 }
148
149 spec.it_value.tv_sec = 5;
150 err = timerfd_settime(timerfd, 0, &spec, NULL);
151 if (err < 0) {
152 perror("timerfd_settime() failed");
153 ksft_exit_fail();
154 }
155
156 if (write(power_state_fd, "mem", strlen("mem")) != strlen("mem")) {
157 perror("entering suspend failed");
158 ksft_exit_fail();
159 }
160
161 close(timerfd);
162 close(power_state_fd);
163}
164
165int main(int argc, char **argv)
166{
167 int opt;
168 bool do_suspend = true;
169 bool succeeded = true;
170 cpu_set_t available_cpus;
171 int err;
172 int cpu;
173
174 while ((opt = getopt(argc, argv, "n")) != -1) {
175 switch (opt) {
176 case 'n':
177 do_suspend = false;
178 break;
179 default:
180 printf("Usage: %s [-n]\n", argv[0]);
181 printf(" -n: do not trigger a suspend/resume cycle before the test\n");
182 return -1;
183 }
184 }
185
186 if (do_suspend)
187 suspend();
188
189 err = sched_getaffinity(0, sizeof(available_cpus), &available_cpus);
190 if (err < 0) {
191 perror("sched_getaffinity() failed");
192 ksft_exit_fail();
193 }
194
195 for (cpu = 0; cpu < CPU_SETSIZE; cpu++) {
196 bool test_success;
197
198 if (!CPU_ISSET(cpu, &available_cpus))
199 continue;
200
201 test_success = run_test(cpu);
202 printf("CPU %d: ", cpu);
203 if (test_success) {
204 printf("[OK]\n");
205 ksft_inc_pass_cnt();
206 } else {
207 printf("[FAILED]\n");
208 ksft_inc_fail_cnt();
209 succeeded = false;
210 }
211 }
212
213 ksft_print_cnts();
214 if (succeeded)
215 ksft_exit_pass();
216 else
217 ksft_exit_fail();
218}
1// SPDX-License-Identifier: GPL-2.0-only
2/*
3 * Copyright (C) 2016 Google, Inc.
4 */
5
6#define _GNU_SOURCE
7
8#include <errno.h>
9#include <fcntl.h>
10#include <sched.h>
11#include <signal.h>
12#include <stdbool.h>
13#include <stdio.h>
14#include <string.h>
15#include <unistd.h>
16#include <sys/ptrace.h>
17#include <sys/stat.h>
18#include <sys/timerfd.h>
19#include <sys/types.h>
20#include <sys/wait.h>
21
22#include "../kselftest.h"
23
24void child(int cpu)
25{
26 cpu_set_t set;
27
28 CPU_ZERO(&set);
29 CPU_SET(cpu, &set);
30 if (sched_setaffinity(0, sizeof(set), &set) != 0) {
31 ksft_print_msg("sched_setaffinity() failed: %s\n",
32 strerror(errno));
33 _exit(1);
34 }
35
36 if (ptrace(PTRACE_TRACEME, 0, NULL, NULL) != 0) {
37 ksft_print_msg("ptrace(PTRACE_TRACEME) failed: %s\n",
38 strerror(errno));
39 _exit(1);
40 }
41
42 if (raise(SIGSTOP) != 0) {
43 ksft_print_msg("raise(SIGSTOP) failed: %s\n", strerror(errno));
44 _exit(1);
45 }
46
47 _exit(0);
48}
49
50int run_test(int cpu)
51{
52 int status;
53 pid_t pid = fork();
54 pid_t wpid;
55
56 if (pid < 0) {
57 ksft_print_msg("fork() failed: %s\n", strerror(errno));
58 return KSFT_FAIL;
59 }
60 if (pid == 0)
61 child(cpu);
62
63 wpid = waitpid(pid, &status, __WALL);
64 if (wpid != pid) {
65 ksft_print_msg("waitpid() failed: %s\n", strerror(errno));
66 return KSFT_FAIL;
67 }
68 if (!WIFSTOPPED(status)) {
69 ksft_print_msg("child did not stop: %s\n", strerror(errno));
70 return KSFT_FAIL;
71 }
72 if (WSTOPSIG(status) != SIGSTOP) {
73 ksft_print_msg("child did not stop with SIGSTOP: %s\n",
74 strerror(errno));
75 return KSFT_FAIL;
76 }
77
78 if (ptrace(PTRACE_SINGLESTEP, pid, NULL, NULL) < 0) {
79 if (errno == EIO) {
80 ksft_print_msg(
81 "ptrace(PTRACE_SINGLESTEP) not supported on this architecture: %s\n",
82 strerror(errno));
83 return KSFT_SKIP;
84 }
85 ksft_print_msg("ptrace(PTRACE_SINGLESTEP) failed: %s\n",
86 strerror(errno));
87 return KSFT_FAIL;
88 }
89
90 wpid = waitpid(pid, &status, __WALL);
91 if (wpid != pid) {
92 ksft_print_msg("waitpid() failed: $s\n", strerror(errno));
93 return KSFT_FAIL;
94 }
95 if (WIFEXITED(status)) {
96 ksft_print_msg("child did not single-step: %s\n",
97 strerror(errno));
98 return KSFT_FAIL;
99 }
100 if (!WIFSTOPPED(status)) {
101 ksft_print_msg("child did not stop: %s\n", strerror(errno));
102 return KSFT_FAIL;
103 }
104 if (WSTOPSIG(status) != SIGTRAP) {
105 ksft_print_msg("child did not stop with SIGTRAP: %s\n",
106 strerror(errno));
107 return KSFT_FAIL;
108 }
109
110 if (ptrace(PTRACE_CONT, pid, NULL, NULL) < 0) {
111 ksft_print_msg("ptrace(PTRACE_CONT) failed: %s\n",
112 strerror(errno));
113 return KSFT_FAIL;
114 }
115
116 wpid = waitpid(pid, &status, __WALL);
117 if (wpid != pid) {
118 ksft_print_msg("waitpid() failed: %s\n", strerror(errno));
119 return KSFT_FAIL;
120 }
121 if (!WIFEXITED(status)) {
122 ksft_print_msg("child did not exit after PTRACE_CONT: %s\n",
123 strerror(errno));
124 return KSFT_FAIL;
125 }
126
127 return KSFT_PASS;
128}
129
130void suspend(void)
131{
132 int power_state_fd;
133 struct sigevent event = {};
134 int timerfd;
135 int err;
136 struct itimerspec spec = {};
137
138 if (getuid() != 0)
139 ksft_exit_skip("Please run the test as root - Exiting.\n");
140
141 power_state_fd = open("/sys/power/state", O_RDWR);
142 if (power_state_fd < 0)
143 ksft_exit_fail_msg(
144 "open(\"/sys/power/state\") failed %s)\n",
145 strerror(errno));
146
147 timerfd = timerfd_create(CLOCK_BOOTTIME_ALARM, 0);
148 if (timerfd < 0)
149 ksft_exit_fail_msg("timerfd_create() failed\n");
150
151 spec.it_value.tv_sec = 5;
152 err = timerfd_settime(timerfd, 0, &spec, NULL);
153 if (err < 0)
154 ksft_exit_fail_msg("timerfd_settime() failed\n");
155
156 if (write(power_state_fd, "mem", strlen("mem")) != strlen("mem"))
157 ksft_exit_fail_msg("Failed to enter Suspend state\n");
158
159 close(timerfd);
160 close(power_state_fd);
161}
162
163int main(int argc, char **argv)
164{
165 int opt;
166 bool do_suspend = true;
167 bool succeeded = true;
168 unsigned int tests = 0;
169 cpu_set_t available_cpus;
170 int err;
171 int cpu;
172
173 ksft_print_header();
174
175 while ((opt = getopt(argc, argv, "n")) != -1) {
176 switch (opt) {
177 case 'n':
178 do_suspend = false;
179 break;
180 default:
181 printf("Usage: %s [-n]\n", argv[0]);
182 printf(" -n: do not trigger a suspend/resume cycle before the test\n");
183 return -1;
184 }
185 }
186
187 err = sched_getaffinity(0, sizeof(available_cpus), &available_cpus);
188 if (err < 0)
189 ksft_exit_fail_msg("sched_getaffinity() failed\n");
190
191 for (cpu = 0; cpu < CPU_SETSIZE; cpu++) {
192 if (!CPU_ISSET(cpu, &available_cpus))
193 continue;
194 tests++;
195 }
196
197 if (do_suspend)
198 suspend();
199
200 ksft_set_plan(tests);
201 for (cpu = 0; cpu < CPU_SETSIZE; cpu++) {
202 int test_success;
203
204 if (!CPU_ISSET(cpu, &available_cpus))
205 continue;
206
207 test_success = run_test(cpu);
208 switch (test_success) {
209 case KSFT_PASS:
210 ksft_test_result_pass("CPU %d\n", cpu);
211 break;
212 case KSFT_SKIP:
213 ksft_test_result_skip("CPU %d\n", cpu);
214 break;
215 case KSFT_FAIL:
216 ksft_test_result_fail("CPU %d\n", cpu);
217 succeeded = false;
218 break;
219 }
220 }
221
222 if (succeeded)
223 ksft_exit_pass();
224 else
225 ksft_exit_fail();
226}