1*5e3eaea3SApple OSS Distributions #include <dispatch/dispatch.h>
2*5e3eaea3SApple OSS Distributions #include <mach-o/dyld.h>
3*5e3eaea3SApple OSS Distributions #include <signal.h>
4*5e3eaea3SApple OSS Distributions #include <sys/kern_sysctl.h>
5*5e3eaea3SApple OSS Distributions #include <sys/sysctl.h>
6*5e3eaea3SApple OSS Distributions #include <sys/kern_memorystatus.h>
7*5e3eaea3SApple OSS Distributions
8*5e3eaea3SApple OSS Distributions #include <darwintest.h>
9*5e3eaea3SApple OSS Distributions #include <darwintest_utils.h>
10*5e3eaea3SApple OSS Distributions
11*5e3eaea3SApple OSS Distributions #include "test_utils.h"
12*5e3eaea3SApple OSS Distributions
13*5e3eaea3SApple OSS Distributions bool
is_development_kernel(void)14*5e3eaea3SApple OSS Distributions is_development_kernel(void)
15*5e3eaea3SApple OSS Distributions {
16*5e3eaea3SApple OSS Distributions static dispatch_once_t is_development_once;
17*5e3eaea3SApple OSS Distributions static bool is_development;
18*5e3eaea3SApple OSS Distributions
19*5e3eaea3SApple OSS Distributions dispatch_once(&is_development_once, ^{
20*5e3eaea3SApple OSS Distributions int dev;
21*5e3eaea3SApple OSS Distributions size_t dev_size = sizeof(dev);
22*5e3eaea3SApple OSS Distributions
23*5e3eaea3SApple OSS Distributions T_QUIET;
24*5e3eaea3SApple OSS Distributions T_ASSERT_POSIX_SUCCESS(sysctlbyname("kern.development", &dev,
25*5e3eaea3SApple OSS Distributions &dev_size, NULL, 0), NULL);
26*5e3eaea3SApple OSS Distributions is_development = (dev != 0);
27*5e3eaea3SApple OSS Distributions });
28*5e3eaea3SApple OSS Distributions
29*5e3eaea3SApple OSS Distributions return is_development;
30*5e3eaea3SApple OSS Distributions }
31*5e3eaea3SApple OSS Distributions
32*5e3eaea3SApple OSS Distributions pid_t
launch_background_helper(const char * variant,bool start_suspended,bool memorystatus_managed)33*5e3eaea3SApple OSS Distributions launch_background_helper(
34*5e3eaea3SApple OSS Distributions const char* variant,
35*5e3eaea3SApple OSS Distributions bool start_suspended,
36*5e3eaea3SApple OSS Distributions bool memorystatus_managed)
37*5e3eaea3SApple OSS Distributions {
38*5e3eaea3SApple OSS Distributions pid_t pid;
39*5e3eaea3SApple OSS Distributions char **launch_tool_args;
40*5e3eaea3SApple OSS Distributions char testpath[PATH_MAX];
41*5e3eaea3SApple OSS Distributions char *variant_cpy = strdup(variant);
42*5e3eaea3SApple OSS Distributions uint32_t testpath_buf_size;
43*5e3eaea3SApple OSS Distributions int ret;
44*5e3eaea3SApple OSS Distributions
45*5e3eaea3SApple OSS Distributions testpath_buf_size = sizeof(testpath);
46*5e3eaea3SApple OSS Distributions ret = _NSGetExecutablePath(testpath, &testpath_buf_size);
47*5e3eaea3SApple OSS Distributions launch_tool_args = (char *[]){
48*5e3eaea3SApple OSS Distributions testpath,
49*5e3eaea3SApple OSS Distributions "-n",
50*5e3eaea3SApple OSS Distributions variant_cpy,
51*5e3eaea3SApple OSS Distributions NULL
52*5e3eaea3SApple OSS Distributions };
53*5e3eaea3SApple OSS Distributions ret = dt_launch_tool(&pid, launch_tool_args, start_suspended, NULL, NULL);
54*5e3eaea3SApple OSS Distributions if (ret != 0) {
55*5e3eaea3SApple OSS Distributions T_LOG("dt_launch tool returned %d with error code %d", ret, errno);
56*5e3eaea3SApple OSS Distributions }
57*5e3eaea3SApple OSS Distributions T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "dt_launch_tool");
58*5e3eaea3SApple OSS Distributions if (memorystatus_managed) {
59*5e3eaea3SApple OSS Distributions set_process_memorystatus_managed(pid);
60*5e3eaea3SApple OSS Distributions }
61*5e3eaea3SApple OSS Distributions free(variant_cpy);
62*5e3eaea3SApple OSS Distributions return pid;
63*5e3eaea3SApple OSS Distributions }
64*5e3eaea3SApple OSS Distributions
65*5e3eaea3SApple OSS Distributions void
set_process_memorystatus_managed(pid_t pid)66*5e3eaea3SApple OSS Distributions set_process_memorystatus_managed(pid_t pid)
67*5e3eaea3SApple OSS Distributions {
68*5e3eaea3SApple OSS Distributions kern_return_t ret = memorystatus_control(MEMORYSTATUS_CMD_SET_PROCESS_IS_MANAGED, pid, 1, NULL, 0);
69*5e3eaea3SApple OSS Distributions T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "memorystatus_control");
70*5e3eaea3SApple OSS Distributions }
71