Mangle Objective-C interfaces correctly (where correctly refers to what gcc does)

git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@66349 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/lib/CodeGen/Mangle.cpp b/lib/CodeGen/Mangle.cpp
index 525f0fe1d62207b98eb7f0a36626d428b17ee58c..490936f625d0b8a40951626593e4b5a7859c30a8 100644 (file)
--- a/lib/CodeGen/Mangle.cpp
+++ b/lib/CodeGen/Mangle.cpp
@@ -18,6 +18,7 @@
 #include "clang/AST/ASTContext.h"
 #include "clang/AST/Decl.h"
 #include "clang/AST/DeclCXX.h"
+#include "clang/AST/DeclObjC.h"
 #include "clang/Basic/SourceManager.h"
 #include "llvm/Support/Compiler.h"
 #include "llvm/Support/raw_ostream.h"
@@ -49,6 +50,7 @@ namespace {
     void mangleType(const ArrayType *T);
     void mangleType(const MemberPointerType *T);
     void mangleType(const TemplateTypeParmType *T);
+    void mangleType(const ObjCInterfaceType *T);
     void mangleExpression(Expr *E);
   };
 }
@@ -377,6 +379,9 @@ void CXXNameMangler::mangleType(QualType T) {
     // GNU extension: vector types
     Out << "U8__vector";
     mangleType(VT->getElementType());
+  } else if (const ObjCInterfaceType *IT = 
+             dyn_cast<ObjCInterfaceType>(T.getTypePtr())) {
+    mangleType(IT);
   }
   // FIXME:  ::= G <type>   # imaginary (C 2000)
   // FIXME:  ::= U <source-name> <type>     # vendor extended type qualifier
@@ -504,6 +509,10 @@ void CXXNameMangler::mangleType(const TemplateTypeParmType *T) {
     Out << 'T' << (T->getIndex() - 1) << '_';
 }
 
+void CXXNameMangler::mangleType(const ObjCInterfaceType *T) {
+  mangleSourceName(T->getDecl()->getIdentifier());
+}
+
 void CXXNameMangler::mangleExpression(Expr *E) {
   assert(false && "Cannot mangle expressions yet");
 }

diff --git a/test/CodeGenObjC/overloadable.m b/test/CodeGenObjC/overloadable.m
new file mode 100644 (file)
index 0000000..5caaca9
--- /dev/null
+++ b/test/CodeGenObjC/overloadable.m
@@ -0,0 +1,9 @@
+// RUN: clang -emit-llvm %s -o %t && 
+
+@class C;
+
+// RUN: grep _Z1fP11objc_object %t | count 1 && 
+void __attribute__((overloadable)) f(C *c) { }
+
+// RUN: grep _Z1fP1C | count 1
+void __attribute__((overloadable)) f(id c) { }