1 /*
2  * Copyright (c) 2023 Apple Computer, Inc. All rights reserved.
3  *
4  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
5  *
6  * This file contains Original Code and/or Modifications of Original Code
7  * as defined in and that are subject to the Apple Public Source License
8  * Version 2.0 (the 'License'). You may not use this file except in
9  * compliance with the License. The rights granted to you under the License
10  * may not be used to create, or enable the creation or redistribution of,
11  * unlawful or unlicensed copies of an Apple operating system, or to
12  * circumvent, violate, or enable the circumvention or violation of, any
13  * terms of an Apple operating system software license agreement.
14  *
15  * Please obtain a copy of the License at
16  * http://www.opensource.apple.com/apsl/ and read it before using this file.
17  *
18  * The Original Code and all software distributed under the License are
19  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
20  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
22  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23  * Please see the License for the specific language governing rights and
24  * limitations under the License.
25  *
26  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
27  */
28 
29 #include <darwintest.h>
30 
31 #include <fcntl.h>
32 #include <stdlib.h>
33 #include <spawn.h>
34 #include <string.h>
35 #include <unistd.h>
36 #include <sys/codesign.h>
37 #include <sys/mman.h>
38 #include <sys/stat.h>
39 #include <sys/wait.h>
40 
41 T_GLOBAL_META(
42 	T_META_NAMESPACE("xnu.vm"),
43 	T_META_RADAR_COMPONENT_NAME("xnu"),
44 	T_META_RADAR_COMPONENT_VERSION("VM"));
45 
46 T_DECL(vm_tainted_executable, "Test that a tainted executable gets killed", T_META_TAG_VM_PREFERRED)
47 {
48 	char tmp_path[] = "/tmp/hell0-XXXXXX";
49 	int fd1, fd2;
50 	struct stat fs;
51 	char *mapaddr1;
52 	size_t fsize;
53 	char *big_sp, *big_cp, *big_ep, *little_cp;
54 	size_t little_len;
55 	char *child_argv[2];
56 	pid_t child_pid;
57 	int child_status;
58 	int cs_status;
59 
60 	T_SETUPBEGIN;
61 	/* copy "./hello" to "/tmp/hell0" */
62 	fd1 = open("./hello", O_RDONLY);
63 	T_QUIET; T_ASSERT_POSIX_SUCCESS(fd1, "open(./hello)");
64 	fd2 = mkstemp(tmp_path);
65 	T_QUIET; T_ASSERT_POSIX_SUCCESS(fd2, "mkstemp(%s)", tmp_path);
66 	T_QUIET; T_ASSERT_POSIX_SUCCESS(fstat(fd1, &fs), NULL);
67 	fsize = (size_t)fs.st_size;
68 	mapaddr1 = mmap(NULL, fsize, PROT_READ, MAP_FILE | MAP_PRIVATE, fd1, 0);
69 	T_QUIET; T_ASSERT_NOTNULL(mapaddr1, NULL);
70 	T_QUIET; T_ASSERT_POSIX_SUCCESS(write(fd2, mapaddr1, fsize), NULL);
71 	/* change "hello, world!" to "hell0, world!" */
72 	big_sp = &mapaddr1[0]; /* start pointer in "big" byte string */
73 	big_ep = &mapaddr1[fsize]; /* end pointer in "big" byte string */
74 	little_cp = "hello, world!"; /* little byte string */
75 	little_len = strlen(little_cp); /* length of little byte string */
76 	big_cp = big_sp; /* start pointer in "big" byte string */
77 	for (;;) {
78 		char zero = '0';
79 		big_cp = memmem(big_cp, (size_t)(big_ep - big_cp),
80 		    little_cp, little_len);
81 		if (big_cp == NULL) {
82 			break;
83 		}
84 		T_LOG("found string at offset 0x%llx", (off_t) (big_cp - big_sp));
85 		T_QUIET; T_ASSERT_POSIX_SUCCESS(pwrite(fd2, &zero, 1,
86 		    (big_cp - big_sp + 4)), NULL);
87 		big_cp += little_len;
88 	}
89 	/* make the new binary "r-x" */
90 	T_QUIET; T_ASSERT_POSIX_SUCCESS(fchmod(fd2, S_IRUSR | S_IXUSR), NULL);
91 	/* cleanup */
92 	T_QUIET; T_ASSERT_POSIX_SUCCESS(close(fd1), NULL);
93 	T_QUIET; T_ASSERT_POSIX_SUCCESS(close(fd2), NULL);
94 	T_QUIET; T_ASSERT_POSIX_SUCCESS(munmap(mapaddr1, fsize), NULL);
95 	T_SETUPEND;
96 	/* spawn the newly-tainted binary */
97 	T_LOG("launching '%s'", tmp_path);
98 	child_argv[0] = tmp_path;
99 	child_argv[1] = NULL;
100 	T_QUIET; T_ASSERT_POSIX_SUCCESS(posix_spawn(&child_pid, tmp_path, NULL, NULL, child_argv, NULL), NULL);
101 	/* check our code-signing policy, assuming the child has same policy */
102 	T_QUIET; T_ASSERT_POSIX_SUCCESS(csops(getpid(), CS_OPS_STATUS, &cs_status, sizeof(cs_status)), NULL);
103 	T_LOG("parent %d cs status 0x%x CS_KILL:%s", getpid(), cs_status,
104 	    (cs_status & CS_KILL) ? "yes" : "no");
105 	/* get child's exit status */
106 	T_QUIET; T_ASSERT_POSIX_SUCCESS(waitpid(child_pid, &child_status, 0), NULL);
107 	T_LOG("child %d exit status 0x%x", child_pid, child_status);
108 	/* we no longer need our modified binary */
109 	T_QUIET; T_ASSERT_POSIX_SUCCESS(unlink(tmp_path), NULL);
110 	if (cs_status & CS_KILL) {
111 		/* check that child got SIGKILL */
112 		T_QUIET; T_ASSERT_TRUE(WIFSIGNALED(child_status), NULL);
113 		T_QUIET; T_ASSERT_TRUE(WTERMSIG(child_status) == SIGKILL, NULL);
114 		T_PASS("enforced process launched from modified binary got SIGKILL");
115 	} else {
116 		/* check that child exited with 0 */
117 		T_QUIET; T_ASSERT_TRUE(WIFEXITED(child_status), NULL);
118 		T_QUIET; T_ASSERT_TRUE(WEXITSTATUS(child_status) == 0, NULL);
119 		T_PASS("non-enforced process launched from modified binary exited with 0");
120 	}
121 }
122