Only auth indirect call operands

TUM-DSE · Apr 30, 2024 · 0a8fd50 · 0a8fd50
1 parent 9550001
commit 0a8fd50
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/llvm/lib/Target/WebAssembly/WebAssemblyPointerAuth.cpp b/llvm/lib/Target/WebAssembly/WebAssemblyPointerAuth.cpp
@@ -211,14 +211,17 @@ bool WebAssemblyPointerAuth::runOnModule(Module &M) {
 }
 
 void WebAssemblyPointerAuth::visitCallBase(llvm::CallBase &CB) {
+  if (!CB.isIndirectCall()) {
+    return;
+  }
   auto *Op = CB.getCalledOperand();
   if (isa<Function>(Op)) {
-    return;
+    llvm_unreachable("Expected Op to be a function pointer.");
   }
 
   // otherwise, we assume Op is a function pointer
   if (!Op->getType()->isPointerTy()) {
-    errs() << "Expected Op to be a function pointer.";
+    llvm_unreachable("Expected Op to be a function pointer.");
   }
 
   auto *PointerAuthIntr =