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