1*4f1223e8SApple OSS Distributions #include <darwintest.h> 2*4f1223e8SApple OSS Distributions #include <darwintest_utils.h> 3*4f1223e8SApple OSS Distributions #include <errno.h> 4*4f1223e8SApple OSS Distributions #include <mach/mach.h> 5*4f1223e8SApple OSS Distributions #include <mach/mach_types.h> 6*4f1223e8SApple OSS Distributions #include <mach/task.h> 7*4f1223e8SApple OSS Distributions #include <mach/mach_error.h> 8*4f1223e8SApple OSS Distributions #include <mach/task_special_ports.h> 9*4f1223e8SApple OSS Distributions 10*4f1223e8SApple OSS Distributions #include <signal.h> 11*4f1223e8SApple OSS Distributions #include <stdio.h> 12*4f1223e8SApple OSS Distributions #include <stdlib.h> 13*4f1223e8SApple OSS Distributions #include <unistd.h> 14*4f1223e8SApple OSS Distributions 15*4f1223e8SApple OSS Distributions T_GLOBAL_META( 16*4f1223e8SApple OSS Distributions T_META_NAMESPACE("xnu.ipc"), 17*4f1223e8SApple OSS Distributions T_META_RUN_CONCURRENTLY(TRUE), 18*4f1223e8SApple OSS Distributions T_META_RADAR_COMPONENT_NAME("xnu"), 19*4f1223e8SApple OSS Distributions T_META_RADAR_COMPONENT_VERSION("IPC"), 20*4f1223e8SApple OSS Distributions T_META_TAG_VM_PREFERRED); 21*4f1223e8SApple OSS Distributions 22*4f1223e8SApple OSS Distributions T_DECL(task_ident, "test task identity token") 23*4f1223e8SApple OSS Distributions { 24*4f1223e8SApple OSS Distributions kern_return_t kr; 25*4f1223e8SApple OSS Distributions task_id_token_t token; 26*4f1223e8SApple OSS Distributions mach_port_t port1, port2; 27*4f1223e8SApple OSS Distributions 28*4f1223e8SApple OSS Distributions kr = task_create_identity_token(mach_task_self(), &token); 29*4f1223e8SApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "task_create_identity_token()"); 30*4f1223e8SApple OSS Distributions 31*4f1223e8SApple OSS Distributions port1 = mach_task_self(); 32*4f1223e8SApple OSS Distributions kr = task_identity_token_get_task_port(token, TASK_FLAVOR_CONTROL, &port2); /* Immovable control port for self */ 33*4f1223e8SApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "task_identity_token_get_task_port() - CONTROL"); 34*4f1223e8SApple OSS Distributions T_EXPECT_EQ(port1, port2, "Control port does not match!"); 35*4f1223e8SApple OSS Distributions 36*4f1223e8SApple OSS Distributions mach_port_deallocate(mach_task_self(), port2); 37*4f1223e8SApple OSS Distributions 38*4f1223e8SApple OSS Distributions kr = task_get_special_port(mach_task_self(), TASK_READ_PORT, &port1); 39*4f1223e8SApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "task_get_special_port() - READ"); 40*4f1223e8SApple OSS Distributions kr = task_identity_token_get_task_port(token, TASK_FLAVOR_READ, &port2); 41*4f1223e8SApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "task_identity_token_get_task_port() - read"); 42*4f1223e8SApple OSS Distributions T_EXPECT_EQ(port1, port2, "Read port does not match!"); 43*4f1223e8SApple OSS Distributions 44*4f1223e8SApple OSS Distributions mach_port_deallocate(mach_task_self(), port1); 45*4f1223e8SApple OSS Distributions mach_port_deallocate(mach_task_self(), port2); 46*4f1223e8SApple OSS Distributions 47*4f1223e8SApple OSS Distributions kr = task_get_special_port(mach_task_self(), TASK_INSPECT_PORT, &port1); 48*4f1223e8SApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "task_get_special_port() - INSPECT"); 49*4f1223e8SApple OSS Distributions kr = task_identity_token_get_task_port(token, TASK_FLAVOR_INSPECT, &port2); 50*4f1223e8SApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "task_identity_token_get_task_port() - inspect"); 51*4f1223e8SApple OSS Distributions T_EXPECT_EQ(port1, port2, "Inspect port does not match!"); 52*4f1223e8SApple OSS Distributions 53*4f1223e8SApple OSS Distributions mach_port_deallocate(mach_task_self(), port1); 54*4f1223e8SApple OSS Distributions mach_port_deallocate(mach_task_self(), port2); 55*4f1223e8SApple OSS Distributions 56*4f1223e8SApple OSS Distributions kr = task_get_special_port(mach_task_self(), TASK_NAME_PORT, &port1); 57*4f1223e8SApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "task_get_special_port() - NAME"); 58*4f1223e8SApple OSS Distributions kr = task_identity_token_get_task_port(token, TASK_FLAVOR_NAME, &port2); 59*4f1223e8SApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "task_identity_token_get_task_port() - name"); 60*4f1223e8SApple OSS Distributions T_EXPECT_EQ(port1, port2, "Name port does not match!"); 61*4f1223e8SApple OSS Distributions 62*4f1223e8SApple OSS Distributions mach_port_deallocate(mach_task_self(), port1); 63*4f1223e8SApple OSS Distributions mach_port_deallocate(mach_task_self(), port2); 64*4f1223e8SApple OSS Distributions 65*4f1223e8SApple OSS Distributions kr = task_identity_token_get_task_port(mach_thread_self(), TASK_FLAVOR_NAME, &port2); 66*4f1223e8SApple OSS Distributions T_EXPECT_NE(kr, KERN_SUCCESS, "task_identity_token_get_task_port() should fail on non-token port"); 67*4f1223e8SApple OSS Distributions 68*4f1223e8SApple OSS Distributions mach_port_deallocate(mach_task_self(), token); 69*4f1223e8SApple OSS Distributions } 70