From 77927cc33c09de292aed3cfc15916d2ae13ad884 Mon Sep 17 00:00:00 2001
From: Chris Lattner <sabre@nondot.org>
Date: Fri, 25 Jan 2008 19:43:26 +0000
Subject: [PATCH] Add support for dispatching an objc message to a variable in
 an initializer list.

llvm-svn: 46367
---
 clang/Parse/ParseInit.cpp     | 17 +++++++++++++++--
 clang/Parse/ParseObjc.cpp     |  2 +-
 clang/test/Parser/objc-init.m |  8 ++++++--
 3 files changed, 22 insertions(+), 5 deletions(-)

diff --git a/clang/Parse/ParseInit.cpp b/clang/Parse/ParseInit.cpp
index 9c5b6d8921b4..45cf86e5b448 100644
--- a/clang/Parse/ParseInit.cpp
+++ b/clang/Parse/ParseInit.cpp
@@ -94,6 +94,8 @@ Parser::ExprResult Parser::ParseInitializerWithPotentialDesignator() {
       // If Objective-C is enabled and this is a typename or other identifier
       // receiver, parse this as a message send expression.
       if (getLang().ObjC1 && isTokObjCMessageIdentifierReceiver()) {
+        // FIXME: Emit ext_gnu_missing_equal_designator for inits like
+        // [4][foo bar].
         IdentifierInfo *Name = Tok.getIdentifierInfo();
         ConsumeToken();
         ExprResult R = ParseObjCMessageExpressionBody(StartLoc, Name, 0);
@@ -101,14 +103,25 @@ Parser::ExprResult Parser::ParseInitializerWithPotentialDesignator() {
       }
       
       // Note that we parse this as an assignment expression, not a constant
-      // expression (allowing *=, =, etc).  Sema needs to validate that the
-      // expression is a constant.
+      // expression (allowing *=, =, etc) to handle the objc case.  Sema needs
+      // to validate that the expression is a constant.
       ExprResult Idx = ParseAssignmentExpression();
       if (Idx.isInvalid) {
         SkipUntil(tok::r_square);
         return Idx;
       }
       
+      // Given an expression, we could either have a designator (if the next
+      // tokens are '...' or ']' or an objc message send.  If this is an objc
+      // message send, handle it now.
+      if (getLang().ObjC1 && Tok.isNot(tok::ellipsis) && 
+          Tok.isNot(tok::r_square)) {
+        // FIXME: Emit ext_gnu_missing_equal_designator for inits like
+        // [4][foo bar].
+        ExprResult R = ParseObjCMessageExpressionBody(StartLoc, 0, Idx.Val);
+        return ParsePostfixExpressionSuffix(R);
+      }
+      
       // Handle the gnu array range extension.
       if (Tok.is(tok::ellipsis)) {
         Diag(Tok, diag::ext_gnu_array_range);
diff --git a/clang/Parse/ParseObjc.cpp b/clang/Parse/ParseObjc.cpp
index e100c259f9f5..c07a16ca4774 100644
--- a/clang/Parse/ParseObjc.cpp
+++ b/clang/Parse/ParseObjc.cpp
@@ -1276,7 +1276,7 @@ Parser::ExprResult Parser::ParseObjCMessageExpression() {
   ExprResult Res = ParseAssignmentExpression();
   if (Res.isInvalid) {
     Diag(Tok, diag::err_invalid_receiver_to_message);
-    SkipUntil(tok::identifier);
+    SkipUntil(tok::r_square);
     return Res;
   }
   return ParseObjCMessageExpressionBody(LBracLoc, 0, Res.Val);
diff --git a/clang/test/Parser/objc-init.m b/clang/test/Parser/objc-init.m
index 43e6c2e26ec3..ce7acaf92534 100644
--- a/clang/test/Parser/objc-init.m
+++ b/clang/test/Parser/objc-init.m
@@ -3,11 +3,15 @@
 
 @interface NSNumber;
 - () METH;
-
 @end
 
-int main() {
+void test1() {
 	id objects[] = {[NSNumber METH]};
+}
+
+void test2(NSNumber x) {
+	id objects[] = {[x METH]};
 	return 0;
 }
 
+
-- 
GitLab