CODEWITHSUNDEEP
c++cprofiling
csteifel
csteifel

Reputation: 2934

Track memory allocation per function

So I know I can track memory allocation with methods of overloading new globally like so: http://www.almostinfinite.com/memtrack.html

However, I was wondering if there was a good way to do this per function so I can get a report of how much is allocated per function. Right now I can get file and lines and what the typeid is as in the link I provided but I would like to find which function is allocating the most.

Upvotes: 4

Views: 514

Answers (2)

Brandon
Brandon

Reputation: 23525

What about doing something like: http://ideone.com/Wqjkrw

#include <iostream>
#include <cstring>

class MemTracker
{
    private:
        static char func_name[100];
        static size_t current_size;

    public:
        MemTracker(const char* FuncName) {strcpy(&func_name[0], FuncName);}
        static void inc(size_t amount) {current_size += amount;}
        static void print() {std::cout<<func_name<<" allocated: "<<current_size<<" bytes.\n";}
        static void reset() {current_size = 0; memset(&func_name[0], 0, sizeof(func_name)/sizeof(char));}
};

char MemTracker::func_name[100] = {0};
size_t MemTracker::current_size = 0;


void* operator new(size_t size)
{
    MemTracker::inc(size);
    return malloc(size);
}

void operator delete(void* ptr)
{
    free(ptr);
}


void FuncOne()
{
    MemTracker(__func__);

    int* i = new int[100];
    delete[] i;
    i = new int[200];
    delete[] i;

    MemTracker::print();
    MemTracker::reset();
}

void FuncTwo()
{
    MemTracker(__func__);
    char* c = new char[1024];
    delete[] c;
    c = new char[2048];
    delete[] c;

    MemTracker::print();
    MemTracker::reset();
}

int main()
{
    FuncOne();
    FuncTwo();
    FuncTwo();
    FuncTwo();
    return 0;
}

Prints:

FuncOne allocated: 1200 bytes.
FuncTwo allocated: 3072 bytes.
FuncTwo allocated: 3072 bytes.
FuncTwo allocated: 3072 bytes.

Upvotes: 2

R Sahu
R Sahu

Reputation: 206747

What platform are you using? There might be platform specific solutions without changing the functions in your code base.

If you are using Microsoft Visual Studio, you can use compiler switches /Gh and /GH to let the compiler call functions _penter and _pexit that you can define. In those functions, you can query how much memory the program is using. There should be enough information in there to figure out how much memory is allocated in each function.

Example code for checking memory usage is provided in this MSDN article.

Upvotes: 0

Related Questions