Open Menu / tests / task_ident_test.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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
#include <darwintest.h>
#include <darwintest_utils.h>
#include <errno.h>
#include <mach/mach.h>
#include <mach/mach_types.h>
#include <mach/task.h>
#include <mach/mach_error.h>
#include <mach/task_special_ports.h>

#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

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

T_DECL(task_ident, "test task identity token")
{
	kern_return_t kr;
	task_id_token_t token;
	mach_port_t port1, port2;

	kr = task_create_identity_token(mach_task_self(), &token);
	T_ASSERT_MACH_SUCCESS(kr, "task_create_identity_token()");

	port1 = mach_task_self();
	kr = task_identity_token_get_task_port(token, TASK_FLAVOR_CONTROL, &port2); /* Immovable control port for self */
	T_ASSERT_MACH_SUCCESS(kr, "task_identity_token_get_task_port() - CONTROL");
	T_EXPECT_EQ(port1, port2, "Control port does not match!");

	mach_port_deallocate(mach_task_self(), port2);

	kr = task_get_special_port(mach_task_self(), TASK_READ_PORT, &port1);
	T_ASSERT_MACH_SUCCESS(kr, "task_get_special_port() - READ");
	kr = task_identity_token_get_task_port(token, TASK_FLAVOR_READ, &port2);
	T_ASSERT_MACH_SUCCESS(kr, "task_identity_token_get_task_port() - read");
	T_EXPECT_EQ(port1, port2, "Read port does not match!");

	mach_port_deallocate(mach_task_self(), port1);
	mach_port_deallocate(mach_task_self(), port2);

	kr = task_get_special_port(mach_task_self(), TASK_INSPECT_PORT, &port1);
	T_ASSERT_MACH_SUCCESS(kr, "task_get_special_port() - INSPECT");
	kr = task_identity_token_get_task_port(token, TASK_FLAVOR_INSPECT, &port2);
	T_ASSERT_MACH_SUCCESS(kr, "task_identity_token_get_task_port() - inspect");
	T_EXPECT_EQ(port1, port2, "Inspect port does not match!");

	mach_port_deallocate(mach_task_self(), port1);
	mach_port_deallocate(mach_task_self(), port2);

	kr = task_get_special_port(mach_task_self(), TASK_NAME_PORT, &port1);
	T_ASSERT_MACH_SUCCESS(kr, "task_get_special_port() - NAME");
	kr = task_identity_token_get_task_port(token, TASK_FLAVOR_NAME, &port2);
	T_ASSERT_MACH_SUCCESS(kr, "task_identity_token_get_task_port() - name");
	T_EXPECT_EQ(port1, port2, "Name port does not match!");

	mach_port_deallocate(mach_task_self(), port1);
	mach_port_deallocate(mach_task_self(), port2);

	kr = task_identity_token_get_task_port(mach_thread_self(), TASK_FLAVOR_NAME, &port2);
	T_EXPECT_NE(kr, KERN_SUCCESS, "task_identity_token_get_task_port() should fail on non-token port");

	mach_port_deallocate(mach_task_self(), token);
}