1*fdd8201dSApple OSS Distributions #include <darwintest.h> 2*fdd8201dSApple OSS Distributions #include "../bsd/sys/proc_info.h" 3*fdd8201dSApple OSS Distributions #include "../libsyscall/wrappers/libproc/libproc.h" 4*fdd8201dSApple OSS Distributions #include <stdio.h> 5*fdd8201dSApple OSS Distributions #include <unistd.h> 6*fdd8201dSApple OSS Distributions #include <TargetConditionals.h> 7*fdd8201dSApple OSS Distributions 8*fdd8201dSApple OSS Distributions T_GLOBAL_META(T_META_RUN_CONCURRENTLY(true)); 9*fdd8201dSApple OSS Distributions 10*fdd8201dSApple OSS Distributions T_DECL(proc_udata_info, "Get and set a proc udata token"){ 11*fdd8201dSApple OSS Distributions uint64_t token = mach_absolute_time(); 12*fdd8201dSApple OSS Distributions proc_info_udata_t udata; 13*fdd8201dSApple OSS Distributions int ret; 14*fdd8201dSApple OSS Distributions 15*fdd8201dSApple OSS Distributions udata = token; 16*fdd8201dSApple OSS Distributions ret = proc_udata_info(getpid(), PROC_UDATA_INFO_SET, &udata, sizeof(udata)); 17*fdd8201dSApple OSS Distributions 18*fdd8201dSApple OSS Distributions #if !TARGET_OS_OSX 19*fdd8201dSApple OSS Distributions T_WITH_ERRNO; 20*fdd8201dSApple OSS Distributions T_ASSERT_EQ_INT(ret, -1, "proc_udata_info PROC_UDATA_INFO_SET returns error on non-supported platforms"); 21*fdd8201dSApple OSS Distributions T_SKIP("Remaining tests are only supported on platforms with CONFIG_PROC_UDATA_STORAGE configured"); 22*fdd8201dSApple OSS Distributions #endif 23*fdd8201dSApple OSS Distributions 24*fdd8201dSApple OSS Distributions T_WITH_ERRNO; 25*fdd8201dSApple OSS Distributions T_ASSERT_EQ_INT(ret, 0, "proc_udata_info PROC_UDATA_INFO_SET"); 26*fdd8201dSApple OSS Distributions 27*fdd8201dSApple OSS Distributions T_LOG("udata set to %#llx", udata); 28*fdd8201dSApple OSS Distributions 29*fdd8201dSApple OSS Distributions bzero(&udata, sizeof(udata)); 30*fdd8201dSApple OSS Distributions ret = proc_udata_info(getpid(), PROC_UDATA_INFO_GET, &udata, sizeof(udata)); 31*fdd8201dSApple OSS Distributions T_WITH_ERRNO; 32*fdd8201dSApple OSS Distributions T_ASSERT_EQ_INT(ret, 0, "proc_udata_info PROC_UDATA_INFO_GET"); 33*fdd8201dSApple OSS Distributions 34*fdd8201dSApple OSS Distributions T_ASSERT_EQ_ULLONG(token, udata, "proc_udata_info(): retrieved value matches token"); 35*fdd8201dSApple OSS Distributions 36*fdd8201dSApple OSS Distributions ret = proc_udata_info(getpid(), PROC_UDATA_INFO_SET, &udata, sizeof(uint32_t)); 37*fdd8201dSApple OSS Distributions T_WITH_ERRNO; 38*fdd8201dSApple OSS Distributions T_ASSERT_EQ_INT(ret, -1, "proc_udata_info PROC_UDATA_INFO_SET with invalid size returned -1"); 39*fdd8201dSApple OSS Distributions T_ASSERT_EQ_INT(errno, EINVAL, "proc_udata_info PROC_UDATA_INFO_SET with invalid size returned EINVAL"); 40*fdd8201dSApple OSS Distributions 41*fdd8201dSApple OSS Distributions ret = proc_udata_info(getppid(), PROC_UDATA_INFO_GET, &udata, sizeof(udata)); 42*fdd8201dSApple OSS Distributions T_WITH_ERRNO; 43*fdd8201dSApple OSS Distributions T_ASSERT_EQ_INT(ret, -1, "proc_udata_info PROC_UDATA_INFO_GET returned -1 on attempt against non-self pid"); 44*fdd8201dSApple OSS Distributions T_ASSERT_EQ_INT(errno, EACCES, "proc_udata_info PROC_UDATA_INFO_GET set errno to EACCES on attempt against non-self pid"); 45*fdd8201dSApple OSS Distributions 46*fdd8201dSApple OSS Distributions ret = proc_udata_info(getppid(), PROC_UDATA_INFO_SET, &udata, sizeof(udata)); 47*fdd8201dSApple OSS Distributions T_WITH_ERRNO; 48*fdd8201dSApple OSS Distributions T_ASSERT_EQ_INT(ret, -1, "proc_udata_info PROC_UDATA_INFO_SET returned -1 on attempt against non-self pid"); 49*fdd8201dSApple OSS Distributions T_ASSERT_EQ_INT(errno, EACCES, "proc_udata_info PROC_UDATA_INFO_SET set errno to EACCES on attempt against non-self pid"); 50*fdd8201dSApple OSS Distributions } 51