1*c54f35caSApple OSS Distributions /* 2*c54f35caSApple OSS Distributions * Copyright (c) 2022 Apple Computer, Inc. All rights reserved. 3*c54f35caSApple OSS Distributions * 4*c54f35caSApple OSS Distributions * @APPLE_OSREFERENCE_LICENSE_HEADER_START@ 5*c54f35caSApple OSS Distributions * 6*c54f35caSApple OSS Distributions * This file contains Original Code and/or Modifications of Original Code 7*c54f35caSApple OSS Distributions * as defined in and that are subject to the Apple Public Source License 8*c54f35caSApple OSS Distributions * Version 2.0 (the 'License'). You may not use this file except in 9*c54f35caSApple OSS Distributions * compliance with the License. The rights granted to you under the License 10*c54f35caSApple OSS Distributions * may not be used to create, or enable the creation or redistribution of, 11*c54f35caSApple OSS Distributions * unlawful or unlicensed copies of an Apple operating system, or to 12*c54f35caSApple OSS Distributions * circumvent, violate, or enable the circumvention or violation of, any 13*c54f35caSApple OSS Distributions * terms of an Apple operating system software license agreement. 14*c54f35caSApple OSS Distributions * 15*c54f35caSApple OSS Distributions * Please obtain a copy of the License at 16*c54f35caSApple OSS Distributions * http://www.opensource.apple.com/apsl/ and read it before using this file. 17*c54f35caSApple OSS Distributions * 18*c54f35caSApple OSS Distributions * The Original Code and all software distributed under the License are 19*c54f35caSApple OSS Distributions * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 20*c54f35caSApple OSS Distributions * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, 21*c54f35caSApple OSS Distributions * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, 22*c54f35caSApple OSS Distributions * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 23*c54f35caSApple OSS Distributions * Please see the License for the specific language governing rights and 24*c54f35caSApple OSS Distributions * limitations under the License. 25*c54f35caSApple OSS Distributions * 26*c54f35caSApple OSS Distributions * @APPLE_OSREFERENCE_LICENSE_HEADER_END@ 27*c54f35caSApple OSS Distributions */ 28*c54f35caSApple OSS Distributions 29*c54f35caSApple OSS Distributions 30*c54f35caSApple OSS Distributions #include <unistd.h> 31*c54f35caSApple OSS Distributions #include <darwintest.h> 32*c54f35caSApple OSS Distributions #include <mach/mach.h> 33*c54f35caSApple OSS Distributions #include <drop_priv.h> 34*c54f35caSApple OSS Distributions T_GLOBAL_META( 35*c54f35caSApple OSS Distributions T_META_NAMESPACE("xnu.ipc"), 36*c54f35caSApple OSS Distributions T_META_RADAR_COMPONENT_NAME("xnu"), 37*c54f35caSApple OSS Distributions T_META_RADAR_COMPONENT_VERSION("IPC"), 38*c54f35caSApple OSS Distributions T_META_RUN_CONCURRENTLY(true)); 39*c54f35caSApple OSS Distributions 40*c54f35caSApple OSS Distributions T_DECL(task_name_for_pid_entitlement, "Test that task_name_for_pid suceeds with entitlement", 41*c54f35caSApple OSS Distributions T_META_ASROOT(false), 42*c54f35caSApple OSS Distributions T_META_CHECK_LEAKS(false)) 43*c54f35caSApple OSS Distributions { 44*c54f35caSApple OSS Distributions kern_return_t kr; 45*c54f35caSApple OSS Distributions mach_port_t tname; 46*c54f35caSApple OSS Distributions pid_t pid; 47*c54f35caSApple OSS Distributions T_SETUPBEGIN; 48*c54f35caSApple OSS Distributions T_ASSERT_NE(getuid(), 0, "test should not be root uid"); 49*c54f35caSApple OSS Distributions T_SETUPEND; 50*c54f35caSApple OSS Distributions // launchd has root uid/gid so we know that we must be hitting the entitlement check here. 51*c54f35caSApple OSS Distributions kr = task_name_for_pid(mach_task_self(), 1, &tname); 52*c54f35caSApple OSS Distributions T_ASSERT_MACH_SUCCESS(kr, "task_name_for_pid should succeed on launchd (pid 1)"); 53*c54f35caSApple OSS Distributions pid_for_task(tname, &pid); 54*c54f35caSApple OSS Distributions T_ASSERT_EQ(pid, 1, "pid_for_task should return pid for launchd (pid 1)"); 55*c54f35caSApple OSS Distributions 56*c54f35caSApple OSS Distributions mach_port_deallocate(mach_task_self(), tname); 57*c54f35caSApple OSS Distributions } 58