1*a325d9c4SApple OSS Distributions #include <darwintest.h>
2*a325d9c4SApple OSS Distributions #include <mach-o/dyld.h>
3*a325d9c4SApple OSS Distributions #include <spawn.h>
4*a325d9c4SApple OSS Distributions #include <unistd.h>
5*a325d9c4SApple OSS Distributions #include <sys/wait.h>
6*a325d9c4SApple OSS Distributions
7*a325d9c4SApple OSS Distributions #include <TargetConditionals.h>
8*a325d9c4SApple OSS Distributions
9*a325d9c4SApple OSS Distributions static void
run_test(const char * name,cpu_type_t type,cpu_subtype_t subtype)10*a325d9c4SApple OSS Distributions run_test(const char *name, cpu_type_t type, cpu_subtype_t subtype)
11*a325d9c4SApple OSS Distributions {
12*a325d9c4SApple OSS Distributions int ret, pid;
13*a325d9c4SApple OSS Distributions posix_spawnattr_t spawnattr;
14*a325d9c4SApple OSS Distributions char path[1024];
15*a325d9c4SApple OSS Distributions uint32_t size = sizeof(path);
16*a325d9c4SApple OSS Distributions cpu_type_t cpuprefs[] = { type };
17*a325d9c4SApple OSS Distributions cpu_type_t subcpuprefs[] = { subtype };
18*a325d9c4SApple OSS Distributions
19*a325d9c4SApple OSS Distributions T_QUIET; T_ASSERT_EQ(_NSGetExecutablePath(path, &size), 0, NULL);
20*a325d9c4SApple OSS Distributions T_QUIET; T_ASSERT_LT(strlcat(path, "_helper", size), (unsigned long)size, NULL);
21*a325d9c4SApple OSS Distributions
22*a325d9c4SApple OSS Distributions ret = posix_spawnattr_init(&spawnattr);
23*a325d9c4SApple OSS Distributions T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "%s: posix_spawnattr_init", name);
24*a325d9c4SApple OSS Distributions
25*a325d9c4SApple OSS Distributions ret = posix_spawnattr_setarchpref_np(&spawnattr, sizeof(cpuprefs) / sizeof(cpuprefs[0]), cpuprefs, subcpuprefs, NULL);
26*a325d9c4SApple OSS Distributions T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "%s: posix_spawnattr_setarchpref_np", name);
27*a325d9c4SApple OSS Distributions
28*a325d9c4SApple OSS Distributions ret = posix_spawn(&pid, path, NULL, &spawnattr, NULL, NULL);
29*a325d9c4SApple OSS Distributions T_ASSERT_EQ(ret, 0, "%s: posix_spawn should succeed", name);
30*a325d9c4SApple OSS Distributions
31*a325d9c4SApple OSS Distributions int wait_ret = 0;
32*a325d9c4SApple OSS Distributions ret = waitpid(pid, &wait_ret, 0);
33*a325d9c4SApple OSS Distributions T_QUIET; T_ASSERT_EQ(ret, pid, "%s: child pid", name);
34*a325d9c4SApple OSS Distributions
35*a325d9c4SApple OSS Distributions T_QUIET; T_ASSERT_EQ(WIFEXITED(wait_ret), 1, "%s: child process should have called exit()", name);
36*a325d9c4SApple OSS Distributions
37*a325d9c4SApple OSS Distributions if (subtype != CPU_SUBTYPE_ANY) {
38*a325d9c4SApple OSS Distributions T_ASSERT_EQ(WEXITSTATUS(wait_ret), subtype, "%s: child process should be running with %d subtype", name, subtype);
39*a325d9c4SApple OSS Distributions }
40*a325d9c4SApple OSS Distributions
41*a325d9c4SApple OSS Distributions ret = posix_spawnattr_destroy(&spawnattr);
42*a325d9c4SApple OSS Distributions T_QUIET; T_ASSERT_EQ(ret, 0, "%s: posix_spawnattr_destroy", name);
43*a325d9c4SApple OSS Distributions }
44*a325d9c4SApple OSS Distributions
45*a325d9c4SApple OSS Distributions T_DECL(posix_spawn_archpref, "verify posix_spawn_setarchpref_np can select slices")
46*a325d9c4SApple OSS Distributions {
47*a325d9c4SApple OSS Distributions #if defined(__x86_64__)
48*a325d9c4SApple OSS Distributions run_test("x86_64", CPU_TYPE_X86_64, CPU_SUBTYPE_X86_64_ALL);
49*a325d9c4SApple OSS Distributions #endif /* defined(__x86_64__) */
50*a325d9c4SApple OSS Distributions /* macOS does not build a arm64 slice */
51*a325d9c4SApple OSS Distributions #if defined(__arm64__) && defined(__LP64__) && !TARGET_OS_OSX
52*a325d9c4SApple OSS Distributions run_test("arm64", CPU_TYPE_ARM64, CPU_SUBTYPE_ARM64_ALL);
53*a325d9c4SApple OSS Distributions #endif /* defined(__arm64__) && defined(__LP64__) */
54*a325d9c4SApple OSS Distributions #if defined(__arm64e__)
55*a325d9c4SApple OSS Distributions run_test("arm64e", CPU_TYPE_ARM64, CPU_SUBTYPE_ARM64E);
56*a325d9c4SApple OSS Distributions #endif /* defined(__arm64e__) */
57*a325d9c4SApple OSS Distributions
58*a325d9c4SApple OSS Distributions #if defined(__x86_64__)
59*a325d9c4SApple OSS Distributions run_test("any (x86_64)", CPU_TYPE_X86_64, CPU_SUBTYPE_ANY);
60*a325d9c4SApple OSS Distributions #elif defined(__arm64__) && defined(__LP64__)
61*a325d9c4SApple OSS Distributions run_test("any (arm64)", CPU_TYPE_ARM64, CPU_SUBTYPE_ANY);
62*a325d9c4SApple OSS Distributions #elif defined(__arm64__)
63*a325d9c4SApple OSS Distributions run_test("any (arm64_32)", CPU_TYPE_ARM64_32, CPU_SUBTYPE_ANY);
64*a325d9c4SApple OSS Distributions #elif defined(__arm__)
65*a325d9c4SApple OSS Distributions run_test("any (arm)", CPU_TYPE_ARM, CPU_SUBTYPE_ANY);
66*a325d9c4SApple OSS Distributions #else
67*a325d9c4SApple OSS Distributions #error unknown architecture
68*a325d9c4SApple OSS Distributions #endif
69*a325d9c4SApple OSS Distributions }
70