1b7bb8442SJohn Stultz /* alarmtimer suspend test
2b7bb8442SJohn Stultz  *		John Stultz (john.stultz@linaro.org)
3b7bb8442SJohn Stultz  *              (C) Copyright Linaro 2013
4b7bb8442SJohn Stultz  *              Licensed under the GPLv2
5b7bb8442SJohn Stultz  *
6b7bb8442SJohn Stultz  *   This test makes sure the alarmtimer & RTC wakeup code is
7b7bb8442SJohn Stultz  *   functioning.
8b7bb8442SJohn Stultz  *
9b7bb8442SJohn Stultz  *  To build:
10b7bb8442SJohn Stultz  *	$ gcc alarmtimer-suspend.c -o alarmtimer-suspend -lrt
11b7bb8442SJohn Stultz  *
12b7bb8442SJohn Stultz  *   This program is free software: you can redistribute it and/or modify
13b7bb8442SJohn Stultz  *   it under the terms of the GNU General Public License as published by
14b7bb8442SJohn Stultz  *   the Free Software Foundation, either version 2 of the License, or
15b7bb8442SJohn Stultz  *   (at your option) any later version.
16b7bb8442SJohn Stultz  *
17b7bb8442SJohn Stultz  *   This program is distributed in the hope that it will be useful,
18b7bb8442SJohn Stultz  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
19b7bb8442SJohn Stultz  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20b7bb8442SJohn Stultz  *   GNU General Public License for more details.
21b7bb8442SJohn Stultz  */
22b7bb8442SJohn Stultz 
23b7bb8442SJohn Stultz 
24b7bb8442SJohn Stultz #include <stdio.h>
25b7bb8442SJohn Stultz #include <unistd.h>
26b7bb8442SJohn Stultz #include <time.h>
27b7bb8442SJohn Stultz #include <string.h>
28b7bb8442SJohn Stultz #include <signal.h>
29b7bb8442SJohn Stultz #include <stdlib.h>
30b7bb8442SJohn Stultz #include <pthread.h>
31b7bb8442SJohn Stultz #ifdef KTEST
32b7bb8442SJohn Stultz #include "../kselftest.h"
33b7bb8442SJohn Stultz #else
34b7bb8442SJohn Stultz static inline int ksft_exit_pass(void)
35b7bb8442SJohn Stultz {
36b7bb8442SJohn Stultz 	exit(0);
37b7bb8442SJohn Stultz }
38b7bb8442SJohn Stultz static inline int ksft_exit_fail(void)
39b7bb8442SJohn Stultz {
40b7bb8442SJohn Stultz 	exit(1);
41b7bb8442SJohn Stultz }
42b7bb8442SJohn Stultz #endif
43b7bb8442SJohn Stultz 
44b7bb8442SJohn Stultz #define CLOCK_REALTIME			0
45b7bb8442SJohn Stultz #define CLOCK_MONOTONIC			1
46b7bb8442SJohn Stultz #define CLOCK_PROCESS_CPUTIME_ID	2
47b7bb8442SJohn Stultz #define CLOCK_THREAD_CPUTIME_ID		3
48b7bb8442SJohn Stultz #define CLOCK_MONOTONIC_RAW		4
49b7bb8442SJohn Stultz #define CLOCK_REALTIME_COARSE		5
50b7bb8442SJohn Stultz #define CLOCK_MONOTONIC_COARSE		6
51b7bb8442SJohn Stultz #define CLOCK_BOOTTIME			7
52b7bb8442SJohn Stultz #define CLOCK_REALTIME_ALARM		8
53b7bb8442SJohn Stultz #define CLOCK_BOOTTIME_ALARM		9
54b7bb8442SJohn Stultz #define CLOCK_HWSPECIFIC		10
55b7bb8442SJohn Stultz #define CLOCK_TAI			11
56b7bb8442SJohn Stultz #define NR_CLOCKIDS			12
57b7bb8442SJohn Stultz 
58b7bb8442SJohn Stultz 
59b7bb8442SJohn Stultz #define NSEC_PER_SEC 1000000000ULL
60acd5705dSJohn Stultz #define UNREASONABLE_LAT (NSEC_PER_SEC * 5) /* hopefully we resume in 5 secs */
61b7bb8442SJohn Stultz 
62b7bb8442SJohn Stultz #define SUSPEND_SECS 15
63b7bb8442SJohn Stultz int alarmcount;
64b7bb8442SJohn Stultz int alarm_clock_id;
65b7bb8442SJohn Stultz struct timespec start_time;
66b7bb8442SJohn Stultz 
67b7bb8442SJohn Stultz 
68b7bb8442SJohn Stultz char *clockstring(int clockid)
69b7bb8442SJohn Stultz {
70b7bb8442SJohn Stultz 	switch (clockid) {
71b7bb8442SJohn Stultz 	case CLOCK_REALTIME:
72b7bb8442SJohn Stultz 		return "CLOCK_REALTIME";
73b7bb8442SJohn Stultz 	case CLOCK_MONOTONIC:
74b7bb8442SJohn Stultz 		return "CLOCK_MONOTONIC";
75b7bb8442SJohn Stultz 	case CLOCK_PROCESS_CPUTIME_ID:
76b7bb8442SJohn Stultz 		return "CLOCK_PROCESS_CPUTIME_ID";
77b7bb8442SJohn Stultz 	case CLOCK_THREAD_CPUTIME_ID:
78b7bb8442SJohn Stultz 		return "CLOCK_THREAD_CPUTIME_ID";
79b7bb8442SJohn Stultz 	case CLOCK_MONOTONIC_RAW:
80b7bb8442SJohn Stultz 		return "CLOCK_MONOTONIC_RAW";
81b7bb8442SJohn Stultz 	case CLOCK_REALTIME_COARSE:
82b7bb8442SJohn Stultz 		return "CLOCK_REALTIME_COARSE";
83b7bb8442SJohn Stultz 	case CLOCK_MONOTONIC_COARSE:
84b7bb8442SJohn Stultz 		return "CLOCK_MONOTONIC_COARSE";
85b7bb8442SJohn Stultz 	case CLOCK_BOOTTIME:
86b7bb8442SJohn Stultz 		return "CLOCK_BOOTTIME";
87b7bb8442SJohn Stultz 	case CLOCK_REALTIME_ALARM:
88b7bb8442SJohn Stultz 		return "CLOCK_REALTIME_ALARM";
89b7bb8442SJohn Stultz 	case CLOCK_BOOTTIME_ALARM:
90b7bb8442SJohn Stultz 		return "CLOCK_BOOTTIME_ALARM";
91b7bb8442SJohn Stultz 	case CLOCK_TAI:
92b7bb8442SJohn Stultz 		return "CLOCK_TAI";
93b7bb8442SJohn Stultz 	};
94b7bb8442SJohn Stultz 	return "UNKNOWN_CLOCKID";
95b7bb8442SJohn Stultz }
96b7bb8442SJohn Stultz 
97b7bb8442SJohn Stultz 
98b7bb8442SJohn Stultz long long timespec_sub(struct timespec a, struct timespec b)
99b7bb8442SJohn Stultz {
100b7bb8442SJohn Stultz 	long long ret = NSEC_PER_SEC * b.tv_sec + b.tv_nsec;
101b7bb8442SJohn Stultz 
102b7bb8442SJohn Stultz 	ret -= NSEC_PER_SEC * a.tv_sec + a.tv_nsec;
103b7bb8442SJohn Stultz 	return ret;
104b7bb8442SJohn Stultz }
105b7bb8442SJohn Stultz 
106b7bb8442SJohn Stultz int final_ret = 0;
107b7bb8442SJohn Stultz 
108b7bb8442SJohn Stultz void sigalarm(int signo)
109b7bb8442SJohn Stultz {
110b7bb8442SJohn Stultz 	long long delta_ns;
111b7bb8442SJohn Stultz 	struct timespec ts;
112b7bb8442SJohn Stultz 
113b7bb8442SJohn Stultz 	clock_gettime(alarm_clock_id, &ts);
114b7bb8442SJohn Stultz 	alarmcount++;
115b7bb8442SJohn Stultz 
116b7bb8442SJohn Stultz 	delta_ns = timespec_sub(start_time, ts);
117b7bb8442SJohn Stultz 	delta_ns -= NSEC_PER_SEC * SUSPEND_SECS * alarmcount;
118b7bb8442SJohn Stultz 
119b7bb8442SJohn Stultz 	printf("ALARM(%i): %ld:%ld latency: %lld ns ", alarmcount, ts.tv_sec,
120b7bb8442SJohn Stultz 							ts.tv_nsec, delta_ns);
121b7bb8442SJohn Stultz 
122b7bb8442SJohn Stultz 	if (delta_ns > UNREASONABLE_LAT) {
123b7bb8442SJohn Stultz 		printf("[FAIL]\n");
124b7bb8442SJohn Stultz 		final_ret = -1;
125b7bb8442SJohn Stultz 	} else
126b7bb8442SJohn Stultz 		printf("[OK]\n");
127b7bb8442SJohn Stultz 
128b7bb8442SJohn Stultz }
129b7bb8442SJohn Stultz 
130b7bb8442SJohn Stultz int main(void)
131b7bb8442SJohn Stultz {
132b7bb8442SJohn Stultz 	timer_t tm1;
133b7bb8442SJohn Stultz 	struct itimerspec its1, its2;
134b7bb8442SJohn Stultz 	struct sigevent se;
135b7bb8442SJohn Stultz 	struct sigaction act;
136b7bb8442SJohn Stultz 	int signum = SIGRTMAX;
137b7bb8442SJohn Stultz 
138b7bb8442SJohn Stultz 	/* Set up signal handler: */
139b7bb8442SJohn Stultz 	sigfillset(&act.sa_mask);
140b7bb8442SJohn Stultz 	act.sa_flags = 0;
141b7bb8442SJohn Stultz 	act.sa_handler = sigalarm;
142b7bb8442SJohn Stultz 	sigaction(signum, &act, NULL);
143b7bb8442SJohn Stultz 
144b7bb8442SJohn Stultz 	/* Set up timer: */
145b7bb8442SJohn Stultz 	memset(&se, 0, sizeof(se));
146b7bb8442SJohn Stultz 	se.sigev_notify = SIGEV_SIGNAL;
147b7bb8442SJohn Stultz 	se.sigev_signo = signum;
148b7bb8442SJohn Stultz 	se.sigev_value.sival_int = 0;
149b7bb8442SJohn Stultz 
150b7bb8442SJohn Stultz 	for (alarm_clock_id = CLOCK_REALTIME_ALARM;
151b7bb8442SJohn Stultz 			alarm_clock_id <= CLOCK_BOOTTIME_ALARM;
152b7bb8442SJohn Stultz 			alarm_clock_id++) {
153b7bb8442SJohn Stultz 
154b7bb8442SJohn Stultz 		alarmcount = 0;
15561171d04SJohn Stultz 		if (timer_create(alarm_clock_id, &se, &tm1) == -1) {
15661171d04SJohn Stultz 			printf("timer_create failled, %s unspported?\n",
15761171d04SJohn Stultz 					clockstring(alarm_clock_id));
15861171d04SJohn Stultz 			break;
15961171d04SJohn Stultz 		}
160b7bb8442SJohn Stultz 
161b7bb8442SJohn Stultz 		clock_gettime(alarm_clock_id, &start_time);
162b7bb8442SJohn Stultz 		printf("Start time (%s): %ld:%ld\n", clockstring(alarm_clock_id),
163b7bb8442SJohn Stultz 				start_time.tv_sec, start_time.tv_nsec);
164b7bb8442SJohn Stultz 		printf("Setting alarm for every %i seconds\n", SUSPEND_SECS);
165b7bb8442SJohn Stultz 		its1.it_value = start_time;
166b7bb8442SJohn Stultz 		its1.it_value.tv_sec += SUSPEND_SECS;
167b7bb8442SJohn Stultz 		its1.it_interval.tv_sec = SUSPEND_SECS;
168b7bb8442SJohn Stultz 		its1.it_interval.tv_nsec = 0;
169b7bb8442SJohn Stultz 
170b7bb8442SJohn Stultz 		timer_settime(tm1, TIMER_ABSTIME, &its1, &its2);
171b7bb8442SJohn Stultz 
172b7bb8442SJohn Stultz 		while (alarmcount < 5)
173b7bb8442SJohn Stultz 			sleep(1); /* First 5 alarms, do nothing */
174b7bb8442SJohn Stultz 
175b7bb8442SJohn Stultz 		printf("Starting suspend loops\n");
176b7bb8442SJohn Stultz 		while (alarmcount < 10) {
177b7bb8442SJohn Stultz 			int ret;
178b7bb8442SJohn Stultz 
179859b1bebSJohn Stultz 			sleep(3);
180b7bb8442SJohn Stultz 			ret = system("echo mem > /sys/power/state");
181b7bb8442SJohn Stultz 			if (ret)
182b7bb8442SJohn Stultz 				break;
183b7bb8442SJohn Stultz 		}
184b7bb8442SJohn Stultz 		timer_delete(tm1);
185b7bb8442SJohn Stultz 	}
186b7bb8442SJohn Stultz 	if (final_ret)
187b7bb8442SJohn Stultz 		return ksft_exit_fail();
188b7bb8442SJohn Stultz 	return ksft_exit_pass();
189b7bb8442SJohn Stultz }
190