xref: /xnu-8792.81.2/tests/posix_spawn_alt_rosetta.c (revision 19c3b8c28c31cb8130e034cfb5df6bf9ba342d90)
1 #include <darwintest.h>
2 #include <mach-o/dyld.h>
3 #include <spawn.h>
4 #include <unistd.h>
5 #include <sys/wait.h>
6 #include <fcntl.h>
7 #include <stdlib.h>
8 
9 #include <TargetConditionals.h>
10 
11 T_DECL(posix_spawn_alt_rosetta, "verify posix_spawn_set_alt_rosetta_np switches to alternative rosetta runtime",
12     T_META_ASROOT(true), T_META_REQUIRES_SYSCTL_EQ("kern.development", 1))
13 {
14 #if TARGET_OS_OSX && defined(__arm64__)
15 	int ret, pid;
16 	posix_spawnattr_t spawnattr;
17 	char path[1024];
18 	uint32_t size = sizeof(path);
19 	cpu_type_t cpuprefs[] = { CPU_TYPE_X86_64 };
20 	cpu_type_t subcpuprefs[] = { CPU_SUBTYPE_ANY };
21 	int wait_ret = 0;
22 
23 	if (access("/Library/Apple/usr/libexec/oah/libRosettaRuntime", O_RDONLY) != 0) {
24 		T_SKIP("Rosetta not installed");
25 		return;
26 	}
27 
28 	if (access("/usr/local/libexec/rosetta/runtime_internal", O_RDONLY) != 0) {
29 		system("mkdir -p /usr/local/libexec/oah");
30 		system("cp /Library/Apple/usr/libexec/rosetta/runtime /usr/local/libexec/rosetta/runtime_internal");
31 	}
32 
33 	T_QUIET; T_ASSERT_EQ(_NSGetExecutablePath(path, &size), 0, NULL);
34 	T_QUIET; T_ASSERT_LT(strlcat(path, "_helper", size), (unsigned long)size, NULL);
35 
36 	ret = posix_spawnattr_init(&spawnattr);
37 	T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "posix_spawnattr_init");
38 
39 	// 1) Run natively
40 
41 	ret = posix_spawn(&pid, path, NULL, &spawnattr, NULL, NULL);
42 	T_ASSERT_EQ(ret, 0, "posix_spawn should succeed");
43 	ret = waitpid(pid, &wait_ret, 0);
44 	T_QUIET; T_ASSERT_EQ(ret, pid, "child pid");
45 	T_QUIET; T_ASSERT_EQ(WIFEXITED(wait_ret), 1, "child process should have called exit()");
46 	T_ASSERT_EQ(WEXITSTATUS(wait_ret), 0, "running natively should return 0");
47 
48 	// 2) Set archpref to run under Rosetta
49 
50 	ret = posix_spawnattr_setarchpref_np(&spawnattr, sizeof(cpuprefs) / sizeof(cpuprefs[0]), cpuprefs, subcpuprefs, NULL);
51 	T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "posix_spawnattr_setarchpref_np");
52 
53 	ret = posix_spawn(&pid, path, NULL, &spawnattr, NULL, NULL);
54 	T_ASSERT_EQ(ret, 0, "posix_spawn should succeed");
55 	ret = waitpid(pid, &wait_ret, 0);
56 	T_QUIET; T_ASSERT_EQ(ret, pid, "child pid");
57 	T_QUIET; T_ASSERT_EQ(WIFEXITED(wait_ret), 1, "child process should have called exit()");
58 	T_ASSERT_EQ(WEXITSTATUS(wait_ret), 1, "running in rosetta should return 1");
59 
60 	// 3) Request alternative Rosetta runtime
61 
62 	ret = posix_spawnattr_set_alt_rosetta_np(&spawnattr, 0);
63 	T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "posix_spawnattr_setarchpref_np");
64 
65 	ret = posix_spawn(&pid, path, NULL, &spawnattr, NULL, NULL);
66 	T_ASSERT_EQ(ret, 0, "posix_spawn should succeed");
67 	ret = waitpid(pid, &wait_ret, 0);
68 	T_QUIET; T_ASSERT_EQ(ret, pid, "child pid");
69 	T_QUIET; T_ASSERT_EQ(WIFEXITED(wait_ret), 1, "child process should have called exit()");
70 	T_ASSERT_EQ(WEXITSTATUS(wait_ret), 2, "running with alternative rosetta runtime should return 2");
71 
72 	ret = posix_spawnattr_destroy(&spawnattr);
73 	T_QUIET; T_ASSERT_EQ(ret, 0, "posix_spawnattr_destroy");
74 #else
75 	T_SKIP("Not arm64 macOS");
76 #endif
77 }
78