[OrcJIT] Support arm64e/ptrauth.

Author: ahmedbougacha

llvm/include/llvm/ExecutionEngine/JITSymbol.h

@@ -62,12 +62,26 @@ template <typename T> T jitTargetAddressToFunction(JITTargetAddress Addr) {
   static_assert(std::is_pointer<T>::value &&
                     std::is_function<std::remove_pointer_t<T>>::value,
                 "T must be a function pointer type");
-  return jitTargetAddressToPointer<T>(Addr);
+  auto VoidPtrResult = jitTargetAddressToPointer<void*>(Addr);
+
+  // If this process uses ptrauth then sign the pointer.
+#if __has_feature(ptrauth_calls)
+  VoidPtrResult =
+    __builtin_ptrauth_sign_unauthenticated(VoidPtrResult, 0, 0);
+#endif
+
+  return reinterpret_cast<T>(VoidPtrResult);
 }
 
 /// Convert a pointer to a JITTargetAddress.
 template <typename T> JITTargetAddress pointerToJITTargetAddress(T *Ptr) {
-  return static_cast<JITTargetAddress>(reinterpret_cast<uintptr_t>(Ptr));
+  // If this process uses ptrauth then sign the pointer.
+  const void *Tmp = reinterpret_cast<const void*>(Ptr);
+#if __has_feature(ptrauth_calls)
+  Tmp = __builtin_ptrauth_strip(Tmp, 0);
+#endif
+
+  return static_cast<JITTargetAddress>(reinterpret_cast<uintptr_t>(Tmp));
 }
 
 /// Flags for symbols in the JIT.

llvm/lib/ExecutionEngine/JITLink/MachO_arm64.cpp

@@ -651,7 +651,6 @@ struct CompactUnwindTraits_MachO_arm64
 
 void link_MachO_arm64(std::unique_ptr<LinkGraph> G,
                       std::unique_ptr<JITLinkContext> Ctx) {
-
   PassConfiguration Config;
 
   if (Ctx->shouldAddDefaultTargetPasses(G->getTargetTriple())) {

llvm/lib/ExecutionEngine/Orc/ExecutionUtils.cpp

@@ -262,8 +262,13 @@ Error DynamicLibrarySearchGenerator::tryToGenerate(
 
     std::string Tmp((*Name).data() + HasGlobalPrefix,
                     (*Name).size() - HasGlobalPrefix);
-    if (void *P = Dylib.getAddressOfSymbol(Tmp.c_str()))
+    if (void *P = Dylib.getAddressOfSymbol(Tmp.c_str())) {
+      // If this process uses ptrauth then strip the pointer.
+#if __has_feature(ptrauth_calls)
+      P = __builtin_ptrauth_strip(P, 0);
+#endif
       NewSymbols[Name] = {ExecutorAddr::fromPtr(P), JITSymbolFlags::Exported};
+    }
   }
 
   if (NewSymbols.empty())

llvm/lib/ExecutionEngine/Orc/LazyReexports.cpp

@@ -116,6 +116,11 @@ createLocalLazyCallThroughManager(const Triple &T, ExecutionSession &ES,
 
   case Triple::aarch64:
   case Triple::aarch64_32:
+    // arm64e doesn't support lazy call.
+    if (T.isArm64e())
+      return make_error<StringError>(
+          std::string("No callback manager available for ") + T.str(),
+          inconvertibleErrorCode());
     return LocalLazyCallThroughManager::Create<OrcAArch64>(ES,
                                                            ErrorHandlerAddr);