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