1 /*
2  * Copyright (c) 2007, Michael Feathers, James Grenning and Bas Vodde
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions are met:
7  *     * Redistributions of source code must retain the above copyright
8  *       notice, this list of conditions and the following disclaimer.
9  *     * Redistributions in binary form must reproduce the above copyright
10  *       notice, this list of conditions and the following disclaimer in the
11  *       documentation and/or other materials provided with the distribution.
12  *     * Neither the name of the <organization> nor the
13  *       names of its contributors may be used to endorse or promote products
14  *       derived from this software without specific prior written permission.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE EARLIER MENTIONED AUTHORS ``AS IS'' AND ANY
17  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
19  * DISCLAIMED. IN NO EVENT SHALL <copyright holder> BE LIABLE FOR ANY
20  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  */
27 
28 #ifndef D_MemoryReportAllocator_h
29 #define D_MemoryReportAllocator_h
30 
31 #include "CppUTest/TestMemoryAllocator.h"
32 
33 class MemoryReportFormatter;
34 
35 class MemoryReportAllocator : public TestMemoryAllocator
36 {
37 protected:
38     TestResult* result_;
39     TestMemoryAllocator* realAllocator_;
40     MemoryReportFormatter* formatter_;
41 public:
42     MemoryReportAllocator();
43     virtual ~MemoryReportAllocator() _destructor_override;
44 
45     virtual void setFormatter(MemoryReportFormatter* formatter);
46     virtual void setTestResult(TestResult* result);
47     virtual void setRealAllocator(TestMemoryAllocator* allocator);
48 
49     virtual TestMemoryAllocator* getRealAllocator();
50 
51     virtual char* alloc_memory(size_t size, const char* file, size_t line) _override;
52     virtual void free_memory(char* memory, size_t size, const char* file, size_t line) _override;
53 
54     virtual const char* name() const _override;
55     virtual const char* alloc_name() const _override;
56     virtual const char* free_name() const _override;
57 
58     virtual TestMemoryAllocator* actualAllocator() _override;
59 };
60 
61 #endif
62