1*e3723e1fSApple OSS Distributions #include <darwintest.h>
2*e3723e1fSApple OSS Distributions #include <darwintest_utils.h>
3*e3723e1fSApple OSS Distributions #include <errno.h>
4*e3723e1fSApple OSS Distributions #include <mach/mach.h>
5*e3723e1fSApple OSS Distributions #include <mach/mach_error.h>
6*e3723e1fSApple OSS Distributions #include <mach/policy.h>
7*e3723e1fSApple OSS Distributions #include <mach/task_info.h>
8*e3723e1fSApple OSS Distributions #include <mach/thread_info.h>
9*e3723e1fSApple OSS Distributions #include <signal.h>
10*e3723e1fSApple OSS Distributions #include <stdio.h>
11*e3723e1fSApple OSS Distributions #include <stdlib.h>
12*e3723e1fSApple OSS Distributions #include <sys/mman.h>
13*e3723e1fSApple OSS Distributions #include <sys/sysctl.h>
14*e3723e1fSApple OSS Distributions #include <unistd.h>
15*e3723e1fSApple OSS Distributions
16*e3723e1fSApple OSS Distributions #include "test_utils.h"
17*e3723e1fSApple OSS Distributions
18*e3723e1fSApple OSS Distributions T_GLOBAL_META(T_META_RUN_CONCURRENTLY(true), T_META_TAG_VM_PREFERRED);
19*e3723e1fSApple OSS Distributions
20*e3723e1fSApple OSS Distributions /* *************************************************************************************
21*e3723e1fSApple OSS Distributions * Test the task_info API.
22*e3723e1fSApple OSS Distributions *
23*e3723e1fSApple OSS Distributions * This is a functional test of the following APIs:
24*e3723e1fSApple OSS Distributions * TASK_BASIC_INFO_32
25*e3723e1fSApple OSS Distributions * TASK_BASIC2_INFO_32
26*e3723e1fSApple OSS Distributions * TASK_BASIC_INFO_64
27*e3723e1fSApple OSS Distributions * TASK_BASIC_INFO_64_2
28*e3723e1fSApple OSS Distributions * TASK_POWER_INFO_V2
29*e3723e1fSApple OSS Distributions * TASK_FLAGS_INFO
30*e3723e1fSApple OSS Distributions * TASK_AFFINITY_TAG_INFO
31*e3723e1fSApple OSS Distributions * TASK_THREAD_TIMES_INFO
32*e3723e1fSApple OSS Distributions * TASK_ABSOLUTE_TIME_INFO
33*e3723e1fSApple OSS Distributions * <rdar://problem/22242021> Add tests to increase code coverage for the task_info API
34*e3723e1fSApple OSS Distributions * *************************************************************************************
35*e3723e1fSApple OSS Distributions */
36*e3723e1fSApple OSS Distributions #define TESTPHYSFOOTPRINTVAL 5
37*e3723e1fSApple OSS Distributions #define CANARY 0x0f0f0f0f0f0f0f0fULL
38*e3723e1fSApple OSS Distributions #if !defined(CONFIG_EMBEDDED)
39*e3723e1fSApple OSS Distributions #define ABSOLUTE_MIN_USER_TIME_DIFF 150
40*e3723e1fSApple OSS Distributions #define ABSOLUTE_MIN_SYSTEM_TIME_DIFF 300
41*e3723e1fSApple OSS Distributions #endif
42*e3723e1fSApple OSS Distributions
43*e3723e1fSApple OSS Distributions enum info_kind { INFO_32, INFO_64, INFO_32_2, INFO_64_2, INFO_MACH, INFO_MAX };
44*e3723e1fSApple OSS Distributions
45*e3723e1fSApple OSS Distributions enum info_get { GET_SUSPEND_COUNT, GET_RESIDENT_SIZE, GET_VIRTUAL_SIZE, GET_USER_TIME, GET_SYS_TIME, GET_POLICY, GET_MAX_RES };
46*e3723e1fSApple OSS Distributions
47*e3723e1fSApple OSS Distributions /*
48*e3723e1fSApple OSS Distributions * This function uses CPU cycles by doing a factorial computation.
49*e3723e1fSApple OSS Distributions */
50*e3723e1fSApple OSS Distributions static void do_factorial_task(void);
51*e3723e1fSApple OSS Distributions
52*e3723e1fSApple OSS Distributions void test_task_basic_info_32(void);
53*e3723e1fSApple OSS Distributions void test_task_basic_info_64(void);
54*e3723e1fSApple OSS Distributions void task_basic_info_32_debug(void);
55*e3723e1fSApple OSS Distributions void task_basic2_info_32_warmup(void);
56*e3723e1fSApple OSS Distributions void test_task_basic_info(enum info_kind kind);
57*e3723e1fSApple OSS Distributions uint64_t info_get(enum info_kind kind, enum info_get get, void * data);
58*e3723e1fSApple OSS Distributions
59*e3723e1fSApple OSS Distributions T_DECL(task_vm_info, "tests task vm info", T_META_ASROOT(true), T_META_LTEPHASE(LTE_POSTINIT))
60*e3723e1fSApple OSS Distributions {
61*e3723e1fSApple OSS Distributions kern_return_t err;
62*e3723e1fSApple OSS Distributions task_vm_info_data_t vm_info;
63*e3723e1fSApple OSS Distributions
64*e3723e1fSApple OSS Distributions mach_msg_type_number_t count = TASK_VM_INFO_COUNT;
65*e3723e1fSApple OSS Distributions
66*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_VM_INFO_PURGEABLE, (task_info_t)&vm_info, &count);
67*e3723e1fSApple OSS Distributions
68*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
69*e3723e1fSApple OSS Distributions
70*e3723e1fSApple OSS Distributions T_EXPECT_NE(vm_info.virtual_size, 0ULL, "task_info return value !=0 for virtual_size\n");
71*e3723e1fSApple OSS Distributions
72*e3723e1fSApple OSS Distributions T_EXPECT_NE(vm_info.phys_footprint, 0ULL, "task_info return value !=0 for phys_footprint\n");
73*e3723e1fSApple OSS Distributions
74*e3723e1fSApple OSS Distributions /*
75*e3723e1fSApple OSS Distributions * Test the REV0 version of TASK_VM_INFO. It should not change the value of phys_footprint.
76*e3723e1fSApple OSS Distributions */
77*e3723e1fSApple OSS Distributions
78*e3723e1fSApple OSS Distributions count = TASK_VM_INFO_REV0_COUNT;
79*e3723e1fSApple OSS Distributions vm_info.phys_footprint = TESTPHYSFOOTPRINTVAL;
80*e3723e1fSApple OSS Distributions vm_info.min_address = CANARY;
81*e3723e1fSApple OSS Distributions vm_info.max_address = CANARY;
82*e3723e1fSApple OSS Distributions
83*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_VM_INFO_PURGEABLE, (task_info_t)&vm_info, &count);
84*e3723e1fSApple OSS Distributions
85*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
86*e3723e1fSApple OSS Distributions
87*e3723e1fSApple OSS Distributions T_EXPECT_EQ(count, TASK_VM_INFO_REV0_COUNT, "task_info count(%d) is equal to TASK_VM_INFO_REV0_COUNT", count);
88*e3723e1fSApple OSS Distributions
89*e3723e1fSApple OSS Distributions T_EXPECT_NE(vm_info.virtual_size, 0ULL, "task_info --rev0 call does not return 0 for virtual_size");
90*e3723e1fSApple OSS Distributions
91*e3723e1fSApple OSS Distributions T_EXPECT_EQ(vm_info.phys_footprint, (unsigned long long)TESTPHYSFOOTPRINTVAL,
92*e3723e1fSApple OSS Distributions "task_info --rev0 call returned value %llu for vm_info.phys_footprint. Expected %u since this value should not be "
93*e3723e1fSApple OSS Distributions "modified by rev0",
94*e3723e1fSApple OSS Distributions vm_info.phys_footprint, TESTPHYSFOOTPRINTVAL);
95*e3723e1fSApple OSS Distributions
96*e3723e1fSApple OSS Distributions T_EXPECT_EQ(vm_info.min_address, CANARY,
97*e3723e1fSApple OSS Distributions "task_info --rev0 call returned value 0x%llx for vm_info.min_address. Expected 0x%llx since this value should not "
98*e3723e1fSApple OSS Distributions "be modified by rev0",
99*e3723e1fSApple OSS Distributions vm_info.min_address, CANARY);
100*e3723e1fSApple OSS Distributions
101*e3723e1fSApple OSS Distributions T_EXPECT_EQ(vm_info.max_address, CANARY,
102*e3723e1fSApple OSS Distributions "task_info --rev0 call returned value 0x%llx for vm_info.max_address. Expected 0x%llx since this value should not "
103*e3723e1fSApple OSS Distributions "be modified by rev0",
104*e3723e1fSApple OSS Distributions vm_info.max_address, CANARY);
105*e3723e1fSApple OSS Distributions
106*e3723e1fSApple OSS Distributions /*
107*e3723e1fSApple OSS Distributions * Test the REV1 version of TASK_VM_INFO.
108*e3723e1fSApple OSS Distributions */
109*e3723e1fSApple OSS Distributions
110*e3723e1fSApple OSS Distributions count = TASK_VM_INFO_REV1_COUNT;
111*e3723e1fSApple OSS Distributions vm_info.phys_footprint = TESTPHYSFOOTPRINTVAL;
112*e3723e1fSApple OSS Distributions vm_info.min_address = CANARY;
113*e3723e1fSApple OSS Distributions vm_info.max_address = CANARY;
114*e3723e1fSApple OSS Distributions
115*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_VM_INFO_PURGEABLE, (task_info_t)&vm_info, &count);
116*e3723e1fSApple OSS Distributions
117*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
118*e3723e1fSApple OSS Distributions
119*e3723e1fSApple OSS Distributions T_EXPECT_EQ(count, TASK_VM_INFO_REV1_COUNT, "task_info count(%d) is equal to TASK_VM_INFO_REV1_COUNT", count);
120*e3723e1fSApple OSS Distributions
121*e3723e1fSApple OSS Distributions T_EXPECT_NE(vm_info.virtual_size, 0ULL, "task_info --rev1 call does not return 0 for virtual_size");
122*e3723e1fSApple OSS Distributions
123*e3723e1fSApple OSS Distributions T_EXPECT_NE(vm_info.phys_footprint, (unsigned long long)TESTPHYSFOOTPRINTVAL,
124*e3723e1fSApple OSS Distributions "task_info --rev1 call returned value %llu for vm_info.phys_footprint. Expected value is anything other than %u "
125*e3723e1fSApple OSS Distributions "since this value should not be modified by rev1",
126*e3723e1fSApple OSS Distributions vm_info.phys_footprint, TESTPHYSFOOTPRINTVAL);
127*e3723e1fSApple OSS Distributions
128*e3723e1fSApple OSS Distributions T_EXPECT_EQ(vm_info.min_address, CANARY,
129*e3723e1fSApple OSS Distributions "task_info --rev1 call returned value 0x%llx for vm_info.min_address. Expected 0x%llx since this value should not "
130*e3723e1fSApple OSS Distributions "be modified by rev1",
131*e3723e1fSApple OSS Distributions vm_info.min_address, CANARY);
132*e3723e1fSApple OSS Distributions
133*e3723e1fSApple OSS Distributions T_EXPECT_EQ(vm_info.max_address, CANARY,
134*e3723e1fSApple OSS Distributions "task_info --rev1 call returned value 0x%llx for vm_info.max_address. Expected 0x%llx since this value should not "
135*e3723e1fSApple OSS Distributions "be modified by rev1",
136*e3723e1fSApple OSS Distributions vm_info.max_address, CANARY);
137*e3723e1fSApple OSS Distributions
138*e3723e1fSApple OSS Distributions /*
139*e3723e1fSApple OSS Distributions * Test the REV2 version of TASK_VM_INFO.
140*e3723e1fSApple OSS Distributions */
141*e3723e1fSApple OSS Distributions
142*e3723e1fSApple OSS Distributions count = TASK_VM_INFO_REV2_COUNT;
143*e3723e1fSApple OSS Distributions vm_info.phys_footprint = TESTPHYSFOOTPRINTVAL;
144*e3723e1fSApple OSS Distributions vm_info.min_address = CANARY;
145*e3723e1fSApple OSS Distributions vm_info.max_address = CANARY;
146*e3723e1fSApple OSS Distributions
147*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_VM_INFO_PURGEABLE, (task_info_t)&vm_info, &count);
148*e3723e1fSApple OSS Distributions
149*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
150*e3723e1fSApple OSS Distributions
151*e3723e1fSApple OSS Distributions T_EXPECT_EQ(count, TASK_VM_INFO_REV2_COUNT, "task_info count(%d) is equal to TASK_VM_INFO_REV2_COUNT\n", count);
152*e3723e1fSApple OSS Distributions
153*e3723e1fSApple OSS Distributions T_EXPECT_NE(vm_info.virtual_size, 0ULL, "task_info --rev2 call does not return 0 for virtual_size\n");
154*e3723e1fSApple OSS Distributions
155*e3723e1fSApple OSS Distributions T_EXPECT_NE(vm_info.phys_footprint, (unsigned long long)TESTPHYSFOOTPRINTVAL,
156*e3723e1fSApple OSS Distributions "task_info --rev2 call returned value %llu for vm_info.phys_footprint. Expected anything other than %u since this "
157*e3723e1fSApple OSS Distributions "value should be modified by rev2",
158*e3723e1fSApple OSS Distributions vm_info.phys_footprint, TESTPHYSFOOTPRINTVAL);
159*e3723e1fSApple OSS Distributions
160*e3723e1fSApple OSS Distributions T_EXPECT_NE(vm_info.min_address, CANARY,
161*e3723e1fSApple OSS Distributions "task_info --rev2 call returned value 0x%llx for vm_info.min_address. Expected anything other than 0x%llx since "
162*e3723e1fSApple OSS Distributions "this value should be modified by rev2",
163*e3723e1fSApple OSS Distributions vm_info.min_address, CANARY);
164*e3723e1fSApple OSS Distributions
165*e3723e1fSApple OSS Distributions T_EXPECT_NE(vm_info.max_address, CANARY,
166*e3723e1fSApple OSS Distributions "task_info --rev2 call returned value 0x%llx for vm_info.max_address. Expected anything other than 0x%llx since "
167*e3723e1fSApple OSS Distributions "this value should be modified by rev2",
168*e3723e1fSApple OSS Distributions vm_info.max_address, CANARY);
169*e3723e1fSApple OSS Distributions
170*e3723e1fSApple OSS Distributions /*
171*e3723e1fSApple OSS Distributions * Test the REV4 version of TASK_VM_INFO.
172*e3723e1fSApple OSS Distributions */
173*e3723e1fSApple OSS Distributions
174*e3723e1fSApple OSS Distributions count = TASK_VM_INFO_REV4_COUNT;
175*e3723e1fSApple OSS Distributions vm_info.phys_footprint = TESTPHYSFOOTPRINTVAL;
176*e3723e1fSApple OSS Distributions vm_info.min_address = CANARY;
177*e3723e1fSApple OSS Distributions vm_info.max_address = CANARY;
178*e3723e1fSApple OSS Distributions vm_info.limit_bytes_remaining = CANARY;
179*e3723e1fSApple OSS Distributions
180*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_VM_INFO_PURGEABLE, (task_info_t)&vm_info, &count);
181*e3723e1fSApple OSS Distributions
182*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
183*e3723e1fSApple OSS Distributions
184*e3723e1fSApple OSS Distributions T_EXPECT_EQ(count, TASK_VM_INFO_REV4_COUNT, "task_info count(%d) is equal to TASK_VM_INFO_REV4_COUNT\n", count);
185*e3723e1fSApple OSS Distributions
186*e3723e1fSApple OSS Distributions T_EXPECT_NE(vm_info.phys_footprint, (unsigned long long)TESTPHYSFOOTPRINTVAL,
187*e3723e1fSApple OSS Distributions "task_info --rev4 call returned value %llu for vm_info.phys_footprint. Expected anything other than %u since this "
188*e3723e1fSApple OSS Distributions "value should be modified by rev4",
189*e3723e1fSApple OSS Distributions vm_info.phys_footprint, TESTPHYSFOOTPRINTVAL);
190*e3723e1fSApple OSS Distributions
191*e3723e1fSApple OSS Distributions T_EXPECT_NE(vm_info.min_address, CANARY,
192*e3723e1fSApple OSS Distributions "task_info --rev4 call returned value 0x%llx for vm_info.min_address. Expected anything other than 0x%llx since "
193*e3723e1fSApple OSS Distributions "this value should be modified by rev4",
194*e3723e1fSApple OSS Distributions vm_info.min_address, CANARY);
195*e3723e1fSApple OSS Distributions
196*e3723e1fSApple OSS Distributions T_EXPECT_NE(vm_info.max_address, CANARY,
197*e3723e1fSApple OSS Distributions "task_info --rev4 call returned value 0x%llx for vm_info.max_address. Expected anything other than 0x%llx since "
198*e3723e1fSApple OSS Distributions "this value should be modified by rev4",
199*e3723e1fSApple OSS Distributions vm_info.max_address, CANARY);
200*e3723e1fSApple OSS Distributions
201*e3723e1fSApple OSS Distributions T_EXPECT_NE(vm_info.limit_bytes_remaining, CANARY,
202*e3723e1fSApple OSS Distributions "task_info --rev4 call returned value 0x%llx for vm_info.limit_bytes_remaining. Expected anything other than 0x%llx since "
203*e3723e1fSApple OSS Distributions "this value should be modified by rev4",
204*e3723e1fSApple OSS Distributions vm_info.limit_bytes_remaining, CANARY);
205*e3723e1fSApple OSS Distributions }
206*e3723e1fSApple OSS Distributions
207*e3723e1fSApple OSS Distributions T_DECL(host_debug_info, "tests host debug info", T_META_ASROOT(true), T_META_LTEPHASE(LTE_POSTINIT))
208*e3723e1fSApple OSS Distributions {
209*e3723e1fSApple OSS Distributions T_SETUPBEGIN;
210*e3723e1fSApple OSS Distributions int is_dev = is_development_kernel();
211*e3723e1fSApple OSS Distributions T_QUIET;
212*e3723e1fSApple OSS Distributions T_ASSERT_TRUE(is_dev, "verify development kernel is running");
213*e3723e1fSApple OSS Distributions T_SETUPEND;
214*e3723e1fSApple OSS Distributions
215*e3723e1fSApple OSS Distributions kern_return_t err;
216*e3723e1fSApple OSS Distributions mach_port_t host;
217*e3723e1fSApple OSS Distributions host_debug_info_internal_data_t debug_info;
218*e3723e1fSApple OSS Distributions mach_msg_type_number_t count = HOST_DEBUG_INFO_INTERNAL_COUNT;
219*e3723e1fSApple OSS Distributions host = mach_host_self();
220*e3723e1fSApple OSS Distributions err = host_info(host, HOST_DEBUG_INFO_INTERNAL, (host_info_t)&debug_info, &count);
221*e3723e1fSApple OSS Distributions
222*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify host_info call succeeded");
223*e3723e1fSApple OSS Distributions }
224*e3723e1fSApple OSS Distributions
225*e3723e1fSApple OSS Distributions T_DECL(task_debug_info, "tests task debug info", T_META_ASROOT(true), T_META_LTEPHASE(LTE_POSTINIT))
226*e3723e1fSApple OSS Distributions {
227*e3723e1fSApple OSS Distributions T_SETUPBEGIN;
228*e3723e1fSApple OSS Distributions int is_dev = is_development_kernel();
229*e3723e1fSApple OSS Distributions T_QUIET;
230*e3723e1fSApple OSS Distributions T_ASSERT_TRUE(is_dev, "verify development kernel is running");
231*e3723e1fSApple OSS Distributions T_SETUPEND;
232*e3723e1fSApple OSS Distributions
233*e3723e1fSApple OSS Distributions kern_return_t err;
234*e3723e1fSApple OSS Distributions task_debug_info_internal_data_t debug_info;
235*e3723e1fSApple OSS Distributions
236*e3723e1fSApple OSS Distributions mach_msg_type_number_t count = TASK_DEBUG_INFO_INTERNAL_COUNT;
237*e3723e1fSApple OSS Distributions
238*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_DEBUG_INFO_INTERNAL, (task_info_t)&debug_info, &count);
239*e3723e1fSApple OSS Distributions
240*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
241*e3723e1fSApple OSS Distributions }
242*e3723e1fSApple OSS Distributions
243*e3723e1fSApple OSS Distributions T_DECL(thread_debug_info, "tests thread debug info", T_META_ASROOT(true), T_META_LTEPHASE(LTE_POSTINIT))
244*e3723e1fSApple OSS Distributions {
245*e3723e1fSApple OSS Distributions T_SETUPBEGIN;
246*e3723e1fSApple OSS Distributions int is_dev = is_development_kernel();
247*e3723e1fSApple OSS Distributions T_QUIET;
248*e3723e1fSApple OSS Distributions T_ASSERT_TRUE(is_dev, "verify development kernel is running");
249*e3723e1fSApple OSS Distributions T_SETUPEND;
250*e3723e1fSApple OSS Distributions
251*e3723e1fSApple OSS Distributions kern_return_t err;
252*e3723e1fSApple OSS Distributions thread_debug_info_internal_data_t debug_info;
253*e3723e1fSApple OSS Distributions
254*e3723e1fSApple OSS Distributions mach_msg_type_number_t count = THREAD_DEBUG_INFO_INTERNAL_COUNT;
255*e3723e1fSApple OSS Distributions
256*e3723e1fSApple OSS Distributions err = thread_info(mach_thread_self(), THREAD_DEBUG_INFO_INTERNAL, (thread_info_t)&debug_info, &count);
257*e3723e1fSApple OSS Distributions
258*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
259*e3723e1fSApple OSS Distributions }
260*e3723e1fSApple OSS Distributions
261*e3723e1fSApple OSS Distributions static void
do_factorial_task()262*e3723e1fSApple OSS Distributions do_factorial_task()
263*e3723e1fSApple OSS Distributions {
264*e3723e1fSApple OSS Distributions int number = 20;
265*e3723e1fSApple OSS Distributions int factorial = 1;
266*e3723e1fSApple OSS Distributions int i;
267*e3723e1fSApple OSS Distributions for (i = 1; i <= number; i++) {
268*e3723e1fSApple OSS Distributions factorial *= i;
269*e3723e1fSApple OSS Distributions }
270*e3723e1fSApple OSS Distributions
271*e3723e1fSApple OSS Distributions return;
272*e3723e1fSApple OSS Distributions }
273*e3723e1fSApple OSS Distributions
274*e3723e1fSApple OSS Distributions T_DECL(task_thread_times_info, "tests task thread times info", T_META_ASROOT(true), T_META_LTEPHASE(LTE_POSTINIT))
275*e3723e1fSApple OSS Distributions {
276*e3723e1fSApple OSS Distributions T_SETUPBEGIN;
277*e3723e1fSApple OSS Distributions int is_dev = is_development_kernel();
278*e3723e1fSApple OSS Distributions T_QUIET;
279*e3723e1fSApple OSS Distributions T_ASSERT_TRUE(is_dev, "verify development kernel is running");
280*e3723e1fSApple OSS Distributions T_SETUPEND;
281*e3723e1fSApple OSS Distributions
282*e3723e1fSApple OSS Distributions kern_return_t err;
283*e3723e1fSApple OSS Distributions task_thread_times_info_data_t thread_times_info_data;
284*e3723e1fSApple OSS Distributions task_thread_times_info_data_t thread_times_info_data_new;
285*e3723e1fSApple OSS Distributions mach_msg_type_number_t count = TASK_THREAD_TIMES_INFO_COUNT;
286*e3723e1fSApple OSS Distributions
287*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_THREAD_TIMES_INFO, (task_info_t)&thread_times_info_data, &count);
288*e3723e1fSApple OSS Distributions
289*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
290*e3723e1fSApple OSS Distributions
291*e3723e1fSApple OSS Distributions do_factorial_task();
292*e3723e1fSApple OSS Distributions
293*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_THREAD_TIMES_INFO, (task_info_t)&thread_times_info_data_new, &count);
294*e3723e1fSApple OSS Distributions
295*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
296*e3723e1fSApple OSS Distributions
297*e3723e1fSApple OSS Distributions /*
298*e3723e1fSApple OSS Distributions * The difference is observed to be less than 30 microseconds for user_time
299*e3723e1fSApple OSS Distributions * and less than 50 microseconds for system_time. This observation was done for over
300*e3723e1fSApple OSS Distributions * 1000 runs.
301*e3723e1fSApple OSS Distributions */
302*e3723e1fSApple OSS Distributions
303*e3723e1fSApple OSS Distributions T_EXPECT_FALSE((thread_times_info_data_new.user_time.seconds - thread_times_info_data.user_time.seconds) != 0 ||
304*e3723e1fSApple OSS Distributions (thread_times_info_data_new.system_time.seconds - thread_times_info_data.system_time.seconds) != 0,
305*e3723e1fSApple OSS Distributions "Tests whether the difference between thread times is greater than the allowed limit");
306*e3723e1fSApple OSS Distributions
307*e3723e1fSApple OSS Distributions /*
308*e3723e1fSApple OSS Distributions * This is a negative case.
309*e3723e1fSApple OSS Distributions */
310*e3723e1fSApple OSS Distributions
311*e3723e1fSApple OSS Distributions count--;
312*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_THREAD_TIMES_INFO, (task_info_t)&thread_times_info_data, &count);
313*e3723e1fSApple OSS Distributions T_ASSERT_MACH_ERROR(err, KERN_INVALID_ARGUMENT,
314*e3723e1fSApple OSS Distributions "Negative test case: task_info should verify that count is at least equal to what is defined in API.");
315*e3723e1fSApple OSS Distributions }
316*e3723e1fSApple OSS Distributions
317*e3723e1fSApple OSS Distributions T_DECL(task_absolutetime_info, "tests task absolute time info", T_META_ASROOT(true), T_META_LTEPHASE(LTE_POSTINIT))
318*e3723e1fSApple OSS Distributions {
319*e3723e1fSApple OSS Distributions T_SETUPBEGIN;
320*e3723e1fSApple OSS Distributions int is_dev = is_development_kernel();
321*e3723e1fSApple OSS Distributions T_QUIET;
322*e3723e1fSApple OSS Distributions T_ASSERT_TRUE(is_dev, "verify development kernel is running");
323*e3723e1fSApple OSS Distributions T_SETUPEND;
324*e3723e1fSApple OSS Distributions
325*e3723e1fSApple OSS Distributions kern_return_t err;
326*e3723e1fSApple OSS Distributions uint64_t user_time_diff, system_time_diff;
327*e3723e1fSApple OSS Distributions task_absolutetime_info_data_t absolute_time_info_data;
328*e3723e1fSApple OSS Distributions task_absolutetime_info_data_t absolute_time_info_data_new;
329*e3723e1fSApple OSS Distributions mach_msg_type_number_t count = TASK_ABSOLUTETIME_INFO_COUNT;
330*e3723e1fSApple OSS Distributions
331*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_ABSOLUTETIME_INFO, (task_info_t)&absolute_time_info_data, &count);
332*e3723e1fSApple OSS Distributions
333*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
334*e3723e1fSApple OSS Distributions
335*e3723e1fSApple OSS Distributions do_factorial_task();
336*e3723e1fSApple OSS Distributions
337*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_ABSOLUTETIME_INFO, (task_info_t)&absolute_time_info_data_new, &count);
338*e3723e1fSApple OSS Distributions
339*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
340*e3723e1fSApple OSS Distributions
341*e3723e1fSApple OSS Distributions user_time_diff = absolute_time_info_data_new.total_user - absolute_time_info_data.total_user;
342*e3723e1fSApple OSS Distributions system_time_diff = absolute_time_info_data_new.total_system - absolute_time_info_data.total_system;
343*e3723e1fSApple OSS Distributions
344*e3723e1fSApple OSS Distributions #if !defined(__arm64__)
345*e3723e1fSApple OSS Distributions /*
346*e3723e1fSApple OSS Distributions * On embedded devices the difference is always zero.
347*e3723e1fSApple OSS Distributions * On non-embedded devices the difference occurs in this range. This was observed over ~10000 runs.
348*e3723e1fSApple OSS Distributions */
349*e3723e1fSApple OSS Distributions
350*e3723e1fSApple OSS Distributions T_EXPECT_FALSE(user_time_diff < ABSOLUTE_MIN_USER_TIME_DIFF || system_time_diff < ABSOLUTE_MIN_SYSTEM_TIME_DIFF,
351*e3723e1fSApple OSS Distributions "Tests whether the difference between thread times is greater than the expected range");
352*e3723e1fSApple OSS Distributions #endif
353*e3723e1fSApple OSS Distributions
354*e3723e1fSApple OSS Distributions if (absolute_time_info_data.threads_user <= 0) {
355*e3723e1fSApple OSS Distributions int precise_time_val = 0;
356*e3723e1fSApple OSS Distributions size_t len = sizeof(size_t);
357*e3723e1fSApple OSS Distributions
358*e3723e1fSApple OSS Distributions T_LOG("User threads time is zero. This should only happen rarely and when precise_user_time is off");
359*e3723e1fSApple OSS Distributions
360*e3723e1fSApple OSS Distributions err = sysctlbyname("kern.precise_user_kernel_time", &precise_time_val, &len, NULL, 0);
361*e3723e1fSApple OSS Distributions
362*e3723e1fSApple OSS Distributions T_EXPECT_POSIX_SUCCESS(err, "performing sysctl to check precise_user_time");
363*e3723e1fSApple OSS Distributions
364*e3723e1fSApple OSS Distributions T_LOG("kern.precise_user_kernel_time val = %d", precise_time_val);
365*e3723e1fSApple OSS Distributions
366*e3723e1fSApple OSS Distributions T_EXPECT_FALSE(precise_time_val, "user thread time should only be zero when precise_user_kernel_time is disabled");
367*e3723e1fSApple OSS Distributions } else {
368*e3723e1fSApple OSS Distributions T_PASS("task_info should return non-zero value for user threads time = %llu", absolute_time_info_data.threads_user);
369*e3723e1fSApple OSS Distributions }
370*e3723e1fSApple OSS Distributions
371*e3723e1fSApple OSS Distributions #if !defined(__arm64__)
372*e3723e1fSApple OSS Distributions /*
373*e3723e1fSApple OSS Distributions * On iOS, system threads are always zero. On OS X this value can be some large positive number.
374*e3723e1fSApple OSS Distributions * There is no real way to estimate the exact amount.
375*e3723e1fSApple OSS Distributions */
376*e3723e1fSApple OSS Distributions T_EXPECT_NE(absolute_time_info_data.threads_system, 0ULL,
377*e3723e1fSApple OSS Distributions "task_info should return non-zero value for system threads time = %llu", absolute_time_info_data.threads_system);
378*e3723e1fSApple OSS Distributions #endif
379*e3723e1fSApple OSS Distributions
380*e3723e1fSApple OSS Distributions /*
381*e3723e1fSApple OSS Distributions * This is a negative case.
382*e3723e1fSApple OSS Distributions */
383*e3723e1fSApple OSS Distributions count--;
384*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_ABSOLUTETIME_INFO, (task_info_t)&absolute_time_info_data_new, &count);
385*e3723e1fSApple OSS Distributions T_ASSERT_MACH_ERROR(err, KERN_INVALID_ARGUMENT,
386*e3723e1fSApple OSS Distributions "Negative test case: task_info should verify that count is at least equal to what is defined in API.");
387*e3723e1fSApple OSS Distributions }
388*e3723e1fSApple OSS Distributions
389*e3723e1fSApple OSS Distributions T_DECL(task_affinity_tag_info, "tests task_affinity_tag_info", T_META_ASROOT(true), T_META_LTEPHASE(LTE_POSTINIT))
390*e3723e1fSApple OSS Distributions {
391*e3723e1fSApple OSS Distributions T_SETUPBEGIN;
392*e3723e1fSApple OSS Distributions int is_dev = is_development_kernel();
393*e3723e1fSApple OSS Distributions T_QUIET;
394*e3723e1fSApple OSS Distributions T_ASSERT_TRUE(is_dev, "verify development kernel is running");
395*e3723e1fSApple OSS Distributions T_SETUPEND;
396*e3723e1fSApple OSS Distributions
397*e3723e1fSApple OSS Distributions kern_return_t err;
398*e3723e1fSApple OSS Distributions task_affinity_tag_info_data_t affinity_tag_info_data;
399*e3723e1fSApple OSS Distributions mach_msg_type_number_t count = TASK_AFFINITY_TAG_INFO_COUNT;
400*e3723e1fSApple OSS Distributions
401*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_AFFINITY_TAG_INFO, (task_info_t)&affinity_tag_info_data, &count);
402*e3723e1fSApple OSS Distributions
403*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
404*e3723e1fSApple OSS Distributions
405*e3723e1fSApple OSS Distributions /*
406*e3723e1fSApple OSS Distributions * The affinity is not set by default, hence expecting a zero value.
407*e3723e1fSApple OSS Distributions */
408*e3723e1fSApple OSS Distributions T_ASSERT_FALSE(affinity_tag_info_data.min != 0 || affinity_tag_info_data.max != 0,
409*e3723e1fSApple OSS Distributions "task_info call returns non-zero min or max value");
410*e3723e1fSApple OSS Distributions
411*e3723e1fSApple OSS Distributions /*
412*e3723e1fSApple OSS Distributions * This is a negative case.
413*e3723e1fSApple OSS Distributions */
414*e3723e1fSApple OSS Distributions count--;
415*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_AFFINITY_TAG_INFO, (task_info_t)&affinity_tag_info_data, &count);
416*e3723e1fSApple OSS Distributions T_ASSERT_MACH_ERROR(err, KERN_INVALID_ARGUMENT,
417*e3723e1fSApple OSS Distributions "Negative test case: task_info should verify that count is at least equal to what is defined in API.");
418*e3723e1fSApple OSS Distributions }
419*e3723e1fSApple OSS Distributions
420*e3723e1fSApple OSS Distributions T_DECL(task_flags_info, "tests task_flags_info", T_META_ASROOT(true), T_META_LTEPHASE(LTE_POSTINIT))
421*e3723e1fSApple OSS Distributions {
422*e3723e1fSApple OSS Distributions T_SETUPBEGIN;
423*e3723e1fSApple OSS Distributions int is_dev = is_development_kernel();
424*e3723e1fSApple OSS Distributions T_QUIET;
425*e3723e1fSApple OSS Distributions T_ASSERT_TRUE(is_dev, "verify development kernel is running");
426*e3723e1fSApple OSS Distributions T_SETUPEND;
427*e3723e1fSApple OSS Distributions
428*e3723e1fSApple OSS Distributions kern_return_t err;
429*e3723e1fSApple OSS Distributions task_flags_info_data_t flags_info_data;
430*e3723e1fSApple OSS Distributions mach_msg_type_number_t count = TASK_FLAGS_INFO_COUNT;
431*e3723e1fSApple OSS Distributions
432*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_FLAGS_INFO, (task_info_t)&flags_info_data, &count);
433*e3723e1fSApple OSS Distributions
434*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
435*e3723e1fSApple OSS Distributions
436*e3723e1fSApple OSS Distributions /* Change for 32-bit arch possibility?*/
437*e3723e1fSApple OSS Distributions T_ASSERT_EQ((flags_info_data.flags & (unsigned int)(~(TF_LP64 | TF_64B_DATA))), 0U,
438*e3723e1fSApple OSS Distributions "task_info should only give out 64-bit addr/data flags");
439*e3723e1fSApple OSS Distributions
440*e3723e1fSApple OSS Distributions /*
441*e3723e1fSApple OSS Distributions * This is a negative case.
442*e3723e1fSApple OSS Distributions */
443*e3723e1fSApple OSS Distributions
444*e3723e1fSApple OSS Distributions count--;
445*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_FLAGS_INFO, (task_info_t)&flags_info_data, &count);
446*e3723e1fSApple OSS Distributions T_ASSERT_MACH_ERROR(err, KERN_INVALID_ARGUMENT,
447*e3723e1fSApple OSS Distributions "Negative test case: task_info should verify that count is at least equal to what is defined in API.");
448*e3723e1fSApple OSS Distributions }
449*e3723e1fSApple OSS Distributions
450*e3723e1fSApple OSS Distributions T_DECL(task_power_info_v2, "tests task_power_info_v2", T_META_ASROOT(true), T_META_LTEPHASE(LTE_POSTINIT),
451*e3723e1fSApple OSS Distributions T_META_TAG_VM_NOT_ELIGIBLE)
452*e3723e1fSApple OSS Distributions {
453*e3723e1fSApple OSS Distributions T_SETUPBEGIN;
454*e3723e1fSApple OSS Distributions int is_dev = is_development_kernel();
455*e3723e1fSApple OSS Distributions T_QUIET;
456*e3723e1fSApple OSS Distributions T_ASSERT_TRUE(is_dev, "verify development kernel is running");
457*e3723e1fSApple OSS Distributions T_SETUPEND;
458*e3723e1fSApple OSS Distributions
459*e3723e1fSApple OSS Distributions kern_return_t err;
460*e3723e1fSApple OSS Distributions task_power_info_v2_data_t power_info_data_v2;
461*e3723e1fSApple OSS Distributions task_power_info_v2_data_t power_info_data_v2_new;
462*e3723e1fSApple OSS Distributions mach_msg_type_number_t count = TASK_POWER_INFO_V2_COUNT;
463*e3723e1fSApple OSS Distributions
464*e3723e1fSApple OSS Distributions sleep(1);
465*e3723e1fSApple OSS Distributions
466*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_POWER_INFO_V2, (task_info_t)&power_info_data_v2, &count);
467*e3723e1fSApple OSS Distributions
468*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
469*e3723e1fSApple OSS Distributions
470*e3723e1fSApple OSS Distributions T_ASSERT_LE(power_info_data_v2.gpu_energy.task_gpu_utilisation, 0ULL,
471*e3723e1fSApple OSS Distributions "verified task_info call shows zero GPU utilization for non-GPU task");
472*e3723e1fSApple OSS Distributions
473*e3723e1fSApple OSS Distributions do_factorial_task();
474*e3723e1fSApple OSS Distributions
475*e3723e1fSApple OSS Distributions /*
476*e3723e1fSApple OSS Distributions * Verify the cpu_energy parameters.
477*e3723e1fSApple OSS Distributions */
478*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_POWER_INFO_V2, (task_info_t)&power_info_data_v2_new, &count);
479*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
480*e3723e1fSApple OSS Distributions
481*e3723e1fSApple OSS Distributions #if !defined(__arm64__)
482*e3723e1fSApple OSS Distributions /*
483*e3723e1fSApple OSS Distributions * iOS does not have system_time.
484*e3723e1fSApple OSS Distributions */
485*e3723e1fSApple OSS Distributions T_ASSERT_GT(power_info_data_v2_new.cpu_energy.total_user, power_info_data_v2.cpu_energy.total_user,
486*e3723e1fSApple OSS Distributions "task_info call returns valid user time");
487*e3723e1fSApple OSS Distributions T_ASSERT_GT(power_info_data_v2_new.cpu_energy.total_system, power_info_data_v2.cpu_energy.total_system,
488*e3723e1fSApple OSS Distributions "task_info call returns valid system time");
489*e3723e1fSApple OSS Distributions #endif
490*e3723e1fSApple OSS Distributions
491*e3723e1fSApple OSS Distributions T_ASSERT_GE(power_info_data_v2.cpu_energy.task_interrupt_wakeups, 1ULL,
492*e3723e1fSApple OSS Distributions "verify task_info call returns non-zero value for interrupt_wakeup (ret value = %llu)",
493*e3723e1fSApple OSS Distributions power_info_data_v2.cpu_energy.task_interrupt_wakeups);
494*e3723e1fSApple OSS Distributions
495*e3723e1fSApple OSS Distributions #if !defined(__arm64__)
496*e3723e1fSApple OSS Distributions if (power_info_data_v2.cpu_energy.task_platform_idle_wakeups != 0) {
497*e3723e1fSApple OSS Distributions T_LOG("task_info call returned %llu for platform_idle_wakeup", power_info_data_v2.cpu_energy.task_platform_idle_wakeups);
498*e3723e1fSApple OSS Distributions }
499*e3723e1fSApple OSS Distributions #endif
500*e3723e1fSApple OSS Distributions
501*e3723e1fSApple OSS Distributions count = TASK_POWER_INFO_V2_COUNT_OLD;
502*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_POWER_INFO_V2, (task_info_t)&power_info_data_v2, &count);
503*e3723e1fSApple OSS Distributions
504*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(err, "verify task_info call succeeded");
505*e3723e1fSApple OSS Distributions
506*e3723e1fSApple OSS Distributions /*
507*e3723e1fSApple OSS Distributions * This is a negative case.
508*e3723e1fSApple OSS Distributions */
509*e3723e1fSApple OSS Distributions count--;
510*e3723e1fSApple OSS Distributions err = task_info(mach_task_self(), TASK_POWER_INFO_V2, (task_info_t)&power_info_data_v2, &count);
511*e3723e1fSApple OSS Distributions
512*e3723e1fSApple OSS Distributions T_ASSERT_MACH_ERROR(err, KERN_INVALID_ARGUMENT,
513*e3723e1fSApple OSS Distributions "Negative test case: task_info should verify that count is at least equal to what is defined in API. Call "
514*e3723e1fSApple OSS Distributions "returns errno %d:%s",
515*e3723e1fSApple OSS Distributions err, mach_error_string(err));
516*e3723e1fSApple OSS Distributions }
517*e3723e1fSApple OSS Distributions
518*e3723e1fSApple OSS Distributions T_DECL(test_task_basic_info_32, "tests TASK_BASIC_INFO_32", T_META_ASROOT(true), T_META_LTEPHASE(LTE_POSTINIT),
519*e3723e1fSApple OSS Distributions T_META_TAG_VM_PREFERRED)
520*e3723e1fSApple OSS Distributions {
521*e3723e1fSApple OSS Distributions test_task_basic_info(INFO_32);
522*e3723e1fSApple OSS Distributions }
523*e3723e1fSApple OSS Distributions
524*e3723e1fSApple OSS Distributions T_DECL(test_task_basic_info_32_2, "tests TASK_BASIC_INFO_32_2", T_META_ASROOT(true), T_META_LTEPHASE(LTE_POSTINIT),
525*e3723e1fSApple OSS Distributions T_META_TAG_VM_PREFERRED)
526*e3723e1fSApple OSS Distributions {
527*e3723e1fSApple OSS Distributions test_task_basic_info(INFO_32_2);
528*e3723e1fSApple OSS Distributions }
529*e3723e1fSApple OSS Distributions
530*e3723e1fSApple OSS Distributions #if defined(__arm64__)
531*e3723e1fSApple OSS Distributions T_DECL(test_task_basic_info_64i_2, "tests TASK_BASIC_INFO_64_2", T_META_ASROOT(true), T_META_LTEPHASE(LTE_POSTINIT),
532*e3723e1fSApple OSS Distributions T_META_TAG_VM_PREFERRED)
533*e3723e1fSApple OSS Distributions {
534*e3723e1fSApple OSS Distributions test_task_basic_info(INFO_64_2);
535*e3723e1fSApple OSS Distributions }
536*e3723e1fSApple OSS Distributions #else
537*e3723e1fSApple OSS Distributions T_DECL(test_task_basic_info_64, "tests TASK_BASIC_INFO_64", T_META_ASROOT(true), T_META_LTEPHASE(LTE_POSTINIT))
538*e3723e1fSApple OSS Distributions {
539*e3723e1fSApple OSS Distributions test_task_basic_info(INFO_64);
540*e3723e1fSApple OSS Distributions }
541*e3723e1fSApple OSS Distributions #endif /* defined(__arm64__) */
542*e3723e1fSApple OSS Distributions
543*e3723e1fSApple OSS Distributions T_DECL(test_mach_task_basic_info, "tests MACH_TASK_BASIC_INFO", T_META_ASROOT(true), T_META_LTEPHASE(LTE_POSTINIT))
544*e3723e1fSApple OSS Distributions {
545*e3723e1fSApple OSS Distributions test_task_basic_info(INFO_MACH);
546*e3723e1fSApple OSS Distributions }
547*e3723e1fSApple OSS Distributions
548*e3723e1fSApple OSS Distributions void
test_task_basic_info(enum info_kind kind)549*e3723e1fSApple OSS Distributions test_task_basic_info(enum info_kind kind)
550*e3723e1fSApple OSS Distributions {
551*e3723e1fSApple OSS Distributions #define BEFORE 0
552*e3723e1fSApple OSS Distributions #define AFTER 1
553*e3723e1fSApple OSS Distributions
554*e3723e1fSApple OSS Distributions T_SETUPBEGIN;
555*e3723e1fSApple OSS Distributions int is_dev = is_development_kernel();
556*e3723e1fSApple OSS Distributions T_QUIET;
557*e3723e1fSApple OSS Distributions T_ASSERT_TRUE(is_dev, "verify development kernel is running");
558*e3723e1fSApple OSS Distributions T_SETUPEND;
559*e3723e1fSApple OSS Distributions
560*e3723e1fSApple OSS Distributions task_info_t info_data[2];
561*e3723e1fSApple OSS Distributions task_basic_info_32_data_t basic_info_32_data[2];
562*e3723e1fSApple OSS Distributions #if defined(__arm64__)
563*e3723e1fSApple OSS Distributions task_basic_info_64_2_data_t basic_info_64_2_data[2];
564*e3723e1fSApple OSS Distributions #else
565*e3723e1fSApple OSS Distributions task_basic_info_64_data_t basic_info_64_data[2];
566*e3723e1fSApple OSS Distributions #endif /* defined(__arm64__) */
567*e3723e1fSApple OSS Distributions mach_task_basic_info_data_t mach_basic_info_data[2];
568*e3723e1fSApple OSS Distributions
569*e3723e1fSApple OSS Distributions kern_return_t kr;
570*e3723e1fSApple OSS Distributions mach_msg_type_number_t count;
571*e3723e1fSApple OSS Distributions task_flavor_t flavor = 0;
572*e3723e1fSApple OSS Distributions integer_t suspend_count;
573*e3723e1fSApple OSS Distributions uint64_t resident_size_diff;
574*e3723e1fSApple OSS Distributions uint64_t virtual_size_diff;
575*e3723e1fSApple OSS Distributions
576*e3723e1fSApple OSS Distributions void * tmp_map = NULL;
577*e3723e1fSApple OSS Distributions pid_t child_pid;
578*e3723e1fSApple OSS Distributions mach_port_name_t child_task;
579*e3723e1fSApple OSS Distributions /*for dt_waitpid*/
580*e3723e1fSApple OSS Distributions int timeout = 10; // change to max timeout
581*e3723e1fSApple OSS Distributions int exit_status = 0;
582*e3723e1fSApple OSS Distributions
583*e3723e1fSApple OSS Distributions switch (kind) {
584*e3723e1fSApple OSS Distributions case INFO_32:
585*e3723e1fSApple OSS Distributions case INFO_32_2:
586*e3723e1fSApple OSS Distributions info_data[BEFORE] = (task_info_t)&basic_info_32_data[BEFORE];
587*e3723e1fSApple OSS Distributions info_data[AFTER] = (task_info_t)&basic_info_32_data[AFTER];
588*e3723e1fSApple OSS Distributions count = TASK_BASIC_INFO_32_COUNT;
589*e3723e1fSApple OSS Distributions flavor = TASK_BASIC_INFO_32;
590*e3723e1fSApple OSS Distributions
591*e3723e1fSApple OSS Distributions if (kind == INFO_32_2) {
592*e3723e1fSApple OSS Distributions flavor = TASK_BASIC2_INFO_32;
593*e3723e1fSApple OSS Distributions }
594*e3723e1fSApple OSS Distributions
595*e3723e1fSApple OSS Distributions break;
596*e3723e1fSApple OSS Distributions #if defined(__arm64__)
597*e3723e1fSApple OSS Distributions case INFO_64:
598*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("invalid basic info kind");
599*e3723e1fSApple OSS Distributions break;
600*e3723e1fSApple OSS Distributions
601*e3723e1fSApple OSS Distributions case INFO_64_2:
602*e3723e1fSApple OSS Distributions info_data[BEFORE] = (task_info_t)&basic_info_64_2_data[BEFORE];
603*e3723e1fSApple OSS Distributions info_data[AFTER] = (task_info_t)&basic_info_64_2_data[AFTER];
604*e3723e1fSApple OSS Distributions count = TASK_BASIC_INFO_64_2_COUNT;
605*e3723e1fSApple OSS Distributions flavor = TASK_BASIC_INFO_64_2;
606*e3723e1fSApple OSS Distributions break;
607*e3723e1fSApple OSS Distributions
608*e3723e1fSApple OSS Distributions #else
609*e3723e1fSApple OSS Distributions case INFO_64:
610*e3723e1fSApple OSS Distributions info_data[BEFORE] = (task_info_t)&basic_info_64_data[BEFORE];
611*e3723e1fSApple OSS Distributions info_data[AFTER] = (task_info_t)&basic_info_64_data[AFTER];
612*e3723e1fSApple OSS Distributions count = TASK_BASIC_INFO_64_COUNT;
613*e3723e1fSApple OSS Distributions flavor = TASK_BASIC_INFO_64;
614*e3723e1fSApple OSS Distributions break;
615*e3723e1fSApple OSS Distributions
616*e3723e1fSApple OSS Distributions case INFO_64_2:
617*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("invalid basic info kind");
618*e3723e1fSApple OSS Distributions break;
619*e3723e1fSApple OSS Distributions #endif /* defined(__arm64__) */
620*e3723e1fSApple OSS Distributions case INFO_MACH:
621*e3723e1fSApple OSS Distributions info_data[BEFORE] = (task_info_t)&mach_basic_info_data[BEFORE];
622*e3723e1fSApple OSS Distributions info_data[AFTER] = (task_info_t)&mach_basic_info_data[AFTER];
623*e3723e1fSApple OSS Distributions count = MACH_TASK_BASIC_INFO_COUNT;
624*e3723e1fSApple OSS Distributions flavor = MACH_TASK_BASIC_INFO;
625*e3723e1fSApple OSS Distributions break;
626*e3723e1fSApple OSS Distributions case INFO_MAX:
627*e3723e1fSApple OSS Distributions default:
628*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("invalid basic info kind");
629*e3723e1fSApple OSS Distributions break;
630*e3723e1fSApple OSS Distributions }
631*e3723e1fSApple OSS Distributions
632*e3723e1fSApple OSS Distributions kr = task_info(mach_task_self(), flavor, info_data[BEFORE], &count);
633*e3723e1fSApple OSS Distributions
634*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "verify task_info succeeded");
635*e3723e1fSApple OSS Distributions
636*e3723e1fSApple OSS Distributions do_factorial_task();
637*e3723e1fSApple OSS Distributions
638*e3723e1fSApple OSS Distributions /*
639*e3723e1fSApple OSS Distributions * Allocate virtual and resident memory.
640*e3723e1fSApple OSS Distributions */
641*e3723e1fSApple OSS Distributions tmp_map = mmap(0, PAGE_SIZE, PROT_WRITE, MAP_ANON | MAP_PRIVATE, -1, 0);
642*e3723e1fSApple OSS Distributions
643*e3723e1fSApple OSS Distributions T_WITH_ERRNO;
644*e3723e1fSApple OSS Distributions T_EXPECT_NE(tmp_map, MAP_FAILED, "verify mmap call is successful");
645*e3723e1fSApple OSS Distributions
646*e3723e1fSApple OSS Distributions memset(tmp_map, 'm', PAGE_SIZE);
647*e3723e1fSApple OSS Distributions
648*e3723e1fSApple OSS Distributions child_pid = fork();
649*e3723e1fSApple OSS Distributions
650*e3723e1fSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(child_pid, "verify process can be forked");
651*e3723e1fSApple OSS Distributions
652*e3723e1fSApple OSS Distributions if (child_pid == 0) {
653*e3723e1fSApple OSS Distributions /*
654*e3723e1fSApple OSS Distributions * This will suspend the child process.
655*e3723e1fSApple OSS Distributions */
656*e3723e1fSApple OSS Distributions kr = task_suspend(mach_task_self());
657*e3723e1fSApple OSS Distributions exit(kr);
658*e3723e1fSApple OSS Distributions }
659*e3723e1fSApple OSS Distributions
660*e3723e1fSApple OSS Distributions /*
661*e3723e1fSApple OSS Distributions * Wait for the child process to suspend itself.
662*e3723e1fSApple OSS Distributions */
663*e3723e1fSApple OSS Distributions sleep(1);
664*e3723e1fSApple OSS Distributions
665*e3723e1fSApple OSS Distributions kr = task_for_pid(mach_task_self(), child_pid, &child_task);
666*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "verify task_for_pid succeeded. check sudo if failed");
667*e3723e1fSApple OSS Distributions
668*e3723e1fSApple OSS Distributions /*
669*e3723e1fSApple OSS Distributions * Verify the suspend_count for child and resume it.
670*e3723e1fSApple OSS Distributions */
671*e3723e1fSApple OSS Distributions
672*e3723e1fSApple OSS Distributions kr = task_info(child_task, flavor, info_data[AFTER], &count);
673*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "verify task_info call succeeded");
674*e3723e1fSApple OSS Distributions
675*e3723e1fSApple OSS Distributions suspend_count = (integer_t)(info_get(kind, GET_SUSPEND_COUNT, info_data[AFTER]));
676*e3723e1fSApple OSS Distributions T_ASSERT_EQ(suspend_count, 1, "verify task_info shows correct suspend_count");
677*e3723e1fSApple OSS Distributions
678*e3723e1fSApple OSS Distributions kr = task_resume(child_task);
679*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "verify task_resume succeeded");
680*e3723e1fSApple OSS Distributions
681*e3723e1fSApple OSS Distributions /*
682*e3723e1fSApple OSS Distributions * reap kr from task_suspend call in child
683*e3723e1fSApple OSS Distributions */
684*e3723e1fSApple OSS Distributions if (dt_waitpid(child_pid, &exit_status, NULL, timeout)) {
685*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(exit_status, "verify child task_suspend is successful");
686*e3723e1fSApple OSS Distributions } else {
687*e3723e1fSApple OSS Distributions T_FAIL("dt_waitpid failed");
688*e3723e1fSApple OSS Distributions }
689*e3723e1fSApple OSS Distributions
690*e3723e1fSApple OSS Distributions kr = task_info(mach_task_self(), flavor, info_data[AFTER], &count);
691*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "verify task_info call succeeded");
692*e3723e1fSApple OSS Distributions
693*e3723e1fSApple OSS Distributions resident_size_diff = info_get(kind, GET_RESIDENT_SIZE, info_data[AFTER]) - info_get(kind, GET_RESIDENT_SIZE, info_data[BEFORE]);
694*e3723e1fSApple OSS Distributions virtual_size_diff = info_get(kind, GET_VIRTUAL_SIZE, info_data[AFTER]) - info_get(kind, GET_VIRTUAL_SIZE, info_data[BEFORE]);
695*e3723e1fSApple OSS Distributions
696*e3723e1fSApple OSS Distributions /*
697*e3723e1fSApple OSS Distributions * INFO_32_2 gets the max resident size instead of the current resident size
698*e3723e1fSApple OSS Distributions * 32 KB tolerance built into test. The returned value is generally between 0 and 16384
699*e3723e1fSApple OSS Distributions *
700*e3723e1fSApple OSS Distributions * max resident size is a discrete field in INFO_MACH, so it's handled differently
701*e3723e1fSApple OSS Distributions */
702*e3723e1fSApple OSS Distributions if (kind == INFO_32_2) {
703*e3723e1fSApple OSS Distributions T_EXPECT_EQ(resident_size_diff % 4096, 0ULL, "verify task_info returns valid max resident_size");
704*e3723e1fSApple OSS Distributions T_EXPECT_GE(resident_size_diff, 0ULL, "verify task_info returns non-negative max resident_size");
705*e3723e1fSApple OSS Distributions T_EXPECT_GE(virtual_size_diff, (unsigned long long)PAGE_SIZE, "verify task_info returns valid virtual_size");
706*e3723e1fSApple OSS Distributions } else {
707*e3723e1fSApple OSS Distributions T_EXPECT_GE(resident_size_diff, (unsigned long long)PAGE_SIZE, "task_info returns valid resident_size");
708*e3723e1fSApple OSS Distributions T_EXPECT_GE(virtual_size_diff, (unsigned long long)PAGE_SIZE, "task_info returns valid virtual_size");
709*e3723e1fSApple OSS Distributions }
710*e3723e1fSApple OSS Distributions
711*e3723e1fSApple OSS Distributions if (kind == INFO_MACH) {
712*e3723e1fSApple OSS Distributions resident_size_diff = info_get(kind, GET_MAX_RES, info_data[AFTER]) - info_get(kind, GET_MAX_RES, info_data[BEFORE]);
713*e3723e1fSApple OSS Distributions T_EXPECT_EQ(resident_size_diff % 4096, 0ULL, "verify task_info returns valid max resident_size");
714*e3723e1fSApple OSS Distributions T_EXPECT_GE(resident_size_diff, 0ULL, "verify task_info returns non-negative max resident_size");
715*e3723e1fSApple OSS Distributions T_EXPECT_GE(info_get(kind, GET_MAX_RES, info_data[AFTER]), info_get(kind, GET_RESIDENT_SIZE, info_data[AFTER]),
716*e3723e1fSApple OSS Distributions "verify max resident size is greater than or equal to curr resident size");
717*e3723e1fSApple OSS Distributions }
718*e3723e1fSApple OSS Distributions
719*e3723e1fSApple OSS Distributions do_factorial_task();
720*e3723e1fSApple OSS Distributions
721*e3723e1fSApple OSS Distributions /*
722*e3723e1fSApple OSS Distributions * These counters give time for threads that have terminated. We dont have any, so checking for zero.
723*e3723e1fSApple OSS Distributions */
724*e3723e1fSApple OSS Distributions
725*e3723e1fSApple OSS Distributions time_value_t * user_tv = (time_value_t *)(info_get(kind, GET_USER_TIME, info_data[BEFORE]));
726*e3723e1fSApple OSS Distributions T_EXPECT_EQ((user_tv->seconds + user_tv->microseconds / 1000000), 0, "verify task_info shows valid user time");
727*e3723e1fSApple OSS Distributions
728*e3723e1fSApple OSS Distributions time_value_t * sys_tv = (time_value_t *)(info_get(kind, GET_SYS_TIME, info_data[BEFORE]));
729*e3723e1fSApple OSS Distributions T_EXPECT_EQ(sys_tv->seconds + (sys_tv->microseconds / 1000000), 0, "verify task_info shows valid system time");
730*e3723e1fSApple OSS Distributions
731*e3723e1fSApple OSS Distributions /*
732*e3723e1fSApple OSS Distributions * The default value for non-kernel tasks is TIMESHARE.
733*e3723e1fSApple OSS Distributions */
734*e3723e1fSApple OSS Distributions
735*e3723e1fSApple OSS Distributions policy_t pt = (policy_t)info_get(kind, GET_POLICY, info_data[BEFORE]);
736*e3723e1fSApple OSS Distributions
737*e3723e1fSApple OSS Distributions T_EXPECT_EQ(pt, POLICY_TIMESHARE, "verify task_info shows valid policy");
738*e3723e1fSApple OSS Distributions
739*e3723e1fSApple OSS Distributions /*
740*e3723e1fSApple OSS Distributions * This is a negative case.
741*e3723e1fSApple OSS Distributions */
742*e3723e1fSApple OSS Distributions
743*e3723e1fSApple OSS Distributions count--;
744*e3723e1fSApple OSS Distributions kr = task_info(mach_task_self(), flavor, info_data[AFTER], &count);
745*e3723e1fSApple OSS Distributions
746*e3723e1fSApple OSS Distributions T_ASSERT_MACH_ERROR(kr, KERN_INVALID_ARGUMENT,
747*e3723e1fSApple OSS Distributions "Negative test case: task_info should verify that count is at least equal to what is defined in API");
748*e3723e1fSApple OSS Distributions
749*e3723e1fSApple OSS Distributions /*
750*e3723e1fSApple OSS Distributions * deallocate memory
751*e3723e1fSApple OSS Distributions */
752*e3723e1fSApple OSS Distributions munmap(tmp_map, PAGE_SIZE);
753*e3723e1fSApple OSS Distributions
754*e3723e1fSApple OSS Distributions return;
755*e3723e1fSApple OSS Distributions
756*e3723e1fSApple OSS Distributions #undef BEFORE
757*e3723e1fSApple OSS Distributions #undef AFTER
758*e3723e1fSApple OSS Distributions }
759*e3723e1fSApple OSS Distributions
760*e3723e1fSApple OSS Distributions T_DECL(test_sigcont_task_suspend_resume,
761*e3723e1fSApple OSS Distributions "test to verify that SIGCONT on task_suspend()-ed process works",
762*e3723e1fSApple OSS Distributions T_META_ASROOT(true),
763*e3723e1fSApple OSS Distributions T_META_LTEPHASE(LTE_POSTINIT))
764*e3723e1fSApple OSS Distributions {
765*e3723e1fSApple OSS Distributions T_SETUPBEGIN;
766*e3723e1fSApple OSS Distributions int is_dev = is_development_kernel();
767*e3723e1fSApple OSS Distributions T_QUIET;
768*e3723e1fSApple OSS Distributions T_ASSERT_TRUE(is_dev, "verify development kernel is running");
769*e3723e1fSApple OSS Distributions T_SETUPEND;
770*e3723e1fSApple OSS Distributions
771*e3723e1fSApple OSS Distributions mach_task_basic_info_data_t mach_basic_info_data;
772*e3723e1fSApple OSS Distributions task_info_t info_data = (task_info_t)&mach_basic_info_data;
773*e3723e1fSApple OSS Distributions
774*e3723e1fSApple OSS Distributions task_debug_info_internal_data_t debug_info;
775*e3723e1fSApple OSS Distributions mach_msg_type_number_t debug_count = TASK_DEBUG_INFO_INTERNAL_COUNT;
776*e3723e1fSApple OSS Distributions
777*e3723e1fSApple OSS Distributions kern_return_t kr;
778*e3723e1fSApple OSS Distributions int posix_ret;
779*e3723e1fSApple OSS Distributions mach_msg_type_number_t count = MACH_TASK_BASIC_INFO_COUNT;
780*e3723e1fSApple OSS Distributions task_flavor_t flavor = MACH_TASK_BASIC_INFO;
781*e3723e1fSApple OSS Distributions integer_t suspend_count;
782*e3723e1fSApple OSS Distributions integer_t debug_suspend_count;
783*e3723e1fSApple OSS Distributions pid_t child_pid = 0;
784*e3723e1fSApple OSS Distributions mach_port_name_t child_task;
785*e3723e1fSApple OSS Distributions /*for dt_waitpid*/
786*e3723e1fSApple OSS Distributions int timeout = 5;
787*e3723e1fSApple OSS Distributions int exit_status = 0;
788*e3723e1fSApple OSS Distributions int signal_no = 0;
789*e3723e1fSApple OSS Distributions
790*e3723e1fSApple OSS Distributions child_pid = fork();
791*e3723e1fSApple OSS Distributions
792*e3723e1fSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(child_pid, "verify process can be forked");
793*e3723e1fSApple OSS Distributions
794*e3723e1fSApple OSS Distributions if (child_pid == 0) {
795*e3723e1fSApple OSS Distributions /*
796*e3723e1fSApple OSS Distributions * This will suspend the child process.
797*e3723e1fSApple OSS Distributions */
798*e3723e1fSApple OSS Distributions kr = task_suspend(mach_task_self());
799*e3723e1fSApple OSS Distributions
800*e3723e1fSApple OSS Distributions /*
801*e3723e1fSApple OSS Distributions * When child resumes, it exits immediately
802*e3723e1fSApple OSS Distributions */
803*e3723e1fSApple OSS Distributions
804*e3723e1fSApple OSS Distributions exit(kr);
805*e3723e1fSApple OSS Distributions }
806*e3723e1fSApple OSS Distributions
807*e3723e1fSApple OSS Distributions /*
808*e3723e1fSApple OSS Distributions * Wait for the child process to suspend itself.
809*e3723e1fSApple OSS Distributions */
810*e3723e1fSApple OSS Distributions sleep(1);
811*e3723e1fSApple OSS Distributions
812*e3723e1fSApple OSS Distributions kr = task_for_pid(mach_task_self(), child_pid, &child_task);
813*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "verify task_for_pid succeeded. check sudo if failed");
814*e3723e1fSApple OSS Distributions
815*e3723e1fSApple OSS Distributions /*
816*e3723e1fSApple OSS Distributions * Verify the suspend_count for child and resume it.
817*e3723e1fSApple OSS Distributions */
818*e3723e1fSApple OSS Distributions
819*e3723e1fSApple OSS Distributions kr = task_info(child_task, flavor, info_data, &count);
820*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "verify task_info call succeeded");
821*e3723e1fSApple OSS Distributions
822*e3723e1fSApple OSS Distributions suspend_count = (integer_t)(info_get(INFO_MACH, GET_SUSPEND_COUNT, info_data));
823*e3723e1fSApple OSS Distributions T_ASSERT_EQ(suspend_count, 1, "verify task_info shows correct suspend_count (1) (actually user stop count) ");
824*e3723e1fSApple OSS Distributions
825*e3723e1fSApple OSS Distributions kr = task_info(child_task, TASK_DEBUG_INFO_INTERNAL, (task_info_t)&debug_info, &debug_count);
826*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "verify task_info call succeeded");
827*e3723e1fSApple OSS Distributions
828*e3723e1fSApple OSS Distributions debug_suspend_count = debug_info.suspend_count;
829*e3723e1fSApple OSS Distributions T_ASSERT_EQ(debug_info.suspend_count, 1, "verify debug_info shows correct suspend_count(1)");
830*e3723e1fSApple OSS Distributions
831*e3723e1fSApple OSS Distributions posix_ret = kill(child_pid, SIGCONT);
832*e3723e1fSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(posix_ret, "verify signal call succeeded");
833*e3723e1fSApple OSS Distributions
834*e3723e1fSApple OSS Distributions /*
835*e3723e1fSApple OSS Distributions * reap kr from task_suspend call in child
836*e3723e1fSApple OSS Distributions */
837*e3723e1fSApple OSS Distributions dt_waitpid(child_pid, &exit_status, &signal_no, timeout);
838*e3723e1fSApple OSS Distributions
839*e3723e1fSApple OSS Distributions T_ASSERT_EQ(signal_no, 0, "child should be resumed and exit without signal");
840*e3723e1fSApple OSS Distributions T_ASSERT_EQ(exit_status, 0, "child should exit with 0");
841*e3723e1fSApple OSS Distributions }
842*e3723e1fSApple OSS Distributions
843*e3723e1fSApple OSS Distributions T_DECL(test_sigcont_task_suspend2_resume,
844*e3723e1fSApple OSS Distributions "test to verify that SIGCONT on task_suspend2()-ed process doesn't work",
845*e3723e1fSApple OSS Distributions T_META_ASROOT(true),
846*e3723e1fSApple OSS Distributions T_META_LTEPHASE(LTE_POSTINIT))
847*e3723e1fSApple OSS Distributions {
848*e3723e1fSApple OSS Distributions T_SETUPBEGIN;
849*e3723e1fSApple OSS Distributions int is_dev = is_development_kernel();
850*e3723e1fSApple OSS Distributions T_QUIET;
851*e3723e1fSApple OSS Distributions T_ASSERT_TRUE(is_dev, "verify development kernel is running");
852*e3723e1fSApple OSS Distributions T_SETUPEND;
853*e3723e1fSApple OSS Distributions
854*e3723e1fSApple OSS Distributions mach_task_basic_info_data_t mach_basic_info_data;
855*e3723e1fSApple OSS Distributions task_info_t info_data = (task_info_t)&mach_basic_info_data;
856*e3723e1fSApple OSS Distributions
857*e3723e1fSApple OSS Distributions task_debug_info_internal_data_t debug_info;
858*e3723e1fSApple OSS Distributions mach_msg_type_number_t debug_count = TASK_DEBUG_INFO_INTERNAL_COUNT;
859*e3723e1fSApple OSS Distributions
860*e3723e1fSApple OSS Distributions kern_return_t kr;
861*e3723e1fSApple OSS Distributions int posix_ret;
862*e3723e1fSApple OSS Distributions mach_msg_type_number_t count = MACH_TASK_BASIC_INFO_COUNT;
863*e3723e1fSApple OSS Distributions task_flavor_t flavor = MACH_TASK_BASIC_INFO;
864*e3723e1fSApple OSS Distributions integer_t suspend_count = 0;
865*e3723e1fSApple OSS Distributions integer_t debug_suspend_count = 0;
866*e3723e1fSApple OSS Distributions pid_t child_pid = 0;
867*e3723e1fSApple OSS Distributions mach_port_name_t child_task;
868*e3723e1fSApple OSS Distributions task_suspension_token_t child_token = 0xFFFFF;
869*e3723e1fSApple OSS Distributions
870*e3723e1fSApple OSS Distributions /*
871*e3723e1fSApple OSS Distributions * for dt_waitpid
872*e3723e1fSApple OSS Distributions * We expect the test to fail right now, so I've set timeout to
873*e3723e1fSApple OSS Distributions * be shorter than we may want it to be when the issue is fixed
874*e3723e1fSApple OSS Distributions */
875*e3723e1fSApple OSS Distributions int timeout = 1;
876*e3723e1fSApple OSS Distributions int exit_status = 0;
877*e3723e1fSApple OSS Distributions int signal_no = 0;
878*e3723e1fSApple OSS Distributions
879*e3723e1fSApple OSS Distributions /* for pipe */
880*e3723e1fSApple OSS Distributions int fd[2];
881*e3723e1fSApple OSS Distributions pipe(fd);
882*e3723e1fSApple OSS Distributions int pipe_msg = 0;
883*e3723e1fSApple OSS Distributions
884*e3723e1fSApple OSS Distributions child_pid = fork();
885*e3723e1fSApple OSS Distributions
886*e3723e1fSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(child_pid, "verify process can be forked %d", child_pid);
887*e3723e1fSApple OSS Distributions
888*e3723e1fSApple OSS Distributions if (child_pid == 0) {
889*e3723e1fSApple OSS Distributions close(fd[1]);
890*e3723e1fSApple OSS Distributions T_LOG("Waiting to read from parent...");
891*e3723e1fSApple OSS Distributions read(fd[0], &pipe_msg, sizeof(pipe_msg));
892*e3723e1fSApple OSS Distributions T_LOG("Done reading from parent, about to exit...");
893*e3723e1fSApple OSS Distributions exit(0);
894*e3723e1fSApple OSS Distributions }
895*e3723e1fSApple OSS Distributions /*
896*e3723e1fSApple OSS Distributions * Wait for child to fork and block on read
897*e3723e1fSApple OSS Distributions */
898*e3723e1fSApple OSS Distributions sleep(1);
899*e3723e1fSApple OSS Distributions
900*e3723e1fSApple OSS Distributions close(fd[0]);
901*e3723e1fSApple OSS Distributions
902*e3723e1fSApple OSS Distributions kr = task_for_pid(mach_task_self(), child_pid, &child_task);
903*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "verify task_for_pid succeeded. check sudo if failed");
904*e3723e1fSApple OSS Distributions
905*e3723e1fSApple OSS Distributions kr = task_info(child_task, TASK_DEBUG_INFO_INTERNAL, (task_info_t)&debug_info, &debug_count);
906*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "verify task_info call succeeded");
907*e3723e1fSApple OSS Distributions
908*e3723e1fSApple OSS Distributions debug_suspend_count = debug_info.suspend_count;
909*e3723e1fSApple OSS Distributions T_EXPECT_EQ(debug_suspend_count, 0, "verify debug_info shows correct (true) suspend_count(0)");
910*e3723e1fSApple OSS Distributions
911*e3723e1fSApple OSS Distributions kr = task_suspend2(child_task, &child_token);
912*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "verify task_suspend2 call succeeded");
913*e3723e1fSApple OSS Distributions
914*e3723e1fSApple OSS Distributions kr = task_info(child_task, TASK_DEBUG_INFO_INTERNAL, (task_info_t)&debug_info, &debug_count);
915*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "verify task_info call succeeded");
916*e3723e1fSApple OSS Distributions
917*e3723e1fSApple OSS Distributions debug_suspend_count = debug_info.suspend_count;
918*e3723e1fSApple OSS Distributions T_ASSERT_EQ(debug_suspend_count, 1, "verify debug_info shows correct (true) suspend_count(1)");
919*e3723e1fSApple OSS Distributions
920*e3723e1fSApple OSS Distributions /*
921*e3723e1fSApple OSS Distributions * Verify the suspend_count for child and resume it.
922*e3723e1fSApple OSS Distributions */
923*e3723e1fSApple OSS Distributions
924*e3723e1fSApple OSS Distributions kr = task_info(child_task, flavor, info_data, &count);
925*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "verify task_info call succeeded");
926*e3723e1fSApple OSS Distributions
927*e3723e1fSApple OSS Distributions suspend_count = (integer_t)(info_get(INFO_MACH, GET_SUSPEND_COUNT, info_data));
928*e3723e1fSApple OSS Distributions T_EXPECT_EQ(suspend_count, 1, "verify task_info shows correct (user_stop_count) suspend_count (1)");
929*e3723e1fSApple OSS Distributions
930*e3723e1fSApple OSS Distributions posix_ret = kill(child_pid, SIGCONT);
931*e3723e1fSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(posix_ret, "verify signal call succeeded");
932*e3723e1fSApple OSS Distributions
933*e3723e1fSApple OSS Distributions kr = task_info(child_task, TASK_DEBUG_INFO_INTERNAL, (task_info_t)&debug_info, &debug_count);
934*e3723e1fSApple OSS Distributions T_EXPECT_MACH_SUCCESS(kr, "verify task_info call succeeded");
935*e3723e1fSApple OSS Distributions
936*e3723e1fSApple OSS Distributions debug_suspend_count = debug_info.suspend_count;
937*e3723e1fSApple OSS Distributions T_EXPECTFAIL_WITH_RADAR(33166654);
938*e3723e1fSApple OSS Distributions T_EXPECT_EQ(debug_suspend_count, 1, "verify debug_info shows correct (true) suspend_count (1)");
939*e3723e1fSApple OSS Distributions
940*e3723e1fSApple OSS Distributions suspend_count = (integer_t)(info_get(INFO_MACH, GET_SUSPEND_COUNT, info_data));
941*e3723e1fSApple OSS Distributions T_ASSERT_EQ(suspend_count, 1, "verify task_info shows correct (user_stop_count) suspend_count (1) after SIG_CONT");
942*e3723e1fSApple OSS Distributions
943*e3723e1fSApple OSS Distributions kr = task_resume(child_task);
944*e3723e1fSApple OSS Distributions T_EXPECTFAIL_WITH_RADAR(33166654);
945*e3723e1fSApple OSS Distributions T_EXPECT_MACH_SUCCESS(kr, "verify task_resume succeeded");
946*e3723e1fSApple OSS Distributions
947*e3723e1fSApple OSS Distributions /*
948*e3723e1fSApple OSS Distributions * reap kr from task_suspend call in child
949*e3723e1fSApple OSS Distributions */
950*e3723e1fSApple OSS Distributions
951*e3723e1fSApple OSS Distributions dt_waitpid(child_pid, &exit_status, &signal_no, timeout);
952*e3723e1fSApple OSS Distributions
953*e3723e1fSApple OSS Distributions T_ASSERT_EQ(signal_no, SIG_DT_TIMEOUT, "dt_waitpid timed out as expected");
954*e3723e1fSApple OSS Distributions
955*e3723e1fSApple OSS Distributions // Resume properly using token and then wait
956*e3723e1fSApple OSS Distributions
957*e3723e1fSApple OSS Distributions kr = task_resume2(child_token);
958*e3723e1fSApple OSS Distributions T_EXPECTFAIL_WITH_RADAR(33166654);
959*e3723e1fSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "verify task_resume2 succeeded");
960*e3723e1fSApple OSS Distributions
961*e3723e1fSApple OSS Distributions write(fd[1], &pipe_msg, sizeof(pipe_msg));
962*e3723e1fSApple OSS Distributions
963*e3723e1fSApple OSS Distributions /*
964*e3723e1fSApple OSS Distributions * reap kr from task_suspend call in child
965*e3723e1fSApple OSS Distributions */
966*e3723e1fSApple OSS Distributions dt_waitpid(child_pid, &exit_status, &signal_no, timeout);
967*e3723e1fSApple OSS Distributions
968*e3723e1fSApple OSS Distributions T_ASSERT_EQ(signal_no, 0, "child should be resumed and no signal should be returned");
969*e3723e1fSApple OSS Distributions T_ASSERT_EQ(exit_status, 0, "child should exit with 0");
970*e3723e1fSApple OSS Distributions }
971*e3723e1fSApple OSS Distributions
972*e3723e1fSApple OSS Distributions uint64_t
info_get(enum info_kind kind,enum info_get get,void * data)973*e3723e1fSApple OSS Distributions info_get(enum info_kind kind, enum info_get get, void * data)
974*e3723e1fSApple OSS Distributions {
975*e3723e1fSApple OSS Distributions switch (get) {
976*e3723e1fSApple OSS Distributions case GET_SUSPEND_COUNT:
977*e3723e1fSApple OSS Distributions switch (kind) {
978*e3723e1fSApple OSS Distributions case INFO_32:
979*e3723e1fSApple OSS Distributions case INFO_32_2:
980*e3723e1fSApple OSS Distributions return (uint64_t)(((task_basic_info_32_t)data)->suspend_count);
981*e3723e1fSApple OSS Distributions #if defined(__arm64__)
982*e3723e1fSApple OSS Distributions case INFO_64:
983*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("illegal info_get %d %d", kind, get);
984*e3723e1fSApple OSS Distributions break;
985*e3723e1fSApple OSS Distributions
986*e3723e1fSApple OSS Distributions case INFO_64_2:
987*e3723e1fSApple OSS Distributions return (uint64_t)(((task_basic_info_64_2_t)data)->suspend_count);
988*e3723e1fSApple OSS Distributions #else
989*e3723e1fSApple OSS Distributions case INFO_64:
990*e3723e1fSApple OSS Distributions return (uint64_t)(((task_basic_info_64_t)data)->suspend_count);
991*e3723e1fSApple OSS Distributions
992*e3723e1fSApple OSS Distributions case INFO_64_2:
993*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("illegal info_get %d %d", kind, get);
994*e3723e1fSApple OSS Distributions break;
995*e3723e1fSApple OSS Distributions #endif /* defined(__arm64__) */
996*e3723e1fSApple OSS Distributions case INFO_MACH:
997*e3723e1fSApple OSS Distributions return (uint64_t)(((mach_task_basic_info_t)data)->suspend_count);
998*e3723e1fSApple OSS Distributions case INFO_MAX:
999*e3723e1fSApple OSS Distributions default:
1000*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("unhandled info_get %d %d", kind, get);
1001*e3723e1fSApple OSS Distributions }
1002*e3723e1fSApple OSS Distributions case GET_RESIDENT_SIZE:
1003*e3723e1fSApple OSS Distributions switch (kind) {
1004*e3723e1fSApple OSS Distributions case INFO_32:
1005*e3723e1fSApple OSS Distributions case INFO_32_2:
1006*e3723e1fSApple OSS Distributions return (uint64_t)(((task_basic_info_32_t)data)->resident_size);
1007*e3723e1fSApple OSS Distributions #if defined(__arm64__)
1008*e3723e1fSApple OSS Distributions case INFO_64:
1009*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("illegal info_get %d %d", kind, get);
1010*e3723e1fSApple OSS Distributions break;
1011*e3723e1fSApple OSS Distributions
1012*e3723e1fSApple OSS Distributions case INFO_64_2:
1013*e3723e1fSApple OSS Distributions return (uint64_t)(((task_basic_info_64_2_t)data)->resident_size);
1014*e3723e1fSApple OSS Distributions #else
1015*e3723e1fSApple OSS Distributions case INFO_64:
1016*e3723e1fSApple OSS Distributions return (uint64_t)(((task_basic_info_64_t)data)->resident_size);
1017*e3723e1fSApple OSS Distributions
1018*e3723e1fSApple OSS Distributions case INFO_64_2:
1019*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("illegal info_get %d %d", kind, get);
1020*e3723e1fSApple OSS Distributions break;
1021*e3723e1fSApple OSS Distributions #endif /* defined(__arm64__) */
1022*e3723e1fSApple OSS Distributions case INFO_MACH:
1023*e3723e1fSApple OSS Distributions return (uint64_t)(((mach_task_basic_info_t)data)->resident_size);
1024*e3723e1fSApple OSS Distributions case INFO_MAX:
1025*e3723e1fSApple OSS Distributions default:
1026*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("unhandled info_get %d %d", kind, get);
1027*e3723e1fSApple OSS Distributions }
1028*e3723e1fSApple OSS Distributions case GET_VIRTUAL_SIZE:
1029*e3723e1fSApple OSS Distributions switch (kind) {
1030*e3723e1fSApple OSS Distributions case INFO_32:
1031*e3723e1fSApple OSS Distributions case INFO_32_2:
1032*e3723e1fSApple OSS Distributions return (uint64_t)(((task_basic_info_32_t)data)->virtual_size);
1033*e3723e1fSApple OSS Distributions #if defined(__arm64__)
1034*e3723e1fSApple OSS Distributions case INFO_64:
1035*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("illegal info_get %d %d", kind, get);
1036*e3723e1fSApple OSS Distributions break;
1037*e3723e1fSApple OSS Distributions
1038*e3723e1fSApple OSS Distributions case INFO_64_2:
1039*e3723e1fSApple OSS Distributions return (uint64_t)(((task_basic_info_64_2_t)data)->virtual_size);
1040*e3723e1fSApple OSS Distributions #else
1041*e3723e1fSApple OSS Distributions case INFO_64:
1042*e3723e1fSApple OSS Distributions return (uint64_t)(((task_basic_info_64_t)data)->virtual_size);
1043*e3723e1fSApple OSS Distributions
1044*e3723e1fSApple OSS Distributions case INFO_64_2:
1045*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("illegal info_get %d %d", kind, get);
1046*e3723e1fSApple OSS Distributions break;
1047*e3723e1fSApple OSS Distributions #endif /* defined(__arm64__) */
1048*e3723e1fSApple OSS Distributions case INFO_MACH:
1049*e3723e1fSApple OSS Distributions return (uint64_t)(((mach_task_basic_info_t)data)->virtual_size);
1050*e3723e1fSApple OSS Distributions
1051*e3723e1fSApple OSS Distributions case INFO_MAX:
1052*e3723e1fSApple OSS Distributions default:
1053*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("unhandled info_get %d %d", kind, get);
1054*e3723e1fSApple OSS Distributions }
1055*e3723e1fSApple OSS Distributions case GET_USER_TIME:
1056*e3723e1fSApple OSS Distributions switch (kind) {
1057*e3723e1fSApple OSS Distributions case INFO_32:
1058*e3723e1fSApple OSS Distributions case INFO_32_2:
1059*e3723e1fSApple OSS Distributions return (uint64_t) &(((task_basic_info_32_t)data)->user_time);
1060*e3723e1fSApple OSS Distributions #if defined(__arm64__)
1061*e3723e1fSApple OSS Distributions case INFO_64:
1062*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("illegal info_get %d %d", kind, get);
1063*e3723e1fSApple OSS Distributions break;
1064*e3723e1fSApple OSS Distributions
1065*e3723e1fSApple OSS Distributions case INFO_64_2:
1066*e3723e1fSApple OSS Distributions return (uint64_t) &(((task_basic_info_64_2_t)data)->user_time);
1067*e3723e1fSApple OSS Distributions #else
1068*e3723e1fSApple OSS Distributions case INFO_64:
1069*e3723e1fSApple OSS Distributions return (uint64_t) &(((task_basic_info_64_t)data)->user_time);
1070*e3723e1fSApple OSS Distributions
1071*e3723e1fSApple OSS Distributions case INFO_64_2:
1072*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("illegal info_get %d %d", kind, get);
1073*e3723e1fSApple OSS Distributions break;
1074*e3723e1fSApple OSS Distributions #endif /* defined(__arm64__) */
1075*e3723e1fSApple OSS Distributions case INFO_MACH:
1076*e3723e1fSApple OSS Distributions return (uint64_t) &(((mach_task_basic_info_t)data)->user_time);
1077*e3723e1fSApple OSS Distributions
1078*e3723e1fSApple OSS Distributions case INFO_MAX:
1079*e3723e1fSApple OSS Distributions default:
1080*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("unhandled info_get %d %d", kind, get);
1081*e3723e1fSApple OSS Distributions }
1082*e3723e1fSApple OSS Distributions case GET_SYS_TIME:
1083*e3723e1fSApple OSS Distributions switch (kind) {
1084*e3723e1fSApple OSS Distributions case INFO_32:
1085*e3723e1fSApple OSS Distributions case INFO_32_2:
1086*e3723e1fSApple OSS Distributions return (uint64_t) &(((task_basic_info_32_t)data)->system_time);
1087*e3723e1fSApple OSS Distributions #if defined(__arm64__)
1088*e3723e1fSApple OSS Distributions case INFO_64:
1089*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("illegal info_get %d %d", kind, get);
1090*e3723e1fSApple OSS Distributions break;
1091*e3723e1fSApple OSS Distributions
1092*e3723e1fSApple OSS Distributions case INFO_64_2:
1093*e3723e1fSApple OSS Distributions return (uint64_t) &(((task_basic_info_64_2_t)data)->system_time);
1094*e3723e1fSApple OSS Distributions #else
1095*e3723e1fSApple OSS Distributions case INFO_64:
1096*e3723e1fSApple OSS Distributions return (uint64_t) &(((task_basic_info_64_t)data)->system_time);
1097*e3723e1fSApple OSS Distributions
1098*e3723e1fSApple OSS Distributions case INFO_64_2:
1099*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("illegal info_get %d %d", kind, get);
1100*e3723e1fSApple OSS Distributions break;
1101*e3723e1fSApple OSS Distributions #endif /* defined(__arm64__) */
1102*e3723e1fSApple OSS Distributions case INFO_MACH:
1103*e3723e1fSApple OSS Distributions return (uint64_t) &(((mach_task_basic_info_t)data)->user_time);
1104*e3723e1fSApple OSS Distributions case INFO_MAX:
1105*e3723e1fSApple OSS Distributions default:
1106*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("unhandled info_get %d %d", kind, get);
1107*e3723e1fSApple OSS Distributions }
1108*e3723e1fSApple OSS Distributions case GET_POLICY:
1109*e3723e1fSApple OSS Distributions switch (kind) {
1110*e3723e1fSApple OSS Distributions case INFO_32:
1111*e3723e1fSApple OSS Distributions case INFO_32_2:
1112*e3723e1fSApple OSS Distributions return (uint64_t)(((task_basic_info_32_t)data)->policy);
1113*e3723e1fSApple OSS Distributions #if defined(__arm64__)
1114*e3723e1fSApple OSS Distributions case INFO_64:
1115*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("illegal info_get %d %d", kind, get);
1116*e3723e1fSApple OSS Distributions break;
1117*e3723e1fSApple OSS Distributions
1118*e3723e1fSApple OSS Distributions case INFO_64_2:
1119*e3723e1fSApple OSS Distributions return (uint64_t)(((task_basic_info_64_2_t)data)->policy);
1120*e3723e1fSApple OSS Distributions #else
1121*e3723e1fSApple OSS Distributions case INFO_64:
1122*e3723e1fSApple OSS Distributions return (uint64_t)(((task_basic_info_64_t)data)->policy);
1123*e3723e1fSApple OSS Distributions
1124*e3723e1fSApple OSS Distributions case INFO_64_2:
1125*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("illegal info_get %d %d", kind, get);
1126*e3723e1fSApple OSS Distributions break;
1127*e3723e1fSApple OSS Distributions #endif /* defined(__arm64__) */
1128*e3723e1fSApple OSS Distributions case INFO_MACH:
1129*e3723e1fSApple OSS Distributions return (uint64_t)(((mach_task_basic_info_t)data)->policy);
1130*e3723e1fSApple OSS Distributions
1131*e3723e1fSApple OSS Distributions case INFO_MAX:
1132*e3723e1fSApple OSS Distributions default:
1133*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("unhandled info_get %d %d", kind, get);
1134*e3723e1fSApple OSS Distributions }
1135*e3723e1fSApple OSS Distributions case GET_MAX_RES:
1136*e3723e1fSApple OSS Distributions switch (kind) {
1137*e3723e1fSApple OSS Distributions case INFO_32:
1138*e3723e1fSApple OSS Distributions case INFO_32_2:
1139*e3723e1fSApple OSS Distributions case INFO_64:
1140*e3723e1fSApple OSS Distributions case INFO_64_2:
1141*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("illegal info_get %d %d", kind, get);
1142*e3723e1fSApple OSS Distributions case INFO_MACH:
1143*e3723e1fSApple OSS Distributions return (uint64_t)(((mach_task_basic_info_t)data)->resident_size_max);
1144*e3723e1fSApple OSS Distributions case INFO_MAX:
1145*e3723e1fSApple OSS Distributions default:
1146*e3723e1fSApple OSS Distributions T_ASSERT_FAIL("unhandled info_get %d %d", kind, get);
1147*e3723e1fSApple OSS Distributions }
1148*e3723e1fSApple OSS Distributions }
1149*e3723e1fSApple OSS Distributions
1150*e3723e1fSApple OSS Distributions __builtin_unreachable();
1151*e3723e1fSApple OSS Distributions }
1152