100 lines · cpp
1// RUN: %clangxx_scudo %s -o %t2// RUN: %run %t valid 2>&13// RUN: %env_scudo_opts=allocator_may_return_null=1 %run %t invalid 2>&14// RUN: %env_scudo_opts=allocator_may_return_null=0 not %run %t invalid 2>&1 | FileCheck %s5 6// Tests that the C++17 aligned new/delete operators are working as expected.7// Currently we do not check the consistency of the alignment on deallocation,8// so this just tests that the APIs work.9 10#include <assert.h>11#include <stdint.h>12#include <stdio.h>13#include <string.h>14 15// Define all new/delete to not depend on the version provided by the platform.16 17namespace std {18struct nothrow_t {};19static const nothrow_t nothrow;20enum class align_val_t : size_t {};21} // namespace std22 23void *operator new(size_t);24void *operator new[](size_t);25void *operator new(size_t, std::nothrow_t const &);26void *operator new[](size_t, std::nothrow_t const &);27void *operator new(size_t, std::align_val_t);28void *operator new[](size_t, std::align_val_t);29void *operator new(size_t, std::align_val_t, std::nothrow_t const &);30void *operator new[](size_t, std::align_val_t, std::nothrow_t const &);31 32void operator delete(void *) throw();33void operator delete[](void *) throw();34void operator delete(void *, std::nothrow_t const &);35void operator delete[](void *, std::nothrow_t const &);36void operator delete(void *, size_t) throw();37void operator delete[](void *, size_t) throw();38void operator delete(void *, std::align_val_t) throw();39void operator delete[](void *, std::align_val_t) throw();40void operator delete(void *, std::align_val_t, std::nothrow_t const &);41void operator delete[](void *, std::align_val_t, std::nothrow_t const &);42void operator delete(void *, size_t, std::align_val_t) throw();43void operator delete[](void *, size_t, std::align_val_t) throw();44 45template <typename T>46inline T *break_optimization(T *arg) {47 __asm__ __volatile__(""48 :49 : "r"(arg)50 : "memory");51 return arg;52}53 54struct S12 {55 int a, b, c;56};57struct alignas(128) S12_128 {58 int a, b, c;59};60struct alignas(256) S12_256 {61 int a, b, c;62};63struct alignas(512) S1024_512 {64 char a[1024];65};66struct alignas(1024) S1024_1024 {67 char a[1024];68};69 70int main(int argc, char **argv) {71 assert(argc == 2);72 73 if (!strcmp(argv[1], "valid")) {74 // Standard use case.75 delete break_optimization(new S12);76 delete break_optimization(new S12_128);77 delete[] break_optimization(new S12_128[4]);78 delete break_optimization(new S12_256);79 delete break_optimization(new S1024_512);80 delete[] break_optimization(new S1024_512[4]);81 delete break_optimization(new S1024_1024);82 83 // Call directly the aligned versions of the operators.84 const size_t alignment = 1U << 8;85 void *p = operator new(1, static_cast<std::align_val_t>(alignment));86 assert((reinterpret_cast<uintptr_t>(p) & (alignment - 1)) == 0);87 operator delete(p, static_cast<std::align_val_t>(alignment));88 }89 if (!strcmp(argv[1], "invalid")) {90 // Alignment must be a power of 2.91 const size_t alignment = (1U << 8) - 1;92 void *p = operator new(1, static_cast<std::align_val_t>(alignment),93 std::nothrow);94 // CHECK: Scudo ERROR: invalid allocation alignment95 assert(!p);96 }97 98 return 0;99}100