CallBase Called Function getName Null Check (#680)

* Null checking getCalledFunction() when calling getName() * minor * pre-commit --------- Co-authored-by: SanthoshMohan <[email protected]>
secure-software-engineering · Nov 15, 2023 · 5ddab3f · 5ddab3f
1 parent 02e832d
commit 5ddab3f
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/lib/PhasarLLVM/DataFlow/IfdsIde/Problems/IDESecureHeapPropagation.cpp b/lib/PhasarLLVM/DataFlow/IfdsIde/Problems/IDESecureHeapPropagation.cpp
@@ -87,9 +87,10 @@ IDESecureHeapPropagation::getCallToRetFlowFunction(
   // Change to CallSite everywhere
   const auto *CS = llvm::cast<llvm::CallBase>(CallSite);
 
-  auto FName = CS->getCalledFunction()->getName();
-  if (FName == InitializerFn) {
-    return generateFromZero(SecureHeapFact::INITIALIZED);
+  if (const auto *Callee = CS->getCalledFunction()) {
+    if (Callee->getName() == InitializerFn) {
+      return generateFromZero(SecureHeapFact::INITIALIZED);
+    }
   }
   return identityFlow();
 }
@@ -147,7 +148,7 @@ IDESecureHeapPropagation::getCallToRetEdgeFunction(
     return SHPGenEdgeFn{l_t::INITIALIZED};
   }
   const auto *CS = llvm::cast<llvm::CallBase>(CallSite);
-  if (CallNode != ZeroValue &&
+  if (CallNode != ZeroValue && CS->getCalledFunction() &&
       CS->getCalledFunction()->getName() == ShutdownFn) {
     // std::cerr << "Kill at " << llvmIRToShortString(callSite) << std::endl;
     return SHPGenEdgeFn{l_t::BOT};