From f7dc91ad5609fab02cb805861dd232cab492340d Mon Sep 17 00:00:00 2001
From: Haojian Wu <hokein.wu@gmail.com>
Date: Fri, 19 Aug 2022 21:55:39 +0200
Subject: [PATCH] [pseudo] Eliminate a false parse of structured binding
 declaration.

Using the guard to implement part of the rule https://eel.is/c++draft/dcl.pre#6.

```
void foo() {
  // can be parsed as
  //   - structured-binding declaration (a false parse)
  //   - assignment expression
  array[index] = value;
}
```

Differential Revision: https://reviews.llvm.org/D132260
---
 clang-tools-extra/pseudo/lib/cxx/CXX.cpp           | 33 ++++++++++++++++++++++
 clang-tools-extra/pseudo/lib/cxx/cxx.bnf           |  2 +-
 .../pseudo/test/cxx/structured-binding.cpp         |  6 ++++
 3 files changed, 40 insertions(+), 1 deletion(-)
 create mode 100644 clang-tools-extra/pseudo/test/cxx/structured-binding.cpp

diff --git a/clang-tools-extra/pseudo/lib/cxx/CXX.cpp b/clang-tools-extra/pseudo/lib/cxx/CXX.cpp
index 668c4d1..46d837a 100644
--- a/clang-tools-extra/pseudo/lib/cxx/CXX.cpp
+++ b/clang-tools-extra/pseudo/lib/cxx/CXX.cpp
@@ -161,6 +161,27 @@ bool guardNextTokenNotElse(const GuardParams &P) {
   return symbolToToken(P.Lookahead) != tok::kw_else;
 }
 
+bool specifiesStructuredBinding(const GuardParams &P) {
+  const auto DSS = P.RHS[0];
+  assert(DSS->symbol() == Symbol::decl_specifier_seq);
+
+  auto Length = P.RHS[1]->startTokenIndex() - DSS->startTokenIndex();
+  for (const auto &T :
+       P.Tokens.tokens().slice(DSS->startTokenIndex(), Length)) {
+    switch (T.Kind) {
+    case clang::tok::kw_static:
+    case clang::tok::kw_thread_local:
+    case clang::tok::kw_auto:
+    case clang::tok::kw_const:
+    case clang::tok::kw_volatile:
+      break;
+    default:
+      return false;
+    }
+  }
+  return true;
+}
+
 // Whether this e.g. decl-specifier contains an "exclusive" type such as a class
 // name, and thus can't combine with a second exclusive type.
 //
@@ -320,6 +341,18 @@ llvm::DenseMap<ExtensionID, RuleGuard> buildGuards() {
       {rule::nested_name_specifier::COLONCOLON,
        TOKEN_GUARD(coloncolon, Tok.prev().Kind != tok::identifier)},
 
+      // Implement C++ [dcl.pre#6]:
+      //   A simple-declaration with an identifier-list is called a structured
+      //   binding declaration ([dcl.struct.bind]). If the decl-specifier-seq
+      //   contains any decl-specifier other than static, thread_­local, auto,
+      //   or cv-qualifiers, the program is ill-formed.
+      {rule::simple_declaration::
+           decl_specifier_seq__ref_qualifier__L_SQUARE__identifier_list__R_SQUARE__initializer__SEMI,
+       specifiesStructuredBinding},
+      {rule::simple_declaration::
+           decl_specifier_seq__L_SQUARE__identifier_list__R_SQUARE__initializer__SEMI,
+       specifiesStructuredBinding},
+
       // The grammar distinguishes (only) user-defined vs plain string literals,
       // where the clang lexer distinguishes (only) encoding types.
       {rule::user_defined_string_literal_chunk::STRING_LITERAL,
diff --git a/clang-tools-extra/pseudo/lib/cxx/cxx.bnf b/clang-tools-extra/pseudo/lib/cxx/cxx.bnf
index 80c1b54..36caf7b 100644
--- a/clang-tools-extra/pseudo/lib/cxx/cxx.bnf
+++ b/clang-tools-extra/pseudo/lib/cxx/cxx.bnf
@@ -330,7 +330,7 @@ block-declaration := opaque-enum-declaration
 nodeclspec-function-declaration := function-declarator ;
 alias-declaration := USING IDENTIFIER = defining-type-id ;
 simple-declaration := decl-specifier-seq init-declarator-list_opt ;
-simple-declaration := decl-specifier-seq ref-qualifier_opt [ identifier-list ] initializer ;
+simple-declaration := decl-specifier-seq ref-qualifier_opt [ identifier-list ] initializer ; [guard]
 static_assert-declaration := STATIC_ASSERT ( constant-expression ) ;
 static_assert-declaration := STATIC_ASSERT ( constant-expression , string-literal ) ;
 empty-declaration := ;
diff --git a/clang-tools-extra/pseudo/test/cxx/structured-binding.cpp b/clang-tools-extra/pseudo/test/cxx/structured-binding.cpp
new file mode 100644
index 0000000..1c68e92
--- /dev/null
+++ b/clang-tools-extra/pseudo/test/cxx/structured-binding.cpp
@@ -0,0 +1,6 @@
+// RUN: clang-pseudo -grammar=cxx -source=%s --start-symbol=statement-seq --print-forest | FileCheck %s
+
+// Verify there is no false parse of the structured binding declaration.
+ABC[post] = abc;
+// CHECK: statement-seq~expression-statement := expression ;
+// CHECK: postfix-expression [ expr-or-braced-init-list ]
-- 
2.7.4