mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
bfbc3a7570
Previously, the test was limited to only x86_64 macos. Remove that limitation by introducing the appropriate cross-platform typedef for mach_header and signing the pointer before casting it to a function pointer.
68 lines
1.9 KiB
C++
68 lines
1.9 KiB
C++
// RUN: %empty-directory(%t)
|
|
// RUN: %target-clang %s -std=c++11 -isysroot %sdk -o %t/main
|
|
// RUN: %target-codesign %t/main
|
|
// RUN: %target-build-swift %S/Inputs/at-main-struct-simple.swift -O -parse-as-library -emit-library -o %t/libHowdy.dylib -module-name Howdy
|
|
// RUN: %target-codesign %t/libHowdy.dylib
|
|
// RUN: %target-run %t/main %t/libHowdy.dylib | %FileCheck %s
|
|
|
|
// REQUIRES: VENDOR=apple
|
|
// REQUIRES: executable_test
|
|
// UNSUPPORTED: remote_run
|
|
|
|
#include <dlfcn.h>
|
|
#include <mach-o/dyld.h>
|
|
#include <mach-o/getsect.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <ptrauth.h>
|
|
|
|
#if __POINTER_WIDTH__ == 64
|
|
using mach_header_platform = mach_header_64;
|
|
#else
|
|
using mach_header_platform = mach_header;
|
|
#endif
|
|
|
|
int main(int argc, char *argv[]) {
|
|
if (argc != 2) {
|
|
printf("no argument!\n");
|
|
return 1;
|
|
}
|
|
auto *dylibPath = argv[1];
|
|
auto *handle = dlopen(dylibPath, RTLD_LAZY);
|
|
if (!handle) {
|
|
printf("no library!\n");
|
|
return 1;
|
|
}
|
|
|
|
using MainFunction = void(int, char *[]);
|
|
MainFunction *mainFunction = nullptr;
|
|
for (int index = 0; index < _dyld_image_count(); ++index) {
|
|
auto *imageName = _dyld_get_image_name(index);
|
|
if (strcmp(dylibPath, imageName)) {
|
|
printf("skipping %s\n", imageName);
|
|
continue;
|
|
}
|
|
auto *header =
|
|
reinterpret_cast<const mach_header_platform *>(_dyld_get_image_header(index));
|
|
size_t size;
|
|
auto *data = getsectiondata(header, "__TEXT", "__swift5_entry", &size);
|
|
int32_t offset = *reinterpret_cast<int32_t *>(data);
|
|
mainFunction = reinterpret_cast<MainFunction *>(
|
|
ptrauth_sign_unauthenticated(
|
|
reinterpret_cast<void *>(
|
|
reinterpret_cast<long>(data) + offset
|
|
),
|
|
ptrauth_key_function_pointer,
|
|
ptrauth_function_pointer_type_discriminator(MainFunction)
|
|
)
|
|
);
|
|
|
|
break;
|
|
}
|
|
if (!mainFunction) {
|
|
printf("no function!");
|
|
return 1;
|
|
}
|
|
mainFunction(argc, argv); // CHECK: howdy mundo
|
|
}
|