Improve invalidation scheme. This should fix llvm#53.

Xazax-hun · Xazax-hun · commit d8497f2173cf · 2019-08-28T17:11:13.000Z
diff --git a/clang/lib/Analysis/LifetimePsetBuilder.cpp b/clang/lib/Analysis/LifetimePsetBuilder.cpp
@@ -692,7 +692,8 @@ class PSetsBuilder : public ConstStmtVisitor<PSetsBuilder, void> {
           PSet ArgPS = derefPSet(getPSet(ObjExpr));
           for (Variable V : ArgPS.vars())
             invalidateVar(
-                V, 1, InvalidationReason::Modified(ObjExpr->getSourceRange()));
+                V, V.getOrder(),
+                InvalidationReason::Modified(ObjExpr->getSourceRange()));
         });
 
     // Bind Pointer return value.
diff --git a/clang/test/Sema/attr-psets.cpp b/clang/test/Sema/attr-psets.cpp
@@ -71,6 +71,12 @@ struct basic_string_view {
 
 using string_view = basic_string_view<char>;
 
+template <typename T>
+struct basic_string {
+    basic_string(const char*);
+    basic_string &operator +=(const basic_string& other);
+};
+
 template <class T>
 struct remove_reference { typedef T type; };
 template <class T>
@@ -956,6 +962,13 @@ void cast(int *p) {
   __lifetime_pset(q);                      // expected-warning {{pset(q) = ((unknown))}}
 }
 
+void doNotInvalidateReference(std::vector<std::basic_string<char>> v) {
+   std::basic_string<char> &r = v[0];
+   __lifetime_pset_ref(r); // expected-warning {{(*v)}}
+   r += "aa";
+   __lifetime_pset_ref(r); // expected-warning {{(*v)}}
+}
+
 // Support CXXOperatorCallExpr on non-member function
 struct S3;
 void operator==(const S3 &A, const S3 &B) { A == B; }
