This includes adding the new TypoExpr-based lazy typo correction to
LookupMemberExprInRecord as an alternative to the existing eager typo
correction.
---
 lib/Sema/SemaExprMember.cpp              | 149 +++++++++++++++++++++++++++++--
 test/SemaCXX/arrow-operator.cpp          |   5 +-
 test/SemaCXX/typo-correction-delayed.cpp |  32 +++++++
 test/SemaCXX/typo-correction-pt2.cpp     |   2 +-
 test/SemaCXX/typo-correction.cpp         |  10 +--
 5 files changed, 182 insertions(+), 16 deletions(-)
 create mode 100644 test/SemaCXX/typo-correction-delayed.cpp

diff --git a/lib/Sema/SemaExprMember.cpp b/lib/Sema/SemaExprMember.cpp
index aca3df1..42ad82a 100644
--- a/lib/Sema/SemaExprMember.cpp
+++ b/lib/Sema/SemaExprMember.cpp
@@ -10,6 +10,7 @@
 //  This file implements semantic analysis member access expressions.
 //
 //===----------------------------------------------------------------------===//
+#include "clang/Sema/Overload.h"
 #include "clang/Sema/SemaInternal.h"
 #include "clang/AST/ASTLambda.h"
 #include "clang/AST/DeclCXX.h"
@@ -569,13 +570,47 @@ class RecordMemberExprValidatorCCC : public CorrectionCandidateCallback {
   const RecordDecl *const Record;
 };
 
+class MemberTypoDiags : public TypoDiagnosticGenerator {
+  Sema &SemaRef;
+  DeclContext *Ctx;
+  DeclarationName Typo;
+  SourceLocation TypoLoc;
+  SourceRange BaseRange;
+  SourceRange ScopeSpecLoc;
+  unsigned DiagnosticID;
+  unsigned NoSuggestDiagnosticID;
+
+public:
+  MemberTypoDiags(Sema &SemaRef, DeclContext *Ctx, DeclarationName Typo,
+                  SourceLocation TypoLoc, SourceRange BaseRange,
+                  CXXScopeSpec &SS, unsigned DiagnosticID,
+                  unsigned NoSuggestDiagnosticID)
+      : SemaRef(SemaRef), Ctx(Ctx), Typo(Typo), TypoLoc(TypoLoc), BaseRange(BaseRange),
+        ScopeSpecLoc(SS.getRange()), DiagnosticID(DiagnosticID),
+        NoSuggestDiagnosticID(NoSuggestDiagnosticID) {}
+
+  void operator()(TypoCorrection &TC) override {
+    if (TC) {
+      assert(!TC.isKeyword() && "Got a keyword as a correction for a member!");
+      bool DroppedSpecifier =
+          TC.WillReplaceSpecifier() &&
+          Typo.getAsString() == TC.getAsString(SemaRef.getLangOpts());
+      SemaRef.diagnoseTypo(TC, SemaRef.PDiag(DiagnosticID) << Typo << Ctx
+                                                           << DroppedSpecifier
+                                                           << ScopeSpecLoc);
+    } else {
+      SemaRef.Diag(TypoLoc, NoSuggestDiagnosticID) << Typo << Ctx << BaseRange;
+    }
+  }
+};
+
 }
 
-static bool
-LookupMemberExprInRecord(Sema &SemaRef, LookupResult &R, 
-                         SourceRange BaseRange, const RecordType *RTy,
-                         SourceLocation OpLoc, CXXScopeSpec &SS,
-                         bool HasTemplateArgs) {
+static bool LookupMemberExprInRecord(
+    Sema &SemaRef, LookupResult &R, SourceRange BaseRange,
+    const RecordType *RTy, SourceLocation OpLoc, CXXScopeSpec &SS,
+    bool HasTemplateArgs, TypoExpr **TE = nullptr,
+    std::unique_ptr<TypoRecoveryCallback> TRC = nullptr) {
   RecordDecl *RDecl = RTy->getDecl();
   if (!SemaRef.isThisOutsideMemberFunctionBody(QualType(RTy, 0)) &&
       SemaRef.RequireCompleteType(OpLoc, QualType(RTy, 0),
@@ -619,6 +654,21 @@ LookupMemberExprInRecord(Sema &SemaRef, LookupResult &R,
   if (!R.empty())
     return false;
 
+  if (TE && SemaRef.getLangOpts().CPlusPlus) {
+    // TODO: C cannot handle TypoExpr nodes because the C code paths do not know
+    // what to do with dependent types e.g. on the LHS of an assigment.
+    *TE = SemaRef.CorrectTypoDelayed(
+        R.getLookupNameInfo(), R.getLookupKind(), nullptr, &SS,
+        llvm::make_unique<RecordMemberExprValidatorCCC>(RTy),
+        llvm::make_unique<MemberTypoDiags>(
+            SemaRef, DC, R.getLookupName(), R.getNameLoc(), BaseRange, SS,
+            diag::err_no_member_suggest, diag::err_no_member),
+        std::move(TRC),
+        Sema::CTK_ErrorRecovery, DC);
+    R.clear();
+    return false;
+  }
+
   // We didn't find anything with the given name, so try to correct
   // for typos.
   DeclarationName Name = R.getLookupName();
@@ -1148,6 +1198,86 @@ Sema::PerformMemberExprBaseConversion(Expr *Base, bool IsArrow) {
   return CheckPlaceholderExpr(Base);
 }
 
+namespace {
+
+class MemberExprTypoRecovery : public TypoRecoveryCallback {
+  Expr *BaseExpr;
+  std::unique_ptr<CXXScopeSpec> SS;
+  SourceLocation OpLoc;
+  bool IsArrow;
+
+public:
+  MemberExprTypoRecovery(Expr *BE, CXXScopeSpec &SS, SourceLocation OpLoc,
+                         bool IsArrow)
+      : BaseExpr(BE), SS(llvm::make_unique<CXXScopeSpec>(SS)),
+        OpLoc(OpLoc), IsArrow(IsArrow) {}
+
+  ExprResult operator()(Sema &SemaRef, TypoExpr *TE, TypoCorrection &TC,
+                        Expr *Parent) override {
+    if (TC.isKeyword())
+      return ExprError();
+
+    LookupResult R(SemaRef, TC.getCorrection(), TC.getCorrectionRange().getBegin(),
+                   TE->Consumer->getLookupResult().getLookupKind());
+    R.suppressDiagnostics();
+
+    QualType BaseType;
+    if (auto *DRE = dyn_cast<DeclRefExpr>(BaseExpr))
+      BaseType = DRE->getDecl()->getType();
+    else if (auto *CE = dyn_cast<CallExpr>(BaseExpr))
+      BaseType = CE->getCallReturnType();
+
+    CallExpr *CE = dyn_cast_or_null<CallExpr>(Parent);
+    if (CE && CE->getCallee() == TE && !BaseType.isNull()) {
+      auto Args = llvm::ArrayRef<Expr*>(CE->getArgs(), CE->getNumArgs());
+      OverloadCandidateSet CandidateSet(CE->getLocStart(),
+                                        OverloadCandidateSet::CSK_Normal);
+      for (NamedDecl *ND : TC) {
+        R.clear();
+        R.addDecl(ND);
+        R.resolveKind();
+        ExprResult MemberRef = SemaRef.BuildMemberReferenceExpr(
+            BaseExpr, BaseExpr->getType(), OpLoc, IsArrow, *SS,
+            SourceLocation(), nullptr, R, nullptr);
+        if (MemberRef.isInvalid())
+          break;
+        SemaRef.AddMethodCandidate(
+            DeclAccessPair::make(ND, AS_none), BaseType.getNonReferenceType(),
+            /*MemberRef.get()->Classify(SemaRef.Context)*/Expr::Classification::makeSimpleLValue(), Args, CandidateSet);
+      }
+      // Perform overload resolution.
+      OverloadCandidateSet::iterator Best;
+      auto result = CandidateSet.BestViableFunction(
+          SemaRef, CE->getLocStart(), Best);
+      switch (result) {
+      case OR_Success:
+        // Make sure both the TypoCorrection and the LookupResult only contain
+        // the correct decl.
+        TC.setCorrectionDecl(Best->Function);
+        R.clear();
+        R.addDecl(Best->Function);
+        break;
+
+      case OR_Ambiguous:
+        break;
+
+      case OR_No_Viable_Function:
+      case OR_Deleted:
+        return ExprError();
+      }
+    } else {
+      for (NamedDecl *ND : TC)
+        R.addDecl(ND);
+    }
+    R.resolveKind();
+    return SemaRef.BuildMemberReferenceExpr(
+        BaseExpr, BaseExpr->getType(), OpLoc, IsArrow, *SS, SourceLocation(),
+        nullptr, R, nullptr);
+  }
+};
+
+}
+
 /// Look up the given member of the given non-type-dependent
 /// expression.  This can return in one of two ways:
 ///  * If it returns a sentinel null-but-valid result, the caller will
@@ -1210,13 +1340,16 @@ static ExprResult LookupMemberExpr(Sema &S, LookupResult &R,
 
   // Handle field access to simple records.
   if (const RecordType *RTy = BaseType->getAs<RecordType>()) {
-    if (LookupMemberExprInRecord(S, R, BaseExpr.get()->getSourceRange(),
-                                 RTy, OpLoc, SS, HasTemplateArgs))
+    TypoExpr *TE = nullptr;
+    if (LookupMemberExprInRecord(S, R, BaseExpr.get()->getSourceRange(), RTy,
+                                 OpLoc, SS, HasTemplateArgs, &TE,
+                                 llvm::make_unique<MemberExprTypoRecovery>(
+                                     BaseExpr.get(), SS, OpLoc, IsArrow)))
       return ExprError();
 
     // Returning valid-but-null is how we indicate to the caller that
     // the lookup result was filled in.
-    return ExprResult((Expr *)nullptr);
+    return TE ? ExprResult(TE) : ExprResult((Expr *)nullptr);
   }
 
   // Handle ivar access to Objective-C objects.
diff --git a/test/SemaCXX/arrow-operator.cpp b/test/SemaCXX/arrow-operator.cpp
index 173ff72..3e32a6b 100644
--- a/test/SemaCXX/arrow-operator.cpp
+++ b/test/SemaCXX/arrow-operator.cpp
@@ -52,14 +52,15 @@ class wrapped_ptr {
 
 class Worker {
  public:
-  void DoSomething();
+  void DoSomething(); // expected-note {{'DoSomething' declared here}}
   void Chuck();
 };
 
 void test() {
   wrapped_ptr<Worker> worker(new Worker);
   worker.DoSomething(); // expected-error {{no member named 'DoSomething' in 'arrow_suggest::wrapped_ptr<arrow_suggest::Worker>'; did you mean to use '->' instead of '.'?}}
-  worker.DoSamething(); // expected-error {{no member named 'DoSamething' in 'arrow_suggest::wrapped_ptr<arrow_suggest::Worker>'}}
+  worker.DoSamething(); // expected-error {{no member named 'DoSamething' in 'arrow_suggest::wrapped_ptr<arrow_suggest::Worker>'; did you mean to use '->' instead of '.'?}} \
+                        // expected-error {{no member named 'DoSamething' in 'arrow_suggest::Worker'; did you mean 'DoSomething'?}}
   worker.Chuck(); // expected-error {{no member named 'Chuck' in 'arrow_suggest::wrapped_ptr<arrow_suggest::Worker>'; did you mean 'Check'?}}
 }
 
diff --git a/test/SemaCXX/typo-correction-delayed.cpp b/test/SemaCXX/typo-correction-delayed.cpp
new file mode 100644
index 0000000..6659f62
--- /dev/null
+++ b/test/SemaCXX/typo-correction-delayed.cpp
@@ -0,0 +1,32 @@
+// RUN: %clang_cc1 -fsyntax-only -verify -Wno-c++11-extensions %s
+
+struct A {};
+struct B {};
+struct D {
+  A fizbin;  // expected-note 2 {{declared here}}
+  A foobar;  // expected-note 2 {{declared here}}
+  B roxbin;  // expected-note 2 {{declared here}}
+  B toobad;  // expected-note 2 {{declared here}}
+  void BooHoo();
+  void FoxBox();
+};
+
+void something(A, B);
+void test() {
+  D obj;
+  something(obj.fixbin,   // expected-error {{did you mean 'fizbin'?}}
+            obj.toobat);  // expected-error {{did you mean 'toobad'?}}
+  something(obj.toobat,   // expected-error {{did you mean 'foobar'?}}
+            obj.fixbin);  // expected-error {{did you mean 'roxbin'?}}
+  something(obj.fixbin,   // expected-error {{did you mean 'fizbin'?}}
+            obj.fixbin);  // expected-error {{did you mean 'roxbin'?}}
+  something(obj.toobat,   // expected-error {{did you mean 'foobar'?}}
+            obj.toobat);  // expected-error {{did you mean 'toobad'?}}
+  // Both members could be corrected to methods, but that isn't valid.
+  something(obj.boohoo,   // expected-error-re {{no member named 'boohoo' in 'D'{{$}}}}
+            obj.foxbox);  // expected-error-re {{no member named 'foxbox' in 'D'{{$}}}}
+  // The first argument has a usable correction but the second doesn't.
+  something(obj.boobar,   // expected-error-re {{no member named 'boobar' in 'D'{{$}}}}
+            obj.foxbox);  // expected-error-re {{no member named 'foxbox' in 'D'{{$}}}}
+}
+
diff --git a/test/SemaCXX/typo-correction-pt2.cpp b/test/SemaCXX/typo-correction-pt2.cpp
index 88a7073..57d4d5b 100644
--- a/test/SemaCXX/typo-correction-pt2.cpp
+++ b/test/SemaCXX/typo-correction-pt2.cpp
@@ -28,7 +28,7 @@ struct A {
 };
 struct B : A {
   using A::CreateFoo;
-  void CreateFoo(int, int);
+  void CreateFoo(int, int);  // expected-note {{'CreateFoo' declared here}}
 };
 void f(B &x) {
   x.Createfoo(0,0);  // expected-error {{no member named 'Createfoo' in 'PR13387::B'; did you mean 'CreateFoo'?}}
diff --git a/test/SemaCXX/typo-correction.cpp b/test/SemaCXX/typo-correction.cpp
index e8160b0..a4f5174 100644
--- a/test/SemaCXX/typo-correction.cpp
+++ b/test/SemaCXX/typo-correction.cpp
@@ -247,7 +247,7 @@ namespace b6956809_test1 {
 
   struct S1 {
     void method(A*);  // no note here
-    void method(B*);
+    void method(B*);  // expected-note{{'method' declared here}}
   };
 
   void test1() {
@@ -258,15 +258,15 @@ namespace b6956809_test1 {
 
   struct S2 {
     S2();
-    void method(A*) const;  // expected-note{{candidate function not viable}}
+    void method(A*) const;
    private:
-    void method(B*);  // expected-note{{candidate function not viable}}
+    void method(B*);
   };
 
   void test2() {
     B b;
     const S2 s;
-    s.methodd(&b);  // expected-error{{no member named 'methodd' in 'b6956809_test1::S2'; did you mean 'method'}}  expected-error{{no matching member function for call to 'method'}}
+    s.methodd(&b);  // expected-error-re{{no member named 'methodd' in 'b6956809_test1::S2'{{$}}}}
   }
 }
 
@@ -274,7 +274,7 @@ namespace b6956809_test2 {
   template<typename T> struct Err { typename T::error n; };  // expected-error{{type 'void *' cannot be used prior to '::' because it has no members}}
   struct S {
     template<typename T> typename Err<T>::type method(T);  // expected-note{{in instantiation of template class 'b6956809_test2::Err<void *>' requested here}}
-    template<typename T> int method(T *);
+    template<typename T> int method(T *);  // expected-note{{'method' declared here}}
   };
 
   void test() {
_______________________________________________
cfe-commits mailing list
[email protected]
http://lists.cs.uiuc.edu/mailman/listinfo/cfe-commits

Reply via email to