From 545d00645c224e3734a11a6b9b95c8f153de1d8d Mon Sep 17 00:00:00 2001
From: Dan Gohman <gohman@apple.com>
Date: Mon, 12 Jul 2010 18:02:04 +0000
Subject: [PATCH] Add a lint check for mismatched return types, inspired by
 PR6944.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@108162 91177308-0d34-0410-b5e6-96231b3b80d8
---
 lib/Analysis/Lint.cpp | 4 ++++
 test/Other/lint.ll    | 9 +++++++++
 2 files changed, 13 insertions(+)

diff --git a/lib/Analysis/Lint.cpp b/lib/Analysis/Lint.cpp
index 60654e61701..9f1b30d2cf4 100644
--- a/lib/Analysis/Lint.cpp
+++ b/lib/Analysis/Lint.cpp
@@ -225,6 +225,10 @@ void Lint::visitCallSite(CallSite CS) {
             "Undefined behavior: Call argument count mismatches callee "
             "argument count", &I);
 
+    Assert1(FT->getReturnType() == I.getType(),
+            "Undefined behavior: Call return type mismatches "
+            "callee return type", &I);
+
     // Check argument types (in case the callee was casted) and attributes.
     // TODO: Verify that caller and callee attributes are compatible.
     Function::arg_iterator PI = F->arg_begin(), PE = F->arg_end();
diff --git a/test/Other/lint.ll b/test/Other/lint.ll
index 9a10abf3c70..dee3d11d2fb 100644
--- a/test/Other/lint.ll
+++ b/test/Other/lint.ll
@@ -154,3 +154,12 @@ exit:
   %x = volatile load i32* %t3
   br label %exit
 }
+
+; CHECK: Call return type mismatches callee return type
+%struct = type { double, double }
+declare i32 @nonstruct_callee() nounwind
+define void @struct_caller() nounwind {
+entry:
+  call %struct bitcast (i32 ()* @foo to %struct ()*)()
+  ret void
+}