Merge pull request #42116 from aschwaighofer/swift_llvm_merge_funcs_objc_msgSend

LLVMMergeFunctions: Must not merge function calls to `objc_sendMsg$...`

Author: aschwaighofer

lib/LLVMPasses/LLVMMergeFunctions.cpp

@@ -120,9 +120,16 @@ static bool isCalleeOperand(const CallInst *CI, unsigned opIdx) {
 static bool canParameterizeCallOperand(const CallInst *CI, unsigned opIdx) {
   if (CI->isInlineAsm())
     return false;
-  if (Function *Callee = CI->getCalledFunction()) {
+
+  Function *Callee = CI->getCalledOperand() ?
+      dyn_cast_or_null<Function>(CI->getCalledOperand()->stripPointerCasts()) :
+      nullptr;
+  if (Callee) {
     if (Callee->isIntrinsic())
       return false;
+    // objc_msgSend stubs must be called, and can't have their address taken.
+    if (Callee->getName().startswith("objc_msgSend$"))
+      return false;
   }
   if (isCalleeOperand(CI, opIdx) &&
       CI->getOperandBundle(LLVMContext::OB_ptrauth).hasValue()) {