Added an option to specify the names of heap freeing functions.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@24863 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/Analysis/DataStructure/Local.cpp b/lib/Analysis/DataStructure/Local.cpp
index 3e6d424..461b1a2 100644
--- a/lib/Analysis/DataStructure/Local.cpp
+++ b/lib/Analysis/DataStructure/Local.cpp
@@ -45,6 +45,12 @@
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 @@
}
}
+ // 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(),