[asan][test] Fix Posix/new_array_cookie_with_new_from_class.cpp to work for arm targets (#208378)

arm targets, and arm64 targets on Darwin, use a 2-pointer array cookie
size. Update the test to support that.

Assisted-by: Claude Code

rdar://181775880
This commit is contained in:
Ian Anderson authored and GitHub committed 2026-07-10 10:50:37 -07:00
1 parent 5b9d51ffcf
commit cc0f756826
1 file changed
+8 -5
@@ -1,16 +1,19 @@
// Test that we do not poison the array cookie if the operator new is defined
// inside the class.
// RUN: %clangxx_asan %s -o %t && %run %t
//
// XFAIL: target=arm{{.*}}
// UNSUPPORTED: darwin-remote
#include <new>
#include <stdlib.h>
#include <stdint.h>
#include <stdio.h>
#include <assert.h>
#if defined(__arm__) || (defined(__aarch64__) && defined(__APPLE__))
constexpr size_t kArrayCookieSize = 2 * sizeof(void *);
#else
constexpr size_t kArrayCookieSize = sizeof(void *);
#endif
struct Foo {
void *operator new(size_t s) { return Allocate(s); }
void *operator new[] (size_t s) { return Allocate(s); }
@@ -34,7 +37,7 @@ int main() {
fprintf(stderr, "foo : %p\n", foo);
fprintf(stderr, "alloc: %p\n", Foo::allocated);
assert(reinterpret_cast<uintptr_t>(foo) ==
reinterpret_cast<uintptr_t>(Foo::allocated) + sizeof(void*));
reinterpret_cast<uintptr_t>(Foo::allocated) + kArrayCookieSize);
*reinterpret_cast<uintptr_t*>(Foo::allocated) = 42;
return 0;
}