]> git.saurik.com Git - apple/xnu.git/blame - tests/kevent_continuous_time.c
xnu-6153.101.6.tar.gz
[apple/xnu.git] / tests / kevent_continuous_time.c
CommitLineData
39037602
A
1#include <stdio.h>
2#include <unistd.h>
3
4#include <mach/mach.h>
5#include <mach/mach_time.h>
6#include <sys/time.h>
7#include <spawn.h>
8#include <sys/wait.h>
9#include <stdio.h>
10#include <unistd.h>
11#include <stdlib.h>
12#include <time.h>
13#include <errno.h>
14#include <sys/event.h>
15
16#include <darwintest.h>
17
18extern char **environ;
19
20static mach_timebase_info_data_t tb_info;
0a7de745 21static const uint64_t one_mil = 1000LL * 1000LL;
39037602
A
22
23#define tick_to_ns(ticks) (((ticks) * tb_info.numer) / (tb_info.denom))
24#define tick_to_ms(ticks) (tick_to_ns(ticks)/one_mil)
25
26#define ns_to_tick(ns) ((ns) * tb_info.denom / tb_info.numer)
27#define ms_to_tick(ms) (ns_to_tick((ms) * one_mil))
28
0a7de745
A
29static uint64_t
30time_delta_ms(void)
31{
39037602
A
32 uint64_t abs_now = mach_absolute_time();
33 uint64_t cnt_now = mach_continuous_time();;
34 return tick_to_ms(cnt_now) - tick_to_ms(abs_now);
35}
36
37static int run_sleep_tests = 0;
38
0a7de745
A
39static int
40trigger_sleep(int for_secs)
41{
42 if (!run_sleep_tests) {
43 return 0;
44 }
39037602
A
45
46 // sleep for 1 seconds each iteration
47 char buf[10];
48 snprintf(buf, 10, "%d", for_secs);
49
50 T_LOG("Sleepeing for %s seconds...", buf);
51
52 int spawn_ret, pid;
53 char *const pmset1_args[] = {"/usr/bin/pmset", "relative", "wake", buf, NULL};
54 T_ASSERT_POSIX_ZERO((spawn_ret = posix_spawn(&pid, pmset1_args[0], NULL, NULL, pmset1_args, environ)), NULL);
0a7de745 55
39037602
A
56 T_ASSERT_EQ(waitpid(pid, &spawn_ret, 0), pid, NULL);
57 T_ASSERT_EQ(spawn_ret, 0, NULL);
58
59 char *const pmset2_args[] = {"/usr/bin/pmset", "sleepnow", NULL};
60 T_ASSERT_POSIX_ZERO((spawn_ret = posix_spawn(&pid, pmset2_args[0], NULL, NULL, pmset2_args, environ)), NULL);
0a7de745 61
39037602
A
62 T_ASSERT_EQ(waitpid(pid, &spawn_ret, 0), pid, NULL);
63 T_ASSERT_EQ(spawn_ret, 0, NULL);
64
65 return 0;
66}
67
68// waits up to 30 seconds for system to sleep
69// returns number of seconds it took for sleep to be entered
70// or -1 if sleep wasn't accomplished
0a7de745
A
71static int
72wait_for_sleep()
73{
74 if (!run_sleep_tests) {
75 return 0;
76 }
39037602
A
77
78 uint64_t before_diff = time_delta_ms();
0a7de745
A
79
80 for (int i = 0; i < 30; i++) {
39037602
A
81 uint64_t after_diff = time_delta_ms();
82
83 // on OSX, there's enough latency between calls to MCT and MAT
84 // when the system is going down for sleep for values to diverge a few ms
0a7de745 85 if (llabs((int64_t)before_diff - (int64_t)after_diff) > 2) {
39037602
A
86 return i + 1;
87 }
0a7de745 88
39037602 89 sleep(1);
0a7de745 90 T_LOG("waited %d seconds for sleep...", i + 1);
39037602
A
91 }
92 return -1;
93}
94
813fb2f6 95T_DECL(kevent_continuous_time_periodic_tick, "kevent(EVFILT_TIMER with NOTE_MACH_CONTINUOUS_TIME)", T_META_LTEPHASE(LTE_POSTINIT)){
39037602
A
96 mach_timebase_info(&tb_info);
97 int kq;
98 T_ASSERT_POSIX_SUCCESS((kq = kqueue()), NULL);
99
d9a64523
A
100 struct kevent64_s kev = {
101 .ident = 1,
102 .filter = EVFILT_TIMER,
103 .flags = EV_ADD | EV_RECEIPT,
104 .fflags = NOTE_SECONDS | NOTE_MACH_CONTINUOUS_TIME,
105 .data = 4,
106 };
107 T_LOG("EV_SET(&kev, 1, EVFILT_TIMER, EV_ADD, NOTE_SECONDS | NOTE_MACH_CONTINUOUS_TIME, 4, 0, 0, 0);");
39037602 108
d9a64523
A
109 T_ASSERT_EQ(kevent64(kq, &kev, 1, &kev, 1, 0, NULL), 1, NULL);
110 T_ASSERT_EQ(0ll, kev.data, "No error returned");
39037602
A
111
112 uint64_t abs_then = mach_absolute_time();
113 uint64_t cnt_then = mach_continuous_time();;
114
115 trigger_sleep(1);
116 int sleep_secs = wait_for_sleep();
117
d9a64523
A
118 T_WITH_ERRNO; T_ASSERT_EQ(kevent64(kq, NULL, 0, &kev, 1, 0, NULL), 1, "kevent() should have returned one event");
119 T_LOG("event = {.ident = %llx, .filter = %d, .flags = %d, .fflags = %d, .data = %lld, .udata = %lld}", kev.ident, kev.filter, kev.flags, kev.fflags, kev.data, kev.udata);
120 T_ASSERT_EQ(kev.flags & EV_ERROR, 0, "event should not have EV_ERROR set: %s", kev.flags & EV_ERROR ? strerror((int)kev.data) : "no error");
39037602
A
121
122 uint64_t abs_now = mach_absolute_time();
123 uint64_t cnt_now = mach_continuous_time();;
124 uint64_t ct_ms_progressed = tick_to_ms(cnt_now - cnt_then);
125 uint64_t ab_ms_progressed = tick_to_ms(abs_now - abs_then);
126
127 T_LOG("ct progressed %llu ms, abs progressed %llu ms", ct_ms_progressed, tick_to_ms(abs_now - abs_then));
128
129 if (run_sleep_tests) {
130 T_ASSERT_GT(llabs((int64_t)ct_ms_progressed - (int64_t)ab_ms_progressed), 500LL, "should have > 500ms difference between MCT and MAT");
131 } else {
132 T_ASSERT_LT(llabs((int64_t)ct_ms_progressed - (int64_t)ab_ms_progressed), 10LL, "should have < 10ms difference between MCT and MAT");
133 }
134
135 if (sleep_secs < 4) {
136 T_ASSERT_LT(llabs((int64_t)ct_ms_progressed - 4000), 100LL, "mach_continuous_time should progress ~4 seconds (+/- 100ms) between sleeps");
137 }
138
139 sleep(1);
140
d9a64523
A
141 kev = (struct kevent64_s){
142 .ident = 1,
143 .filter = EVFILT_TIMER,
144 .flags = EV_DELETE | EV_RECEIPT,
145 };
146 T_LOG("EV_SET(&kev, 1, EVFILT_TIMER, EV_DELETE, 0, 0, 0);");
147 T_ASSERT_EQ(kevent64(kq, &kev, 1, &kev, 1, 0, NULL), 1, NULL);
148 T_ASSERT_EQ(0ll, kev.data, "No error returned");
39037602
A
149
150 T_ASSERT_POSIX_ZERO(close(kq), NULL);
151}
152
813fb2f6 153T_DECL(kevent_continuous_time_absolute, "kevent(EVFILT_TIMER with NOTE_MACH_CONTINUOUS_TIME and NOTE_ABSOLUTE)", T_META_LTEPHASE(LTE_POSTINIT)){
39037602
A
154 mach_timebase_info(&tb_info);
155
156 int kq;
157 T_ASSERT_POSIX_SUCCESS((kq = kqueue()), NULL);
158
159 struct timeval tv;
160 gettimeofday(&tv, NULL);
d9a64523 161 int64_t nowus = (int64_t)tv.tv_sec * USEC_PER_SEC + (int64_t)tv.tv_usec;
0a7de745 162 int64_t fire_at = (3 * USEC_PER_SEC) + nowus;
39037602
A
163
164 uint64_t cnt_now = mach_continuous_time();
165 uint64_t cnt_then = cnt_now + ms_to_tick(3000);
166
167 T_LOG("currently is %llu, firing at %llu", nowus, fire_at);
168
d9a64523
A
169 struct kevent64_s kev = {
170 .ident = 2,
171 .filter = EVFILT_TIMER,
172 .flags = EV_ADD | EV_RECEIPT,
173 .fflags = NOTE_MACH_CONTINUOUS_TIME | NOTE_ABSOLUTE | NOTE_USECONDS,
174 .data = fire_at,
175 };
176 T_LOG("EV_SET(&kev, 2, EVFILT_TIMER, EV_ADD, NOTE_MACH_CONTINUOUS_TIME | NOTE_ABSOLUTE | NOTE_USECONDS, fire_at, 0);");
39037602 177
d9a64523
A
178 T_ASSERT_EQ(kevent64(kq, &kev, 1, &kev, 1, 0, NULL), 1, NULL);
179 T_ASSERT_EQ(0ll, kev.data, "No error returned");
39037602
A
180
181 T_LOG("testing NOTE_MACH_CONTINUOUS_TIME | NOTE_ABSOLUTE between sleep");
182
183 trigger_sleep(1);
184
185 struct timespec timeout = {
186 .tv_sec = 10,
d9a64523 187 .tv_nsec = 0,
39037602
A
188 };
189 struct kevent64_s event = {0};
190 T_ASSERT_EQ(kevent64(kq, NULL, 0, &event, 1, 0, &timeout), 1, "kevent() should have returned one event");
191 T_LOG("event = {.ident = %llx, .filter = %d, .flags = %d, .fflags = %d, .data = %lld, .udata = %lld}", event.ident, event.filter, event.flags, event.fflags, event.data, event.udata);
192 T_ASSERT_EQ(event.flags & EV_ERROR, 0, "event should not have EV_ERROR set: %s", event.flags & EV_ERROR ? strerror((int)event.data) : "no error");
193
194 uint64_t elapsed_ms = tick_to_ms(mach_continuous_time() - cnt_now);
195 int64_t missed_by = tick_to_ns((int64_t)mach_continuous_time() - (int64_t)cnt_then) / 1000000;
196
197 // ~1/2 second is about as good as we'll get
198 T_ASSERT_LT(llabs(missed_by), 500LL, "timer should pop 3 sec in the future, popped after %lldms", elapsed_ms);
199
200 T_ASSERT_EQ(event.data, 1LL, NULL);
201
202 T_ASSERT_EQ(event.ident, 2ULL, NULL);
203
204 // try getting a periodic tick out of kq
205 T_ASSERT_EQ(kevent64(kq, NULL, 0, &event, 1, 0, &timeout), 0, NULL);
206 T_ASSERT_EQ(event.flags & EV_ERROR, 0, "event should not have EV_ERROR set: %s", event.flags & EV_ERROR ? strerror((int)event.data) : "no error");
207
208 T_ASSERT_POSIX_ZERO(close(kq), NULL);
209}
210
813fb2f6 211T_DECL(kevent_continuous_time_pops, "kevent(EVFILT_TIMER with NOTE_MACH_CONTINUOUS_TIME with multiple pops)", T_META_LTEPHASE(LTE_POSTINIT)){
39037602
A
212 // have to throttle rate at which pmset is called
213 sleep(2);
214
215 mach_timebase_info(&tb_info);
216
217 int kq;
218 T_ASSERT_POSIX_SUCCESS((kq = kqueue()), NULL);
219
220 // test that periodic ticks accumulate while asleep
d9a64523
A
221 struct kevent64_s kev = {
222 .ident = 3,
223 .filter = EVFILT_TIMER,
224 .flags = EV_ADD | EV_RECEIPT,
225 .fflags = NOTE_MACH_CONTINUOUS_TIME,
226 .data = 100,
227 };
228 T_LOG("EV_SET(&kev, 3, EVFILT_TIMER, EV_ADD, NOTE_MACH_CONTINUOUS_TIME, 100, 0);");
39037602
A
229
230 // wait for first pop, then sleep
d9a64523
A
231 T_ASSERT_EQ(kevent64(kq, &kev, 1, &kev, 1, 0, NULL), 1, NULL);
232 T_ASSERT_EQ(0ll, kev.data, "No error returned");
39037602
A
233
234 struct kevent64_s event = {0};
235 T_ASSERT_EQ(kevent64(kq, NULL, 0, &event, 1, 0, NULL), 1, "kevent() should have returned one event");
236 T_LOG("event = {.ident = %llx, .filter = %d, .flags = %d, .fflags = %d, .data = %lld, .udata = %llu}", event.ident, event.filter, event.flags, event.fflags, event.data, event.udata);
237 T_ASSERT_EQ(event.flags & EV_ERROR, 0, "should not have EV_ERROR set: %s", event.flags & EV_ERROR ? strerror((int)event.data) : "no error");
238 T_ASSERT_EQ(event.ident, 3ULL, NULL);
239
240 uint64_t cnt_then = mach_continuous_time();
241 trigger_sleep(2);
242
243 int sleep_secs = 0;
0a7de745 244 if (run_sleep_tests) {
39037602 245 sleep_secs = wait_for_sleep();
0a7de745 246 } else {
39037602
A
247 // simulate 2 seconds of system "sleep"
248 sleep(2);
249 }
250
251 uint64_t cnt_now = mach_continuous_time();
252
253 uint64_t ms_elapsed = tick_to_ms(cnt_now - cnt_then);
0a7de745 254 if (run_sleep_tests) {
39037602
A
255 T_ASSERT_LT(llabs((int64_t)ms_elapsed - 2000LL), 500LL, "slept for %llums, expected 2000ms (astris is connected?)", ms_elapsed);
256 }
257
258 T_ASSERT_EQ(kevent64(kq, NULL, 0, &event, 1, 0, NULL), 1, "kevent() should have returned one event");
259 T_LOG("event = {.ident = %llx, .filter = %d, .flags = %d, .fflags = %d, .data = %lld, .udata = %llu}", event.ident, event.filter, event.flags, event.fflags, event.data, event.udata);
260 T_ASSERT_EQ(event.ident, 3ULL, NULL);
261
262 uint64_t expected_pops = ms_elapsed / 100;
263 uint64_t got_pops = (uint64_t)event.data;
264
265 T_ASSERT_GE(got_pops, expected_pops - 1, "tracking pops while asleep");
266 T_ASSERT_POSIX_ZERO(close(kq), NULL);
267}