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