1*19c3b8c2SApple OSS Distributions #include <sys/sysctl.h>
2*19c3b8c2SApple OSS Distributions #include <signal.h>
3*19c3b8c2SApple OSS Distributions #include <darwintest.h>
4*19c3b8c2SApple OSS Distributions #include <darwintest_utils.h>
5*19c3b8c2SApple OSS Distributions
6*19c3b8c2SApple OSS Distributions T_GLOBAL_META(
7*19c3b8c2SApple OSS Distributions T_META_NAMESPACE("xnu.vm"),
8*19c3b8c2SApple OSS Distributions T_META_RADAR_COMPONENT_NAME("xnu"),
9*19c3b8c2SApple OSS Distributions T_META_RADAR_COMPONENT_VERSION("zalloc"),
10*19c3b8c2SApple OSS Distributions T_META_ASROOT(YES));
11*19c3b8c2SApple OSS Distributions
12*19c3b8c2SApple OSS Distributions static int64_t
run_sysctl_test(const char * t,int64_t value)13*19c3b8c2SApple OSS Distributions run_sysctl_test(const char *t, int64_t value)
14*19c3b8c2SApple OSS Distributions {
15*19c3b8c2SApple OSS Distributions char name[1024];
16*19c3b8c2SApple OSS Distributions int64_t result = 0;
17*19c3b8c2SApple OSS Distributions size_t s = sizeof(value);
18*19c3b8c2SApple OSS Distributions int rc;
19*19c3b8c2SApple OSS Distributions
20*19c3b8c2SApple OSS Distributions snprintf(name, sizeof(name), "debug.test.%s", t);
21*19c3b8c2SApple OSS Distributions rc = sysctlbyname(name, &result, &s, &value, s);
22*19c3b8c2SApple OSS Distributions T_ASSERT_POSIX_SUCCESS(rc, "sysctlbyname(%s)", t);
23*19c3b8c2SApple OSS Distributions return result;
24*19c3b8c2SApple OSS Distributions }
25*19c3b8c2SApple OSS Distributions
26*19c3b8c2SApple OSS Distributions T_DECL(basic_zone_test, "General zalloc test",
27*19c3b8c2SApple OSS Distributions T_META_CHECK_LEAKS(false))
28*19c3b8c2SApple OSS Distributions {
29*19c3b8c2SApple OSS Distributions T_EXPECT_EQ(1ull, run_sysctl_test("zone_basic_test", 0), "zone_basic_test");
30*19c3b8c2SApple OSS Distributions }
31*19c3b8c2SApple OSS Distributions
32*19c3b8c2SApple OSS Distributions T_DECL(read_only_zone_test, "Read-only zalloc test",
33*19c3b8c2SApple OSS Distributions T_META_CHECK_LEAKS(false))
34*19c3b8c2SApple OSS Distributions {
35*19c3b8c2SApple OSS Distributions T_EXPECT_EQ(1ull, run_sysctl_test("zone_ro_basic_test", 0), "zone_ro_basic_test");
36*19c3b8c2SApple OSS Distributions }
37*19c3b8c2SApple OSS Distributions
38*19c3b8c2SApple OSS Distributions T_DECL(zone_stress_test, "Zone stress test of edge cases",
39*19c3b8c2SApple OSS Distributions T_META_CHECK_LEAKS(false))
40*19c3b8c2SApple OSS Distributions {
41*19c3b8c2SApple OSS Distributions T_EXPECT_EQ(1ull, run_sysctl_test("zone_stress_test", 0), "zone_stress_test");
42*19c3b8c2SApple OSS Distributions }
43*19c3b8c2SApple OSS Distributions
44*19c3b8c2SApple OSS Distributions #define ZLOG_ZONE "data.kalloc.128"
45*19c3b8c2SApple OSS Distributions
46*19c3b8c2SApple OSS Distributions T_DECL(zlog_smoke_test, "check that zlog functions at all",
47*19c3b8c2SApple OSS Distributions T_META_BOOTARGS_SET("zlog1=" ZLOG_ZONE))
48*19c3b8c2SApple OSS Distributions {
49*19c3b8c2SApple OSS Distributions char *cmd[] = { "/usr/local/bin/zlog", "-l", "-z", ZLOG_ZONE, NULL };
50*19c3b8c2SApple OSS Distributions dispatch_semaphore_t sema = dispatch_semaphore_create(0);
51*19c3b8c2SApple OSS Distributions int status = 0;
52*19c3b8c2SApple OSS Distributions pid_t pid;
53*19c3b8c2SApple OSS Distributions
54*19c3b8c2SApple OSS Distributions pid = dt_launch_tool_pipe(cmd, false, NULL,
55*19c3b8c2SApple OSS Distributions ^bool (char *d, size_t s, dt_pipe_data_handler_context_t *ctx) {
56*19c3b8c2SApple OSS Distributions (void)ctx;
57*19c3b8c2SApple OSS Distributions if (strstr(d, "active refs") && strstr(d, "operation type: ")) {
58*19c3b8c2SApple OSS Distributions T_PASS("found line [%.*s]", (int)(s - 1), d);
59*19c3b8c2SApple OSS Distributions dispatch_semaphore_signal(sema);
60*19c3b8c2SApple OSS Distributions }
61*19c3b8c2SApple OSS Distributions return false;
62*19c3b8c2SApple OSS Distributions }, ^bool (char *d, size_t s, dt_pipe_data_handler_context_t *ctx) {
63*19c3b8c2SApple OSS Distributions /* Forward errors to stderror for debugging */
64*19c3b8c2SApple OSS Distributions (void)ctx;
65*19c3b8c2SApple OSS Distributions fwrite(d, 1, s, stderr);
66*19c3b8c2SApple OSS Distributions return false;
67*19c3b8c2SApple OSS Distributions }, BUFFER_PATTERN_LINE, NULL);
68*19c3b8c2SApple OSS Distributions
69*19c3b8c2SApple OSS Distributions dt_waitpid(pid, &status, NULL, 0);
70*19c3b8c2SApple OSS Distributions if (WIFEXITED(status)) {
71*19c3b8c2SApple OSS Distributions T_LOG("waitpid for %d returned with status %d",
72*19c3b8c2SApple OSS Distributions pid, WEXITSTATUS(status));
73*19c3b8c2SApple OSS Distributions } else {
74*19c3b8c2SApple OSS Distributions int sig = WTERMSIG(status);
75*19c3b8c2SApple OSS Distributions T_LOG("waitpid for %d killed by signal %d/%s",
76*19c3b8c2SApple OSS Distributions pid, sig, sys_signame[sig]);
77*19c3b8c2SApple OSS Distributions }
78*19c3b8c2SApple OSS Distributions T_ASSERT_TRUE(WIFEXITED(status) && WEXITSTATUS(status) == 0,
79*19c3b8c2SApple OSS Distributions "zlog exited cleanly");
80*19c3b8c2SApple OSS Distributions
81*19c3b8c2SApple OSS Distributions /* work around rdar://84948713 */
82*19c3b8c2SApple OSS Distributions T_ASSERT_EQ(dispatch_semaphore_wait(sema,
83*19c3b8c2SApple OSS Distributions dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC)), 0L,
84*19c3b8c2SApple OSS Distributions "found the line we wanted");
85*19c3b8c2SApple OSS Distributions dispatch_release(sema);
86*19c3b8c2SApple OSS Distributions }
87