1*5e3eaea3SApple OSS Distributions #include <string.h>
2*5e3eaea3SApple OSS Distributions #include <stdlib.h>
3*5e3eaea3SApple OSS Distributions #include <mach/mach.h>
4*5e3eaea3SApple OSS Distributions #include <mach_debug/mach_debug.h>
5*5e3eaea3SApple OSS Distributions #include <darwintest.h>
6*5e3eaea3SApple OSS Distributions
7*5e3eaea3SApple OSS Distributions T_GLOBAL_META(
8*5e3eaea3SApple OSS Distributions T_META_NAMESPACE("xnu.vm"),
9*5e3eaea3SApple OSS Distributions T_META_RADAR_COMPONENT_NAME("xnu"),
10*5e3eaea3SApple OSS Distributions T_META_RADAR_COMPONENT_VERSION("zalloc"),
11*5e3eaea3SApple OSS Distributions T_META_CHECK_LEAKS(false),
12*5e3eaea3SApple OSS Distributions T_META_RUN_CONCURRENTLY(true)
13*5e3eaea3SApple OSS Distributions );
14*5e3eaea3SApple OSS Distributions
15*5e3eaea3SApple OSS Distributions static void run_test(void);
16*5e3eaea3SApple OSS Distributions
17*5e3eaea3SApple OSS Distributions static void
run_test(void)18*5e3eaea3SApple OSS Distributions run_test(void)
19*5e3eaea3SApple OSS Distributions {
20*5e3eaea3SApple OSS Distributions kern_return_t kr;
21*5e3eaea3SApple OSS Distributions uint64_t size, i;
22*5e3eaea3SApple OSS Distributions mach_zone_name_t *name = NULL;
23*5e3eaea3SApple OSS Distributions unsigned int nameCnt = 0;
24*5e3eaea3SApple OSS Distributions mach_zone_info_t *info = NULL;
25*5e3eaea3SApple OSS Distributions unsigned int infoCnt = 0;
26*5e3eaea3SApple OSS Distributions mach_memory_info_t *wiredInfo = NULL;
27*5e3eaea3SApple OSS Distributions unsigned int wiredInfoCnt = 0;
28*5e3eaea3SApple OSS Distributions const char kalloc_str[] = "kalloc.";
29*5e3eaea3SApple OSS Distributions const char type_str[] = "type";
30*5e3eaea3SApple OSS Distributions size_t kt_name_len = strlen(kalloc_str) + strlen(type_str);
31*5e3eaea3SApple OSS Distributions
32*5e3eaea3SApple OSS Distributions kr = mach_memory_info(mach_host_self(),
33*5e3eaea3SApple OSS Distributions &name, &nameCnt, &info, &infoCnt,
34*5e3eaea3SApple OSS Distributions &wiredInfo, &wiredInfoCnt);
35*5e3eaea3SApple OSS Distributions T_QUIET; T_ASSERT_MACH_SUCCESS(kr, "mach_memory_info");
36*5e3eaea3SApple OSS Distributions T_QUIET; T_ASSERT_EQ(nameCnt, infoCnt, "zone name and info counts don't match");
37*5e3eaea3SApple OSS Distributions
38*5e3eaea3SApple OSS Distributions /* Match the names of the kalloc zones against their element sizes. */
39*5e3eaea3SApple OSS Distributions for (i = 0; i < nameCnt; i++) {
40*5e3eaea3SApple OSS Distributions const char *z_name = &name[i].mzn_name;
41*5e3eaea3SApple OSS Distributions if (strncmp(z_name, kalloc_str, strlen(kalloc_str)) == 0) {
42*5e3eaea3SApple OSS Distributions const char *size_ptr = strrchr(z_name, '.') + 1;
43*5e3eaea3SApple OSS Distributions T_QUIET; T_ASSERT_NOTNULL(size_ptr, "couldn't find size in name");
44*5e3eaea3SApple OSS Distributions size = strtoul(size_ptr, NULL, 10);
45*5e3eaea3SApple OSS Distributions T_LOG("ZONE NAME: %-25s ELEMENT SIZE: %llu", z_name, size);
46*5e3eaea3SApple OSS Distributions T_QUIET; T_ASSERT_EQ(size, info[i].mzi_elem_size, "kalloc zone name and element size don't match");
47*5e3eaea3SApple OSS Distributions }
48*5e3eaea3SApple OSS Distributions }
49*5e3eaea3SApple OSS Distributions
50*5e3eaea3SApple OSS Distributions if ((name != NULL) && (nameCnt != 0)) {
51*5e3eaea3SApple OSS Distributions kr = vm_deallocate(mach_task_self(), (vm_address_t) name,
52*5e3eaea3SApple OSS Distributions (vm_size_t) (nameCnt * sizeof *name));
53*5e3eaea3SApple OSS Distributions T_QUIET; T_ASSERT_MACH_SUCCESS(kr, "vm_deallocate name");
54*5e3eaea3SApple OSS Distributions }
55*5e3eaea3SApple OSS Distributions
56*5e3eaea3SApple OSS Distributions if ((info != NULL) && (infoCnt != 0)) {
57*5e3eaea3SApple OSS Distributions kr = vm_deallocate(mach_task_self(), (vm_address_t) info,
58*5e3eaea3SApple OSS Distributions (vm_size_t) (infoCnt * sizeof *info));
59*5e3eaea3SApple OSS Distributions T_QUIET; T_ASSERT_MACH_SUCCESS(kr, "vm_deallocate info");
60*5e3eaea3SApple OSS Distributions }
61*5e3eaea3SApple OSS Distributions
62*5e3eaea3SApple OSS Distributions if ((wiredInfo != NULL) && (wiredInfoCnt != 0)) {
63*5e3eaea3SApple OSS Distributions kr = vm_deallocate(mach_task_self(), (vm_address_t) wiredInfo,
64*5e3eaea3SApple OSS Distributions (vm_size_t) (wiredInfoCnt * sizeof *wiredInfo));
65*5e3eaea3SApple OSS Distributions T_QUIET; T_ASSERT_MACH_SUCCESS(kr, "vm_deallocate wiredInfo");
66*5e3eaea3SApple OSS Distributions }
67*5e3eaea3SApple OSS Distributions
68*5e3eaea3SApple OSS Distributions T_END;
69*5e3eaea3SApple OSS Distributions }
70*5e3eaea3SApple OSS Distributions
71*5e3eaea3SApple OSS Distributions T_DECL( verify_kalloc_config,
72*5e3eaea3SApple OSS Distributions "verifies that the kalloc zones are configured correctly",
73*5e3eaea3SApple OSS Distributions T_META_ASROOT(true))
74*5e3eaea3SApple OSS Distributions {
75*5e3eaea3SApple OSS Distributions run_test();
76*5e3eaea3SApple OSS Distributions }
77