123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220 |
- /*
- * Copyright (C) 2016 Google, Inc.
- *
- * This software is licensed under the terms of the GNU General Public
- * License version 2, as published by the Free Software Foundation, and
- * may be copied, distributed, and modified under those terms.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- */
- #define _GNU_SOURCE
- #include <errno.h>
- #include <fcntl.h>
- #include <sched.h>
- #include <signal.h>
- #include <stdbool.h>
- #include <stdio.h>
- #include <string.h>
- #include <unistd.h>
- #include <sys/ptrace.h>
- #include <sys/stat.h>
- #include <sys/timerfd.h>
- #include <sys/types.h>
- #include <sys/wait.h>
- #include "../kselftest.h"
- void child(int cpu)
- {
- cpu_set_t set;
- CPU_ZERO(&set);
- CPU_SET(cpu, &set);
- if (sched_setaffinity(0, sizeof(set), &set) != 0) {
- ksft_print_msg("sched_setaffinity() failed: %s\n",
- strerror(errno));
- _exit(1);
- }
- if (ptrace(PTRACE_TRACEME, 0, NULL, NULL) != 0) {
- ksft_print_msg("ptrace(PTRACE_TRACEME) failed: %s\n",
- strerror(errno));
- _exit(1);
- }
- if (raise(SIGSTOP) != 0) {
- ksft_print_msg("raise(SIGSTOP) failed: %s\n", strerror(errno));
- _exit(1);
- }
- _exit(0);
- }
- bool run_test(int cpu)
- {
- int status;
- pid_t pid = fork();
- pid_t wpid;
- if (pid < 0) {
- ksft_print_msg("fork() failed: %s\n", strerror(errno));
- return false;
- }
- if (pid == 0)
- child(cpu);
- wpid = waitpid(pid, &status, __WALL);
- if (wpid != pid) {
- ksft_print_msg("waitpid() failed: %s\n", strerror(errno));
- return false;
- }
- if (!WIFSTOPPED(status)) {
- ksft_print_msg("child did not stop: %s\n", strerror(errno));
- return false;
- }
- if (WSTOPSIG(status) != SIGSTOP) {
- ksft_print_msg("child did not stop with SIGSTOP: %s\n",
- strerror(errno));
- return false;
- }
- if (ptrace(PTRACE_SINGLESTEP, pid, NULL, NULL) < 0) {
- if (errno == EIO) {
- ksft_exit_skip(
- "ptrace(PTRACE_SINGLESTEP) not supported on this architecture: %s\n",
- strerror(errno));
- }
- ksft_print_msg("ptrace(PTRACE_SINGLESTEP) failed: %s\n",
- strerror(errno));
- return false;
- }
- wpid = waitpid(pid, &status, __WALL);
- if (wpid != pid) {
- ksft_print_msg("waitpid() failed: $s\n", strerror(errno));
- return false;
- }
- if (WIFEXITED(status)) {
- ksft_print_msg("child did not single-step: %s\n",
- strerror(errno));
- return false;
- }
- if (!WIFSTOPPED(status)) {
- ksft_print_msg("child did not stop: %s\n", strerror(errno));
- return false;
- }
- if (WSTOPSIG(status) != SIGTRAP) {
- ksft_print_msg("child did not stop with SIGTRAP: %s\n",
- strerror(errno));
- return false;
- }
- if (ptrace(PTRACE_CONT, pid, NULL, NULL) < 0) {
- ksft_print_msg("ptrace(PTRACE_CONT) failed: %s\n",
- strerror(errno));
- return false;
- }
- wpid = waitpid(pid, &status, __WALL);
- if (wpid != pid) {
- ksft_print_msg("waitpid() failed: %s\n", strerror(errno));
- return false;
- }
- if (!WIFEXITED(status)) {
- ksft_print_msg("child did not exit after PTRACE_CONT: %s\n",
- strerror(errno));
- return false;
- }
- return true;
- }
- void suspend(void)
- {
- int power_state_fd;
- struct sigevent event = {};
- int timerfd;
- int err;
- struct itimerspec spec = {};
- if (getuid() != 0)
- ksft_exit_skip("Please run the test as root - Exiting.\n");
- power_state_fd = open("/sys/power/state", O_RDWR);
- if (power_state_fd < 0)
- ksft_exit_fail_msg(
- "open(\"/sys/power/state\") failed %s)\n",
- strerror(errno));
- timerfd = timerfd_create(CLOCK_BOOTTIME_ALARM, 0);
- if (timerfd < 0)
- ksft_exit_fail_msg("timerfd_create() failed\n");
- spec.it_value.tv_sec = 5;
- err = timerfd_settime(timerfd, 0, &spec, NULL);
- if (err < 0)
- ksft_exit_fail_msg("timerfd_settime() failed\n");
- if (write(power_state_fd, "mem", strlen("mem")) != strlen("mem"))
- ksft_exit_fail_msg("Failed to enter Suspend state\n");
- close(timerfd);
- close(power_state_fd);
- }
- int main(int argc, char **argv)
- {
- int opt;
- bool do_suspend = true;
- bool succeeded = true;
- cpu_set_t available_cpus;
- int err;
- int cpu;
- ksft_print_header();
- while ((opt = getopt(argc, argv, "n")) != -1) {
- switch (opt) {
- case 'n':
- do_suspend = false;
- break;
- default:
- printf("Usage: %s [-n]\n", argv[0]);
- printf(" -n: do not trigger a suspend/resume cycle before the test\n");
- return -1;
- }
- }
- if (do_suspend)
- suspend();
- err = sched_getaffinity(0, sizeof(available_cpus), &available_cpus);
- if (err < 0)
- ksft_exit_fail_msg("sched_getaffinity() failed\n");
- for (cpu = 0; cpu < CPU_SETSIZE; cpu++) {
- bool test_success;
- if (!CPU_ISSET(cpu, &available_cpus))
- continue;
- test_success = run_test(cpu);
- if (test_success) {
- ksft_test_result_pass("CPU %d\n", cpu);
- } else {
- ksft_test_result_fail("CPU %d\n", cpu);
- succeeded = false;
- }
- }
- if (succeeded)
- ksft_exit_pass();
- else
- ksft_exit_fail();
- }
|