1*2c2f96dcSApple OSS Distributions /* 2*2c2f96dcSApple OSS Distributions * Copyright (c) 2019 Apple Computer, Inc. All rights reserved. 3*2c2f96dcSApple OSS Distributions * 4*2c2f96dcSApple OSS Distributions * @APPLE_OSREFERENCE_LICENSE_HEADER_START@ 5*2c2f96dcSApple OSS Distributions * 6*2c2f96dcSApple OSS Distributions * This file contains Original Code and/or Modifications of Original Code 7*2c2f96dcSApple OSS Distributions * as defined in and that are subject to the Apple Public Source License 8*2c2f96dcSApple OSS Distributions * Version 2.0 (the 'License'). You may not use this file except in 9*2c2f96dcSApple OSS Distributions * compliance with the License. The rights granted to you under the License 10*2c2f96dcSApple OSS Distributions * may not be used to create, or enable the creation or redistribution of, 11*2c2f96dcSApple OSS Distributions * unlawful or unlicensed copies of an Apple operating system, or to 12*2c2f96dcSApple OSS Distributions * circumvent, violate, or enable the circumvention or violation of, any 13*2c2f96dcSApple OSS Distributions * terms of an Apple operating system software license agreement. 14*2c2f96dcSApple OSS Distributions * 15*2c2f96dcSApple OSS Distributions * Please obtain a copy of the License at 16*2c2f96dcSApple OSS Distributions * http://www.opensource.apple.com/apsl/ and read it before using this file. 17*2c2f96dcSApple OSS Distributions * 18*2c2f96dcSApple OSS Distributions * The Original Code and all software distributed under the License are 19*2c2f96dcSApple OSS Distributions * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 20*2c2f96dcSApple OSS Distributions * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, 21*2c2f96dcSApple OSS Distributions * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, 22*2c2f96dcSApple OSS Distributions * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 23*2c2f96dcSApple OSS Distributions * Please see the License for the specific language governing rights and 24*2c2f96dcSApple OSS Distributions * limitations under the License. 25*2c2f96dcSApple OSS Distributions * 26*2c2f96dcSApple OSS Distributions * @APPLE_OSREFERENCE_LICENSE_HEADER_END@ 27*2c2f96dcSApple OSS Distributions */ 28*2c2f96dcSApple OSS Distributions #include <darwintest.h> 29*2c2f96dcSApple OSS Distributions #include <pthread.h> 30*2c2f96dcSApple OSS Distributions #include <ptrauth.h> 31*2c2f96dcSApple OSS Distributions #include <mach/machine/thread_state.h> 32*2c2f96dcSApple OSS Distributions #include <sys/types.h> 33*2c2f96dcSApple OSS Distributions #include <sys/sysctl.h> 34*2c2f96dcSApple OSS Distributions #include <sys/code_signing.h> 35*2c2f96dcSApple OSS Distributions #include <stdlib.h> 36*2c2f96dcSApple OSS Distributions #include "exc_helpers.h" 37*2c2f96dcSApple OSS Distributions 38*2c2f96dcSApple OSS Distributions T_GLOBAL_META( 39*2c2f96dcSApple OSS Distributions T_META_NAMESPACE("xnu.arm"), 40*2c2f96dcSApple OSS Distributions T_META_RADAR_COMPONENT_NAME("xnu"), 41*2c2f96dcSApple OSS Distributions T_META_RADAR_COMPONENT_VERSION("arm"), 42*2c2f96dcSApple OSS Distributions T_META_OWNER("ghackmann"), 43*2c2f96dcSApple OSS Distributions T_META_RUN_CONCURRENTLY(true) 44*2c2f96dcSApple OSS Distributions ); 45*2c2f96dcSApple OSS Distributions 46*2c2f96dcSApple OSS Distributions 47*2c2f96dcSApple OSS Distributions T_DECL(thread_set_state_corrupted_pc, 48*2c2f96dcSApple OSS Distributions "Test that ptrauth failures in thread_set_state() poison the respective register.") 49*2c2f96dcSApple OSS Distributions { 50*2c2f96dcSApple OSS Distributions #if !__arm64e__ 51*2c2f96dcSApple OSS Distributions T_SKIP("Running on non-arm64e target, skipping..."); 52*2c2f96dcSApple OSS Distributions #else 53*2c2f96dcSApple OSS Distributions mach_port_t thread; 54*2c2f96dcSApple OSS Distributions kern_return_t err = thread_create(mach_task_self(), &thread); 55*2c2f96dcSApple OSS Distributions T_QUIET; T_ASSERT_EQ(err, KERN_SUCCESS, "Created thread"); 56*2c2f96dcSApple OSS Distributions 57*2c2f96dcSApple OSS Distributions arm_thread_state64_t state; 58*2c2f96dcSApple OSS Distributions mach_msg_type_number_t count = ARM_THREAD_STATE64_COUNT; 59*2c2f96dcSApple OSS Distributions err = thread_get_state(mach_thread_self(), ARM_THREAD_STATE64, (thread_state_t)&state, &count); 60*2c2f96dcSApple OSS Distributions T_QUIET; T_ASSERT_EQ(err, KERN_SUCCESS, "Got own thread state"); 61*2c2f96dcSApple OSS Distributions 62*2c2f96dcSApple OSS Distributions void *corrupted_pc = (void *)((uintptr_t)state.__opaque_pc ^ 0x4); 63*2c2f96dcSApple OSS Distributions state.__opaque_pc = corrupted_pc; 64*2c2f96dcSApple OSS Distributions err = thread_set_state(thread, ARM_THREAD_STATE64, (thread_state_t)&state, count); 65*2c2f96dcSApple OSS Distributions T_QUIET; T_ASSERT_EQ(err, KERN_SUCCESS, "Set child thread's PC to a corrupted pointer"); 66*2c2f96dcSApple OSS Distributions 67*2c2f96dcSApple OSS Distributions err = thread_get_state(thread, ARM_THREAD_STATE64, (thread_state_t)&state, &count); 68*2c2f96dcSApple OSS Distributions T_QUIET; T_ASSERT_EQ(err, KERN_SUCCESS, "Got child's thread state"); 69*2c2f96dcSApple OSS Distributions T_EXPECT_NE(state.__opaque_pc, corrupted_pc, "thread_set_state() with a corrupted PC should poison the PC value"); 70*2c2f96dcSApple OSS Distributions 71*2c2f96dcSApple OSS Distributions err = thread_terminate(thread); 72*2c2f96dcSApple OSS Distributions T_QUIET; T_EXPECT_EQ(err, KERN_SUCCESS, "Terminated thread"); 73*2c2f96dcSApple OSS Distributions #endif // __arm64e__ 74*2c2f96dcSApple OSS Distributions } 75*2c2f96dcSApple OSS Distributions 76