commit 220b45c95edc0ea86cd157426e0edc7f6a288620
author Argyrios Kyrtzidis <akyrtzi@gmail.com> Wed Nov 16 02:34:55 2011 +0000
committer Argyrios Kyrtzidis <akyrtzi@gmail.com> Wed Nov 16 02:34:55 2011 +0000
tree 2027b4e759c790e1c26aab6771473ec7ce1fdbe3
parent 32380594b6774af451cf34af79b086d6f920d052

[libclang] In lazyCreateDiags, recreate the diagnostic set if the number of diagnostics
in the ASTUnit changed.

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

tools/libclang/CIndexDiagnostic.cpp

diff --git a/tools/libclang/CIndexDiagnostic.cpp b/tools/libclang/CIndexDiagnostic.cpp
index 436ea37..50641fa 100644
--- a/tools/libclang/CIndexDiagnostic.cpp
+++ b/tools/libclang/CIndexDiagnostic.cpp
@@ -39,9 +39,22 @@
 
 CXDiagnosticImpl::~CXDiagnosticImpl() {}
 
-static CXDiagnosticSetImpl *lazyCreateDiags(CXTranslationUnit TU) {
+static CXDiagnosticSetImpl *lazyCreateDiags(CXTranslationUnit TU,
+                                            bool checkIfChanged = false) {
+  ASTUnit *AU = static_cast<ASTUnit *>(TU->TUData);
+
+  if (TU->Diagnostics && checkIfChanged) {
+    CXDiagnosticSetImpl *
+      Set = static_cast<CXDiagnosticSetImpl*>(TU->Diagnostics);
+    if (AU->stored_diag_size() != Set->getNumDiagnostics()) {
+      // Diagnostics in the ASTUnit were updated, reset the associated
+      // diagnostics.
+      delete Set;
+      TU->Diagnostics = 0;
+    }
+  }
+
   if (!TU->Diagnostics) {
-    ASTUnit *AU = static_cast<ASTUnit *>(TU->TUData);
     CXDiagnosticSetImpl *Set = new CXDiagnosticSetImpl();
     TU->Diagnostics = Set;
     
@@ -63,7 +76,7 @@
 unsigned clang_getNumDiagnostics(CXTranslationUnit Unit) {
   if (!Unit->TUData)
     return 0;
-  return lazyCreateDiags(Unit)->getNumDiagnostics();
+  return lazyCreateDiags(Unit, /*checkIfChanged=*/true)->getNumDiagnostics();
 }
 
 CXDiagnostic clang_getDiagnostic(CXTranslationUnit Unit, unsigned Index) {