polish the -finstrument-functions implementation, patch by Nelson Elhage!

llvm-svn: 106618

1 files changed, 8 insertions, 3 deletions

diff --git a/clang/lib/CodeGen/CodeGenFunction.cpp b/clang/lib/CodeGen/CodeGenFunction.cpp
index 8a0b41a7cad..704de3531c8 100644
--- a/clang/lib/CodeGen/CodeGenFunction.cpp
+++ b/clang/lib/CodeGen/CodeGenFunction.cpp
@@ -180,11 +180,14 @@ void CodeGenFunction::EmitFunctionInstrumentation(const char *Fn) {
   if (!ShouldInstrumentFunction())
     return;
 
+  const llvm::PointerType *PointerTy;
   const llvm::FunctionType *FunctionTy;
   std::vector<const llvm::Type*> ProfileFuncArgs;
 
-  ProfileFuncArgs.push_back(CurFn->getType());
-  ProfileFuncArgs.push_back(llvm::Type::getInt8PtrTy(VMContext));
+  // void __cyg_profile_func_{enter,exit} (void *this_fn, void *call_site);
+  PointerTy = llvm::Type::getInt8PtrTy(VMContext);
+  ProfileFuncArgs.push_back(PointerTy);
+  ProfileFuncArgs.push_back(PointerTy);
   FunctionTy = llvm::FunctionType::get(
     llvm::Type::getVoidTy(VMContext),
     ProfileFuncArgs, false);
@@ -195,7 +198,9 @@ void CodeGenFunction::EmitFunctionInstrumentation(const char *Fn) {
     llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), 0),
     "callsite");
 
-  Builder.CreateCall2(F, CurFn, CallSite);
+  Builder.CreateCall2(F,
+                      llvm::ConstantExpr::getBitCast(CurFn, PointerTy),
+                      CallSite);
 }
 
 void CodeGenFunction::StartFunction(GlobalDecl GD, QualType RetTy,