Coverage mapping: fix mapping for objective-c message expression

llvm-svn: 216081

1 files changed, 10 insertions, 0 deletions

diff --git a/clang/lib/CodeGen/CoverageMappingGen.cpp b/clang/lib/CodeGen/CoverageMappingGen.cpp
index 836d09f7343..4c5f3751167 100644
--- a/clang/lib/CodeGen/CoverageMappingGen.cpp
+++ b/clang/lib/CodeGen/CoverageMappingGen.cpp
@@ -1042,6 +1042,16 @@ struct CounterCoverageMappingBuilder
   void VisitImaginaryLiteral(const ImaginaryLiteral *E) {
     mapToken(E->getLocStart());
   }
+
+  void VisitObjCMessageExpr(const ObjCMessageExpr *E) {
+    mapToken(E->getLeftLoc());
+    for (Stmt::const_child_range I = static_cast<const Stmt*>(E)->children(); I;
+         ++I) {
+      if (*I)
+        this->Visit(*I);
+    }
+    mapToken(E->getRightLoc());
+  }
 };
 }