Added an option to specify the names of heap freeing functions.

llvm-svn: 24863
This commit is contained in:
John Criswell 2005-12-19 19:54:23 +00:00
parent aa0fed0148
commit 45bb70d8f9
1 changed files with 20 additions and 0 deletions

View File

@ -45,6 +45,12 @@ AllocList("alloc-list",
cl::desc("List of functions that allocate memory from the heap"),
cl::CommaSeparated);
static cl::list<std::string>
FreeList("free-list",
cl::value_desc("list"),
cl::desc("List of functions that free memory from the heap"),
cl::CommaSeparated);
namespace llvm {
namespace DS {
// isPointerType - Return true if this type is big enough to hold a pointer.
@ -567,6 +573,20 @@ void GraphBuilder::visitCallSite(CallSite CS) {
}
}
// Determine if the called function is one of the specified heap
// free functions
for (cl::list<std::string>::iterator FreeFunc = FreeList.begin(),
LastFreeFunc = FreeList.end();
FreeFunc != LastFreeFunc;
++FreeFunc) {
if (F->getName() == *(FreeFunc)) {
// Mark that the node is written to...
if (DSNode *N = getValueDest(*(CS.getArgument(0))).getNode())
N->setModifiedMarker()->setHeapNodeMarker();
return;
}
}
if (F->getName() == "calloc" || F->getName() == "posix_memalign" ||
F->getName() == "memalign" || F->getName() == "valloc") {
setDestTo(*CS.getInstruction(),