Open Menu / tests / exec_set_proc_name.c
Loading...
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#include <spawn.h>
#include <libproc.h>
#include <darwintest.h>
#include <darwintest_utils.h>
#include <mach-o/dyld.h>


T_GLOBAL_META(
	T_META_NAMESPACE("xnu.spawn"),
	T_META_RADAR_COMPONENT_NAME("xnu"),
	T_META_RADAR_COMPONENT_VERSION("spawn"),
	T_META_RUN_CONCURRENTLY(TRUE));

static void
check_myself(char *name)
{
	struct proc_bsdinfo pinfo = {0};
	int ret = proc_pidinfo(getpid(), PROC_PIDTBSDINFO, 0, &pinfo, sizeof(pinfo));
	T_ASSERT_POSIX_SUCCESS(ret, "proc_pidinfo");

	T_LOG("my process name is '%s' (comm is '%s')", pinfo.pbi_name, pinfo.pbi_comm);

	char *found = strstr(pinfo.pbi_name, "exec_set_proc_name");
	T_ASSERT_NOTNULL(found, "proc name of %s", name);
}

T_HELPER_DECL(spawned_helper, "spawned helper")
{
	check_myself("child");
}

T_DECL(set_proc_name, "check process name is correct", T_META_TAG_VM_PREFERRED)
{
	int pid, ret, status;

	check_myself("parent");

	char binpath[MAXPATHLEN];
	uint32_t size = sizeof(binpath);
	ret = _NSGetExecutablePath(binpath, &size);
	T_QUIET; T_ASSERT_EQ(ret, 0, "get binary path");

	ret = dt_launch_tool(&pid, (char *[]) { binpath, "-n", "spawned_helper", NULL }, false, NULL, NULL);
	T_ASSERT_POSIX_ZERO(ret, "posix_spawn");

	ret = waitpid(pid, &status, 0);
	T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "waitpid");
	T_ASSERT_TRUE(WIFEXITED(status), "child exited");
	T_ASSERT_EQ(WEXITSTATUS(status), 0, "child exit code");
}