CComHeapPtr.cpp File Reference

#include "atltest.h"
#include <atlbase.h>

Include dependency graph for CComHeapPtr.cpp:

Go to the source code of this file.

Classes
struct	CMallocSpy

Functions
static PDWORD	test_Alloc (DWORD value)
	START_TEST (CComHeapPtr)

Variables
static LONG	g_OpenAllocations = 0
static LONG	g_Reallocations = 0
static CMallocSpy	g_Spy

Function Documentation

START_TEST()

START_TEST

(

CComHeapPtr

)

Definition at line 74 of file CComHeapPtr.cpp.

{
    HRESULT hr = CoRegisterMallocSpy(&g_Spy);
    ok(SUCCEEDED(hr), "Expected CoRegisterMallocSpy to succeed, but it failed: 0x%lx\n", hr);

    {
        ok(g_OpenAllocations == 0, "Expected there to be 0 allocations, was: %ld\n", g_OpenAllocations);
        CComHeapPtr<DWORD> heapPtr1;
        ok(g_OpenAllocations == 0, "Expected there to be 0 allocations, was: %ld\n", g_OpenAllocations);
        CComHeapPtr<DWORD> heapPtr2(test_Alloc(0x11111111));
        ok(g_OpenAllocations == 1, "Expected there to be 1 allocations, was: %ld\n", g_OpenAllocations);

        ok((PDWORD)heapPtr1 == NULL, "Expected heapPtr1 to be NULL, was: 0x%p\n", (PDWORD)heapPtr1);
        ok((PDWORD)heapPtr2 != NULL, "Expected heapPtr2 to not be NULL\n");
        ok(*heapPtr2 == 0x11111111, "Expected *heapPtr2 to be 0x11111111, but was: 0x%lx\n", *heapPtr2);

        {
            ok(g_OpenAllocations == 1, "Expected there to be 1 allocations, was: %ld\n", g_OpenAllocations);
            CComHeapPtr<DWORD> heapPtrSteal1(heapPtr1);
            ok(g_OpenAllocations == 1, "Expected there to be 1 allocations, was: %ld\n", g_OpenAllocations);
            ok((PDWORD)heapPtr1 == NULL, "Expected heapPtr1 to be NULL, was: 0x%p\n", (PDWORD)heapPtr1);
            ok((PDWORD)heapPtrSteal1 == NULL, "Expected heapPtrSteal1 to be NULL, was: 0x%p\n", (PDWORD)heapPtrSteal1);
            CComHeapPtr<DWORD> heapPtrSteal2(heapPtr2);
            ok(g_OpenAllocations == 1, "Expected there to be 1 allocations, was: %ld\n", g_OpenAllocations);
            ok((PDWORD)heapPtr2 == NULL, "Expected heapPtr2 to be NULL, was: 0x%p\n", (PDWORD)heapPtr2);
            ok((PDWORD)heapPtrSteal2 != NULL, "Expected heapPtrSteal2 to not be NULL\n");
            ok(*heapPtrSteal2 == 0x11111111, "Expected *heapPtrSteal2 to be 0x11111111, but was: 0x%lx\n", *heapPtrSteal2);
        }
        ok(g_OpenAllocations == 0, "Expected there to be 0 allocations, was: %ld\n", g_OpenAllocations);

        ok(heapPtr1.Allocate(1), "Expected Allocate to succeed\n");
        ok(g_OpenAllocations == 1, "Expected there to be 1 allocations, was: %ld\n", g_OpenAllocations);
        ok(g_Reallocations == 0, "Expected there to be 0 reallocations, was: %ld\n", g_Reallocations);

        *heapPtr1 = 0x22222222;
        ok(*heapPtr1 == 0x22222222, "Expected *heapPtr1 to be 0x22222222, but was: 0x%lx\n", *heapPtr1);

        ok(heapPtr1.Reallocate(2), "Expected Reallocate to succeed\n");
        heapPtr1[1] = 0x33333333;
        ok(*heapPtr1 == 0x22222222, "Expected *heapPtr1 to be 0x22222222, but was: 0x%lx\n", *heapPtr1);
        ok(g_Reallocations == 1, "Expected there to be 1 reallocations, was: %ld\n", g_Reallocations);
        ok(g_OpenAllocations == 1, "Expected there to be 1 allocations, was: %ld\n", g_OpenAllocations);

        heapPtr2 = heapPtr1;
        ok(g_OpenAllocations == 1, "Expected there to be 1 allocations, was: %ld\n", g_OpenAllocations);
        ok(*heapPtr2 == 0x22222222, "Expected *heapPtr2 to be 0x33333333, but was: 0x%lx\n", *heapPtr2);
        ok(heapPtr2[1] == 0x33333333, "Expected heapPtr2[1] to be 0x33333333, but was: 0x%lx\n", heapPtr2[1]);
        ok((PDWORD)heapPtr1 == NULL, "Expected heapPtr1 to be NULL, was: 0x%p\n", (PDWORD)heapPtr1);
    }
    ok(g_OpenAllocations == 0, "Expected there to be 0 allocations, was: %ld\n", g_OpenAllocations);

    hr = CoRevokeMallocSpy();
    ok(SUCCEEDED(hr), "Expected CoRevokeMallocSpy to succeed, but it failed: 0x%lx\n", hr);
}

test_Alloc()

static PDWORD test_Alloc ( DWORD  value )

static

Definition at line 16 of file CComHeapPtr.cpp.

{
    PDWORD ptr = (PDWORD)::CoTaskMemAlloc(sizeof(DWORD));
    *ptr = value;
    return ptr;
}

Referenced by START_TEST().

Variable Documentation

g_OpenAllocations

LONG g_OpenAllocations = 0

static

Definition at line 24 of file CComHeapPtr.cpp.

Referenced by CMallocSpy::PostAlloc(), CMallocSpy::PostFree(), and START_TEST().

g_Reallocations

LONG g_Reallocations = 0

static

Definition at line 25 of file CComHeapPtr.cpp.

Referenced by CMallocSpy::PostRealloc(), and START_TEST().

g_Spy

CMallocSpy g_Spy

static

Definition at line 71 of file CComHeapPtr.cpp.

Referenced by START_TEST().