[NFC][llvm-dwarfdump] Don't calculate unnecessary stats

[djtodoro]

Summary

Small optimization of the code -- No need to calculate any stats for NULL nodes, and also no need to call the collectStatsForDie() if it is the CU itself.

[aprantl]

This definitely won't hurt, but it's a bit inconsistent. The NULL check is inside collectStatsRecursive(), but the CU check is outside of collectStatsForDie(). It would be nice to be consistent and have all functions handle all DIE types with potentially an early exit at the top rather sometimes having the burden to sanitize the input at the call site.

[djtodoro]

• Use early exit in both funcs

[djtodoro]

In D96871#2568791, @aprantl wrote:

This definitely won't hurt, but it's a bit inconsistent. The NULL check is inside collectStatsRecursive(), but the CU check is outside of collectStatsForDie(). It would be nice to be consistent and have all functions handle all DIE types with potentially an early exit at the top rather sometimes having the burden to sanitize the input at the call site.

I agree. Thanks.

[djtodoro]

@aprantl I guess this is ok now ? :)

[djtodoro]

Thanks!

llvm/tools/llvm-dwarfdump/Statistics.cpp

static void collectStatsRecursive(DWARFDie Die, std::string FnPrefix,
                                   std::string VarPrefix, uint64_t BytesInScope,
                                   uint32_t InlineDepth,
                                   StringMap<PerFunctionStats> &FnStatMap,
                                   GlobalStats &GlobalStats,
                                   LocationStats &LocStats,
                                   InlinedVarsTyMap &GlobalInlinedFnInfo,
                                   InlinedFnInstacesTy &InlinedFnsToBeProcessed,
                                   InlinedVarsTy *InlinedVarsPtr = nullptr) {
+  // Skip NULL nodes.
+  if (Die.isNULL())
+    return;
+
   const dwarf::Tag Tag = Die.getTag();
   // Skip function types.
   if (Tag == dwarf::DW_TAG_subroutine_type)
     return;
 
   // Handle any kind of lexical scope.
   const bool IsFunction = Tag == dwarf::DW_TAG_subprogram;
   const bool IsBlock = Tag == dwarf::DW_TAG_lexical_block;
if (IsFunction || IsInlinedFunction || IsBlock) {
 
     if (BytesInThisScope) {
       BytesInScope = BytesInThisScope;
       if (IsFunction)
         GlobalStats.FunctionSize += BytesInThisScope;
       else if (IsInlinedFunction && InlineDepth == 0)
         GlobalStats.InlineFunctionSize += BytesInThisScope;
     }
-  } else {
+  } else if (Tag != dwarf::DW_TAG_compile_unit) {
     // Not a scope, visit the Die itself. It could be a variable.
     collectStatsForDie(Die, FnPrefix, VarPrefix, BytesInScope, InlineDepth,
                        FnStatMap, GlobalStats, LocStats, InlinedVarsPtr);
   }
 
   // Set InlineDepth correctly for child recursion
   if (IsFunction)
     InlineDepth = 0;