Null checking getCalledFunction() when calling getName()

fabianbs96 · Oct 31, 2023 · 8c8906d · 8c8906d
1 parent 9179cf0
commit 8c8906d
Showing 1 changed file with 6 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,11 @@ 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 (CS->getCalledFunction()) {
+    auto FName = CS->getCalledFunction()->getName();
+    if (FName == InitializerFn) {
+      return generateFromZero(SecureHeapFact::INITIALIZED);
+    }
   }
   return identityFlow();
 }
@@ -147,7 +149,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};