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