commit ed4b4bcc36da84fbda9a813df59540faef5ff1f4
author Vitaly Buka <vitalybuka@google.com> Sat Nov 19 01:51:52 2022 -0800
committer Copybara-Service <copybara-worker@google.com> Sat Nov 19 01:56:12 2022 -0800
tree 12098c20a51fce501c28a20e11c8968912f33926
parent abd52b73c5c85189f4e2d9f8258156bc817fe555

[test][asan] Simplify __sanitizer_verify_contiguous_container test

GitOrigin-RevId: e37f8e588cd8a1d89f6cfbb829b8c9b6d8014b0f

test/asan/TestCases/contiguous_container.cpp

diff --git a/test/asan/TestCases/contiguous_container.cpp b/test/asan/TestCases/contiguous_container.cpp
index d786780..d6ec523 100644
--- a/test/asan/TestCases/contiguous_container.cpp
+++ b/test/asan/TestCases/contiguous_container.cpp
@@ -48,23 +48,31 @@
     // bytes after the end.
     for (; cur != end + kGranularity; ++cur)
       assert(__asan_address_is_poisoned(cur) == poison_buffer);
-    assert(__sanitizer_verify_contiguous_container(beg, mid, end));
-    assert(NULL ==
-           __sanitizer_contiguous_container_find_bad_address(beg, mid, end));
-    size_t distance = (end > RoundDown(end)) ? kGranularity + 1 : 1;
-    if (mid >= beg + distance) {
-      assert(
-          !__sanitizer_verify_contiguous_container(beg, mid - distance, end));
-      assert(mid - distance ==
-             __sanitizer_contiguous_container_find_bad_address(
-                 beg, mid - distance, end));
-    }
+  }
 
-    if (mid + distance <= end) {
-      assert(
-          !__sanitizer_verify_contiguous_container(beg, mid + distance, end));
-      assert(mid == __sanitizer_contiguous_container_find_bad_address(
-                        beg, mid + distance, end));
+  for (int i = 0; i <= capacity; i++) {
+    old_mid = mid;
+    mid = beg + i;
+    __sanitizer_annotate_contiguous_container(beg, end, old_mid, mid);
+
+    for (char *cur = std::max(beg, mid - 2 * kGranularity);
+         cur <= std::min(end, mid + 2 * kGranularity); ++cur) {
+      if (cur == mid ||
+          // Any mid in the last unaligned granule is OK, if bytes after the
+          // storage are not poisoned.
+          (!poison_buffer && RoundDown(end) <= std::min(cur, mid))) {
+        assert(__sanitizer_verify_contiguous_container(beg, cur, end));
+        assert(NULL == __sanitizer_contiguous_container_find_bad_address(
+                           beg, cur, end));
+      } else if (cur < mid) {
+        assert(!__sanitizer_verify_contiguous_container(beg, cur, end));
+        assert(cur == __sanitizer_contiguous_container_find_bad_address(
+                          beg, cur, end));
+      } else {
+        assert(!__sanitizer_verify_contiguous_container(beg, cur, end));
+        assert(mid == __sanitizer_contiguous_container_find_bad_address(
+                          beg, cur, end));
+      }
     }
   }