main.cxx   [plain text]

// BUILD:  $CXX main.cxx  -o $BUILD_DIR/operator-new.exe

// RUN:  ./operator-new.exe

#include <stdio.h>
#include <new>



//
// This test case verifies that calling operator new[] in libstdc++.dylib
// will turn around and call operator new in this main exectuable
//

static void* myLastNewAllocation;
static void* myLastDelete;

// Note: this is not weak.  That is specifically suppported
void* operator new(size_t s) throw (std::bad_alloc)
{
  myLastNewAllocation = malloc(s);
  return myLastNewAllocation;
}

struct Foo {
    int  bytes[10];
};

// Note: this is weak and because it is in main executable should override OS
__attribute__((weak))
void operator delete(void* p) throw()
{
    myLastDelete = p;
    ::free(p);
}

int main()
{
	printf("[BEGIN] operator-new\n");

    // test that OS's operator new[] redirects to my operator new
    myLastNewAllocation = NULL;
    char* stuff = new char[24];
    if ( (void*)stuff != myLastNewAllocation ) {
        printf("[FAIL] operator-new system array allocator not redirected through my operator new\n");
        return 0;
    }

    // test that program uses my operator new
    myLastNewAllocation = NULL;
    Foo* foo = new Foo();
    if ( (void*)foo != myLastNewAllocation ) {
        printf("[FAIL] operator-new allocation not redirected though my operator new\n");
        return 0;
    }

    //
    delete foo;
    if ( (void*)foo != myLastDelete ) {
        printf("[FAIL] operator-new deallocation not redirected though my operator delete\n");
        return 0;
    }

    printf("[PASS] operator-new\n");
	return 0;
}