[llvm] e433fc3 - [NFC] Add error checking for InlinedAts.

[Shubham Sandeep Rastogi via llvm-commits]

Author: Shubham Sandeep Rastogi
Date: 2025-02-04T18:45:22-08:00
New Revision: e433fc3ce3155860e5f07c8bbc790b117a45e33f

URL: https://github.com/llvm/llvm-project/commit/e433fc3ce3155860e5f07c8bbc790b117a45e33f
DIFF: https://github.com/llvm/llvm-project/commit/e433fc3ce3155860e5f07c8bbc790b117a45e33f.diff

LOG: [NFC] Add error checking for InlinedAts.

The DroppedVariableStats::calculateDroppedStatsAndPrint should check if
it's InlinedAts stack contains the the function name that is being
accessed to make sure that a pass did not create a new function
declaration which may then lead to a crash. For example, in hot-cold
splitting, the Module before the pass will not contain the newly created
cold function and can cause a crash when trying to access the InlinedAts
stack with the function name of the newly created cold function.

Added: 
    

Modified: 
    llvm/include/llvm/Passes/DroppedVariableStats.h

Removed: 
    


################################################################################
diff  --git a/llvm/include/llvm/Passes/DroppedVariableStats.h b/llvm/include/llvm/Passes/DroppedVariableStats.h
index c4de849ca7554c..e2e91891e24c19 100644
--- a/llvm/include/llvm/Passes/DroppedVariableStats.h
+++ b/llvm/include/llvm/Passes/DroppedVariableStats.h
@@ -96,6 +96,8 @@ class DroppedVariableStats {
     DenseSet<VarID> &DebugVariablesBeforeSet =
         DbgVariables.DebugVariablesBefore;
     DenseSet<VarID> &DebugVariablesAfterSet = DbgVariables.DebugVariablesAfter;
+    if (InlinedAts.back().find(FuncName) == InlinedAts.back().end())
+      return;
     DenseMap<VarID, DILocation *> &InlinedAtsMap = InlinedAts.back()[FuncName];
     // Find an Instruction that shares the same scope as the dropped #dbg_value
     // or has a scope that is the child of the scope of the #dbg_value, and has