LLVMMisc.cpp: Check if identifier exists in SymbolTable

Signed-off-by: Marcos Paulo de Souza <[email protected]>
SUSE · Jul 31, 2024 · 95b5227 · 95b5227
1 parent 4309abc
commit 95b5227
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/libcextract/LLVMMisc.cpp b/libcextract/LLVMMisc.cpp
@@ -217,6 +217,10 @@ DeclContextLookupResult Get_Decl_From_Symtab(ASTUnit *ast, const IdentifierInfo
 DeclContextLookupResult Get_Decl_From_Symtab(ASTUnit *ast, const StringRef &name)
 {
   IdentifierTable &symtab = ast->getPreprocessor().getIdentifierTable();
-  IdentifierInfo &info = symtab.get(name);
-  return Get_Decl_From_Symtab(ast, &info);
+  auto info = symtab.find(name);
+  /* Return an empty DeclContextLookupResult if the identifier is not found */
+  if (info == symtab.end())
+    return DeclContextLookupResult();
+
+  return Get_Decl_From_Symtab(ast, info->getValue());
 }