1*d4514f0bSApple OSS Distributions #include <unistd.h>
2*d4514f0bSApple OSS Distributions #include <stdio.h>
3*d4514f0bSApple OSS Distributions
4*d4514f0bSApple OSS Distributions #include <darwintest.h>
5*d4514f0bSApple OSS Distributions #include <darwintest_utils.h>
6*d4514f0bSApple OSS Distributions
7*d4514f0bSApple OSS Distributions T_GLOBAL_META(
8*d4514f0bSApple OSS Distributions /*
9*d4514f0bSApple OSS Distributions * We're going to corrupt shared library text, so don't
10*d4514f0bSApple OSS Distributions * run with other tests.
11*d4514f0bSApple OSS Distributions */
12*d4514f0bSApple OSS Distributions T_META_RUN_CONCURRENTLY(false),
13*d4514f0bSApple OSS Distributions T_META_REQUIRES_SYSCTL_NE("kern.page_protection_type", 2)
14*d4514f0bSApple OSS Distributions );
15*d4514f0bSApple OSS Distributions
16*d4514f0bSApple OSS Distributions /*
17*d4514f0bSApple OSS Distributions * No system(3c) on watchOS, so provide our own.
18*d4514f0bSApple OSS Distributions * returns -1 if fails to run
19*d4514f0bSApple OSS Distributions * returns 0 if process exits normally.
20*d4514f0bSApple OSS Distributions * returns +n if process exits due to signal N
21*d4514f0bSApple OSS Distributions */
22*d4514f0bSApple OSS Distributions static int
my_system(const char * command)23*d4514f0bSApple OSS Distributions my_system(const char *command)
24*d4514f0bSApple OSS Distributions {
25*d4514f0bSApple OSS Distributions pid_t pid;
26*d4514f0bSApple OSS Distributions int status = 0;
27*d4514f0bSApple OSS Distributions int signal = 0;
28*d4514f0bSApple OSS Distributions int err;
29*d4514f0bSApple OSS Distributions const char *argv[] = {
30*d4514f0bSApple OSS Distributions "/bin/sh",
31*d4514f0bSApple OSS Distributions "-c",
32*d4514f0bSApple OSS Distributions command,
33*d4514f0bSApple OSS Distributions NULL
34*d4514f0bSApple OSS Distributions };
35*d4514f0bSApple OSS Distributions
36*d4514f0bSApple OSS Distributions if (dt_launch_tool(&pid, (char **)(void *)argv, FALSE, NULL, NULL)) {
37*d4514f0bSApple OSS Distributions return -1;
38*d4514f0bSApple OSS Distributions }
39*d4514f0bSApple OSS Distributions
40*d4514f0bSApple OSS Distributions err = dt_waitpid(pid, &status, &signal, 30);
41*d4514f0bSApple OSS Distributions if (err) {
42*d4514f0bSApple OSS Distributions return 0;
43*d4514f0bSApple OSS Distributions }
44*d4514f0bSApple OSS Distributions
45*d4514f0bSApple OSS Distributions return signal;
46*d4514f0bSApple OSS Distributions }
47*d4514f0bSApple OSS Distributions
48*d4514f0bSApple OSS Distributions
49*d4514f0bSApple OSS Distributions /*
50*d4514f0bSApple OSS Distributions * The tests are run in the following order:
51*d4514f0bSApple OSS Distributions *
52*d4514f0bSApple OSS Distributions * - call foo
53*d4514f0bSApple OSS Distributions * - corrupt foo, then call foo
54*d4514f0bSApple OSS Distributions * - call foo
55*d4514f0bSApple OSS Distributions *
56*d4514f0bSApple OSS Distributions * - call atan
57*d4514f0bSApple OSS Distributions * - corrupt atan, then call atan
58*d4514f0bSApple OSS Distributions * - call atan
59*d4514f0bSApple OSS Distributions *
60*d4514f0bSApple OSS Distributions * The first and last of each should exit normally. The middle one should exit with SIGILL.
61*d4514f0bSApple OSS Distributions *
62*d4514f0bSApple OSS Distributions * atan() was picked as a shared region function that isn't likely used by any normal daemons.
63*d4514f0bSApple OSS Distributions */
64*d4514f0bSApple OSS Distributions T_DECL(text_corruption_recovery, "test detection/recovery of text corruption",
65*d4514f0bSApple OSS Distributions T_META_IGNORECRASHES(".*text_corruption_helper.*"),
66*d4514f0bSApple OSS Distributions T_META_ASROOT(true))
67*d4514f0bSApple OSS Distributions {
68*d4514f0bSApple OSS Distributions int ret;
69*d4514f0bSApple OSS Distributions
70*d4514f0bSApple OSS Distributions ret = my_system("./text_corruption_helper foo");
71*d4514f0bSApple OSS Distributions T_QUIET; T_ASSERT_EQ(ret, 0, "First call of foo");
72*d4514f0bSApple OSS Distributions
73*d4514f0bSApple OSS Distributions ret = my_system("./text_corruption_helper Xfoo");
74*d4514f0bSApple OSS Distributions T_QUIET; T_ASSERT_EQ(ret, SIGILL, "Call of corrupted foo");
75*d4514f0bSApple OSS Distributions
76*d4514f0bSApple OSS Distributions ret = my_system("./text_corruption_helper foo");
77*d4514f0bSApple OSS Distributions T_QUIET; T_ASSERT_EQ(ret, 0, "Fixed call of foo");
78*d4514f0bSApple OSS Distributions
79*d4514f0bSApple OSS Distributions ret = my_system("./text_corruption_helper atan");
80*d4514f0bSApple OSS Distributions T_QUIET; T_ASSERT_EQ(ret, 0, "First call of atan");
81*d4514f0bSApple OSS Distributions
82*d4514f0bSApple OSS Distributions ret = my_system("./text_corruption_helper Xatan");
83*d4514f0bSApple OSS Distributions T_QUIET; T_ASSERT_EQ(ret, SIGILL, "Call of corrupted atan");
84*d4514f0bSApple OSS Distributions
85*d4514f0bSApple OSS Distributions ret = my_system("./text_corruption_helper atan");
86*d4514f0bSApple OSS Distributions T_QUIET; T_ASSERT_EQ(ret, 0, "Fixed call of atan");
87*d4514f0bSApple OSS Distributions }
88